wechat wss demo

wss-server_3.js 6.9KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310
  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. wsConnect.send(buffer.toString());
  138. console.log(' wssclients[0].send ');
  139. }
  140. });
  141. //number = 5
  142. } else {
  143. strData = '';
  144. }
  145. console.log('============mqtt rece finish');
  146. console.log(' ');
  147. //wsConnect.send(Buffer.from(message));
  148. // client.end();
  149. });
  150. //
  151. //0
  152. wsConnect.on('message', function(message)
  153. {
  154. console.log('wss rece: '+message);
  155. //to k or m
  156. var obj = JSON.parse(message.toString());
  157. var vdata =obj ['data'];
  158. console.log( 'obj.data:' + vdata);
  159. console.log( '~~~~~~~~~~~~~~~~~~~~~');
  160. message = vdata ;
  161. var rawData;
  162. var dataType;
  163. var event = 0;
  164. var bufferID = stringToArray('GA300_1_A1001001_5');
  165. console.log('message.length:' + message.length);
  166. if (message.length < 128) {
  167. dataType = 0;
  168. rawData = stringToArray(message );
  169. console.log('rawData:' + rawData);
  170. var blockData = new Uint8Array(25 + rawData.length + 2);
  171. blockData[0] = dataType;
  172. blockData.set(bufferID,1);
  173. blockData.set(rawData, 25);
  174. var crc16 = calc_crc16(blockData, 25 + rawData.length)
  175. blockData.set(intToUint8ArrayCRC(crc16), 25 + rawData.length);
  176. console.log('blockData: ' + blockData);
  177. var packedData = new Uint8Array(32 + blockData.length);
  178. packedData.set(bufferID);
  179. packedData.set(intToUint8Array(event), 24);
  180. packedData.set(intToUint8Array(25+rawData.length+2), 26);
  181. packedData.set(intToUint8Array(0), 28);
  182. packedData[30] = 1;
  183. packedData[31] = 1;
  184. packedData.set(blockData, 32);
  185. console.log('~~~~~~~~~~~~~~~~~~');
  186. console.log('~~~~~~~~~~~~~~~~~~');
  187. console.log('~~~~~~~~~~~~~~~~~~');
  188. client.publish('CAD9A3D3EUWWNBKY111A/iotm/5',Buffer.from(packedData));
  189. client.publish('CAD9A3D3EUWWNBKY111A/iotk/5',Buffer.from(packedData));
  190. console.log('~~~~~~~~~~~~~~~~~~');
  191. } else {
  192. dataType = 1;
  193. zlib.deflate(message, (err, buffer) => {
  194. if (!err) {
  195. console.log('deflate OK'+buffer.toString('base64'));
  196. console.log('rawData:' + buffer);
  197. rawData=buffer;
  198. var blockData = new Uint8Array(25 + rawData.length + 2);
  199. blockData[0] = dataType;
  200. blockData.set(bufferID,1);
  201. blockData.set(rawData, 25);
  202. var crc16 = calc_crc16(blockData, 25 + rawData.length)
  203. blockData.set(intToUint8ArrayCRC(crc16), 25 + rawData.length);
  204. console.log('blockData: ' + blockData);
  205. var packedData = new Uint8Array(32 + blockData.length);
  206. packedData.set(bufferID);
  207. packedData.set(intToUint8Array(event), 24);
  208. packedData.set(intToUint8Array(25+rawData.length+2), 26);
  209. packedData.set(intToUint8Array(0), 28);
  210. packedData[30] = 1;
  211. packedData[31] = 1;
  212. packedData.set(blockData, 32);
  213. console.log('~~~~~~~~~~~~~~~~~~');
  214. console.log('~~~~~~~~~~~~~~~~~~');
  215. console.log('~~~~~~~~~~~~~~~~~~');
  216. client.publish('CAD9A3D3EUWWNBKY111A/iotm/5',Buffer.from(packedData));
  217. client.publish('CAD9A3D3EUWWNBKY111A/iotk/5',Buffer.from(packedData));
  218. console.log('~~~~~~~~~~~~~~~~~~');
  219. }
  220. });
  221. }
  222. // client.publish('CAD9A3D3EUWWNBKY111A/iotm/5',rawData);
  223. //client.publish('CAD9A3D3EUWWNBKY111A/iotk/5',rawData);
  224. //client.subscribe("CAD9A3D3EUWWNBKY111A/wxxm/#");
  225. //wsConnect.send(message);
  226. });
  227. });
  228. //wss finish
  229. //0