mirror of
https://github.com/cesanta/mongoose.git
synced 2026-10-14 10:20:01 +00:00
246 lines
8.1 KiB
C
246 lines
8.1 KiB
C
// SPDX-FileCopyrightText: 2026 Cesanta Software Limited
|
|
// SPDX-License-Identifier: GPL-2.0-only or commercial
|
|
|
|
#include "mongoose.h"
|
|
|
|
#if MG_ARCH == MG_ARCH_UNIX || MG_ARCH == MG_ARCH_WIN32
|
|
#define HTTP_ADDR "http://0.0.0.0:8000"
|
|
#else
|
|
#define HTTP_ADDR "http://0.0.0.0:80"
|
|
#endif
|
|
|
|
// Control panel
|
|
static bool s_led1, s_led2, s_led3;
|
|
|
|
static struct mg_field fields_leds[] = {
|
|
{"led1", MG_VAL_BOOL, &s_led1, sizeof(s_led1)},
|
|
{"led2", MG_VAL_BOOL, &s_led2, sizeof(s_led2)},
|
|
{"led3", MG_VAL_BOOL, &s_led3, sizeof(s_led3)},
|
|
{NULL, 0, NULL, 0},
|
|
};
|
|
|
|
static void sync_leds(bool is_write) {
|
|
if (is_write) {
|
|
// gpio_write(LED1_PIN, s_led1);
|
|
// gpio_write(LED2_PIN, s_led2);
|
|
// gpio_write(LED3_PIN, s_led3);
|
|
} else {
|
|
// s_led1 = gpio_read(LED1_PIN);
|
|
// s_led2 = gpio_read(LED2_PIN);
|
|
// s_led3 = gpio_read(LED3_PIN);
|
|
}
|
|
}
|
|
|
|
// Read-only device Metrics
|
|
static int s_ram = 32, s_cpu = 7;
|
|
static double s_temperature = 24.8;
|
|
|
|
static struct mg_field fields_metrics[] = {
|
|
{"ram", MG_VAL_INT, &s_ram, sizeof(s_ram)},
|
|
{"cpu", MG_VAL_INT, &s_cpu, sizeof(s_cpu)},
|
|
{"temperature", MG_VAL_DBL, &s_temperature, sizeof(s_temperature)},
|
|
{NULL, 0, NULL, 0},
|
|
};
|
|
|
|
static void sync_metrics(bool is_write) {
|
|
if (is_write) return;
|
|
s_ram = 25 + (rand() % 16); // Sumulate metric change
|
|
s_cpu = 7 + (rand() % 21);
|
|
s_temperature = 14.8 + ((double) rand() / RAND_MAX) * 20.0;
|
|
}
|
|
|
|
// Read-write device settings
|
|
static double s_volume = 17.2;
|
|
static int s_log_level = MG_LL_DEBUG;
|
|
static char s_name[20] = "Dublin";
|
|
|
|
static struct mg_field fields_settings[] = {
|
|
{"volume", MG_VAL_DBL, &s_volume, sizeof(s_volume)},
|
|
{"name", MG_VAL_STR, &s_name, sizeof(s_name)},
|
|
{"log_level", MG_VAL_INT, &s_log_level, sizeof(s_log_level)},
|
|
{NULL, 0, NULL, 0},
|
|
};
|
|
|
|
static void sync_settings(bool is_write) {
|
|
if (is_write) {
|
|
mg_log_level = s_log_level;
|
|
} else {
|
|
s_log_level = mg_log_level;
|
|
}
|
|
}
|
|
|
|
#define NUM_POINTS 7 // How many data points to send
|
|
|
|
struct point {
|
|
uint16_t x; // X coordinate of the data point, usually this is time
|
|
uint16_t y1; // We can plot multiple values per graph,
|
|
uint16_t y2; // let's plot 2 values: y1 and y2.
|
|
};
|
|
|
|
// Serialise chart data into a string: [[x,y1,y2],[x,y1,y2],...]
|
|
// One uint16_t takes 5 character max, so one entry is 19 bytes max.
|
|
// We could be more efficient by storing hex or base64-encoded data, but
|
|
// let's keep it simple. However if we do, update data parsing logic in
|
|
// the web_root/main.js
|
|
#define POINT_SIZE 19
|
|
static char s_chart1_data[NUM_POINTS * POINT_SIZE + 2 + 1];
|
|
|
|
static struct mg_field fields_chart1[] = {
|
|
{"data", MG_VAL_RAW, s_chart1_data, sizeof(s_chart1_data)},
|
|
{NULL, 0, NULL, 0},
|
|
};
|
|
|
|
// Randomly generate graph points and serialise them into a string
|
|
void sync_chart1(bool is_write) {
|
|
size_t i, len = 0;
|
|
if (is_write) return;
|
|
len += mg_snprintf(s_chart1_data + len, sizeof(s_chart1_data) - len, "[");
|
|
for (i = 0; i < NUM_POINTS; i++) {
|
|
struct point p;
|
|
p.x = (uint16_t) i;
|
|
p.y1 = 85 - (uint16_t) ((49 * i) / NUM_POINTS) + (rand() % 9);
|
|
p.y2 = 20 + (uint16_t) ((69 * i) / NUM_POINTS) + (rand() % 5);
|
|
len += mg_snprintf(s_chart1_data + len, sizeof(s_chart1_data) - len,
|
|
"%s[%hu,%hu,%hu]", i > 0 ? "," : "", p.x, p.y1, p.y2);
|
|
}
|
|
len += mg_snprintf(s_chart1_data + len, sizeof(s_chart1_data) - len, "]");
|
|
}
|
|
|
|
struct file {
|
|
struct file *next;
|
|
char *name;
|
|
size_t size;
|
|
} *s_file_list;
|
|
|
|
static void file_add(struct mg_str name, size_t size) {
|
|
struct file *f = mg_calloc(1, sizeof(*f));
|
|
f->name = mg_strdup(name).buf;
|
|
f->size = size;
|
|
f->next = s_file_list;
|
|
s_file_list = f;
|
|
}
|
|
|
|
static void file_del(struct mg_str name) {
|
|
struct file **head, *f;
|
|
for (head = &s_file_list, f = *head; f; head = &(*head)->next, f = *head) {
|
|
if (mg_strcmp(mg_str(f->name), name) == 0) {
|
|
MG_INFO(("Deleting %s", f->name));
|
|
*head = f->next;
|
|
mg_free(f->name);
|
|
mg_free(f);
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
// Files array: [{"name": "foo.txt", "size": 1234}]
|
|
static char s_files[1024];
|
|
static struct mg_field fields_files[] = {
|
|
{"data", MG_VAL_RAW, s_files, sizeof(s_files)},
|
|
{NULL, 0, NULL, 0},
|
|
};
|
|
static void sync_files(bool is_write) {
|
|
size_t len = 0;
|
|
struct file *f;
|
|
if (is_write) return;
|
|
len += mg_snprintf(s_files + len, sizeof(s_files) - len, "[");
|
|
for (f = s_file_list; f; f = f->next) {
|
|
len += mg_snprintf(s_files + len, sizeof(s_files) - len, "%s{%m:%m,%m:%u}",
|
|
len > 1 ? "," : "", MG_ESC("name"), MG_ESC(f->name),
|
|
MG_ESC("size"), f->size);
|
|
}
|
|
len += mg_snprintf(s_files + len, sizeof(s_files) - len, "]");
|
|
}
|
|
|
|
static struct mg_field_set field_sets[] = {
|
|
{"leds", fields_leds, sync_leds, 0, 0},
|
|
{"metrics", fields_metrics, sync_metrics, 0, 0},
|
|
{"settings", fields_settings, sync_settings, 0, 0},
|
|
{"chart1", fields_chart1, sync_chart1, 0, 0},
|
|
{"files", fields_files, sync_files, 0, 0},
|
|
{0, 0, 0, 0, 0},
|
|
};
|
|
static struct mg_dash s_dashboard = {field_sets, NULL};
|
|
|
|
struct upload_state {
|
|
char marker; // Tells that we're a file upload connection
|
|
size_t expected; // POST data length, bytes
|
|
size_t received; // Already received bytes
|
|
void *ctx; // OTA context
|
|
};
|
|
|
|
static void file_ev_handler(struct mg_connection *c, int ev, void *ev_data) {
|
|
struct upload_state *us = (struct upload_state *) c->data;
|
|
if ((ev == MG_EV_HTTP_HDRS || ev == MG_EV_HTTP_MSG) && us->marker == 0) {
|
|
struct mg_http_message *hm = (struct mg_http_message *) ev_data;
|
|
if (hm->uri.len > 7) hm->uri.len -= 7, hm->uri.buf += 7; // Strip /files/
|
|
if (sizeof(*us) > sizeof(c->data)) {
|
|
mg_error(c,
|
|
"FAILURE: sizeof(c->data) == %lu, need %lu."
|
|
" Set #define MG_DATA_SIZE XXX",
|
|
sizeof(c->data), sizeof(*us));
|
|
} else if (mg_strcmp(hm->query, mg_str("action=delete")) == 0) {
|
|
// Query string ?action=delete - file deletion
|
|
file_del(hm->uri);
|
|
mg_dash_send_change(c->mgr, &s_dashboard, "files");
|
|
mg_http_reply(c, 200, NULL, "true");
|
|
} else if (mg_path_is_sane(hm->uri) == false) {
|
|
// Bad file name, return error. Protect from traversal, etc
|
|
mg_http_reply(c, 400, NULL, "Bad name");
|
|
} else if (mg_match(hm->method, mg_str("POST"), NULL) ||
|
|
mg_match(hm->method, mg_str("PUT"), NULL)) {
|
|
// File upload
|
|
char path[128];
|
|
mg_snprintf(path, sizeof(path), "/tmp/%.*s", hm->uri.len, hm->uri.buf);
|
|
us->ctx = fopen(path, "wb+");
|
|
if (us->ctx == NULL) {
|
|
mg_http_reply(c, 500, NULL, "File upload error %d\n", errno);
|
|
} else {
|
|
file_add(hm->uri, hm->body.len);
|
|
MG_DEBUG(("Starting upload, [%.*s] %lu", hm->uri.len, hm->uri.buf, hm->body.len));
|
|
us->marker = 'U';
|
|
us->expected = hm->body.len;
|
|
us->received = 0;
|
|
mg_iobuf_del(&c->recv, 0, hm->head.len);
|
|
c->pfn = NULL;
|
|
c->fn = file_ev_handler;
|
|
mg_call(c, MG_EV_READ, &c->recv.len);
|
|
}
|
|
} else {
|
|
// Serve file
|
|
char path[128];
|
|
mg_snprintf(path, sizeof(path), "/tmp/%.*s", hm->uri.len, hm->uri.buf);
|
|
mg_http_serve_file(c, ev_data, path, NULL);
|
|
}
|
|
} else if (ev == MG_EV_READ && us->marker == 'U') {
|
|
// Write uploaded data
|
|
MG_DEBUG(("Uploading.. recv.len=%u", c->recv.len));
|
|
us->received += c->recv.len;
|
|
fwrite(c->recv.buf, 1, c->recv.len, us->ctx);
|
|
c->recv.len = 0;
|
|
if (us->received >= us->expected) {
|
|
fclose(us->ctx);
|
|
MG_DEBUG(("Uploaded %lu", us->expected));
|
|
mg_http_reply(c, 200, NULL, "%lu uploaded\n", us->expected);
|
|
mg_dash_send_change(c->mgr, &s_dashboard, "files");
|
|
memset(us, 0, sizeof(*us));
|
|
c->is_draining = 1;
|
|
}
|
|
}
|
|
}
|
|
|
|
void mg_dash_init(struct mg_mgr *mgr) {
|
|
file_add(mg_str("device-config.json"), 1234);
|
|
file_add(mg_str("device-log-2026-04-25.txt"), 1327854);
|
|
MG_DASH_REGISTER_CUSTOM_HANDLER(&s_dashboard, "/files/#", file_ev_handler);
|
|
mg_http_listen(mgr, HTTP_ADDR, mg_dash_ev_handler, &s_dashboard);
|
|
}
|
|
|
|
void mg_dash_poll(struct mg_mgr *mgr) {
|
|
// Send metrics change periodically
|
|
static uint64_t timer = 0;
|
|
if (mg_timer_expired(&timer, 30000, mg_now())) {
|
|
mg_dash_send_change(mgr, &s_dashboard, "metrics");
|
|
}
|
|
}
|