from util import print_msg, print_error, user_dir, format_satoshis
from bitcoin import *
-
+from account import *
# AES encryption
EncodeAES = lambda secret, s: base64.b64encode(aes.encryptData(secret,s))
self.seed_version = config.get('seed_version', SEED_VERSION)
self.gap_limit = config.get('gap_limit', 5)
self.use_change = config.get('use_change',True)
- self.fee = int(config.get('fee_per_kb',20000))
+ self.fee = int(config.get('fee_per_kb',50000))
self.num_zeros = int(config.get('num_zeros',0))
self.use_encryption = config.get('use_encryption', False)
self.seed = config.get('seed', '') # encrypted
self.frozen_addresses = config.get('frozen_addresses',[])
self.prioritized_addresses = config.get('prioritized_addresses',[])
self.addressbook = config.get('contacts', [])
+
self.imported_keys = config.get('imported_keys',{})
self.history = config.get('addr_history',{}) # address -> list(txid, height)
- self.accounts = config.get('accounts', {}) # this should not include public keys
- self.SequenceClass = ElectrumSequence
- self.sequences = {}
- self.sequences[0] = self.SequenceClass(self.config.get('master_public_key'))
- if self.accounts.get(0) is None:
- self.accounts[0] = { 0:[], 1:[], 'name':'Main account' }
+ self.master_public_keys = config.get('master_public_keys',{})
+ self.master_private_keys = config.get('master_private_keys', {})
+
+ self.load_accounts(config)
self.transactions = {}
tx = config.get('transactions',{})
if self.seed_version != SEED_VERSION:
raise ValueError("This wallet seed is deprecated. Please run upgrade.py for a diagnostic.")
- for tx_hash in self.transactions.keys():
- self.update_tx_outputs(tx_hash)
+ for tx_hash, tx in self.transactions.items():
+ if self.check_new_tx(tx_hash, tx):
+ self.update_tx_outputs(tx_hash)
+ else:
+ print_error("unreferenced tx", tx_hash)
+ self.transactions.pop(tx_hash)
def set_up_to_date(self,b):
def import_key(self, sec, password):
# check password
seed = self.decode_seed(password)
- address = address_from_private_key(sec)
+ try:
+ address = address_from_private_key(sec)
+ except:
+ raise BaseException('Invalid private key')
if self.is_mine(address):
raise BaseException('Address already in wallet')
self.config.set_key('imported_keys', self.imported_keys, True)
return address
+ def delete_imported_key(self, addr):
+ if addr in self.imported_keys:
+ self.imported_keys.pop(addr)
+ self.config.set_key('imported_keys', self.imported_keys, True)
+
def init_seed(self, seed):
if self.seed: raise BaseException("a seed exists")
seed = random_seed(128)
self.seed = seed
+
def save_seed(self):
self.config.set_key('seed', self.seed, True)
self.config.set_key('seed_version', self.seed_version, True)
- mpk = self.SequenceClass.mpk_from_seed(self.seed)
- self.init_sequence(mpk)
+
+ master_k, master_c, master_K, master_cK = bip32_init(self.seed)
+
+ # normal accounts
+ k0, c0, K0, cK0 = bip32_private_derivation(master_k, master_c, "m/", "m/0'/")
+ # p2sh 2of2
+ k1, c1, K1, cK1 = bip32_private_derivation(master_k, master_c, "m/", "m/1'/")
+ k2, c2, K2, cK2 = bip32_private_derivation(master_k, master_c, "m/", "m/2'/")
+ # p2sh 2of3
+ k3, c3, K3, cK3 = bip32_private_derivation(master_k, master_c, "m/", "m/3'/")
+ k4, c4, K4, cK4 = bip32_private_derivation(master_k, master_c, "m/", "m/4'/")
+ k5, c5, K5, cK5 = bip32_private_derivation(master_k, master_c, "m/", "m/5'/")
+
+ self.master_public_keys = {
+ "m/0'/": (c0, K0, cK0),
+ "m/1'/": (c1, K1, cK1),
+ "m/2'/": (c2, K2, cK2),
+ "m/3'/": (c3, K3, cK3),
+ "m/4'/": (c4, K4, cK4),
+ "m/5'/": (c5, K5, cK5)
+ }
+
+ self.master_private_keys = {
+ "m/0'/": k0,
+ "m/1'/": k1,
+ "m/2'/": k2,
+ "m/3'/": k3,
+ "m/4'/": k4,
+ "m/5'/": k5
+ }
+
+ self.config.set_key('master_public_keys', self.master_public_keys, True)
+ self.config.set_key('master_private_keys', self.master_private_keys, True)
+
+ # create default account
+ self.create_account('Main account')
+
+
+ def account_id(self, account_type, i):
+ if account_type is None:
+ return "m/0'/%d"%i
+ elif account_type == '2of2':
+ return "m/1'/%d & m/2'/%d"%(i,i)
+ elif account_type == '2of3':
+ return "m/3'/%d & m/4'/%d & m/5'/%d"%(i,i,i)
+ else:
+ raise BaseException('unknown account type')
+
+
+ def num_accounts(self, account_type):
+ keys = self.accounts.keys()
+ i = 0
+ while True:
+ account_id = self.account_id(account_type, i)
+ if account_id not in keys: break
+ i += 1
+ return i
+
+
+ def create_account(self, name, account_type = None):
+ i = self.num_accounts(account_type)
+ account_id = self.account_id(account_type,i)
+
+ if account_type is None:
+ master_c0, master_K0, _ = self.master_public_keys["m/0'/"]
+ c0, K0, cK0 = bip32_public_derivation(master_c0.decode('hex'), master_K0.decode('hex'), "m/0'/", "m/0'/%d"%i)
+ account = BIP32_Account({ 'c':c0, 'K':K0, 'cK':cK0 })
+
+ elif account_type == '2of2':
+ master_c1, master_K1, _ = self.master_public_keys["m/1'/"]
+ c1, K1, cK1 = bip32_public_derivation(master_c1.decode('hex'), master_K1.decode('hex'), "m/1'/", "m/1'/%d"%i)
+ master_c2, master_K2, _ = self.master_public_keys["m/2'/"]
+ c2, K2, cK2 = bip32_public_derivation(master_c2.decode('hex'), master_K2.decode('hex'), "m/2'/", "m/2'/%d"%i)
+ account = BIP32_Account_2of2({ 'c':c1, 'K':K1, 'cK':cK1, 'c2':c2, 'K2':K2, 'cK2':cK2 })
+
+ elif account_type == '2of3':
+ master_c3, master_K3, _ = self.master_public_keys["m/3'/"]
+ c3, K3, cK3 = bip32_public_derivation(master_c3.decode('hex'), master_K3.decode('hex'), "m/3'/", "m/3'/%d"%i)
+ master_c4, master_K4, _ = self.master_public_keys["m/4'/"]
+ c4, K4, cK4 = bip32_public_derivation(master_c4.decode('hex'), master_K4.decode('hex'), "m/4'/", "m/4'/%d"%i)
+ master_c5, master_K5, _ = self.master_public_keys["m/5'/"]
+ c5, K5, cK5 = bip32_public_derivation(master_c5.decode('hex'), master_K5.decode('hex'), "m/5'/", "m/5'/%d"%i)
+ account = BIP32_Account_2of3({ 'c':c3, 'K':K3, 'cK':cK3, 'c2':c4, 'K2':K4, 'cK2':cK4, 'c3':c5, 'K3':K5, 'cK3':cK5 })
+
+ self.accounts[account_id] = account
+ self.save_accounts()
+ self.labels[account_id] = name
+ self.config.set_key('labels', self.labels, True)
+
+
+ def save_accounts(self):
+ d = {}
+ for k, v in self.accounts.items():
+ d[k] = v.dump()
+ self.config.set_key('accounts', d, True)
+
+
+ def load_accounts(self, config):
+ d = config.get('accounts', {})
+ self.accounts = {}
+ for k, v in d.items():
+ if '&' in k:
+ self.accounts[k] = BIP32_Account_2of2(v)
+ else:
+ self.accounts[k] = BIP32_Account(v)
- def init_sequence(self, mpk):
- self.config.set_key('master_public_key', mpk, True)
- self.sequences[0] = self.SequenceClass(mpk)
- self.accounts[0] = { 0:[], 1:[], 'name':'Main account' }
- self.config.set_key('accounts', self.accounts, True)
def addresses(self, include_change = True):
return s[0] == 1
def get_master_public_key(self):
+ raise
return self.config.get("master_public_key")
+ def get_master_private_key(self, account, password):
+ master_k = pw_decode( self.master_private_keys[account], password)
+ master_c, master_K, master_Kc = self.master_public_keys[account]
+ try:
+ K, Kc = get_pubkeys_from_secret(master_k.decode('hex'))
+ assert K.encode('hex') == master_K
+ except:
+ raise BaseException("Invalid password")
+ return master_k
+
+
def get_address_index(self, address):
if address in self.imported_keys.keys():
return -1, None
for account in self.accounts.keys():
for for_change in [0,1]:
- addresses = self.accounts[account][for_change]
+ addresses = self.accounts[account].get_addresses(for_change)
for addr in addresses:
if address == addr:
return account, (for_change, addresses.index(addr))
def get_public_key(self, address):
account, sequence = self.get_address_index(address)
- return self.sequences[account].get_pubkey( sequence )
+ return self.accounts[account].get_pubkey( *sequence )
def decode_seed(self, password):
seed = pw_decode(self.seed, password)
- self.sequences[0].check_seed(seed)
+ #todo: #self.sequences[0].check_seed(seed)
return seed
+
def get_private_key(self, address, password):
- return self.get_private_keys([address], password).get(address)
+ out = []
+ if address in self.imported_keys.keys():
+ out.append( pw_decode( self.imported_keys[address], password ) )
+ else:
+ account, sequence = self.get_address_index(address)
+ # assert address == self.accounts[account].get_address(*sequence)
+ l = account.split("&")
+ for s in l:
+ s = s.strip()
+ m = re.match("(m/\d+'/)(\d+)", s)
+ if m:
+ root = m.group(1)
+ if root not in self.master_private_keys.keys(): continue
+ num = int(m.group(2))
+ master_k = self.get_master_private_key(root, password)
+ master_c, _, _ = self.master_public_keys[root]
+ pk = bip32_private_key( (num,) + sequence, master_k.decode('hex'), master_c.decode('hex'))
+ out.append(pk)
+
+ return out
- def get_private_keys(self, addresses, password):
- if not self.seed: return {}
- # decode seed in any case, in order to test the password
- seed = self.decode_seed(password)
- out = {}
- l_sequences = []
- l_addresses = []
- for address in addresses:
- if address in self.imported_keys.keys():
- out[address] = pw_decode( self.imported_keys[address], password )
- else:
- account, sequence = self.get_address_index(address)
- if account == 0:
- l_sequences.append(sequence)
- l_addresses.append(address)
- pk = self.sequences[0].get_private_keys(l_sequences, seed)
- for i, address in enumerate(l_addresses): out[address] = pk[i]
- return out
def signrawtransaction(self, tx, input_info, private_keys, password):
+ import deserialize
unspent_coins = self.get_unspent_coins()
seed = self.decode_seed(password)
- # convert private_keys to dict
- pk = {}
+ # build a list of public/private keys
+ keypairs = {}
for sec in private_keys:
- address = address_from_private_key(sec)
- pk[address] = sec
- private_keys = pk
+ pubkey = public_key_from_private_key(sec)
+ keypairs[ pubkey ] = sec
+
for txin in tx.inputs:
# convert to own format
# if neither, we might want to get it from the server..
raise
- # find the address:
+ # find the address and fill private_keys
if txin.get('KeyID'):
account, name, sequence = txin.get('KeyID')
- if name != 'Electrum': continue
- sec = self.sequences[account].get_private_key(sequence, seed)
- addr = self.sequences[account].get_address(sequence)
+ if name != 'BIP32': continue
+ sec = self.accounts[account].get_private_key(sequence, seed)
+ pubkey = self.accounts[account].get_pubkey(sequence)
txin['address'] = addr
- private_keys[addr] = sec
+ keypairs[pubkey] = [sec]
- elif txin.get("redeemScript"):
- txin['address'] = hash_160_to_bc_address(hash_160(txin.get("redeemScript").decode('hex')), 5)
+ redeem_script = txin.get("redeemScript")
+ if redeem_script:
+ num, redeem_pubkeys = deserialize.parse_redeemScript(redeem_script)
+ addr = hash_160_to_bc_address(hash_160(redeem_script.decode('hex')), 5)
+ txin['address'] = addr
elif txin.get("raw_output_script"):
- import deserialize
addr = deserialize.get_address_from_output_script(txin.get("raw_output_script").decode('hex'))
sec = self.get_private_key(addr, password)
+ pubkey = public_key_from_private_key(sec)
if sec:
- private_keys[addr] = sec
+ keypairs[pubkey] = [sec]
txin['address'] = addr
- tx.sign( private_keys )
+ tx.sign( keypairs )
def sign_message(self, address, message, password):
sec = self.get_private_key(address, password)
print_error("Verification error: {0}".format(e))
return False
- def create_new_address(self, account, for_change):
- addresses = self.accounts[account][for_change]
- n = len(addresses)
- address = self.get_new_address( account, for_change, n)
- self.accounts[account][for_change].append(address)
- self.history[address] = []
- print_msg(address)
- return address
-
-
- def get_new_address(self, account, for_change, n):
- return self.sequences[account].get_address((for_change, n))
- print address
- return address
def change_gap_limit(self, value):
if value >= self.gap_limit:
self.gap_limit = value
- self.save()
+ self.config.set_key('gap_limit', self.gap_limit, True)
self.interface.poke('synchronizer')
return True
self.accounts[key][0] = addresses
self.gap_limit = value
- self.save()
+ self.config.set_key('gap_limit', self.gap_limit, True)
+ self.save_accounts()
return True
else:
return False
nmax = 0
for account in self.accounts.values():
- addresses = account[0]
+ addresses = account.get_addresses(0)
k = self.num_unused_trailing_addresses(addresses)
for a in addresses[0:-k]:
if self.history.get(a):
def synchronize_sequence(self, account, for_change):
limit = self.gap_limit_for_change if for_change else self.gap_limit
- addresses = self.accounts[account][for_change]
new_addresses = []
while True:
+ addresses = account.get_addresses(for_change)
if len(addresses) < limit:
- new_addresses.append( self.create_new_address(account, for_change) )
+ address = account.create_new_address(for_change)
+ self.history[address] = []
+ new_addresses.append( address )
continue
+
if map( lambda a: self.address_is_old(a), addresses[-limit:] ) == limit*[False]:
break
else:
- new_addresses.append( self.create_new_address(account, for_change) )
+ address = account.create_new_address(for_change)
+ self.history[address] = []
+ new_addresses.append( address )
+
return new_addresses
def synchronize(self):
new = []
- for account in self.accounts.keys():
+ for account in self.accounts.values():
new += self.synchronize_account(account)
+ if new:
+ self.save_accounts()
+ self.config.set_key('addr_history', self.history, True)
return new
return self.history.values() != [[]] * len(self.history)
+ def add_contact(self, address, label=None):
+ self.addressbook.append(address)
+ self.config.set_key('contacts', self.addressbook, True)
+ if label:
+ self.labels[address] = label
+ self.config.set_key('labels', self.labels, True)
+
+ def delete_contact(self, addr):
+ if addr in self.addressbook:
+ self.addressbook.remove(addr)
+ self.config.set_key('addressbook', self.addressbook, True)
+
+
def fill_addressbook(self):
for tx_hash, tx in self.transactions.items():
is_relevant, is_send, _, _ = self.get_tx_value(tx)
def get_accounts(self):
accounts = {}
for k, account in self.accounts.items():
- accounts[k] = account.get('name')
+ accounts[k] = self.labels.get(k, 'unnamed')
if self.imported_keys:
accounts[-1] = 'Imported keys'
return accounts
o = self.imported_keys.keys()
else:
ac = self.accounts[a]
- o = ac[0][:]
- if include_change: o += ac[1]
+ o = ac.get_addresses(0)
+ if include_change: o += ac.get_addresses(1)
return o
def get_imported_balance(self):
if not self.use_change or account == -1:
change_addr = inputs[-1]['address']
else:
- change_addr = self.accounts[account][1][-self.gap_limit_for_change]
+ change_addr = self.accounts[account].get_addresses(1)[-self.gap_limit_for_change]
# Insert the change output at a random position in the outputs
posn = random.randint(0, len(outputs))
def receive_tx_callback(self, tx_hash, tx, tx_height):
-
if not self.check_new_tx(tx_hash, tx):
# may happen due to pruning
print_error("received transaction that is no longer referenced in history", tx_hash)
with self.transaction_lock:
self.transactions[tx_hash] = tx
+
+ self.interface.pending_transactions_for_notifications.append(tx)
+
+ self.save_transactions()
if self.verifier and tx_height>0:
self.verifier.add(tx_hash, tx_height)
self.update_tx_outputs(tx_hash)
- self.save()
+ def save_transactions(self):
+ tx = {}
+ for k,v in self.transactions.items():
+ tx[k] = str(v)
+ self.config.set_key('transactions', tx, True)
def receive_history_callback(self, addr, hist):
with self.lock:
self.history[addr] = hist
- self.save()
+ self.config.set_key('addr_history', self.history, True)
if hist != ['*']:
for tx_hash, tx_height in hist:
tx = Transaction.from_io(inputs, outputs)
- pk_addresses = []
- for i in range(len(tx.inputs)):
- txin = tx.inputs[i]
+
+ keypairs = {}
+ for i, txin in enumerate(tx.inputs):
address = txin['address']
if address in self.imported_keys.keys():
pk_addresses.append(address)
continue
account, sequence = self.get_address_index(address)
- txin['KeyID'] = (account, 'Electrum', sequence) # used by the server to find the key
- pk_addr, redeemScript = self.sequences[account].get_input_info(sequence)
- if redeemScript: txin['redeemScript'] = redeemScript
- pk_addresses.append(pk_addr)
+ txin['KeyID'] = (account, 'BIP32', sequence) # used by the server to find the key
+
+ redeemScript = self.accounts[account].redeem_script(sequence)
+ if redeemScript:
+ txin['redeemScript'] = redeemScript
+ assert address == self.accounts[account].get_address(*sequence)
+ else:
+ txin['redeemPubkey'] = self.accounts[account].get_pubkey(*sequence)
+
+ private_keys = self.get_private_key(address, password)
+ for sec in private_keys:
+ pubkey = public_key_from_private_key(sec)
+ keypairs[ pubkey ] = sec
- # get all private keys at once.
- if self.seed:
- private_keys = self.get_private_keys(pk_addresses, password)
- tx.sign(private_keys)
+ tx.sign(keypairs)
for address, x in outputs:
if address not in self.addressbook and not self.is_mine(address):
self.imported_keys[k] = c
self.config.set_key('imported_keys', self.imported_keys, True)
+ for k, v in self.master_private_keys.items():
+ b = pw_decode(v, old_password)
+ c = pw_encode(b, new_password)
+ self.master_private_keys[k] = c
+ self.config.set_key('master_private_keys', self.master_private_keys, True)
+
def freeze(self,addr):
if self.is_mine(addr) and addr not in self.frozen_addresses:
else:
return False
+ def set_fee(self, fee):
+ if self.fee != fee:
+ self.fee = fee
+ self.config.set_key('fee_per_kb', self.fee, True)
+
+
def save(self):
+ print_error("Warning: wallet.save() is deprecated")
tx = {}
for k,v in self.transactions.items():
tx[k] = str(v)
s = {
'use_change': self.use_change,
'fee_per_kb': self.fee,
- 'accounts': self.accounts,
'addr_history': self.history,
'labels': self.labels,
'contacts': self.addressbook,
threading.Thread.__init__(self)
self.daemon = True
self.wallet = wallet
+ wallet.synchronizer = self
self.interface = self.wallet.interface
self.interface.register_channel('synchronizer')
self.wallet.interface.register_callback('connected', lambda: self.wallet.set_up_to_date(False))
if self.was_updated and not requested_tx:
self.interface.trigger_callback('updated')
- self.was_updated = False
-
+ self.interface.trigger_callback("new_transaction") # Updated gets called too many times from other places as well; if we use that signal we get the notification three times
+
+ self.was_updated = False