wallet: allow message sign/verify for subaddresses

This commit is contained in:
moneromooo-monero 2019-10-31 14:09:19 +00:00
parent 4c9fd8d86d
commit d64e5aa719
No known key found for this signature in database
GPG key ID: 686F07454D6CEFC3
7 changed files with 67 additions and 19 deletions

View file

@ -43,7 +43,8 @@ from framework.wallet import Wallet
class MessageSigningTest():
def run_test(self):
self.create()
self.check_signing()
self.check_signing(False)
self.check_signing(True)
def create(self):
print('Creating wallets')
@ -65,20 +66,34 @@ class MessageSigningTest():
assert res.address == self.address[i]
assert res.seed == seeds[i]
def check_signing(self):
print('Signing/verifing messages')
def check_signing(self, subaddress):
print('Signing/verifing messages with ' + ('subaddress' if subaddress else 'standard address'))
messages = ['foo', '']
if subaddress:
address = []
for i in range(2):
res = self.wallet[i].create_account()
if i == 0:
account_index = res.account_index
res = self.wallet[i].create_address(account_index = account_index)
if i == 0:
address_index = res.address_index
address.append(res.address)
else:
address = [self.address[0], self.address[1]]
account_index = 0
address_index = 0
for message in messages:
res = self.wallet[0].sign(message)
res = self.wallet[0].sign(message, account_index = account_index, address_index = address_index)
signature = res.signature
for i in range(2):
res = self.wallet[i].verify(message, self.address[0], signature)
res = self.wallet[i].verify(message, address[0], signature)
assert res.good
res = self.wallet[i].verify('different', self.address[0], signature)
res = self.wallet[i].verify('different', address[0], signature)
assert not res.good
res = self.wallet[i].verify(message, self.address[1], signature)
res = self.wallet[i].verify(message, address[1], signature)
assert not res.good
res = self.wallet[i].verify(message, self.address[0], signature + 'x')
res = self.wallet[i].verify(message, address[0], signature + 'x')
assert not res.good
if __name__ == '__main__':