#include "MyMesh.h" #include #include #if defined(MESHCORE_ESP32_FULL_PROFILE) #include #endif #include #include #include #ifdef WITH_WEBCONFIG #include #endif static uint32_t nextRadioApplyRetryDelay(uint8_t& failure_count) { uint8_t shift = failure_count < 5 ? failure_count : 5; if (failure_count < 6) failure_count++; uint32_t delay_ms = 1000UL << shift; return delay_ms > 30000UL ? 30000UL : delay_ms; } #if defined(WITH_MQTT_NEIGHBORS) #include // kSyncedClockEpoch #endif #define ANON_REQ_TYPE_REGIONS 0x01 // client side of the anon-regions scope query (neighbors feature) #define REPLY_DELAY_MILLIS 1500 #define PUSH_NOTIFY_DELAY_MILLIS 2000 #define SYNC_PUSH_INTERVAL 1200 #define PUSH_ACK_TIMEOUT_FLOOD 12000 #define PUSH_TIMEOUT_BASE 4000 #define PUSH_ACK_TIMEOUT_FACTOR 2000 #define POST_SYNC_DELAY_SECS 6 #define FIRMWARE_VER_LEVEL 1 #define REQ_TYPE_GET_STATUS 0x01 // same as _GET_STATS #define REQ_TYPE_KEEP_ALIVE 0x02 #define REQ_TYPE_GET_TELEMETRY_DATA 0x03 #define REQ_TYPE_GET_ACCESS_LIST 0x05 #define RESP_SERVER_LOGIN_OK 0 // response to ANON_REQ // Best-effort bound for the queued CLI reply before OTA blocks the loop and // reboots. Do not let a busy or duty-limited channel stall the update forever. #define OTA_TX_DRAIN_TIMEOUT_MS 5000 #define LAZY_CONTACTS_WRITE_DELAY 5000 struct ServerStats { uint16_t batt_milli_volts; uint16_t curr_tx_queue_len; int16_t noise_floor; int16_t last_rssi; uint32_t n_packets_recv; uint32_t n_packets_sent; uint32_t total_air_time_secs; uint32_t total_up_time_secs; uint32_t n_sent_flood, n_sent_direct; uint32_t n_recv_flood, n_recv_direct; uint16_t err_events; // was 'n_full_events' int16_t last_snr; // x 4 uint16_t n_direct_dups, n_flood_dups; uint16_t n_posted, n_post_push; }; void MyMesh::addPost(ClientInfo *client, const char *postData) { storePost(client->id, postData); } void MyMesh::addSystemPost(const char *postData) { if (!postData || postData[0] == 0) return; MESH_DEBUG_PRINTLN("room.post: addSystemPost: %s", postData); storePost(self_id, postData); } void MyMesh::storePost(const mesh::Identity &author, const char *postData) { int idx = next_post_idx; // TODO: suggested postData format: /<descrption> posts[idx].author = author; // add to cyclic queue StrHelper::strncpy(posts[idx].text, postData, MAX_POST_TEXT_LEN); posts[idx].post_timestamp = getRTCClock()->getCurrentTimeUnique(); MESH_DEBUG_PRINTLN("room.post: storePost idx=%d text=%s", idx, posts[idx].text); MESH_DEBUG_PRINTLN("room.post: timestamp=%u", posts[idx].post_timestamp); next_post_idx = (next_post_idx + 1) % MAX_UNSYNCED_POSTS; next_push = futureMillis(PUSH_NOTIFY_DELAY_MILLIS); _num_posted++; // stats MESH_DEBUG_PRINTLN("room.post: next_post_idx=%d num_posted=%d push scheduled", next_post_idx, _num_posted); } void MyMesh::pushPostToClient(ClientInfo *client, PostInfo &post) { MESH_DEBUG_PRINTLN("room.post: pushPostToClient text=%s", post.text); int len = 0; memcpy(&reply_data[len], &post.post_timestamp, 4); len += 4; // this is a PAST timestamp... but should be accepted by client uint8_t attempt; getRNG()->random(&attempt, 1); // need this for re-tries, so packet hash (and ACK) will be different reply_data[len++] = (TXT_TYPE_SIGNED_PLAIN << 2) | (attempt & 3); // 'signed' plain text // encode prefix of post.author.pub_key memcpy(&reply_data[len], post.author.pub_key, 4); len += 4; // just first 4 bytes int text_len = strlen(post.text); memcpy(&reply_data[len], post.text, text_len); len += text_len; uint8_t message_participants[2 * PUB_KEY_SIZE]; memcpy(message_participants, client->id.pub_key, PUB_KEY_SIZE); memcpy(&message_participants[PUB_KEY_SIZE], post.author.pub_key, PUB_KEY_SIZE); uint8_t message_retry_key[MAX_HASH_SIZE]; mesh::Utils::sha256(message_retry_key, sizeof(message_retry_key), message_participants, sizeof(message_participants), (const uint8_t*)post.text, text_len); // calc expected ACK reply mesh::Utils::sha256((uint8_t *)&client->extra.room.pending_ack, 4, reply_data, len, client->id.pub_key, PUB_KEY_SIZE); client->extra.room.push_post_timestamp = post.post_timestamp; auto reply = createDatagram(PAYLOAD_TYPE_TXT_MSG, client->id, client->shared_secret, reply_data, len); bool sent = false; if (reply) { if (client->out_path_len == OUT_PATH_UNKNOWN) { unsigned long delay_millis = 0; sent = sendFloodScoped(default_scope, reply, delay_millis, _prefs.path_hash_mode + 1); // REVISIT if (sent) { client->extra.room.ack_timeout = futureMillis(PUSH_ACK_TIMEOUT_FLOOD); } } else { sent = sendDirect(reply, client->out_path, client->out_path_len); if (sent) { uint8_t path_hash_count = client->out_path_len & 63; client->extra.room.ack_timeout = futureMillis( PUSH_TIMEOUT_BASE + PUSH_ACK_TIMEOUT_FACTOR * (path_hash_count + 1)); } } if (sent) { replaceActiveMessageRetries(reply, message_retry_key, post.post_timestamp); _num_post_pushes++; // stats } } if (!sent) { client->extra.room.pending_ack = 0; MESH_DEBUG_PRINTLN("Unable to push post to client"); } } uint8_t MyMesh::getUnsyncedCount(ClientInfo *client) { uint8_t count = 0; for (int k = 0; k < MAX_UNSYNCED_POSTS; k++) { if (posts[k].post_timestamp > client->extra.room.sync_since // is new post for this Client? && !posts[k].author.matches(client->id)) { // don't push posts to the author count++; } } return count; } bool MyMesh::processAck(const uint8_t *data) { for (int i = 0; i < acl.getNumClients(); i++) { auto client = acl.getClientByIdx(i); if (client->extra.room.pending_ack && memcmp(data, &client->extra.room.pending_ack, 4) == 0) { // got an ACK from Client! client->extra.room.pending_ack = 0; // clear this, so next push can happen client->extra.room.push_failures = 0; client->extra.room.sync_since = client->extra.room.push_post_timestamp; // advance Client's SINCE timestamp, to sync next post return true; } } return false; } mesh::Packet *MyMesh::createSelfAdvert() { uint8_t app_data[MAX_ADVERT_DATA_SIZE]; uint8_t app_data_len = _cli.buildAdvertData(ADV_TYPE_ROOM, app_data); return createAdvert(self_id, app_data, app_data_len); } File MyMesh::openAppend(const char *fname) { #if defined(NRF52_PLATFORM) return _fs->open(fname, FILE_O_WRITE); #elif defined(RP2040_PLATFORM) return _fs->open(fname, "a"); #else return _fs->open(fname, "a", true); #endif } int MyMesh::handleRequest(ClientInfo *sender, uint32_t sender_timestamp, uint8_t *payload, size_t payload_len) { // uint32_t now = getRTCClock()->getCurrentTimeUnique(); // memcpy(reply_data, &now, 4); // response packets always prefixed with timestamp memcpy(reply_data, &sender_timestamp, 4); // reflect sender_timestamp back in response packet (kind of like a 'tag') if (payload[0] == REQ_TYPE_GET_STATUS) { ServerStats stats; stats.batt_milli_volts = board.getBattMilliVolts(); stats.curr_tx_queue_len = _mgr->getOutboundCount(0xFFFFFFFF); stats.noise_floor = (int16_t)_radio->getNoiseFloor(); stats.last_rssi = (int16_t)radio_driver.getLastRSSI(); stats.n_packets_recv = radio_driver.getPacketsRecv(); stats.n_packets_sent = radio_driver.getPacketsSent(); stats.total_air_time_secs = getTotalAirTime() / 1000; stats.total_up_time_secs = uptime_millis / 1000; stats.n_sent_flood = getNumSentFlood(); stats.n_sent_direct = getNumSentDirect(); stats.n_recv_flood = getNumRecvFlood(); stats.n_recv_direct = getNumRecvDirect(); stats.err_events = _err_flags; stats.last_snr = (int16_t)(radio_driver.getLastSNR() * 4); stats.n_direct_dups = ((SimpleMeshTables *)getTables())->getNumDirectDups(); stats.n_flood_dups = ((SimpleMeshTables *)getTables())->getNumFloodDups(); stats.n_posted = _num_posted; stats.n_post_push = _num_post_pushes; memcpy(&reply_data[4], &stats, sizeof(stats)); return 4 + sizeof(stats); } if (payload[0] == REQ_TYPE_GET_TELEMETRY_DATA) { uint8_t perm_mask = ~(payload[1]); // NEW: first reserved byte (of 4), is now inverse mask to apply to permissions telemetry.reset(); telemetry.addVoltage(TELEM_CHANNEL_SELF, (float)board.getBattMilliVolts() / 1000.0f); // query other sensors -- target specific if ((sender->permissions & PERM_ACL_ROLE_MASK) == PERM_ACL_GUEST) { perm_mask = 0x00; // just base telemetry allowed } sensors.querySensors(perm_mask, telemetry); // This default temperature will be overridden by external sensors (if any) float temperature = board.getMCUTemperature(); if(!isnan(temperature)) { // Supported boards with built-in temperature sensor. ESP32-C3 may return NAN telemetry.addTemperature(TELEM_CHANNEL_SELF, temperature); // Built-in MCU Temperature } uint8_t tlen = telemetry.getSize(); memcpy(&reply_data[4], telemetry.getBuffer(), tlen); return 4 + tlen; // reply_len } if (payload[0] == REQ_TYPE_GET_ACCESS_LIST && sender->isAdmin()) { uint8_t res1 = payload[1]; // reserved for future (extra query params) uint8_t res2 = payload[2]; if (res1 == 0 && res2 == 0) { uint8_t ofs = 4; for (int i = 0; i < acl.getNumClients() && ofs + 7 <= sizeof(reply_data) - 4; i++) { auto c = acl.getClientByIdx(i); if (!c->isAdmin()) continue; // skip non-Admin entries memcpy(&reply_data[ofs], c->id.pub_key, 6); ofs += 6; // just 6-byte pub_key prefix reply_data[ofs++] = c->permissions; } return ofs; } } return 0; // unknown command } void MyMesh::logRxRaw(float snr, float rssi, const uint8_t raw[], int len) { #if MESH_PACKET_LOGGING if (mesh::isUsbLoggingEnabled()) { // Logging builds prefer backpressure over silently losing a packet record. Serial.print(getLogDateTime()); Serial.print(" RAW: "); mesh::Utils::printHex(Serial, raw, len); Serial.println(); } #endif #ifdef WITH_MQTT_BRIDGE if (_prefs.bridge_enabled) { // Store raw radio data for MQTT messages (same as repeater) if (bridge) bridge->storeRawRadioData(raw, len, snr, rssi); } #endif } void MyMesh::logRx(mesh::Packet *pkt, int len, float score) { #ifdef WITH_MQTT_BRIDGE // MQTT bridge: always feed RX packets - bridge decides based on mqtt.rx setting if (_prefs.bridge_enabled && bridge) bridge->onPacketReceived(pkt); #endif if (_logging) { File f = openAppend(PACKET_LOG_FILE); if (f) { f.print(getLogDateTime()); f.printf(": RX, len=%d (type=%d, route=%s, payload_len=%d) SNR=%d RSSI=%d score=%d", len, pkt->getPayloadType(), pkt->isRouteDirect() ? "D" : "F", pkt->payload_len, (int)_radio->getLastSNR(), (int)_radio->getLastRSSI(), (int)(score * 1000)); if (pkt->getPayloadType() == PAYLOAD_TYPE_PATH || pkt->getPayloadType() == PAYLOAD_TYPE_REQ || pkt->getPayloadType() == PAYLOAD_TYPE_RESPONSE || pkt->getPayloadType() == PAYLOAD_TYPE_TXT_MSG) { f.printf(" [%02X -> %02X]\n", (uint32_t)pkt->payload[1], (uint32_t)pkt->payload[0]); } else { f.printf("\n"); } f.close(); } } } void MyMesh::logTx(mesh::Packet *pkt, int len) { #if defined(WITH_MQTT_NEIGHBORS) if (neighbor_discover_active && pkt == neighbor_discover_request && neighbor_discover_next < neighbor_discover_count) { NeighborDiscoverEntry& entry = neighbor_discover[neighbor_discover_next]; if (entry.status == ND_QUEUED) { entry.status = ND_PENDING; neighbor_discover_queried_count++; neighbor_discover_request = NULL; neighbor_discover_until = futureMillis(neighborDiscoverQueryTimeoutMs()); } } #endif #ifdef WITH_MQTT_BRIDGE // MQTT bridge: always feed TX packets - bridge decides based on mqtt.tx setting if (_prefs.bridge_enabled && bridge) bridge->sendPacket(pkt); #endif if (_logging) { File f = openAppend(PACKET_LOG_FILE); if (f) { f.print(getLogDateTime()); f.printf(": TX, len=%d (type=%d, route=%s, payload_len=%d)", len, pkt->getPayloadType(), pkt->isRouteDirect() ? "D" : "F", pkt->payload_len); if (pkt->getPayloadType() == PAYLOAD_TYPE_PATH || pkt->getPayloadType() == PAYLOAD_TYPE_REQ || pkt->getPayloadType() == PAYLOAD_TYPE_RESPONSE || pkt->getPayloadType() == PAYLOAD_TYPE_TXT_MSG) { f.printf(" [%02X -> %02X]\n", (uint32_t)pkt->payload[1], (uint32_t)pkt->payload[0]); } else { f.printf("\n"); } f.close(); } } } void MyMesh::logTxFail(mesh::Packet *pkt, int len) { #if defined(WITH_MQTT_NEIGHBORS) if (neighbor_discover_active && pkt == neighbor_discover_request && neighbor_discover_next < neighbor_discover_count) { NeighborDiscoverEntry& entry = neighbor_discover[neighbor_discover_next]; if (entry.status == ND_QUEUED) { entry.status = ND_SEND_FAILED; neighbor_discover_request = NULL; neighbor_discover_until = 0; } } #endif if (_logging) { File f = openAppend(PACKET_LOG_FILE); if (f) { f.print(getLogDateTime()); f.printf(": TX FAIL!, len=%d (type=%d, route=%s, payload_len=%d)\n", len, pkt->getPayloadType(), pkt->isRouteDirect() ? "D" : "F", pkt->payload_len); f.close(); } } } int MyMesh::calcRxDelay(float score, uint32_t air_time) const { if (_prefs.rx_delay_base <= 0.0f) return 0; return (int)((powf(_prefs.rx_delay_base, 0.85f - score) - 1.0f) * air_time); } #if defined(MESHCORE_ESP32_FULL_PROFILE) bool MyMesh::evaluateFloodRuleTiming(const mesh::Packet* packet, bool& fast_track) { fast_track = false; if (packet == NULL || !packet->isRouteFlood()) return false; bool incoming_region_allowed = false; RegionEntry* incoming_region = NULL; if (packet->getRouteType() == ROUTE_TYPE_TRANSPORT_FLOOD) { incoming_region = region_map.findMatch(packet, REGION_DENY_FLOOD); incoming_region_allowed = incoming_region != NULL; } else if (packet->getRouteType() == ROUTE_TYPE_FLOOD) { incoming_region_allowed = (region_map.getWildcard().flags & REGION_DENY_FLOOD) == 0; } mesh::Packet candidate = *packet; uint32_t match_mask = flood_rules.evaluate( packet, isTempRadioActive(), incoming_region_allowed, incoming_region); bool scope_set = false; return flood_rules.applyScope(&candidate, match_mask, scope_set, fast_track, false); } static const uint8_t ROOM_MAX_LOOP_MINIMAL[] = { 0, 4, 2, 1 }; static const uint8_t ROOM_MAX_LOOP_MODERATE[] = { 0, 2, 1, 1 }; static const uint8_t ROOM_MAX_LOOP_STRICT[] = { 0, 1, 1, 1 }; bool MyMesh::isLooped(const mesh::Packet* packet, const uint8_t max_counters[]) const { uint8_t hash_size = packet->getPathHashSize(); uint8_t hash_count = packet->getPathHashCount(); uint8_t occurrences = 0; const uint8_t* path = packet->path; while (hash_count > 0) { if (self_id.isHashMatch(path, hash_size)) occurrences++; hash_count--; path += hash_size; } return occurrences >= max_counters[hash_size]; } int MyMesh::calcRxDelayForPacket(const mesh::Packet* packet, float score, uint32_t air_time) { if (packet != NULL && packet->getPayloadType() == PAYLOAD_TYPE_OTA && isTempRadioActive()) return 0; bool fast_track = false; if (!evaluateFloodRuleTiming(packet, fast_track)) { return calcRxDelay(score, air_time); } if (fast_track) return 0; float slow_base = FloodFilterPolicy::slowScopeRxDelayBase(_prefs.rx_delay_base); return (int)((powf(slow_base, 0.85f - score) - 1.0f) * air_time); } uint32_t MyMesh::getSlowFloodRuleRetransmitDelay( const mesh::Packet* packet) { uint32_t airtime = _radio->getEstAirtimeFor( packet->getPathByteLen() + packet->payload_len + 2); uint32_t max_delay = FloodFilterPolicy::slowScopeMaxDelay(airtime); return getRNG()->nextInt(0, max_delay + 1); } #endif const char *MyMesh::getLogDateTime() { static char tmp[32]; uint32_t now = getRTCClock()->getCurrentTime(); DateTime dt = DateTime(now); sprintf(tmp, "%02d:%02d:%02d - %d/%d/%d U", dt.hour(), dt.minute(), dt.second(), dt.day(), dt.month(), dt.year()); return tmp; } uint32_t MyMesh::getRetransmitDelay(const mesh::Packet *packet) { uint32_t t = (_radio->getEstAirtimeFor(packet->getPathByteLen() + packet->payload_len + 2) * _prefs.tx_delay_factor); return getRNG()->nextInt(0, 5*t + 1); } uint32_t MyMesh::getDirectRetransmitDelay(const mesh::Packet *packet) { uint32_t t = (_radio->getEstAirtimeFor(packet->getPathByteLen() + packet->payload_len + 2) * _prefs.direct_tx_delay_factor); return getRNG()->nextInt(0, 5*t + 1); } bool MyMesh::allowDirectRetry(const mesh::Packet* packet, const uint8_t* next_hop_hash, uint8_t next_hop_hash_len) const { (void)packet; (void)next_hop_hash; (void)next_hop_hash_len; return _prefs.direct_retry_enabled != 0; } void MyMesh::configureDirectRetryPacket(mesh::Packet* retry, const mesh::Packet* original, uint8_t retry_attempt) { if (!_prefs.direct_retry_cr_enabled) { if (retry != NULL) retry->tx_cr = 0; return; } mesh::Mesh::configureDirectRetryPacket(retry, original, retry_attempt); } uint32_t MyMesh::getDirectRetryEchoDelay(const mesh::Packet* packet) const { uint32_t base_ms = constrain((uint32_t)_prefs.direct_retry_base_ms, (uint32_t)10, (uint32_t)5000); return base_ms + getDirectRetryPacketAirtimeDelay(packet); } uint8_t MyMesh::getDirectRetryMaxAttempts(const mesh::Packet* packet) const { if (packet != NULL && packet->getPayloadType() == PAYLOAD_TYPE_TXT_MSG) return 21; return constrain(_prefs.direct_retry_attempts, 1, 15); } uint32_t MyMesh::getDirectRetryAttemptDelay(const mesh::Packet* packet, uint8_t attempt_idx) { uint32_t step_ms = constrain((uint32_t)_prefs.direct_retry_step_ms, (uint32_t)0, (uint32_t)5000); return getDirectRetransmitDelay(packet) + getDirectRetryEchoDelay(packet) + ((uint32_t)attempt_idx * step_ms); } bool MyMesh::allowFloodRetry(const mesh::Packet* packet) const { if (_prefs.disable_fwd || _prefs.flood_retry_attempts == 0) return false; return packet == NULL || packet->getPayloadType() != PAYLOAD_TYPE_ADVERT || _prefs.flood_retry_advert_enabled; } uint8_t MyMesh::getFloodRetryMaxPathLength(const mesh::Packet* packet) const { uint8_t general_gate = _prefs.flood_retry_max_path == FLOOD_RETRY_PATH_GATE_DISABLED ? FLOOD_RETRY_PATH_GATE_DISABLED : constrain(_prefs.flood_retry_max_path, 0, 63); uint8_t group_data_gate = _prefs.flood_retry_group_max_path == FLOOD_RETRY_PATH_GATE_DISABLED ? FLOOD_RETRY_PATH_GATE_DISABLED : constrain(_prefs.flood_retry_group_max_path, 0, 63); return applyGroupDataFloodRetryPathGate(packet, general_gate, group_data_gate); } uint8_t MyMesh::getFloodRetryMaxAttempts(const mesh::Packet* packet) const { if (_prefs.disable_fwd) return 0; uint8_t attempts = constrain(_prefs.flood_retry_attempts, 0, 15); uint16_t scaled_attempts = attempts; uint8_t hops = packet != NULL ? packet->getPathHashCount() : 0; if (hops == 0) { scaled_attempts = (uint16_t)attempts * 2U; } else if (hops == 1) { scaled_attempts = (((uint16_t)attempts * 3U) + 1U) / 2U; } return scaled_attempts > 15 ? 15 : (uint8_t)scaled_attempts; } void MyMesh::onRetryConfigChanged() { if (!_prefs.direct_retry_enabled) cancelAllDirectRetries(); if (_prefs.disable_fwd || _prefs.flood_retry_attempts == 0) cancelAllFloodRetries(); } bool MyMesh::allowPacketForward(const mesh::Packet *packet) { if (_prefs.disable_fwd) return false; if (packet->isRouteFlood()) { if (mesh::isFloodHopLimitExceeded(packet, _prefs.flood_max, _prefs.flood_max_unscoped, _prefs.flood_max_advert)) { return false; } #if defined(MESHCORE_ESP32_FULL_PROFILE) if (flood_rules.shouldBlock(packet, recv_pkt_rule_match_mask, _ms->getMillis())) { return false; } #endif } #if defined(MESHCORE_ESP32_FULL_PROFILE) if (packet->isRouteFlood()) { if (recv_pkt_region == NULL && !recv_pkt_regionless_scope_set) { MESH_DEBUG_PRINTLN( "allowPacketForward: unknown transport code, or wildcard not allowed for FLOOD packet"); return false; } if (_prefs.loop_detect != LOOP_DETECT_OFF) { const uint8_t* maximums = _prefs.loop_detect == LOOP_DETECT_MINIMAL ? ROOM_MAX_LOOP_MINIMAL : (_prefs.loop_detect == LOOP_DETECT_MODERATE ? ROOM_MAX_LOOP_MODERATE : ROOM_MAX_LOOP_STRICT); if (isLooped(packet, maximums)) { MESH_DEBUG_PRINTLN( "allowPacketForward: FLOOD packet loop detected!"); return false; } } flood_rules.commitRates(packet, recv_pkt_rule_match_mask, _ms->getMillis()); } #endif _clock_sync.observeAcceptedFlood(packet); return true; } mesh::DispatcherAction MyMesh::onRecvPacket(mesh::Packet* pkt) { #if defined(MESHCORE_ESP32_FULL_PROFILE) bool scope_changed = false; bool fast_track_scope_change = false; recv_pkt_regionless_scope_set = false; recv_pkt_rule_match_mask = 0; if (pkt->isRouteFlood()) { bool incoming_region_allowed = false; RegionEntry* incoming_region = NULL; if (pkt->getRouteType() == ROUTE_TYPE_TRANSPORT_FLOOD) { incoming_region = region_map.findMatch(pkt, REGION_DENY_FLOOD); incoming_region_allowed = incoming_region != NULL; } else if (pkt->getRouteType() == ROUTE_TYPE_FLOOD) { incoming_region_allowed = (region_map.getWildcard().flags & REGION_DENY_FLOOD) == 0; } recv_pkt_rule_match_mask = flood_rules.evaluate( pkt, isTempRadioActive(), incoming_region_allowed, incoming_region); bool scope_set = false; scope_changed = flood_rules.applyScope( pkt, recv_pkt_rule_match_mask, scope_set, fast_track_scope_change); recv_pkt_regionless_scope_set = scope_set; } #endif if (pkt->getRouteType() == ROUTE_TYPE_TRANSPORT_FLOOD) { recv_pkt_region = region_map.findMatch(pkt, REGION_DENY_FLOOD); } else if (pkt->getRouteType() == ROUTE_TYPE_FLOOD) { if (region_map.getWildcard().flags & REGION_DENY_FLOOD) { recv_pkt_region = NULL; } else { recv_pkt_region = ®ion_map.getWildcard(); } } else { recv_pkt_region = NULL; } mesh::DispatcherAction action = Mesh::onRecvPacket(pkt); #if defined(MESHCORE_ESP32_FULL_PROFILE) if (scope_changed && action != ACTION_RELEASE && action != ACTION_MANUAL_HOLD) { if (fast_track_scope_change) { action = ACTION_RETRANSMIT(0); } else { uint8_t priority = (action >> 24) - 1; action = ACTION_RETRANSMIT_DELAYED( priority, getSlowFloodRuleRetransmitDelay(pkt)); } } #endif return action; } void MyMesh::onAnonDataRecv(mesh::Packet *packet, const uint8_t *secret, const mesh::Identity &sender, uint8_t *data, size_t len) { if (packet->getPayloadType() == PAYLOAD_TYPE_ANON_REQ) { // received an initial request by a possible admin // client (unknown at this stage) uint32_t sender_timestamp, sender_sync_since; memcpy(&sender_timestamp, data, 4); memcpy(&sender_sync_since, &data[4], 4); // sender's "sync messags SINCE x" timestamp data[len] = 0; // ensure null terminator ClientInfo* client = NULL; if (data[8] == 0) { // blank password, just check if sender is in ACL client = acl.getClient(sender.pub_key, PUB_KEY_SIZE); if (client == NULL) { #if MESH_DEBUG MESH_DEBUG_PRINTLN("Login, sender not in ACL"); #endif } } if (client == NULL) { uint8_t perm; if (strcmp((char *)&data[8], _prefs.password) == 0) { // check for valid admin password perm = PERM_ACL_ADMIN; } else { if (strcmp((char *)&data[8], _prefs.guest_password) == 0) { // check the room/public password perm = PERM_ACL_READ_WRITE; } else if (_prefs.allow_read_only) { perm = PERM_ACL_GUEST; } else { MESH_DEBUG_PRINTLN("Incorrect room password"); return; // no response. Client will timeout } } client = acl.putClient(sender, 0); // add to known clients (if not already known) if (client == NULL) { MESH_DEBUG_PRINTLN("Login rejected: ACL is full of protected contacts"); return; } if (sender_timestamp <= client->last_timestamp) { MESH_DEBUG_PRINTLN("possible replay attack!"); return; } MESH_DEBUG_PRINTLN("Login success!"); client->last_timestamp = sender_timestamp; client->extra.room.sync_since = sender_sync_since; client->extra.room.pending_ack = 0; client->extra.room.push_failures = 0; client->last_activity = getRTCClock()->getCurrentTime(); client->permissions &= ~0x03; client->permissions |= perm; memcpy(client->shared_secret, secret, PUB_KEY_SIZE); dirty_contacts_expiry = futureMillis(LAZY_CONTACTS_WRITE_DELAY); } if (packet->isRouteFlood()) { client->out_path_len = OUT_PATH_UNKNOWN; // need to rediscover out_path } uint32_t now = getRTCClock()->getCurrentTimeUnique(); memcpy(reply_data, &now, 4); // response packets always prefixed with timestamp // TODO: maybe reply with count of messages waiting to be synced for THIS client? reply_data[4] = RESP_SERVER_LOGIN_OK; reply_data[5] = 0; // Legacy: was recommended keep-alive interval (secs / 16) reply_data[6] = (client->isAdmin() ? 1 : (client->permissions == 0 ? 2 : 0)); // LEGACY: reply_data[7] = getUnsyncedCount(client); reply_data[7] = client->permissions; // NEW getRNG()->random(&reply_data[8], 4); // random blob to help packet-hash uniqueness reply_data[12] = FIRMWARE_VER_LEVEL; // New field next_push = futureMillis(PUSH_NOTIFY_DELAY_MILLIS); // delay next push, give RESPONSE packet time to arrive first if (packet->isRouteFlood()) { // let this sender know path TO here, so they can use sendDirect(), and ALSO encode the response mesh::Packet *path = createPathReturn(sender, client->shared_secret, packet->path, packet->path_len, PAYLOAD_TYPE_RESPONSE, reply_data, 13); if (path) sendFloodReply(path, SERVER_RESPONSE_DELAY, packet->getPathHashSize()); } else { mesh::Packet *reply = createDatagram(PAYLOAD_TYPE_RESPONSE, sender, client->shared_secret, reply_data, 13); if (reply) { if (client->out_path_len != OUT_PATH_UNKNOWN) { // we have an out_path, so send DIRECT sendDirect(reply, client->out_path, client->out_path_len, SERVER_RESPONSE_DELAY); } else { sendFloodReply(reply, SERVER_RESPONSE_DELAY, packet->getPathHashSize()); } } } } } int MyMesh::searchPeersByHash(const uint8_t *hash) { int n = 0; #if defined(WITH_MQTT_NEIGHBORS) if (neighbor_discover_active) { for (int i = 0; i < neighbor_discover_count && n < MAX_CLIENTS; i++) { auto& entry = neighbor_discover[i]; // ACL clients already have a matching peer entry and shared secret. Adding // a second overlay entry would decrypt first and intercept their normal // CLI/request traffic for the duration of discovery. if (acl.getClient(entry.id.pub_key, PUB_KEY_SIZE) != nullptr) continue; if (entry.heard_timestamp > 0 && entry.id.isHashMatch(hash)) { matching_peer_indexes[n++] = NEIGHBOR_DISCOVER_PEER_BASE + i; } } } #endif for (int i = 0; i < acl.getNumClients() && n < MAX_CLIENTS; i++) { if (acl.getClientByIdx(i)->id.isHashMatch(hash)) { matching_peer_indexes[n++] = i; // store the INDEXES of matching contacts (for subsequent 'peer' methods) } } return n; } void MyMesh::getPeerSharedSecret(uint8_t *dest_secret, int peer_idx) { int i = matching_peer_indexes[peer_idx]; #if defined(WITH_MQTT_NEIGHBORS) if (neighbor_discover_active && i >= NEIGHBOR_DISCOVER_PEER_BASE) { int oi = i - NEIGHBOR_DISCOVER_PEER_BASE; if (oi >= 0 && oi < neighbor_discover_count) { self_id.calcSharedSecret(dest_secret, neighbor_discover[oi].id); return; } } #endif if (i >= 0 && i < acl.getNumClients()) { // lookup pre-calculated shared_secret memcpy(dest_secret, acl.getClientByIdx(i)->shared_secret, PUB_KEY_SIZE); } else { MESH_DEBUG_PRINTLN("getPeerSharedSecret: Invalid peer idx: %d", i); } } void MyMesh::onPeerDataRecv(mesh::Packet *packet, uint8_t type, int sender_idx, const uint8_t *secret, uint8_t *data, size_t len) { int i = matching_peer_indexes[sender_idx]; #if defined(WITH_MQTT_NEIGHBORS) // Overlay response: a heard neighbour (not an ACL client) answering our // anon-regions scope query. Consume it and stop -- it is not a client packet. if (neighbor_discover_active && i >= NEIGHBOR_DISCOVER_PEER_BASE) { int oi = i - NEIGHBOR_DISCOVER_PEER_BASE; if (type == PAYLOAD_TYPE_RESPONSE && oi >= 0 && oi < neighbor_discover_count) { handleNeighborDiscoverResponse(oi, data, len); } return; } #endif if (i < 0 || i >= acl.getNumClients()) { // get from our known_clients table (sender SHOULD already be known in this context) MESH_DEBUG_PRINTLN("onPeerDataRecv: invalid peer idx: %d", i); return; } auto client = acl.getClientByIdx(i); #if defined(WITH_MQTT_NEIGHBORS) // A neighbour that IS an ACL client resolves to a normal index above, so a // scope-query response from it lands here -- match it against the overlay. if (neighbor_discover_active && type == PAYLOAD_TYPE_RESPONSE) { for (int oi = 0; oi < neighbor_discover_count; oi++) { if (client->id.matches(neighbor_discover[oi].id) && handleNeighborDiscoverResponse(oi, data, len)) { return; } } } #endif if (type == PAYLOAD_TYPE_TXT_MSG && len > 5) { // a CLI command or new Post uint32_t sender_timestamp; memcpy(&sender_timestamp, data, 4); // timestamp (by sender's RTC clock - which could be wrong) uint8_t flags = (data[4] >> 2); // message attempt number, and other flags if (!(flags == TXT_TYPE_PLAIN || flags == TXT_TYPE_CLI_DATA)) { MESH_DEBUG_PRINTLN("onPeerDataRecv: unsupported command flags received: flags=%02x", (uint32_t)flags); return; } // len can be > original length, but 'text' will be padded with zeroes data[len] = 0; // need to make a C string again, with null terminator const char* text = (const char*)&data[5]; const size_t text_len = strlen(text); uint8_t temp[5 + mesh::RemoteCliReplyCache::MAX_REPLY_TEXT + 1]; temp[5] = 0; bool send_ack = false; if (flags == TXT_TYPE_PLAIN) { const bool is_guest = (client->permissions & PERM_ACL_ROLE_MASK) == PERM_ACL_GUEST; if (is_guest) { if (sender_timestamp < client->extra.room.last_post_timestamp) { MESH_DEBUG_PRINTLN("onPeerDataRecv: stale room post detected"); return; } client->extra.room.last_post_timestamp = sender_timestamp; } else { uint8_t message_fingerprint[MAX_HASH_SIZE]; mesh::Utils::sha256(message_fingerprint, sizeof(message_fingerprint), client->id.pub_key, PUB_KEY_SIZE, (const uint8_t*)text, text_len); const auto replay_decision = recent_room_posts.classifyAndRemember( message_fingerprint, sender_timestamp, client->extra.room.last_post_timestamp); using ReplayDecision = mesh::LogicalMessageCache<ROOM_MESSAGE_CACHE_SIZE>::ReplayDecision; if (replay_decision == ReplayDecision::StaleOrMismatched) { MESH_DEBUG_PRINTLN("onPeerDataRecv: stale or mismatched room post detected"); return; } if (replay_decision == ReplayDecision::NewMessage) { addPost(client, text); } // Exact retries are ACKed again regardless of whether newer posts have // advanced this client's room-post replay timestamp. send_ack = true; } } else { // TXT_TYPE_CLI_DATA uint32_t request_id = sender_timestamp; mesh::RemoteCliRequest::parse(data, len, 5, request_id); const uint32_t command_fingerprint = mesh::RemoteCliReplyCache::fingerprint(text, text_len); const char* cached_response = NULL; const bool cached_retry = remote_cli_reply_cache.lookup( client->id.pub_key, request_id, command_fingerprint, &cached_response); if (sender_timestamp < client->last_timestamp && !cached_retry) { MESH_DEBUG_PRINTLN("onPeerDataRecv: possible replay attack detected"); return; } const bool repeated_timestamp = sender_timestamp == client->last_timestamp; if (sender_timestamp > client->last_timestamp) { client->last_timestamp = sender_timestamp; } if (cached_retry) { MESH_DEBUG_PRINTLN("onPeerDataRecv: replaying cached remote CLI reply"); size_t cached_len = strlen(cached_response); if (cached_len > mesh::RemoteCliReplyCache::MAX_REPLY_TEXT) { cached_len = mesh::RemoteCliReplyCache::MAX_REPLY_TEXT; } memcpy(&temp[5], cached_response, cached_len); temp[5 + cached_len] = 0; temp[4] = (TXT_TYPE_CLI_DATA << 2); } else if (repeated_timestamp) { MESH_DEBUG_PRINTLN("onPeerDataRecv: duplicate remote CLI request has no cached reply"); return; } else if (client->isAdmin()) { handleCommand(sender_timestamp, (char*)text, (char*)&temp[5], i, packet->getPathHashSize()); temp[5 + mesh::RemoteCliReplyCache::MAX_REPLY_TEXT] = 0; if (temp[5] == 0) strcpy((char*)&temp[5], "OK"); remote_cli_reply_cache.remember(client->id.pub_key, request_id, command_fingerprint, (char*)&temp[5]); temp[4] = (TXT_TYPE_CLI_DATA << 2); } else { const char* error = "Err - admin permission required"; strcpy((char*)&temp[5], error); remote_cli_reply_cache.remember(client->id.pub_key, request_id, command_fingerprint, error); temp[4] = (TXT_TYPE_CLI_DATA << 2); } // CLI_DATA replies are the result signal; no separate ACK is expected. } uint32_t now = getRTCClock()->getCurrentTimeUnique(); client->last_activity = now; client->extra.room.push_failures = 0; // reset so push can resume (if prev failed) uint32_t delay_millis; if (send_ack) { uint32_t ack_hash; // prove receipt of this timestamp, attempt, text, and sender mesh::Utils::sha256((uint8_t*)&ack_hash, 4, data, 5 + text_len, client->id.pub_key, PUB_KEY_SIZE); if (client->out_path_len == OUT_PATH_UNKNOWN) { mesh::Packet *ack = createAck(ack_hash); if (ack) sendFloodReply(ack, TXT_ACK_DELAY, packet->getPathHashSize()); delay_millis = TXT_ACK_DELAY + REPLY_DELAY_MILLIS; } else { uint32_t d = TXT_ACK_DELAY; if (getExtraAckTransmitCount() > 0) { mesh::Packet *a1 = createMultiAck(ack_hash, 1); if (a1) sendDirect(a1, client->out_path, client->out_path_len, d); d += 300; } mesh::Packet *a2 = createAck(ack_hash); if (a2) sendDirect(a2, client->out_path, client->out_path_len, d); delay_millis = d + REPLY_DELAY_MILLIS; } } else { delay_millis = 0; } int reply_text_len = strlen((char*)&temp[5]); if (reply_text_len > 0) { if (now == sender_timestamp) { // WORKAROUND: the two timestamps need to be different, in the CLI view now++; } memcpy(temp, &now, 4); // mostly an extra blob to help make packet_hash unique // calc expected ACK reply // mesh::Utils::sha256((uint8_t *)&expected_ack_crc, 4, temp, 5 + reply_text_len, self_id.pub_key, // PUB_KEY_SIZE); auto reply = createDatagram(PAYLOAD_TYPE_TXT_MSG, client->id, secret, temp, 5 + reply_text_len); if (reply) { if (client->out_path_len == OUT_PATH_UNKNOWN) { sendFloodReply(reply, delay_millis + SERVER_RESPONSE_DELAY, packet->getPathHashSize()); } else { sendDirect(reply, client->out_path, client->out_path_len, delay_millis + SERVER_RESPONSE_DELAY); } } } } else if (type == PAYLOAD_TYPE_REQ && len >= 5) { uint32_t sender_timestamp; memcpy(&sender_timestamp, data, 4); // timestamp (by sender's RTC clock - which could be wrong) if (sender_timestamp < client->last_timestamp) { // prevent replay attacks MESH_DEBUG_PRINTLN("onPeerDataRecv: possible replay attack detected"); } else { client->last_timestamp = sender_timestamp; uint32_t now = getRTCClock()->getCurrentTime(); client->last_activity = now; // <-- THIS will keep client connection alive client->extra.room.push_failures = 0; // reset so push can resume (if prev failed) if (data[4] == REQ_TYPE_KEEP_ALIVE && packet->isRouteDirect()) { // request type uint32_t forceSince = 0; if (len >= 9) { // optional - last post_timestamp client received memcpy(&forceSince, &data[5], 4); // NOTE: this may be 0, if part of decrypted PADDING! } else { memcpy(&data[5], &forceSince, 4); // make sure there are zeroes in payload (for ack_hash calc below) } if (forceSince > 0) { client->extra.room.sync_since = forceSince; // force-update the 'sync since' } client->extra.room.pending_ack = 0; // TODO: Throttle KEEP_ALIVE requests! // if client sends too quickly, evict() // RULE: only send keep_alive response DIRECT! if (client->out_path_len != OUT_PATH_UNKNOWN) { uint32_t ack_hash; // calc ACK to prove to sender that we got request mesh::Utils::sha256((uint8_t *)&ack_hash, 4, data, 9, client->id.pub_key, PUB_KEY_SIZE); auto reply = createAck(ack_hash); if (reply) { reply->payload[reply->payload_len++] = getUnsyncedCount(client); // NEW: add unsynced counter to end of ACK packet sendDirect(reply, client->out_path, client->out_path_len, SERVER_RESPONSE_DELAY); } } } else { int reply_len = handleRequest(client, sender_timestamp, &data[4], len - 4); if (reply_len > 0) { // valid command if (packet->isRouteFlood()) { // let this sender know path TO here, so they can use sendDirect(), and ALSO encode the response mesh::Packet *path = createPathReturn(client->id, secret, packet->path, packet->path_len, PAYLOAD_TYPE_RESPONSE, reply_data, reply_len); if (path) sendFloodReply(path, SERVER_RESPONSE_DELAY, packet->getPathHashSize()); } else { mesh::Packet *reply = createDatagram(PAYLOAD_TYPE_RESPONSE, client->id, secret, reply_data, reply_len); if (reply) { if (client->out_path_len != OUT_PATH_UNKNOWN) { // we have an out_path, so send DIRECT sendDirect(reply, client->out_path, client->out_path_len, SERVER_RESPONSE_DELAY); } else { sendFloodReply(reply, SERVER_RESPONSE_DELAY, packet->getPathHashSize()); } } } } } } } } bool MyMesh::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) { // TODO: prevent replay attacks int i = matching_peer_indexes[sender_idx]; if (i >= 0 && i < acl.getNumClients()) { // get from our known_clients table (sender SHOULD already be known in this context) MESH_DEBUG_PRINTLN("PATH to client, path_len=%d", (uint32_t)path_len); auto client = acl.getClientByIdx(i); client->out_path_len = mesh::Packet::copyPath(client->out_path, path, path_len); // store a copy of path, for sendDirect() client->last_activity = getRTCClock()->getCurrentTime(); } else { MESH_DEBUG_PRINTLN("onPeerPathRecv: invalid peer idx: %d", i); } if (extra_type == PAYLOAD_TYPE_ACK && extra_len >= 4) { // also got an encoded ACK! processAck(extra); } // NOTE: no reciprocal path send!! return false; } void MyMesh::onAckRecv(mesh::Packet *packet, uint32_t ack_crc) { if (processAck((uint8_t *)&ack_crc)) { packet->markDoNotRetransmit(); // ACK was for this node, so don't retransmit } } void MyMesh::onAdvertRecv(mesh::Packet* packet, const mesh::Identity& id, uint32_t timestamp, const uint8_t* app_data, size_t app_data_len) { mesh::Mesh::onAdvertRecv(packet, id, timestamp, app_data, app_data_len); _clock_sync.observeVerifiedAdvert(packet, id, timestamp); #if defined(WITH_MQTT_NEIGHBORS) bool is_share = packet->hasTransportCodes() && packet->transport_codes[0] == 0 && packet->transport_codes[1] == 0; if (packet->getPathHashCount() == 0 && !is_share) { AdvertDataParser parser(app_data, app_data_len); if (parser.isValid() && parser.getType() == ADV_TYPE_REPEATER) { putNeighbour(id, timestamp, packet->getSNR()); } } #endif } void MyMesh::onGroupPacketRecv(mesh::Packet* packet) { _clock_sync.observeGroupPacket(packet); } #if defined(WITH_MQTT_NEIGHBORS) #define CTL_TYPE_NODE_DISCOVER_REQ 0x80 #define CTL_TYPE_NODE_DISCOVER_RESP 0x90 void MyMesh::putNeighbour(const mesh::Identity &id, uint32_t timestamp, float snr) { // find existing neighbour, else use least recently updated uint32_t oldest_timestamp = 0xFFFFFFFF; NeighbourInfo *neighbour = &neighbours[0]; for (int i = 0; i < MAX_NEIGHBOURS; i++) { // if neighbour already known, we should update it if (id.matches(neighbours[i].id)) { neighbour = &neighbours[i]; break; } // otherwise we should update the least recently updated neighbour if (neighbours[i].heard_timestamp < oldest_timestamp) { neighbour = &neighbours[i]; oldest_timestamp = neighbour->heard_timestamp; } } // update neighbour info neighbour->id = id; neighbour->advert_timestamp = timestamp; neighbour->heard_timestamp = getRTCClock()->getCurrentTime(); neighbour->snr = (int8_t)(snr * 4); } void MyMesh::onControlDataRecv(mesh::Packet* packet) { uint8_t type = packet->payload[0] & 0xF0; // just test upper 4 bits // A room server is ADV_TYPE_ROOM, so it does NOT answer node-discover requests // (those filter for repeaters). It only records repeater responses to its own // discovery, to build the neighbour table. if (type == CTL_TYPE_NODE_DISCOVER_RESP && packet->payload_len >= 6) { uint8_t node_type = packet->payload[0] & 0x0F; if (node_type != ADV_TYPE_REPEATER) { return; } if (packet->payload_len < 6 + PUB_KEY_SIZE) { MESH_DEBUG_PRINTLN("onControlDataRecv: DISCOVER_RESP pubkey too short: %d", (uint32_t)packet->payload_len); return; } if (pending_discover_tag == 0 || millisHasNowPassed(pending_discover_until)) { pending_discover_tag = 0; return; } uint32_t tag; memcpy(&tag, &packet->payload[2], 4); if (tag != pending_discover_tag) { return; } mesh::Identity id(&packet->payload[6]); if (id.matches(self_id)) { return; } putNeighbour(id, getRTCClock()->getCurrentTime(), packet->getSNR()); } } void MyMesh::sendNodeDiscoverReq() { uint8_t data[10]; data[0] = CTL_TYPE_NODE_DISCOVER_REQ; // prefix_only=0 data[1] = (1 << ADV_TYPE_REPEATER); getRNG()->random(&data[2], 4); // tag memcpy(&pending_discover_tag, &data[2], 4); pending_discover_until = futureMillis(60000); uint32_t since = 0; memcpy(&data[6], &since, 4); auto pkt = createControlData(data, sizeof(data)); if (pkt) { sendZeroHop(pkt); } } #endif // WITH_MQTT_NEIGHBORS MyMesh::MyMesh(mesh::MainBoard &board, mesh::Radio &radio, mesh::MillisecondClock &ms, mesh::RNG &rng, mesh::RTCClock &rtc, mesh::MeshTables &tables) : mesh::Mesh(radio, ms, rng, rtc, *createObserverPacketManager(32), tables), region_map(key_store), temp_map(key_store), _cli(board, rtc, sensors, region_map, acl, &_prefs, this), _clock_sync(radio, ms, rtc, acl, sensors, _prefs.tx_delay_factor, this), telemetry(MAX_PACKET_PAYLOAD - 4) #ifdef WITH_MQTT_BRIDGE , bridge(nullptr) #endif { last_millis = 0; uptime_millis = 0; next_local_advert = next_flood_advert = 0; dirty_contacts_expiry = 0; _logging = false; region_load_active = false; set_radio_at = revert_radio_at = 0; active_cr = LORA_CR; temp_radio_applied = false; saved_radio_apply_pending = false; radio_apply_retry_at = 0; radio_apply_failures = 0; recv_pkt_region = NULL; #if defined(MESHCORE_ESP32_FULL_PROFILE) recv_pkt_rule_match_mask = 0; recv_pkt_regionless_scope_set = false; #endif // defaults _prefs.airtime_factor = 1.0; // one half _prefs.rx_delay_base = 0.0f; // off by default, was 10.0 _prefs.tx_delay_factor = 0.5f; // was 0.25f; _prefs.direct_tx_delay_factor = 0.2f; // was zero StrHelper::strncpy(_prefs.node_name, ADVERT_NAME, sizeof(_prefs.node_name)); _prefs.node_lat = ADVERT_LAT; _prefs.node_lon = ADVERT_LON; StrHelper::strncpy(_prefs.password, ADMIN_PASSWORD, sizeof(_prefs.password)); _prefs.freq = LORA_FREQ; _prefs.sf = LORA_SF; _prefs.bw = LORA_BW; _prefs.cr = LORA_CR; _prefs.tx_power_dbm = LORA_TX_POWER; _prefs.disable_fwd = 1; _prefs.advert_interval = 1; // default to 2 minutes for NEW installs _prefs.flood_advert_interval = 47; // 47 hours _prefs.flood_max = 64; _prefs.flood_max_unscoped = 64; _prefs.flood_max_advert = 8; _prefs.flood_channel_data_enabled = 1; _prefs.flood_retry_group_max_path = FLOOD_RETRY_GROUP_MAX_PATH_DEFAULT; _prefs.interference_threshold = 0; // disabled _prefs.radio_fem_rxgain = 1; // LoRa FEM RX gain on by default (FEM boards) _prefs.cad_enabled = DEFAULT_CAD_ENABLED; // Cascade defaults CAD on; target default remains off _prefs.powersaving_enabled = DEFAULT_POWERSAVING_ENABLED ? 1 : 0; _prefs.rx_powersaving_enabled = DEFAULT_RXPS_ENABLED ? 1 : 0; _prefs.rx_ps_level = DEFAULT_RXPS_LEVEL; _prefs.rx_ps_preamble = DEFAULT_RXPS_PREAMBLE; _prefs.rx_ps_rx_us = RX_POWERSAVING_DEFAULT_RX_US; _prefs.rx_ps_sleep_us = RX_POWERSAVING_DEFAULT_SLEEP_US; recalcRxPowerSavingFromLevel(_prefs.rx_ps_level, _prefs.sf, _prefs.bw, _prefs.rx_ps_preamble, &_prefs.rx_ps_rx_us, &_prefs.rx_ps_sleep_us); #ifdef ROOM_PASSWORD StrHelper::strncpy(_prefs.guest_password, ROOM_PASSWORD, sizeof(_prefs.guest_password)); #endif // GPS defaults _prefs.gps_enabled = 0; _prefs.gps_interval = 0; _prefs.advert_loc_policy = ADVERT_LOC_PREFS; #if defined(USE_SX1262) || defined(USE_SX1268) || defined(USE_LR1110) \ || defined(USE_LR2021) || defined(SX126X_RX_BOOSTED_GAIN) \ || defined(RX_BOOSTED_GAIN) _prefs.rx_boosted_gain = mesh::radio::configuredRxBoostedGainDefault(); #endif _prefs.radio_fem_rxgain = 1; _prefs.radio_fem_txgain = 0; // Observer defaults (alert.*, etc.) moved to applyMQTTDefaults() - they live // in /mqtt_prefs now, not NodePrefs. // bridge defaults (same as repeater) _prefs.bridge_enabled = 1; // enabled _prefs.bridge_delay = 500; // milliseconds _prefs.bridge_pkt_src = 1; // logRx (RX packets) _prefs.bridge_baud = 115200; // baud rate _prefs.bridge_channel = 1; // channel 1 // MQTT/WiFi/timezone defaults live in /mqtt_prefs now (see applyMQTTDefaults). next_post_idx = 0; next_client_idx = 0; next_push = 0; memset(posts, 0, sizeof(posts)); _num_posted = _num_post_pushes = 0; #if defined(WITH_MQTT_NEIGHBORS) pending_discover_tag = 0; pending_discover_until = 0; neighbor_discover_count = 0; neighbor_discover_next = 0; neighbor_discover_publish_count = 0; neighbor_discover_queried_count = 0; neighbor_discover_json_size = 0; neighbor_discover_truncated = false; neighbor_discover_active = false; neighbor_table_refresh_active = false; neighbor_table_refresh_periodic = false; neighbor_discover_until = 0; neighbor_discover_request = NULL; next_neighbors_publish = 0; self_scopes_buf[0] = 0; self_default_scope_buf[0] = 0; neighbor_discover_origin[0] = 0; memset(neighbours, 0, sizeof(neighbours)); #endif memset(default_scope.key, 0, sizeof(default_scope.key)); } void MyMesh::begin(FILESYSTEM *fs) { mesh::Mesh::begin(); _fs = fs; // load persisted prefs _cli.loadPrefs(_fs); acl.load(_fs, self_id); region_map.load(_fs); _clock_sync.begin(_fs); #if defined(MESHCORE_ESP32_FULL_PROFILE) flood_rules.begin(_fs, ®ion_map); #endif // establish default-scope { RegionEntry* r = region_map.getDefaultRegion(); if (r) { region_map.getTransportKeysFor(*r, &default_scope, 1); } else { #ifdef DEFAULT_FLOOD_SCOPE_NAME r = region_map.findByName(DEFAULT_FLOOD_SCOPE_NAME); if (r == NULL) { r = region_map.putRegion(DEFAULT_FLOOD_SCOPE_NAME, 0); // auto-create the default scope region if (r) { r->flags = 0; } // Allow-flood } if (r) { region_map.setDefaultRegion(r); region_map.getTransportKeysFor(*r, &default_scope, 1); } #endif } } saved_radio_apply_pending = !applySavedRadioParams(); if (!saved_radio_apply_pending) { radio_driver.setTxPower(_prefs.tx_power_dbm); radio_driver.setRxBoostedGainMode(_prefs.rx_boosted_gain); } const bool fem_gain_changed = board.canControlLoRaFemLna() && board.isLoRaFemLnaEnabled() != (_prefs.radio_fem_rxgain != 0); if (board.setLoRaFemLnaEnabled(_prefs.radio_fem_rxgain) && fem_gain_changed) { _radio->recalibrateNoiseFloor(); } board.setLoRaFemPaGainEnabled(_prefs.radio_fem_txgain); setRxPowerSaving(_prefs.rx_powersaving_enabled, _prefs.rx_ps_rx_us, _prefs.rx_ps_sleep_us); updateAdvertTimer(); updateFloodAdvertTimer(); board.setAdcMultiplier(_prefs.adc_multiplier); #if ENV_INCLUDE_GPS == 1 applyGpsPrefs(); #endif #ifdef WITH_MQTT_BRIDGE if (_prefs.bridge_enabled) { // Defer construction to avoid static init crashes on ESP32 classic MQTTNodeInfo node_info; node_info.node_name = _prefs.node_name; node_info.freq = &_prefs.freq; node_info.bw = &_prefs.bw; node_info.sf = &_prefs.sf; node_info.cr = &_prefs.cr; node_info.repeat_flag = &_prefs.disable_fwd; node_info.repeat_when_nonzero = false; bridge = new MQTTBridge(node_info, _cli.getObserverPrefs(), getRTCClock(), &self_id); if (bridge) { // Set device public key for MQTT topics char device_id[65]; mesh::LocalIdentity self_id = getSelfId(); mesh::Utils::toHex(device_id, self_id.pub_key, PUB_KEY_SIZE); MESH_DEBUG_PRINTLN("Setting device ID: %s", device_id); bridge->setDeviceID(device_id); // Set firmware version bridge->setFirmwareVersion(getFirmwareVer()); // Set board model bridge->setBoardModel(_cli.getBoard()->getManufacturerName()); // Set build date bridge->setBuildDate(getBuildDate()); // Set stats sources for automatic stats collection bridge->setStatsSources(this, _radio, _cli.getBoard(), _ms); bridge->begin(); } } #endif // Wire fault-alert reporter. begin() is safe regardless of bridge state. // Passing `this` as the callbacks lets the reporter resolve a TransportKey // scope (alert.region override, falling back to default_scope) so alert // floods ride the same scope as adverts/channel messages. #ifdef WITH_MQTT_BRIDGE _alerter.begin(&_prefs, _cli.getObserverPrefs(), this, this); _alerter.setBridge(bridge); #endif #if defined(WITH_WEBCONFIG) && !defined(WEBCONFIG_NO_AUTO_AP) bool start_webui = WebConfigServer::loadEnabled(false); #ifdef WITH_MQTT_BRIDGE start_webui = start_webui || _cli.getObserverPrefs()->wifi_ssid[0] == 0; if (start_webui && _cli.getObserverPrefs()->wifi_ssid[0] == 0) { if (bridge && bridge->isRunning()) bridge->end(); } #endif if (start_webui) { char wc_reply[160]; startWebConfig(false, wc_reply); Serial.println(wc_reply); } #endif } bool MyMesh::applySavedRadioParams() { uint32_t timings[2] = {_prefs.rx_ps_rx_us, _prefs.rx_ps_sleep_us}; const uint32_t* applied_timings = _prefs.rx_powersaving_enabled && radio_driver.supportsRxPowerSaving() ? timings : NULL; if (!radio_driver.setParams( _prefs.freq, _prefs.bw, _prefs.sf, _prefs.cr, applied_timings)) { return false; } active_cr = _prefs.cr; return true; } bool MyMesh::sendFloodScoped(const TransportKey& scope, mesh::Packet* pkt, uint32_t delay_millis, uint8_t path_hash_size) { if (scope.isNull()) { return sendFlood(pkt, delay_millis, path_hash_size); } else { uint16_t codes[2]; codes[0] = scope.calcTransportCode(pkt); codes[1] = 0; // REVISIT: set to 'home' Region, for sender/return region? return sendFlood(pkt, codes, delay_millis, path_hash_size); } } bool MyMesh::resolveAlertScope(TransportKey& dest) { // Same resolution policy as simple_repeater: alert.region > default_scope. #ifdef WITH_MQTT_BRIDGE const char* alert_region = _cli.getObserverPrefs()->alert_region; if (alert_region[0]) { auto r = region_map.findByNamePrefix(alert_region); if (r && region_map.getTransportKeysFor(*r, &dest, 1) > 0 && !dest.isNull()) { return true; } } #endif if (!default_scope.isNull()) { dest = default_scope; return true; } return false; } void MyMesh::sendFloodReply(mesh::Packet* packet, unsigned long delay_millis, uint8_t path_hash_size) { TransportKey req_scope; bool is_wildcard = recv_pkt_region != NULL && recv_pkt_region->isWildcard(); bool req_scope_known = recv_pkt_region != NULL && !is_wildcard && region_map.getTransportKeysFor(*recv_pkt_region, &req_scope, 1) > 0; switch (mesh::chooseReplyScope(req_scope_known, is_wildcard, !default_scope.isNull())) { case mesh::REPLY_SCOPE_REQUEST: sendFloodScoped(req_scope, packet, delay_millis, path_hash_size); // reply with same scope as request break; case mesh::REPLY_SCOPE_DEFAULT: // requester's scope is unknown: DIRECT request (no transport codes), or code matched no Region. // un-scoped would be dropped at hop 0 by repeaters running flood.max.unscoped=0 sendFloodScoped(default_scope, packet, delay_millis, path_hash_size); break; case mesh::REPLY_SCOPE_NONE: sendFlood(packet, delay_millis, path_hash_size); // send un-scoped break; } } void MyMesh::applyTempRadioParams(float freq, float bw, uint8_t sf, uint8_t cr, int timeout_mins) { radio_apply_retry_at = 0; radio_apply_failures = 0; set_radio_at = futureMillis(2000); // give CLI reply some time to be sent back, before applying temp radio params pending_freq = freq; pending_bw = bw; pending_sf = sf; pending_cr = cr; revert_radio_at = futureMillis(2000 + timeout_mins * 60 * 1000); // schedule when to revert radio params } bool MyMesh::scheduleNormalRadio() { // A one-tick deadline cancels a not-yet-applied window or restores an active // one as soon as the command reply has drained from the outbound queue. set_radio_at = 0; revert_radio_at = futureMillis(1); radio_apply_retry_at = 0; radio_apply_failures = 0; return true; } bool MyMesh::formatFileSystem() { #if defined(NRF52_PLATFORM) return InternalFS.format(); #elif defined(RP2040_PLATFORM) return LittleFS.format(); #elif defined(ESP32) return SPIFFS.format(); #else #error "need to implement file system erase" return false; #endif } void MyMesh::sendSelfAdvertisement(int delay_millis, bool flood) { mesh::Packet *pkt = createSelfAdvert(); if (pkt) { if (flood) { sendFloodScoped(default_scope, pkt, delay_millis, _prefs.path_hash_mode + 1); } else { sendZeroHop(pkt, delay_millis); } } else { MESH_DEBUG_PRINTLN("ERROR: unable to create advertisement packet!"); } } void MyMesh::updateAdvertTimer() { if (_prefs.advert_interval > 0) { // schedule local advert timer next_local_advert = futureMillis((int)((uint32_t)_prefs.advert_interval * 2 * 60 * 1000)); } else { next_local_advert = 0; // stop the timer } } void MyMesh::updateFloodAdvertTimer() { if (_prefs.flood_advert_interval > 0) { // schedule flood advert timer next_flood_advert = futureMillis(((uint32_t)_prefs.flood_advert_interval) * 60 * 60 * 1000); } else { next_flood_advert = 0; // stop the timer } } void MyMesh::dumpLogFile() { #if defined(RP2040_PLATFORM) File f = _fs->open(PACKET_LOG_FILE, "r"); #else File f = _fs->open(PACKET_LOG_FILE); #endif if (f) { while (f.available()) { int c = f.read(); if (c < 0) break; Serial.print((char)c); } f.close(); } } void MyMesh::setTxPower(int8_t power_dbm) { radio_driver.setTxPower(power_dbm); } bool MyMesh::setRxPowerSaving(bool enable, uint32_t rx_us, uint32_t sleep_us) { return radio_driver.setRxPowerSaving(enable, rx_us, sleep_us); } void MyMesh::getRxPsWatchdogCounts(uint32_t* soft, uint32_t* hard) { *soft = radio_driver.getRxPsWatchdogSoftCount(); *hard = radio_driver.getRxPsWatchdogHardCount(); } bool MyMesh::setRxBoostedGain(bool enable) { return radio_driver.setRxBoostedGainMode(enable); } void MyMesh::saveIdentity(const mesh::LocalIdentity &new_id) { #if defined(NRF52_PLATFORM) || defined(STM32_PLATFORM) IdentityStore store(*_fs, ""); #elif defined(ESP32) IdentityStore store(*_fs, "/identity"); #elif defined(RP2040_PLATFORM) IdentityStore store(*_fs, "/identity"); #else #error "need to define saveIdentity()" #endif store.save("_main", new_id); } void MyMesh::startRegionsLoad() { temp_map.resetFrom(region_map); // rebuild regions in a temp instance memset(load_stack, 0, sizeof(load_stack)); load_stack[0] = &temp_map.getWildcard(); region_load_active = true; } bool MyMesh::saveRegions() { return region_map.save(_fs); } void MyMesh::onDefaultRegionChanged(const RegionEntry* r) { if (r) { region_map.getTransportKeysFor(*r, &default_scope, 1); } else { memset(default_scope.key, 0, sizeof(default_scope.key)); } } void MyMesh::clearStats() { radio_driver.resetStats(); resetStats(); ((SimpleMeshTables *)getTables())->resetStats(); } #ifdef WITH_WEBCONFIG void MyMesh::getNodeSnapshot(WebConfigServer::NodeSnapshot& s) { memset(&s, 0, sizeof(s)); StrHelper::strncpy(s.name, _prefs.node_name, sizeof(s.name)); StrHelper::strncpy(s.admin_password, _prefs.password, sizeof(s.admin_password)); s.lat = _prefs.node_lat; s.lon = _prefs.node_lon; s.freq = _prefs.freq; s.bw = _prefs.bw; s.sf = _prefs.sf; s.cr = _prefs.cr; s.tx_power = _prefs.tx_power_dbm; s.airtime_factor = _prefs.airtime_factor; s.rx_delay = _prefs.rx_delay_base; s.tx_delay = _prefs.tx_delay_factor; s.cad = _prefs.cad_enabled; s.rx_gain = _prefs.rx_boosted_gain; s.fem_rx_gain = board.isLoRaFemLnaEnabled(); s.rx_ps_enabled = _prefs.rx_powersaving_enabled; s.rx_ps_level = _prefs.rx_ps_level; s.rx_ps_preamble = _prefs.rx_ps_preamble; s.rx_ps_rx_us = _prefs.rx_ps_rx_us; s.rx_ps_sleep_us = _prefs.rx_ps_sleep_us; s.power_saving = _prefs.powersaving_enabled; s.repeat = !_prefs.disable_fwd; s.advert_interval = _prefs.advert_interval * 2; s.flood_advert_interval = _prefs.flood_advert_interval; s.flood_max = _prefs.flood_max; s.flood_max_advert = _prefs.flood_max_advert; s.flood_max_unscoped = _prefs.flood_max_unscoped; s.loop_detect = _prefs.loop_detect; s.capabilities = WebConfigServer::CAP_LOCATION | WebConfigServer::CAP_AIRTIME | WebConfigServer::CAP_DELAYS | WebConfigServer::CAP_CAD | WebConfigServer::CAP_RX_GAIN | WebConfigServer::CAP_REPEAT | WebConfigServer::CAP_ADVERT | WebConfigServer::CAP_FLOOD | WebConfigServer::CAP_LOOP | WebConfigServer::CAP_WIFI_POWER_SAVE | WebConfigServer::CAP_POWER_SAVING; if (board.canControlLoRaFemLna()) { s.capabilities |= WebConfigServer::CAP_FEM_RX_GAIN; } if (radio_driver.supportsRxPowerSaving()) { s.capabilities |= WebConfigServer::CAP_RX_POWER_SAVING; } } bool MyMesh::startWebConfig(bool force_ap, char* reply) { if (_cli.getBoard()->isOTAUpdateRunning()) { strcpy(reply, "Err: OTA server is running - 'stop ota' first"); return true; } if (_webconfig && (_webconfig->isRunning() || _webconfig->isStopping())) { strcpy(reply, _webconfig->isStopping() ? "Err: webconfig still stopping, retry shortly" : "Err: webconfig already running"); return true; } if (!_webconfig) { void* mqtt_prefs = nullptr; bool owns_wifi = true; #ifdef WITH_MQTT_BRIDGE mqtt_prefs = _cli.getObserverPrefs(); owns_wifi = false; #endif _webconfig = new WebConfigServer(this, mqtt_prefs, owns_wifi, self_id.pub_key, getFirmwareVer(), getBuildDate(), getRole(), _cli.getBoard()->getManufacturerName()); if (!_webconfig) { strcpy(reply, "Err: not enough memory for webconfig"); return true; } } if (force_ap) { #ifdef WITH_MQTT_BRIDGE if (bridge && bridge->isRunning()) { strcpy(reply, "Err: MQTT bridge is running - 'set bridge off' first"); return true; } #endif _webconfig->startSetupMode(reply); } else { _webconfig->startAutoMode(reply); } return true; } bool MyMesh::setWebUIEnabled(bool enabled, char* reply) { if (!WebConfigServer::saveEnabled(enabled)) { strcpy(reply, "Error: failed to save webui setting"); return true; } if (enabled) { if (_webconfig && (_webconfig->isRunning() || _webconfig->isStopping())) { strcpy(reply, "OK - webui on (already active)"); } else { startWebConfig(false, reply); if (strncmp(reply, "WebConfig", 9) == 0) { char tmp[160]; StrHelper::strncpy(tmp, reply, sizeof(tmp)); snprintf(reply, 160, "OK - webui on; %s", tmp); } } } else { if (_webconfig && _webconfig->isRunning()) _webconfig->requestStop(); strcpy(reply, "OK - webui off"); } return true; } bool MyMesh::getWebUIStatus(char* reply) const { const bool enabled = WebConfigServer::loadEnabled(false); if (!_webconfig || (!_webconfig->isRunning() && !_webconfig->isStopping())) { snprintf(reply, 160, "> %s, inactive", enabled ? "on" : "off"); } else if (_webconfig->mode() == WebConfigServer::MODE_SETUP) { char ssid[33], ip[16]; WebConfigServer::getSetupInfo(ssid, sizeof(ssid), ip, sizeof(ip)); snprintf(reply, 160, "> %s, setup AP %s http://%s/", enabled ? "on" : "off", ssid, ip); } else if (_webconfig->mode() == WebConfigServer::MODE_CONNECTING) { snprintf(reply, 160, "> %s, connecting to WiFi", enabled ? "on" : "off"); } else { snprintf(reply, 160, "> %s, http://%s/", enabled ? "on" : "off", WiFi.localIP().toString().c_str()); } return true; } bool MyMesh::getWiFiSSID(char* reply) const { return WebConfigServer::formatWiFiSSID(reply, 160); } bool MyMesh::getWiFiStatus(char* reply) const { return WebConfigServer::formatWiFiStatus(reply, 160); } bool MyMesh::getWiFiPowerSave(char* reply) const { return WebConfigServer::formatWiFiPowerSave(reply, 160); } bool MyMesh::getWiFiCLI(char* reply) const { return WebConfigServer::formatWiFiCliStatus(reply, 160); } bool MyMesh::setWiFiSSID(const char* value, char* reply) { if (WebConfigServer::setStandaloneWiFiSSID(value, reply, 160)) { const bool was_running = _webconfig && _webconfig->isRunning(); if (was_running) _webconfig->requestStop(); if (_webconfig) _webconfig->reloadStandaloneWiFi(); if (was_running) { strcpy(reply, "OK - WiFi SSID saved; WebConfig stopping, start again to apply"); } } return true; } bool MyMesh::setWiFiPassword(const char* value, char* reply) { if (WebConfigServer::setStandaloneWiFiPassword(value, reply, 160)) { const bool was_running = _webconfig && _webconfig->isRunning(); if (was_running) _webconfig->requestStop(); if (_webconfig) _webconfig->reloadStandaloneWiFi(); if (was_running) { strcpy(reply, "OK - WiFi password saved; WebConfig stopping, start again to apply"); } } return true; } bool MyMesh::setWiFiPowerSave(const char* value, char* reply) { if (WebConfigServer::setStandaloneWiFiPowerSave(value, reply, 160) && _webconfig) { _webconfig->reloadStandaloneWiFi(); } return true; } bool MyMesh::setWiFiCLI(const char* value, char* reply) { WebConfigServer::setWiFiCliEnabled(value, reply, 160); return true; } bool MyMesh::stopWebConfig(char* reply) { if (!_webconfig || !_webconfig->isRunning()) { strcpy(reply, "Err: webconfig not running"); return true; } _webconfig->requestStop(); strcpy(reply, "OK - webconfig stopping"); return true; } void MyMesh::onConfigBatchEnd() { _wc_batch_active = false; #ifdef WITH_MQTT_BRIDGE if (_wc_restart_pending) { _wc_restart_pending = false; _wc_slot_restart_mask = 0; restartBridge(); return; } const uint8_t mask = _wc_slot_restart_mask; _wc_slot_restart_mask = 0; for (int i = 0; i < RUNTIME_MQTT_SLOTS; i++) { if (mask & (1U << i)) restartBridgeSlot(i); } #else _wc_restart_pending = false; _wc_slot_restart_mask = 0; #endif } void MyMesh::buildStatsJson(char* buf, size_t buf_size) { char ip[20] = ""; int wifi_rssi = 0; if (WiFi.status() == WL_CONNECTED) { strncpy(ip, WiFi.localIP().toString().c_str(), sizeof(ip) - 1); wifi_rssi = WiFi.RSSI(); } else if (_webconfig && _webconfig->mode() == WebConfigServer::MODE_SETUP) { strncpy(ip, WiFi.softAPIP().toString().c_str(), sizeof(ip) - 1); } char snr_text[16]; StrHelper::ftoaFixed(snr_text, sizeof(snr_text), radio_driver.getLastSNR(), 1); int pos = snprintf(buf, buf_size, "{\"uptime_s\":%lu,\"batt_mv\":%u," "\"heap_free\":%lu,\"heap_min\":%lu,\"heap_max_alloc\":%lu," "\"noise\":%d,\"rssi\":%d,\"snr\":%s," "\"airtime_s\":%lu,\"rx_airtime_s\":%lu," "\"recv\":%lu,\"sent\":%lu,\"rx_err\":%lu," "\"sent_flood\":%lu,\"sent_direct\":%lu,\"recv_flood\":%lu,\"recv_direct\":%lu," "\"tx_queue\":%d,\"wifi_rssi\":%d,\"ip\":\"%s\",\"mqtt_queue\":%d,\"slots\":[", (unsigned long)(uptime_millis / 1000), (unsigned)board.getBattMilliVolts(), (unsigned long)ESP.getFreeHeap(), (unsigned long)ESP.getMinFreeHeap(), (unsigned long)ESP.getMaxAllocHeap(), (int)_radio->getNoiseFloor(), (int)radio_driver.getLastRSSI(), snr_text, (unsigned long)(getTotalAirTime() / 1000), (unsigned long)(getReceiveAirTime() / 1000), (unsigned long)radio_driver.getPacketsRecv(), (unsigned long)radio_driver.getPacketsSent(), (unsigned long)radio_driver.getPacketsRecvErrors(), (unsigned long)getNumSentFlood(), (unsigned long)getNumSentDirect(), (unsigned long)getNumRecvFlood(), (unsigned long)getNumRecvDirect(), (int)_mgr->getOutboundCount(0xFFFFFFFF), wifi_rssi, ip, #ifdef WITH_MQTT_BRIDGE bridge ? bridge->getQueueSize() : 0); #else 0); #endif if (pos < 0 || pos >= static_cast<int>(buf_size) - 3) return; bool first = true; #ifdef WITH_MQTT_BRIDGE for (int i = 0; i < RUNTIME_MQTT_SLOTS; i++) { MQTTBridge::SlotStatusSnapshot status; if (!MQTTBridge::getSlotStatusSnapshot(i, &status)) continue; int written; if (status.has_publish_counts) { written = snprintf(buf + pos, buf_size - pos, "%s{\"n\":%d,\"name\":\"%s\",\"state\":\"%s\",\"ok\":%lu,\"err\":%lu}", first ? "" : ",", i + 1, status.name, status.state, status.publish_ok, status.publish_err); } else { written = snprintf(buf + pos, buf_size - pos, "%s{\"n\":%d,\"name\":\"%s\",\"state\":\"%s\"}", first ? "" : ",", i + 1, status.name, status.state); } if (written < 0 || written >= static_cast<int>(buf_size - pos)) break; pos += written; first = false; } #else (void)first; #endif snprintf(buf + pos, buf_size - pos, "]}"); } #endif void MyMesh::formatNeighborsReply(char *reply) { #if defined(WITH_MQTT_NEIGHBORS) char *dp = reply; // create copy of neighbours list, skipping empty entries so we can sort it separately from main list int16_t neighbours_count = 0; NeighbourInfo* sorted_neighbours[MAX_NEIGHBOURS]; for (int i = 0; i < MAX_NEIGHBOURS; i++) { auto neighbour = &neighbours[i]; if (neighbour->heard_timestamp > 0) { sorted_neighbours[neighbours_count] = neighbour; neighbours_count++; } } // sort neighbours newest to oldest std::sort(sorted_neighbours, sorted_neighbours + neighbours_count, [](const NeighbourInfo* a, const NeighbourInfo* b) { return a->heard_timestamp > b->heard_timestamp; // desc }); for (int i = 0; i < neighbours_count && dp - reply < 134; i++) { NeighbourInfo *neighbour = sorted_neighbours[i]; // add new line if not first item if (i > 0) *dp++ = '\n'; char hex[10]; // get 4 bytes of neighbour id as hex mesh::Utils::toHex(hex, neighbour->id.pub_key, 4); // add next neighbour uint32_t secs_ago = getRTCClock()->getCurrentTime() - neighbour->heard_timestamp; sprintf(dp, "%s:%d:%d", hex, secs_ago, neighbour->snr); while (*dp) dp++; // find end of string } if (dp == reply) { // no neighbours, need empty response strcpy(dp, "-none-"); dp += 6; } *dp = 0; // null terminator #else strcpy(reply, "not supported"); #endif } void MyMesh::removeNeighbor(const uint8_t *pubkey, int key_len) { #if defined(WITH_MQTT_NEIGHBORS) for (int i = 0; i < MAX_NEIGHBOURS; i++) { NeighbourInfo *neighbour = &neighbours[i]; if (memcmp(neighbour->id.pub_key, pubkey, key_len) == 0) { neighbours[i] = NeighbourInfo(); // clear neighbour entry } } #else (void)pubkey; (void)key_len; #endif } void MyMesh::formatStatsReply(char *reply) { StatsFormatHelper::formatCoreStats(reply, board, *_ms, _err_flags, _mgr); } void MyMesh::formatRadioStatsReply(char *reply) { StatsFormatHelper::formatRadioStats(reply, _radio, radio_driver, getTotalAirTime(), getReceiveAirTime()); } void MyMesh::formatRadioDiagReply(char *reply) { StatsFormatHelper::formatRadioDiag(reply, _radio, radio_driver, *_ms, _err_flags, hasOutbound()); } void MyMesh::formatPacketStatsReply(char *reply) { StatsFormatHelper::formatPacketStats(reply, radio_driver, getNumSentFlood(), getNumSentDirect(), getNumRecvFlood(), getNumRecvDirect()); } #if defined(ESP32_PLATFORM) || defined(USER_GPIO_CONTROL) void MyMesh::onUserGpioTimerCompleted(uint8_t pin, uint8_t state, uint32_t request_id) { int client_index; uint8_t path_hash_size; uint8_t client_tag[UserGpioReplyTracker::CLIENT_TAG_SIZE]; if (!_gpio_reply_tracker.takeRoute(pin, request_id, client_index, path_hash_size, client_tag)) { MESH_DEBUG_PRINTLN("GPIO %u timer complete: %s", pin, UserGpio::stateName((UserGpio::State)state)); return; } ClientInfo* client = NULL; if (client_index >= 0 && client_index < acl.getNumClients()) { ClientInfo* indexed = acl.getClientByIdx(client_index); if (UserGpioReplyTracker::matchesClient(indexed->id.pub_key, client_tag)) { client = indexed; } } if (client == NULL) { for (int i = 0; i < acl.getNumClients(); i++) { ClientInfo* candidate = acl.getClientByIdx(i); if (UserGpioReplyTracker::matchesClient(candidate->id.pub_key, client_tag)) { client = candidate; break; } } } if (client == NULL) return; uint8_t data[72]; uint32_t timestamp = getRTCClock()->getCurrentTimeUnique(); if (timestamp == request_id) timestamp++; memcpy(data, ×tamp, 4); data[4] = (TXT_TYPE_CLI_DATA << 2); const int text_len = snprintf((char*)&data[5], sizeof(data) - 5, "> GPIO %u timer complete: %s", pin, UserGpio::stateName((UserGpio::State)state)); if (text_len <= 0) return; mesh::Packet* packet = createDatagram(PAYLOAD_TYPE_TXT_MSG, client->id, client->shared_secret, data, 5 + (size_t)text_len); if (!packet) return; if (client->out_path_len == OUT_PATH_UNKNOWN) { sendFlood(packet, SERVER_RESPONSE_DELAY, path_hash_size); } else { sendDirect(packet, client->out_path, client->out_path_len, SERVER_RESPONSE_DELAY); } } #endif void MyMesh::handleCommand(uint32_t sender_timestamp, char *command, char *reply, int gpio_client_index, uint8_t gpio_path_hash_size) { #if defined(ESP32_PLATFORM) || defined(USER_GPIO_CONTROL) const uint8_t* gpio_client_key = gpio_client_index >= 0 && gpio_client_index < acl.getNumClients() ? acl.getClientByIdx(gpio_client_index)->id.pub_key : NULL; _gpio_reply_tracker.beginCommand(gpio_client_index, gpio_path_hash_size, gpio_client_key); #endif if (region_load_active) { if (StrHelper::isBlank(command)) { // empty/blank line, signal to terminate 'load' operation region_load_active = false; // resetFrom() preserves the selected IDs. Reject a replacement that // omitted either selected region instead of leaving a dangling ID. bool missing_default = region_map.getDefaultRegion() != NULL && temp_map.getDefaultRegion() == NULL; bool missing_home = region_map.getHomeRegion() != NULL && temp_map.getHomeRegion() == NULL; if (!missing_default && !missing_home) { region_map = temp_map; sprintf(reply, "OK - loaded %d regions", region_map.getCount()); } else { strcpy(reply, "Err - invalid region map; previous map retained"); } } else { char *np = command; while (*np == ' ') np++; // skip indent int indent = np - command; char *ep = np; while (RegionMap::is_name_char(*ep)) ep++; if (*ep) { *ep++ = 0; } // set null terminator for end of name while (*ep && *ep != 'F') ep++; // look for (optional) flags if (indent > 0 && indent < 8 && strlen(np) > 0) { auto parent = load_stack[indent - 1]; if (parent) { auto old = region_map.findByName(np); auto nw = temp_map.putRegion(np, parent->id, old ? old->id : 0); // carry-over the current ID (if name already exists) if (nw) { nw->flags = old ? old->flags : (*ep == 'F' ? 0 : REGION_DENY_FLOOD); // carry-over flags from curr load_stack[indent] = nw; // keep pointers to parent regions, to resolve parent_id's } } } reply[0] = 0; } return; } while (*command == ' ') command++; // skip leading spaces if (strlen(command) > 4 && command[2] == '|') { // optional prefix (for companion radio CLI) memcpy(reply, command, 3); // reflect the prefix back reply += 3; command += 3; } mesh::cli::normalizeCommandVerb(command); // handle ACL related commands if (memcmp(command, "setperm ", 8) == 0) { // format: setperm {pubkey-hex} {permissions-int8} char* hex = &command[8]; char* sp = strchr(hex, ' '); // look for separator char if (sp == NULL) { strcpy(reply, "Err - bad params"); } else { size_t hex_len = (size_t)(sp - hex); *sp++ = 0; // replace space with null terminator uint8_t pubkey[PUB_KEY_SIZE]; if (hex_len > 0 && hex_len <= PUB_KEY_SIZE * 2 && (hex_len & 1) == 0 && mesh::Utils::fromHex(pubkey, (int)(hex_len / 2), hex)) { uint8_t perms = atoi(sp); if (acl.applyPermissions(self_id, pubkey, (int)(hex_len / 2), perms)) { dirty_contacts_expiry = futureMillis(LAZY_CONTACTS_WRITE_DELAY); // trigger acl.save() strcpy(reply, "OK"); } else { strcpy(reply, "Err - invalid params"); } } else { strcpy(reply, "Err - bad pubkey"); } } } else if (sender_timestamp == 0 && strcmp(command, "get acl") == 0) { Serial.println("ACL:"); for (int i = 0; i < acl.getNumClients(); i++) { auto c = acl.getClientByIdx(i); if (c->permissions == 0) continue; // skip deleted (or guest) entries Serial.printf("%02X ", c->permissions); mesh::Utils::printHex(Serial, c->id.pub_key, PUB_KEY_SIZE); Serial.printf("\n"); } reply[0] = 0; #if defined(WITH_MQTT_NEIGHBORS) } else if (memcmp(command, "discover.neighbors", 18) == 0) { const char* sub = command + 18; while (*sub == ' ') sub++; if (*sub != 0) { strcpy(reply, "Err - discover.neighbors has no options"); } else { sendNodeDiscoverReq(); strcpy(reply, "OK - Discover sent"); } } else if (memcmp(command, "discover.scopes", 15) == 0) { const char* sub = command + 15; while (*sub == ' ') sub++; if (*sub != 0) { strcpy(reply, "Err - discover.scopes has no options"); } else if (pending_discover_tag != 0 && !millisHasNowPassed(pending_discover_until) && !neighbor_discover_active) { // A zero-hop table refresh is already collecting; queue the scope pass // behind it (as a manual, non-periodic request) rather than starting a // second refresh. if (!neighborDiscoverReady(reply)) { // reply already set by neighborDiscoverReady } else { neighbor_table_refresh_active = true; neighbor_table_refresh_periodic = false; long remaining_ms = (long)(pending_discover_until - futureMillis(0)); unsigned remaining_secs = remaining_ms > 0 ? (unsigned)(((unsigned long)remaining_ms + 999UL) / 1000UL) : 0; sprintf(reply, "OK - scopes queued (%us discovery remaining)", remaining_secs); MESH_DEBUG_PRINTLN("Neighbor scopes queued behind active discovery (%us remaining)", remaining_secs); } } else if (!startNeighborDiscover(reply)) { // reply already set by startNeighborDiscover } #elif defined(WITH_MQTT_BRIDGE) } else if (memcmp(command, "discover.scopes", 15) == 0) { strcpy(reply, "Err - neighbors not enabled in this build"); #endif } else if (strncmp(command, "room.post", 9) == 0) { char* msg = command + 9; while (*msg == ' ') msg++; if (*msg == 0) { snprintf(reply, MAX_POST_TEXT_LEN, "ERR empty message"); } else { addSystemPost(msg); snprintf(reply, MAX_POST_TEXT_LEN, "OK"); } } #if defined(MESHCORE_ESP32_FULL_PROFILE) else if (flood_rules.handleCommand(command, reply)) { // handled by the FULL-profile persistent flood rule engine } #endif else if (_clock_sync.handleCommand(command, reply)) { // handled by the role-independent mesh clock synchronizer } else { _cli.handleCommand(sender_timestamp, command, reply); // common CLI commands } } bool MyMesh::saveFilter(ClientInfo* client) { return client->isAdmin(); // only save Admins } void MyMesh::loop() { // Check radio FIRST to ensure we don't miss incoming packets // MQTT processing can take time, so we prioritize radio reception mesh::Mesh::loop(); _cli.loop(); _clock_sync.loop(); #ifdef WITH_MQTT_BRIDGE // bridge.loop() is now handled by FreeRTOS task on Core 0 - no need to call it here #endif if (millisHasNowPassed(next_push) && acl.getNumClients() > 0) { // check for ACK timeouts for (int i = 0; i < acl.getNumClients(); i++) { auto c = acl.getClientByIdx(i); if (c->extra.room.pending_ack && millisHasNowPassed(c->extra.room.ack_timeout)) { c->extra.room.push_failures++; c->extra.room.pending_ack = 0; // reset (TODO: keep prev expected_ack's in a list, incase they arrive LATER, after we retry) MESH_DEBUG_PRINTLN("pending ACK timed out: push_failures: %d", (uint32_t)c->extra.room.push_failures); } } // check next Round-Robin client, and sync next new post auto client = acl.getClientByIdx(next_client_idx); bool did_push = false; if (client->extra.room.pending_ack == 0 && client->last_activity != 0 && client->extra.room.push_failures < 3) { // not already waiting for ACK, AND not evicted, AND retries not max MESH_DEBUG_PRINTLN("loop - checking for client %02X", (uint32_t)client->id.pub_key[0]); uint32_t now = getRTCClock()->getCurrentTime(); for (int k = 0, idx = next_post_idx; k < MAX_UNSYNCED_POSTS; k++) { auto p = &posts[idx]; if (now >= p->post_timestamp + POST_SYNC_DELAY_SECS && p->post_timestamp > client->extra.room.sync_since // is new post for this Client? && !p->author.matches(client->id)) { // don't push posts to the author // push this post to Client, then wait for ACK pushPostToClient(client, *p); did_push = true; MESH_DEBUG_PRINTLN("loop - pushed to client %02X: %s", (uint32_t)client->id.pub_key[0], p->text); break; } idx = (idx + 1) % MAX_UNSYNCED_POSTS; // wrap to start of cyclic queue } } else { MESH_DEBUG_PRINTLN("loop - skipping busy (or evicted) client %02X", (uint32_t)client->id.pub_key[0]); } next_client_idx = (next_client_idx + 1) % acl.getNumClients(); // round robin polling for each client if (did_push) { next_push = futureMillis(SYNC_PUSH_INTERVAL); } else { // were no unsynced posts for curr client, so process next client much quicker! (in next loop()) next_push = futureMillis(SYNC_PUSH_INTERVAL / 8); } } if (next_flood_advert && millisHasNowPassed(next_flood_advert)) { mesh::Packet *pkt = createSelfAdvert(); uint32_t delay_millis = 0; if (pkt) sendFloodScoped(default_scope, pkt, delay_millis, _prefs.path_hash_mode + 1); updateFloodAdvertTimer(); // schedule next flood advert updateAdvertTimer(); // also schedule local advert (so they don't overlap) } else if (next_local_advert && millisHasNowPassed(next_local_advert)) { mesh::Packet *pkt = createSelfAdvert(); if (pkt) sendZeroHop(pkt); updateAdvertTimer(); // schedule next local advert } const bool revert_radio_due = revert_radio_at && millisHasNowPassed(revert_radio_at); const bool radio_apply_ready = !radio_apply_retry_at || millisHasNowPassed(radio_apply_retry_at); bool radio_apply_failed = false; if (revert_radio_due && !temp_radio_applied) { // The temporary window ended before it could be applied. Drop both timers // so a previously busy radio cannot switch to the expired channel later. set_radio_at = revert_radio_at = 0; radio_apply_retry_at = 0; radio_apply_failures = 0; MESH_DEBUG_PRINTLN("Temp radio params expired before apply"); } else if (revert_radio_due && !hasOutbound() && radio_apply_ready) { if (applySavedRadioParams()) { if (saved_radio_apply_pending) { radio_driver.setTxPower(_prefs.tx_power_dbm); radio_driver.setRxBoostedGainMode(_prefs.rx_boosted_gain); } set_radio_at = revert_radio_at = 0; temp_radio_applied = false; saved_radio_apply_pending = false; radio_apply_retry_at = 0; radio_apply_failures = 0; MESH_DEBUG_PRINTLN("Radio params restored"); } else { radio_apply_failed = true; } } else if (set_radio_at && millisHasNowPassed(set_radio_at) && !hasOutbound() && radio_apply_ready) { uint32_t rx_us = _prefs.rx_ps_rx_us; uint32_t sleep_us = _prefs.rx_ps_sleep_us; bool timing_ok = true; if (_prefs.rx_powersaving_enabled && _prefs.rx_ps_level != 0) { uint32_t preamble = _prefs.rx_ps_preamble ? _prefs.rx_ps_preamble : (pending_sf <= 8 ? 32UL : 16UL); timing_ok = CommonCLI::calculateRxPowerSavingLevel( _prefs.rx_ps_level, pending_sf, pending_bw, preamble, &rx_us, &sleep_us); } uint32_t timings[2] = {rx_us, sleep_us}; const uint32_t* applied_timings = _prefs.rx_powersaving_enabled && radio_driver.supportsRxPowerSaving() ? timings : NULL; if (timing_ok && radio_driver.setParams( pending_freq, pending_bw, pending_sf, pending_cr, applied_timings)) { set_radio_at = 0; active_cr = pending_cr; temp_radio_applied = true; radio_apply_retry_at = 0; radio_apply_failures = 0; MESH_DEBUG_PRINTLN("Temp radio params"); } else { // A failed setParams() may have applied only a prefix of the tuple. // Ensure expiry restores the complete saved configuration. saved_radio_apply_pending = true; radio_apply_failed = true; } } if (saved_radio_apply_pending && !temp_radio_applied && !hasOutbound() && radio_apply_ready && !radio_apply_failed) { if (applySavedRadioParams()) { radio_driver.setTxPower(_prefs.tx_power_dbm); radio_driver.setRxBoostedGainMode(_prefs.rx_boosted_gain); saved_radio_apply_pending = false; radio_apply_retry_at = 0; radio_apply_failures = 0; } else { radio_apply_failed = true; } } if (radio_apply_failed) { radio_apply_retry_at = futureMillis(nextRadioApplyRetryDelay(radio_apply_failures)); } #ifdef WITH_WEBCONFIG if (_webconfig) { _webconfig->tick(millis()); if (!_webconfig->isRunning() && !_webconfig->isStopping()) { delete _webconfig; _webconfig = nullptr; } } #endif #if defined(WITH_MQTT_BRIDGE) && defined(OTA_MANIFEST_BASE) if (_ota_update_at && millisHasNowPassed(_ota_update_at)) { // deferred `ota update` _ota_update_at = 0; // clear timer // The "Beginning update..." reply has now been queued. Flush it before OTA // blocks the loop until reboot, then free a running bridge for heap headroom. // Remember its state: an OTA request must not enable MQTT that an operator // had deliberately stopped. Serial.println("OTA: starting update"); const bool bridge_was_running = bridge && bridge->isRunning(); drainOutbound(OTA_TX_DRAIN_TIMEOUT_MS); bool may_flash = true; if (bridge_was_running) { setBridgeState(false); // OTA must not write after a forced/timed-out MQTT shutdown: its TLS/heap // ownership is uncertain until a subsequent clean start/stop cycle. may_flash = bridge && bridge->canFlashAfterStop(); if (!may_flash) { Serial.println("OTA: aborted, MQTT stop did not complete cleanly"); } } char ota_reply[160]; if (may_flash && !_cli.getBoard()->otaFromManifest(getFirmwareVer(), false, ota_reply)) { Serial.print("OTA: aborted - "); Serial.println(ota_reply); may_flash = false; } // Successful otaFromManifest() reboots and never returns. Restore only a // bridge that was running before this attempt; leave an intentionally // stopped bridge stopped after any OTA refusal or download failure. if (!may_flash && bridge_was_running) { Serial.println("OTA: resuming bridge"); setBridgeState(true); } } #endif // is pending dirty contacts write needed? if (dirty_contacts_expiry && millisHasNowPassed(dirty_contacts_expiry)) { acl.save(_fs, MyMesh::saveFilter); dirty_contacts_expiry = 0; } // TODO: periodically check for OLD/inactive entries in known_clients[], and evict // update uptime uint32_t now = millis(); uptime_millis += now - last_millis; last_millis = now; #ifdef WITH_MQTT_BRIDGE _alerter.onLoop(now); #endif #if defined(WITH_MQTT_NEIGHBORS) // Two-stage periodic neighbors publication: // stage 1 - zero-hop node-discover refreshes the neighbour table (60s window) // stage 2 - anon-regions scope query per neighbour (startNeighborDiscover) // then the table JSON is published and the next cycle is rescheduled. bool periodic_neighbors_enabled = _cli.getObserverPrefs()->mqtt_neighbors_enabled; if (neighbor_discover_active) { loopNeighborDiscover(); } else if (neighbor_table_refresh_active) { if (neighbor_table_refresh_periodic && !periodic_neighbors_enabled) { // periodic switched off mid-refresh -> cancel (leave pending_discover_tag alone) neighbor_table_refresh_active = false; neighbor_table_refresh_periodic = false; next_neighbors_publish = 0; } else if (pending_discover_tag == 0 || millisHasNowPassed(pending_discover_until)) { // 60s zero-hop window done -> begin the per-neighbour scope queries bool was_periodic = neighbor_table_refresh_periodic; pending_discover_tag = 0; neighbor_table_refresh_active = false; neighbor_table_refresh_periodic = false; char tmp_reply[80]; const char* origin_str = was_periodic ? "periodic" : "manual"; if (startNeighborDiscover(tmp_reply)) { MESH_DEBUG_PRINTLN("MQTT %s %s", origin_str, tmp_reply); } else { if (periodic_neighbors_enabled) { next_neighbors_publish = futureMillis(_cli.getObserverPrefs()->mqtt_neighbors_interval); } MESH_DEBUG_PRINTLN("MQTT %s neighbor scope discovery failed: %s", origin_str, tmp_reply); } } } else if (periodic_neighbors_enabled && bridge && bridge->isRunning()) { if (next_neighbors_publish == 0 || (next_neighbors_publish != 0 && millisHasNowPassed(next_neighbors_publish))) { if (pending_discover_tag == 0 || millisHasNowPassed(pending_discover_until)) { pending_discover_tag = 0; sendNodeDiscoverReq(); MESH_DEBUG_PRINTLN("MQTT periodic neighbor table refresh started"); } else { MESH_DEBUG_PRINTLN("MQTT periodic refresh joined active neighbor discovery"); } neighbor_table_refresh_active = true; neighbor_table_refresh_periodic = true; } } // Report the schedule state back to the bridge for `get mqtt.status`. if (bridge) { if (neighbor_discover_active || neighbor_table_refresh_active) { bridge->setNeighborsSchedule(MQTTBridge::NBR_ACTIVE, 0); } else if (next_neighbors_publish == 0 || millisHasNowPassed(next_neighbors_publish)) { bridge->setNeighborsSchedule(MQTTBridge::NBR_DUE, 0); } else { long remaining_ms = (long)(next_neighbors_publish - futureMillis(0)); uint32_t remaining_secs = remaining_ms > 0 ? (uint32_t)(remaining_ms / 1000) : 0; bridge->setNeighborsSchedule(MQTTBridge::NBR_SCHEDULED, remaining_secs); } } #endif #ifdef WITH_SNMP // Push radio stats to SNMP agent every 2 seconds if (_snmp_agent.isRunning()) { static unsigned long last_snmp_stats = 0; if (now - last_snmp_stats >= 2000) { last_snmp_stats = now; _snmp_agent.updateRadioStats( radio_driver.getPacketsRecv(), radio_driver.getPacketsSent(), radio_driver.getPacketsRecvErrors(), (int16_t)_radio->getNoiseFloor(), (int16_t)radio_driver.getLastRSSI(), (int16_t)(radio_driver.getLastSNR() * 4), getNumSentFlood(), getNumSentDirect(), getNumRecvFlood(), getNumRecvDirect(), getTotalAirTime() / 1000, uptime_millis / 1000); } } #endif } bool MyMesh::isMillisTimerDue(unsigned long timestamp) const { return timestamp && millisHasNowPassed(timestamp); } uint32_t MyMesh::limitSleepToMillisTimer(unsigned long timestamp, uint32_t sleep_secs) const { if (!timestamp || sleep_secs == 0) { return sleep_secs; } unsigned long now = millis(); if ((long)(now - timestamp) >= 0) { return 0; } unsigned long remaining_ms = timestamp - now; uint32_t remaining_secs = (remaining_ms + 999UL) / 1000UL; return remaining_secs < sleep_secs ? remaining_secs : sleep_secs; } uint32_t MyMesh::getPowerSaveSleepSeconds(uint32_t max_secs) const { if (max_secs == 0 || hasPendingWork()) { return 0; } uint32_t sleep_secs = max_secs; uint32_t queue_delay_ms; if (getNextQueueWakeDelay(queue_delay_ms)) { uint32_t queue_delay_secs = (queue_delay_ms + 999UL) / 1000UL; if (queue_delay_secs < sleep_secs) sleep_secs = queue_delay_secs; } uint32_t retry_delay_ms; if (getNextRetryWakeDelay(retry_delay_ms)) { uint32_t retry_delay_secs = (retry_delay_ms + 999UL) / 1000UL; if (retry_delay_secs < sleep_secs) sleep_secs = retry_delay_secs; } if (acl.getNumClients() > 0) { sleep_secs = limitSleepToMillisTimer(next_push, sleep_secs); } sleep_secs = limitSleepToMillisTimer(next_flood_advert, sleep_secs); sleep_secs = limitSleepToMillisTimer(next_local_advert, sleep_secs); const bool radio_apply_backoff = radio_apply_retry_at && !millisHasNowPassed(radio_apply_retry_at); if (radio_apply_backoff) { sleep_secs = limitSleepToMillisTimer(radio_apply_retry_at, sleep_secs); } else { sleep_secs = limitSleepToMillisTimer(set_radio_at, sleep_secs); sleep_secs = limitSleepToMillisTimer(revert_radio_at, sleep_secs); } sleep_secs = limitSleepToMillisTimer(dirty_contacts_expiry, sleep_secs); return sleep_secs; } // To check if there is pending work bool MyMesh::hasPendingWork() const { #if defined(WITH_BRIDGE) if (bridge && bridge->isRunning()) return true; // bridge needs WiFi radio, can't sleep #endif if (radio_driver.isWatchdogObserving()) return true; // keep MCU awake for one radio duty cycle if (radio_driver.isCalibratingNoiseFloor()) return true; // keep MCU awake for the noise-floor window if (hasQueuedWorkDue() || hasRetryWorkDue()) return true; if (acl.getNumClients() > 0 && isMillisTimerDue(next_push)) return true; if (isMillisTimerDue(next_flood_advert) || isMillisTimerDue(next_local_advert)) return true; const bool radio_apply_backoff = radio_apply_retry_at && !isMillisTimerDue(radio_apply_retry_at); if (!radio_apply_backoff && (isMillisTimerDue(set_radio_at) || isMillisTimerDue(revert_radio_at) || (saved_radio_apply_pending && !temp_radio_applied))) return true; return isMillisTimerDue(dirty_contacts_expiry); } #if defined(WITH_MQTT_NEIGHBORS) #include "helpers/MQTTMessageBuilder.h" #if defined(ESP_PLATFORM) #include <esp_heap_caps.h> #endif // This node's own non-flood scope names, same source the anon-regions server // reply uses. Empty string when the node has no scoped regions. void MyMesh::getLocalScopes(char* buf, size_t len) { if (!buf || len == 0) return; buf[0] = 0; region_map.exportNamesTo(buf, (int)len, REGION_DENY_FLOOD); } // Client side of the anon-regions request (the server side is handleAnonRegionsReq). // Inner payload: {tag(4)}{ANON_REQ_TYPE_REGIONS}{0x00 = zero-hop reply path}. mesh::Packet* MyMesh::sendAnonRegionsReq(const mesh::Identity& target, uint32_t& tag) { // RxReservePacketManager keeps a four-packet emergency floor. Preflight one // extra free packet so its void queue API cannot silently shed this request. if (_mgr->getFreeCount() < NEIGHBOR_DISCOVER_MIN_FREE_PACKETS) return NULL; uint8_t secret[PUB_KEY_SIZE]; self_id.calcSharedSecret(secret, target); tag = getRTCClock()->getCurrentTimeUnique(); uint8_t inner[6]; memcpy(inner, &tag, 4); inner[4] = ANON_REQ_TYPE_REGIONS; inner[5] = 0x00; // request a zero-hop reply path mesh::Packet* pkt = createAnonDatagram(PAYLOAD_TYPE_ANON_REQ, self_id, target, secret, inner, sizeof(inner)); if (!pkt) return NULL; sendDirect(pkt, NULL, 0, 0); return pkt; } bool MyMesh::cancelNeighborDiscoverRequest() { if (!neighbor_discover_request) return false; for (int i = _mgr->getOutboundTotal() - 1; i >= 0; i--) { if (_mgr->getOutboundByIdx(i) == neighbor_discover_request) { mesh::Packet* pkt = _mgr->removeOutboundByIdx(i); if (pkt) releasePacket(pkt); neighbor_discover_request = NULL; return true; } } return false; } // This timer starts after the request finishes transmitting. Allow the server // delay, the responder's full CAD deferral window plus one maximum retry // overshoot, and airtime for one priority-0 packet ahead of the response plus // the response itself. The radio estimate scales with SF, bandwidth, coding // rate, and preamble. uint32_t MyMesh::neighborDiscoverQueryTimeoutMs() const { uint32_t response_airtime = _radio->getEstAirtimeFor(MAX_PACKET_PAYLOAD + 2); return SERVER_RESPONSE_DELAY + getCADFailMaxDuration() + 360UL + response_airtime * 2UL; } void MyMesh::resetNeighborDiscoverJsonBudget() { getLocalScopes(self_scopes_buf, sizeof(self_scopes_buf)); { // No default region means this node floods unscoped, i.e. the wildcard. RegionEntry* def = region_map.getDefaultRegion(); const char* def_name = (def && def->name[0]) ? def->name : "*"; if (*def_name == '#') def_name++; // match how self.scopes renders names strncpy(self_default_scope_buf, def_name, sizeof(self_default_scope_buf) - 1); self_default_scope_buf[sizeof(self_default_scope_buf) - 1] = 0; } MQTTBridge::getEffectiveMqttOrigin( _prefs.node_name, _cli.getObserverPrefs(), neighbor_discover_origin, sizeof(neighbor_discover_origin)); char self_pubkey_hex[65]; mesh::Utils::toHex(self_pubkey_hex, self_id.pub_key, PUB_KEY_SIZE); char timestamp[40]; MQTTMessageBuilder::formatIsoTimestampForMqtt( getRTCClock()->getCurrentTime(), 0, nullptr, timestamp, sizeof(timestamp)); neighbor_discover_publish_count = 0; neighbor_discover_queried_count = 0; neighbor_discover_truncated = false; neighbor_discover_json_size = MQTTMessageBuilder::measureNeighborsMessageBase( neighbor_discover_origin, self_pubkey_hex, timestamp, self_scopes_buf, self_default_scope_buf, neighbor_discover_count); } // Account for one terminal result. The base measurement reserves maximum-width // progress metadata; UINT32_MAX likewise reserves the widest heard-age value. // If this result cannot fit, stop before transmitting another scope request. bool MyMesh::completeNeighborDiscoverEntry() { NeighborDiscoverEntry& entry = neighbor_discover[neighbor_discover_next]; char pubkey_hex[65]; mesh::Utils::toHex(pubkey_hex, entry.id.pub_key, PUB_KEY_SIZE); MQTTMessageBuilder::NeighborsMessageEntry measured = { pubkey_hex, entry.snr / 4.0f, UINT32_MAX, entry.scopes, entry.status == ND_RESPONDED ? "responded" : (entry.status == ND_SEND_FAILED ? "send_failed" : "timeout") }; size_t added = MQTTMessageBuilder::measureNeighborsMessageEntry(measured); if (neighbor_discover_publish_count > 0) added++; // array comma if (neighbor_discover_json_size + added >= MQTTBridge::NEIGHBORS_JSON_BUFFER_SIZE || neighbor_discover_publish_count >= MQTTBridge::NEIGHBORS_MAX_PUBLISH_ENTRIES) { neighbor_discover_truncated = true; finishNeighborDiscover(); return false; } neighbor_discover_json_size += added; neighbor_discover_publish_count++; neighbor_discover_next++; return true; } // Match a RESPONSE against the pending overlay entry by tag; copy its scope // string (payload after the 8-byte {tag}{clock} header) into the entry. bool MyMesh::handleNeighborDiscoverResponse(int overlay_idx, const uint8_t* data, size_t len) { if (overlay_idx < 0 || overlay_idx >= neighbor_discover_count) return false; NeighborDiscoverEntry& entry = neighbor_discover[overlay_idx]; if (entry.status != ND_PENDING || len < 8) return false; uint32_t tag; memcpy(&tag, data, 4); if (tag != entry.tag) return false; size_t scope_len = len - 8; if (scope_len >= sizeof(entry.scopes)) { scope_len = sizeof(entry.scopes) - 1; } memcpy(entry.scopes, &data[8], scope_len); entry.scopes[scope_len] = 0; entry.status = ND_RESPONDED; // A zero-hop reply is proof we heard this neighbour now, so re-stamp both the // snapshot and the live table; a stamp taken before time sync heals here. entry.heard_timestamp = getRTCClock()->getCurrentTime(); touchNeighbourHeard(entry.id, entry.heard_timestamp); return true; } // Refresh a live neighbour's heard time only: a scope reply carries no advert // timestamp or SNR to update. void MyMesh::touchNeighbourHeard(const mesh::Identity& id, uint32_t heard_timestamp) { for (int i = 0; i < MAX_NEIGHBOURS; i++) { if (id.matches(neighbours[i].id)) { neighbours[i].heard_timestamp = heard_timestamp; return; } } } // A heard age is a wall-clock delta, so it only means something when both stamps // share a clock epoch. An entry heard before the clock was set holds the unset // default, which a synced clock turns into a ~2-year age; report those as // unknown instead. See UPSTREAM_BUGS.md for the monotonic fix. static bool neighborHeardAgeUsable(uint32_t heard_timestamp, uint32_t now_secs) { if (heard_timestamp == 0 || now_secs < heard_timestamp) return false; // Never synced: the stamp shares this clock's boot epoch, so the delta holds. if (now_secs < MQTTConnectionPolicy::kSyncedClockEpoch) return true; return heard_timestamp >= MQTTConnectionPolicy::kSyncedClockEpoch; } // Publish-ordering: usable ages first, then most recently heard, then stronger // SNR, then pubkey. The JSON builder drops the tail if the buffer fills, so the // head must be the most useful entries. static bool neighborPublishEntryComesBefore( const MQTTMessageBuilder::NeighborsMessageEntry& lhs, const MQTTMessageBuilder::NeighborsMessageEntry& rhs) { if (lhs.heard_unknown != rhs.heard_unknown) { return !lhs.heard_unknown; } if (lhs.heard_secs_ago != rhs.heard_secs_ago) { return lhs.heard_secs_ago < rhs.heard_secs_ago; // newer first } if (lhs.snr != rhs.snr) { return lhs.snr > rhs.snr; // stronger first when equally recent } return strcmp(lhs.pubkey_hex, rhs.pubkey_hex) < 0; } #if defined(ESP_PLATFORM) // Neighbors allocations prefer PSRAM where it exists and otherwise come from // internal DRAM, so MQTT_NEIGHBORS_WITHOUT_PSRAM boards can build the table too. #if defined(BOARD_HAS_PSRAM) static const uint32_t kNeighborsAllocCaps = MALLOC_CAP_SPIRAM | MALLOC_CAP_8BIT; #else static const uint32_t kNeighborsAllocCaps = MALLOC_CAP_INTERNAL | MALLOC_CAP_8BIT; #endif static void* neighborsAlloc(size_t size) { if (size == 0) return nullptr; void* p = heap_caps_malloc(size, kNeighborsAllocCaps); #if defined(BOARD_HAS_PSRAM) if (!p) p = heap_caps_malloc(size, MALLOC_CAP_INTERNAL | MALLOC_CAP_8BIT); #endif return p; } static void neighborsFree(void* ptr) { if (ptr) heap_caps_free(ptr); } // ArduinoJson v7 JsonDocument has no real capacity cap (DynamicJsonDocument(N) // is a no-op shim). Keep the pool off internal DRAM and soft-cap peak growth to // the publish buffer size. used only rises on allocate -- conservative for this // single-shot doc (overflow path removes+breaks, so no further growth after free). struct NeighborsDocAllocator : ArduinoJson::Allocator { size_t used = 0; static const size_t kBudget = MQTTBridge::NEIGHBORS_DOC_POOL_BUDGET; void* allocate(size_t size) override { if (used >= kBudget || size > kBudget - used) return nullptr; void* p = neighborsAlloc(size); if (p) used += size; return p; } void deallocate(void* ptr) override { neighborsFree(ptr); } void* reallocate(void* ptr, size_t new_size) override { size_t old_size = ptr ? heap_caps_get_allocated_size(ptr) : 0; size_t next_used = (used >= old_size) ? (used - old_size) : 0; if (next_used >= kBudget || new_size > kBudget - next_used) return nullptr; void* p = heap_caps_realloc(ptr, new_size, kNeighborsAllocCaps); #if defined(BOARD_HAS_PSRAM) if (!p) p = heap_caps_realloc(ptr, new_size, MALLOC_CAP_INTERNAL | MALLOC_CAP_8BIT); #endif if (p) used = next_used + new_size; return p; } }; #else static void* neighborsAlloc(size_t size) { return size ? malloc(size) : nullptr; } static void neighborsFree(void* ptr) { free(ptr); } #endif // Build the neighbors-table JSON and hand it to the bridge, then reschedule. void MyMesh::finishNeighborDiscover() { char self_pubkey_hex[65]; mesh::Utils::toHex(self_pubkey_hex, self_id.pub_key, PUB_KEY_SIZE); char timestamp[40]; MQTTMessageBuilder::formatIsoTimestampForMqtt(getRTCClock()->getCurrentTime(), 0, nullptr, timestamp, sizeof(timestamp)); // The entry table plus one hex string each reaches ~4.5 KB at MAX_NEIGHBOURS, // which does not fit the mesh loop task's 8 KB stack, so both share a single // heap block sized to this pass. Publishing is skipped if either alloc fails. const int publish_count = neighbor_discover_publish_count; const size_t hex_size = PUB_KEY_SIZE * 2 + 1; const size_t entries_bytes = sizeof(MQTTMessageBuilder::NeighborsMessageEntry) * publish_count; void* scratch = neighborsAlloc(entries_bytes + hex_size * publish_count); char* json_buf = (char*)neighborsAlloc(MQTTBridge::NEIGHBORS_JSON_BUFFER_SIZE); if (json_buf && (scratch || publish_count == 0)) { auto* entries = (MQTTMessageBuilder::NeighborsMessageEntry*)scratch; char* pubkey_hex = (char*)scratch + entries_bytes; uint32_t now_secs = getRTCClock()->getCurrentTime(); for (int i = 0; i < publish_count; i++) { auto& entry = neighbor_discover[i]; char* hex = &pubkey_hex[i * hex_size]; mesh::Utils::toHex(hex, entry.id.pub_key, PUB_KEY_SIZE); entries[i].pubkey_hex = hex; entries[i].snr = entry.snr / 4.0f; bool heard_known = neighborHeardAgeUsable(entry.heard_timestamp, now_secs); entries[i].heard_unknown = !heard_known; entries[i].heard_secs_ago = heard_known ? (now_secs - entry.heard_timestamp) : 0; entries[i].scopes = entry.scopes; switch (entry.status) { case ND_RESPONDED: entries[i].status = "responded"; break; case ND_SEND_FAILED: entries[i].status = "send_failed"; break; default: entries[i].status = "timeout"; break; } } // insertion sort: most useful first (JSON builder drops the tail on overflow) for (int i = 1; i < publish_count; i++) { MQTTMessageBuilder::NeighborsMessageEntry entry = entries[i]; int j = i; while (j > 0 && neighborPublishEntryComesBefore(entry, entries[j - 1])) { entries[j] = entries[j - 1]; j--; } entries[j] = entry; } #if defined(ESP_PLATFORM) NeighborsDocAllocator doc_alloc; JsonDocument doc(&doc_alloc); #else JsonDocument doc; #endif int json_len = MQTTMessageBuilder::buildNeighborsMessage( doc, neighbor_discover_origin, self_pubkey_hex, timestamp, self_scopes_buf, self_default_scope_buf, entries, publish_count, json_buf, MQTTBridge::NEIGHBORS_JSON_BUFFER_SIZE, neighbor_discover_count, neighbor_discover_queried_count, neighbor_discover_truncated); if (json_len > 0 && bridge) { bridge->requestPublishNeighbors(json_buf, (size_t)json_len); } } neighborsFree(scratch); neighborsFree(json_buf); neighbor_discover_active = false; neighbor_discover_count = 0; neighbor_discover_next = 0; neighbor_discover_publish_count = 0; neighbor_discover_queried_count = 0; neighbor_discover_json_size = 0; neighbor_discover_truncated = false; neighbor_discover_until = 0; neighbor_discover_request = NULL; if (_cli.getObserverPrefs()->mqtt_neighbors_enabled) { next_neighbors_publish = futureMillis(_cli.getObserverPrefs()->mqtt_neighbors_interval); } } // Advance the newest-first scope-query phase. Keep only one request in flight so // its responder gets a clear reply opportunity and the packet pool stays free. void MyMesh::loopNeighborDiscover() { if (!neighbor_discover_active) return; if (neighbor_discover_next >= neighbor_discover_count) { finishNeighborDiscover(); return; } NeighborDiscoverEntry& entry = neighbor_discover[neighbor_discover_next]; if (entry.status == ND_QUEUED) { if (!millisHasNowPassed(neighbor_discover_until)) return; if (cancelNeighborDiscoverRequest()) { entry.status = ND_SEND_FAILED; completeNeighborDiscoverEntry(); return; } if (isCurrentOutbound(neighbor_discover_request)) { neighbor_discover_until = futureMillis(neighborDiscoverQueryTimeoutMs()); return; } neighbor_discover_request = NULL; // packet manager already shed it entry.status = ND_SEND_FAILED; completeNeighborDiscoverEntry(); return; } if (entry.status == ND_PENDING) { if (!millisHasNowPassed(neighbor_discover_until)) return; entry.status = ND_TIMEOUT; completeNeighborDiscoverEntry(); return; } if (entry.status == ND_RESPONDED || entry.status == ND_SEND_FAILED || entry.status == ND_TIMEOUT) { completeNeighborDiscoverEntry(); return; } if (entry.status != ND_UNSENT) { neighbor_discover_next++; return; } uint32_t tag; mesh::Packet* request = sendAnonRegionsReq(entry.id, tag); if (request) { entry.tag = tag; entry.status = ND_QUEUED; neighbor_discover_request = request; neighbor_discover_until = futureMillis(NEIGHBOR_DISCOVER_QUEUE_TIMEOUT_MS); } else { entry.status = ND_SEND_FAILED; completeNeighborDiscoverEntry(); } } // Shared precondition for starting a discovery: usable buffers + bridge running. // PSRAM builds size their neighbors buffers for PSRAM, so a board whose PSRAM // failed to init must not silently spend that much internal DRAM here. // MQTT_NEIGHBORS_WITHOUT_PSRAM builds are already sized for internal DRAM. bool MyMesh::neighborDiscoverReady(char* reply) { #if defined(ESP_PLATFORM) && defined(BOARD_HAS_PSRAM) if (!psramFound()) { strcpy(reply, "Err - PSRAM not available"); return false; } #endif if (!bridge || !bridge->isRunning()) { strcpy(reply, "Err - MQTT bridge not running"); return false; } return true; } // Snapshot the neighbor table newest-first. loopNeighborDiscover() emits one // anon-regions query at a time so hidden responders do not reply as a burst. bool MyMesh::startNeighborDiscover(char* reply) { if (neighbor_discover_active) { strcpy(reply, "Err - neighbor discover already active"); return false; } if (!neighborDiscoverReady(reply)) { return false; // reply already set } neighbor_discover_count = 0; for (int i = 0; i < MAX_NEIGHBOURS; i++) { if (neighbours[i].heard_timestamp > 0) { NeighborDiscoverEntry& entry = neighbor_discover[neighbor_discover_count]; entry.id = neighbours[i].id; entry.heard_timestamp = neighbours[i].heard_timestamp; entry.snr = neighbours[i].snr; entry.scopes[0] = 0; entry.tag = 0; entry.status = ND_UNSENT; neighbor_discover_count++; } } // Query the freshest/strongest entries first; pubkey makes ties deterministic. for (int i = 1; i < neighbor_discover_count; i++) { NeighborDiscoverEntry entry = neighbor_discover[i]; int j = i; while (j > 0) { auto& rhs = neighbor_discover[j - 1]; bool before = entry.heard_timestamp > rhs.heard_timestamp || (entry.heard_timestamp == rhs.heard_timestamp && entry.snr > rhs.snr) || (entry.heard_timestamp == rhs.heard_timestamp && entry.snr == rhs.snr && memcmp(entry.id.pub_key, rhs.id.pub_key, PUB_KEY_SIZE) < 0); if (!before) break; neighbor_discover[j] = neighbor_discover[j - 1]; j--; } neighbor_discover[j] = entry; } neighbor_discover_next = 0; resetNeighborDiscoverJsonBudget(); neighbor_discover_active = true; neighbor_discover_until = 0; neighbor_discover_request = NULL; if (neighbor_discover_count == 0) { finishNeighborDiscover(); strcpy(reply, "OK - neighbor discover started (0 neighbors, self only)"); } else { loopNeighborDiscover(); // queue the first request now sprintf(reply, "OK - neighbor discover started (%u neighbors)", (unsigned)neighbor_discover_count); } return true; } #endif // WITH_MQTT_NEIGHBORS