mirror of
https://github.com/Zxilly/UA2F.git
synced 2025-12-26 21:18:06 +00:00
98 lines
2.4 KiB
Python
98 lines
2.4 KiB
Python
import atexit
|
|
import http.server
|
|
import os
|
|
import socketserver
|
|
import subprocess
|
|
import sys
|
|
import threading
|
|
import time
|
|
|
|
import requests
|
|
|
|
PORT = 37491
|
|
|
|
|
|
class MyHandler(http.server.SimpleHTTPRequestHandler):
|
|
def do_GET(self):
|
|
user_agent = self.headers.get('User-Agent')
|
|
|
|
# assert user_agent only contains F
|
|
if not all([c == 'F' for c in user_agent]):
|
|
print("UA2F does not work!")
|
|
exit(1)
|
|
else:
|
|
print("Got a full F user agent with length", len(user_agent))
|
|
|
|
self.send_response(200)
|
|
self.end_headers()
|
|
self.wfile.write(b"Hello, world!")
|
|
|
|
|
|
def start_server():
|
|
with socketserver.TCPServer(("", PORT), MyHandler, True) as httpd:
|
|
httpd.serve_forever()
|
|
atexit.register(httpd.shutdown)
|
|
|
|
|
|
def start_ua2f(u: str):
|
|
p = subprocess.Popen([u])
|
|
atexit.register(lambda: p.kill())
|
|
|
|
|
|
# iptables 设置函数
|
|
def setup_iptables():
|
|
os.system(f"sudo iptables -A OUTPUT -p tcp --dport {PORT} -j NFQUEUE --queue-num 10010")
|
|
|
|
|
|
def cleanup_iptables():
|
|
os.system(f"sudo iptables -D OUTPUT -p tcp --dport {PORT} -j NFQUEUE --queue-num 10010")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if os.name != 'posix':
|
|
raise Exception("This script only supports Linux")
|
|
|
|
if os.geteuid() != 0:
|
|
raise Exception("This script requires root privileges")
|
|
|
|
ua2f = sys.argv[1]
|
|
|
|
setup_iptables()
|
|
|
|
server_thread = threading.Thread(target=start_server)
|
|
server_thread.daemon = True
|
|
server_thread.start()
|
|
|
|
print(f"Starting server on port {PORT}")
|
|
|
|
ua2f_thread = threading.Thread(target=start_ua2f, args=(ua2f,))
|
|
ua2f_thread.daemon = True
|
|
ua2f_thread.start()
|
|
|
|
print(f"Starting UA2F: {ua2f}")
|
|
|
|
time.sleep(2)
|
|
|
|
normal_ua = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) "
|
|
"Chrome/126.0.0.0 Safari/537.36 Edg/126.0.0.0"
|
|
|
|
response = requests.get(f"http://localhost:{PORT}", headers={
|
|
"User-Agent": normal_ua
|
|
})
|
|
assert response.ok
|
|
print("Tested with a normal user agent with length", len(normal_ua))
|
|
|
|
time.sleep(1)
|
|
|
|
random_ua = "Some random user agent"
|
|
response = requests.get(f"http://localhost:{PORT}", headers={
|
|
"User-Agent": random_ua
|
|
})
|
|
assert response.ok
|
|
print("Tested with a random user agent with length", len(random_ua))
|
|
|
|
time.sleep(1) # wait for process
|
|
|
|
# clean
|
|
cleanup_iptables()
|