ceremonyclient/nekryptology/pkg/ot/base/simplest/ot_test.go
2023-07-05 00:32:28 -05:00

86 lines
2.5 KiB
Go

//
// Copyright Coinbase, Inc. All Rights Reserved.
//
// SPDX-License-Identifier: Apache-2.0
//
package simplest_test
import (
"crypto/rand"
"crypto/sha256"
"fmt"
"testing"
"github.com/stretchr/testify/require"
"source.quilibrium.com/quilibrium/monorepo/nekryptology/pkg/core/curves"
"source.quilibrium.com/quilibrium/monorepo/nekryptology/pkg/ot/base/simplest"
"source.quilibrium.com/quilibrium/monorepo/nekryptology/pkg/ot/ottest"
)
func TestOtOnMultipleCurves(t *testing.T) {
curveInstances := []*curves.Curve{
curves.K256(),
curves.P256(),
}
for _, curve := range curveInstances {
batchSize := 256
hashKeySeed := [32]byte{}
_, err := rand.Read(hashKeySeed[:])
require.NoError(t, err)
sender, receiver, err := ottest.RunSimplestOT(curve, batchSize, hashKeySeed)
require.NoError(t, err)
for i := 0; i < batchSize; i++ {
require.Equal(t, receiver.OneTimePadDecryptionKey[i], sender.OneTimePadEncryptionKeys[i][receiver.RandomChoiceBits[i]])
}
// Transfer messages
messages := make([][2][32]byte, batchSize)
for i := 0; i < batchSize; i++ {
messages[i] = [2][32]byte{
sha256.Sum256([]byte(fmt.Sprintf("message[%d][0]", i))),
sha256.Sum256([]byte(fmt.Sprintf("message[%d][1]", i))),
}
}
ciphertexts, err := sender.Encrypt(messages)
require.NoError(t, err)
decrypted, err := receiver.Decrypt(ciphertexts)
require.NoError(t, err)
for i := 0; i < batchSize; i++ {
choice := receiver.RandomChoiceBits[i]
require.Equal(t, messages[i][choice], decrypted[i])
require.NotEqual(t, messages[i][1-choice], decrypted[i])
}
}
}
func TestOTStreaming(t *testing.T) {
batchSize := 256
curve := curves.K256()
hashKeySeed := [32]byte{}
_, err := rand.Read(hashKeySeed[:])
require.NoError(t, err)
sender, err := simplest.NewSender(curve, batchSize, hashKeySeed)
require.Nil(t, err)
receiver, err := simplest.NewReceiver(curve, batchSize, hashKeySeed)
require.Nil(t, err)
senderPipe, receiverPipe := simplest.NewPipeWrappers()
errorsChannel := make(chan error, 2) // warning: if one party errors, the other will sit there forever. add timeouts.
go func() {
errorsChannel <- simplest.SenderStreamOTRun(sender, senderPipe)
}()
go func() {
errorsChannel <- simplest.ReceiverStreamOTRun(receiver, receiverPipe)
}()
for i := 0; i < 2; i++ {
require.Nil(t, <-errorsChannel)
}
for i := 0; i < batchSize; i++ {
require.Equal(t, receiver.Output.OneTimePadDecryptionKey[i], sender.Output.OneTimePadEncryptionKeys[i][receiver.Output.RandomChoiceBits[i]])
}
}