SensorMesh.h 4.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134
  1. #pragma once
  2. #include <Arduino.h> // needed for PlatformIO
  3. #include <Mesh.h>
  4. #if defined(NRF52_PLATFORM) || defined(STM32_PLATFORM)
  5. #include <InternalFileSystem.h>
  6. #elif defined(RP2040_PLATFORM)
  7. #include <LittleFS.h>
  8. #elif defined(ESP32)
  9. #include <SPIFFS.h>
  10. #endif
  11. #include <helpers/ArduinoHelpers.h>
  12. #include <helpers/StaticPoolPacketManager.h>
  13. #include <helpers/SimpleMeshTables.h>
  14. #include <helpers/IdentityStore.h>
  15. #include <helpers/AdvertDataHelpers.h>
  16. #include <helpers/TxtDataHelpers.h>
  17. #include <helpers/CommonCLI.h>
  18. #include <RTClib.h>
  19. #include <target.h>
  20. struct ContactInfo {
  21. mesh::Identity id;
  22. uint8_t type; // 1 = admin, 0 = guest
  23. uint8_t flags;
  24. int8_t out_path_len;
  25. uint8_t out_path[MAX_PATH_SIZE];
  26. uint8_t shared_secret[PUB_KEY_SIZE];
  27. uint32_t last_timestamp; // by THEIR clock (transient)
  28. uint32_t last_activity; // by OUR clock (transient)
  29. bool isAdmin() const { return type != 0; }
  30. };
  31. #ifndef FIRMWARE_BUILD_DATE
  32. #define FIRMWARE_BUILD_DATE "2 Jul 2025"
  33. #endif
  34. #ifndef FIRMWARE_VERSION
  35. #define FIRMWARE_VERSION "v1.7.2"
  36. #endif
  37. #define FIRMWARE_ROLE "sensor"
  38. #ifndef MAX_CONTACTS
  39. #define MAX_CONTACTS 32
  40. #endif
  41. #define MAX_SEARCH_RESULTS 8
  42. class SensorMesh : public mesh::Mesh, public CommonCLICallbacks {
  43. public:
  44. SensorMesh(mesh::MainBoard& board, mesh::Radio& radio, mesh::MillisecondClock& ms, mesh::RNG& rng, mesh::RTCClock& rtc, mesh::MeshTables& tables);
  45. void begin(FILESYSTEM* fs);
  46. CommonCLI* getCLI() { return &_cli; }
  47. void loop();
  48. // CommonCLI callbacks
  49. const char* getFirmwareVer() override { return FIRMWARE_VERSION; }
  50. const char* getBuildDate() override { return FIRMWARE_BUILD_DATE; }
  51. const char* getRole() override { return FIRMWARE_ROLE; }
  52. const char* getNodeName() { return _prefs.node_name; }
  53. NodePrefs* getNodePrefs() { return &_prefs; }
  54. void savePrefs() override { _cli.savePrefs(_fs); }
  55. bool formatFileSystem() override;
  56. void sendSelfAdvertisement(int delay_millis) override;
  57. void updateAdvertTimer() override;
  58. void updateFloodAdvertTimer() override;
  59. void setLoggingOn(bool enable) override { }
  60. void eraseLogFile() override { }
  61. void dumpLogFile() override { }
  62. void setTxPower(uint8_t power_dbm) override;
  63. void formatNeighborsReply(char *reply) override {
  64. strcpy(reply, "not supported");
  65. }
  66. const uint8_t* getSelfIdPubKey() override { return self_id.pub_key; }
  67. void clearStats() override { }
  68. protected:
  69. // telemetry data queries
  70. float getVoltage(uint8_t channel) { return 0.0f; } // TODO: extract from curr telemetry buffer
  71. // alerts
  72. struct Trigger {
  73. bool triggered;
  74. uint32_t time;
  75. Trigger() { triggered = false; time = 0; }
  76. };
  77. void alertIfLow(Trigger& t, float value, float threshold, const char* text);
  78. void alertIfHigh(Trigger& t, float value, float threshold, const char* text);
  79. virtual void checkForAlerts() = 0; // for app to implement
  80. // Mesh overrides
  81. float getAirtimeBudgetFactor() const override;
  82. bool allowPacketForward(const mesh::Packet* packet) override;
  83. int calcRxDelay(float score, uint32_t air_time) const override;
  84. uint32_t getRetransmitDelay(const mesh::Packet* packet) override;
  85. uint32_t getDirectRetransmitDelay(const mesh::Packet* packet) override;
  86. int getInterferenceThreshold() const override;
  87. int getAGCResetInterval() const override;
  88. void onAnonDataRecv(mesh::Packet* packet, uint8_t type, const mesh::Identity& sender, uint8_t* data, size_t len) override;
  89. int searchPeersByHash(const uint8_t* hash) override;
  90. void getPeerSharedSecret(uint8_t* dest_secret, int peer_idx) override;
  91. void onAdvertRecv(mesh::Packet* packet, const mesh::Identity& id, uint32_t timestamp, const uint8_t* app_data, size_t app_data_len);
  92. void onPeerDataRecv(mesh::Packet* packet, uint8_t type, int sender_idx, const uint8_t* secret, uint8_t* data, size_t len) override;
  93. bool onPeerPathRecv(mesh::Packet* packet, int sender_idx, const uint8_t* secret, uint8_t* path, uint8_t path_len, uint8_t extra_type, uint8_t* extra, uint8_t extra_len) override;
  94. private:
  95. FILESYSTEM* _fs;
  96. unsigned long next_local_advert, next_flood_advert;
  97. NodePrefs _prefs;
  98. CommonCLI _cli;
  99. uint8_t reply_data[MAX_PACKET_PAYLOAD];
  100. ContactInfo contacts[MAX_CONTACTS];
  101. int num_contacts;
  102. unsigned long dirty_contacts_expiry;
  103. CayenneLPP telemetry;
  104. uint32_t last_read_time;
  105. int matching_peer_indexes[MAX_SEARCH_RESULTS];
  106. void loadContacts();
  107. void saveContacts();
  108. int handleRequest(ContactInfo& sender, uint32_t sender_timestamp, uint8_t* payload, size_t payload_len);
  109. mesh::Packet* createSelfAdvert();
  110. ContactInfo* putContact(const mesh::Identity& id);
  111. void sendAlert(const char* text) { } // TODO
  112. };