Version 1.41

Fixed one bug with generating of message ID.

README.md

@@ -4,14 +4,15 @@ A simple library for creating ESP-NOW based Mesh network for ESP8266/ESP32.
 
 ## Features
 
-1. The maximum size of transmitted data is 200 bytes. Currently only unencrypted messages.
+1. The maximum size of transmitted data is 200 bytes.
-2. All nodes are not visible to the network scanner (for the ESP_NOW mode only).
+2. Encrypted and unencrypted messages. Simple XOR crypting.
-3. Not required a pre-pairings for data transfer.
+3. All nodes are not visible to the network scanner.
-4. Broadcast or unicast data transmission.
+4. Not required a pre-pairings for data transfer.
-5. There are no periodic/synchronous messages on the network. All devices are in "silent mode" and do not "hum" into the air (for the ESP_NOW mode only).
+5. Broadcast or unicast data transmissions.
-6. Each node has its own independent routing table, updated only as needed.
+6. There are no periodic/synchronous messages on the network. All devices are in "silent mode" and do not "hum" into the air.
-7. Each node will receive/send a message if it "sees" at least one device on the network.
+7. Each node has its own independent routing table, updated only as needed.
-8. The number of devices on the network and the area of use is not limited (hypothetically). :-)
+8. Each node will receive/send a message if it "sees" at least one device on the network.
+9. The number of devices on the network and the area of use is not limited (hypothetically). :-)
 
 ## Testing
 
@@ -22,6 +23,11 @@ A simple library for creating ESP-NOW based Mesh network for ESP8266/ESP32.
 5. Voila. ;-)
 6. P.S. Uncomment #define PRINT_LOG in ZHNetwork.h for display to serial port the full operation log.
 
+## Notes
+
+1. Possibility uses WiFi AP or STA modes at the same time with ESP-NOW using the standard libraries.
+2. For correct work at ESP-NOW + STA mode your WiFi router must be set on channel 1 and set gateway mode.
+
 ## Function descriptions
 
 ### Sets the callback function for processing a received broadcast message
@@ -54,76 +60,35 @@ Note. Called only at broadcast or unicast with confirm message. Status will alwa
 
 ```cpp
 myNet.setOnConfirmReceivingCallback(onConfirmReceiving);
-void onConfirmReceiving(const uint8_t *target, const bool status)
+void onConfirmReceiving(const uint8_t *target, const uint16_t id, const bool status)
 {
     // Do something when receiving a delivery/undelivery confirm message.
 }
 ```
 
-### Sets one of the three possibility operating modes
+### ESP-NOW Mesh network initialization
-
-* ESP_NOW. Default mode. ESP-NOW Mesh network only.
-* ESP_NOW_AP. ESP-NOW Mesh network + access point.
-* ESP_NOW_STA. ESP-NOW Mesh network + connect to your WiFi router.
-
-Attention! For correct work on ESP_NOW_STA mode at ESP8266 your WiFi router must be set on channel 1.
-
-```cpp
-myNet.setWorkMode(ESP_NOW);
-```
-
-### Gets used operating mode
-
-```cpp
-myNet.getWorkMode();
-```
-
-### Sets ESP-NOW Mesh network name
 
 1-20 characters.
 
 Note. If network name not set node will work with all ESP-NOW networks. If set node will work with only one network.
 
 ```cpp
-myNet.setNetName("ZHNetwork");
+myNet.begin("ZHNetwork");
-```
+myNet.begin("ZHNetwork", true); // Gateway mode.
-
-### Gets used ESP-NOW Mesh network name
-
-```cpp
-myNet.getNetName();
-```
-
-### Sets WiFi ssid and password for ESP_NOW_STA mode
-
-Note. Must be called before Mesh network initialization.
-
-```cpp
-myNet.setStaSetting("SSID", "PASSWORD");
-```
-
-### Sets access point ssid and password for ESP_NOW_AP mode
-
-Note. Must be called before Mesh network initialization.
-
-```cpp
-myNet.setApSetting("SSID", "PASSWORD");
-```
-
-### ESP-NOW Mesh network initialization
-
-```cpp
-myNet.begin();
 ```
 
 ### Sends broadcast message to all nodes
 
+Returns message ID.
+
 ```cpp
 myNet.sendBroadcastMessage("Hello world!");
 ```
 
 ### Sends unicast message to node
 
+Returns message ID.
+
 ```cpp
 myNet.sendUnicastMessage("Hello world!", target); // Without confirm.
 myNet.sendUnicastMessage("Hello world!", target, true); // With confirm.
@@ -143,12 +108,6 @@ myNet.maintenance();
 myNet.getNodeMac();
 ```
 
-### Gets node IP address
-
-```cpp
-myNet.getNodeIp();
-```
-
 ### Gets version of this library
 
 ```cpp
@@ -168,6 +127,14 @@ uint8_t mac[6]
 myNet.stringToMac(string, mac);
 ```
 
+### Sets crypt key
+
+1-20 characters.
+
+```cpp
+myNet.setCryptKey("VERY_LONG_CRYPT_KEY"); 
+```
+
 ### Sets max number of attempts to send message
 
 1-10. 3 default value.
