gitea-mirror/hate_crack
1
0
Fork 0
mirror of https://github.com/trustedsec/hate_crack.git synced 2026-07-28 14:47:22 -07:00
Code Issues Packages Projects Releases Wiki Activity

fix(api): skip final sleep in backoff loop, fold probe into stream, guard resp.close

Browse Source 
Fix 1: _with_hashmob_backoff no longer sleeps after the last attempt —
breaks out of the loop immediately so we don't waste up to 300s before
giving up.

Fix 2: download_hashmob_wordlist and download_hashmob_rule now open a
single streaming connection inside _attempt() and pass the already-open
response to _stream_response_to_file, eliminating the probe-then-reopen
pattern that caused 429 responses on the second request to bypass the
backoff retry machinery.

Fix 3: HashviewAPI.download_wordlist wraps _stream_response_to_file in
try/finally so resp.close() is guaranteed even on KeyboardInterrupt.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
Justin Bollinger
2026-04-24 23:35:27 -04:00
co-authored by Claude Sonnet 4.6
parent d11194cf6e
commit 11b575d5a6
1 changed files with 16 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hate_crack/api.py
+16 -22
View File
@@ -153,9 +153,10 @@ def _with_hashmob_backoff(
penalty = base_delay
for attempt in range(max_attempts):
try:
result = fn()
return result
return fn()
except _Hashmob429:
if attempt == max_attempts - 1:
break
print(f"[!] Rate limit hit (429). Backing off for {penalty} seconds...")
time.sleep(penalty)
penalty = min(penalty + step, max_delay)
@@ -1182,8 +1183,10 @@ class HashviewAPI:
output_file = os.path.join(dest_dir, output_file)
os.makedirs(os.path.dirname(output_file), exist_ok=True)
ok = _stream_response_to_file(resp, output_file, label=output_file)
resp.close()
try:
ok = _stream_response_to_file(resp, output_file, label=output_file)
finally:
resp.close()
if ok:
return {"output_file": output_file, "size": os.path.getsize(output_file)}
return {"output_file": output_file, "size": 0}
@@ -1452,9 +1455,7 @@ def download_hashmob_wordlist(file_name, out_path):
def _attempt():
_hashmob_limiter.wait()
with requests.get(
url, headers=headers, stream=True, timeout=60, allow_redirects=True
) as r:
with requests.get(url, headers=headers, stream=True, timeout=60, allow_redirects=True) as r:
if r.status_code == 429:
raise _Hashmob429()
r.raise_for_status()
@@ -1472,8 +1473,7 @@ def download_hashmob_wordlist(file_name, out_path):
return _streamed_download(real_url, out_path, label=file_name)
print("Error: Received HTML instead of file. Possible permission or quota issue.")
return False
# Normal binary download — re-request now that we've confirmed the URL is good
return _streamed_download(url, out_path, headers=headers, label=file_name)
return _stream_response_to_file(r, out_path, label=file_name)
try:
return _with_hashmob_backoff(_attempt)
@@ -1588,24 +1588,18 @@ def download_hashmob_rule(file_name, out_path):
def _attempt():
_hashmob_limiter.wait()
with requests.get(
primary_url, headers=headers, stream=True, timeout=60, allow_redirects=True
) as r:
with requests.get(primary_url, headers=headers, stream=True, timeout=60, allow_redirects=True) as r:
if r.status_code == 429:
raise _Hashmob429()
if r.status_code == 404 and alt_url:
print(
f"[i] Hashmob rule not found at primary URL, trying fallback: {alt_url}"
)
with requests.get(
alt_url, headers=headers, stream=True, timeout=60, allow_redirects=True
) as r_alt:
if r_alt.status_code == 429:
print(f"[i] Hashmob rule not found at primary URL, trying fallback: {alt_url}")
with requests.get(alt_url, headers=headers, stream=True, timeout=60, allow_redirects=True) as r2:
if r2.status_code == 429:
raise _Hashmob429()
r_alt.raise_for_status()
return _streamed_download(alt_url, out_path, headers=headers, label=file_name)
r2.raise_for_status()
return _stream_response_to_file(r2, out_path, label=file_name)
r.raise_for_status()
return _streamed_download(primary_url, out_path, headers=headers, label=file_name)
return _stream_response_to_file(r, out_path, label=file_name)
try:
return _with_hashmob_backoff(_attempt)