diff --git a/src/background/p2p.js b/src/background/p2p.js index 26619620..b02e816b 100644 --- a/src/background/p2p.js +++ b/src/background/p2p.js @@ -19,20 +19,27 @@ class p2p { socket.on('message', (message) => { if(message.type && this.messageHandlers[message.type]) { + // responce only to rats messages + if(message.type != 'protocol' && !socket.rats) + return + this.messageHandlers[message.type](message.data, (data) => { socket.sendMessage({ id: message.id, data }); - }, socket._socket) + }, socket) } }); }) // check protocol - this.on('protocol', (data, callback, socket) => { + this.on('protocol', (data, callback, socketObject) => { if(!data || data.protocol != 'rats') return + const { _socket: socket } = socketObject + socketObject.rats = true + callback({ protocol: 'rats', peers: this.peersList().slice(0, 4).map(peer => ({address: peer.address, port: peer.port}))