@@ -17,7 +17,8 @@ import (
1717
1818type multiSignerSetup struct {
1919 privKeys [][]byte
20- pubKeys [][]byte
20+ pubKeysBytes [][]byte
21+ pubKeys []crypto.PublicKey
2122 partialSignatures [][][]byte
2223 messages []string
2324 aggSignatures [][]byte
@@ -51,19 +52,19 @@ func TestBls_VerifyBLSMultiSig(t *testing.T) {
5152 setupKOSK , multiSignerKOSK := createMultiSigSetupKOSK (uint16 (numMessages ), numMessages )
5253 setupKOSK .aggSignatures = aggregateSignatures (setupKOSK , multiSignerKOSK )
5354
54- for i := 0 ; i < len (setupKOSK .pubKeys ); i ++ {
55- fmt .Println (hex .EncodeToString (setupKOSK .pubKeys [i ]))
55+ for i := 0 ; i < len (setupKOSK .pubKeysBytes ); i ++ {
56+ fmt .Println (hex .EncodeToString (setupKOSK .pubKeysBytes [i ]))
5657 }
5758
5859 for i := 0 ; i < numMessages ; i ++ {
5960 fmt .Println (setupKOSK .messages [i ])
6061 fmt .Println (hex .EncodeToString (setupKOSK .aggSignatures [i ]))
6162
62- assert .Nil (t , b .VerifyAggregatedSig (setupKOSK .pubKeys , []byte (setupKOSK .messages [i ]), setupKOSK .aggSignatures [i ]))
63+ assert .Nil (t , b .VerifyAggregatedSig (setupKOSK .pubKeysBytes , []byte (setupKOSK .messages [i ]), setupKOSK .aggSignatures [i ]))
6364 changedSig := make ([]byte , len (setupKOSK .aggSignatures [i ]))
6465 copy (changedSig , setupKOSK .aggSignatures [i ])
6566 changedSig [0 ] += 1
66- assert .NotNil (t , b .VerifyAggregatedSig (setupKOSK .pubKeys , []byte (setupKOSK .messages [i ]), changedSig ))
67+ assert .NotNil (t , b .VerifyAggregatedSig (setupKOSK .pubKeysBytes , []byte (setupKOSK .messages [i ]), changedSig ))
6768 }
6869}
6970
@@ -84,26 +85,28 @@ func splitString(t testing.TB, str string) ([]byte, []byte, []byte) {
8485func createKeysAndMultiSignerBlsKOSK (
8586 grSize uint16 ,
8687 suite crypto.Suite ,
87- ) ([][]byte , [][]byte , crypto.MultiSigner ) {
88+ ) ([][]byte , []crypto. PublicKey , [] []byte , crypto.MultiSigner ) {
8889
89- kg , privKeys , pubKeys := createMultiSignerSetup (grSize , suite )
90+ kg , pubKeys , privKeys , pubKeysBytes := createMultiSignerSetup (grSize , suite )
9091 llSigner := & llsig.BlsMultiSignerKOSK {}
9192 multiSigner , _ := multisig .NewBLSMultisig (llSigner , kg )
9293
93- return privKeys , pubKeys , multiSigner
94+ return privKeys , pubKeys , pubKeysBytes , multiSigner
9495}
9596
96- func createMultiSignerSetup (grSize uint16 , suite crypto.Suite ) (crypto.KeyGenerator , [][]byte , [][]byte ) {
97+ func createMultiSignerSetup (grSize uint16 , suite crypto.Suite ) (crypto.KeyGenerator , []crypto. PublicKey , [] []byte , [][]byte ) {
9798 kg := signing .NewKeyGenerator (suite )
9899 privKeys := make ([][]byte , grSize )
99- pubKeys := make ([][]byte , grSize )
100+ pubKeysBytes := make ([][]byte , grSize )
101+ pubKeys := make ([]crypto.PublicKey , grSize )
100102
101103 for i := uint16 (0 ); i < grSize ; i ++ {
102104 sk , pk := kg .GeneratePair ()
103105 privKeys [i ], _ = sk .ToByteArray ()
104- pubKeys [i ], _ = pk .ToByteArray ()
106+ pubKeysBytes [i ], _ = pk .ToByteArray ()
107+ pubKeys [i ] = pk
105108 }
106- return kg , privKeys , pubKeys
109+ return kg , pubKeys , privKeys , pubKeysBytes
107110}
108111
109112func createSignaturesShares (privKeys [][]byte , multiSigner crypto.MultiSigner , message []byte ) [][]byte {
@@ -119,7 +122,7 @@ func createMultiSigSetupKOSK(numSigners uint16, numMessages int) (*multiSignerSe
119122 var multiSigner crypto.MultiSigner
120123 setup := & multiSignerSetup {}
121124 suite := mcl .NewSuiteBLS12 ()
122- setup .privKeys , setup .pubKeys , multiSigner = createKeysAndMultiSignerBlsKOSK (numSigners , suite )
125+ setup .privKeys , setup .pubKeys , setup . pubKeysBytes , multiSigner = createKeysAndMultiSignerBlsKOSK (numSigners , suite )
123126 setup .messages , setup .partialSignatures = createMessagesAndPartialSignatures (numMessages , setup .privKeys , multiSigner )
124127
125128 return setup , multiSigner
0 commit comments