node-mitmproxy1/lib/mitmproxy/createUpgradeHandler.js

Node.js MITM Proxy

'use strict';

var http = require('http');
var https = require('https');
var util = require('../common/util');

// copy from node-http-proxy.  ^_^

// create connectHandler function
module.exports = function createUpgradeHandler() {

    // return
    return function upgradeHandler(req, cltSocket, head, ssl) {
        var clientOptions = util.getOptionsFormRequest(req, ssl);
        var proxyReq = (ssl ? https : http).request(clientOptions);
        proxyReq.on('error', function (e) {
            console.error(e);
        });
        proxyReq.on('response', function (res) {
            // if upgrade event isn't going to happen, close the socket
            if (!res.upgrade) cltSocket.end();
        });

        proxyReq.on('upgrade', function (proxyRes, proxySocket, proxyHead) {
            proxySocket.on('error', function (e) {
                console.error(e);
            });

            cltSocket.on('error', function () {
                proxySocket.end();
            });

            proxySocket.setTimeout(0);
            proxySocket.setNoDelay(true);

            proxySocket.setKeepAlive(true, 0);

            if (proxyHead && proxyHead.length) proxySocket.unshift(proxyHead);

            cltSocket.write(Object.keys(proxyRes.headers).reduce(function (head, key) {
                var value = proxyRes.headers[key];

                if (!Array.isArray(value)) {
                    head.push(key + ': ' + value);
                    return head;
                }

                for (var i = 0; i < value.length; i++) {
                    head.push(key + ': ' + value[i]);
                }
                return head;
            }, ['HTTP/1.1 101 Switching Protocols']).join('\r\n') + '\r\n\r\n');

            proxySocket.pipe(cltSocket).pipe(proxySocket);
        });
        proxyReq.end();
    };
};