net.c 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619
  1. // Copyright (c) 2023 Cesanta Software Limited
  2. // All rights reserved
  3. #include <syslog.h>
  4. #include "net.h"
  5. #include "appl.h"
  6. #include "modbus.h"
  7. // Authenticated user.
  8. // A user can be authenticated by:
  9. // - a name:pass pair, passed in a header Authorization: Basic .....
  10. // - an access_token, passed in a header Cookie: access_token=....
  11. // When a user is shown a login screen, she enters a user:pass. If successful,
  12. // a server responds with a http-only access_token cookie set.
  13. struct user {
  14. const char *name, *pass, *access_token;
  15. };
  16. // Settings
  17. struct settings {
  18. bool log_enabled;
  19. int log_level;
  20. long brightness;
  21. char *device_name;
  22. };
  23. static struct settings s_settings = {true, 1, 57, NULL};
  24. static const char *s_json_header =
  25. "Content-Type: application/json\r\n"
  26. "Cache-Control: no-cache\r\n";
  27. static uint64_t s_boot_timestamp = 0; // Updated by SNTP
  28. // This is for newlib and TLS (mbedTLS)
  29. uint64_t mg_now(void) {
  30. return mg_millis() + s_boot_timestamp;
  31. }
  32. int ui_event_next(int no, struct ui_event *e) {
  33. if (no < 0 || no >= MAX_EVENTS_NO) return 0;
  34. srand((unsigned) no);
  35. e->type = (uint8_t) rand() % 4;
  36. e->prio = (uint8_t) rand() % 3;
  37. e->timestamp =
  38. (unsigned long) ((int64_t) mg_now() - 86400 * 1000 /* one day back */ +
  39. no * 300 * 1000 /* 5 mins between alerts */ +
  40. 1000 * (rand() % 300) /* randomize event time */) /
  41. 1000UL;
  42. mg_snprintf(e->text, MAX_EVENT_TEXT_SIZE, "event#%d", no);
  43. return no + 1;
  44. }
  45. // SNTP connection event handler. When we get a response from an SNTP server,
  46. // adjust s_boot_timestamp. We'll get a valid time from that point on
  47. static void sfn(struct mg_connection *c, int ev, void *ev_data) {
  48. uint64_t *expiration_time = (uint64_t *) c->data;
  49. if (ev == MG_EV_OPEN) {
  50. *expiration_time = mg_millis() + 3000; // Store expiration time in 3s
  51. } else if (ev == MG_EV_SNTP_TIME) {
  52. uint64_t t = *(uint64_t *) ev_data;
  53. s_boot_timestamp = t - mg_millis();
  54. c->is_closing = 1;
  55. } else if (ev == MG_EV_POLL) {
  56. if (mg_millis() > *expiration_time) c->is_closing = 1;
  57. }
  58. }
  59. static void timer_sntp_fn(void *param) { // SNTP timer function. Sync up time
  60. mg_sntp_connect(param, "udp://time.google.com:123", sfn, NULL);
  61. }
  62. // Parse HTTP requests, return authenticated user or NULL
  63. static struct user *authenticate(struct mg_http_message *hm) {
  64. // In production, make passwords strong and tokens randomly generated
  65. // In this example, user list is kept in RAM. In production, it can
  66. // be backed by file, database, or some other method.
  67. static struct user users[] = {
  68. {"admin", "admin", "admin_token"},
  69. {"user1", "user1", "user1_token"},
  70. {"user2", "user2", "user2_token"},
  71. {NULL, NULL, NULL},
  72. };
  73. char user[64], pass[64];
  74. struct user *u, *result = NULL;
  75. mg_http_creds(hm, user, sizeof(user), pass, sizeof(pass));
  76. MG_VERBOSE(("user [%s] pass [%s]", user, pass));
  77. if (user[0] != '\0' && pass[0] != '\0') {
  78. // Both user and password is set, search by user/password
  79. for (u = users; result == NULL && u->name != NULL; u++)
  80. if (strcmp(user, u->name) == 0 && strcmp(pass, u->pass) == 0) result = u;
  81. } else if (user[0] == '\0') {
  82. // Only password is set, search by token
  83. for (u = users; result == NULL && u->name != NULL; u++)
  84. if (strcmp(pass, u->access_token) == 0) result = u;
  85. }
  86. return result;
  87. }
  88. static void handle_login(struct mg_connection *c, struct user *u) {
  89. char cookie[256];
  90. mg_snprintf(cookie, sizeof(cookie),
  91. "Set-Cookie: access_token=%s; Path=/; "
  92. "%sHttpOnly; SameSite=Lax; Max-Age=%d\r\n",
  93. u->access_token, c->is_tls ? "Secure; " : "", 3600 * 24);
  94. mg_http_reply(c, 200, cookie, "{%m:%m}", MG_ESC("user"), MG_ESC(u->name));
  95. }
  96. static void handle_logout(struct mg_connection *c) {
  97. char cookie[256];
  98. mg_snprintf(cookie, sizeof(cookie),
  99. "Set-Cookie: access_token=; Path=/; "
  100. "Expires=Thu, 01 Jan 1970 00:00:00 UTC; "
  101. "%sHttpOnly; Max-Age=0; \r\n",
  102. c->is_tls ? "Secure; " : "");
  103. mg_http_reply(c, 200, cookie, "true\n");
  104. }
  105. static void handle_debug(struct mg_connection *c, struct mg_http_message *hm) {
  106. int level = mg_json_get_long(hm->body, "$.level", MG_LL_DEBUG);
  107. mg_log_set(level);
  108. mg_http_reply(c, 200, "", "Debug level set to %d\n", level);
  109. }
  110. static size_t print_int_arr(void (*out)(char, void *), void *ptr, va_list *ap) {
  111. size_t i, len = 0, num = va_arg(*ap, size_t); // Number of items in the array
  112. int *arr = va_arg(*ap, int *); // Array ptr
  113. for (i = 0; i < num; i++) {
  114. len += mg_xprintf(out, ptr, "%s%d", i == 0 ? "" : ",", arr[i]);
  115. }
  116. return len;
  117. }
  118. static void handle_stats_get(struct mg_connection *c) { //处理 HTTP 请求并返回 JSON 数据
  119. int points[] = {21, 22, 22, 19, 18, 20, 23, 23, 22, 22, 22, 23, 22};
  120. mg_http_reply(c, 200, s_json_header, "{%m:%d,%m:%d,%m:[%M]}\n",
  121. MG_ESC("temperature"), 21, //
  122. MG_ESC("humidity"), 67, //
  123. MG_ESC("points"), print_int_arr,
  124. sizeof(points) / sizeof(points[0]), points);
  125. }
  126. static size_t print_events(void (*out)(char, void *), void *ptr, va_list *ap) {
  127. size_t len = 0;
  128. struct ui_event ev;
  129. int pageno = va_arg(*ap, int);
  130. int no = (pageno - 1) * EVENTS_PER_PAGE;
  131. int end = no + EVENTS_PER_PAGE;
  132. while ((no = ui_event_next(no, &ev)) != 0 && no <= end) {
  133. len += mg_xprintf(out, ptr, "%s{%m:%lu,%m:%d,%m:%d,%m:%m}\n", //
  134. len == 0 ? "" : ",", //
  135. MG_ESC("time"), ev.timestamp, //
  136. MG_ESC("type"), ev.type, //
  137. MG_ESC("prio"), ev.prio, //
  138. MG_ESC("text"), MG_ESC(ev.text));
  139. }
  140. return len;
  141. }
  142. static void handle_events_get(struct mg_connection *c,
  143. struct mg_http_message *hm) {
  144. int pageno = mg_json_get_long(hm->body, "$.page", 1);
  145. mg_http_reply(c, 200, s_json_header, "{%m:[%M], %m:%d}\n", MG_ESC("arr"),
  146. print_events, pageno, MG_ESC("totalCount"), MAX_EVENTS_NO);
  147. }
  148. static void handle_settings_set(struct mg_connection *c, struct mg_str body) {
  149. struct Settings_t* set = &APPL.Set.s;
  150. struct chanmqtt_t* mqtt = &APPL.chanmqtt[4];//用mqtt4就改成4
  151. int cmd;
  152. int param;
  153. bool ok = true;
  154. int rc;
  155. long val;
  156. int i;
  157. char buf[128];
  158. char* p;
  159. val = mg_json_get_long(body, "$.cmd", -1);
  160. if(val != -1){
  161. cmd = val;
  162. syslog(LOG_INFO,"%s, cmd:%d", __func__, cmd);
  163. switch (cmd)
  164. {
  165. case SETTINGS_CMD_SET_CLOUD_USERNAME:
  166. p = mg_json_get_str(body, "$.param");
  167. if( p != NULL){
  168. strcpy(set->szCloudUserName, p);
  169. free(p);
  170. if(appl_cfg_save() != 0){
  171. appl_cfg_set_err();
  172. ok = false;
  173. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_CLOUD_USERNAME, appl_cfg_save Fail", __func__);
  174. }
  175. }else{
  176. ok = false;
  177. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_CLOUD_USERNAME, mg_json_get_str Fail", __func__);
  178. }
  179. break;
  180. case SETTINGS_CMD_SET_CLOUD_PASSWD:
  181. p = mg_json_get_str(body, "$.param");
  182. if( p != NULL){
  183. strcpy(set->szCloudPasswd, p);
  184. free(p);
  185. if(appl_cfg_save() != 0){
  186. appl_cfg_set_err();
  187. ok = false;
  188. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_CLOUD_PASSWD, appl_cfg_save Fail", __func__);
  189. }
  190. }else{
  191. ok = false;
  192. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_CLOUD_PASSWD, mg_json_get_str Fail", __func__);
  193. }
  194. break;
  195. case SETTINGS_CMD_SET_CLOUD_URL:
  196. p = mg_json_get_str(body, "$.param");
  197. if( p != NULL){
  198. strcpy(set->szCloudUrl, p);
  199. free(p);
  200. if(appl_cfg_save() != 0){
  201. appl_cfg_set_err();
  202. ok = false;
  203. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_CLOUD_URL, appl_cfg_save Fail", __func__);
  204. }
  205. }else{
  206. ok = false;
  207. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_CLOUD_URL, mg_json_get_str Fail", __func__);
  208. }
  209. break;
  210. case SETTINGS_CMD_SET_CLOUD_CLIENTID:
  211. p = mg_json_get_str(body, "$.param");
  212. if( p != NULL){
  213. strcpy(set->szClientId, p);
  214. free(p);
  215. if(appl_cfg_save() != 0){
  216. appl_cfg_set_err();
  217. ok = false;
  218. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_CLOUD_CLIENTID, appl_cfg_save Fail", __func__);
  219. }
  220. }else{
  221. ok = false;
  222. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_CLOUD_CLIENTID, mg_json_get_str Fail", __func__);
  223. }
  224. break;
  225. case SETTINGS_CMD_SET_UPLOADHIHGSPEED:
  226. val = mg_json_get_long(body, "$.param", -1);
  227. if( val == -1 ){
  228. ok = false;
  229. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_UPLOADHIHGSPEED, mg_json_get_long Fail", __func__);
  230. }else{
  231. set->UploadHighSpeed = val;
  232. if(appl_cfg_save() != 0){
  233. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_UPLOADHIHGSPEED, appl_cfg_save Fail", __func__);
  234. appl_cfg_set_err();
  235. ok = false;
  236. }else{
  237. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_UPLOADHIHGSPEED, set->UploadHighSpeed is set to %d", __func__, set->UploadHighSpeed);
  238. }
  239. }
  240. break;
  241. case SETTINGS_CMD_SET_UPLOADMEDIUMSPEED:
  242. val = mg_json_get_long(body, "$.param", -1);
  243. if( val == -1 ){
  244. ok = false;
  245. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_UPLOADMEDIUMSPEED, mg_json_get_long Fail", __func__);
  246. }else{
  247. set->UploadMediumSpeed = val;
  248. if(appl_cfg_save() != 0){
  249. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_UPLOADMEDIUMSPEED, appl_cfg_save Fail", __func__);
  250. appl_cfg_set_err();
  251. ok = false;
  252. }else{
  253. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_UPLOADMEDIUMSPEED, set->UploadMediumSpeed is set to %d", __func__, set->UploadMediumSpeed);
  254. }
  255. }
  256. break;
  257. case SETTINGS_CMD_SET_UPLOADSLOWSPEED:
  258. val = mg_json_get_long(body, "$.param", -1);
  259. if( val == -1 ){
  260. ok = false;
  261. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_UPLOADSLOWSPEED, mg_json_get_long Fail", __func__);
  262. }else{
  263. set->UploadSlowSpeed = val;
  264. if(appl_cfg_save() != 0){
  265. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_UPLOADSLOWSPEED, appl_cfg_save Fail", __func__);
  266. appl_cfg_set_err();
  267. ok = false;
  268. }else{
  269. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_UPLOADSLOWSPEED, set->UploadSlowSpeed is set to %d", __func__, set->UploadSlowSpeed);
  270. }
  271. }
  272. break;
  273. case SETTINGS_CMD_REGISTER:
  274. mqtt->Cmd = CMD_MQTT_REGISTER;
  275. break;
  276. case SETTINGS_CMD_SET_DATAKEEPDAY:
  277. val = mg_json_get_long(body, "$.param", -1);
  278. if( val == -1 ){
  279. ok = false;
  280. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_DATAKEEPDAY, mg_json_get_long Fail", __func__);
  281. }else{
  282. set->DataKeepDay = val;
  283. if(appl_cfg_save() != 0){
  284. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_DATAKEEPDAY, appl_cfg_save Fail", __func__);
  285. appl_cfg_set_err();
  286. ok = false;
  287. }else{
  288. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_DATAKEEPDAY, set->DataKeepDay is set to %d", __func__, set->DataKeepDay);
  289. }
  290. }
  291. break;
  292. case SETTINGS_CMD_TEST:
  293. syslog(LOG_INFO,"%s, Get Test Cmd", __func__);
  294. rc = appl_snap_rmdir("./snap/test");
  295. syslog(LOG_INFO,"%s, rc:%d", __func__, rc);
  296. break;
  297. default:
  298. break;
  299. }
  300. }else{
  301. ok = false;
  302. syslog(LOG_INFO,"%s, Get Cmd Fail", __func__);
  303. }
  304. mg_http_reply(c, 200, s_json_header,
  305. "{%m:%s,%m:%m}", //
  306. MG_ESC("status"), ok ? "true" : "false", //
  307. MG_ESC("message"), MG_ESC(ok ? "Success" : "Failed"));
  308. }
  309. static void handle_ctl_get(struct mg_connection *c) {
  310. struct Settings_t* set = &APPL.Set.s;
  311. struct Snap_t* snap = &APPL.Snap;
  312. mg_http_reply(c, 200, s_json_header, "{\
  313. %m:%m,%m:%m,\
  314. %m:%m}\n",
  315. /*1*/MG_ESC("数据记录状态"), MG_ESC(snap->szState),MG_ESC("参数配置状态"), MG_ESC(set->szState),
  316. /*2*/MG_ESC("软件版本"),MG_ESC(VERSION));
  317. }
  318. static void handle_settings_get(struct mg_connection *c) {
  319. struct Settings_t* set = &APPL.Set.s;
  320. mg_http_reply(c, 200, s_json_header, "{\
  321. %m:%m,%m:%m,%m:%m,%m:%m,\
  322. %m:%d,%m:%d,%m:%d,%m:%d}\n",
  323. MG_ESC("Cloud用户名"),MG_ESC(set->szCloudUserName),MG_ESC("Cloud密码"),MG_ESC(set->szCloudPasswd),MG_ESC("Cloud Url"),MG_ESC(set->szCloudUrl),MG_ESC("Cloud Client ID"),MG_ESC(set->szClientId),
  324. MG_ESC("数据保留天数"),set->DataKeepDay,MG_ESC("上传高速"),set->UploadHighSpeed,MG_ESC("上传中速"),set->UploadMediumSpeed,MG_ESC("上传低速"),set->UploadSlowSpeed);
  325. }
  326. static void handle_comm_get(struct mg_connection *c) {
  327. struct chan485_t* c1 = &APPL.chan485[1];
  328. struct chan485_t* c2 = &APPL.chan485[2];
  329. struct chan485_t* c3 = &APPL.chan485[3];
  330. struct chan485_t* c4 = &APPL.chan485[4];
  331. struct chancan_t* c5 = &APPL.chancan[1];
  332. struct chancan_t* c6 = &APPL.chancan[2];
  333. struct chanmqtt_t* c7 = &APPL.chanmqtt[1];
  334. struct chanmqtt_t* c8 = &APPL.chanmqtt[2];
  335. struct chanmqtt_t* c9 = &APPL.chanmqtt[3];
  336. struct chanmqtt_t* c10 = &APPL.chanmqtt[4];
  337. mg_http_reply(c, 200, s_json_header, "{\
  338. %m:{%m:%m,%m:%m,%m:%d,%m:%m,%m:%m,%m:%lld,%m:%lld,%m:%lld,%m:%lld},\
  339. %m:{%m:%m,%m:%m,%m:%d,%m:%m,%m:%m,%m:%lld,%m:%lld,%m:%lld,%m:%lld},\
  340. %m:{%m:%m,%m:%m,%m:%d,%m:%m,%m:%m,%m:%lld,%m:%lld,%m:%lld,%m:%lld},\
  341. %m:{%m:%m,%m:%m,%m:%d,%m:%m,%m:%m,%m:%lld,%m:%lld,%m:%lld,%m:%lld},\
  342. %m:{%m:%m,%m:%m,%m:%m,%m:%m,%m:%lld,%m:%lld,%m:%lld,%m:%lld},\
  343. %m:{%m:%m,%m:%m,%m:%m,%m:%m,%m:%lld,%m:%lld,%m:%lld,%m:%lld},\
  344. %m:{%m:%m,%m:%lld,%m:%lld,%m:%lld},\
  345. %m:{%m:%m,%m:%lld,%m:%lld,%m:%lld}}\n",
  346. MG_ESC("1#485"),MG_ESC("状态"), MG_ESC(c1->szstate),MG_ESC("故障码"),MG_ESC(c1->szerr),MG_ESC("波特率"),c1->baud,MG_ESC("设备名"), MG_ESC(c1->szdev),
  347. MG_ESC("备注"), MG_ESC(c1->szinfo), MG_ESC("循环次数"),c1->loopcnt, MG_ESC("循环时间"),c1->looptime, MG_ESC("请求次数"),c1->reqcnt, MG_ESC("失败次数"),c1->failcnt,
  348. MG_ESC("2#485"),MG_ESC("状态"), MG_ESC(c2->szstate),MG_ESC("故障码"),MG_ESC(c2->szerr),MG_ESC("波特率"),c2->baud,MG_ESC("设备名"),MG_ESC(c2->szdev),
  349. MG_ESC("备注"),MG_ESC(c2->szinfo),MG_ESC("循环次数"),c2->loopcnt,MG_ESC("循环时间"),c2->looptime,MG_ESC("请求次数"),c2->reqcnt,MG_ESC("失败次数"),c2->failcnt,
  350. MG_ESC("3#485"),MG_ESC("状态"), MG_ESC(c3->szstate),MG_ESC("故障码"),MG_ESC(c3->szerr),MG_ESC("波特率"),c3->baud,MG_ESC("设备名"),MG_ESC(c3->szdev),
  351. MG_ESC("备注"),MG_ESC(c3->szinfo),MG_ESC("循环次数"),c3->loopcnt,MG_ESC("循环时间"),c3->looptime,MG_ESC("请求次数"),c3->reqcnt,MG_ESC("失败次数"),c3->failcnt,
  352. MG_ESC("4#485"),MG_ESC("状态"), MG_ESC(c4->szstate),MG_ESC("故障码"),MG_ESC(c4->szerr),MG_ESC("波特率"),c4->baud,MG_ESC("设备名"),MG_ESC(c4->szdev),
  353. MG_ESC("备注"),MG_ESC(c4->szinfo),MG_ESC("循环次数"),c4->loopcnt,MG_ESC("循环时间"),c4->looptime,MG_ESC("请求次数"),c4->reqcnt,MG_ESC("失败次数"),c4->failcnt,
  354. MG_ESC("1#CAN"),MG_ESC("状态"), MG_ESC(c5->szState),MG_ESC("故障码"),MG_ESC(c5->szErr),MG_ESC("设备名"),MG_ESC(c5->szdev),
  355. MG_ESC("备注"),MG_ESC(c5->szinfo),MG_ESC("循环次数"),c5->Loopcnt,MG_ESC("循环时间"),c5->LoopTime,MG_ESC("发送帧数"),c5->WrCnt,MG_ESC("接收帧数"),c5->RdCnt,
  356. MG_ESC("2#CAN"),MG_ESC("状态"), MG_ESC(c6->szState),MG_ESC("故障码"),MG_ESC(c6->szErr),MG_ESC("设备名"),MG_ESC(c6->szdev),
  357. MG_ESC("备注"),MG_ESC(c6->szinfo),MG_ESC("循环次数"),c6->Loopcnt,MG_ESC("循环时间"),c6->LoopTime,MG_ESC("发送帧数"),c6->WrCnt,MG_ESC("接收帧数"),c6->RdCnt,
  358. MG_ESC("2#MQTT"),MG_ESC("状态"),MG_ESC(c8->szState),MG_ESC("发送计数"), c8->TotalSend,MG_ESC("接收计数"),c8->TotalRecv,MG_ESC("重连计数"), c8->TotalReconn,
  359. MG_ESC("4#MQTT"),MG_ESC("状态"),MG_ESC(c10->szState),MG_ESC("发送计数"), c10->TotalSend,MG_ESC("接收计数"), c10->TotalRecv,MG_ESC("重连计数"), c10->TotalReconn);
  360. }
  361. static void handle_comm_set(struct mg_connection *c, struct mg_str body) {
  362. struct chan485_t* ch = NULL;
  363. int idx = mg_json_get_long(body, "$.index", 0);
  364. int cmd = mg_json_get_long(body, "$.cmd", 0);
  365. printf("%s idx:%d cmd:%d\n", __func__, idx, cmd);
  366. if( idx >= 1 && idx <= 4 ){
  367. ch = &APPL.chan485[idx];
  368. ch->Cmd = cmd;
  369. mg_http_reply(c, 200, s_json_header, "true\n");
  370. }else{
  371. mg_http_reply(c, 403, s_json_header, "false\n");
  372. }
  373. }
  374. static void handle_meter_get(struct mg_connection *c) {
  375. struct Dtsd1352_t* m1 = &APPL.Dtsd1352[1]; //1#关口电表
  376. //struct Dtsd1352_t* m2 = &APPL.Dtsd1352[2]; //2#关口电表
  377. mg_http_reply(c, 200, s_json_header, "{\
  378. %m:{%m:%m,%m:%m,%m:%lld,%m:%f,%m:%f,%m:%f,%m:%f,%m:%f,%m:%f,%m:%f,%m:%f,%m:%f,%m:%f,%m:%f,%m:%f,%m:%f,%m:%d,%m:%d}}\n",
  379. /*1*/MG_ESC("关口电表"),MG_ESC("通信状态"), MG_ESC(m1->szCommState),MG_ESC("数据更新时间"),MG_ESC(m1->szLastUpdate),MG_ESC("通信失败次数"),m1->CommFailTotalCnt,
  380. MG_ESC("组合有功功率"), m1->com_active_p, MG_ESC("正向有功总电能"),m1->pos_active_e, MG_ESC("反向有功总电能"),m1->neg_active_e,
  381. MG_ESC("功率因数"),m1->pwr_factor, MG_ESC("正向有功需量"),m1->pos_active_dem,MG_ESC("反向有功需量"),m1->neg_active_dem,
  382. MG_ESC("A相电压"),m1->ua,MG_ESC("B相电压"),m1->ub,MG_ESC("C相电压"),m1->uc,MG_ESC("A相电流"),m1->ia,MG_ESC("B相电流"),m1->ib,MG_ESC("C相电流"),m1->ic,MG_ESC("频率"),m1->freq,MG_ESC("PT"),m1->PT,MG_ESC("CT"),m1->CT); //别忘了,
  383. /*2*//* MG_ESC("2#关口电表"),MG_ESC("通信状态"), MG_ESC(m2->szCommState),MG_ESC("数据更新时间"),MG_ESC(m2->szLastUpdate),MG_ESC("通信失败次数"),m2->CommFailTotalCnt,
  384. MG_ESC("组合有功功率"), m2->com_active_p, MG_ESC("正向有功总电能"),m2->pos_active_e, MG_ESC("反向有功总电能"),m2->neg_active_e,
  385. MG_ESC("功率因数"),m2->pwr_factor, MG_ESC("正向有功需量"),m2->pos_active_dem,MG_ESC("反向有功需量"),m2->neg_active_dem,
  386. MG_ESC("A相电压"),m2->ua,MG_ESC("B相电压"),m2->ub,MG_ESC("C相电压"),m2->uc,MG_ESC("A相电流"),m2->ia,MG_ESC("B相电流"),m2->ib,MG_ESC("C相电流"),m2->ic,MG_ESC("频率"),m2->freq,MG_ESC("PT"),m2->PT,MG_ESC("CT"),m2->CT); */
  387. }
  388. static void handel_meter_set(struct mg_connection *c, struct mg_str body){
  389. bool ok = true;
  390. int rc;
  391. long meter_nmb = mg_json_get_long(body, "$.meter_nmb", -1);
  392. long pt_ct = mg_json_get_long(body, "$.pt_ct", -1);;
  393. long value = mg_json_get_long(body, "$.value", -1);
  394. unsigned short start = 0x008D;
  395. unsigned short nbr = 1;
  396. unsigned short data[256];
  397. struct Dtsd1352_t* m = &APPL.Dtsd1352[meter_nmb]; // 选择电表(关口电表)
  398. if(meter_nmb != -1){
  399. syslog(LOG_INFO,"%s, meter_nmb:%d", __func__, meter_nmb);
  400. }else{
  401. ok = false;
  402. syslog(LOG_INFO,"%s, Get meter_nmb Fail", __func__);
  403. }
  404. if(pt_ct != -1){
  405. syslog(LOG_INFO,"%s, meter_nmb:%d", __func__, pt_ct);
  406. switch (pt_ct){
  407. case PT:
  408. data[0x008D - start] = value;
  409. rc = modbus_write_registers(m->ctx, start, nbr, data);
  410. break;
  411. case CT:
  412. data[0x008E - start] = value;
  413. rc = modbus_write_registers(m->ctx, start, nbr, data);
  414. break;
  415. default:
  416. break;
  417. }
  418. }else{
  419. ok = false;
  420. syslog(LOG_INFO,"%s, Get meter_nmb Fail", __func__);
  421. }
  422. mg_http_reply(c, 200, s_json_header,
  423. "{%m:%s,%m:%m}", //
  424. MG_ESC("status"), ok ? "true" : "false", //
  425. MG_ESC("message"), MG_ESC(ok ? "Success" : "Failed"));
  426. }
  427. static void handle_firmware_upload(struct mg_connection *c,
  428. struct mg_http_message *hm) {
  429. char name[64], offset[20], total[20];
  430. struct mg_str data = hm->body;
  431. long ofs = -1, tot = -1;
  432. name[0] = offset[0] = '\0';
  433. mg_http_get_var(&hm->query, "name", name, sizeof(name));
  434. mg_http_get_var(&hm->query, "offset", offset, sizeof(offset));
  435. mg_http_get_var(&hm->query, "total", total, sizeof(total));
  436. MG_INFO(("File %s, offset %s, len %lu", name, offset, data.len));
  437. if ((ofs = mg_json_get_long(mg_str(offset), "$", -1)) < 0 ||
  438. (tot = mg_json_get_long(mg_str(total), "$", -1)) < 0) {
  439. mg_http_reply(c, 500, "", "offset and total not set\n");
  440. } else if (ofs == 0 && mg_ota_begin((size_t) tot) == false) {
  441. mg_http_reply(c, 500, "", "mg_ota_begin(%ld) failed\n", tot);
  442. } else if (data.len > 0 && mg_ota_write(data.ptr, data.len) == false) {
  443. mg_http_reply(c, 500, "", "mg_ota_write(%lu) @%ld failed\n", data.len, ofs);
  444. mg_ota_end();
  445. } else if (data.len == 0 && mg_ota_end() == false) {
  446. mg_http_reply(c, 500, "", "mg_ota_end() failed\n", tot);
  447. } else {
  448. mg_http_reply(c, 200, s_json_header, "true\n");
  449. if (data.len == 0) {
  450. // Successful mg_ota_end() called, schedule device reboot
  451. mg_timer_add(c->mgr, 500, 0, (void (*)(void *)) mg_device_reset, NULL);
  452. }
  453. }
  454. }
  455. static void handle_firmware_commit(struct mg_connection *c) {
  456. mg_http_reply(c, 200, s_json_header, "%s\n",
  457. mg_ota_commit() ? "true" : "false");
  458. }
  459. static void handle_firmware_rollback(struct mg_connection *c) {
  460. mg_http_reply(c, 200, s_json_header, "%s\n",
  461. mg_ota_rollback() ? "true" : "false");
  462. }
  463. static size_t print_status(void (*out)(char, void *), void *ptr, va_list *ap) {
  464. int fw = va_arg(*ap, int);
  465. return mg_xprintf(out, ptr, "{%m:%d,%m:%c%lx%c,%m:%u,%m:%u}\n",
  466. MG_ESC("status"), mg_ota_status(fw), MG_ESC("crc32"), '"',
  467. mg_ota_crc32(fw), '"', MG_ESC("size"), mg_ota_size(fw),
  468. MG_ESC("timestamp"), mg_ota_timestamp(fw));
  469. }
  470. static void handle_firmware_status(struct mg_connection *c) {
  471. mg_http_reply(c, 200, s_json_header, "[%M,%M]\n", print_status,
  472. MG_FIRMWARE_CURRENT, print_status, MG_FIRMWARE_PREVIOUS);
  473. }
  474. static void handle_device_reset(struct mg_connection *c) {
  475. mg_http_reply(c, 200, s_json_header, "true\n");
  476. mg_timer_add(c->mgr, 500, 0, (void (*)(void *)) mg_device_reset, NULL);
  477. }
  478. static void handle_device_eraselast(struct mg_connection *c) {
  479. size_t ss = mg_flash_sector_size(), size = mg_flash_size();
  480. char *base = (char *) mg_flash_start(), *last = base + size - ss;
  481. if (mg_flash_bank() == 2) last -= size / 2;
  482. mg_flash_erase(last);
  483. mg_http_reply(c, 200, s_json_header, "true\n");
  484. }
  485. // HTTP request handler function
  486. static void fn(struct mg_connection *c, int ev, void *ev_data) {
  487. if (ev == MG_EV_ACCEPT) {
  488. if (c->fn_data != NULL) { // TLS listener!
  489. struct mg_tls_opts opts = {0};
  490. opts.cert = mg_unpacked("/certs/server_cert.pem");
  491. opts.key = mg_unpacked("/certs/server_key.pem");
  492. mg_tls_init(c, &opts);
  493. }
  494. } else if (ev == MG_EV_HTTP_MSG) {
  495. struct mg_http_message *hm = (struct mg_http_message *) ev_data;
  496. struct user *u = authenticate(hm);
  497. if ( 0 ) { // mg_http_match_uri(hm, "/api/#") && u == NULL
  498. mg_http_reply(c, 403, "", "Not Authorised\n");
  499. } else if (mg_http_match_uri(hm, "/api/login")) {
  500. handle_login(c, u);
  501. } else if (mg_http_match_uri(hm, "/api/logout")) {
  502. handle_logout(c);
  503. } else if (mg_http_match_uri(hm, "/api/debug")) {
  504. handle_debug(c, hm);
  505. } else if (mg_http_match_uri(hm, "/api/settings/get")) {
  506. handle_settings_get(c);
  507. } else if (mg_http_match_uri(hm, "/api/settings/set")) {
  508. handle_settings_set(c, hm->body);
  509. }else if (mg_http_match_uri(hm, "/api/stats/get")) {
  510. handle_stats_get(c);
  511. } else if (mg_http_match_uri(hm, "/api/events/get")) {
  512. handle_events_get(c, hm);
  513. } else if (mg_http_match_uri(hm, "/api/comm/get")) {
  514. handle_comm_get(c);
  515. } else if (mg_http_match_uri(hm, "/api/comm/set")) {
  516. handle_comm_set(c, hm->body);
  517. } else if (mg_http_match_uri(hm, "/api/meter/get")) {
  518. handle_meter_get(c);
  519. } else if (mg_http_match_uri(hm, "/api/meter/set")) {
  520. handel_meter_set(c, hm->body);
  521. } else if (mg_http_match_uri(hm, "/api/ctl/get")) {
  522. handle_ctl_get(c);
  523. } else if (mg_http_match_uri(hm, "/api/firmware/upload")) {
  524. handle_firmware_upload(c, hm);
  525. } else if (mg_http_match_uri(hm, "/api/firmware/commit")) {
  526. handle_firmware_commit(c);
  527. } else if (mg_http_match_uri(hm, "/api/firmware/rollback")) {
  528. handle_firmware_rollback(c);
  529. } else if (mg_http_match_uri(hm, "/api/firmware/status")) {
  530. handle_firmware_status(c);
  531. } else if (mg_http_match_uri(hm, "/api/device/reset")) {
  532. handle_device_reset(c);
  533. } else if (mg_http_match_uri(hm, "/api/device/eraselast")) {
  534. handle_device_eraselast(c);
  535. } else {
  536. struct mg_http_serve_opts opts;
  537. memset(&opts, 0, sizeof(opts));
  538. #if MG_ARCH == MG_ARCH_UNIX || MG_ARCH == MG_ARCH_WIN32
  539. opts.root_dir = "web_root"; // On workstations, use filesystem
  540. #else
  541. opts.root_dir = "/web_root"; // On embedded, use packed files
  542. opts.fs = &mg_fs_packed;
  543. #endif
  544. mg_http_serve_dir(c, ev_data, &opts);
  545. }
  546. MG_DEBUG(("%lu %.*s %.*s -> %.*s", c->id, (int) hm->method.len,
  547. hm->method.ptr, (int) hm->uri.len, hm->uri.ptr, (int) 3,
  548. &c->send.buf[9]));
  549. }
  550. }
  551. void web_init(struct mg_mgr *mgr) {
  552. s_settings.device_name = strdup("My Device");
  553. mg_http_listen(mgr, HTTP_URL, fn, NULL);
  554. mg_http_listen(mgr, HTTPS_URL, fn, (void *) 1);
  555. mg_timer_add(mgr, 3600 * 1000, MG_TIMER_RUN_NOW | MG_TIMER_REPEAT,
  556. timer_sntp_fn, mgr);
  557. }