@@ -217,11 +184,11 @@ myNet.getMaxWaitingTimeForRoutingInfo();
 
 void onBroadcastReceiving(const char *data, const uint8_t *sender);
 void onUnicastReceiving(const char *data, const uint8_t *sender);
-void onConfirmReceiving(const uint8_t *target, const bool status);
+void onConfirmReceiving(const uint8_t *target, const uint16_t id, const bool status);
 
 ZHNetwork myNet;
 
-uint64_t messagelastTime{0};
+uint64_t messageLastTime{0};
 uint16_t messageTimerDelay{5000};
 const uint8_t target[6]{0xA8, 0x48, 0xFA, 0xDC, 0x5B, 0xFA};
 
@@ -229,27 +196,13 @@ void setup()
 {
   Serial.begin(115200);
   Serial.println();
-  // *** ESP-NOW mode only.
+  myNet.begin("ZHNetwork");
-  myNet.setWorkMode(ESP_NOW);
+  // myNet.setCryptKey("VERY_LONG_CRYPT_KEY"); // If encryption is used, the key must be set same of all another ESP-NOW devices in network.
-  // *** Or ESP-NOW + access point mode.
-  // myNet.setWorkMode(ESP_NOW_AP);
-  // myNet.setApSetting("ESP NODE TEST", "12345678");
-  // *** Or ESP-NOW + connect to your router mode.
-  // myNet.setWorkMode(ESP_NOW_STA);
-  // myNet.setStaSetting("SSID", "PASSWORD");
-  // ***
-  myNet.setNetName("ZHNetwork");                   // Optional.
-  myNet.setMaxNumberOfAttempts(3);                 // Optional.
-  myNet.setMaxWaitingTimeBetweenTransmissions(50); // Optional.
-  myNet.setMaxWaitingTimeForRoutingInfo(500);      // Optional.
-  myNet.begin();
   myNet.setOnBroadcastReceivingCallback(onBroadcastReceiving);
   myNet.setOnUnicastReceivingCallback(onUnicastReceiving);
   myNet.setOnConfirmReceivingCallback(onConfirmReceiving);
   Serial.print("MAC: ");
   Serial.print(myNet.getNodeMac());
-  Serial.print(". IP: ");
-  Serial.print(myNet.getNodeIp());
   Serial.print(". Firmware version: ");
   Serial.print(myNet.getFirmwareVersion());
   Serial.println(".");
@@ -257,19 +210,23 @@ void setup()
 
 void loop()
 {
-  if ((millis() - messagelastTime) > messageTimerDelay)
+  if ((millis() - messageLastTime) > messageTimerDelay)
   {
     Serial.println("Broadcast message sended.");
     myNet.sendBroadcastMessage("Hello world!");
+
     Serial.print("Unicast message to MAC ");
     Serial.print(myNet.macToString(target));
     Serial.println(" sended.");
     myNet.sendUnicastMessage("Hello world!", target);
+    
     Serial.print("Unicast with confirm message to MAC ");
     Serial.print(myNet.macToString(target));
+    Serial.print(" ID ");
+    Serial.print(myNet.sendUnicastMessage("Hello world!", target, true));
     Serial.println(" sended.");
-    myNet.sendUnicastMessage("Hello world!", target, true);
+
-    messagelastTime = millis();
+    messageLastTime = millis();
   }
   myNet.maintenance();
 }
@@ -282,6 +239,7 @@ void onBroadcastReceiving(const char *data, const uint8_t *sender)
   Serial.print("Message: ");
   Serial.println(data);
 }
+
 void onUnicastReceiving(const char *data, const uint8_t *sender)
 {
   Serial.print("Unicast message from MAC ");
@@ -291,10 +249,14 @@ void onUnicastReceiving(const char *data, const uint8_t *sender)
   Serial.println(data);
 }
 
