added new network module
This commit is contained in:
parent
4ac270475b
commit
1cdd9477c3
|
@ -1,4 +1,4 @@
|
||||||
idf_component_register(SRCS "https_client.c" "mesh_ota.c"
|
idf_component_register(SRCS "mesh_network.c" "mesh_ota.c"
|
||||||
INCLUDE_DIRS "include"
|
INCLUDE_DIRS "include"
|
||||||
REQUIRES nvs_flash
|
REQUIRES nvs_flash
|
||||||
esp_http_client
|
esp_http_client
|
||||||
|
|
|
@ -1,292 +0,0 @@
|
||||||
#include "https_client.h"
|
|
||||||
|
|
||||||
static const char *TAG = "https_client";
|
|
||||||
|
|
||||||
static const char *REQUEST = "GET " CONFIG_OTA_HTTPS_URL " HTTP/1.1\r\n"
|
|
||||||
"Host: "CONFIG_OTA_HTTPS_SERVER_COMMON_NAME"\r\n"
|
|
||||||
"User-Agent: esp-idf/1.0 esp32\r\n"
|
|
||||||
"Authorization: Basic " CONFIG_OTA_HTTPS_AUTH "\r\n"
|
|
||||||
"\r\n";
|
|
||||||
|
|
||||||
|
|
||||||
static HTTPS_Client_t sHTTPS_ClientConfig;
|
|
||||||
|
|
||||||
https_client_ret_t https_clientInitEmbedTLS();
|
|
||||||
https_client_ret_t https_clientConnectToServer();
|
|
||||||
https_client_ret_t https_clientValidateServer();
|
|
||||||
https_client_ret_t https_clientSendRequest();
|
|
||||||
|
|
||||||
https_client_ret_t https_clientInitialize()
|
|
||||||
{
|
|
||||||
https_client_ret_t i32RetHTTPClient = HTTPS_CLIENT_OK;
|
|
||||||
|
|
||||||
i32RetHTTPClient = https_clientInitEmbedTLS();
|
|
||||||
|
|
||||||
if(i32RetHTTPClient == HTTPS_CLIENT_OK)
|
|
||||||
{
|
|
||||||
i32RetHTTPClient = https_clientConnectToServer();
|
|
||||||
}
|
|
||||||
|
|
||||||
if(i32RetHTTPClient == HTTPS_CLIENT_OK)
|
|
||||||
{
|
|
||||||
i32RetHTTPClient = https_clientValidateServer();
|
|
||||||
}
|
|
||||||
|
|
||||||
if(i32RetHTTPClient == HTTPS_CLIENT_OK)
|
|
||||||
{
|
|
||||||
i32RetHTTPClient = https_clientSendRequest();
|
|
||||||
}
|
|
||||||
|
|
||||||
switch (i32RetHTTPClient)
|
|
||||||
{
|
|
||||||
case HTTPS_CLIENT_ERROR_INIT_EMBEDTLS:
|
|
||||||
ESP_LOGE(TAG, "Unable to initialize EmbedTLS");
|
|
||||||
i32RetHTTPClient = HTTPS_CLIENT_ERROR;
|
|
||||||
break;
|
|
||||||
case HTTPS_CLIENT_ERROR_INIT_CONNECT_TWO_SERVER:
|
|
||||||
ESP_LOGE(TAG, "Unable to connect to server");
|
|
||||||
i32RetHTTPClient = HTTPS_CLIENT_ERROR;
|
|
||||||
break;
|
|
||||||
case HTTPS_CLIENT_ERROR_INIT_VALIDATE_SERVER:
|
|
||||||
ESP_LOGE(TAG, "Unable to validate the server");
|
|
||||||
i32RetHTTPClient = HTTPS_CLIENT_ERROR;
|
|
||||||
break;
|
|
||||||
case HTTPS_CLIENT_ERROR_INIT_SEND_REQUEST:
|
|
||||||
ESP_LOGE(TAG, "Unable to send request to server");
|
|
||||||
i32RetHTTPClient = HTTPS_CLIENT_ERROR;
|
|
||||||
break;
|
|
||||||
case HTTPS_CLIENT_OK:
|
|
||||||
ESP_LOGI(TAG, "HTTPS Client successfully initialized");
|
|
||||||
i32RetHTTPClient = HTTPS_CLIENT_OK;
|
|
||||||
break;
|
|
||||||
default:
|
|
||||||
i32RetHTTPClient = HTTPS_CLIENT_ERROR;
|
|
||||||
ESP_LOGE(TAG, "Unknown error while init https client");
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
return i32RetHTTPClient;
|
|
||||||
}
|
|
||||||
|
|
||||||
https_client_ret_t https_clientRetrieveData(char* pu8Data, uint32_t* pu32DataLenght, uint32_t* pu32BytesRead)
|
|
||||||
{
|
|
||||||
https_client_ret_t i32RetHTTPClient = HTTPS_CLIENT_OK;
|
|
||||||
int32_t i32RetRetrieveData = ESP_OK;
|
|
||||||
bool bRetriveData = true;
|
|
||||||
|
|
||||||
bzero(pu8Data, *pu32DataLenght);
|
|
||||||
*pu32BytesRead = 0U;
|
|
||||||
|
|
||||||
while (bRetriveData)
|
|
||||||
{
|
|
||||||
//Reading HTTP response
|
|
||||||
i32RetRetrieveData = mbedtls_ssl_read(&sHTTPS_ClientConfig.ssl, (unsigned char *)(pu8Data+(*pu32BytesRead)), ((*pu32DataLenght)-(*pu32BytesRead)));
|
|
||||||
|
|
||||||
if(i32RetRetrieveData > 0)
|
|
||||||
{
|
|
||||||
//Data received
|
|
||||||
*pu32BytesRead = *pu32BytesRead + i32RetRetrieveData;
|
|
||||||
|
|
||||||
if(*pu32DataLenght > 0)
|
|
||||||
{
|
|
||||||
//buffer not full yet --> read some more
|
|
||||||
bRetriveData = true;
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
//buffer full --> stop reading
|
|
||||||
bRetriveData = false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if(i32RetRetrieveData == 0)
|
|
||||||
{
|
|
||||||
//all data read --> stop reading
|
|
||||||
bRetriveData = false;
|
|
||||||
pu32BytesRead = 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
if(i32RetRetrieveData == MBEDTLS_ERR_SSL_PEER_CLOSE_NOTIFY)
|
|
||||||
{
|
|
||||||
//connection is going to be closed
|
|
||||||
i32RetHTTPClient = HTTPS_CLIENT_ERROR;
|
|
||||||
bRetriveData = false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return i32RetHTTPClient;
|
|
||||||
}
|
|
||||||
|
|
||||||
https_client_ret_t https_clientDeinitialize()
|
|
||||||
{
|
|
||||||
https_client_ret_t i32RetHTTPClient = HTTPS_CLIENT_OK;
|
|
||||||
|
|
||||||
i32RetHTTPClient = mbedtls_ssl_close_notify(&sHTTPS_ClientConfig.ssl); //close session
|
|
||||||
|
|
||||||
if(i32RetHTTPClient != ESP_OK)
|
|
||||||
{
|
|
||||||
ESP_LOGE(TAG, "mbedtls_ssl_close_notify returned 0x%x", i32RetHTTPClient);
|
|
||||||
}
|
|
||||||
|
|
||||||
mbedtls_ssl_session_reset(&sHTTPS_ClientConfig.ssl); //reset embedssl
|
|
||||||
mbedtls_net_free(&sHTTPS_ClientConfig.server_fd); //free ram
|
|
||||||
|
|
||||||
return i32RetHTTPClient;
|
|
||||||
}
|
|
||||||
|
|
||||||
https_client_ret_t https_clientInitEmbedTLS() {
|
|
||||||
|
|
||||||
https_client_ret_t i32RetHTTPClient = HTTPS_CLIENT_OK;
|
|
||||||
int32_t i32RetEmbedTLS = ESP_OK;
|
|
||||||
|
|
||||||
mbedtls_ssl_init(&sHTTPS_ClientConfig.ssl);
|
|
||||||
mbedtls_x509_crt_init(&sHTTPS_ClientConfig.cacert);
|
|
||||||
mbedtls_ctr_drbg_init(&sHTTPS_ClientConfig.ctr_drbg);
|
|
||||||
mbedtls_ssl_config_init(&sHTTPS_ClientConfig.conf);
|
|
||||||
mbedtls_entropy_init(&sHTTPS_ClientConfig.entropy);
|
|
||||||
|
|
||||||
i32RetEmbedTLS = mbedtls_ctr_drbg_seed(&sHTTPS_ClientConfig.ctr_drbg, mbedtls_entropy_func, &sHTTPS_ClientConfig.entropy, NULL, 0);
|
|
||||||
|
|
||||||
if(i32RetEmbedTLS!= ESP_OK)
|
|
||||||
{
|
|
||||||
ESP_LOGE(TAG, "mbedtls_ctr_drbg_seed returned %d", i32RetEmbedTLS);
|
|
||||||
}
|
|
||||||
|
|
||||||
if(i32RetEmbedTLS == ESP_OK)
|
|
||||||
{
|
|
||||||
//Attaching the certificate bundle
|
|
||||||
i32RetEmbedTLS = esp_crt_bundle_attach(&sHTTPS_ClientConfig.conf);
|
|
||||||
if(i32RetEmbedTLS != ESP_OK)
|
|
||||||
{
|
|
||||||
ESP_LOGE(TAG, "esp_crt_bundle_attach returned 0x%x\n\n", i32RetEmbedTLS);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if(i32RetEmbedTLS == ESP_OK)
|
|
||||||
{
|
|
||||||
//Setting hostname for TLS session.
|
|
||||||
i32RetEmbedTLS = mbedtls_ssl_set_hostname(&sHTTPS_ClientConfig.ssl, CONFIG_OTA_HTTPS_SERVER_COMMON_NAME);
|
|
||||||
// Hostname set here should match CN in server certificate
|
|
||||||
if(i32RetEmbedTLS != ESP_OK)
|
|
||||||
{
|
|
||||||
ESP_LOGE(TAG, "mbedtls_ssl_set_hostname returned 0x%x", i32RetEmbedTLS);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if(i32RetEmbedTLS == ESP_OK)
|
|
||||||
{
|
|
||||||
//Setting up the SSL/TLS structure
|
|
||||||
i32RetEmbedTLS = mbedtls_ssl_config_defaults(&sHTTPS_ClientConfig.conf,
|
|
||||||
MBEDTLS_SSL_IS_CLIENT,
|
|
||||||
MBEDTLS_SSL_TRANSPORT_STREAM,
|
|
||||||
MBEDTLS_SSL_PRESET_DEFAULT);
|
|
||||||
|
|
||||||
if(i32RetEmbedTLS != ESP_OK)
|
|
||||||
{
|
|
||||||
ESP_LOGE(TAG, "mbedtls_ssl_config_defaults returned %d", i32RetEmbedTLS);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if(i32RetEmbedTLS == ESP_OK)
|
|
||||||
{
|
|
||||||
mbedtls_ssl_conf_authmode(&sHTTPS_ClientConfig.conf, MBEDTLS_SSL_VERIFY_REQUIRED);
|
|
||||||
mbedtls_ssl_conf_ca_chain(&sHTTPS_ClientConfig.conf, &sHTTPS_ClientConfig.cacert, NULL);
|
|
||||||
mbedtls_ssl_conf_rng(&sHTTPS_ClientConfig.conf, mbedtls_ctr_drbg_random, &sHTTPS_ClientConfig.ctr_drbg);
|
|
||||||
|
|
||||||
i32RetEmbedTLS = mbedtls_ssl_setup(&sHTTPS_ClientConfig.ssl, &sHTTPS_ClientConfig.conf);
|
|
||||||
if(i32RetEmbedTLS != ESP_OK)
|
|
||||||
{
|
|
||||||
ESP_LOGE(TAG, "mbedtls_ssl_setup returned -0x%x\n\n", i32RetEmbedTLS);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if(i32RetEmbedTLS == ESP_OK)
|
|
||||||
{
|
|
||||||
mbedtls_net_init(&sHTTPS_ClientConfig.server_fd);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (i32RetEmbedTLS != ESP_OK)
|
|
||||||
{
|
|
||||||
i32RetHTTPClient = HTTPS_CLIENT_ERROR_INIT_EMBEDTLS;
|
|
||||||
}
|
|
||||||
|
|
||||||
return i32RetHTTPClient;
|
|
||||||
}
|
|
||||||
|
|
||||||
https_client_ret_t https_clientConnectToServer()
|
|
||||||
{
|
|
||||||
https_client_ret_t i32RetHTTPClient = HTTPS_CLIENT_OK;
|
|
||||||
int32_t i32RetServerConnect = ESP_OK;
|
|
||||||
|
|
||||||
//Connecting to server
|
|
||||||
i32RetServerConnect = mbedtls_net_connect(&sHTTPS_ClientConfig.server_fd, CONFIG_OTA_HTTPS_SERVER_COMMON_NAME, CONFIG_OTA_HTTPS_SERVER_PORT, MBEDTLS_NET_PROTO_TCP);
|
|
||||||
if (i32RetServerConnect != ESP_OK)
|
|
||||||
{
|
|
||||||
ESP_LOGE(TAG, "mbedtls_net_connect returned %x", i32RetServerConnect);
|
|
||||||
}
|
|
||||||
|
|
||||||
if(i32RetServerConnect == ESP_OK)
|
|
||||||
{
|
|
||||||
mbedtls_ssl_set_bio(&sHTTPS_ClientConfig.ssl, &sHTTPS_ClientConfig.server_fd, mbedtls_net_send, mbedtls_net_recv, NULL);
|
|
||||||
|
|
||||||
//Performing the SSL/TLS handshake
|
|
||||||
while ((i32RetServerConnect = mbedtls_ssl_handshake(&sHTTPS_ClientConfig.ssl)) != 0)
|
|
||||||
{
|
|
||||||
if ((i32RetServerConnect != MBEDTLS_ERR_SSL_WANT_READ) && (i32RetServerConnect != MBEDTLS_ERR_SSL_WANT_WRITE))
|
|
||||||
{
|
|
||||||
ESP_LOGE(TAG, "mbedtls_ssl_handshake returned 0x%x", i32RetServerConnect);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if(i32RetServerConnect != ESP_OK)
|
|
||||||
{
|
|
||||||
i32RetHTTPClient = HTTPS_CLIENT_ERROR_INIT_CONNECT_TWO_SERVER;
|
|
||||||
}
|
|
||||||
return i32RetHTTPClient;
|
|
||||||
}
|
|
||||||
|
|
||||||
https_client_ret_t https_clientValidateServer()
|
|
||||||
{
|
|
||||||
https_client_ret_t i32RetHTTPClient = HTTPS_CLIENT_OK;
|
|
||||||
int32_t i32RetValidateServer = ESP_OK;
|
|
||||||
|
|
||||||
//Verifying peer X.509 certificate
|
|
||||||
if ((i32RetValidateServer = mbedtls_ssl_get_verify_result(&sHTTPS_ClientConfig.ssl)) != 0)
|
|
||||||
{
|
|
||||||
ESP_LOGE(TAG, "Failed to verify peer certificate!");
|
|
||||||
}
|
|
||||||
|
|
||||||
if(i32RetValidateServer != ESP_OK)
|
|
||||||
{
|
|
||||||
i32RetHTTPClient = HTTPS_CLIENT_ERROR_INIT_VALIDATE_SERVER;
|
|
||||||
}
|
|
||||||
return i32RetHTTPClient;
|
|
||||||
}
|
|
||||||
|
|
||||||
https_client_ret_t https_clientSendRequest()
|
|
||||||
{
|
|
||||||
https_client_ret_t i32RetHTTPClient = HTTPS_CLIENT_OK;
|
|
||||||
int32_t i32RetSendRequest = ESP_OK;
|
|
||||||
uint32_t u32WrittenBytes = 0;
|
|
||||||
bool bWrite = true; //flag to stop loop
|
|
||||||
|
|
||||||
//Writing HTTP request
|
|
||||||
while((u32WrittenBytes < strlen(REQUEST)) && bWrite)
|
|
||||||
{
|
|
||||||
i32RetSendRequest = mbedtls_ssl_write(&sHTTPS_ClientConfig.ssl,
|
|
||||||
(const unsigned char *)REQUEST + u32WrittenBytes,
|
|
||||||
strlen(REQUEST) - u32WrittenBytes);
|
|
||||||
if (i32RetSendRequest >= 0)
|
|
||||||
{
|
|
||||||
//bytes written
|
|
||||||
u32WrittenBytes += i32RetSendRequest;
|
|
||||||
} else if (i32RetSendRequest != MBEDTLS_ERR_SSL_WANT_WRITE && i32RetSendRequest != MBEDTLS_ERR_SSL_WANT_READ) {
|
|
||||||
ESP_LOGE(TAG, "mbedtls_ssl_write returned 0x%x", i32RetSendRequest);
|
|
||||||
bWrite = false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if(bWrite == false)
|
|
||||||
{
|
|
||||||
i32RetHTTPClient = HTTPS_CLIENT_ERROR_INIT_SEND_REQUEST;
|
|
||||||
}
|
|
||||||
return i32RetHTTPClient;
|
|
||||||
}
|
|
|
@ -1,75 +0,0 @@
|
||||||
#ifndef H_HTTPS_CLIENT
|
|
||||||
#define H_HTTPS_CLIENT
|
|
||||||
|
|
||||||
#include <string.h>
|
|
||||||
#include <stdlib.h>
|
|
||||||
#include "freertos/FreeRTOS.h"
|
|
||||||
#include "freertos/task.h"
|
|
||||||
#include "esp_wifi.h"
|
|
||||||
#include "esp_event.h"
|
|
||||||
#include "esp_log.h"
|
|
||||||
#include "esp_system.h"
|
|
||||||
#include "nvs_flash.h"
|
|
||||||
#include "esp_netif.h"
|
|
||||||
|
|
||||||
#include "lwip/err.h"
|
|
||||||
#include "lwip/sockets.h"
|
|
||||||
#include "lwip/sys.h"
|
|
||||||
#include "lwip/netdb.h"
|
|
||||||
#include "lwip/dns.h"
|
|
||||||
|
|
||||||
#include "mbedtls/platform.h"
|
|
||||||
#include "mbedtls/net_sockets.h"
|
|
||||||
#include "mbedtls/esp_debug.h"
|
|
||||||
#include "mbedtls/ssl.h"
|
|
||||||
#include "mbedtls/entropy.h"
|
|
||||||
#include "mbedtls/ctr_drbg.h"
|
|
||||||
#include "mbedtls/error.h"
|
|
||||||
#include "mbedtls/certs.h"
|
|
||||||
#include "esp_crt_bundle.h"
|
|
||||||
|
|
||||||
#ifndef CONFIG_OTA_HTTPS_URL
|
|
||||||
#define CONFIG_OTA_HTTPS_URL "https://exmaple.com/theImage.bin"
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#ifndef CONFIG_OTA_HTTPS_SERVER_PORT
|
|
||||||
#define CONFIG_OTA_HTTPS_SERVER_PORT "443"
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#ifndef CONFIG_OTA_HTTPS_AUTH
|
|
||||||
#define CONFIG_OTA_HTTPS_AUTH "base64(user:password)"
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#ifndef CONFIG_OTA_HTTPS_SERVER_COMMON_NAME
|
|
||||||
#define CONFIG_OTA_HTTPS_SERVER_COMMON_NAME "exmaple.com"
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#define HTTPS_CLIENT_OK 0
|
|
||||||
#define HTTPS_CLIENT_ERROR -1
|
|
||||||
#define HTTPS_CLIENT_ERROR_INIT_EMBEDTLS -2
|
|
||||||
#define HTTPS_CLIENT_ERROR_INIT_CONNECT_TWO_SERVER -3
|
|
||||||
#define HTTPS_CLIENT_ERROR_INIT_VALIDATE_SERVER -4
|
|
||||||
#define HTTPS_CLIENT_ERROR_INIT_SEND_REQUEST -5
|
|
||||||
|
|
||||||
struct HTTPS_Client
|
|
||||||
{
|
|
||||||
mbedtls_entropy_context entropy;
|
|
||||||
mbedtls_ctr_drbg_context ctr_drbg;
|
|
||||||
mbedtls_ssl_context ssl;
|
|
||||||
mbedtls_x509_crt cacert;
|
|
||||||
mbedtls_ssl_config conf;
|
|
||||||
mbedtls_net_context server_fd;
|
|
||||||
};
|
|
||||||
|
|
||||||
typedef int32_t https_client_ret_t;
|
|
||||||
typedef struct HTTPS_Client HTTPS_Client_t;
|
|
||||||
|
|
||||||
https_client_ret_t https_clientInitialize();
|
|
||||||
https_client_ret_t https_clientRetrieveData(char* pu8Data, uint32_t* pu32DataLenght, uint32_t* pu32BytesRead);
|
|
||||||
https_client_ret_t https_clientDeinitialize();
|
|
||||||
|
|
||||||
#endif /* H_HTTPS_CLIENT */
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -0,0 +1,28 @@
|
||||||
|
#ifndef H_MESH_NETWORK
|
||||||
|
#define H_MESH_NETWORK
|
||||||
|
|
||||||
|
#include <string.h>
|
||||||
|
#include "esp_wifi.h"
|
||||||
|
#include "esp_system.h"
|
||||||
|
#include "esp_event.h"
|
||||||
|
#include "esp_log.h"
|
||||||
|
#include "esp_mesh.h"
|
||||||
|
#include "esp_mesh_internal.h"
|
||||||
|
|
||||||
|
|
||||||
|
#define RX_SIZE (1234)
|
||||||
|
#define TX_SIZE (1234)
|
||||||
|
|
||||||
|
esp_err_t errMeshNetworkInitialize();
|
||||||
|
esp_err_t errMeshNetworkInitializeWiFi();
|
||||||
|
esp_err_t errMeshNetworkInitializeRouter(mesh_cfg_t* cfg);
|
||||||
|
|
||||||
|
|
||||||
|
bool bCheckMACEquality(uint8_t* pu8aMAC, uint8_t* pu8bMAC);
|
||||||
|
esp_err_t errGetChildren(mesh_addr_t children[], uint16_t* pu16ChildrenSize);
|
||||||
|
esp_err_t errSendPacket(mesh_addr_t* dest, struct ota_mesh_packet* packet);
|
||||||
|
void vMeshEventHandler(void *arg, esp_event_base_t event_base, int32_t i32EventID, void* vpEventData);
|
||||||
|
void vIPEventHandler(void *arg, esp_event_base_t event_base, int32_t i32EventID, void *event_data);
|
||||||
|
|
||||||
|
#endif /* H_MESH_NETWORK */
|
||||||
|
|
|
@ -13,10 +13,38 @@
|
||||||
#include "esp_ota_ops.h"
|
#include "esp_ota_ops.h"
|
||||||
#include "esp_partition.h"
|
#include "esp_partition.h"
|
||||||
|
|
||||||
#include "https_client.h"
|
#include "mesh_network.h"
|
||||||
|
|
||||||
|
#define ERASE_NVS //erase non volatile storage if full
|
||||||
|
|
||||||
|
struct ota_mesh_packet
|
||||||
|
{
|
||||||
|
enum ota_mesh_packet_type
|
||||||
|
{
|
||||||
|
APP_Version_Request,
|
||||||
|
APP_Version_Response,
|
||||||
|
OTA_Data,
|
||||||
|
OTA_ACK,
|
||||||
|
OTA_Complete
|
||||||
|
} type;
|
||||||
|
|
||||||
|
uint8_t au8Payload[1024];
|
||||||
|
};
|
||||||
|
|
||||||
|
#define ERROR_CHECK(x) if (err == ESP_OK) \
|
||||||
|
{ \
|
||||||
|
err = (x); \
|
||||||
|
if (err != ESP_OK) \
|
||||||
|
{ \
|
||||||
|
ESP_LOGE(LOG_TAG, "%s failed with error: %d -> %s", #x, err, esp_err_to_name(err)); \
|
||||||
|
} \
|
||||||
|
} \
|
||||||
|
|
||||||
|
|
||||||
bool bNewerVersion(const char* pu8Local, const char* pu8Remote);
|
bool bNewerVersion(const char* pu8Local, const char* pu8Remote);
|
||||||
esp_err_t errExtractVersionNumber(const char* pu8Data, uint32_t* pu32DataLenght, char* pc8RemoteVersionNumber);
|
esp_err_t errExtractVersionNumber(const char* pu8Data, uint32_t* pu32DataLenght, char* pc8RemoteVersionNumber);
|
||||||
esp_err_t errFindImageStart(const char* pu8Data, uint32_t* pu32DataLenght, uint32_t* pu32StartOffset);
|
esp_err_t errFindImageStart(const char* pu8Data, uint32_t* pu32DataLenght, uint32_t* pu32StartOffset);
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#endif /* H_MESH_OTA */
|
#endif /* H_MESH_OTA */
|
||||||
|
|
|
@ -0,0 +1,444 @@
|
||||||
|
|
||||||
|
#include "mesh_ota.h"
|
||||||
|
|
||||||
|
static const char *LOG_TAG = "mesh_network";
|
||||||
|
static const uint8_t MESH_ID[6] = { 0x77, 0x77, 0x77, 0x77, 0x77, 0x77};
|
||||||
|
static uint8_t tx_buf[TX_SIZE] = { 0, };
|
||||||
|
static uint8_t rx_buf[RX_SIZE] = { 0, };
|
||||||
|
static bool is_mesh_connected = false;
|
||||||
|
static mesh_addr_t mesh_parent_addr;
|
||||||
|
|
||||||
|
static int mesh_layer = -1;
|
||||||
|
static esp_netif_t *netif_sta = NULL;
|
||||||
|
|
||||||
|
uint8_t ownMAC[6];
|
||||||
|
|
||||||
|
|
||||||
|
esp_err_t errMeshNetworkInitialize()
|
||||||
|
{
|
||||||
|
esp_err_t err;
|
||||||
|
|
||||||
|
err = nvs_flash_init(); //init non-volatile storage
|
||||||
|
|
||||||
|
#ifdef ERASE_NVS
|
||||||
|
if(err == ESP_ERR_NVS_NO_FREE_PAGES) //check if storage is full
|
||||||
|
{
|
||||||
|
ERROR_CHECK(nvs_flash_erase());
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
// tcpip initialization
|
||||||
|
ERROR_CHECK(esp_netif_init());
|
||||||
|
|
||||||
|
//event initialization
|
||||||
|
ERROR_CHECK(esp_event_loop_create_default());
|
||||||
|
|
||||||
|
//create network interfaces for mesh (only station instance saved for further manipulation, soft AP instance ignored
|
||||||
|
ERROR_CHECK(esp_netif_create_default_wifi_mesh_netifs(&netif_sta, NULL));
|
||||||
|
|
||||||
|
//wifi initialization
|
||||||
|
ERROR_CHECK(errMeshNetworkInitializeWiFi());
|
||||||
|
|
||||||
|
//mesh initialization
|
||||||
|
ERROR_CHECK(esp_mesh_init());
|
||||||
|
|
||||||
|
//mesh initialization
|
||||||
|
ERROR_CHECK(esp_event_handler_register(MESH_EVENT, ESP_EVENT_ANY_ID, &vMeshEventHandler, NULL));
|
||||||
|
|
||||||
|
//set mesh topology
|
||||||
|
ERROR_CHECK(esp_mesh_set_topology(CONFIG_MESH_TOPOLOGY));
|
||||||
|
|
||||||
|
//set mesh max layer according to the topology
|
||||||
|
ERROR_CHECK(esp_mesh_set_max_layer(CONFIG_MESH_MAX_LAYER));
|
||||||
|
ERROR_CHECK(esp_mesh_set_vote_percentage(1));
|
||||||
|
ERROR_CHECK(esp_mesh_set_xon_qsize(128));
|
||||||
|
|
||||||
|
//Disable mesh PS function
|
||||||
|
ERROR_CHECK(esp_mesh_disable_ps());
|
||||||
|
ERROR_CHECK(esp_mesh_set_ap_assoc_expire(10));
|
||||||
|
|
||||||
|
mesh_cfg_t cfg = MESH_INIT_CONFIG_DEFAULT();
|
||||||
|
|
||||||
|
/* mesh ID */
|
||||||
|
memcpy((uint8_t *) &cfg.mesh_id, MESH_ID, 6);
|
||||||
|
|
||||||
|
ERROR_CHECK(errMeshNetworkInitializeRouter(&cfg));
|
||||||
|
|
||||||
|
/* mesh softAP */
|
||||||
|
ERROR_CHECK(esp_mesh_set_ap_authmode(CONFIG_MESH_AP_AUTHMODE));
|
||||||
|
cfg.mesh_ap.max_connection = CONFIG_MESH_AP_CONNECTIONS;
|
||||||
|
memcpy((uint8_t *) &cfg.mesh_ap.password, CONFIG_MESH_AP_PASSWD,
|
||||||
|
strlen(CONFIG_MESH_AP_PASSWD));
|
||||||
|
ERROR_CHECK(esp_mesh_set_config(&cfg));
|
||||||
|
|
||||||
|
/* mesh start */
|
||||||
|
ERROR_CHECK(esp_mesh_start());
|
||||||
|
|
||||||
|
ERROR_CHECK(esp_base_mac_addr_get(ownMAC))
|
||||||
|
|
||||||
|
//debug info
|
||||||
|
ESP_LOGD(LOG_TAG, "mesh starts successfully, heap:%d, %s<%d>%s, ps:%d\n", esp_get_minimum_free_heap_size(),
|
||||||
|
esp_mesh_is_root_fixed() ? "root fixed" : "root not fixed",
|
||||||
|
esp_mesh_get_topology(), esp_mesh_get_topology() ? "(chain)":"(tree)", esp_mesh_is_ps_enabled());
|
||||||
|
|
||||||
|
|
||||||
|
ESP_LOGI(LOG_TAG, "Node MAC: \"%x:%x:%x:%x:%x:%x\" ", ownMAC[0], ownMAC[1], ownMAC[2], ownMAC[3], ownMAC[4], ownMAC[5]);
|
||||||
|
|
||||||
|
return ESP_OK;
|
||||||
|
}
|
||||||
|
|
||||||
|
esp_err_t errMeshNetworkInitializeWiFi()
|
||||||
|
{
|
||||||
|
//wifi initialization
|
||||||
|
esp_err_t err = ESP_OK;
|
||||||
|
wifi_init_config_t config = WIFI_INIT_CONFIG_DEFAULT();
|
||||||
|
ERROR_CHECK(esp_wifi_init(&config));
|
||||||
|
ERROR_CHECK(esp_event_handler_register(IP_EVENT, IP_EVENT_STA_GOT_IP, &vIPEventHandler, NULL));
|
||||||
|
ERROR_CHECK(esp_wifi_set_storage(WIFI_STORAGE_FLASH));
|
||||||
|
ERROR_CHECK(esp_wifi_start());
|
||||||
|
return err;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
esp_err_t errMeshNetworkInitializeRouter(mesh_cfg_t* cfg)
|
||||||
|
{
|
||||||
|
//router initialization
|
||||||
|
esp_err_t err = ESP_OK;
|
||||||
|
(*cfg).channel = CONFIG_MESH_CHANNEL;
|
||||||
|
(*cfg).router.ssid_len = strlen(CONFIG_MESH_ROUTER_SSID);
|
||||||
|
memcpy((uint8_t *) &(*cfg).router.ssid, CONFIG_MESH_ROUTER_SSID, (*cfg).router.ssid_len);
|
||||||
|
memcpy((uint8_t *) &(*cfg).router.password, CONFIG_MESH_ROUTER_PASSWD,
|
||||||
|
strlen(CONFIG_MESH_ROUTER_PASSWD));
|
||||||
|
return err;
|
||||||
|
}
|
||||||
|
|
||||||
|
//returns true if MAC address is equal
|
||||||
|
bool bCheckMACEquality(uint8_t* pu8aMAC, uint8_t* pu8bMAC)
|
||||||
|
{
|
||||||
|
bool bRet = true;
|
||||||
|
uint8_t index = 0;
|
||||||
|
|
||||||
|
while ((index < 6) && (bRet == true))
|
||||||
|
{
|
||||||
|
if(pu8aMAC[index] != pu8bMAC[index])
|
||||||
|
{
|
||||||
|
bRet = false;
|
||||||
|
}
|
||||||
|
index++;
|
||||||
|
}
|
||||||
|
return bRet;
|
||||||
|
}
|
||||||
|
|
||||||
|
esp_err_t errGetChildren(mesh_addr_t children[], uint16_t* pu16ChildrenSize)
|
||||||
|
{
|
||||||
|
esp_err_t err = ESP_OK;
|
||||||
|
int route_table_size = 0;
|
||||||
|
*pu16ChildrenSize = 0;
|
||||||
|
mesh_addr_t route_table[CONFIG_MESH_ROUTE_TABLE_SIZE];
|
||||||
|
ERROR_CHECK(esp_mesh_get_routing_table((mesh_addr_t *) &route_table, (CONFIG_MESH_ROUTE_TABLE_SIZE * 6), &route_table_size));
|
||||||
|
|
||||||
|
if (err == ESP_OK)
|
||||||
|
{
|
||||||
|
for(uint16_t index = 0; index < esp_mesh_get_routing_table_size(); index++)
|
||||||
|
{
|
||||||
|
if(! (bCheckMACEquality(ownMAC, route_table[index].addr)) )
|
||||||
|
{
|
||||||
|
//child node
|
||||||
|
//ESP_LOGI(MESH_TAG, "adding Node: \"0x%x, 0x%x, 0x%x, 0x%x, 0x%x, 0x%x\" ", route_table[index].addr[0], route_table[index].addr[1], route_table[index].addr[2], route_table[index].addr[3], route_table[index].addr[4], route_table[index].addr[5]);
|
||||||
|
children[*pu16ChildrenSize] = route_table[index];
|
||||||
|
*pu16ChildrenSize = (*pu16ChildrenSize)+1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return err;
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
esp_err_t errSendPacket(mesh_addr_t* dest, struct ota_mesh_packet* packet)
|
||||||
|
{
|
||||||
|
esp_err_t err;
|
||||||
|
mesh_data_t data;
|
||||||
|
data.data = tx_buf;
|
||||||
|
data.size = sizeof(tx_buf);
|
||||||
|
data.proto = MESH_PROTO_BIN;
|
||||||
|
data.tos = MESH_TOS_P2P;
|
||||||
|
memcpy(tx_buf, (uint8_t *)packet, sizeof(struct ota_mesh_packet));
|
||||||
|
err = esp_mesh_send(dest, &data, MESH_DATA_P2P, NULL, 0);
|
||||||
|
return err;
|
||||||
|
}
|
||||||
|
*/
|
||||||
|
|
||||||
|
void esp_mesh_p2p_rx_main(void *arg)
|
||||||
|
{
|
||||||
|
esp_err_t err;
|
||||||
|
mesh_addr_t from;
|
||||||
|
mesh_data_t data;
|
||||||
|
int flag = 0;
|
||||||
|
data.data = rx_buf;
|
||||||
|
data.size = RX_SIZE;
|
||||||
|
|
||||||
|
while (true)
|
||||||
|
{
|
||||||
|
data.size = RX_SIZE;
|
||||||
|
err = esp_mesh_recv(&from, &data, portMAX_DELAY, &flag, NULL, 0);
|
||||||
|
if (err != ESP_OK || !data.size)
|
||||||
|
{
|
||||||
|
ESP_LOGE(LOG_TAG, "err:0x%x, size:%d", err, data.size);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
struct ota_mesh_packet packet;
|
||||||
|
memcpy(&packet, (uint8_t *)rx_buf, sizeof(struct ota_mesh_packet));
|
||||||
|
|
||||||
|
switch (packet.type)
|
||||||
|
{
|
||||||
|
case APP_Version_Request:
|
||||||
|
ESP_LOGI(LOG_TAG, "recv: APP_Version_Request");
|
||||||
|
packet.type=APP_Version_Response;
|
||||||
|
packet.au8Payload[0] = 42; //TODO get current running version
|
||||||
|
// ESP_ERROR_CHECK (esp_mesh_send_packet(&from, &packet)); //send back to parent
|
||||||
|
break;
|
||||||
|
case APP_Version_Response:
|
||||||
|
ESP_LOGI(LOG_TAG, "recv: APP_Version_Response - App Version %i: ", packet.au8Payload[0]);
|
||||||
|
//check if node is out-dated
|
||||||
|
// esp_mesh_ota_send(&from);
|
||||||
|
break;
|
||||||
|
case OTA_Data:
|
||||||
|
ESP_LOGI(LOG_TAG, "recv: OTA_Data");
|
||||||
|
//esp_mesh_ota_receive(&from, &packet);
|
||||||
|
packet.type=OTA_ACK;
|
||||||
|
//ESP_ERROR_CHECK (esp_mesh_send_packet(&from, &packet)); //send back to parent
|
||||||
|
break;
|
||||||
|
case OTA_ACK:
|
||||||
|
ESP_LOGI(LOG_TAG, "recv: OTA_ACK");
|
||||||
|
//esp_mesh_ota_send(&from);
|
||||||
|
break;
|
||||||
|
case OTA_Complete:
|
||||||
|
ESP_LOGI(LOG_TAG, "recv: OTA_Complete");
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
ESP_LOGE(LOG_TAG, "recv: something");
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
} //end while
|
||||||
|
vTaskDelete(NULL);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
esp_err_t esp_mesh_comm_p2p_start(void)
|
||||||
|
{
|
||||||
|
static bool is_comm_p2p_started = false;
|
||||||
|
if (!is_comm_p2p_started)
|
||||||
|
{
|
||||||
|
is_comm_p2p_started = true;
|
||||||
|
xTaskCreate(esp_mesh_p2p_rx_main, "MPRX", 7000, NULL, 5, NULL);
|
||||||
|
}
|
||||||
|
return ESP_OK;
|
||||||
|
}
|
||||||
|
|
||||||
|
void vMeshEventHandler(void *arg, esp_event_base_t event_base, int32_t i32EventID, void* vpEventData)
|
||||||
|
{
|
||||||
|
mesh_addr_t id = {0,};
|
||||||
|
static uint16_t last_layer = 0;
|
||||||
|
|
||||||
|
switch (i32EventID)
|
||||||
|
{
|
||||||
|
case MESH_EVENT_STARTED:
|
||||||
|
{
|
||||||
|
esp_mesh_get_id(&id);
|
||||||
|
ESP_LOGI(LOG_TAG, "<MESH_EVENT_MESH_STARTED>ID:"MACSTR"", MAC2STR(id.addr));
|
||||||
|
is_mesh_connected = false;
|
||||||
|
mesh_layer = esp_mesh_get_layer();
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case MESH_EVENT_STOPPED: {
|
||||||
|
ESP_LOGI(LOG_TAG, "<MESH_EVENT_STOPPED>");
|
||||||
|
is_mesh_connected = false;
|
||||||
|
mesh_layer = esp_mesh_get_layer();
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case MESH_EVENT_CHILD_CONNECTED: {
|
||||||
|
mesh_event_child_connected_t *child_connected = (mesh_event_child_connected_t *)vpEventData;
|
||||||
|
ESP_LOGI(LOG_TAG, "<MESH_EVENT_CHILD_CONNECTED>aid:%d, "MACSTR"",
|
||||||
|
child_connected->aid,
|
||||||
|
MAC2STR(child_connected->mac));
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case MESH_EVENT_CHILD_DISCONNECTED: {
|
||||||
|
mesh_event_child_disconnected_t *child_disconnected = (mesh_event_child_disconnected_t *)vpEventData;
|
||||||
|
ESP_LOGI(LOG_TAG, "<MESH_EVENT_CHILD_DISCONNECTED>aid:%d, "MACSTR"",
|
||||||
|
child_disconnected->aid,
|
||||||
|
MAC2STR(child_disconnected->mac));
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case MESH_EVENT_ROUTING_TABLE_ADD: {
|
||||||
|
mesh_event_routing_table_change_t *routing_table = (mesh_event_routing_table_change_t *)vpEventData;
|
||||||
|
ESP_LOGW(LOG_TAG, "<MESH_EVENT_ROUTING_TABLE_ADD>add %d, new:%d, layer:%d",
|
||||||
|
routing_table->rt_size_change,
|
||||||
|
routing_table->rt_size_new, mesh_layer);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case MESH_EVENT_ROUTING_TABLE_REMOVE: {
|
||||||
|
mesh_event_routing_table_change_t *routing_table = (mesh_event_routing_table_change_t *)vpEventData;
|
||||||
|
ESP_LOGW(LOG_TAG, "<MESH_EVENT_ROUTING_TABLE_REMOVE>remove %d, new:%d, layer:%d",
|
||||||
|
routing_table->rt_size_change,
|
||||||
|
routing_table->rt_size_new, mesh_layer);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case MESH_EVENT_NO_PARENT_FOUND: {
|
||||||
|
mesh_event_no_parent_found_t *no_parent = (mesh_event_no_parent_found_t *)vpEventData;
|
||||||
|
ESP_LOGI(LOG_TAG, "<MESH_EVENT_NO_PARENT_FOUND>scan times:%d",
|
||||||
|
no_parent->scan_times);
|
||||||
|
}
|
||||||
|
/* TODO handler for the failure */
|
||||||
|
break;
|
||||||
|
case MESH_EVENT_PARENT_CONNECTED: {
|
||||||
|
mesh_event_connected_t *connected = (mesh_event_connected_t *)vpEventData;
|
||||||
|
esp_mesh_get_id(&id);
|
||||||
|
mesh_layer = connected->self_layer;
|
||||||
|
memcpy(&mesh_parent_addr.addr, connected->connected.bssid, 6);
|
||||||
|
ESP_LOGI(LOG_TAG,
|
||||||
|
"<MESH_EVENT_PARENT_CONNECTED>layer:%d-->%d, parent:"MACSTR"%s, ID:"MACSTR", duty:%d",
|
||||||
|
last_layer, mesh_layer, MAC2STR(mesh_parent_addr.addr),
|
||||||
|
esp_mesh_is_root() ? "<ROOT>" :
|
||||||
|
(mesh_layer == 2) ? "<layer2>" : "", MAC2STR(id.addr), connected->duty);
|
||||||
|
last_layer = mesh_layer;
|
||||||
|
// mesh_connected_indicator(mesh_layer);
|
||||||
|
is_mesh_connected = true;
|
||||||
|
if (esp_mesh_is_root()) {
|
||||||
|
esp_netif_dhcpc_start(netif_sta);
|
||||||
|
}
|
||||||
|
esp_mesh_comm_p2p_start();//start receiving
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case MESH_EVENT_PARENT_DISCONNECTED: {
|
||||||
|
mesh_event_disconnected_t *disconnected = (mesh_event_disconnected_t *)vpEventData;
|
||||||
|
ESP_LOGI(LOG_TAG,
|
||||||
|
"<MESH_EVENT_PARENT_DISCONNECTED>reason:%d",
|
||||||
|
disconnected->reason);
|
||||||
|
is_mesh_connected = false;
|
||||||
|
// mesh_disconnected_indicator();
|
||||||
|
mesh_layer = esp_mesh_get_layer();
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case MESH_EVENT_LAYER_CHANGE: {
|
||||||
|
mesh_event_layer_change_t *layer_change = (mesh_event_layer_change_t *)vpEventData;
|
||||||
|
mesh_layer = layer_change->new_layer;
|
||||||
|
ESP_LOGI(LOG_TAG, "<MESH_EVENT_LAYER_CHANGE>layer:%d-->%d%s",
|
||||||
|
last_layer, mesh_layer,
|
||||||
|
esp_mesh_is_root() ? "<ROOT>" :
|
||||||
|
(mesh_layer == 2) ? "<layer2>" : "");
|
||||||
|
last_layer = mesh_layer;
|
||||||
|
// mesh_connected_indicator(mesh_layer);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case MESH_EVENT_ROOT_ADDRESS: {
|
||||||
|
mesh_event_root_address_t *root_addr = (mesh_event_root_address_t *)vpEventData;
|
||||||
|
ESP_LOGI(LOG_TAG, "<MESH_EVENT_ROOT_ADDRESS>root address:"MACSTR"",
|
||||||
|
MAC2STR(root_addr->addr));
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case MESH_EVENT_VOTE_STARTED: {
|
||||||
|
mesh_event_vote_started_t *vote_started = (mesh_event_vote_started_t *)vpEventData;
|
||||||
|
ESP_LOGI(LOG_TAG,
|
||||||
|
"<MESH_EVENT_VOTE_STARTED>attempts:%d, reason:%d, rc_addr:"MACSTR"",
|
||||||
|
vote_started->attempts,
|
||||||
|
vote_started->reason,
|
||||||
|
MAC2STR(vote_started->rc_addr.addr));
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case MESH_EVENT_VOTE_STOPPED: {
|
||||||
|
ESP_LOGI(LOG_TAG, "<MESH_EVENT_VOTE_STOPPED>");
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case MESH_EVENT_ROOT_SWITCH_REQ: {
|
||||||
|
mesh_event_root_switch_req_t *switch_req = (mesh_event_root_switch_req_t *)vpEventData;
|
||||||
|
ESP_LOGI(LOG_TAG,
|
||||||
|
"<MESH_EVENT_ROOT_SWITCH_REQ>reason:%d, rc_addr:"MACSTR"",
|
||||||
|
switch_req->reason,
|
||||||
|
MAC2STR( switch_req->rc_addr.addr));
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case MESH_EVENT_ROOT_SWITCH_ACK: {
|
||||||
|
/* new root */
|
||||||
|
mesh_layer = esp_mesh_get_layer();
|
||||||
|
esp_mesh_get_parent_bssid(&mesh_parent_addr);
|
||||||
|
ESP_LOGI(LOG_TAG, "<MESH_EVENT_ROOT_SWITCH_ACK>layer:%d, parent:"MACSTR"", mesh_layer, MAC2STR(mesh_parent_addr.addr));
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case MESH_EVENT_TODS_STATE: {
|
||||||
|
mesh_event_toDS_state_t *toDs_state = (mesh_event_toDS_state_t *)vpEventData;
|
||||||
|
ESP_LOGI(LOG_TAG, "<MESH_EVENT_TODS_REACHABLE>state:%d", *toDs_state);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case MESH_EVENT_ROOT_FIXED: {
|
||||||
|
mesh_event_root_fixed_t *root_fixed = (mesh_event_root_fixed_t *)vpEventData;
|
||||||
|
ESP_LOGI(LOG_TAG, "<MESH_EVENT_ROOT_FIXED>%s",
|
||||||
|
root_fixed->is_fixed ? "fixed" : "not fixed");
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case MESH_EVENT_ROOT_ASKED_YIELD: {
|
||||||
|
mesh_event_root_conflict_t *root_conflict = (mesh_event_root_conflict_t *)vpEventData;
|
||||||
|
ESP_LOGI(LOG_TAG,
|
||||||
|
"<MESH_EVENT_ROOT_ASKED_YIELD>"MACSTR", rssi:%d, capacity:%d",
|
||||||
|
MAC2STR(root_conflict->addr),
|
||||||
|
root_conflict->rssi,
|
||||||
|
root_conflict->capacity);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case MESH_EVENT_CHANNEL_SWITCH: {
|
||||||
|
mesh_event_channel_switch_t *channel_switch = (mesh_event_channel_switch_t *)vpEventData;
|
||||||
|
ESP_LOGI(LOG_TAG, "<MESH_EVENT_CHANNEL_SWITCH>new channel:%d", channel_switch->channel);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case MESH_EVENT_SCAN_DONE: {
|
||||||
|
mesh_event_scan_done_t *scan_done = (mesh_event_scan_done_t *)vpEventData;
|
||||||
|
ESP_LOGI(LOG_TAG, "<MESH_EVENT_SCAN_DONE>number:%d",
|
||||||
|
scan_done->number);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case MESH_EVENT_NETWORK_STATE: {
|
||||||
|
mesh_event_network_state_t *network_state = (mesh_event_network_state_t *)vpEventData;
|
||||||
|
ESP_LOGI(LOG_TAG, "<MESH_EVENT_NETWORK_STATE>is_rootless:%d",
|
||||||
|
network_state->is_rootless);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case MESH_EVENT_STOP_RECONNECTION: {
|
||||||
|
ESP_LOGI(LOG_TAG, "<MESH_EVENT_STOP_RECONNECTION>");
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case MESH_EVENT_FIND_NETWORK: {
|
||||||
|
mesh_event_find_network_t *find_network = (mesh_event_find_network_t *)vpEventData;
|
||||||
|
ESP_LOGI(LOG_TAG, "<MESH_EVENT_FIND_NETWORK>new channel:%d, router BSSID:"MACSTR"",
|
||||||
|
find_network->channel, MAC2STR(find_network->router_bssid));
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case MESH_EVENT_ROUTER_SWITCH: {
|
||||||
|
mesh_event_router_switch_t *router_switch = (mesh_event_router_switch_t *)vpEventData;
|
||||||
|
ESP_LOGI(LOG_TAG, "<MESH_EVENT_ROUTER_SWITCH>new router:%s, channel:%d, "MACSTR"",
|
||||||
|
router_switch->ssid, router_switch->channel, MAC2STR(router_switch->bssid));
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case MESH_EVENT_PS_PARENT_DUTY: {
|
||||||
|
mesh_event_ps_duty_t *ps_duty = (mesh_event_ps_duty_t *)vpEventData;
|
||||||
|
ESP_LOGI(LOG_TAG, "<MESH_EVENT_PS_PARENT_DUTY>duty:%d", ps_duty->duty);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case MESH_EVENT_PS_CHILD_DUTY: {
|
||||||
|
mesh_event_ps_duty_t *ps_duty = (mesh_event_ps_duty_t *)vpEventData;
|
||||||
|
ESP_LOGI(LOG_TAG, "<MESH_EVENT_PS_CHILD_DUTY>cidx:%d, "MACSTR", duty:%d", ps_duty->child_connected.aid-1,
|
||||||
|
MAC2STR(ps_duty->child_connected.mac), ps_duty->duty);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
ESP_LOGI(LOG_TAG, "unknown id:%d", i32EventID);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void vIPEventHandler(void *arg, esp_event_base_t event_base, int32_t i32EventID, void *vpEventData)
|
||||||
|
{
|
||||||
|
ip_event_got_ip_t *event = (ip_event_got_ip_t *) vpEventData;
|
||||||
|
ESP_LOGI(LOG_TAG, "<IP_EVENT_STA_GOT_IP>IP:" IPSTR, IP2STR(&event->ip_info.ip));
|
||||||
|
}
|
|
@ -126,3 +126,112 @@ esp_err_t errExtractVersionNumber(const char* pu8Data, uint32_t* pu32DataLenght,
|
||||||
}
|
}
|
||||||
return err;
|
return err;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
|
||||||
|
|
||||||
|
esp_err_t esp_mesh_ota_send(mesh_addr_t* dest)
|
||||||
|
{
|
||||||
|
esp_err_t err = ESP_OK;
|
||||||
|
|
||||||
|
static uint32_t u32index;
|
||||||
|
|
||||||
|
const esp_partition_t * currentPartition = esp_ota_get_boot_partition();
|
||||||
|
|
||||||
|
if((*currentPartition).subtype == 0)
|
||||||
|
{
|
||||||
|
|
||||||
|
int data_read = 0;
|
||||||
|
|
||||||
|
struct ota_mesh_packet packet;
|
||||||
|
packet.type=OTA_Data;
|
||||||
|
|
||||||
|
if(u32index == 1024)
|
||||||
|
{
|
||||||
|
//all data read
|
||||||
|
data_read = 0;
|
||||||
|
u32index = 0;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
ESP_LOGI(MESH_TAG, "OTA-Data read: %i", u32index);
|
||||||
|
err = esp_partition_read(currentPartition, (1024*u32index), packet.au8Payload, 1024 );
|
||||||
|
ESP_ERROR_CHECK(err);
|
||||||
|
data_read = 1024;
|
||||||
|
u32index++;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (data_read > 0)
|
||||||
|
{
|
||||||
|
//send ota fragemnt to node
|
||||||
|
esp_mesh_send_packet(dest, &packet);
|
||||||
|
}
|
||||||
|
|
||||||
|
ESP_ERROR_CHECK(err);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
ESP_LOGI(MESH_TAG, "Subtype: %d", (*currentPartition).subtype);
|
||||||
|
}
|
||||||
|
return err;
|
||||||
|
}
|
||||||
|
|
||||||
|
esp_err_t esp_mesh_ota_receive(mesh_addr_t* dest, struct ota_mesh_packet* packet)
|
||||||
|
{
|
||||||
|
esp_err_t err = ESP_OK;
|
||||||
|
static esp_ota_handle_t otaHandle;
|
||||||
|
static uint32_t u32index;
|
||||||
|
|
||||||
|
const esp_partition_t * currentPartition = esp_ota_get_boot_partition();
|
||||||
|
const esp_partition_t * otaPartition = esp_ota_get_next_update_partition(currentPartition);
|
||||||
|
|
||||||
|
if(u32index == 0)
|
||||||
|
{
|
||||||
|
//first run
|
||||||
|
|
||||||
|
err = esp_ota_begin(otaPartition, OTA_SIZE_UNKNOWN, &otaHandle);
|
||||||
|
ESP_ERROR_CHECK(err);
|
||||||
|
}
|
||||||
|
|
||||||
|
ESP_LOGI(MESH_TAG, "OTA-Data write: %i", u32index);
|
||||||
|
err = esp_ota_write(otaHandle, packet->au8Payload, 1024);
|
||||||
|
|
||||||
|
if(err != ESP_OK)
|
||||||
|
{
|
||||||
|
ESP_LOGE(MESH_TAG, "OTA-Data write error: %i at %i", err, u32index);
|
||||||
|
}
|
||||||
|
|
||||||
|
ESP_ERROR_CHECK(err);
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
if(u32index >= 1023)
|
||||||
|
{
|
||||||
|
//ota update complete
|
||||||
|
|
||||||
|
ESP_LOGI(MESH_TAG, "OTA-Data complete arrived: %i", u32index);
|
||||||
|
err = esp_ota_end(otaHandle);
|
||||||
|
ESP_ERROR_CHECK(err);
|
||||||
|
esp_app_desc_t otaPartitionDesc;
|
||||||
|
err = esp_ota_get_partition_description(otaPartition, &otaPartitionDesc);
|
||||||
|
ESP_ERROR_CHECK(err);
|
||||||
|
ESP_LOGI(MESH_TAG, "otaPartition project_name: %s", (otaPartitionDesc).project_name);
|
||||||
|
|
||||||
|
err = esp_ota_set_boot_partition(otaPartition);
|
||||||
|
ESP_ERROR_CHECK(err);
|
||||||
|
|
||||||
|
struct ota_mesh_packet retPacket;
|
||||||
|
retPacket.type=OTA_Complete;
|
||||||
|
ESP_ERROR_CHECK (esp_mesh_send_packet(dest, &retPacket)); //send back to parent
|
||||||
|
|
||||||
|
//check if this node has children --> Update them
|
||||||
|
|
||||||
|
esp_restart();
|
||||||
|
}
|
||||||
|
|
||||||
|
u32index++;
|
||||||
|
|
||||||
|
|
||||||
|
return err;
|
||||||
|
}
|
||||||
|
*/
|
636
main/main.c
636
main/main.c
|
@ -13,648 +13,26 @@
|
||||||
|
|
||||||
#include "mesh_ota.h"
|
#include "mesh_ota.h"
|
||||||
|
|
||||||
#define RX_SIZE (1234)
|
static const char *LOG_TAG = "esp_main";
|
||||||
#define TX_SIZE (1234)
|
|
||||||
|
|
||||||
static const char *MESH_TAG = "mesh_main";
|
|
||||||
static const uint8_t MESH_ID[6] = { 0x77, 0x77, 0x77, 0x77, 0x77, 0x77};
|
|
||||||
static uint8_t tx_buf[TX_SIZE] = { 0, };
|
|
||||||
static uint8_t rx_buf[RX_SIZE] = { 0, };
|
|
||||||
static bool is_mesh_connected = false;
|
|
||||||
static mesh_addr_t mesh_parent_addr;
|
|
||||||
static uint8_t ownMAC[6];
|
|
||||||
static int mesh_layer = -1;
|
|
||||||
static esp_netif_t *netif_sta = NULL;
|
|
||||||
|
|
||||||
struct ota_mesh_packet
|
|
||||||
{
|
|
||||||
enum ota_mesh_packet_type
|
|
||||||
{
|
|
||||||
APP_Version_Request,
|
|
||||||
APP_Version_Response,
|
|
||||||
OTA_Data,
|
|
||||||
OTA_ACK,
|
|
||||||
OTA_Complete
|
|
||||||
} type;
|
|
||||||
|
|
||||||
uint8_t au8Payload[1024];
|
|
||||||
};
|
|
||||||
|
|
||||||
/*******************************************************
|
|
||||||
* Function Declarations
|
|
||||||
*******************************************************/
|
|
||||||
|
|
||||||
esp_err_t esp_mesh_send_packet(mesh_addr_t* dest, struct ota_mesh_packet* packet);
|
|
||||||
|
|
||||||
/*******************************************************
|
|
||||||
* Function Definitions
|
|
||||||
*******************************************************/
|
|
||||||
|
|
||||||
esp_err_t esp_mesh_ota_send(mesh_addr_t* dest)
|
|
||||||
{
|
|
||||||
esp_err_t err = ESP_OK;
|
|
||||||
|
|
||||||
static uint32_t u32index;
|
|
||||||
|
|
||||||
const esp_partition_t * currentPartition = esp_ota_get_boot_partition();
|
|
||||||
|
|
||||||
if((*currentPartition).subtype == 0)
|
|
||||||
{
|
|
||||||
|
|
||||||
int data_read = 0;
|
|
||||||
|
|
||||||
struct ota_mesh_packet packet;
|
|
||||||
packet.type=OTA_Data;
|
|
||||||
|
|
||||||
if(u32index == 1024)
|
|
||||||
{
|
|
||||||
//all data read
|
|
||||||
data_read = 0;
|
|
||||||
u32index = 0;
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
ESP_LOGI(MESH_TAG, "OTA-Data read: %i", u32index);
|
|
||||||
err = esp_partition_read(currentPartition, (1024*u32index), packet.au8Payload, 1024 );
|
|
||||||
ESP_ERROR_CHECK(err);
|
|
||||||
data_read = 1024;
|
|
||||||
u32index++;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (data_read > 0)
|
|
||||||
{
|
|
||||||
//send ota fragemnt to node
|
|
||||||
esp_mesh_send_packet(dest, &packet);
|
|
||||||
}
|
|
||||||
|
|
||||||
ESP_ERROR_CHECK(err);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
ESP_LOGI(MESH_TAG, "Subtype: %d", (*currentPartition).subtype);
|
|
||||||
}
|
|
||||||
return err;
|
|
||||||
}
|
|
||||||
|
|
||||||
esp_err_t esp_mesh_ota_receive(mesh_addr_t* dest, struct ota_mesh_packet* packet)
|
|
||||||
{
|
|
||||||
esp_err_t err = ESP_OK;
|
|
||||||
static esp_ota_handle_t otaHandle;
|
|
||||||
static uint32_t u32index;
|
|
||||||
|
|
||||||
const esp_partition_t * currentPartition = esp_ota_get_boot_partition();
|
|
||||||
const esp_partition_t * otaPartition = esp_ota_get_next_update_partition(currentPartition);
|
|
||||||
|
|
||||||
if(u32index == 0)
|
|
||||||
{
|
|
||||||
//first run
|
|
||||||
|
|
||||||
err = esp_ota_begin(otaPartition, OTA_SIZE_UNKNOWN, &otaHandle);
|
|
||||||
ESP_ERROR_CHECK(err);
|
|
||||||
}
|
|
||||||
|
|
||||||
ESP_LOGI(MESH_TAG, "OTA-Data write: %i", u32index);
|
|
||||||
err = esp_ota_write(otaHandle, packet->au8Payload, 1024);
|
|
||||||
|
|
||||||
if(err != ESP_OK)
|
|
||||||
{
|
|
||||||
ESP_LOGE(MESH_TAG, "OTA-Data write error: %i at %i", err, u32index);
|
|
||||||
}
|
|
||||||
|
|
||||||
ESP_ERROR_CHECK(err);
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
if(u32index >= 1023)
|
|
||||||
{
|
|
||||||
//ota update complete
|
|
||||||
|
|
||||||
ESP_LOGI(MESH_TAG, "OTA-Data complete arrived: %i", u32index);
|
|
||||||
err = esp_ota_end(otaHandle);
|
|
||||||
ESP_ERROR_CHECK(err);
|
|
||||||
esp_app_desc_t otaPartitionDesc;
|
|
||||||
err = esp_ota_get_partition_description(otaPartition, &otaPartitionDesc);
|
|
||||||
ESP_ERROR_CHECK(err);
|
|
||||||
ESP_LOGI(MESH_TAG, "otaPartition project_name: %s", (otaPartitionDesc).project_name);
|
|
||||||
|
|
||||||
err = esp_ota_set_boot_partition(otaPartition);
|
|
||||||
ESP_ERROR_CHECK(err);
|
|
||||||
|
|
||||||
struct ota_mesh_packet retPacket;
|
|
||||||
retPacket.type=OTA_Complete;
|
|
||||||
ESP_ERROR_CHECK (esp_mesh_send_packet(dest, &retPacket)); //send back to parent
|
|
||||||
|
|
||||||
//check if this node has children --> Update them
|
|
||||||
|
|
||||||
esp_restart();
|
|
||||||
}
|
|
||||||
|
|
||||||
u32index++;
|
|
||||||
|
|
||||||
|
|
||||||
return err;
|
|
||||||
}
|
|
||||||
|
|
||||||
//returns true if MAC address is equal
|
|
||||||
bool esp_mesh_check_MAC_Equality(uint8_t* aMAC, uint8_t* bMAC)
|
|
||||||
{
|
|
||||||
for (uint8_t index = 0; index < 6; index++)
|
|
||||||
{
|
|
||||||
if(aMAC[index] != bMAC[index])
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
esp_err_t esp_mesh_get_Children(mesh_addr_t children[], uint16_t* pu16ChildrenSize)
|
|
||||||
{
|
|
||||||
*pu16ChildrenSize = 0;
|
|
||||||
mesh_addr_t route_table[CONFIG_MESH_ROUTE_TABLE_SIZE];
|
|
||||||
int route_table_size = 0;
|
|
||||||
esp_mesh_get_routing_table((mesh_addr_t *) &route_table, CONFIG_MESH_ROUTE_TABLE_SIZE * 6, &route_table_size);
|
|
||||||
for(uint16_t index = 0; index < esp_mesh_get_routing_table_size(); index++)
|
|
||||||
{
|
|
||||||
if(! (esp_mesh_check_MAC_Equality(ownMAC, route_table[index].addr)) )
|
|
||||||
{
|
|
||||||
//child node
|
|
||||||
// ESP_LOGI(MESH_TAG, "adding Node: \"0x%x, 0x%x, 0x%x, 0x%x, 0x%x, 0x%x\" ", route_table[index].addr[0], route_table[index].addr[1], route_table[index].addr[2], route_table[index].addr[3], route_table[index].addr[4], route_table[index].addr[5]);
|
|
||||||
children[*pu16ChildrenSize] = route_table[index];
|
|
||||||
*pu16ChildrenSize = (*pu16ChildrenSize)+1;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return ESP_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
esp_err_t esp_mesh_send_packet(mesh_addr_t* dest, struct ota_mesh_packet* packet)
|
|
||||||
{
|
|
||||||
esp_err_t err;
|
|
||||||
mesh_data_t data;
|
|
||||||
data.data = tx_buf;
|
|
||||||
data.size = sizeof(tx_buf);
|
|
||||||
data.proto = MESH_PROTO_BIN;
|
|
||||||
data.tos = MESH_TOS_P2P;
|
|
||||||
|
|
||||||
memcpy(tx_buf, (uint8_t *)packet, sizeof(struct ota_mesh_packet));
|
|
||||||
|
|
||||||
err = esp_mesh_send(dest, &data, MESH_DATA_P2P, NULL, 0);
|
|
||||||
|
|
||||||
return err;
|
|
||||||
}
|
|
||||||
|
|
||||||
void esp_mesh_p2p_rx_main(void *arg)
|
|
||||||
{
|
|
||||||
esp_err_t err;
|
|
||||||
mesh_addr_t from;
|
|
||||||
mesh_data_t data;
|
|
||||||
int flag = 0;
|
|
||||||
data.data = rx_buf;
|
|
||||||
data.size = RX_SIZE;
|
|
||||||
|
|
||||||
while (true)
|
|
||||||
{
|
|
||||||
data.size = RX_SIZE;
|
|
||||||
err = esp_mesh_recv(&from, &data, portMAX_DELAY, &flag, NULL, 0);
|
|
||||||
if (err != ESP_OK || !data.size)
|
|
||||||
{
|
|
||||||
ESP_LOGE(MESH_TAG, "err:0x%x, size:%d", err, data.size);
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
struct ota_mesh_packet packet;
|
|
||||||
memcpy(&packet, (uint8_t *)rx_buf, sizeof(struct ota_mesh_packet));
|
|
||||||
|
|
||||||
switch (packet.type)
|
|
||||||
{
|
|
||||||
case APP_Version_Request:
|
|
||||||
ESP_LOGI(MESH_TAG, "recv: APP_Version_Request");
|
|
||||||
packet.type=APP_Version_Response;
|
|
||||||
packet.au8Payload[0] = 42; //TODO get current running version
|
|
||||||
ESP_ERROR_CHECK (esp_mesh_send_packet(&from, &packet)); //send back to parent
|
|
||||||
break;
|
|
||||||
case APP_Version_Response:
|
|
||||||
ESP_LOGI(MESH_TAG, "recv: APP_Version_Response - App Version %i: ", packet.au8Payload[0]);
|
|
||||||
//check if node is out-dated
|
|
||||||
esp_mesh_ota_send(&from);
|
|
||||||
break;
|
|
||||||
case OTA_Data:
|
|
||||||
ESP_LOGI(MESH_TAG, "recv: OTA_Data");
|
|
||||||
esp_mesh_ota_receive(&from, &packet);
|
|
||||||
packet.type=OTA_ACK;
|
|
||||||
ESP_ERROR_CHECK (esp_mesh_send_packet(&from, &packet)); //send back to parent
|
|
||||||
break;
|
|
||||||
case OTA_ACK:
|
|
||||||
ESP_LOGI(MESH_TAG, "recv: OTA_ACK");
|
|
||||||
esp_mesh_ota_send(&from);
|
|
||||||
break;
|
|
||||||
case OTA_Complete:
|
|
||||||
ESP_LOGI(MESH_TAG, "recv: OTA_Complete");
|
|
||||||
break;
|
|
||||||
default:
|
|
||||||
ESP_LOGE(MESH_TAG, "recv: something");
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
} //end while
|
|
||||||
vTaskDelete(NULL);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
esp_err_t esp_mesh_comm_p2p_start(void)
|
|
||||||
{
|
|
||||||
static bool is_comm_p2p_started = false;
|
|
||||||
if (!is_comm_p2p_started)
|
|
||||||
{
|
|
||||||
is_comm_p2p_started = true;
|
|
||||||
xTaskCreate(esp_mesh_p2p_rx_main, "MPRX", 7000, NULL, 5, NULL);
|
|
||||||
}
|
|
||||||
return ESP_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
void mesh_event_handler(void *arg, esp_event_base_t event_base,
|
|
||||||
int32_t event_id, void *event_data)
|
|
||||||
{
|
|
||||||
mesh_addr_t id = {0,};
|
|
||||||
static uint16_t last_layer = 0;
|
|
||||||
|
|
||||||
switch (event_id)
|
|
||||||
{
|
|
||||||
case MESH_EVENT_STARTED:
|
|
||||||
{
|
|
||||||
esp_mesh_get_id(&id);
|
|
||||||
ESP_LOGI(MESH_TAG, "<MESH_EVENT_MESH_STARTED>ID:"MACSTR"", MAC2STR(id.addr));
|
|
||||||
is_mesh_connected = false;
|
|
||||||
mesh_layer = esp_mesh_get_layer();
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case MESH_EVENT_STOPPED: {
|
|
||||||
ESP_LOGI(MESH_TAG, "<MESH_EVENT_STOPPED>");
|
|
||||||
is_mesh_connected = false;
|
|
||||||
mesh_layer = esp_mesh_get_layer();
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case MESH_EVENT_CHILD_CONNECTED: {
|
|
||||||
mesh_event_child_connected_t *child_connected = (mesh_event_child_connected_t *)event_data;
|
|
||||||
ESP_LOGI(MESH_TAG, "<MESH_EVENT_CHILD_CONNECTED>aid:%d, "MACSTR"",
|
|
||||||
child_connected->aid,
|
|
||||||
MAC2STR(child_connected->mac));
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case MESH_EVENT_CHILD_DISCONNECTED: {
|
|
||||||
mesh_event_child_disconnected_t *child_disconnected = (mesh_event_child_disconnected_t *)event_data;
|
|
||||||
ESP_LOGI(MESH_TAG, "<MESH_EVENT_CHILD_DISCONNECTED>aid:%d, "MACSTR"",
|
|
||||||
child_disconnected->aid,
|
|
||||||
MAC2STR(child_disconnected->mac));
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case MESH_EVENT_ROUTING_TABLE_ADD: {
|
|
||||||
mesh_event_routing_table_change_t *routing_table = (mesh_event_routing_table_change_t *)event_data;
|
|
||||||
ESP_LOGW(MESH_TAG, "<MESH_EVENT_ROUTING_TABLE_ADD>add %d, new:%d, layer:%d",
|
|
||||||
routing_table->rt_size_change,
|
|
||||||
routing_table->rt_size_new, mesh_layer);
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case MESH_EVENT_ROUTING_TABLE_REMOVE: {
|
|
||||||
mesh_event_routing_table_change_t *routing_table = (mesh_event_routing_table_change_t *)event_data;
|
|
||||||
ESP_LOGW(MESH_TAG, "<MESH_EVENT_ROUTING_TABLE_REMOVE>remove %d, new:%d, layer:%d",
|
|
||||||
routing_table->rt_size_change,
|
|
||||||
routing_table->rt_size_new, mesh_layer);
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case MESH_EVENT_NO_PARENT_FOUND: {
|
|
||||||
mesh_event_no_parent_found_t *no_parent = (mesh_event_no_parent_found_t *)event_data;
|
|
||||||
ESP_LOGI(MESH_TAG, "<MESH_EVENT_NO_PARENT_FOUND>scan times:%d",
|
|
||||||
no_parent->scan_times);
|
|
||||||
}
|
|
||||||
/* TODO handler for the failure */
|
|
||||||
break;
|
|
||||||
case MESH_EVENT_PARENT_CONNECTED: {
|
|
||||||
mesh_event_connected_t *connected = (mesh_event_connected_t *)event_data;
|
|
||||||
esp_mesh_get_id(&id);
|
|
||||||
mesh_layer = connected->self_layer;
|
|
||||||
memcpy(&mesh_parent_addr.addr, connected->connected.bssid, 6);
|
|
||||||
ESP_LOGI(MESH_TAG,
|
|
||||||
"<MESH_EVENT_PARENT_CONNECTED>layer:%d-->%d, parent:"MACSTR"%s, ID:"MACSTR", duty:%d",
|
|
||||||
last_layer, mesh_layer, MAC2STR(mesh_parent_addr.addr),
|
|
||||||
esp_mesh_is_root() ? "<ROOT>" :
|
|
||||||
(mesh_layer == 2) ? "<layer2>" : "", MAC2STR(id.addr), connected->duty);
|
|
||||||
last_layer = mesh_layer;
|
|
||||||
// mesh_connected_indicator(mesh_layer);
|
|
||||||
is_mesh_connected = true;
|
|
||||||
if (esp_mesh_is_root()) {
|
|
||||||
esp_netif_dhcpc_start(netif_sta);
|
|
||||||
}
|
|
||||||
esp_mesh_comm_p2p_start();//start receiving
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case MESH_EVENT_PARENT_DISCONNECTED: {
|
|
||||||
mesh_event_disconnected_t *disconnected = (mesh_event_disconnected_t *)event_data;
|
|
||||||
ESP_LOGI(MESH_TAG,
|
|
||||||
"<MESH_EVENT_PARENT_DISCONNECTED>reason:%d",
|
|
||||||
disconnected->reason);
|
|
||||||
is_mesh_connected = false;
|
|
||||||
// mesh_disconnected_indicator();
|
|
||||||
mesh_layer = esp_mesh_get_layer();
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case MESH_EVENT_LAYER_CHANGE: {
|
|
||||||
mesh_event_layer_change_t *layer_change = (mesh_event_layer_change_t *)event_data;
|
|
||||||
mesh_layer = layer_change->new_layer;
|
|
||||||
ESP_LOGI(MESH_TAG, "<MESH_EVENT_LAYER_CHANGE>layer:%d-->%d%s",
|
|
||||||
last_layer, mesh_layer,
|
|
||||||
esp_mesh_is_root() ? "<ROOT>" :
|
|
||||||
(mesh_layer == 2) ? "<layer2>" : "");
|
|
||||||
last_layer = mesh_layer;
|
|
||||||
// mesh_connected_indicator(mesh_layer);
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case MESH_EVENT_ROOT_ADDRESS: {
|
|
||||||
mesh_event_root_address_t *root_addr = (mesh_event_root_address_t *)event_data;
|
|
||||||
ESP_LOGI(MESH_TAG, "<MESH_EVENT_ROOT_ADDRESS>root address:"MACSTR"",
|
|
||||||
MAC2STR(root_addr->addr));
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case MESH_EVENT_VOTE_STARTED: {
|
|
||||||
mesh_event_vote_started_t *vote_started = (mesh_event_vote_started_t *)event_data;
|
|
||||||
ESP_LOGI(MESH_TAG,
|
|
||||||
"<MESH_EVENT_VOTE_STARTED>attempts:%d, reason:%d, rc_addr:"MACSTR"",
|
|
||||||
vote_started->attempts,
|
|
||||||
vote_started->reason,
|
|
||||||
MAC2STR(vote_started->rc_addr.addr));
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case MESH_EVENT_VOTE_STOPPED: {
|
|
||||||
ESP_LOGI(MESH_TAG, "<MESH_EVENT_VOTE_STOPPED>");
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
case MESH_EVENT_ROOT_SWITCH_REQ: {
|
|
||||||
mesh_event_root_switch_req_t *switch_req = (mesh_event_root_switch_req_t *)event_data;
|
|
||||||
ESP_LOGI(MESH_TAG,
|
|
||||||
"<MESH_EVENT_ROOT_SWITCH_REQ>reason:%d, rc_addr:"MACSTR"",
|
|
||||||
switch_req->reason,
|
|
||||||
MAC2STR( switch_req->rc_addr.addr));
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case MESH_EVENT_ROOT_SWITCH_ACK: {
|
|
||||||
/* new root */
|
|
||||||
mesh_layer = esp_mesh_get_layer();
|
|
||||||
esp_mesh_get_parent_bssid(&mesh_parent_addr);
|
|
||||||
ESP_LOGI(MESH_TAG, "<MESH_EVENT_ROOT_SWITCH_ACK>layer:%d, parent:"MACSTR"", mesh_layer, MAC2STR(mesh_parent_addr.addr));
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case MESH_EVENT_TODS_STATE: {
|
|
||||||
mesh_event_toDS_state_t *toDs_state = (mesh_event_toDS_state_t *)event_data;
|
|
||||||
ESP_LOGI(MESH_TAG, "<MESH_EVENT_TODS_REACHABLE>state:%d", *toDs_state);
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case MESH_EVENT_ROOT_FIXED: {
|
|
||||||
mesh_event_root_fixed_t *root_fixed = (mesh_event_root_fixed_t *)event_data;
|
|
||||||
ESP_LOGI(MESH_TAG, "<MESH_EVENT_ROOT_FIXED>%s",
|
|
||||||
root_fixed->is_fixed ? "fixed" : "not fixed");
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case MESH_EVENT_ROOT_ASKED_YIELD: {
|
|
||||||
mesh_event_root_conflict_t *root_conflict = (mesh_event_root_conflict_t *)event_data;
|
|
||||||
ESP_LOGI(MESH_TAG,
|
|
||||||
"<MESH_EVENT_ROOT_ASKED_YIELD>"MACSTR", rssi:%d, capacity:%d",
|
|
||||||
MAC2STR(root_conflict->addr),
|
|
||||||
root_conflict->rssi,
|
|
||||||
root_conflict->capacity);
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case MESH_EVENT_CHANNEL_SWITCH: {
|
|
||||||
mesh_event_channel_switch_t *channel_switch = (mesh_event_channel_switch_t *)event_data;
|
|
||||||
ESP_LOGI(MESH_TAG, "<MESH_EVENT_CHANNEL_SWITCH>new channel:%d", channel_switch->channel);
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case MESH_EVENT_SCAN_DONE: {
|
|
||||||
mesh_event_scan_done_t *scan_done = (mesh_event_scan_done_t *)event_data;
|
|
||||||
ESP_LOGI(MESH_TAG, "<MESH_EVENT_SCAN_DONE>number:%d",
|
|
||||||
scan_done->number);
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case MESH_EVENT_NETWORK_STATE: {
|
|
||||||
mesh_event_network_state_t *network_state = (mesh_event_network_state_t *)event_data;
|
|
||||||
ESP_LOGI(MESH_TAG, "<MESH_EVENT_NETWORK_STATE>is_rootless:%d",
|
|
||||||
network_state->is_rootless);
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case MESH_EVENT_STOP_RECONNECTION: {
|
|
||||||
ESP_LOGI(MESH_TAG, "<MESH_EVENT_STOP_RECONNECTION>");
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case MESH_EVENT_FIND_NETWORK: {
|
|
||||||
mesh_event_find_network_t *find_network = (mesh_event_find_network_t *)event_data;
|
|
||||||
ESP_LOGI(MESH_TAG, "<MESH_EVENT_FIND_NETWORK>new channel:%d, router BSSID:"MACSTR"",
|
|
||||||
find_network->channel, MAC2STR(find_network->router_bssid));
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case MESH_EVENT_ROUTER_SWITCH: {
|
|
||||||
mesh_event_router_switch_t *router_switch = (mesh_event_router_switch_t *)event_data;
|
|
||||||
ESP_LOGI(MESH_TAG, "<MESH_EVENT_ROUTER_SWITCH>new router:%s, channel:%d, "MACSTR"",
|
|
||||||
router_switch->ssid, router_switch->channel, MAC2STR(router_switch->bssid));
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case MESH_EVENT_PS_PARENT_DUTY: {
|
|
||||||
mesh_event_ps_duty_t *ps_duty = (mesh_event_ps_duty_t *)event_data;
|
|
||||||
ESP_LOGI(MESH_TAG, "<MESH_EVENT_PS_PARENT_DUTY>duty:%d", ps_duty->duty);
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case MESH_EVENT_PS_CHILD_DUTY: {
|
|
||||||
mesh_event_ps_duty_t *ps_duty = (mesh_event_ps_duty_t *)event_data;
|
|
||||||
ESP_LOGI(MESH_TAG, "<MESH_EVENT_PS_CHILD_DUTY>cidx:%d, "MACSTR", duty:%d", ps_duty->child_connected.aid-1,
|
|
||||||
MAC2STR(ps_duty->child_connected.mac), ps_duty->duty);
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
default:
|
|
||||||
ESP_LOGI(MESH_TAG, "unknown id:%d", event_id);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void ip_event_handler(void *arg, esp_event_base_t event_base,
|
|
||||||
int32_t event_id, void *event_data)
|
|
||||||
{
|
|
||||||
ip_event_got_ip_t *event = (ip_event_got_ip_t *) event_data;
|
|
||||||
ESP_LOGI(MESH_TAG, "<IP_EVENT_STA_GOT_IP>IP:" IPSTR, IP2STR(&event->ip_info.ip));
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
static void test(void *pvParameters)
|
|
||||||
{
|
|
||||||
esp_err_t err;
|
|
||||||
uint32_t u32BufferLenght = 1024U;
|
|
||||||
char buffer[1024U];
|
|
||||||
uint32_t u32BytesRead = 0;
|
|
||||||
char pcRemoteVersionNumber[12];
|
|
||||||
const esp_partition_t * currentPartition;
|
|
||||||
const esp_partition_t * otaPartition;
|
|
||||||
static esp_ota_handle_t otaHandle;
|
|
||||||
uint32_t u32StartOffset;
|
|
||||||
esp_app_desc_t otaPartitionDesc;
|
|
||||||
|
|
||||||
ESP_LOGI(MESH_TAG, "Hello World");
|
|
||||||
currentPartition = esp_ota_get_boot_partition();
|
|
||||||
ESP_LOGI(MESH_TAG, "Type: %d", (*currentPartition).subtype);
|
|
||||||
ESP_LOGI(MESH_TAG, "Start address: %d", (*currentPartition).address);
|
|
||||||
ESP_LOGI(MESH_TAG, "Size: %d", (*currentPartition).size);
|
|
||||||
ESP_LOGI(MESH_TAG, "Encrypted: %d", (*currentPartition).encrypted);
|
|
||||||
|
|
||||||
esp_app_desc_t curPartitionDesc;
|
|
||||||
err = esp_ota_get_partition_description(currentPartition, &curPartitionDesc);
|
|
||||||
ESP_ERROR_CHECK(err);
|
|
||||||
ESP_LOGI(MESH_TAG, "currentPartition project_name: %s", (curPartitionDesc).project_name);
|
|
||||||
ESP_LOGI(MESH_TAG, "currentPartition version: %s", (curPartitionDesc).version);
|
|
||||||
ESP_LOGI(MESH_TAG, "currentPartition Timestamp: %s %s", (curPartitionDesc).date, (curPartitionDesc).time);
|
|
||||||
|
|
||||||
https_clientInitialize();
|
|
||||||
https_clientRetrieveData(buffer, &u32BufferLenght, &u32BytesRead);
|
|
||||||
ESP_LOGI(MESH_TAG, "Data received: %i", u32BytesRead);
|
|
||||||
err = errExtractVersionNumber(buffer, &u32BytesRead, pcRemoteVersionNumber);
|
|
||||||
|
|
||||||
if(err == ESP_OK)
|
|
||||||
{
|
|
||||||
if(bNewerVersion((curPartitionDesc).version, pcRemoteVersionNumber))
|
|
||||||
{
|
|
||||||
ESP_LOGI(MESH_TAG, "Newer Version available");
|
|
||||||
//write ota
|
|
||||||
otaPartition= esp_ota_get_next_update_partition(currentPartition);
|
|
||||||
|
|
||||||
err = errFindImageStart(buffer, &u32BufferLenght, &u32StartOffset);
|
|
||||||
|
|
||||||
ESP_LOGI(MESH_TAG, "first byte offset: %i", u32StartOffset);
|
|
||||||
ESP_LOGI(MESH_TAG, "first byte: %x", buffer[u32StartOffset]);
|
|
||||||
|
|
||||||
err = esp_ota_begin(otaPartition, OTA_SIZE_UNKNOWN, &otaHandle);
|
|
||||||
ESP_ERROR_CHECK(err);
|
|
||||||
|
|
||||||
do
|
|
||||||
{
|
|
||||||
ESP_LOGI(MESH_TAG, "OTA-Data written: %i", u32BytesRead);
|
|
||||||
err = esp_ota_write(otaHandle, (const void*) buffer+u32StartOffset, (u32BytesRead-u32StartOffset));
|
|
||||||
u32StartOffset = 0U;
|
|
||||||
https_clientRetrieveData(buffer, &u32BufferLenght, &u32BytesRead);
|
|
||||||
}
|
|
||||||
while (u32BytesRead > 0);
|
|
||||||
|
|
||||||
err = esp_ota_end(otaHandle);
|
|
||||||
ESP_ERROR_CHECK(err);
|
|
||||||
|
|
||||||
err = esp_ota_get_partition_description(otaPartition, &otaPartitionDesc);
|
|
||||||
ESP_ERROR_CHECK(err);
|
|
||||||
ESP_LOGI(MESH_TAG, "otaPartition project_name: %s", (otaPartitionDesc).project_name);
|
|
||||||
err = esp_ota_set_boot_partition(otaPartition);
|
|
||||||
ESP_ERROR_CHECK(err);
|
|
||||||
//esp_restart();
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
ESP_LOGI(MESH_TAG, "NO newer Version available");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
ESP_LOGI(MESH_TAG, "errExtractVersionNumber failed: %i", err);
|
|
||||||
}
|
|
||||||
|
|
||||||
https_clientDeinitialize();
|
|
||||||
|
|
||||||
ESP_LOGI(MESH_TAG, "\nend of task\n");
|
|
||||||
while(1)
|
|
||||||
{
|
|
||||||
vTaskDelay(1000 / portTICK_PERIOD_MS);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
void app_main(void)
|
void app_main(void)
|
||||||
{
|
{
|
||||||
ESP_LOGI(MESH_TAG, "hardcoded: 0.0.1");
|
esp_err_t err = ESP_OK;
|
||||||
/*
|
ESP_LOGI(LOG_TAG, "hardcoded: 0.0.1");
|
||||||
err = nvs_flash_erase();
|
err = errMeshNetworkInitialize();
|
||||||
if(err != ESP_OK){
|
ESP_ERROR_CHECK(err);
|
||||||
ESP_LOGI(MESH_TAG, "Error: %x", err);
|
|
||||||
while(1){
|
|
||||||
|
|
||||||
}
|
//start ota
|
||||||
}
|
|
||||||
*/
|
|
||||||
|
|
||||||
ESP_ERROR_CHECK(nvs_flash_init());
|
//start app
|
||||||
|
|
||||||
/* tcpip initialization */
|
|
||||||
ESP_ERROR_CHECK(esp_netif_init());
|
|
||||||
|
|
||||||
/* event initialization */
|
|
||||||
ESP_ERROR_CHECK(esp_event_loop_create_default());
|
|
||||||
|
|
||||||
/* create network interfaces for mesh (only station instance saved for further manipulation, soft AP instance ignored */
|
|
||||||
ESP_ERROR_CHECK(esp_netif_create_default_wifi_mesh_netifs(&netif_sta, NULL));
|
|
||||||
|
|
||||||
/* wifi initialization */
|
|
||||||
wifi_init_config_t config = WIFI_INIT_CONFIG_DEFAULT();
|
|
||||||
ESP_ERROR_CHECK(esp_wifi_init(&config));
|
|
||||||
ESP_ERROR_CHECK(esp_event_handler_register(IP_EVENT, IP_EVENT_STA_GOT_IP, &ip_event_handler, NULL));
|
|
||||||
ESP_ERROR_CHECK(esp_wifi_set_storage(WIFI_STORAGE_FLASH));
|
|
||||||
ESP_ERROR_CHECK(esp_wifi_start());
|
|
||||||
|
|
||||||
/* mesh initialization */
|
|
||||||
ESP_ERROR_CHECK(esp_mesh_init());
|
|
||||||
ESP_ERROR_CHECK(esp_event_handler_register(MESH_EVENT, ESP_EVENT_ANY_ID, &mesh_event_handler, NULL));
|
|
||||||
|
|
||||||
/* set mesh topology */
|
|
||||||
ESP_ERROR_CHECK(esp_mesh_set_topology(CONFIG_MESH_TOPOLOGY));
|
|
||||||
|
|
||||||
/* set mesh max layer according to the topology */
|
|
||||||
ESP_ERROR_CHECK(esp_mesh_set_max_layer(CONFIG_MESH_MAX_LAYER));
|
|
||||||
ESP_ERROR_CHECK(esp_mesh_set_vote_percentage(1));
|
|
||||||
ESP_ERROR_CHECK(esp_mesh_set_xon_qsize(128));
|
|
||||||
|
|
||||||
/* Disable mesh PS function */
|
|
||||||
ESP_ERROR_CHECK(esp_mesh_disable_ps());
|
|
||||||
ESP_ERROR_CHECK(esp_mesh_set_ap_assoc_expire(10));
|
|
||||||
|
|
||||||
mesh_cfg_t cfg = MESH_INIT_CONFIG_DEFAULT();
|
|
||||||
|
|
||||||
/* mesh ID */
|
|
||||||
memcpy((uint8_t *) &cfg.mesh_id, MESH_ID, 6);
|
|
||||||
|
|
||||||
/* router */
|
|
||||||
cfg.channel = CONFIG_MESH_CHANNEL;
|
|
||||||
cfg.router.ssid_len = strlen(CONFIG_MESH_ROUTER_SSID);
|
|
||||||
memcpy((uint8_t *) &cfg.router.ssid, CONFIG_MESH_ROUTER_SSID, cfg.router.ssid_len);
|
|
||||||
memcpy((uint8_t *) &cfg.router.password, CONFIG_MESH_ROUTER_PASSWD,
|
|
||||||
strlen(CONFIG_MESH_ROUTER_PASSWD));
|
|
||||||
|
|
||||||
/* mesh softAP */
|
|
||||||
ESP_ERROR_CHECK(esp_mesh_set_ap_authmode(CONFIG_MESH_AP_AUTHMODE));
|
|
||||||
cfg.mesh_ap.max_connection = CONFIG_MESH_AP_CONNECTIONS;
|
|
||||||
memcpy((uint8_t *) &cfg.mesh_ap.password, CONFIG_MESH_AP_PASSWD,
|
|
||||||
strlen(CONFIG_MESH_AP_PASSWD));
|
|
||||||
ESP_ERROR_CHECK(esp_mesh_set_config(&cfg));
|
|
||||||
|
|
||||||
ESP_ERROR_CHECK(esp_base_mac_addr_get(ownMAC));
|
|
||||||
|
|
||||||
ESP_LOGI(MESH_TAG, "Own MAC: \"0x%x, 0x%x, 0x%x, 0x%x, 0x%x, 0x%x\" ", ownMAC[0], ownMAC[1], ownMAC[2], ownMAC[3], ownMAC[4], ownMAC[5]);
|
|
||||||
|
|
||||||
/* mesh start */
|
|
||||||
ESP_ERROR_CHECK(esp_mesh_start());
|
|
||||||
|
|
||||||
ESP_LOGI(MESH_TAG, "mesh starts successfully, heap:%d, %s<%d>%s, ps:%d\n", esp_get_minimum_free_heap_size(),
|
|
||||||
esp_mesh_is_root_fixed() ? "root fixed" : "root not fixed",
|
|
||||||
esp_mesh_get_topology(), esp_mesh_get_topology() ? "(chain)":"(tree)", esp_mesh_is_ps_enabled());
|
|
||||||
|
|
||||||
for(;;)
|
|
||||||
{
|
|
||||||
if(gpio_get_level(0) == 0){
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
vTaskDelay(1000 / portTICK_PERIOD_MS);
|
|
||||||
}
|
|
||||||
xTaskCreate(&test, "test_task", 8192, NULL, 5, NULL);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue