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.
		
		
		
		
		
			
		
			
				
					180 lines
				
				5.9 KiB
			
		
		
			
		
	
	
					180 lines
				
				5.9 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 dns_1 = __importDefault(require("dns")); | ||
|  | const tls_1 = __importDefault(require("tls")); | ||
|  | const url_1 = __importDefault(require("url")); | ||
|  | const debug_1 = __importDefault(require("debug")); | ||
|  | const agent_base_1 = require("agent-base"); | ||
|  | const socks_1 = require("socks"); | ||
|  | const debug = debug_1.default('socks-proxy-agent'); | ||
|  | function dnsLookup(host) { | ||
|  |     return new Promise((resolve, reject) => { | ||
|  |         dns_1.default.lookup(host, (err, res) => { | ||
|  |             if (err) { | ||
|  |                 reject(err); | ||
|  |             } | ||
|  |             else { | ||
|  |                 resolve(res); | ||
|  |             } | ||
|  |         }); | ||
|  |     }); | ||
|  | } | ||
|  | function parseSocksProxy(opts) { | ||
|  |     let port = 0; | ||
|  |     let lookup = false; | ||
|  |     let type = 5; | ||
|  |     // Prefer `hostname` over `host`, because of `url.parse()`
 | ||
|  |     const host = opts.hostname || opts.host; | ||
|  |     if (!host) { | ||
|  |         throw new TypeError('No "host"'); | ||
|  |     } | ||
|  |     if (typeof opts.port === 'number') { | ||
|  |         port = opts.port; | ||
|  |     } | ||
|  |     else if (typeof opts.port === 'string') { | ||
|  |         port = parseInt(opts.port, 10); | ||
|  |     } | ||
|  |     // From RFC 1928, Section 3: https://tools.ietf.org/html/rfc1928#section-3
 | ||
|  |     // "The SOCKS service is conventionally located on TCP port 1080"
 | ||
|  |     if (!port) { | ||
|  |         port = 1080; | ||
|  |     } | ||
|  |     // figure out if we want socks v4 or v5, based on the "protocol" used.
 | ||
|  |     // Defaults to 5.
 | ||
|  |     if (opts.protocol) { | ||
|  |         switch (opts.protocol.replace(':', '')) { | ||
|  |             case 'socks4': | ||
|  |                 lookup = true; | ||
|  |             // pass through
 | ||
|  |             case 'socks4a': | ||
|  |                 type = 4; | ||
|  |                 break; | ||
|  |             case 'socks5': | ||
|  |                 lookup = true; | ||
|  |             // pass through
 | ||
|  |             case 'socks': // no version specified, default to 5h
 | ||
|  |             case 'socks5h': | ||
|  |                 type = 5; | ||
|  |                 break; | ||
|  |             default: | ||
|  |                 throw new TypeError(`A "socks" protocol must be specified! Got: ${opts.protocol}`); | ||
|  |         } | ||
|  |     } | ||
|  |     if (typeof opts.type !== 'undefined') { | ||
|  |         if (opts.type === 4 || opts.type === 5) { | ||
|  |             type = opts.type; | ||
|  |         } | ||
|  |         else { | ||
|  |             throw new TypeError(`"type" must be 4 or 5, got: ${opts.type}`); | ||
|  |         } | ||
|  |     } | ||
|  |     const proxy = { | ||
|  |         host, | ||
|  |         port, | ||
|  |         type | ||
|  |     }; | ||
|  |     let userId = opts.userId || opts.username; | ||
|  |     let password = opts.password; | ||
|  |     if (opts.auth) { | ||
|  |         const auth = opts.auth.split(':'); | ||
|  |         userId = auth[0]; | ||
|  |         password = auth[1]; | ||
|  |     } | ||
|  |     if (userId) { | ||
|  |         Object.defineProperty(proxy, 'userId', { | ||
|  |             value: userId, | ||
|  |             enumerable: false | ||
|  |         }); | ||
|  |     } | ||
|  |     if (password) { | ||
|  |         Object.defineProperty(proxy, 'password', { | ||
|  |             value: password, | ||
|  |             enumerable: false | ||
|  |         }); | ||
|  |     } | ||
|  |     return { lookup, proxy }; | ||
|  | } | ||
|  | /** | ||
|  |  * The `SocksProxyAgent`. | ||
|  |  * | ||
|  |  * @api public | ||
|  |  */ | ||
|  | class SocksProxyAgent 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 TypeError('a SOCKS proxy server `host` and `port` must be specified!'); | ||
|  |         } | ||
|  |         super(opts); | ||
|  |         const parsedProxy = parseSocksProxy(opts); | ||
|  |         this.lookup = parsedProxy.lookup; | ||
|  |         this.proxy = parsedProxy.proxy; | ||
|  |     } | ||
|  |     /** | ||
|  |      * Initiates a SOCKS connection to the specified SOCKS proxy server, | ||
|  |      * which in turn connects to the specified remote host and port. | ||
|  |      * | ||
|  |      * @api protected | ||
|  |      */ | ||
|  |     callback(req, opts) { | ||
|  |         return __awaiter(this, void 0, void 0, function* () { | ||
|  |             const { lookup, proxy } = this; | ||
|  |             let { host, port, timeout } = opts; | ||
|  |             if (!host) { | ||
|  |                 throw new Error('No `host` defined!'); | ||
|  |             } | ||
|  |             if (lookup) { | ||
|  |                 // Client-side DNS resolution for "4" and "5" socks proxy versions.
 | ||
|  |                 host = yield dnsLookup(host); | ||
|  |             } | ||
|  |             const socksOpts = { | ||
|  |                 proxy, | ||
|  |                 destination: { host, port }, | ||
|  |                 command: 'connect', | ||
|  |                 timeout | ||
|  |             }; | ||
|  |             debug('Creating socks proxy connection: %o', socksOpts); | ||
|  |             const { socket } = yield socks_1.SocksClient.createConnection(socksOpts); | ||
|  |             debug('Successfully created socks proxy connection'); | ||
|  |             if (opts.secureEndpoint) { | ||
|  |                 // The proxy is connecting to a TLS server, so upgrade
 | ||
|  |                 // this socket connection to a TLS connection.
 | ||
|  |                 debug('Upgrading socket connection to TLS'); | ||
|  |                 const servername = opts.servername || host; | ||
|  |                 return tls_1.default.connect(Object.assign(Object.assign({}, omit(opts, 'host', 'hostname', 'path', 'port')), { socket, | ||
|  |                     servername })); | ||
|  |             } | ||
|  |             return socket; | ||
|  |         }); | ||
|  |     } | ||
|  | } | ||
|  | exports.default = SocksProxyAgent; | ||
|  | function omit(obj, ...keys) { | ||
|  |     const ret = {}; | ||
|  |     let key; | ||
|  |     for (key in obj) { | ||
|  |         if (!keys.includes(key)) { | ||
|  |             ret[key] = obj[key]; | ||
|  |         } | ||
|  |     } | ||
|  |     return ret; | ||
|  | } | ||
|  | //# sourceMappingURL=agent.js.map
 |