sonr/crypto/ot/base/simplest/ot_test.go
Prad Nukala 807b2e86ec
feature/1220 origin handle exists method (#1241)
* feat: add docs and CI workflow for publishing to onsonr.dev

* (refactor): Move hway,motr executables to their own repos

* feat: simplify devnet and testnet configurations

* refactor: update import path for didcrypto package

* docs(networks): Add README with project overview, architecture, and community links

* refactor: Move network configurations to deploy directory

* build: update golang version to 1.23

* refactor: move logger interface to appropriate package

* refactor: Move devnet configuration to networks/devnet

* chore: improve release process with date variable

* (chore): Move Crypto Library

* refactor: improve code structure and readability in DID module

* feat: integrate Trunk CI checks

* ci: optimize CI workflow by removing redundant build jobs

---------

Co-authored-by: Darp Alakun <i@prad.nu>
2025-01-06 17:06:10 +00:00

86 lines
2.4 KiB
Go
Executable File

//
// 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"
"github.com/onsonr/sonr/crypto/core/curves"
"github.com/onsonr/sonr/crypto/ot/base/simplest"
"github.com/onsonr/sonr/crypto/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]])
}
}