124 lines
3.5 KiB
Python
124 lines
3.5 KiB
Python
|
from electrum.plugins import BasePlugin, hook
|
||
|
|
from electrum_gui.qt.util import WaitingDialog
|
||
|
|
from electrum.util import print_msg
|
||
|
|
|
||
|
|
from PyQt4.QtGui import *
|
||
|
|
from PyQt4.QtCore import *
|
||
|
|
|
||
|
|
import subprocess
|
||
|
|
import traceback
|
||
|
|
import zlib
|
||
|
|
import json
|
||
|
|
|
||
|
|
try:
|
||
|
|
subprocess.check_call(['amodem-cli', '--help'])
|
||
|
|
print_msg('Audio MODEM is enabled.')
|
||
|
|
amodem_available = True
|
||
|
|
except subprocess.CalledProcessError:
|
||
|
|
print_msg('Audio MODEM is not found.')
|
||
|
|
amodem_available = False
|
||
|
|
|
||
|
|
|
||
|
|
def send(parent, blob):
|
||
|
|
print_msg('Sending:', repr(blob))
|
||
|
|
blob = zlib.compress(blob)
|
||
|
|
def sender_thread():
|
||
|
|
try:
|
||
|
|
args = ['amodem-cli', 'send', '-vv']
|
||
|
|
p = subprocess.Popen(args, stdin=subprocess.PIPE)
|
||
|
|
p.stdin.write(blob)
|
||
|
|
p.stdin.close()
|
||
|
|
p.wait()
|
||
|
|
except Exception:
|
||
|
|
traceback.print_exc()
|
||
|
|
p.kill()
|
||
|
|
|
||
|
|
return WaitingDialog(
|
||
|
|
parent=parent, message='Sending transaction to Audio MODEM...',
|
||
|
|
run_task=sender_thread)
|
||
|
|
|
||
|
|
|
||
|
|
def recv(parent):
|
||
|
|
def receiver_thread():
|
||
|
|
import subprocess
|
||
|
|
try:
|
||
|
|
args = ['amodem-cli', 'recv', '-vv']
|
||
|
|
p = subprocess.Popen(args, stdout=subprocess.PIPE)
|
||
|
|
return p.stdout.read()
|
||
|
|
except Exception:
|
||
|
|
traceback.print_exc()
|
||
|
|
p.kill()
|
||
|
|
|
||
|
|
def on_success(blob):
|
||
|
|
blob = zlib.decompress(blob)
|
||
|
|
print_msg('Received:', repr(blob))
|
||
|
|
parent.setText(blob)
|
||
|
|
|
||
|
|
return WaitingDialog(
|
||
|
|
parent=parent, message='Receiving transaction from Audio MODEM...',
|
||
|
|
run_task=receiver_thread, on_success=on_success)
|
||
|
|
|
||
|
|
|
||
|
|
class Plugin(BasePlugin):
|
||
|
|
|
||
|
|
def fullname(self):
|
||
|
|
return 'Audio MODEM'
|
||
|
|
|
||
|
|
def description(self):
|
||
|
|
return ('Provides support for air-gapped transaction signing.\n\n'
|
||
|
|
'Requires http://github.com/romanz/amodem/')
|
||
|
|
|
||
|
|
def is_available(self):
|
||
|
|
return amodem_available
|
||
|
|
|
||
|
|
is_enabled = is_available
|
||
|
|
|
||
|
|
@hook
|
||
|
|
def transaction_dialog(self, dialog):
|
||
|
|
b = QPushButton()
|
||
|
|
b.setIcon(QIcon(":icons/speaker.png"))
|
||
|
|
|
||
|
|
def handler():
|
||
|
|
blob = json.dumps(dialog.tx.as_dict())
|
||
|
|
self.sender = send(parent=dialog, blob=blob)
|
||
|
|
self.sender.start()
|
||
|
|
b.clicked.connect(handler)
|
||
|
|
dialog.buttons.insertWidget(1, b)
|
||
|
|
|
||
|
|
@hook
|
||
|
|
def scan_text_edit(self, parent):
|
||
|
|
def handler():
|
||
|
|
self.receiver = recv(parent=parent)
|
||
|
|
self.receiver.start()
|
||
|
|
button = add_button(parent=parent, icon_name=':icons/microphone.png')
|
||
|
|
button.clicked.connect(handler)
|
||
|
|
|
||
|
|
@hook
|
||
|
|
def show_text_edit(self, parent):
|
||
|
|
def handler():
|
||
|
|
blob = str(parent.toPlainText())
|
||
|
|
self.sender = send(parent=parent, blob=blob)
|
||
|
|
self.sender.start()
|
||
|
|
button = add_button(parent=parent, icon_name=':icons/speaker.png')
|
||
|
|
button.clicked.connect(handler)
|
||
|
|
|
||
|
|
|
||
|
|
def add_button(parent, icon_name):
|
||
|
|
audio_button = QToolButton(parent)
|
||
|
|
audio_button.setIcon(QIcon(icon_name))
|
||
|
|
audio_button.setStyleSheet("QToolButton { border: none; padding: 0px; }")
|
||
|
|
audio_button.setVisible(True)
|
||
|
|
|
||
|
|
parent_resizeEvent = parent.resizeEvent
|
||
|
|
|
||
|
|
def resizeEvent(e):
|
||
|
|
result = parent_resizeEvent(e)
|
||
|
|
qr_button = parent.button
|
||
|
|
left = qr_button.geometry().left() - audio_button.sizeHint().width()
|
||
|
|
top = qr_button.geometry().top()
|
||
|
|
audio_button.move(left, top)
|
||
|
|
return result
|
||
|
|
|
||
|
|
parent.resizeEvent = resizeEvent
|
||
|
|
return audio_button
|