net.c 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718
  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) {
  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[2];
  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_LOCAL_USERNAME:
  226. p = mg_json_get_str(body, "$.param");
  227. if( p != NULL){
  228. strcpy(set->szCloudLocalUserName, p);
  229. free(p);
  230. if(appl_cfg_save() != 0){
  231. appl_cfg_set_err();
  232. ok = false;
  233. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_LOCAL_USERNAME, appl_cfg_save Fail", __func__);
  234. }
  235. }else{
  236. ok = false;
  237. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_LOCAL_USERNAME, mg_json_get_str Fail", __func__);
  238. }
  239. break;
  240. case SETTINGS_CMD_SET_LOCAL_PASSWD:
  241. p = mg_json_get_str(body, "$.param");
  242. if( p != NULL){
  243. strcpy(set->szCloudLocalPasswd, p);
  244. free(p);
  245. if(appl_cfg_save() != 0){
  246. appl_cfg_set_err();
  247. ok = false;
  248. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_LOCAL_PASSWD, appl_cfg_save Fail", __func__);
  249. }
  250. }else{
  251. ok = false;
  252. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_LOCAL_PASSWD, mg_json_get_str Fail", __func__);
  253. }
  254. break;
  255. case SETTINGS_CMD_SET_LOCAL_URL:
  256. p = mg_json_get_str(body, "$.param");
  257. if( p != NULL){
  258. strcpy(set->szCloudLocalUrl, p);
  259. free(p);
  260. if(appl_cfg_save() != 0){
  261. appl_cfg_set_err();
  262. ok = false;
  263. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_LOCAL_URL, appl_cfg_save Fail", __func__);
  264. }
  265. }else{
  266. ok = false;
  267. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_LOCAL_URL, mg_json_get_str Fail", __func__);
  268. }
  269. break;
  270. case SETTINGS_CMD_SET_LOCAL_CLIENTID:
  271. p = mg_json_get_str(body, "$.param");
  272. if( p != NULL){
  273. strcpy(set->szLocalClientId, p);
  274. free(p);
  275. if(appl_cfg_save() != 0){
  276. appl_cfg_set_err();
  277. ok = false;
  278. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_LOCAL_CLIENTID, appl_cfg_save Fail", __func__);
  279. }
  280. }else{
  281. ok = false;
  282. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_LOCAL_CLIENTID, mg_json_get_str Fail", __func__);
  283. }
  284. break;
  285. case SETTINGS_CMD_SET_UPLOADHIHGSPEED:
  286. val = mg_json_get_long(body, "$.param", -1);
  287. if( val == -1 ){
  288. ok = false;
  289. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_UPLOADHIHGSPEED, mg_json_get_long Fail", __func__);
  290. }else{
  291. set->UploadHighSpeed = val;
  292. if(appl_cfg_save() != 0){
  293. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_UPLOADHIHGSPEED, appl_cfg_save Fail", __func__);
  294. appl_cfg_set_err();
  295. ok = false;
  296. }else{
  297. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_UPLOADHIHGSPEED, set->UploadHighSpeed is set to %d", __func__, set->UploadHighSpeed);
  298. }
  299. }
  300. break;
  301. case SETTINGS_CMD_SET_UPLOADMEDIUMSPEED:
  302. val = mg_json_get_long(body, "$.param", -1);
  303. if( val == -1 ){
  304. ok = false;
  305. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_UPLOADMEDIUMSPEED, mg_json_get_long Fail", __func__);
  306. }else{
  307. set->UploadMediumSpeed = val;
  308. if(appl_cfg_save() != 0){
  309. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_UPLOADMEDIUMSPEED, appl_cfg_save Fail", __func__);
  310. appl_cfg_set_err();
  311. ok = false;
  312. }else{
  313. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_UPLOADMEDIUMSPEED, set->UploadMediumSpeed is set to %d", __func__, set->UploadMediumSpeed);
  314. }
  315. }
  316. break;
  317. case SETTINGS_CMD_SET_UPLOADSLOWSPEED:
  318. val = mg_json_get_long(body, "$.param", -1);
  319. if( val == -1 ){
  320. ok = false;
  321. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_UPLOADSLOWSPEED, mg_json_get_long Fail", __func__);
  322. }else{
  323. set->UploadSlowSpeed = val;
  324. if(appl_cfg_save() != 0){
  325. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_UPLOADSLOWSPEED, appl_cfg_save Fail", __func__);
  326. appl_cfg_set_err();
  327. ok = false;
  328. }else{
  329. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_UPLOADSLOWSPEED, set->UploadSlowSpeed is set to %d", __func__, set->UploadSlowSpeed);
  330. }
  331. }
  332. break;
  333. case SETTINGS_CMD_REGISTER:
  334. mqtt->Cmd = CMD_MQTT_REGISTER;
  335. break;
  336. case SETTINGS_CMD_SET_DATAKEEPDAY:
  337. val = mg_json_get_long(body, "$.param", -1);
  338. if( val == -1 ){
  339. ok = false;
  340. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_DATAKEEPDAY, mg_json_get_long Fail", __func__);
  341. }else{
  342. set->DataKeepDay = val;
  343. if(appl_cfg_save() != 0){
  344. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_DATAKEEPDAY, appl_cfg_save Fail", __func__);
  345. appl_cfg_set_err();
  346. ok = false;
  347. }else{
  348. syslog(LOG_INFO, "%s, SETTINGS_CMD_SET_DATAKEEPDAY, set->DataKeepDay is set to %d", __func__, set->DataKeepDay);
  349. }
  350. }
  351. break;
  352. case SEETINGS_CMD_METER_DATA_FILE_NAME:
  353. p = mg_json_get_str(body, "$.param");
  354. if( p != NULL){
  355. strcpy(set->meterDataLoadFileName, p);
  356. free(p);
  357. if(appl_cfg_save() != 0){
  358. appl_cfg_set_err();
  359. ok = false;
  360. syslog(LOG_INFO, "%s, SEETINGS_CMD_METER_DATA_FILE_NAME, appl_cfg_save Fail", __func__);
  361. }
  362. }else{
  363. ok = false;
  364. syslog(LOG_INFO, "%s, SEETINGS_CMD_METER_DATA_FILE_NAME, mg_json_get_str Fail", __func__);
  365. }
  366. break;
  367. case SETTINGS_CMD_TEST:
  368. syslog(LOG_INFO,"%s, Get Test Cmd", __func__);
  369. rc = appl_snap_rmdir("./snap/test");
  370. syslog(LOG_INFO,"%s, rc:%d", __func__, rc);
  371. break;
  372. default:
  373. break;
  374. }
  375. }else{
  376. ok = false;
  377. syslog(LOG_INFO,"%s, Get Cmd Fail", __func__);
  378. }
  379. mg_http_reply(c, 200, s_json_header,
  380. "{%m:%s,%m:%m}", //
  381. MG_ESC("status"), ok ? "true" : "false", //
  382. MG_ESC("message"), MG_ESC(ok ? "Success" : "Failed"));
  383. }
  384. static void handle_ctl_get(struct mg_connection *c) {
  385. struct Settings_t* set = &APPL.Set.s;
  386. struct Snap_t* snap = &APPL.Snap;
  387. mg_http_reply(c, 200, s_json_header, "{\
  388. %m:%m,%m:%m,\
  389. %m:%m}\n",
  390. /*1*/MG_ESC("数据记录状态"), MG_ESC(snap->szState),MG_ESC("参数配置状态"), MG_ESC(set->szState),
  391. /*2*/MG_ESC("软件版本"),MG_ESC(VERSION));
  392. }
  393. static void handle_settings_get(struct mg_connection *c) {
  394. struct Settings_t* set = &APPL.Set.s;
  395. mg_http_reply(c, 200, s_json_header, "{\
  396. %m:%m,%m:%m,%m:%m,%m:%m,\
  397. %m:%m,%m:%m,%m:%m,%m:%m,\
  398. %m:%d,%m:%d,%m:%d,%m:%d}\n",
  399. 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),
  400. MG_ESC("Local用户名"),MG_ESC(set->szCloudLocalUserName),MG_ESC("Local密码"),MG_ESC(set->szCloudLocalPasswd),MG_ESC("Local Url"),MG_ESC(set->szCloudLocalUrl),MG_ESC("Local Client ID"),MG_ESC(set->szLocalClientId),
  401. MG_ESC("数据保留天数"),set->DataKeepDay,MG_ESC("上传高速"),set->UploadHighSpeed,MG_ESC("上传中速"),set->UploadMediumSpeed,MG_ESC("上传低速"),set->UploadSlowSpeed);
  402. }
  403. static void handle_comm_get(struct mg_connection *c) {
  404. struct chan485_t* c1 = &APPL.chan485[1];
  405. struct chan485_t* c2 = &APPL.chan485[2];
  406. struct chan485_t* c3 = &APPL.chan485[3];
  407. struct chan485_t* c4 = &APPL.chan485[4];
  408. struct chancan_t* c5 = &APPL.chancan[1];
  409. struct chancan_t* c6 = &APPL.chancan[2];
  410. struct chanmqtt_t* c7 = &APPL.chanmqtt[1];
  411. struct chanmqtt_t* c8 = &APPL.chanmqtt[2];
  412. struct chanmqtt_t* c9 = &APPL.chanmqtt[3];
  413. struct chanmqtt_t* c10 = &APPL.chanmqtt[4];
  414. mg_http_reply(c, 200, s_json_header, "{\
  415. %m:{%m:%m,%m:%m,%m:%d,%m:%m,%m:%m,%m:%lld,%m:%lld,%m:%lld,%m:%lld},\
  416. %m:{%m:%m,%m:%m,%m:%d,%m:%m,%m:%m,%m:%lld,%m:%lld,%m:%lld,%m:%lld},\
  417. %m:{%m:%m,%m:%m,%m:%d,%m:%m,%m:%m,%m:%lld,%m:%lld,%m:%lld,%m:%lld},\
  418. %m:{%m:%m,%m:%m,%m:%d,%m:%m,%m:%m,%m:%lld,%m:%lld,%m:%lld,%m:%lld},\
  419. %m:{%m:%m,%m:%m,%m:%m,%m:%m,%m:%lld,%m:%lld,%m:%lld,%m:%lld},\
  420. %m:{%m:%m,%m:%m,%m:%m,%m:%m,%m:%lld,%m:%lld,%m:%lld,%m:%lld},\
  421. %m:{%m:%m,%m:%lld,%m:%lld,%m:%lld},\
  422. %m:{%m:%m,%m:%lld,%m:%lld,%m:%lld}}\n",
  423. 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),
  424. MG_ESC("备注"), MG_ESC(c1->szinfo), MG_ESC("循环次数"),c1->loopcnt, MG_ESC("循环时间"),c1->looptime, MG_ESC("请求次数"),c1->reqcnt, MG_ESC("失败次数"),c1->failcnt,
  425. 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),
  426. MG_ESC("备注"),MG_ESC(c2->szinfo),MG_ESC("循环次数"),c2->loopcnt,MG_ESC("循环时间"),c2->looptime,MG_ESC("请求次数"),c2->reqcnt,MG_ESC("失败次数"),c2->failcnt,
  427. 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),
  428. MG_ESC("备注"),MG_ESC(c3->szinfo),MG_ESC("循环次数"),c3->loopcnt,MG_ESC("循环时间"),c3->looptime,MG_ESC("请求次数"),c3->reqcnt,MG_ESC("失败次数"),c3->failcnt,
  429. 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),
  430. MG_ESC("备注"),MG_ESC(c4->szinfo),MG_ESC("循环次数"),c4->loopcnt,MG_ESC("循环时间"),c4->looptime,MG_ESC("请求次数"),c4->reqcnt,MG_ESC("失败次数"),c4->failcnt,
  431. MG_ESC("1#CAN"),MG_ESC("状态"), MG_ESC(c5->szState),MG_ESC("故障码"),MG_ESC(c5->szErr),MG_ESC("设备名"),MG_ESC(c5->szdev),
  432. MG_ESC("备注"),MG_ESC(c5->szinfo),MG_ESC("循环次数"),c5->Loopcnt,MG_ESC("循环时间"),c5->LoopTime,MG_ESC("发送帧数"),c5->WrCnt,MG_ESC("接收帧数"),c5->RdCnt,
  433. MG_ESC("2#CAN"),MG_ESC("状态"), MG_ESC(c6->szState),MG_ESC("故障码"),MG_ESC(c6->szErr),MG_ESC("设备名"),MG_ESC(c6->szdev),
  434. MG_ESC("备注"),MG_ESC(c6->szinfo),MG_ESC("循环次数"),c6->Loopcnt,MG_ESC("循环时间"),c6->LoopTime,MG_ESC("发送帧数"),c6->WrCnt,MG_ESC("接收帧数"),c6->RdCnt,
  435. MG_ESC("2#MQTT"),MG_ESC("状态"),MG_ESC(c8->szState),MG_ESC("发送计数"), c8->TotalSend,MG_ESC("接收计数"),c8->TotalRecv,MG_ESC("重连计数"), c8->TotalReconn,
  436. MG_ESC("4#MQTT"),MG_ESC("状态"),MG_ESC(c10->szState),MG_ESC("发送计数"), c10->TotalSend,MG_ESC("接收计数"), c10->TotalRecv,MG_ESC("重连计数"), c10->TotalReconn);
  437. }
  438. static void handle_comm_set(struct mg_connection *c, struct mg_str body) {
  439. struct chan485_t* ch = NULL;
  440. int idx = mg_json_get_long(body, "$.index", 0);
  441. int cmd = mg_json_get_long(body, "$.cmd", 0);
  442. printf("%s idx:%d cmd:%d\n", __func__, idx, cmd);
  443. if( idx >= 1 && idx <= 4 ){
  444. ch = &APPL.chan485[idx];
  445. ch->Cmd = cmd;
  446. mg_http_reply(c, 200, s_json_header, "true\n");
  447. }else{
  448. mg_http_reply(c, 403, s_json_header, "false\n");
  449. }
  450. }
  451. static void handle_meter_get(struct mg_connection *c) {
  452. struct Dtsd1352_t* m; // 关口电表
  453. char buf[8096];
  454. char temp_buf[1024];
  455. int i = 0;
  456. buf[0] = 0;
  457. for(i = 1;i < appl_get_meter_config_nbr();i ++){
  458. m = &APPL.Dtsd1352[i];
  459. if(i > 1)
  460. strcat(buf,",");
  461. mg_snprintf(temp_buf,1024,
  462. "%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}",
  463. MG_ESC(m->szinfo),MG_ESC("通信状态"), MG_ESC(m->szCommState),MG_ESC("数据更新时间"),MG_ESC(m->szLastUpdate),MG_ESC("通信失败次数"),m->CommFailTotalCnt,
  464. MG_ESC("组合有功功率"), m->com_active_p, MG_ESC("正向有功总电能"),m->pos_active_e, MG_ESC("反向有功总电能"),m->neg_active_e,
  465. MG_ESC("功率因数"),m->pwr_factor, MG_ESC("正向有功需量"),m->pos_active_dem,MG_ESC("反向有功需量"),m->neg_active_dem,
  466. MG_ESC("A相电压"),m->ua,MG_ESC("B相电压"),m->ub,MG_ESC("C相电压"),m->uc,MG_ESC("A相电流"),m->ia,MG_ESC("B相电流"),m->ib,MG_ESC("C相电流"),m->ic,MG_ESC("频率"),m->freq,MG_ESC("PT"),m->PT,MG_ESC("CT"),m->CT);
  467. strcat(buf,temp_buf);
  468. }
  469. mg_http_reply(c, 200, s_json_header,"{%s}",buf);
  470. // mg_http_reply(c, 200, s_json_header, "{\
  471. // %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",
  472. // /*1*/MG_ESC("关口电表"),MG_ESC("通信状态"), MG_ESC(m1->szCommState),MG_ESC("数据更新时间"),MG_ESC(m1->szLastUpdate),MG_ESC("通信失败次数"),m1->CommFailTotalCnt,
  473. // MG_ESC("组合有功功率"), m1->com_active_p, MG_ESC("正向有功总电能"),m1->pos_active_e, MG_ESC("反向有功总电能"),m1->neg_active_e,
  474. // MG_ESC("功率因数"),m1->pwr_factor, MG_ESC("正向有功需量"),m1->pos_active_dem,MG_ESC("反向有功需量"),m1->neg_active_dem,
  475. // 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);
  476. }
  477. static void handle_meter_set(struct mg_connection *c, struct mg_str body){
  478. bool ok = true;
  479. int rc;
  480. long meter_nmb = mg_json_get_long(body, "$.meter_nmb", -1);
  481. long pt_ct = mg_json_get_long(body, "$.pt_ct", -1);;
  482. long value = mg_json_get_long(body, "$.value", -1);
  483. unsigned short start = 0x008D;
  484. unsigned short nbr = 1;
  485. unsigned short data[256];
  486. struct Dtsd1352_t* m = &APPL.Dtsd1352[meter_nmb]; // 选择电表(关口电表)
  487. if(meter_nmb != -1){
  488. syslog(LOG_INFO,"%s, meter_nmb:%d", __func__, meter_nmb);
  489. }else{
  490. ok = false;
  491. syslog(LOG_INFO,"%s, Get meter_nmb Fail", __func__);
  492. }
  493. if(pt_ct != -1){
  494. syslog(LOG_INFO,"%s, meter_nmb:%d", __func__, pt_ct);
  495. switch (pt_ct){
  496. case PT:
  497. data[0x008D - start] = value;
  498. rc = modbus_write_registers(m->ctx, start, nbr, data);
  499. break;
  500. case CT:
  501. data[0x008E - start] = value;
  502. rc = modbus_write_registers(m->ctx, start, nbr, data);
  503. break;
  504. default:
  505. break;
  506. }
  507. }else{
  508. ok = false;
  509. syslog(LOG_INFO,"%s, Get meter_nmb Fail", __func__);
  510. }
  511. mg_http_reply(c, 200, s_json_header,
  512. "{%m:%s,%m:%m}", //
  513. MG_ESC("status"), ok ? "true" : "false", //
  514. MG_ESC("message"), MG_ESC(ok ? "Success" : "Failed"));
  515. }
  516. static void handle_firmware_upload(struct mg_connection *c,
  517. struct mg_http_message *hm) {
  518. char name[64], offset[20], total[20];
  519. struct mg_str data = hm->body;
  520. long ofs = -1, tot = -1;
  521. name[0] = offset[0] = '\0';
  522. mg_http_get_var(&hm->query, "name", name, sizeof(name));
  523. mg_http_get_var(&hm->query, "offset", offset, sizeof(offset));
  524. mg_http_get_var(&hm->query, "total", total, sizeof(total));
  525. MG_INFO(("File %s, offset %s, len %lu", name, offset, data.len));
  526. if ((ofs = mg_json_get_long(mg_str(offset), "$", -1)) < 0 ||
  527. (tot = mg_json_get_long(mg_str(total), "$", -1)) < 0) {
  528. mg_http_reply(c, 500, "", "offset and total not set\n");
  529. } else if (ofs == 0 && mg_ota_begin((size_t) tot) == false) {
  530. mg_http_reply(c, 500, "", "mg_ota_begin(%ld) failed\n", tot);
  531. } else if (data.len > 0 && mg_ota_write(data.ptr, data.len) == false) {
  532. mg_http_reply(c, 500, "", "mg_ota_write(%lu) @%ld failed\n", data.len, ofs);
  533. mg_ota_end();
  534. } else if (data.len == 0 && mg_ota_end() == false) {
  535. mg_http_reply(c, 500, "", "mg_ota_end() failed\n", tot);
  536. } else {
  537. mg_http_reply(c, 200, s_json_header, "true\n");
  538. if (data.len == 0) {
  539. // Successful mg_ota_end() called, schedule device reboot
  540. mg_timer_add(c->mgr, 500, 0, (void (*)(void *)) mg_device_reset, NULL);
  541. }
  542. }
  543. }
  544. static void handle_firmware_commit(struct mg_connection *c) {
  545. mg_http_reply(c, 200, s_json_header, "%s\n",
  546. mg_ota_commit() ? "true" : "false");
  547. }
  548. static void handle_firmware_rollback(struct mg_connection *c) {
  549. mg_http_reply(c, 200, s_json_header, "%s\n",
  550. mg_ota_rollback() ? "true" : "false");
  551. }
  552. static size_t print_status(void (*out)(char, void *), void *ptr, va_list *ap) {
  553. int fw = va_arg(*ap, int);
  554. return mg_xprintf(out, ptr, "{%m:%d,%m:%c%lx%c,%m:%u,%m:%u}\n",
  555. MG_ESC("status"), mg_ota_status(fw), MG_ESC("crc32"), '"',
  556. mg_ota_crc32(fw), '"', MG_ESC("size"), mg_ota_size(fw),
  557. MG_ESC("timestamp"), mg_ota_timestamp(fw));
  558. }
  559. static void handle_firmware_status(struct mg_connection *c) {
  560. mg_http_reply(c, 200, s_json_header, "[%M,%M]\n", print_status,
  561. MG_FIRMWARE_CURRENT, print_status, MG_FIRMWARE_PREVIOUS);
  562. }
  563. static void handle_device_reset(struct mg_connection *c) {
  564. mg_http_reply(c, 200, s_json_header, "true\n");
  565. mg_timer_add(c->mgr, 500, 0, (void (*)(void *)) mg_device_reset, NULL);
  566. }
  567. static void handle_device_eraselast(struct mg_connection *c) {
  568. size_t ss = mg_flash_sector_size(), size = mg_flash_size();
  569. char *base = (char *) mg_flash_start(), *last = base + size - ss;
  570. if (mg_flash_bank() == 2) last -= size / 2;
  571. mg_flash_erase(last);
  572. mg_http_reply(c, 200, s_json_header, "true\n");
  573. }
  574. // HTTP request handler function
  575. static void fn(struct mg_connection *c, int ev, void *ev_data) {
  576. if (ev == MG_EV_ACCEPT) {
  577. if (c->fn_data != NULL) { // TLS listener!
  578. struct mg_tls_opts opts = {0};
  579. opts.cert = mg_unpacked("/certs/server_cert.pem");
  580. opts.key = mg_unpacked("/certs/server_key.pem");
  581. mg_tls_init(c, &opts);
  582. }
  583. } else if (ev == MG_EV_HTTP_MSG) {
  584. struct mg_http_message *hm = (struct mg_http_message *) ev_data;
  585. struct user *u = authenticate(hm);
  586. if ( 0 ) { // mg_http_match_uri(hm, "/api/#") && u == NULL
  587. mg_http_reply(c, 403, "", "Not Authorised\n");
  588. } else if (mg_http_match_uri(hm, "/api/login")) {
  589. handle_login(c, u);
  590. } else if (mg_http_match_uri(hm, "/api/logout")) {
  591. handle_logout(c);
  592. } else if (mg_http_match_uri(hm, "/api/debug")) {
  593. handle_debug(c, hm);
  594. } else if (mg_http_match_uri(hm, "/api/settings/get")) {
  595. handle_settings_get(c);
  596. } else if (mg_http_match_uri(hm, "/api/settings/set")) {
  597. handle_settings_set(c, hm->body);
  598. }else if (mg_http_match_uri(hm, "/api/stats/get")) {
  599. handle_stats_get(c);
  600. } else if (mg_http_match_uri(hm, "/api/events/get")) {
  601. handle_events_get(c, hm);
  602. } else if (mg_http_match_uri(hm, "/api/comm/get")) {
  603. handle_comm_get(c);
  604. } else if (mg_http_match_uri(hm, "/api/comm/set")) {
  605. handle_comm_set(c, hm->body);
  606. } else if (mg_http_match_uri(hm, "/api/meter/get")) {
  607. handle_meter_get(c);
  608. } else if (mg_http_match_uri(hm, "/api/meter/set")) {
  609. handle_meter_set(c,hm->body);
  610. } else if (mg_http_match_uri(hm, "/api/ctl/get")) {
  611. handle_ctl_get(c);
  612. } else if (mg_http_match_uri(hm, "/api/firmware/upload")) {
  613. handle_firmware_upload(c, hm);
  614. } else if (mg_http_match_uri(hm, "/api/firmware/commit")) {
  615. handle_firmware_commit(c);
  616. } else if (mg_http_match_uri(hm, "/api/firmware/rollback")) {
  617. handle_firmware_rollback(c);
  618. } else if (mg_http_match_uri(hm, "/api/firmware/status")) {
  619. handle_firmware_status(c);
  620. } else if (mg_http_match_uri(hm, "/api/device/reset")) {
  621. handle_device_reset(c);
  622. } else if (mg_http_match_uri(hm, "/api/device/eraselast")) {
  623. handle_device_eraselast(c);
  624. } else {
  625. struct mg_http_serve_opts opts;
  626. memset(&opts, 0, sizeof(opts));
  627. #if MG_ARCH == MG_ARCH_UNIX || MG_ARCH == MG_ARCH_WIN32
  628. opts.root_dir = "web_root"; // On workstations, use filesystem
  629. #else
  630. opts.root_dir = "/web_root"; // On embedded, use packed files
  631. opts.fs = &mg_fs_packed;
  632. #endif
  633. mg_http_serve_dir(c, ev_data, &opts);
  634. }
  635. MG_DEBUG(("%lu %.*s %.*s -> %.*s", c->id, (int) hm->method.len,
  636. hm->method.ptr, (int) hm->uri.len, hm->uri.ptr, (int) 3,
  637. &c->send.buf[9]));
  638. }
  639. }
  640. void web_init(struct mg_mgr *mgr) {
  641. s_settings.device_name = strdup("My Device");
  642. mg_http_listen(mgr, HTTP_URL, fn, NULL);
  643. mg_http_listen(mgr, HTTPS_URL, fn, (void *) 1);
  644. mg_timer_add(mgr, 3600 * 1000, MG_TIMER_RUN_NOW | MG_TIMER_REPEAT,
  645. timer_sntp_fn, mgr);
  646. }