-void onConfirmReceiving(const uint8_t *target, const bool status)
+void onConfirmReceiving(const uint8_t *target, const uint16_t id, const bool status)
 {
   Serial.print("Message to MAC ");
   Serial.print(myNet.macToString(target));
+  Serial.print(" ID ");
+  Serial.print(id);
   Serial.println(status ? " delivered." : " undelivered.");
 }
 ```
+
+Any feedback via [e-mail](mailto:github@zh.com.ru) would be appreciated. Or... [Buy me a coffee](https://paypal.me/aZholtikov).

examples/Receiver/main.cpp

@@ -9,26 +9,12 @@ void setup()
 {
   Serial.begin(115200);
   Serial.println();
-  // *** ESP-NOW mode only.
+  myNet.begin("ZHNetwork");
-  myNet.setWorkMode(ESP_NOW);
+  // myNet.setCryptKey("VERY_LONG_CRYPT_KEY"); // If encryption is used, the key must be set same of all another ESP-NOW devices in network.
-  // *** Or ESP-NOW + access point mode.
-  // myNet.setWorkMode(ESP_NOW_AP);
-  // myNet.setApSetting("ESP NODE TEST", "12345678");
-  // *** Or ESP-NOW + connect to your router mode.
-  // myNet.setWorkMode(ESP_NOW_STA);
-  // myNet.setStaSetting("SSID", "PASSWORD");
-  // ***
-  myNet.setNetName("ZHNetwork");                   // Optional.
-  myNet.setMaxNumberOfAttempts(3);                 // Optional.
-  myNet.setMaxWaitingTimeBetweenTransmissions(50); // Optional.
-  myNet.setMaxWaitingTimeForRoutingInfo(500);      // Optional.
-  myNet.begin();
   myNet.setOnBroadcastReceivingCallback(onBroadcastReceiving);
   myNet.setOnUnicastReceivingCallback(onUnicastReceiving);
   Serial.print("MAC: ");
   Serial.print(myNet.getNodeMac());
-  Serial.print(". IP: ");
-  Serial.print(myNet.getNodeIp());
   Serial.print(". Firmware version: ");
   Serial.print(myNet.getFirmwareVersion());
   Serial.println(".");
@@ -47,6 +33,7 @@ void onBroadcastReceiving(const char *data, const uint8_t *sender)
   Serial.print("Message: ");
   Serial.println(data);
 }
+
 void onUnicastReceiving(const char *data, const uint8_t *sender)
 {
   Serial.print("Unicast message from MAC ");

examples/Transmitter/main.cpp

@@ -1,10 +1,10 @@
 #include "ZHNetwork.h"
 
-void onConfirmReceiving(const uint8_t *target, const bool status);
+void onConfirmReceiving(const uint8_t *target, const uint16_t id, const bool status);
 
 ZHNetwork myNet;
 
-uint64_t messagelastTime{0};
+uint64_t messageLastTime{0};
 uint16_t messageTimerDelay{5000};
 const uint8_t target[6]{0xA8, 0x48, 0xFA, 0xDC, 0x5B, 0xFA};
 
@@ -12,25 +12,11 @@ void setup()
 {
   Serial.begin(115200);
   Serial.println();
-  // *** ESP-NOW mode only.
+  myNet.begin("ZHNetwork");
-  myNet.setWorkMode(ESP_NOW);
+  // myNet.setCryptKey("VERY_LONG_CRYPT_KEY"); // If encryption is used, the key must be set same of all another ESP-NOW devices in network.
-  // *** Or ESP-NOW + access point mode.
-  // myNet.setWorkMode(ESP_NOW_AP);
-  // myNet.setApSetting("ESP NODE TEST", "12345678");
-  // *** Or ESP-NOW + connect to your router mode.
-  // myNet.setWorkMode(ESP_NOW_STA);
-  // myNet.setStaSetting("SSID", "PASSWORD");
-  // ***
-  myNet.setNetName("ZHNetwork");                   // Optional.
-  myNet.setMaxNumberOfAttempts(3);                 // Optional.
-  myNet.setMaxWaitingTimeBetweenTransmissions(50); // Optional.
-  myNet.setMaxWaitingTimeForRoutingInfo(500);      // Optional.
-  myNet.begin();
   myNet.setOnConfirmReceivingCallback(onConfirmReceiving);
   Serial.print("MAC: ");
   Serial.print(myNet.getNodeMac());
-  Serial.print(". IP: ");
-  Serial.print(myNet.getNodeIp());
   Serial.print(". Firmware version: ");
   Serial.print(myNet.getFirmwareVersion());
   Serial.println(".");
@@ -38,26 +24,32 @@ void setup()
 
 void loop()
 {
-  if ((millis() - messagelastTime) > messageTimerDelay)
+  if ((millis() - messageLastTime) > messageTimerDelay)
   {
     Serial.println("Broadcast message sended.");
     myNet.sendBroadcastMessage("Hello world!");
+
     Serial.print("Unicast message to MAC ");
     Serial.print(myNet.macToString(target));
     Serial.println(" sended.");
     myNet.sendUnicastMessage("Hello world!", target);
+
     Serial.print("Unicast with confirm message to MAC ");
     Serial.print(myNet.macToString(target));
+    Serial.print(" ID ");
+    Serial.print(myNet.sendUnicastMessage("Hello world!", target, true));
     Serial.println(" sended.");
-    myNet.sendUnicastMessage("Hello world!", target, true);
+
-    messagelastTime = millis();
+    messageLastTime = millis();
   }
   myNet.maintenance();
 }
 
-void onConfirmReceiving(const uint8_t *target, const bool status)
+void onConfirmReceiving(const uint8_t *target, const uint16_t id, const bool status)
 {
   Serial.print("Message to MAC ");
   Serial.print(myNet.macToString(target));
+  Serial.print(" ID ");
+  Serial.print(id);
   Serial.println(status ? " delivered." : " undelivered.");
 }

library.properties

@@ -1,5 +1,5 @@
 name=ZHNetwork
-version=1.01
+version=1.41
 author=Alexey Zholtikov
 maintainer=Alexey Zholtikov
 sentence=ESP-NOW based Mesh network for ESP8266/ESP32

src/ZHNetwork.cpp

@@ -1,31 +1,19 @@
 #include "ZHNetwork.h"
 
-routing_vector_t routingVector;
+routing_vector_t ZHNetwork::routingVector;
-incoming_queue_t queueForIncomingData;
+confirmation_vector_t ZHNetwork::confirmationVector;
-outgoing_queue_t queueForOutgoingData;
+incoming_queue_t ZHNetwork::queueForIncomingData;
-waiting_queue_t queueForRoutingVectorWaiting;
+outgoing_queue_t ZHNetwork::queueForOutgoingData;
+waiting_queue_t ZHNetwork::queueForRoutingVectorWaiting;
 
-const char *firmware PROGMEM{"1.0"};
+bool ZHNetwork::criticalProcessSemaphore{false};
-const uint8_t broadcastMAC[6]{0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF};
+bool ZHNetwork::sentMessageSemaphore{false};
-
+bool ZHNetwork::confirmReceivingSemaphore{false};
-bool criticalProcessSemaphore{false};
+bool ZHNetwork::confirmReceiving{false};
-bool sentMessageSemaphore{false};
+char ZHNetwork::netName_[20]{0};
-bool confirmReceivingSemaphore{false};
+char ZHNetwork::key_[20]{0};
-bool confirmReceiving{false};
+uint8_t ZHNetwork::localMAC[6]{0};
-uint8_t localMAC[6]{0};
+uint16_t ZHNetwork::lastMessageID[10]{0};
-uint8_t numberOfAttemptsToSend{1};
-uint16_t lastMessageID[10]{0};
-uint64_t lastMessageSentTime{0};
-
-work_mode_t workMode_{ESP_NOW};
-char netName_[20]{0};
-char apSsid_[32]{"ESP-NOW NODE"};
-char apPassword_[64]{0};
-char staSsid_[32]{0};
-char staPassword_[64]{0};
-uint8_t maxNumberOfAttempts_{3};
-uint8_t maxWaitingTimeBetweenTransmissions_{50};
-uint16_t maxTimeForRoutingInfoWaiting_{500};
 
 ZHNetwork &ZHNetwork::setOnBroadcastReceivingCallback(on_message_t onBroadcastReceivingCallback)
 {
@@ -45,104 +33,41 @@ ZHNetwork &ZHNetwork::setOnConfirmReceivingCallback(on_confirm_t onConfirmReceiv
     return *this;
 }
 
-error_code_t ZHNetwork::setWorkMode(const work_mode_t workMode)
+error_code_t ZHNetwork::begin(const char *netName, const bool gateway)
 {
-    if (workMode < ESP_NOW || workMode > ESP_NOW_STA)
+#if defined(ESP8266)
-        return ERROR;
+    randomSeed(os_random());
-    workMode_ = workMode;
+#endif
-    return SUCCESS;
+#if defined(ESP32)
-}
+    randomSeed(esp_random());
-
+#endif
-work_mode_t ZHNetwork::getWorkMode()
+    if (strlen(netName) >= 1 && strlen(netName) <= 20)
-{
+        strcpy(netName_, netName);
-    return workMode_;
-}
-
-error_code_t ZHNetwork::setNetName(const char *netName)
-{
-    if (strlen(netName) < 1 || strlen(netName) > 20)
-        return ERROR;
-    memset(&netName_, 0, strlen(netName));
-    strcpy(netName_, netName);
-    return SUCCESS;
-}
-
-String ZHNetwork::getNetName()
-{
-    return String(netName_);
-}
-
-error_code_t ZHNetwork::setStaSetting(const char *ssid, const char *password)
-{
-    if (strlen(ssid) < 1 || strlen(ssid) > 32 || strlen(password) > 64)
-        return ERROR;
-    memset(&staSsid_, 0, strlen(ssid));
-    strcpy(staSsid_, ssid);
-    memset(&staPassword_, 0, strlen(password));
-    strcpy(staPassword_, password);
-    return SUCCESS;
-}
-
-error_code_t ZHNetwork::setApSetting(const char *ssid, const char *password)
-{
-    if (strlen(ssid) < 1 || strlen(ssid) > 32 || strlen(password) < 8 || strlen(password) > 64)
-        return ERROR;
-    memset(&apSsid_, 0, strlen(ssid));
-    strcpy(apSsid_, ssid);
-    memset(&apPassword_, 0, strlen(password));
-    strcpy(apPassword_, password);
-    return SUCCESS;
-}
-
-error_code_t ZHNetwork::begin()
-{
-    randomSeed(analogRead(0));
 #ifdef PRINT_LOG
     Serial.begin(115200);
 #endif
-    switch (workMode_)
+    WiFi.mode(gateway ? WIFI_AP_STA : WIFI_STA);
-    {
-    case ESP_NOW:
-        WiFi.mode(WIFI_STA);
-        break;
-    case ESP_NOW_AP:
-        WiFi.mode(WIFI_AP_STA);
-        WiFi.softAP(apSsid_, apPassword_);
-        break;
-    case ESP_NOW_STA:
-        WiFi.mode(WIFI_STA);
-        WiFi.begin(staSsid_, staPassword_);
-        while (WiFi.status() != WL_CONNECTED)
-        {
-            if (WiFi.status() == WL_NO_SSID_AVAIL || WiFi.status() == WL_CONNECT_FAILED)
-                return ERROR;
-            delay(500);
-        }
-        break;
-    default:
-        break;
-    }
     esp_now_init();
 #if defined(ESP8266)
-    wifi_get_macaddr(STATION_IF, localMAC);
+    wifi_get_macaddr(gateway ? SOFTAP_IF : STATION_IF, localMAC);
     esp_now_set_self_role(ESP_NOW_ROLE_COMBO);
 #endif
 #if defined(ESP32)
-    esp_wifi_get_mac((wifi_interface_t)ESP_IF_WIFI_STA, localMAC);
+    esp_wifi_get_mac(gateway ? (wifi_interface_t)ESP_IF_WIFI_AP : (wifi_interface_t)ESP_IF_WIFI_STA, localMAC);
 #endif
     esp_now_register_send_cb(onDataSent);
     esp_now_register_recv_cb(onDataReceive);
     return SUCCESS;
 }
 
-void ZHNetwork::sendBroadcastMessage(const char *data)
+uint16_t ZHNetwork::sendBroadcastMessage(const char *data)
 {
-    broadcastMessage(data, broadcastMAC, BROADCAST);
+    return broadcastMessage(data, broadcastMAC, BROADCAST);
 }
 
-void ZHNetwork::sendUnicastMessage(const char *data, const uint8_t *target, const bool confirm)
+uint16_t ZHNetwork::sendUnicastMessage(const char *data, const uint8_t *target, const bool confirm)
 {
-    unicastMessage(data, target, localMAC, confirm ? UNICAST_WITH_CONFIRM : UNICAST);
+    return unicastMessage(data, target, localMAC, confirm ? UNICAST_WITH_CONFIRM : UNICAST);
 }
 
 void ZHNetwork::maintenance()
@@ -162,7 +87,15 @@ void ZHNetwork::maintenance()
             esp_now_del_peer(outgoingData.intermediateTargetMAC);
 #endif
             if (onConfirmReceivingCallback && macToString(outgoingData.transmittedData.originalSenderMAC) == macToString(localMAC) && outgoingData.transmittedData.messageType == BROADCAST)
-                onConfirmReceivingCallback(outgoingData.transmittedData.originalTargetMAC, true);
+                onConfirmReceivingCallback(outgoingData.transmittedData.originalTargetMAC, outgoingData.transmittedData.messageID, true);
+            if (macToString(outgoingData.transmittedData.originalSenderMAC) == macToString(localMAC) && outgoingData.transmittedData.messageType == UNICAST_WITH_CONFIRM)
+            {
+                confirmation_waiting_data_t confirmationData;
+                confirmationData.time = millis();
+                memcpy(&confirmationData.targetMAC, &outgoingData.transmittedData.originalTargetMAC, 6);
+                memcpy(&confirmationData.messageID, &outgoingData.transmittedData.messageID, 2);
+                confirmationVector.push_back(confirmationData);
+            }
         }
         else
         {
@@ -193,7 +126,6 @@ void ZHNetwork::maintenance()
                     }
                 }
                 waiting_data_t waitingData;
-                esp_memset(&waitingData, 0, sizeof(waiting_data_t));
                 waitingData.time = millis();
                 memcpy(&waitingData.intermediateTargetMAC, &outgoingData.intermediateTargetMAC, 6);
                 memcpy(&waitingData.transmittedData, &outgoingData.transmittedData, sizeof(transmitted_data_t));
@@ -266,7 +198,12 @@ void ZHNetwork::maintenance()
             Serial.println(F(" received."));
 #endif
             if (onBroadcastReceivingCallback)
+            {
+                if (key_[0])
+                    for (uint8_t i{0}; i < strlen(incomingData.transmittedData.message); ++i)
+                        incomingData.transmittedData.message[i] = incomingData.transmittedData.message[i] ^ key_[i % strlen(key_)];
                 onBroadcastReceivingCallback(incomingData.transmittedData.message, incomingData.transmittedData.originalSenderMAC);
+            }
             forward = true;
             break;
         case UNICAST:
@@ -282,7 +219,12 @@ void ZHNetwork::maintenance()
             if (macToString(incomingData.transmittedData.originalTargetMAC) == macToString(localMAC))
             {
                 if (onUnicastReceivingCallback)
+                {
+                    if (key_[0])
+                        for (uint8_t i{0}; i < strlen(incomingData.transmittedData.message); ++i)
+                            incomingData.transmittedData.message[i] = incomingData.transmittedData.message[i] ^ key_[i % strlen(key_)];
                     onUnicastReceivingCallback(incomingData.transmittedData.message, incomingData.transmittedData.originalSenderMAC);
+                }
             }
             else
                 unicastMessage(incomingData.transmittedData.message, incomingData.transmittedData.originalTargetMAC, incomingData.transmittedData.originalSenderMAC, UNICAST);
@@ -300,8 +242,17 @@ void ZHNetwork::maintenance()
             if (macToString(incomingData.transmittedData.originalTargetMAC) == macToString(localMAC))
             {
                 if (onUnicastReceivingCallback)
+                {
+                    if (key_[0])
+                        for (uint8_t i{0}; i < strlen(incomingData.transmittedData.message); ++i)
+                            incomingData.transmittedData.message[i] = incomingData.transmittedData.message[i] ^ key_[i % strlen(key_)];
                     onUnicastReceivingCallback(incomingData.transmittedData.message, incomingData.transmittedData.originalSenderMAC);
-                unicastMessage("", incomingData.transmittedData.originalSenderMAC, localMAC, DELIVERY_CONFIRM_RESPONSE);
+                }
+                confirmation_id_t id;
+                memcpy(&id.messageID, &incomingData.transmittedData.messageID, 2);
+                char temp[sizeof(transmitted_data_t::message)];
+                memcpy(&temp, &id, sizeof(transmitted_data_t::message));
+                unicastMessage(temp, incomingData.transmittedData.originalSenderMAC, localMAC, DELIVERY_CONFIRM_RESPONSE);
             }
             else
                 unicastMessage(incomingData.transmittedData.message, incomingData.transmittedData.originalTargetMAC, incomingData.transmittedData.originalSenderMAC, UNICAST_WITH_CONFIRM);
@@ -319,7 +270,17 @@ void ZHNetwork::maintenance()
             if (macToString(incomingData.transmittedData.originalTargetMAC) == macToString(localMAC))
             {
                 if (onConfirmReceivingCallback)
-                    onConfirmReceivingCallback(incomingData.transmittedData.originalSenderMAC, true);
+                {
+                    confirmation_id_t id;
+                    memcpy(&id.messageID, &incomingData.transmittedData.message, 2);
+                    for (uint16_t i{0}; i < confirmationVector.size(); ++i)
+                    {
+                        confirmation_waiting_data_t confirmationData = confirmationVector[i];
+                        if (confirmationData.messageID == id.messageID)
+                            confirmationVector.erase(confirmationVector.begin() + i);
+                    }
+                    onConfirmReceivingCallback(incomingData.transmittedData.originalSenderMAC, id.messageID, true);
+                }
             }
             else
                 unicastMessage(incomingData.transmittedData.message, incomingData.transmittedData.originalTargetMAC, incomingData.transmittedData.originalSenderMAC, DELIVERY_CONFIRM_RESPONSE);
@@ -413,7 +374,6 @@ void ZHNetwork::maintenance()
             {
                 queueForRoutingVectorWaiting.pop();
                 outgoing_data_t outgoingData;
-                esp_memset(&outgoingData, 0, sizeof(outgoing_data_t));
                 memcpy(&outgoingData.transmittedData, &waitingData.transmittedData, sizeof(transmitted_data_t));
                 memcpy(&outgoingData.intermediateTargetMAC, &routingTable.intermediateTargetMAC, 6);
                 queueForOutgoingData.push(outgoingData);
@@ -458,7 +418,21 @@ void ZHNetwork::maintenance()
 #endif
             if (waitingData.transmittedData.messageType == UNICAST_WITH_CONFIRM && macToString(waitingData.transmittedData.originalSenderMAC) == macToString(localMAC))
                 if (onConfirmReceivingCallback)
-                    onConfirmReceivingCallback(waitingData.transmittedData.originalTargetMAC, false);
+                    onConfirmReceivingCallback(waitingData.transmittedData.originalTargetMAC, waitingData.transmittedData.messageID, false);
+        }
+    }
+    if (confirmationVector.size())
+    {
+        for (uint16_t i{0}; i < confirmationVector.size(); ++i)
+        {
+            confirmation_waiting_data_t confirmationData = confirmationVector[i];
+            if ((millis() - confirmationData.time) > maxTimeForRoutingInfoWaiting_)
+            {
+                confirmationVector.erase(confirmationVector.begin() + i);
+                broadcastMessage("", confirmationData.targetMAC, SEARCH_REQUEST);
+                if (onConfirmReceivingCallback)
+                    onConfirmReceivingCallback(confirmationData.targetMAC, confirmationData.messageID, false);
+            }
         }
     }
 }
@@ -468,15 +442,6 @@ String ZHNetwork::getNodeMac()
     return macToString(localMAC);
 }
 
-IPAddress ZHNetwork::getNodeIp()
-{
-    if (workMode_ == ESP_NOW_AP)
-        return WiFi.softAPIP();
-    if (workMode_ == ESP_NOW_STA)
-        return WiFi.localIP();
-    return IPAddress(0, 0, 0, 0);
-}
-
 String ZHNetwork::getFirmwareVersion()
 {
     return firmware;
@@ -504,6 +469,13 @@ uint8_t *ZHNetwork::stringToMac(const String &string, uint8_t *mac)
     return mac;
 }
 
+error_code_t ZHNetwork::setCryptKey(const char *key)
+{
+    if (strlen(key) >= 1 && strlen(key) <= 20)
+        strcpy(key_, key);
+    return SUCCESS;
+}
+
 error_code_t ZHNetwork::setMaxNumberOfAttempts(const uint8_t maxNumberOfAttempts)
 {
     if (maxNumberOfAttempts < 1 || maxNumberOfAttempts > 10)
@@ -598,16 +570,18 @@ void IRAM_ATTR ZHNetwork::onDataReceive(uint8_t *mac, uint8_t *data, uint8_t len
     criticalProcessSemaphore = false;
 }
 
-void ZHNetwork::broadcastMessage(const char *data, const uint8_t *target, message_type_t type)
+uint16_t ZHNetwork::broadcastMessage(const char *data, const uint8_t *target, message_type_t type)
 {
     outgoing_data_t outgoingData;
-    esp_memset(&outgoingData, 0, sizeof(outgoing_data_t));
     outgoingData.transmittedData.messageType = type;
     outgoingData.transmittedData.messageID = ((uint16_t)random(32767) << 8) | (uint16_t)random(32767);
     memcpy(&outgoingData.transmittedData.netName, &netName_, 20);
     memcpy(&outgoingData.transmittedData.originalTargetMAC, target, 6);
     memcpy(&outgoingData.transmittedData.originalSenderMAC, &localMAC, 6);
     strcpy(outgoingData.transmittedData.message, data);
+    if (key_[0] && outgoingData.transmittedData.messageType == BROADCAST)
+        for (uint8_t i{0}; i < strlen(outgoingData.transmittedData.message); ++i)
+            outgoingData.transmittedData.message[i] = outgoingData.transmittedData.message[i] ^ key_[i % strlen(key_)];
     memcpy(&outgoingData.intermediateTargetMAC, &broadcastMAC, 6);
     queueForOutgoingData.push(outgoingData);
 #ifdef PRINT_LOG
@@ -631,18 +605,21 @@ void ZHNetwork::broadcastMessage(const char *data, const uint8_t *target, messag
     Serial.print(macToString(outgoingData.transmittedData.originalTargetMAC));
     Serial.println(F(" added to queue."));
 #endif
+    return outgoingData.transmittedData.messageID;
 }
 
-void ZHNetwork::unicastMessage(const char *data, const uint8_t *target, const uint8_t *sender, message_type_t type)
+uint16_t ZHNetwork::unicastMessage(const char *data, const uint8_t *target, const uint8_t *sender, message_type_t type)
 {
     outgoing_data_t outgoingData;
-    esp_memset(&outgoingData, 0, sizeof(outgoing_data_t));
     outgoingData.transmittedData.messageType = type;
     outgoingData.transmittedData.messageID = ((uint16_t)random(32767) << 8) | (uint16_t)random(32767);
     memcpy(&outgoingData.transmittedData.netName, &netName_, 20);
     memcpy(&outgoingData.transmittedData.originalTargetMAC, target, 6);
     memcpy(&outgoingData.transmittedData.originalSenderMAC, sender, 6);
     strcpy(outgoingData.transmittedData.message, data);
+    if (key_[0] && macToString(outgoingData.transmittedData.originalSenderMAC) == macToString(localMAC) && outgoingData.transmittedData.messageType != DELIVERY_CONFIRM_RESPONSE)
+        for (uint8_t i{0}; i < strlen(outgoingData.transmittedData.message); ++i)
+            outgoingData.transmittedData.message[i] = outgoingData.transmittedData.message[i] ^ key_[i % strlen(key_)];
     for (uint16_t i{0}; i < routingVector.size(); ++i)
     {
         routing_table_t routingTable = routingVector[i];
@@ -678,7 +655,7 @@ void ZHNetwork::unicastMessage(const char *data, const uint8_t *target, const ui
             Serial.print(macToString(outgoingData.intermediateTargetMAC));
             Serial.println(F(" added to queue."));
 #endif
-            return;
+            return outgoingData.transmittedData.messageID;
         }
     }
     memcpy(&outgoingData.intermediateTargetMAC, target, 6);
@@ -711,4 +688,5 @@ void ZHNetwork::unicastMessage(const char *data, const uint8_t *target, const ui
     Serial.print(macToString(outgoingData.intermediateTargetMAC));
     Serial.println(F(" added to queue."));
 #endif
+    return outgoingData.transmittedData.messageID;
 }

src/ZHNetwork.h

@@ -15,54 +15,53 @@
 
 // #define PRINT_LOG // Uncomment to display to serial port the full operation log.
 
-#if defined(ESP8266)
-#define esp_memset memset // Just for remove the compiler notice for ESP8266 at "memset". I don't know why this is happening...
-#endif
-#if defined(ESP32)
-#define esp_memset memset
-#endif
-
 typedef struct
 {
-    uint8_t messageType;
+    uint8_t messageType{0};
-    uint16_t messageID;
+    uint16_t messageID{0};
-    char netName[20];
+    char netName[20]{0};
-    uint8_t originalTargetMAC[6];
+    uint8_t originalTargetMAC[6]{0};
-    uint8_t originalSenderMAC[6];
+    uint8_t originalSenderMAC[6]{0};
-    char message[200];
+    char message[200]{0};
 } transmitted_data_t;
 
 typedef struct
 {
-    uint8_t intermediateTargetMAC[6];
+    uint8_t intermediateTargetMAC[6]{0};
     transmitted_data_t transmittedData;
 } outgoing_data_t;
 
 typedef struct
 {
-    uint8_t intermediateSenderMAC[6];
+    uint8_t intermediateSenderMAC[6]{0};
     transmitted_data_t transmittedData;
 } incoming_data_t;
 
 typedef struct
 {
-    uint64_t time;
+    uint64_t time{0};
-    uint8_t intermediateTargetMAC[6];
+    uint8_t intermediateTargetMAC[6]{0};
     transmitted_data_t transmittedData;
 } waiting_data_t;
 
 typedef struct
 {
-    uint8_t originalTargetMAC[6];
+    uint8_t originalTargetMAC[6]{0};
-    uint8_t intermediateTargetMAC[6];
+    uint8_t intermediateTargetMAC[6]{0};
 } routing_table_t;
 
-typedef enum
+typedef struct
 {
-    ESP_NOW = 1,
+    uint16_t messageID{0};
-    ESP_NOW_AP,
+    char empty[198]{0}; // Just only to prevent compiler warnings.
-    ESP_NOW_STA
+} confirmation_id_t;
-} work_mode_t;
+
+typedef struct
+{
+    uint64_t time{0};
+    uint8_t targetMAC[6]{0};
+    uint16_t messageID{0};
+} confirmation_waiting_data_t;
 
 typedef enum
 {
@@ -81,8 +80,9 @@ typedef enum // Just for further development.
 } error_code_t;
 
 typedef std::function<void(const char *, const uint8_t *)> on_message_t;
-typedef std::function<void(const uint8_t *, const bool)> on_confirm_t;
+typedef std::function<void(const uint8_t *, const uint16_t, const bool)> on_confirm_t;
 typedef std::vector<routing_table_t> routing_vector_t;
+typedef std::vector<confirmation_waiting_data_t> confirmation_vector_t;
 typedef std::queue<outgoing_data_t> outgoing_queue_t;
 typedef std::queue<incoming_data_t> incoming_queue_t;
 typedef std::queue<waiting_data_t> waiting_queue_t;
@@ -94,30 +94,21 @@ public:
     ZHNetwork &setOnUnicastReceivingCallback(on_message_t onUnicastReceivingCallback);
     ZHNetwork &setOnConfirmReceivingCallback(on_confirm_t onConfirmReceivingCallback);
 
-    error_code_t setWorkMode(const work_mode_t workMode);
+    error_code_t begin(const char *netName = "", const bool gateway = false);
-    work_mode_t getWorkMode(void);
 
-    error_code_t setNetName(const char *netName);
+    uint16_t sendBroadcastMessage(const char *data);
-    String getNetName(void);
+    uint16_t sendUnicastMessage(const char *data, const uint8_t *target, const bool confirm = false);
-
-    error_code_t setStaSetting(const char *ssid, const char *password);
-    error_code_t setApSetting(const char *ssid, const char *password);
-
-    error_code_t begin(void);
-
-    void sendBroadcastMessage(const char *data);
-    void sendUnicastMessage(const char *data, const uint8_t *target, const bool confirm = false);
 
     void maintenance(void);
 
     String getNodeMac(void);
-    IPAddress getNodeIp(void);
     String getFirmwareVersion(void);
     String readErrorCode(error_code_t code); // Just for further development.
 
     static String macToString(const uint8_t *mac);
     uint8_t *stringToMac(const String &string, uint8_t *mac);
 
+    error_code_t setCryptKey(const char *key = "");
     error_code_t setMaxNumberOfAttempts(const uint8_t maxNumberOfAttempts);
     uint8_t getMaxNumberOfAttempts(void);
     error_code_t setMaxWaitingTimeBetweenTransmissions(const uint8_t maxWaitingTimeBetweenTransmissions);
@@ -126,6 +117,29 @@ public:
     uint16_t getMaxWaitingTimeForRoutingInfo(void);
 
 private:
+    static routing_vector_t routingVector;
+    static confirmation_vector_t confirmationVector;
+    static incoming_queue_t queueForIncomingData;
+    static outgoing_queue_t queueForOutgoingData;
+    static waiting_queue_t queueForRoutingVectorWaiting;
+
+    static bool criticalProcessSemaphore;
+    static bool sentMessageSemaphore;
+    static bool confirmReceivingSemaphore;
+    static bool confirmReceiving;
+    static uint8_t localMAC[6];
+    static uint16_t lastMessageID[10];
+    static char netName_[20];
+    static char key_[20];
+
+    const char *firmware{"1.41"};
+    const uint8_t broadcastMAC[6]{0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF};
+    uint8_t maxNumberOfAttempts_{3};
+    uint8_t maxWaitingTimeBetweenTransmissions_{50};
+    uint8_t numberOfAttemptsToSend{1};
+    uint16_t maxTimeForRoutingInfoWaiting_{500};
+    uint32_t lastMessageSentTime{0};
+
 #if defined(ESP8266)
     static void onDataSent(uint8_t *mac, uint8_t status);
     static void onDataReceive(uint8_t *mac, uint8_t *data, uint8_t length);
@@ -134,8 +148,8 @@ private:
     static void onDataSent(const uint8_t *mac, esp_now_send_status_t status);
     static void onDataReceive(const uint8_t *mac, const uint8_t *data, int length);
 #endif
-    void broadcastMessage(const char *data, const uint8_t *target, message_type_t type);
+    uint16_t broadcastMessage(const char *data, const uint8_t *target, message_type_t type);
-    void unicastMessage(const char *data, const uint8_t *target, const uint8_t *sender, message_type_t type);
+    uint16_t unicastMessage(const char *data, const uint8_t *target, const uint8_t *sender, message_type_t type);
     on_message_t onBroadcastReceivingCallback;
     on_message_t onUnicastReceivingCallback;
     on_confirm_t onConfirmReceivingCallback;