Загрузить файлы в «/»
This commit is contained in:
@@ -0,0 +1,102 @@
|
||||
'use strict';
|
||||
/*
|
||||
* proxy-forwarder.js — локальный прокси-«перекладчик» на слот.
|
||||
*
|
||||
* ЗАЧЕМ. Прокси у browser-context в CDP фиксируется при создании контекста —
|
||||
* сменить его на лету нельзя, только пересоздать контекст (а это прогрев ~10с и
|
||||
* нестабильность в Node при частом destroy). Поэтому контекст навсегда смотрит в
|
||||
* ОДИН локальный адрес (127.0.0.1:PORT), а за этим адресом стоит вот этот
|
||||
* форвардер, у которого upstream (реальный резидентский прокси) можно менять
|
||||
* одной переменной. Смена IP аккаунта = setUpstream(...) + reload страницы,
|
||||
* БЕЗ пересоздания контекста → ~2-5с вместо ~10-18с.
|
||||
*
|
||||
* Держит только то, что реально нужно минту: HTTPS через CONNECT-туннель и
|
||||
* обычный HTTP forward. Upstream — HTTP-прокси с Basic-авторизацией (формат
|
||||
* плейнпрокси host:port:user:pass). SOCKS не поддерживаем: наши чат-прокси HTTP.
|
||||
*
|
||||
* Зависимостей нет — только встроенные http/net.
|
||||
*/
|
||||
|
||||
import http from 'http';
|
||||
import net from 'net';
|
||||
|
||||
/**
|
||||
* Поднять форвардер на 127.0.0.1:port.
|
||||
* @returns {{ server: http.Server, setUpstream: (u: {host,port,user?,pass?}|null)=>void, getUpstream: ()=>object|null, close: ()=>void }}
|
||||
*/
|
||||
export function startForwarder(port) {
|
||||
// upstream живёт в замыкании: setUpstream меняет его атомарно, новые
|
||||
// соединения (после reload страницы) идут уже через новый прокси.
|
||||
let upstream = null;
|
||||
|
||||
const authHeader = (u) =>
|
||||
u && u.user ? 'Basic ' + Buffer.from(`${u.user}:${u.pass || ''}`).toString('base64') : null;
|
||||
|
||||
const server = http.createServer((req, res) => {
|
||||
// Обычный HTTP (не CONNECT). Твич весь на https, так что путь редкий, но
|
||||
// держим, чтобы случайный http-ресурс не ломал страницу.
|
||||
const u = upstream;
|
||||
if (!u) { res.writeHead(502); res.end('no upstream'); return; }
|
||||
const headers = { ...req.headers };
|
||||
const a = authHeader(u);
|
||||
if (a) headers['proxy-authorization'] = a;
|
||||
const preq = http.request(
|
||||
{ host: u.host, port: u.port, method: req.method, path: req.url, headers },
|
||||
(pres) => { res.writeHead(pres.statusCode || 502, pres.headers); pres.pipe(res); },
|
||||
);
|
||||
preq.on('error', () => { try { res.writeHead(502); res.end(); } catch (_) {} });
|
||||
req.pipe(preq);
|
||||
});
|
||||
|
||||
// HTTPS: браузер шлёт CONNECT host:443 — открываем туннель до цели ЧЕРЕЗ upstream.
|
||||
server.on('connect', (req, clientSocket, head) => {
|
||||
const u = upstream;
|
||||
if (!u) { try { clientSocket.end('HTTP/1.1 502 Bad Gateway\r\n\r\n'); } catch (_) {} return; }
|
||||
|
||||
const upSock = net.connect(u.port, u.host, () => {
|
||||
let line = `CONNECT ${req.url} HTTP/1.1\r\nHost: ${req.url}\r\n`;
|
||||
const a = authHeader(u);
|
||||
if (a) line += `Proxy-Authorization: ${a}\r\n`;
|
||||
line += 'Connection: keep-alive\r\n\r\n';
|
||||
upSock.write(line);
|
||||
});
|
||||
|
||||
let header = Buffer.alloc(0);
|
||||
let piped = false;
|
||||
const onData = (d) => {
|
||||
if (piped) return;
|
||||
header = Buffer.concat([header, d]);
|
||||
const idx = header.indexOf('\r\n\r\n');
|
||||
if (idx === -1) return; // ответ upstream ещё не дочитан
|
||||
const status = header.slice(0, header.indexOf('\r\n')).toString();
|
||||
upSock.removeListener('data', onData);
|
||||
if (!/^HTTP\/1\.[01] 200/.test(status)) { // upstream отказал (авторизация/цель)
|
||||
try { clientSocket.end(`HTTP/1.1 502 Bad Gateway\r\n\r\n`); } catch (_) {}
|
||||
try { upSock.end(); } catch (_) {}
|
||||
return;
|
||||
}
|
||||
piped = true;
|
||||
try { clientSocket.write('HTTP/1.1 200 Connection Established\r\n\r\n'); } catch (_) {}
|
||||
const rest = header.slice(idx + 4); // тело, прилипшее к заголовку
|
||||
if (rest.length) clientSocket.write(rest);
|
||||
if (head && head.length) upSock.write(head);
|
||||
upSock.pipe(clientSocket);
|
||||
clientSocket.pipe(upSock);
|
||||
};
|
||||
upSock.on('data', onData);
|
||||
|
||||
const kill = () => { try { upSock.destroy(); } catch (_) {} try { clientSocket.destroy(); } catch (_) {} };
|
||||
upSock.on('error', kill);
|
||||
clientSocket.on('error', kill);
|
||||
});
|
||||
|
||||
server.on('clientError', (_e, sock) => { try { sock.destroy(); } catch (_) {} });
|
||||
server.listen(port, '127.0.0.1');
|
||||
|
||||
return {
|
||||
server,
|
||||
setUpstream(u) { upstream = u; },
|
||||
getUpstream() { return upstream; },
|
||||
close() { try { server.close(); } catch (_) {} },
|
||||
};
|
||||
}
|
||||
Reference in New Issue
Block a user