void ICACHE_FLASH_ATTR
mesh_device_set_root(struct mesh_device_mac_type *root)
{
if (!g_mesh_device_init)
mesh_device_list_init();
/*
* the first time to set root
*/
if (g_node_list.scale == 0) {
MESH_DEMO_PRINT("new root:" MACSTR "\n", MAC2STR((uint8_t *)root));
MESH_DEV_MEMCPY(&g_node_list.root, root, sizeof(*root));
g_node_list.scale = 1;
return;
}
/*
* root device is the same to the current node,
* we don't need to modify anything
*/
if (!MESH_DEV_MEMCMP(&g_node_list.root, root, sizeof(*root)))
return;
/*
* switch root device, so the mac address list is stale
* we need to free the stale the mac list
*/
MESH_DEMO_PRINT("switch root:" MACSTR "to root:" MACSTR "\n",
MAC2STR((uint8_t *)&g_node_list.root), MAC2STR((uint8_t *)root));
mesh_device_list_release();
MESH_DEV_MEMCPY(&g_node_list.root, root, sizeof(*root));
g_node_list.scale = 1;
}
void ICACHE_FLASH_ATTR
mesh_device_disp_mac_list()
{
uint16_t idx = 0;
if (g_node_list.scale < 1)
return;
MESH_DEMO_PRINT("=====mac list info=====\n");
MESH_DEMO_PRINT("root: " MACSTR "\n", MAC2STR(g_node_list.root.mac));
if (g_node_list.scale < 2)
return;
for (idx = 0; idx < g_node_list.scale - 1; idx ++)
MESH_DEMO_PRINT("idx:%d, " MACSTR "\n", idx, MAC2STR(g_node_list.list[idx].mac));
MESH_DEMO_PRINT("=====mac list end======\n");
}
void ICACHE_FLASH_ATTR esp_mesh_demo_con_cb(void *arg)
{
static os_timer_t tst_timer;
struct espconn *server = (struct espconn *)arg;
MESH_DEMO_PRINT("esp_mesh_demo_con_cb\n");
if (server != &g_ser_conn) {
MESH_DEMO_PRINT("con_cb, para err\n");
return;
}
os_timer_disarm(&tst_timer);
os_timer_setfn(&tst_timer, (os_timer_func_t *)esp_mesh_demo_test, NULL);
os_timer_arm(&tst_timer, 7000, true);
}
bool ICACHE_FLASH_ATTR
mesh_device_get_root(const struct mesh_device_mac_type **root)
{
if (!g_mesh_device_init) {
MESH_DEMO_PRINT("please init mesh device list firstly\n");
return false;
}
if (g_node_list.scale == 0) {
MESH_DEMO_PRINT("no mac in current mac list\n");
return false;
}
if (!root)
return false;
*root = &g_node_list.root;
return true;
}
bool ICACHE_FLASH_ATTR
mesh_device_add(struct mesh_device_mac_type *nodes, uint16_t count)
{
#define MESH_DEV_STEP (10)
uint16_t idx = 0;
uint16_t rest = g_node_list.size + 1 - g_node_list.scale;
if (!g_mesh_device_init)
mesh_device_list_init();
if (g_node_list.size == 0)
rest = 0;
if (rest < count) {
/*
* current list is limited
* we need to re-allocate buffer for mac list
*/
uint16_t size = g_node_list.size + rest + MESH_DEV_STEP;
uint8_t *buf = (uint8_t *)MESH_DEV_ZALLOC(size * sizeof(*nodes));
if (!buf) {
MESH_DEMO_PRINT("mesh add alloc buf fail\n");
return false;
}
/*
* move the current list to new buffer
*/
if (g_node_list.list && g_node_list.scale > 1)
MESH_DEV_MEMCPY(buf, g_node_list.list,
(g_node_list.scale - 1) * sizeof(*nodes));
if (g_node_list.list)
MESH_DEV_FREE(g_node_list.list);
g_node_list.list = (struct mesh_device_mac_type *)buf;
g_node_list.size = size;
}
while (idx < count) {
if (!mesh_search_device(nodes + idx)) { // not in list, add it into list
MESH_DEV_MEMCPY(g_node_list.list + g_node_list.scale - 1,
nodes + idx, sizeof(*nodes));
g_node_list.scale ++;
}
idx ++;
}
return true;
}
bool ICACHE_FLASH_ATTR
mesh_device_get_mac_list(const struct mesh_device_mac_type **list,
uint16_t *count)
{
if (!g_mesh_device_init) {
MESH_DEMO_PRINT("please init mesh device list firstly\n");
return false;
}
if (!list || !count)
return false;
if (g_node_list.scale < 2) {
*list = NULL;
*count = 0;
return true;
}
*list = g_node_list.list;
*count = g_node_list.scale - 1;
}
void ICACHE_FLASH_ATTR esp_recv_entrance(void *arg, char *pdata, uint16_t len)
{
uint8_t *src = NULL, *dst = NULL;
uint8_t *resp = "{\"rsp_key\":\"rsp_key_value\"}";
struct mesh_header_format *header = (struct mesh_header_format *)pdata;
MESH_DEMO_PRINT("recv entrance\n");
if (!pdata)
return;
/*
* process packet
* call packet_parser(...)
* general packet_parser demo
*/
mesh_packet_parser(arg, pdata, len);
/*
* then build response packet,
* and give resonse to controller
* the following code is response demo
*/
#if 0
if (!espconn_mesh_get_src_addr(header, &src) || !espconn_mesh_get_dst_addr(header, &dst)) {
MESH_DEMO_PRINT("get addr fail\n");
return;
}
header = (struct mesh_header_format *)espconn_mesh_create_packet(
src, // destiny address
dst, // source address
false, // not p2p packet
true, // piggyback congest request
M_PROTO_JSON, // packe with JSON format
MESH_DEMO_STRLEN(resp), // data length
false, // no option
0, // option len
false, // no frag
0, // frag type, this packet doesn't use frag
false, // more frag
0, // frag index
0); // frag length
if (!header) {
MESH_DEMO_PRINT("create packet fail\n");
return;
}
if (!espconn_mesh_set_usr_data(header, resp, MESH_DEMO_STRLEN(resp))) {
MESH_DEMO_PRINT("set user data fail\n");
MESH_DEMO_FREE(header);
return;
}
if (espconn_mesh_sent(&g_ser_conn, (uint8_t *)header, header->len)) {
MESH_DEMO_PRINT("mesh is busy\n");
MESH_DEMO_FREE(header);
return;
}
MESH_DEMO_FREE(header);
#endif
}
bool ICACHE_FLASH_ATTR esp_mesh_demo_init()
{
struct station_config config;
// print version of mesh
espconn_mesh_print_ver();
/*
* set the AP password of mesh node
*/
if (!espconn_mesh_encrypt_init(MESH_AUTH, MESH_PASSWD, MESH_DEMO_STRLEN(MESH_PASSWD))) {
MESH_DEMO_PRINT("set pw fail\n");
return false;
}
/*
* if you want set max_hop > 4
* please make the heap is avaliable
* mac_route_table_size = (4^max_hop - 1)/3 * 6
*/
if (!espconn_mesh_set_max_hops(MESH_MAX_HOP)) {
MESH_DEMO_PRINT("fail, max_hop:%d\n", espconn_mesh_get_max_hops());
return false;
}
/*
* mesh_ssid_prefix
* mesh_group_id and mesh_ssid_prefix represent mesh network
*/
if (!espconn_mesh_set_ssid_prefix(MESH_SSID_PREFIX, MESH_DEMO_STRLEN(MESH_SSID_PREFIX))) {
MESH_DEMO_PRINT("set prefix fail\n");
return false;
}
/*
* mesh_group_id
* mesh_group_id and mesh_ssid_prefix represent mesh network
*/
if (!espconn_mesh_group_id_init((uint8_t *)MESH_GROUP_ID, sizeof(MESH_GROUP_ID))) {
MESH_DEMO_PRINT("set grp id fail\n");
return false;
}
if (!espconn_mesh_server_init((struct ip_addr *)server_ip, server_port)) {
MESH_DEMO_PRINT("server_init fail\n");
return false;
}
/*
* please change MESH_ROUTER_SSID and MESH_ROUTER_PASSWD according to your router
*/
MESH_DEMO_MEMSET(&config, 0, sizeof(config));
MESH_DEMO_MEMCPY(config.ssid, MESH_ROUTER_SSID, MESH_DEMO_STRLEN(MESH_ROUTER_SSID));
MESH_DEMO_MEMCPY(config.password, MESH_ROUTER_PASSWD, MESH_DEMO_STRLEN(MESH_ROUTER_PASSWD));
/*
* if you use router with hide ssid, you MUST set bssid in config,
* otherwise, node will fail to connect router.
*
* if you use normal router, please pay no attention to the bssid,
* and you don't need to modify the bssid, mesh will ignore the bssid.
*/
config.bssid_set = 1;
MESH_DEMO_MEMCPY(config.bssid, MESH_ROUTER_BSSID, sizeof(config.bssid));
/*
* you can use esp-touch(smart configure) to sent information about router AP to mesh node
* if you donn't use esp-touch, you should use espconn_mesh_set_router to set router for mesh node
*/
if (!espconn_mesh_set_router(&config)) {
MESH_DEMO_PRINT("set_router fail\n");
return false;
}
return true;
}
void ICACHE_FLASH_ATTR esp_mesh_demo_test()
{
uint8_t src[6];
uint8_t dst[6];
struct mesh_header_format *header = NULL;
/*
* this is ucast test case
*/
/*
* the mesh data can be any content
* it can be string(json/http), or binary(MQTT).
*/
char *tst_data = "{\"req_key\":\"req_key_val\"}\r\n";
// uint8_t tst_data[] = {'a', 'b', 'c', 'd'};}
// uint8_t tst_data[] = {0x01, 0x02, 0x03, 0x04, 0x00};
MESH_DEMO_PRINT("free heap:%u\n", system_get_free_heap_size());
if (!wifi_get_macaddr(STATION_IF, src)) {
MESH_DEMO_PRINT("get sta mac fail\n");
return;
}
MESH_DEMO_MEMCPY(dst, server_ip, sizeof(server_ip));
MESH_DEMO_MEMCPY(dst + sizeof(server_ip), &server_port, sizeof(server_port));
header = (struct mesh_header_format *)espconn_mesh_create_packet(
dst, // destiny address
src, // source address
false, // not p2p packet
true, // piggyback congest request
M_PROTO_JSON, // packe with JSON format
MESH_DEMO_STRLEN(tst_data), // data length
false, // no option
0, // option len
false, // no frag
0, // frag type, this packet doesn't use frag
false, // more frag
0, // frag index
0); // frag length
if (!header) {
MESH_DEMO_PRINT("create packet fail\n");
return;
}
if (!espconn_mesh_set_usr_data(header, tst_data, MESH_DEMO_STRLEN(tst_data))) {
MESH_DEMO_PRINT("set user data fail\n");
MESH_DEMO_FREE(header);
return;
}
if (espconn_mesh_sent(&g_ser_conn, (uint8_t *)header, header->len)) {
MESH_DEMO_PRINT("mesh is busy\n");
MESH_DEMO_FREE(header);
/*
* if fail, we re-connect mesh
*/
espconn_mesh_connect(&g_ser_conn);
return;
}
MESH_DEMO_FREE(header);
}
void ICACHE_FLASH_ATTR mesh_enable_cb(int8_t res)
{
MESH_DEMO_PRINT("mesh_enable_cb\n");
if (res == MESH_OP_FAILURE) {
MESH_DEMO_PRINT("enable mesh fail, re-enable\n");
espconn_mesh_enable(mesh_enable_cb, MESH_ONLINE);
return;
}
if (espconn_mesh_get_usr_context() &&
espconn_mesh_is_root() &&
res == MESH_LOCAL_SUC)
goto TEST_SCENARIO;
/*
* try to estable user virtual connect
* user can to use the virtual connect to sent packet to any node, server or mobile.
* if you want to sent packet to one node in mesh, please build p2p packet
* if you want to sent packet to server/mobile, please build normal packet (uincast packet)
* if you want to sent bcast/mcast packet, please build bcast/mcast packet
*/
MESH_DEMO_MEMSET(&g_ser_conn, 0 ,sizeof(g_ser_conn));
MESH_DEMO_MEMSET(&ser_tcp, 0, sizeof(ser_tcp));
MESH_DEMO_MEMCPY(ser_tcp.remote_ip, server_ip, sizeof(server_ip));
ser_tcp.remote_port = server_port;
ser_tcp.local_port = espconn_port();
g_ser_conn.proto.tcp = &ser_tcp;
if (espconn_regist_connectcb(&g_ser_conn, esp_mesh_demo_con_cb)) {
MESH_DEMO_PRINT("regist con_cb err\n");
espconn_mesh_disable(NULL);
return;
}
if (espconn_regist_recvcb(&g_ser_conn, esp_recv_entrance)) {
MESH_DEMO_PRINT("regist recv_cb err\n");
espconn_mesh_disable(NULL);
return;
}
/*
* regist the other callback
* sent_cb, reconnect_cb, disconnect_cb
* if you donn't need the above cb, you donn't need to register them.
*/
if (espconn_mesh_connect(&g_ser_conn)) {
MESH_DEMO_PRINT("connect err\n");
if (espconn_mesh_is_root())
espconn_mesh_enable(mesh_enable_cb, MESH_LOCAL);
else
espconn_mesh_enable(mesh_enable_cb, MESH_ONLINE);
return;
}
TEST_SCENARIO:
mesh_device_list_init();
mesh_topo_test_init();
mesh_json_mcast_test_init();
mesh_json_bcast_test_init();
mesh_json_p2p_test_init();
}
