network_manager.c 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732
  1. /*
  2. Copyright (c) 2017-2021 Sebastien L
  3. */
  4. #ifdef NETWORK_MANAGER_LOG_LEVEL
  5. #define LOG_LOCAL_LEVEL NETWORK_MANAGER_LOG_LEVEL
  6. #endif
  7. #include "network_manager.h"
  8. #include <stdbool.h>
  9. #include <stdio.h>
  10. #include <stdlib.h>
  11. #include <string.h>
  12. #include "network_ethernet.h"
  13. #include "network_status.h"
  14. #include "network_wifi.h"
  15. #include "dns_server.h"
  16. #include "esp_log.h"
  17. #include "esp_system.h"
  18. #include "freertos/FreeRTOS.h"
  19. #include "platform_esp32.h"
  20. #include "esp_netif.h"
  21. #include "freertos/task.h"
  22. #include "cJSON.h"
  23. #include "cmd_system.h"
  24. #include "esp_app_format.h"
  25. #include "esp_event.h"
  26. #include "esp_ota_ops.h"
  27. #include "esp_wifi.h"
  28. #include "esp_wifi_types.h"
  29. #include "lwip/api.h"
  30. #include "lwip/err.h"
  31. #include "lwip/ip4_addr.h"
  32. #include "lwip/netdb.h"
  33. #include "mdns.h"
  34. #include "messaging.h"
  35. #include "platform_config.h"
  36. #include "tools.h"
  37. #include "trace.h"
  38. #include "accessors.h"
  39. #include "esp_err.h"
  40. #include "http_server_handlers.h"
  41. #include "network_manager.h"
  42. QueueHandle_t network_queue;
  43. BaseType_t network_task_handle;
  44. static const char TAG[] = "network";
  45. static TaskHandle_t task_network_manager = NULL;
  46. RTC_NOINIT_ATTR static bool s_wifi_prioritized = false;
  47. extern esp_reset_reason_t xReason;
  48. typedef struct network_callback {
  49. network_status_reached_cb cb;
  50. nm_state_t state;
  51. int sub_state;
  52. const char* from;
  53. SLIST_ENTRY(network_callback)
  54. next; //!< next callback
  55. } network_callback_t;
  56. /** linked list of command structures */
  57. static SLIST_HEAD(cb_list, network_callback) s_cb_list;
  58. network_t NM;
  59. //! Create and initialize the array of state machines.
  60. state_machine_t* const SM[] = {(state_machine_t*)&NM};
  61. static void network_timer_cb(void* timer_id);
  62. int get_root_id(const state_t * state);
  63. const state_t* get_root( const state_t* const state);
  64. static void network_task(void* pvParameters);
  65. void network_start_stop_dhcp(esp_netif_t* netif, bool start) {
  66. tcpip_adapter_dhcp_status_t status;
  67. esp_err_t err = ESP_OK;
  68. ESP_LOGD(TAG, "Checking if DHCP client for STA interface is running");
  69. if (!netif) {
  70. ESP_LOGE(TAG, "Invalid adapter. Cannot start/stop dhcp. ");
  71. return;
  72. }
  73. if((err=esp_netif_dhcpc_get_status(netif, &status))!=ESP_OK){
  74. ESP_LOGE(TAG,"Error retrieving dhcp status : %s", esp_err_to_name(err));
  75. return;
  76. }
  77. switch (status)
  78. {
  79. case ESP_NETIF_DHCP_STARTED:
  80. if(start){
  81. ESP_LOGD(TAG, "DHCP client already started");
  82. }
  83. else {
  84. ESP_LOGI(TAG, "Stopping DHCP client");
  85. ESP_ERROR_CHECK_WITHOUT_ABORT(esp_netif_dhcpc_stop(netif));
  86. }
  87. break;
  88. case ESP_NETIF_DHCP_STOPPED:
  89. if(start){
  90. ESP_LOGI(TAG, "Starting DHCP client");
  91. ESP_ERROR_CHECK_WITHOUT_ABORT(esp_netif_dhcpc_start(netif));
  92. }
  93. else {
  94. ESP_LOGI(TAG, "DHCP client already started");
  95. }
  96. break;
  97. case ESP_NETIF_DHCP_INIT:
  98. if(start){
  99. ESP_LOGI(TAG, "Starting DHCP client");
  100. ESP_ERROR_CHECK_WITHOUT_ABORT(esp_netif_dhcpc_start(netif));
  101. }
  102. else {
  103. ESP_LOGI(TAG, "Stopping DHCP client");
  104. ESP_ERROR_CHECK_WITHOUT_ABORT(esp_netif_dhcpc_stop(netif));
  105. }
  106. break;
  107. default:
  108. ESP_LOGW(TAG,"Unknown DHCP status");
  109. break;
  110. }
  111. }
  112. void network_start_stop_dhcps(esp_netif_t* netif, bool start) {
  113. tcpip_adapter_dhcp_status_t status;
  114. esp_err_t err = ESP_OK;
  115. ESP_LOGD(TAG, "Checking if DHCP server is running");
  116. if (!netif) {
  117. ESP_LOGE(TAG, "Invalid adapter. Cannot start/stop dhcp server. ");
  118. return;
  119. }
  120. if((err=esp_netif_dhcps_get_status(netif, &status))!=ESP_OK){
  121. ESP_LOGE(TAG,"Error retrieving dhcp server status : %s", esp_err_to_name(err));
  122. return;
  123. }
  124. switch (status)
  125. {
  126. case ESP_NETIF_DHCP_STARTED:
  127. if(start){
  128. ESP_LOGD(TAG, "DHCP server already started");
  129. }
  130. else {
  131. ESP_LOGI(TAG, "Stopping DHCP server");
  132. ESP_ERROR_CHECK_WITHOUT_ABORT(esp_netif_dhcps_stop(netif));
  133. }
  134. break;
  135. case ESP_NETIF_DHCP_STOPPED:
  136. if(start){
  137. ESP_LOGI(TAG, "Starting DHCP server");
  138. ESP_ERROR_CHECK_WITHOUT_ABORT(esp_netif_dhcps_start(netif));
  139. }
  140. else {
  141. ESP_LOGI(TAG, "DHCP server already stopped");
  142. }
  143. break;
  144. case ESP_NETIF_DHCP_INIT:
  145. if(start){
  146. ESP_LOGI(TAG, "Starting DHCP server");
  147. ESP_ERROR_CHECK_WITHOUT_ABORT(esp_netif_dhcps_start(netif));
  148. }
  149. else {
  150. ESP_LOGI(TAG, "Stopping DHCP server");
  151. ESP_ERROR_CHECK_WITHOUT_ABORT(esp_netif_dhcps_stop(netif));
  152. }
  153. break;
  154. default:
  155. ESP_LOGW(TAG,"Unknown DHCP status");
  156. break;
  157. }
  158. }
  159. /*********************************************************************************************
  160. * String conversion routines
  161. */
  162. #define ADD_ROOT(name,...) CASE_TO_STR(name);
  163. #define ADD_ROOT_LEAF(name,...) CASE_TO_STR(name);
  164. #define ADD_LEAF(name,...) CASE_TO_STR(name);
  165. #define ADD_EVENT(name) CASE_TO_STR(name);
  166. #define ADD_FIRST_EVENT(name) CASE_TO_STR(name);
  167. static const char* state_to_string(const state_t * state) {
  168. if(!state) {
  169. return "";
  170. }
  171. switch (state->Parent?state->Parent->Id:state->Id) {
  172. ALL_NM_STATE
  173. default:
  174. break;
  175. }
  176. return "Unknown";
  177. }
  178. static const char* wifi_state_to_string(mn_wifi_active_state_t state) {
  179. switch (state) {
  180. ALL_WIFI_STATE(,)
  181. default:
  182. break;
  183. }
  184. return "Unknown";
  185. }
  186. static const char* eth_state_to_string(mn_eth_active_state_t state) {
  187. switch (state) {
  188. ALL_ETH_STATE(,)
  189. default:
  190. break;
  191. }
  192. return "Unknown";
  193. }
  194. static const char* wifi_configuring_state_to_string(mn_wifi_configuring_state_t state) {
  195. switch (state) {
  196. ALL_WIFI_CONFIGURING_STATE(,)
  197. default:
  198. break;
  199. }
  200. return "Unknown";
  201. }
  202. static const char* sub_state_to_string(const state_t * state) {
  203. if(!state) {
  204. return "N/A";
  205. }
  206. int root_id = get_root_id(state);
  207. switch (root_id)
  208. {
  209. case NETWORK_ETH_ACTIVE_STATE:
  210. return eth_state_to_string(state->Id);
  211. break;
  212. case NETWORK_WIFI_ACTIVE_STATE:
  213. return wifi_state_to_string(state->Id);
  214. case NETWORK_WIFI_CONFIGURING_ACTIVE_STATE:
  215. return wifi_configuring_state_to_string(state->Id);
  216. default:
  217. break;
  218. }
  219. return "*";
  220. }
  221. static const char* event_to_string(network_event_t state) {
  222. switch (state) {
  223. ALL_NM_EVENTS
  224. default:
  225. break;
  226. }
  227. return "Unknown";
  228. }
  229. #undef ADD_EVENT
  230. #undef ADD_FIRST_EVENT
  231. #undef ADD_ROOT
  232. #undef ADD_ROOT_LEAF
  233. #undef ADD_LEAF
  234. typedef struct {
  235. int parent_state;
  236. int sub_state_last ;
  237. } max_sub_states_t;
  238. static const max_sub_states_t state_max[] = {
  239. { .parent_state = NETWORK_INSTANTIATED_STATE, .sub_state_last = 0 },
  240. {.parent_state = NETWORK_ETH_ACTIVE_STATE, .sub_state_last = TOTAL_ETH_ACTIVE_STATE-1 },
  241. {.parent_state = NETWORK_WIFI_ACTIVE_STATE, .sub_state_last = TOTAL_WIFI_ACTIVE_STATE-1 },
  242. {.parent_state = WIFI_CONFIGURING_STATE, .sub_state_last = TOTAL_WIFI_CONFIGURING_STATE-1 },
  243. {.parent_state = WIFI_CONFIGURING_STATE, .sub_state_last = TOTAL_WIFI_CONFIGURING_STATE-1 },
  244. {.parent_state =-1}
  245. };
  246. void network_start() {
  247. if(xReason == ESP_RST_POWERON ){
  248. ESP_LOGD(TAG, "Power on reset, initializing wifi priotitized to false");
  249. s_wifi_prioritized = false;
  250. }
  251. ESP_LOGD(TAG, " Creating message queue");
  252. network_queue = xQueueCreate(3, sizeof(queue_message));
  253. ESP_LOGD(TAG, " Creating network manager task");
  254. network_task_handle = xTaskCreate(&network_task, "network", 4096, NULL, WIFI_MANAGER_TASK_PRIORITY, &task_network_manager);
  255. }
  256. static void event_logger(uint32_t state_machine, uint32_t state, uint32_t event) {
  257. ESP_LOGI(TAG, "Handling network manager event state Id %d->[%s]", state, event_to_string(event));
  258. }
  259. static const char * get_state_machine_result_string(state_machine_result_t result) {
  260. switch(result) {
  261. case EVENT_HANDLED:
  262. return "EVENT_HANDLED";
  263. case EVENT_UN_HANDLED:
  264. return "EVENT_UN_HANDLED";
  265. case TRIGGERED_TO_SELF:
  266. return "TRIGGERED_TO_SELF";
  267. }
  268. return "Unknown";
  269. }
  270. static void result_logger(uint32_t state, state_machine_result_t result) {
  271. ESP_LOGD(TAG, "Network Manager Result: %s, New State id: %d", get_state_machine_result_string(result) , state);
  272. }
  273. static void network_task(void* pvParameters) {
  274. queue_message msg;
  275. BaseType_t xStatus;
  276. initialize_network_handlers((state_machine_t*)&NM);
  277. network_async(EN_START);
  278. /* main processing loop */
  279. for (;;) {
  280. xStatus = xQueueReceive(network_queue, &msg, portMAX_DELAY);
  281. if (xStatus == pdPASS) {
  282. // pass the event to the sync processor
  283. NM.event_parameters = &msg;
  284. NM.Machine.Event = msg.trigger;
  285. if (dispatch_event(SM, 1, event_logger, result_logger) == EVENT_UN_HANDLED) {
  286. network_manager_format_from_to_states(ESP_LOG_ERROR,"Unhandled Event",NULL,NM.Machine.State,msg.trigger,false,"network manager");
  287. }
  288. } /* end of if status=pdPASS */
  289. } /* end of for loop */
  290. vTaskDelete(NULL);
  291. }
  292. int get_max_substate(nm_state_t state){
  293. for(int i=0;state_max[i].parent_state!=-1;i++){
  294. if(state_max[i].parent_state == state){
  295. return state_max[i].sub_state_last;
  296. }
  297. }
  298. return -1;
  299. }
  300. esp_err_t network_register_state_callback(nm_state_t state,int sub_state, const char* from, network_status_reached_cb cb) {
  301. network_callback_t* item = NULL;
  302. if (!cb) {
  303. return ESP_ERR_INVALID_ARG;
  304. }
  305. item = calloc(1, sizeof(*item));
  306. if (item == NULL) {
  307. return ESP_ERR_NO_MEM;
  308. }
  309. if(sub_state != -1 && sub_state>get_max_substate(state)){
  310. // sub state has to be valid
  311. return ESP_ERR_INVALID_ARG;
  312. }
  313. item->state = state;
  314. item->cb = cb;
  315. item->from = from;
  316. item->sub_state=sub_state;
  317. network_callback_t* last = SLIST_FIRST(&s_cb_list);
  318. if (last == NULL) {
  319. SLIST_INSERT_HEAD(&s_cb_list, item, next);
  320. } else {
  321. network_callback_t* it;
  322. while ((it = SLIST_NEXT(last, next)) != NULL) {
  323. last = it;
  324. }
  325. SLIST_INSERT_AFTER(last, item, next);
  326. }
  327. return ESP_OK;
  328. }
  329. const state_t* get_root( const state_t* const state){
  330. if(!state) return NULL;
  331. return state->Parent==NULL?state: get_root(state->Parent);
  332. }
  333. int get_root_id(const state_t * state){
  334. if(!state) return -1;
  335. return state->Parent==NULL?state->Id: get_root_id(state->Parent);
  336. }
  337. static bool is_root_state(const state_t * state){
  338. return state->Parent==NULL;
  339. }
  340. static bool is_current_state(const state_t* state, nm_state_t state_id, int sub_state_id){
  341. return get_root(state)->Id == state_id && (sub_state_id==-1 || (!is_root_state(state) && state->Id == sub_state_id) );
  342. }
  343. void network_execute_cb(state_machine_t* const state_machine, const char * caller) {
  344. network_callback_t* it;
  345. SLIST_FOREACH(it, &s_cb_list, next) {
  346. if (is_current_state(state_machine->State,it->state, it->sub_state)) {
  347. char * cb_prefix= messaging_alloc_format_string("BEGIN Executing Callback %s", it->from) ;
  348. NETWORK_DEBUG_STATE_MACHINE(true,STR_OR_BLANK(cb_prefix),state_machine,false, STR_OR_BLANK(caller));
  349. FREE_AND_NULL(cb_prefix);
  350. it->cb((nm_state_t)get_root(state_machine->State)->Id, is_root_state(state_machine->State)?-1:state_machine->State->Id);
  351. cb_prefix= messaging_alloc_format_string("END Executing Callback %s", it->from) ;
  352. NETWORK_DEBUG_STATE_MACHINE(false,STR_OR_BLANK(cb_prefix),state_machine,false, STR_OR_BLANK(caller));
  353. FREE_AND_NULL(cb_prefix);
  354. }
  355. }
  356. }
  357. bool network_is_wifi_prioritized() {
  358. eth_config_t eth_config;
  359. config_eth_init(&eth_config);
  360. // char* prioritize = (char*)config_alloc_get_default(NVS_TYPE_STR, "prio_wifi", "N", 0);
  361. // bool result = strcasecmp("N", prioritize);
  362. bool result = s_wifi_prioritized;
  363. if(result){
  364. result = network_wifi_get_known_count()>0 || !eth_config.valid;
  365. ESP_LOGD(TAG,"Wifi is prioritized with %d known access points.%s %s",network_wifi_get_known_count(),eth_config.valid?" And a valid ethernet adapter":"",result?"Wifi prioritized":"Ethernet prioritized");
  366. }
  367. return result;
  368. }
  369. void network_prioritize_wifi(bool activate) {
  370. if(s_wifi_prioritized == activate) return;
  371. s_wifi_prioritized = activate;
  372. ESP_LOGI(TAG,"Wifi is %s prioritized",activate?"":"not");
  373. // if (network_is_wifi_prioritized() != activate) {
  374. // ESP_LOGW(TAG, "Wifi will %s be prioritized on next boot", activate ? "" : "NOT");
  375. // config_set_value(NVS_TYPE_STR, "prio_wifi", activate ? "Y" : "N");
  376. // }
  377. }
  378. void network_manager_format_state_machine(esp_log_level_t level, const char* prefix, state_machine_t* state_machine, bool show_source, const char * caller) {
  379. state_t const* source_state = NULL;
  380. state_t const* current_state = NULL;
  381. network_event_t event = -1;
  382. MEMTRACE_PRINT_DELTA();
  383. if (state_machine) {
  384. source_state = ((network_t *)state_machine)->source_state;
  385. current_state = state_machine->State;
  386. event = state_machine->Event;
  387. network_manager_format_from_to_states(level, prefix, source_state, current_state, event, show_source,caller);
  388. }
  389. else {
  390. ESP_LOG_LEVEL(level, TAG, "%s - %s -> [%s]",
  391. STR_OR_BLANK(caller),
  392. prefix,
  393. event_to_string(event));
  394. }
  395. }
  396. void network_manager_format_from_to_states(esp_log_level_t level, const char* prefix, const state_t * from_state,const state_t * current_state, network_event_t event,bool show_source, const char * caller) {
  397. const char* source_state = "";
  398. const char* source_sub_state = "";
  399. const char* state = "N/A";
  400. const char* sub_state = "N/A";
  401. if (current_state) {
  402. state = state_to_string(current_state);
  403. sub_state = sub_state_to_string(current_state);
  404. }
  405. if (!from_state) {
  406. source_state = "N/A";
  407. } else {
  408. source_state = state_to_string(from_state);
  409. source_sub_state = sub_state_to_string(from_state);
  410. }
  411. if (show_source) {
  412. ESP_LOG_LEVEL(level, TAG, "%s %s %s(%s)->%s(%s) [%s]",
  413. STR_OR_BLANK(caller),
  414. prefix,
  415. source_state,
  416. source_sub_state,
  417. state,
  418. sub_state,
  419. event_to_string(event));
  420. } else {
  421. ESP_LOG_LEVEL(level, TAG, "%s %s %s(%s) [%s]",
  422. STR_OR_BLANK(caller),
  423. prefix,
  424. state,
  425. sub_state,
  426. event_to_string(event));
  427. }
  428. }
  429. void network_async(network_event_t trigger) {
  430. queue_message msg;
  431. memset(&msg,0x00,sizeof(msg));
  432. msg.trigger = trigger;
  433. ESP_LOGD(TAG, "Posting event %s directly", event_to_string(trigger));
  434. xQueueSendToBack(network_queue, &msg, portMAX_DELAY);
  435. }
  436. void network_async_fail() {
  437. network_async(EN_FAIL);
  438. }
  439. void network_async_success() {
  440. network_async(EN_SUCCESS);
  441. }
  442. void network_async_connected(){
  443. network_async(EN_CONNECTED);
  444. }
  445. void network_async_link_up() {
  446. network_async(EN_LINK_UP);
  447. }
  448. void network_async_link_down() {
  449. network_async(EN_LINK_DOWN);
  450. }
  451. void network_async_configure() {
  452. network_async(EN_CONFIGURE);
  453. }
  454. void network_async_got_ip() {
  455. network_async(EN_GOT_IP);
  456. }
  457. void network_async_eth_got_ip() {
  458. network_async(EN_ETH_GOT_IP);
  459. }
  460. void network_async_timer() {
  461. network_async(EN_TIMER);
  462. }
  463. void network_async_start() {
  464. network_async(EN_START);
  465. }
  466. void network_async_scan() {
  467. network_async(EN_SCAN);
  468. }
  469. void network_async_update_status() {
  470. network_async(EN_UPDATE_STATUS);
  471. }
  472. void network_async_delete() {
  473. network_async(EN_DELETE);
  474. }
  475. void network_async_scan_done() {
  476. network_async(EN_SCAN_DONE);
  477. }
  478. void network_async_connect(const char * ssid, const char * password) {
  479. queue_message msg;
  480. memset(&msg,0x00,sizeof(msg));
  481. msg.trigger = EN_CONNECT_NEW;
  482. msg.ssid = strdup_psram(ssid);
  483. if(password && strlen(password) >0){
  484. msg.password = strdup_psram(password);
  485. }
  486. ESP_LOGD(TAG, "Posting event %s", event_to_string(msg.trigger));
  487. xQueueSendToBack(network_queue, &msg, portMAX_DELAY);
  488. }
  489. void network_async_lost_connection(wifi_event_sta_disconnected_t* disconnected_event) {
  490. queue_message msg;
  491. memset(&msg,0x00,sizeof(msg));
  492. msg.trigger = EN_LOST_CONNECTION;
  493. ESP_LOGD(TAG, "Posting event %s", event_to_string(msg.trigger));
  494. msg.disconnected_event = malloc_init_external(sizeof(wifi_event_sta_disconnected_t));
  495. if(msg.disconnected_event){
  496. memcpy(msg.disconnected_event, disconnected_event,sizeof(wifi_event_sta_disconnected_t));
  497. xQueueSendToBack(network_queue, &msg, portMAX_DELAY);
  498. }
  499. else {
  500. ESP_LOGE(TAG,"Unable to post lost connection event.");
  501. }
  502. }
  503. void network_async_reboot(reboot_type_t rtype) {
  504. queue_message msg;
  505. memset(&msg,0x00,sizeof(msg));
  506. msg.trigger = EN_REBOOT;
  507. msg.rtype = rtype;
  508. ESP_LOGD(TAG, "Posting event %s - type %d", event_to_string(msg.trigger),rtype);
  509. xQueueSendToBack(network_queue, &msg, portMAX_DELAY);
  510. }
  511. void network_reboot_ota(char* url) {
  512. queue_message msg;
  513. memset(&msg,0x00,sizeof(msg));
  514. if (url == NULL) {
  515. msg.trigger = EN_REBOOT;
  516. msg.rtype = OTA;
  517. ESP_LOGD(TAG, "Posting event %s - type %d", event_to_string(msg.trigger),msg.rtype);
  518. } else {
  519. msg.trigger = EN_REBOOT_URL;
  520. ESP_LOGD(TAG, "Posting event %s - type reboot URL", event_to_string(msg.trigger));
  521. msg.strval = strdup_psram(url);
  522. }
  523. xQueueSendToBack(network_queue, &msg, portMAX_DELAY);
  524. }
  525. network_t* network_get_state_machine() {
  526. return &NM;
  527. }
  528. static void network_timer_cb(void* timer_id) {
  529. network_async_timer();
  530. }
  531. esp_netif_t* network_get_active_interface() {
  532. if (NM.wifi_ap_netif && (network_wifi_is_ap_mode() || network_wifi_is_ap_sta_mode())) {
  533. return NM.wifi_ap_netif;
  534. } else if (NM.wifi_netif && network_wifi_is_sta_mode()) {
  535. return NM.wifi_netif;
  536. }
  537. return NM.eth_netif;
  538. }
  539. bool network_is_interface_connected(esp_netif_t* interface) {
  540. esp_err_t err = ESP_OK;
  541. tcpip_adapter_ip_info_t ipInfo;
  542. if(!interface){
  543. return false;
  544. }
  545. err = network_get_ip_info_for_netif(interface, &ipInfo);
  546. if(err != ESP_OK){
  547. ESP_LOGD(TAG,"network_get_ip_info_for_netif returned %s", esp_err_to_name(err));
  548. }
  549. return ((err == ESP_OK) && (ipInfo.ip.addr != IPADDR_ANY));
  550. }
  551. static esp_netif_t* get_connected_interface() {
  552. esp_netif_t* interface = NULL;
  553. for (int i = 0; i < 4; i++) {
  554. switch (i) {
  555. case 0:
  556. // try the active interface
  557. interface = network_get_active_interface();
  558. break;
  559. case 1:
  560. interface = NM.wifi_ap_netif;
  561. break;
  562. case 2:
  563. interface = NM.wifi_netif;
  564. break;
  565. case 3:
  566. interface = NM.eth_netif;
  567. break;
  568. default:
  569. break;
  570. }
  571. if (interface && network_is_interface_connected(interface)) {
  572. ESP_LOGD(TAG,"Found connected interface in iteration #%d",i);
  573. return interface;
  574. }
  575. }
  576. ESP_LOGD(TAG,"No connected interface found");
  577. return NULL;
  578. }
  579. esp_err_t network_get_ip_info_for_netif(esp_netif_t* netif, tcpip_adapter_ip_info_t* ipInfo) {
  580. esp_netif_ip_info_t loc_ip_info;
  581. if (!ipInfo ) {
  582. ESP_LOGE(TAG, "Invalid pointer for ipInfo");
  583. return ESP_ERR_INVALID_ARG;
  584. }
  585. if (!netif) {
  586. ESP_LOGE(TAG, "Invalid pointer for netif");
  587. return ESP_ERR_INVALID_ARG;
  588. }
  589. memset(ipInfo,0x00,sizeof(tcpip_adapter_ip_info_t));
  590. esp_err_t err= esp_netif_get_ip_info(netif, &loc_ip_info);
  591. if(err==ESP_OK){
  592. ip4_addr_set(&(ipInfo->ip),&loc_ip_info.ip);
  593. ip4_addr_set(&(ipInfo->gw),&loc_ip_info.gw);
  594. ip4_addr_set(&(ipInfo->netmask),&loc_ip_info.netmask);
  595. }
  596. return err;
  597. }
  598. esp_err_t network_get_ip_info(tcpip_adapter_ip_info_t* ipInfo) {
  599. esp_netif_t* netif= get_connected_interface();
  600. if(netif){
  601. return network_get_ip_info_for_netif(netif,ipInfo);
  602. }
  603. return ESP_FAIL;
  604. }
  605. esp_err_t network_get_hostname(const char** hostname) {
  606. return esp_netif_get_hostname(get_connected_interface(), hostname);
  607. }
  608. void network_set_timer(uint16_t duration) {
  609. if (duration > 0) {
  610. if (!NM.state_timer) {
  611. ESP_LOGD(TAG, "Starting new pulse check timer with period of %u ms.", duration);
  612. NM.state_timer = xTimerCreate("background STA", pdMS_TO_TICKS(duration), pdFALSE, NULL, network_timer_cb);
  613. } else {
  614. ESP_LOGD(TAG, "Changing the pulse timer period to %u ms.", duration);
  615. xTimerChangePeriod(NM.state_timer, pdMS_TO_TICKS(duration), portMAX_DELAY);
  616. }
  617. xTimerStart(NM.state_timer, portMAX_DELAY);
  618. } else if (NM.state_timer) {
  619. ESP_LOGD(TAG, "Stopping timer");
  620. xTimerStop(NM.state_timer, portMAX_DELAY);
  621. }
  622. }
  623. void network_ip_event_handler(void* arg, esp_event_base_t event_base, int32_t event_id, void* event_data) {
  624. ip_event_got_ip_t* s = NULL;
  625. esp_netif_ip_info_t* ip_info = NULL;
  626. if (event_base != IP_EVENT)
  627. return;
  628. switch (event_id) {
  629. case IP_EVENT_ETH_GOT_IP:
  630. case IP_EVENT_STA_GOT_IP:
  631. s = (ip_event_got_ip_t*)event_data;
  632. ip_info = &s->ip_info;
  633. ESP_LOGI(TAG, "Got an IP address from interface %s. IP=" IPSTR ", Gateway=" IPSTR ", NetMask=" IPSTR ", %s",
  634. event_id == IP_EVENT_ETH_GOT_IP ? "Eth" : event_id == IP_EVENT_STA_GOT_IP ? "Wifi"
  635. : "Unknown",
  636. IP2STR(&ip_info->ip),
  637. IP2STR(&ip_info->gw),
  638. IP2STR(&ip_info->netmask),
  639. s->ip_changed ? "Address was changed" : "Address unchanged");
  640. network_async(event_id == IP_EVENT_ETH_GOT_IP ? EN_ETH_GOT_IP : EN_GOT_IP);
  641. break;
  642. case IP_EVENT_STA_LOST_IP:
  643. ESP_LOGD(TAG, "IP_EVENT_STA_LOST_IP");
  644. break;
  645. case IP_EVENT_AP_STAIPASSIGNED:
  646. ESP_LOGD(TAG, "IP_EVENT_AP_STAIPASSIGNED");
  647. break;
  648. case IP_EVENT_GOT_IP6:
  649. ESP_LOGD(TAG, "IP_EVENT_GOT_IP6");
  650. break;
  651. default:
  652. break;
  653. }
  654. }
  655. void network_set_hostname(esp_netif_t* interface) {
  656. esp_err_t err;
  657. ESP_LOGD(TAG, "Retrieving host name from nvs");
  658. char* host_name = (char*)config_alloc_get(NVS_TYPE_STR, "host_name");
  659. if (host_name == NULL) {
  660. ESP_LOGE(TAG, "Could not retrieve host name from nvs");
  661. } else {
  662. ESP_LOGD(TAG, "Setting host name to : %s", host_name);
  663. if ((err = esp_netif_set_hostname(interface, host_name)) != ESP_OK) {
  664. ESP_LOGE(TAG, "Unable to set host name. Error: %s", esp_err_to_name(err));
  665. }
  666. free(host_name);
  667. }
  668. }
  669. #define LOCAL_MAC_SIZE 20
  670. char* network_manager_alloc_get_mac_string(uint8_t mac[6]) {
  671. char* macStr = malloc_init_external(LOCAL_MAC_SIZE);
  672. if(macStr){
  673. snprintf(macStr, LOCAL_MAC_SIZE, MACSTR, MAC2STR(mac));
  674. }
  675. return macStr;
  676. }