goforward/Client.js at master · WeaselScience/goforward · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
import tls from 'tls';
import net from 'net';
import EventEmitter from 'events';
import debug from 'debug';
import ControlClient from './ControlClient';
import find from 'lodash/find';

const log = debug('fwdizer:Client');

export default class Client extends EventEmitter {
    expose({
        localPort,
        externalPort,
        callback = () => {}
    }) {
        log(`Exposing local port ${localPort} on remote port ${externalPort}`);

        return new Promise((resolve, reject) => {
            this.control.emit('createRouter', {
                externalPort
            }, ({
                error,
                internalPort
            }) => {
                if (error) {
                    log(`createRouter error: ${error.toString()}`);

                    reject(error);
                } else {
                    log(`createRouter success ${externalPort} - ${
                        internalPort
                    } - ${localPort}`);

                    this.exposures.push({
                        externalPort,
                        internalPort,
                        localPort
                    });

                    resolve({
                        internalPort
                    });
                }
            });
        });
    }

    constructor({
        host,
        port,
        tlsConfig
    }) {
        super();


        this.exposures = [];


        log('Creating');


        const control = this.control = ControlClient({
            host,
            port,
            tlsConfig
        });

        control.on('requestTunnel', ({
            externalPort,
            internalPort
        }) => {
            log('Processing a request for a new tunnel');

            const exposure = find(this.exposures, {
                externalPort
            });

            const localSocket = net.createConnection(exposure.localPort);

            const remoteSocket = tls.connect({
                ...tlsConfig,
                rejectUnauthorized: true,
                host,
                port: exposure.internalPort
            });

            localSocket.pipe(remoteSocket);
            remoteSocket.pipe(localSocket);
        });
    }

    selfDestruct() {
        log('Self-destructing');

        this.emit('dead');
    }
}