You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

106 lines
2.3 KiB

common = require("../common");
assert = common.assert;
var util = require("util"),
fs = require("fs"),
http = require("http"),
url = require("url");
var chunk = '01234567890123456789';
// Produce a very large response.
var chargen = http.createServer(function (req, res) {
var len = parseInt(req.headers['x-len'], 10);
assert.ok(len > 0);
res.writeHead(200, {"transfer-encoding":"chunked"});
for (var i=0; i<len; i++) {
if (i % 1000 == 0) common.print(',');
res.write(chunk);
}
res.end();
});
chargen.listen(9000, ready);
// Proxy to the chargen server.
var proxy = http.createServer(function (req, res) {
var c = http.createClient(9000, 'localhost');
var len = parseInt(req.headers['x-len'], 10);
assert.ok(len > 0);
var sent = 0;
c.addListener('error', function (e) {
console.log('proxy client error. sent ' + sent);
throw e;
});
var proxy_req = c.request(req.method, req.url, req.headers);
proxy_req.addListener('response', function(proxy_res) {
res.writeHead(proxy_res.statusCode, proxy_res.headers);
var count = 0;
proxy_res.addListener('data', function(d) {
if (count++ % 1000 == 0) common.print('.');
res.write(d);
sent += d.length;
assert.ok(sent <= (len*chunk.length));
});
proxy_res.addListener('end', function() {
res.end();
});
});
proxy_req.end();
});
proxy.listen(9001, ready);
var done = false;
function call_chargen(list) {
if (list.length > 0) {
var len = list.shift();
common.debug("calling chargen for " + len + " chunks.");
var recved = 0;
var req = http.createClient(9001, 'localhost').request('/', {'x-len': len});
req.addListener('response', function(res) {
res.addListener('data', function(d) {
recved += d.length;
assert.ok(recved <= (len*chunk.length));
});
res.addListener('end', function() {
assert.ok(recved <= (len*chunk.length));
common.debug("end for " + len + " chunks.");
call_chargen(list);
});
});
req.end();
} else {
console.log("End of list. closing servers");
proxy.close();
chargen.close();
done = true;
}
}
serversRunning = 0;
function ready () {
if (++serversRunning < 2) return;
call_chargen([ 100, 1000, 10000, 100000, 1000000 ]);
}
process.addListener('exit', function () {
assert.ok(done);
});