|
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348 |
- /* jshint node:true */
- 'use strict';
- /**
- * @fileOverview
- * Global proxy settings.
- */
- var globalTunnel = exports;
- exports.constructor = function() {};
-
- var http = require('http');
- var https = require('https');
- var urlParse = require('url').parse;
- var urlStringify = require('url').format;
-
- var pick = require('lodash/pick');
- var assign = require('lodash/assign');
- var clone = require('lodash/clone');
- var tunnel = require('tunnel');
- var npmConfig = require('npm-conf');
- var encodeUrl = require('encodeurl');
-
- var agents = require('./lib/agents');
- exports.agents = agents;
-
- var ENV_VAR_PROXY_SEARCH_ORDER = [
- 'https_proxy',
- 'HTTPS_PROXY',
- 'http_proxy',
- 'HTTP_PROXY'
- ];
- var NPM_CONFIG_PROXY_SEARCH_ORDER = ['https-proxy', 'http-proxy', 'proxy'];
-
- // Save the original settings for restoration later.
- var ORIGINALS = {
- http: pick(http, 'globalAgent', ['request', 'get']),
- https: pick(https, 'globalAgent', ['request', 'get']),
- env: pick(process.env, ENV_VAR_PROXY_SEARCH_ORDER)
- };
-
- var loggingEnabled =
- process &&
- process.env &&
- process.env.DEBUG &&
- process.env.DEBUG.toLowerCase().indexOf('global-tunnel') !== -1 &&
- console &&
- typeof console.log === 'function';
-
- function log(message) {
- if (loggingEnabled) {
- console.log('DEBUG global-tunnel: ' + message);
- }
- }
-
- function resetGlobals() {
- assign(http, ORIGINALS.http);
- assign(https, ORIGINALS.https);
- var val;
- for (var key in ORIGINALS.env) {
- if (Object.prototype.hasOwnProperty.call(ORIGINALS.env, key)) {
- val = ORIGINALS.env[key];
- if (val !== null && val !== undefined) {
- process.env[key] = val;
- }
- }
- }
- }
-
- /**
- * Parses the de facto `http_proxy` environment.
- */
- function tryParse(url) {
- if (!url) {
- return null;
- }
-
- var parsed = urlParse(url);
-
- return {
- protocol: parsed.protocol,
- host: parsed.hostname,
- port: parseInt(parsed.port, 10),
- proxyAuth: parsed.auth
- };
- }
-
- // Stringifies the normalized parsed config
- function stringifyProxy(conf) {
- return encodeUrl(
- urlStringify({
- protocol: conf.protocol,
- hostname: conf.host,
- port: conf.port,
- auth: conf.proxyAuth
- })
- );
- }
-
- globalTunnel.isProxying = false;
- globalTunnel.proxyUrl = null;
- globalTunnel.proxyConfig = null;
-
- function findEnvVarProxy() {
- var i;
- var key;
- var val;
- var result;
- for (i = 0; i < ENV_VAR_PROXY_SEARCH_ORDER.length; i++) {
- key = ENV_VAR_PROXY_SEARCH_ORDER[i];
- val = process.env[key];
- if (val !== null && val !== undefined) {
- // Get the first non-empty
- result = result || val;
- // Delete all
- // NB: we do it here to prevent double proxy handling (and for example path change)
- // by us and the `request` module or other sub-dependencies
- delete process.env[key];
- log('Found proxy in environment variable ' + ENV_VAR_PROXY_SEARCH_ORDER[i]);
- }
- }
-
- if (!result) {
- // __GLOBAL_TUNNEL_DEPENDENCY_NPMCONF__ is a hook to override the npm-conf module
- var config =
- (global.__GLOBAL_TUNNEL_DEPENDENCY_NPMCONF__ &&
- global.__GLOBAL_TUNNEL_DEPENDENCY_NPMCONF__()) ||
- npmConfig();
-
- for (i = 0; i < NPM_CONFIG_PROXY_SEARCH_ORDER.length && !val; i++) {
- val = config.get(NPM_CONFIG_PROXY_SEARCH_ORDER[i]);
- }
-
- if (val) {
- log('Found proxy in npm config ' + NPM_CONFIG_PROXY_SEARCH_ORDER[i]);
- result = val;
- }
- }
-
- return result;
- }
-
- /**
- * Overrides the node http/https `globalAgent`s to use the configured proxy.
- *
- * If the config is empty, the `http_proxy` environment variable is checked.
- * If that's not present, the NPM `http-proxy` configuration is checked.
- * If neither are present no proxying will be enabled.
- *
- * @param {object} conf - Options
- * @param {string} conf.host - Hostname or IP of the HTTP proxy to use
- * @param {int} conf.port - TCP port of the proxy
- * @param {string} [conf.protocol='http'] - The protocol of the proxy, 'http' or 'https'
- * @param {string} [conf.proxyAuth] - Credentials for the proxy in the form userId:password
- * @param {string} [conf.connect='https'] - Which protocols will use the CONNECT method 'neither', 'https' or 'both'
- * @param {int} [conf.sockets=5] Maximum number of TCP sockets to use in each pool. There are two different pools for HTTP and HTTPS
- * @param {object} [conf.httpsOptions] - HTTPS options
- */
- globalTunnel.initialize = function(conf) {
- // Don't do anything if already proxying.
- // To change the settings `.end()` should be called first.
- if (globalTunnel.isProxying) {
- log('Already proxying');
- return;
- }
-
- try {
- // This has an effect of also removing the proxy config
- // from the global env to prevent other modules (like request) doing
- // double handling
- var envVarProxy = findEnvVarProxy();
-
- if (conf && typeof conf === 'string') {
- // Passed string - parse it as a URL
- conf = tryParse(conf);
- } else if (conf) {
- // Passed object - take it but clone for future mutations
- conf = clone(conf);
- } else if (envVarProxy) {
- // Nothing passed - parse from the env
- conf = tryParse(envVarProxy);
- } else {
- log('No configuration found, not proxying');
- // No config - do nothing
- return;
- }
-
- log('Proxy configuration to be used is ' + JSON.stringify(conf, null, 2));
-
- if (!conf.host) {
- throw new Error('upstream proxy host is required');
- }
- if (!conf.port) {
- throw new Error('upstream proxy port is required');
- }
-
- if (conf.protocol === undefined) {
- conf.protocol = 'http:'; // Default to proxy speaking http
- }
- if (!/:$/.test(conf.protocol)) {
- conf.protocol += ':';
- }
-
- if (!conf.connect) {
- conf.connect = 'https'; // Just HTTPS by default
- }
-
- if (['both', 'neither', 'https'].indexOf(conf.connect) < 0) {
- throw new Error('valid connect options are "neither", "https", or "both"');
- }
-
- var connectHttp = conf.connect === 'both';
- var connectHttps = conf.connect !== 'neither';
-
- if (conf.httpsOptions) {
- conf.innerHttpsOpts = conf.httpsOptions;
- conf.outerHttpsOpts = conf.innerHttpsOpts;
- }
-
- http.globalAgent = globalTunnel._makeAgent(conf, 'http', connectHttp);
- https.globalAgent = globalTunnel._makeAgent(conf, 'https', connectHttps);
-
- http.request = globalTunnel._makeHttp('request', http, 'http');
- https.request = globalTunnel._makeHttp('request', https, 'https');
- http.get = globalTunnel._makeHttp('get', http, 'http');
- https.get = globalTunnel._makeHttp('get', https, 'https');
-
- globalTunnel.isProxying = true;
- globalTunnel.proxyUrl = stringifyProxy(conf);
- globalTunnel.proxyConfig = clone(conf);
- } catch (e) {
- resetGlobals();
- throw e;
- }
- };
-
- var _makeAgent = function(conf, innerProtocol, useCONNECT) {
- log('Creating proxying agent');
- var outerProtocol = conf.protocol;
- innerProtocol += ':';
-
- var opts = {
- proxy: pick(conf, 'host', 'port', 'protocol', 'localAddress', 'proxyAuth'),
- maxSockets: conf.sockets
- };
- opts.proxy.innerProtocol = innerProtocol;
-
- if (useCONNECT) {
- if (conf.proxyHttpsOptions) {
- assign(opts.proxy, conf.proxyHttpsOptions);
- }
- if (conf.originHttpsOptions) {
- assign(opts, conf.originHttpsOptions);
- }
-
- if (outerProtocol === 'https:') {
- if (innerProtocol === 'https:') {
- return tunnel.httpsOverHttps(opts);
- }
- return tunnel.httpOverHttps(opts);
- }
- if (innerProtocol === 'https:') {
- return tunnel.httpsOverHttp(opts);
- }
- return tunnel.httpOverHttp(opts);
- }
- if (conf.originHttpsOptions) {
- throw new Error('originHttpsOptions must be combined with a tunnel:true option');
- }
- if (conf.proxyHttpsOptions) {
- // NB: not opts.
- assign(opts, conf.proxyHttpsOptions);
- }
-
- if (outerProtocol === 'https:') {
- return new agents.OuterHttpsAgent(opts);
- }
- return new agents.OuterHttpAgent(opts);
- };
-
- /**
- * Construct an agent based on:
- * - is the connection to the proxy secure?
- * - is the connection to the origin secure?
- * - the address of the proxy
- */
- globalTunnel._makeAgent = function(conf, innerProtocol, useCONNECT) {
- var agent = _makeAgent(conf, innerProtocol, useCONNECT);
- // Set the protocol to match that of the target request type
- agent.protocol = innerProtocol + ':';
- return agent;
- };
-
- /**
- * Override for http/https, makes sure to default the agent
- * to the global agent. Due to how node implements it in lib/http.js, the
- * globalAgent we define won't get used (node uses a module-scoped variable,
- * not the exports field).
- * @param {string} method 'request' or 'get', http/https methods
- * @param {string|object} options http/https request url or options
- * @param {function} [cb]
- * @private
- */
- globalTunnel._makeHttp = function(method, httpOrHttps, protocol) {
- return function(options, callback) {
- if (typeof options === 'string') {
- options = urlParse(options);
- } else {
- options = clone(options);
- }
-
- // Respect the default agent provided by node's lib/https.js
- if (
- (options.agent === null || options.agent === undefined) &&
- typeof options.createConnection !== 'function' &&
- (options.host || options.hostname)
- ) {
- options.agent = options._defaultAgent || httpOrHttps.globalAgent;
- }
-
- // Set the default port ourselves to prevent Node doing it based on the proxy agent protocol
- if (options.protocol === 'https:' || (!options.protocol && protocol === 'https')) {
- options.port = options.port || 443;
- }
- if (options.protocol === 'http:' || (!options.protocol && protocol === 'http')) {
- options.port = options.port || 80;
- }
-
- log(
- 'Requesting to ' +
- (options.protocol || protocol) +
- '//' +
- (options.host || options.hostname) +
- ':' +
- options.port
- );
-
- return ORIGINALS[protocol][method].call(httpOrHttps, options, callback);
- };
- };
-
- /**
- * Restores global http/https agents.
- */
- globalTunnel.end = function() {
- resetGlobals();
- globalTunnel.isProxying = false;
- globalTunnel.proxyUrl = null;
- globalTunnel.proxyConfig = null;
- };
|