wraps connections in a domain
This commit is contained in:
parent
ef2ef80164
commit
85217346c0
|
@ -1,23 +1,34 @@
|
||||||
const socks = require('socksv5');
|
const socks = require('socksv5');
|
||||||
const SOCKS5Server = socks.Server;
|
const SOCKS5Server = socks.Server;
|
||||||
|
const domain = require('domain');
|
||||||
|
|
||||||
class SOCKSServer extends SOCKS5Server{
|
class SOCKSServer extends SOCKS5Server{
|
||||||
constructor(tor_pool, logger) {
|
constructor(tor_pool, logger) {
|
||||||
super((info, accept, deny) => {
|
let handleConnection = (info, accept, deny) => {
|
||||||
|
let d = domain.create();
|
||||||
|
|
||||||
let inbound_socket = accept(true);
|
let inbound_socket = accept(true);
|
||||||
|
d.add(inbound_socket);
|
||||||
var outbound_socket;
|
var outbound_socket;
|
||||||
let buffer = [];
|
let buffer = [];
|
||||||
|
|
||||||
let socks_port = (tor_pool.next().socks_port);
|
let socks_port = (tor_pool.next().socks_port);
|
||||||
logger && logger.info(`[socks]: ${info.srcAddr}:${info.srcPort} → 127.0.0.1:${socks_port} → ${info.dstAddr}:${info.dstPort}`)
|
logger && logger.info(`[socks]: ${info.srcAddr}:${info.srcPort} → 127.0.0.1:${socks_port} → ${info.dstAddr}:${info.dstPort}`)
|
||||||
|
|
||||||
let onClose = () => {
|
let onClose = (error) => {
|
||||||
inbound_socket && inbound_socket.end();
|
inbound_socket && inbound_socket.end();
|
||||||
outbound_socket && outbound_socket.end();
|
outbound_socket && outbound_socket.end();
|
||||||
|
|
||||||
inbound_socket = outbound_socket = buffer = void(0);
|
inbound_socket = outbound_socket = buffer = void(0);
|
||||||
|
|
||||||
|
if (error)
|
||||||
|
this.logger.error(`[socks]: an error occured: ${error.message}`)
|
||||||
|
|
||||||
|
d.exit();
|
||||||
};
|
};
|
||||||
|
|
||||||
|
d.on('error', onClose);
|
||||||
|
|
||||||
let onInboundData = (data) => buffer.push(data)
|
let onInboundData = (data) => buffer.push(data)
|
||||||
|
|
||||||
if (!inbound_socket) return;
|
if (!inbound_socket) return;
|
||||||
|
@ -26,6 +37,7 @@ class SOCKSServer extends SOCKS5Server{
|
||||||
inbound_socket.on('data', onInboundData);
|
inbound_socket.on('data', onInboundData);
|
||||||
inbound_socket.on('error', onClose);
|
inbound_socket.on('error', onClose);
|
||||||
|
|
||||||
|
d.run(() => {
|
||||||
socks.connect({
|
socks.connect({
|
||||||
host: info.dstAddr,
|
host: info.dstAddr,
|
||||||
port: info.dstPort,
|
port: info.dstPort,
|
||||||
|
@ -35,6 +47,7 @@ class SOCKSServer extends SOCKS5Server{
|
||||||
auths: [ socks.auth.None() ]
|
auths: [ socks.auth.None() ]
|
||||||
}, ($outbound_socket) => {
|
}, ($outbound_socket) => {
|
||||||
outbound_socket = $outbound_socket;
|
outbound_socket = $outbound_socket;
|
||||||
|
d.add(outbound_socket);
|
||||||
outbound_socket && outbound_socket.on('close', onClose);
|
outbound_socket && outbound_socket.on('close', onClose);
|
||||||
|
|
||||||
inbound_socket && inbound_socket.removeListener('data', onInboundData);
|
inbound_socket && inbound_socket.removeListener('data', onInboundData);
|
||||||
|
@ -53,6 +66,9 @@ class SOCKSServer extends SOCKS5Server{
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
});
|
});
|
||||||
|
};
|
||||||
|
|
||||||
|
super(handleConnection);
|
||||||
|
|
||||||
this.logger = logger;
|
this.logger = logger;
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue