curve/elgamal/curve.go

package elgamal

import (
	"context"
	"fmt"
	"log"
	"math/big"

	"git.ali33.ru/fcg-xvii/curve/v2/tools"
)

func NewCurve(c *tools.Curve, ctx context.Context) (*Curve, error) {
	cc := &Curve{
		Curve: c,
	}
	return cc, cc.setupDataTable(ctx)
}

type Curve struct {
	*tools.Curve
	dTable map[byte]*tools.Point
}

func (s *Curve) bytePoint(p *tools.Point) (byte, error) {
	for key, val := range s.dTable {
		if val.IsEqual(p) {
			return key, nil
		}
	}
	return 0, fmt.Errorf("Не найден символ, соответствующий точке %v", p)
}

// Заполнение таблицы байт -> точка на кривой
func (s *Curve) setupDataTable(ctx context.Context) error {
	s.dTable = make(map[byte]*tools.Point)
	dx := big.NewInt(1)
	for i := byte(0); i <= 254; i++ {
		// поиск точки
		p, _, err := s.SearchClosePoints(dx, ctx)
		if err != nil {
			return err
		}
		// check already exists - todo
		dx = tools.Add64(p.X(), 2)
		s.dTable[i] = p
		log.Println(i, p)
	}
	return nil
}