mirrors
/
blackmagic-esp32-s2
mirror of https://github.com/flipperdevices/blackmagic-esp32-s2.git
1
0
Fork 0
Code Releases Wiki Activity
blackmagic-esp32-s2/main/network-http.c

643 lines
20 KiB
C
Raw Blame History

#include <esp_http_server.h>
#include <svelte-portal.h>
#include <esp_log.h>
#include <cJSON.h>
#include <esp_wifi.h>
#include <esp_system.h>
#include "network.h"
#include "nvs.h"
#include "nvs-config.h"
#include "led.h"
#include "helpers.h"
#define TAG "network-http"
#define JSON_ERROR(error_text) "{\"error\": \"" error_text "\"}"
#define JSON_RESULT(result_text) "{\"result\": \"" result_text "\"}"
#define WIFI_SCAN_SIZE 20
static const char* get_auth_mode(int authmode) {
switch(authmode) {
case WIFI_AUTH_OPEN:
return "OPEN";
case WIFI_AUTH_WEP:
return "WEP";
case WIFI_AUTH_WPA_PSK:
return "WPA_PSK";
case WIFI_AUTH_WPA2_PSK:
return "WPA2_PSK";
case WIFI_AUTH_WPA_WPA2_PSK:
return "WPA_WPA2_PSK";
case WIFI_AUTH_WPA2_ENTERPRISE:
return "WPA2_ENTERPRISE";
case WIFI_AUTH_WPA3_PSK:
return "WPA3_PSK";
case WIFI_AUTH_WPA2_WPA3_PSK:
return "WPA2_WPA3_PSK";
default:
return "UNKNOWN";
}
}
static const char* get_pairwise_cipher(int pairwise_cipher) {
switch(pairwise_cipher) {
case WIFI_CIPHER_TYPE_NONE:
return "NONE";
case WIFI_CIPHER_TYPE_WEP40:
return "WEP40";
case WIFI_CIPHER_TYPE_WEP104:
return "WEP104";
case WIFI_CIPHER_TYPE_TKIP:
return "TKIP";
case WIFI_CIPHER_TYPE_CCMP:
return "CCMP";
case WIFI_CIPHER_TYPE_TKIP_CCMP:
return "TKIP_CCMP";
default:
return "UNKNOWN";
}
}
static const char* get_group_cipher(int group_cipher) {
switch(group_cipher) {
case WIFI_CIPHER_TYPE_NONE:
return "NONE";
case WIFI_CIPHER_TYPE_WEP40:
return "WEP40";
case WIFI_CIPHER_TYPE_WEP104:
return "WEP104";
case WIFI_CIPHER_TYPE_TKIP:
return "TKIP";
case WIFI_CIPHER_TYPE_CCMP:
return "CCMP";
case WIFI_CIPHER_TYPE_TKIP_CCMP:
return "TKIP_CCMP";
default:
return "UNKNOWN";
}
}
typedef struct {
const char* uri;
const char* type;
const uint8_t* addr_start;
const uint8_t* addr_end;
} HttpResource;
const HttpResource http_resources[] = {
{
.uri = "/",
.type = "text/html",
.addr_start = index_html_start,
.addr_end = index_html_end,
},
{
.uri = "/index.html",
.type = "text/html",
.addr_start = index_html_start,
.addr_end = index_html_end,
},
{
.uri = "/build/bundle.css",
.type = "text/css",
.addr_start = build_bundle_css_start,
.addr_end = build_bundle_css_end,
},
{
.uri = "/build/bundle.js",
.type = "application/javascript",
.addr_start = build_bundle_js_start,
.addr_end = build_bundle_js_end,
},
{
.uri = "/assets/ega8.otf",
.type = "application/x-font-opentype",
.addr_start = assets_ega8_otf_start,
.addr_end = assets_ega8_otf_end,
},
{
.uri = "/assets/favicon.ico",
.type = "image/x-icon",
.addr_start = assets_favicon_ico_start,
.addr_end = assets_favicon_ico_end,
},
{
.uri = "/favicon.ico",
.type = "image/x-icon",
.addr_start = assets_favicon_ico_start,
.addr_end = assets_favicon_ico_end,
},
};
static esp_err_t http_common_get_handler(httpd_req_t* req) {
const uint8_t* file_start = NULL;
const uint8_t* file_end = NULL;
for(size_t i = 0; i < COUNT_OF(http_resources); i++) {
if(strcmp(http_resources[i].uri, req->uri) == 0) {
file_start = http_resources[i].addr_start;
file_end = http_resources[i].addr_end;
httpd_resp_set_type(req, http_resources[i].type);
break;
}
}
if(file_start == NULL) {
ESP_LOGE(TAG, "file not exist: %s", req->uri);
httpd_resp_send_err(req, HTTPD_404_NOT_FOUND, "file not exist");
return ESP_FAIL;
}
esp_err_t err =
httpd_resp_send_chunk(req, (const char*)file_start, (ssize_t)(file_end - file_start));
if(err != ESP_OK) {
ESP_LOGE(TAG, "file sending failed, %d", err);
// Abort
httpd_resp_sendstr_chunk(req, NULL);
httpd_resp_send_err(req, HTTPD_500_INTERNAL_SERVER_ERROR, "file sending failed");
return ESP_FAIL;
} else {
// HTTP response completion
httpd_resp_send_chunk(req, NULL, 0);
}
ESP_LOGI(TAG, "file sending complete");
return ESP_OK;
}
static esp_err_t system_ping_handler(httpd_req_t* req) {
httpd_resp_set_hdr(req, "Access-Control-Allow-Origin", "*");
httpd_resp_set_type(req, "application/json");
httpd_resp_sendstr(req, JSON_RESULT("OK"));
return ESP_OK;
}
static esp_err_t system_tasks_handler(httpd_req_t* req) {
httpd_resp_set_hdr(req, "Access-Control-Allow-Origin", "*");
httpd_resp_set_type(req, "application/json");
uint32_t task_count = uxTaskGetNumberOfTasks();
TaskStatus_t* tasks = malloc(task_count * sizeof(TaskStatus_t));
task_count = uxTaskGetSystemState(tasks, task_count, NULL);
cJSON* root = cJSON_CreateObject();
cJSON_AddNumberToObject(root, "count", task_count);
cJSON* array = cJSON_AddArrayToObject(root, "list");
for(uint32_t i = 0; i < task_count; i++) {
cJSON* object = cJSON_CreateObject();
cJSON_AddNumberToObject(object, "handle", (uint32_t)tasks[i].xHandle);
cJSON_AddStringToObject(object, "name", tasks[i].pcTaskName);
cJSON_AddNumberToObject(object, "number", (uint32_t)tasks[i].xTaskNumber);
switch(tasks[i].eCurrentState) {
case eRunning:
cJSON_AddStringToObject(object, "state", "Running");
break;
case eReady:
cJSON_AddStringToObject(object, "state", "Ready");
break;
case eBlocked:
cJSON_AddStringToObject(object, "state", "Blocked");
break;
case eSuspended:
cJSON_AddStringToObject(object, "state", "Suspended");
break;
case eDeleted:
cJSON_AddStringToObject(object, "state", "Deleted");
break;
case eInvalid:
cJSON_AddStringToObject(object, "state", "Invalid");
break;
}
cJSON_AddNumberToObject(object, "current_priority", (uint32_t)tasks[i].uxCurrentPriority);
cJSON_AddNumberToObject(object, "base_priority", (uint32_t)tasks[i].uxBasePriority);
cJSON_AddNumberToObject(object, "runtime", (uint32_t)tasks[i].ulRunTimeCounter);
cJSON_AddNumberToObject(object, "stack_base", (uint32_t)tasks[i].pxStackBase);
cJSON_AddNumberToObject(object, "watermark", (uint32_t)tasks[i].usStackHighWaterMark);
cJSON_AddItemToArray(array, object);
}
const char* json_text = cJSON_Print(root);
httpd_resp_sendstr(req, json_text);
free((void*)json_text);
cJSON_Delete(root);
free((void*)tasks);
return ESP_OK;
}
static esp_err_t system_reboot(httpd_req_t* req) {
httpd_resp_sendstr(req, JSON_RESULT("OK"));
esp_restart();
return ESP_OK;
}
static esp_err_t system_info_get_handler(httpd_req_t* req) {
httpd_resp_set_hdr(req, "Access-Control-Allow-Origin", "*");
httpd_resp_set_type(req, "application/json");
cJSON* root = cJSON_CreateObject();
esp_chip_info_t chip_info;
esp_chip_info(&chip_info);
cJSON_AddStringToObject(root, "idf_version", IDF_VER);
switch(chip_info.model) {
case CHIP_ESP32:
cJSON_AddStringToObject(root, "model", "ESP32");
break;
case CHIP_ESP32S2:
cJSON_AddStringToObject(root, "model", "ESP32-S2");
break;
case CHIP_ESP32H2:
cJSON_AddStringToObject(root, "model", "ESP32-H2");
break;
case CHIP_ESP32S3:
cJSON_AddStringToObject(root, "model", "ESP32-S3");
break;
case CHIP_ESP32C3:
cJSON_AddStringToObject(root, "model", "ESP32-C3");
break;
default:
cJSON_AddStringToObject(root, "model", "UNKNOWN");
break;
}
cJSON_AddNumberToObject(root, "revision", chip_info.revision);
cJSON_AddNumberToObject(root, "cores", chip_info.cores);
multi_heap_info_t info;
heap_caps_get_info(&info, MALLOC_CAP_DEFAULT);
cJSON* heap = cJSON_AddObjectToObject(root, "heap");
cJSON_AddNumberToObject(heap, "total_free_bytes", info.total_free_bytes);
cJSON_AddNumberToObject(heap, "total_allocated_bytes", info.total_allocated_bytes);
cJSON_AddNumberToObject(heap, "largest_free_block", info.largest_free_block);
cJSON_AddNumberToObject(heap, "minimum_free_bytes", info.minimum_free_bytes);
// ip addr
cJSON_AddNumberToObject(root, "ip", network_get_ip());
// mac
uint8_t mac_addr[6] = {0};
ESP_ERROR_CHECK(esp_read_mac(mac_addr, ESP_MAC_WIFI_STA));
cJSON* mac = cJSON_AddArrayToObject(root, "mac");
cJSON_AddItemToArray(mac, cJSON_CreateNumber(mac_addr[0]));
cJSON_AddItemToArray(mac, cJSON_CreateNumber(mac_addr[1]));
cJSON_AddItemToArray(mac, cJSON_CreateNumber(mac_addr[2]));
cJSON_AddItemToArray(mac, cJSON_CreateNumber(mac_addr[3]));
cJSON_AddItemToArray(mac, cJSON_CreateNumber(mac_addr[4]));
cJSON_AddItemToArray(mac, cJSON_CreateNumber(mac_addr[5]));
const char* json_text = cJSON_Print(root);
httpd_resp_sendstr(req, json_text);
free((void*)json_text);
cJSON_Delete(root);
return ESP_OK;
}
static esp_err_t wifi_get_credenitals_handler(httpd_req_t* req) {
httpd_resp_set_hdr(req, "Access-Control-Allow-Origin", "*");
httpd_resp_set_type(req, "application/json");
cJSON* root = cJSON_CreateObject();
mstring_t* ap_ssid = mstring_alloc();
mstring_t* ap_pass = mstring_alloc();
mstring_t* sta_ssid = mstring_alloc();
mstring_t* sta_pass = mstring_alloc();
WiFiMode wifi_mode;
nvs_config_get_wifi_mode(&wifi_mode);
nvs_config_get_ap_ssid(ap_ssid);
nvs_config_get_ap_pass(ap_pass);
nvs_config_get_sta_ssid(sta_ssid);
nvs_config_get_sta_pass(sta_pass);
cJSON_AddStringToObject(root, "ap_ssid", mstring_get_cstr(ap_ssid));
cJSON_AddStringToObject(root, "ap_pass", mstring_get_cstr(ap_pass));
cJSON_AddStringToObject(root, "sta_ssid", mstring_get_cstr(sta_ssid));
cJSON_AddStringToObject(root, "sta_pass", mstring_get_cstr(sta_pass));
switch(wifi_mode) {
case WiFiModeAP:
cJSON_AddStringToObject(root, "wifi_mode", CFG_WIFI_MODE_AP);
break;
case WiFiModeSTA:
cJSON_AddStringToObject(root, "wifi_mode", CFG_WIFI_MODE_STA);
break;
}
const char* json_text = cJSON_Print(root);
httpd_resp_sendstr(req, json_text);
free((void*)json_text);
cJSON_Delete(root);
mstring_free(ap_ssid);
mstring_free(ap_pass);
mstring_free(sta_ssid);
mstring_free(sta_pass);
return ESP_OK;
}
static esp_err_t wifi_set_credenitals_handler(httpd_req_t* req) {
httpd_resp_set_hdr(req, "Access-Control-Allow-Origin", "*");
int total_length = req->content_len;
int cur_len = 0;
char* buffer = malloc(256);
mstring_t* ap_ssid = mstring_alloc();
mstring_t* ap_pass = mstring_alloc();
mstring_t* sta_ssid = mstring_alloc();
mstring_t* sta_pass = mstring_alloc();
mstring_t* wifi_mode = mstring_alloc();
const char* error_text = JSON_ERROR("unknown error");
int received = 0;
httpd_resp_set_type(req, "application/json");
if(total_length >= 256) {
error_text = JSON_ERROR("request too long");
goto err_fail;
}
while(cur_len < total_length) {
received = httpd_req_recv(req, buffer + cur_len, total_length);
if(received <= 0) {
error_text = JSON_ERROR("cannot receive request data");
goto err_fail;
}
cur_len += received;
}
buffer[total_length] = '\0';
cJSON* root = cJSON_Parse(buffer);
if(cJSON_GetObjectItem(root, "ap_ssid") != NULL) {
mstring_set(ap_ssid, cJSON_GetObjectItem(root, "ap_ssid")->valuestring);
} else {
cJSON_Delete(root);
error_text = JSON_ERROR("request dont have [ap_ssid] field");
goto err_fail;
}
if(cJSON_GetObjectItem(root, "ap_pass") != NULL) {
mstring_set(ap_pass, cJSON_GetObjectItem(root, "ap_pass")->valuestring);
} else {
cJSON_Delete(root);
error_text = JSON_ERROR("request dont have [ap_pass] field");
goto err_fail;
}
if(cJSON_GetObjectItem(root, "sta_ssid") != NULL) {
mstring_set(sta_ssid, cJSON_GetObjectItem(root, "sta_ssid")->valuestring);
} else {
cJSON_Delete(root);
error_text = JSON_ERROR("request dont have [sta_ssid] field");
goto err_fail;
}
if(cJSON_GetObjectItem(root, "sta_pass") != NULL) {
mstring_set(sta_pass, cJSON_GetObjectItem(root, "sta_pass")->valuestring);
} else {
cJSON_Delete(root);
error_text = JSON_ERROR("request dont have [sta_pass] field");
goto err_fail;
}
if(cJSON_GetObjectItem(root, "wifi_mode") != NULL) {
mstring_set(wifi_mode, cJSON_GetObjectItem(root, "wifi_mode")->valuestring);
} else {
cJSON_Delete(root);
error_text = JSON_ERROR("request dont have [wifi_mode] field");
goto err_fail;
}
cJSON_Delete(root);
if(strcmp(mstring_get_cstr(wifi_mode), CFG_WIFI_MODE_AP) != 0 &&
strcmp(mstring_get_cstr(wifi_mode), CFG_WIFI_MODE_STA) != 0) {
error_text = JSON_ERROR("invalid value in [wifi_mode]");
goto err_fail;
}
if(nvs_config_set_ap_ssid(ap_ssid) != ESP_OK) {
error_text = JSON_ERROR("invalid value in [ap_ssid]");
goto err_fail;
}
if(nvs_config_set_ap_pass(ap_pass) != ESP_OK) {
error_text = JSON_ERROR("invalid value in [ap_pass]");
goto err_fail;
}
if(nvs_config_set_sta_ssid(sta_ssid) != ESP_OK) {
error_text = JSON_ERROR("invalid value in [sta_ssid]");
goto err_fail;
}
if(nvs_config_set_sta_pass(sta_pass) != ESP_OK) {
error_text = JSON_ERROR("invalid value in [sta_pass]");
goto err_fail;
}
if(strcmp(mstring_get_cstr(wifi_mode), CFG_WIFI_MODE_AP) == 0) {
if(nvs_config_set_wifi_mode(WiFiModeAP) != ESP_OK) {
error_text = JSON_ERROR("invalid value in [sta_pass]");
goto err_fail;
}
} else {
if(nvs_config_set_wifi_mode(WiFiModeSTA) != ESP_OK) {
error_text = JSON_ERROR("invalid value in [sta_pass]");
goto err_fail;
}
}
httpd_resp_sendstr(req, JSON_RESULT("WIFI settings saved"));
return ESP_OK;
err_fail:
httpd_resp_send_err(req, HTTPD_500_INTERNAL_SERVER_ERROR, error_text);
free(buffer);
mstring_free(ap_ssid);
mstring_free(ap_pass);
mstring_free(sta_ssid);
mstring_free(sta_pass);
mstring_free(wifi_mode);
return ESP_FAIL;
}
static esp_err_t wifi_list_get_handler(httpd_req_t* req) {
httpd_resp_set_hdr(req, "Access-Control-Allow-Origin", "*");
uint16_t number = WIFI_SCAN_SIZE;
wifi_ap_record_t* ap_info = calloc(WIFI_SCAN_SIZE, sizeof(wifi_ap_record_t));
uint16_t ap_count = 0;
memset(ap_info, 0, sizeof(wifi_ap_record_t));
esp_wifi_scan_start(NULL, true);
ESP_ERROR_CHECK(esp_wifi_scan_get_ap_records(&number, ap_info));
ESP_ERROR_CHECK(esp_wifi_scan_get_ap_num(&ap_count));
httpd_resp_set_type(req, "application/json");
cJSON* root = cJSON_CreateObject();
cJSON* array = cJSON_AddArrayToObject(root, "net_list");
for(int i = 0; (i < WIFI_SCAN_SIZE) && (i < ap_count); i++) {
cJSON* object = cJSON_CreateObject();
cJSON_AddStringToObject(object, "ssid", (const char*)&ap_info[i].ssid);
cJSON_AddNumberToObject(object, "channel", ap_info[i].primary);
cJSON_AddNumberToObject(object, "rssi", ap_info[i].rssi);
cJSON_AddStringToObject(object, "auth", get_auth_mode(ap_info[i].authmode));
if(ap_info[i].authmode != WIFI_AUTH_WEP) {
cJSON_AddStringToObject(
object, "pairwise_cipher", get_pairwise_cipher(ap_info[i].pairwise_cipher));
cJSON_AddStringToObject(
object, "group_cipher", get_group_cipher(ap_info[i].group_cipher));
}
cJSON_AddItemToArray(array, object);
}
cJSON_AddNumberToObject(root, "total", ap_count);
const char* json_text = cJSON_Print(root);
httpd_resp_sendstr(req, json_text);
free((void*)json_text);
cJSON_Delete(root);
free(ap_info);
return ESP_OK;
}
static esp_err_t gpio_led_set_handler(httpd_req_t* req) {
httpd_resp_set_hdr(req, "Access-Control-Allow-Origin", "*");
int total_length = req->content_len;
int current_length = 0;
char* buffer = malloc(256);
const char* error_text = JSON_ERROR("unknown error");
int received = 0;
int32_t led_red = -1;
int32_t led_green = -1;
int32_t led_blue = -1;
httpd_resp_set_type(req, "application/json");
if(total_length >= 256) {
error_text = JSON_ERROR("request too long");
goto err_fail;
}
while(current_length < total_length) {
received = httpd_req_recv(req, buffer + current_length, total_length);
if(received <= 0) {
error_text = JSON_ERROR("cannot receive request data");
goto err_fail;
}
current_length += received;
}
buffer[total_length] = '\0';
cJSON* root = cJSON_Parse(buffer);
cJSON* element;
element = cJSON_GetObjectItem(root, "red");
if(element != NULL && element->type == cJSON_Number) {
led_red = element->valuedouble;
}
element = cJSON_GetObjectItem(root, "green");
if(element != NULL && element->type == cJSON_Number) {
led_green = element->valuedouble;
}
element = cJSON_GetObjectItem(root, "blue");
if(element != NULL && element->type == cJSON_Number) {
led_blue = element->valuedouble;
}
cJSON_Delete(root);
if((led_red < 0 && led_green < 0 && led_blue < 0) ||
(led_red > 255 || led_green > 255 || led_blue > 255)) {
error_text = JSON_ERROR("[red], [green], [blue] must contain numbers in the range 0-255");
goto err_fail;
}
if(led_red >= 0) {
led_set_red(led_red);
}
if(led_green >= 0) {
led_set_green(led_green);
}
if(led_blue >= 0) {
led_set_blue(led_blue);
}
httpd_resp_sendstr(req, JSON_RESULT("OK"));
return ESP_OK;
err_fail:
httpd_resp_send_err(req, HTTPD_500_INTERNAL_SERVER_ERROR, error_text);
free(buffer);
return ESP_FAIL;
}
const httpd_uri_t uri_handlers[] = {
/*************** SYSTEM ***************/
{.uri = "/api/v1/system/ping",
.method = HTTP_GET,
.handler = system_ping_handler,
.user_ctx = NULL},
{.uri = "/api/v1/system/tasks",
.method = HTTP_GET,
.handler = system_tasks_handler,
.user_ctx = NULL},
{.uri = "/api/v1/system/info",
.method = HTTP_GET,
.handler = system_info_get_handler,
.user_ctx = NULL},
{.uri = "/api/v1/system/reboot",
.method = HTTP_POST,
.handler = system_reboot,
.user_ctx = NULL},
/*************** GPIO ***************/
{.uri = "/api/v1/gpio/led",
.method = HTTP_POST,
.handler = gpio_led_set_handler,
.user_ctx = NULL},
/*************** WIFI ***************/
{.uri = "/api/v1/wifi/list",
.method = HTTP_GET,
.handler = wifi_list_get_handler,
.user_ctx = NULL},
{.uri = "/api/v1/wifi/set_credenitals",
.method = HTTP_POST,
.handler = wifi_set_credenitals_handler,
.user_ctx = NULL},
{.uri = "/api/v1/wifi/get_credenitals",
.method = HTTP_GET,
.handler = wifi_get_credenitals_handler,
.user_ctx = NULL},
/*************** HTTP ***************/
{.uri = "/*", .method = HTTP_GET, .handler = http_common_get_handler, .user_ctx = NULL},
};
void network_http_server_init(void) {
ESP_LOGI(TAG, "init rest server");
httpd_handle_t server = NULL;
httpd_config_t config = HTTPD_DEFAULT_CONFIG();
config.max_uri_handlers = COUNT_OF(uri_handlers);
config.uri_match_fn = httpd_uri_match_wildcard;
ESP_LOGI(TAG, "starting http server");
if(httpd_start(&server, &config) != ESP_OK) {
ESP_LOGE(TAG, "start http server failed");
return;
}
for(size_t i = 0; i < COUNT_OF(uri_handlers); i++) {
ESP_ERROR_CHECK(httpd_register_uri_handler(server, &uri_handlers[i]));
}
ESP_LOGI(TAG, "init rest server done");
}
View Git Blame Copy Permalink