1
const http = require('http');
2
const httpProxy = require('http-proxy');
3
const tunnel = require('tunnel');
4
const { SocksProxyAgent } = require('socks');
5

6
// -------------------------- 配置区 --------------------------
7
const LISTEN_PORT = 3000;
8
const PROXY_POOL = [
9
  'socks5://user1:pass1@proxy1.example.com:1080',
10
  'https://user2:pass2@proxy2.example.com:443',
11
  'socks5://proxy3.example.com:1080',
12
  'https://proxy4.example.com:8443'
13
];
14
let currentIndex = 0;
15
const MAX_RETRY = PROXY_POOL.length;
16
const REQUEST_TIMEOUT = 800 * 1000; // 超时时间设置为 800 秒（转为毫秒）
17
// -------------------------- 配置区结束 --------------------------
18

19
const proxy = httpProxy.createProxyServer({});
20

21
function getAgentByIndex(index) {
22
  const proxyUrl = PROXY_POOL[index];
23
  const urlObj = new URL(proxyUrl);
24

25
  if (urlObj.protocol === 'socks5:') {
26
    return new SocksProxyAgent(proxyUrl, { timeout: REQUEST_TIMEOUT });
27
  } else if (urlObj.protocol === 'https:') {
28
    const proxyOpts = {
29
      host: urlObj.hostname,
30
      port: parseInt(urlObj.port),
31
      proxyAuth: urlObj.username && urlObj.password
32
        ? `${urlObj.username}:${urlObj.password}`
33
        : undefined,
34
      timeout: REQUEST_TIMEOUT
35
    };
36
    return tunnel.httpsOverHttps({ proxy: proxyOpts });
37
  }
38
  throw new Error(`不支持的代理协议: ${urlObj.protocol}`);
39
}
40

41
function getNextIndex() {
42
  currentIndex = (currentIndex + 1) % PROXY_POOL.length;
43
  return currentIndex;
44
}
45

46
function handleHttpRequestWithRetry(req, res, retryCount = 0) {
47
  if (retryCount >= MAX_RETRY) {
48
    res.writeHead(500, { 'Content-Type': 'text/plain' });
49
    return res.end(`所有代理均不可用，重试次数: ${retryCount}`);
50
  }
51

52
  const currentTryIndex = (currentIndex + retryCount) % PROXY_POOL.length;
53
  const agent = getAgentByIndex(currentTryIndex);
54

55
  try {
56
    proxy.web(req, res, {
57
      target: req.url,
58
      agent,
59
      prependPath: false,
60
      changeOrigin: true,
61
      timeout: REQUEST_TIMEOUT // HTTP 请求转发超时
62
    });
63

64
    proxy.once('proxyRes', () => {
65
      getNextIndex();
66
    });
67
  } catch (err) {
68
    console.error(`代理 ${PROXY_POOL[currentTryIndex]} 失败: ${err.message}`);
69
    handleHttpRequestWithRetry(req, res, retryCount + 1);
70
  }
71
}
72

73
function handleHttpsConnectWithRetry(req, socket, head, retryCount = 0) {
74
  if (retryCount >= MAX_RETRY) {
75
    socket.write('HTTP/1.1 500 All Proxies Unavailable\r\n\r\n');
76
    return socket.end();
77
  }
78

79
  const currentTryIndex = (currentIndex + retryCount) % PROXY_POOL.length;
80
  const agent = getAgentByIndex(currentTryIndex);
81
  const [host, port] = req.url.split(':');
82
  const opts = {
83
    host,
84
    port: port || 443,
85
    timeout: REQUEST_TIMEOUT // HTTPS 隧道连接超时
86
  };
87

88
  agent.createConnection(opts, (err, conn) => {
89
    if (err) {
90
      console.error(`代理 ${PROXY_POOL[currentTryIndex]} 失败: ${err.message}`);
91
      return handleHttpsConnectWithRetry(req, socket, head, retryCount + 1);
92
    }
93

94
    socket.write('HTTP/1.1 200 Connection Established\r\n\r\n');
95
    conn.write(head);
96
    conn.pipe(socket);
97
    socket.pipe(conn);
98
    getNextIndex();
99
  });
100
}
101

102
const server = http.createServer((req, res) => {
103
  handleHttpRequestWithRetry(req, res);
104
});
105

106
server.on('connect', (req, socket, head) => {
107
  handleHttpsConnectWithRetry(req, socket, head);
108
});
109

110
proxy.on('error', (err, req, res) => {
111
  if (!res.headersSent) {
112
    const retryCount = parseInt(req.headers['x-retry-count'] || 0) + 1;
113
    req.headers['x-retry-count'] = retryCount;
114
    handleHttpRequestWithRetry(req, res, retryCount);
115
  }
116
});
117

118
server.listen(LISTEN_PORT, () => {
119
  console.log(`✅ 带失败重试的混合代理服务启动: http://localhost:${LISTEN_PORT}`);
120
  console.log(`🔄 代理池(${PROXY_POOL.length}个): ${PROXY_POOL.join(', ')}`);
121
  console.log(`⏱️  请求超时时间: ${REQUEST_TIMEOUT / 1000} 秒`);
122
  console.log(`🔧 最大重试次数: ${MAX_RETRY}`);
123
});