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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
/* global arguments */
"use strict";
var CC = Components.Constructor;
const BinaryInputStream = CC(
"@mozilla.org/binaryinputstream;1",
"nsIBinaryInputStream",
"setInputStream"
);
const ProtocolProxyService = CC(
"@mozilla.org/network/protocol-proxy-service;1",
"nsIProtocolProxyService"
);
var sts = Cc["@mozilla.org/network/socket-transport-service;1"].getService(
Ci.nsISocketTransportService
);
function waitForStream(stream, streamType) {
return new Promise((resolve, reject) => {
stream = stream.QueryInterface(streamType);
if (!stream) {
reject("stream didn't implement given stream type");
}
let currentThread = Cc["@mozilla.org/thread-manager;1"].getService()
.currentThread;
stream.asyncWait(
stream => {
resolve(stream);
},
0,
0,
currentThread
);
});
}
async function launchConnection(
socks_vers,
socks_port,
dest_host,
dest_port,
dns
) {
let pi_flags = 0;
if (dns == "remote") {
pi_flags = Ci.nsIProxyInfo.TRANSPARENT_PROXY_RESOLVES_HOST;
}
let pps = new ProtocolProxyService();
let pi = pps.newProxyInfo(
socks_vers,
"localhost",
socks_port,
"",
"",
pi_flags,
-1,
null
);
let trans = sts.createTransport([], dest_host, dest_port, pi, null);
let input = trans.openInputStream(0, 0, 0);
let output = trans.openOutputStream(0, 0, 0);
input = await waitForStream(input, Ci.nsIAsyncInputStream);
let bin = new BinaryInputStream(input);
let data = bin.readBytes(5);
let response;
if (data == "PING!") {
print("client: got ping, sending pong.");
response = "PONG!";
} else {
print("client: wrong data from server:", data);
response = "Error: wrong data received.";
}
output = await waitForStream(output, Ci.nsIAsyncOutputStream);
output.write(response, response.length);
output.close();
input.close();
}
async function run(args) {
for (let arg of args) {
print("client: running test", arg);
let test = arg.split("|");
await launchConnection(
test[0],
parseInt(test[1]),
test[2],
parseInt(test[3]),
test[4]
);
}
}
var satisfied = false;
run(arguments).then(() => (satisfied = true));
var mainThread = Cc["@mozilla.org/thread-manager;1"].getService().mainThread;
while (!satisfied) {
mainThread.processNextEvent(true);
}
|
