mirror of
https://github.com/sonr-io/sonr.git
synced 2026-08-04 18:31:41 +00:00
Prad NukalaandGitHub
89989fa102
- **refactor: move session-related code to middleware package** - **refactor: update PKL build process and adjust related configurations** - **feat: integrate base.cosmos.v1 Genesis module** - **refactor: pass session context to modal rendering functions** - **refactor: move nebula package to app directory and update templ version** - **refactor: Move home section video view to dedicated directory** - **refactor: remove unused views file** - **refactor: move styles and UI components to global scope** - **refactor: Rename images.go to cdn.go** - **feat: Add Empty State Illustrations** - **refactor: Consolidate Vault Index Logic** - **fix: References to App.wasm and remove Vault Directory embedded CDN files** - **refactor: Move CDN types to Models** - **fix: Correct line numbers in templ error messages for arch_templ.go** - **refactor: use common types for peer roles** - **refactor: move common types and ORM to a shared package** - **fix: Config import dwn** - **refactor: move nebula directory to app** - **feat: Rebuild nebula** - **fix: correct file paths in panels templates** - **feat: Remove duplicate types** - **refactor: Move dwn to pkg/core** - **refactor: Binary Structure** - **feat: Introduce Crypto Pkg** - **fix: Broken Process Start** - **feat: Update pkg/* structure** - **feat: Refactor PKL Structure** - **build: update pkl build process** - **chore: Remove Empty Files** - **refactor: remove unused macaroon package** - **feat: Add WebAwesome Components** - **refactor: consolidate build and generation tasks into a single taskfile, remove redundant makefile targets** - **refactor: refactor server and move components to pkg/core/dwn** - **build: update go modules** - **refactor: move gateway logic into dedicated hway command** - **feat: Add KSS (Krawczyk-Song-Song) MPC cryptography module** - **feat: Implement MPC-based JWT signing and UCAN token generation** - **feat: add support for MPC-based JWT signing** - **feat: Implement MPC-based UCAN capabilities for smart accounts** - **feat: add address field to keyshareSource** - **feat: Add comprehensive MPC test suite for keyshares, UCAN tokens, and token attenuations** - **refactor: improve MPC keyshare management and signing process** - **feat: enhance MPC capability hierarchy documentation** - **refactor: rename GenerateKeyshares function to NewKeyshareSource for clarity** - **refactor: remove unused Ethereum address computation** - **feat: Add HasHandle and IsAuthenticated methods to HTTPContext** - **refactor: Add context.Context support to session HTTPContext** - **refactor: Resolve context interface conflicts in HTTPContext** - **feat: Add session ID context key and helper functions** - **feat: Update WebApp Page Rendering** - **refactor: Simplify context management by using single HTTPContext key** - **refactor: Simplify HTTPContext creation and context management in session middleware** - **refactor: refactor session middleware to use a single data structure** - **refactor: Simplify HTTPContext implementation and session data handling** - **refactor: Improve session context handling and prevent nil pointer errors** - **refactor: Improve session context handling with nil safety and type support** - **refactor: improve session data injection** - **feat: add full-screen modal component and update registration flow** - **chore: add .air.toml to .gitignore** - **feat: add Air to devbox and update dependencies**
318 lines
9.5 KiB
Go
Executable File
318 lines
9.5 KiB
Go
Executable File
//
|
|
// Copyright Coinbase, Inc. All Rights Reserved.
|
|
//
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
package bbs
|
|
|
|
import (
|
|
crand "crypto/rand"
|
|
"testing"
|
|
|
|
"github.com/gtank/merlin"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/onsonr/sonr/pkg/crypto/core/curves"
|
|
"github.com/onsonr/sonr/pkg/crypto/signatures/common"
|
|
)
|
|
|
|
func TestPokSignatureProofSomeMessagesRevealed(t *testing.T) {
|
|
curve := curves.BLS12381(&curves.PointBls12381G2{})
|
|
pk, sk, err := NewKeys(curve)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, sk)
|
|
require.NotNil(t, pk)
|
|
require.False(t, sk.value.IsZero())
|
|
require.False(t, pk.value.IsIdentity())
|
|
_, ok := pk.value.(*curves.PointBls12381G2)
|
|
require.True(t, ok)
|
|
generators, err := new(MessageGenerators).Init(pk, 4)
|
|
require.NoError(t, err)
|
|
msgs := []curves.Scalar{
|
|
curve.Scalar.New(2),
|
|
curve.Scalar.New(3),
|
|
curve.Scalar.New(4),
|
|
curve.Scalar.New(5),
|
|
}
|
|
|
|
sig, err := sk.Sign(generators, msgs)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, sig)
|
|
require.False(t, sig.a.IsIdentity())
|
|
require.False(t, sig.e.IsZero())
|
|
require.False(t, sig.s.IsZero())
|
|
|
|
proofMsgs := []common.ProofMessage{
|
|
&common.ProofSpecificMessage{
|
|
Message: msgs[0],
|
|
},
|
|
&common.ProofSpecificMessage{
|
|
Message: msgs[1],
|
|
},
|
|
&common.RevealedMessage{
|
|
Message: msgs[2],
|
|
},
|
|
&common.RevealedMessage{
|
|
Message: msgs[3],
|
|
},
|
|
}
|
|
|
|
pok, err := NewPokSignature(sig, generators, proofMsgs, crand.Reader)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, pok)
|
|
nonce := curve.Scalar.Random(crand.Reader)
|
|
transcript := merlin.NewTranscript("TestPokSignatureProofWorks")
|
|
pok.GetChallengeContribution(transcript)
|
|
transcript.AppendMessage([]byte("nonce"), nonce.Bytes())
|
|
okm := transcript.ExtractBytes([]byte("signature proof of knowledge"), 64)
|
|
challenge, err := curve.Scalar.SetBytesWide(okm)
|
|
require.NoError(t, err)
|
|
|
|
pokSig, err := pok.GenerateProof(challenge)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, pokSig)
|
|
require.True(t, pokSig.VerifySigPok(pk))
|
|
|
|
revealedMsgs := map[int]curves.Scalar{
|
|
2: msgs[2],
|
|
3: msgs[3],
|
|
}
|
|
// Manual verify to show how when used in conjunction with other ZKPs
|
|
transcript = merlin.NewTranscript("TestPokSignatureProofWorks")
|
|
pokSig.GetChallengeContribution(generators, revealedMsgs, challenge, transcript)
|
|
transcript.AppendMessage([]byte("nonce"), nonce.Bytes())
|
|
okm = transcript.ExtractBytes([]byte("signature proof of knowledge"), 64)
|
|
vChallenge, err := curve.Scalar.SetBytesWide(okm)
|
|
require.NoError(t, err)
|
|
require.Equal(t, challenge.Cmp(vChallenge), 0)
|
|
|
|
// Use the all-inclusive method
|
|
transcript = merlin.NewTranscript("TestPokSignatureProofWorks")
|
|
require.True(t, pokSig.Verify(revealedMsgs, pk, generators, nonce, challenge, transcript))
|
|
}
|
|
|
|
func TestPokSignatureProofAllMessagesRevealed(t *testing.T) {
|
|
curve := curves.BLS12381(&curves.PointBls12381G2{})
|
|
pk, sk, err := NewKeys(curve)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, sk)
|
|
require.NotNil(t, pk)
|
|
require.False(t, sk.value.IsZero())
|
|
require.False(t, pk.value.IsIdentity())
|
|
_, ok := pk.value.(*curves.PointBls12381G2)
|
|
require.True(t, ok)
|
|
generators, err := new(MessageGenerators).Init(pk, 4)
|
|
require.NoError(t, err)
|
|
msgs := []curves.Scalar{
|
|
curve.Scalar.New(2),
|
|
curve.Scalar.New(3),
|
|
curve.Scalar.New(4),
|
|
curve.Scalar.New(5),
|
|
}
|
|
|
|
sig, err := sk.Sign(generators, msgs)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, sig)
|
|
require.False(t, sig.a.IsIdentity())
|
|
require.False(t, sig.e.IsZero())
|
|
require.False(t, sig.s.IsZero())
|
|
|
|
proofMsgs := []common.ProofMessage{
|
|
&common.RevealedMessage{
|
|
Message: msgs[0],
|
|
},
|
|
&common.RevealedMessage{
|
|
Message: msgs[1],
|
|
},
|
|
&common.RevealedMessage{
|
|
Message: msgs[2],
|
|
},
|
|
&common.RevealedMessage{
|
|
Message: msgs[3],
|
|
},
|
|
}
|
|
|
|
pok, err := NewPokSignature(sig, generators, proofMsgs, crand.Reader)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, pok)
|
|
nonce := curve.Scalar.Random(crand.Reader)
|
|
transcript := merlin.NewTranscript("TestPokSignatureProofWorks")
|
|
pok.GetChallengeContribution(transcript)
|
|
transcript.AppendMessage([]byte("nonce"), nonce.Bytes())
|
|
okm := transcript.ExtractBytes([]byte("signature proof of knowledge"), 64)
|
|
challenge, err := curve.Scalar.SetBytesWide(okm)
|
|
require.NoError(t, err)
|
|
|
|
pokSig, err := pok.GenerateProof(challenge)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, pokSig)
|
|
require.True(t, pokSig.VerifySigPok(pk))
|
|
|
|
revealedMsgs := map[int]curves.Scalar{
|
|
0: msgs[0],
|
|
1: msgs[1],
|
|
2: msgs[2],
|
|
3: msgs[3],
|
|
}
|
|
// Manual verify to show how when used in conjunction with other ZKPs
|
|
transcript = merlin.NewTranscript("TestPokSignatureProofWorks")
|
|
pokSig.GetChallengeContribution(generators, revealedMsgs, challenge, transcript)
|
|
transcript.AppendMessage([]byte("nonce"), nonce.Bytes())
|
|
okm = transcript.ExtractBytes([]byte("signature proof of knowledge"), 64)
|
|
vChallenge, err := curve.Scalar.SetBytesWide(okm)
|
|
require.NoError(t, err)
|
|
require.Equal(t, challenge.Cmp(vChallenge), 0)
|
|
|
|
// Use the all-inclusive method
|
|
transcript = merlin.NewTranscript("TestPokSignatureProofWorks")
|
|
require.True(t, pokSig.Verify(revealedMsgs, pk, generators, nonce, challenge, transcript))
|
|
}
|
|
|
|
func TestPokSignatureProofAllMessagesHidden(t *testing.T) {
|
|
curve := curves.BLS12381(&curves.PointBls12381G2{})
|
|
pk, sk, err := NewKeys(curve)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, sk)
|
|
require.NotNil(t, pk)
|
|
require.False(t, sk.value.IsZero())
|
|
require.False(t, pk.value.IsIdentity())
|
|
_, ok := pk.value.(*curves.PointBls12381G2)
|
|
require.True(t, ok)
|
|
generators, err := new(MessageGenerators).Init(pk, 4)
|
|
require.NoError(t, err)
|
|
msgs := []curves.Scalar{
|
|
curve.Scalar.New(2),
|
|
curve.Scalar.New(3),
|
|
curve.Scalar.New(4),
|
|
curve.Scalar.New(5),
|
|
}
|
|
|
|
sig, err := sk.Sign(generators, msgs)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, sig)
|
|
require.False(t, sig.a.IsIdentity())
|
|
require.False(t, sig.e.IsZero())
|
|
require.False(t, sig.s.IsZero())
|
|
|
|
proofMsgs := []common.ProofMessage{
|
|
&common.ProofSpecificMessage{
|
|
Message: msgs[0],
|
|
},
|
|
&common.ProofSpecificMessage{
|
|
Message: msgs[1],
|
|
},
|
|
&common.ProofSpecificMessage{
|
|
Message: msgs[2],
|
|
},
|
|
&common.ProofSpecificMessage{
|
|
Message: msgs[3],
|
|
},
|
|
}
|
|
|
|
pok, err := NewPokSignature(sig, generators, proofMsgs, crand.Reader)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, pok)
|
|
nonce := curve.Scalar.Random(crand.Reader)
|
|
transcript := merlin.NewTranscript("TestPokSignatureProofWorks")
|
|
pok.GetChallengeContribution(transcript)
|
|
transcript.AppendMessage([]byte("nonce"), nonce.Bytes())
|
|
okm := transcript.ExtractBytes([]byte("signature proof of knowledge"), 64)
|
|
challenge, err := curve.Scalar.SetBytesWide(okm)
|
|
require.NoError(t, err)
|
|
|
|
pokSig, err := pok.GenerateProof(challenge)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, pokSig)
|
|
require.True(t, pokSig.VerifySigPok(pk))
|
|
|
|
revealedMsgs := map[int]curves.Scalar{}
|
|
|
|
// Manual verify to show how when used in conjunction with other ZKPs
|
|
transcript = merlin.NewTranscript("TestPokSignatureProofWorks")
|
|
pokSig.GetChallengeContribution(generators, revealedMsgs, challenge, transcript)
|
|
transcript.AppendMessage([]byte("nonce"), nonce.Bytes())
|
|
okm = transcript.ExtractBytes([]byte("signature proof of knowledge"), 64)
|
|
vChallenge, err := curve.Scalar.SetBytesWide(okm)
|
|
require.NoError(t, err)
|
|
require.Equal(t, challenge.Cmp(vChallenge), 0)
|
|
|
|
// Use the all-inclusive method
|
|
transcript = merlin.NewTranscript("TestPokSignatureProofWorks")
|
|
require.True(t, pokSig.Verify(revealedMsgs, pk, generators, nonce, challenge, transcript))
|
|
}
|
|
|
|
func TestPokSignatureProofMarshalBinary(t *testing.T) {
|
|
curve := curves.BLS12381(&curves.PointBls12381G2{})
|
|
pk, sk, err := NewKeys(curve)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, sk)
|
|
require.NotNil(t, pk)
|
|
require.False(t, sk.value.IsZero())
|
|
require.False(t, pk.value.IsIdentity())
|
|
_, ok := pk.value.(*curves.PointBls12381G2)
|
|
require.True(t, ok)
|
|
generators, err := new(MessageGenerators).Init(pk, 4)
|
|
require.NoError(t, err)
|
|
msgs := []curves.Scalar{
|
|
curve.Scalar.New(2),
|
|
curve.Scalar.New(3),
|
|
curve.Scalar.New(4),
|
|
curve.Scalar.New(5),
|
|
}
|
|
|
|
sig, err := sk.Sign(generators, msgs)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, sig)
|
|
require.False(t, sig.a.IsIdentity())
|
|
require.False(t, sig.e.IsZero())
|
|
require.False(t, sig.s.IsZero())
|
|
|
|
proofMsgs := []common.ProofMessage{
|
|
&common.ProofSpecificMessage{
|
|
Message: msgs[0],
|
|
},
|
|
&common.ProofSpecificMessage{
|
|
Message: msgs[1],
|
|
},
|
|
&common.RevealedMessage{
|
|
Message: msgs[2],
|
|
},
|
|
&common.RevealedMessage{
|
|
Message: msgs[3],
|
|
},
|
|
}
|
|
|
|
pok, err := NewPokSignature(sig, generators, proofMsgs, crand.Reader)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, pok)
|
|
nonce := curve.Scalar.Random(crand.Reader)
|
|
transcript := merlin.NewTranscript("TestPokSignatureProofMarshalBinary")
|
|
pok.GetChallengeContribution(transcript)
|
|
transcript.AppendMessage([]byte("nonce"), nonce.Bytes())
|
|
challenge, err := curve.Scalar.SetBytesWide(transcript.ExtractBytes([]byte("signature proof of knowledge"), 64))
|
|
require.NoError(t, err)
|
|
|
|
pokSig, err := pok.GenerateProof(challenge)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, pokSig)
|
|
|
|
data, err := pokSig.MarshalBinary()
|
|
require.NoError(t, err)
|
|
require.NotNil(t, data)
|
|
pokSig2 := new(PokSignatureProof).Init(curve)
|
|
err = pokSig2.UnmarshalBinary(data)
|
|
require.NoError(t, err)
|
|
require.True(t, pokSig.aPrime.Equal(pokSig2.aPrime))
|
|
require.True(t, pokSig.aBar.Equal(pokSig2.aBar))
|
|
require.True(t, pokSig.d.Equal(pokSig2.d))
|
|
require.Equal(t, len(pokSig.proof1), len(pokSig2.proof1))
|
|
require.Equal(t, len(pokSig.proof2), len(pokSig2.proof2))
|
|
for i, p := range pokSig.proof1 {
|
|
require.Equal(t, p.Cmp(pokSig2.proof1[i]), 0)
|
|
}
|
|
for i, p := range pokSig.proof2 {
|
|
require.Equal(t, p.Cmp(pokSig2.proof2[i]), 0)
|
|
}
|
|
}
|