fix(p2p): disconnect on protocol bad handshake on server side

This commit is contained in:
Alexey Kasyanchuk 2018-03-14 14:01:19 +03:00
parent 13bc8d1d51
commit 349a01557e

View File

@ -15,6 +15,9 @@ class p2p {
this.tcpServer = net.createServer(); this.tcpServer = net.createServer();
this.tcpServer.maxConnections = config.p2pConnections * 2; this.tcpServer.maxConnections = config.p2pConnections * 2;
this.tcpServer.on('connection', (socket) => { this.tcpServer.on('connection', (socket) => {
this.tcpServer.getConnections((err,con) => {
console.log('server connected', con, 'max', this.tcpServer.maxConnections)
})
socket = new JsonSocket(socket); socket = new JsonSocket(socket);
socket.on('error', (err) => {}) socket.on('error', (err) => {})
socket.on('message', (message) => { socket.on('message', (message) => {
@ -32,12 +35,15 @@ class p2p {
}, socket) }, socket)
} }
}); });
socket.protocolTimeout = setTimeout(() => socket._socket.destroy(), 7000)
}) })
// check protocol // check protocol
this.on('protocol', (data, callback, socketObject) => { this.on('protocol', (data, callback, socketObject) => {
if(!data || data.protocol != 'rats') if(!data || data.protocol != 'rats')
return return
// protocol ok
clearTimeout(socketObject.protocolTimeout)
const { _socket: socket } = socketObject const { _socket: socket } = socketObject
socketObject.rats = true socketObject.rats = true
@ -188,6 +194,8 @@ class p2p {
{ {
this.size--; this.size--;
this.send('peer', this.size) this.send('peer', this.size)
// trying reconnect once
setTimeout(() => this.add(this.addr(address)), 5000)
} }
this.peers.splice(index, 1); this.peers.splice(index, 1);
@ -221,6 +229,11 @@ class p2p {
return peers.map(peer => ({address: peer.address, port: peer.port})) return peers.map(peer => ({address: peer.address, port: peer.port}))
} }
addr(peer)
{
return {address: peer.address, port: peer.port}
}
find(peer) find(peer)
{ {
return this.peersList().find((localPeer) => { return this.peersList().find((localPeer) => {