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
"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
|