You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
54 lines
1.3 KiB
54 lines
1.3 KiB
'use strict'
|
|
const net = require('net');
|
|
const tls = require('tls');
|
|
const TlsSocketWrapper = require('./TlsSocketWrapper.js');
|
|
const TIMEOUT = 5000
|
|
|
|
const getSocket = (protocol, options) => {
|
|
switch(protocol){
|
|
case 'tcp':
|
|
return new net.Socket();
|
|
case 'tls':
|
|
case 'ssl':
|
|
if (!tls) throw new Error('tls package could not be loaded');
|
|
return new TlsSocketWrapper({ tls, verbose: false });
|
|
}
|
|
throw new Error('unknown protocol')
|
|
}
|
|
|
|
const initSocket = (self, protocol, options) => {
|
|
const conn = getSocket(protocol, options);
|
|
conn.setTimeout(TIMEOUT)
|
|
conn.setEncoding('utf8')
|
|
conn.setKeepAlive(true, 0)
|
|
conn.setNoDelay(true)
|
|
conn.on('connect', () => {
|
|
conn.setTimeout(0)
|
|
self.onConnect()
|
|
})
|
|
conn.on('close', (e) => {
|
|
self.onClose(e)
|
|
})
|
|
conn.on('timeout', () => {
|
|
//self.onClose(e);
|
|
/*const e = new Error('ETIMEDOUT')
|
|
e.errorno = 'ETIMEDOUT'
|
|
e.code = 'ETIMEDOUT'
|
|
e.connect = false
|
|
conn.emit('error', e)*/
|
|
})
|
|
conn.on('data', (chunk) => {
|
|
conn.setTimeout(0)
|
|
self.onRecv(chunk)
|
|
})
|
|
conn.on('end', (e) => {
|
|
conn.setTimeout(0)
|
|
self.onEnd(e)
|
|
})
|
|
conn.on('error', (e) => {
|
|
self.onError(e)
|
|
})
|
|
return conn
|
|
}
|
|
|
|
module.exports = initSocket
|
|
|