Browse Source
unify most qml and qt wizard code for trustedcoin, separate non-GUI trustedcoin wizard definition to trustedcoin.pymaster
Sander van Grieken
2 years ago
8 changed files with 558 additions and 355 deletions
@ -0,0 +1,248 @@
|
||||
import threading |
||||
import socket |
||||
import base64 |
||||
|
||||
from PyQt5.QtCore import pyqtSignal, pyqtProperty, pyqtSlot |
||||
|
||||
from electrum.i18n import _ |
||||
from electrum.bip32 import BIP32Node |
||||
|
||||
from .trustedcoin import (server, ErrorConnectingServer, MOBILE_DISCLAIMER, TrustedCoinException) |
||||
from electrum.gui.qt_common.plugins import PluginQObject |
||||
|
||||
|
||||
class QSignalObject(PluginQObject): |
||||
canSignWithoutServerChanged = pyqtSignal() |
||||
_canSignWithoutServer = False |
||||
termsAndConditionsRetrieved = pyqtSignal([str], arguments=['message']) |
||||
termsAndConditionsError = pyqtSignal([str], arguments=['message']) |
||||
otpError = pyqtSignal([str], arguments=['message']) |
||||
otpSuccess = pyqtSignal() |
||||
disclaimerChanged = pyqtSignal() |
||||
keystoreChanged = pyqtSignal() |
||||
otpSecretChanged = pyqtSignal() |
||||
_otpSecret = '' |
||||
shortIdChanged = pyqtSignal() |
||||
_shortId = '' |
||||
billingModelChanged = pyqtSignal() |
||||
_billingModel = [] |
||||
|
||||
_remoteKeyState = '' |
||||
remoteKeyStateChanged = pyqtSignal() |
||||
remoteKeyError = pyqtSignal([str], arguments=['message']) |
||||
|
||||
requestOtp = pyqtSignal() |
||||
|
||||
def __init__(self, plugin, wizard, parent): |
||||
super().__init__(plugin, parent) |
||||
self.wizard = wizard |
||||
|
||||
@pyqtProperty(str, notify=disclaimerChanged) |
||||
def disclaimer(self): |
||||
return '\n\n'.join(MOBILE_DISCLAIMER) |
||||
|
||||
@pyqtProperty(bool, notify=canSignWithoutServerChanged) |
||||
def canSignWithoutServer(self): |
||||
return self._canSignWithoutServer |
||||
|
||||
@pyqtProperty('QVariantMap', notify=keystoreChanged) |
||||
def keystore(self): |
||||
return self._keystore |
||||
|
||||
@pyqtProperty(str, notify=otpSecretChanged) |
||||
def otpSecret(self): |
||||
return self._otpSecret |
||||
|
||||
@pyqtProperty(str, notify=shortIdChanged) |
||||
def shortId(self): |
||||
return self._shortId |
||||
|
||||
@pyqtSlot(str) |
||||
def otpSubmit(self, otp): |
||||
self._plugin.on_otp(otp) |
||||
|
||||
@pyqtProperty(str, notify=remoteKeyStateChanged) |
||||
def remoteKeyState(self): |
||||
return self._remoteKeyState |
||||
|
||||
@remoteKeyState.setter |
||||
def remoteKeyState(self, new_state): |
||||
if self._remoteKeyState != new_state: |
||||
self._remoteKeyState = new_state |
||||
self.remoteKeyStateChanged.emit() |
||||
|
||||
@pyqtProperty('QVariantList', notify=billingModelChanged) |
||||
def billingModel(self): |
||||
return self._billingModel |
||||
|
||||
def updateBillingInfo(self, wallet): |
||||
billingModel = [] |
||||
|
||||
price_per_tx = wallet.price_per_tx |
||||
for k, v in sorted(price_per_tx.items()): |
||||
if k == 1: |
||||
continue |
||||
item = { |
||||
'text': 'Pay every %d transactions' % k, |
||||
'value': k, |
||||
'sats_per_tx': v / k |
||||
} |
||||
billingModel.append(item) |
||||
|
||||
self._billingModel = billingModel |
||||
self.billingModelChanged.emit() |
||||
|
||||
@pyqtSlot() |
||||
def fetchTermsAndConditions(self): |
||||
def fetch_task(): |
||||
try: |
||||
self.plugin.logger.debug('TOS') |
||||
tos = server.get_terms_of_service() |
||||
except ErrorConnectingServer as e: |
||||
self.termsAndConditionsError.emit(_('Error connecting to server')) |
||||
except Exception as e: |
||||
self.termsAndConditionsError.emit('%s: %s' % (_('Error'), repr(e))) |
||||
else: |
||||
self.termsAndConditionsRetrieved.emit(tos) |
||||
finally: |
||||
self._busy = False |
||||
self.busyChanged.emit() |
||||
|
||||
self._busy = True |
||||
self.busyChanged.emit() |
||||
t = threading.Thread(target=fetch_task) |
||||
t.daemon = True |
||||
t.start() |
||||
|
||||
@pyqtSlot(str) |
||||
def createKeystore(self, email): |
||||
self.remoteKeyState = '' |
||||
self._otpSecret = '' |
||||
self.otpSecretChanged.emit() |
||||
|
||||
wizard_data = self.wizard._current.wizard_data |
||||
|
||||
xprv1, xpub1, xprv2, xpub2, xpub3, short_id = self.plugin.create_keys(wizard_data) |
||||
|
||||
def create_remote_key_task(): |
||||
try: |
||||
self.plugin.logger.debug('create remote key') |
||||
r = server.create(xpub1, xpub2, email) |
||||
|
||||
otp_secret = r['otp_secret'] |
||||
_xpub3 = r['xpubkey_cosigner'] |
||||
_id = r['id'] |
||||
except (socket.error, ErrorConnectingServer) as e: |
||||
self.remoteKeyState = 'error' |
||||
self.remoteKeyError.emit(f'Network error: {str(e)}') |
||||
except TrustedCoinException as e: |
||||
if e.status_code == 409: |
||||
self.remoteKeyState = 'wallet_known' |
||||
self._shortId = short_id |
||||
self.shortIdChanged.emit() |
||||
else: |
||||
self.remoteKeyState = 'error' |
||||
self.logger.warning(str(e)) |
||||
self.remoteKeyError.emit(f'Service error: {str(e)}') |
||||
except (KeyError, TypeError) as e: # catch any assumptions |
||||
self.remoteKeyState = 'error' |
||||
self.remoteKeyError.emit(f'Error: {str(e)}') |
||||
self.logger.error(str(e)) |
||||
else: |
||||
if short_id != _id: |
||||
self.remoteKeyState = 'error' |
||||
self.logger.error("unexpected trustedcoin short_id: expected {}, received {}".format(short_id, _id)) |
||||
self.remoteKeyError.emit('Unexpected short_id') |
||||
return |
||||
if xpub3 != _xpub3: |
||||
self.remoteKeyState = 'error' |
||||
self.logger.error("unexpected trustedcoin xpub3: expected {}, received {}".format(xpub3, _xpub3)) |
||||
self.remoteKeyError.emit('Unexpected trustedcoin xpub3') |
||||
return |
||||
self.remoteKeyState = 'new' |
||||
self._otpSecret = otp_secret |
||||
self.otpSecretChanged.emit() |
||||
self._shortId = short_id |
||||
self.shortIdChanged.emit() |
||||
finally: |
||||
self._busy = False |
||||
self.busyChanged.emit() |
||||
|
||||
self._busy = True |
||||
self.busyChanged.emit() |
||||
|
||||
t = threading.Thread(target=create_remote_key_task) |
||||
t.daemon = True |
||||
t.start() |
||||
|
||||
@pyqtSlot() |
||||
def resetOtpSecret(self): |
||||
self.remoteKeyState = '' |
||||
|
||||
wizard_data = self.wizard._current.wizard_data |
||||
|
||||
xprv1, xpub1, xprv2, xpub2, xpub3, short_id = self.plugin.create_keys(wizard_data) |
||||
|
||||
def reset_otp_task(): |
||||
try: |
||||
self.plugin.logger.debug('reset_otp') |
||||
r = server.get_challenge(short_id) |
||||
challenge = r.get('challenge') |
||||
message = 'TRUSTEDCOIN CHALLENGE: ' + challenge |
||||
|
||||
def f(xprv): |
||||
rootnode = BIP32Node.from_xkey(xprv) |
||||
key = rootnode.subkey_at_private_derivation((0, 0)).eckey |
||||
sig = key.sign_message(message, True) |
||||
return base64.b64encode(sig).decode() |
||||
|
||||
signatures = [f(x) for x in [xprv1, xprv2]] |
||||
r = server.reset_auth(short_id, challenge, signatures) |
||||
otp_secret = r.get('otp_secret') |
||||
except (socket.error, ErrorConnectingServer) as e: |
||||
self.remoteKeyState = 'error' |
||||
self.remoteKeyError.emit(f'Network error: {str(e)}') |
||||
except Exception as e: |
||||
self.remoteKeyState = 'error' |
||||
self.remoteKeyError.emit(f'Error: {str(e)}') |
||||
else: |
||||
self.remoteKeyState = 'reset' |
||||
self._otpSecret = otp_secret |
||||
self.otpSecretChanged.emit() |
||||
finally: |
||||
self._busy = False |
||||
self.busyChanged.emit() |
||||
|
||||
self._busy = True |
||||
self.busyChanged.emit() |
||||
|
||||
t = threading.Thread(target=reset_otp_task, daemon=True) |
||||
t.start() |
||||
|
||||
@pyqtSlot(str, int) |
||||
def checkOtp(self, short_id, otp): |
||||
def check_otp_task(): |
||||
try: |
||||
self.plugin.logger.debug(f'check OTP, shortId={short_id}, otp={otp}') |
||||
server.auth(short_id, otp) |
||||
except TrustedCoinException as e: |
||||
if e.status_code == 400: # invalid OTP |
||||
self.plugin.logger.debug('Invalid one-time password.') |
||||
self.otpError.emit(_('Invalid one-time password.')) |
||||
else: |
||||
self.plugin.logger.error(str(e)) |
||||
self.otpError.emit(f'Service error: {str(e)}') |
||||
except Exception as e: |
||||
self.plugin.logger.error(str(e)) |
||||
self.otpError.emit(f'Error: {str(e)}') |
||||
else: |
||||
self.plugin.logger.debug('OTP verify success') |
||||
self.otpSuccess.emit() |
||||
finally: |
||||
self._busy = False |
||||
self.busyChanged.emit() |
||||
|
||||
self._busy = True |
||||
self.busyChanged.emit() |
||||
t = threading.Thread(target=check_otp_task, daemon=True) |
||||
t.start() |
||||
Loading…
Reference in new issue