123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547 |
- #!/usr/bin/env python
- #
- # mmgen = Multi-Mode GENerator, command-line Bitcoin cold storage solution
- # Copyright (C)2013-2016 Philemon <mmgen-py@yandex.com>
- #
- # This program is free software: you can redistribute it and/or modify
- # it under the terms of the GNU General Public License as published by
- # the Free Software Foundation, either version 3 of the License, or
- # (at your option) any later version.
- #
- # This program is distributed in the hope that it will be useful,
- # but WITHOUT ANY WARRANTY; without even the implied warranty of
- # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- # GNU General Public License for more details.
- #
- # You should have received a copy of the GNU General Public License
- # along with this program. If not, see <http://www.gnu.org/licenses/>.
- """
- addr.py: Address generation/display routines for the MMGen suite
- """
- from hashlib import sha256, sha512
- from mmgen.common import *
- from mmgen.bitcoin import privnum2addr,hex2wif,wif2hex
- from mmgen.obj import *
- from mmgen.tx import *
- from mmgen.tw import *
- pnm = g.proj_name
- def _test_for_keyconv(silent=False):
- no_keyconv_errmsg = """
- Executable '{kconv}' unavailable. Please install '{kconv}' from the {vgen}
- package on your system or specify the secp256k1 library.
- """.format(kconv=g.keyconv_exec, vgen='vanitygen')
- from subprocess import check_output,STDOUT
- try:
- check_output([g.keyconv_exec, '-G'],stderr=STDOUT)
- except:
- if not silent: msg(no_keyconv_errmsg.strip())
- return False
- return True
- def _test_for_secp256k1(silent=False):
- no_secp256k1_errmsg = """
- secp256k1 library unavailable. Will use '{kconv}', or failing that, the (slow)
- internal ECDSA library for address generation.
- """.format(kconv=g.keyconv_exec)
- try:
- from mmgen.secp256k1 import priv2pub
- assert priv2pub(os.urandom(32),1)
- except:
- if not silent: msg(no_secp256k1_errmsg.strip())
- return False
- return True
- def _wif2addr_python(wif):
- privhex = wif2hex(wif)
- if not privhex: return False
- return privnum2addr(int(privhex,16),wif[0] != ('5','9')[g.testnet])
- def _wif2addr_keyconv(wif):
- if wif[0] == ('5','9')[g.testnet]:
- from subprocess import check_output
- return check_output(['keyconv', wif]).split()[1]
- else:
- return _wif2addr_python(wif)
- def _wif2addr_secp256k1(wif):
- return _privhex2addr_secp256k1(wif2hex(wif),wif[0] != ('5','9')[g.testnet])
- def _privhex2addr_python(privhex,compressed=False):
- return privnum2addr(int(privhex,16),compressed)
- def _privhex2addr_keyconv(privhex,compressed=False):
- if compressed:
- return privnum2addr(int(privhex,16),compressed)
- else:
- from subprocess import check_output
- return check_output(['keyconv', hex2wif(privhex,compressed=False)]).split()[1]
- def _privhex2addr_secp256k1(privhex,compressed=False):
- from mmgen.secp256k1 import priv2pub
- from mmgen.bitcoin import hexaddr2addr,pubhex2hexaddr
- from binascii import hexlify,unhexlify
- pubkey = priv2pub(unhexlify(privhex),int(compressed))
- return hexaddr2addr(pubhex2hexaddr(hexlify(pubkey)))
- def _keygen_selector(generator=None):
- if generator:
- if generator == 3 and _test_for_secp256k1(): return 2
- elif generator in (2,3) and _test_for_keyconv(): return 1
- else:
- if opt.key_generator == 3 and _test_for_secp256k1(): return 2
- elif opt.key_generator in (2,3) and _test_for_keyconv(): return 1
- msg('Using (slow) internal ECDSA library for address generation')
- return 0
- def get_wif2addr_f(generator=None):
- gen = _keygen_selector(generator=generator)
- return (_wif2addr_python,_wif2addr_keyconv,_wif2addr_secp256k1)[gen]
- def get_privhex2addr_f(generator=None):
- gen = _keygen_selector(generator=generator)
- return (_privhex2addr_python,_privhex2addr_keyconv,_privhex2addr_secp256k1)[gen]
- class AddrListEntry(MMGenListItem):
- attrs = 'idx','addr','label','wif','sec'
- label = MMGenListItemAttr('label','MMGenAddrLabel')
- idx = MMGenListItemAttr('idx','AddrIdx')
- class AddrListChksum(str,Hilite):
- color = 'pink'
- trunc_ok = False
- def __new__(cls,addrlist):
- lines=[' '.join([str(e.idx),e.addr]+([e.wif] if addrlist.has_keys else []))
- for e in addrlist.data]
- return str.__new__(cls,make_chksum_N(' '.join(lines), nchars=16, sep=True))
- class AddrListID(str,Hilite):
- color = 'green'
- trunc_ok = False
- def __new__(cls,addrlist):
- try: int(addrlist.data[0].idx)
- except:
- s = '(no idxs)'
- else:
- idxs = [e.idx for e in addrlist.data]
- prev = idxs[0]
- ret = prev,
- for i in idxs[1:]:
- if i == prev + 1:
- if i == idxs[-1]: ret += '-', i
- else:
- if prev != ret[-1]: ret += '-', prev
- ret += ',', i
- prev = i
- s = ''.join([str(i) for i in ret])
- return str.__new__(cls,'%s[%s]' % (addrlist.seed_id,s))
- class AddrList(MMGenObject): # Address info for a single seed ID
- msgs = {
- 'file_header': """
- # {pnm} address file
- #
- # This file is editable.
- # Everything following a hash symbol '#' is a comment and ignored by {pnm}.
- # A text label of {n} characters or less may be added to the right of each
- # address, and it will be appended to the bitcoind wallet label upon import.
- # The label may contain any printable ASCII symbol.
- """.strip().format(n=MMGenAddrLabel.max_len,pnm=pnm),
- 'record_chksum': """
- Record this checksum: it will be used to verify the address file in the future
- """.strip(),
- 'check_chksum': 'Check this value against your records',
- 'removed_dups': """
- Removed %s duplicate wif key%s from keylist (also in {pnm} key-address file
- """.strip().format(pnm=pnm)
- }
- data_desc = 'address'
- file_desc = 'addresses'
- gen_desc = 'address'
- gen_desc_pl = 'es'
- gen_addrs = True
- gen_keys = False
- has_keys = False
- ext = 'addrs'
- def __init__(self,addrfile='',sid='',adata=[],seed='',addr_idxs='',src='',
- addrlist='',keylist='',do_chksum=True,chksum_only=False):
- self.update_msgs()
- if addrfile: # data from MMGen address file
- (sid,adata) = self.parse_file(addrfile)
- elif sid and adata: # data from tracking wallet
- do_chksum = False
- elif seed and addr_idxs: # data from seed + idxs
- sid,src = seed.sid,'gen'
- adata = self.generate(seed,addr_idxs)
- elif addrlist: # data from flat address list
- sid = None
- adata = [AddrListEntry(addr=a) for a in addrlist]
- elif keylist: # data from flat key list
- sid,do_chksum = None,False
- adata = [AddrListEntry(wif=k) for k in keylist]
- elif seed or addr_idxs:
- die(3,'Must specify both seed and addr indexes')
- elif sid or adata:
- die(3,'Must specify both seed_id and adata')
- else:
- die(3,'Incorrect arguments for %s' % type(self).__name__)
- # sid,adata now set
- self.seed_id = sid
- self.data = adata
- self.num_addrs = len(adata)
- self.fmt_data = ''
- self.id_str = None
- self.chksum = None
- if type(self) == KeyList:
- self.id_str = AddrListID(self)
- return
- if do_chksum:
- self.chksum = AddrListChksum(self)
- if chksum_only:
- Msg(self.chksum)
- else:
- self.id_str = AddrListID(self)
- qmsg('Checksum for %s data %s: %s' %
- (self.data_desc,self.id_str.hl(),self.chksum.hl()))
- qmsg(self.msgs[('check_chksum','record_chksum')[src=='gen']])
- def update_msgs(self):
- if type(self).msgs and type(self) != AddrList:
- for k in AddrList.msgs:
- if k not in self.msgs:
- self.msgs[k] = AddrList.msgs[k]
- def generate(self,seed,addrnums):
- assert type(addrnums) is AddrIdxList
- self.seed_id = SeedID(seed=seed)
- seed = seed.get_data()
- if self.gen_addrs:
- privhex2addr_f = get_privhex2addr_f()
- t_addrs,num,pos,out = len(addrnums),0,0,[]
- while pos != t_addrs:
- seed = sha512(seed).digest()
- num += 1 # round
- if num != addrnums[pos]: continue
- pos += 1
- qmsg_r('\rGenerating %s #%s (%s of %s)' % (self.gen_desc,num,pos,t_addrs))
- e = AddrListEntry(idx=num)
- # Secret key is double sha256 of seed hash round /num/
- sec = sha256(sha256(seed).digest()).hexdigest()
- if self.gen_addrs:
- e.addr = privhex2addr_f(sec,compressed=False)
- if self.gen_keys:
- e.wif = hex2wif(sec,compressed=False)
- if opt.b16: e.sec = sec
- out.append(e)
- qmsg('\r%s: %s %s%s generated%s' % (
- self.seed_id.hl(),t_addrs,self.gen_desc,suf(t_addrs,self.gen_desc_pl),' '*15))
- return out
- def encrypt(self):
- from mmgen.crypto import mmgen_encrypt
- self.fmt_data = mmgen_encrypt(self.fmt_data,'new key list','')
- self.ext += '.'+g.mmenc_ext
- def write_to_file(self,ask_tty=True,ask_write_default_yes=False,binary=False):
- fn = '{}.{}'.format(self.id_str,self.ext)
- ask_tty = self.has_keys and not opt.quiet
- write_data_to_file(fn,self.fmt_data,self.file_desc,ask_tty=ask_tty,binary=binary)
- def idxs(self):
- return [e.idx for e in self.data]
- def addrs(self):
- return ['%s:%s'%(self.seed_id,e.idx) for e in self.data]
- def addrpairs(self):
- return [(e.idx,e.addr) for e in self.data]
- def btcaddrs(self):
- return [e.addr for e in self.data]
- def comments(self):
- return [e.label for e in self.data]
- def entry(self,idx):
- for e in self.data:
- if idx == e.idx: return e
- def btcaddr(self,idx):
- for e in self.data:
- if idx == e.idx: return e.addr
- def comment(self,idx):
- for e in self.data:
- if idx == e.idx: return e.label
- def set_comment(self,idx,comment):
- for e in self.data:
- if idx == e.idx:
- e.label = comment
- def make_reverse_dict(self,btcaddrs):
- d,b = {},btcaddrs
- for e in self.data:
- try:
- d[b[b.index(e.addr)]] = ('%s:%s'%(self.seed_id,e.idx),e.label)
- except: pass
- return d
- def flat_list(self):
- class AddrListFlatEntry(AddrListEntry):
- attrs = 'mmid','addr','wif'
- return [AddrListFlatEntry(
- mmid='{}:{}'.format(self.seed_id,e.idx),
- addr=e.addr,
- wif=e.wif)
- for e in self.data]
- def remove_dups(self,cmplist,key='wif'):
- pop_list = []
- for n,d in enumerate(self.data):
- if getattr(d,key) == None: continue
- for e in cmplist.data:
- if getattr(e,key) and getattr(e,key) == getattr(d,key):
- pop_list.append(n)
- for n in reversed(pop_list): self.data.pop(n)
- if pop_list:
- vmsg(self.msgs['removed_dups'] % (len(pop_list),suf(removed,'k')))
- def add_wifs(self,al_key):
- for d in self.data:
- for e in al_key.data:
- if e.addr and e.wif and e.addr == d.addr:
- d.wif = e.wif
- def list_missing(self,key):
- return [d for d in self.data if not getattr(d,key)]
- def get(self,key):
- return [getattr(d,key) for d in self.data if getattr(d,key)]
- def get_addrs(self): return self.get('addr')
- def get_wifs(self): return self.get('wif')
- def generate_addrs(self):
- wif2addr_f = get_wif2addr_f()
- d = self.data
- for n,e in enumerate(d,1):
- qmsg_r('\rGenerating addresses from keylist: %s/%s' % (n,len(d)))
- e.addr = wif2addr_f(e.wif)
- qmsg('\rGenerated addresses from keylist: %s/%s ' % (n,len(d)))
- def format(self,enable_comments=False):
- def check_attrs(key,desc):
- for e in self.data:
- if not getattr(e,key):
- die(3,'missing %s in addr data' % desc)
- if type(self) != KeyList: check_attrs('addr','addresses')
- if self.has_keys:
- if opt.b16: check_attrs('sec','hex keys')
- check_attrs('wif','wif keys')
- out = [self.msgs['file_header']+'\n']
- if self.chksum:
- out.append('# {} data checksum for {}: {}'.format(
- self.data_desc.capitalize(),self.id_str,self.chksum))
- out.append('# Record this value to a secure location.\n')
- out.append('%s {' % self.seed_id)
- fs = ' {:<%s} {:<34}{}' % len(str(self.data[-1].idx))
- for e in self.data:
- c = ' '+e.label if enable_comments and e.label else ''
- if type(self) == KeyList:
- out.append(fs.format(e.idx, 'wif: '+e.wif,c))
- else: # First line with idx
- out.append(fs.format(e.idx, e.addr,c))
- if self.has_keys:
- if opt.b16: out.append(fs.format('', 'hex: '+e.sec,c))
- out.append(fs.format('', 'wif: '+e.wif,c))
- out.append('}')
- self.fmt_data = '\n'.join([l.rstrip() for l in out]) + '\n'
- def parse_file_body(self,lines):
- if self.has_keys and len(lines) % 2:
- return 'Key-address file has odd number of lines'
- ret = []
- while lines:
- l = lines.pop(0)
- d = l.split(None,2)
- if not is_mmgen_idx(d[0]):
- return "'%s': invalid address num. in line: '%s'" % (d[0],l)
- if not is_btc_addr(d[1]):
- return "'%s': invalid Bitcoin address" % d[1]
- if len(d) != 3: d.append('')
- a = AddrListEntry(idx=int(d[0]),addr=d[1],label=d[2])
- if self.has_keys:
- l = lines.pop(0)
- d = l.split(None,2)
- if d[0] != 'wif:':
- return "Invalid key line in file: '%s'" % l
- if not is_wif(d[1]):
- return "'%s': invalid Bitcoin key" % d[1]
- a.wif = d[1]
- ret.append(a)
- if self.has_keys and keypress_confirm('Check key-to-address validity?'):
- wif2addr_f = get_wif2addr_f()
- llen = len(ret)
- for n,e in enumerate(ret):
- msg_r('\rVerifying keys %s/%s' % (n+1,llen))
- if e.addr != wif2addr_f(e.wif):
- return "Key doesn't match address!\n %s\n %s" % (e.wif,e.addr)
- msg(' - done')
- return ret
- def parse_file(self,fn,buf=[],exit_on_error=True):
- lines = get_lines_from_file(fn,self.data_desc+' data',trim_comments=True)
- try:
- sid,obrace = lines[0].split()
- except:
- errmsg = "Invalid first line: '%s'" % lines[0]
- else:
- cbrace = lines[-1]
- if obrace != '{':
- errmsg = "'%s': invalid first line" % lines[0]
- elif cbrace != '}':
- errmsg = "'%s': invalid last line" % cbrace
- elif not is_mmgen_seed_id(sid):
- errmsg = "'%s': invalid Seed ID" % sid
- else:
- ret = self.parse_file_body(lines[1:-1])
- if type(ret) == list:
- return sid,ret
- else:
- errmsg = ret
- if exit_on_error: die(3,errmsg)
- msg(errmsg)
- return False
- class KeyAddrList(AddrList):
- data_desc = 'key-address'
- file_desc = 'secret keys'
- gen_desc = 'key/address pair'
- gen_desc_pl = 's'
- gen_addrs = True
- gen_keys = True
- has_keys = True
- ext = 'akeys'
- class KeyList(AddrList):
- msgs = {
- 'file_header': """
- # {pnm} key file
- #
- # This file is editable.
- # Everything following a hash symbol '#' is a comment and ignored by {pnm}.
- """.strip().format(pnm=pnm)
- }
- data_desc = 'key'
- file_desc = 'secret keys'
- gen_desc = 'key'
- gen_desc_pl = 's'
- gen_addrs = False
- gen_keys = True
- has_keys = True
- ext = 'keys'
- class AddrData(MMGenObject):
- msgs = {
- 'too_many_acct_addresses': """
- ERROR: More than one address found for account: '%s'.
- Your 'wallet.dat' file appears to have been altered by a non-{pnm} program.
- Please restore your tracking wallet from a backup or create a new one and
- re-import your addresses.
- """.strip().format(pnm=pnm)
- }
- def __init__(self,source=None):
- self.sids = {}
- if source == 'tw': self.add_tw_data()
- def seed_ids(self):
- return self.sids.keys()
- def addrlist(self,sid):
- # TODO: Validate sid
- if sid in self.sids:
- return self.sids[sid]
- def mmaddr2btcaddr(self,mmaddr):
- btcaddr = ''
- sid,idx = mmaddr.split(':')
- if sid in self.seed_ids():
- btcaddr = self.addrlist(sid).btcaddr(int(idx))
- return btcaddr
- def add_tw_data(self):
- vmsg_r('Getting address data from tracking wallet...')
- c = bitcoin_connection()
- accts = c.listaccounts(0,True)
- data,i = {},0
- alists = c.getaddressesbyaccount([[k] for k in accts],batch=True)
- for acct,addrlist in zip(accts,alists):
- maddr,label = parse_tw_acct_label(acct)
- if maddr:
- i += 1
- if len(addrlist) != 1:
- die(2,self.msgs['too_many_acct_addresses'] % acct)
- seed_id,idx = maddr.split(':')
- if seed_id not in data:
- data[seed_id] = []
- data[seed_id].append(AddrListEntry(idx=idx,addr=addrlist[0],label=label))
- vmsg('{n} {pnm} addresses found, {m} accounts total'.format(
- n=i,pnm=pnm,m=len(accts)))
- for sid in data:
- self.add(AddrList(sid=sid,adata=data[sid]))
- def add(self,addrlist):
- if type(addrlist) == AddrList:
- self.sids[addrlist.seed_id] = addrlist
- return True
- else:
- raise TypeError, 'Error: object %s is not of type AddrList' % repr(addrlist)
- def make_reverse_dict(self,btcaddrs):
- d = {}
- for sid in self.sids:
- d.update(self.sids[sid].make_reverse_dict(btcaddrs))
- return d
|