ceremonyclient/nekryptology/pkg/ot/extension/kos/stream.go

58 lines
2.3 KiB
Go
Raw Normal View History

2023-07-05 05:32:28 +00:00
package kos
import (
"encoding/gob"
"io"
"github.com/pkg/errors"
"source.quilibrium.com/quilibrium/monorepo/nekryptology/pkg/core/curves"
"source.quilibrium.com/quilibrium/monorepo/nekryptology/pkg/ot/base/simplest"
)
// ReceiverStreamCOtRun exposes an end-to-end "streaming" version of the cOT process for the receiver.
// this is similar to what we're also doing in the base OT side. the user only passes an arbitrary `ReadWriter` here,
// together with the relevant inputs (namely a choice vector); this method handles all parts of the process,
// including both encoding / decoding and writing to / reading from the stream.
func ReceiverStreamCOtRun(receiver *Receiver, hashKeySeed [simplest.DigestSize]byte, choice []byte, rw io.ReadWriter) error {
enc := gob.NewEncoder(rw)
dec := gob.NewDecoder(rw)
firstMessage, err := receiver.Round1Initialize(hashKeySeed, choice)
if err != nil {
return errors.Wrap(err, "computing first message in receiver stream cOT")
}
if err = enc.Encode(firstMessage); err != nil {
return errors.Wrap(err, "encoding first message in receiver stream cOT")
}
responseTau := &Round2Output{}
if err = dec.Decode(responseTau); err != nil {
return errors.Wrap(err, "decoding responseTau in receiver stream OT")
}
if err = receiver.Round3Transfer(responseTau); err != nil {
return errors.Wrap(err, "error during round 3 in receiver stream OT")
}
return nil
}
// SenderStreamCOtRun exposes the end-to-end "streaming" version of cOT for the sender.
// the sender should pass an arbitrary ReadWriter together with their input; this will handle the whole process,
// including all component methods, plus reading to and writing from the network.
func SenderStreamCOtRun(sender *Sender, hashKeySeed [simplest.DigestSize]byte, input [][]curves.Scalar, rw io.ReadWriter) error {
enc := gob.NewEncoder(rw)
dec := gob.NewDecoder(rw)
firstMessage := &Round1Output{}
if err := dec.Decode(firstMessage); err != nil {
return errors.Wrap(err, "decoding first message in sender stream cOT")
}
responseTau, err := sender.Round2Transfer(hashKeySeed, input, firstMessage)
if err != nil {
return errors.Wrap(err, "error in round 2 in sender stream cOT")
}
if err = enc.Encode(responseTau); err != nil {
return errors.Wrap(err, "encoding responseTau in sender stream cOT")
}
return nil
}