You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
					145 lines
				
				6.4 KiB
			
		
		
			
		
	
	
					145 lines
				
				6.4 KiB
			| 
								 
											3 years ago
										 
									 | 
							
								"use strict";
							 | 
						||
| 
								 | 
							
								var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
							 | 
						||
| 
								 | 
							
								    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
							 | 
						||
| 
								 | 
							
								    return new (P || (P = Promise))(function (resolve, reject) {
							 | 
						||
| 
								 | 
							
								        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
							 | 
						||
| 
								 | 
							
								        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
							 | 
						||
| 
								 | 
							
								        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
							 | 
						||
| 
								 | 
							
								        step((generator = generator.apply(thisArg, _arguments || [])).next());
							 | 
						||
| 
								 | 
							
								    });
							 | 
						||
| 
								 | 
							
								};
							 | 
						||
| 
								 | 
							
								var __importDefault = (this && this.__importDefault) || function (mod) {
							 | 
						||
| 
								 | 
							
								    return (mod && mod.__esModule) ? mod : { "default": mod };
							 | 
						||
| 
								 | 
							
								};
							 | 
						||
| 
								 | 
							
								Object.defineProperty(exports, "__esModule", { value: true });
							 | 
						||
| 
								 | 
							
								const net_1 = __importDefault(require("net"));
							 | 
						||
| 
								 | 
							
								const tls_1 = __importDefault(require("tls"));
							 | 
						||
| 
								 | 
							
								const url_1 = __importDefault(require("url"));
							 | 
						||
| 
								 | 
							
								const debug_1 = __importDefault(require("debug"));
							 | 
						||
| 
								 | 
							
								const once_1 = __importDefault(require("@tootallnate/once"));
							 | 
						||
| 
								 | 
							
								const agent_base_1 = require("agent-base");
							 | 
						||
| 
								 | 
							
								const debug = debug_1.default('http-proxy-agent');
							 | 
						||
