You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
36 lines
1.2 KiB
36 lines
1.2 KiB
import electrum, getpass, base64,ast,sys
|
|
|
|
|
|
try:
|
|
path = sys.argv[1]
|
|
except:
|
|
path = None
|
|
wallet = electrum.Wallet(path)
|
|
try:
|
|
wallet.read()
|
|
print "ok"
|
|
except BaseException:
|
|
if wallet.version == 1 and wallet.use_encryption:
|
|
# version 1 used pycrypto for wallet encryption
|
|
import Crypto
|
|
from Crypto.Cipher import AES
|
|
BLOCK_SIZE = 32
|
|
PADDING = '{'
|
|
pad = lambda s: s + (BLOCK_SIZE - len(s) % BLOCK_SIZE) * PADDING
|
|
EncodeAES = lambda secret, s: base64.b64encode(AES.new(secret).encrypt(pad(s)))
|
|
DecodeAES = lambda secret, e: AES.new(secret).decrypt(base64.b64decode(e)).rstrip(PADDING)
|
|
|
|
print "please enter your password"
|
|
password = getpass.getpass("Password:")
|
|
secret = electrum.Hash(password)
|
|
try:
|
|
seed = DecodeAES( secret, wallet.seed )
|
|
private_keys = ast.literal_eval( DecodeAES( secret, wallet.private_keys ) )
|
|
except:
|
|
print "sorry"
|
|
exit(1)
|
|
wallet.version = 2
|
|
wallet.seed = wallet.pw_encode( seed, password)
|
|
wallet.private_keys = wallet.pw_encode( repr( private_keys ), password)
|
|
wallet.save()
|
|
print "upgrade successful"
|
|
|