Compare commits

...

3 Commits

2 changed files with 24 additions and 20 deletions

View File

@ -164,17 +164,19 @@ class MultiSocketProxy extends proxy_js_1.default {
decode: this._server ? compact_encoding_1.json.decode : socketEncoding.decode, decode: this._server ? compact_encoding_1.json.decode : socketEncoding.decode,
}, },
async onmessage(m) { async onmessage(m) {
if (self._allowedPorts.length && if (self._server) {
!self._allowedPorts.includes(m.port)) { if (self._allowedPorts.length &&
self.get(await self._getPublicKey(peer)).messages.errorSocket.send({ !self._allowedPorts.includes(m.port)) {
id: m.id, self.get(await self._getPublicKey(peer)).messages.errorSocket.send({
err: new Error(`port ${m.port} not allowed`), id: m.id,
}); err: new Error(`port ${m.port} not allowed`),
return; });
return;
}
} }
m = m; m = m;
if (self._server) { if (self._server) {
new self.socketClass(nextSocketId(), m, self, self.get(await self._getPublicKey(peer)), m).connect(); new self.socketClass(nextSocketId(), m.id, self, self.get(await self._getPublicKey(peer)), m).connect();
return; return;
} }
const socket = self._sockets.get(m.id); const socket = self._sockets.get(m.id);

View File

@ -204,17 +204,19 @@ export default class MultiSocketProxy extends Proxy {
decode: this._server ? json.decode : socketEncoding.decode, decode: this._server ? json.decode : socketEncoding.decode,
}, },
async onmessage(m: SocketRequest | TcpSocketConnectOpts) { async onmessage(m: SocketRequest | TcpSocketConnectOpts) {
if ( if (self._server) {
self._allowedPorts.length && if (
!self._allowedPorts.includes((m as TcpSocketConnectOpts).port) self._allowedPorts.length &&
) { !self._allowedPorts.includes((m as TcpSocketConnectOpts).port)
self.get(await self._getPublicKey(peer)).messages.errorSocket.send({ ) {
id: (m as SocketRequest).id, self.get(await self._getPublicKey(peer)).messages.errorSocket.send({
err: new Error( id: (m as SocketRequest).id,
`port ${(m as TcpSocketConnectOpts).port} not allowed` err: new Error(
), `port ${(m as TcpSocketConnectOpts).port} not allowed`
}); ),
return; });
return;
}
} }
m = m as SocketRequest; m = m as SocketRequest;
@ -222,7 +224,7 @@ export default class MultiSocketProxy extends Proxy {
if (self._server) { if (self._server) {
new self.socketClass( new self.socketClass(
nextSocketId(), nextSocketId(),
m, m.id,
self, self,
self.get(await self._getPublicKey(peer)) as PeerEntity, self.get(await self._getPublicKey(peer)) as PeerEntity,
m m