From 5d40faf39e7b9781a7e5bf8ff8c2fe4b3324e38b Mon Sep 17 00:00:00 2001
From: afrokick <devbyru@gmail.com>
Date: Mon, 1 Apr 2019 15:23:45 +0300
Subject: [PATCH] fix src of null

---
 src/messageHandler/index.js           | 2 +-
 src/services/webSocketServer/index.js | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/messageHandler/index.js b/src/messageHandler/index.js
index edf0f7c..cd85653 100644
--- a/src/messageHandler/index.js
+++ b/src/messageHandler/index.js
@@ -24,7 +24,7 @@ module.exports = (client, message) => {
 const handleTransmission = (client, message) => {
   transmissionHandler(client, {
     type: message.type,
-    src: client.getId(),
+    src: message.src,
     dst: message.dst,
     payload: message.payload
   });
diff --git a/src/services/webSocketServer/index.js b/src/services/webSocketServer/index.js
index 410df3a..93db3cb 100644
--- a/src/services/webSocketServer/index.js
+++ b/src/services/webSocketServer/index.js
@@ -130,6 +130,8 @@ class WebSocketServer extends EventEmitter {
       try {
         const message = JSON.parse(data);
 
+        message.src = client.getId();
+
         this.emit('message', client, message);
       } catch (e) {
         logger.error('Invalid message', data);