102 lines
1.8 KiB
JavaScript
102 lines
1.8 KiB
JavaScript
/*!
|
|
* tcp.js - tcp backend for bcoin
|
|
* Copyright (c) 2014-2017, Christopher Jeffrey (MIT License).
|
|
* https://github.com/bcoin-org/bcoin
|
|
*/
|
|
|
|
/* eslint prefer-arrow-callback: "off" */
|
|
|
|
'use strict';
|
|
|
|
const EventEmitter = require('events');
|
|
const net = require('net');
|
|
const socks = require('./socks');
|
|
|
|
/**
|
|
* @exports net/tcp
|
|
*/
|
|
|
|
const tcp = exports;
|
|
|
|
/**
|
|
* Create a TCP socket and connect.
|
|
* @param {Number} port
|
|
* @param {String} host
|
|
* @param {String?} proxy
|
|
* @returns {Object}
|
|
*/
|
|
|
|
tcp.createSocket = function createSocket(port, host, proxy) {
|
|
if (proxy)
|
|
return socks.connect(proxy, port, host);
|
|
return net.connect(port, host);
|
|
};
|
|
|
|
/**
|
|
* Create a TCP server.
|
|
* @returns {Object}
|
|
*/
|
|
|
|
tcp.createServer = function createServer() {
|
|
const server = new net.Server();
|
|
const ee = new EventEmitter();
|
|
|
|
ee.listen = function listen(port, host) {
|
|
return new Promise((resolve, reject) => {
|
|
server.once('error', reject);
|
|
server.listen(port, host, () => {
|
|
server.removeListener('error', reject);
|
|
resolve();
|
|
});
|
|
});
|
|
};
|
|
|
|
ee.close = function close() {
|
|
return new Promise((resolve, reject) => {
|
|
server.close(wrap(resolve, reject));
|
|
});
|
|
};
|
|
|
|
ee.address = function address() {
|
|
return server.address();
|
|
};
|
|
|
|
Object.defineProperty(ee, 'maxConnections', {
|
|
get() {
|
|
return server.maxConnections;
|
|
},
|
|
set(value) {
|
|
server.maxConnections = value;
|
|
return server.maxConnections;
|
|
}
|
|
});
|
|
|
|
server.on('listening', () => {
|
|
ee.emit('listening');
|
|
});
|
|
|
|
server.on('connection', (socket) => {
|
|
ee.emit('connection', socket);
|
|
});
|
|
|
|
server.on('error', (err) => {
|
|
ee.emit('error', err);
|
|
});
|
|
|
|
return ee;
|
|
};
|
|
|
|
/*
|
|
* Helpers
|
|
*/
|
|
|
|
function wrap(resolve, reject) {
|
|
return function(err, result) {
|
|
if (err) {
|
|
reject(err);
|
|
return;
|
|
}
|
|
resolve(result);
|
|
};
|
|
}
|