2022-05-14 05:15:33 +08:00
|
|
|
// Copyright (c) 2020-2022 Cesanta Software Limited
|
|
|
|
// All rights reserved
|
|
|
|
|
|
|
|
#include "mongoose.h"
|
|
|
|
|
2022-09-09 18:11:53 +08:00
|
|
|
#if !defined(MQTT_SERVER)
|
2022-06-03 14:13:08 +08:00
|
|
|
#define MQTT_SERVER "mqtt://broker.hivemq.com:1883"
|
2022-09-09 18:11:53 +08:00
|
|
|
#endif
|
2022-06-03 14:13:08 +08:00
|
|
|
#define MQTT_PUBLISH_TOPIC "mg/my_device"
|
|
|
|
#define MQTT_SUBSCRIBE_TOPIC "mg/#"
|
|
|
|
|
2022-08-20 07:03:15 +08:00
|
|
|
static time_t s_boot_timestamp = 0; // Updated by SNTP
|
2022-12-28 05:12:55 +08:00
|
|
|
#ifndef DISABLE_ROUTING
|
2022-08-20 07:03:15 +08:00
|
|
|
static struct mg_connection *s_sntp_conn = NULL; // SNTP connection
|
2022-12-28 05:12:55 +08:00
|
|
|
#endif
|
2022-08-20 07:03:15 +08:00
|
|
|
|
2022-10-05 06:27:38 +08:00
|
|
|
// Define a system time alternative
|
|
|
|
time_t ourtime(time_t *tp) {
|
2022-08-20 07:03:15 +08:00
|
|
|
time_t t = s_boot_timestamp + (time_t) (mg_millis() / 1000);
|
|
|
|
if (tp != NULL) *tp = t;
|
|
|
|
return t;
|
|
|
|
}
|
|
|
|
|
2022-05-14 05:15:33 +08:00
|
|
|
// Authenticated user.
|
|
|
|
// A user can be authenticated by:
|
|
|
|
// - a name:pass pair
|
|
|
|
// - a token
|
|
|
|
// When a user is shown a login screen, she enters a user:pass. If successful,
|
|
|
|
// a server returns user info which includes token. From that point on,
|
|
|
|
// client can use token for authentication. Tokens could be refreshed/changed
|
|
|
|
// on a server side, forcing clients to re-login.
|
|
|
|
struct user {
|
|
|
|
const char *name, *pass, *token;
|
|
|
|
};
|
|
|
|
|
|
|
|
// This is a configuration structure we're going to show on a dashboard
|
|
|
|
static struct config {
|
2022-06-03 14:13:08 +08:00
|
|
|
char *url, *pub, *sub; // MQTT settings
|
|
|
|
} s_config;
|
|
|
|
|
|
|
|
static struct mg_connection *s_mqtt = NULL; // MQTT connection
|
2022-06-05 21:59:59 +08:00
|
|
|
static bool s_connected = false; // MQTT connection established
|
2022-05-14 05:15:33 +08:00
|
|
|
|
2022-06-03 14:13:08 +08:00
|
|
|
// Try to update a single configuration value
|
|
|
|
static void update_config(struct mg_str *body, const char *name, char **value) {
|
2022-05-14 05:15:33 +08:00
|
|
|
char buf[256];
|
2022-06-03 14:13:08 +08:00
|
|
|
if (mg_http_get_var(body, name, buf, sizeof(buf)) > 0) {
|
|
|
|
free(*value);
|
|
|
|
*value = strdup(buf);
|
2022-05-14 05:15:33 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Parse HTTP requests, return authenticated user or NULL
|
|
|
|
static struct user *getuser(struct mg_http_message *hm) {
|
|
|
|
// In production, make passwords strong and tokens randomly generated
|
|
|
|
// In this example, user list is kept in RAM. In production, it can
|
|
|
|
// be backed by file, database, or some other method.
|
|
|
|
static struct user users[] = {
|
|
|
|
{"admin", "pass0", "admin_token"},
|
|
|
|
{"user1", "pass1", "user1_token"},
|
|
|
|
{"user2", "pass2", "user2_token"},
|
|
|
|
{NULL, NULL, NULL},
|
|
|
|
};
|
|
|
|
char user[256], pass[256];
|
|
|
|
struct user *u;
|
|
|
|
mg_http_creds(hm, user, sizeof(user), pass, sizeof(pass));
|
|
|
|
if (user[0] != '\0' && pass[0] != '\0') {
|
|
|
|
// Both user and password is set, search by user/password
|
|
|
|
for (u = users; u->name != NULL; u++)
|
|
|
|
if (strcmp(user, u->name) == 0 && strcmp(pass, u->pass) == 0) return u;
|
|
|
|
} else if (user[0] == '\0') {
|
|
|
|
// Only password is set, search by token
|
|
|
|
for (u = users; u->name != NULL; u++)
|
|
|
|
if (strcmp(pass, u->token) == 0) return u;
|
|
|
|
}
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Notify all config watchers about the config change
|
2022-08-13 21:52:10 +08:00
|
|
|
static void send_notification(struct mg_mgr *mgr, const char *fmt, ...) {
|
2022-05-14 05:15:33 +08:00
|
|
|
struct mg_connection *c;
|
|
|
|
for (c = mgr->conns; c != NULL; c = c->next) {
|
2023-01-09 18:58:07 +08:00
|
|
|
if (c->data[0] == 'W') {
|
2022-08-13 21:52:10 +08:00
|
|
|
va_list ap;
|
|
|
|
va_start(ap, fmt);
|
|
|
|
mg_ws_vprintf(c, WEBSOCKET_OP_TEXT, fmt, &ap);
|
|
|
|
va_end(ap);
|
|
|
|
}
|
2022-05-14 05:15:33 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-05-23 22:27:19 +08:00
|
|
|
// Send simulated metrics data to the dashboard, for chart rendering
|
2022-06-03 14:13:08 +08:00
|
|
|
static void timer_metrics_fn(void *param) {
|
2022-08-13 21:52:10 +08:00
|
|
|
send_notification(param, "{%Q:%Q,%Q:[%lu, %d]}", "name", "metrics", "data",
|
2022-10-05 06:27:38 +08:00
|
|
|
(unsigned long) ourtime(NULL),
|
2022-08-13 21:52:10 +08:00
|
|
|
10 + (int) ((double) rand() * 10 / RAND_MAX));
|
2022-05-23 22:27:19 +08:00
|
|
|
}
|
|
|
|
|
2022-12-28 05:12:55 +08:00
|
|
|
#ifndef DISABLE_ROUTING
|
|
|
|
|
2022-06-03 14:13:08 +08:00
|
|
|
// MQTT event handler function
|
2022-06-09 02:53:29 +08:00
|
|
|
static void mqtt_fn(struct mg_connection *c, int ev, void *ev_data, void *fnd) {
|
2022-08-26 21:20:29 +08:00
|
|
|
if (ev == MG_EV_CONNECT && mg_url_is_ssl(s_config.url)) {
|
|
|
|
struct mg_tls_opts opts = {.ca = "ca.pem",
|
|
|
|
.srvname = mg_url_host(s_config.url)};
|
|
|
|
mg_tls_init(c, &opts);
|
|
|
|
} else if (ev == MG_EV_MQTT_OPEN) {
|
2022-06-05 21:59:59 +08:00
|
|
|
s_connected = true;
|
2022-06-09 02:53:29 +08:00
|
|
|
c->is_hexdumping = 1;
|
|
|
|
mg_mqtt_sub(s_mqtt, mg_str(s_config.sub), 2);
|
2022-08-13 21:52:10 +08:00
|
|
|
send_notification(c->mgr, "{%Q:%Q,%Q:null}", "name", "config", "data");
|
2022-06-03 14:13:08 +08:00
|
|
|
} else if (ev == MG_EV_MQTT_MSG) {
|
2022-06-09 02:53:29 +08:00
|
|
|
struct mg_mqtt_message *mm = ev_data;
|
2022-08-13 21:52:10 +08:00
|
|
|
send_notification(c->mgr, "{%Q:%Q,%Q:{%Q: %.*Q, %Q: %.*Q, %Q: %d}}", "name",
|
|
|
|
"message", "data", "topic", (int) mm->topic.len,
|
|
|
|
mm->topic.ptr, "data", (int) mm->data.len, mm->data.ptr,
|
|
|
|
"qos", (int) mm->qos);
|
2022-06-09 02:53:29 +08:00
|
|
|
} else if (ev == MG_EV_MQTT_CMD) {
|
|
|
|
struct mg_mqtt_message *mm = (struct mg_mqtt_message *) ev_data;
|
2022-09-09 18:11:53 +08:00
|
|
|
MG_DEBUG(("%lu cmd %d qos %d", c->id, mm->cmd, mm->qos));
|
2022-06-03 14:13:08 +08:00
|
|
|
} else if (ev == MG_EV_CLOSE) {
|
|
|
|
s_mqtt = NULL;
|
2022-06-05 21:59:59 +08:00
|
|
|
if (s_connected) {
|
|
|
|
s_connected = false;
|
2022-08-13 21:52:10 +08:00
|
|
|
send_notification(c->mgr, "{%Q:%Q,%Q:null}", "name", "config", "data");
|
2022-06-05 21:59:59 +08:00
|
|
|
}
|
2022-06-03 14:13:08 +08:00
|
|
|
}
|
2022-06-05 21:59:59 +08:00
|
|
|
(void) fnd;
|
2022-06-03 14:13:08 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// Keep MQTT connection open - reconnect if closed
|
2022-06-05 21:59:59 +08:00
|
|
|
static void timer_mqtt_fn(void *param) {
|
2022-06-03 14:13:08 +08:00
|
|
|
struct mg_mgr *mgr = (struct mg_mgr *) param;
|
|
|
|
if (s_mqtt == NULL) {
|
2022-11-08 02:23:47 +08:00
|
|
|
struct mg_mqtt_opts opts;
|
|
|
|
memset(&opts, 0, sizeof(opts));
|
2022-06-05 21:59:59 +08:00
|
|
|
s_mqtt = mg_mqtt_connect(mgr, s_config.url, &opts, mqtt_fn, NULL);
|
2022-06-03 14:13:08 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-08-20 07:03:15 +08:00
|
|
|
// SNTP connection event handler. When we get a response from an SNTP server,
|
|
|
|
// adjust s_boot_timestamp. We'll get a valid time from that point on
|
|
|
|
static void sfn(struct mg_connection *c, int ev, void *ev_data, void *fn_data) {
|
|
|
|
if (ev == MG_EV_SNTP_TIME) {
|
|
|
|
uint64_t t = *(uint64_t *) ev_data;
|
|
|
|
s_boot_timestamp = (time_t) ((t - mg_millis()) / 1000);
|
|
|
|
c->is_closing = 1;
|
|
|
|
} else if (ev == MG_EV_CLOSE) {
|
|
|
|
s_sntp_conn = NULL;
|
|
|
|
}
|
|
|
|
(void) fn_data;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void timer_sntp_fn(void *param) { // SNTP timer function. Sync up time
|
|
|
|
struct mg_mgr *mgr = (struct mg_mgr *) param;
|
|
|
|
if (s_sntp_conn == NULL && s_boot_timestamp == 0) {
|
|
|
|
s_sntp_conn = mg_sntp_connect(mgr, NULL, sfn, NULL);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-12-28 05:12:55 +08:00
|
|
|
#endif
|
|
|
|
|
2022-05-14 05:15:33 +08:00
|
|
|
// HTTP request handler function
|
2022-05-15 21:40:58 +08:00
|
|
|
void device_dashboard_fn(struct mg_connection *c, int ev, void *ev_data,
|
|
|
|
void *fn_data) {
|
2022-05-23 22:27:19 +08:00
|
|
|
if (ev == MG_EV_OPEN && c->is_listening) {
|
2022-06-03 14:13:08 +08:00
|
|
|
mg_timer_add(c->mgr, 1000, MG_TIMER_REPEAT, timer_metrics_fn, c->mgr);
|
2022-12-28 05:12:55 +08:00
|
|
|
#ifndef DISABLE_ROUTING
|
2022-06-03 14:13:08 +08:00
|
|
|
mg_timer_add(c->mgr, 1000, MG_TIMER_REPEAT, timer_mqtt_fn, c->mgr);
|
2022-08-20 07:03:15 +08:00
|
|
|
mg_timer_add(c->mgr, 1000, MG_TIMER_REPEAT, timer_sntp_fn, c->mgr);
|
2022-12-28 05:12:55 +08:00
|
|
|
#endif
|
2022-06-03 14:13:08 +08:00
|
|
|
s_config.url = strdup(MQTT_SERVER);
|
|
|
|
s_config.pub = strdup(MQTT_PUBLISH_TOPIC);
|
|
|
|
s_config.sub = strdup(MQTT_SUBSCRIBE_TOPIC);
|
2022-05-23 22:27:19 +08:00
|
|
|
} else if (ev == MG_EV_HTTP_MSG) {
|
2022-05-14 05:15:33 +08:00
|
|
|
struct mg_http_message *hm = (struct mg_http_message *) ev_data;
|
|
|
|
struct user *u = getuser(hm);
|
|
|
|
// MG_INFO(("%p [%.*s] auth %s", c->fd, (int) hm->uri.len, hm->uri.ptr,
|
|
|
|
// u ? u->name : "NULL"));
|
2022-05-18 02:29:43 +08:00
|
|
|
if (mg_http_match_uri(hm, "/api/hi")) {
|
|
|
|
mg_http_reply(c, 200, "", "hi\n"); // Testing endpoint
|
2022-10-27 15:12:12 +08:00
|
|
|
} else if (mg_http_match_uri(hm, "/api/debug")) {
|
|
|
|
int level = mg_json_get_long(hm->body, "$.level", MG_LL_DEBUG);
|
|
|
|
mg_log_set(level);
|
|
|
|
mg_http_reply(c, 200, "", "Debug level set to %d\n", level);
|
2022-05-18 02:29:43 +08:00
|
|
|
} else if (u == NULL && mg_http_match_uri(hm, "/api/#")) {
|
2022-05-14 05:15:33 +08:00
|
|
|
// All URIs starting with /api/ must be authenticated
|
2022-08-13 21:52:10 +08:00
|
|
|
mg_http_reply(c, 403, "", "Denied\n");
|
2022-05-14 05:15:33 +08:00
|
|
|
} else if (mg_http_match_uri(hm, "/api/config/get")) {
|
2022-12-28 05:12:55 +08:00
|
|
|
#ifdef DISABLE_ROUTING
|
|
|
|
mg_http_reply(c, 200, NULL, "{%Q:%Q,%Q:%Q,%Q:%Q}\n", "url", s_config.url,
|
|
|
|
"pub", s_config.pub, "sub", s_config.sub);
|
|
|
|
#else
|
2022-06-09 04:44:01 +08:00
|
|
|
mg_http_reply(c, 200, NULL, "{%Q:%Q,%Q:%Q,%Q:%Q,%Q:%s}\n", "url",
|
|
|
|
s_config.url, "pub", s_config.pub, "sub", s_config.sub,
|
|
|
|
"connected", s_connected ? "true" : "false");
|
2022-12-28 05:12:55 +08:00
|
|
|
#endif
|
2022-05-14 05:15:33 +08:00
|
|
|
} else if (mg_http_match_uri(hm, "/api/config/set")) {
|
|
|
|
// Admins only
|
|
|
|
if (strcmp(u->name, "admin") == 0) {
|
2022-06-03 14:13:08 +08:00
|
|
|
update_config(&hm->body, "url", &s_config.url);
|
|
|
|
update_config(&hm->body, "pub", &s_config.pub);
|
|
|
|
update_config(&hm->body, "sub", &s_config.sub);
|
2022-06-05 21:59:59 +08:00
|
|
|
if (s_mqtt) s_mqtt->is_closing = 1; // Ask to disconnect from MQTT
|
2022-08-13 21:52:10 +08:00
|
|
|
send_notification(fn_data, "{%Q:%Q,%Q:null}", "name", "config", "data");
|
|
|
|
mg_http_reply(c, 200, "", "ok\n");
|
2022-05-14 05:15:33 +08:00
|
|
|
} else {
|
2022-08-13 21:52:10 +08:00
|
|
|
mg_http_reply(c, 403, "", "Denied\n");
|
2022-05-14 05:15:33 +08:00
|
|
|
}
|
|
|
|
} else if (mg_http_match_uri(hm, "/api/message/send")) {
|
|
|
|
char buf[256];
|
2022-06-05 21:59:59 +08:00
|
|
|
if (s_connected &&
|
|
|
|
mg_http_get_var(&hm->body, "message", buf, sizeof(buf)) > 0) {
|
|
|
|
mg_mqtt_pub(s_mqtt, mg_str(s_config.pub), mg_str(buf), 1, false);
|
2022-05-14 05:15:33 +08:00
|
|
|
}
|
2022-08-13 21:52:10 +08:00
|
|
|
mg_http_reply(c, 200, "", "ok\n");
|
2022-05-14 05:15:33 +08:00
|
|
|
} else if (mg_http_match_uri(hm, "/api/watch")) {
|
2023-01-09 18:58:07 +08:00
|
|
|
c->data[0] = 'W'; // Mark ourselves as a event listener
|
2022-06-05 21:59:59 +08:00
|
|
|
mg_ws_upgrade(c, hm, NULL);
|
2022-05-14 05:15:33 +08:00
|
|
|
} else if (mg_http_match_uri(hm, "/api/login")) {
|
2022-06-09 04:44:01 +08:00
|
|
|
mg_http_reply(c, 200, NULL, "{%Q:%Q,%Q:%Q}\n", "user", u->name, "token",
|
|
|
|
u->token);
|
2022-05-14 05:15:33 +08:00
|
|
|
} else {
|
2022-11-08 02:23:47 +08:00
|
|
|
struct mg_http_serve_opts opts;
|
|
|
|
memset(&opts, 0, sizeof(opts));
|
2022-08-03 03:06:19 +08:00
|
|
|
#if 1
|
2022-05-14 05:15:33 +08:00
|
|
|
opts.root_dir = "/web_root";
|
|
|
|
opts.fs = &mg_fs_packed;
|
|
|
|
#else
|
|
|
|
opts.root_dir = "web_root";
|
|
|
|
#endif
|
|
|
|
mg_http_serve_dir(c, ev_data, &opts);
|
|
|
|
}
|
2022-09-09 18:11:53 +08:00
|
|
|
MG_DEBUG(("%lu %.*s %.*s -> %.*s", c->id, (int) hm->method.len,
|
|
|
|
hm->method.ptr, (int) hm->uri.len, hm->uri.ptr, (int) 3,
|
|
|
|
&c->send.buf[9]));
|
2022-05-14 05:15:33 +08:00
|
|
|
}
|
|
|
|
}
|