mirror of
https://github.com/http-party/node-http-proxy.git
synced 2025-12-08 20:59:18 +00:00
164 lines
4.3 KiB
JavaScript
164 lines
4.3 KiB
JavaScript
var http = require('http'),
|
||
https = require('https'),
|
||
web_o = require('./web-outgoing'),
|
||
common = require('../common'),
|
||
passes = exports;
|
||
|
||
web_o = Object.keys(web_o).map(function(pass) {
|
||
return web_o[pass];
|
||
});
|
||
|
||
/*!
|
||
* Array of passes.
|
||
*
|
||
* A `pass` is just a function that is executed on `req, res, options`
|
||
* so that you can easily add new checks while still keeping the base
|
||
* flexible.
|
||
*/
|
||
|
||
[ // <--
|
||
|
||
/**
|
||
* Sets `content-length` to '0' if request is of DELETE type.
|
||
*
|
||
* @param {ClientRequest} Req Request object
|
||
* @param {IncomingMessage} Res Response object
|
||
* @param {Object} Options Config object passed to the proxy
|
||
*
|
||
* @api private
|
||
*/
|
||
|
||
function deleteLength(req, res, options) {
|
||
if(req.method === 'DELETE' && !req.headers['content-length']) {
|
||
req.headers['content-length'] = '0';
|
||
}
|
||
},
|
||
|
||
/**
|
||
* Sets timeout in request socket if it was specified in options.
|
||
*
|
||
* @param {ClientRequest} Req Request object
|
||
* @param {IncomingMessage} Res Response object
|
||
* @param {Object} Options Config object passed to the proxy
|
||
*
|
||
* @api private
|
||
*/
|
||
|
||
function timeout(req, res, options) {
|
||
if(options.timeout) {
|
||
req.socket.setTimeout(options.timeout);
|
||
}
|
||
},
|
||
|
||
/**
|
||
* Sets `x-forwarded-*` headers if specified in config.
|
||
*
|
||
* @param {ClientRequest} Req Request object
|
||
* @param {IncomingMessage} Res Response object
|
||
* @param {Object} Options Config object passed to the proxy
|
||
*
|
||
* @api private
|
||
*/
|
||
|
||
function XHeaders(req, res, options) {
|
||
if(!options.xfwd) return;
|
||
|
||
var values = {
|
||
for : req.connection.remoteAddress || req.socket.remoteAddress,
|
||
port : common.getPort(req),
|
||
proto: req.isSpdy ? 'https' : (req.connection.pair ? 'https' : 'http')
|
||
};
|
||
|
||
['for', 'port', 'proto'].forEach(function(header) {
|
||
req.headers['x-forwarded-' + header] =
|
||
(req.headers['x-forwarded-' + header] || '') +
|
||
(req.headers['x-forwarded-' + header] ? ',' : '') +
|
||
values[header];
|
||
});
|
||
},
|
||
|
||
/**
|
||
* Does the actual proxying. If `forward` is enabled fires up
|
||
* a ForwardStream, same happens for ProxyStream. The request
|
||
* just dies otherwise.
|
||
*
|
||
* @param {ClientRequest} Req Request object
|
||
* @param {IncomingMessage} Res Response object
|
||
* @param {Object} Options Config object passed to the proxy
|
||
*
|
||
* @api private
|
||
*/
|
||
|
||
function stream(req, res, options, _, server, clb) {
|
||
|
||
// And we begin!
|
||
if (!clb) {
|
||
server.emit('start', req, res, options.target)
|
||
}
|
||
if(options.forward) {
|
||
// If forward enable, so just pipe the request
|
||
var forwardReq = (options.forward.protocol === 'https:' ? https : http).request(
|
||
common.setupOutgoing(options.ssl || {}, options, req, 'forward')
|
||
);
|
||
(options.buffer || req).pipe(forwardReq);
|
||
if(!options.target) { return res.end(); }
|
||
}
|
||
|
||
// Request initalization
|
||
var proxyReq = (options.target.protocol === 'https:' ? https : http).request(
|
||
common.setupOutgoing(options.ssl || {}, options, req)
|
||
);
|
||
|
||
// allow outgoing socket to timeout so that we could
|
||
// show an error page at the initial request
|
||
if(options.targetTimeout) {
|
||
proxyReq.setTimeout(options.targetTimeout, function() {
|
||
proxyReq.abort();
|
||
});
|
||
}
|
||
|
||
// Ensure we abort proxy if request is aborted
|
||
req.on('aborted', function () {
|
||
proxyReq.abort();
|
||
});
|
||
|
||
// Handle errors on incoming request as well as it makes sense to
|
||
req.on('error', proxyError);
|
||
|
||
// Error Handler
|
||
proxyReq.on('error', proxyError);
|
||
|
||
function proxyError (err){
|
||
if (clb) {
|
||
clb(err, req, res);
|
||
} else {
|
||
server.emit('error', err, req, res);
|
||
}
|
||
}
|
||
|
||
(options.buffer || req).pipe(proxyReq);
|
||
|
||
proxyReq.on('response', function(proxyRes) {
|
||
if(server) { server.emit('proxyRes', proxyRes, req, res); }
|
||
for(var i=0; i < web_o.length; i++) {
|
||
if(web_o[i](req, res, proxyRes)) { break; }
|
||
}
|
||
|
||
// Allow us to listen when the proxy has completed
|
||
proxyRes.on('end', function () {
|
||
if (!clb) {
|
||
server.emit('end', req, res, proxyRes);
|
||
}
|
||
})
|
||
|
||
proxyRes.pipe(res);
|
||
});
|
||
|
||
//proxyReq.end();
|
||
}
|
||
|
||
] // <--
|
||
.forEach(function(func) {
|
||
passes[func.name] = func;
|
||
});
|