JSON validation logic for DT which used in establishing connection

Author: 0nlyoung7

lib/server/channel-server.js

@@ -968,9 +968,6 @@ ChannelServer.prototype.startup = function() {
     }
 
     if (_mode == 'CHANNEL_ONLY') { // without session socket Server.
-      console.log(ChannelServer.proc);
-      console.log(this.proc);
-      console.log(self.proc);
       var _us = self.channels[_app + '^' + _channel];
 
       if (!_us) {
@@ -1167,6 +1164,14 @@ ChannelServer.prototype.startup = function() {
 
     socket.join(_room);
 
+    // DT 
+    var err = serverUtils.validJsonParams(socket.handshake.query, ['DT']);
+    if (err) {
+      socket.emit( "connect_error", err );
+      socket.disconnect();
+      return;
+    }
+
     socket._userId = socket.handshake.query.U;
     socket._deviceId = socket.handshake.query.D;
 

lib/server/utils.js

@@ -48,6 +48,56 @@ exports.validSocketParams = function(params, paramArray) {
   return false;
 };
 
+exports.validJsonParams = function(params, paramArray) {
+  for (var i in paramArray) {
+    var param = params[paramArray[i]];
+
+    if ( param && typeof param == 'object') {
+      return false;
+    } else if ( param && typeof param == 'string' ){
+
+      var json = parseJson( param );
+
+      if( !json ) {
+        return {
+          status: 'error',
+          message: '[' + paramArray[i] + '] must be JSON format'
+        };
+      }
+
+      return false;
+    } else if( param ) {
+      return {
+        status: 'error',
+        message: '[' + paramArray[i] + '] must be JSON format'
+      };
+    }
+  }
+
+  return false;
+};
+
+var parseJson = function( instance ){
+  var json;
+  try{
+
+    json = JSON.parse( instance );
+
+    if( typeof json == 'string' ){
+      json = parseJson( json );
+    }
+
+  } catch(err){
+    json = null;
+  }
+
+  return json;
+};
+
+exports.parseJson = function( instance ){
+  return parseJson( instance );
+};
+
 exports.regExpEscape = function(s) {
   return s.replace(/[-\/\\^$*+?.()|[\]{}]/g, '\\$&');
 };