evan/rats-search
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat(secure): responce only to rats messages

Browse Source
This commit is contained in:
Alexey Kasyanchuk 2018-03-03 05:55:14 +03:00
parent c2b030dddc
commit 65354947ba
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/background/p2p.js
View File

@ -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}))