Fork of the espurna firmware for `mhsw` switches
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

284 lines
8.7 KiB

  1. /*
  2. RFM69 MODULE
  3. Copyright (C) 2016-2017 by Xose Pérez <xose dot perez at gmail dot com>
  4. */
  5. #if RFM69_SUPPORT
  6. #include "libs/RFM69Wrap.h"
  7. #define RFM69_PACKET_SEPARATOR ':'
  8. // -----------------------------------------------------------------------------
  9. // Locals
  10. // -----------------------------------------------------------------------------
  11. RFM69Wrap * _rfm69_radio;
  12. struct _node_t {
  13. unsigned long count = 0;
  14. unsigned long missing = 0;
  15. unsigned long duplicates = 0;
  16. unsigned char lastPacketID = 0;
  17. };
  18. _node_t _rfm69_node_info[RFM69_MAX_NODES];
  19. unsigned char _rfm69_node_count;
  20. unsigned long _rfm69_packet_count;
  21. // -----------------------------------------------------------------------------
  22. // WEB
  23. // -----------------------------------------------------------------------------
  24. #if WEB_SUPPORT
  25. void _rfm69WebSocketOnConnected(JsonObject& root) {
  26. root["rfm69Visible"] = 1;
  27. root["rfm69Topic"] = getSetting("rfm69Topic", RFM69_DEFAULT_TOPIC);
  28. root["packetCount"] = _rfm69_packet_count;
  29. root["nodeCount"] = _rfm69_node_count;
  30. JsonArray& mappings = root.createNestedArray("mapping");
  31. for (unsigned char i=0; i<RFM69_MAX_TOPICS; i++) {
  32. unsigned char node = getSetting("node", i, 0).toInt();
  33. if (0 == node) break;
  34. JsonObject& mapping = mappings.createNestedObject();
  35. mapping["node"] = node;
  36. mapping["key"] = getSetting("key", i, "");
  37. mapping["topic"] = getSetting("topic", i, "");
  38. }
  39. }
  40. bool _rfm69WebSocketOnKeyCheck(const char * key, JsonVariant& value) {
  41. if (strncmp(key, "rfm69", 5) == 0) return true;
  42. if (strncmp(key, "node", 4) == 0) return true;
  43. if (strncmp(key, "key", 3) == 0) return true;
  44. if (strncmp(key, "topic", 5) == 0) return true;
  45. return false;
  46. }
  47. void _rfm69WebSocketOnAction(uint32_t client_id, const char * action, JsonObject& data) {
  48. if (strcmp(action, "clear-counts") == 0) _rfm69Clear();
  49. }
  50. #endif // WEB_SUPPORT
  51. void _rfm69CleanNodes(unsigned char num) {
  52. // Look for the last defined node
  53. int i = 0;
  54. while (i < num) {
  55. if (getSetting("node", i, 0).toInt() == 0) break;
  56. if (getSetting("key", i, "").length() == 0) break;
  57. if (getSetting("topic", i, "").length() == 0) break;
  58. ++i;
  59. }
  60. // Delete all other settings
  61. while (i < WIFI_MAX_NETWORKS) {
  62. delSetting("node", i);
  63. delSetting("key", i);
  64. delSetting("topic", i);
  65. ++i;
  66. }
  67. }
  68. void _rfm69Configure() {
  69. _rfm69CleanNodes(RFM69_MAX_TOPICS);
  70. }
  71. // -----------------------------------------------------------------------------
  72. // Radio
  73. // -----------------------------------------------------------------------------
  74. void _rfm69Debug(const char * level, packet_t * data) {
  75. DEBUG_MSG_P(
  76. PSTR("[RFM69] %s: messageID:%05d senderID:%03d targetID:%03d packetID:%03d rssi:%-04d key:%s value:%s\n"),
  77. level,
  78. data->messageID,
  79. data->senderID,
  80. data->targetID,
  81. data->packetID,
  82. data->rssi,
  83. data->key,
  84. data->value
  85. );
  86. }
  87. void _rfm69Process(packet_t * data) {
  88. // Is node beyond RFM69_MAX_NODES?
  89. if (data->senderID >= RFM69_MAX_NODES) return;
  90. // Count seen nodes
  91. if (_rfm69_node_info[data->senderID].count == 0) ++_rfm69_node_count;
  92. // Detect duplicates and missing packets
  93. // packetID==0 means device is not sending packetID info
  94. if (data->packetID > 0) {
  95. if (_rfm69_node_info[data->senderID].count > 0) {
  96. unsigned char gap = data->packetID - _rfm69_node_info[data->senderID].lastPacketID;
  97. if (gap == 0) {
  98. _rfm69_node_info[data->senderID].duplicates = _rfm69_node_info[data->senderID].duplicates + 1;
  99. //_rfm69Debug("DUP", data);
  100. return;
  101. }
  102. if ((gap > 1) && (data->packetID > 1)) {
  103. _rfm69_node_info[data->senderID].missing = _rfm69_node_info[data->senderID].missing + gap - 1;
  104. DEBUG_MSG_P(PSTR("[RFM69] %u missing packets detected\n"), gap - 1);
  105. }
  106. }
  107. }
  108. _rfm69Debug("OK ", data);
  109. _rfm69_node_info[data->senderID].lastPacketID = data->packetID;
  110. _rfm69_node_info[data->senderID].count = _rfm69_node_info[data->senderID].count + 1;
  111. // Send info to websocket clients
  112. {
  113. char buffer[200];
  114. snprintf_P(
  115. buffer,
  116. sizeof(buffer) - 1,
  117. PSTR("{\"nodeCount\": %d, \"packetCount\": %lu, \"packet\": {\"senderID\": %u, \"targetID\": %u, \"packetID\": %u, \"key\": \"%s\", \"value\": \"%s\", \"rssi\": %d, \"duplicates\": %d, \"missing\": %d}}"),
  118. _rfm69_node_count, _rfm69_packet_count,
  119. data->senderID, data->targetID, data->packetID, data->key, data->value, data->rssi,
  120. _rfm69_node_info[data->senderID].duplicates , _rfm69_node_info[data->senderID].missing);
  121. wsSend(buffer);
  122. }
  123. // If we are the target of the message, forward it via MQTT, otherwise quit
  124. if (!RFM69_PROMISCUOUS_SENDS && (RFM69_GATEWAY_ID != data->targetID)) return;
  125. // Try to find a matching mapping
  126. for (unsigned int i=0; i<RFM69_MAX_TOPICS; i++) {
  127. unsigned char node = getSetting("node", i, 0).toInt();
  128. if (0 == node) break;
  129. if ((node == data->senderID) && (getSetting("key", i, "").equals(data->key))) {
  130. mqttSendRaw((char *) getSetting("topic", i, "").c_str(), (char *) String(data->value).c_str());
  131. return;
  132. }
  133. }
  134. // Mapping not found, use default topic
  135. String topic = getSetting("rfm69Topic", RFM69_DEFAULT_TOPIC);
  136. if (topic.length() > 0) {
  137. topic.replace("{node}", String(data->senderID));
  138. topic.replace("{key}", String(data->key));
  139. mqttSendRaw((char *) topic.c_str(), (char *) String(data->value).c_str());
  140. }
  141. }
  142. void _rfm69Loop() {
  143. if (_rfm69_radio->receiveDone()) {
  144. uint8_t senderID = _rfm69_radio->SENDERID;
  145. uint8_t targetID = _rfm69_radio->TARGETID;
  146. int16_t rssi = _rfm69_radio->RSSI;
  147. uint8_t length = _rfm69_radio->DATALEN;
  148. char buffer[length + 1];
  149. strncpy(buffer, (const char *) _rfm69_radio->DATA, length);
  150. buffer[length] = 0;
  151. // Do not send ACKs in promiscuous mode,
  152. // we want to listen without being heard
  153. if (!RFM69_PROMISCUOUS) {
  154. if (_rfm69_radio->ACKRequested()) _rfm69_radio->sendACK();
  155. }
  156. uint8_t parts = 1;
  157. for (uint8_t i=0; i<length; i++) {
  158. if (buffer[i] == RFM69_PACKET_SEPARATOR) ++parts;
  159. }
  160. if (parts > 1) {
  161. char sep[2] = {RFM69_PACKET_SEPARATOR, 0};
  162. uint8_t packetID = 0;
  163. char * key = strtok(buffer, sep);
  164. char * value = strtok(NULL, sep);
  165. if (parts > 2) {
  166. char * packet = strtok(NULL, sep);
  167. packetID = atoi(packet);
  168. }
  169. packet_t message;
  170. message.messageID = ++_rfm69_packet_count;
  171. message.packetID = packetID;
  172. message.senderID = senderID;
  173. message.targetID = targetID;
  174. message.key = key;
  175. message.value = value;
  176. message.rssi = rssi;
  177. _rfm69Process(&message);
  178. }
  179. }
  180. }
  181. void _rfm69Clear() {
  182. for(unsigned int i=0; i<RFM69_MAX_NODES; i++) {
  183. _rfm69_node_info[i].duplicates = 0;
  184. _rfm69_node_info[i].missing = 0;
  185. _rfm69_node_info[i].count = 0;
  186. }
  187. _rfm69_node_count = 0;
  188. _rfm69_packet_count = 0;
  189. }
  190. // -----------------------------------------------------------------------------
  191. // RFM69
  192. // -----------------------------------------------------------------------------
  193. void rfm69Setup() {
  194. delay(10);
  195. _rfm69Configure();
  196. _rfm69_radio = new RFM69Wrap(RFM69_CS_PIN, RFM69_IRQ_PIN, RFM69_IS_RFM69HW, digitalPinToInterrupt(RFM69_IRQ_PIN));
  197. _rfm69_radio->initialize(RFM69_FREQUENCY, RFM69_NODE_ID, RFM69_NETWORK_ID);
  198. _rfm69_radio->encrypt(RFM69_ENCRYPTKEY);
  199. _rfm69_radio->promiscuous(RFM69_PROMISCUOUS);
  200. _rfm69_radio->enableAutoPower(0);
  201. if (RFM69_IS_RFM69HW) _rfm69_radio->setHighPower();
  202. DEBUG_MSG_P(PSTR("[RFM69] Worning at %u MHz\n"), RFM69_FREQUENCY == RF69_433MHZ ? 433 : RFM69_FREQUENCY == RF69_868MHZ ? 868 : 915);
  203. DEBUG_MSG_P(PSTR("[RFM69] Node %u\n"), RFM69_NODE_ID);
  204. DEBUG_MSG_P(PSTR("[RFM69] Network %u\n"), RFM69_NETWORK_ID);
  205. DEBUG_MSG_P(PSTR("[RFM69] Promiscuous mode %s\n"), RFM69_PROMISCUOUS ? "ON" : "OFF");
  206. #if WEB_SUPPORT
  207. wsRegister()
  208. .onConnected(_rfm69WebSocketOnConnected)
  209. .onAction(_rfm69WebSocketOnAction)
  210. .onKeyCheck(_rfm69WebSocketOnKeyCheck);
  211. #endif
  212. // Main callbacks
  213. espurnaRegisterLoop(_rfm69Loop);
  214. espurnaRegisterReload(_rfm69Configure);
  215. }
  216. #endif // RFM69_SUPPORT