Exemple #1
0
 def test_generate_signature(self):
     raw_hex_data = '523c5fcf74823831756f0bcb3634234f10b3beb1c05595058534577752ad2d9f'
     account = Account(raw_hex_data, SignatureScheme.SHA256withECDSA)
     msg = 'test'.encode('utf-8')
     signature = account.generate_signature(msg)
     result = account.verify_signature(msg, signature)
     self.assertEqual(True, result)
Exemple #2
0
 def add_multi_sign_transaction(self, m: int, pub_keys: List[bytes]
                                or List[str], signer: Account):
     """
     This interface is used to generate an Transaction object which has multi signature.
     """
     for index, pk in enumerate(pub_keys):
         if isinstance(pk, str):
             pub_keys[index] = pk.encode('ascii')
     pub_keys = ProgramBuilder.sort_public_keys(pub_keys)
     tx_hash = self.hash256()
     sig_data = signer.generate_signature(tx_hash)
     if self.sig_list is None or len(self.sig_list) == 0:
         self.sig_list = []
     elif len(self.sig_list) >= TX_MAX_SIG_SIZE:
         raise SDKException(
             ErrorCode.param_err(
                 'the number of transaction signatures should not be over 16'
             ))
     else:
         for i in range(len(self.sig_list)):
             if self.sig_list[i].public_keys == pub_keys:
                 if len(self.sig_list[i].sig_data) + 1 > len(pub_keys):
                     raise SDKException(
                         ErrorCode.param_err('too more sigData'))
                 if self.sig_list[i].m != m:
                     raise SDKException(ErrorCode.param_err('M error'))
                 self.sig_list[i].sig_data.append(sig_data)
                 return
     sig = Sig(pub_keys, m, [sig_data])
     self.sig_list.append(sig)
 def test_signature(self):
     for strength in self.strengths:
         master_key = HDPrivateKey.master_key_from_mnemonic(
             self.mnemonic.generate(strength))
         acct = Account(master_key.hex())
         signature = acct.generate_signature(self.msg)
         self.assertTrue(acct.verify_signature(self.msg, signature))
         root_sk = HDPrivateKey.from_path(master_key)[-1]
         root_pk = root_sk.public_key
         bip32_root_pk = root_pk.b58encode()
         for index in range(10):
             child_sk = HDPrivateKey.from_path(root_sk, f'0/{index}')[-1]
             child_pk = HDPublicKey.from_path(
                 HDPublicKey.b58decode(bip32_root_pk), f'0/{index}')[-1]
             child_acct = Account(child_sk.hex())
             signature = child_acct.generate_signature(self.msg)
             handler = SignatureHandler(SignatureScheme.SHA256withECDSA)
             handler.verify_signature(child_pk.hex(), self.msg, signature)
Exemple #4
0
 def add_sign_transaction(self, signer: Account):
     """
     This interface is used to add signature into the transaction.
     """
     if self.sig_list is None or len(self.sig_list) == 0:
         self.sig_list = []
     elif len(self.sig_list) >= TX_MAX_SIG_SIZE:
         raise SDKException(
             ErrorCode.param_err(
                 'the number of transaction signatures should not be over 16'
             ))
     tx_hash = self.hash256()
     sig_data = signer.generate_signature(tx_hash)
     sig = Sig([signer.get_public_key_bytes()], 1, [sig_data])
     self.sig_list.append(sig)
Exemple #5
0
 def generate_signature(self, iss: Account, verify_kid: bool = True):
     if not isinstance(self.__head, Header) or not isinstance(
             self.__payload, Payload):
         raise SDKException(
             ErrorCode.other_error('Please set claim parameters first.'))
     if verify_kid:
         key_index = int(self.__head.kid.split('-')[1])
         result = self.__sdk.native_vm.ont_id().verify_signature(
             iss.get_ont_id(), key_index, iss)
         if not result:
             raise SDKException(
                 ErrorCode.other_error('Issuer account error.'))
     b64_head = self.__head.to_base64()
     b64_payload = self.__payload.to_base64()
     msg = f'{b64_head}.{b64_payload}'.encode('utf-8')
     self.__signature = iss.generate_signature(msg)
     return self.__signature