Riccardo Spagni
10 years ago
2 changed files with 261 additions and 1 deletions
@ -0,0 +1,259 @@ |
|||
from electrum_gui.qt.util import EnterButton |
|||
from electrum.plugins import BasePlugin, hook |
|||
from electrum.i18n import _ |
|||
from PyQt4.QtGui import * |
|||
from PyQt4.QtCore import * |
|||
|
|||
import re |
|||
|
|||
import dns.name |
|||
import dns.query |
|||
import dns.dnssec |
|||
import dns.message |
|||
import dns.resolver |
|||
import dns.rdatatype |
|||
from dns.exception import DNSException |
|||
|
|||
|
|||
class Plugin(BasePlugin): |
|||
def fullname(self): |
|||
return 'OpenAlias' |
|||
|
|||
def description(self): |
|||
return 'Import contacts by OpenAlias.' |
|||
|
|||
def __init__(self, gui, name): |
|||
BasePlugin.__init__(self, gui, name) |
|||
self._is_available = True |
|||
|
|||
@hook |
|||
def init_qt(self, gui): |
|||
self.gui = gui |
|||
self.win = gui.main_window |
|||
|
|||
def requires_settings(self): |
|||
return True |
|||
|
|||
def settings_widget(self, window): |
|||
return EnterButton(_('Settings'), self.settings_dialog) |
|||
|
|||
@hook |
|||
def before_send(self): |
|||
''' |
|||
Change URL to address before making a send. |
|||
IMPORTANT: |
|||
return False to continue execution of the send |
|||
return True to stop execution of the send |
|||
''' |
|||
if self.win.payto_e.is_multiline(): # only supports single line entries atm |
|||
return False |
|||
url = str(self.win.payto_e.toPlainText()) |
|||
|
|||
if not '.' in url: |
|||
return False |
|||
|
|||
data = self.resolve(url) |
|||
|
|||
if not data: |
|||
return True |
|||
|
|||
if not self.validate_dnssec(url): |
|||
msgBox = QMessageBox() |
|||
msgBox.setText(_('No valid DNSSEC trust chain!')) |
|||
msgBox.setInformativeText(_('Do you wish to continue?')) |
|||
msgBox.setStandardButtons(QMessageBox.Cancel | QMessageBox.Ok) |
|||
msgBox.setDefaultButton(QMessageBox.Cancel) |
|||
reply = msgBox.exec_() |
|||
if reply != QMessageBox.Ok: |
|||
return True |
|||
|
|||
(address, name) = data |
|||
self.win.payto_e.setText(address) |
|||
if self.config.get('openalias_autoadd') == 'checked': |
|||
self.win.wallet.add_contact(address, name) |
|||
return False |
|||
|
|||
def settings_dialog(self): |
|||
'''Settings dialog.''' |
|||
d = QDialog() |
|||
d.setWindowTitle("Settings") |
|||
layout = QGridLayout(d) |
|||
layout.addWidget(QLabel(_('Automatically add to contacts')), 0, 0) |
|||
autoadd_checkbox = QCheckBox() |
|||
autoadd_checkbox.setEnabled(True) |
|||
autoadd_checkbox.setChecked(self.config.get('openalias_autoadd', 'unchecked') != 'unchecked') |
|||
layout.addWidget(autoadd_checkbox, 0, 1) |
|||
ok_button = QPushButton(_("OK")) |
|||
ok_button.clicked.connect(d.accept) |
|||
layout.addWidget(ok_button, 1, 1) |
|||
|
|||
def on_change_autoadd(checked): |
|||
if checked: |
|||
self.config.set_key('openalias_autoadd', 'checked') |
|||
else: |
|||
self.config.set_key('openalias_autoadd', 'unchecked') |
|||
|
|||
autoadd_checkbox.stateChanged.connect(on_change_autoadd) |
|||
|
|||
return bool(d.exec_()) |
|||
|
|||
def openalias_contact_dialog(self): |
|||
'''Previous version using a get contact button from settings, currently unused.''' |
|||
d = QDialog(self.win) |
|||
vbox = QVBoxLayout(d) |
|||
vbox.addWidget(QLabel(_('Openalias Contact') + ':')) |
|||
|
|||
grid = QGridLayout() |
|||
line1 = QLineEdit() |
|||
grid.addWidget(QLabel(_("URL")), 1, 0) |
|||
grid.addWidget(line1, 1, 1) |
|||
|
|||
vbox.addLayout(grid) |
|||
vbox.addLayout(ok_cancel_buttons(d)) |
|||
|
|||
if not d.exec_(): |
|||
return |
|||
|
|||
url = str(line1.text()) |
|||
|
|||
if not '.' in url: |
|||
QMessageBox.warning(self.win, _('Error'), _('Invalid URL'), _('OK')) |
|||
return |
|||
|
|||
data = self.resolve(url) |
|||
|
|||
if not data: |
|||
return |
|||
|
|||
if not self.validate_dnssec(url): |
|||
msgBox = QMessageBox() |
|||
msgBox.setText("No valid DNSSEC trust chain!") |
|||
msgBox.setInformativeText("Do you wish to continue?") |
|||
msgBox.setStandardButtons(QMessageBox.Cancel | QMessageBox.Ok) |
|||
msgBox.setDefaultButton(QMessageBox.Cancel) |
|||
reply = msgBox.exec_() |
|||
if reply != QMessageBox.Ok: |
|||
return |
|||
|
|||
(address, name) = data |
|||
|
|||
d2 = QDialog(self.win) |
|||
vbox2 = QVBoxLayout(d2) |
|||
grid2 = QGridLayout() |
|||
grid2.addWidget(QLabel(url), 1, 1) |
|||
if name: |
|||
grid2.addWidget(QLabel('Name: '), 2, 0) |
|||
grid2.addWidget(QLabel(name), 2, 1) |
|||
|
|||
grid2.addWidget(QLabel('Address: '), 4, 0) |
|||
grid2.addWidget(QLabel(address), 4, 1) |
|||
|
|||
vbox2.addLayout(grid2) |
|||
vbox2.addLayout(ok_cancel_buttons(d2)) |
|||
|
|||
if not d2.exec_(): |
|||
return |
|||
|
|||
self.win.wallet.add_contact(address) |
|||
|
|||
try: |
|||
label = url + " (" + name + ")" |
|||
except Exception: |
|||
pass |
|||
|
|||
if label: |
|||
self.win.wallet.set_label(address, label) |
|||
|
|||
self.win.update_contacts_tab() |
|||
self.win.update_history_tab() |
|||
self.win.update_completions() |
|||
self.win.tabs.setCurrentIndex(3) |
|||
|
|||
def resolve(self, url): |
|||
'''Resolve OpenAlias address using url.''' |
|||
prefix = 'btc' |
|||
retries = 3 |
|||
err = None |
|||
for i in range(0, retries): |
|||
try: |
|||
records = dns.resolver.query(url, 'TXT') |
|||
for record in records: |
|||
string = record.strings[0] |
|||
if string.startswith('oa1:' + prefix): |
|||
address = self.find_regex(string, r'recipient_address=([A-Za-z0-9]+)') |
|||
name = self.find_regex(string, r'recipient_name=([^;]+)') |
|||
if not address: |
|||
continue |
|||
return (address, name) |
|||
QMessageBox.warning(self.win, _('Error'), _('No OpenAlias record found.'), _('OK')) |
|||
return 0 |
|||
except dns.resolver.NXDOMAIN: |
|||
err = _('No such domain.') |
|||
continue |
|||
except dns.resolver.Timeout: |
|||
err = _('Timed out while resolving.') |
|||
continue |
|||
except DNSException: |
|||
err = _('Unhandled exception.') |
|||
continue |
|||
except: |
|||
err = _('Unknown error.') |
|||
continue |
|||
break |
|||
if err: |
|||
QMessageBox.warning(self.win, _('Error'), err, _('OK')) |
|||
return 0 |
|||
|
|||
def find_regex(self, haystack, needle): |
|||
regex = re.compile(needle) |
|||
try: |
|||
return regex.search(haystack).groups()[0] |
|||
except AttributeError: |
|||
return None |
|||
|
|||
def validate_dnssec(self, url): |
|||
default = dns.resolver.get_default_resolver() |
|||
ns = default.nameservers[0] |
|||
|
|||
parts = url.split('.') |
|||
|
|||
for i in xrange(len(parts), 0, -1): |
|||
sub = '.'.join(parts[i - 1:]) |
|||
|
|||
query = dns.message.make_query(sub, dns.rdatatype.NS) |
|||
response = dns.query.udp(query, ns) |
|||
|
|||
if response.rcode() != dns.rcode.NOERROR: |
|||
return 0 |
|||
|
|||
if len(response.authority) > 0: |
|||
rrset = response.authority[0] |
|||
else: |
|||
rrset = response.answer[0] |
|||
|
|||
rr = rrset[0] |
|||
if rr.rdtype == dns.rdatatype.SOA: |
|||
#Same server is authoritative, don't check again |
|||
continue |
|||
|
|||
query = dns.message.make_query(sub, |
|||
dns.rdatatype.DNSKEY, |
|||
want_dnssec=True) |
|||
response = dns.query.udp(query, ns) |
|||
|
|||
if response.rcode() != 0: |
|||
return 0 |
|||
# HANDLE QUERY FAILED (SERVER ERROR OR NO DNSKEY RECORD) |
|||
|
|||
# answer should contain two RRSET: DNSKEY and RRSIG(DNSKEY) |
|||
answer = response.answer |
|||
if len(answer) != 2: |
|||
return 0 |
|||
|
|||
# the DNSKEY should be self signed, validate it |
|||
name = dns.name.from_text(sub) |
|||
try: |
|||
dns.dnssec.validate(answer[0], answer[1], {name: answer[0]}) |
|||
except dns.dnssec.ValidationFailure: |
|||
return 0 |
|||
return 1 |
|||
Loading…
Reference in new issue