initial commit

test_503_2.py

@@ -0,0 +1,61 @@
+import urllib.request, time, hashlib, re, os, subprocess
+from curl_cffi import requests
+
+def solve_challenge(html):
+    m = re.search(r"\['([A-F0-9]{40})'", html)
+    if not m: return None, None
+    c = m.group(1)
+    n1 = int(c[0], 16)
+    i = 0
+    t0 = time.time()
+    while True:
+        d = hashlib.sha1((c + str(i)).encode()).digest()
+        if d[n1] == 0xb0 and d[n1+1] == 0x0b: break
+        i += 1
+    return c + str(i), f"{time.time() - t0:.3f}"
+
+yaml_config = """
+mode: global
+mixed-port: 10899
+bind-address: '127.0.0.1'
+proxies:
+"""
+proxy_str = ""
+with open("proxies.txt") as f:
+    for line in f:
+        if "ss," in line:
+            proxy_str = line.strip()
+            break
+parts = proxy_str.split("=", 1)
+cfg = parts[1].strip().split(",")
+params = dict(p.strip().split("=", 1) for p in cfg[3:] if "=" in p)
+yaml_config += f"""  - name: "test"
+    type: ss
+    server: "{cfg[1].strip()}"
+    port: {cfg[2].strip()}
+    cipher: "{params.get('encrypt-method', '')}"
+    password: "{params.get('password', '')}"
+"""
+if "obfs" in params:
+    yaml_config += f"    plugin: obfs\n    plugin-opts:\n      mode: {params['obfs']}\n      host: {params['obfs-host']}\n"
+with open("test_503.yaml", "w") as f: f.write(yaml_config)
+
+proc = subprocess.Popen(["./.venv/bin/mihomo", "-f", "test_503.yaml"], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+time.sleep(2)
+
+s = requests.Session()
+p = {"http": "socks5://127.0.0.1:10899", "https": "socks5://127.0.0.1:10899"}
+try:
+    r1 = s.get("https://zlib.li/", proxies=p, impersonate="chrome", timeout=10)
+    print("Initial:", r1.status_code)
+    
+    if r1.status_code == 503 and "Checking your browser" in r1.text:
+        ct, cti = solve_challenge(r1.text)
+        s.cookies.set("c_token", ct, domain="zlib.li", path="/")
+        s.cookies.set("c_time", cti, domain="zlib.li", path="/")
+        r1 = s.get("https://zlib.li/", proxies=p, impersonate="chrome", timeout=10)
+        print("After PoW:", r1.status_code)
+        if r1.status_code == 503:
+            print(r1.text[:500])
+finally:
+    proc.terminate()