Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- /* Mesh Internal Transceiver Example
- This example code is in the Public Domain (or CC0 licensed, at your option.)
- Unless required by applicable law or agreed to in writing, this
- software is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
- CONDITIONS OF ANY KIND, either express or implied.
- */
- #include <string.h>
- #include "esp_wifi.h"
- #include "esp_system.h"
- #include "esp_event_loop.h"
- #include "esp_log.h"
- #include "esp_mesh.h"
- #include "esp_mesh_internal.h"
- #include "mesh_light.h"
- #include "nvs_flash.h"
- #include "mdf_wifi_mesh.h"
- /*******************************************************
- * Macros
- *******************************************************/
- //#define MESH_P2P_TOS_OFF
- //#define MESH_ROOT_TO_GROUP
- /*******************************************************
- * Constants
- *******************************************************/
- #define RX_SIZE (1500)
- #define TX_SIZE (1460)
- /*******************************************************
- * Variable Definitions
- *******************************************************/
- 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 slave_buf[TX_SIZE] = { 0x3E,0x3E, 0x48,0x65,0x6c,0x6c,0x6f,0x5f,0x4d,0x65,0x73,0x68,0x21, 0x00}; // 0x480x650x6c0x6c0x6f0x5f0x4d0x650x730x680x21
- static uint8_t rx_buf[RX_SIZE] = { 0, };
- static bool is_running = true;
- static bool is_mesh_connected = false;
- static mesh_addr_t mesh_parent_addr;
- static int mesh_layer = -1;
- static bool gotIP = false;
- mesh_light_ctl_t light_on = {
- .cmd = MESH_CONTROL_CMD,
- .on = 1,
- .token_id = MESH_TOKEN_ID,
- .token_value = MESH_TOKEN_VALUE,
- };
- mesh_light_ctl_t light_off = {
- .cmd = MESH_CONTROL_CMD,
- .on = 0,
- .token_id = MESH_TOKEN_ID,
- .token_value = MESH_TOKEN_VALUE,
- };
- /*******************************************************
- * Function Declarations
- *******************************************************/
- /*******************************************************
- * Function Definitions
- *******************************************************/
- void mesh_send_to_group(const mesh_data_t *data, const mesh_addr_t *group, int seqno)
- {
- esp_err_t err = esp_mesh_send(group, data,
- MESH_DATA_P2P | MESH_DATA_GROUP,
- NULL, 0);
- if (err) {
- ESP_LOGE(MESH_TAG,
- "[GROUP:%d][L:%d][rtableSize:%d]parent:"MACSTR" to "MACSTR", heap:%d[err:0x%x, proto:%d, tos:%d]",
- seqno, mesh_layer, esp_mesh_get_routing_table_size(),
- MAC2STR(mesh_parent_addr.addr), MAC2STR(group->addr),
- esp_get_free_heap_size(), err, data->proto, data->tos);
- } else {
- ESP_LOGW(MESH_TAG,
- "[GROUP:%d][L:%d][rtableSize:%d]parent:"MACSTR" to "MACSTR", heap:%d[err:0x%x, proto:%d, tos:%d]",
- seqno, mesh_layer, esp_mesh_get_routing_table_size(),
- MAC2STR(mesh_parent_addr.addr), MAC2STR(group->addr),
- esp_get_free_heap_size(), err, data->proto, data->tos);
- }
- }
- void send2Server(mesh_addr_t* s, mesh_data_t *d){
- if(!gotIP) return;
- ESP_LOGE(MESH_TAG, "Send to server");
- esp_mesh_send(s, d, MESH_DATA_TODS, NULL , 0);
- }
- void esp_mesh_p2p_tx_main(void *arg)
- {
- int i;
- esp_err_t err;
- int send_count = 0;
- mesh_addr_t route_table[CONFIG_MESH_ROUTE_TABLE_SIZE];
- int route_table_size = 0;
- mesh_data_t data;
- data.data = tx_buf;
- data.size = sizeof(tx_buf);
- data.proto = MESH_PROTO_BIN;
- mesh_addr_t serverAddr;
- IP4_ADDR(&serverAddr.mip.ip4, 192,168,0,132);
- serverAddr.mip.port = 8089;
- #ifdef MESH_P2P_TOS_OFF
- data.tos = MESH_TOS_DEF;
- #endif /* MESH_P2P_TOS_OFF */
- is_running = true;
- while (is_running) {
- /* non-root do nothing but print */
- send2Server(&serverAddr, &data);
- ESP_LOGE(MESH_TAG, "Routing table");
- for (i = 0; i < route_table_size; i++) {
- ESP_LOGE(MESH_TAG,MACSTR,MAC2STR(route_table[i].addr));
- }
- if (!esp_mesh_is_root()) {
- ESP_LOGI(MESH_TAG, "layer:%d, rtableSize:%d, %s", mesh_layer,
- esp_mesh_get_routing_table_size(),
- (is_mesh_connected && esp_mesh_is_root()) ? "ROOT" : is_mesh_connected ? "NODE" : "DISCONNECT");
- vTaskDelay(10 * 1000 / portTICK_RATE_MS);
- mesh_addr_t parent;
- err = esp_mesh_get_parent_bssid(&parent);
- if(err) continue;
- ESP_LOGI(MESH_TAG, "got parent");
- // warning. begin of copypaste
- /*static int packet_id = 0;
- esp_err_t ret = ESP_OK;
- int mesh_flag = MESH_DATA_TODS;
- wifi_mesh_data_type_t data_type = {
- .proto = MESH_PROTO_HTTP,
- };
- mesh_data_t mesh_data = {
- .proto = MESH_PROTO_HTTP,
- .tos = MESH_TOS_DEF,
- .data = slave_buf,
- .size = sizeof(slave_buf),
- };
- wifi_mesh_head_data_t mesh_head_data = {
- .size = sizeof(slave_buf),
- .seq = 0,
- .id = packet_id++,
- };
- mesh_opt_t mesh_opt = {
- .len = sizeof(wifi_mesh_head_data_t),
- .val = (void *) &mesh_head_data,
- .type = MESH_OPT_RECV_DS_ADDR,
- };
- ret = esp_mesh_send(&serverAddr, &mesh_data, mesh_flag, &mesh_opt, 1);*/
- // end of copypaste
- continue;
- }
- esp_mesh_get_routing_table((mesh_addr_t *) &route_table,
- CONFIG_MESH_ROUTE_TABLE_SIZE * 6, &route_table_size);
- if (send_count && !(send_count % 100)) {
- ESP_LOGI(MESH_TAG, "size:%d/%d,send_count:%d", route_table_size,
- esp_mesh_get_routing_table_size(), send_count);
- }
- send_count++;
- tx_buf[25] = (send_count >> 24) & 0xff;
- tx_buf[24] = (send_count >> 16) & 0xff;
- tx_buf[23] = (send_count >> 8) & 0xff;
- tx_buf[22] = (send_count >> 0) & 0xff;
- if (send_count % 2) {
- memcpy(tx_buf, (uint8_t *)&light_on, sizeof(light_on));
- } else {
- memcpy(tx_buf, (uint8_t *)&light_off, sizeof(light_off));
- }
- #ifdef MESH_ROOT_TO_GROUP
- mesh_send_to_group(&data, (mesh_addr_t *)MESH_GROUP_ID, send_count);
- vTaskDelay(1 * 1000 / portTICK_RATE_MS);
- continue;
- #endif
- for (i = 0; i < route_table_size; i++) {
- err = esp_mesh_send(&route_table[i], &data, MESH_DATA_P2P, NULL, 0);
- if (err) {
- ESP_LOGE(MESH_TAG,
- "[ROOT-2-UNICAST:%d][L:%d]parent:"MACSTR" to "MACSTR", heap:%d[err:0x%x, proto:%d, tos:%d]",
- send_count, mesh_layer, MAC2STR(mesh_parent_addr.addr),
- MAC2STR(route_table[i].addr), esp_get_free_heap_size(),
- err, data.proto, data.tos);
- } else if (!(send_count % 100)) {
- ESP_LOGW(MESH_TAG,
- "[ROOT-2-UNICAST:%d][L:%d][rtableSize:%d]parent:"MACSTR" to "MACSTR", heap:%d[err:0x%x, proto:%d, tos:%d]",
- send_count, mesh_layer,
- esp_mesh_get_routing_table_size(),
- MAC2STR(mesh_parent_addr.addr),
- MAC2STR(route_table[i].addr), esp_get_free_heap_size(),
- err, data.proto, data.tos);
- }
- }
- /* if route_table_size is less than 10, add delay to avoid watchdog in this task. */
- if (route_table_size < 10) {
- vTaskDelay(1 * 1000 / portTICK_RATE_MS);
- }
- }
- vTaskDelete(NULL);
- }
- void esp_mesh_p2p_rx_main(void *arg)
- {
- int recv_count = 0;
- esp_err_t err;
- mesh_addr_t from;
- int send_count = 0;
- mesh_data_t data;
- int flag = 0;
- data.data = rx_buf;
- data.size = RX_SIZE;
- is_running = true;
- while (is_running) {
- 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;
- }
- /* extract send count */
- if (data.size >= sizeof(send_count)) {
- send_count = (data.data[25] << 24) | (data.data[24] << 16)
- | (data.data[23] << 8) | data.data[22];
- }
- recv_count++;
- /* process light control */
- if(data.data[0] == 62){
- ESP_LOGW(MESH_TAG, "GOT MESSAGE FROM NODE:%s", data.data);
- }
- mesh_light_process(&from, data.data, data.size);
- if (!(recv_count % 1)) {
- ESP_LOGW(MESH_TAG,
- "[#RX:%d/%d][L:%d] parent:"MACSTR", receive from "MACSTR", size:%d, heap:%d, flag:%d[err:0x%x, proto:%d, tos:%d]",
- recv_count, send_count, mesh_layer,
- MAC2STR(mesh_parent_addr.addr), MAC2STR(from.addr),
- data.size, esp_get_free_heap_size(), flag, err, data.proto,
- data.tos);
- }
- }
- 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_tx_main, "MPTX", 3072, NULL, 5, NULL);
- xTaskCreate(esp_mesh_p2p_rx_main, "MPRX", 3072, NULL, 5, NULL);
- }
- return ESP_OK;
- }
- void esp_mesh_event_handler(mesh_event_t event)
- {
- #ifdef MESH_ROOT_TO_GROUP
- mesh_addr_t group;
- #endif
- static uint8_t last_layer = 0;
- ESP_LOGD(MESH_TAG, "esp_event_handler:%d", event.id);
- switch (event.id) {
- case MESH_EVENT_STARTED:
- ESP_LOGI(MESH_TAG, "<MESH_EVENT_STARTED>heap:%d", esp_get_free_heap_size());
- is_mesh_connected = false;
- mesh_layer = esp_mesh_get_layer();
- break;
- case MESH_EVENT_STOPPED:
- ESP_LOGI(MESH_TAG, "<MESH_EVENT_STOPPED>heap:%d", esp_get_free_heap_size());
- is_mesh_connected = false;
- mesh_layer = esp_mesh_get_layer();
- break;
- case MESH_EVENT_CHILD_CONNECTED:
- ESP_LOGI(MESH_TAG, "<MESH_EVENT_CHILD_CONNECTED>aid:%d, "MACSTR"",
- event.info.child_connected.aid,
- MAC2STR(event.info.child_connected.mac));
- break;
- case MESH_EVENT_CHILD_DISCONNECTED:
- ESP_LOGI(MESH_TAG, "<MESH_EVENT_CHILD_DISCONNECTED>aid:%d, "MACSTR"",
- event.info.child_disconnected.aid,
- MAC2STR(event.info.child_disconnected.mac));
- break;
- case MESH_EVENT_ROUTING_TABLE_ADD:
- ESP_LOGW(MESH_TAG, "<MESH_EVENT_ROUTING_TABLE_ADD>add %d, new:%d",
- event.info.routing_table.rt_size_change,
- event.info.routing_table.rt_size_new);
- break;
- case MESH_EVENT_ROUTING_TABLE_REMOVE:
- ESP_LOGW(MESH_TAG, "<MESH_EVENT_ROUTING_TABLE_REMOVE>remove %d, new:%d",
- event.info.routing_table.rt_size_change,
- event.info.routing_table.rt_size_new);
- break;
- case MESH_EVENT_NO_PARNET_FOUND:
- ESP_LOGI(MESH_TAG, "<MESH_EVENT_NO_PARNET_FOUND>scan times:%d",
- event.info.no_parent.scan_times);
- /* TODO handler for the failure */
- break;
- case MESH_EVENT_PARENT_CONNECTED:
- mesh_layer = event.info.connected.self_layer;
- memcpy(&mesh_parent_addr.addr, event.info.connected.connected.bssid, 6);
- ESP_LOGI(MESH_TAG,
- "<MESH_EVENT_PARENT_CONNECTED>layer:%d-->%d, parent:"MACSTR"%s",
- last_layer, mesh_layer, MAC2STR(mesh_parent_addr.addr),
- esp_mesh_is_root() ? "<ROOT>" :
- (mesh_layer == 2) ? "<layer2>" : "");
- last_layer = mesh_layer;
- mesh_connected_indicator(mesh_layer);
- is_mesh_connected = true;
- if (esp_mesh_is_root()) {
- tcpip_adapter_dhcpc_start(TCPIP_ADAPTER_IF_STA);
- }
- #ifdef MESH_ROOT_TO_GROUP
- if (mesh_layer == 3) {
- ESP_ERROR_CHECK(
- esp_mesh_set_group_id((mesh_addr_t * ) MESH_GROUP_ID, 1))
- ESP_ERROR_CHECK(esp_mesh_get_group_list(&group, 1))
- ESP_LOGI(MESH_TAG, "num:%d, group "MACSTR"\n",
- esp_mesh_get_group_num(), MAC2STR(group.addr));
- }
- #endif
- esp_mesh_comm_p2p_start();
- break;
- case MESH_EVENT_PARENT_DISCONNECTED:
- ESP_LOGI(MESH_TAG,
- "<MESH_EVENT_PARENT_DISCONNECTED>reason:%d",
- event.info.disconnected.reason);
- is_mesh_connected = false;
- mesh_disconnected_indicator();
- break;
- case MESH_EVENT_LAYER_CHANGE:
- mesh_layer = event.info.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:
- ESP_LOGI(MESH_TAG, "<MESH_EVENT_ROOT_ADDRESS>root address:"MACSTR"",
- MAC2STR(event.info.root_addr.addr));
- break;
- case MESH_EVENT_ROOT_GOT_IP:
- /* root starts to connect to server */
- ESP_LOGI(MESH_TAG,
- "<MESH_EVENT_ROOT_GOT_IP>sta ip: " IPSTR ", mask: " IPSTR ", gw: " IPSTR,
- IP2STR(&event.info.got_ip.ip_info.ip),
- IP2STR(&event.info.got_ip.ip_info.netmask),
- IP2STR(&event.info.got_ip.ip_info.gw));
- gotIP = true;
- break;
- case MESH_EVENT_ROOT_LOST_IP:
- ESP_LOGI(MESH_TAG, "<MESH_EVENT_ROOT_LOST_IP>");
- break;
- case MESH_EVENT_VOTE_STARTED:
- ESP_LOGI(MESH_TAG,
- "<MESH_EVENT_VOTE_STARTED>attempts:%d, reason:%d, rc_addr:"MACSTR"",
- event.info.vote_started.attempts,
- event.info.vote_started.reason,
- MAC2STR(event.info.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:
- ESP_LOGI(MESH_TAG,
- "<MESH_EVENT_ROOT_SWITCH_REQ>reason:%d, rc_addr:"MACSTR"",
- event.info.switch_req.reason,
- MAC2STR( event.info.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:
- ESP_LOGI(MESH_TAG, "<MESH_EVENT_TODS_REACHABLE>state:%d",
- event.info.toDS_state);
- break;
- case MESH_EVENT_ROOT_FIXED:
- ESP_LOGI(MESH_TAG, "<MESH_EVENT_ROOT_FIXED>%s",
- event.info.root_fixed.is_fixed ? "fixed" : "not fixed");
- break;
- default:
- ESP_LOGI(MESH_TAG, "unknown id:%d", event.id);
- break;
- }
- }
- void app_main(void)
- {
- ESP_ERROR_CHECK(mesh_light_init());
- ESP_ERROR_CHECK(nvs_flash_init());
- /* tcpip initialization */
- tcpip_adapter_init();
- /* for mesh
- * stop DHCP server on softAP interface by default
- * stop DHCP client on station interface by default
- * */
- ESP_ERROR_CHECK(tcpip_adapter_dhcps_stop(TCPIP_ADAPTER_IF_AP));
- ESP_ERROR_CHECK(tcpip_adapter_dhcpc_stop(TCPIP_ADAPTER_IF_STA));
- #if 0
- /* static ip settings */
- tcpip_adapter_ip_info_t sta_ip;
- sta_ip.ip.addr = ipaddr_addr("192.168.1.102");
- sta_ip.gw.addr = ipaddr_addr("192.168.1.1");
- sta_ip.netmask.addr = ipaddr_addr("255.255.255.0");
- tcpip_adapter_set_ip_info(WIFI_IF_STA, &sta_ip);
- #endif
- /* wifi initialization */
- ESP_ERROR_CHECK(esp_event_loop_init(NULL, NULL));
- wifi_init_config_t config = WIFI_INIT_CONFIG_DEFAULT();
- ESP_ERROR_CHECK(esp_wifi_init(&config));
- 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_mesh_set_max_layer(CONFIG_MESH_MAX_LAYER));
- ESP_ERROR_CHECK(esp_mesh_set_ap_authmode(CONFIG_MESH_AP_AUTHMODE));
- ESP_ERROR_CHECK(esp_mesh_set_vote_percentage(1));
- ESP_ERROR_CHECK(esp_mesh_set_ap_assoc_expire(10));
- #ifdef MESH_FIX_ROOT
- ESP_ERROR_CHECK(esp_mesh_fix_root(1));
- #endif
- mesh_cfg_t cfg = MESH_INIT_CONFIG_DEFAULT();
- /* mesh ID */
- memcpy((uint8_t *) &cfg.mesh_id, MESH_ID, 6);
- /* mesh event callback */
- cfg.event_cb = &esp_mesh_event_handler;
- /* 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 */
- 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));
- /* set RSSI threshold for connecting to the root */
- mesh_switch_parent_t switch_paras ;
- ESP_ERROR_CHECK(esp_mesh_get_switch_parent_paras(&switch_paras));
- switch_paras.backoff_rssi = -45;
- ESP_ERROR_CHECK(esp_mesh_set_switch_parent_paras(&switch_paras));
- #ifdef MESH_SET_PARENT
- /* parent */
- wifi_config_t parent = {
- .sta = {
- .ssid = CONFIG_MESH_PARENT_SSID,
- .channel = CONFIG_MESH_CHANNEL,
- },
- };
- ESP_ERROR_CHECK(esp_mesh_set_parent(&parent, MESH_ROOT, 1));
- #endif
- /* mesh start */
- ESP_ERROR_CHECK(esp_mesh_start());
- ESP_LOGI(MESH_TAG, "mesh starts successfully, heap:%d, %s\n", esp_get_free_heap_size(),
- esp_mesh_is_root_fixed() ? "root fixed" : "root not fixed");
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement