LocalReverse/demo.py at main · FatFatYoung/LocalReverse · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
# -*- coding: utf-8 -*-
"""
本地代理演示版 (Demo)
功能：将本地 80 端口请求直接转发到 https://github.com/FatFatYoung
用途：演示如何无需修改浏览器地址栏即可通过 localhost 访问特定网站
注意：必须使用管理员权限运行（因为占用 80 端口）
"""
from http.server import HTTPServer, BaseHTTPRequestHandler
import requests
import sys
import ctypes

# --- 核心配置 ---
TARGET_URL = 'https://github.com/FatFatYoung'
LISTEN_PORT = 80
SECRET_HEADER = {'X-Local-Proxy': 'DemoMode'}
# ----------------

class ProxyHandler(BaseHTTPRequestHandler):
    def do_GET(self):
        # 1. 构建目标 URL
        target = f"{TARGET_URL}{self.path}"
        try:
            # 2. 带着暗号头转发请求
            resp = requests.get(target, headers=SECRET_HEADER, allow_redirects=False, timeout=30)

            # 3. 将服务器的响应返回给浏览器
            self.send_response(resp.status_code)
            for name, value in resp.headers.items():
                if name.lower() not in ['content-encoding', 'transfer-encoding', 'content-length']:
                    self.send_header(name, value)
            self.end_headers()
            self.wfile.write(resp.content)
        except Exception as e:
            self.send_response(500)
            self.send_header('Content-Type', 'text/html; charset=utf-8')
            self.end_headers()
            error_html = f"""
            <!DOCTYPE html>
            <html><head><meta charset="utf-8"><title>代理演示错误</title></head>
            <body style="font-family:Microsoft YaHei;padding:50px;">
            <h1 style="color:red;">代理错误</h1>
            <p>请求地址：{target}</p>
            <p>错误信息：{str(e)}</p>
            </body></html>
            """.encode('utf-8')
            self.wfile.write(error_html)

    def do_POST(self):
        self.do_GET()

    def do_HEAD(self):
        self.do_GET()

    def log_message(self, format, *args):
        # 在控制台输出日志
        print(f"[{self.log_date_time_string()}] {format % args}")
        sys.stdout.flush()

def is_admin():
    """检查是否以管理员权限运行"""
    try:
        return ctypes.windll.shell32.IsUserAnAdmin()
    except:
        return False

def main():
    print("=" * 60)
    print("本地代理演示版 (Demo) -> 80 端口")
    print("=" * 60)
    print()
    print(f"目标网站：{TARGET_URL}")
    print(f"本地地址：http://127.0.0.1 (端口 80)")
    print()

    # 检查管理员权限
    if not is_admin():
        print("[错误] 需要使用管理员权限运行（80 端口需要管理员权限）")
        print()
        print("解决方法：")
        print("  1. 右键点击本程序")
        print("  2. 选择「以管理员身份运行」")
        print()
        input("按回车键退出...")
        sys.exit(1)

    print("[OK] 管理员权限检查通过")
    print("正在监听 80 端口...")
    print("在浏览器访问 http://127.0.0.1 即可看到 GitHub 主页")
    print()
    print("按 Ctrl+C 停止服务...")
    print("=" * 60)

    try:
        # 绑定 80 端口
        server = HTTPServer(('127.0.0.1', LISTEN_PORT), ProxyHandler)
        server.serve_forever()
    except KeyboardInterrupt:
        print("\n正在停止服务...")
        server.shutdown()
        print("服务已停止")
    except OSError as e:
        print(f"[错误] 无法启动：{e}")
        input("按回车键退出...")

if __name__ == '__main__':
    main()