wechat wss demo

wss-server-4.js 7.7KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377
  1. 'use strict';
  2. const encoding = require("encoding");
  3. const zlib = require('zlib');
  4. function stringToArray(bufferString) {
  5. // let uint8Array = new TextEncoder("utf-8").encode(bufferString);
  6. //var resultBuffer = encoding.convert(nameString, 'ASCII', 'UTF-8');
  7. var uint8Array = encoding.convert(bufferString, 'ASCII', 'UTF-8');
  8. return uint8Array;
  9. }
  10. function intToUint8Array(value) {
  11. console.log('value:' + value);
  12. var arr = new Uint8Array(2);
  13. arr[0] = (value >> 8);
  14. arr[1] = value % 0x100;
  15. return arr;
  16. }
  17. function intToUint8ArrayCRC(value) {
  18. console.log('value:' + value);
  19. var arr = new Uint8Array(2);
  20. arr[0] = value % 0x100;
  21. arr[1] = (value >> 8);
  22. return arr;
  23. }
  24. function calc_crc16(data, len) {
  25. var crc = 0xffff;
  26. var LSB;
  27. for (var i = 0; i < len; i++) {
  28. crc ^= data[i];
  29. for (var j = 0; j < 8; j++) {
  30. LSB = crc & 1;
  31. crc = crc >> 1;
  32. if (LSB != 0) crc ^= 0xa001;
  33. }
  34. }
  35. return crc;
  36. }
  37. //wss start
  38. var fs = require('fs');
  39. var cfg = {
  40. ssl: true,
  41. port: 443,
  42. ssl_key: 'ssl.key', ssl_cert: 'ssl.crt'
  43. };
  44. var httpServ = (cfg.ssl) ? require('https') : require('http');
  45. const WebSocketServer = require('ws').Server;
  46. var app = null;
  47. var processRequest = function(req, res)
  48. {
  49. res.writeHead(200);
  50. res.end('All glory to WebSockets!\n');
  51. };
  52. if (cfg.ssl) {
  53. app = httpServ.createServer({
  54. key: fs.readFileSync(cfg.ssl_key),
  55. cert: fs.readFileSync(cfg.ssl_cert)
  56. }, processRequest).listen(cfg.port);
  57. } else {
  58. app = httpServ.createServer(processRequest).listen(cfg.port);
  59. }
  60. const wss = new WebSocketServer({ server: app });
  61. wss.on('connection', function(wsConnect)
  62. {
  63. //0
  64. //
  65. //mqtt
  66. var mqtt = require('mqtt');
  67. var opt = {
  68. port:1883,
  69. clientId: 'phonetest'
  70. };
  71. function uint8ArrayToInt(arr) {
  72. console.log('arr[0]:' + arr[0]);
  73. console.log('arr[1]:' + arr[1]);
  74. return arr[0] * 0x100 + arr[1];
  75. }
  76. var client = mqtt.connect('tcp://iot.twgigatech.cn:1883');
  77. client.on('connect', function ()
  78. {
  79. console.log('connect MQTT server');
  80. //client.subscribe("CAD9A3D3EUWWNBKY111A/iotk");
  81. client.subscribe("CAD9A3D3EUWWNBKY111A/wxxk/#");
  82. client.subscribe("CAD9A3D3EUWWNBKY111A/wxxm/#");
  83. /*
  84. client.subscribe("CAD9A3D3EUWWNBKY111A/wxxm/5");
  85. client.subscribe("CAD9A3D3EUWWNBKY111A/wxxm/6");
  86. client.subscribe("CAD9A3D3EUWWNBKY111A/wxxm/7");
  87. client.subscribe("CAD9A3D3EUWWNBKY111A/wxxm/4");
  88. client.subscribe("CAD9A3D3EUWWNBKY111A/wxxm/3");
  89. client.subscribe("CAD9A3D3EUWWNBKY111A/wxxm/2");
  90. client.subscribe("CAD9A3D3EUWWNBKY111A/wxxm/1");
  91. client.subscribe("CAD9A3D3EUWWNBKY111A/wxxm/8");
  92. */
  93. // client.publish('CAD9A3D3EUWWNBKY/iotk','test');
  94. });
  95. client.on('message', function (topic, message) {
  96. console.log(' ');
  97. console.log(' topic = '+topic.toString());
  98. console.log('==============mqtt rece start ');
  99. //zlib.inflate(message, function(err, buffer) {
  100. /*zlib.unzip(message, function(err, buffer) {
  101. if (!err)
  102. {
  103. console.log('mqtt unzip: '+buffer.toString('base64'));
  104. }
  105. });
  106. */
  107. console.log('mqtt rece: '+message.toString());
  108. console.log('=======================');
  109. console.log('messagelen:' + message.length.toString());
  110. let lenBin = message.subarray(26, 28);
  111. let len = uint8ArrayToInt(lenBin);
  112. console.log('len:' + len);
  113. let blockData = message.subarray(32, message.length);
  114. console.log('blockData:' + blockData);
  115. let dataType = blockData[0];
  116. console.log('dataType:' + dataType);
  117. var rawData = blockData.subarray(25, len-2);
  118. console.log('rawData:' + rawData);
  119. console.log('rawData len:' + rawData.length.toString());
  120. var strData;
  121. if(dataType == 0)
  122. {
  123. strData = arrayToString(rawData);
  124. } else if (dataType == 1)
  125. {
  126. // strData = zlib.inflate(rawData, function(err, buffer)
  127. zlib.inflate(rawData, function(err, buffer)
  128. {
  129. if (!err)
  130. {
  131. console.log('buffer= '+buffer);
  132. // buffer.length.toString();
  133. console.log('mqtt inflate OK ');
  134. console.log('wss.clients size~~~~~~~ '+ wss.clients.size);
  135. //wss.send(buffer);
  136. //wssclients[0].send(buffer);
  137. var jsdata = buffer.toString();
  138. if (topic.toString().indexOf('wxxm')!=-1)
  139. {
  140. console.log(' wxxm find ');
  141. var jssdata = {
  142. "cmd": "wxxIotM"//,
  143. //"data":""
  144. };
  145. //jssdata['data']=jsdata ;
  146. jssdata['data']=JSON.parse(jsdata) ;
  147. try
  148. {
  149. wsConnect.send(JSON.stringify(jssdata ));
  150. }
  151. catch(e)
  152. {
  153. console.log(e)
  154. }
  155. console.log(JSON.stringify(jssdata ));
  156. }
  157. if (topic.toString().indexOf('wxxk')!=-1)
  158. {
  159. console.log(' wxxk find');
  160. var jssdata = {
  161. "cmd": "wxxIotK"//,
  162. //"data":""
  163. };
  164. //jssdata['data']=jsdata ;
  165. jssdata['data']=JSON.parse(jsdata) ;
  166. try
  167. {
  168. wsConnect.send(JSON.stringify(jssdata ));
  169. }
  170. catch(e)
  171. {
  172. console.log(e)
  173. }
  174. console.log(JSON.stringify(jssdata ));
  175. }
  176. console.log(' wssclients[0].send ');
  177. }
  178. });
  179. //number = 5
  180. } else {
  181. strData = '';
  182. }
  183. console.log('============mqtt rece finish');
  184. console.log(' ');
  185. //wsConnect.send(Buffer.from(message));
  186. // client.end();
  187. });
  188. //
  189. //0
  190. wsConnect.on('message', function(message)
  191. {
  192. console.log('wss rece: '+message);
  193. //to k or m
  194. var obj = JSON.parse(message.toString());
  195. var vdata =obj ['data'];
  196. console.log( 'obj.data:' + vdata);
  197. console.log( '~~~~~~~~~~~~~~~~~~~~~');
  198. message = vdata ;
  199. var rawData;
  200. var dataType;
  201. var event = 0;
  202. var bufferID = stringToArray('GA300_1_A1001001_5');
  203. console.log('message.length:' + message.length);
  204. try
  205. {
  206. if (message.length < 128) {
  207. dataType = 0;
  208. rawData = stringToArray(message );
  209. console.log('rawData:' + rawData);
  210. var blockData = new Uint8Array(25 + rawData.length + 2);
  211. blockData[0] = dataType;
  212. blockData.set(bufferID,1);
  213. blockData.set(rawData, 25);
  214. var crc16 = calc_crc16(blockData, 25 + rawData.length)
  215. blockData.set(intToUint8ArrayCRC(crc16), 25 + rawData.length);
  216. console.log('blockData: ' + blockData);
  217. var packedData = new Uint8Array(32 + blockData.length);
  218. packedData.set(bufferID);
  219. packedData.set(intToUint8Array(event), 24);
  220. packedData.set(intToUint8Array(25+rawData.length+2), 26);
  221. packedData.set(intToUint8Array(0), 28);
  222. packedData[30] = 1;
  223. packedData[31] = 1;
  224. packedData.set(blockData, 32);
  225. console.log('~~~~~~~~~~~~~~~~~~');
  226. console.log('~~~~~~~~~~~~~~~~~~');
  227. console.log('~~~~~~~~~~~~~~~~~~');
  228. client.publish('CAD9A3D3EUWWNBKY111A/iotm/5',Buffer.from(packedData));
  229. client.publish('CAD9A3D3EUWWNBKY111A/iotk/5',Buffer.from(packedData));
  230. console.log('~~~~~~~~~~~~~~~~~~');
  231. } else {
  232. dataType = 1;
  233. zlib.deflate(message, (err, buffer) => {
  234. try
  235. {
  236. if (!err) {
  237. console.log('deflate OK'+buffer.toString('base64'));
  238. console.log('rawData:' + buffer);
  239. rawData=buffer;
  240. var blockData = new Uint8Array(25 + rawData.length + 2);
  241. blockData[0] = dataType;
  242. blockData.set(bufferID,1);
  243. blockData.set(rawData, 25);
  244. var crc16 = calc_crc16(blockData, 25 + rawData.length)
  245. blockData.set(intToUint8ArrayCRC(crc16), 25 + rawData.length);
  246. console.log('blockData: ' + blockData);
  247. var packedData = new Uint8Array(32 + blockData.length);
  248. packedData.set(bufferID);
  249. packedData.set(intToUint8Array(event), 24);
  250. packedData.set(intToUint8Array(25+rawData.length+2), 26);
  251. packedData.set(intToUint8Array(0), 28);
  252. packedData[30] = 1;
  253. packedData[31] = 1;
  254. packedData.set(blockData, 32);
  255. console.log('~~~~~~~~~~~~~~~~~~');
  256. console.log('~~~~~~~~~~~~~~~~~~');
  257. console.log('~~~~~~~~~~~~~~~~~~');
  258. client.publish('CAD9A3D3EUWWNBKY111A/iotm/5',Buffer.from(packedData));
  259. client.publish('CAD9A3D3EUWWNBKY111A/iotk/5',Buffer.from(packedData));
  260. console.log('~~~~~~~~~~~~~~~~~~');
  261. }
  262. }
  263. catch(e){
  264. console.log(e)
  265. }
  266. });
  267. }
  268. }
  269. catch(e){
  270. console.log(e)
  271. }
  272. //client.publish('CAD9A3D3EUWWNBKY111A/iotm/5',rawData);
  273. //client.publish('CAD9A3D3EUWWNBKY111A/iotk/5',rawData);
  274. //client.subscribe("CAD9A3D3EUWWNBKY111A/wxxm/#");
  275. //wsConnect.send(message);
  276. });
  277. });
  278. //wss finish
  279. //0
  280. //0