nodejs/test/parallel/test-tls-socket-close.js

'use strict';
const common = require('../common');
if (!common.hasCrypto)
  common.skip('missing crypto');

const assert = require('assert');
const tls = require('tls');
const net = require('net');
const Countdown = require('../common/countdown');
const fixtures = require('../common/fixtures');

const key = fixtures.readKey('agent2-key.pem');
const cert = fixtures.readKey('agent2-cert.pem');

let serverTlsSocket;
const tlsServer = tls.createServer({ cert, key }, (socket) => {
  serverTlsSocket = socket;
  socket.on('data', (chunk) => {
    assert.strictEqual(chunk[0], 46);
    socket.write('.');
  });
  socket.on('close', dec);
});

// A plain net server, that manually passes connections to the TLS
// server to be upgraded.
let netSocket;
let netSocketCloseEmitted = false;
const netServer = net.createServer((socket) => {
  netSocket = socket;
  tlsServer.emit('connection', socket);
  socket.on('close', common.mustCall(() => {
    netSocketCloseEmitted = true;
    assert.strictEqual(serverTlsSocket.destroyed, true);
  }));
}).listen(0, common.mustCall(() => {
  connectClient(netServer);
}));

const countdown = new Countdown(2, () => {
  netServer.close();
});

// A client that connects, sends one message, and closes the raw connection:
function connectClient(server) {
  const clientTlsSocket = tls.connect({
    host: 'localhost',
    port: server.address().port,
    rejectUnauthorized: false
  });

  clientTlsSocket.write('.');

  clientTlsSocket.on('data', (chunk) => {
    assert.strictEqual(chunk[0], 46);

    netSocket.destroy();
    assert.strictEqual(netSocket.destroyed, true);

    setImmediate(() => {
      // Close callbacks are executed after `setImmediate()` callbacks.
      assert.strictEqual(netSocketCloseEmitted, false);
      assert.strictEqual(serverTlsSocket.destroyed, false);
    });
  });

  clientTlsSocket.on('close', dec);
}

function dec() {
  countdown.dec();
}