esp_app_main.c 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511
  1. /*
  2. * Squeezelite for esp32
  3. *
  4. * (c) Sebastien 2019
  5. * Philippe G. 2019, philippe_44@outlook.com
  6. *
  7. * This software is released under the MIT License.
  8. * https://opensource.org/licenses/MIT
  9. *
  10. */
  11. #include "platform_esp32.h"
  12. #include "led.h"
  13. #include <stdio.h>
  14. #include <string.h>
  15. #include "freertos/FreeRTOS.h"
  16. #include "driver/gpio.h"
  17. #include "driver/spi_master.h"
  18. #include "freertos/task.h"
  19. #include "esp_system.h"
  20. #include "esp_spi_flash.h"
  21. #include "esp_wifi.h"
  22. #include <esp_event.h>
  23. #include "nvs_flash.h"
  24. #include "esp_log.h"
  25. #include "freertos/event_groups.h"
  26. #include "mdns.h"
  27. #include "lwip/api.h"
  28. #include "lwip/err.h"
  29. #include "lwip/netdb.h"
  30. #include "nvs_utilities.h"
  31. #include "trace.h"
  32. #include "wifi_manager.h"
  33. #include "squeezelite-ota.h"
  34. #include <math.h>
  35. #include "audio_controls.h"
  36. #include "platform_config.h"
  37. #include "telnet.h"
  38. #include "messaging.h"
  39. #include "gds.h"
  40. #include "gds_default_if.h"
  41. #include "gds_draw.h"
  42. #include "gds_text.h"
  43. #include "gds_font.h"
  44. #include "display.h"
  45. #include "accessors.h"
  46. #include "cmd_system.h"
  47. static const char certs_namespace[] = "certificates";
  48. static const char certs_key[] = "blob";
  49. static const char certs_version[] = "version";
  50. const char unknown_string_placeholder[] = "unknown";
  51. const char null_string_placeholder[] = "null";
  52. EventGroupHandle_t wifi_event_group;
  53. bool bypass_wifi_manager=false;
  54. const int CONNECTED_BIT = BIT0;
  55. #define JOIN_TIMEOUT_MS (10000)
  56. #define LOCAL_MAC_SIZE 20
  57. static const char TAG[] = "esp_app_main";
  58. #define DEFAULT_HOST_NAME "squeezelite"
  59. char * fwurl = NULL;
  60. RTC_NOINIT_ATTR uint32_t RebootCounter ;
  61. static bool bWifiConnected=false;
  62. extern const uint8_t server_cert_pem_start[] asm("_binary_github_pem_start");
  63. extern const uint8_t server_cert_pem_end[] asm("_binary_github_pem_end");
  64. // as an exception _init function don't need include
  65. extern void services_init(void);
  66. extern void display_init(char *welcome);
  67. const char * str_or_unknown(const char * str) { return (str?str:unknown_string_placeholder); }
  68. const char * str_or_null(const char * str) { return (str?str:null_string_placeholder); }
  69. bool is_recovery_running;
  70. /* brief this is an exemple of a callback that you can setup in your own app to get notified of wifi manager event */
  71. void cb_connection_got_ip(void *pvParameter){
  72. static ip4_addr_t ip;
  73. tcpip_adapter_ip_info_t ipInfo;
  74. tcpip_adapter_get_ip_info(TCPIP_ADAPTER_IF_STA, &ipInfo);
  75. if (ip.addr && ipInfo.ip.addr != ip.addr) {
  76. ESP_LOGW(TAG, "IP change, need to reboot");
  77. if(!wait_for_commit()){
  78. ESP_LOGW(TAG,"Unable to commit configuration. ");
  79. }
  80. esp_restart();
  81. }
  82. ip.addr = ipInfo.ip.addr;
  83. ESP_LOGI(TAG, "I have a connection!");
  84. messaging_post_message(MESSAGING_INFO,MESSAGING_CLASS_SYSTEM,"Wifi connected");
  85. xEventGroupSetBits(wifi_event_group, CONNECTED_BIT);
  86. bWifiConnected=true;
  87. led_unpush(LED_GREEN);
  88. }
  89. void cb_connection_sta_disconnected(void *pvParameter){
  90. led_blink_pushed(LED_GREEN, 250, 250);
  91. messaging_post_message(MESSAGING_WARNING,MESSAGING_CLASS_SYSTEM,"Wifi disconnected");
  92. bWifiConnected=false;
  93. xEventGroupClearBits(wifi_event_group, CONNECTED_BIT);
  94. }
  95. bool wait_for_wifi(){
  96. bool connected=(xEventGroupGetBits(wifi_event_group) & CONNECTED_BIT)!=0;
  97. if(!connected){
  98. ESP_LOGD(TAG,"Waiting for WiFi...");
  99. connected = (xEventGroupWaitBits(wifi_event_group, CONNECTED_BIT,
  100. pdFALSE, pdTRUE, JOIN_TIMEOUT_MS / portTICK_PERIOD_MS)& CONNECTED_BIT)!=0;
  101. if(!connected){
  102. ESP_LOGW(TAG,"wifi timeout.");
  103. }
  104. else
  105. {
  106. ESP_LOGI(TAG,"WiFi Connected!");
  107. }
  108. }
  109. return connected;
  110. }
  111. char * process_ota_url(){
  112. ESP_LOGI(TAG,"Checking for update url");
  113. char * fwurl=config_alloc_get(NVS_TYPE_STR, "fwurl");
  114. if(fwurl!=NULL)
  115. {
  116. ESP_LOGD(TAG,"Deleting nvs entry for Firmware URL %s", fwurl);
  117. config_delete_key("fwurl");
  118. }
  119. return fwurl;
  120. }
  121. esp_log_level_t get_log_level_from_char(char * level){
  122. if(!strcasecmp(level, "NONE" )) { return ESP_LOG_NONE ;}
  123. if(!strcasecmp(level, "ERROR" )) { return ESP_LOG_ERROR ;}
  124. if(!strcasecmp(level, "WARN" )) { return ESP_LOG_WARN ;}
  125. if(!strcasecmp(level, "INFO" )) { return ESP_LOG_INFO ;}
  126. if(!strcasecmp(level, "DEBUG" )) { return ESP_LOG_DEBUG ;}
  127. if(!strcasecmp(level, "VERBOSE" )) { return ESP_LOG_VERBOSE;}
  128. return ESP_LOG_WARN;
  129. }
  130. void set_log_level(char * tag, char * level){
  131. esp_log_level_set(tag, get_log_level_from_char(level));
  132. }
  133. esp_err_t update_certificates(bool force){
  134. nvs_handle handle;
  135. esp_err_t esp_err;
  136. esp_app_desc_t running_app_info;
  137. ESP_LOGI(TAG, "About to check if certificates need to be updated in flash");
  138. esp_err = nvs_open_from_partition(settings_partition, certs_namespace, NVS_READWRITE, &handle);
  139. if (esp_err != ESP_OK) {
  140. LOG_SEND(MESSAGING_INFO,"Unable to update HTTPS certificates. Could not open NVS namespace %s. Error %s", certs_namespace, esp_err_to_name(esp_err));
  141. return esp_err;
  142. }
  143. const esp_partition_t *running = esp_ota_get_running_partition();
  144. ESP_LOGI(TAG, "Running partition [%s] type %d subtype %d (offset 0x%08x)", running->label, running->type, running->subtype, running->address);
  145. if (esp_ota_get_partition_description(running, &running_app_info) == ESP_OK) {
  146. ESP_LOGI(TAG, "Running version: %s", running_app_info.version);
  147. }
  148. size_t len=0;
  149. char *str=NULL;
  150. bool changed=false;
  151. if ( (esp_err= nvs_get_str(handle, certs_version, NULL, &len)) == ESP_OK) {
  152. str=(char *)malloc(len+1);
  153. if(str){
  154. memset(str,0x00,len+1);
  155. if ( (esp_err = nvs_get_str(handle, certs_version, str, &len)) == ESP_OK) {
  156. ESP_LOGI(TAG,"Certificate version: %s", str);
  157. }
  158. }
  159. }
  160. if(str!=NULL && running->subtype !=ESP_PARTITION_SUBTYPE_APP_FACTORY){
  161. // If certificates were found in nvs, only update if we're not
  162. // running recovery. This will prevent rolling back to an older version
  163. if(strcmp((char *)running_app_info.version,(char *)str )){
  164. // Versions are different
  165. ESP_LOGW(TAG,"Found a different software version. Updating certificates");
  166. changed=true;
  167. }
  168. free(str);
  169. }
  170. else if(str==NULL){
  171. ESP_LOGW(TAG,"No certificate found. Adding certificates");
  172. changed=true;
  173. }
  174. if(changed || force){
  175. esp_err = nvs_set_blob(handle, certs_key, server_cert_pem_start, (server_cert_pem_end-server_cert_pem_start));
  176. if(esp_err!=ESP_OK){
  177. log_send_messaging(MESSAGING_ERROR,"Failed to store certificate data: %s", esp_err_to_name(esp_err));
  178. }
  179. else {
  180. esp_err = nvs_set_str(handle, certs_version, running_app_info.version);
  181. if(esp_err!=ESP_OK){
  182. log_send_messaging(MESSAGING_ERROR,"Unable to update HTTPS Certificates version: %s",esp_err_to_name(esp_err));
  183. }
  184. else {
  185. esp_err = nvs_commit(handle);
  186. if(esp_err!=ESP_OK){
  187. log_send_messaging(MESSAGING_ERROR,"Failed to commit certificates changes : %s",esp_err_to_name(esp_err));
  188. }
  189. else {
  190. log_send_messaging(MESSAGING_INFO,"HTTPS Certificates were updated with version: %s",running_app_info.version);
  191. }
  192. }
  193. }
  194. }
  195. nvs_close(handle);
  196. return ESP_OK;
  197. }
  198. const char * get_certificate(){
  199. nvs_handle handle;
  200. esp_err_t esp_err;
  201. char *blob =NULL;
  202. //
  203. ESP_LOGD(TAG, "Fetching certificate.");
  204. esp_err = nvs_open_from_partition(settings_partition, certs_namespace, NVS_READONLY, &handle);
  205. if(esp_err == ESP_OK){
  206. size_t len;
  207. esp_err = nvs_get_blob(handle, certs_key, NULL, &len);
  208. if( esp_err == ESP_OK) {
  209. blob = (char *) heap_caps_malloc(len+1, (MALLOC_CAP_INTERNAL|MALLOC_CAP_8BIT));
  210. if(!blob){
  211. log_send_messaging(MESSAGING_ERROR,"Unable to retrieve HTTPS certificates. %s","Memory allocation failed");
  212. return "";
  213. }
  214. memset(blob,0x00,len+1);
  215. esp_err = nvs_get_blob(handle, certs_key, blob, &len);
  216. if ( esp_err == ESP_OK) {
  217. ESP_LOGD(TAG,"Certificates content is %d bytes long: ", len);
  218. }
  219. else {
  220. log_send_messaging(MESSAGING_ERROR,"Unable to retrieve HTTPS certificates. Get blob failed: %s", esp_err_to_name(esp_err));
  221. }
  222. }
  223. else{
  224. log_send_messaging(MESSAGING_ERROR,"Unable to retrieve HTTPS certificates. Get blob failed: %s",esp_err_to_name(esp_err));
  225. }
  226. nvs_close(handle);
  227. }
  228. else{
  229. log_send_messaging(MESSAGING_ERROR,"Unable to retrieve HTTPS certificates. NVS name space %s open failed: %s",certs_namespace, esp_err_to_name(esp_err));
  230. }
  231. return blob;
  232. }
  233. #define DEFAULT_NAME_WITH_MAC(var,defval) char var[strlen(defval)+sizeof(macStr)]; strcpy(var,defval); strcat(var,macStr)
  234. void register_default_nvs(){
  235. uint8_t mac[6];
  236. char macStr[LOCAL_MAC_SIZE+1];
  237. char default_command_line[strlen(CONFIG_DEFAULT_COMMAND_LINE)+sizeof(macStr)];
  238. esp_read_mac((uint8_t *)&mac, ESP_MAC_WIFI_STA);
  239. snprintf(macStr, LOCAL_MAC_SIZE-1,"-%x%x%x", mac[3], mac[4], mac[5]);
  240. DEFAULT_NAME_WITH_MAC(default_bt_name,CONFIG_BT_NAME);
  241. ESP_LOGD(TAG,"Registering default value for key %s, value %s", "bt_name", default_bt_name);
  242. config_set_default(NVS_TYPE_STR, "bt_name", default_bt_name, 0);
  243. DEFAULT_NAME_WITH_MAC(default_host_name,DEFAULT_HOST_NAME);
  244. ESP_LOGD(TAG,"Registering default value for key %s, value %s", "host_name", default_host_name);
  245. config_set_default(NVS_TYPE_STR, "host_name", default_host_name, 0);
  246. DEFAULT_NAME_WITH_MAC(default_airplay_name,CONFIG_AIRPLAY_NAME);
  247. ESP_LOGD(TAG,"Registering default value for key %s, value %s", "airplay_name",default_airplay_name);
  248. config_set_default(NVS_TYPE_STR, "airplay_name",default_airplay_name , 0);
  249. DEFAULT_NAME_WITH_MAC(default_ap_name,CONFIG_DEFAULT_AP_SSID);
  250. ESP_LOGD(TAG,"Registering default value for key %s, value %s", "ap_ssid", default_ap_name);
  251. config_set_default(NVS_TYPE_STR, "ap_ssid",default_ap_name , 0);
  252. strncpy(default_command_line, CONFIG_DEFAULT_COMMAND_LINE,sizeof(default_command_line)-1);
  253. strncat(default_command_line, " -n ",sizeof(default_command_line)-1);
  254. strncat(default_command_line, default_host_name,sizeof(default_command_line)-1);
  255. ESP_LOGD(TAG,"Registering default value for key %s, value %s", "autoexec", "1");
  256. config_set_default(NVS_TYPE_STR,"autoexec","1", 0);
  257. ESP_LOGD(TAG,"Registering default value for key %s, value %s", "autoexec1",default_command_line);
  258. config_set_default(NVS_TYPE_STR,"autoexec1",default_command_line,0);
  259. ESP_LOGD(TAG,"Registering default value for key %s, value %s", "a2dp_sink_name", CONFIG_A2DP_SINK_NAME);
  260. config_set_default(NVS_TYPE_STR, "a2dp_sink_name", CONFIG_A2DP_SINK_NAME, 0);
  261. ESP_LOGD(TAG,"Registering default value for key %s, value %s", "bt_sink_pin", STR(CONFIG_BT_SINK_PIN));
  262. config_set_default(NVS_TYPE_STR, "a2dp_ctmt", STR(CONFIG_A2DP_CONNECT_TIMEOUT_MS), 0);
  263. ESP_LOGD(TAG,"Registering default value for key %s, value %s", "bt_sink_pin", STR(CONFIG_A2DP_CONNECT_TIMEOUT_MS));
  264. config_set_default(NVS_TYPE_STR, "a2dp_ctrld", STR(CONFIG_A2DP_CONTROL_DELAY_MS), 0);
  265. ESP_LOGD(TAG,"Registering default value for key %s, value %s", "bt_sink_pin", STR(CONFIG_A2DP_CONTROL_DELAY_MS));
  266. config_set_default(NVS_TYPE_STR, "bt_sink_pin", STR(CONFIG_BT_SINK_PIN), 0);
  267. ESP_LOGD(TAG,"Registering default value for key %s, value %s", "release_url", CONFIG_SQUEEZELITE_ESP32_RELEASE_URL);
  268. config_set_default(NVS_TYPE_STR, "release_url", CONFIG_SQUEEZELITE_ESP32_RELEASE_URL, 0);
  269. ESP_LOGD(TAG,"Registering default value for key %s, value %s","ap_ip_address",CONFIG_DEFAULT_AP_IP );
  270. config_set_default(NVS_TYPE_STR, "ap_ip_address",CONFIG_DEFAULT_AP_IP , 0);
  271. ESP_LOGD(TAG,"Registering default value for key %s, value %s", "ap_ip_gateway",CONFIG_DEFAULT_AP_GATEWAY );
  272. config_set_default(NVS_TYPE_STR, "ap_ip_gateway",CONFIG_DEFAULT_AP_GATEWAY , 0);
  273. ESP_LOGD(TAG,"Registering default value for key %s, value %s","ap_ip_netmask",CONFIG_DEFAULT_AP_NETMASK );
  274. config_set_default(NVS_TYPE_STR, "ap_ip_netmask",CONFIG_DEFAULT_AP_NETMASK , 0);
  275. ESP_LOGD(TAG,"Registering default value for key %s, value %s", "ap_channel",STR(CONFIG_DEFAULT_AP_CHANNEL));
  276. config_set_default(NVS_TYPE_STR, "ap_channel",STR(CONFIG_DEFAULT_AP_CHANNEL) , 0);
  277. ESP_LOGD(TAG,"Registering default value for key %s, value %s", "ap_pwd", CONFIG_DEFAULT_AP_PASSWORD);
  278. config_set_default(NVS_TYPE_STR, "ap_pwd", CONFIG_DEFAULT_AP_PASSWORD, 0);
  279. ESP_LOGD(TAG,"Registering default value for key %s, value %s", "airplay_port", CONFIG_AIRPLAY_PORT);
  280. config_set_default(NVS_TYPE_STR, "airplay_port", CONFIG_AIRPLAY_PORT, 0);
  281. ESP_LOGD(TAG,"Registering default value for key %s, value %s", "a2dp_dev_name", CONFIG_A2DP_DEV_NAME);
  282. config_set_default(NVS_TYPE_STR, "a2dp_dev_name", CONFIG_A2DP_DEV_NAME, 0);
  283. ESP_LOGD(TAG,"Registering default value for key %s, value %s", "bypass_wm", "0");
  284. config_set_default(NVS_TYPE_STR, "bypass_wm", "0", 0);
  285. ESP_LOGD(TAG,"Registering default Audio control board type %s, value ","actrls_config");
  286. config_set_default(NVS_TYPE_STR, "actrls_config", "", 0);
  287. ESP_LOGD(TAG,"Registering default value for key %s", "lms_ctrls_raw");
  288. config_set_default(NVS_TYPE_STR, "lms_ctrls_raw", "n", 0);
  289. ESP_LOGD(TAG,"Registering default Audio control board type %s, value %s", "rotary_config", CONFIG_ROTARY_ENCODER);
  290. config_set_default(NVS_TYPE_STR, "rotary_config", CONFIG_ROTARY_ENCODER, 0);
  291. char number_buffer[101] = {};
  292. snprintf(number_buffer,sizeof(number_buffer)-1,"%u",OTA_FLASH_ERASE_BLOCK);
  293. ESP_LOGD(TAG,"Registering default value for key %s, value %s", "ota_erase_blk", number_buffer);
  294. config_set_default(NVS_TYPE_STR, "ota_erase_blk", number_buffer, 0);
  295. snprintf(number_buffer,sizeof(number_buffer)-1,"%u",OTA_STACK_SIZE);
  296. ESP_LOGD(TAG,"Registering default value for key %s, value %s", "ota_stack", number_buffer);
  297. config_set_default(NVS_TYPE_STR, "ota_stack", number_buffer, 0);
  298. snprintf(number_buffer,sizeof(number_buffer)-1,"%d",OTA_TASK_PRIOTITY);
  299. ESP_LOGD(TAG,"Registering default value for key %s, value %s", "ota_prio", number_buffer);
  300. config_set_default(NVS_TYPE_STR, "ota_prio", number_buffer, 0);
  301. ESP_LOGD(TAG,"Registering default value for key %s, value %s", "enable_bt_sink", STR(CONFIG_BT_SINK));
  302. config_set_default(NVS_TYPE_STR, "enable_bt_sink", STR(CONFIG_BT_SINK), 0);
  303. ESP_LOGD(TAG,"Registering default value for key %s, value %s", "enable_airplay", STR(CONFIG_AIRPLAY_SINK));
  304. config_set_default(NVS_TYPE_STR, "enable_airplay", STR(CONFIG_AIRPLAY_SINK), 0);
  305. ESP_LOGD(TAG,"Registering default value for key %s, value %s", "display_config", CONFIG_DISPLAY_CONFIG);
  306. config_set_default(NVS_TYPE_STR, "display_config", CONFIG_DISPLAY_CONFIG, 0);
  307. ESP_LOGD(TAG,"Registering default value for key %s, value %s", "i2c_config", CONFIG_I2C_CONFIG);
  308. config_set_default(NVS_TYPE_STR, "i2c_config", CONFIG_I2C_CONFIG, 0);
  309. ESP_LOGD(TAG,"Registering default value for key %s, value %s", "spi_config", CONFIG_SPI_CONFIG);
  310. config_set_default(NVS_TYPE_STR, "spi_config", CONFIG_SPI_CONFIG, 0);
  311. ESP_LOGD(TAG,"Registering default value for key %s, value %s", "set_GPIO", CONFIG_SET_GPIO);
  312. config_set_default(NVS_TYPE_STR, "set_GPIO", CONFIG_SET_GPIO, 0);
  313. ESP_LOGD(TAG,"Registering default value for key %s", "led_brightness");
  314. config_set_default(NVS_TYPE_STR, "led_brightness", "", 0);
  315. ESP_LOGD(TAG,"Registering default value for key %s", "spdif_config");
  316. config_set_default(NVS_TYPE_STR, "spdif_config", "", 0);
  317. ESP_LOGD(TAG,"Registering default value for key %s", "dac_config");
  318. config_set_default(NVS_TYPE_STR, "dac_config", "", 0);
  319. //todo: add dac_config for known targets
  320. ESP_LOGD(TAG,"Registering default value for key %s", "dac_controlset");
  321. config_set_default(NVS_TYPE_STR, "dac_controlset", "", 0);
  322. ESP_LOGD(TAG,"Registering default value for key %s", "jack_mutes_amp");
  323. config_set_default(NVS_TYPE_STR, "jack_mutes_amp", "n", 0);
  324. ESP_LOGD(TAG,"Registering default value for key %s", "bat_config");
  325. config_set_default(NVS_TYPE_STR, "bat_config", "", 0);
  326. ESP_LOGD(TAG,"Registering default value for key %s", "metadata_config");
  327. config_set_default(NVS_TYPE_STR, "metadata_config", "", 0);
  328. ESP_LOGD(TAG,"Registering default value for key %s", "telnet_enable");
  329. config_set_default(NVS_TYPE_STR, "telnet_enable", "", 0);
  330. ESP_LOGD(TAG,"Registering default value for key %s", "telnet_buffer");
  331. config_set_default(NVS_TYPE_STR, "telnet_buffer", "40000", 0);
  332. ESP_LOGD(TAG,"Registering default value for key %s", "telnet_block");
  333. config_set_default(NVS_TYPE_STR, "telnet_block", "500", 0);
  334. ESP_LOGD(TAG,"Registering default value for key %s", "stats");
  335. config_set_default(NVS_TYPE_STR, "stats", "n", 0);
  336. ESP_LOGD(TAG,"Registering default value for key %s", "rel_api");
  337. config_set_default(NVS_TYPE_STR, "rel_api", CONFIG_RELEASE_API, 0);
  338. wait_for_commit();
  339. ESP_LOGD(TAG,"Done setting default values in nvs.");
  340. }
  341. uint32_t halSTORAGE_RebootCounterRead(void) { return RebootCounter ; }
  342. uint32_t halSTORAGE_RebootCounterUpdate(int32_t xValue) { return (RebootCounter = (xValue != 0) ? (RebootCounter + xValue) : 0) ; }
  343. void handle_ap_connect(){
  344. start_telnet(NULL);
  345. halSTORAGE_RebootCounterUpdate(0);
  346. }
  347. void app_main()
  348. {
  349. const esp_partition_t *running = esp_ota_get_running_partition();
  350. is_recovery_running = (running->subtype == ESP_PARTITION_SUBTYPE_APP_FACTORY);
  351. esp_reset_reason_t xReason = esp_reset_reason();
  352. ESP_LOGI(TAG,"Reset reason is: %u", xReason);
  353. if(!is_recovery_running && xReason != ESP_RST_SW && xReason != ESP_RST_POWERON ) {
  354. /* unscheduled restart (HW, Watchdog or similar) thus increment dynamic
  355. * counter then log current boot statistics as a warning */
  356. uint32_t Counter = halSTORAGE_RebootCounterUpdate(1) ; // increment counter
  357. ESP_LOGI(TAG,"Reboot counter=%u\n", Counter) ;
  358. if (Counter == 5) {
  359. // before we change the partition, update the info for current running partition.
  360. halSTORAGE_RebootCounterUpdate(0);
  361. guided_factory();
  362. }
  363. }
  364. char * fwurl = NULL;
  365. ESP_LOGI(TAG,"Starting app_main");
  366. initialize_nvs();
  367. ESP_LOGI(TAG,"Setting up telnet.");
  368. init_telnet(); // align on 32 bits boundaries
  369. ESP_LOGI(TAG,"Setting up config subsystem.");
  370. config_init();
  371. ESP_LOGD(TAG,"Creating event group for wifi");
  372. wifi_event_group = xEventGroupCreate();
  373. ESP_LOGD(TAG,"Clearing CONNECTED_BIT from wifi group");
  374. xEventGroupClearBits(wifi_event_group, CONNECTED_BIT);
  375. ESP_LOGI(TAG,"Registering default values");
  376. register_default_nvs();
  377. ESP_LOGI(TAG,"Configuring services");
  378. services_init();
  379. ESP_LOGI(TAG,"Initializing display");
  380. display_init("SqueezeESP32");
  381. if(is_recovery_running && display){
  382. GDS_ClearExt(display, true);
  383. GDS_SetFont(display, &Font_droid_sans_fallback_15x17 );
  384. GDS_TextPos(display, GDS_FONT_MEDIUM, GDS_TEXT_CENTERED, GDS_TEXT_CLEAR | GDS_TEXT_UPDATE, "RECOVERY");
  385. }
  386. ESP_LOGI(TAG,"Checking if certificates need to be updated");
  387. update_certificates(false);
  388. ESP_LOGD(TAG,"Getting firmware OTA URL (if any)");
  389. fwurl = process_ota_url();
  390. ESP_LOGD(TAG,"Getting value for WM bypass, nvs 'bypass_wm'");
  391. char * bypass_wm = config_alloc_get_default(NVS_TYPE_STR, "bypass_wm", "0", 0);
  392. if(bypass_wm==NULL)
  393. {
  394. ESP_LOGE(TAG, "Unable to retrieve the Wifi Manager bypass flag");
  395. bypass_wifi_manager = false;
  396. }
  397. else {
  398. bypass_wifi_manager=(strcmp(bypass_wm,"1")==0 ||strcasecmp(bypass_wm,"y")==0);
  399. }
  400. if(!is_recovery_running){
  401. ESP_LOGD(TAG,"Getting audio control mapping ");
  402. char *actrls_config = config_alloc_get_default(NVS_TYPE_STR, "actrls_config", NULL, 0);
  403. if (actrls_init(actrls_config) == ESP_OK) {
  404. ESP_LOGD(TAG,"Initializing audio control buttons type %s", actrls_config);
  405. } else {
  406. ESP_LOGD(TAG,"No audio control buttons");
  407. }
  408. if (actrls_config) free(actrls_config);
  409. }
  410. /* start the wifi manager */
  411. ESP_LOGD(TAG,"Blinking led");
  412. led_blink_pushed(LED_GREEN, 250, 250);
  413. if(bypass_wifi_manager){
  414. ESP_LOGW(TAG,"*******************************************************************************************");
  415. ESP_LOGW(TAG,"* wifi manager is disabled. Please use wifi commands to connect to your wifi access point.");
  416. ESP_LOGW(TAG,"*******************************************************************************************");
  417. }
  418. else {
  419. ESP_LOGI(TAG,"Starting Wifi Manager");
  420. wifi_manager_start();
  421. wifi_manager_set_callback(EVENT_STA_GOT_IP, &cb_connection_got_ip);
  422. wifi_manager_set_callback(EVENT_STA_DISCONNECTED, &cb_connection_sta_disconnected);
  423. /* Start the telnet service after we are certain that the network stack has been properly initialized.
  424. * This can be either after we're started the AP mode, or after we've started the STA mode */
  425. wifi_manager_set_callback(ORDER_START_AP, &handle_ap_connect);
  426. wifi_manager_set_callback(ORDER_CONNECT_STA, &handle_ap_connect);
  427. }
  428. console_start();
  429. if(fwurl && strlen(fwurl)>0){
  430. if(is_recovery_running){
  431. while(!bWifiConnected){
  432. wait_for_wifi();
  433. taskYIELD();
  434. }
  435. ESP_LOGI(TAG,"Updating firmware from link: %s",fwurl);
  436. start_ota(fwurl, NULL, 0);
  437. }
  438. else {
  439. ESP_LOGE(TAG,"Restarted to application partition. We're not going to perform OTA!");
  440. }
  441. free(fwurl);
  442. }
  443. messaging_post_message(MESSAGING_INFO,MESSAGING_CLASS_SYSTEM,"System started");
  444. }