From 2356abd1b43b8aafcc576fdcd544720d9ed5ad17 Mon Sep 17 00:00:00 2001 From: The MMGen Project Date: Sun, 17 May 2026 18:03:37 +0000 Subject: [PATCH] tool_cmd.find_incog_data(): use `itertools.count()` in loop --- mmgen/tool/fileutil.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/mmgen/tool/fileutil.py b/mmgen/tool/fileutil.py index 14ea2187..6330af93 100755 --- a/mmgen/tool/fileutil.py +++ b/mmgen/tool/fileutil.py @@ -38,15 +38,18 @@ class tool_cmd(tool_cmd_base): "Use an Incog ID to find hidden incognito wallet data" from hashlib import sha256 + from itertools import count ivsize, bsize, mod = (Crypto.aesctr_iv_len, 4096, 4096*8) - n, carry = 0, b' '*ivsize + carry = b' ' * ivsize flgs = os.O_RDONLY|os.O_BINARY if gc.platform == 'win32' else os.O_RDONLY f = os.open(filename, flgs) + for ch in incog_id: if ch not in '0123456789ABCDEF': die(2, f'{incog_id!r}: invalid Incog ID') - while True: + + for n in count(step=bsize): d = os.read(f, bsize) if not d: break @@ -60,7 +63,6 @@ class tool_cmd(tool_cmd_base): import sys sys.exit(0) carry = d[len(d)-ivsize:] - n += bsize if not n % mod: msg_r(f'\rSearched: {n} bytes')