single-tls-tunnel/test/support/MockClient.js

A single port TLS tunnel implementation to support a single downstream client proxy

var http = require('http'),
    crypto = require('crypto'),
    tls = require('tls'),
    MultiplexStream = require('multiplex-stream'),
    util = require('util'),
    EventEmitter = require('events').EventEmitter;

function MockClient(options) {
  var self = this,
      connection,
      socket;

  self.connect = function(callback){
    var request = http.request({
      port: options.port,
      headers: {
        'Connection': 'Upgrade',
        'Upgrade': 'TLS'
      }
    });
    request.on('upgrade', function(res, sock, upgradeHead) {
      socket = sock;
      var securePair = tls.createSecurePair(
       crypto.createCredentials({
         key: options.key,
         cert: options.cert,
         ca: options.ca
       }),
       false,
       true,
       options.rejectUnauthorized
      );
      connection = securePair.cleartext;

      socket.pipe(securePair.encrypted).pipe(socket);

      securePair.on('secure', function() {
        self.multiplex = new MultiplexStream();
        connection.pipe(self.multiplex).pipe(connection);
        callback();
      });
    });
    request.end();
  };

  self.end = function() {
    connection.on('end', function() {
      self.emit('end');
    });
    connection.end();
  };

  self.destroy = function() {
    socket.destroy();
  };
}
util.inherits(MockClient, EventEmitter);

module.exports = MockClient;