upgrade.py

   1 import electrum, base64, ast, sys, os
   2 from version import SEED_VERSION
   3
   4 try:
   5     from lib import prompt_password
   6     from lib.util import print_error
   7 except ImportError:
   8     from electrum import prompt_password
   9     from electrum.util import print_error
  10
  11 if __name__ == "__main__":
  12     try:
  13         path = sys.argv[1]
  14     except:
  15         # backward compatibility: look for wallet file in the default data directory
  16         if "HOME" in os.environ:
  17             wallet_dir = os.path.join( os.environ["HOME"], '.electrum')
  18         elif "LOCALAPPDATA" in os.environ:
  19             wallet_dir = os.path.join( os.environ["LOCALAPPDATA"], 'Electrum' )
  20         elif "APPDATA" in os.environ:
  21             wallet_dir = os.path.join( os.environ["APPDATA"],  'Electrum' )
  22         else:
  23             raise BaseException("No home directory found in environment variables.")
  24         path = os.path.join( wallet_dir, 'electrum.dat')
  25
  26     try:
  27         f = open(path,"r")
  28         data = f.read()
  29         f.close()
  30     except:
  31         print_error("Error: File not found: " + path)
  32         exit(1)
  33
  34     try:
  35         x = ast.literal_eval(data)
  36     except:
  37         print_error("Error: Could not parse wallet")
  38         exit(1)
  39
  40     # version <= 0.33 uses a tuple
  41     if type(x) == tuple:
  42         seed_version, use_encryption, fee, host, port, blocks, seed, all_addresses, private_keys, change_indexes, status, history, labels, addressbook = x
  43
  44         print "walet path =",path
  45         print "seed version =", seed_version
  46
  47         if seed_version == 1 and use_encryption:
  48             # version 1 used pycrypto for wallet encryption
  49             import Crypto
  50             from Crypto.Cipher import AES
  51             BLOCK_SIZE = 32
  52             PADDING = '{'
  53             pad = lambda s: s + (BLOCK_SIZE - len(s) % BLOCK_SIZE) * PADDING
  54             EncodeAES = lambda secret, s: base64.b64encode(AES.new(secret).encrypt(pad(s)))
  55             DecodeAES = lambda secret, e: AES.new(secret).decrypt(base64.b64decode(e)).rstrip(PADDING)
  56
  57             print "Please enter your password"
  58             password = prompt_password("Password:")
  59             secret = electrum.Hash(password)
  60             try:
  61                 seed = DecodeAES( secret, wallet.seed )
  62                 private_keys = ast.literal_eval( DecodeAES( secret, wallet.private_keys ) )
  63             except:
  64                 print_error("Error: Password does not decrypt this wallet.")
  65                 exit(1)
  66             seed_version = 2
  67             s = repr( (seed_version, use_encryption, fee, host, port, blocks, seed, all_addresses, private_keys, change_indexes, status, history, labels, addressbook ))
  68             f = open(path,"w")
  69             data = f.read()
  70             f.close()
  71             print "Wallet is now unencrypted."
  72
  73         print """This wallet is deprecated.
  74 Please create a new wallet, open the old wallet with Electrum 0.33, and send your coins to your new wallet.
  75 We apologize for the inconvenience. We try to keep this kind of upgrades as rare as possible."""
  76
  77
  78
  79