mirror of
https://github.com/cryptoadvance/specter-desktop.git
synced 2026-08-13 12:33:29 +02:00
556ec5f961
* automate pip-packages-gen + upload for gittags release documentation proper package including and static + templates * Change import- and start logic, removed cli.py * project-renaming to cryptoadvance.specter propoer packaging include proper dependencies * Adding requirements.txt * Specify maintainers and thanks to contributors * Removing test-url for pip-uload. This is now live!
140 lines
4.2 KiB
Python
140 lines
4.2 KiB
Python
import re
|
|
|
|
# From: https://github.com/bitcoin/bitcoin/blob/master/src/script/descriptor.cpp
|
|
|
|
def PolyMod(c, val):
|
|
c0 = c >> 35
|
|
c = ((c & 0x7ffffffff) << 5) ^ val
|
|
if (c0 & 1):
|
|
c ^= 0xf5dee51989
|
|
if (c0 & 2):
|
|
c ^= 0xa9fdca3312
|
|
if (c0 & 4):
|
|
c ^= 0x1bab10e32d
|
|
if (c0 & 8):
|
|
c ^= 0x3706b1677a
|
|
if (c0 & 16):
|
|
c ^= 0x644d626ffd
|
|
return c
|
|
|
|
def DescriptorChecksum(desc):
|
|
INPUT_CHARSET = "0123456789()[],'/*abcdefgh@:$%{}IJKLMNOPQRSTUVWXYZ&+-.;<=>?!^_|~ijklmnopqrstuvwxyzABCDEFGH`#\"\\ ";
|
|
CHECKSUM_CHARSET = "qpzry9x8gf2tvdw0s3jn54khce6mua7l";
|
|
|
|
c = 1
|
|
cls = 0
|
|
clscount = 0
|
|
for ch in desc:
|
|
pos = INPUT_CHARSET.find(ch)
|
|
if pos == -1:
|
|
return ""
|
|
c = PolyMod(c, pos & 31)
|
|
cls = cls * 3 + (pos >> 5)
|
|
clscount += 1
|
|
if clscount == 3:
|
|
c = PolyMod(c, cls)
|
|
cls = 0
|
|
clscount = 0
|
|
if clscount > 0:
|
|
c = PolyMod(c, cls)
|
|
for j in range (0, 8):
|
|
c = PolyMod(c, 0)
|
|
c ^= 1
|
|
|
|
ret = [None] * 8
|
|
for j in range(0, 8):
|
|
ret[j] = CHECKSUM_CHARSET[(c >> (5 * (7 - j))) & 31]
|
|
return ''.join(ret)
|
|
|
|
def AddChecksum(desc):
|
|
return desc + "#" + DescriptorChecksum(desc)
|
|
|
|
class Descriptor:
|
|
def __init__(self, origin_fingerprint, origin_path, base_key, path_suffix, testnet, sh_wpkh, wpkh):
|
|
self.origin_fingerprint = origin_fingerprint
|
|
self.origin_path = origin_path
|
|
self.path_suffix = path_suffix
|
|
self.base_key = base_key
|
|
self.testnet = testnet
|
|
self.sh_wpkh = sh_wpkh
|
|
self.wpkh = wpkh
|
|
self.m_path = None
|
|
|
|
if origin_path:
|
|
self.m_path_base = "m" + origin_path
|
|
self.m_path = "m" + origin_path + (path_suffix or "")
|
|
|
|
@classmethod
|
|
def parse(cls, desc, testnet = False):
|
|
sh_wpkh = None
|
|
wpkh = None
|
|
origin_fingerprint = None
|
|
origin_path = None
|
|
base_key_and_path_match = None
|
|
base_key = None
|
|
path_suffix = None
|
|
|
|
# Check the checksum
|
|
check_split = desc.split('#')
|
|
if len(check_split) > 2:
|
|
return None
|
|
if len(check_split) == 2:
|
|
if len(check_split[1]) != 8:
|
|
return None
|
|
checksum = DescriptorChecksum(check_split[0])
|
|
if not checksum.strip():
|
|
return None
|
|
if checksum != check_split[1]:
|
|
return None
|
|
desc = check_split[0]
|
|
|
|
if desc.startswith("sh(wpkh("):
|
|
sh_wpkh = True
|
|
elif desc.startswith("wpkh("):
|
|
wpkh = True
|
|
|
|
origin_match = re.search(r"\[(.*)\]", desc)
|
|
if origin_match:
|
|
origin = origin_match.group(1)
|
|
match = re.search(r"^([0-9a-fA-F]{8})(\/.*)", origin)
|
|
if match:
|
|
origin_fingerprint = match.group(1)
|
|
origin_path = match.group(2)
|
|
# Replace h with '
|
|
origin_path = origin_path.replace('h', '\'')
|
|
|
|
base_key_and_path_match = re.search(r"\[.*\](\w+)([\/\)][\d'\/\*]*)", desc)
|
|
else:
|
|
base_key_and_path_match = re.search(r"\((\w+)([\/\)][\d'\/\*]*)", desc)
|
|
|
|
if base_key_and_path_match:
|
|
base_key = base_key_and_path_match.group(1)
|
|
path_suffix = base_key_and_path_match.group(2)
|
|
if path_suffix == ")":
|
|
path_suffix = None
|
|
else:
|
|
if origin_match == None:
|
|
return None
|
|
|
|
return cls(origin_fingerprint, origin_path, base_key, path_suffix, testnet, sh_wpkh, wpkh)
|
|
|
|
|
|
def serialize(self):
|
|
descriptor_open = 'pkh('
|
|
descriptor_close = ')'
|
|
origin = ''
|
|
path_suffix = ''
|
|
|
|
if self.wpkh == True:
|
|
descriptor_open = 'wpkh('
|
|
elif self.sh_wpkh == True:
|
|
descriptor_open = 'sh(wpkh('
|
|
descriptor_close = '))'
|
|
|
|
if self.origin_fingerprint and self.origin_path:
|
|
origin = '[' + self.origin_fingerprint + self.origin_path + ']'
|
|
|
|
if self.path_suffix:
|
|
path_suffix = self.path_suffix
|
|
|
|
return AddChecksum(descriptor_open + origin + self.base_key + path_suffix + descriptor_close)
|