1 | 'use strict';
|
2 |
|
3 | const { Agent: HttpAgent } = require('http');
|
4 | const { Agent: HttpsAgent } = require('https');
|
5 | const { connect: tlsConnect } = require('tls');
|
6 |
|
7 | let Client;
|
8 |
|
9 | for (const ctor of [HttpAgent, HttpsAgent]) {
|
10 | class SSHAgent extends ctor {
|
11 | constructor(connectCfg, agentOptions) {
|
12 | super(agentOptions);
|
13 |
|
14 | this._connectCfg = connectCfg;
|
15 | this._defaultSrcIP = (agentOptions && agentOptions.srcIP) || 'localhost';
|
16 | }
|
17 |
|
18 | createConnection(options, cb) {
|
19 | const srcIP = (options && options.localAddress) || this._defaultSrcIP;
|
20 | const srcPort = (options && options.localPort) || 0;
|
21 | const dstIP = options.host;
|
22 | const dstPort = options.port;
|
23 |
|
24 | if (Client === undefined)
|
25 | Client = require('./client.js');
|
26 |
|
27 | const client = new Client();
|
28 | let triedForward = false;
|
29 | client.on('ready', () => {
|
30 | client.forwardOut(srcIP, srcPort, dstIP, dstPort, (err, stream) => {
|
31 | triedForward = true;
|
32 | if (err) {
|
33 | client.end();
|
34 | return cb(err);
|
35 | }
|
36 | stream.once('close', () => client.end());
|
37 | cb(null, decorateStream(stream, ctor, options));
|
38 | });
|
39 | }).on('error', cb).on('close', () => {
|
40 | if (!triedForward)
|
41 | cb(new Error('Unexpected connection close'));
|
42 | }).connect(this._connectCfg);
|
43 | }
|
44 | }
|
45 |
|
46 | exports[ctor === HttpAgent ? 'SSHTTPAgent' : 'SSHTTPSAgent'] = SSHAgent;
|
47 | }
|
48 |
|
49 | function noop() {}
|
50 |
|
51 | function decorateStream(stream, ctor, options) {
|
52 | if (ctor === HttpAgent) {
|
53 |
|
54 | stream.setKeepAlive = noop;
|
55 | stream.setNoDelay = noop;
|
56 | stream.setTimeout = noop;
|
57 | stream.ref = noop;
|
58 | stream.unref = noop;
|
59 | stream.destroySoon = stream.destroy;
|
60 | return stream;
|
61 | }
|
62 |
|
63 |
|
64 | options.socket = stream;
|
65 | const wrapped = tlsConnect(options);
|
66 |
|
67 |
|
68 |
|
69 | const onClose = (() => {
|
70 | let called = false;
|
71 | return () => {
|
72 | if (called)
|
73 | return;
|
74 | called = true;
|
75 | if (stream.isPaused())
|
76 | stream.resume();
|
77 | };
|
78 | })();
|
79 |
|
80 |
|
81 | wrapped.on('end', onClose).on('close', onClose);
|
82 |
|
83 | return wrapped;
|
84 | }
|