319 lines
11 KiB
C
319 lines
11 KiB
C
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <sdkconfig.h>
|
|
#include <freertos/FreeRTOS.h>
|
|
#include <freertos/task.h>
|
|
#include <freertos/queue.h>
|
|
#include <esp_system.h>
|
|
#include <esp_err.h>
|
|
#include <esp_log.h>
|
|
#include <nvs_flash.h>
|
|
#include <nvs.h>
|
|
#include "appfs.h"
|
|
#include "ili9341.h"
|
|
#include "pax_gfx.h"
|
|
#include "menu.h"
|
|
#include "rp2040.h"
|
|
#include "appfs_wrapper.h"
|
|
#include "hardware.h"
|
|
#include "system_wrapper.h"
|
|
#include "bootscreen.h"
|
|
#include "wifi_connect.h"
|
|
#include "wifi_connection.h"
|
|
#include "wifi_ota.h"
|
|
#include "graphics_wrapper.h"
|
|
|
|
static const char *TAG = "wifi menu";
|
|
|
|
typedef enum action {
|
|
ACTION_NONE,
|
|
ACTION_BACK,
|
|
ACTION_SHOW,
|
|
ACTION_SCAN,
|
|
ACTION_MANUAL
|
|
} menu_wifi_action_t;
|
|
|
|
void render_wifi_help(pax_buf_t* pax_buffer) {
|
|
pax_background(pax_buffer, 0xFFFFFF);
|
|
pax_noclip(pax_buffer);
|
|
pax_draw_text(pax_buffer, 0xFF000000, NULL, 18, 5, 240 - 19, "[A] accept [B] back");
|
|
}
|
|
|
|
void wifi_show(xQueueHandle buttonQueue, pax_buf_t* pax_buffer, ILI9341* ili9341);
|
|
void wifi_setup(xQueueHandle buttonQueue, pax_buf_t* pax_buffer, ILI9341* ili9341, bool scan);
|
|
wifi_ap_record_t *wifi_scan_results(xQueueHandle buttonQueue, pax_buf_t* pax_buffer, ILI9341* ili9341, size_t num_aps, wifi_ap_record_t *aps);
|
|
|
|
void menu_wifi(xQueueHandle buttonQueue, pax_buf_t* pax_buffer, ILI9341* ili9341) {
|
|
menu_t* menu = menu_alloc("WiFi configuration");
|
|
menu_insert_item(menu, "Show current settings", NULL, (void*) ACTION_SHOW, -1);
|
|
menu_insert_item(menu, "Scan for networks", NULL, (void*) ACTION_SCAN, -1);
|
|
menu_insert_item(menu, "Configure manually", NULL, (void*) ACTION_MANUAL, -1);
|
|
|
|
bool render = true;
|
|
menu_wifi_action_t action = ACTION_NONE;
|
|
|
|
render_wifi_help(pax_buffer);
|
|
|
|
while (1) {
|
|
rp2040_input_message_t buttonMessage = {0};
|
|
if (xQueueReceive(buttonQueue, &buttonMessage, 16 / portTICK_PERIOD_MS) == pdTRUE) {
|
|
uint8_t pin = buttonMessage.input;
|
|
bool value = buttonMessage.state;
|
|
switch(pin) {
|
|
case RP2040_INPUT_JOYSTICK_DOWN:
|
|
if (value) {
|
|
menu_navigate_next(menu);
|
|
render = true;
|
|
}
|
|
break;
|
|
case RP2040_INPUT_JOYSTICK_UP:
|
|
if (value) {
|
|
menu_navigate_previous(menu);
|
|
render = true;
|
|
}
|
|
break;
|
|
case RP2040_INPUT_BUTTON_HOME:
|
|
case RP2040_INPUT_BUTTON_BACK:
|
|
if (value) {
|
|
action = ACTION_BACK;
|
|
}
|
|
break;
|
|
case RP2040_INPUT_BUTTON_ACCEPT:
|
|
case RP2040_INPUT_JOYSTICK_PRESS:
|
|
case RP2040_INPUT_BUTTON_SELECT:
|
|
case RP2040_INPUT_BUTTON_START:
|
|
if (value) {
|
|
action = (menu_wifi_action_t) menu_get_callback_args(menu, menu_get_position(menu));
|
|
}
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (render) {
|
|
menu_render(pax_buffer, menu, 0, 0, 320, 220, 0xFF2f55a8);
|
|
ili9341_write(ili9341, pax_buffer->buf);
|
|
render = false;
|
|
}
|
|
|
|
if (action != ACTION_NONE) {
|
|
if (action == ACTION_SHOW) {
|
|
wifi_show(buttonQueue, pax_buffer, ili9341);
|
|
} else if (action == ACTION_SCAN) {
|
|
wifi_setup(buttonQueue, pax_buffer, ili9341, true);
|
|
} else if (action == ACTION_MANUAL) {
|
|
wifi_setup(buttonQueue, pax_buffer, ili9341, false);
|
|
} else if (action == ACTION_BACK) {
|
|
break;
|
|
}
|
|
render = true;
|
|
action = ACTION_NONE;
|
|
render_wifi_help(pax_buffer);
|
|
}
|
|
}
|
|
|
|
menu_free(menu);
|
|
}
|
|
|
|
void wifi_show(xQueueHandle buttonQueue, pax_buf_t* pax_buffer, ILI9341* ili9341) {
|
|
nvs_handle_t handle;
|
|
|
|
nvs_open("system", NVS_READWRITE, &handle);
|
|
char ssid[33] = "<not set>";
|
|
char password[33] = "<not set>";
|
|
size_t requiredSize;
|
|
|
|
esp_err_t res = nvs_get_str(handle, "wifi.ssid", NULL, &requiredSize);
|
|
if ((res == ESP_OK) && (requiredSize < sizeof(ssid))) {
|
|
res = nvs_get_str(handle, "wifi.ssid", ssid, &requiredSize);
|
|
}
|
|
|
|
res = nvs_get_str(handle, "wifi.password", NULL, &requiredSize);
|
|
if ((res == ESP_OK) && (requiredSize < sizeof(password))) {
|
|
res = nvs_get_str(handle, "wifi.password", password, &requiredSize);
|
|
}
|
|
|
|
nvs_close(handle);
|
|
|
|
char buffer[300];
|
|
pax_noclip(pax_buffer);
|
|
pax_background(pax_buffer, 0xFFFFFF);
|
|
snprintf(buffer, sizeof(buffer), "WiFi SSID:\n%s\nWiFi password:\n%s", ssid, password);
|
|
pax_draw_text(pax_buffer, 0xFF000000, NULL, 18, 0, 0, buffer);
|
|
ili9341_write(ili9341, pax_buffer->buf);
|
|
|
|
bool quit = false;
|
|
while (!quit) {
|
|
rp2040_input_message_t buttonMessage = {0};
|
|
if (xQueueReceive(buttonQueue, &buttonMessage, 16 / portTICK_PERIOD_MS) == pdTRUE) {
|
|
uint8_t pin = buttonMessage.input;
|
|
bool value = buttonMessage.state;
|
|
switch(pin) {
|
|
case RP2040_INPUT_BUTTON_HOME:
|
|
case RP2040_INPUT_BUTTON_BACK:
|
|
case RP2040_INPUT_BUTTON_ACCEPT:
|
|
case RP2040_INPUT_JOYSTICK_PRESS:
|
|
case RP2040_INPUT_BUTTON_SELECT:
|
|
case RP2040_INPUT_BUTTON_START:
|
|
if (value) quit = true;
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
wifi_ap_record_t *wifi_scan_results(xQueueHandle buttonQueue, pax_buf_t* pax_buffer, ILI9341* ili9341, size_t num_aps, wifi_ap_record_t *aps) {
|
|
menu_t *menu = menu_alloc("Select network");
|
|
wifi_ap_record_t *picked = NULL;
|
|
|
|
render_wifi_help(pax_buffer);
|
|
|
|
for (size_t i = 0; i < num_aps; i++) {
|
|
menu_insert_item(menu, (const char*) aps[i].ssid, NULL, (void *) (i + 1), -1);
|
|
}
|
|
|
|
bool render = true;
|
|
size_t selection = 0;
|
|
while (1) {
|
|
rp2040_input_message_t buttonMessage = {0};
|
|
selection = -1;
|
|
if (xQueueReceive(buttonQueue, &buttonMessage, 16 / portTICK_PERIOD_MS) == pdTRUE) {
|
|
uint8_t pin = buttonMessage.input;
|
|
bool value = buttonMessage.state;
|
|
switch(pin) {
|
|
case RP2040_INPUT_JOYSTICK_DOWN:
|
|
if (value) {
|
|
menu_navigate_next(menu);
|
|
render = true;
|
|
}
|
|
break;
|
|
case RP2040_INPUT_JOYSTICK_UP:
|
|
if (value) {
|
|
menu_navigate_previous(menu);
|
|
render = true;
|
|
}
|
|
break;
|
|
case RP2040_INPUT_BUTTON_HOME:
|
|
case RP2040_INPUT_BUTTON_BACK:
|
|
if (value) {
|
|
selection = 0;
|
|
}
|
|
break;
|
|
case RP2040_INPUT_BUTTON_ACCEPT:
|
|
case RP2040_INPUT_JOYSTICK_PRESS:
|
|
case RP2040_INPUT_BUTTON_SELECT:
|
|
case RP2040_INPUT_BUTTON_START:
|
|
if (value) {
|
|
selection = (size_t) menu_get_callback_args(menu, menu_get_position(menu));
|
|
}
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (render) {
|
|
menu_render(pax_buffer, menu, 0, 0, 320, 220, 0xFF2f55a8);
|
|
ili9341_write(ili9341, pax_buffer->buf);
|
|
render = false;
|
|
}
|
|
|
|
if (selection != (size_t) -1) {
|
|
if (selection == 0) {
|
|
break;
|
|
} else {
|
|
// You picked one, yay!
|
|
picked = &aps[selection-1];
|
|
break;
|
|
}
|
|
render = true;
|
|
selection = -1;
|
|
render_wifi_help(pax_buffer);
|
|
}
|
|
}
|
|
|
|
menu_free(menu);
|
|
return picked;
|
|
}
|
|
|
|
// Sorts WiFi APs by RSSI (best RSSI first in the list).
|
|
static int wifi_ap_sorter(const void *a0, const void *b0) {
|
|
const wifi_ap_record_t *a = a0;
|
|
const wifi_ap_record_t *b = b0;
|
|
return b->rssi - a->rssi;
|
|
}
|
|
|
|
void wifi_setup(xQueueHandle buttonQueue, pax_buf_t* pax_buffer, ILI9341* ili9341, bool scan) {
|
|
|
|
char ssid[33] = {0};
|
|
char password[33] = {0};
|
|
nvs_handle_t handle;
|
|
nvs_open("system", NVS_READWRITE, &handle);
|
|
bool accepted = true;
|
|
|
|
/* ==== scanning phase ==== */
|
|
if (scan) {
|
|
// Scan for networks.
|
|
wifi_ap_record_t *aps;
|
|
size_t n_aps = wifi_scan(&aps);
|
|
|
|
// Sort them by RSSI.
|
|
qsort(aps, n_aps, sizeof(wifi_ap_record_t), wifi_ap_sorter);
|
|
|
|
// Make a de-duplicated list.
|
|
wifi_ap_record_t *dedup = malloc(sizeof(wifi_ap_record_t) * n_aps);
|
|
size_t n_dedup = 0;
|
|
for (size_t i = 0; i < n_aps; ) {
|
|
for (size_t x = 0; x < n_dedup; x++) {
|
|
if (!strcmp((const char *) aps[i].ssid, (const char *) dedup[x].ssid)) goto cont;
|
|
}
|
|
dedup[n_dedup] = aps[i];
|
|
n_dedup ++;
|
|
cont:
|
|
i++;
|
|
}
|
|
|
|
// Open a little menu for picking a network.
|
|
wifi_ap_record_t *pick = wifi_scan_results(buttonQueue, pax_buffer, ili9341, n_dedup, dedup);
|
|
if (!pick) {
|
|
nvs_close(handle);
|
|
return;
|
|
}
|
|
// Copy the SSID in.
|
|
memcpy(ssid, pick->ssid, 33);
|
|
} else {
|
|
size_t requiredSize;
|
|
esp_err_t res = nvs_get_str(handle, "wifi.ssid", NULL, &requiredSize);
|
|
if (res != ESP_OK) {
|
|
strcpy(ssid, "");
|
|
strcpy(password, "");
|
|
} else if (requiredSize < sizeof(ssid)) {
|
|
res = nvs_get_str(handle, "wifi.ssid", ssid, &requiredSize);
|
|
if (res != ESP_OK) strcpy(ssid, "");
|
|
res = nvs_get_str(handle, "wifi.password", NULL, &requiredSize);
|
|
if (res != ESP_OK) {
|
|
strcpy(password, "");
|
|
} else if (requiredSize < sizeof(password)) {
|
|
res = nvs_get_str(handle, "wifi.password", password, &requiredSize);
|
|
if (res != ESP_OK) strcpy(password, "");
|
|
}
|
|
}
|
|
accepted = keyboard(buttonQueue, pax_buffer, ili9341, 30, 30, pax_buffer->width - 60, pax_buffer->height - 60, "WiFi SSID", "Press HOME to exit", ssid, sizeof(ssid));
|
|
}
|
|
|
|
/* ==== manual entering phase ==== */
|
|
if (accepted) {
|
|
accepted = keyboard(buttonQueue, pax_buffer, ili9341, 30, 30, pax_buffer->width - 60, pax_buffer->height - 60, "WiFi password", "Press HOME to exit", password, sizeof(password));
|
|
}
|
|
if (accepted) {
|
|
nvs_set_str(handle, "wifi.ssid", ssid);
|
|
nvs_set_str(handle, "wifi.password", password);
|
|
display_boot_screen(pax_buffer, ili9341, "WiFi settings stored");
|
|
}
|
|
nvs_set_u8(handle, "wifi.use_ent", 0);
|
|
nvs_close(handle);
|
|
}
|
|
|