2020-12-05 19:26:32 +08:00
|
|
|
// Copyright (c) 2004-2013 Sergey Lyubka
|
2021-03-09 16:21:29 +08:00
|
|
|
// Copyright (c) 2013-2021 Cesanta Software Limited
|
2020-12-05 19:26:32 +08:00
|
|
|
// All rights reserved
|
|
|
|
//
|
|
|
|
// This software is dual-licensed: you can redistribute it and/or modify
|
|
|
|
// it under the terms of the GNU General Public License version 2 as
|
|
|
|
// published by the Free Software Foundation. For the terms of this
|
|
|
|
// license, see <http://www.gnu.org/licenses/>.
|
|
|
|
//
|
|
|
|
// You are free to use this software under the terms of the GNU General
|
|
|
|
// Public License, but WITHOUT ANY WARRANTY; without even the implied
|
|
|
|
// warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
|
|
|
|
// See the GNU General Public License for more details.
|
|
|
|
//
|
|
|
|
// Alternatively, you can license this software under a commercial
|
|
|
|
// license, as set out in <https://www.cesanta.com/license>.
|
2019-03-19 20:59:13 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
#include "mongoose.h"
|
2017-03-24 02:39:57 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
#ifdef MG_ENABLE_LINES
|
|
|
|
#line 1 "src/private.h"
|
2017-03-24 02:39:57 +08:00
|
|
|
#endif
|
2020-12-05 19:26:32 +08:00
|
|
|
void mg_connect_resolved(struct mg_connection *);
|
2017-03-24 02:39:57 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
#ifdef MG_ENABLE_LINES
|
|
|
|
#line 1 "src/base64.c"
|
2014-02-20 00:38:53 +08:00
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
#include <string.h>
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
static int mg_b64idx(int c) {
|
|
|
|
if (c < 26) {
|
|
|
|
return c + 'A';
|
|
|
|
} else if (c < 52) {
|
|
|
|
return c - 26 + 'a';
|
|
|
|
} else if (c < 62) {
|
|
|
|
return c - 52 + '0';
|
2015-11-01 00:26:19 +08:00
|
|
|
} else {
|
2020-12-05 19:26:32 +08:00
|
|
|
return c == 62 ? '+' : '/';
|
2015-11-01 00:26:19 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2014-05-13 06:22:57 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
static int mg_b64rev(int c) {
|
|
|
|
if (c >= 'A' && c <= 'Z') {
|
|
|
|
return c - 'A';
|
|
|
|
} else if (c >= 'a' && c <= 'z') {
|
|
|
|
return c + 26 - 'a';
|
|
|
|
} else if (c >= '0' && c <= '9') {
|
|
|
|
return c + 52 - '0';
|
|
|
|
} else if (c == '+') {
|
|
|
|
return 62;
|
|
|
|
} else if (c == '/') {
|
|
|
|
return 63;
|
|
|
|
} else if (c == '=') {
|
|
|
|
return 64;
|
|
|
|
} else {
|
|
|
|
return -1;
|
2014-10-09 16:26:45 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
int mg_base64_update(unsigned char ch, char *to, int n) {
|
2021-05-29 01:45:35 +08:00
|
|
|
int rem = (n & 3) % 3;
|
2020-12-05 19:26:32 +08:00
|
|
|
if (rem == 0) {
|
2021-05-29 01:30:42 +08:00
|
|
|
to[n] = (char) mg_b64idx(ch >> 2);
|
|
|
|
to[++n] = (char) ((ch & 3) << 4);
|
2020-12-05 19:26:32 +08:00
|
|
|
} else if (rem == 1) {
|
2021-05-29 01:30:42 +08:00
|
|
|
to[n] = (char) mg_b64idx(to[n] | (ch >> 4));
|
|
|
|
to[++n] = (char) ((ch & 15) << 2);
|
2020-12-05 19:26:32 +08:00
|
|
|
} else {
|
2021-05-29 01:30:42 +08:00
|
|
|
to[n] = (char) mg_b64idx(to[n] | (ch >> 6));
|
|
|
|
to[++n] = (char) mg_b64idx(ch & 63);
|
2020-12-05 19:26:32 +08:00
|
|
|
n++;
|
|
|
|
}
|
|
|
|
return n;
|
2015-11-01 00:26:19 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
int mg_base64_final(char *to, int n) {
|
|
|
|
int saved = n;
|
|
|
|
// printf("---[%.*s]\n", n, to);
|
|
|
|
if (n & 3) n = mg_base64_update(0, to, n);
|
|
|
|
if ((saved & 3) == 2) n--;
|
|
|
|
// printf(" %d[%.*s]\n", n, n, to);
|
|
|
|
while (n & 3) to[n++] = '=';
|
|
|
|
to[n] = '\0';
|
|
|
|
return n;
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
int mg_base64_encode(const unsigned char *p, int n, char *to) {
|
|
|
|
int i, len = 0;
|
|
|
|
for (i = 0; i < n; i++) len = mg_base64_update(p[i], to, len);
|
|
|
|
len = mg_base64_final(to, len);
|
|
|
|
return len;
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
int mg_base64_decode(const char *src, int n, char *dst) {
|
|
|
|
const char *end = src + n;
|
|
|
|
int len = 0;
|
|
|
|
while (src + 3 < end) {
|
|
|
|
int a = mg_b64rev(src[0]), b = mg_b64rev(src[1]), c = mg_b64rev(src[2]),
|
|
|
|
d = mg_b64rev(src[3]);
|
|
|
|
if (a == 64 || a < 0 || b == 64 || b < 0 || c < 0 || d < 0) return 0;
|
2021-05-29 01:30:42 +08:00
|
|
|
dst[len++] = (char) ((a << 2) | (b >> 4));
|
2020-12-05 19:26:32 +08:00
|
|
|
if (src[2] != '=') {
|
2021-05-29 01:30:42 +08:00
|
|
|
dst[len++] = (char) ((b << 4) | (c >> 2));
|
|
|
|
if (src[3] != '=') dst[len++] = (char) ((c << 6) | d);
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
src += 4;
|
|
|
|
}
|
|
|
|
dst[len] = '\0';
|
|
|
|
return len;
|
2014-09-10 01:07:55 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
#ifdef MG_ENABLE_LINES
|
|
|
|
#line 1 "src/dns.c"
|
2017-11-24 18:21:45 +08:00
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2019-01-26 12:37:17 +08:00
|
|
|
|
2017-11-24 18:21:45 +08:00
|
|
|
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
struct dns_data {
|
|
|
|
struct dns_data *next;
|
|
|
|
struct mg_connection *c;
|
|
|
|
unsigned long expire;
|
|
|
|
uint16_t txnid;
|
|
|
|
};
|
2017-11-24 18:21:45 +08:00
|
|
|
|
2020-12-22 17:44:59 +08:00
|
|
|
static struct dns_data *s_reqs; // Active DNS requests
|
|
|
|
|
|
|
|
static void mg_sendnsreq(struct mg_connection *, struct mg_str *, int,
|
|
|
|
struct mg_dns *, bool);
|
|
|
|
|
|
|
|
static void mg_dns_free(struct dns_data *d) {
|
|
|
|
LIST_DELETE(struct dns_data, &s_reqs, d);
|
2020-12-05 19:26:32 +08:00
|
|
|
free(d);
|
|
|
|
}
|
2017-11-24 18:21:45 +08:00
|
|
|
|
2020-12-22 17:44:59 +08:00
|
|
|
void mg_resolve_cancel(struct mg_connection *c) {
|
|
|
|
struct dns_data *tmp, *d;
|
|
|
|
for (d = s_reqs; d != NULL; d = tmp) {
|
2020-12-05 19:26:32 +08:00
|
|
|
tmp = d->next;
|
2020-12-22 17:44:59 +08:00
|
|
|
if (d->c == c) mg_dns_free(d);
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
}
|
2017-11-24 18:21:45 +08:00
|
|
|
|
2020-12-28 13:25:29 +08:00
|
|
|
static size_t mg_dns_parse_name_depth(const uint8_t *s, size_t len, size_t ofs,
|
|
|
|
char *to, size_t tolen, int depth) {
|
2020-12-05 19:26:32 +08:00
|
|
|
size_t i = 0, j = 0;
|
2020-12-18 03:14:17 +08:00
|
|
|
if (tolen > 0) to[0] = '\0';
|
2020-12-12 01:35:58 +08:00
|
|
|
if (depth > 5) return 0;
|
2020-12-28 13:25:29 +08:00
|
|
|
while (ofs + i + 1 < len) {
|
|
|
|
size_t n = s[ofs + i];
|
|
|
|
if (n == 0) {
|
2020-12-05 19:26:32 +08:00
|
|
|
i++;
|
|
|
|
break;
|
|
|
|
}
|
2020-12-28 13:25:29 +08:00
|
|
|
if (n & 0xc0) {
|
|
|
|
size_t ptr = (((n & 0x3f) << 8) | s[ofs + i + 1]); // 12 is hdr len
|
|
|
|
if (ptr + 1 < len && (s[ptr] & 0xc0) == 0 &&
|
|
|
|
mg_dns_parse_name_depth(s, len, ptr, to, tolen, depth + 1) == 0)
|
|
|
|
return 0;
|
|
|
|
i += 2;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
if (ofs + i + n + 1 >= len) return 0;
|
|
|
|
if (j > 0) {
|
|
|
|
if (j < tolen) to[j] = '.';
|
|
|
|
j++;
|
|
|
|
}
|
|
|
|
if (j + n < tolen) memcpy(&to[j], &s[ofs + i + 1], n);
|
2020-12-05 19:26:32 +08:00
|
|
|
j += n;
|
|
|
|
i += n + 1;
|
2020-12-28 13:25:29 +08:00
|
|
|
if (j < tolen) to[j] = '\0'; // Zero-terminate this chunk
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
2020-12-18 03:14:17 +08:00
|
|
|
if (tolen > 0) to[tolen - 1] = '\0'; // Make sure make sure it is nul-term
|
2020-12-05 19:26:32 +08:00
|
|
|
return i;
|
|
|
|
}
|
2017-12-17 00:21:51 +08:00
|
|
|
|
2020-12-28 13:25:29 +08:00
|
|
|
size_t mg_dns_parse_name(const uint8_t *s, size_t n, size_t ofs, char *dst,
|
|
|
|
size_t dstlen) {
|
|
|
|
return mg_dns_parse_name_depth(s, n, ofs, dst, dstlen, 0);
|
|
|
|
}
|
|
|
|
|
|
|
|
size_t mg_dns_parse_rr(const uint8_t *buf, size_t len, size_t ofs,
|
|
|
|
bool is_question, struct mg_dns_rr *rr) {
|
|
|
|
const uint8_t *s = buf + ofs, *e = &buf[len];
|
|
|
|
|
|
|
|
memset(rr, 0, sizeof(*rr));
|
2021-04-14 02:42:47 +08:00
|
|
|
if (len < sizeof(struct mg_dns_header)) return 0; // Too small
|
2020-12-18 17:48:28 +08:00
|
|
|
if (len > 512) return 0; // Too large, we don't expect that
|
2020-12-28 13:25:29 +08:00
|
|
|
if (s >= e) return 0; // Overflow
|
|
|
|
|
2020-12-28 14:32:55 +08:00
|
|
|
if ((rr->nlen = (uint16_t) mg_dns_parse_name(buf, len, ofs, NULL, 0)) == 0)
|
|
|
|
return 0;
|
2020-12-28 13:25:29 +08:00
|
|
|
s += rr->nlen + 4;
|
|
|
|
if (s > e) return 0;
|
2021-05-29 01:30:42 +08:00
|
|
|
rr->atype = (uint16_t)(((uint16_t) s[-4] << 8) | s[-3]);
|
|
|
|
rr->aclass = (uint16_t)(((uint16_t) s[-2] << 8) | s[-1]);
|
2021-05-29 01:45:35 +08:00
|
|
|
if (is_question) return (size_t)(rr->nlen + 4);
|
2020-12-28 13:25:29 +08:00
|
|
|
|
|
|
|
s += 6;
|
|
|
|
if (s > e) return 0;
|
2021-05-29 01:30:42 +08:00
|
|
|
rr->alen = (uint16_t)(((uint16_t) s[-2] << 8) | s[-1]);
|
2020-12-28 13:25:29 +08:00
|
|
|
if (s + rr->alen > e) return 0;
|
2021-05-29 01:30:42 +08:00
|
|
|
return (size_t)(rr->nlen + rr->alen + 10);
|
2020-12-28 13:25:29 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
bool mg_dns_parse(const uint8_t *buf, size_t len, struct mg_dns_message *dm) {
|
|
|
|
const struct mg_dns_header *h = (struct mg_dns_header *) buf;
|
|
|
|
struct mg_dns_rr rr;
|
|
|
|
size_t i, n, ofs = sizeof(*h);
|
|
|
|
memset(dm, 0, sizeof(*dm));
|
|
|
|
|
|
|
|
if (len < sizeof(*h)) return 0; // Too small, headers dont fit
|
2020-12-25 12:32:56 +08:00
|
|
|
if (mg_ntohs(h->num_questions) > 1) return 0; // Sanity
|
|
|
|
if (mg_ntohs(h->num_answers) > 10) return 0; // Sanity
|
2020-12-28 13:25:29 +08:00
|
|
|
dm->txnid = mg_ntohs(h->txnid);
|
|
|
|
|
2020-12-12 01:35:58 +08:00
|
|
|
for (i = 0; i < mg_ntohs(h->num_questions); i++) {
|
2020-12-28 13:25:29 +08:00
|
|
|
if ((n = mg_dns_parse_rr(buf, len, ofs, true, &rr)) == 0) return false;
|
|
|
|
// LOG(LL_INFO, ("Q %zu %zu", ofs, n));
|
|
|
|
ofs += n;
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
for (i = 0; i < mg_ntohs(h->num_answers); i++) {
|
2020-12-28 13:25:29 +08:00
|
|
|
// LOG(LL_INFO, ("A -- %zu %zu %s", ofs, n, dm->name));
|
|
|
|
if ((n = mg_dns_parse_rr(buf, len, ofs, false, &rr)) == 0) return false;
|
|
|
|
mg_dns_parse_name(buf, len, ofs, dm->name, sizeof(dm->name));
|
|
|
|
ofs += n;
|
|
|
|
|
|
|
|
if (rr.alen == 4 && rr.atype == 1 && rr.aclass == 1) {
|
2020-12-22 17:44:59 +08:00
|
|
|
dm->addr.is_ip6 = false;
|
2020-12-28 13:25:29 +08:00
|
|
|
memcpy(&dm->addr.ip, &buf[ofs - 4], 4);
|
2020-12-22 17:44:59 +08:00
|
|
|
dm->resolved = true;
|
|
|
|
break; // Return success
|
2020-12-28 13:25:29 +08:00
|
|
|
} else if (rr.alen == 16 && rr.atype == 28 && rr.aclass == 1) {
|
2020-12-22 17:44:59 +08:00
|
|
|
dm->addr.is_ip6 = true;
|
2020-12-28 13:25:29 +08:00
|
|
|
memcpy(&dm->addr.ip6, &buf[ofs - 16], 16);
|
2020-12-21 00:55:33 +08:00
|
|
|
dm->resolved = true;
|
|
|
|
break; // Return success
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
}
|
2020-12-21 00:55:33 +08:00
|
|
|
return true;
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
2017-11-24 18:21:45 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
static void dns_cb(struct mg_connection *c, int ev, void *ev_data,
|
|
|
|
void *fn_data) {
|
2020-12-22 17:44:59 +08:00
|
|
|
struct dns_data *d, *tmp;
|
2020-12-05 19:26:32 +08:00
|
|
|
if (ev == MG_EV_POLL) {
|
|
|
|
unsigned long now = *(unsigned long *) ev_data;
|
2021-01-21 18:00:18 +08:00
|
|
|
for (d = s_reqs; d != NULL; d = tmp) {
|
2020-12-05 19:26:32 +08:00
|
|
|
tmp = d->next;
|
|
|
|
// LOG(LL_DEBUG, ("%lu %lu dns poll", d->expire, now));
|
|
|
|
if (now > d->expire) mg_error(d->c, "DNS timeout");
|
|
|
|
}
|
|
|
|
} else if (ev == MG_EV_READ) {
|
|
|
|
struct mg_dns_message dm;
|
|
|
|
int resolved = 0;
|
2020-12-21 00:55:33 +08:00
|
|
|
if (mg_dns_parse(c->recv.buf, c->recv.len, &dm) == false) {
|
|
|
|
char *s = mg_hexdump(c->recv.buf, c->recv.len);
|
|
|
|
LOG(LL_ERROR, ("Unexpected DNS response:\n%s\n", s));
|
|
|
|
free(s);
|
|
|
|
} else {
|
2020-12-25 00:52:58 +08:00
|
|
|
LOG(LL_VERBOSE_DEBUG, ("%s %d", dm.name, dm.resolved));
|
2020-12-22 17:44:59 +08:00
|
|
|
for (d = s_reqs; d != NULL; d = tmp) {
|
2020-12-05 19:26:32 +08:00
|
|
|
tmp = d->next;
|
2020-12-21 00:55:33 +08:00
|
|
|
// LOG(LL_INFO, ("d %p %hu %hu", d, d->txnid, dm.txnid));
|
2020-12-05 19:26:32 +08:00
|
|
|
if (dm.txnid != d->txnid) continue;
|
|
|
|
if (d->c->is_resolving) {
|
|
|
|
d->c->is_resolving = 0;
|
2020-12-21 00:55:33 +08:00
|
|
|
if (dm.resolved) {
|
2021-01-27 00:30:44 +08:00
|
|
|
#if MG_ENABLE_LOG
|
2021-01-24 21:57:40 +08:00
|
|
|
char buf[100];
|
2021-01-27 00:30:44 +08:00
|
|
|
#endif
|
2020-12-22 17:44:59 +08:00
|
|
|
dm.addr.port = d->c->peer.port; // Save port
|
|
|
|
d->c->peer = dm.addr; // Copy resolved address
|
2021-01-24 21:57:40 +08:00
|
|
|
LOG(LL_DEBUG, ("%lu %s resolved to %s", d->c->id, dm.name,
|
|
|
|
mg_ntoa(&d->c->peer, buf, sizeof(buf))));
|
2020-12-21 00:55:33 +08:00
|
|
|
mg_connect_resolved(d->c);
|
2020-12-22 17:44:59 +08:00
|
|
|
#if MG_ENABLE_IPV6
|
|
|
|
} else if (dm.addr.is_ip6 == false && dm.name[0] != '\0') {
|
|
|
|
struct mg_str x = mg_str(dm.name);
|
|
|
|
mg_sendnsreq(d->c, &x, c->mgr->dnstimeout, &c->mgr->dns6, true);
|
|
|
|
#endif
|
2020-12-21 00:55:33 +08:00
|
|
|
} else {
|
|
|
|
mg_error(d->c, "%s DNS lookup failed", dm.name);
|
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
} else {
|
2020-12-21 20:26:44 +08:00
|
|
|
LOG(LL_ERROR, ("%lu already resolved", d->c->id));
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
2020-12-22 17:44:59 +08:00
|
|
|
mg_dns_free(d);
|
2020-12-05 19:26:32 +08:00
|
|
|
resolved = 1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (!resolved) LOG(LL_ERROR, ("stray DNS reply"));
|
|
|
|
c->recv.len = 0;
|
|
|
|
} else if (ev == MG_EV_CLOSE) {
|
2020-12-22 17:44:59 +08:00
|
|
|
for (d = s_reqs; d != NULL; d = tmp) {
|
2020-12-05 19:26:32 +08:00
|
|
|
tmp = d->next;
|
2020-12-22 17:44:59 +08:00
|
|
|
mg_dns_free(d);
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
}
|
2021-01-21 18:00:18 +08:00
|
|
|
(void) fn_data;
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
void mg_dns_send(struct mg_connection *c, const struct mg_str *name,
|
2020-12-22 17:44:59 +08:00
|
|
|
uint16_t txnid, bool ipv6) {
|
2020-12-05 19:26:32 +08:00
|
|
|
struct {
|
|
|
|
struct mg_dns_header header;
|
|
|
|
uint8_t data[256];
|
|
|
|
} pkt;
|
|
|
|
size_t i, n;
|
|
|
|
memset(&pkt, 0, sizeof(pkt));
|
2020-12-28 13:25:29 +08:00
|
|
|
pkt.header.txnid = mg_htons(txnid);
|
2020-12-05 19:26:32 +08:00
|
|
|
pkt.header.flags = mg_htons(0x100);
|
2020-12-22 17:44:59 +08:00
|
|
|
pkt.header.num_questions = mg_htons(1);
|
2020-12-05 19:26:32 +08:00
|
|
|
for (i = n = 0; i < sizeof(pkt.data) - 5; i++) {
|
|
|
|
if (name->ptr[i] == '.' || i >= name->len) {
|
|
|
|
pkt.data[n] = (uint8_t)(i - n);
|
|
|
|
memcpy(&pkt.data[n + 1], name->ptr + n, i - n);
|
|
|
|
n = i + 1;
|
|
|
|
}
|
|
|
|
if (i >= name->len) break;
|
|
|
|
}
|
|
|
|
memcpy(&pkt.data[n], "\x00\x00\x01\x00\x01", 5); // A query
|
|
|
|
n += 5;
|
2020-12-22 17:44:59 +08:00
|
|
|
if (ipv6) pkt.data[n - 3] = 0x1c; // AAAA query
|
|
|
|
// memcpy(&pkt.data[n], "\xc0\x0c\x00\x1c\x00\x01", 6); // AAAA query
|
|
|
|
// n += 6;
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_send(c, &pkt, sizeof(pkt.header) + n);
|
|
|
|
#if 0
|
|
|
|
// Immediately after A query, send AAAA query. Whatever reply comes first,
|
|
|
|
// we'll use it. Note: we cannot send two queries in a single packet.
|
|
|
|
// https://stackoverflow.com/questions/4082081/requesting-a-and-aaaa-records-in-single-dns-query
|
|
|
|
pkt.data[n - 3] = 0x1c; // AAAA query
|
|
|
|
mg_send(c, &pkt, sizeof(pkt.header) + n);
|
|
|
|
#endif
|
|
|
|
}
|
2018-07-13 04:28:51 +08:00
|
|
|
|
2020-12-22 17:44:59 +08:00
|
|
|
static void mg_sendnsreq(struct mg_connection *c, struct mg_str *name, int ms,
|
|
|
|
struct mg_dns *dnsc, bool ipv6) {
|
2020-12-05 19:26:32 +08:00
|
|
|
struct dns_data *d = NULL;
|
2020-12-22 17:44:59 +08:00
|
|
|
if (dnsc->url == NULL) {
|
|
|
|
mg_error(c, "DNS server URL is NULL. Call mg_mgr_init()");
|
|
|
|
} else if (dnsc->c == NULL) {
|
|
|
|
dnsc->c = mg_connect(c->mgr, dnsc->url, NULL, NULL);
|
|
|
|
if (dnsc->c != NULL) {
|
|
|
|
dnsc->c->pfn = dns_cb;
|
|
|
|
snprintf(dnsc->c->label, sizeof(dnsc->c->label), "%s", "DNS");
|
|
|
|
// dnsc->c->is_hexdumping = 1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (dnsc->c == NULL) {
|
|
|
|
mg_error(c, "resolver");
|
|
|
|
} else if ((d = (struct dns_data *) calloc(1, sizeof(*d))) == NULL) {
|
|
|
|
mg_error(c, "resolve OOM");
|
|
|
|
} else {
|
2021-04-22 22:21:16 +08:00
|
|
|
#if MG_ENABLE_LOG
|
|
|
|
char buf[100];
|
|
|
|
#endif
|
2021-05-29 01:45:35 +08:00
|
|
|
d->txnid = s_reqs ? (uint16_t)(s_reqs->txnid + 1) : 1;
|
2020-12-22 17:44:59 +08:00
|
|
|
d->next = s_reqs;
|
|
|
|
s_reqs = d;
|
2021-05-29 01:30:42 +08:00
|
|
|
d->expire = mg_millis() + (unsigned long) ms;
|
2020-12-22 17:44:59 +08:00
|
|
|
d->c = c;
|
|
|
|
c->is_resolving = 1;
|
2021-04-22 22:21:16 +08:00
|
|
|
LOG(LL_VERBOSE_DEBUG,
|
|
|
|
("%lu resolving %.*s @ %s, txnid %hu", c->id, (int) name->len,
|
|
|
|
name->ptr, mg_ntoa(&dnsc->c->peer, buf, sizeof(buf)), d->txnid));
|
2020-12-22 17:44:59 +08:00
|
|
|
mg_dns_send(dnsc->c, name, d->txnid, ipv6);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void mg_resolve(struct mg_connection *c, struct mg_str *name, int ms) {
|
2020-12-21 00:55:33 +08:00
|
|
|
if (mg_aton(*name, &c->peer)) {
|
2020-12-05 19:26:32 +08:00
|
|
|
// name is an IP address, do not fire name resolution
|
|
|
|
mg_connect_resolved(c);
|
|
|
|
} else {
|
|
|
|
// name is not an IP, send DNS resolution request
|
2020-12-22 17:44:59 +08:00
|
|
|
mg_sendnsreq(c, name, ms, &c->mgr->dns4, false);
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
}
|
2018-07-13 04:28:51 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
#ifdef MG_ENABLE_LINES
|
|
|
|
#line 1 "src/event.c"
|
2018-07-13 04:28:51 +08:00
|
|
|
#endif
|
|
|
|
|
2017-11-24 18:21:45 +08:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
void mg_call(struct mg_connection *c, int ev, void *ev_data) {
|
|
|
|
if (c->pfn != NULL) c->pfn(c, ev, ev_data, c->pfn_data);
|
|
|
|
if (c->fn != NULL) c->fn(c, ev, ev_data, c->fn_data);
|
|
|
|
}
|
2017-11-24 18:21:45 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
void mg_error(struct mg_connection *c, const char *fmt, ...) {
|
|
|
|
char mem[256], *buf = mem;
|
|
|
|
va_list ap;
|
|
|
|
va_start(ap, fmt);
|
|
|
|
mg_vasprintf(&buf, sizeof(mem), fmt, ap);
|
|
|
|
va_end(ap);
|
2020-12-21 20:26:44 +08:00
|
|
|
LOG(LL_ERROR, ("%lu %s", c->id, buf));
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_call(c, MG_EV_ERROR, buf);
|
|
|
|
if (buf != mem) free(buf);
|
|
|
|
c->is_closing = 1;
|
|
|
|
}
|
2017-11-24 18:21:45 +08:00
|
|
|
|
2021-07-24 10:44:00 +08:00
|
|
|
#ifdef MG_ENABLE_LINES
|
2021-07-29 04:11:07 +08:00
|
|
|
#line 1 "src/fs_packed.c"
|
2021-07-24 10:44:00 +08:00
|
|
|
#endif
|
|
|
|
|
|
|
|
|
2021-07-26 16:05:34 +08:00
|
|
|
struct packed_file {
|
|
|
|
const char *data;
|
|
|
|
size_t size;
|
|
|
|
size_t pos;
|
|
|
|
};
|
|
|
|
|
2021-07-30 20:19:20 +08:00
|
|
|
const char *mg_unpack(const char *path, size_t *size, time_t *mtime);
|
|
|
|
const char *mg_unlist(size_t no);
|
|
|
|
#if MG_ENABLE_PACKED_FS
|
|
|
|
#else
|
|
|
|
const char *mg_unpack(const char *path, size_t *size, time_t *mtime) {
|
|
|
|
(void) path, (void) size, (void) mtime;
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
const char *mg_unlist(size_t no) {
|
|
|
|
(void) no;
|
2021-07-26 16:05:34 +08:00
|
|
|
return NULL;
|
|
|
|
}
|
2021-07-30 20:19:20 +08:00
|
|
|
#endif
|
2021-07-26 16:05:34 +08:00
|
|
|
|
2021-07-30 20:19:20 +08:00
|
|
|
static int is_dir_prefix(const char *prefix, size_t n, const char *path) {
|
2021-07-31 00:41:23 +08:00
|
|
|
return n < strlen(path) && memcmp(prefix, path, n) == 0 && path[n] == '/';
|
|
|
|
//(n == 0 || path[n] == MG_DIRSEP);
|
2021-07-30 20:19:20 +08:00
|
|
|
}
|
|
|
|
|
2021-07-29 21:21:20 +08:00
|
|
|
static int packed_stat(const char *path, size_t *size, time_t *mtime) {
|
2021-07-30 20:19:20 +08:00
|
|
|
const char *p;
|
|
|
|
size_t i, n = strlen(path);
|
|
|
|
if (mg_unpack(path, size, mtime)) return MG_FS_READ; // Regular file
|
|
|
|
// Scan all files. If `path` is a dir prefix for any of them, it's a dir
|
|
|
|
for (i = 0; (p = mg_unlist(i)) != NULL; i++) {
|
|
|
|
if (is_dir_prefix(path, n, p)) return MG_FS_DIR;
|
|
|
|
}
|
|
|
|
return 0;
|
2021-07-26 16:05:34 +08:00
|
|
|
}
|
|
|
|
|
2021-07-31 00:41:23 +08:00
|
|
|
static void packed_list(const char *dir, void (*fn)(const char *, void *),
|
2021-07-29 04:11:07 +08:00
|
|
|
void *userdata) {
|
2021-07-31 00:41:23 +08:00
|
|
|
char buf[256], tmp[sizeof(buf)];
|
|
|
|
const char *path, *begin, *end;
|
|
|
|
size_t i, n = strlen(dir);
|
|
|
|
tmp[0] = '\0'; // Previously listed entry
|
|
|
|
for (i = 0; (path = mg_unlist(i)) != NULL; i++) {
|
|
|
|
if (!is_dir_prefix(dir, n, path)) continue;
|
|
|
|
begin = &path[n + 1];
|
|
|
|
end = strchr(begin, '/');
|
|
|
|
if (end == NULL) end = begin + strlen(begin);
|
|
|
|
snprintf(buf, sizeof(buf), "%.*s", (int) (end - begin), begin);
|
|
|
|
buf[sizeof(buf) - 1] = '\0';
|
|
|
|
// If this entry has been already listed, skip
|
|
|
|
// NOTE: we're assuming that file list is sorted alphabetically
|
|
|
|
if (strcmp(buf, tmp) == 0) continue;
|
|
|
|
fn(buf, userdata); // Not yet listed, call user function
|
|
|
|
strcpy(tmp, buf); // And save this entry as listed
|
2021-07-30 20:19:20 +08:00
|
|
|
}
|
2021-07-26 16:05:34 +08:00
|
|
|
}
|
|
|
|
|
2021-07-29 04:11:07 +08:00
|
|
|
static struct mg_fd *packed_open(const char *path, int flags) {
|
2021-07-26 16:05:34 +08:00
|
|
|
size_t size = 0;
|
2021-07-30 20:19:20 +08:00
|
|
|
const char *data = mg_unpack(path, &size, NULL);
|
2021-07-29 04:11:07 +08:00
|
|
|
struct packed_file *fp = NULL;
|
|
|
|
struct mg_fd *fd = NULL;
|
2021-07-26 16:05:34 +08:00
|
|
|
if (data == NULL) return NULL;
|
2021-07-29 04:11:07 +08:00
|
|
|
if (flags & MG_FS_WRITE) return NULL;
|
2021-07-29 21:21:20 +08:00
|
|
|
fp = (struct packed_file *) calloc(1, sizeof(*fp));
|
|
|
|
fd = (struct mg_fd *) calloc(1, sizeof(*fd));
|
2021-07-29 04:11:07 +08:00
|
|
|
fp->size = size;
|
|
|
|
fp->data = data;
|
|
|
|
fd->fd = fp;
|
|
|
|
fd->fs = &mg_fs_packed;
|
|
|
|
return fd;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void packed_close(struct mg_fd *fd) {
|
|
|
|
if (fd) free(fd->fd), free(fd);
|
|
|
|
}
|
|
|
|
|
|
|
|
static size_t packed_read(void *fd, void *buf, size_t len) {
|
|
|
|
struct packed_file *fp = (struct packed_file *) fd;
|
|
|
|
if (fp->pos + len > fp->size) len = fp->size - fp->pos;
|
|
|
|
memcpy(buf, &fp->data[fp->pos], len);
|
|
|
|
fp->pos += len;
|
|
|
|
return len;
|
|
|
|
}
|
|
|
|
|
|
|
|
static size_t packed_write(void *fd, const void *buf, size_t len) {
|
|
|
|
(void) fd, (void) buf, (void) len;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static size_t packed_seek(void *fd, size_t offset) {
|
|
|
|
struct packed_file *fp = (struct packed_file *) fd;
|
|
|
|
fp->pos = offset;
|
|
|
|
if (fp->pos > fp->size) fp->pos = fp->size;
|
|
|
|
return fp->pos;
|
|
|
|
}
|
|
|
|
|
2021-09-15 14:43:48 +08:00
|
|
|
struct mg_fs mg_fs_packed = {packed_stat, packed_list, packed_open,
|
|
|
|
packed_close, packed_read, packed_write,
|
|
|
|
packed_seek};
|
2021-07-29 04:11:07 +08:00
|
|
|
|
|
|
|
#ifdef MG_ENABLE_LINES
|
|
|
|
#line 1 "src/fs_posix.c"
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
2021-07-29 21:21:20 +08:00
|
|
|
#if defined(FOPEN_MAX)
|
2021-09-16 18:16:10 +08:00
|
|
|
static int p_stat(const char *path, size_t *size, time_t *mtime) {
|
2021-07-29 04:11:07 +08:00
|
|
|
#ifdef _WIN32
|
|
|
|
struct _stati64 st;
|
|
|
|
wchar_t tmp[PATH_MAX];
|
|
|
|
MultiByteToWideChar(CP_UTF8, 0, path, -1, tmp, sizeof(tmp) / sizeof(tmp[0]));
|
|
|
|
if (_wstati64(tmp, &st) != 0) return 0;
|
|
|
|
#else
|
|
|
|
struct stat st;
|
|
|
|
if (stat(path, &st) != 0) return 0;
|
|
|
|
#endif
|
|
|
|
if (size) *size = (size_t) st.st_size;
|
2021-07-29 21:21:20 +08:00
|
|
|
if (mtime) *mtime = st.st_mtime;
|
2021-07-29 04:11:07 +08:00
|
|
|
return MG_FS_READ | MG_FS_WRITE | (S_ISDIR(st.st_mode) ? MG_FS_DIR : 0);
|
|
|
|
}
|
|
|
|
|
2021-07-29 21:21:20 +08:00
|
|
|
#ifdef _WIN32
|
|
|
|
struct dirent {
|
|
|
|
char d_name[MAX_PATH];
|
|
|
|
};
|
|
|
|
|
|
|
|
typedef struct win32_dir {
|
|
|
|
HANDLE handle;
|
|
|
|
WIN32_FIND_DATAW info;
|
|
|
|
struct dirent result;
|
|
|
|
} DIR;
|
|
|
|
|
|
|
|
int gettimeofday(struct timeval *tv, void *tz) {
|
|
|
|
FILETIME ft;
|
|
|
|
unsigned __int64 tmpres = 0;
|
|
|
|
|
|
|
|
if (tv != NULL) {
|
|
|
|
GetSystemTimeAsFileTime(&ft);
|
|
|
|
tmpres |= ft.dwHighDateTime;
|
|
|
|
tmpres <<= 32;
|
|
|
|
tmpres |= ft.dwLowDateTime;
|
|
|
|
tmpres /= 10; // convert into microseconds
|
|
|
|
tmpres -= (int64_t) 11644473600000000;
|
|
|
|
tv->tv_sec = (long) (tmpres / 1000000UL);
|
|
|
|
tv->tv_usec = (long) (tmpres % 1000000UL);
|
|
|
|
}
|
|
|
|
(void) tz;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int to_wchar(const char *path, wchar_t *wbuf, size_t wbuf_len) {
|
|
|
|
int ret;
|
|
|
|
char buf[MAX_PATH * 2], buf2[MAX_PATH * 2], *p;
|
|
|
|
strncpy(buf, path, sizeof(buf));
|
|
|
|
buf[sizeof(buf) - 1] = '\0';
|
|
|
|
// Trim trailing slashes. Leave backslash for paths like "X:\"
|
|
|
|
p = buf + strlen(buf) - 1;
|
|
|
|
while (p > buf && p[-1] != ':' && (p[0] == '\\' || p[0] == '/')) *p-- = '\0';
|
|
|
|
memset(wbuf, 0, wbuf_len * sizeof(wchar_t));
|
|
|
|
ret = MultiByteToWideChar(CP_UTF8, 0, buf, -1, wbuf, (int) wbuf_len);
|
|
|
|
// Convert back to Unicode. If doubly-converted string does not match the
|
|
|
|
// original, something is fishy, reject.
|
|
|
|
WideCharToMultiByte(CP_UTF8, 0, wbuf, (int) wbuf_len, buf2, sizeof(buf2),
|
|
|
|
NULL, NULL);
|
|
|
|
if (strcmp(buf, buf2) != 0) {
|
|
|
|
wbuf[0] = L'\0';
|
|
|
|
ret = 0;
|
|
|
|
}
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
DIR *opendir(const char *name) {
|
|
|
|
DIR *d = NULL;
|
|
|
|
wchar_t wpath[MAX_PATH];
|
|
|
|
DWORD attrs;
|
|
|
|
|
|
|
|
if (name == NULL) {
|
|
|
|
SetLastError(ERROR_BAD_ARGUMENTS);
|
|
|
|
} else if ((d = (DIR *) calloc(1, sizeof(*d))) == NULL) {
|
|
|
|
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
|
|
|
|
} else {
|
|
|
|
to_wchar(name, wpath, sizeof(wpath) / sizeof(wpath[0]));
|
|
|
|
attrs = GetFileAttributesW(wpath);
|
|
|
|
if (attrs != 0Xffffffff && (attrs & FILE_ATTRIBUTE_DIRECTORY)) {
|
|
|
|
(void) wcscat(wpath, L"\\*");
|
|
|
|
d->handle = FindFirstFileW(wpath, &d->info);
|
|
|
|
d->result.d_name[0] = '\0';
|
|
|
|
} else {
|
|
|
|
free(d);
|
|
|
|
d = NULL;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return d;
|
|
|
|
}
|
|
|
|
|
|
|
|
int closedir(DIR *d) {
|
|
|
|
int result = 0;
|
|
|
|
if (d != NULL) {
|
|
|
|
if (d->handle != INVALID_HANDLE_VALUE)
|
|
|
|
result = FindClose(d->handle) ? 0 : -1;
|
|
|
|
free(d);
|
|
|
|
} else {
|
|
|
|
result = -1;
|
|
|
|
SetLastError(ERROR_BAD_ARGUMENTS);
|
|
|
|
}
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
struct dirent *readdir(DIR *d) {
|
|
|
|
struct dirent *result = NULL;
|
|
|
|
if (d != NULL) {
|
|
|
|
memset(&d->result, 0, sizeof(d->result));
|
|
|
|
if (d->handle != INVALID_HANDLE_VALUE) {
|
|
|
|
result = &d->result;
|
|
|
|
WideCharToMultiByte(CP_UTF8, 0, d->info.cFileName, -1, result->d_name,
|
|
|
|
sizeof(result->d_name), NULL, NULL);
|
|
|
|
if (!FindNextFileW(d->handle, &d->info)) {
|
|
|
|
FindClose(d->handle);
|
|
|
|
d->handle = INVALID_HANDLE_VALUE;
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
SetLastError(ERROR_FILE_NOT_FOUND);
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
SetLastError(ERROR_BAD_ARGUMENTS);
|
|
|
|
}
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2021-09-16 18:16:10 +08:00
|
|
|
static void p_list(const char *dir, void (*fn)(const char *, void *),
|
|
|
|
void *userdata) {
|
2021-07-29 21:21:20 +08:00
|
|
|
#if MG_ENABLE_DIRLIST
|
2021-07-29 04:11:07 +08:00
|
|
|
struct dirent *dp;
|
|
|
|
DIR *dirp;
|
2021-07-29 21:21:20 +08:00
|
|
|
if ((dirp = (opendir(dir))) == NULL) return;
|
|
|
|
while ((dp = readdir(dirp)) != NULL) {
|
|
|
|
if (!strcmp(dp->d_name, ".") || !strcmp(dp->d_name, "..")) continue;
|
|
|
|
fn(dp->d_name, userdata);
|
2021-07-29 04:11:07 +08:00
|
|
|
}
|
2021-07-29 21:21:20 +08:00
|
|
|
closedir(dirp);
|
|
|
|
#else
|
|
|
|
(void) dir, (void) fn, (void) userdata;
|
|
|
|
#endif
|
2021-07-29 04:11:07 +08:00
|
|
|
}
|
|
|
|
|
2021-09-16 18:16:10 +08:00
|
|
|
static struct mg_fd *p_open(const char *path, int flags) {
|
2021-09-02 17:20:55 +08:00
|
|
|
const char *mode = flags == (MG_FS_READ | MG_FS_WRITE) ? "r+b"
|
|
|
|
: flags & MG_FS_READ ? "rb"
|
|
|
|
: flags & MG_FS_WRITE ? "wb"
|
|
|
|
: "";
|
2021-07-29 04:11:07 +08:00
|
|
|
void *fp = NULL;
|
|
|
|
struct mg_fd *fd = NULL;
|
|
|
|
#ifdef _WIN32
|
|
|
|
wchar_t b1[PATH_MAX], b2[10];
|
|
|
|
MultiByteToWideChar(CP_UTF8, 0, path, -1, b1, sizeof(b1) / sizeof(b1[0]));
|
|
|
|
MultiByteToWideChar(CP_UTF8, 0, mode, -1, b2, sizeof(b2) / sizeof(b2[0]));
|
|
|
|
fp = (void *) _wfopen(b1, b2);
|
|
|
|
#else
|
|
|
|
fp = (void *) fopen(path, mode);
|
|
|
|
#endif
|
|
|
|
if (fp == NULL) return NULL;
|
2021-07-29 21:21:20 +08:00
|
|
|
fd = (struct mg_fd *) calloc(1, sizeof(*fd));
|
2021-07-29 04:11:07 +08:00
|
|
|
fd->fd = fp;
|
|
|
|
fd->fs = &mg_fs_posix;
|
|
|
|
return fd;
|
|
|
|
}
|
|
|
|
|
2021-09-16 18:16:10 +08:00
|
|
|
static void p_close(struct mg_fd *fd) {
|
2021-07-29 21:21:20 +08:00
|
|
|
if (fd != NULL) fclose((FILE *) fd->fd), free(fd);
|
2021-07-29 04:11:07 +08:00
|
|
|
}
|
|
|
|
|
2021-09-16 18:16:10 +08:00
|
|
|
static size_t p_read(void *fp, void *buf, size_t len) {
|
2021-07-29 04:11:07 +08:00
|
|
|
return fread(buf, 1, len, (FILE *) fp);
|
|
|
|
}
|
|
|
|
|
2021-09-16 18:16:10 +08:00
|
|
|
static size_t p_write(void *fp, const void *buf, size_t len) {
|
2021-07-29 04:11:07 +08:00
|
|
|
return fwrite(buf, 1, len, (FILE *) fp);
|
2021-07-26 16:05:34 +08:00
|
|
|
}
|
2021-07-29 04:11:07 +08:00
|
|
|
|
2021-09-16 18:16:10 +08:00
|
|
|
static size_t p_seek(void *fp, size_t offset) {
|
2021-07-29 04:11:07 +08:00
|
|
|
#if _FILE_OFFSET_BITS == 64 || _POSIX_C_SOURCE >= 200112L || \
|
|
|
|
_XOPEN_SOURCE >= 600
|
|
|
|
fseeko((FILE *) fp, (off_t) offset, SEEK_SET);
|
2021-07-26 16:05:34 +08:00
|
|
|
#else
|
2021-07-29 04:11:07 +08:00
|
|
|
fseek((FILE *) fp, (long) offset, SEEK_SET);
|
|
|
|
#endif
|
|
|
|
return (size_t) ftell((FILE *) fp);
|
|
|
|
}
|
|
|
|
#else
|
2021-09-16 18:16:10 +08:00
|
|
|
static char *p_realpath(const char *path, char *resolved_path) {
|
2021-07-29 04:11:07 +08:00
|
|
|
(void) path, (void) resolved_path;
|
2021-07-26 16:05:34 +08:00
|
|
|
return NULL;
|
|
|
|
}
|
2021-07-29 04:11:07 +08:00
|
|
|
|
2021-09-16 18:16:10 +08:00
|
|
|
static int p_stat(const char *path, size_t *size, time_t *mtime) {
|
2021-07-29 04:11:07 +08:00
|
|
|
(void) path, (void) size, (void) mtime;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2021-09-16 18:16:10 +08:00
|
|
|
static void p_list(const char *path, void (*fn)(const char *, void *),
|
|
|
|
void *userdata) {
|
2021-07-29 04:11:07 +08:00
|
|
|
(void) path, (void) fn, (void) userdata;
|
|
|
|
}
|
|
|
|
|
2021-09-16 18:16:10 +08:00
|
|
|
static struct mg_fd *p_open(const char *path, int flags) {
|
2021-07-29 04:11:07 +08:00
|
|
|
(void) path, (void) flags;
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2021-09-16 18:16:10 +08:00
|
|
|
static void p_close(struct mg_fd *fd) {
|
2021-07-29 04:11:07 +08:00
|
|
|
(void) fd;
|
|
|
|
}
|
|
|
|
|
2021-09-16 18:16:10 +08:00
|
|
|
static size_t p_read(void *fd, void *buf, size_t len) {
|
2021-07-29 04:11:07 +08:00
|
|
|
(void) fd, (void) buf, (void) len;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2021-09-16 18:16:10 +08:00
|
|
|
static size_t p_write(void *fd, const void *buf, size_t len) {
|
2021-07-29 04:11:07 +08:00
|
|
|
(void) fd, (void) buf, (void) len;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2021-09-16 18:16:10 +08:00
|
|
|
static size_t p_seek(void *fd, size_t offset) {
|
2021-07-29 04:11:07 +08:00
|
|
|
(void) fd, (void) offset;
|
|
|
|
return (size_t) ~0;
|
|
|
|
}
|
2021-07-26 16:05:34 +08:00
|
|
|
#endif
|
2021-07-24 10:44:00 +08:00
|
|
|
|
2021-09-16 18:16:10 +08:00
|
|
|
struct mg_fs mg_fs_posix = {p_stat, p_list, p_open, p_close,
|
|
|
|
p_read, p_write, p_seek};
|
2021-07-29 04:11:07 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
#ifdef MG_ENABLE_LINES
|
|
|
|
#line 1 "src/http.c"
|
2017-11-24 18:21:45 +08:00
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2018-01-15 21:15:59 +08:00
|
|
|
|
2017-11-24 18:21:45 +08:00
|
|
|
|
2020-12-11 01:46:26 +08:00
|
|
|
|
2021-01-03 01:57:51 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
// Multipart POST example:
|
2021-03-17 15:43:29 +08:00
|
|
|
// --xyz
|
|
|
|
// Content-Disposition: form-data; name="val"
|
|
|
|
//
|
|
|
|
// abcdef
|
|
|
|
// --xyz
|
|
|
|
// Content-Disposition: form-data; name="foo"; filename="a.txt"
|
|
|
|
// Content-Type: text/plain
|
|
|
|
//
|
|
|
|
// hello world
|
|
|
|
//
|
|
|
|
// --xyz--
|
|
|
|
size_t mg_http_next_multipart(struct mg_str body, size_t ofs,
|
|
|
|
struct mg_http_part *part) {
|
|
|
|
struct mg_str cd = mg_str_n("Content-Disposition", 19);
|
|
|
|
const char *s = body.ptr;
|
|
|
|
size_t b = ofs, h1, h2, b1, b2, max = body.len;
|
|
|
|
|
|
|
|
// Init part params
|
|
|
|
if (part != NULL) part->name = part->filename = part->body = mg_str_n(0, 0);
|
|
|
|
|
|
|
|
// Skip boundary
|
|
|
|
while (b + 2 < max && s[b] != '\r' && s[b + 1] != '\n') b++;
|
|
|
|
if (b <= ofs || b + 2 >= max) return 0;
|
|
|
|
// LOG(LL_INFO, ("B: %zu %zu [%.*s]", ofs, b - ofs, (int) (b - ofs), s));
|
|
|
|
|
|
|
|
// Skip headers
|
|
|
|
h1 = h2 = b + 2;
|
|
|
|
for (;;) {
|
|
|
|
while (h2 + 2 < max && s[h2] != '\r' && s[h2 + 1] != '\n') h2++;
|
|
|
|
if (h2 == h1) break;
|
|
|
|
if (h2 + 2 >= max) return 0;
|
|
|
|
// LOG(LL_INFO, ("Header: [%.*s]", (int) (h2 - h1), &s[h1]));
|
|
|
|
if (part != NULL && h1 + cd.len + 2 < h2 && s[h1 + cd.len] == ':' &&
|
|
|
|
mg_ncasecmp(&s[h1], cd.ptr, cd.len) == 0) {
|
|
|
|
struct mg_str v = mg_str_n(&s[h1 + cd.len + 2], h2 - (h1 + cd.len + 2));
|
|
|
|
part->name = mg_http_get_header_var(v, mg_str_n("name", 4));
|
|
|
|
part->filename = mg_http_get_header_var(v, mg_str_n("filename", 8));
|
|
|
|
}
|
|
|
|
h1 = h2 = h2 + 2;
|
|
|
|
}
|
|
|
|
b1 = b2 = h2 + 2;
|
2021-03-17 21:28:36 +08:00
|
|
|
while (b2 + 2 + (b - ofs) + 2 < max && !(s[b2] == '\r' && s[b2 + 1] == '\n' &&
|
|
|
|
memcmp(&s[b2 + 2], s, b - ofs) == 0))
|
2021-03-17 15:43:29 +08:00
|
|
|
b2++;
|
|
|
|
|
|
|
|
if (b2 + 2 >= max) return 0;
|
|
|
|
if (part != NULL) part->body = mg_str_n(&s[b1], b2 - b1);
|
|
|
|
// LOG(LL_INFO, ("Body: [%.*s]", (int) (b2 - b1), &s[b1]));
|
|
|
|
return b2 + 2;
|
2021-03-15 21:20:53 +08:00
|
|
|
}
|
2017-11-24 18:21:45 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
void mg_http_bauth(struct mg_connection *c, const char *user,
|
|
|
|
const char *pass) {
|
|
|
|
struct mg_str u = mg_str(user), p = mg_str(pass);
|
|
|
|
size_t need = c->send.len + 36 + (u.len + p.len) * 2;
|
|
|
|
if (c->send.size < need) mg_iobuf_resize(&c->send, need);
|
|
|
|
if (c->send.size >= need) {
|
2021-05-29 01:30:42 +08:00
|
|
|
int i, n = 0;
|
2020-12-05 19:26:32 +08:00
|
|
|
char *buf = (char *) &c->send.buf[c->send.len + 21];
|
|
|
|
memcpy(&buf[-21], "Authorization: Basic ", 21); // DON'T use mg_send!
|
2021-05-29 01:30:42 +08:00
|
|
|
for (i = 0; i < (int) u.len; i++) {
|
|
|
|
n = mg_base64_update(((unsigned char *) u.ptr)[i], buf, n);
|
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
if (p.len > 0) {
|
|
|
|
n = mg_base64_update(':', buf, n);
|
2021-05-29 01:30:42 +08:00
|
|
|
for (i = 0; i < (int) p.len; i++) {
|
|
|
|
n = mg_base64_update(((unsigned char *) p.ptr)[i], buf, n);
|
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
n = mg_base64_final(buf, n);
|
2021-05-29 01:30:42 +08:00
|
|
|
c->send.len += 21 + (size_t) n + 2;
|
2020-12-05 19:26:32 +08:00
|
|
|
memcpy(&c->send.buf[c->send.len - 2], "\r\n", 2);
|
|
|
|
} else {
|
2020-12-21 20:26:44 +08:00
|
|
|
LOG(LL_ERROR, ("%lu %s cannot resize iobuf %d->%d ", c->id, c->label,
|
2020-12-05 19:26:32 +08:00
|
|
|
(int) c->send.size, (int) need));
|
|
|
|
}
|
2020-08-29 03:29:13 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
int mg_http_get_var(const struct mg_str *buf, const char *name, char *dst,
|
2021-05-29 01:30:42 +08:00
|
|
|
size_t dst_len) {
|
2020-12-05 19:26:32 +08:00
|
|
|
const char *p, *e, *s;
|
|
|
|
size_t name_len;
|
|
|
|
int len;
|
2020-08-29 03:29:13 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
if (dst == NULL || dst_len == 0) {
|
|
|
|
len = -2; // Bad destination
|
|
|
|
} else if (buf->ptr == NULL || name == NULL || buf->len == 0) {
|
|
|
|
len = -1; // Bad source
|
|
|
|
dst[0] = '\0';
|
2017-11-24 18:21:45 +08:00
|
|
|
} else {
|
2020-12-05 19:26:32 +08:00
|
|
|
name_len = strlen(name);
|
|
|
|
e = buf->ptr + buf->len;
|
|
|
|
len = -4; // Name does not exist
|
|
|
|
dst[0] = '\0';
|
|
|
|
for (p = buf->ptr; p + name_len < e; p++) {
|
|
|
|
if ((p == buf->ptr || p[-1] == '&') && p[name_len] == '=' &&
|
|
|
|
!mg_ncasecmp(name, p, name_len)) {
|
|
|
|
p += name_len + 1;
|
2021-08-08 00:22:47 +08:00
|
|
|
s = (const char *) memchr(p, '&', (size_t) (e - p));
|
2020-12-05 19:26:32 +08:00
|
|
|
if (s == NULL) s = e;
|
2021-08-08 00:22:47 +08:00
|
|
|
len = mg_url_decode(p, (size_t) (s - p), dst, dst_len, 1);
|
2021-02-03 10:13:05 +08:00
|
|
|
if (len < 0) len = -3; // Failed to decode
|
2020-12-05 19:26:32 +08:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
2017-11-24 18:21:45 +08:00
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
return len;
|
2017-11-24 18:21:45 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
int mg_url_decode(const char *src, size_t src_len, char *dst, size_t dst_len,
|
|
|
|
int is_form_url_encoded) {
|
|
|
|
size_t i, j;
|
2020-12-11 17:35:50 +08:00
|
|
|
for (i = j = 0; i < src_len && j + 1 < dst_len; i++, j++) {
|
2020-12-05 19:26:32 +08:00
|
|
|
if (src[i] == '%') {
|
2020-12-11 17:35:50 +08:00
|
|
|
// Use `i + 2 < src_len`, not `i < src_len - 2`, note small src_len
|
|
|
|
if (i + 2 < src_len && isxdigit(*(const unsigned char *) (src + i + 1)) &&
|
2020-12-05 19:26:32 +08:00
|
|
|
isxdigit(*(const unsigned char *) (src + i + 2))) {
|
|
|
|
mg_unhex(src + i + 1, 2, (uint8_t *) &dst[j]);
|
|
|
|
i += 2;
|
|
|
|
} else {
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
} else if (is_form_url_encoded && src[i] == '+') {
|
|
|
|
dst[j] = ' ';
|
|
|
|
} else {
|
|
|
|
dst[j] = src[i];
|
|
|
|
}
|
|
|
|
}
|
2020-12-11 17:35:50 +08:00
|
|
|
if (j < dst_len) dst[j] = '\0'; // Null-terminate the destination
|
2021-02-03 10:13:05 +08:00
|
|
|
return i >= src_len && j < dst_len ? (int) j : -1;
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
2019-01-26 12:37:17 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
int mg_http_get_request_len(const unsigned char *buf, size_t buf_len) {
|
|
|
|
size_t i;
|
|
|
|
for (i = 0; i < buf_len; i++) {
|
|
|
|
if (!isprint(buf[i]) && buf[i] != '\r' && buf[i] != '\n' && buf[i] < 128)
|
|
|
|
return -1;
|
|
|
|
if ((i > 0 && buf[i] == '\n' && buf[i - 1] == '\n') ||
|
|
|
|
(i > 3 && buf[i] == '\n' && buf[i - 1] == '\r' && buf[i - 2] == '\n'))
|
|
|
|
return (int) i + 1;
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
2019-01-26 12:37:17 +08:00
|
|
|
|
2021-02-10 05:16:33 +08:00
|
|
|
static const char *skip(const char *s, const char *e, const char *d,
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_str *v) {
|
|
|
|
v->ptr = s;
|
2021-02-10 05:16:33 +08:00
|
|
|
while (s < e && *s != '\n' && strchr(d, *s) == NULL) s++;
|
2021-08-08 00:22:47 +08:00
|
|
|
v->len = (size_t) (s - v->ptr);
|
2021-02-10 05:16:33 +08:00
|
|
|
while (s < e && strchr(d, *s) != NULL) s++;
|
2020-12-05 19:26:32 +08:00
|
|
|
return s;
|
|
|
|
}
|
2017-11-24 18:21:45 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_str *mg_http_get_header(struct mg_http_message *h, const char *name) {
|
|
|
|
size_t i, n = strlen(name), max = sizeof(h->headers) / sizeof(h->headers[0]);
|
|
|
|
for (i = 0; i < max && h->headers[i].name.len > 0; i++) {
|
|
|
|
struct mg_str *k = &h->headers[i].name, *v = &h->headers[i].value;
|
|
|
|
if (n == k->len && mg_ncasecmp(k->ptr, name, n) == 0) return v;
|
2019-01-26 12:37:17 +08:00
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
return NULL;
|
|
|
|
}
|
2019-01-26 12:37:17 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
void mg_http_parse_headers(const char *s, const char *end,
|
|
|
|
struct mg_http_header *h, int max_headers) {
|
|
|
|
int i;
|
|
|
|
for (i = 0; i < max_headers; i++) {
|
|
|
|
struct mg_str k, v, tmp;
|
|
|
|
const char *he = skip(s, end, "\n", &tmp);
|
|
|
|
s = skip(s, he, ": \r\n", &k);
|
|
|
|
s = skip(s, he, "\r\n", &v);
|
|
|
|
if (k.len == tmp.len) continue;
|
|
|
|
while (v.len > 0 && v.ptr[v.len - 1] == ' ') v.len--; // Trim spaces
|
|
|
|
if (k.len == 0) break;
|
|
|
|
// LOG(LL_INFO, ("--HH [%.*s] [%.*s] [%.*s]", (int) tmp.len - 1, tmp.ptr,
|
|
|
|
//(int) k.len, k.ptr, (int) v.len, v.ptr));
|
|
|
|
h[i].name = k;
|
|
|
|
h[i].value = v;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
int mg_http_parse(const char *s, size_t len, struct mg_http_message *hm) {
|
|
|
|
int is_response, req_len = mg_http_get_request_len((unsigned char *) s, len);
|
|
|
|
const char *end = s + req_len, *qs;
|
|
|
|
struct mg_str *cl;
|
2019-01-26 12:37:17 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
memset(hm, 0, sizeof(*hm));
|
|
|
|
if (req_len <= 0) return req_len;
|
|
|
|
|
2021-02-09 21:27:17 +08:00
|
|
|
hm->message.ptr = hm->head.ptr = s;
|
|
|
|
hm->body.ptr = end;
|
2021-05-29 01:30:42 +08:00
|
|
|
hm->head.len = (size_t) req_len;
|
2021-03-07 19:56:20 +08:00
|
|
|
hm->chunk.ptr = end;
|
2020-12-05 19:26:32 +08:00
|
|
|
hm->message.len = hm->body.len = (size_t) ~0; // Set body length to infinite
|
|
|
|
|
|
|
|
// Parse request line
|
|
|
|
s = skip(s, end, " ", &hm->method);
|
|
|
|
s = skip(s, end, " ", &hm->uri);
|
|
|
|
s = skip(s, end, "\r\n", &hm->proto);
|
2021-02-10 19:09:13 +08:00
|
|
|
|
2021-05-29 17:07:05 +08:00
|
|
|
// Sanity check. Allow protocol/reason to be empty
|
|
|
|
if (hm->method.len == 0 || hm->uri.len == 0) return -1;
|
2020-12-05 19:26:32 +08:00
|
|
|
|
|
|
|
// If URI contains '?' character, setup query string
|
|
|
|
if ((qs = (const char *) memchr(hm->uri.ptr, '?', hm->uri.len)) != NULL) {
|
|
|
|
hm->query.ptr = qs + 1;
|
2021-08-08 00:22:47 +08:00
|
|
|
hm->query.len = (size_t) (&hm->uri.ptr[hm->uri.len] - (qs + 1));
|
|
|
|
hm->uri.len = (size_t) (qs - hm->uri.ptr);
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
mg_http_parse_headers(s, end, hm->headers,
|
|
|
|
sizeof(hm->headers) / sizeof(hm->headers[0]));
|
|
|
|
if ((cl = mg_http_get_header(hm, "Content-Length")) != NULL) {
|
2020-12-23 19:58:08 +08:00
|
|
|
hm->body.len = (size_t) mg_to64(*cl);
|
2021-05-29 01:30:42 +08:00
|
|
|
hm->message.len = (size_t) req_len + hm->body.len;
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// mg_http_parse() is used to parse both HTTP requests and HTTP
|
|
|
|
// responses. If HTTP response does not have Content-Length set, then
|
|
|
|
// body is read until socket is closed, i.e. body.len is infinite (~0).
|
|
|
|
//
|
|
|
|
// For HTTP requests though, according to
|
|
|
|
// http://tools.ietf.org/html/rfc7231#section-8.1.3,
|
|
|
|
// only POST and PUT methods have defined body semantics.
|
|
|
|
// Therefore, if Content-Length is not specified and methods are
|
|
|
|
// not one of PUT or POST, set body length to 0.
|
|
|
|
//
|
|
|
|
// So, if it is HTTP request, and Content-Length is not set,
|
|
|
|
// and method is not (PUT or POST) then reset body length to zero.
|
|
|
|
is_response = mg_ncasecmp(hm->method.ptr, "HTTP/", 5) == 0;
|
|
|
|
if (hm->body.len == (size_t) ~0 && !is_response &&
|
|
|
|
mg_vcasecmp(&hm->method, "PUT") != 0 &&
|
|
|
|
mg_vcasecmp(&hm->method, "POST") != 0) {
|
|
|
|
hm->body.len = 0;
|
2021-05-29 01:30:42 +08:00
|
|
|
hm->message.len = (size_t) req_len;
|
2017-11-24 18:21:45 +08:00
|
|
|
}
|
|
|
|
|
2020-12-25 01:01:49 +08:00
|
|
|
// The 204 (No content) responses also have 0 body length
|
|
|
|
if (hm->body.len == (size_t) ~0 && is_response &&
|
|
|
|
mg_vcasecmp(&hm->uri, "204") == 0) {
|
|
|
|
hm->body.len = 0;
|
2021-05-29 01:30:42 +08:00
|
|
|
hm->message.len = (size_t) req_len;
|
2020-12-25 01:01:49 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
return req_len;
|
2017-11-24 18:21:45 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
static void mg_http_vprintf_chunk(struct mg_connection *c, const char *fmt,
|
|
|
|
va_list ap) {
|
|
|
|
char mem[256], *buf = mem;
|
|
|
|
int len = mg_vasprintf(&buf, sizeof(mem), fmt, ap);
|
|
|
|
mg_printf(c, "%X\r\n", len);
|
2021-05-29 01:30:42 +08:00
|
|
|
mg_send(c, buf, len > 0 ? (size_t) len : 0);
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_send(c, "\r\n", 2);
|
|
|
|
if (buf != mem) free(buf);
|
|
|
|
}
|
|
|
|
|
|
|
|
void mg_http_printf_chunk(struct mg_connection *c, const char *fmt, ...) {
|
2017-11-24 18:21:45 +08:00
|
|
|
va_list ap;
|
|
|
|
va_start(ap, fmt);
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_http_vprintf_chunk(c, fmt, ap);
|
2017-11-24 18:21:45 +08:00
|
|
|
va_end(ap);
|
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
void mg_http_write_chunk(struct mg_connection *c, const char *buf, size_t len) {
|
2021-04-29 16:26:37 +08:00
|
|
|
mg_printf(c, "%lX\r\n", (unsigned long) len);
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_send(c, buf, len);
|
|
|
|
mg_send(c, "\r\n", 2);
|
2017-11-24 18:21:45 +08:00
|
|
|
}
|
|
|
|
|
2021-04-29 16:17:24 +08:00
|
|
|
// clang-format off
|
|
|
|
static const char *mg_http_status_code_str(int status_code) {
|
|
|
|
switch (status_code) {
|
|
|
|
case 100: return "Continue";
|
|
|
|
case 101: return "Switching Protocols";
|
|
|
|
case 102: return "Processing";
|
|
|
|
case 200: return "OK";
|
|
|
|
case 201: return "Created";
|
|
|
|
case 202: return "Accepted";
|
|
|
|
case 203: return "Non-authoritative Information";
|
|
|
|
case 204: return "No Content";
|
|
|
|
case 205: return "Reset Content";
|
|
|
|
case 206: return "Partial Content";
|
|
|
|
case 207: return "Multi-Status";
|
|
|
|
case 208: return "Already Reported";
|
|
|
|
case 226: return "IM Used";
|
|
|
|
case 300: return "Multiple Choices";
|
|
|
|
case 301: return "Moved Permanently";
|
|
|
|
case 302: return "Found";
|
|
|
|
case 303: return "See Other";
|
|
|
|
case 304: return "Not Modified";
|
|
|
|
case 305: return "Use Proxy";
|
|
|
|
case 307: return "Temporary Redirect";
|
|
|
|
case 308: return "Permanent Redirect";
|
|
|
|
case 400: return "Bad Request";
|
|
|
|
case 401: return "Unauthorized";
|
|
|
|
case 402: return "Payment Required";
|
|
|
|
case 403: return "Forbidden";
|
|
|
|
case 404: return "Not Found";
|
|
|
|
case 405: return "Method Not Allowed";
|
|
|
|
case 406: return "Not Acceptable";
|
|
|
|
case 407: return "Proxy Authentication Required";
|
|
|
|
case 408: return "Request Timeout";
|
|
|
|
case 409: return "Conflict";
|
|
|
|
case 410: return "Gone";
|
|
|
|
case 411: return "Length Required";
|
|
|
|
case 412: return "Precondition Failed";
|
|
|
|
case 413: return "Payload Too Large";
|
|
|
|
case 414: return "Request-URI Too Long";
|
|
|
|
case 415: return "Unsupported Media Type";
|
|
|
|
case 416: return "Requested Range Not Satisfiable";
|
|
|
|
case 417: return "Expectation Failed";
|
|
|
|
case 418: return "I'm a teapot";
|
|
|
|
case 421: return "Misdirected Request";
|
|
|
|
case 422: return "Unprocessable Entity";
|
|
|
|
case 423: return "Locked";
|
|
|
|
case 424: return "Failed Dependency";
|
|
|
|
case 426: return "Upgrade Required";
|
|
|
|
case 428: return "Precondition Required";
|
|
|
|
case 429: return "Too Many Requests";
|
|
|
|
case 431: return "Request Header Fields Too Large";
|
|
|
|
case 444: return "Connection Closed Without Response";
|
|
|
|
case 451: return "Unavailable For Legal Reasons";
|
|
|
|
case 499: return "Client Closed Request";
|
|
|
|
case 500: return "Internal Server Error";
|
|
|
|
case 501: return "Not Implemented";
|
|
|
|
case 502: return "Bad Gateway";
|
|
|
|
case 503: return "Service Unavailable";
|
|
|
|
case 504: return "Gateway Timeout";
|
|
|
|
case 505: return "HTTP Version Not Supported";
|
|
|
|
case 506: return "Variant Also Negotiates";
|
|
|
|
case 507: return "Insufficient Storage";
|
|
|
|
case 508: return "Loop Detected";
|
|
|
|
case 510: return "Not Extended";
|
|
|
|
case 511: return "Network Authentication Required";
|
|
|
|
case 599: return "Network Connect Timeout Error";
|
|
|
|
default: return "OK";
|
|
|
|
}
|
|
|
|
}
|
|
|
|
// clang-format on
|
|
|
|
|
2020-12-18 06:45:22 +08:00
|
|
|
void mg_http_reply(struct mg_connection *c, int code, const char *headers,
|
|
|
|
const char *fmt, ...) {
|
2021-08-08 00:22:47 +08:00
|
|
|
char mem[256], *buf = mem;
|
2020-12-05 19:26:32 +08:00
|
|
|
va_list ap;
|
|
|
|
int len;
|
|
|
|
va_start(ap, fmt);
|
|
|
|
len = mg_vasprintf(&buf, sizeof(mem), fmt, ap);
|
|
|
|
va_end(ap);
|
2021-04-29 16:17:24 +08:00
|
|
|
mg_printf(c, "HTTP/1.1 %d %s\r\n%sContent-Length: %d\r\n\r\n", code,
|
|
|
|
mg_http_status_code_str(code), headers == NULL ? "" : headers, len);
|
2021-05-29 01:30:42 +08:00
|
|
|
mg_send(c, buf, len > 0 ? (size_t) len : 0);
|
2020-12-05 19:26:32 +08:00
|
|
|
if (buf != mem) free(buf);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void http_cb(struct mg_connection *, int, void *, void *);
|
|
|
|
static void restore_http_cb(struct mg_connection *c) {
|
2021-07-29 21:21:20 +08:00
|
|
|
struct mg_fd *fd = (struct mg_fd *) c->pfn_data;
|
|
|
|
if (fd != NULL) fd->fs->close(fd);
|
2021-03-07 19:56:20 +08:00
|
|
|
c->pfn_data = NULL;
|
2020-12-05 19:26:32 +08:00
|
|
|
c->pfn = http_cb;
|
|
|
|
}
|
|
|
|
|
2021-07-29 21:21:20 +08:00
|
|
|
char *mg_http_etag(char *buf, size_t len, size_t size, time_t mtime) {
|
|
|
|
snprintf(buf, len, "\"%lx." MG_INT64_FMT "\"", (unsigned long) mtime,
|
|
|
|
(int64_t) size);
|
2020-12-05 19:26:32 +08:00
|
|
|
return buf;
|
2018-01-15 21:15:59 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
int mg_http_upload(struct mg_connection *c, struct mg_http_message *hm,
|
|
|
|
const char *dir) {
|
|
|
|
char offset[40] = "", name[200] = "", path[256];
|
|
|
|
mg_http_get_var(&hm->query, "offset", offset, sizeof(offset));
|
|
|
|
mg_http_get_var(&hm->query, "name", name, sizeof(name));
|
|
|
|
if (name[0] == '\0') {
|
2020-12-18 06:45:22 +08:00
|
|
|
mg_http_reply(c, 400, "", "%s", "name required");
|
2020-12-05 19:26:32 +08:00
|
|
|
return -1;
|
|
|
|
} else {
|
|
|
|
FILE *fp;
|
|
|
|
size_t oft = strtoul(offset, NULL, 0);
|
|
|
|
snprintf(path, sizeof(path), "%s%c%s", dir, MG_DIRSEP, name);
|
|
|
|
LOG(LL_DEBUG,
|
|
|
|
("%p %d bytes @ %d [%s]", c->fd, (int) hm->body.len, (int) oft, name));
|
2021-07-23 05:46:33 +08:00
|
|
|
if ((fp = fopen(path, oft == 0 ? "wb" : "ab")) == NULL) {
|
2020-12-18 06:45:22 +08:00
|
|
|
mg_http_reply(c, 400, "", "fopen(%s): %d", name, errno);
|
2020-12-05 19:26:32 +08:00
|
|
|
return -2;
|
|
|
|
} else {
|
|
|
|
fwrite(hm->body.ptr, 1, hm->body.len, fp);
|
|
|
|
fclose(fp);
|
2020-12-18 06:45:22 +08:00
|
|
|
mg_http_reply(c, 200, "", "");
|
2021-05-29 01:30:42 +08:00
|
|
|
return (int) hm->body.len;
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2017-11-24 18:21:45 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
static void static_cb(struct mg_connection *c, int ev, void *ev_data,
|
|
|
|
void *fn_data) {
|
|
|
|
if (ev == MG_EV_WRITE || ev == MG_EV_POLL) {
|
2021-07-29 21:21:20 +08:00
|
|
|
struct mg_fd *fd = (struct mg_fd *) fn_data;
|
2020-12-05 19:26:32 +08:00
|
|
|
// Read to send IO buffer directly, avoid extra on-stack buffer
|
|
|
|
size_t n, max = 2 * MG_IO_SIZE;
|
|
|
|
if (c->send.size < max) mg_iobuf_resize(&c->send, max);
|
|
|
|
if (c->send.len >= c->send.size) return; // Rate limit
|
2021-07-29 21:21:20 +08:00
|
|
|
n = fd->fs->read(fd->fd, c->send.buf + c->send.len,
|
|
|
|
c->send.size - c->send.len);
|
2020-12-05 19:26:32 +08:00
|
|
|
if (n > 0) c->send.len += n;
|
|
|
|
if (c->send.len < c->send.size) restore_http_cb(c);
|
|
|
|
} else if (ev == MG_EV_CLOSE) {
|
|
|
|
restore_http_cb(c);
|
|
|
|
}
|
|
|
|
(void) ev_data;
|
2017-11-24 18:21:45 +08:00
|
|
|
}
|
2016-04-22 23:12:42 +08:00
|
|
|
|
2021-07-29 04:11:07 +08:00
|
|
|
static struct mg_str guess_content_type(struct mg_str path, const char *extra) {
|
|
|
|
// clang-format off
|
|
|
|
struct mimeentry { struct mg_str extension, value; };
|
|
|
|
#define MIME_ENTRY(a, b) {{a, sizeof(a) - 1 }, { b, sizeof(b) - 1 }}
|
|
|
|
// clang-format on
|
|
|
|
const struct mimeentry tab[] = {
|
|
|
|
MIME_ENTRY("html", "text/html; charset=utf-8"),
|
|
|
|
MIME_ENTRY("htm", "text/html; charset=utf-8"),
|
|
|
|
MIME_ENTRY("css", "text/css; charset=utf-8"),
|
|
|
|
MIME_ENTRY("js", "text/javascript; charset=utf-8"),
|
|
|
|
MIME_ENTRY("gif", "image/gif"),
|
|
|
|
MIME_ENTRY("png", "image/png"),
|
|
|
|
MIME_ENTRY("woff", "font/woff"),
|
|
|
|
MIME_ENTRY("ttf", "font/ttf"),
|
|
|
|
MIME_ENTRY("aac", "audio/aac"),
|
|
|
|
MIME_ENTRY("avi", "video/x-msvideo"),
|
|
|
|
MIME_ENTRY("azw", "application/vnd.amazon.ebook"),
|
|
|
|
MIME_ENTRY("bin", "application/octet-stream"),
|
|
|
|
MIME_ENTRY("bmp", "image/bmp"),
|
|
|
|
MIME_ENTRY("bz", "application/x-bzip"),
|
|
|
|
MIME_ENTRY("bz2", "application/x-bzip2"),
|
|
|
|
MIME_ENTRY("csv", "text/csv"),
|
|
|
|
MIME_ENTRY("doc", "application/msword"),
|
|
|
|
MIME_ENTRY("epub", "application/epub+zip"),
|
|
|
|
MIME_ENTRY("exe", "application/octet-stream"),
|
|
|
|
MIME_ENTRY("gz", "application/gzip"),
|
|
|
|
MIME_ENTRY("ico", "image/x-icon"),
|
|
|
|
MIME_ENTRY("json", "application/json"),
|
|
|
|
MIME_ENTRY("mid", "audio/mid"),
|
|
|
|
MIME_ENTRY("mjs", "text/javascript"),
|
|
|
|
MIME_ENTRY("mov", "video/quicktime"),
|
|
|
|
MIME_ENTRY("mp3", "audio/mpeg"),
|
|
|
|
MIME_ENTRY("mp4", "video/mp4"),
|
|
|
|
MIME_ENTRY("mpeg", "video/mpeg"),
|
|
|
|
MIME_ENTRY("mpg", "video/mpeg"),
|
|
|
|
MIME_ENTRY("ogg", "application/ogg"),
|
|
|
|
MIME_ENTRY("pdf", "application/pdf"),
|
|
|
|
MIME_ENTRY("rar", "application/rar"),
|
|
|
|
MIME_ENTRY("rtf", "application/rtf"),
|
|
|
|
MIME_ENTRY("shtml", "text/html; charset=utf-8"),
|
|
|
|
MIME_ENTRY("svg", "image/svg+xml"),
|
|
|
|
MIME_ENTRY("tar", "application/tar"),
|
|
|
|
MIME_ENTRY("tgz", "application/tar-gz"),
|
|
|
|
MIME_ENTRY("txt", "text/plain; charset=utf-8"),
|
|
|
|
MIME_ENTRY("wasm", "application/wasm"),
|
|
|
|
MIME_ENTRY("wav", "audio/wav"),
|
|
|
|
MIME_ENTRY("weba", "audio/webm"),
|
|
|
|
MIME_ENTRY("webm", "video/webm"),
|
|
|
|
MIME_ENTRY("webp", "image/webp"),
|
|
|
|
MIME_ENTRY("xls", "application/excel"),
|
|
|
|
MIME_ENTRY("xml", "application/xml"),
|
|
|
|
MIME_ENTRY("xsl", "application/xml"),
|
|
|
|
MIME_ENTRY("zip", "application/zip"),
|
|
|
|
MIME_ENTRY("3gp", "video/3gpp"),
|
|
|
|
MIME_ENTRY("7z", "application/x-7z-compressed"),
|
|
|
|
MIME_ENTRY("7z", "application/x-7z-compressed"),
|
|
|
|
{{0, 0}, {0, 0}},
|
|
|
|
};
|
|
|
|
size_t i = 0;
|
|
|
|
struct mg_str k, v, s = mg_str(extra);
|
|
|
|
|
|
|
|
// Shrink path to its extension only
|
|
|
|
while (i < path.len && path.ptr[path.len - i - 1] != '.') i++;
|
|
|
|
path.ptr += path.len - i;
|
|
|
|
path.len = i;
|
|
|
|
|
|
|
|
// Process user-provided mime type overrides, if any
|
2021-08-28 13:54:56 +08:00
|
|
|
while (mg_commalist(&s, &k, &v)) {
|
2021-07-29 04:11:07 +08:00
|
|
|
if (mg_strcmp(path, k) == 0) return v;
|
2020-12-15 19:58:30 +08:00
|
|
|
}
|
2021-07-29 04:11:07 +08:00
|
|
|
|
|
|
|
// Process built-in mime types
|
|
|
|
for (i = 0; tab[i].extension.ptr != NULL; i++) {
|
|
|
|
if (mg_strcmp(path, tab[i].extension) == 0) return tab[i].value;
|
|
|
|
}
|
|
|
|
|
|
|
|
return mg_str("text/plain; charset=utf-8");
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
|
2021-04-29 18:00:57 +08:00
|
|
|
static int getrange(struct mg_str *s, int64_t *a, int64_t *b) {
|
2021-05-29 01:30:42 +08:00
|
|
|
size_t i, numparsed = 0;
|
2021-04-29 18:00:57 +08:00
|
|
|
LOG(LL_INFO, ("%.*s", (int) s->len, s->ptr));
|
2021-05-29 01:30:42 +08:00
|
|
|
for (i = 0; i + 6 < s->len; i++) {
|
2021-04-29 18:00:57 +08:00
|
|
|
if (memcmp(&s->ptr[i], "bytes=", 6) == 0) {
|
|
|
|
struct mg_str p = mg_str_n(s->ptr + i + 6, s->len - i - 6);
|
|
|
|
if (p.len > 0 && p.ptr[0] >= '0' && p.ptr[0] <= '9') numparsed++;
|
|
|
|
*a = mg_to64(p);
|
|
|
|
// LOG(LL_INFO, ("PPP [%.*s] %d", (int) p.len, p.ptr, numparsed));
|
|
|
|
while (p.len && p.ptr[0] >= '0' && p.ptr[0] <= '9') p.ptr++, p.len--;
|
|
|
|
if (p.len && p.ptr[0] == '-') p.ptr++, p.len--;
|
|
|
|
*b = mg_to64(p);
|
|
|
|
if (p.len > 0 && p.ptr[0] >= '0' && p.ptr[0] <= '9') numparsed++;
|
|
|
|
// LOG(LL_INFO, ("PPP [%.*s] %d", (int) p.len, p.ptr, numparsed));
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
2021-05-29 01:30:42 +08:00
|
|
|
return (int) numparsed;
|
2021-04-29 18:00:57 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
void mg_http_serve_file(struct mg_connection *c, struct mg_http_message *hm,
|
2021-07-29 04:11:07 +08:00
|
|
|
const char *path, struct mg_http_serve_opts *opts) {
|
2020-12-05 19:26:32 +08:00
|
|
|
char etag[64];
|
2021-07-29 21:21:20 +08:00
|
|
|
struct mg_fs *fs = opts->fs == NULL ? &mg_fs_posix : opts->fs;
|
|
|
|
struct mg_fd *fd = fs->open(path, MG_FS_READ);
|
|
|
|
size_t size = 0;
|
|
|
|
time_t mtime = 0;
|
|
|
|
struct mg_str *inm = NULL;
|
|
|
|
|
|
|
|
if (fd == NULL || fs->stat(path, &size, &mtime) == 0) {
|
2021-09-17 18:07:20 +08:00
|
|
|
LOG(LL_DEBUG, ("404 [%s] %p", path, (void *) fd));
|
2020-12-18 06:45:22 +08:00
|
|
|
mg_http_reply(c, 404, "", "%s", "Not found\n");
|
2021-07-29 21:21:20 +08:00
|
|
|
fs->close(fd);
|
|
|
|
// NOTE: mg_http_etag() call should go first!
|
|
|
|
} else if (mg_http_etag(etag, sizeof(etag), size, mtime) != NULL &&
|
|
|
|
(inm = mg_http_get_header(hm, "If-None-Match")) != NULL &&
|
|
|
|
mg_vcasecmp(inm, etag) == 0) {
|
|
|
|
fs->close(fd);
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_printf(c, "HTTP/1.1 304 Not Modified\r\nContent-Length: 0\r\n\r\n");
|
|
|
|
} else {
|
2021-04-29 18:00:57 +08:00
|
|
|
int n, status = 200;
|
2021-05-29 01:30:42 +08:00
|
|
|
char range[100] = "";
|
2021-07-29 21:21:20 +08:00
|
|
|
int64_t r1 = 0, r2 = 0, cl = (int64_t) size;
|
2021-07-29 04:11:07 +08:00
|
|
|
struct mg_str mime = guess_content_type(mg_str(path), opts->mime_types);
|
2021-04-29 18:00:57 +08:00
|
|
|
|
|
|
|
// Handle Range header
|
|
|
|
struct mg_str *rh = mg_http_get_header(hm, "Range");
|
|
|
|
if (rh != NULL && (n = getrange(rh, &r1, &r2)) > 0 && r1 >= 0 && r2 >= 0) {
|
|
|
|
// If range is specified like "400-", set second limit to content len
|
|
|
|
if (n == 1) r2 = cl - 1;
|
|
|
|
if (r1 > r2 || r2 >= cl) {
|
|
|
|
status = 416;
|
|
|
|
cl = 0;
|
|
|
|
snprintf(range, sizeof(range),
|
2021-07-29 21:21:20 +08:00
|
|
|
"Content-Range: bytes */" MG_INT64_FMT "\r\n", (int64_t) size);
|
2021-04-29 18:00:57 +08:00
|
|
|
} else {
|
|
|
|
status = 206;
|
|
|
|
cl = r2 - r1 + 1;
|
|
|
|
snprintf(range, sizeof(range),
|
|
|
|
"Content-Range: bytes " MG_INT64_FMT "-" MG_INT64_FMT
|
|
|
|
"/" MG_INT64_FMT "\r\n",
|
2021-07-29 21:21:20 +08:00
|
|
|
r1, r1 + cl - 1, (int64_t) size);
|
|
|
|
fs->seek(fd->fd, (size_t) r1);
|
2021-04-29 18:00:57 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_printf(c,
|
2021-07-29 04:11:07 +08:00
|
|
|
"HTTP/1.1 %d %s\r\nContent-Type: %.*s\r\n"
|
2021-04-29 18:00:57 +08:00
|
|
|
"Etag: %s\r\nContent-Length: " MG_INT64_FMT "\r\n%s%s\r\n",
|
2021-07-29 04:11:07 +08:00
|
|
|
status, mg_http_status_code_str(status), (int) mime.len, mime.ptr,
|
|
|
|
etag, cl, range, opts->extra_headers ? opts->extra_headers : "");
|
2020-12-05 19:26:32 +08:00
|
|
|
if (mg_vcasecmp(&hm->method, "HEAD") == 0) {
|
2021-08-05 17:39:24 +08:00
|
|
|
c->is_draining = 1;
|
2021-07-29 21:21:20 +08:00
|
|
|
fs->close(fd);
|
2020-12-05 19:26:32 +08:00
|
|
|
} else {
|
2021-03-07 19:56:20 +08:00
|
|
|
c->pfn = static_cb;
|
2021-07-29 21:21:20 +08:00
|
|
|
c->pfn_data = fd;
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2017-01-24 20:41:44 +08:00
|
|
|
|
2021-07-29 21:21:20 +08:00
|
|
|
struct printdirentrydata {
|
|
|
|
struct mg_connection *c;
|
|
|
|
struct mg_http_message *hm;
|
|
|
|
struct mg_http_serve_opts *opts;
|
|
|
|
const char *dir;
|
2017-01-24 20:41:44 +08:00
|
|
|
};
|
2016-12-23 20:24:27 +08:00
|
|
|
|
2021-07-29 21:21:20 +08:00
|
|
|
static void printdirentry(const char *name, void *userdata) {
|
|
|
|
struct printdirentrydata *d = (struct printdirentrydata *) userdata;
|
|
|
|
struct mg_fs *fs = d->opts->fs == NULL ? &mg_fs_posix : d->opts->fs;
|
|
|
|
size_t size = 0;
|
|
|
|
time_t mtime = 0;
|
|
|
|
char path[MG_PATH_MAX], sz[64], mod[64];
|
|
|
|
int flags, n = 0;
|
|
|
|
|
2021-07-31 00:41:23 +08:00
|
|
|
// LOG(LL_DEBUG, ("[%s] [%s]", d->dir, name));
|
2021-07-30 20:19:20 +08:00
|
|
|
if (snprintf(path, sizeof(path), "%s%c%s", d->dir, '/', name) < 0) {
|
2021-07-29 21:21:20 +08:00
|
|
|
LOG(LL_ERROR, ("%s truncated", name));
|
|
|
|
} else if ((flags = fs->stat(path, &size, &mtime)) == 0) {
|
|
|
|
LOG(LL_ERROR, ("%lu stat(%s): %d", d->c->id, path, errno));
|
2020-12-05 19:26:32 +08:00
|
|
|
} else {
|
2021-07-29 21:21:20 +08:00
|
|
|
const char *slash = flags & MG_FS_DIR ? "/" : "";
|
|
|
|
struct tm t;
|
|
|
|
if (flags & MG_FS_DIR) {
|
|
|
|
snprintf(sz, sizeof(sz), "%s", "[DIR]");
|
|
|
|
} else if (size < 1024) {
|
|
|
|
snprintf(sz, sizeof(sz), "%d", (int) size);
|
|
|
|
} else if (size < 0x100000) {
|
|
|
|
snprintf(sz, sizeof(sz), "%.1fk", (double) size / 1024.0);
|
|
|
|
} else if (size < 0x40000000) {
|
|
|
|
snprintf(sz, sizeof(sz), "%.1fM", (double) size / 1048576);
|
2020-12-05 19:26:32 +08:00
|
|
|
} else {
|
2021-07-29 21:21:20 +08:00
|
|
|
snprintf(sz, sizeof(sz), "%.1fG", (double) size / 1073741824);
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
2021-07-29 21:21:20 +08:00
|
|
|
strftime(mod, sizeof(mod), "%d-%b-%Y %H:%M", localtime_r(&mtime, &t));
|
|
|
|
n = (int) mg_url_encode(name, strlen(name), path, sizeof(path));
|
|
|
|
mg_printf(d->c,
|
|
|
|
" <tr><td><a href=\"%.*s%s\">%s%s</a></td>"
|
|
|
|
"<td>%s</td><td>%s</td></tr>\n",
|
|
|
|
n, path, slash, name, slash, mod, sz);
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static void listdir(struct mg_connection *c, struct mg_http_message *hm,
|
2021-03-10 15:56:14 +08:00
|
|
|
struct mg_http_serve_opts *opts, char *dir) {
|
2021-02-08 06:43:49 +08:00
|
|
|
static const char *sort_js_code =
|
|
|
|
"<script>function srt(tb, sc, so, d) {"
|
|
|
|
"var tr = Array.prototype.slice.call(tb.rows, 0),"
|
|
|
|
"tr = tr.sort(function (a, b) { var c1 = a.cells[sc], c2 = b.cells[sc],"
|
|
|
|
"n1 = c1.getAttribute('name'), n2 = c2.getAttribute('name'), "
|
|
|
|
"t1 = a.cells[2].getAttribute('name'), "
|
|
|
|
"t2 = b.cells[2].getAttribute('name'); "
|
|
|
|
"return so * (t1 < 0 && t2 >= 0 ? -1 : t2 < 0 && t1 >= 0 ? 1 : "
|
|
|
|
"n1 ? parseInt(n2) - parseInt(n1) : "
|
|
|
|
"c1.textContent.trim().localeCompare(c2.textContent.trim())); });";
|
|
|
|
static const char *sort_js_code2 =
|
|
|
|
"for (var i = 0; i < tr.length; i++) tb.appendChild(tr[i]); "
|
|
|
|
"if (!d) window.location.hash = ('sc=' + sc + '&so=' + so); "
|
|
|
|
"};"
|
|
|
|
"window.onload = function() {"
|
|
|
|
"var tb = document.getElementById('tb');"
|
|
|
|
"var m = /sc=([012]).so=(1|-1)/.exec(window.location.hash) || [0, 2, 1];"
|
|
|
|
"var sc = m[1], so = m[2]; document.onclick = function(ev) { "
|
|
|
|
"var c = ev.target.rel; if (c) {if (c == sc) so *= -1; srt(tb, c, so); "
|
|
|
|
"sc = c; ev.preventDefault();}};"
|
|
|
|
"srt(tb, sc, so, true);"
|
|
|
|
"}"
|
|
|
|
"</script>";
|
2021-07-29 21:21:20 +08:00
|
|
|
struct mg_fs *fs = opts->fs == NULL ? &mg_fs_posix : opts->fs;
|
|
|
|
struct printdirentrydata d = {c, hm, opts, dir};
|
|
|
|
char tmp[10];
|
|
|
|
size_t off, n;
|
2016-04-18 23:04:45 +08:00
|
|
|
|
2021-07-29 21:21:20 +08:00
|
|
|
mg_printf(c,
|
|
|
|
"HTTP/1.1 200 OK\r\n"
|
|
|
|
"Content-Type: text/html; charset=utf-8\r\n"
|
|
|
|
"%s"
|
|
|
|
"Content-Length: \r\n\r\n",
|
|
|
|
opts->extra_headers == NULL ? "" : opts->extra_headers);
|
|
|
|
off = c->send.len; // Start of body
|
|
|
|
mg_printf(c,
|
|
|
|
"<!DOCTYPE html><html><head><title>Index of %.*s</title>%s%s"
|
|
|
|
"<style>th,td {text-align: left; padding-right: 1em; "
|
|
|
|
"font-family: monospace; }</style></head>"
|
|
|
|
"<body><h1>Index of %.*s</h1><table cellpadding=\"0\"><thead>"
|
|
|
|
"<tr><th><a href=\"#\" rel=\"0\">Name</a></th><th>"
|
|
|
|
"<a href=\"#\" rel=\"1\">Modified</a></th>"
|
|
|
|
"<th><a href=\"#\" rel=\"2\">Size</a></th></tr>"
|
|
|
|
"<tr><td colspan=\"3\"><hr></td></tr>"
|
|
|
|
"</thead>"
|
|
|
|
"<tbody id=\"tb\">\n",
|
|
|
|
(int) hm->uri.len, hm->uri.ptr, sort_js_code, sort_js_code2,
|
|
|
|
(int) hm->uri.len, hm->uri.ptr);
|
|
|
|
|
|
|
|
fs->list(dir, printdirentry, &d);
|
|
|
|
mg_printf(c,
|
|
|
|
"</tbody><tfoot><tr><td colspan=\"3\"><hr></td></tr></tfoot>"
|
|
|
|
"</table><address>Mongoose v.%s</address></body></html>\n",
|
|
|
|
MG_VERSION);
|
|
|
|
n = (size_t) snprintf(tmp, sizeof(tmp), "%lu",
|
|
|
|
(unsigned long) (c->send.len - off));
|
|
|
|
if (n > sizeof(tmp)) n = 0;
|
|
|
|
memcpy(c->send.buf + off - 10, tmp, n); // Set content length
|
|
|
|
}
|
|
|
|
|
2021-09-15 14:43:48 +08:00
|
|
|
static void remove_double_dots(char *s) {
|
|
|
|
char *p = s;
|
|
|
|
while (*s != '\0') {
|
|
|
|
*p++ = *s++;
|
|
|
|
if (s[-1] == '/' || s[-1] == '\\') {
|
|
|
|
while (s[0] != '\0') {
|
|
|
|
if (s[0] == '/' || s[0] == '\\') {
|
|
|
|
s++;
|
|
|
|
} else if (s[0] == '.' && s[1] == '.') {
|
|
|
|
s += 2;
|
|
|
|
} else {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
*p = '\0';
|
|
|
|
}
|
|
|
|
|
|
|
|
// Resolve requested file into `path` and return its fs->stat() result
|
2021-09-20 19:24:44 +08:00
|
|
|
static int uri_to_path2(struct mg_connection *c, struct mg_http_message *hm,
|
|
|
|
struct mg_fs *fs, struct mg_str url, struct mg_str dir,
|
|
|
|
char *path, size_t path_size) {
|
2021-07-29 21:21:20 +08:00
|
|
|
int flags = 0, tmp;
|
2021-09-15 14:43:48 +08:00
|
|
|
// Append URI to the root_dir, and sanitize it
|
2021-09-20 19:24:44 +08:00
|
|
|
size_t n = (size_t) snprintf(path, path_size, "%.*s", (int) dir.len, dir.ptr);
|
2021-09-15 14:43:48 +08:00
|
|
|
if (n > path_size) n = path_size;
|
2021-09-20 19:24:44 +08:00
|
|
|
path[path_size - 1] = '\0';
|
|
|
|
if ((fs->stat(path, NULL, NULL) & MG_FS_DIR) == 0) {
|
|
|
|
mg_http_reply(c, 400, "", "Invalid web root [%.*s]\n", (int) dir.len,
|
|
|
|
dir.ptr);
|
|
|
|
} else {
|
|
|
|
if (n + 2 < path_size) path[n++] = '/', path[n] = '\0';
|
|
|
|
mg_url_decode(hm->uri.ptr + url.len, hm->uri.len - url.len, path + n,
|
|
|
|
path_size - n, 0);
|
|
|
|
path[path_size - 1] = '\0'; // Double-check
|
|
|
|
remove_double_dots(path);
|
|
|
|
n = strlen(path);
|
|
|
|
LOG(LL_DEBUG, ("--> %s", path));
|
|
|
|
while (n > 0 && path[n - 1] == '/') path[--n] = 0; // Trim trailing slashes
|
|
|
|
flags = fs->stat(path, NULL, NULL); // Does it exist?
|
|
|
|
if (flags == 0) {
|
|
|
|
mg_http_reply(c, 404, "", "Not found\n"); // Does not exist, doh
|
|
|
|
} else if (flags & MG_FS_DIR) {
|
|
|
|
if (((snprintf(path + n, path_size - n, "/index.html") > 0 &&
|
|
|
|
(tmp = fs->stat(path, NULL, NULL)) != 0) ||
|
|
|
|
(snprintf(path + n, path_size - n, "/index.shtml") > 0 &&
|
|
|
|
(tmp = fs->stat(path, NULL, NULL)) != 0))) {
|
|
|
|
flags = tmp;
|
|
|
|
} else {
|
|
|
|
path[n] = '\0'; // Remove appended index file name
|
|
|
|
}
|
2021-07-29 21:21:20 +08:00
|
|
|
}
|
2021-07-25 05:00:37 +08:00
|
|
|
}
|
2021-07-29 21:21:20 +08:00
|
|
|
return flags;
|
2021-07-25 05:00:37 +08:00
|
|
|
}
|
|
|
|
|
2021-09-20 19:24:44 +08:00
|
|
|
static int uri_to_path(struct mg_connection *c, struct mg_http_message *hm,
|
|
|
|
struct mg_http_serve_opts *opts, char *path,
|
|
|
|
size_t path_size) {
|
|
|
|
struct mg_fs *fs = opts->fs == NULL ? &mg_fs_posix : opts->fs;
|
|
|
|
struct mg_str k, v, s = mg_str(opts->root_dir), u = {0, 0}, p = {0, 0};
|
|
|
|
while (mg_commalist(&s, &k, &v)) {
|
|
|
|
if (v.len == 0) v = k, k = mg_str("/");
|
|
|
|
if (hm->uri.len < k.len) continue;
|
|
|
|
if (mg_strcmp(k, mg_str_n(hm->uri.ptr, k.len)) != 0) continue;
|
|
|
|
u = k, p = v;
|
|
|
|
}
|
|
|
|
return uri_to_path2(c, hm, fs, u, p, path, path_size);
|
|
|
|
}
|
|
|
|
|
2021-07-25 05:00:37 +08:00
|
|
|
void mg_http_serve_dir(struct mg_connection *c, struct mg_http_message *hm,
|
|
|
|
struct mg_http_serve_opts *opts) {
|
2021-09-15 14:43:48 +08:00
|
|
|
char path[MG_PATH_MAX] = "";
|
|
|
|
const char *sp = opts->ssi_pattern;
|
2021-09-20 19:24:44 +08:00
|
|
|
int flags = uri_to_path(c, hm, opts, path, sizeof(path));
|
|
|
|
if (flags == 0) return;
|
|
|
|
LOG(LL_DEBUG, ("%.*s %s %d", (int) hm->uri.len, hm->uri.ptr, path, flags));
|
|
|
|
if (flags & MG_FS_DIR) {
|
|
|
|
listdir(c, hm, opts, path);
|
|
|
|
} else if (sp != NULL && mg_globmatch(sp, strlen(sp), path, strlen(path))) {
|
|
|
|
mg_http_serve_ssi(c, opts->root_dir, path);
|
|
|
|
} else {
|
|
|
|
mg_http_serve_file(c, hm, path, opts);
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
}
|
2015-12-16 08:52:37 +08:00
|
|
|
|
2021-02-15 22:42:55 +08:00
|
|
|
static bool mg_is_url_safe(int c) {
|
|
|
|
return (c >= '0' && c <= '9') || (c >= 'a' && c <= 'z') ||
|
|
|
|
(c >= 'A' && c <= 'Z') || c == '.' || c == '_' || c == '-' || c == '~';
|
|
|
|
}
|
|
|
|
|
2021-05-29 01:30:42 +08:00
|
|
|
size_t mg_url_encode(const char *s, size_t sl, char *buf, size_t len) {
|
2021-02-15 22:42:55 +08:00
|
|
|
size_t i, n = 0;
|
|
|
|
for (i = 0; i < sl; i++) {
|
|
|
|
int c = *(unsigned char *) &s[i];
|
|
|
|
if (n + 4 >= len) return 0;
|
|
|
|
if (mg_is_url_safe(c)) {
|
|
|
|
buf[n++] = s[i];
|
|
|
|
} else {
|
|
|
|
buf[n++] = '%';
|
|
|
|
mg_hex(&s[i], 1, &buf[n]);
|
|
|
|
n += 2;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return n;
|
|
|
|
}
|
|
|
|
|
2021-05-29 01:30:42 +08:00
|
|
|
void mg_http_creds(struct mg_http_message *hm, char *user, size_t userlen,
|
|
|
|
char *pass, size_t passlen) {
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_str *v = mg_http_get_header(hm, "Authorization");
|
|
|
|
user[0] = pass[0] = '\0';
|
|
|
|
if (v != NULL && v->len > 6 && memcmp(v->ptr, "Basic ", 6) == 0) {
|
|
|
|
char buf[256];
|
|
|
|
int n = mg_base64_decode(v->ptr + 6, (int) v->len - 6, buf);
|
2021-05-29 01:30:42 +08:00
|
|
|
const char *p = (const char *) memchr(buf, ':', n > 0 ? (size_t) n : 0);
|
2020-12-05 19:26:32 +08:00
|
|
|
if (p != NULL) {
|
|
|
|
snprintf(user, userlen, "%.*s", (int) (p - buf), buf);
|
|
|
|
snprintf(pass, passlen, "%.*s", n - (int) (p - buf) - 1, p + 1);
|
|
|
|
}
|
|
|
|
} else if (v != NULL && v->len > 7 && memcmp(v->ptr, "Bearer ", 7) == 0) {
|
|
|
|
snprintf(pass, passlen, "%.*s", (int) v->len - 7, v->ptr + 7);
|
|
|
|
} else if ((v = mg_http_get_header(hm, "Cookie")) != NULL) {
|
2021-03-17 15:43:29 +08:00
|
|
|
struct mg_str t = mg_http_get_header_var(*v, mg_str_n("access_token", 12));
|
|
|
|
if (t.len > 0) snprintf(pass, passlen, "%.*s", (int) t.len, t.ptr);
|
2020-12-05 19:26:32 +08:00
|
|
|
} else {
|
|
|
|
mg_http_get_var(&hm->query, "access_token", pass, passlen);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-03-17 15:43:29 +08:00
|
|
|
static struct mg_str stripquotes(struct mg_str s) {
|
|
|
|
return s.len > 1 && s.ptr[0] == '"' && s.ptr[s.len - 1] == '"'
|
|
|
|
? mg_str_n(s.ptr + 1, s.len - 2)
|
|
|
|
: s;
|
|
|
|
}
|
|
|
|
|
|
|
|
struct mg_str mg_http_get_header_var(struct mg_str s, struct mg_str v) {
|
|
|
|
size_t i;
|
|
|
|
for (i = 0; i + v.len + 2 < s.len; i++) {
|
|
|
|
if (s.ptr[i + v.len] == '=' && memcmp(&s.ptr[i], v.ptr, v.len) == 0) {
|
2021-03-17 21:28:36 +08:00
|
|
|
const char *p = &s.ptr[i + v.len + 1], *b = p, *x = &s.ptr[s.len];
|
|
|
|
int q = p < x && *p == '"' ? 1 : 0;
|
|
|
|
while (p < x && (q ? p == b || *p != '"' : *p != ';' && *p != ' ')) p++;
|
|
|
|
// LOG(LL_INFO, ("[%.*s] [%.*s] [%.*s]", (int) s.len, s.ptr, (int) v.len,
|
|
|
|
// v.ptr, (int) (p - b), b));
|
2021-08-08 00:22:47 +08:00
|
|
|
return stripquotes(mg_str_n(b, (size_t) (p - b + q)));
|
2021-03-17 15:43:29 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
return mg_str_n(NULL, 0);
|
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
bool mg_http_match_uri(const struct mg_http_message *hm, const char *glob) {
|
|
|
|
return mg_globmatch(glob, strlen(glob), hm->uri.ptr, hm->uri.len);
|
|
|
|
}
|
|
|
|
|
2021-03-08 23:41:16 +08:00
|
|
|
static size_t get_chunk_length(const char *buf, size_t len, size_t *ll) {
|
|
|
|
size_t i = 0, n;
|
|
|
|
while (i < len && buf[i] != '\r' && i != '\n') i++;
|
|
|
|
n = mg_unhexn((char *) buf, i);
|
|
|
|
while (i < len && (buf[i] == '\r' || i == '\n')) i++;
|
|
|
|
// LOG(LL_INFO, ("len %zu i %zu n %zu ", len, i, n));
|
|
|
|
if (ll != NULL) *ll = i + 1;
|
|
|
|
if (i < len && i + n + 2 < len) return i + n + 3;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Walk through all chunks in the chunked body. For each chunk, fire
|
|
|
|
// an MG_EV_HTTP_CHUNK event.
|
|
|
|
static void walkchunks(struct mg_connection *c, struct mg_http_message *hm,
|
2021-05-29 01:30:42 +08:00
|
|
|
size_t reqlen) {
|
2021-03-09 16:21:29 +08:00
|
|
|
size_t off = 0, bl, ll;
|
2021-03-09 17:54:02 +08:00
|
|
|
while (off + reqlen < c->recv.len) {
|
2021-03-08 23:41:16 +08:00
|
|
|
char *buf = (char *) &c->recv.buf[reqlen];
|
|
|
|
size_t memo = c->recv.len;
|
|
|
|
size_t cl = get_chunk_length(&buf[off], memo - reqlen - off, &ll);
|
|
|
|
// LOG(LL_INFO, ("len %zu off %zu cl %zu ll %zu", len, off, cl, ll));
|
|
|
|
if (cl == 0) break;
|
|
|
|
hm->chunk = mg_str_n(&buf[off + ll], cl < ll + 2 ? 0 : cl - ll - 2);
|
|
|
|
mg_call(c, MG_EV_HTTP_CHUNK, hm);
|
|
|
|
// Increase offset only if user has not deleted this chunk
|
|
|
|
if (memo == c->recv.len) off += cl;
|
|
|
|
if (cl <= 5) {
|
2021-03-09 16:21:29 +08:00
|
|
|
// Zero chunk - last one. Prepare body - cut off chunk lengths
|
|
|
|
off = bl = 0;
|
2021-03-09 17:54:02 +08:00
|
|
|
while (off + reqlen < c->recv.len) {
|
2021-05-29 01:30:42 +08:00
|
|
|
char *buf2 = (char *) &c->recv.buf[reqlen];
|
|
|
|
size_t memo2 = c->recv.len;
|
|
|
|
size_t cl2 = get_chunk_length(&buf2[off], memo2 - reqlen - off, &ll);
|
2021-08-23 14:40:35 +08:00
|
|
|
size_t n = cl2 < ll + 2 ? 0 : cl2 - ll - 2;
|
2021-05-29 01:30:42 +08:00
|
|
|
memmove(buf2 + bl, buf2 + off + ll, n);
|
2021-03-09 16:21:29 +08:00
|
|
|
bl += n;
|
2021-05-29 01:30:42 +08:00
|
|
|
off += cl2;
|
|
|
|
if (cl2 <= 5) break;
|
2021-03-09 16:21:29 +08:00
|
|
|
}
|
2021-03-19 21:13:24 +08:00
|
|
|
// LOG(LL_INFO, ("BL->%d del %d off %d", (int) bl, (int) del, (int) off));
|
|
|
|
c->recv.len -= off - bl;
|
2021-03-09 16:21:29 +08:00
|
|
|
// Set message length to indicate we've received
|
2021-03-08 23:41:16 +08:00
|
|
|
// everything, to fire MG_EV_HTTP_MSG
|
2021-03-09 16:21:29 +08:00
|
|
|
hm->message.len = bl + reqlen;
|
|
|
|
hm->body.len = bl;
|
2021-03-08 23:41:16 +08:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static bool mg_is_chunked(struct mg_http_message *hm) {
|
|
|
|
struct mg_str needle = mg_str_n("chunked", 7);
|
|
|
|
struct mg_str *te = mg_http_get_header(hm, "Transfer-Encoding");
|
|
|
|
return te != NULL && mg_strstr(*te, needle) != NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
void mg_http_delete_chunk(struct mg_connection *c, struct mg_http_message *hm) {
|
|
|
|
struct mg_str ch = hm->chunk;
|
|
|
|
if (mg_is_chunked(hm)) {
|
|
|
|
ch.len += 4; // \r\n before and after the chunk
|
|
|
|
ch.ptr -= 2;
|
|
|
|
while (ch.ptr > hm->body.ptr && *ch.ptr != '\n') ch.ptr--, ch.len++;
|
|
|
|
}
|
|
|
|
{
|
|
|
|
const char *end = &ch.ptr[ch.len];
|
2021-08-08 00:22:47 +08:00
|
|
|
size_t n = (size_t) (end - (char *) c->recv.buf);
|
2021-05-29 01:30:42 +08:00
|
|
|
if (c->recv.len > n) {
|
2021-08-08 00:22:47 +08:00
|
|
|
memmove((char *) ch.ptr, end, (size_t) (c->recv.len - n));
|
2021-05-29 01:30:42 +08:00
|
|
|
}
|
2021-03-08 23:41:16 +08:00
|
|
|
// LOG(LL_INFO, ("DELETING CHUNK: %zu %zu %zu\n%.*s", c->recv.len, n,
|
|
|
|
// ch.len, (int) ch.len, ch.ptr));
|
|
|
|
}
|
|
|
|
c->recv.len -= ch.len;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void http_cb(struct mg_connection *c, int ev, void *evd, void *fnd) {
|
2020-12-05 19:26:32 +08:00
|
|
|
if (ev == MG_EV_READ || ev == MG_EV_CLOSE) {
|
|
|
|
struct mg_http_message hm;
|
|
|
|
for (;;) {
|
|
|
|
int n = mg_http_parse((char *) c->recv.buf, c->recv.len, &hm);
|
2021-03-08 23:41:16 +08:00
|
|
|
bool is_chunked = n > 0 && mg_is_chunked(&hm);
|
2020-12-05 19:26:32 +08:00
|
|
|
if (ev == MG_EV_CLOSE) {
|
|
|
|
hm.message.len = c->recv.len;
|
2021-08-08 00:22:47 +08:00
|
|
|
hm.body.len = hm.message.len - (size_t) (hm.body.ptr - hm.message.ptr);
|
2021-05-29 01:30:42 +08:00
|
|
|
} else if (is_chunked && n > 0) {
|
|
|
|
walkchunks(c, &hm, (size_t) n);
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
2021-03-08 23:41:16 +08:00
|
|
|
// LOG(LL_INFO,
|
|
|
|
//("---->%d %d\n%.*s", n, is_chunked, (int) c->recv.len, c->recv.buf));
|
2020-12-05 19:26:32 +08:00
|
|
|
if (n < 0 && ev == MG_EV_READ) {
|
2021-03-19 21:13:24 +08:00
|
|
|
mg_error(c, "HTTP parse:\n%.*s", (int) c->recv.len, c->recv.buf);
|
2020-12-05 19:26:32 +08:00
|
|
|
break;
|
|
|
|
} else if (n > 0 && (size_t) c->recv.len >= hm.message.len) {
|
|
|
|
mg_call(c, MG_EV_HTTP_MSG, &hm);
|
2021-08-28 14:16:38 +08:00
|
|
|
mg_iobuf_del(&c->recv, 0, hm.message.len);
|
2020-12-05 19:26:32 +08:00
|
|
|
} else {
|
2021-03-08 23:41:16 +08:00
|
|
|
if (n > 0 && !is_chunked) {
|
2021-05-29 01:30:42 +08:00
|
|
|
hm.chunk =
|
|
|
|
mg_str_n((char *) &c->recv.buf[n], c->recv.len - (size_t) n);
|
2021-03-08 23:41:16 +08:00
|
|
|
mg_call(c, MG_EV_HTTP_CHUNK, &hm);
|
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2021-03-08 23:41:16 +08:00
|
|
|
(void) fnd;
|
|
|
|
(void) evd;
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
struct mg_connection *mg_http_connect(struct mg_mgr *mgr, const char *url,
|
|
|
|
mg_event_handler_t fn, void *fn_data) {
|
|
|
|
struct mg_connection *c = mg_connect(mgr, url, fn, fn_data);
|
2021-03-07 19:56:20 +08:00
|
|
|
if (c != NULL) c->pfn = http_cb;
|
2020-12-05 19:26:32 +08:00
|
|
|
return c;
|
|
|
|
}
|
|
|
|
|
|
|
|
struct mg_connection *mg_http_listen(struct mg_mgr *mgr, const char *url,
|
|
|
|
mg_event_handler_t fn, void *fn_data) {
|
|
|
|
struct mg_connection *c = mg_listen(mgr, url, fn, fn_data);
|
2021-03-07 19:56:20 +08:00
|
|
|
if (c != NULL) c->pfn = http_cb;
|
2020-12-05 19:26:32 +08:00
|
|
|
return c;
|
|
|
|
}
|
|
|
|
|
|
|
|
#ifdef MG_ENABLE_LINES
|
|
|
|
#line 1 "src/iobuf.c"
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#include <string.h>
|
|
|
|
|
2021-04-29 15:49:23 +08:00
|
|
|
// Not using memset for zeroing memory, cause it can be dropped by compiler
|
|
|
|
// See https://github.com/cesanta/mongoose/pull/1265
|
|
|
|
static void zeromem(volatile unsigned char *buf, size_t len) {
|
|
|
|
if (buf != NULL) {
|
|
|
|
while (len--) *buf++ = 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-01-29 20:32:34 +08:00
|
|
|
int mg_iobuf_resize(struct mg_iobuf *io, size_t new_size) {
|
|
|
|
int ok = 1;
|
2020-12-05 19:26:32 +08:00
|
|
|
if (new_size == 0) {
|
2021-04-29 15:49:23 +08:00
|
|
|
zeromem(io->buf, io->size);
|
2020-12-11 01:39:40 +08:00
|
|
|
free(io->buf);
|
|
|
|
io->buf = NULL;
|
|
|
|
io->len = io->size = 0;
|
2020-12-05 19:26:32 +08:00
|
|
|
} else if (new_size != io->size) {
|
2021-05-18 00:36:57 +08:00
|
|
|
// NOTE(lsm): do not use realloc here. Use calloc/free only, to ease the
|
2020-12-05 19:26:32 +08:00
|
|
|
// porting to some obscure platforms like FreeRTOS
|
2021-05-18 00:36:57 +08:00
|
|
|
void *p = calloc(1, new_size);
|
2020-12-05 19:26:32 +08:00
|
|
|
if (p != NULL) {
|
2021-03-22 19:30:48 +08:00
|
|
|
size_t len = new_size < io->len ? new_size : io->len;
|
|
|
|
if (len > 0) memcpy(p, io->buf, len);
|
2021-04-29 15:49:23 +08:00
|
|
|
zeromem(io->buf, io->size);
|
2020-12-05 19:26:32 +08:00
|
|
|
free(io->buf);
|
|
|
|
io->buf = (unsigned char *) p;
|
|
|
|
io->size = new_size;
|
|
|
|
} else {
|
2021-01-29 20:32:34 +08:00
|
|
|
ok = 0;
|
2020-12-05 19:26:32 +08:00
|
|
|
LOG(LL_ERROR,
|
|
|
|
("%lu->%lu", (unsigned long) io->size, (unsigned long) new_size));
|
|
|
|
}
|
|
|
|
}
|
2021-01-29 20:32:34 +08:00
|
|
|
return ok;
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
|
2021-01-29 20:32:34 +08:00
|
|
|
int mg_iobuf_init(struct mg_iobuf *io, size_t size) {
|
2021-02-15 22:42:55 +08:00
|
|
|
return mg_iobuf_resize(io, size);
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
|
2021-08-28 15:08:54 +08:00
|
|
|
size_t mg_iobuf_add(struct mg_iobuf *io, size_t ofs, const void *buf,
|
|
|
|
size_t len, size_t chunk_size) {
|
2021-05-29 16:44:07 +08:00
|
|
|
size_t new_size = io->len + len;
|
|
|
|
if (new_size > io->size) {
|
2021-05-29 16:46:50 +08:00
|
|
|
new_size += chunk_size; // Make sure that io->size
|
|
|
|
new_size -= new_size % chunk_size; // is aligned by chunk_size boundary
|
|
|
|
mg_iobuf_resize(io, new_size); // Attempt to realloc
|
2021-05-29 16:44:07 +08:00
|
|
|
if (new_size != io->size) len = 0; // Realloc failure, append nothing
|
|
|
|
}
|
2021-08-28 15:08:54 +08:00
|
|
|
if (ofs < io->len) memmove(io->buf + ofs + len, io->buf + ofs, io->len - ofs);
|
|
|
|
if (buf != NULL) memmove(io->buf + ofs, buf, len);
|
|
|
|
if (ofs > io->len) io->len += ofs - io->len;
|
2020-12-05 19:26:32 +08:00
|
|
|
io->len += len;
|
|
|
|
return len;
|
|
|
|
}
|
|
|
|
|
2021-08-27 17:25:24 +08:00
|
|
|
size_t mg_iobuf_del(struct mg_iobuf *io, size_t ofs, size_t len) {
|
|
|
|
if (ofs > io->len) ofs = io->len;
|
|
|
|
if (ofs + len > io->len) len = io->len - ofs;
|
|
|
|
memmove(io->buf + ofs, io->buf + ofs + len, io->len - ofs - len);
|
2021-08-30 18:39:19 +08:00
|
|
|
zeromem(io->buf + io->len - len, len);
|
2020-12-05 19:26:32 +08:00
|
|
|
io->len -= len;
|
|
|
|
return len;
|
|
|
|
}
|
|
|
|
|
|
|
|
void mg_iobuf_free(struct mg_iobuf *io) {
|
2021-02-15 22:42:55 +08:00
|
|
|
mg_iobuf_resize(io, 0);
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
#ifdef MG_ENABLE_LINES
|
|
|
|
#line 1 "src/log.c"
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#if MG_ENABLE_LOG
|
2021-05-29 01:30:42 +08:00
|
|
|
static void mg_log_stdout(const void *buf, size_t len, void *userdata) {
|
2020-12-05 19:26:32 +08:00
|
|
|
(void) userdata;
|
|
|
|
fwrite(buf, 1, len, stdout);
|
|
|
|
}
|
|
|
|
|
|
|
|
static const char *s_spec = "2";
|
2021-05-29 01:30:42 +08:00
|
|
|
static void (*s_fn)(const void *, size_t, void *) = mg_log_stdout;
|
2020-12-05 19:26:32 +08:00
|
|
|
static void *s_fn_param = NULL;
|
|
|
|
|
|
|
|
void mg_log_set(const char *spec) {
|
2021-01-20 19:10:02 +08:00
|
|
|
LOG(LL_DEBUG, ("Setting log level to %s", spec));
|
2020-12-05 19:26:32 +08:00
|
|
|
s_spec = spec;
|
2015-12-16 08:52:37 +08:00
|
|
|
}
|
2017-09-13 00:08:43 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
bool mg_log_prefix(int level, const char *file, int line, const char *fname) {
|
|
|
|
// static unsigned long seq;
|
|
|
|
int max = LL_INFO;
|
|
|
|
struct mg_str k, v, s = mg_str(s_spec);
|
|
|
|
const char *p = strrchr(file, '/');
|
2021-05-29 01:30:42 +08:00
|
|
|
|
|
|
|
if (s_fn == NULL) return false;
|
|
|
|
|
2021-04-14 04:02:05 +08:00
|
|
|
if (p == NULL) p = strrchr(file, '\\');
|
2020-12-05 19:26:32 +08:00
|
|
|
p = p == NULL ? file : p + 1;
|
2017-09-13 00:08:43 +08:00
|
|
|
|
2021-08-28 13:54:56 +08:00
|
|
|
while (mg_commalist(&s, &k, &v)) {
|
2020-12-05 19:26:32 +08:00
|
|
|
if (v.len == 0) max = atoi(k.ptr);
|
|
|
|
if (v.len > 0 && strncmp(p, k.ptr, k.len) == 0) max = atoi(v.ptr);
|
|
|
|
}
|
2017-09-13 00:08:43 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
if (level <= max) {
|
|
|
|
char timebuf[21], buf[50] = "";
|
|
|
|
time_t t = time(NULL);
|
2021-04-29 16:50:33 +08:00
|
|
|
struct tm tmp, *tm = gmtime_r(&t, &tmp);
|
2021-09-13 20:48:02 +08:00
|
|
|
int n;
|
|
|
|
(void) tmp;
|
2020-12-05 19:26:32 +08:00
|
|
|
strftime(timebuf, sizeof(timebuf), "%Y-%m-%d %H:%M:%S", tm);
|
2021-09-13 20:48:02 +08:00
|
|
|
n = snprintf(buf, sizeof(buf), "%s %d %s:%d:%s", timebuf, level, p, line,
|
2020-12-05 19:26:32 +08:00
|
|
|
fname);
|
|
|
|
if (n < 0 || n > (int) sizeof(buf) - 2) n = sizeof(buf) - 2;
|
|
|
|
while (n < (int) sizeof(buf) - 1) buf[n++] = ' ';
|
|
|
|
s_fn(buf, sizeof(buf) - 1, s_fn_param);
|
|
|
|
return true;
|
|
|
|
} else {
|
|
|
|
return false;
|
2017-09-13 00:08:43 +08:00
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
void mg_log(const char *fmt, ...) {
|
|
|
|
char mem[256], *buf = mem;
|
|
|
|
va_list ap;
|
|
|
|
int len = 0;
|
|
|
|
va_start(ap, fmt);
|
|
|
|
len = mg_vasprintf(&buf, sizeof(mem), fmt, ap);
|
|
|
|
va_end(ap);
|
2021-05-29 01:30:42 +08:00
|
|
|
s_fn(buf, len > 0 ? (size_t) len : 0, s_fn_param);
|
2020-12-05 19:26:32 +08:00
|
|
|
s_fn("\n", 1, s_fn_param);
|
|
|
|
if (buf != mem) free(buf);
|
|
|
|
}
|
2017-09-13 00:08:43 +08:00
|
|
|
|
2021-05-29 01:30:42 +08:00
|
|
|
void mg_log_set_callback(void (*fn)(const void *, size_t, void *), void *fnd) {
|
2020-12-05 19:26:32 +08:00
|
|
|
s_fn = fn;
|
2021-05-29 01:30:42 +08:00
|
|
|
s_fn_param = fnd;
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
#endif
|
2016-08-24 20:46:30 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
#ifdef MG_ENABLE_LINES
|
|
|
|
#line 1 "src/md5.c"
|
|
|
|
#endif
|
|
|
|
#include <string.h>
|
|
|
|
|
2020-12-24 05:20:48 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
#if MG_ENABLE_MD5
|
2016-08-24 20:46:30 +08:00
|
|
|
#if !defined(BYTE_ORDER) && defined(__BYTE_ORDER)
|
|
|
|
#define BYTE_ORDER __BYTE_ORDER
|
|
|
|
#ifndef LITTLE_ENDIAN
|
|
|
|
#define LITTLE_ENDIAN __LITTLE_ENDIAN
|
|
|
|
#endif /* LITTLE_ENDIAN */
|
|
|
|
#ifndef BIG_ENDIAN
|
|
|
|
#define BIG_ENDIAN __LITTLE_ENDIAN
|
|
|
|
#endif /* BIG_ENDIAN */
|
|
|
|
#endif /* BYTE_ORDER */
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
static void mg_byte_reverse(unsigned char *buf, unsigned longs) {
|
2015-11-01 00:26:19 +08:00
|
|
|
/* Forrest: MD5 expect LITTLE_ENDIAN, swap if BIG_ENDIAN */
|
|
|
|
#if BYTE_ORDER == BIG_ENDIAN
|
|
|
|
do {
|
|
|
|
uint32_t t = (uint32_t)((unsigned) buf[3] << 8 | buf[2]) << 16 |
|
|
|
|
((unsigned) buf[1] << 8 | buf[0]);
|
|
|
|
*(uint32_t *) buf = t;
|
|
|
|
buf += 4;
|
|
|
|
} while (--longs);
|
|
|
|
#else
|
|
|
|
(void) buf;
|
|
|
|
(void) longs;
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
#define F1(x, y, z) (z ^ (x & (y ^ z)))
|
|
|
|
#define F2(x, y, z) F1(z, x, y)
|
|
|
|
#define F3(x, y, z) (x ^ y ^ z)
|
|
|
|
#define F4(x, y, z) (y ^ (x | ~z))
|
|
|
|
|
|
|
|
#define MD5STEP(f, w, x, y, z, data, s) \
|
|
|
|
(w += f(x, y, z) + data, w = w << s | w >> (32 - s), w += x)
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
/*
|
2015-11-01 00:26:19 +08:00
|
|
|
* Start MD5 accumulation. Set bit count to 0 and buffer to mysterious
|
|
|
|
* initialization constants.
|
2015-09-08 19:49:03 +08:00
|
|
|
*/
|
2020-12-05 19:26:32 +08:00
|
|
|
void mg_md5_init(mg_md5_ctx *ctx) {
|
2015-11-01 00:26:19 +08:00
|
|
|
ctx->buf[0] = 0x67452301;
|
|
|
|
ctx->buf[1] = 0xefcdab89;
|
|
|
|
ctx->buf[2] = 0x98badcfe;
|
|
|
|
ctx->buf[3] = 0x10325476;
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
ctx->bits[0] = 0;
|
|
|
|
ctx->bits[1] = 0;
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
static void mg_md5_transform(uint32_t buf[4], uint32_t const in[16]) {
|
2020-12-24 05:20:48 +08:00
|
|
|
uint32_t a, b, c, d;
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
a = buf[0];
|
|
|
|
b = buf[1];
|
|
|
|
c = buf[2];
|
|
|
|
d = buf[3];
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
MD5STEP(F1, a, b, c, d, in[0] + 0xd76aa478, 7);
|
|
|
|
MD5STEP(F1, d, a, b, c, in[1] + 0xe8c7b756, 12);
|
|
|
|
MD5STEP(F1, c, d, a, b, in[2] + 0x242070db, 17);
|
|
|
|
MD5STEP(F1, b, c, d, a, in[3] + 0xc1bdceee, 22);
|
|
|
|
MD5STEP(F1, a, b, c, d, in[4] + 0xf57c0faf, 7);
|
|
|
|
MD5STEP(F1, d, a, b, c, in[5] + 0x4787c62a, 12);
|
|
|
|
MD5STEP(F1, c, d, a, b, in[6] + 0xa8304613, 17);
|
|
|
|
MD5STEP(F1, b, c, d, a, in[7] + 0xfd469501, 22);
|
|
|
|
MD5STEP(F1, a, b, c, d, in[8] + 0x698098d8, 7);
|
|
|
|
MD5STEP(F1, d, a, b, c, in[9] + 0x8b44f7af, 12);
|
|
|
|
MD5STEP(F1, c, d, a, b, in[10] + 0xffff5bb1, 17);
|
|
|
|
MD5STEP(F1, b, c, d, a, in[11] + 0x895cd7be, 22);
|
|
|
|
MD5STEP(F1, a, b, c, d, in[12] + 0x6b901122, 7);
|
|
|
|
MD5STEP(F1, d, a, b, c, in[13] + 0xfd987193, 12);
|
|
|
|
MD5STEP(F1, c, d, a, b, in[14] + 0xa679438e, 17);
|
|
|
|
MD5STEP(F1, b, c, d, a, in[15] + 0x49b40821, 22);
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
MD5STEP(F2, a, b, c, d, in[1] + 0xf61e2562, 5);
|
|
|
|
MD5STEP(F2, d, a, b, c, in[6] + 0xc040b340, 9);
|
|
|
|
MD5STEP(F2, c, d, a, b, in[11] + 0x265e5a51, 14);
|
|
|
|
MD5STEP(F2, b, c, d, a, in[0] + 0xe9b6c7aa, 20);
|
|
|
|
MD5STEP(F2, a, b, c, d, in[5] + 0xd62f105d, 5);
|
|
|
|
MD5STEP(F2, d, a, b, c, in[10] + 0x02441453, 9);
|
|
|
|
MD5STEP(F2, c, d, a, b, in[15] + 0xd8a1e681, 14);
|
|
|
|
MD5STEP(F2, b, c, d, a, in[4] + 0xe7d3fbc8, 20);
|
|
|
|
MD5STEP(F2, a, b, c, d, in[9] + 0x21e1cde6, 5);
|
|
|
|
MD5STEP(F2, d, a, b, c, in[14] + 0xc33707d6, 9);
|
|
|
|
MD5STEP(F2, c, d, a, b, in[3] + 0xf4d50d87, 14);
|
|
|
|
MD5STEP(F2, b, c, d, a, in[8] + 0x455a14ed, 20);
|
|
|
|
MD5STEP(F2, a, b, c, d, in[13] + 0xa9e3e905, 5);
|
|
|
|
MD5STEP(F2, d, a, b, c, in[2] + 0xfcefa3f8, 9);
|
|
|
|
MD5STEP(F2, c, d, a, b, in[7] + 0x676f02d9, 14);
|
|
|
|
MD5STEP(F2, b, c, d, a, in[12] + 0x8d2a4c8a, 20);
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
MD5STEP(F3, a, b, c, d, in[5] + 0xfffa3942, 4);
|
|
|
|
MD5STEP(F3, d, a, b, c, in[8] + 0x8771f681, 11);
|
|
|
|
MD5STEP(F3, c, d, a, b, in[11] + 0x6d9d6122, 16);
|
|
|
|
MD5STEP(F3, b, c, d, a, in[14] + 0xfde5380c, 23);
|
|
|
|
MD5STEP(F3, a, b, c, d, in[1] + 0xa4beea44, 4);
|
|
|
|
MD5STEP(F3, d, a, b, c, in[4] + 0x4bdecfa9, 11);
|
|
|
|
MD5STEP(F3, c, d, a, b, in[7] + 0xf6bb4b60, 16);
|
|
|
|
MD5STEP(F3, b, c, d, a, in[10] + 0xbebfbc70, 23);
|
|
|
|
MD5STEP(F3, a, b, c, d, in[13] + 0x289b7ec6, 4);
|
|
|
|
MD5STEP(F3, d, a, b, c, in[0] + 0xeaa127fa, 11);
|
|
|
|
MD5STEP(F3, c, d, a, b, in[3] + 0xd4ef3085, 16);
|
|
|
|
MD5STEP(F3, b, c, d, a, in[6] + 0x04881d05, 23);
|
|
|
|
MD5STEP(F3, a, b, c, d, in[9] + 0xd9d4d039, 4);
|
|
|
|
MD5STEP(F3, d, a, b, c, in[12] + 0xe6db99e5, 11);
|
|
|
|
MD5STEP(F3, c, d, a, b, in[15] + 0x1fa27cf8, 16);
|
|
|
|
MD5STEP(F3, b, c, d, a, in[2] + 0xc4ac5665, 23);
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
MD5STEP(F4, a, b, c, d, in[0] + 0xf4292244, 6);
|
|
|
|
MD5STEP(F4, d, a, b, c, in[7] + 0x432aff97, 10);
|
|
|
|
MD5STEP(F4, c, d, a, b, in[14] + 0xab9423a7, 15);
|
|
|
|
MD5STEP(F4, b, c, d, a, in[5] + 0xfc93a039, 21);
|
|
|
|
MD5STEP(F4, a, b, c, d, in[12] + 0x655b59c3, 6);
|
|
|
|
MD5STEP(F4, d, a, b, c, in[3] + 0x8f0ccc92, 10);
|
|
|
|
MD5STEP(F4, c, d, a, b, in[10] + 0xffeff47d, 15);
|
|
|
|
MD5STEP(F4, b, c, d, a, in[1] + 0x85845dd1, 21);
|
|
|
|
MD5STEP(F4, a, b, c, d, in[8] + 0x6fa87e4f, 6);
|
|
|
|
MD5STEP(F4, d, a, b, c, in[15] + 0xfe2ce6e0, 10);
|
|
|
|
MD5STEP(F4, c, d, a, b, in[6] + 0xa3014314, 15);
|
|
|
|
MD5STEP(F4, b, c, d, a, in[13] + 0x4e0811a1, 21);
|
|
|
|
MD5STEP(F4, a, b, c, d, in[4] + 0xf7537e82, 6);
|
|
|
|
MD5STEP(F4, d, a, b, c, in[11] + 0xbd3af235, 10);
|
|
|
|
MD5STEP(F4, c, d, a, b, in[2] + 0x2ad7d2bb, 15);
|
|
|
|
MD5STEP(F4, b, c, d, a, in[9] + 0xeb86d391, 21);
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
buf[0] += a;
|
|
|
|
buf[1] += b;
|
|
|
|
buf[2] += c;
|
|
|
|
buf[3] += d;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
void mg_md5_update(mg_md5_ctx *ctx, const unsigned char *buf, size_t len) {
|
2015-11-01 00:26:19 +08:00
|
|
|
uint32_t t;
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
t = ctx->bits[0];
|
|
|
|
if ((ctx->bits[0] = t + ((uint32_t) len << 3)) < t) ctx->bits[1]++;
|
|
|
|
ctx->bits[1] += (uint32_t) len >> 29;
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
t = (t >> 3) & 0x3f;
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
if (t) {
|
|
|
|
unsigned char *p = (unsigned char *) ctx->in + t;
|
|
|
|
|
|
|
|
t = 64 - t;
|
|
|
|
if (len < t) {
|
|
|
|
memcpy(p, buf, len);
|
|
|
|
return;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-11-01 00:26:19 +08:00
|
|
|
memcpy(p, buf, t);
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_byte_reverse(ctx->in, 16);
|
|
|
|
mg_md5_transform(ctx->buf, (uint32_t *) ctx->in);
|
2015-11-01 00:26:19 +08:00
|
|
|
buf += t;
|
|
|
|
len -= t;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
while (len >= 64) {
|
|
|
|
memcpy(ctx->in, buf, 64);
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_byte_reverse(ctx->in, 16);
|
|
|
|
mg_md5_transform(ctx->buf, (uint32_t *) ctx->in);
|
2015-11-01 00:26:19 +08:00
|
|
|
buf += 64;
|
|
|
|
len -= 64;
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
memcpy(ctx->in, buf, len);
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
void mg_md5_final(mg_md5_ctx *ctx, unsigned char digest[16]) {
|
2015-11-01 00:26:19 +08:00
|
|
|
unsigned count;
|
|
|
|
unsigned char *p;
|
|
|
|
uint32_t *a;
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
count = (ctx->bits[0] >> 3) & 0x3F;
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
p = ctx->in + count;
|
|
|
|
*p++ = 0x80;
|
|
|
|
count = 64 - 1 - count;
|
|
|
|
if (count < 8) {
|
|
|
|
memset(p, 0, count);
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_byte_reverse(ctx->in, 16);
|
|
|
|
mg_md5_transform(ctx->buf, (uint32_t *) ctx->in);
|
2015-11-01 00:26:19 +08:00
|
|
|
memset(ctx->in, 0, 56);
|
|
|
|
} else {
|
|
|
|
memset(p, 0, count - 8);
|
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_byte_reverse(ctx->in, 14);
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
a = (uint32_t *) ctx->in;
|
|
|
|
a[14] = ctx->bits[0];
|
|
|
|
a[15] = ctx->bits[1];
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_md5_transform(ctx->buf, (uint32_t *) ctx->in);
|
|
|
|
mg_byte_reverse((unsigned char *) ctx->buf, 4);
|
2015-11-01 00:26:19 +08:00
|
|
|
memcpy(digest, ctx->buf, 16);
|
|
|
|
memset((char *) ctx, 0, sizeof(*ctx));
|
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
#ifdef MG_ENABLE_LINES
|
|
|
|
#line 1 "src/mqtt.c"
|
2015-11-01 00:26:19 +08:00
|
|
|
#endif
|
|
|
|
|
2016-10-14 02:57:02 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
|
|
|
|
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#define MQTT_CLEAN_SESSION 0x02
|
|
|
|
#define MQTT_HAS_WILL 0x04
|
|
|
|
#define MQTT_WILL_RETAIN 0x20
|
|
|
|
#define MQTT_HAS_PASSWORD 0x40
|
|
|
|
#define MQTT_HAS_USER_NAME 0x80
|
|
|
|
#define MQTT_GET_WILL_QOS(flags) (((flags) &0x18) >> 3)
|
|
|
|
#define MQTT_SET_WILL_QOS(flags, qos) (flags) = ((flags) & ~0x18) | ((qos) << 3)
|
|
|
|
|
|
|
|
enum { MQTT_OK, MQTT_INCOMPLETE, MQTT_MALFORMED };
|
|
|
|
|
2020-12-16 04:29:47 +08:00
|
|
|
void mg_mqtt_send_header(struct mg_connection *c, uint8_t cmd, uint8_t flags,
|
|
|
|
uint32_t len) {
|
2020-12-07 11:47:37 +08:00
|
|
|
uint8_t buf[1 + sizeof(len)], *vlen = &buf[1];
|
2021-08-28 14:16:38 +08:00
|
|
|
buf[0] = (uint8_t) ((cmd << 4) | flags);
|
2020-12-05 19:26:32 +08:00
|
|
|
do {
|
|
|
|
*vlen = len % 0x80;
|
|
|
|
len /= 0x80;
|
|
|
|
if (len > 0) *vlen |= 0x80;
|
|
|
|
vlen++;
|
|
|
|
} while (len > 0 && vlen < &buf[sizeof(buf)]);
|
2021-08-28 14:16:38 +08:00
|
|
|
mg_send(c, buf, (size_t) (vlen - buf));
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
static void mg_send_u16(struct mg_connection *c, uint16_t value) {
|
|
|
|
mg_send(c, &value, sizeof(value));
|
|
|
|
}
|
|
|
|
|
2021-08-01 21:10:21 +08:00
|
|
|
void mg_mqtt_login(struct mg_connection *c, const char *url,
|
|
|
|
struct mg_mqtt_opts *opts) {
|
2020-12-05 19:26:32 +08:00
|
|
|
uint32_t total_len = 7 + 1 + 2 + 2;
|
2021-08-28 14:16:38 +08:00
|
|
|
uint16_t flags = (uint16_t) (((uint16_t) opts->qos & 3) << 3);
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_str user = mg_url_user(url);
|
|
|
|
struct mg_str pass = mg_url_pass(url);
|
|
|
|
|
|
|
|
if (user.len > 0) {
|
|
|
|
total_len += 2 + (uint32_t) user.len;
|
|
|
|
flags |= MQTT_HAS_USER_NAME;
|
|
|
|
}
|
|
|
|
if (pass.len > 0) {
|
|
|
|
total_len += 2 + (uint32_t) pass.len;
|
|
|
|
flags |= MQTT_HAS_PASSWORD;
|
|
|
|
}
|
|
|
|
if (opts->will_topic.len > 0 && opts->will_message.len > 0) {
|
|
|
|
total_len +=
|
|
|
|
4 + (uint32_t) opts->will_topic.len + (uint32_t) opts->will_message.len;
|
|
|
|
flags |= MQTT_HAS_WILL;
|
|
|
|
}
|
|
|
|
if (opts->clean || opts->client_id.len == 0) flags |= MQTT_CLEAN_SESSION;
|
|
|
|
if (opts->will_retain) flags |= MQTT_WILL_RETAIN;
|
|
|
|
total_len += (uint32_t) opts->client_id.len;
|
|
|
|
|
2020-12-16 04:29:47 +08:00
|
|
|
mg_mqtt_send_header(c, MQTT_CMD_CONNECT, 0, total_len);
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_send(c, "\00\04MQTT\04", 7);
|
|
|
|
mg_send(c, &flags, 1);
|
|
|
|
// keepalive == 0 means "do not disconnect us!"
|
|
|
|
mg_send_u16(c, mg_htons((uint16_t) opts->keepalive));
|
|
|
|
mg_send_u16(c, mg_htons((uint16_t) opts->client_id.len));
|
|
|
|
mg_send(c, opts->client_id.ptr, opts->client_id.len);
|
|
|
|
if (flags & MQTT_HAS_WILL) {
|
|
|
|
mg_send_u16(c, mg_htons((uint16_t) opts->will_topic.len));
|
|
|
|
mg_send(c, opts->will_topic.ptr, opts->will_topic.len);
|
|
|
|
mg_send_u16(c, mg_htons((uint16_t) opts->will_message.len));
|
|
|
|
mg_send(c, opts->will_message.ptr, opts->will_message.len);
|
|
|
|
}
|
|
|
|
if (user.len > 0) {
|
|
|
|
mg_send_u16(c, mg_htons((uint16_t) user.len));
|
|
|
|
mg_send(c, user.ptr, user.len);
|
|
|
|
}
|
|
|
|
if (pass.len > 0) {
|
|
|
|
mg_send_u16(c, mg_htons((uint16_t) pass.len));
|
|
|
|
mg_send(c, pass.ptr, pass.len);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-06-11 02:27:19 +08:00
|
|
|
void mg_mqtt_pub(struct mg_connection *c, struct mg_str *topic,
|
|
|
|
struct mg_str *data, int qos, bool retain) {
|
2021-08-28 14:16:38 +08:00
|
|
|
uint8_t flags = (uint8_t) (((qos & 3) << 1) | (retain ? 1 : 0));
|
2020-12-05 19:26:32 +08:00
|
|
|
uint32_t total_len = 2 + (uint32_t) topic->len + (uint32_t) data->len;
|
2020-12-21 20:26:44 +08:00
|
|
|
LOG(LL_DEBUG, ("%lu [%.*s] -> [%.*s]", c->id, (int) topic->len,
|
2020-12-05 19:26:32 +08:00
|
|
|
(char *) topic->ptr, (int) data->len, (char *) data->ptr));
|
|
|
|
if (MQTT_GET_QOS(flags) > 0) total_len += 2;
|
2020-12-16 04:29:47 +08:00
|
|
|
mg_mqtt_send_header(c, MQTT_CMD_PUBLISH, flags, total_len);
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_send_u16(c, mg_htons((uint16_t) topic->len));
|
|
|
|
mg_send(c, topic->ptr, topic->len);
|
|
|
|
if (MQTT_GET_QOS(flags) > 0) {
|
|
|
|
static uint16_t s_id;
|
|
|
|
if (++s_id == 0) s_id++;
|
|
|
|
mg_send_u16(c, mg_htons(s_id));
|
|
|
|
}
|
|
|
|
mg_send(c, data->ptr, data->len);
|
|
|
|
}
|
|
|
|
|
2021-06-11 02:27:19 +08:00
|
|
|
void mg_mqtt_sub(struct mg_connection *c, struct mg_str *topic, int qos) {
|
2020-12-05 19:26:32 +08:00
|
|
|
static uint16_t s_id;
|
2021-06-11 02:27:19 +08:00
|
|
|
uint8_t qos_ = qos & 3;
|
2020-12-05 19:26:32 +08:00
|
|
|
uint32_t total_len = 2 + (uint32_t) topic->len + 2 + 1;
|
2021-06-11 02:27:19 +08:00
|
|
|
mg_mqtt_send_header(c, MQTT_CMD_SUBSCRIBE, (uint8_t) MQTT_QOS(qos_),
|
2020-12-16 04:29:47 +08:00
|
|
|
total_len);
|
2020-12-05 19:26:32 +08:00
|
|
|
if (++s_id == 0) ++s_id;
|
|
|
|
mg_send_u16(c, mg_htons(s_id));
|
|
|
|
mg_send_u16(c, mg_htons((uint16_t) topic->len));
|
|
|
|
mg_send(c, topic->ptr, topic->len);
|
2021-06-11 02:27:19 +08:00
|
|
|
mg_send(c, &qos_, sizeof(qos_));
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
|
2020-12-16 18:14:00 +08:00
|
|
|
int mg_mqtt_parse(const uint8_t *buf, size_t len, struct mg_mqtt_message *m) {
|
2020-12-05 19:26:32 +08:00
|
|
|
uint8_t lc = 0, *p, *end;
|
2020-12-16 18:14:00 +08:00
|
|
|
uint32_t n = 0, len_len = 0;
|
2020-12-05 19:26:32 +08:00
|
|
|
|
2020-12-16 04:29:47 +08:00
|
|
|
memset(m, 0, sizeof(*m));
|
2020-12-16 18:14:00 +08:00
|
|
|
m->dgram.ptr = (char *) buf;
|
|
|
|
if (len < 2) return MQTT_INCOMPLETE;
|
2021-08-28 14:16:38 +08:00
|
|
|
m->cmd = (uint8_t) (buf[0] >> 4);
|
2020-12-16 18:14:00 +08:00
|
|
|
m->qos = (buf[0] >> 1) & 3;
|
|
|
|
|
|
|
|
n = len_len = 0;
|
|
|
|
p = (uint8_t *) buf + 1;
|
2021-08-28 14:16:38 +08:00
|
|
|
while ((size_t) (p - buf) < len) {
|
2020-12-05 19:26:32 +08:00
|
|
|
lc = *((uint8_t *) p++);
|
2021-08-28 14:16:38 +08:00
|
|
|
n += (uint32_t) ((lc & 0x7f) << 7 * len_len);
|
2020-12-05 19:26:32 +08:00
|
|
|
len_len++;
|
|
|
|
if (!(lc & 0x80)) break;
|
2020-12-11 17:35:50 +08:00
|
|
|
if (len_len >= 4) return MQTT_MALFORMED;
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
2020-12-16 18:14:00 +08:00
|
|
|
end = p + n;
|
|
|
|
if (lc & 0x80 || end > buf + len) return MQTT_INCOMPLETE;
|
2021-08-28 14:16:38 +08:00
|
|
|
m->dgram.len = (size_t) (end - buf);
|
2020-12-05 19:26:32 +08:00
|
|
|
|
|
|
|
switch (m->cmd) {
|
|
|
|
case MQTT_CMD_CONNACK:
|
|
|
|
if (end - p < 2) return MQTT_MALFORMED;
|
2020-12-16 18:14:00 +08:00
|
|
|
m->ack = p[1];
|
2020-12-05 19:26:32 +08:00
|
|
|
break;
|
|
|
|
case MQTT_CMD_PUBACK:
|
|
|
|
case MQTT_CMD_PUBREC:
|
|
|
|
case MQTT_CMD_PUBREL:
|
|
|
|
case MQTT_CMD_PUBCOMP:
|
|
|
|
case MQTT_CMD_SUBACK:
|
|
|
|
if (p + 2 > end) return MQTT_MALFORMED;
|
2021-08-28 14:16:38 +08:00
|
|
|
m->id = (uint16_t) ((((uint16_t) p[0]) << 8) | p[1]);
|
2020-12-05 19:26:32 +08:00
|
|
|
break;
|
2020-12-16 04:29:47 +08:00
|
|
|
case MQTT_CMD_SUBSCRIBE: {
|
|
|
|
if (p + 2 > end) return MQTT_MALFORMED;
|
2021-08-28 14:16:38 +08:00
|
|
|
m->id = (uint16_t) ((((uint16_t) p[0]) << 8) | p[1]);
|
2020-12-16 04:29:47 +08:00
|
|
|
p += 2;
|
|
|
|
break;
|
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
case MQTT_CMD_PUBLISH: {
|
|
|
|
if (p + 2 > end) return MQTT_MALFORMED;
|
2021-08-28 14:16:38 +08:00
|
|
|
m->topic.len = (uint16_t) ((((uint16_t) p[0]) << 8) | p[1]);
|
2020-12-16 18:14:00 +08:00
|
|
|
m->topic.ptr = (char *) p + 2;
|
|
|
|
p += 2 + m->topic.len;
|
2020-12-05 19:26:32 +08:00
|
|
|
if (p > end) return MQTT_MALFORMED;
|
|
|
|
if (m->qos > 0) {
|
|
|
|
if (p + 2 > end) return MQTT_MALFORMED;
|
2021-08-28 14:16:38 +08:00
|
|
|
m->id = (uint16_t) ((((uint16_t) p[0]) << 8) | p[1]);
|
2020-12-05 19:26:32 +08:00
|
|
|
p += 2;
|
|
|
|
}
|
|
|
|
if (p > end) return MQTT_MALFORMED;
|
2020-12-16 18:14:00 +08:00
|
|
|
m->data.ptr = (char *) p;
|
2021-08-28 14:16:38 +08:00
|
|
|
m->data.len = (size_t) (end - p);
|
2020-12-16 04:29:47 +08:00
|
|
|
break;
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
default:
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
return MQTT_OK;
|
|
|
|
}
|
|
|
|
|
2021-05-29 01:30:42 +08:00
|
|
|
static size_t mg_mqtt_next_topic(struct mg_mqtt_message *msg,
|
|
|
|
struct mg_str *topic, uint8_t *qos,
|
|
|
|
size_t pos) {
|
2020-12-16 04:29:47 +08:00
|
|
|
unsigned char *buf = (unsigned char *) msg->dgram.ptr + pos;
|
2021-05-29 01:30:42 +08:00
|
|
|
size_t new_pos;
|
|
|
|
if (pos >= msg->dgram.len) return 0;
|
2020-12-16 04:29:47 +08:00
|
|
|
|
2021-08-28 14:16:38 +08:00
|
|
|
topic->len = (size_t) (((unsigned) buf[0]) << 8 | buf[1]);
|
2020-12-16 04:29:47 +08:00
|
|
|
topic->ptr = (char *) buf + 2;
|
2021-03-09 17:54:02 +08:00
|
|
|
new_pos = pos + 2 + topic->len + (qos == NULL ? 0 : 1);
|
2021-05-29 01:30:42 +08:00
|
|
|
if ((size_t) new_pos > msg->dgram.len) return 0;
|
2021-03-09 17:54:02 +08:00
|
|
|
if (qos != NULL) *qos = buf[2 + topic->len];
|
2020-12-16 04:29:47 +08:00
|
|
|
return new_pos;
|
|
|
|
}
|
|
|
|
|
2021-05-29 01:30:42 +08:00
|
|
|
size_t mg_mqtt_next_sub(struct mg_mqtt_message *msg, struct mg_str *topic,
|
|
|
|
uint8_t *qos, size_t pos) {
|
2021-03-09 17:54:02 +08:00
|
|
|
uint8_t tmp;
|
|
|
|
return mg_mqtt_next_topic(msg, topic, qos == NULL ? &tmp : qos, pos);
|
|
|
|
}
|
|
|
|
|
2021-05-29 01:30:42 +08:00
|
|
|
size_t mg_mqtt_next_unsub(struct mg_mqtt_message *msg, struct mg_str *topic,
|
|
|
|
size_t pos) {
|
2021-03-09 17:54:02 +08:00
|
|
|
return mg_mqtt_next_topic(msg, topic, NULL, pos);
|
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
static void mqtt_cb(struct mg_connection *c, int ev, void *ev_data,
|
|
|
|
void *fn_data) {
|
|
|
|
if (ev == MG_EV_READ) {
|
|
|
|
for (;;) {
|
2020-12-16 04:29:47 +08:00
|
|
|
struct mg_mqtt_message mm;
|
|
|
|
int rc = mg_mqtt_parse(c->recv.buf, c->recv.len, &mm);
|
2020-12-05 19:26:32 +08:00
|
|
|
if (rc == MQTT_MALFORMED) {
|
2020-12-21 20:26:44 +08:00
|
|
|
LOG(LL_ERROR, ("%lu MQTT malformed message", c->id));
|
2020-12-05 19:26:32 +08:00
|
|
|
c->is_closing = 1;
|
|
|
|
break;
|
|
|
|
} else if (rc == MQTT_OK) {
|
|
|
|
LOG(LL_VERBOSE_DEBUG,
|
2020-12-16 04:29:47 +08:00
|
|
|
("%p MQTT CMD %d len %d [%.*s]", c->fd, mm.cmd, (int) mm.dgram.len,
|
|
|
|
(int) mm.data.len, mm.data.ptr));
|
|
|
|
switch (mm.cmd) {
|
2020-12-05 19:26:32 +08:00
|
|
|
case MQTT_CMD_CONNACK:
|
2020-12-16 04:29:47 +08:00
|
|
|
mg_call(c, MG_EV_MQTT_OPEN, &mm.ack);
|
|
|
|
if (mm.ack == 0) {
|
2020-12-21 20:26:44 +08:00
|
|
|
LOG(LL_INFO, ("%lu Connected", c->id));
|
2020-12-05 19:26:32 +08:00
|
|
|
} else {
|
2020-12-21 20:26:44 +08:00
|
|
|
LOG(LL_ERROR, ("%lu MQTT auth failed, code %d", c->id, mm.ack));
|
2020-12-05 19:26:32 +08:00
|
|
|
c->is_closing = 1;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
case MQTT_CMD_PUBLISH: {
|
2020-12-21 20:26:44 +08:00
|
|
|
LOG(LL_DEBUG, ("%lu [%.*s] -> [%.*s]", c->id, (int) mm.topic.len,
|
2020-12-16 04:29:47 +08:00
|
|
|
mm.topic.ptr, (int) mm.data.len, mm.data.ptr));
|
|
|
|
mg_call(c, MG_EV_MQTT_MSG, &mm);
|
2020-12-05 19:26:32 +08:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
2020-12-16 04:29:47 +08:00
|
|
|
mg_call(c, MG_EV_MQTT_CMD, &mm);
|
2021-08-28 14:16:38 +08:00
|
|
|
mg_iobuf_del(&c->recv, 0, mm.dgram.len);
|
2020-12-05 19:26:32 +08:00
|
|
|
} else {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
(void) ev_data;
|
|
|
|
(void) fn_data;
|
|
|
|
}
|
|
|
|
|
2021-03-07 20:21:59 +08:00
|
|
|
void mg_mqtt_ping(struct mg_connection *nc) {
|
|
|
|
mg_mqtt_send_header(nc, MQTT_CMD_PINGREQ, 0, 0);
|
|
|
|
}
|
|
|
|
|
|
|
|
void mg_mqtt_pong(struct mg_connection *nc) {
|
|
|
|
mg_mqtt_send_header(nc, MQTT_CMD_PINGRESP, 0, 0);
|
|
|
|
}
|
|
|
|
|
|
|
|
void mg_mqtt_disconnect(struct mg_connection *nc) {
|
|
|
|
mg_mqtt_send_header(nc, MQTT_CMD_DISCONNECT, 0, 0);
|
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_connection *mg_mqtt_connect(struct mg_mgr *mgr, const char *url,
|
|
|
|
struct mg_mqtt_opts *opts,
|
|
|
|
mg_event_handler_t fn, void *fn_data) {
|
|
|
|
struct mg_connection *c = mg_connect(mgr, url, fn, fn_data);
|
|
|
|
if (c != NULL) {
|
|
|
|
struct mg_mqtt_opts empty;
|
|
|
|
memset(&empty, 0, sizeof(empty));
|
2021-08-01 21:10:21 +08:00
|
|
|
mg_mqtt_login(c, url, opts == NULL ? &empty : opts);
|
2020-12-05 19:26:32 +08:00
|
|
|
c->pfn = mqtt_cb;
|
|
|
|
}
|
|
|
|
return c;
|
|
|
|
}
|
|
|
|
|
2020-12-16 04:29:47 +08:00
|
|
|
struct mg_connection *mg_mqtt_listen(struct mg_mgr *mgr, const char *url,
|
|
|
|
mg_event_handler_t fn, void *fn_data) {
|
|
|
|
struct mg_connection *c = mg_listen(mgr, url, fn, fn_data);
|
|
|
|
if (c != NULL) c->pfn = mqtt_cb, c->pfn_data = mgr;
|
|
|
|
return c;
|
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
#ifdef MG_ENABLE_LINES
|
|
|
|
#line 1 "src/net.c"
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
int mg_vprintf(struct mg_connection *c, const char *fmt, va_list ap) {
|
|
|
|
char mem[256], *buf = mem;
|
|
|
|
int len = mg_vasprintf(&buf, sizeof(mem), fmt, ap);
|
2021-05-29 01:30:42 +08:00
|
|
|
len = mg_send(c, buf, len > 0 ? (size_t) len : 0);
|
2020-12-05 19:26:32 +08:00
|
|
|
if (buf != mem) free(buf);
|
|
|
|
return len;
|
|
|
|
}
|
|
|
|
|
|
|
|
int mg_printf(struct mg_connection *c, const char *fmt, ...) {
|
|
|
|
int len = 0;
|
|
|
|
va_list ap;
|
|
|
|
va_start(ap, fmt);
|
|
|
|
len = mg_vprintf(c, fmt, ap);
|
|
|
|
va_end(ap);
|
|
|
|
return len;
|
|
|
|
}
|
|
|
|
|
|
|
|
char *mg_straddr(struct mg_connection *c, char *buf, size_t len) {
|
2020-12-24 16:05:54 +08:00
|
|
|
char tmp[100];
|
|
|
|
const char *fmt = c->peer.is_ip6 ? "[%s]:%d" : "%s:%d";
|
|
|
|
mg_ntoa(&c->peer, tmp, sizeof(tmp));
|
|
|
|
snprintf(buf, len, fmt, tmp, (int) mg_ntohs(c->peer.port));
|
2020-12-05 19:26:32 +08:00
|
|
|
return buf;
|
|
|
|
}
|
|
|
|
|
2020-12-21 00:55:33 +08:00
|
|
|
char *mg_ntoa(const struct mg_addr *addr, char *buf, size_t len) {
|
2020-12-24 16:05:54 +08:00
|
|
|
if (addr->is_ip6) {
|
|
|
|
uint16_t *p = (uint16_t *) addr->ip6;
|
|
|
|
snprintf(buf, len, "%x:%x:%x:%x:%x:%x:%x:%x", mg_htons(p[0]),
|
|
|
|
mg_htons(p[1]), mg_htons(p[2]), mg_htons(p[3]), mg_htons(p[4]),
|
|
|
|
mg_htons(p[5]), mg_htons(p[6]), mg_htons(p[7]));
|
|
|
|
} else {
|
|
|
|
uint8_t p[4];
|
|
|
|
memcpy(p, &addr->ip, sizeof(p));
|
|
|
|
snprintf(buf, len, "%d.%d.%d.%d", (int) p[0], (int) p[1], (int) p[2],
|
|
|
|
(int) p[3]);
|
|
|
|
}
|
2020-12-21 00:55:33 +08:00
|
|
|
return buf;
|
|
|
|
}
|
|
|
|
|
2020-12-22 17:44:59 +08:00
|
|
|
static bool mg_atonl(struct mg_str str, struct mg_addr *addr) {
|
2020-12-23 18:15:09 +08:00
|
|
|
if (mg_vcasecmp(&str, "localhost") != 0) return false;
|
2020-12-22 17:44:59 +08:00
|
|
|
addr->ip = mg_htonl(0x7f000001);
|
|
|
|
addr->is_ip6 = false;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
static bool mg_aton4(struct mg_str str, struct mg_addr *addr) {
|
|
|
|
uint8_t data[4] = {0, 0, 0, 0};
|
|
|
|
size_t i, num_dots = 0;
|
|
|
|
for (i = 0; i < str.len; i++) {
|
|
|
|
if (str.ptr[i] >= '0' && str.ptr[i] <= '9') {
|
|
|
|
int octet = data[num_dots] * 10 + (str.ptr[i] - '0');
|
|
|
|
if (octet > 255) return false;
|
2021-05-29 01:30:42 +08:00
|
|
|
data[num_dots] = (uint8_t) octet;
|
2020-12-22 17:44:59 +08:00
|
|
|
} else if (str.ptr[i] == '.') {
|
|
|
|
if (num_dots >= 3 || i == 0 || str.ptr[i - 1] == '.') return false;
|
|
|
|
num_dots++;
|
|
|
|
} else {
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (num_dots != 3 || str.ptr[i - 1] == '.') return false;
|
|
|
|
memcpy(&addr->ip, data, sizeof(data));
|
|
|
|
addr->is_ip6 = false;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2021-09-19 02:38:59 +08:00
|
|
|
static bool mg_v4mapped(struct mg_str str, struct mg_addr *addr) {
|
|
|
|
int i;
|
|
|
|
if (str.len < 14) return false;
|
|
|
|
if (str.ptr[0] != ':' || str.ptr[1] != ':' || str.ptr[6] != ':') return false;
|
|
|
|
for (i = 2; i < 6; i++) {
|
|
|
|
if (str.ptr[i] != 'f' && str.ptr[i] != 'F') return false;
|
|
|
|
}
|
|
|
|
if (!mg_aton4(mg_str_n(&str.ptr[7], str.len - 7), addr)) return false;
|
|
|
|
memset(addr->ip6, 0, sizeof(addr->ip6));
|
|
|
|
addr->ip6[10] = addr->ip6[11] = 255;
|
|
|
|
memcpy(&addr->ip6[12], &addr->ip, 4);
|
|
|
|
addr->is_ip6 = true;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2020-12-22 17:44:59 +08:00
|
|
|
static bool mg_aton6(struct mg_str str, struct mg_addr *addr) {
|
|
|
|
size_t i, j = 0, n = 0, dc = 42;
|
2021-09-21 15:10:27 +08:00
|
|
|
if (str.len > 2 && str.ptr[0] == '[') str.ptr++, str.len -= 2;
|
2021-09-19 02:38:59 +08:00
|
|
|
if (mg_v4mapped(str, addr)) return true;
|
2020-12-22 17:44:59 +08:00
|
|
|
for (i = 0; i < str.len; i++) {
|
|
|
|
if ((str.ptr[i] >= '0' && str.ptr[i] <= '9') ||
|
|
|
|
(str.ptr[i] >= 'a' && str.ptr[i] <= 'f') ||
|
|
|
|
(str.ptr[i] >= 'A' && str.ptr[i] <= 'F')) {
|
|
|
|
unsigned long val;
|
|
|
|
if (i > j + 3) return false;
|
|
|
|
// LOG(LL_DEBUG, ("%zu %zu [%.*s]", i, j, (int) (i - j + 1),
|
|
|
|
// &str.ptr[j]));
|
|
|
|
val = mg_unhexn(&str.ptr[j], i - j + 1);
|
2021-09-19 02:38:59 +08:00
|
|
|
addr->ip6[n] = (uint8_t) ((val >> 8) & 255);
|
|
|
|
addr->ip6[n + 1] = (uint8_t) (val & 255);
|
2020-12-22 17:44:59 +08:00
|
|
|
} else if (str.ptr[i] == ':') {
|
|
|
|
j = i + 1;
|
|
|
|
if (i > 0 && str.ptr[i - 1] == ':') {
|
|
|
|
dc = n; // Double colon
|
|
|
|
if (i > 1 && str.ptr[i - 2] == ':') return false;
|
|
|
|
} else if (i > 0) {
|
|
|
|
n += 2;
|
2020-12-21 00:55:33 +08:00
|
|
|
}
|
2020-12-22 17:44:59 +08:00
|
|
|
if (n > 14) return false;
|
|
|
|
addr->ip6[n] = addr->ip6[n + 1] = 0; // For trailing ::
|
|
|
|
} else {
|
|
|
|
return false;
|
2020-12-21 00:55:33 +08:00
|
|
|
}
|
|
|
|
}
|
2020-12-22 17:44:59 +08:00
|
|
|
if (n < 14 && dc == 42) return false;
|
|
|
|
if (n < 14) {
|
|
|
|
memmove(&addr->ip6[dc + (14 - n)], &addr->ip6[dc], n - dc + 2);
|
|
|
|
memset(&addr->ip6[dc], 0, 14 - n);
|
|
|
|
}
|
|
|
|
addr->is_ip6 = true;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool mg_aton(struct mg_str str, struct mg_addr *addr) {
|
|
|
|
// LOG(LL_INFO, ("[%.*s]", (int) str.len, str.ptr));
|
|
|
|
return mg_atonl(str, addr) || mg_aton4(str, addr) || mg_aton6(str, addr);
|
2020-12-21 00:55:33 +08:00
|
|
|
}
|
|
|
|
|
2020-12-21 20:26:44 +08:00
|
|
|
void mg_mgr_free(struct mg_mgr *mgr) {
|
|
|
|
struct mg_connection *c;
|
|
|
|
for (c = mgr->conns; c != NULL; c = c->next) c->is_closing = 1;
|
|
|
|
mg_mgr_poll(mgr, 0);
|
2021-05-18 00:36:57 +08:00
|
|
|
#if MG_ARCH == MG_ARCH_FREERTOS_TCP
|
2020-12-21 20:26:44 +08:00
|
|
|
FreeRTOS_DeleteSocketSet(mgr->ss);
|
|
|
|
#endif
|
|
|
|
LOG(LL_INFO, ("All connections closed"));
|
|
|
|
}
|
|
|
|
|
|
|
|
void mg_mgr_init(struct mg_mgr *mgr) {
|
2021-05-12 15:43:34 +08:00
|
|
|
memset(mgr, 0, sizeof(*mgr));
|
2021-03-13 20:34:26 +08:00
|
|
|
#if defined(_WIN32) && MG_ENABLE_WINSOCK
|
2021-05-12 16:25:21 +08:00
|
|
|
// clang-format off
|
|
|
|
{ WSADATA data; WSAStartup(MAKEWORD(2, 2), &data); }
|
|
|
|
// clang-format on
|
2021-05-12 15:43:34 +08:00
|
|
|
#elif MG_ARCH == MG_ARCH_FREERTOS_TCP
|
2020-12-21 20:26:44 +08:00
|
|
|
mgr->ss = FreeRTOS_CreateSocketSet();
|
|
|
|
#elif defined(__unix) || defined(__unix__) || defined(__APPLE__)
|
|
|
|
// Ignore SIGPIPE signal, so if client cancels the request, it
|
|
|
|
// won't kill the whole process.
|
|
|
|
signal(SIGPIPE, SIG_IGN);
|
|
|
|
#endif
|
|
|
|
mgr->dnstimeout = 3000;
|
2020-12-22 17:44:59 +08:00
|
|
|
mgr->dns4.url = "udp://8.8.8.8:53";
|
|
|
|
mgr->dns6.url = "udp://[2001:4860:4860::8888]:53";
|
2020-12-21 20:26:44 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
#ifdef MG_ENABLE_LINES
|
|
|
|
#line 1 "src/sha1.c"
|
|
|
|
#endif
|
|
|
|
/* Copyright(c) By Steve Reid <steve@edmweb.com> */
|
|
|
|
/* 100% Public Domain */
|
|
|
|
|
|
|
|
#include <string.h>
|
|
|
|
|
|
|
|
/*
|
|
|
|
* clang with std=-c99 uses __LITTLE_ENDIAN, by default
|
|
|
|
* while for ex, RTOS gcc - LITTLE_ENDIAN, by default
|
|
|
|
* it depends on __USE_BSD, but let's have everything
|
|
|
|
*/
|
|
|
|
#if !defined(BYTE_ORDER) && defined(__BYTE_ORDER)
|
|
|
|
#define BYTE_ORDER __BYTE_ORDER
|
|
|
|
#ifndef LITTLE_ENDIAN
|
|
|
|
#define LITTLE_ENDIAN __LITTLE_ENDIAN
|
|
|
|
#endif /* LITTLE_ENDIAN */
|
|
|
|
#ifndef BIG_ENDIAN
|
|
|
|
#define BIG_ENDIAN __LITTLE_ENDIAN
|
|
|
|
#endif /* BIG_ENDIAN */
|
|
|
|
#endif /* BYTE_ORDER */
|
|
|
|
|
|
|
|
union char64long16 {
|
|
|
|
unsigned char c[64];
|
|
|
|
uint32_t l[16];
|
2015-11-01 00:26:19 +08:00
|
|
|
};
|
|
|
|
|
|
|
|
#define rol(value, bits) (((value) << (bits)) | ((value) >> (32 - (bits))))
|
|
|
|
|
|
|
|
static uint32_t blk0(union char64long16 *block, int i) {
|
|
|
|
/* Forrest: SHA expect BIG_ENDIAN, swap if LITTLE_ENDIAN */
|
|
|
|
#if BYTE_ORDER == LITTLE_ENDIAN
|
|
|
|
block->l[i] =
|
|
|
|
(rol(block->l[i], 24) & 0xFF00FF00) | (rol(block->l[i], 8) & 0x00FF00FF);
|
|
|
|
#endif
|
|
|
|
return block->l[i];
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
/* Avoid redefine warning (ARM /usr/include/sys/ucontext.h define R0~R4) */
|
|
|
|
#undef blk
|
|
|
|
#undef R0
|
|
|
|
#undef R1
|
|
|
|
#undef R2
|
|
|
|
#undef R3
|
|
|
|
#undef R4
|
|
|
|
|
|
|
|
#define blk(i) \
|
|
|
|
(block->l[i & 15] = rol(block->l[(i + 13) & 15] ^ block->l[(i + 8) & 15] ^ \
|
|
|
|
block->l[(i + 2) & 15] ^ block->l[i & 15], \
|
|
|
|
1))
|
|
|
|
#define R0(v, w, x, y, z, i) \
|
|
|
|
z += ((w & (x ^ y)) ^ y) + blk0(block, i) + 0x5A827999 + rol(v, 5); \
|
|
|
|
w = rol(w, 30);
|
|
|
|
#define R1(v, w, x, y, z, i) \
|
|
|
|
z += ((w & (x ^ y)) ^ y) + blk(i) + 0x5A827999 + rol(v, 5); \
|
|
|
|
w = rol(w, 30);
|
|
|
|
#define R2(v, w, x, y, z, i) \
|
|
|
|
z += (w ^ x ^ y) + blk(i) + 0x6ED9EBA1 + rol(v, 5); \
|
|
|
|
w = rol(w, 30);
|
|
|
|
#define R3(v, w, x, y, z, i) \
|
|
|
|
z += (((w | x) & y) | (w & x)) + blk(i) + 0x8F1BBCDC + rol(v, 5); \
|
|
|
|
w = rol(w, 30);
|
|
|
|
#define R4(v, w, x, y, z, i) \
|
|
|
|
z += (w ^ x ^ y) + blk(i) + 0xCA62C1D6 + rol(v, 5); \
|
|
|
|
w = rol(w, 30);
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
void mg_sha1_transform(uint32_t state[5], const unsigned char buffer[64]) {
|
2015-11-01 00:26:19 +08:00
|
|
|
uint32_t a, b, c, d, e;
|
|
|
|
union char64long16 block[1];
|
|
|
|
|
|
|
|
memcpy(block, buffer, 64);
|
|
|
|
a = state[0];
|
|
|
|
b = state[1];
|
|
|
|
c = state[2];
|
|
|
|
d = state[3];
|
|
|
|
e = state[4];
|
|
|
|
R0(a, b, c, d, e, 0);
|
|
|
|
R0(e, a, b, c, d, 1);
|
|
|
|
R0(d, e, a, b, c, 2);
|
|
|
|
R0(c, d, e, a, b, 3);
|
|
|
|
R0(b, c, d, e, a, 4);
|
|
|
|
R0(a, b, c, d, e, 5);
|
|
|
|
R0(e, a, b, c, d, 6);
|
|
|
|
R0(d, e, a, b, c, 7);
|
|
|
|
R0(c, d, e, a, b, 8);
|
|
|
|
R0(b, c, d, e, a, 9);
|
|
|
|
R0(a, b, c, d, e, 10);
|
|
|
|
R0(e, a, b, c, d, 11);
|
|
|
|
R0(d, e, a, b, c, 12);
|
|
|
|
R0(c, d, e, a, b, 13);
|
|
|
|
R0(b, c, d, e, a, 14);
|
|
|
|
R0(a, b, c, d, e, 15);
|
|
|
|
R1(e, a, b, c, d, 16);
|
|
|
|
R1(d, e, a, b, c, 17);
|
|
|
|
R1(c, d, e, a, b, 18);
|
|
|
|
R1(b, c, d, e, a, 19);
|
|
|
|
R2(a, b, c, d, e, 20);
|
|
|
|
R2(e, a, b, c, d, 21);
|
|
|
|
R2(d, e, a, b, c, 22);
|
|
|
|
R2(c, d, e, a, b, 23);
|
|
|
|
R2(b, c, d, e, a, 24);
|
|
|
|
R2(a, b, c, d, e, 25);
|
|
|
|
R2(e, a, b, c, d, 26);
|
|
|
|
R2(d, e, a, b, c, 27);
|
|
|
|
R2(c, d, e, a, b, 28);
|
|
|
|
R2(b, c, d, e, a, 29);
|
|
|
|
R2(a, b, c, d, e, 30);
|
|
|
|
R2(e, a, b, c, d, 31);
|
|
|
|
R2(d, e, a, b, c, 32);
|
|
|
|
R2(c, d, e, a, b, 33);
|
|
|
|
R2(b, c, d, e, a, 34);
|
|
|
|
R2(a, b, c, d, e, 35);
|
|
|
|
R2(e, a, b, c, d, 36);
|
|
|
|
R2(d, e, a, b, c, 37);
|
|
|
|
R2(c, d, e, a, b, 38);
|
|
|
|
R2(b, c, d, e, a, 39);
|
|
|
|
R3(a, b, c, d, e, 40);
|
|
|
|
R3(e, a, b, c, d, 41);
|
|
|
|
R3(d, e, a, b, c, 42);
|
|
|
|
R3(c, d, e, a, b, 43);
|
|
|
|
R3(b, c, d, e, a, 44);
|
|
|
|
R3(a, b, c, d, e, 45);
|
|
|
|
R3(e, a, b, c, d, 46);
|
|
|
|
R3(d, e, a, b, c, 47);
|
|
|
|
R3(c, d, e, a, b, 48);
|
|
|
|
R3(b, c, d, e, a, 49);
|
|
|
|
R3(a, b, c, d, e, 50);
|
|
|
|
R3(e, a, b, c, d, 51);
|
|
|
|
R3(d, e, a, b, c, 52);
|
|
|
|
R3(c, d, e, a, b, 53);
|
|
|
|
R3(b, c, d, e, a, 54);
|
|
|
|
R3(a, b, c, d, e, 55);
|
|
|
|
R3(e, a, b, c, d, 56);
|
|
|
|
R3(d, e, a, b, c, 57);
|
|
|
|
R3(c, d, e, a, b, 58);
|
|
|
|
R3(b, c, d, e, a, 59);
|
|
|
|
R4(a, b, c, d, e, 60);
|
|
|
|
R4(e, a, b, c, d, 61);
|
|
|
|
R4(d, e, a, b, c, 62);
|
|
|
|
R4(c, d, e, a, b, 63);
|
|
|
|
R4(b, c, d, e, a, 64);
|
|
|
|
R4(a, b, c, d, e, 65);
|
|
|
|
R4(e, a, b, c, d, 66);
|
|
|
|
R4(d, e, a, b, c, 67);
|
|
|
|
R4(c, d, e, a, b, 68);
|
|
|
|
R4(b, c, d, e, a, 69);
|
|
|
|
R4(a, b, c, d, e, 70);
|
|
|
|
R4(e, a, b, c, d, 71);
|
|
|
|
R4(d, e, a, b, c, 72);
|
|
|
|
R4(c, d, e, a, b, 73);
|
|
|
|
R4(b, c, d, e, a, 74);
|
|
|
|
R4(a, b, c, d, e, 75);
|
|
|
|
R4(e, a, b, c, d, 76);
|
|
|
|
R4(d, e, a, b, c, 77);
|
|
|
|
R4(c, d, e, a, b, 78);
|
|
|
|
R4(b, c, d, e, a, 79);
|
|
|
|
state[0] += a;
|
|
|
|
state[1] += b;
|
|
|
|
state[2] += c;
|
|
|
|
state[3] += d;
|
|
|
|
state[4] += e;
|
|
|
|
/* Erase working structures. The order of operations is important,
|
|
|
|
* used to ensure that compiler doesn't optimize those out. */
|
|
|
|
memset(block, 0, sizeof(block));
|
|
|
|
a = b = c = d = e = 0;
|
|
|
|
(void) a;
|
|
|
|
(void) b;
|
|
|
|
(void) c;
|
|
|
|
(void) d;
|
|
|
|
(void) e;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
void mg_sha1_init(mg_sha1_ctx *context) {
|
2015-11-01 00:26:19 +08:00
|
|
|
context->state[0] = 0x67452301;
|
|
|
|
context->state[1] = 0xEFCDAB89;
|
|
|
|
context->state[2] = 0x98BADCFE;
|
|
|
|
context->state[3] = 0x10325476;
|
|
|
|
context->state[4] = 0xC3D2E1F0;
|
|
|
|
context->count[0] = context->count[1] = 0;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
void mg_sha1_update(mg_sha1_ctx *context, const unsigned char *data,
|
|
|
|
size_t len) {
|
|
|
|
size_t i, j;
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
j = context->count[0];
|
2020-12-05 19:26:32 +08:00
|
|
|
if ((context->count[0] += (uint32_t) len << 3) < j) context->count[1]++;
|
|
|
|
context->count[1] += (uint32_t)(len >> 29);
|
2015-11-01 00:26:19 +08:00
|
|
|
j = (j >> 3) & 63;
|
|
|
|
if ((j + len) > 63) {
|
|
|
|
memcpy(&context->buffer[j], data, (i = 64 - j));
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_sha1_transform(context->state, context->buffer);
|
2015-11-01 00:26:19 +08:00
|
|
|
for (; i + 63 < len; i += 64) {
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_sha1_transform(context->state, &data[i]);
|
2015-11-01 00:26:19 +08:00
|
|
|
}
|
|
|
|
j = 0;
|
|
|
|
} else
|
|
|
|
i = 0;
|
|
|
|
memcpy(&context->buffer[j], &data[i], len - i);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
void mg_sha1_final(unsigned char digest[20], mg_sha1_ctx *context) {
|
2015-11-01 00:26:19 +08:00
|
|
|
unsigned i;
|
|
|
|
unsigned char finalcount[8], c;
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
for (i = 0; i < 8; i++) {
|
|
|
|
finalcount[i] = (unsigned char) ((context->count[(i >= 4 ? 0 : 1)] >>
|
|
|
|
((3 - (i & 3)) * 8)) &
|
|
|
|
255);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-11-01 00:26:19 +08:00
|
|
|
c = 0200;
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_sha1_update(context, &c, 1);
|
2015-11-01 00:26:19 +08:00
|
|
|
while ((context->count[0] & 504) != 448) {
|
|
|
|
c = 0000;
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_sha1_update(context, &c, 1);
|
2015-11-01 00:26:19 +08:00
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_sha1_update(context, finalcount, 8);
|
2015-11-01 00:26:19 +08:00
|
|
|
for (i = 0; i < 20; i++) {
|
|
|
|
digest[i] =
|
|
|
|
(unsigned char) ((context->state[i >> 2] >> ((3 - (i & 3)) * 8)) & 255);
|
|
|
|
}
|
|
|
|
memset(context, '\0', sizeof(*context));
|
|
|
|
memset(&finalcount, '\0', sizeof(finalcount));
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
void mg_hmac_sha1(const unsigned char *key, size_t keylen,
|
2015-11-01 00:26:19 +08:00
|
|
|
const unsigned char *data, size_t datalen,
|
|
|
|
unsigned char out[20]) {
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_sha1_ctx ctx;
|
2015-11-01 00:26:19 +08:00
|
|
|
unsigned char buf1[64], buf2[64], tmp_key[20], i;
|
|
|
|
|
|
|
|
if (keylen > sizeof(buf1)) {
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_sha1_init(&ctx);
|
|
|
|
mg_sha1_update(&ctx, key, keylen);
|
|
|
|
mg_sha1_final(tmp_key, &ctx);
|
2015-11-01 00:26:19 +08:00
|
|
|
key = tmp_key;
|
|
|
|
keylen = sizeof(tmp_key);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
memset(buf1, 0, sizeof(buf1));
|
|
|
|
memset(buf2, 0, sizeof(buf2));
|
|
|
|
memcpy(buf1, key, keylen);
|
|
|
|
memcpy(buf2, key, keylen);
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
for (i = 0; i < sizeof(buf1); i++) {
|
|
|
|
buf1[i] ^= 0x36;
|
|
|
|
buf2[i] ^= 0x5c;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_sha1_init(&ctx);
|
|
|
|
mg_sha1_update(&ctx, buf1, sizeof(buf1));
|
|
|
|
mg_sha1_update(&ctx, data, datalen);
|
|
|
|
mg_sha1_final(out, &ctx);
|
2015-11-01 00:26:19 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_sha1_init(&ctx);
|
|
|
|
mg_sha1_update(&ctx, buf2, sizeof(buf2));
|
|
|
|
mg_sha1_update(&ctx, out, 20);
|
|
|
|
mg_sha1_final(out, &ctx);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
#ifdef MG_ENABLE_LINES
|
|
|
|
#line 1 "src/sntp.c"
|
2017-04-11 18:32:19 +08:00
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
|
|
|
|
|
|
|
|
#define SNTP_INTERVAL_SEC (3600)
|
2021-07-24 18:40:52 +08:00
|
|
|
#define SNTP_TIME_OFFSET 2208988800UL
|
2020-12-05 19:26:32 +08:00
|
|
|
|
|
|
|
static unsigned long s_sntmp_next;
|
|
|
|
|
|
|
|
int mg_sntp_parse(const unsigned char *buf, size_t len, struct timeval *tv) {
|
2020-12-11 21:16:51 +08:00
|
|
|
int mode = len > 0 ? buf[0] & 7 : 0, res = -1;
|
2020-12-05 19:26:32 +08:00
|
|
|
if (len < 48) {
|
|
|
|
LOG(LL_ERROR, ("%s", "corrupt packet"));
|
|
|
|
} else if ((buf[0] & 0x38) >> 3 != 4) {
|
|
|
|
LOG(LL_ERROR, ("%s", "wrong version"));
|
|
|
|
} else if (mode != 4 && mode != 5) {
|
|
|
|
LOG(LL_ERROR, ("%s", "not a server reply"));
|
|
|
|
} else if (buf[1] == 0) {
|
|
|
|
LOG(LL_ERROR, ("%s", "server sent a kiss of death"));
|
|
|
|
} else {
|
|
|
|
uint32_t *data = (uint32_t *) &buf[40];
|
2021-08-08 05:55:13 +08:00
|
|
|
tv->tv_sec = (time_t) (mg_ntohl(data[0]) - SNTP_TIME_OFFSET);
|
2021-05-29 01:30:42 +08:00
|
|
|
tv->tv_usec = (suseconds_t) mg_ntohl(data[1]);
|
|
|
|
s_sntmp_next = (unsigned long) (tv->tv_sec + SNTP_INTERVAL_SEC);
|
2020-12-05 19:26:32 +08:00
|
|
|
res = 0;
|
|
|
|
}
|
|
|
|
return res;
|
2017-04-11 18:32:19 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
static void sntp_cb(struct mg_connection *c, int ev, void *evd, void *fnd) {
|
|
|
|
if (ev == MG_EV_READ) {
|
|
|
|
struct timeval tv = {0, 0};
|
|
|
|
if (mg_sntp_parse(c->recv.buf, c->recv.len, &tv) == 0) {
|
|
|
|
mg_call(c, MG_EV_SNTP_TIME, &tv);
|
|
|
|
LOG(LL_DEBUG, ("%u.%u, next at %lu", (unsigned) tv.tv_sec,
|
|
|
|
(unsigned) tv.tv_usec, s_sntmp_next));
|
|
|
|
}
|
|
|
|
c->recv.len = 0; // Clear receive buffer
|
2021-08-08 05:55:13 +08:00
|
|
|
} else if (ev == MG_EV_CONNECT) {
|
2021-05-29 01:30:42 +08:00
|
|
|
mg_sntp_send(c, (unsigned long) time(NULL));
|
2020-12-05 19:26:32 +08:00
|
|
|
} else if (ev == MG_EV_CLOSE) {
|
2017-04-11 18:32:19 +08:00
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
(void) fnd;
|
|
|
|
(void) evd;
|
2017-04-11 18:32:19 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
void mg_sntp_send(struct mg_connection *c, unsigned long utc) {
|
|
|
|
if (c->is_resolving) {
|
2020-12-21 20:26:44 +08:00
|
|
|
LOG(LL_ERROR, ("%lu wait until resolved", c->id));
|
2020-12-05 19:26:32 +08:00
|
|
|
} else if (utc > s_sntmp_next) {
|
|
|
|
uint8_t buf[48] = {0};
|
|
|
|
s_sntmp_next = utc + SNTP_INTERVAL_SEC;
|
|
|
|
buf[0] = (3 << 6) | (4 << 3) | 3;
|
|
|
|
mg_send(c, buf, sizeof(buf));
|
|
|
|
LOG(LL_DEBUG,
|
|
|
|
("%p request sent, ct %lu, next at %lu", c->fd, utc, s_sntmp_next));
|
2017-04-11 18:32:19 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_connection *mg_sntp_connect(struct mg_mgr *mgr, const char *url,
|
|
|
|
mg_event_handler_t fn, void *fnd) {
|
|
|
|
struct mg_connection *c = NULL;
|
|
|
|
if (url == NULL) url = "udp://time.google.com:123";
|
|
|
|
if ((c = mg_connect(mgr, url, fn, fnd)) != NULL) c->pfn = sntp_cb;
|
|
|
|
return c;
|
2017-04-11 18:32:19 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
#ifdef MG_ENABLE_LINES
|
|
|
|
#line 1 "src/sock.c"
|
|
|
|
#endif
|
2017-04-11 18:32:19 +08:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2018-12-06 04:46:00 +08:00
|
|
|
|
2017-04-11 18:32:19 +08:00
|
|
|
|
2018-11-23 22:54:44 +08:00
|
|
|
|
2018-12-06 04:46:00 +08:00
|
|
|
|
2017-04-11 18:32:19 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
#if MG_ENABLE_SOCKET
|
2021-03-13 20:34:26 +08:00
|
|
|
#if defined(_WIN32) && MG_ENABLE_WINSOCK
|
2020-12-05 19:26:32 +08:00
|
|
|
#define MG_SOCK_ERRNO WSAGetLastError()
|
2021-01-20 19:10:02 +08:00
|
|
|
#ifndef SO_EXCLUSIVEADDRUSE
|
|
|
|
#define SO_EXCLUSIVEADDRUSE ((int) (~SO_REUSEADDR))
|
|
|
|
#endif
|
2021-05-11 16:12:06 +08:00
|
|
|
#elif MG_ARCH == MG_ARCH_FREERTOS_TCP
|
2020-12-05 19:26:32 +08:00
|
|
|
#define MG_SOCK_ERRNO errno
|
|
|
|
typedef Socket_t SOCKET;
|
|
|
|
#define INVALID_SOCKET FREERTOS_INVALID_SOCKET
|
|
|
|
#else
|
|
|
|
#define MG_SOCK_ERRNO errno
|
2021-03-13 20:34:26 +08:00
|
|
|
#ifndef closesocket
|
2020-12-05 19:26:32 +08:00
|
|
|
#define closesocket(x) close(x)
|
2021-03-13 20:34:26 +08:00
|
|
|
#endif
|
2020-12-05 19:26:32 +08:00
|
|
|
#define INVALID_SOCKET (-1)
|
|
|
|
typedef int SOCKET;
|
|
|
|
#endif
|
2017-04-11 18:32:19 +08:00
|
|
|
|
2021-08-06 17:23:08 +08:00
|
|
|
#define FD(c_) ((SOCKET) (size_t) (c_)->fd)
|
|
|
|
#define S2PTR(s_) ((void *) (size_t) (s_))
|
2021-03-13 20:34:26 +08:00
|
|
|
|
2020-12-13 00:48:47 +08:00
|
|
|
#ifndef MSG_NONBLOCKING
|
|
|
|
#define MSG_NONBLOCKING 0
|
|
|
|
#endif
|
|
|
|
|
2021-08-08 00:48:51 +08:00
|
|
|
#ifndef AF_INET6
|
|
|
|
#define AF_INET6 10
|
|
|
|
#endif
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
union usa {
|
|
|
|
struct sockaddr sa;
|
|
|
|
struct sockaddr_in sin;
|
|
|
|
#if MG_ENABLE_IPV6
|
|
|
|
struct sockaddr_in6 sin6;
|
|
|
|
#endif
|
|
|
|
};
|
2017-04-11 18:32:19 +08:00
|
|
|
|
2021-08-08 00:22:47 +08:00
|
|
|
static socklen_t tousa(struct mg_addr *a, union usa *usa) {
|
|
|
|
socklen_t len = sizeof(usa->sin);
|
|
|
|
memset(usa, 0, sizeof(*usa));
|
|
|
|
usa->sin.sin_family = AF_INET;
|
|
|
|
usa->sin.sin_port = a->port;
|
|
|
|
*(uint32_t *) &usa->sin.sin_addr = a->ip;
|
2020-12-22 17:44:59 +08:00
|
|
|
#if MG_ENABLE_IPV6
|
|
|
|
if (a->is_ip6) {
|
2021-08-08 00:22:47 +08:00
|
|
|
usa->sin.sin_family = AF_INET6;
|
|
|
|
usa->sin6.sin6_port = a->port;
|
|
|
|
memcpy(&usa->sin6.sin6_addr, a->ip6, sizeof(a->ip6));
|
|
|
|
len = sizeof(usa->sin6);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
return len;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void tomgaddr(union usa *usa, struct mg_addr *a, bool is_ip6) {
|
|
|
|
a->is_ip6 = is_ip6;
|
|
|
|
a->port = usa->sin.sin_port;
|
|
|
|
memcpy(&a->ip, &usa->sin.sin_addr, sizeof(a->ip));
|
|
|
|
#if MG_ENABLE_IPV6
|
|
|
|
if (is_ip6) {
|
|
|
|
memcpy(a->ip6, &usa->sin6.sin6_addr, sizeof(a->ip6));
|
|
|
|
a->port = usa->sin6.sin6_port;
|
2020-12-22 17:44:59 +08:00
|
|
|
}
|
|
|
|
#endif
|
2017-04-11 18:32:19 +08:00
|
|
|
}
|
|
|
|
|
2021-05-29 01:30:42 +08:00
|
|
|
static bool mg_sock_would_block(void) {
|
2020-12-05 19:26:32 +08:00
|
|
|
int err = MG_SOCK_ERRNO;
|
2021-05-29 01:30:42 +08:00
|
|
|
return err == EINPROGRESS || err == EWOULDBLOCK
|
2020-12-05 19:26:32 +08:00
|
|
|
#ifndef WINCE
|
2021-05-29 01:30:42 +08:00
|
|
|
|| err == EAGAIN || err == EINTR
|
2020-12-05 19:26:32 +08:00
|
|
|
#endif
|
2021-03-13 20:34:26 +08:00
|
|
|
#if defined(_WIN32) && MG_ENABLE_WINSOCK
|
2021-05-29 01:30:42 +08:00
|
|
|
|| err == WSAEINTR || err == WSAEWOULDBLOCK
|
2020-12-05 19:26:32 +08:00
|
|
|
#endif
|
|
|
|
;
|
2017-04-11 18:32:19 +08:00
|
|
|
}
|
|
|
|
|
2021-05-29 01:30:42 +08:00
|
|
|
static struct mg_connection *alloc_conn(struct mg_mgr *mgr, bool is_client,
|
2020-12-05 19:26:32 +08:00
|
|
|
SOCKET fd) {
|
|
|
|
struct mg_connection *c = (struct mg_connection *) calloc(1, sizeof(*c));
|
|
|
|
if (c != NULL) {
|
|
|
|
c->is_client = is_client;
|
2021-08-01 05:26:43 +08:00
|
|
|
c->fd = S2PTR(fd);
|
2020-12-05 19:26:32 +08:00
|
|
|
c->mgr = mgr;
|
2020-12-21 20:26:44 +08:00
|
|
|
c->id = ++mgr->nextid;
|
2017-04-11 18:32:19 +08:00
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
return c;
|
2017-04-11 18:32:19 +08:00
|
|
|
}
|
|
|
|
|
2021-05-29 01:30:42 +08:00
|
|
|
static long mg_sock_send(struct mg_connection *c, const void *buf, size_t len) {
|
2021-08-08 05:55:13 +08:00
|
|
|
long n = send(FD(c), (char *) buf, len, MSG_NONBLOCKING);
|
2021-05-29 01:30:42 +08:00
|
|
|
return n == 0 ? -1 : n < 0 && mg_sock_would_block() ? 0 : n;
|
2017-04-11 18:32:19 +08:00
|
|
|
}
|
|
|
|
|
2021-05-29 01:30:42 +08:00
|
|
|
bool mg_send(struct mg_connection *c, const void *buf, size_t len) {
|
|
|
|
return c->is_udp ? mg_sock_send(c, buf, len) > 0
|
2021-08-28 15:08:54 +08:00
|
|
|
: mg_iobuf_add(&c->send, c->send.len, buf, len, MG_IO_SIZE);
|
2017-07-03 22:55:24 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
static void mg_set_non_blocking_mode(SOCKET fd) {
|
2021-03-13 20:34:26 +08:00
|
|
|
#if defined(_WIN32) && MG_ENABLE_WINSOCK
|
2020-12-05 19:26:32 +08:00
|
|
|
unsigned long on = 1;
|
|
|
|
ioctlsocket(fd, FIONBIO, &on);
|
2021-05-11 16:12:06 +08:00
|
|
|
#elif MG_ARCH == MG_ARCH_FREERTOS_TCP
|
2020-12-05 19:26:32 +08:00
|
|
|
const BaseType_t off = 0;
|
|
|
|
setsockopt(fd, 0, FREERTOS_SO_RCVTIMEO, &off, sizeof(off));
|
|
|
|
setsockopt(fd, 0, FREERTOS_SO_SNDTIMEO, &off, sizeof(off));
|
2021-08-11 14:56:46 +08:00
|
|
|
#elif MG_ARCH == MG_ARCH_FREERTOS_LWIP
|
2021-08-28 14:16:38 +08:00
|
|
|
lwip_fcntl(fd, F_SETFL, O_NONBLOCK);
|
2021-09-14 14:28:28 +08:00
|
|
|
#elif MG_ARCH == MG_ARCH_AZURERTOS
|
|
|
|
fcntl(fd, F_SETFL, O_NONBLOCK);
|
2020-12-05 19:26:32 +08:00
|
|
|
#else
|
2021-07-14 05:49:36 +08:00
|
|
|
fcntl(fd, F_SETFL, fcntl(fd, F_GETFL, 0) | O_NONBLOCK, FD_CLOEXEC);
|
2020-12-05 19:26:32 +08:00
|
|
|
#endif
|
2017-04-11 18:32:19 +08:00
|
|
|
}
|
|
|
|
|
2021-07-19 15:47:38 +08:00
|
|
|
SOCKET mg_open_listener(const char *url, struct mg_addr *addr) {
|
2020-12-22 18:16:31 +08:00
|
|
|
SOCKET fd = INVALID_SOCKET;
|
2021-09-23 01:52:42 +08:00
|
|
|
int s_err = 0; // Memoized socket error, in case closesocket() overrides it
|
2021-07-19 15:47:38 +08:00
|
|
|
memset(addr, 0, sizeof(*addr));
|
|
|
|
addr->port = mg_htons(mg_url_port(url));
|
|
|
|
if (!mg_aton(mg_url_host(url), addr)) {
|
2020-12-22 18:16:31 +08:00
|
|
|
LOG(LL_ERROR, ("invalid listening URL: %s", url));
|
|
|
|
} else {
|
2021-08-08 00:22:47 +08:00
|
|
|
union usa usa;
|
|
|
|
socklen_t slen = tousa(addr, &usa);
|
|
|
|
int on = 1, af = addr->is_ip6 ? AF_INET6 : AF_INET;
|
2020-12-22 18:16:31 +08:00
|
|
|
int type = strncmp(url, "udp:", 4) == 0 ? SOCK_DGRAM : SOCK_STREAM;
|
|
|
|
int proto = type == SOCK_DGRAM ? IPPROTO_UDP : IPPROTO_TCP;
|
2021-08-28 14:16:38 +08:00
|
|
|
(void) on;
|
2017-07-06 00:14:34 +08:00
|
|
|
|
2020-12-22 18:16:31 +08:00
|
|
|
if ((fd = socket(af, type, proto)) != INVALID_SOCKET &&
|
2021-08-28 14:16:38 +08:00
|
|
|
#if (!defined(_WIN32) || !defined(SO_EXCLUSIVEADDRUSE)) && \
|
|
|
|
(!defined(LWIP_SOCKET) || (defined(LWIP_SOCKET) && SO_REUSE == 1))
|
2021-08-11 14:56:46 +08:00
|
|
|
// 1. SO_RESUSEADDR is not enabled on Windows because the semantics of
|
|
|
|
// SO_REUSEADDR on UNIX and Windows is different. On Windows,
|
2021-08-28 14:16:38 +08:00
|
|
|
// SO_REUSEADDR allows to bind a socket to a port without error even
|
|
|
|
// if the port is already open by another program. This is not the
|
|
|
|
// behavior SO_REUSEADDR was designed for, and leads to hard-to-track
|
|
|
|
// failure scenarios. Therefore, SO_REUSEADDR was disabled on Windows
|
|
|
|
// unless SO_EXCLUSIVEADDRUSE is supported and set on a socket.
|
|
|
|
// 2. In case of LWIP, SO_REUSEADDR should be explicitly enabled, by
|
|
|
|
// defining
|
2021-08-11 14:56:46 +08:00
|
|
|
// SO_REUSE (in lwipopts.h), otherwise the code below will compile
|
2021-08-28 14:16:38 +08:00
|
|
|
// but won't work! (setsockopt will return EINVAL)
|
2020-12-26 03:24:51 +08:00
|
|
|
!setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, (char *) &on, sizeof(on)) &&
|
2020-12-05 19:26:32 +08:00
|
|
|
#endif
|
|
|
|
#if defined(_WIN32) && defined(SO_EXCLUSIVEADDRUSE) && !defined(WINCE)
|
2020-12-22 18:16:31 +08:00
|
|
|
// "Using SO_REUSEADDR and SO_EXCLUSIVEADDRUSE"
|
2021-01-20 19:10:02 +08:00
|
|
|
//! setsockopt(fd, SOL_SOCKET, SO_BROADCAST, (char *) &on, sizeof(on))
|
|
|
|
//! &&
|
2020-12-26 03:24:51 +08:00
|
|
|
!setsockopt(fd, SOL_SOCKET, SO_EXCLUSIVEADDRUSE, (char *) &on,
|
2020-12-22 18:16:31 +08:00
|
|
|
sizeof(on)) &&
|
2020-12-05 19:26:32 +08:00
|
|
|
#endif
|
2020-12-22 18:16:31 +08:00
|
|
|
bind(fd, &usa.sa, slen) == 0 &&
|
|
|
|
// NOTE(lsm): FreeRTOS uses backlog value as a connection limit
|
2021-09-14 14:28:28 +08:00
|
|
|
(type == SOCK_DGRAM || listen(fd, MG_SOCK_LISTEN_BACKLOG_SIZE) == 0)) {
|
2021-07-19 15:47:38 +08:00
|
|
|
// In case port was set to 0, get the real port number
|
|
|
|
if (getsockname(fd, &usa.sa, &slen) == 0) {
|
|
|
|
addr->port = usa.sin.sin_port;
|
|
|
|
#if MG_ENABLE_IPV6
|
|
|
|
if (addr->is_ip6) addr->port = usa.sin6.sin6_port;
|
|
|
|
#endif
|
|
|
|
}
|
2020-12-22 18:16:31 +08:00
|
|
|
mg_set_non_blocking_mode(fd);
|
2020-12-26 03:24:51 +08:00
|
|
|
} else if (fd != INVALID_SOCKET) {
|
2021-09-23 01:52:42 +08:00
|
|
|
s_err = MG_SOCK_ERRNO;
|
2020-12-22 18:16:31 +08:00
|
|
|
closesocket(fd);
|
|
|
|
fd = INVALID_SOCKET;
|
|
|
|
}
|
2019-12-30 03:29:52 +08:00
|
|
|
}
|
2021-05-18 00:36:57 +08:00
|
|
|
if (fd == INVALID_SOCKET) {
|
2021-09-23 03:38:28 +08:00
|
|
|
if (s_err == 0) s_err = MG_SOCK_ERRNO;
|
2021-09-23 01:52:42 +08:00
|
|
|
LOG(LL_ERROR, ("Failed to listen on %s, errno %d", url, s_err));
|
2021-05-18 00:36:57 +08:00
|
|
|
}
|
2019-12-30 03:29:52 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
return fd;
|
2019-01-17 03:44:59 +08:00
|
|
|
}
|
|
|
|
|
2021-05-29 01:30:42 +08:00
|
|
|
static long mg_sock_recv(struct mg_connection *c, void *buf, size_t len) {
|
|
|
|
long n = 0;
|
|
|
|
if (c->is_udp) {
|
|
|
|
union usa usa;
|
2021-08-08 00:22:47 +08:00
|
|
|
socklen_t slen = tousa(&c->peer, &usa);
|
2021-05-29 01:30:42 +08:00
|
|
|
n = recvfrom(FD(c), (char *) buf, len, 0, &usa.sa, &slen);
|
2021-08-08 00:22:47 +08:00
|
|
|
if (n > 0) tomgaddr(&usa, &c->peer, slen != sizeof(usa.sin));
|
2020-12-05 19:26:32 +08:00
|
|
|
} else {
|
2021-05-29 01:30:42 +08:00
|
|
|
n = recv(FD(c), (char *) buf, len, MSG_NONBLOCKING);
|
2018-07-07 21:45:52 +08:00
|
|
|
}
|
2021-05-29 01:30:42 +08:00
|
|
|
return n == 0 ? -1 : n < 0 && mg_sock_would_block() ? 0 : n;
|
2018-07-07 21:45:52 +08:00
|
|
|
}
|
2018-10-29 22:42:32 +08:00
|
|
|
|
2021-05-29 01:30:42 +08:00
|
|
|
// NOTE(lsm): do only one iteration of reads, cause some systems
|
|
|
|
// (e.g. FreeRTOS stack) return 0 instead of -1/EWOULDBLOCK when no data
|
|
|
|
static void read_conn(struct mg_connection *c) {
|
|
|
|
if (c->recv.len >= MG_MAX_RECV_BUF_SIZE) {
|
|
|
|
mg_error(c, "max_recv_buf_size reached");
|
|
|
|
} else if (c->recv.size - c->recv.len < MG_IO_SIZE &&
|
|
|
|
!mg_iobuf_resize(&c->recv, c->recv.size + MG_IO_SIZE)) {
|
|
|
|
mg_error(c, "oom");
|
|
|
|
} else {
|
|
|
|
char *buf = (char *) &c->recv.buf[c->recv.len];
|
|
|
|
size_t len = c->recv.size - c->recv.len;
|
|
|
|
long n = c->is_tls ? mg_tls_recv(c, buf, len) : mg_sock_recv(c, buf, len);
|
|
|
|
LOG(n > 0 ? LL_VERBOSE_DEBUG : LL_DEBUG,
|
|
|
|
("%-3lu %d%d%d%d%d%d%d%d%d%d%d%d%d%d %7ld %ld/%ld err %d", c->id,
|
|
|
|
c->is_listening, c->is_client, c->is_accepted, c->is_resolving,
|
|
|
|
c->is_connecting, c->is_tls, c->is_tls_hs, c->is_udp, c->is_websocket,
|
|
|
|
c->is_hexdumping, c->is_draining, c->is_closing, c->is_readable,
|
|
|
|
c->is_writable, (long) c->recv.len, n, (long) len, MG_SOCK_ERRNO));
|
|
|
|
if (n == 0) {
|
|
|
|
// Do nothing
|
|
|
|
} else if (n < 0) {
|
|
|
|
c->is_closing = 1; // Error, or normal termination
|
|
|
|
} else if (n > 0) {
|
|
|
|
struct mg_str evd = mg_str_n(buf, (size_t) n);
|
|
|
|
if (c->is_hexdumping) {
|
|
|
|
char *s = mg_hexdump(buf, (size_t) n);
|
|
|
|
LOG(LL_INFO, ("\n-- %lu %s %s %ld\n%s", c->id, c->label, "<-", n, s));
|
|
|
|
free(s);
|
|
|
|
}
|
|
|
|
c->recv.len += (size_t) n;
|
|
|
|
mg_call(c, MG_EV_READ, &evd);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static void write_conn(struct mg_connection *c) {
|
|
|
|
char *buf = (char *) c->send.buf;
|
|
|
|
size_t len = c->send.len;
|
|
|
|
long n = c->is_tls ? mg_tls_send(c, buf, len) : mg_sock_send(c, buf, len);
|
|
|
|
|
|
|
|
LOG(n > 0 ? LL_VERBOSE_DEBUG : LL_DEBUG,
|
|
|
|
("%-3lu %d%d%d%d%d%d%d%d%d%d%d%d%d%d %7ld %ld err %d", c->id,
|
|
|
|
c->is_listening, c->is_client, c->is_accepted, c->is_resolving,
|
|
|
|
c->is_connecting, c->is_tls, c->is_tls_hs, c->is_udp, c->is_websocket,
|
|
|
|
c->is_hexdumping, c->is_draining, c->is_closing, c->is_readable,
|
|
|
|
c->is_writable, (long) c->send.len, n, MG_SOCK_ERRNO));
|
|
|
|
|
|
|
|
if (n == 0) {
|
|
|
|
// Do nothing
|
|
|
|
} else if (n < 0) {
|
|
|
|
c->is_closing = 1; // Error, or normal termination
|
|
|
|
} else if (n > 0) {
|
|
|
|
// Hexdump before deleting
|
|
|
|
if (c->is_hexdumping) {
|
|
|
|
char *s = mg_hexdump(buf, (size_t) n);
|
|
|
|
LOG(LL_INFO, ("\n-- %lu %s %s %ld\n%s", c->id, c->label, "<-", n, s));
|
|
|
|
free(s);
|
|
|
|
}
|
2021-08-28 14:16:38 +08:00
|
|
|
mg_iobuf_del(&c->send, 0, (size_t) n);
|
2020-12-05 19:26:32 +08:00
|
|
|
if (c->send.len == 0) mg_iobuf_resize(&c->send, 0);
|
2021-05-29 01:30:42 +08:00
|
|
|
mg_call(c, MG_EV_WRITE, &n);
|
|
|
|
// if (c->send.len == 0) mg_iobuf_resize(&c->send, 0);
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
2018-10-29 22:42:32 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2020-12-22 17:44:59 +08:00
|
|
|
static void close_conn(struct mg_connection *c) {
|
2021-08-05 17:39:24 +08:00
|
|
|
mg_resolve_cancel(c); // Close any pending DNS query
|
2021-04-09 23:55:12 +08:00
|
|
|
LIST_DELETE(struct mg_connection, &c->mgr->conns, c);
|
2020-12-22 17:44:59 +08:00
|
|
|
if (c == c->mgr->dns4.c) c->mgr->dns4.c = NULL;
|
|
|
|
if (c == c->mgr->dns6.c) c->mgr->dns6.c = NULL;
|
2021-08-05 17:39:24 +08:00
|
|
|
// Order of operations is important. `MG_EV_CLOSE` event must be fired
|
|
|
|
// before we deallocate received data, see #1331
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_call(c, MG_EV_CLOSE, NULL);
|
2020-12-21 20:26:44 +08:00
|
|
|
LOG(LL_DEBUG, ("%lu closed", c->id));
|
2020-12-05 19:26:32 +08:00
|
|
|
if (FD(c) != INVALID_SOCKET) {
|
|
|
|
closesocket(FD(c));
|
2021-05-11 16:12:06 +08:00
|
|
|
#if MG_ARCH == MG_ARCH_FREERTOS_TCP
|
2020-12-22 17:44:59 +08:00
|
|
|
FreeRTOS_FD_CLR(c->fd, c->mgr->ss, eSELECT_ALL);
|
Change from using #ifdef to #if for features tests
"#if FOO" still works with simple -DFOO, but gives more flexibility.
Specifically, if user expressed no preference (FOO is not defined),
we can apply reasonable defaults (this is the legitimate use of ifdef).
In short, from now on, please use
#if MG_ENABLE_FOO
instead of
#ifdef MG_ENABLE_FOO
Since we are all used to #ifdef, this change also adds a precommit check
to police this. Specifically, in *.h and *.c files that are Copyright Cesanta,
"ifdef" and "if defined()" are not allowed to be used with macros that contain
ENABLE or DISABLE, unless the like also contains "ifdef-ok".
Hence, if you are sure you want to use ifdef, use this:
#ifdef MG_ENABLE_FOO /* ifdef-ok */
PUBLISHED_FROM=9be829448f53cff575d6cae8b9945fb12531c15a
2016-10-14 01:55:08 +08:00
|
|
|
#endif
|
2021-08-05 17:39:24 +08:00
|
|
|
c->fd = S2PTR(INVALID_SOCKET);
|
2015-11-01 00:26:19 +08:00
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_tls_free(c);
|
2021-08-05 17:39:24 +08:00
|
|
|
mg_iobuf_free(&c->recv);
|
|
|
|
mg_iobuf_free(&c->send);
|
2020-12-05 19:26:32 +08:00
|
|
|
memset(c, 0, sizeof(*c));
|
|
|
|
free(c);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
static void setsockopts(struct mg_connection *c) {
|
2021-09-14 14:28:28 +08:00
|
|
|
#if MG_ARCH == MG_ARCH_FREERTOS_TCP || MG_ARCH == MG_ARCH_AZURERTOS
|
2021-05-19 07:00:32 +08:00
|
|
|
(void) c;
|
2015-11-01 00:26:19 +08:00
|
|
|
#else
|
2020-12-26 03:24:51 +08:00
|
|
|
int on = 1;
|
2020-12-05 19:26:32 +08:00
|
|
|
#if !defined(SOL_TCP)
|
|
|
|
#define SOL_TCP IPPROTO_TCP
|
2016-06-10 01:33:22 +08:00
|
|
|
#endif
|
2020-12-26 03:24:51 +08:00
|
|
|
setsockopt(FD(c), SOL_TCP, TCP_NODELAY, (char *) &on, sizeof(on));
|
2020-12-05 19:26:32 +08:00
|
|
|
#if defined(TCP_QUICKACK)
|
2020-12-26 03:24:51 +08:00
|
|
|
setsockopt(FD(c), SOL_TCP, TCP_QUICKACK, (char *) &on, sizeof(on));
|
2016-11-09 06:29:07 +08:00
|
|
|
#endif
|
2020-12-26 03:24:51 +08:00
|
|
|
setsockopt(FD(c), SOL_SOCKET, SO_KEEPALIVE, (char *) &on, sizeof(on));
|
2020-12-05 19:26:32 +08:00
|
|
|
#if ESP32 || ESP8266 || defined(__linux__)
|
|
|
|
int idle = 60;
|
|
|
|
setsockopt(FD(c), IPPROTO_TCP, TCP_KEEPIDLE, &idle, sizeof(idle));
|
|
|
|
#endif
|
2021-03-04 19:09:57 +08:00
|
|
|
#if !defined(_WIN32) && !defined(__QNX__)
|
2020-12-26 03:24:51 +08:00
|
|
|
{
|
|
|
|
int cnt = 3, intvl = 20;
|
|
|
|
setsockopt(FD(c), IPPROTO_TCP, TCP_KEEPCNT, &cnt, sizeof(cnt));
|
|
|
|
setsockopt(FD(c), IPPROTO_TCP, TCP_KEEPINTVL, &intvl, sizeof(intvl));
|
|
|
|
}
|
2016-11-12 04:34:00 +08:00
|
|
|
#endif
|
2017-06-13 23:22:43 +08:00
|
|
|
#endif
|
2016-11-12 04:34:00 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
void mg_connect_resolved(struct mg_connection *c) {
|
|
|
|
char buf[40];
|
|
|
|
int type = c->is_udp ? SOCK_DGRAM : SOCK_STREAM;
|
2021-08-08 05:55:13 +08:00
|
|
|
int rc, af = c->peer.is_ip6 ? AF_INET6 : AF_INET;
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_straddr(c, buf, sizeof(buf));
|
2021-08-01 05:26:43 +08:00
|
|
|
c->fd = S2PTR(socket(af, type, 0));
|
2020-12-21 20:26:44 +08:00
|
|
|
if (FD(c) == INVALID_SOCKET) {
|
|
|
|
mg_error(c, "socket(): %d", MG_SOCK_ERRNO);
|
2021-08-08 05:55:13 +08:00
|
|
|
} else {
|
2021-08-08 00:22:47 +08:00
|
|
|
union usa usa;
|
|
|
|
socklen_t slen = tousa(&c->peer, &usa);
|
2021-08-08 05:55:13 +08:00
|
|
|
if (c->is_udp == 0) mg_set_non_blocking_mode(FD(c));
|
|
|
|
if (c->is_udp == 0) setsockopts(c);
|
|
|
|
mg_call(c, MG_EV_RESOLVE, NULL);
|
|
|
|
if ((rc = connect(FD(c), &usa.sa, slen)) == 0) {
|
|
|
|
mg_call(c, MG_EV_CONNECT, NULL);
|
|
|
|
} else if (mg_sock_would_block()) {
|
|
|
|
c->is_connecting = 1;
|
2021-05-29 01:30:42 +08:00
|
|
|
} else {
|
|
|
|
mg_error(c, "connect: %d", MG_SOCK_ERRNO);
|
2016-11-12 04:34:00 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_connection *mg_connect(struct mg_mgr *mgr, const char *url,
|
|
|
|
mg_event_handler_t fn, void *fn_data) {
|
|
|
|
struct mg_connection *c = NULL;
|
2020-12-21 20:26:44 +08:00
|
|
|
if ((c = alloc_conn(mgr, 1, INVALID_SOCKET)) == NULL) {
|
|
|
|
LOG(LL_ERROR, ("OOM"));
|
2020-12-05 19:26:32 +08:00
|
|
|
} else {
|
|
|
|
struct mg_str host = mg_url_host(url);
|
|
|
|
LIST_ADD_HEAD(struct mg_connection, &mgr->conns, c);
|
2020-12-21 20:26:44 +08:00
|
|
|
c->is_udp = (strncmp(url, "udp:", 4) == 0);
|
2020-12-05 19:26:32 +08:00
|
|
|
c->peer.port = mg_htons(mg_url_port(url));
|
|
|
|
c->fn = fn;
|
|
|
|
c->fn_data = fn_data;
|
2020-12-21 20:26:44 +08:00
|
|
|
LOG(LL_DEBUG, ("%lu -> %s", c->id, url));
|
2020-12-22 17:44:59 +08:00
|
|
|
mg_resolve(c, &host, mgr->dnstimeout);
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
return c;
|
2018-03-27 07:25:18 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
static void accept_conn(struct mg_mgr *mgr, struct mg_connection *lsn) {
|
|
|
|
struct mg_connection *c = NULL;
|
|
|
|
union usa usa;
|
2021-01-25 18:35:45 +08:00
|
|
|
socklen_t sa_len = sizeof(usa);
|
2020-12-05 19:26:32 +08:00
|
|
|
SOCKET fd = accept(FD(lsn), &usa.sa, &sa_len);
|
|
|
|
if (fd == INVALID_SOCKET) {
|
2021-09-14 14:28:28 +08:00
|
|
|
#if MG_ARCH == MG_ARCH_AZURERTOS
|
|
|
|
// AzureRTOS, in non-block socket mode can mark listening socket readable
|
|
|
|
// even it is not. See comment for 'select' func implementation in nx_bsd.c
|
|
|
|
// That's not an error, just should try later
|
|
|
|
if (MG_SOCK_ERRNO != EAGAIN)
|
|
|
|
#endif
|
2020-12-21 20:26:44 +08:00
|
|
|
LOG(LL_ERROR, ("%lu accept failed, errno %d", lsn->id, MG_SOCK_ERRNO));
|
2021-05-19 07:00:32 +08:00
|
|
|
#if (!defined(_WIN32) && (MG_ARCH != MG_ARCH_FREERTOS_TCP))
|
2021-05-12 15:43:34 +08:00
|
|
|
} else if ((long) fd >= FD_SETSIZE) {
|
2020-12-05 19:26:32 +08:00
|
|
|
LOG(LL_ERROR, ("%ld > %ld", (long) fd, (long) FD_SETSIZE));
|
|
|
|
closesocket(fd);
|
|
|
|
#endif
|
|
|
|
} else if ((c = alloc_conn(mgr, 0, fd)) == NULL) {
|
2020-12-21 20:26:44 +08:00
|
|
|
LOG(LL_ERROR, ("%lu OOM", lsn->id));
|
2020-12-05 19:26:32 +08:00
|
|
|
closesocket(fd);
|
|
|
|
} else {
|
|
|
|
char buf[40];
|
2021-08-08 00:22:47 +08:00
|
|
|
tomgaddr(&usa, &c->peer, sa_len != sizeof(usa.sin));
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_straddr(c, buf, sizeof(buf));
|
2020-12-21 20:26:44 +08:00
|
|
|
LOG(LL_DEBUG, ("%lu accepted %s", c->id, buf));
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_set_non_blocking_mode(FD(c));
|
|
|
|
setsockopts(c);
|
|
|
|
LIST_ADD_HEAD(struct mg_connection, &mgr->conns, c);
|
|
|
|
c->is_accepted = 1;
|
|
|
|
c->is_hexdumping = lsn->is_hexdumping;
|
|
|
|
c->pfn = lsn->pfn;
|
|
|
|
c->pfn_data = lsn->pfn_data;
|
|
|
|
c->fn = lsn->fn;
|
|
|
|
c->fn_data = lsn->fn_data;
|
|
|
|
mg_call(c, MG_EV_ACCEPT, NULL);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-08-08 00:22:47 +08:00
|
|
|
static bool mg_socketpair(SOCKET sp[2], union usa usa[2]) {
|
|
|
|
socklen_t n = sizeof(usa[0].sin);
|
|
|
|
bool result = false;
|
2018-03-27 07:25:18 +08:00
|
|
|
|
2021-08-08 00:22:47 +08:00
|
|
|
(void) memset(&usa[0], 0, sizeof(usa[0]));
|
|
|
|
usa[0].sin.sin_family = AF_INET;
|
2021-08-08 00:48:51 +08:00
|
|
|
*(uint32_t *) &usa->sin.sin_addr = mg_htonl(0x7f000001); // 127.0.0.1
|
2021-08-08 00:22:47 +08:00
|
|
|
usa[1] = usa[0];
|
2018-03-27 07:25:18 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
if ((sp[0] = socket(AF_INET, SOCK_DGRAM, 0)) != INVALID_SOCKET &&
|
|
|
|
(sp[1] = socket(AF_INET, SOCK_DGRAM, 0)) != INVALID_SOCKET &&
|
2021-08-08 00:22:47 +08:00
|
|
|
bind(sp[0], &usa[0].sa, n) == 0 && bind(sp[1], &usa[1].sa, n) == 0 &&
|
|
|
|
getsockname(sp[0], &usa[0].sa, &n) == 0 &&
|
|
|
|
getsockname(sp[1], &usa[1].sa, &n) == 0 &&
|
|
|
|
connect(sp[0], &usa[1].sa, n) == 0 &&
|
|
|
|
connect(sp[1], &usa[0].sa, n) == 0) {
|
|
|
|
mg_set_non_blocking_mode(sp[1]); // Set close-on-exec
|
|
|
|
result = true;
|
2020-12-05 19:26:32 +08:00
|
|
|
} else {
|
|
|
|
if (sp[0] != INVALID_SOCKET) closesocket(sp[0]);
|
|
|
|
if (sp[1] != INVALID_SOCKET) closesocket(sp[1]);
|
2021-08-08 00:22:47 +08:00
|
|
|
sp[0] = sp[1] = INVALID_SOCKET;
|
2016-11-26 03:20:59 +08:00
|
|
|
}
|
|
|
|
|
2021-08-08 00:22:47 +08:00
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
2021-08-12 02:17:04 +08:00
|
|
|
void mg_mgr_wakeup(struct mg_connection *c) {
|
|
|
|
LOG(LL_INFO, ("skt: %p", c->pfn_data));
|
|
|
|
send((SOCKET) (size_t) c->pfn_data, "\x01", 1, MSG_NONBLOCKING);
|
2021-08-08 00:22:47 +08:00
|
|
|
}
|
|
|
|
|
2021-08-12 02:17:04 +08:00
|
|
|
static void pf1(struct mg_connection *c, int ev, void *ev_data, void *fn_data) {
|
|
|
|
if (ev == MG_EV_READ) mg_iobuf_free(&c->recv);
|
|
|
|
(void) ev_data, (void) fn_data;
|
2021-08-08 00:22:47 +08:00
|
|
|
}
|
|
|
|
|
2021-08-12 02:17:04 +08:00
|
|
|
struct mg_connection *mg_mkpipe(struct mg_mgr *mgr, mg_event_handler_t fn,
|
|
|
|
void *fn_data) {
|
2021-08-08 00:22:47 +08:00
|
|
|
union usa usa[2];
|
|
|
|
SOCKET sp[2] = {INVALID_SOCKET, INVALID_SOCKET};
|
2021-08-12 02:17:04 +08:00
|
|
|
struct mg_connection *c = NULL;
|
|
|
|
if (!mg_socketpair(sp, usa)) {
|
|
|
|
LOG(LL_ERROR, ("Cannot create socket pair"));
|
|
|
|
} else if ((c = alloc_conn(mgr, false, sp[1])) == NULL) {
|
|
|
|
closesocket(sp[0]);
|
|
|
|
closesocket(sp[1]);
|
|
|
|
LOG(LL_ERROR, ("OOM"));
|
|
|
|
} else {
|
|
|
|
LOG(LL_INFO, ("pipe %lu", (unsigned long) sp[0]));
|
|
|
|
tomgaddr(&usa[0], &c->peer, false);
|
|
|
|
c->is_udp = 1;
|
|
|
|
c->pfn = pf1;
|
|
|
|
c->pfn_data = (void *) (size_t) sp[0];
|
|
|
|
c->fn = fn;
|
|
|
|
c->fn_data = fn_data;
|
|
|
|
LIST_ADD_HEAD(struct mg_connection, &mgr->conns, c);
|
|
|
|
}
|
|
|
|
return c;
|
2016-11-26 03:20:59 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_connection *mg_listen(struct mg_mgr *mgr, const char *url,
|
|
|
|
mg_event_handler_t fn, void *fn_data) {
|
|
|
|
struct mg_connection *c = NULL;
|
2021-05-29 01:30:42 +08:00
|
|
|
bool is_udp = strncmp(url, "udp:", 4) == 0;
|
2021-07-19 15:47:38 +08:00
|
|
|
struct mg_addr addr;
|
|
|
|
SOCKET fd = mg_open_listener(url, &addr);
|
2020-12-05 19:26:32 +08:00
|
|
|
if (fd == INVALID_SOCKET) {
|
2021-08-08 05:55:13 +08:00
|
|
|
LOG(LL_ERROR, ("Failed: %s, errno %d", url, MG_SOCK_ERRNO));
|
2020-12-05 19:26:32 +08:00
|
|
|
} else if ((c = alloc_conn(mgr, 0, fd)) == NULL) {
|
|
|
|
LOG(LL_ERROR, ("OOM %s", url));
|
|
|
|
closesocket(fd);
|
|
|
|
} else {
|
2021-07-19 15:47:38 +08:00
|
|
|
memcpy(&c->peer, &addr, sizeof(struct mg_addr));
|
2021-08-01 05:26:43 +08:00
|
|
|
c->fd = S2PTR(fd);
|
2020-12-05 19:26:32 +08:00
|
|
|
c->is_listening = 1;
|
|
|
|
c->is_udp = is_udp;
|
|
|
|
LIST_ADD_HEAD(struct mg_connection, &mgr->conns, c);
|
|
|
|
c->fn = fn;
|
|
|
|
c->fn_data = fn_data;
|
2021-07-19 15:47:38 +08:00
|
|
|
LOG(LL_INFO,
|
|
|
|
("%lu accepting on %s (port %u)", c->id, url, mg_ntohs(c->peer.port)));
|
2016-11-26 03:20:59 +08:00
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
return c;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void mg_iotest(struct mg_mgr *mgr, int ms) {
|
2021-05-11 16:12:06 +08:00
|
|
|
#if MG_ARCH == MG_ARCH_FREERTOS_TCP
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_connection *c;
|
|
|
|
for (c = mgr->conns; c != NULL; c = c->next) {
|
2021-05-19 07:00:32 +08:00
|
|
|
if (c->is_closing || c->is_resolving || FD(c) == INVALID_SOCKET) continue;
|
|
|
|
FreeRTOS_FD_SET(c->fd, mgr->ss, eSELECT_READ | eSELECT_EXCEPT);
|
2020-12-05 19:26:32 +08:00
|
|
|
if (c->is_connecting || (c->send.len > 0 && c->is_tls_hs == 0))
|
|
|
|
FreeRTOS_FD_SET(c->fd, mgr->ss, eSELECT_WRITE);
|
2016-11-26 03:20:59 +08:00
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
FreeRTOS_select(mgr->ss, pdMS_TO_TICKS(ms));
|
|
|
|
for (c = mgr->conns; c != NULL; c = c->next) {
|
|
|
|
EventBits_t bits = FreeRTOS_FD_ISSET(c->fd, mgr->ss);
|
|
|
|
c->is_readable = bits & (eSELECT_READ | eSELECT_EXCEPT) ? 1 : 0;
|
|
|
|
c->is_writable = bits & eSELECT_WRITE ? 1 : 0;
|
2021-05-19 07:00:32 +08:00
|
|
|
FreeRTOS_FD_CLR(c->fd, mgr->ss,
|
|
|
|
eSELECT_READ | eSELECT_EXCEPT | eSELECT_WRITE);
|
2016-11-26 03:20:59 +08:00
|
|
|
}
|
|
|
|
#else
|
2020-12-05 19:26:32 +08:00
|
|
|
struct timeval tv = {ms / 1000, (ms % 1000) * 1000};
|
|
|
|
struct mg_connection *c;
|
|
|
|
fd_set rset, wset;
|
|
|
|
SOCKET maxfd = 0;
|
|
|
|
int rc;
|
|
|
|
|
|
|
|
FD_ZERO(&rset);
|
|
|
|
FD_ZERO(&wset);
|
|
|
|
|
|
|
|
for (c = mgr->conns; c != NULL; c = c->next) {
|
|
|
|
if (c->is_closing || c->is_resolving || FD(c) == INVALID_SOCKET) continue;
|
|
|
|
FD_SET(FD(c), &rset);
|
|
|
|
if (FD(c) > maxfd) maxfd = FD(c);
|
|
|
|
if (c->is_connecting || (c->send.len > 0 && c->is_tls_hs == 0))
|
|
|
|
FD_SET(FD(c), &wset);
|
|
|
|
}
|
|
|
|
|
2021-08-01 05:26:43 +08:00
|
|
|
if ((rc = select((int) maxfd + 1, &rset, &wset, NULL, &tv)) < 0) {
|
2020-12-05 19:26:32 +08:00
|
|
|
LOG(LL_DEBUG, ("select: %d %d", rc, MG_SOCK_ERRNO));
|
|
|
|
FD_ZERO(&rset);
|
|
|
|
FD_ZERO(&wset);
|
|
|
|
}
|
|
|
|
|
|
|
|
for (c = mgr->conns; c != NULL; c = c->next) {
|
|
|
|
// TLS might have stuff buffered, so dig everything
|
|
|
|
c->is_readable = c->is_tls && c->is_readable
|
|
|
|
? 1
|
|
|
|
: FD(c) != INVALID_SOCKET && FD_ISSET(FD(c), &rset);
|
|
|
|
c->is_writable = FD(c) != INVALID_SOCKET && FD_ISSET(FD(c), &wset);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
2020-12-12 01:35:58 +08:00
|
|
|
static void connect_conn(struct mg_connection *c) {
|
|
|
|
int rc = 0;
|
2021-05-12 15:43:34 +08:00
|
|
|
#if MG_ARCH != MG_ARCH_FREERTOS_TCP
|
2020-12-12 01:35:58 +08:00
|
|
|
socklen_t len = sizeof(rc);
|
|
|
|
if (getsockopt(FD(c), SOL_SOCKET, SO_ERROR, (char *) &rc, &len)) rc = 1;
|
2021-05-12 15:43:34 +08:00
|
|
|
#endif
|
2020-12-12 01:35:58 +08:00
|
|
|
if (rc == EAGAIN || rc == EWOULDBLOCK) rc = 0;
|
|
|
|
c->is_connecting = 0;
|
|
|
|
if (rc) {
|
2020-12-12 06:58:50 +08:00
|
|
|
char buf[40];
|
|
|
|
mg_error(c, "error connecting to %s", mg_straddr(c, buf, sizeof(buf)));
|
2020-12-12 01:35:58 +08:00
|
|
|
} else {
|
2020-12-18 23:13:19 +08:00
|
|
|
if (c->is_tls_hs) mg_tls_handshake(c);
|
|
|
|
mg_call(c, MG_EV_CONNECT, NULL);
|
2020-12-12 01:35:58 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
void mg_mgr_poll(struct mg_mgr *mgr, int ms) {
|
|
|
|
struct mg_connection *c, *tmp;
|
|
|
|
unsigned long now;
|
|
|
|
|
|
|
|
mg_iotest(mgr, ms);
|
|
|
|
now = mg_millis();
|
|
|
|
mg_timer_poll(now);
|
|
|
|
|
|
|
|
for (c = mgr->conns; c != NULL; c = tmp) {
|
|
|
|
tmp = c->next;
|
|
|
|
mg_call(c, MG_EV_POLL, &now);
|
|
|
|
LOG(LL_VERBOSE_DEBUG,
|
2020-12-21 20:26:44 +08:00
|
|
|
("%lu %c%c %c%c%c%c%c", c->id, c->is_readable ? 'r' : '-',
|
2020-12-05 19:26:32 +08:00
|
|
|
c->is_writable ? 'w' : '-', c->is_tls ? 'T' : 't',
|
|
|
|
c->is_connecting ? 'C' : 'c', c->is_tls_hs ? 'H' : 'h',
|
|
|
|
c->is_resolving ? 'R' : 'r', c->is_closing ? 'C' : 'c'));
|
|
|
|
if (c->is_resolving || c->is_closing) {
|
|
|
|
// Do nothing
|
2020-12-28 13:25:29 +08:00
|
|
|
} else if (c->is_listening && c->is_udp == 0) {
|
2020-12-05 19:26:32 +08:00
|
|
|
if (c->is_readable) accept_conn(mgr, c);
|
|
|
|
} else if (c->is_connecting) {
|
2020-12-18 22:29:30 +08:00
|
|
|
if (c->is_readable || c->is_writable) connect_conn(c);
|
2020-12-05 19:26:32 +08:00
|
|
|
} else if (c->is_tls_hs) {
|
2020-12-18 23:13:19 +08:00
|
|
|
if ((c->is_readable || c->is_writable)) mg_tls_handshake(c);
|
2020-12-05 19:26:32 +08:00
|
|
|
} else {
|
2021-05-29 01:30:42 +08:00
|
|
|
if (c->is_readable) read_conn(c);
|
2020-12-05 19:26:32 +08:00
|
|
|
if (c->is_writable) write_conn(c);
|
2016-11-15 16:59:04 +08:00
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
|
|
|
|
if (c->is_draining && c->send.len == 0) c->is_closing = 1;
|
2020-12-22 17:44:59 +08:00
|
|
|
if (c->is_closing) close_conn(c);
|
2018-03-27 07:25:18 +08:00
|
|
|
}
|
2016-11-12 04:34:00 +08:00
|
|
|
}
|
2017-01-11 20:25:57 +08:00
|
|
|
#endif
|
2016-10-12 02:20:28 +08:00
|
|
|
|
2021-01-03 01:57:51 +08:00
|
|
|
#ifdef MG_ENABLE_LINES
|
|
|
|
#line 1 "src/ssi.c"
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
|
|
|
2021-02-08 06:43:49 +08:00
|
|
|
|
2021-01-03 01:57:51 +08:00
|
|
|
#ifndef MG_MAX_SSI_DEPTH
|
|
|
|
#define MG_MAX_SSI_DEPTH 5
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#if MG_ENABLE_SSI
|
|
|
|
static char *mg_ssi(const char *path, const char *root, int depth) {
|
|
|
|
struct mg_iobuf b = {NULL, 0, 0};
|
2021-07-23 05:46:33 +08:00
|
|
|
FILE *fp = fopen(path, "rb");
|
2021-01-03 01:57:51 +08:00
|
|
|
if (fp != NULL) {
|
|
|
|
char buf[BUFSIZ], arg[sizeof(buf)];
|
|
|
|
int ch, intag = 0;
|
|
|
|
size_t len = 0, align = MG_IO_SIZE;
|
|
|
|
while ((ch = fgetc(fp)) != EOF) {
|
|
|
|
if (intag && ch == '>' && buf[len - 1] == '-' && buf[len - 2] == '-') {
|
2021-05-29 01:30:42 +08:00
|
|
|
buf[len++] = (char) (ch & 0xff);
|
2021-01-03 01:57:51 +08:00
|
|
|
if (sscanf(buf, "<!--#include file=\"%[^\"]", arg)) {
|
2021-03-17 20:22:19 +08:00
|
|
|
char tmp[MG_PATH_MAX + BUFSIZ + 10],
|
|
|
|
*p = (char *) path + strlen(path), *data;
|
2021-01-03 01:57:51 +08:00
|
|
|
while (p > path && p[-1] != MG_DIRSEP && p[-1] != '/') p--;
|
|
|
|
snprintf(tmp, sizeof(tmp), "%.*s%s", (int) (p - path), path, arg);
|
|
|
|
if (depth < MG_MAX_SSI_DEPTH &&
|
|
|
|
(data = mg_ssi(tmp, root, depth + 1)) != NULL) {
|
2021-08-28 15:08:54 +08:00
|
|
|
mg_iobuf_add(&b, b.len, data, strlen(data), align);
|
2021-01-03 01:57:51 +08:00
|
|
|
free(data);
|
|
|
|
} else {
|
|
|
|
LOG(LL_ERROR, ("%s: file=%s error or too deep", path, arg));
|
|
|
|
}
|
|
|
|
} else if (sscanf(buf, "<!--#include virtual=\"%[^\"]", arg)) {
|
2021-03-17 20:22:19 +08:00
|
|
|
char tmp[MG_PATH_MAX + BUFSIZ + 10], *data;
|
2021-01-03 01:57:51 +08:00
|
|
|
snprintf(tmp, sizeof(tmp), "%s%s", root, arg);
|
|
|
|
if (depth < MG_MAX_SSI_DEPTH &&
|
|
|
|
(data = mg_ssi(tmp, root, depth + 1)) != NULL) {
|
2021-08-28 15:08:54 +08:00
|
|
|
mg_iobuf_add(&b, b.len, data, strlen(data), align);
|
2021-01-03 01:57:51 +08:00
|
|
|
free(data);
|
|
|
|
} else {
|
|
|
|
LOG(LL_ERROR, ("%s: virtual=%s error or too deep", path, arg));
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
// Unknown SSI tag
|
|
|
|
LOG(LL_INFO, ("Unknown SSI tag: %.*s", (int) len, buf));
|
2021-08-28 15:08:54 +08:00
|
|
|
mg_iobuf_add(&b, b.len, buf, len, align);
|
2021-01-03 01:57:51 +08:00
|
|
|
}
|
|
|
|
intag = 0;
|
|
|
|
len = 0;
|
|
|
|
} else if (ch == '<') {
|
|
|
|
intag = 1;
|
2021-08-28 15:08:54 +08:00
|
|
|
if (len > 0) mg_iobuf_add(&b, b.len, buf, len, align);
|
2021-01-03 01:57:51 +08:00
|
|
|
len = 0;
|
2021-05-29 01:30:42 +08:00
|
|
|
buf[len++] = (char) (ch & 0xff);
|
2021-01-03 01:57:51 +08:00
|
|
|
} else if (intag) {
|
|
|
|
if (len == 5 && strncmp(buf, "<!--#", 5) != 0) {
|
|
|
|
intag = 0;
|
|
|
|
} else if (len >= sizeof(buf) - 2) {
|
|
|
|
LOG(LL_ERROR, ("%s: SSI tag is too large", path));
|
|
|
|
len = 0;
|
|
|
|
}
|
2021-05-29 01:30:42 +08:00
|
|
|
buf[len++] = (char) (ch & 0xff);
|
2021-01-03 01:57:51 +08:00
|
|
|
} else {
|
2021-05-29 01:30:42 +08:00
|
|
|
buf[len++] = (char) (ch & 0xff);
|
2021-01-03 01:57:51 +08:00
|
|
|
if (len >= sizeof(buf)) {
|
2021-08-28 15:08:54 +08:00
|
|
|
mg_iobuf_add(&b, b.len, buf, len, align);
|
2021-01-03 01:57:51 +08:00
|
|
|
len = 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2021-08-28 15:08:54 +08:00
|
|
|
if (len > 0) mg_iobuf_add(&b, b.len, buf, len, align);
|
|
|
|
if (b.len > 0) mg_iobuf_add(&b, b.len, "", 1, align); // nul-terminate
|
2021-01-03 01:57:51 +08:00
|
|
|
fclose(fp);
|
|
|
|
}
|
|
|
|
(void) depth;
|
|
|
|
(void) root;
|
|
|
|
return (char *) b.buf;
|
|
|
|
}
|
|
|
|
|
|
|
|
void mg_http_serve_ssi(struct mg_connection *c, const char *root,
|
|
|
|
const char *fullpath) {
|
2021-07-20 18:33:38 +08:00
|
|
|
const char *headers = "Content-Type: text/html; charset=utf-8\r\n";
|
2021-01-03 01:57:51 +08:00
|
|
|
char *data = mg_ssi(fullpath, root, 0);
|
2021-07-20 18:33:38 +08:00
|
|
|
mg_http_reply(c, 200, headers, "%s", data == NULL ? "" : data);
|
2021-01-03 01:57:51 +08:00
|
|
|
free(data);
|
|
|
|
}
|
2021-07-29 21:21:20 +08:00
|
|
|
#else
|
|
|
|
void mg_http_serve_ssi(struct mg_connection *c, const char *root,
|
|
|
|
const char *fullpath) {
|
|
|
|
mg_http_reply(c, 501, NULL, "SSI not enabled");
|
|
|
|
(void) root, (void) fullpath;
|
|
|
|
}
|
2021-01-03 01:57:51 +08:00
|
|
|
#endif
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
#ifdef MG_ENABLE_LINES
|
|
|
|
#line 1 "src/str.c"
|
2017-05-10 00:07:00 +08:00
|
|
|
#endif
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
#include <stdlib.h>
|
2016-10-20 03:12:12 +08:00
|
|
|
|
2021-07-15 09:18:32 +08:00
|
|
|
struct mg_str mg_str_s(const char *s) {
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_str str = {s, s == NULL ? 0 : strlen(s)};
|
|
|
|
return str;
|
2019-04-15 22:22:03 +08:00
|
|
|
}
|
2016-10-20 03:12:12 +08:00
|
|
|
|
2020-12-07 11:47:37 +08:00
|
|
|
struct mg_str mg_str_n(const char *s, size_t n) {
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_str str = {s, n};
|
|
|
|
return str;
|
|
|
|
}
|
2016-11-09 06:29:07 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
int mg_lower(const char *s) {
|
|
|
|
return tolower(*(const unsigned char *) s);
|
|
|
|
}
|
2017-02-16 23:59:26 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
int mg_ncasecmp(const char *s1, const char *s2, size_t len) {
|
|
|
|
int diff = 0;
|
|
|
|
if (len > 0) do {
|
|
|
|
diff = mg_lower(s1++) - mg_lower(s2++);
|
|
|
|
} while (diff == 0 && s1[-1] != '\0' && --len > 0);
|
|
|
|
return diff;
|
|
|
|
}
|
2017-02-09 01:34:28 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
int mg_casecmp(const char *s1, const char *s2) {
|
|
|
|
return mg_ncasecmp(s1, s2, (size_t) ~0);
|
2016-10-12 02:20:28 +08:00
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
|
|
|
|
int mg_vcmp(const struct mg_str *s1, const char *s2) {
|
|
|
|
size_t n2 = strlen(s2), n1 = s1->len;
|
|
|
|
int r = strncmp(s1->ptr, s2, (n1 < n2) ? n1 : n2);
|
|
|
|
if (r == 0) return (int) (n1 - n2);
|
|
|
|
return r;
|
2016-10-12 02:20:28 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
int mg_vcasecmp(const struct mg_str *str1, const char *str2) {
|
|
|
|
size_t n2 = strlen(str2), n1 = str1->len;
|
|
|
|
int r = mg_ncasecmp(str1->ptr, str2, (n1 < n2) ? n1 : n2);
|
|
|
|
if (r == 0) return (int) (n1 - n2);
|
|
|
|
return r;
|
2016-10-12 02:20:28 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_str mg_strdup(const struct mg_str s) {
|
|
|
|
struct mg_str r = {NULL, 0};
|
|
|
|
if (s.len > 0 && s.ptr != NULL) {
|
2021-05-18 00:36:57 +08:00
|
|
|
char *sc = (char *) calloc(1, s.len + 1);
|
2020-12-05 19:26:32 +08:00
|
|
|
if (sc != NULL) {
|
|
|
|
memcpy(sc, s.ptr, s.len);
|
|
|
|
sc[s.len] = '\0';
|
|
|
|
r.ptr = sc;
|
|
|
|
r.len = s.len;
|
2017-02-09 01:34:28 +08:00
|
|
|
}
|
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
return r;
|
2016-10-12 02:20:28 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
int mg_strcmp(const struct mg_str str1, const struct mg_str str2) {
|
|
|
|
size_t i = 0;
|
|
|
|
while (i < str1.len && i < str2.len) {
|
|
|
|
int c1 = str1.ptr[i];
|
|
|
|
int c2 = str2.ptr[i];
|
|
|
|
if (c1 < c2) return -1;
|
|
|
|
if (c1 > c2) return 1;
|
|
|
|
i++;
|
2018-09-28 18:47:58 +08:00
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
if (i < str1.len) return 1;
|
|
|
|
if (i < str2.len) return -1;
|
|
|
|
return 0;
|
2016-10-12 02:20:28 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
const char *mg_strstr(const struct mg_str haystack,
|
|
|
|
const struct mg_str needle) {
|
|
|
|
size_t i;
|
|
|
|
if (needle.len > haystack.len) return NULL;
|
|
|
|
for (i = 0; i <= haystack.len - needle.len; i++) {
|
|
|
|
if (memcmp(haystack.ptr + i, needle.ptr, needle.len) == 0) {
|
|
|
|
return haystack.ptr + i;
|
|
|
|
}
|
2016-10-12 02:20:28 +08:00
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
return NULL;
|
2016-10-12 02:20:28 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_str mg_strstrip(struct mg_str s) {
|
|
|
|
while (s.len > 0 && isspace((int) *s.ptr)) s.ptr++, s.len--;
|
|
|
|
while (s.len > 0 && isspace((int) *(s.ptr + s.len - 1))) s.len--;
|
|
|
|
return s;
|
2017-05-10 00:07:00 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
#ifdef MG_ENABLE_LINES
|
|
|
|
#line 1 "src/timer.c"
|
2017-05-10 00:07:00 +08:00
|
|
|
#endif
|
2020-12-05 19:26:32 +08:00
|
|
|
// Copyright (c) Cesanta Software Limited
|
|
|
|
// All rights reserved
|
2017-02-09 01:34:28 +08:00
|
|
|
|
2016-10-12 02:20:28 +08:00
|
|
|
|
|
|
|
|
2017-05-10 00:07:00 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_timer *g_timers;
|
|
|
|
|
2021-05-29 01:30:42 +08:00
|
|
|
void mg_timer_init(struct mg_timer *t, unsigned long ms, unsigned flags,
|
|
|
|
void (*fn)(void *), void *arg) {
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_timer tmp = {ms, flags, fn, arg, 0UL, g_timers};
|
|
|
|
*t = tmp;
|
|
|
|
g_timers = t;
|
|
|
|
if (flags & MG_TIMER_RUN_NOW) fn(arg);
|
2017-05-10 00:07:00 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
void mg_timer_free(struct mg_timer *t) {
|
|
|
|
struct mg_timer **head = &g_timers;
|
|
|
|
while (*head && *head != t) head = &(*head)->next;
|
|
|
|
if (*head) *head = t->next;
|
2017-02-16 20:07:06 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
void mg_timer_poll(unsigned long now_ms) {
|
|
|
|
// If time goes back (wrapped around), reset timers
|
|
|
|
struct mg_timer *t, *tmp;
|
|
|
|
static unsigned long oldnow; // Timestamp in a previous invocation
|
|
|
|
if (oldnow > now_ms) { // If it is wrapped, reset timers
|
|
|
|
for (t = g_timers; t != NULL; t = t->next) t->expire = 0;
|
2017-06-21 02:25:43 +08:00
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
oldnow = now_ms;
|
|
|
|
|
|
|
|
for (t = g_timers; t != NULL; t = tmp) {
|
|
|
|
tmp = t->next;
|
|
|
|
if (t->expire == 0) t->expire = now_ms + t->period_ms;
|
|
|
|
if (t->expire > now_ms) continue;
|
|
|
|
t->fn(t->arg);
|
|
|
|
// Try to tick timers with the given period as accurate as possible,
|
|
|
|
// even if this polling function is called with some random period.
|
|
|
|
t->expire = now_ms - t->expire > (unsigned long) t->period_ms
|
|
|
|
? now_ms + t->period_ms
|
|
|
|
: t->expire + t->period_ms;
|
|
|
|
if (!(t->flags & MG_TIMER_REPEAT)) mg_timer_free(t);
|
2016-10-12 02:20:28 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
#ifdef MG_ENABLE_LINES
|
|
|
|
#line 1 "src/tls.c"
|
|
|
|
#endif
|
2017-05-10 00:07:00 +08:00
|
|
|
|
2016-10-12 02:20:28 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
#if MG_ENABLE_MBEDTLS ///////////////////////////////////////// MBEDTLS
|
2017-05-10 00:07:00 +08:00
|
|
|
|
|
|
|
|
|
|
|
|
2016-10-12 02:20:28 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
#include <mbedtls/debug.h>
|
|
|
|
#include <mbedtls/ssl.h>
|
2020-12-07 13:46:47 +08:00
|
|
|
|
2021-01-25 18:08:15 +08:00
|
|
|
#ifndef EXTERN_C
|
2020-12-07 13:46:47 +08:00
|
|
|
#ifdef __cplusplus
|
|
|
|
#define EXTERN_C extern "C"
|
2020-12-07 13:18:07 +08:00
|
|
|
#else
|
2020-12-07 13:46:47 +08:00
|
|
|
#define EXTERN_C
|
2020-12-07 12:56:50 +08:00
|
|
|
#endif
|
2021-01-25 18:08:15 +08:00
|
|
|
#endif
|
2017-04-12 23:43:11 +08:00
|
|
|
|
2021-07-14 00:28:07 +08:00
|
|
|
#if MBEDTLS_VERSION_NUMBER >= 0x03000000
|
|
|
|
#define RNG , rng_get, NULL
|
|
|
|
#else
|
|
|
|
#define RNG
|
|
|
|
#endif
|
|
|
|
|
2021-01-28 16:45:24 +08:00
|
|
|
// Different versions have those in different files, so declare here
|
2020-12-07 13:46:47 +08:00
|
|
|
EXTERN_C int mbedtls_net_recv(void *, unsigned char *, size_t);
|
|
|
|
EXTERN_C int mbedtls_net_send(void *, const unsigned char *, size_t);
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_tls {
|
|
|
|
char *cafile; // CA certificate path
|
|
|
|
mbedtls_x509_crt ca; // Parsed CA certificate
|
|
|
|
mbedtls_x509_crt cert; // Parsed certificate
|
|
|
|
mbedtls_ssl_context ssl; // SSL/TLS context
|
|
|
|
mbedtls_ssl_config conf; // SSL-TLS config
|
|
|
|
mbedtls_pk_context pk; // Private key context
|
2017-05-10 00:07:00 +08:00
|
|
|
};
|
|
|
|
|
2021-05-29 01:30:42 +08:00
|
|
|
void mg_tls_handshake(struct mg_connection *c) {
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_tls *tls = (struct mg_tls *) c->tls;
|
|
|
|
int rc;
|
|
|
|
mbedtls_ssl_set_bio(&tls->ssl, &c->fd, mbedtls_net_send, mbedtls_net_recv, 0);
|
|
|
|
rc = mbedtls_ssl_handshake(&tls->ssl);
|
|
|
|
if (rc == 0) { // Success
|
2020-12-21 20:26:44 +08:00
|
|
|
LOG(LL_DEBUG, ("%lu success", c->id));
|
2020-12-18 23:13:19 +08:00
|
|
|
c->is_tls_hs = 0;
|
2020-12-05 19:26:32 +08:00
|
|
|
} else if (rc == MBEDTLS_ERR_SSL_WANT_READ ||
|
|
|
|
rc == MBEDTLS_ERR_SSL_WANT_WRITE) { // Still pending
|
2020-12-21 20:26:44 +08:00
|
|
|
LOG(LL_VERBOSE_DEBUG, ("%lu pending, %d%d %d (-%#x)", c->id,
|
2020-12-18 22:29:30 +08:00
|
|
|
c->is_connecting, c->is_tls_hs, rc, -rc));
|
2020-12-05 19:26:32 +08:00
|
|
|
} else {
|
|
|
|
mg_error(c, "TLS handshake: -%#x", -rc); // Error
|
2016-10-12 02:20:28 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
static int mbed_rng(void *ctx, unsigned char *buf, size_t len) {
|
|
|
|
mg_random(buf, len);
|
|
|
|
(void) ctx;
|
|
|
|
return 0;
|
2016-10-12 02:20:28 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
static void debug_cb(void *c, int lev, const char *s, int n, const char *s2) {
|
2021-05-29 01:30:42 +08:00
|
|
|
n = (int) strlen(s2) - 1;
|
2020-12-05 19:26:32 +08:00
|
|
|
LOG(LL_VERBOSE_DEBUG, ("%p %.*s", ((struct mg_connection *) c)->fd, n, s2));
|
|
|
|
(void) s;
|
|
|
|
(void) c;
|
|
|
|
(void) lev;
|
|
|
|
}
|
|
|
|
|
2021-07-14 00:28:07 +08:00
|
|
|
#if MBEDTLS_VERSION_NUMBER >= 0x03000000
|
|
|
|
static int rng_get(void *p_rng, unsigned char *buf, size_t len) {
|
|
|
|
(void) p_rng;
|
|
|
|
mg_random(buf, len);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2021-05-29 01:30:42 +08:00
|
|
|
void mg_tls_init(struct mg_connection *c, struct mg_tls_opts *opts) {
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_tls *tls = (struct mg_tls *) calloc(1, sizeof(*tls));
|
|
|
|
int rc = 0;
|
2021-08-29 23:53:29 +08:00
|
|
|
const char *ca = opts->ca == NULL ? "-"
|
|
|
|
: opts->ca[0] == '-' ? "(emb)"
|
|
|
|
: opts->ca;
|
|
|
|
const char *cert = opts->cert == NULL ? "-"
|
|
|
|
: opts->cert[0] == '-' ? "(emb)"
|
|
|
|
: opts->cert;
|
|
|
|
const char *certkey = opts->certkey == NULL ? "-"
|
|
|
|
: opts->certkey[0] == '-' ? "(emb)"
|
|
|
|
: opts->certkey;
|
2021-01-26 20:16:58 +08:00
|
|
|
if (tls == NULL) {
|
|
|
|
mg_error(c, "TLS OOM");
|
|
|
|
goto fail;
|
|
|
|
}
|
2021-08-01 05:26:43 +08:00
|
|
|
LOG(LL_DEBUG,
|
|
|
|
("%lu Setting TLS, CA: %s, cert: %s, key: %s", c->id, ca, cert, certkey));
|
2020-12-05 19:26:32 +08:00
|
|
|
mbedtls_ssl_init(&tls->ssl);
|
|
|
|
mbedtls_ssl_config_init(&tls->conf);
|
2021-04-22 22:21:16 +08:00
|
|
|
mbedtls_x509_crt_init(&tls->ca);
|
|
|
|
mbedtls_x509_crt_init(&tls->cert);
|
|
|
|
mbedtls_pk_init(&tls->pk);
|
2020-12-05 19:26:32 +08:00
|
|
|
mbedtls_ssl_conf_dbg(&tls->conf, debug_cb, c);
|
2021-04-22 22:21:16 +08:00
|
|
|
//#if !defined(ESP_PLATFORM)
|
|
|
|
// mbedtls_debug_set_threshold(5);
|
|
|
|
//#endif
|
2020-12-05 19:26:32 +08:00
|
|
|
if ((rc = mbedtls_ssl_config_defaults(
|
|
|
|
&tls->conf,
|
|
|
|
c->is_client ? MBEDTLS_SSL_IS_CLIENT : MBEDTLS_SSL_IS_SERVER,
|
|
|
|
MBEDTLS_SSL_TRANSPORT_STREAM, MBEDTLS_SSL_PRESET_DEFAULT)) != 0) {
|
|
|
|
mg_error(c, "tls defaults %#x", -rc);
|
|
|
|
goto fail;
|
|
|
|
}
|
|
|
|
mbedtls_ssl_conf_rng(&tls->conf, mbed_rng, c);
|
|
|
|
if (opts->ca == NULL || strcmp(opts->ca, "*") == 0) {
|
|
|
|
mbedtls_ssl_conf_authmode(&tls->conf, MBEDTLS_SSL_VERIFY_NONE);
|
|
|
|
}
|
|
|
|
if (opts->ca != NULL && opts->ca[0] != '\0') {
|
|
|
|
#if defined(MBEDTLS_X509_CA_CHAIN_ON_DISK)
|
|
|
|
tls->cafile = strdup(opts->ca);
|
|
|
|
rc = mbedtls_ssl_conf_ca_chain_file(&tls->conf, tls->cafile, NULL);
|
|
|
|
if (rc != 0) {
|
2021-02-09 01:50:00 +08:00
|
|
|
mg_error(c, "parse on-disk chain(%s) err %#x", ca, -rc);
|
2020-12-05 19:26:32 +08:00
|
|
|
goto fail;
|
|
|
|
}
|
|
|
|
#else
|
2021-02-09 01:50:00 +08:00
|
|
|
rc = opts->ca[0] == '-'
|
|
|
|
? mbedtls_x509_crt_parse(&tls->ca, (uint8_t *) opts->ca,
|
|
|
|
strlen(opts->ca) + 1)
|
|
|
|
: mbedtls_x509_crt_parse_file(&tls->ca, opts->ca);
|
|
|
|
if (rc != 0) {
|
|
|
|
mg_error(c, "parse(%s) err %#x", ca, -rc);
|
2020-12-05 19:26:32 +08:00
|
|
|
goto fail;
|
|
|
|
}
|
|
|
|
mbedtls_ssl_conf_ca_chain(&tls->conf, &tls->ca, NULL);
|
|
|
|
#endif
|
2021-01-30 21:03:11 +08:00
|
|
|
if (opts->srvname.len > 0) {
|
2021-04-07 02:07:15 +08:00
|
|
|
char mem[128], *buf = mem;
|
2021-08-01 05:26:43 +08:00
|
|
|
mg_asprintf(&buf, sizeof(mem), "%.*s", (int) opts->srvname.len,
|
|
|
|
opts->srvname.ptr);
|
2021-01-30 21:03:11 +08:00
|
|
|
mbedtls_ssl_set_hostname(&tls->ssl, buf);
|
2021-04-07 02:07:15 +08:00
|
|
|
if (buf != mem) free(buf);
|
2021-01-30 21:03:11 +08:00
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
mbedtls_ssl_conf_authmode(&tls->conf, MBEDTLS_SSL_VERIFY_REQUIRED);
|
|
|
|
}
|
|
|
|
if (opts->cert != NULL && opts->cert[0] != '\0') {
|
|
|
|
const char *key = opts->certkey;
|
2021-03-17 22:23:45 +08:00
|
|
|
if (key == NULL) {
|
|
|
|
key = opts->cert;
|
|
|
|
certkey = cert;
|
|
|
|
}
|
|
|
|
rc = opts->cert[0] == '-'
|
|
|
|
? mbedtls_x509_crt_parse(&tls->cert, (uint8_t *) opts->cert,
|
|
|
|
strlen(opts->cert) + 1)
|
|
|
|
: mbedtls_x509_crt_parse_file(&tls->cert, opts->cert);
|
|
|
|
if (rc != 0) {
|
|
|
|
mg_error(c, "parse(%s) err %#x", cert, -rc);
|
2020-12-05 19:26:32 +08:00
|
|
|
goto fail;
|
|
|
|
}
|
2021-07-14 00:28:07 +08:00
|
|
|
rc = key[0] == '-' ? mbedtls_pk_parse_key(&tls->pk, (uint8_t *) key,
|
|
|
|
strlen(key) + 1, NULL, 0 RNG)
|
|
|
|
: mbedtls_pk_parse_keyfile(&tls->pk, key, NULL RNG);
|
2021-03-17 22:23:45 +08:00
|
|
|
if (rc != 0) {
|
|
|
|
mg_error(c, "tls key(%s) %#x", certkey, -rc);
|
2020-12-05 19:26:32 +08:00
|
|
|
goto fail;
|
|
|
|
}
|
|
|
|
rc = mbedtls_ssl_conf_own_cert(&tls->conf, &tls->cert, &tls->pk);
|
|
|
|
if (rc != 0) {
|
|
|
|
mg_error(c, "own cert %#x", -rc);
|
|
|
|
goto fail;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if ((rc = mbedtls_ssl_setup(&tls->ssl, &tls->conf)) != 0) {
|
|
|
|
mg_error(c, "setup err %#x", -rc);
|
|
|
|
goto fail;
|
|
|
|
}
|
|
|
|
c->tls = tls;
|
|
|
|
c->is_tls = 1;
|
|
|
|
c->is_tls_hs = 1;
|
2020-12-18 23:13:19 +08:00
|
|
|
if (c->is_client && c->is_resolving == 0 && c->is_connecting == 0) {
|
|
|
|
mg_tls_handshake(c);
|
|
|
|
}
|
2021-05-29 01:30:42 +08:00
|
|
|
return;
|
2020-12-05 19:26:32 +08:00
|
|
|
fail:
|
|
|
|
c->is_closing = 1;
|
|
|
|
free(tls);
|
2017-05-10 00:07:00 +08:00
|
|
|
}
|
|
|
|
|
2021-05-29 01:30:42 +08:00
|
|
|
long mg_tls_recv(struct mg_connection *c, void *buf, size_t len) {
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_tls *tls = (struct mg_tls *) c->tls;
|
2021-05-29 01:30:42 +08:00
|
|
|
long n = mbedtls_ssl_read(&tls->ssl, (unsigned char *) buf, len);
|
|
|
|
return n == 0 ? -1 : n == MBEDTLS_ERR_SSL_WANT_READ ? 0 : n;
|
2016-10-12 02:20:28 +08:00
|
|
|
}
|
|
|
|
|
2021-05-29 01:30:42 +08:00
|
|
|
long mg_tls_send(struct mg_connection *c, const void *buf, size_t len) {
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_tls *tls = (struct mg_tls *) c->tls;
|
2021-05-29 01:30:42 +08:00
|
|
|
long n = mbedtls_ssl_write(&tls->ssl, (unsigned char *) buf, len);
|
|
|
|
return n == 0 ? -1 : n == MBEDTLS_ERR_SSL_WANT_WRITE ? 0 : n;
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
2016-10-12 02:20:28 +08:00
|
|
|
|
2021-05-29 01:30:42 +08:00
|
|
|
void mg_tls_free(struct mg_connection *c) {
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_tls *tls = (struct mg_tls *) c->tls;
|
2021-05-29 01:30:42 +08:00
|
|
|
if (tls == NULL) return;
|
2020-12-05 19:26:32 +08:00
|
|
|
free(tls->cafile);
|
|
|
|
mbedtls_ssl_free(&tls->ssl);
|
|
|
|
mbedtls_pk_free(&tls->pk);
|
2021-04-22 22:21:16 +08:00
|
|
|
mbedtls_x509_crt_free(&tls->ca);
|
|
|
|
mbedtls_x509_crt_free(&tls->cert);
|
2020-12-05 19:26:32 +08:00
|
|
|
mbedtls_ssl_config_free(&tls->conf);
|
|
|
|
free(tls);
|
2021-04-22 22:21:16 +08:00
|
|
|
c->tls = NULL;
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
#elif MG_ENABLE_OPENSSL ///////////////////////////////////////// OPENSSL
|
2016-10-12 02:20:28 +08:00
|
|
|
|
2021-01-28 16:45:24 +08:00
|
|
|
#include <openssl/err.h>
|
2020-12-05 19:26:32 +08:00
|
|
|
#include <openssl/ssl.h>
|
2021-01-24 01:47:51 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_tls {
|
|
|
|
SSL_CTX *ctx;
|
|
|
|
SSL *ssl;
|
2016-10-12 02:20:28 +08:00
|
|
|
};
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
static int mg_tls_err(struct mg_tls *tls, int res) {
|
|
|
|
int err = SSL_get_error(tls->ssl, res);
|
|
|
|
// We've just fetched the last error from the queue.
|
|
|
|
// Now we need to clear the error queue. If we do not, then the following
|
|
|
|
// can happen (actually reported):
|
|
|
|
// - A new connection is accept()-ed with cert error (e.g. self-signed cert)
|
|
|
|
// - Since all accept()-ed connections share listener's context,
|
|
|
|
// - *ALL* SSL accepted connection report read error on the next poll cycle.
|
|
|
|
// Thus a single errored connection can close all the rest, unrelated ones.
|
|
|
|
// Clearing the error keeps the shared SSL_CTX in an OK state.
|
2021-08-30 18:45:30 +08:00
|
|
|
|
|
|
|
if (err != 0) ERR_print_errors_fp(stderr);
|
2020-12-05 19:26:32 +08:00
|
|
|
ERR_clear_error();
|
|
|
|
if (err == SSL_ERROR_WANT_READ) return 0;
|
|
|
|
if (err == SSL_ERROR_WANT_WRITE) return 0;
|
|
|
|
return err;
|
|
|
|
}
|
|
|
|
|
2021-05-29 01:30:42 +08:00
|
|
|
void mg_tls_init(struct mg_connection *c, struct mg_tls_opts *opts) {
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_tls *tls = (struct mg_tls *) calloc(1, sizeof(*tls));
|
|
|
|
const char *id = "mongoose";
|
|
|
|
static unsigned char s_initialised = 0;
|
|
|
|
int rc;
|
2021-01-26 20:16:58 +08:00
|
|
|
|
|
|
|
if (tls == NULL) {
|
|
|
|
mg_error(c, "TLS OOM");
|
|
|
|
goto fail;
|
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
if (!s_initialised) {
|
|
|
|
SSL_library_init();
|
|
|
|
s_initialised++;
|
|
|
|
}
|
2020-12-21 20:26:44 +08:00
|
|
|
LOG(LL_DEBUG, ("%lu Setting TLS, CA: %s, cert: %s, key: %s", c->id,
|
2020-12-05 19:26:32 +08:00
|
|
|
opts->ca == NULL ? "null" : opts->ca,
|
|
|
|
opts->cert == NULL ? "null" : opts->cert,
|
|
|
|
opts->certkey == NULL ? "null" : opts->certkey));
|
|
|
|
tls->ctx = c->is_client ? SSL_CTX_new(SSLv23_client_method())
|
|
|
|
: SSL_CTX_new(SSLv23_server_method());
|
|
|
|
if ((tls->ssl = SSL_new(tls->ctx)) == NULL) {
|
|
|
|
mg_error(c, "SSL_new");
|
|
|
|
goto fail;
|
|
|
|
}
|
2021-05-29 01:30:42 +08:00
|
|
|
SSL_set_session_id_context(tls->ssl, (const uint8_t *) id,
|
|
|
|
(unsigned) strlen(id));
|
2020-12-05 19:26:32 +08:00
|
|
|
// Disable deprecated protocols
|
|
|
|
SSL_set_options(tls->ssl, SSL_OP_NO_SSLv2);
|
|
|
|
SSL_set_options(tls->ssl, SSL_OP_NO_SSLv3);
|
|
|
|
SSL_set_options(tls->ssl, SSL_OP_NO_TLSv1);
|
2021-07-16 19:10:48 +08:00
|
|
|
SSL_set_options(tls->ssl, SSL_OP_NO_TLSv1_1);
|
2020-12-05 19:26:32 +08:00
|
|
|
#ifdef MG_ENABLE_OPENSSL_NO_COMPRESSION
|
|
|
|
SSL_set_options(tls->ssl, SSL_OP_NO_COMPRESSION);
|
|
|
|
#endif
|
|
|
|
#ifdef MG_ENABLE_OPENSSL_CIPHER_SERVER_PREFERENCE
|
|
|
|
SSL_set_options(tls->ssl, SSL_OP_CIPHER_SERVER_PREFERENCE);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
if (opts->ca != NULL && opts->ca[0] != '\0') {
|
2021-01-30 21:03:11 +08:00
|
|
|
SSL_set_verify(tls->ssl, SSL_VERIFY_PEER | SSL_VERIFY_FAIL_IF_NO_PEER_CERT,
|
|
|
|
NULL);
|
2020-12-05 19:26:32 +08:00
|
|
|
if ((rc = SSL_CTX_load_verify_locations(tls->ctx, opts->ca, NULL)) != 1) {
|
|
|
|
mg_error(c, "parse(%s): err %d", opts->ca, mg_tls_err(tls, rc));
|
|
|
|
goto fail;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (opts->cert != NULL && opts->cert[0] != '\0') {
|
|
|
|
const char *key = opts->certkey;
|
|
|
|
if (key == NULL) key = opts->cert;
|
|
|
|
if ((rc = SSL_use_certificate_file(tls->ssl, opts->cert, 1)) != 1) {
|
|
|
|
mg_error(c, "Invalid SSL cert, err %d", mg_tls_err(tls, rc));
|
|
|
|
goto fail;
|
|
|
|
} else if ((rc = SSL_use_PrivateKey_file(tls->ssl, key, 1)) != 1) {
|
|
|
|
mg_error(c, "Invalid SSL key, err %d", mg_tls_err(tls, rc));
|
|
|
|
goto fail;
|
2021-07-14 01:05:57 +08:00
|
|
|
#if OPENSSL_VERSION_NUMBER > 0x10100000L
|
2020-12-05 19:26:32 +08:00
|
|
|
} else if ((rc = SSL_use_certificate_chain_file(tls->ssl, opts->cert)) !=
|
|
|
|
1) {
|
|
|
|
mg_error(c, "Invalid CA, err %d", mg_tls_err(tls, rc));
|
|
|
|
goto fail;
|
2020-12-17 20:12:46 +08:00
|
|
|
#endif
|
2020-12-05 19:26:32 +08:00
|
|
|
} else {
|
|
|
|
SSL_set_mode(tls->ssl, SSL_MODE_ACCEPT_MOVING_WRITE_BUFFER);
|
|
|
|
#if OPENSSL_VERSION_NUMBER > 0x10002000L
|
|
|
|
SSL_set_ecdh_auto(tls->ssl, 1);
|
|
|
|
#endif
|
2016-10-12 02:20:28 +08:00
|
|
|
}
|
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
if (opts->ciphers != NULL) SSL_set_cipher_list(tls->ssl, opts->ciphers);
|
2021-01-30 21:03:11 +08:00
|
|
|
if (opts->srvname.len > 0) {
|
2021-04-07 02:07:15 +08:00
|
|
|
char mem[128], *buf = mem;
|
2021-08-01 05:26:43 +08:00
|
|
|
mg_asprintf(&buf, sizeof(mem), "%.*s", (int) opts->srvname.len,
|
|
|
|
opts->srvname.ptr);
|
2021-01-30 21:03:11 +08:00
|
|
|
SSL_set_tlsext_host_name(tls->ssl, buf);
|
2021-04-07 02:07:15 +08:00
|
|
|
if (buf != mem) free(buf);
|
2021-01-30 21:03:11 +08:00
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
c->tls = tls;
|
|
|
|
c->is_tls = 1;
|
|
|
|
c->is_tls_hs = 1;
|
2020-12-18 23:13:19 +08:00
|
|
|
if (c->is_client && c->is_resolving == 0 && c->is_connecting == 0) {
|
|
|
|
mg_tls_handshake(c);
|
|
|
|
}
|
2020-12-21 05:45:58 +08:00
|
|
|
c->is_hexdumping = 1;
|
2021-01-24 21:57:40 +08:00
|
|
|
LOG(LL_DEBUG, ("%lu SSL %s OK", c->id, c->is_accepted ? "accept" : "client"));
|
2021-05-29 01:30:42 +08:00
|
|
|
return;
|
2020-12-05 19:26:32 +08:00
|
|
|
fail:
|
|
|
|
c->is_closing = 1;
|
|
|
|
free(tls);
|
2016-10-12 02:20:28 +08:00
|
|
|
}
|
|
|
|
|
2021-05-29 01:30:42 +08:00
|
|
|
void mg_tls_handshake(struct mg_connection *c) {
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_tls *tls = (struct mg_tls *) c->tls;
|
|
|
|
int rc;
|
2021-08-01 05:26:43 +08:00
|
|
|
SSL_set_fd(tls->ssl, (int) (size_t) c->fd);
|
2020-12-05 19:26:32 +08:00
|
|
|
rc = c->is_client ? SSL_connect(tls->ssl) : SSL_accept(tls->ssl);
|
|
|
|
if (rc == 1) {
|
2020-12-21 20:26:44 +08:00
|
|
|
LOG(LL_DEBUG, ("%lu success", c->id));
|
2020-12-21 05:45:58 +08:00
|
|
|
c->is_tls_hs = 0;
|
2020-12-05 19:26:32 +08:00
|
|
|
} else {
|
2021-08-29 23:53:29 +08:00
|
|
|
int code = mg_tls_err(tls, rc);
|
2020-12-05 19:26:32 +08:00
|
|
|
if (code != 0) mg_error(c, "tls hs: rc %d, err %d", rc, code);
|
|
|
|
}
|
2016-10-12 02:20:28 +08:00
|
|
|
}
|
|
|
|
|
2021-05-29 01:30:42 +08:00
|
|
|
void mg_tls_free(struct mg_connection *c) {
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_tls *tls = (struct mg_tls *) c->tls;
|
2021-05-29 01:30:42 +08:00
|
|
|
if (tls == NULL) return;
|
2020-12-05 19:26:32 +08:00
|
|
|
SSL_free(tls->ssl);
|
|
|
|
SSL_CTX_free(tls->ctx);
|
|
|
|
free(tls);
|
2021-04-22 22:21:16 +08:00
|
|
|
c->tls = NULL;
|
2016-10-12 02:20:28 +08:00
|
|
|
}
|
|
|
|
|
2021-05-29 01:30:42 +08:00
|
|
|
long mg_tls_recv(struct mg_connection *c, void *buf, size_t len) {
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_tls *tls = (struct mg_tls *) c->tls;
|
2021-05-29 01:30:42 +08:00
|
|
|
int n = SSL_read(tls->ssl, buf, (int) len);
|
|
|
|
return n == 0 ? -1 : n < 0 && mg_tls_err(tls, n) == 0 ? 0 : n;
|
2016-10-12 02:20:28 +08:00
|
|
|
}
|
|
|
|
|
2021-05-29 01:30:42 +08:00
|
|
|
long mg_tls_send(struct mg_connection *c, const void *buf, size_t len) {
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_tls *tls = (struct mg_tls *) c->tls;
|
2021-05-29 01:30:42 +08:00
|
|
|
int n = SSL_write(tls->ssl, buf, (int) len);
|
|
|
|
return n == 0 ? -1 : n < 0 && mg_tls_err(tls, n) == 0 ? 0 : n;
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
#else ////////////////////////////////////////// NO TLS
|
|
|
|
|
2021-05-29 01:30:42 +08:00
|
|
|
void mg_tls_init(struct mg_connection *c, struct mg_tls_opts *opts) {
|
2020-12-05 19:26:32 +08:00
|
|
|
(void) opts;
|
|
|
|
mg_error(c, "TLS is not enabled");
|
|
|
|
}
|
2021-05-29 01:30:42 +08:00
|
|
|
void mg_tls_handshake(struct mg_connection *c) {
|
|
|
|
(void) c;
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
2021-05-29 01:30:42 +08:00
|
|
|
void mg_tls_free(struct mg_connection *c) {
|
|
|
|
(void) c;
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
2021-05-29 01:30:42 +08:00
|
|
|
long mg_tls_recv(struct mg_connection *c, void *buf, size_t len) {
|
2020-12-05 19:26:32 +08:00
|
|
|
return c == NULL || buf == NULL || len == 0 ? 0 : -1;
|
|
|
|
}
|
2021-05-29 01:30:42 +08:00
|
|
|
long mg_tls_send(struct mg_connection *c, const void *buf, size_t len) {
|
2020-12-05 19:26:32 +08:00
|
|
|
return c == NULL || buf == NULL || len == 0 ? 0 : -1;
|
2016-10-12 02:20:28 +08:00
|
|
|
}
|
|
|
|
|
2016-10-25 22:59:53 +08:00
|
|
|
#endif
|
2017-12-29 06:54:59 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
#ifdef MG_ENABLE_LINES
|
|
|
|
#line 1 "src/url.c"
|
2016-10-25 22:59:53 +08:00
|
|
|
#endif
|
2020-12-05 19:26:32 +08:00
|
|
|
|
2021-09-21 15:10:27 +08:00
|
|
|
#include <stdlib.h>
|
2020-12-22 17:44:59 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
struct url {
|
2021-05-29 01:30:42 +08:00
|
|
|
size_t key, user, pass, host, port, uri, end;
|
2020-12-05 19:26:32 +08:00
|
|
|
};
|
|
|
|
|
|
|
|
int mg_url_is_ssl(const char *url) {
|
|
|
|
return strncmp(url, "wss:", 4) == 0 || strncmp(url, "https:", 6) == 0 ||
|
|
|
|
strncmp(url, "mqtts:", 6) == 0 || strncmp(url, "ssl:", 4) == 0 ||
|
|
|
|
strncmp(url, "tls:", 4) == 0;
|
2016-10-12 02:20:28 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
static struct url urlparse(const char *url) {
|
2021-05-29 01:30:42 +08:00
|
|
|
size_t i;
|
2020-12-05 19:26:32 +08:00
|
|
|
struct url u;
|
|
|
|
memset(&u, 0, sizeof(u));
|
|
|
|
for (i = 0; url[i] != '\0'; i++) {
|
2021-03-24 23:49:35 +08:00
|
|
|
if (i > 0 && u.host == 0 && url[i - 1] == '/' && url[i] == '/') {
|
2020-12-05 19:26:32 +08:00
|
|
|
u.host = i + 1;
|
|
|
|
u.port = 0;
|
2020-12-22 17:44:59 +08:00
|
|
|
} else if (url[i] == ']') {
|
|
|
|
u.port = 0; // IPv6 URLs, like http://[::1]/bar
|
2021-07-31 01:34:55 +08:00
|
|
|
} else if (url[i] == ':' && u.port == 0 && u.uri == 0) {
|
2020-12-05 19:26:32 +08:00
|
|
|
u.port = i + 1;
|
2021-03-24 23:49:35 +08:00
|
|
|
} else if (url[i] == '@' && u.user == 0 && u.pass == 0) {
|
2020-12-05 19:26:32 +08:00
|
|
|
u.user = u.host;
|
|
|
|
u.pass = u.port;
|
|
|
|
u.host = i + 1;
|
|
|
|
u.port = 0;
|
2021-03-24 23:49:35 +08:00
|
|
|
} else if (u.host && u.uri == 0 && url[i] == '/') {
|
2020-12-05 19:26:32 +08:00
|
|
|
u.uri = i;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
u.end = i;
|
|
|
|
#if 0
|
|
|
|
printf("[%s] %d %d %d %d %d\n", url, u.user, u.pass, u.host, u.port, u.uri);
|
2016-10-14 21:31:02 +08:00
|
|
|
#endif
|
2020-12-05 19:26:32 +08:00
|
|
|
return u;
|
|
|
|
}
|
2016-10-14 21:31:02 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_str mg_url_host(const char *url) {
|
|
|
|
struct url u = urlparse(url);
|
2021-05-29 01:30:42 +08:00
|
|
|
size_t n = u.port ? u.port - u.host - 1
|
|
|
|
: u.uri ? u.uri - u.host
|
|
|
|
: u.end - u.host;
|
2020-12-22 17:44:59 +08:00
|
|
|
struct mg_str s = mg_str_n(url + u.host, n);
|
|
|
|
return s;
|
2016-10-14 21:31:02 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
const char *mg_url_uri(const char *url) {
|
|
|
|
struct url u = urlparse(url);
|
|
|
|
return u.uri ? url + u.uri : "/";
|
2016-10-17 17:03:59 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
unsigned short mg_url_port(const char *url) {
|
|
|
|
struct url u = urlparse(url);
|
|
|
|
unsigned short port = 0;
|
2021-01-19 02:39:22 +08:00
|
|
|
if (memcmp(url, "http:", 5) == 0 || memcmp(url, "ws:", 3) == 0) port = 80;
|
2020-12-05 19:26:32 +08:00
|
|
|
if (memcmp(url, "wss:", 4) == 0 || memcmp(url, "https:", 6) == 0) port = 443;
|
|
|
|
if (memcmp(url, "mqtt:", 5) == 0) port = 1883;
|
|
|
|
if (memcmp(url, "mqtts:", 6) == 0) port = 8883;
|
2021-05-29 01:57:47 +08:00
|
|
|
if (u.port) port = (unsigned short) atoi(url + u.port);
|
2020-12-05 19:26:32 +08:00
|
|
|
return port;
|
|
|
|
}
|
|
|
|
|
|
|
|
struct mg_str mg_url_user(const char *url) {
|
|
|
|
struct url u = urlparse(url);
|
|
|
|
struct mg_str s = mg_str("");
|
|
|
|
if (u.user && (u.pass || u.host)) {
|
2021-05-29 01:30:42 +08:00
|
|
|
size_t n = u.pass ? u.pass - u.user - 1 : u.host - u.user - 1;
|
2020-12-05 19:26:32 +08:00
|
|
|
s = mg_str_n(url + u.user, n);
|
2016-10-17 17:03:59 +08:00
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
return s;
|
2016-10-17 17:03:59 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_str mg_url_pass(const char *url) {
|
|
|
|
struct url u = urlparse(url);
|
2021-05-29 01:30:42 +08:00
|
|
|
struct mg_str s = mg_str_n("", 0UL);
|
2020-12-05 19:26:32 +08:00
|
|
|
if (u.pass && u.host) {
|
2021-05-29 01:30:42 +08:00
|
|
|
size_t n = u.host - u.pass - 1;
|
2020-12-05 19:26:32 +08:00
|
|
|
s = mg_str_n(url + u.pass, n);
|
2016-10-17 17:03:59 +08:00
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
return s;
|
2016-10-17 17:03:59 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
#ifdef MG_ENABLE_LINES
|
|
|
|
#line 1 "src/util.c"
|
2016-10-14 21:31:02 +08:00
|
|
|
#endif
|
2016-11-09 06:29:07 +08:00
|
|
|
|
|
|
|
|
2021-01-27 01:43:31 +08:00
|
|
|
|
2021-05-12 16:25:21 +08:00
|
|
|
|
2021-03-15 17:16:34 +08:00
|
|
|
char *mg_file_read(const char *path, size_t *sizep) {
|
2020-12-05 19:26:32 +08:00
|
|
|
FILE *fp;
|
|
|
|
char *data = NULL;
|
2021-07-24 03:18:34 +08:00
|
|
|
size_t size = 0;
|
2021-07-23 05:46:33 +08:00
|
|
|
if ((fp = fopen(path, "rb")) != NULL) {
|
2021-07-24 03:18:34 +08:00
|
|
|
fseek(fp, 0, SEEK_END);
|
|
|
|
size = (size_t) ftell(fp);
|
|
|
|
rewind(fp);
|
2021-05-18 00:36:57 +08:00
|
|
|
data = (char *) calloc(1, size + 1);
|
2020-12-05 19:26:32 +08:00
|
|
|
if (data != NULL) {
|
|
|
|
if (fread(data, 1, size, fp) != size) {
|
|
|
|
free(data);
|
|
|
|
data = NULL;
|
|
|
|
} else {
|
|
|
|
data[size] = '\0';
|
2021-03-15 17:16:34 +08:00
|
|
|
if (sizep != NULL) *sizep = size;
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
fclose(fp);
|
|
|
|
}
|
|
|
|
return data;
|
2016-10-26 19:38:15 +08:00
|
|
|
}
|
|
|
|
|
2021-01-27 00:30:44 +08:00
|
|
|
bool mg_file_write(const char *path, const void *buf, size_t len) {
|
|
|
|
bool result = false;
|
2020-12-05 19:26:32 +08:00
|
|
|
FILE *fp;
|
2021-02-03 10:13:05 +08:00
|
|
|
char tmp[MG_PATH_MAX];
|
2020-12-05 19:26:32 +08:00
|
|
|
snprintf(tmp, sizeof(tmp), "%s.%d", path, rand());
|
2021-07-23 05:46:33 +08:00
|
|
|
fp = fopen(tmp, "wb");
|
2020-12-05 19:26:32 +08:00
|
|
|
if (fp != NULL) {
|
2021-01-27 00:30:44 +08:00
|
|
|
result = fwrite(buf, 1, len, fp) == len;
|
2020-12-05 19:26:32 +08:00
|
|
|
fclose(fp);
|
2021-01-27 00:30:44 +08:00
|
|
|
if (result) {
|
|
|
|
remove(path);
|
|
|
|
rename(tmp, path);
|
|
|
|
} else {
|
|
|
|
remove(tmp);
|
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
2021-01-27 00:30:44 +08:00
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool mg_file_printf(const char *path, const char *fmt, ...) {
|
|
|
|
char tmp[256], *buf = tmp;
|
|
|
|
bool result;
|
2021-05-29 01:30:42 +08:00
|
|
|
int len;
|
2021-01-27 00:30:44 +08:00
|
|
|
va_list ap;
|
|
|
|
va_start(ap, fmt);
|
|
|
|
len = mg_vasprintf(&buf, sizeof(tmp), fmt, ap);
|
|
|
|
va_end(ap);
|
2021-05-29 01:30:42 +08:00
|
|
|
result = mg_file_write(path, buf, len > 0 ? (size_t) len : 0);
|
2021-01-27 00:30:44 +08:00
|
|
|
if (buf != tmp) free(buf);
|
|
|
|
return result;
|
2016-10-26 19:38:15 +08:00
|
|
|
}
|
|
|
|
|
2021-07-30 20:19:20 +08:00
|
|
|
#if MG_ENABLE_CUSTOM_RANDOM
|
|
|
|
#else
|
2020-12-05 19:26:32 +08:00
|
|
|
void mg_random(void *buf, size_t len) {
|
2021-02-09 01:50:00 +08:00
|
|
|
bool done = false;
|
2021-07-14 01:07:38 +08:00
|
|
|
unsigned char *p = (unsigned char *) buf;
|
2021-07-14 00:28:07 +08:00
|
|
|
#if MG_ARCH == MG_ARCH_ESP32
|
|
|
|
while (len--) *p++ = (unsigned char) (esp_random() & 255);
|
2021-10-05 02:47:24 +08:00
|
|
|
done = true;
|
2021-07-14 00:28:07 +08:00
|
|
|
#elif MG_ARCH == MG_ARCH_WIN32
|
2021-07-24 10:44:00 +08:00
|
|
|
#elif MG_ARCH_UNIX
|
2021-07-23 05:46:33 +08:00
|
|
|
FILE *fp = fopen("/dev/urandom", "rb");
|
2021-02-09 03:43:01 +08:00
|
|
|
if (fp != NULL) {
|
2021-03-22 19:30:48 +08:00
|
|
|
if (fread(buf, 1, len, fp) == len) done = true;
|
2021-02-09 03:43:01 +08:00
|
|
|
fclose(fp);
|
2021-02-09 01:50:00 +08:00
|
|
|
}
|
|
|
|
#endif
|
2021-10-05 02:47:24 +08:00
|
|
|
// If everything above did not work, fallback to a pseudo random generator
|
|
|
|
while (!done && len--) *p++ = (unsigned char) (rand() & 255);
|
2016-10-26 19:38:15 +08:00
|
|
|
}
|
2021-07-30 20:19:20 +08:00
|
|
|
#endif
|
2016-10-26 19:38:15 +08:00
|
|
|
|
2021-05-29 01:30:42 +08:00
|
|
|
bool mg_globmatch(const char *s1, size_t n1, const char *s2, size_t n2) {
|
|
|
|
size_t i = 0, j = 0, ni = 0, nj = 0;
|
2020-12-05 19:26:32 +08:00
|
|
|
while (i < n1 || j < n2) {
|
|
|
|
if (i < n1 && j < n2 && (s1[i] == '?' || s2[j] == s1[i])) {
|
|
|
|
i++, j++;
|
|
|
|
} else if (i < n1 && (s1[i] == '*' || s1[i] == '#')) {
|
|
|
|
ni = i, nj = j + 1, i++;
|
2021-09-15 14:43:48 +08:00
|
|
|
} else if (nj > 0 && nj <= n2 && (s1[ni] == '#' || s2[j] != '/')) {
|
2020-12-05 19:26:32 +08:00
|
|
|
i = ni, j = nj;
|
|
|
|
} else {
|
2021-09-15 14:43:48 +08:00
|
|
|
// printf(">>: [%s] [%s] %d %d %d %d\n", s1, s2, i, j, ni, nj);
|
2020-12-05 19:26:32 +08:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return true;
|
2016-10-26 19:38:15 +08:00
|
|
|
}
|
|
|
|
|
2021-05-29 01:30:42 +08:00
|
|
|
static size_t mg_nce(const char *s, size_t n, size_t ofs, size_t *koff,
|
|
|
|
size_t *klen, size_t *voff, size_t *vlen) {
|
|
|
|
size_t kvlen, kl;
|
|
|
|
for (kvlen = 0; ofs + kvlen < n && s[ofs + kvlen] != ',';) kvlen++;
|
2020-12-05 19:26:32 +08:00
|
|
|
for (kl = 0; kl < kvlen && s[ofs + kl] != '=';) kl++;
|
|
|
|
if (koff != NULL) *koff = ofs;
|
|
|
|
if (klen != NULL) *klen = kl;
|
|
|
|
if (voff != NULL) *voff = kl < kvlen ? ofs + kl + 1 : 0;
|
|
|
|
if (vlen != NULL) *vlen = kl < kvlen ? kvlen - kl - 1 : 0;
|
2021-05-29 01:30:42 +08:00
|
|
|
ofs += kvlen + 1;
|
|
|
|
return ofs > n ? n : ofs;
|
2016-10-26 19:38:15 +08:00
|
|
|
}
|
|
|
|
|
2021-08-28 13:54:56 +08:00
|
|
|
bool mg_commalist(struct mg_str *s, struct mg_str *k, struct mg_str *v) {
|
2021-05-29 01:30:42 +08:00
|
|
|
size_t koff = 0, klen = 0, voff = 0, vlen = 0;
|
|
|
|
size_t off = mg_nce(s->ptr, s->len, 0, &koff, &klen, &voff, &vlen);
|
2020-12-05 19:26:32 +08:00
|
|
|
if (k != NULL) *k = mg_str_n(s->ptr + koff, klen);
|
|
|
|
if (v != NULL) *v = mg_str_n(s->ptr + voff, vlen);
|
|
|
|
*s = mg_str_n(s->ptr + off, s->len - off);
|
|
|
|
return off > 0;
|
2016-10-26 19:38:15 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
uint32_t mg_ntohl(uint32_t net) {
|
|
|
|
uint8_t data[4] = {0, 0, 0, 0};
|
|
|
|
memcpy(&data, &net, sizeof(data));
|
2021-05-29 01:30:42 +08:00
|
|
|
return (((uint32_t) data[3]) << 0) | (((uint32_t) data[2]) << 8) |
|
|
|
|
(((uint32_t) data[1]) << 16) | (((uint32_t) data[0]) << 24);
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
2016-10-26 19:38:15 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
uint16_t mg_ntohs(uint16_t net) {
|
|
|
|
uint8_t data[2] = {0, 0};
|
|
|
|
memcpy(&data, &net, sizeof(data));
|
2021-08-25 14:59:31 +08:00
|
|
|
return (uint16_t) ((uint16_t) data[1] | (((uint16_t) data[0]) << 8));
|
2016-10-26 19:38:15 +08:00
|
|
|
}
|
|
|
|
|
2020-12-22 17:44:59 +08:00
|
|
|
char *mg_hexdump(const void *buf, size_t len) {
|
2020-12-05 19:26:32 +08:00
|
|
|
const unsigned char *p = (const unsigned char *) buf;
|
2020-12-22 17:44:59 +08:00
|
|
|
size_t i, idx, n = 0, ofs = 0, dlen = len * 5 + 100;
|
2021-05-18 00:36:57 +08:00
|
|
|
char ascii[17] = "", *dst = (char *) calloc(1, dlen);
|
2020-12-05 19:26:32 +08:00
|
|
|
if (dst == NULL) return dst;
|
|
|
|
for (i = 0; i < len; i++) {
|
|
|
|
idx = i % 16;
|
|
|
|
if (idx == 0) {
|
2021-05-29 01:30:42 +08:00
|
|
|
if (i > 0 && dlen > n)
|
|
|
|
n += (size_t) snprintf(dst + n, dlen - n, " %s\n", ascii);
|
|
|
|
if (dlen > n)
|
|
|
|
n += (size_t) snprintf(dst + n, dlen - n, "%04x ", (int) (i + ofs));
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
if (dlen < n) break;
|
2021-05-29 01:30:42 +08:00
|
|
|
n += (size_t) snprintf(dst + n, dlen - n, " %02x", p[i]);
|
|
|
|
ascii[idx] = (char) (p[i] < 0x20 || p[i] > 0x7e ? '.' : p[i]);
|
2020-12-05 19:26:32 +08:00
|
|
|
ascii[idx + 1] = '\0';
|
2016-10-26 19:38:15 +08:00
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
while (i++ % 16) {
|
2021-05-29 01:30:42 +08:00
|
|
|
if (n < dlen) n += (size_t) snprintf(dst + n, dlen - n, "%s", " ");
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
2021-05-29 01:30:42 +08:00
|
|
|
if (n < dlen) n += (size_t) snprintf(dst + n, dlen - n, " %s\n", ascii);
|
2020-12-05 19:26:32 +08:00
|
|
|
if (n > dlen - 1) n = dlen - 1;
|
|
|
|
dst[n] = '\0';
|
|
|
|
return dst;
|
2016-10-26 19:38:15 +08:00
|
|
|
}
|
|
|
|
|
2021-05-29 01:30:42 +08:00
|
|
|
char *mg_hex(const void *buf, size_t len, char *to) {
|
2020-12-05 19:26:32 +08:00
|
|
|
const unsigned char *p = (const unsigned char *) buf;
|
|
|
|
static const char *hex = "0123456789abcdef";
|
2021-05-29 01:30:42 +08:00
|
|
|
size_t i = 0;
|
2020-12-05 19:26:32 +08:00
|
|
|
for (; len--; p++) {
|
|
|
|
to[i++] = hex[p[0] >> 4];
|
|
|
|
to[i++] = hex[p[0] & 0x0f];
|
|
|
|
}
|
|
|
|
to[i] = '\0';
|
|
|
|
return to;
|
2016-10-26 19:38:15 +08:00
|
|
|
}
|
|
|
|
|
2021-05-29 01:30:42 +08:00
|
|
|
static unsigned char mg_unhex_nimble(unsigned char c) {
|
2021-08-25 14:59:31 +08:00
|
|
|
return (c >= '0' && c <= '9') ? (unsigned char) (c - '0')
|
|
|
|
: (c >= 'A' && c <= 'F') ? (unsigned char) (c - '7')
|
|
|
|
: (unsigned char) (c - 'W');
|
2021-05-29 01:30:42 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
unsigned long mg_unhexn(const char *s, size_t len) {
|
2020-12-05 19:26:32 +08:00
|
|
|
unsigned long i = 0, v = 0;
|
2021-05-29 01:30:42 +08:00
|
|
|
for (i = 0; i < len; i++) v <<= 4, v |= mg_unhex_nimble(((uint8_t *) s)[i]);
|
2020-12-05 19:26:32 +08:00
|
|
|
return v;
|
2016-10-26 19:38:15 +08:00
|
|
|
}
|
|
|
|
|
2021-05-29 01:30:42 +08:00
|
|
|
void mg_unhex(const char *buf, size_t len, unsigned char *to) {
|
|
|
|
size_t i;
|
2020-12-05 19:26:32 +08:00
|
|
|
for (i = 0; i < len; i += 2) {
|
|
|
|
to[i >> 1] = (unsigned char) mg_unhexn(&buf[i], 2);
|
2016-10-26 19:38:15 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
int mg_vasprintf(char **buf, size_t size, const char *fmt, va_list ap) {
|
|
|
|
va_list ap_copy;
|
|
|
|
int len;
|
2016-10-26 19:38:15 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
va_copy(ap_copy, ap);
|
|
|
|
len = vsnprintf(*buf, size, fmt, ap_copy);
|
|
|
|
va_end(ap_copy);
|
2016-10-26 19:38:15 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
if (len < 0) {
|
|
|
|
// eCos and Windows are not standard-compliant and return -1 when
|
|
|
|
// the buffer is too small. Keep allocating larger buffers until we
|
|
|
|
// succeed or out of memory.
|
2020-12-07 16:54:58 +08:00
|
|
|
// LCOV_EXCL_START
|
|
|
|
*buf = NULL;
|
2020-12-05 19:26:32 +08:00
|
|
|
while (len < 0) {
|
|
|
|
free(*buf);
|
|
|
|
if (size == 0) size = 5;
|
|
|
|
size *= 2;
|
2021-05-18 00:36:57 +08:00
|
|
|
if ((*buf = (char *) calloc(1, size)) == NULL) {
|
2020-12-05 19:26:32 +08:00
|
|
|
len = -1;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
va_copy(ap_copy, ap);
|
|
|
|
len = vsnprintf(*buf, size - 1, fmt, ap_copy);
|
|
|
|
va_end(ap_copy);
|
|
|
|
}
|
|
|
|
// Microsoft version of vsnprintf() is not always null-terminated, so put
|
|
|
|
// the terminator manually
|
2020-12-25 03:10:54 +08:00
|
|
|
if (*buf != NULL) (*buf)[len] = 0;
|
2020-12-07 16:54:58 +08:00
|
|
|
// LCOV_EXCL_STOP
|
2020-12-05 19:26:32 +08:00
|
|
|
} else if (len >= (int) size) {
|
|
|
|
/// Standard-compliant code path. Allocate a buffer that is large enough
|
2021-05-29 01:30:42 +08:00
|
|
|
if ((*buf = (char *) calloc(1, (size_t) len + 1)) == NULL) {
|
2020-12-05 19:26:32 +08:00
|
|
|
len = -1; // LCOV_EXCL_LINE
|
|
|
|
} else { // LCOV_EXCL_LINE
|
|
|
|
va_copy(ap_copy, ap);
|
2021-05-29 01:30:42 +08:00
|
|
|
len = vsnprintf(*buf, (size_t) len + 1, fmt, ap_copy);
|
2020-12-05 19:26:32 +08:00
|
|
|
va_end(ap_copy);
|
|
|
|
}
|
2016-10-26 19:38:15 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
return len;
|
|
|
|
}
|
2016-10-26 19:38:15 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
int mg_asprintf(char **buf, size_t size, const char *fmt, ...) {
|
|
|
|
int ret;
|
|
|
|
va_list ap;
|
|
|
|
va_start(ap, fmt);
|
|
|
|
ret = mg_vasprintf(buf, size, fmt, ap);
|
|
|
|
va_end(ap);
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
2020-12-23 19:58:08 +08:00
|
|
|
int64_t mg_to64(struct mg_str str) {
|
2021-03-22 00:14:44 +08:00
|
|
|
int64_t result = 0, neg = 1, max = 922337203685477570 /* INT64_MAX/10-10 */;
|
2020-12-23 19:58:08 +08:00
|
|
|
size_t i = 0;
|
|
|
|
while (i < str.len && (str.ptr[i] == ' ' || str.ptr[i] == '\t')) i++;
|
|
|
|
if (i < str.len && str.ptr[i] == '-') neg = -1, i++;
|
|
|
|
while (i < str.len && str.ptr[i] >= '0' && str.ptr[i] <= '9') {
|
2021-03-17 21:46:50 +08:00
|
|
|
if (result > max) return 0;
|
2020-12-05 19:26:32 +08:00
|
|
|
result *= 10;
|
2020-12-23 19:58:08 +08:00
|
|
|
result += (str.ptr[i] - '0');
|
|
|
|
i++;
|
2016-10-26 19:38:15 +08:00
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
return result * neg;
|
|
|
|
}
|
2016-10-26 19:38:15 +08:00
|
|
|
|
2021-03-08 23:41:16 +08:00
|
|
|
uint32_t mg_crc32(uint32_t crc, const char *buf, size_t len) {
|
|
|
|
int i;
|
|
|
|
crc = ~crc;
|
|
|
|
while (len--) {
|
|
|
|
crc ^= *(unsigned char *) buf++;
|
|
|
|
for (i = 0; i < 8; i++) crc = crc & 1 ? (crc >> 1) ^ 0xedb88320 : crc >> 1;
|
|
|
|
}
|
|
|
|
return ~crc;
|
|
|
|
}
|
|
|
|
|
2021-08-25 14:59:31 +08:00
|
|
|
static int isbyte(int n) {
|
|
|
|
return n >= 0 && n <= 255;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int parse_net(const char *spec, uint32_t *net, uint32_t *mask) {
|
|
|
|
int n, a, b, c, d, slash = 32, len = 0;
|
|
|
|
if ((sscanf(spec, "%d.%d.%d.%d/%d%n", &a, &b, &c, &d, &slash, &n) == 5 ||
|
|
|
|
sscanf(spec, "%d.%d.%d.%d%n", &a, &b, &c, &d, &n) == 4) &&
|
|
|
|
isbyte(a) && isbyte(b) && isbyte(c) && isbyte(d) && slash >= 0 &&
|
|
|
|
slash < 33) {
|
|
|
|
len = n;
|
|
|
|
*net = ((uint32_t) a << 24) | ((uint32_t) b << 16) | ((uint32_t) c << 8) |
|
|
|
|
(uint32_t) d;
|
|
|
|
*mask = slash ? (uint32_t) (0xffffffffU << (32 - slash)) : (uint32_t) 0;
|
|
|
|
}
|
|
|
|
return len;
|
|
|
|
}
|
|
|
|
|
|
|
|
int mg_check_ip_acl(struct mg_str acl, uint32_t remote_ip) {
|
|
|
|
struct mg_str k, v;
|
|
|
|
int allowed = acl.len == 0 ? '+' : '-'; // If any ACL is set, deny by default
|
2021-08-28 13:54:56 +08:00
|
|
|
while (mg_commalist(&acl, &k, &v)) {
|
2021-08-25 14:59:31 +08:00
|
|
|
uint32_t net, mask;
|
|
|
|
if (k.ptr[0] != '+' && k.ptr[0] != '-') return -1;
|
|
|
|
if (parse_net(&k.ptr[1], &net, &mask) == 0) return -2;
|
|
|
|
if ((remote_ip & mask) == net) allowed = k.ptr[0];
|
|
|
|
}
|
|
|
|
return allowed == '+';
|
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
double mg_time(void) {
|
|
|
|
#if MG_ARCH == MG_ARCH_WIN32
|
|
|
|
SYSTEMTIME sysnow;
|
|
|
|
FILETIME ftime;
|
|
|
|
GetLocalTime(&sysnow);
|
|
|
|
SystemTimeToFileTime(&sysnow, &ftime);
|
|
|
|
/*
|
|
|
|
* 1. VC 6.0 doesn't support conversion uint64 -> double, so, using int64
|
|
|
|
* This should not cause a problems in this (21th) century
|
|
|
|
* 2. Windows FILETIME is a number of 100-nanosecond intervals since January
|
|
|
|
* 1, 1601 while time_t is a number of _seconds_ since January 1, 1970 UTC,
|
|
|
|
* thus, we need to convert to seconds and adjust amount (subtract 11644473600
|
|
|
|
* seconds)
|
|
|
|
*/
|
|
|
|
return (double) (((int64_t) ftime.dwLowDateTime +
|
|
|
|
((int64_t) ftime.dwHighDateTime << 32)) /
|
|
|
|
10000000.0) -
|
|
|
|
11644473600;
|
2021-09-14 14:28:28 +08:00
|
|
|
#elif MG_ARCH == MG_ARCH_FREERTOS_TCP || MG_ARCH == MG_ARCH_AZURERTOS
|
2021-05-12 15:43:34 +08:00
|
|
|
return mg_millis() / 1000.0;
|
2020-12-05 19:26:32 +08:00
|
|
|
#else
|
|
|
|
struct timeval tv;
|
|
|
|
if (gettimeofday(&tv, NULL /* tz */) != 0) return 0;
|
|
|
|
return (double) tv.tv_sec + (((double) tv.tv_usec) / 1000000.0);
|
|
|
|
#endif /* _WIN32 */
|
2016-10-26 19:38:15 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
void mg_usleep(unsigned long usecs) {
|
|
|
|
#if MG_ARCH == MG_ARCH_WIN32
|
|
|
|
Sleep(usecs / 1000);
|
|
|
|
#elif MG_ARCH == MG_ARCH_ESP8266
|
|
|
|
ets_delay_us(usecs);
|
2021-05-29 06:49:26 +08:00
|
|
|
#elif MG_ARCH == MG_ARCH_FREERTOS_TCP || MG_ARCH == MG_ARCH_FREERTOS_LWIP
|
2021-05-12 15:43:34 +08:00
|
|
|
vTaskDelay(pdMS_TO_TICKS(usecs / 1000));
|
2021-09-14 14:28:28 +08:00
|
|
|
#elif MG_ARCH == MG_ARCH_AZURERTOS
|
|
|
|
tx_thread_sleep((usecs / 1000) * TX_TIMER_TICKS_PER_SECOND);
|
2020-12-05 19:26:32 +08:00
|
|
|
#else
|
2021-05-29 01:30:42 +08:00
|
|
|
usleep((useconds_t) usecs);
|
2020-12-05 19:26:32 +08:00
|
|
|
#endif
|
2016-10-26 19:38:15 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
unsigned long mg_millis(void) {
|
|
|
|
#if MG_ARCH == MG_ARCH_WIN32
|
|
|
|
return GetTickCount();
|
|
|
|
#elif MG_ARCH == MG_ARCH_ESP32
|
|
|
|
return esp_timer_get_time() / 1000;
|
|
|
|
#elif MG_ARCH == MG_ARCH_ESP8266
|
2020-12-27 09:29:42 +08:00
|
|
|
return xTaskGetTickCount() * portTICK_PERIOD_MS;
|
2021-05-29 06:49:26 +08:00
|
|
|
#elif MG_ARCH == MG_ARCH_FREERTOS_TCP || MG_ARCH == MG_ARCH_FREERTOS_LWIP
|
2020-12-05 19:26:32 +08:00
|
|
|
return xTaskGetTickCount() * portTICK_PERIOD_MS;
|
2021-09-14 14:28:28 +08:00
|
|
|
#elif MG_ARCH == MG_ARCH_AZURERTOS
|
|
|
|
return tx_time_get() * (1000 /* MS per SEC */ / TX_TIMER_TICKS_PER_SECOND);
|
2020-12-05 19:26:32 +08:00
|
|
|
#else
|
|
|
|
struct timespec ts;
|
|
|
|
clock_gettime(CLOCK_REALTIME, &ts);
|
2021-05-29 01:30:42 +08:00
|
|
|
return (unsigned long) ((uint64_t) ts.tv_sec * 1000 +
|
|
|
|
(uint64_t) ts.tv_nsec / 1000000);
|
2020-12-05 19:26:32 +08:00
|
|
|
#endif
|
|
|
|
}
|
2016-10-26 19:38:15 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
#ifdef MG_ENABLE_LINES
|
|
|
|
#line 1 "src/ws.c"
|
|
|
|
#endif
|
2016-10-26 19:38:15 +08:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2016-11-09 06:29:07 +08:00
|
|
|
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
struct ws_msg {
|
|
|
|
uint8_t flags;
|
|
|
|
size_t header_len;
|
|
|
|
size_t data_len;
|
|
|
|
};
|
2016-11-09 19:15:06 +08:00
|
|
|
|
2021-08-04 18:38:32 +08:00
|
|
|
static void ws_handshake(struct mg_connection *c, const struct mg_str *wskey,
|
|
|
|
const struct mg_str *wsproto, const char *fmt,
|
|
|
|
va_list ap) {
|
2020-12-05 19:26:32 +08:00
|
|
|
const char *magic = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
|
|
|
|
unsigned char sha[20], b64_sha[30];
|
2021-01-17 06:48:43 +08:00
|
|
|
char mem[128], *buf = mem;
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_sha1_ctx sha_ctx;
|
|
|
|
mg_sha1_init(&sha_ctx);
|
2021-08-04 18:38:32 +08:00
|
|
|
mg_sha1_update(&sha_ctx, (unsigned char *) wskey->ptr, wskey->len);
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_sha1_update(&sha_ctx, (unsigned char *) magic, 36);
|
|
|
|
mg_sha1_final(sha, &sha_ctx);
|
|
|
|
mg_base64_encode(sha, sizeof(sha), (char *) b64_sha);
|
2021-01-17 06:48:43 +08:00
|
|
|
buf[0] = '\0';
|
|
|
|
if (fmt != NULL) mg_vasprintf(&buf, sizeof(mem), fmt, ap);
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_printf(c,
|
|
|
|
"HTTP/1.1 101 Switching Protocols\r\n"
|
|
|
|
"Upgrade: websocket\r\n"
|
|
|
|
"Connection: Upgrade\r\n"
|
2021-01-17 06:48:43 +08:00
|
|
|
"Sec-WebSocket-Accept: %s\r\n"
|
2021-08-04 18:38:32 +08:00
|
|
|
"%s",
|
2021-01-17 06:48:43 +08:00
|
|
|
b64_sha, buf);
|
|
|
|
if (buf != mem) free(buf);
|
2021-08-04 18:38:32 +08:00
|
|
|
if (wsproto != NULL) {
|
|
|
|
mg_printf(c, "Sec-WebSocket-Protocol: %.*s\r\n", (int) wsproto->len,
|
|
|
|
wsproto->ptr);
|
|
|
|
}
|
|
|
|
mg_send(c, "\r\n", 2);
|
2016-11-09 19:15:06 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
static size_t ws_process(uint8_t *buf, size_t len, struct ws_msg *msg) {
|
|
|
|
size_t i, n = 0, mask_len = 0;
|
|
|
|
memset(msg, 0, sizeof(*msg));
|
|
|
|
if (len >= 2) {
|
2021-08-27 17:25:24 +08:00
|
|
|
n = buf[1] & 0x7f; // Frame length
|
|
|
|
mask_len = buf[1] & 128 ? 4 : 0; // last bit is a mask bit
|
|
|
|
msg->flags = buf[0];
|
2020-12-05 19:26:32 +08:00
|
|
|
if (n < 126 && len >= mask_len) {
|
|
|
|
msg->data_len = n;
|
|
|
|
msg->header_len = 2 + mask_len;
|
|
|
|
} else if (n == 126 && len >= 4 + mask_len) {
|
|
|
|
msg->header_len = 4 + mask_len;
|
|
|
|
msg->data_len = mg_ntohs(*(uint16_t *) &buf[2]);
|
|
|
|
} else if (len >= 10 + mask_len) {
|
|
|
|
msg->header_len = 10 + mask_len;
|
|
|
|
msg->data_len =
|
2021-08-11 11:52:39 +08:00
|
|
|
(size_t) (((uint64_t) mg_ntohl(*(uint32_t *) &buf[2])) << 32) +
|
2020-12-05 19:26:32 +08:00
|
|
|
mg_ntohl(*(uint32_t *) &buf[6]);
|
|
|
|
}
|
|
|
|
}
|
2021-09-14 17:26:03 +08:00
|
|
|
// Sanity check, and integer overflow protection for the boundary check below
|
|
|
|
// data_len should not be larger than 1 Gb
|
2021-09-17 17:56:58 +08:00
|
|
|
if (msg->data_len > 1024 * 1024 * 1024) return 0;
|
2020-12-05 19:26:32 +08:00
|
|
|
if (msg->header_len + msg->data_len > len) return 0;
|
|
|
|
if (mask_len > 0) {
|
|
|
|
uint8_t *p = buf + msg->header_len, *m = p - mask_len;
|
|
|
|
for (i = 0; i < msg->data_len; i++) p[i] ^= m[i & 3];
|
|
|
|
}
|
|
|
|
return msg->header_len + msg->data_len;
|
|
|
|
}
|
|
|
|
|
2021-08-01 21:10:21 +08:00
|
|
|
static size_t mkhdr(size_t len, int op, bool is_client, uint8_t *buf) {
|
|
|
|
size_t n = 0;
|
2021-08-27 17:25:24 +08:00
|
|
|
buf[0] = (uint8_t) (op | 128);
|
2020-12-05 19:26:32 +08:00
|
|
|
if (len < 126) {
|
2021-08-01 21:10:21 +08:00
|
|
|
buf[1] = (unsigned char) len;
|
|
|
|
n = 2;
|
2020-12-24 00:18:49 +08:00
|
|
|
} else if (len < 65536) {
|
2020-12-05 19:26:32 +08:00
|
|
|
uint16_t tmp = mg_htons((uint16_t) len);
|
2021-08-01 21:10:21 +08:00
|
|
|
buf[1] = 126;
|
|
|
|
memcpy(&buf[2], &tmp, sizeof(tmp));
|
|
|
|
n = 4;
|
2020-12-05 19:26:32 +08:00
|
|
|
} else {
|
|
|
|
uint32_t tmp;
|
2021-08-01 21:10:21 +08:00
|
|
|
buf[1] = 127;
|
2021-08-11 11:52:39 +08:00
|
|
|
tmp = mg_htonl((uint32_t) ((uint64_t) len >> 32));
|
2021-08-01 21:10:21 +08:00
|
|
|
memcpy(&buf[2], &tmp, sizeof(tmp));
|
2021-08-11 11:52:39 +08:00
|
|
|
tmp = mg_htonl((uint32_t) (len & 0xffffffff));
|
2021-08-01 21:10:21 +08:00
|
|
|
memcpy(&buf[6], &tmp, sizeof(tmp));
|
|
|
|
n = 10;
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
2021-08-01 21:10:21 +08:00
|
|
|
if (is_client) {
|
|
|
|
buf[1] |= 1 << 7; // Set masking flag
|
|
|
|
mg_random(&buf[n], 4);
|
|
|
|
n += 4;
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
2021-08-01 21:10:21 +08:00
|
|
|
return n;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void mg_ws_mask(struct mg_connection *c, size_t len) {
|
2021-01-02 20:20:08 +08:00
|
|
|
if (c->is_client && c->send.buf != NULL) {
|
2021-08-01 21:10:21 +08:00
|
|
|
size_t i;
|
|
|
|
uint8_t *p = c->send.buf + c->send.len - len, *mask = p - 4;
|
2020-12-05 19:26:32 +08:00
|
|
|
for (i = 0; i < len; i++) p[i] ^= mask[i & 3];
|
2016-11-09 06:29:07 +08:00
|
|
|
}
|
2021-08-01 21:10:21 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
size_t mg_ws_send(struct mg_connection *c, const char *buf, size_t len,
|
|
|
|
int op) {
|
|
|
|
uint8_t header[14];
|
|
|
|
size_t header_len = mkhdr(len, op, c->is_client, header);
|
|
|
|
mg_send(c, header, header_len);
|
|
|
|
LOG(LL_VERBOSE_DEBUG, ("WS out: %d [%.*s]", (int) len, (int) len, buf));
|
|
|
|
mg_send(c, buf, len);
|
|
|
|
mg_ws_mask(c, len);
|
2020-12-05 19:26:32 +08:00
|
|
|
return header_len + len;
|
|
|
|
}
|
2016-11-09 06:29:07 +08:00
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
static void mg_ws_cb(struct mg_connection *c, int ev, void *ev_data,
|
|
|
|
void *fn_data) {
|
|
|
|
struct ws_msg msg;
|
2021-08-27 17:25:24 +08:00
|
|
|
size_t ofs = (size_t) c->pfn_data;
|
2016-11-09 06:29:07 +08:00
|
|
|
|
2021-08-27 17:25:24 +08:00
|
|
|
// assert(ofs < c->recv.len);
|
2020-12-05 19:26:32 +08:00
|
|
|
if (ev == MG_EV_READ) {
|
|
|
|
if (!c->is_websocket && c->is_client) {
|
|
|
|
int n = mg_http_get_request_len(c->recv.buf, c->recv.len);
|
|
|
|
if (n < 0) {
|
|
|
|
c->is_closing = 1; // Some just, not an HTTP request
|
|
|
|
} else if (n > 0) {
|
|
|
|
if (n < 15 || memcmp(c->recv.buf + 9, "101", 3) != 0) {
|
2020-12-21 20:26:44 +08:00
|
|
|
LOG(LL_ERROR,
|
|
|
|
("%lu WS handshake error: %.*s", c->id, 15, c->recv.buf));
|
2020-12-05 19:26:32 +08:00
|
|
|
c->is_closing = 1;
|
|
|
|
} else {
|
2021-01-17 19:16:19 +08:00
|
|
|
struct mg_http_message hm;
|
|
|
|
mg_http_parse((char *) c->recv.buf, c->recv.len, &hm);
|
2020-12-05 19:26:32 +08:00
|
|
|
c->is_websocket = 1;
|
2021-01-17 19:16:19 +08:00
|
|
|
mg_call(c, MG_EV_WS_OPEN, &hm);
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
2021-08-28 14:16:38 +08:00
|
|
|
mg_iobuf_del(&c->recv, 0, (size_t) n);
|
2020-12-05 19:26:32 +08:00
|
|
|
} else {
|
|
|
|
return; // A request is not yet received
|
|
|
|
}
|
|
|
|
}
|
2017-08-17 22:07:45 +08:00
|
|
|
|
2021-08-27 17:25:24 +08:00
|
|
|
while (ws_process(c->recv.buf + ofs, c->recv.len - ofs, &msg) > 0) {
|
|
|
|
char *s = (char *) c->recv.buf + ofs + msg.header_len;
|
2020-12-23 19:58:08 +08:00
|
|
|
struct mg_ws_message m = {{s, msg.data_len}, msg.flags};
|
2021-08-27 17:25:24 +08:00
|
|
|
size_t len = msg.header_len + msg.data_len;
|
|
|
|
uint8_t final = msg.flags & 128, op = msg.flags & 15;
|
|
|
|
// LOG(LL_VERBOSE_DEBUG, ("fin %d op %d len %d [%.*s]", final, op,
|
|
|
|
// (int) m.data.len, (int) m.data.len, m.data.ptr));
|
|
|
|
switch (op) {
|
2021-06-11 02:15:50 +08:00
|
|
|
case WEBSOCKET_OP_CONTINUE:
|
|
|
|
mg_call(c, MG_EV_WS_CTL, &m);
|
|
|
|
break;
|
2020-12-05 19:26:32 +08:00
|
|
|
case WEBSOCKET_OP_PING:
|
|
|
|
LOG(LL_DEBUG, ("%s", "WS PONG"));
|
|
|
|
mg_ws_send(c, s, msg.data_len, WEBSOCKET_OP_PONG);
|
2020-12-23 19:58:08 +08:00
|
|
|
mg_call(c, MG_EV_WS_CTL, &m);
|
2020-12-05 19:26:32 +08:00
|
|
|
break;
|
|
|
|
case WEBSOCKET_OP_PONG:
|
2020-12-23 19:58:08 +08:00
|
|
|
mg_call(c, MG_EV_WS_CTL, &m);
|
2020-12-05 19:26:32 +08:00
|
|
|
break;
|
2021-06-11 02:15:50 +08:00
|
|
|
case WEBSOCKET_OP_TEXT:
|
|
|
|
case WEBSOCKET_OP_BINARY:
|
2021-08-27 17:25:24 +08:00
|
|
|
if (final) mg_call(c, MG_EV_WS_MSG, &m);
|
2021-06-11 02:15:50 +08:00
|
|
|
break;
|
2020-12-23 19:58:08 +08:00
|
|
|
case WEBSOCKET_OP_CLOSE:
|
2020-12-21 20:26:44 +08:00
|
|
|
LOG(LL_ERROR, ("%lu Got WS CLOSE", c->id));
|
2020-12-23 19:58:08 +08:00
|
|
|
mg_call(c, MG_EV_WS_CTL, &m);
|
2020-12-05 19:26:32 +08:00
|
|
|
c->is_closing = 1;
|
|
|
|
break;
|
2021-06-11 02:15:50 +08:00
|
|
|
default:
|
|
|
|
// Per RFC6455, close conn when an unknown op is recvd
|
2021-08-27 17:25:24 +08:00
|
|
|
mg_error(c, "unknown WS op %d", op);
|
2021-06-11 02:15:50 +08:00
|
|
|
break;
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
2021-08-27 17:25:24 +08:00
|
|
|
|
|
|
|
// Handle fragmented frames: strip header, keep in c->recv
|
|
|
|
if (final == 0 || op == 0) {
|
|
|
|
if (op) ofs++, len--, msg.header_len--; // First frame
|
|
|
|
mg_iobuf_del(&c->recv, ofs, msg.header_len); // Strip header
|
|
|
|
len -= msg.header_len;
|
|
|
|
ofs += len;
|
|
|
|
c->pfn_data = (void *) ofs;
|
|
|
|
// LOG(LL_INFO, ("FRAG %d [%.*s]", (int) ofs, (int) ofs, c->recv.buf));
|
|
|
|
}
|
|
|
|
// Remove non-fragmented frame
|
|
|
|
if (final && op) mg_iobuf_del(&c->recv, ofs, len);
|
|
|
|
// Last chunk of the fragmented frame
|
|
|
|
if (final && !op) {
|
|
|
|
m.flags = c->recv.buf[0];
|
|
|
|
m.data = mg_str_n((char *) &c->recv.buf[1], (size_t) (ofs - 1));
|
|
|
|
mg_call(c, MG_EV_WS_MSG, &m);
|
|
|
|
mg_iobuf_del(&c->recv, 0, ofs);
|
|
|
|
ofs = 0;
|
|
|
|
c->pfn_data = NULL;
|
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
(void) fn_data;
|
|
|
|
(void) ev_data;
|
2017-08-17 22:07:45 +08:00
|
|
|
}
|
|
|
|
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_connection *mg_ws_connect(struct mg_mgr *mgr, const char *url,
|
|
|
|
mg_event_handler_t fn, void *fn_data,
|
|
|
|
const char *fmt, ...) {
|
|
|
|
struct mg_connection *c = mg_connect(mgr, url, fn, fn_data);
|
|
|
|
if (c != NULL) {
|
|
|
|
char nonce[16], key[30], mem1[128], mem2[256], *buf1 = mem1, *buf2 = mem2;
|
|
|
|
struct mg_str host = mg_url_host(url);
|
|
|
|
int n1 = 0, n2 = 0;
|
|
|
|
if (fmt != NULL) {
|
|
|
|
va_list ap;
|
|
|
|
va_start(ap, fmt);
|
|
|
|
n1 = mg_vasprintf(&buf1, sizeof(mem1), fmt, ap);
|
|
|
|
va_end(ap);
|
|
|
|
}
|
|
|
|
// Send handshake request
|
|
|
|
// mg_url_host(url, host, sizeof(host));
|
|
|
|
mg_random(nonce, sizeof(nonce));
|
|
|
|
mg_base64_encode((unsigned char *) nonce, sizeof(nonce), key);
|
|
|
|
// LOG(LL_DEBUG, "%p [%s]", uri, uri == NULL ? "???" : uri);
|
|
|
|
n2 = mg_asprintf(&buf2, sizeof(mem2),
|
|
|
|
"GET %s HTTP/1.1\r\n"
|
|
|
|
"Upgrade: websocket\r\n"
|
|
|
|
"Host: %.*s\r\n"
|
|
|
|
"Connection: Upgrade\r\n"
|
|
|
|
"%.*s"
|
|
|
|
"Sec-WebSocket-Version: 13\r\n"
|
|
|
|
"Sec-WebSocket-Key: %s\r\n"
|
|
|
|
"\r\n",
|
|
|
|
mg_url_uri(url), (int) host.len, host.ptr, n1, buf1, key);
|
2021-05-29 01:30:42 +08:00
|
|
|
mg_send(c, buf2, n2 > 0 ? (size_t) n2 : 0);
|
2020-12-05 19:26:32 +08:00
|
|
|
if (buf1 != mem1) free(buf1);
|
|
|
|
if (buf2 != mem2) free(buf2);
|
|
|
|
c->pfn = mg_ws_cb;
|
2021-08-27 17:25:24 +08:00
|
|
|
c->pfn_data = NULL;
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
return c;
|
2017-08-17 22:07:45 +08:00
|
|
|
}
|
|
|
|
|
2021-01-17 06:48:43 +08:00
|
|
|
void mg_ws_upgrade(struct mg_connection *c, struct mg_http_message *hm,
|
|
|
|
const char *fmt, ...) {
|
2020-12-05 19:26:32 +08:00
|
|
|
struct mg_str *wskey = mg_http_get_header(hm, "Sec-WebSocket-Key");
|
|
|
|
c->pfn = mg_ws_cb;
|
2021-08-27 17:25:24 +08:00
|
|
|
c->pfn_data = NULL;
|
2021-08-04 06:37:50 +08:00
|
|
|
if (wskey == NULL) {
|
|
|
|
mg_http_reply(c, 426, "", "WS upgrade expected\n");
|
|
|
|
c->is_draining = 1;
|
|
|
|
} else {
|
2021-08-04 18:38:32 +08:00
|
|
|
struct mg_str *wsproto = mg_http_get_header(hm, "Sec-WebSocket-Protocol");
|
2021-01-17 06:48:43 +08:00
|
|
|
va_list ap;
|
|
|
|
va_start(ap, fmt);
|
2021-08-04 18:38:32 +08:00
|
|
|
ws_handshake(c, wskey, wsproto, fmt, ap);
|
2021-01-17 06:48:43 +08:00
|
|
|
va_end(ap);
|
2021-08-04 06:37:50 +08:00
|
|
|
c->is_websocket = 1;
|
2021-08-11 11:52:39 +08:00
|
|
|
mg_call(c, MG_EV_WS_OPEN, hm);
|
2021-01-17 06:48:43 +08:00
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
2021-08-01 21:10:21 +08:00
|
|
|
|
|
|
|
size_t mg_ws_wrap(struct mg_connection *c, size_t len, int op) {
|
|
|
|
uint8_t header[14], *p;
|
|
|
|
size_t header_len = mkhdr(len, op, c->is_client, header);
|
|
|
|
|
|
|
|
// NOTE: order of operations is important!
|
2021-08-28 15:08:54 +08:00
|
|
|
mg_iobuf_add(&c->send, c->send.len, NULL, header_len, MG_IO_SIZE);
|
|
|
|
p = &c->send.buf[c->send.len - len]; // p points to data
|
|
|
|
memmove(p, p - header_len, len); // Shift data
|
|
|
|
memcpy(p - header_len, header, header_len); // Prepend header
|
|
|
|
mg_ws_mask(c, len); // Mask data
|
2021-08-01 21:10:21 +08:00
|
|
|
|
|
|
|
return c->send.len;
|
|
|
|
}
|