Add recover wallet
This commit is contained in:
jackjack
parent
a82edee764
commit
3f22df2577
219
pywallet.py
219
pywallet.py
@ -53,6 +53,18 @@ aversions[111] = 'Testnet';
|
||||
wallet_dir = ""
|
||||
wallet_name = ""
|
||||
|
||||
ko = 1e3
|
||||
kio = 1024
|
||||
Mo = 1e6
|
||||
Mio = 1024 ** 2
|
||||
Go = 1e9
|
||||
Gio = 1024 ** 3
|
||||
To = 1e12
|
||||
Tio = 1024 ** 4
|
||||
|
||||
prekeys = ["82011302010104".decode('hex'), "82011202010104".decode('hex')]
|
||||
postkeys = ["a081a530".decode('hex'), "81a530".decode('hex')]
|
||||
|
||||
def iais(a):
|
||||
if a>= 2:
|
||||
return 's'
|
||||
@ -483,6 +495,154 @@ def parse_setting(setting, vds):
|
||||
class SerializationError(Exception):
|
||||
""" Thrown when there's a problem deserializing or serializing """
|
||||
|
||||
def ts():
|
||||
return int(time.mktime(datetime.now().timetuple()))
|
||||
|
||||
def check_postkeys(key, postkeys):
|
||||
for i in postkeys:
|
||||
if key[:len(i)] == i:
|
||||
return True
|
||||
return False
|
||||
|
||||
def one_element_in(a, string):
|
||||
for i in a:
|
||||
if i in string:
|
||||
return True
|
||||
return False
|
||||
|
||||
def first_read(device, size, prekeys, inc=10000):
|
||||
t0 = ts()-1
|
||||
try:
|
||||
fd = os.open (device, os.O_RDONLY)
|
||||
except:
|
||||
print("Can't open %s, check the path or try as root"%device)
|
||||
exit(0)
|
||||
prekey = prekeys[0]
|
||||
data = ""
|
||||
i = 0
|
||||
data = os.read (fd, i)
|
||||
before_contained_key = False
|
||||
contains_key = False
|
||||
ranges = []
|
||||
|
||||
while i < int(size):
|
||||
if i%(10*Mio) > 0 and i%(10*Mio) <= inc:
|
||||
print("\n%.2f/%.2f Go"%(i/1e9, size/1e9))
|
||||
t = ts()
|
||||
speed = i/(t-t0)
|
||||
ETAts = size/speed + t0
|
||||
d = datetime.fromtimestamp(ETAts)
|
||||
print(d.strftime(" ETA: %H:%M:%S"))
|
||||
|
||||
data = os.read (fd, inc)
|
||||
contains_key = one_element_in(prekeys, data)
|
||||
|
||||
if not before_contained_key and contains_key:
|
||||
ranges.append(i)
|
||||
|
||||
if before_contained_key and not contains_key:
|
||||
ranges.append(i)
|
||||
|
||||
before_contained_key = contains_key
|
||||
|
||||
i += inc
|
||||
|
||||
os.close (fd)
|
||||
return ranges
|
||||
|
||||
def shrink_intervals(device, ranges, prekeys, inc=1000):
|
||||
prekey = prekeys[0]
|
||||
nranges = []
|
||||
fd = os.open (device, os.O_RDONLY)
|
||||
for j in range(len(ranges)/2):
|
||||
before_contained_key = False
|
||||
contains_key = False
|
||||
bi = ranges[2*j]
|
||||
bf = ranges[2*j+1]
|
||||
|
||||
mini_blocks = []
|
||||
k = bi
|
||||
while k <= bf + len(prekey) + 1:
|
||||
mini_blocks.append(k)
|
||||
k += inc
|
||||
mini_blocks.append(k)
|
||||
|
||||
for k in range(len(mini_blocks)/2):
|
||||
mini_blocks[2*k] -= len(prekey) +1
|
||||
mini_blocks[2*k+1] += len(prekey) +1
|
||||
|
||||
|
||||
bi = mini_blocks[2*k]
|
||||
bf = mini_blocks[2*k+1]
|
||||
|
||||
os.lseek(fd, bi, 0)
|
||||
|
||||
data = os.read(fd, bf-bi+1)
|
||||
contains_key = one_element_in(prekeys, data)
|
||||
|
||||
if not before_contained_key and contains_key:
|
||||
nranges.append(bi)
|
||||
|
||||
if before_contained_key and not contains_key:
|
||||
nranges.append(bi+len(prekey) +1+len(prekey) +1)
|
||||
|
||||
before_contained_key = contains_key
|
||||
|
||||
os.close (fd)
|
||||
|
||||
return nranges
|
||||
|
||||
def find_offsets(device, ranges, prekeys):
|
||||
prekey = prekeys[0]
|
||||
list_offsets = []
|
||||
to_read = 0
|
||||
fd = os.open (device, os.O_RDONLY)
|
||||
for i in range(len(ranges)/2):
|
||||
bi = ranges[2*i]-len(prekey)-1
|
||||
os.lseek(fd, bi, 0)
|
||||
bf = ranges[2*i+1]+len(prekey)+1
|
||||
to_read += bf-bi+1
|
||||
buf = ""
|
||||
for j in range(len(prekey)):
|
||||
buf += "\x00"
|
||||
curs = bi
|
||||
|
||||
while curs <= bf:
|
||||
data = os.read(fd, 1)
|
||||
buf = buf[1:] + data
|
||||
if buf in prekeys:
|
||||
list_offsets.append(curs)
|
||||
curs += 1
|
||||
|
||||
os.close (fd)
|
||||
|
||||
return [to_read, list_offsets]
|
||||
|
||||
def read_keys(device, list_offsets):
|
||||
found_hexkeys = []
|
||||
fd = os.open (device, os.O_RDONLY)
|
||||
for offset in list_offsets:
|
||||
os.lseek(fd, offset+1, 0)
|
||||
data = os.read(fd, 40)
|
||||
hexkey = data[1:33].encode('hex')
|
||||
after_key = data[33:39].encode('hex')
|
||||
if hexkey not in found_hexkeys and check_postkeys(after_key.decode('hex'), postkeys):
|
||||
found_hexkeys.append(hexkey)
|
||||
|
||||
os.close (fd)
|
||||
|
||||
return found_hexkeys
|
||||
|
||||
def read_device_size(size):
|
||||
if size[-2] == 'i':
|
||||
unit = size[-3:]
|
||||
value = float(size[:-3])
|
||||
else:
|
||||
unit = size[-2:]
|
||||
value = float(size[:-2])
|
||||
exec 'unit = %s' % unit
|
||||
return int(value * unit)
|
||||
|
||||
class KEY:
|
||||
|
||||
def __init__ (self):
|
||||
@ -903,6 +1063,23 @@ def update_wallet(db, type, data):
|
||||
print("data dictionary: %r"%data)
|
||||
traceback.print_exc()
|
||||
|
||||
def create_new_wallet(db_env, walletfile, version):
|
||||
db_out = DB(db_env)
|
||||
|
||||
try:
|
||||
r = db_out.open(walletfile, "main", DB_BTREE, DB_CREATE)
|
||||
except DBError:
|
||||
r = True
|
||||
|
||||
if r is not None:
|
||||
logging.error("Couldn't open %s."%walletfile)
|
||||
sys.exit(1)
|
||||
|
||||
db_out.put("0776657273696f6e".decode('hex'), ("%08x"%version).decode('hex')[::-1])
|
||||
|
||||
db_out.close()
|
||||
|
||||
|
||||
def rewrite_wallet(db_env, walletfile, destFileName, pre_put_callback=None):
|
||||
db = open_wallet(db_env, walletfile)
|
||||
|
||||
@ -1675,6 +1852,18 @@ if __name__ == '__main__':
|
||||
parser.add_option("--port", dest="port",
|
||||
help="port of web interface (defaults to 8989)")
|
||||
|
||||
parser.add_option("--recover", dest="recover", action="store_true",
|
||||
help="recover your deleted keys, use with recov_size and recov_device")
|
||||
|
||||
parser.add_option("--recov_device", dest="recov_device",
|
||||
help="device to read (e.g. /dev/sda1)")
|
||||
|
||||
parser.add_option("--recov_size", dest="recov_size",
|
||||
help="number of bytes to read (e.g. 20Mo or 50Gio)")
|
||||
|
||||
parser.add_option("--recov_outputdir", dest="recov_outputdir",
|
||||
help="output directory where the recovered wallet will be put")
|
||||
|
||||
# parser.add_option("--forcerun", dest="forcerun",
|
||||
# action="store_true",
|
||||
# help="run even if pywallet detects bitcoin is running")
|
||||
@ -1700,6 +1889,35 @@ if __name__ == '__main__':
|
||||
# if options.forcerun is None:
|
||||
# exit(0)
|
||||
|
||||
if options.recover:
|
||||
if options.recov_size is None or options.recov_device is None or options.recov_outputdir is None:
|
||||
print("You must provide the device, the number of bytes to read and the output directory")
|
||||
exit(0)
|
||||
device = options.recov_device
|
||||
size = read_device_size(options.recov_size)
|
||||
r = first_read(device, size, prekeys, 10000)
|
||||
r = shrink_intervals(device, r, prekeys, 1000)
|
||||
[to_read, o] = find_offsets(device, r, prekeys)
|
||||
print("%dko to read"%(to_read/1000))
|
||||
keys = read_keys(device, o)
|
||||
print("%d key%s found"%(len(keys), iais(len(keys))))
|
||||
|
||||
db_env = create_env(options.recov_outputdir)
|
||||
recov_wallet_name = "recovered_wallet_%s.dat"%ts()
|
||||
create_new_wallet(db_env, recov_wallet_name, 32500)
|
||||
|
||||
db = open_wallet(db_env, recov_wallet_name, True)
|
||||
i = 0
|
||||
for sec in keys:
|
||||
print("\nImporting key %4d/%d:"%(i+1, len(keys)))
|
||||
importprivkey(db, sec, "recovered: %s"%sec, None, True)
|
||||
i += 1
|
||||
db.close()
|
||||
|
||||
print("\n\nThe new wallet %s/%s contains the %d recovered key%s\n"%(options.recov_outputdir, recov_wallet_name, len(keys), iais(len(keys))))
|
||||
|
||||
exit(0)
|
||||
|
||||
|
||||
if 'bsddb' in missing_dep:
|
||||
print("pywallet needs 'bsddb' package to run, please install it")
|
||||
@ -1787,3 +2005,4 @@ if __name__ == '__main__':
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user