forked from tewarid/node-tcp-relay
-
Notifications
You must be signed in to change notification settings - Fork 0
/
relayc.js
71 lines (63 loc) · 2.2 KB
/
relayc.js
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
#!/usr/bin/env node
var argv = require("optimist")
.usage('Usage: $0 --host [host] --port [port] --relayHost [host] --relayPort [port] [--numConn [count]]')
.demand(['host', 'port', 'relayHost', 'relayPort'])
.argv;
console.log(argv);
var net = require("net");
function connect() {
var relaySocket = new net.Socket();
var serverSocket = undefined;
relaySocket.connect(argv.relayPort, argv.relayHost, function () {
console.log("relay socket established");
relaySocket.on("data", function (data) {
if (serverSocket == undefined) {
serverSocket = new net.Socket();
serverSocket.connect(argv.port, argv.host, function () {
console.log("server socket established");
serverSocket.write(data);
});
serverSocket.on("data", function (data) {
try {
relaySocket.write(data);
} catch (ex) {
console.log(ex);
}
});
serverSocket.on("close", function (had_error) {
console.log("server socket closed");
relaySocket.end();
});
serverSocket.on("error", function (exception) {
console.log("server socket error");
console.log(exception);
relaySocket.end();
});
connect();
console.log("next relay socket established");
} else {
try {
serverSocket.write(data);
} catch (ex) {
console.log(ex);
}
}
});
relaySocket.on("close", function (had_error) {
console.log("relay socket closed");
if (serverSocket != undefined)
serverSocket.end();
});
relaySocket.on("error", function (exception) {
console.log("relay socket error");
console.log(exception);
});
});
}
var count = argv.numConn;
if (count == undefined) {
count = 1;
}
for (var i = 0; i < count; i++) {
connect();
}