diff --git a/src/transports/ipc.js b/src/transports/ipc.js index 91d1eb2..10f3928 100644 --- a/src/transports/ipc.js +++ b/src/transports/ipc.js @@ -246,6 +246,11 @@ export default class IPCServer { onMessage(socket, msg) { if (process.env.ARRPC_DEBUG) log('message', msg); - this.handlers.message(socket, msg); + if (!msg.args || !msg.cmd) { + log("Invaild payload!") + return + } else { + this.handlers.message(socket, msg); + } } } diff --git a/src/transports/websocket.js b/src/transports/websocket.js index bdd0c36..7f8f57e 100644 --- a/src/transports/websocket.js +++ b/src/transports/websocket.js @@ -115,7 +115,11 @@ export default class WSServer { } onMessage(socket, msg) { - if (process.env.ARRPC_DEBUG) log('message', JSON.parse(msg)); - this.handlers.message(socket, JSON.parse(msg)); + try { + if (process.env.ARRPC_DEBUG) log('message', JSON.parse(msg)); + this.handlers.message(socket, JSON.parse(msg)); + } catch { + log("Invalid Payload!") + } } }