network_manager.c 24 KB

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