3 var net = require('net')
5 , http = require('http')
6 , https = require('https')
7 , events = require('events')
8 , assert = require('assert')
9 , util = require('util')
12 exports.httpOverHttp = httpOverHttp
13 exports.httpsOverHttp = httpsOverHttp
14 exports.httpOverHttps = httpOverHttps
15 exports.httpsOverHttps = httpsOverHttps
18 function httpOverHttp(options) {
19 var agent = new TunnelingAgent(options)
20 agent.request = http.request
24 function httpsOverHttp(options) {
25 var agent = new TunnelingAgent(options)
26 agent.request = http.request
27 agent.createSocket = createSecureSocket
31 function httpOverHttps(options) {
32 var agent = new TunnelingAgent(options)
33 agent.request = https.request
37 function httpsOverHttps(options) {
38 var agent = new TunnelingAgent(options)
39 agent.request = https.request
40 agent.createSocket = createSecureSocket
45 function TunnelingAgent(options) {
47 self.options = options || {}
48 self.proxyOptions = self.options.proxy || {}
49 self.maxSockets = self.options.maxSockets || http.Agent.defaultMaxSockets
53 self.on('free', function onFree(socket, host, port) {
54 for (var i = 0, len = self.requests.length; i < len; ++i) {
55 var pending = self.requests[i]
56 if (pending.host === host && pending.port === port) {
57 // Detect the request to connect same origin server,
58 // reuse the connection.
59 self.requests.splice(i, 1)
60 pending.request.onSocket(socket)
65 self.removeSocket(socket)
68 util.inherits(TunnelingAgent, events.EventEmitter)
70 TunnelingAgent.prototype.addRequest = function addRequest(req, host, port) {
73 if (self.sockets.length >= this.maxSockets) {
74 // We are over limit so we'll add it to the queue.
75 self.requests.push({host: host, port: port, request: req})
79 // If we are under maxSockets create a new one.
80 self.createSocket({host: host, port: port, request: req}, function(socket) {
81 socket.on('free', onFree)
82 socket.on('close', onCloseOrRemove)
83 socket.on('agentRemove', onCloseOrRemove)
87 self.emit('free', socket, host, port)
90 function onCloseOrRemove(err) {
92 socket.removeListener('free', onFree)
93 socket.removeListener('close', onCloseOrRemove)
94 socket.removeListener('agentRemove', onCloseOrRemove)
99 TunnelingAgent.prototype.createSocket = function createSocket(options, cb) {
102 self.sockets.push(placeholder)
104 var connectOptions = mergeOptions({}, self.proxyOptions,
106 , path: options.host + ':' + options.port
110 if (connectOptions.proxyAuth) {
111 connectOptions.headers = connectOptions.headers || {}
112 connectOptions.headers['Proxy-Authorization'] = 'Basic ' +
113 new Buffer(connectOptions.proxyAuth).toString('base64')
116 debug('making CONNECT request')
117 var connectReq = self.request(connectOptions)
118 connectReq.useChunkedEncodingByDefault = false // for v0.6
119 connectReq.once('response', onResponse) // for v0.6
120 connectReq.once('upgrade', onUpgrade) // for v0.6
121 connectReq.once('connect', onConnect) // for v0.7 or later
122 connectReq.once('error', onError)
125 function onResponse(res) {
126 // Very hacky. This is necessary to avoid http-parser leaks.
130 function onUpgrade(res, socket, head) {
132 process.nextTick(function() {
133 onConnect(res, socket, head)
137 function onConnect(res, socket, head) {
138 connectReq.removeAllListeners()
139 socket.removeAllListeners()
141 if (res.statusCode === 200) {
142 assert.equal(head.length, 0)
143 debug('tunneling connection has established')
144 self.sockets[self.sockets.indexOf(placeholder)] = socket
147 debug('tunneling socket could not be established, statusCode=%d', res.statusCode)
148 var error = new Error('tunneling socket could not be established, ' + 'statusCode=' + res.statusCode)
149 error.code = 'ECONNRESET'
150 options.request.emit('error', error)
151 self.removeSocket(placeholder)
155 function onError(cause) {
156 connectReq.removeAllListeners()
158 debug('tunneling socket could not be established, cause=%s\n', cause.message, cause.stack)
159 var error = new Error('tunneling socket could not be established, ' + 'cause=' + cause.message)
160 error.code = 'ECONNRESET'
161 options.request.emit('error', error)
162 self.removeSocket(placeholder)
166 TunnelingAgent.prototype.removeSocket = function removeSocket(socket) {
167 var pos = this.sockets.indexOf(socket)
168 if (pos === -1) return
170 this.sockets.splice(pos, 1)
172 var pending = this.requests.shift()
174 // If we have pending requests and a socket gets closed a new one
175 // needs to be created to take over in the pool for the one that closed.
176 this.createSocket(pending, function(socket) {
177 pending.request.onSocket(socket)
182 function createSecureSocket(options, cb) {
184 TunnelingAgent.prototype.createSocket.call(self, options, function(socket) {
185 // 0 is dummy port for v0.6
186 var secureSocket = tls.connect(0, mergeOptions({}, self.options,
187 { servername: options.host
196 function mergeOptions(target) {
197 for (var i = 1, len = arguments.length; i < len; ++i) {
198 var overrides = arguments[i]
199 if (typeof overrides === 'object') {
200 var keys = Object.keys(overrides)
201 for (var j = 0, keyLen = keys.length; j < keyLen; ++j) {
203 if (overrides[k] !== undefined) {
204 target[k] = overrides[k]
214 if (process.env.NODE_DEBUG && /\btunnel\b/.test(process.env.NODE_DEBUG)) {
216 var args = Array.prototype.slice.call(arguments)
217 if (typeof args[0] === 'string') {
218 args[0] = 'TUNNEL: ' + args[0]
220 args.unshift('TUNNEL:')
222 console.error.apply(console, args)
225 debug = function() {}
227 exports.debug = debug // for test