index.js 6.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241
  1. 'use strict'
  2. var net = require('net')
  3. , tls = require('tls')
  4. , http = require('http')
  5. , https = require('https')
  6. , events = require('events')
  7. , assert = require('assert')
  8. , util = require('util')
  9. ;
  10. exports.httpOverHttp = httpOverHttp
  11. exports.httpsOverHttp = httpsOverHttp
  12. exports.httpOverHttps = httpOverHttps
  13. exports.httpsOverHttps = httpsOverHttps
  14. function httpOverHttp(options) {
  15. var agent = new TunnelingAgent(options)
  16. agent.request = http.request
  17. return agent
  18. }
  19. function httpsOverHttp(options) {
  20. var agent = new TunnelingAgent(options)
  21. agent.request = http.request
  22. agent.createSocket = createSecureSocket
  23. return agent
  24. }
  25. function httpOverHttps(options) {
  26. var agent = new TunnelingAgent(options)
  27. agent.request = https.request
  28. return agent
  29. }
  30. function httpsOverHttps(options) {
  31. var agent = new TunnelingAgent(options)
  32. agent.request = https.request
  33. agent.createSocket = createSecureSocket
  34. return agent
  35. }
  36. function TunnelingAgent(options) {
  37. var self = this
  38. self.options = options || {}
  39. self.proxyOptions = self.options.proxy || {}
  40. self.maxSockets = self.options.maxSockets || http.Agent.defaultMaxSockets
  41. self.requests = []
  42. self.sockets = []
  43. self.on('free', function onFree(socket, host, port) {
  44. for (var i = 0, len = self.requests.length; i < len; ++i) {
  45. var pending = self.requests[i]
  46. if (pending.host === host && pending.port === port) {
  47. // Detect the request to connect same origin server,
  48. // reuse the connection.
  49. self.requests.splice(i, 1)
  50. pending.request.onSocket(socket)
  51. return
  52. }
  53. }
  54. socket.destroy()
  55. self.removeSocket(socket)
  56. })
  57. }
  58. util.inherits(TunnelingAgent, events.EventEmitter)
  59. TunnelingAgent.prototype.addRequest = function addRequest(req, options) {
  60. var self = this
  61. // Legacy API: addRequest(req, host, port, path)
  62. if (typeof options === 'string') {
  63. options = {
  64. host: options,
  65. port: arguments[2],
  66. path: arguments[3]
  67. };
  68. }
  69. if (self.sockets.length >= this.maxSockets) {
  70. // We are over limit so we'll add it to the queue.
  71. self.requests.push({host: options.host, port: options.port, request: req})
  72. return
  73. }
  74. // If we are under maxSockets create a new one.
  75. self.createConnection({host: options.host, port: options.port, request: req})
  76. }
  77. TunnelingAgent.prototype.createConnection = function createConnection(pending) {
  78. var self = this
  79. self.createSocket(pending, function(socket) {
  80. socket.on('free', onFree)
  81. socket.on('close', onCloseOrRemove)
  82. socket.on('agentRemove', onCloseOrRemove)
  83. pending.request.onSocket(socket)
  84. function onFree() {
  85. self.emit('free', socket, pending.host, pending.port)
  86. }
  87. function onCloseOrRemove(err) {
  88. self.removeSocket(socket)
  89. socket.removeListener('free', onFree)
  90. socket.removeListener('close', onCloseOrRemove)
  91. socket.removeListener('agentRemove', onCloseOrRemove)
  92. }
  93. })
  94. }
  95. TunnelingAgent.prototype.createSocket = function createSocket(options, cb) {
  96. var self = this
  97. var placeholder = {}
  98. self.sockets.push(placeholder)
  99. var connectOptions = mergeOptions({}, self.proxyOptions,
  100. { method: 'CONNECT'
  101. , path: options.host + ':' + options.port
  102. , agent: false
  103. }
  104. )
  105. if (connectOptions.proxyAuth) {
  106. connectOptions.headers = connectOptions.headers || {}
  107. connectOptions.headers['Proxy-Authorization'] = 'Basic ' +
  108. new Buffer(connectOptions.proxyAuth).toString('base64')
  109. }
  110. debug('making CONNECT request')
  111. var connectReq = self.request(connectOptions)
  112. connectReq.useChunkedEncodingByDefault = false // for v0.6
  113. connectReq.once('response', onResponse) // for v0.6
  114. connectReq.once('upgrade', onUpgrade) // for v0.6
  115. connectReq.once('connect', onConnect) // for v0.7 or later
  116. connectReq.once('error', onError)
  117. connectReq.end()
  118. function onResponse(res) {
  119. // Very hacky. This is necessary to avoid http-parser leaks.
  120. res.upgrade = true
  121. }
  122. function onUpgrade(res, socket, head) {
  123. // Hacky.
  124. process.nextTick(function() {
  125. onConnect(res, socket, head)
  126. })
  127. }
  128. function onConnect(res, socket, head) {
  129. connectReq.removeAllListeners()
  130. socket.removeAllListeners()
  131. if (res.statusCode === 200) {
  132. assert.equal(head.length, 0)
  133. debug('tunneling connection has established')
  134. self.sockets[self.sockets.indexOf(placeholder)] = socket
  135. cb(socket)
  136. } else {
  137. debug('tunneling socket could not be established, statusCode=%d', res.statusCode)
  138. var error = new Error('tunneling socket could not be established, ' + 'statusCode=' + res.statusCode)
  139. error.code = 'ECONNRESET'
  140. options.request.emit('error', error)
  141. self.removeSocket(placeholder)
  142. }
  143. }
  144. function onError(cause) {
  145. connectReq.removeAllListeners()
  146. debug('tunneling socket could not be established, cause=%s\n', cause.message, cause.stack)
  147. var error = new Error('tunneling socket could not be established, ' + 'cause=' + cause.message)
  148. error.code = 'ECONNRESET'
  149. options.request.emit('error', error)
  150. self.removeSocket(placeholder)
  151. }
  152. }
  153. TunnelingAgent.prototype.removeSocket = function removeSocket(socket) {
  154. var pos = this.sockets.indexOf(socket)
  155. if (pos === -1) return
  156. this.sockets.splice(pos, 1)
  157. var pending = this.requests.shift()
  158. if (pending) {
  159. // If we have pending requests and a socket gets closed a new one
  160. // needs to be created to take over in the pool for the one that closed.
  161. this.createConnection(pending)
  162. }
  163. }
  164. function createSecureSocket(options, cb) {
  165. var self = this
  166. TunnelingAgent.prototype.createSocket.call(self, options, function(socket) {
  167. // 0 is dummy port for v0.6
  168. var secureSocket = tls.connect(0, mergeOptions({}, self.options,
  169. { servername: options.host
  170. , socket: socket
  171. }
  172. ))
  173. self.sockets[self.sockets.indexOf(socket)] = secureSocket
  174. cb(secureSocket)
  175. })
  176. }
  177. function mergeOptions(target) {
  178. for (var i = 1, len = arguments.length; i < len; ++i) {
  179. var overrides = arguments[i]
  180. if (typeof overrides === 'object') {
  181. var keys = Object.keys(overrides)
  182. for (var j = 0, keyLen = keys.length; j < keyLen; ++j) {
  183. var k = keys[j]
  184. if (overrides[k] !== undefined) {
  185. target[k] = overrides[k]
  186. }
  187. }
  188. }
  189. }
  190. return target
  191. }
  192. var debug
  193. if (process.env.NODE_DEBUG && /\btunnel\b/.test(process.env.NODE_DEBUG)) {
  194. debug = function() {
  195. var args = Array.prototype.slice.call(arguments)
  196. if (typeof args[0] === 'string') {
  197. args[0] = 'TUNNEL: ' + args[0]
  198. } else {
  199. args.unshift('TUNNEL:')
  200. }
  201. console.error.apply(console, args)
  202. }
  203. } else {
  204. debug = function() {}
  205. }
  206. exports.debug = debug // for test