diff options
| -rw-r--r-- | .editorconfig | 4 | ||||
| -rw-r--r-- | static/solver.py | 28 |
2 files changed, 20 insertions, 12 deletions
diff --git a/.editorconfig b/.editorconfig index 093c47b..b51a366 100644 --- a/.editorconfig +++ b/.editorconfig @@ -17,3 +17,7 @@ indent_size = 2 [*.html] tab_size = 2 indent_size = 2 + +[*.py] +tab_size = 4 +indent_size = 4 diff --git a/static/solver.py b/static/solver.py index 83841fb..d067f8b 100644 --- a/static/solver.py +++ b/static/solver.py @@ -7,23 +7,27 @@ import sys import struct def validate_bit_zeros(bs: bytes, n: int) -> bool: - q, r = divmod(n, 8) - if any(b != 0 for b in bs[:q]): - return False - if r and (bs[q] & (0xFF << (8 - r))): - return False - return True + q, r = divmod(n, 8) + if any(b != 0 for b in bs[:q]): + return False + if r and (bs[q] & (0xFF << (8 - r))): + return False + return True + +if len(sys.argv) < 3: + print(f"usage: {sys.argv[0]} <base64 identifier> <difficulty>", file=sys.stderr) + sys.exit(1) decoded = base64.b64decode(sys.argv[1]) difficulty = int(sys.argv[2]) next_val = 0 while True: - h = hashlib.sha256(decoded + struct.pack("Q", next_val)).digest() - if validate_bit_zeros(h, difficulty): - break - next_val = (next_val + 1) & 0xFFFFFFFFFFFFFFFF - if next_val == 0: - raise ValueError("overflow") + h = hashlib.sha256(decoded + struct.pack("Q", next_val)).digest() + if validate_bit_zeros(h, difficulty): + break + next_val = (next_val + 1) & 0xFFFFFFFFFFFFFFFF + if next_val == 0: + raise ValueError("overflow") print(base64.b64encode(struct.pack("Q", next_val)).decode()) |