Skip to content

Instantly share code, notes, and snippets.

@couto
Forked from grantmichaels/proxy.js
Last active December 15, 2015 22:49
Show Gist options
  • Save couto/5335990 to your computer and use it in GitHub Desktop.
Save couto/5335990 to your computer and use it in GitHub Desktop.
make it a little friendlier for the terminal
var net = require("net"),
args = {};
process.argv.forEach(function (val, index) {
var dest, listen;
if (val === '--destination') {
dest = process.argv[index + 1].split(':');
args.destination = dest[0];
args.destPort = dest[1] || 80;
}
if (val === '--listen') {
listen = process.argv[index + 1];
args.listen = listen;
}
})
var Proxy = function () {};
Proxy.to = function (dst_host, dst_port) {
var proxy = new Proxy();
proxy.dstHost = dst_host;
proxy.dstPort = dst_port;
proxy.mid = [];
return proxy;
};
Proxy.prototype.use = function (fn) {
this.mid.push(fn);
return this;
}
Proxy.prototype.handle = function (data, src, dst) {
var next = Proxy.noop,
i = this.mid.length - 1;
for (i; i >= 0; i--) {
next = this.mid[i].bind(this, data, src, dst, next);
}
next();
return this;
};
Proxy.prototype.listen = function(port) {
var self = this;
this.server = net.createServer(function (sockA) {
var preb = [],
sockB;
sockA.on("data", pre = function(data) {
preb.push(data);
});
sockB = net.connect(self.dstPort, self.dstHost, function () {
sockA.removeListener("data", pre);
while (preb.length) {
self.handle(preb.shift(), sockA, sockB);
}
sockA.on("data", function (data) {
self.handle(data, sockA, sockB);
});
sockB.on("data", function (data) {
self.handle(data, sockB, sockA);
});
sockA.on("end", function () {
sockB.removeAllListeners();
sockB.end();
});
sockB.on("end", function () {
sockA.removeAllListeners();
sockA.end();
});
});
});
this.server.listen(port, function () {
console.log('\033[0;32m ✔ \033[0m' +
'redirecting from:\033[0;36m localhost:' +
port +
'\033[0m to: \033[0;36m' +
self.dstHost +
':' +
self.dstPort +
'\033[0m'
);
});
return this;
};
// built-in middlewares
Proxy.noop = function () {};
Proxy.send = function (data, src, dst, next) {
dst.write(data);
next();
};
Proxy.log = function (data, src, dst, next) {
console.log(" → " + src.remoteAddress + ":" + src.remotePort + " ▸ " + dst.remoteAddress + ":" + dst.remotePort);
console.log(" → " data.toString());
next();
};
// redirect all connections on port 8080 to google.com:80
// node proxy.js --destination google.com:80 --listen 8080
Proxy
.to(args.destination, args.destPort)
.use(Proxy.log)
.use(Proxy.send)
.listen(args.listen || 3000);
@couto
Copy link
Author

couto commented Apr 8, 2013

Simple TCP Proxy.
I used this to make two Virtual Machines, under the same host, talk to each other.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment