agent.js 6.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289
  1. 'use strict'
  2. var assert = require('assert')
  3. var http = require('http')
  4. var https = require('https')
  5. var net = require('net')
  6. var util = require('util')
  7. var transport = require('spdy-transport')
  8. var debug = require('debug')('spdy:client')
  9. // Node.js 0.10 and 0.12 support
  10. Object.assign = process.versions.modules >= 46
  11. ? Object.assign // eslint-disable-next-line
  12. : util._extend
  13. var EventEmitter = require('events').EventEmitter
  14. var spdy = require('../spdy')
  15. var mode = /^v0\.8\./.test(process.version)
  16. ? 'rusty'
  17. : /^v0\.(9|10)\./.test(process.version)
  18. ? 'old'
  19. : /^v0\.12\./.test(process.version)
  20. ? 'normal'
  21. : 'modern'
  22. var proto = {}
  23. function instantiate (base) {
  24. function Agent (options) {
  25. this._init(base, options)
  26. }
  27. util.inherits(Agent, base)
  28. Agent.create = function create (options) {
  29. return new Agent(options)
  30. }
  31. Object.keys(proto).forEach(function (key) {
  32. Agent.prototype[key] = proto[key]
  33. })
  34. return Agent
  35. }
  36. proto._init = function _init (base, options) {
  37. base.call(this, options)
  38. var state = {}
  39. this._spdyState = state
  40. state.host = options.host
  41. state.options = options.spdy || {}
  42. state.secure = this instanceof https.Agent
  43. state.fallback = false
  44. state.createSocket = this._getCreateSocket()
  45. state.socket = null
  46. state.connection = null
  47. // No chunked encoding
  48. this.keepAlive = false
  49. var self = this
  50. this._connect(options, function (err, connection) {
  51. if (err) {
  52. return self.emit('error', err)
  53. }
  54. state.connection = connection
  55. self.emit('_connect')
  56. })
  57. }
  58. proto._getCreateSocket = function _getCreateSocket () {
  59. // Find super's `createSocket` method
  60. var createSocket
  61. var cons = this.constructor.super_
  62. do {
  63. createSocket = cons.prototype.createSocket
  64. if (cons.super_ === EventEmitter || !cons.super_) {
  65. break
  66. }
  67. cons = cons.super_
  68. } while (!createSocket)
  69. if (!createSocket) {
  70. createSocket = http.Agent.prototype.createSocket
  71. }
  72. assert(createSocket, '.createSocket() method not found')
  73. return createSocket
  74. }
  75. proto._connect = function _connect (options, callback) {
  76. var state = this._spdyState
  77. var protocols = state.options.protocols || [
  78. 'h2',
  79. 'spdy/3.1', 'spdy/3', 'spdy/2',
  80. 'http/1.1', 'http/1.0'
  81. ]
  82. // TODO(indutny): reconnect automatically?
  83. var socket = this.createConnection(Object.assign({
  84. NPNProtocols: protocols,
  85. ALPNProtocols: protocols,
  86. servername: options.servername || options.host
  87. }, options))
  88. state.socket = socket
  89. socket.setNoDelay(true)
  90. function onError (err) {
  91. return callback(err)
  92. }
  93. socket.on('error', onError)
  94. socket.on(state.secure ? 'secureConnect' : 'connect', function () {
  95. socket.removeListener('error', onError)
  96. var protocol
  97. if (state.secure) {
  98. protocol = socket.npnProtocol ||
  99. socket.alpnProtocol ||
  100. state.options.protocol
  101. } else {
  102. protocol = state.options.protocol
  103. }
  104. // HTTP server - kill socket and switch to the fallback mode
  105. if (!protocol || protocol === 'http/1.1' || protocol === 'http/1.0') {
  106. debug('activating fallback')
  107. socket.destroy()
  108. state.fallback = true
  109. return
  110. }
  111. debug('connected protocol=%j', protocol)
  112. var connection = transport.connection.create(socket, Object.assign({
  113. protocol: /spdy/.test(protocol) ? 'spdy' : 'http2',
  114. isServer: false
  115. }, state.options.connection || {}))
  116. // Set version when we are certain
  117. if (protocol === 'h2') {
  118. connection.start(4)
  119. } else if (protocol === 'spdy/3.1') {
  120. connection.start(3.1)
  121. } else if (protocol === 'spdy/3') {
  122. connection.start(3)
  123. } else if (protocol === 'spdy/2') {
  124. connection.start(2)
  125. } else {
  126. socket.destroy()
  127. callback(new Error('Unexpected protocol: ' + protocol))
  128. return
  129. }
  130. if (state.options['x-forwarded-for'] !== undefined) {
  131. connection.sendXForwardedFor(state.options['x-forwarded-for'])
  132. }
  133. callback(null, connection)
  134. })
  135. }
  136. proto._createSocket = function _createSocket (req, options, callback) {
  137. var state = this._spdyState
  138. if (state.fallback) { return state.createSocket(req, options) }
  139. var handle = spdy.handle.create(null, null, state.socket)
  140. var socketOptions = {
  141. handle: handle,
  142. allowHalfOpen: true
  143. }
  144. var socket
  145. if (state.secure) {
  146. socket = new spdy.Socket(state.socket, socketOptions)
  147. } else {
  148. socket = new net.Socket(socketOptions)
  149. }
  150. handle.assignSocket(socket)
  151. handle.assignClientRequest(req)
  152. // Create stream only once `req.end()` is called
  153. var self = this
  154. handle.once('needStream', function () {
  155. if (state.connection === null) {
  156. self.once('_connect', function () {
  157. handle.setStream(self._createStream(req, handle))
  158. })
  159. } else {
  160. handle.setStream(self._createStream(req, handle))
  161. }
  162. })
  163. // Yes, it is in reverse
  164. req.on('response', function (res) {
  165. handle.assignRequest(res)
  166. })
  167. handle.assignResponse(req)
  168. // Handle PUSH
  169. req.addListener('newListener', spdy.request.onNewListener)
  170. // For v0.8
  171. socket.readable = true
  172. socket.writable = true
  173. if (callback) {
  174. return callback(null, socket)
  175. }
  176. return socket
  177. }
  178. if (mode === 'modern' || mode === 'normal') {
  179. proto.createSocket = proto._createSocket
  180. } else {
  181. proto.createSocket = function createSocket (name, host, port, addr, req) {
  182. var state = this._spdyState
  183. if (state.fallback) {
  184. return state.createSocket(name, host, port, addr, req)
  185. }
  186. return this._createSocket(req, {
  187. host: host,
  188. port: port
  189. })
  190. }
  191. }
  192. proto._createStream = function _createStream (req, handle) {
  193. var state = this._spdyState
  194. var self = this
  195. return state.connection.reserveStream({
  196. method: req.method,
  197. path: req.path,
  198. headers: req._headers,
  199. host: state.host
  200. }, function (err, stream) {
  201. if (err) {
  202. return self.emit('error', err)
  203. }
  204. stream.on('response', function (status, headers) {
  205. handle.emitResponse(status, headers)
  206. })
  207. })
  208. }
  209. // Public APIs
  210. proto.close = function close (callback) {
  211. var state = this._spdyState
  212. if (state.connection === null) {
  213. this.once('_connect', function () {
  214. this.close(callback)
  215. })
  216. return
  217. }
  218. state.connection.end(callback)
  219. }
  220. exports.Agent = instantiate(https.Agent)
  221. exports.PlainAgent = instantiate(http.Agent)
  222. exports.create = function create (base, options) {
  223. if (typeof base === 'object') {
  224. options = base
  225. base = null
  226. }
  227. if (base) {
  228. return instantiate(base).create(options)
  229. }
  230. if (options.spdy && options.spdy.plain) {
  231. return exports.PlainAgent.create(options)
  232. } else { return exports.Agent.create(options) }
  233. }