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 }