| 
								 | 
							
								function isHTTPS(protocol) {
							 | 
						||
| 
								 | 
							
								    return typeof protocol === 'string' ? /^https:?$/i.test(protocol) : false;
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								/**
							 | 
						||
| 
								 | 
							
								 * The `HttpProxyAgent` implements an HTTP Agent subclass that connects
							 | 
						||
| 
								 | 
							
								 * to the specified "HTTP proxy server" in order to proxy HTTP requests.
							 | 
						||
| 
								 | 
							
								 *
							 | 
						||
| 
								 | 
							
								 * @api public
							 | 
						||
| 
								 | 
							
								 */
							 | 
						||
| 
								 | 
							
								class HttpProxyAgent extends agent_base_1.Agent {
							 | 
						||
| 
								 | 
							
								    constructor(_opts) {
							 | 
						||
| 
								 | 
							
								        let opts;
							 | 
						||
| 
								 | 
							
								        if (typeof _opts === 'string') {
							 | 
						||
| 
								 | 
							
								            opts = url_1.default.parse(_opts);
							 | 
						||
| 
								 | 
							
								        }
							 | 
						||
| 
								 | 
							
								        else {
							 | 
						||
| 
								 | 
							
								            opts = _opts;
							 | 
						||
| 
								 | 
							
								        }
							 | 
						||
| 
								 | 
							
								        if (!opts) {
							 | 
						||
| 
								 | 
							
								            throw new Error('an HTTP(S) proxy server `host` and `port` must be specified!');
							 | 
						||
| 
								 | 
							
								        }
							 | 
						||
| 
								 | 
							
								        debug('Creating new HttpProxyAgent instance: %o', opts);
							 | 
						||
| 
								 | 
							
								        super(opts);
							 | 
						||
| 
								 | 
							
								        const proxy = Object.assign({}, opts);
							 | 
						||
| 
								 | 
							
								        // If `true`, then connect to the proxy server over TLS.
							 | 
						||
| 
								 | 
							
								        // Defaults to `false`.
							 | 
						||
| 
								 | 
							
								        this.secureProxy = opts.secureProxy || isHTTPS(proxy.protocol);
							 | 
						||
| 
								 | 
							
								        // Prefer `hostname` over `host`, and set the `port` if needed.
							 | 
						||
| 
								 | 
							
								        proxy.host = proxy.hostname || proxy.host;
							 | 
						||
| 
								 | 
							
								        if (typeof proxy.port === 'string') {
							 | 
						||
| 
								 | 
							
								            proxy.port = parseInt(proxy.port, 10);
							 | 
						||
| 
								 | 
							
								        }
							 | 
						||
| 
								 | 
							
								        if (!proxy.port && proxy.host) {
							 | 
						||
| 
								 | 
							
								            proxy.port = this.secureProxy ? 443 : 80;
							 | 
						||
| 
								 | 
							
								        }
							 | 
						||
| 
								 | 
							
								        if (proxy.host && proxy.path) {
							 | 
						||
| 
								 | 
							
								            // If both a `host` and `path` are specified then it's most likely
							 | 
						||
| 
								 | 
							
								            // the result of a `url.parse()` call... we need to remove the
							 | 
						||
| 
								 | 
							
								            // `path` portion so that `net.connect()` doesn't attempt to open
							 | 
						||
| 
								 | 
							
								            // that as a Unix socket file.
							 | 
						||
| 
								 | 
							
								            delete proxy.path;
							 | 
						||
| 
								 | 
							
								            delete proxy.pathname;
							 | 
						||
| 
								 | 
							
								        }
							 | 
						||
| 
								 | 
							
								        this.proxy = proxy;
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								    /**
							 | 
						||
| 
								 | 
							
								     * Called when the node-core HTTP client library is creating a
							 | 
						||
| 
								 | 
							
								     * new HTTP request.
							 | 
						||
| 
								 | 
							
								     *
							 | 
						||
| 
								 | 
							
								     * @api protected
							 | 
						||
| 
								 | 
							
								     */
							 | 
						||
| 
								 | 
							
								    callback(req, opts) {
							 | 
						||
| 
								 | 
							
								        return __awaiter(this, void 0, void 0, function* () {
							 | 
						||
| 
								 | 
							
								            const { proxy, secureProxy } = this;
							 | 
						||
| 
								 | 
							
								            const parsed = url_1.default.parse(req.path);
							 | 
						||
| 
								 | 
							
								            if (!parsed.protocol) {
							 | 
						||
| 
								 | 
							
								                parsed.protocol = 'http:';
							 | 
						||
| 
								 | 
							
								            }
							 | 
						||
| 
								 | 
							
								            if (!parsed.hostname) {
							 | 
						||
| 
								 | 
							
								                parsed.hostname = opts.hostname || opts.host || null;
							 | 
						||
| 
								 | 
							
								            }
							 | 
						||
| 
								 | 
							
								            if (parsed.port == null && typeof opts.port) {
							 | 
						||
| 
								 | 
							
								                parsed.port = String(opts.port);
							 | 
						||
| 
								 | 
							
								            }
							 | 
						||
| 
								 | 
							
								            if (parsed.port === '80') {
							 | 
						||
| 
								 | 
							
								                // if port is 80, then we can remove the port so that the
							 | 
						||
| 
								 | 
							
								                // ":80" portion is not on the produced URL
							 | 
						||
| 
								 | 
							
								                delete parsed.port;
							 | 
						||
| 
								 | 
							
								            }
							 | 
						||
| 
								 | 
							
								            // Change the `http.ClientRequest` instance's "path" field
							 | 
						||
| 
								 | 
							
								            // to the absolute path of the URL that will be requested.
							 | 
						||
| 
								 | 
							
								            req.path = url_1.default.format(parsed);
							 | 
						||
| 
								 | 
							
								            // Inject the `Proxy-Authorization` header if necessary.
							 | 
						||
| 
								 | 
							
								            if (proxy.auth) {
							 | 
						||
| 
								 | 
							
								                req.setHeader('Proxy-Authorization', `Basic ${Buffer.from(proxy.auth).toString('base64')}`);
							 | 
						||
| 
								 | 
							
								            }
							 | 
						||
| 
								 | 
							
								            // Create a socket connection to the proxy server.
							 | 
						||
| 
								 | 
							
								            let socket;
							 | 
						||
| 
								 | 
							
								            if (secureProxy) {
							 | 
						||
| 
								 | 
							
								                debug('Creating `tls.Socket`: %o', proxy);
							 | 
						||
| 
								 | 
							
								                socket = tls_1.default.connect(proxy);
							 | 
						||
| 
								 | 
							
								            }
							 | 
						||
| 
								 | 
							
								            else {
							 | 
						||
| 
								 | 
							
								                debug('Creating `net.Socket`: %o', proxy);
							 | 
						||
| 
								 | 
							
								                socket = net_1.default.connect(proxy);
							 | 
						||
| 
								 | 
							
								            }
							 | 
						||
| 
								 | 
							
								            // At this point, the http ClientRequest's internal `_header` field
							 | 
						||
| 
								 | 
							
								            // might have already been set. If this is the case then we'll need
							 | 
						||
| 
								 | 
							
								            // to re-generate the string since we just changed the `req.path`.
							 | 
						||
| 
								 | 
							
								            if (req._header) {
							 | 
						||
| 
								 | 
							
								                let first;
							 | 
						||
| 
								 | 
							
								                let endOfHeaders;
							 | 
						||
| 
								 | 
							
								                debug('Regenerating stored HTTP header string for request');
							 | 
						||
| 
								 | 
							
								                req._header = null;
							 | 
						||
| 
								 | 
							
								                req._implicitHeader();
							 | 
						||
| 
								 | 
							
								                if (req.output && req.output.length > 0) {
							 | 
						||
| 
								 | 
							
								                    // Node < 12
							 | 
						||
| 
								 | 
							
								                    debug('Patching connection write() output buffer with updated header');
							 | 
						||
| 
								 | 
							
								                    first = req.output[0];
							 | 
						||
| 
								 | 
							
								                    endOfHeaders = first.indexOf('\r\n\r\n') + 4;
							 | 
						||
| 
								 | 
							
								                    req.output[0] = req._header + first.substring(endOfHeaders);
							 | 
						||
| 
								 | 
							
								                    debug('Output buffer: %o', req.output);
							 | 
						||
| 
								 | 
							
								                }
							 | 
						||
| 
								 | 
							
								                else if (req.outputData && req.outputData.length > 0) {
							 | 
						||
| 
								 | 
							
								                    // Node >= 12
							 | 
						||
| 
								 | 
							
								                    debug('Patching connection write() output buffer with updated header');
							 | 
						||
| 
								 | 
							
								                    first = req.outputData[0].data;
							 | 
						||
| 
								 | 
							
								                    endOfHeaders = first.indexOf('\r\n\r\n') + 4;
							 | 
						||
| 
								 | 
							
								                    req.outputData[0].data =
							 | 
						||
| 
								 | 
							
								                        req._header + first.substring(endOfHeaders);
							 | 
						||
| 
								 | 
							
								                    debug('Output buffer: %o', req.outputData[0].data);
							 | 
						||
| 
								 | 
							
								                }
							 | 
						||
| 
								 | 
							
								            }
							 | 
						||
| 
								 | 
							
								            // Wait for the socket's `connect` event, so that this `callback()`
							 | 
						||
| 
								 | 
							
								            // function throws instead of the `http` request machinery. This is
							 | 
						||
| 
								 | 
							
								            // important for i.e. `PacProxyAgent` which determines a failed proxy
							 | 
						||
| 
								 | 
							
								            // connection via the `callback()` function throwing.
							 | 
						||
| 
								 | 
							
								            yield once_1.default(socket, 'connect');
							 | 
						||
| 
								 | 
							
								            return socket;
							 | 
						||
| 
								 | 
							
								        });
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								exports.default = HttpProxyAgent;
							 | 
						||
| 
								 | 
							
								//# sourceMappingURL=agent.js.map
							 |