112 строки
2.3 KiB
JavaScript
112 строки
2.3 KiB
JavaScript
var common = require('../common');
|
|
var assert = require('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 len = parseInt(req.headers['x-len'], 10);
|
|
assert.ok(len > 0);
|
|
|
|
var sent = 0;
|
|
|
|
|
|
function onError(e) {
|
|
console.log('proxy client error. sent ' + sent);
|
|
throw e;
|
|
}
|
|
|
|
var proxy_req = http.request({
|
|
host: 'localhost',
|
|
port: 9000,
|
|
method: req.method,
|
|
path: req.url,
|
|
headers: req.headers
|
|
}, function(proxy_res) {
|
|
res.writeHead(proxy_res.statusCode, proxy_res.headers);
|
|
|
|
var count = 0;
|
|
|
|
proxy_res.on('data', function(d) {
|
|
if (count++ % 1000 == 0) common.print('.');
|
|
res.write(d);
|
|
sent += d.length;
|
|
assert.ok(sent <= (len * chunk.length));
|
|
});
|
|
|
|
proxy_res.on('end', function() {
|
|
res.end();
|
|
});
|
|
|
|
});
|
|
proxy_req.on('error', onError);
|
|
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.request({
|
|
port: 9001,
|
|
host: 'localhost',
|
|
path: '/',
|
|
headers: {'x-len': len}
|
|
}, function(res) {
|
|
|
|
res.on('data', function(d) {
|
|
recved += d.length;
|
|
assert.ok(recved <= (len * chunk.length));
|
|
});
|
|
|
|
res.on('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;
|
|
}
|
|
}
|
|
|
|
var serversRunning = 0;
|
|
function ready() {
|
|
if (++serversRunning < 2) return;
|
|
call_chargen([100, 1000, 10000, 100000, 1000000]);
|
|
}
|
|
|
|
process.on('exit', function() {
|
|
assert.ok(done);
|
|
});
|