2015-09-08 19:49:03 +08:00
|
|
|
#include "mongoose.h"
|
|
|
|
#ifdef NS_MODULE_LINES
|
2015-12-15 02:45:18 +08:00
|
|
|
#line 1 "./src/internal.h"
|
2015-09-08 19:49:03 +08:00
|
|
|
/**/
|
2014-08-08 04:01:50 +08:00
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Copyright (c) 2014 Cesanta Software Limited
|
|
|
|
* All rights reserved
|
|
|
|
*/
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_INTERNAL_HEADER_INCLUDED
|
|
|
|
#define MG_INTERNAL_HEADER_INCLUDED
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_MALLOC
|
|
|
|
#define MG_MALLOC malloc
|
2015-03-26 01:10:58 +08:00
|
|
|
#endif
|
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_CALLOC
|
|
|
|
#define MG_CALLOC calloc
|
2015-03-20 23:33:02 +08:00
|
|
|
#endif
|
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_REALLOC
|
|
|
|
#define MG_REALLOC realloc
|
2015-03-21 18:04:57 +08:00
|
|
|
#endif
|
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_FREE
|
|
|
|
#define MG_FREE free
|
2015-02-24 18:12:09 +08:00
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
#ifndef MBUF_REALLOC
|
2015-09-19 19:38:46 +08:00
|
|
|
#define MBUF_REALLOC MG_REALLOC
|
2014-02-20 00:38:53 +08:00
|
|
|
#endif
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
#ifndef MBUF_FREE
|
2015-09-19 19:38:46 +08:00
|
|
|
#define MBUF_FREE MG_FREE
|
2014-02-20 00:38:53 +08:00
|
|
|
#endif
|
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#define MG_SET_PTRPTR(_ptr, _v) \
|
2015-09-08 19:49:03 +08:00
|
|
|
do { \
|
|
|
|
if (_ptr) *(_ptr) = _v; \
|
|
|
|
} while (0)
|
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_INTERNAL
|
|
|
|
#define MG_INTERNAL static
|
2014-09-10 01:07:55 +08:00
|
|
|
#endif
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#if !defined(MG_MGR_EV_MGR)
|
2015-09-25 23:17:33 +08:00
|
|
|
/*
|
|
|
|
* Switches between different methods of handling sockets. Supported values:
|
|
|
|
* 0 - select()
|
|
|
|
* 1 - epoll() (Linux only)
|
|
|
|
*/
|
2015-09-19 19:38:46 +08:00
|
|
|
#define MG_MGR_EV_MGR 0 /* select() */
|
2014-02-20 00:38:53 +08:00
|
|
|
#endif
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
#ifdef PICOTCP
|
|
|
|
#define NO_LIBC
|
2015-09-19 19:38:46 +08:00
|
|
|
#define MG_DISABLE_FILESYSTEM
|
|
|
|
#define MG_DISABLE_POPEN
|
|
|
|
#define MG_DISABLE_CGI
|
|
|
|
#define MG_DISABLE_DIRECTORY_LISTING
|
|
|
|
#define MG_DISABLE_SOCKETPAIR
|
|
|
|
#define MG_DISABLE_PFS
|
2014-02-20 00:38:53 +08:00
|
|
|
#endif
|
|
|
|
|
2015-12-12 22:16:48 +08:00
|
|
|
/* Amalgamated: #include "mongoose/src/net.h" */
|
|
|
|
/* Amalgamated: #include "mongoose/src/http.h" */
|
|
|
|
|
|
|
|
#define MG_CTL_MSG_MESSAGE_SIZE 8192
|
2014-10-09 16:26:45 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* internals that need to be accessible in unit tests */
|
2015-10-22 20:04:35 +08:00
|
|
|
MG_INTERNAL struct mg_connection *mg_do_connect(struct mg_connection *nc,
|
|
|
|
int proto,
|
|
|
|
union socket_address *sa);
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
MG_INTERNAL int mg_parse_address(const char *str, union socket_address *sa,
|
2015-09-08 19:49:03 +08:00
|
|
|
int *proto, char *host, size_t host_len);
|
2015-10-29 06:12:55 +08:00
|
|
|
MG_INTERNAL void mg_call(struct mg_connection *nc,
|
|
|
|
mg_event_handler_t ev_handler, int ev, void *ev_data);
|
2015-12-22 15:30:04 +08:00
|
|
|
MG_INTERNAL void mg_forward(struct mg_connection *from,
|
|
|
|
struct mg_connection *to);
|
2015-09-19 19:38:46 +08:00
|
|
|
MG_INTERNAL void mg_add_conn(struct mg_mgr *mgr, struct mg_connection *c);
|
|
|
|
MG_INTERNAL void mg_remove_conn(struct mg_connection *c);
|
2015-12-12 22:16:48 +08:00
|
|
|
MG_INTERNAL size_t recv_avail_size(struct mg_connection *conn, size_t max);
|
|
|
|
MG_INTERNAL struct mg_connection *mg_create_connection(
|
|
|
|
struct mg_mgr *mgr, mg_event_handler_t callback,
|
|
|
|
struct mg_add_sock_opts opts);
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_DISABLE_FILESYSTEM
|
2015-12-22 15:30:04 +08:00
|
|
|
MG_INTERNAL int find_index_file(char *path, size_t path_len, const char *list,
|
|
|
|
cs_stat_t *stp);
|
2014-02-20 00:38:53 +08:00
|
|
|
#endif
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
#ifdef _WIN32
|
|
|
|
void to_wchar(const char *path, wchar_t *wbuf, size_t wbuf_len);
|
2014-02-20 00:38:53 +08:00
|
|
|
#endif
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Reassemble the content of the buffer (buf, blen) which should be
|
|
|
|
* in the HTTP chunked encoding, by collapsing data chunks to the
|
|
|
|
* beginning of the buffer.
|
|
|
|
*
|
|
|
|
* If chunks get reassembled, modify hm->body to point to the reassembled
|
2015-09-19 19:38:46 +08:00
|
|
|
* body and fire MG_EV_HTTP_CHUNK event. If handler sets MG_F_DELETE_CHUNK
|
2015-09-08 19:49:03 +08:00
|
|
|
* in nc->flags, delete reassembled body from the mbuf.
|
|
|
|
*
|
|
|
|
* Return reassembled body size.
|
|
|
|
*/
|
2015-09-19 19:38:46 +08:00
|
|
|
MG_INTERNAL size_t mg_handle_chunked(struct mg_connection *nc,
|
2015-09-08 19:49:03 +08:00
|
|
|
struct http_message *hm, char *buf,
|
|
|
|
size_t blen);
|
|
|
|
|
2015-10-31 21:44:54 +08:00
|
|
|
#ifndef MG_DISABLE_FILESYSTEM
|
|
|
|
MG_INTERNAL time_t mg_parse_date_string(const char *datetime);
|
2015-11-18 06:25:04 +08:00
|
|
|
MG_INTERNAL int mg_is_not_modified(struct http_message *hm, cs_stat_t *st);
|
2015-10-31 21:44:54 +08:00
|
|
|
#endif
|
|
|
|
|
2015-12-12 22:16:48 +08:00
|
|
|
struct ctl_msg {
|
|
|
|
mg_event_handler_t callback;
|
|
|
|
char message[MG_CTL_MSG_MESSAGE_SIZE];
|
|
|
|
};
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Forward declarations for testing. */
|
2015-12-22 15:30:04 +08:00
|
|
|
extern void *(*test_malloc)(size_t size);
|
|
|
|
extern void *(*test_calloc)(size_t count, size_t size);
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#endif /* MG_INTERNAL_HEADER_INCLUDED */
|
2015-09-08 19:49:03 +08:00
|
|
|
#ifdef NS_MODULE_LINES
|
2015-12-15 02:45:18 +08:00
|
|
|
#line 1 "./src/../../common/base64.c"
|
2015-09-08 19:49:03 +08:00
|
|
|
/**/
|
2014-02-20 00:38:53 +08:00
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Copyright (c) 2014 Cesanta Software Limited
|
|
|
|
* All rights reserved
|
|
|
|
*/
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
#ifndef EXCLUDE_COMMON
|
|
|
|
|
2015-12-12 22:05:54 +08:00
|
|
|
/* Amalgamated: #include "common/base64.h" */
|
2015-09-08 19:49:03 +08:00
|
|
|
#include <string.h>
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
/* ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/ */
|
2014-11-07 19:36:02 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
#define NUM_UPPERCASES ('Z' - 'A' + 1)
|
|
|
|
#define NUM_LETTERS (NUM_UPPERCASES * 2)
|
|
|
|
#define NUM_DIGITS ('9' - '0' + 1)
|
2014-09-10 01:07:55 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
/*
|
|
|
|
* Emit a base64 code char.
|
|
|
|
*
|
|
|
|
* Doesn't use memory, thus it's safe to use to safely dump memory in crashdumps
|
|
|
|
*/
|
|
|
|
static void cs_base64_emit_code(struct cs_base64_ctx *ctx, int v) {
|
|
|
|
if (v < NUM_UPPERCASES) {
|
|
|
|
ctx->b64_putc(v + 'A', ctx->user_data);
|
|
|
|
} else if (v < (NUM_LETTERS)) {
|
|
|
|
ctx->b64_putc(v - NUM_UPPERCASES + 'a', ctx->user_data);
|
|
|
|
} else if (v < (NUM_LETTERS + NUM_DIGITS)) {
|
|
|
|
ctx->b64_putc(v - NUM_LETTERS + '0', ctx->user_data);
|
|
|
|
} else {
|
|
|
|
ctx->b64_putc(v - NUM_LETTERS - NUM_DIGITS == 0 ? '+' : '/',
|
|
|
|
ctx->user_data);
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2014-05-13 06:22:57 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
static void cs_base64_emit_chunk(struct cs_base64_ctx *ctx) {
|
|
|
|
int a, b, c;
|
|
|
|
|
|
|
|
a = ctx->chunk[0];
|
|
|
|
b = ctx->chunk[1];
|
|
|
|
c = ctx->chunk[2];
|
|
|
|
|
|
|
|
cs_base64_emit_code(ctx, a >> 2);
|
|
|
|
cs_base64_emit_code(ctx, ((a & 3) << 4) | (b >> 4));
|
|
|
|
if (ctx->chunk_size > 1) {
|
|
|
|
cs_base64_emit_code(ctx, (b & 15) << 2 | (c >> 6));
|
|
|
|
}
|
|
|
|
if (ctx->chunk_size > 2) {
|
|
|
|
cs_base64_emit_code(ctx, c & 63);
|
2014-10-09 16:26:45 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
void cs_base64_init(struct cs_base64_ctx *ctx, cs_base64_putc_t b64_putc,
|
|
|
|
void *user_data) {
|
|
|
|
ctx->chunk_size = 0;
|
|
|
|
ctx->b64_putc = b64_putc;
|
|
|
|
ctx->user_data = user_data;
|
|
|
|
}
|
|
|
|
|
|
|
|
void cs_base64_update(struct cs_base64_ctx *ctx, const char *str, size_t len) {
|
|
|
|
const unsigned char *src = (const unsigned char *) str;
|
|
|
|
size_t i;
|
|
|
|
for (i = 0; i < len; i++) {
|
|
|
|
ctx->chunk[ctx->chunk_size++] = src[i];
|
|
|
|
if (ctx->chunk_size == 3) {
|
|
|
|
cs_base64_emit_chunk(ctx);
|
|
|
|
ctx->chunk_size = 0;
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
void cs_base64_finish(struct cs_base64_ctx *ctx) {
|
|
|
|
if (ctx->chunk_size > 0) {
|
|
|
|
int i;
|
|
|
|
memset(&ctx->chunk[ctx->chunk_size], 0, 3 - ctx->chunk_size);
|
|
|
|
cs_base64_emit_chunk(ctx);
|
|
|
|
for (i = 0; i < (3 - ctx->chunk_size); i++) {
|
|
|
|
ctx->b64_putc('=', ctx->user_data);
|
|
|
|
}
|
|
|
|
}
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
#define BASE64_ENCODE_BODY \
|
|
|
|
static const char *b64 = \
|
|
|
|
"ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"; \
|
|
|
|
int i, j, a, b, c; \
|
|
|
|
\
|
|
|
|
for (i = j = 0; i < src_len; i += 3) { \
|
|
|
|
a = src[i]; \
|
|
|
|
b = i + 1 >= src_len ? 0 : src[i + 1]; \
|
|
|
|
c = i + 2 >= src_len ? 0 : src[i + 2]; \
|
|
|
|
\
|
|
|
|
BASE64_OUT(b64[a >> 2]); \
|
|
|
|
BASE64_OUT(b64[((a & 3) << 4) | (b >> 4)]); \
|
|
|
|
if (i + 1 < src_len) { \
|
|
|
|
BASE64_OUT(b64[(b & 15) << 2 | (c >> 6)]); \
|
|
|
|
} \
|
|
|
|
if (i + 2 < src_len) { \
|
|
|
|
BASE64_OUT(b64[c & 63]); \
|
|
|
|
} \
|
|
|
|
} \
|
|
|
|
\
|
|
|
|
while (j % 4 != 0) { \
|
|
|
|
BASE64_OUT('='); \
|
|
|
|
} \
|
|
|
|
BASE64_FLUSH()
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
#define BASE64_OUT(ch) \
|
|
|
|
do { \
|
|
|
|
dst[j++] = (ch); \
|
|
|
|
} while (0)
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
#define BASE64_FLUSH() \
|
|
|
|
do { \
|
|
|
|
dst[j++] = '\0'; \
|
|
|
|
} while (0)
|
2015-02-13 03:12:37 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
void cs_base64_encode(const unsigned char *src, int src_len, char *dst) {
|
|
|
|
BASE64_ENCODE_BODY;
|
|
|
|
}
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
#undef BASE64_OUT
|
|
|
|
#undef BASE64_FLUSH
|
|
|
|
|
|
|
|
#define BASE64_OUT(ch) \
|
|
|
|
do { \
|
|
|
|
fprintf(f, "%c", (ch)); \
|
|
|
|
j++; \
|
|
|
|
} while (0)
|
|
|
|
|
|
|
|
#define BASE64_FLUSH()
|
|
|
|
|
|
|
|
void cs_fprint_base64(FILE *f, const unsigned char *src, int src_len) {
|
|
|
|
BASE64_ENCODE_BODY;
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
#undef BASE64_OUT
|
|
|
|
#undef BASE64_FLUSH
|
|
|
|
|
|
|
|
/* Convert one byte of encoded base64 input stream to 6-bit chunk */
|
|
|
|
static unsigned char from_b64(unsigned char ch) {
|
|
|
|
/* Inverse lookup map */
|
|
|
|
static const unsigned char tab[128] = {
|
|
|
|
255, 255, 255, 255,
|
|
|
|
255, 255, 255, 255, /* 0 */
|
|
|
|
255, 255, 255, 255,
|
|
|
|
255, 255, 255, 255, /* 8 */
|
|
|
|
255, 255, 255, 255,
|
|
|
|
255, 255, 255, 255, /* 16 */
|
|
|
|
255, 255, 255, 255,
|
|
|
|
255, 255, 255, 255, /* 24 */
|
|
|
|
255, 255, 255, 255,
|
|
|
|
255, 255, 255, 255, /* 32 */
|
|
|
|
255, 255, 255, 62,
|
|
|
|
255, 255, 255, 63, /* 40 */
|
|
|
|
52, 53, 54, 55,
|
|
|
|
56, 57, 58, 59, /* 48 */
|
|
|
|
60, 61, 255, 255,
|
|
|
|
255, 200, 255, 255, /* 56 '=' is 200, on index 61 */
|
|
|
|
255, 0, 1, 2,
|
|
|
|
3, 4, 5, 6, /* 64 */
|
|
|
|
7, 8, 9, 10,
|
|
|
|
11, 12, 13, 14, /* 72 */
|
|
|
|
15, 16, 17, 18,
|
|
|
|
19, 20, 21, 22, /* 80 */
|
|
|
|
23, 24, 25, 255,
|
|
|
|
255, 255, 255, 255, /* 88 */
|
|
|
|
255, 26, 27, 28,
|
|
|
|
29, 30, 31, 32, /* 96 */
|
|
|
|
33, 34, 35, 36,
|
|
|
|
37, 38, 39, 40, /* 104 */
|
|
|
|
41, 42, 43, 44,
|
|
|
|
45, 46, 47, 48, /* 112 */
|
|
|
|
49, 50, 51, 255,
|
|
|
|
255, 255, 255, 255, /* 120 */
|
|
|
|
};
|
|
|
|
return tab[ch & 127];
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
int cs_base64_decode(const unsigned char *s, int len, char *dst) {
|
|
|
|
unsigned char a, b, c, d;
|
|
|
|
int orig_len = len;
|
|
|
|
while (len >= 4 && (a = from_b64(s[0])) != 255 &&
|
|
|
|
(b = from_b64(s[1])) != 255 && (c = from_b64(s[2])) != 255 &&
|
|
|
|
(d = from_b64(s[3])) != 255) {
|
|
|
|
s += 4;
|
|
|
|
len -= 4;
|
|
|
|
if (a == 200 || b == 200) break; /* '=' can't be there */
|
|
|
|
*dst++ = a << 2 | b >> 4;
|
|
|
|
if (c == 200) break;
|
|
|
|
*dst++ = b << 4 | c >> 2;
|
|
|
|
if (d == 200) break;
|
|
|
|
*dst++ = c << 6 | d;
|
2014-09-10 01:07:55 +08:00
|
|
|
}
|
2015-11-01 00:26:19 +08:00
|
|
|
*dst = 0;
|
|
|
|
return orig_len - len;
|
2014-09-10 01:07:55 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
#endif /* EXCLUDE_COMMON */
|
|
|
|
#ifdef NS_MODULE_LINES
|
2015-12-15 02:45:18 +08:00
|
|
|
#line 1 "./src/../../common/cs_dbg.c"
|
2015-09-08 19:49:03 +08:00
|
|
|
/**/
|
|
|
|
#endif
|
2015-12-12 22:05:54 +08:00
|
|
|
/* Amalgamated: #include "common/cs_dbg.h" */
|
2015-11-20 00:20:45 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
#include <stdarg.h>
|
|
|
|
#include <stdio.h>
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-20 00:20:45 +08:00
|
|
|
enum cs_log_level s_cs_log_level =
|
|
|
|
#ifdef CS_ENABLE_DEBUG
|
2015-12-07 23:56:13 +08:00
|
|
|
LL_VERBOSE_DEBUG;
|
2015-11-20 00:20:45 +08:00
|
|
|
#else
|
|
|
|
LL_ERROR;
|
|
|
|
#endif
|
|
|
|
|
|
|
|
void cs_log_printf(const char *fmt, ...) {
|
2015-11-01 00:26:19 +08:00
|
|
|
va_list ap;
|
|
|
|
va_start(ap, fmt);
|
|
|
|
vfprintf(stderr, fmt, ap);
|
|
|
|
va_end(ap);
|
|
|
|
fputc('\n', stderr);
|
|
|
|
fflush(stderr);
|
|
|
|
}
|
2015-11-20 00:20:45 +08:00
|
|
|
|
|
|
|
void cs_log_set_level(enum cs_log_level level) {
|
|
|
|
s_cs_log_level = level;
|
|
|
|
}
|
2015-11-01 00:26:19 +08:00
|
|
|
#ifdef NS_MODULE_LINES
|
2015-12-15 02:45:18 +08:00
|
|
|
#line 1 "./src/../../common/cs_dirent.c"
|
2015-11-01 00:26:19 +08:00
|
|
|
/**/
|
2014-02-20 00:38:53 +08:00
|
|
|
#endif
|
2015-11-01 00:26:19 +08:00
|
|
|
/*
|
|
|
|
* Copyright (c) 2015 Cesanta Software Limited
|
|
|
|
* All rights reserved
|
|
|
|
*/
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
#ifndef EXCLUDE_COMMON
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-12-12 22:05:54 +08:00
|
|
|
/* Amalgamated: #include "common/osdep.h" */
|
|
|
|
/* Amalgamated: #include "common/cs_dirent.h" */
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
/*
|
|
|
|
* This file contains POSIX opendir/closedir/readdir API implementation
|
|
|
|
* for systems which do not natively support it (e.g. Windows).
|
|
|
|
*/
|
|
|
|
|
|
|
|
#ifndef MG_FREE
|
|
|
|
#define MG_FREE free
|
2014-02-20 00:38:53 +08:00
|
|
|
#endif
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
#ifndef MG_MALLOC
|
|
|
|
#define MG_MALLOC malloc
|
|
|
|
#endif
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
#ifdef _WIN32
|
|
|
|
DIR *opendir(const char *name) {
|
|
|
|
DIR *dir = NULL;
|
|
|
|
wchar_t wpath[MAX_PATH];
|
|
|
|
DWORD attrs;
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
if (name == NULL) {
|
|
|
|
SetLastError(ERROR_BAD_ARGUMENTS);
|
|
|
|
} else if ((dir = (DIR *) MG_MALLOC(sizeof(*dir))) == NULL) {
|
|
|
|
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
|
|
|
|
} else {
|
|
|
|
to_wchar(name, wpath, ARRAY_SIZE(wpath));
|
|
|
|
attrs = GetFileAttributesW(wpath);
|
|
|
|
if (attrs != 0xFFFFFFFF && (attrs & FILE_ATTRIBUTE_DIRECTORY)) {
|
|
|
|
(void) wcscat(wpath, L"\\*");
|
|
|
|
dir->handle = FindFirstFileW(wpath, &dir->info);
|
|
|
|
dir->result.d_name[0] = '\0';
|
|
|
|
} else {
|
|
|
|
MG_FREE(dir);
|
|
|
|
dir = NULL;
|
|
|
|
}
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
return dir;
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
int closedir(DIR *dir) {
|
|
|
|
int result = 0;
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
if (dir != NULL) {
|
|
|
|
if (dir->handle != INVALID_HANDLE_VALUE)
|
|
|
|
result = FindClose(dir->handle) ? 0 : -1;
|
|
|
|
MG_FREE(dir);
|
|
|
|
} else {
|
|
|
|
result = -1;
|
|
|
|
SetLastError(ERROR_BAD_ARGUMENTS);
|
|
|
|
}
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
return result;
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
struct dirent *readdir(DIR *dir) {
|
|
|
|
struct dirent *result = 0;
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
if (dir) {
|
|
|
|
if (dir->handle != INVALID_HANDLE_VALUE) {
|
|
|
|
result = &dir->result;
|
|
|
|
(void) WideCharToMultiByte(CP_UTF8, 0, dir->info.cFileName, -1,
|
|
|
|
result->d_name, sizeof(result->d_name), NULL,
|
|
|
|
NULL);
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
if (!FindNextFileW(dir->handle, &dir->info)) {
|
|
|
|
(void) FindClose(dir->handle);
|
|
|
|
dir->handle = INVALID_HANDLE_VALUE;
|
|
|
|
}
|
2014-08-02 00:35:59 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
} else {
|
|
|
|
SetLastError(ERROR_FILE_NOT_FOUND);
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
SetLastError(ERROR_BAD_ARGUMENTS);
|
2014-08-02 00:35:59 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
return result;
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
2015-11-01 00:26:19 +08:00
|
|
|
#endif
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-12-08 22:16:56 +08:00
|
|
|
#ifdef CS_ENABLE_SPIFFS
|
|
|
|
|
|
|
|
DIR *opendir(const char *dir_name) {
|
|
|
|
DIR *dir = NULL;
|
|
|
|
extern spiffs fs;
|
|
|
|
|
|
|
|
if (dir_name != NULL && (dir = (DIR *) malloc(sizeof(*dir))) != NULL &&
|
|
|
|
SPIFFS_opendir(&fs, (char *) dir_name, &dir->dh) == NULL) {
|
|
|
|
free(dir);
|
|
|
|
dir = NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
return dir;
|
|
|
|
}
|
|
|
|
|
|
|
|
int closedir(DIR *dir) {
|
|
|
|
if (dir != NULL) {
|
|
|
|
SPIFFS_closedir(&dir->dh);
|
|
|
|
free(dir);
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
struct dirent *readdir(DIR *dir) {
|
|
|
|
return SPIFFS_readdir(&dir->dh, &dir->de);
|
|
|
|
}
|
|
|
|
|
|
|
|
/* SPIFFs doesn't support directory operations */
|
|
|
|
int rmdir(const char *path) {
|
|
|
|
(void) path;
|
|
|
|
return ENOTDIR;
|
|
|
|
}
|
|
|
|
|
|
|
|
int mkdir(const char *path, mode_t mode) {
|
|
|
|
(void) path;
|
|
|
|
(void) mode;
|
|
|
|
/* for spiffs supports only root dir, which comes from mongoose as '.' */
|
|
|
|
return (strlen(path) == 1 && *path == '.') ? 0 : ENOTDIR;
|
|
|
|
}
|
|
|
|
|
|
|
|
#endif /* CS_ENABLE_SPIFFS */
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
#endif /* EXCLUDE_COMMON */
|
|
|
|
#ifdef NS_MODULE_LINES
|
2015-12-16 08:52:37 +08:00
|
|
|
#line 1 "./src/../../common/cs_time.c"
|
|
|
|
/**/
|
|
|
|
#endif
|
|
|
|
#ifndef _WIN32
|
|
|
|
#include <stddef.h>
|
|
|
|
#ifndef MG_CC3200
|
|
|
|
#include <sys/time.h>
|
|
|
|
#endif
|
|
|
|
#else
|
|
|
|
#include <windows.h>
|
|
|
|
#endif
|
|
|
|
|
|
|
|
double cs_time() {
|
|
|
|
double now;
|
|
|
|
#ifndef _WIN32
|
|
|
|
struct timeval tv;
|
|
|
|
if (gettimeofday(&tv, NULL /* tz */) != 0) return 0;
|
|
|
|
now = (double) tv.tv_sec + (((double) tv.tv_usec) / 1000000.0);
|
|
|
|
#else
|
|
|
|
now = GetTickCount() / 1000.0;
|
|
|
|
#endif
|
|
|
|
return now;
|
|
|
|
}
|
|
|
|
#ifdef NS_MODULE_LINES
|
2015-12-15 02:45:18 +08:00
|
|
|
#line 1 "./src/../deps/frozen/frozen.c"
|
2015-09-08 19:49:03 +08:00
|
|
|
/**/
|
2014-04-16 23:45:31 +08:00
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
2015-11-01 00:26:19 +08:00
|
|
|
* Copyright (c) 2004-2013 Sergey Lyubka <valenok@gmail.com>
|
|
|
|
* Copyright (c) 2013 Cesanta Software Limited
|
|
|
|
* All rights reserved
|
2015-09-08 19:49:03 +08:00
|
|
|
*
|
2015-11-01 00:26:19 +08:00
|
|
|
* This library 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/>.
|
2015-09-08 19:49:03 +08:00
|
|
|
*
|
2015-11-01 00:26:19 +08:00
|
|
|
* You are free to use this library 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 library under a commercial
|
|
|
|
* license, as set out in <https://www.cesanta.com/license>.
|
2015-09-08 19:49:03 +08:00
|
|
|
*/
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
#define _CRT_SECURE_NO_WARNINGS /* Disable deprecation warning in VS2005+ */
|
2014-09-10 01:07:55 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
#include <stdio.h>
|
|
|
|
#include <stdlib.h>
|
|
|
|
#include <string.h>
|
|
|
|
#include <stdarg.h>
|
|
|
|
/* Amalgamated: #include "frozen.h" */
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
#ifdef _WIN32
|
|
|
|
#define snprintf _snprintf
|
2014-02-20 00:38:53 +08:00
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
#ifndef FROZEN_REALLOC
|
|
|
|
#define FROZEN_REALLOC realloc
|
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
#ifndef FROZEN_FREE
|
|
|
|
#define FROZEN_FREE free
|
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
struct frozen {
|
|
|
|
const char *end;
|
|
|
|
const char *cur;
|
|
|
|
struct json_token *tokens;
|
|
|
|
int max_tokens;
|
|
|
|
int num_tokens;
|
|
|
|
int do_realloc;
|
|
|
|
};
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
static int parse_object(struct frozen *f);
|
|
|
|
static int parse_value(struct frozen *f);
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
#define EXPECT(cond, err_code) do { if (!(cond)) return (err_code); } while (0)
|
|
|
|
#define TRY(expr) do { int _n = expr; if (_n < 0) return _n; } while (0)
|
|
|
|
#define END_OF_STRING (-1)
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
static int left(const struct frozen *f) {
|
|
|
|
return f->end - f->cur;
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
static int is_space(int ch) {
|
|
|
|
return ch == ' ' || ch == '\t' || ch == '\r' || ch == '\n';
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
static void skip_whitespaces(struct frozen *f) {
|
|
|
|
while (f->cur < f->end && is_space(*f->cur)) f->cur++;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
static int cur(struct frozen *f) {
|
|
|
|
skip_whitespaces(f);
|
|
|
|
return f->cur >= f->end ? END_OF_STRING : * (unsigned char *) f->cur;
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
static int test_and_skip(struct frozen *f, int expected) {
|
|
|
|
int ch = cur(f);
|
|
|
|
if (ch == expected) { f->cur++; return 0; }
|
|
|
|
return ch == END_OF_STRING ? JSON_STRING_INCOMPLETE : JSON_STRING_INVALID;
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
static int is_alpha(int ch) {
|
|
|
|
return (ch >= 'a' && ch <= 'z') || (ch >= 'A' && ch <= 'Z');
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
static int is_digit(int ch) {
|
|
|
|
return ch >= '0' && ch <= '9';
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
static int is_hex_digit(int ch) {
|
|
|
|
return is_digit(ch) || (ch >= 'a' && ch <= 'f') || (ch >= 'A' && ch <= 'F');
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
static int get_escape_len(const char *s, int len) {
|
|
|
|
switch (*s) {
|
|
|
|
case 'u':
|
|
|
|
return len < 6 ? JSON_STRING_INCOMPLETE :
|
|
|
|
is_hex_digit(s[1]) && is_hex_digit(s[2]) &&
|
|
|
|
is_hex_digit(s[3]) && is_hex_digit(s[4]) ? 5 : JSON_STRING_INVALID;
|
|
|
|
case '"': case '\\': case '/': case 'b':
|
|
|
|
case 'f': case 'n': case 'r': case 't':
|
|
|
|
return len < 2 ? JSON_STRING_INCOMPLETE : 1;
|
|
|
|
default:
|
|
|
|
return JSON_STRING_INVALID;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
static int capture_ptr(struct frozen *f, const char *ptr, enum json_type type) {
|
|
|
|
if (f->do_realloc && f->num_tokens >= f->max_tokens) {
|
|
|
|
int new_size = f->max_tokens == 0 ? 100 : f->max_tokens * 2;
|
|
|
|
void *p = FROZEN_REALLOC(f->tokens, new_size * sizeof(f->tokens[0]));
|
|
|
|
if (p == NULL) return JSON_TOKEN_ARRAY_TOO_SMALL;
|
|
|
|
f->max_tokens = new_size;
|
|
|
|
f->tokens = (struct json_token *) p;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-11-01 00:26:19 +08:00
|
|
|
if (f->tokens == NULL || f->max_tokens == 0) return 0;
|
|
|
|
if (f->num_tokens >= f->max_tokens) return JSON_TOKEN_ARRAY_TOO_SMALL;
|
|
|
|
f->tokens[f->num_tokens].ptr = ptr;
|
|
|
|
f->tokens[f->num_tokens].type = type;
|
|
|
|
f->num_tokens++;
|
|
|
|
return 0;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
static int capture_len(struct frozen *f, int token_index, const char *ptr) {
|
|
|
|
if (f->tokens == 0 || f->max_tokens == 0) return 0;
|
|
|
|
EXPECT(token_index >= 0 && token_index < f->max_tokens, JSON_STRING_INVALID);
|
|
|
|
f->tokens[token_index].len = ptr - f->tokens[token_index].ptr;
|
|
|
|
f->tokens[token_index].num_desc = (f->num_tokens - 1) - token_index;
|
|
|
|
return 0;
|
|
|
|
}
|
2015-10-20 17:31:23 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
/* identifier = letter { letter | digit | '_' } */
|
|
|
|
static int parse_identifier(struct frozen *f) {
|
|
|
|
EXPECT(is_alpha(cur(f)), JSON_STRING_INVALID);
|
|
|
|
TRY(capture_ptr(f, f->cur, JSON_TYPE_STRING));
|
|
|
|
while (f->cur < f->end &&
|
|
|
|
(*f->cur == '_' || is_alpha(*f->cur) || is_digit(*f->cur))) {
|
|
|
|
f->cur++;
|
2015-10-20 17:31:23 +08:00
|
|
|
}
|
2015-11-01 00:26:19 +08:00
|
|
|
capture_len(f, f->num_tokens - 1, f->cur);
|
|
|
|
return 0;
|
2015-10-20 17:31:23 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
static int get_utf8_char_len(unsigned char ch) {
|
|
|
|
if ((ch & 0x80) == 0) return 1;
|
|
|
|
switch (ch & 0xf0) {
|
|
|
|
case 0xf0: return 4;
|
|
|
|
case 0xe0: return 3;
|
|
|
|
default: return 2;
|
|
|
|
}
|
|
|
|
}
|
2015-10-20 17:31:23 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
/* string = '"' { quoted_printable_chars } '"' */
|
|
|
|
static int parse_string(struct frozen *f) {
|
|
|
|
int n, ch = 0, len = 0;
|
|
|
|
TRY(test_and_skip(f, '"'));
|
|
|
|
TRY(capture_ptr(f, f->cur, JSON_TYPE_STRING));
|
|
|
|
for (; f->cur < f->end; f->cur += len) {
|
|
|
|
ch = * (unsigned char *) f->cur;
|
|
|
|
len = get_utf8_char_len((unsigned char) ch);
|
|
|
|
EXPECT(ch >= 32 && len > 0, JSON_STRING_INVALID); /* No control chars */
|
|
|
|
EXPECT(len < left(f), JSON_STRING_INCOMPLETE);
|
|
|
|
if (ch == '\\') {
|
|
|
|
EXPECT((n = get_escape_len(f->cur + 1, left(f))) > 0, n);
|
|
|
|
len += n;
|
|
|
|
} else if (ch == '"') {
|
|
|
|
capture_len(f, f->num_tokens - 1, f->cur);
|
|
|
|
f->cur++;
|
|
|
|
break;
|
|
|
|
};
|
|
|
|
}
|
|
|
|
return ch == '"' ? 0 : JSON_STRING_INCOMPLETE;
|
|
|
|
}
|
2015-10-20 17:31:23 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
/* number = [ '-' ] digit+ [ '.' digit+ ] [ ['e'|'E'] ['+'|'-'] digit+ ] */
|
|
|
|
static int parse_number(struct frozen *f) {
|
|
|
|
int ch = cur(f);
|
|
|
|
TRY(capture_ptr(f, f->cur, JSON_TYPE_NUMBER));
|
|
|
|
if (ch == '-') f->cur++;
|
|
|
|
EXPECT(f->cur < f->end, JSON_STRING_INCOMPLETE);
|
|
|
|
EXPECT(is_digit(f->cur[0]), JSON_STRING_INVALID);
|
|
|
|
while (f->cur < f->end && is_digit(f->cur[0])) f->cur++;
|
|
|
|
if (f->cur < f->end && f->cur[0] == '.') {
|
|
|
|
f->cur++;
|
|
|
|
EXPECT(f->cur < f->end, JSON_STRING_INCOMPLETE);
|
|
|
|
EXPECT(is_digit(f->cur[0]), JSON_STRING_INVALID);
|
|
|
|
while (f->cur < f->end && is_digit(f->cur[0])) f->cur++;
|
2015-10-20 17:31:23 +08:00
|
|
|
}
|
2015-11-01 00:26:19 +08:00
|
|
|
if (f->cur < f->end && (f->cur[0] == 'e' || f->cur[0] == 'E')) {
|
|
|
|
f->cur++;
|
|
|
|
EXPECT(f->cur < f->end, JSON_STRING_INCOMPLETE);
|
|
|
|
if ((f->cur[0] == '+' || f->cur[0] == '-')) f->cur++;
|
|
|
|
EXPECT(f->cur < f->end, JSON_STRING_INCOMPLETE);
|
|
|
|
EXPECT(is_digit(f->cur[0]), JSON_STRING_INVALID);
|
|
|
|
while (f->cur < f->end && is_digit(f->cur[0])) f->cur++;
|
|
|
|
}
|
|
|
|
capture_len(f, f->num_tokens - 1, f->cur);
|
|
|
|
return 0;
|
|
|
|
}
|
2015-10-20 17:31:23 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
/* array = '[' [ value { ',' value } ] ']' */
|
|
|
|
static int parse_array(struct frozen *f) {
|
|
|
|
int ind;
|
|
|
|
TRY(test_and_skip(f, '['));
|
|
|
|
TRY(capture_ptr(f, f->cur - 1, JSON_TYPE_ARRAY));
|
|
|
|
ind = f->num_tokens - 1;
|
|
|
|
while (cur(f) != ']') {
|
|
|
|
TRY(parse_value(f));
|
|
|
|
if (cur(f) == ',') f->cur++;
|
|
|
|
}
|
|
|
|
TRY(test_and_skip(f, ']'));
|
|
|
|
capture_len(f, ind, f->cur);
|
|
|
|
return 0;
|
|
|
|
}
|
2015-10-20 17:31:23 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
static int compare(const char *s, const char *str, int len) {
|
|
|
|
int i = 0;
|
|
|
|
while (i < len && s[i] == str[i]) i++;
|
|
|
|
return i == len ? 1 : 0;
|
2015-10-20 17:31:23 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
static int expect(struct frozen *f, const char *s, int len, enum json_type t) {
|
|
|
|
int i, n = left(f);
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
TRY(capture_ptr(f, f->cur, t));
|
|
|
|
for (i = 0; i < len; i++) {
|
|
|
|
if (i >= n) return JSON_STRING_INCOMPLETE;
|
|
|
|
if (f->cur[i] != s[i]) return JSON_STRING_INVALID;
|
|
|
|
}
|
|
|
|
f->cur += len;
|
|
|
|
TRY(capture_len(f, f->num_tokens - 1, f->cur));
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
return 0;
|
|
|
|
}
|
2015-09-12 23:58:45 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
/* value = 'null' | 'true' | 'false' | number | string | array | object */
|
|
|
|
static int parse_value(struct frozen *f) {
|
|
|
|
int ch = cur(f);
|
2015-09-12 23:58:45 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
switch (ch) {
|
|
|
|
case '"': TRY(parse_string(f)); break;
|
|
|
|
case '{': TRY(parse_object(f)); break;
|
|
|
|
case '[': TRY(parse_array(f)); break;
|
|
|
|
case 'n': TRY(expect(f, "null", 4, JSON_TYPE_NULL)); break;
|
|
|
|
case 't': TRY(expect(f, "true", 4, JSON_TYPE_TRUE)); break;
|
|
|
|
case 'f': TRY(expect(f, "false", 5, JSON_TYPE_FALSE)); break;
|
|
|
|
case '-': case '0': case '1': case '2': case '3': case '4':
|
|
|
|
case '5': case '6': case '7': case '8': case '9':
|
|
|
|
TRY(parse_number(f));
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
return ch == END_OF_STRING ? JSON_STRING_INCOMPLETE : JSON_STRING_INVALID;
|
|
|
|
}
|
2015-09-12 23:58:45 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* key = identifier | string */
|
|
|
|
static int parse_key(struct frozen *f) {
|
|
|
|
int ch = cur(f);
|
|
|
|
#if 0
|
|
|
|
printf("%s 1 [%.*s]\n", __func__, (int) (f->end - f->cur), f->cur);
|
|
|
|
#endif
|
|
|
|
if (is_alpha(ch)) {
|
|
|
|
TRY(parse_identifier(f));
|
|
|
|
} else if (ch == '"') {
|
|
|
|
TRY(parse_string(f));
|
2015-09-12 23:58:45 +08:00
|
|
|
} else {
|
2015-11-01 00:26:19 +08:00
|
|
|
return ch == END_OF_STRING ? JSON_STRING_INCOMPLETE : JSON_STRING_INVALID;
|
2015-09-12 23:58:45 +08:00
|
|
|
}
|
2015-11-01 00:26:19 +08:00
|
|
|
return 0;
|
2015-09-12 23:58:45 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
/* pair = key ':' value */
|
|
|
|
static int parse_pair(struct frozen *f) {
|
|
|
|
TRY(parse_key(f));
|
|
|
|
TRY(test_and_skip(f, ':'));
|
|
|
|
TRY(parse_value(f));
|
|
|
|
return 0;
|
|
|
|
}
|
2015-09-12 23:58:45 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
/* object = '{' pair { ',' pair } '}' */
|
|
|
|
static int parse_object(struct frozen *f) {
|
|
|
|
int ind;
|
|
|
|
TRY(test_and_skip(f, '{'));
|
|
|
|
TRY(capture_ptr(f, f->cur - 1, JSON_TYPE_OBJECT));
|
|
|
|
ind = f->num_tokens - 1;
|
|
|
|
while (cur(f) != '}') {
|
|
|
|
TRY(parse_pair(f));
|
|
|
|
if (cur(f) == ',') f->cur++;
|
2015-09-12 23:58:45 +08:00
|
|
|
}
|
2015-11-01 00:26:19 +08:00
|
|
|
TRY(test_and_skip(f, '}'));
|
|
|
|
capture_len(f, ind, f->cur);
|
|
|
|
return 0;
|
2015-09-12 23:58:45 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
static int doit(struct frozen *f) {
|
|
|
|
if (f->cur == 0 || f->end < f->cur) return JSON_STRING_INVALID;
|
|
|
|
if (f->end == f->cur) return JSON_STRING_INCOMPLETE;
|
|
|
|
TRY(parse_object(f));
|
|
|
|
TRY(capture_ptr(f, f->cur, JSON_TYPE_EOF));
|
|
|
|
capture_len(f, f->num_tokens, f->cur);
|
|
|
|
return 0;
|
2015-09-12 23:58:45 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
/* json = object */
|
|
|
|
int parse_json(const char *s, int s_len, struct json_token *arr, int arr_len) {
|
|
|
|
struct frozen frozen;
|
2015-09-08 23:46:54 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
memset(&frozen, 0, sizeof(frozen));
|
|
|
|
frozen.end = s + s_len;
|
|
|
|
frozen.cur = s;
|
|
|
|
frozen.tokens = arr;
|
|
|
|
frozen.max_tokens = arr_len;
|
2015-09-08 23:46:54 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
TRY(doit(&frozen));
|
2015-09-08 23:46:54 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
return frozen.cur - s;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
struct json_token *parse_json2(const char *s, int s_len) {
|
|
|
|
struct frozen frozen;
|
2015-09-08 23:46:54 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
memset(&frozen, 0, sizeof(frozen));
|
|
|
|
frozen.end = s + s_len;
|
|
|
|
frozen.cur = s;
|
|
|
|
frozen.do_realloc = 1;
|
2015-09-08 23:46:54 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
if (doit(&frozen) < 0) {
|
|
|
|
FROZEN_FREE((void *) frozen.tokens);
|
|
|
|
frozen.tokens = NULL;
|
|
|
|
}
|
|
|
|
return frozen.tokens;
|
|
|
|
}
|
2015-09-08 23:46:54 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
static int path_part_len(const char *p) {
|
|
|
|
int i = 0;
|
|
|
|
while (p[i] != '\0' && p[i] != '[' && p[i] != '.') i++;
|
|
|
|
return i;
|
2015-09-08 23:46:54 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
struct json_token *find_json_token(struct json_token *toks, const char *path) {
|
|
|
|
while (path != 0 && path[0] != '\0') {
|
|
|
|
int i, ind2 = 0, ind = -1, skip = 2, n = path_part_len(path);
|
|
|
|
if (path[0] == '[') {
|
|
|
|
if (toks->type != JSON_TYPE_ARRAY || !is_digit(path[1])) return 0;
|
|
|
|
for (ind = 0, n = 1; path[n] != ']' && path[n] != '\0'; n++) {
|
|
|
|
if (!is_digit(path[n])) return 0;
|
|
|
|
ind *= 10;
|
|
|
|
ind += path[n] - '0';
|
|
|
|
}
|
|
|
|
if (path[n++] != ']') return 0;
|
|
|
|
skip = 1; /* In objects, we skip 2 elems while iterating, in arrays 1. */
|
|
|
|
} else if (toks->type != JSON_TYPE_OBJECT) return 0;
|
|
|
|
toks++;
|
|
|
|
for (i = 0; i < toks[-1].num_desc; i += skip, ind2++) {
|
|
|
|
/* ind == -1 indicated that we're iterating an array, not object */
|
|
|
|
if (ind == -1 && toks[i].type != JSON_TYPE_STRING) return 0;
|
|
|
|
if (ind2 == ind ||
|
|
|
|
(ind == -1 && toks[i].len == n && compare(path, toks[i].ptr, n))) {
|
|
|
|
i += skip - 1;
|
|
|
|
break;
|
|
|
|
};
|
|
|
|
if (toks[i - 1 + skip].type == JSON_TYPE_ARRAY ||
|
|
|
|
toks[i - 1 + skip].type == JSON_TYPE_OBJECT) {
|
|
|
|
i += toks[i - 1 + skip].num_desc;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (i == toks[-1].num_desc) return 0;
|
|
|
|
path += n;
|
|
|
|
if (path[0] == '.') path++;
|
|
|
|
if (path[0] == '\0') return &toks[i];
|
|
|
|
toks += i;
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
2015-09-08 23:46:54 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
int json_emit_long(char *buf, int buf_len, long int value) {
|
|
|
|
char tmp[20];
|
|
|
|
int n = snprintf(tmp, sizeof(tmp), "%ld", value);
|
|
|
|
strncpy(buf, tmp, buf_len > 0 ? buf_len : 0);
|
|
|
|
return n;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
int json_emit_double(char *buf, int buf_len, double value) {
|
|
|
|
char tmp[20];
|
|
|
|
int n = snprintf(tmp, sizeof(tmp), "%g", value);
|
|
|
|
strncpy(buf, tmp, buf_len > 0 ? buf_len : 0);
|
|
|
|
return n;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
int json_emit_quoted_str(char *s, int s_len, const char *str, int len) {
|
|
|
|
const char *begin = s, *end = s + s_len, *str_end = str + len;
|
|
|
|
char ch;
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
#define EMIT(x) do { if (s < end) *s = x; s++; } while (0)
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
EMIT('"');
|
|
|
|
while (str < str_end) {
|
|
|
|
ch = *str++;
|
|
|
|
switch (ch) {
|
|
|
|
case '"': EMIT('\\'); EMIT('"'); break;
|
|
|
|
case '\\': EMIT('\\'); EMIT('\\'); break;
|
|
|
|
case '\b': EMIT('\\'); EMIT('b'); break;
|
|
|
|
case '\f': EMIT('\\'); EMIT('f'); break;
|
|
|
|
case '\n': EMIT('\\'); EMIT('n'); break;
|
|
|
|
case '\r': EMIT('\\'); EMIT('r'); break;
|
|
|
|
case '\t': EMIT('\\'); EMIT('t'); break;
|
|
|
|
default: EMIT(ch);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
EMIT('"');
|
|
|
|
if (s < end) {
|
|
|
|
*s = '\0';
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
return s - begin;
|
|
|
|
}
|
|
|
|
|
|
|
|
int json_emit_unquoted_str(char *buf, int buf_len, const char *str, int len) {
|
|
|
|
if (buf_len > 0 && len > 0) {
|
|
|
|
int n = len < buf_len ? len : buf_len;
|
|
|
|
memcpy(buf, str, n);
|
|
|
|
if (n < buf_len) {
|
|
|
|
buf[n] = '\0';
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-11-01 00:26:19 +08:00
|
|
|
return len;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
int json_emit_va(char *s, int s_len, const char *fmt, va_list ap) {
|
|
|
|
const char *end = s + s_len, *str, *orig = s;
|
|
|
|
size_t len;
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
while (*fmt != '\0') {
|
|
|
|
switch (*fmt) {
|
|
|
|
case '[': case ']': case '{': case '}': case ',': case ':':
|
|
|
|
case ' ': case '\r': case '\n': case '\t':
|
|
|
|
if (s < end) {
|
|
|
|
*s = *fmt;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-11-01 00:26:19 +08:00
|
|
|
s++;
|
|
|
|
break;
|
|
|
|
case 'i':
|
|
|
|
s += json_emit_long(s, end - s, va_arg(ap, long));
|
|
|
|
break;
|
|
|
|
case 'f':
|
|
|
|
s += json_emit_double(s, end - s, va_arg(ap, double));
|
|
|
|
break;
|
|
|
|
case 'v':
|
|
|
|
str = va_arg(ap, char *);
|
|
|
|
len = va_arg(ap, size_t);
|
|
|
|
s += json_emit_quoted_str(s, end - s, str, len);
|
|
|
|
break;
|
|
|
|
case 'V':
|
|
|
|
str = va_arg(ap, char *);
|
|
|
|
len = va_arg(ap, size_t);
|
|
|
|
s += json_emit_unquoted_str(s, end - s, str, len);
|
|
|
|
break;
|
|
|
|
case 's':
|
|
|
|
str = va_arg(ap, char *);
|
|
|
|
s += json_emit_quoted_str(s, end - s, str, strlen(str));
|
|
|
|
break;
|
|
|
|
case 'S':
|
|
|
|
str = va_arg(ap, char *);
|
|
|
|
s += json_emit_unquoted_str(s, end - s, str, strlen(str));
|
|
|
|
break;
|
|
|
|
case 'T':
|
|
|
|
s += json_emit_unquoted_str(s, end - s, "true", 4);
|
|
|
|
break;
|
|
|
|
case 'F':
|
|
|
|
s += json_emit_unquoted_str(s, end - s, "false", 5);
|
|
|
|
break;
|
|
|
|
case 'N':
|
|
|
|
s += json_emit_unquoted_str(s, end - s, "null", 4);
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
return 0;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-11-01 00:26:19 +08:00
|
|
|
fmt++;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
/* Best-effort to 0-terminate generated string */
|
|
|
|
if (s < end) {
|
|
|
|
*s = '\0';
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
return s - orig;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
int json_emit(char *buf, int buf_len, const char *fmt, ...) {
|
|
|
|
int len;
|
2015-09-08 19:49:03 +08:00
|
|
|
va_list ap;
|
2015-11-01 00:26:19 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
va_start(ap, fmt);
|
2015-11-01 00:26:19 +08:00
|
|
|
len = json_emit_va(buf, buf_len, fmt, ap);
|
2015-09-08 19:49:03 +08:00
|
|
|
va_end(ap);
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
return len;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
#ifdef NS_MODULE_LINES
|
2015-12-15 02:45:18 +08:00
|
|
|
#line 1 "./src/../../common/md5.c"
|
2015-09-08 19:49:03 +08:00
|
|
|
/**/
|
|
|
|
#endif
|
|
|
|
/*
|
2015-11-01 00:26:19 +08:00
|
|
|
* This code implements the MD5 message-digest algorithm.
|
|
|
|
* The algorithm is due to Ron Rivest. This code was
|
|
|
|
* written by Colin Plumb in 1993, no copyright is claimed.
|
|
|
|
* This code is in the public domain; do with it what you wish.
|
|
|
|
*
|
|
|
|
* Equivalent code is available from RSA Data Security, Inc.
|
|
|
|
* This code has been tested against that, and is equivalent,
|
|
|
|
* except that you don't need to include two pages of legalese
|
|
|
|
* with every copy.
|
|
|
|
*
|
|
|
|
* To compute the message digest of a chunk of bytes, declare an
|
|
|
|
* MD5Context structure, pass it to MD5Init, call MD5Update as
|
|
|
|
* needed on buffers full of bytes, and then call MD5Final, which
|
|
|
|
* will fill a supplied 16-byte array with the digest.
|
2015-09-08 19:49:03 +08:00
|
|
|
*/
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
#if !defined(DISABLE_MD5) && !defined(EXCLUDE_COMMON)
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-12-12 22:05:54 +08:00
|
|
|
/* Amalgamated: #include "common/md5.h" */
|
2015-11-01 00:26:19 +08:00
|
|
|
|
|
|
|
#ifndef CS_ENABLE_NATIVE_MD5
|
|
|
|
static void byteReverse(unsigned char *buf, unsigned longs) {
|
|
|
|
/* 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
|
|
|
*/
|
2015-11-01 00:26:19 +08:00
|
|
|
void MD5_Init(MD5_CTX *ctx) {
|
|
|
|
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
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
static void MD5Transform(uint32_t buf[4], uint32_t const in[16]) {
|
|
|
|
register 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
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
void MD5_Update(MD5_CTX *ctx, const unsigned char *buf, size_t len) {
|
|
|
|
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);
|
|
|
|
byteReverse(ctx->in, 16);
|
|
|
|
MD5Transform(ctx->buf, (uint32_t *) ctx->in);
|
|
|
|
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);
|
|
|
|
byteReverse(ctx->in, 16);
|
|
|
|
MD5Transform(ctx->buf, (uint32_t *) ctx->in);
|
|
|
|
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
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
void MD5_Final(unsigned char digest[16], MD5_CTX *ctx) {
|
|
|
|
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);
|
|
|
|
byteReverse(ctx->in, 16);
|
|
|
|
MD5Transform(ctx->buf, (uint32_t *) ctx->in);
|
|
|
|
memset(ctx->in, 0, 56);
|
|
|
|
} else {
|
|
|
|
memset(p, 0, count - 8);
|
|
|
|
}
|
|
|
|
byteReverse(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
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
MD5Transform(ctx->buf, (uint32_t *) ctx->in);
|
|
|
|
byteReverse((unsigned char *) ctx->buf, 4);
|
|
|
|
memcpy(digest, ctx->buf, 16);
|
|
|
|
memset((char *) ctx, 0, sizeof(*ctx));
|
|
|
|
}
|
2015-11-12 23:18:51 +08:00
|
|
|
#endif /* CS_ENABLE_NATIVE_MD5 */
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
/*
|
|
|
|
* Stringify binary data. Output buffer size must be 2 * size_of_input + 1
|
|
|
|
* because each byte of input takes 2 bytes in string representation
|
|
|
|
* plus 1 byte for the terminating \0 character.
|
|
|
|
*/
|
|
|
|
void cs_to_hex(char *to, const unsigned char *p, size_t len) {
|
|
|
|
static const char *hex = "0123456789abcdef";
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
for (; len--; p++) {
|
|
|
|
*to++ = hex[p[0] >> 4];
|
|
|
|
*to++ = hex[p[0] & 0x0f];
|
|
|
|
}
|
|
|
|
*to = '\0';
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
char *cs_md5(char buf[33], ...) {
|
|
|
|
unsigned char hash[16];
|
|
|
|
const unsigned char *p;
|
|
|
|
va_list ap;
|
|
|
|
MD5_CTX ctx;
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
MD5_Init(&ctx);
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
va_start(ap, buf);
|
|
|
|
while ((p = va_arg(ap, const unsigned char *) ) != NULL) {
|
|
|
|
size_t len = va_arg(ap, size_t);
|
|
|
|
MD5_Update(&ctx, p, len);
|
|
|
|
}
|
|
|
|
va_end(ap);
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
MD5_Final(hash, &ctx);
|
|
|
|
cs_to_hex(buf, hash, sizeof(hash));
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
return buf;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
#endif /* EXCLUDE_COMMON */
|
|
|
|
#ifdef NS_MODULE_LINES
|
2015-12-15 02:45:18 +08:00
|
|
|
#line 1 "./src/../../common/mbuf.c"
|
2015-11-01 00:26:19 +08:00
|
|
|
/**/
|
|
|
|
#endif
|
|
|
|
/*
|
|
|
|
* Copyright (c) 2014 Cesanta Software Limited
|
|
|
|
* All rights reserved
|
|
|
|
*/
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
#ifndef EXCLUDE_COMMON
|
|
|
|
|
|
|
|
#include <assert.h>
|
|
|
|
#include <string.h>
|
2015-12-12 22:05:54 +08:00
|
|
|
/* Amalgamated: #include "common/mbuf.h" */
|
2015-11-01 00:26:19 +08:00
|
|
|
|
|
|
|
#ifndef MBUF_REALLOC
|
|
|
|
#define MBUF_REALLOC realloc
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#ifndef MBUF_FREE
|
|
|
|
#define MBUF_FREE free
|
|
|
|
#endif
|
|
|
|
|
|
|
|
void mbuf_init(struct mbuf *mbuf, size_t initial_size) {
|
|
|
|
mbuf->len = mbuf->size = 0;
|
|
|
|
mbuf->buf = NULL;
|
|
|
|
mbuf_resize(mbuf, initial_size);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
void mbuf_free(struct mbuf *mbuf) {
|
|
|
|
if (mbuf->buf != NULL) {
|
|
|
|
MBUF_FREE(mbuf->buf);
|
|
|
|
mbuf_init(mbuf, 0);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
void mbuf_resize(struct mbuf *a, size_t new_size) {
|
|
|
|
if (new_size > a->size || (new_size < a->size && new_size >= a->len)) {
|
|
|
|
char *buf = (char *) MBUF_REALLOC(a->buf, new_size);
|
|
|
|
/*
|
|
|
|
* In case realloc fails, there's not much we can do, except keep things as
|
|
|
|
* they are. Note that NULL is a valid return value from realloc when
|
|
|
|
* size == 0, but that is covered too.
|
|
|
|
*/
|
|
|
|
if (buf == NULL && new_size != 0) return;
|
|
|
|
a->buf = buf;
|
|
|
|
a->size = new_size;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
void mbuf_trim(struct mbuf *mbuf) {
|
|
|
|
mbuf_resize(mbuf, mbuf->len);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
size_t mbuf_insert(struct mbuf *a, size_t off, const void *buf, size_t len) {
|
|
|
|
char *p = NULL;
|
|
|
|
|
|
|
|
assert(a != NULL);
|
|
|
|
assert(a->len <= a->size);
|
|
|
|
assert(off <= a->len);
|
|
|
|
|
|
|
|
/* check overflow */
|
|
|
|
if (~(size_t) 0 - (size_t) a->buf < len) return 0;
|
|
|
|
|
|
|
|
if (a->len + len <= a->size) {
|
|
|
|
memmove(a->buf + off + len, a->buf + off, a->len - off);
|
|
|
|
if (buf != NULL) {
|
|
|
|
memcpy(a->buf + off, buf, len);
|
|
|
|
}
|
|
|
|
a->len += len;
|
|
|
|
} else if ((p = (char *) MBUF_REALLOC(
|
|
|
|
a->buf, (a->len + len) * MBUF_SIZE_MULTIPLIER)) != NULL) {
|
|
|
|
a->buf = p;
|
|
|
|
memmove(a->buf + off + len, a->buf + off, a->len - off);
|
|
|
|
if (buf != NULL) {
|
|
|
|
memcpy(a->buf + off, buf, len);
|
|
|
|
}
|
|
|
|
a->len += len;
|
|
|
|
a->size = a->len * MBUF_SIZE_MULTIPLIER;
|
|
|
|
} else {
|
|
|
|
len = 0;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-11-01 00:26:19 +08:00
|
|
|
|
|
|
|
return len;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
size_t mbuf_append(struct mbuf *a, const void *buf, size_t len) {
|
|
|
|
return mbuf_insert(a, a->len, buf, len);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
void mbuf_remove(struct mbuf *mb, size_t n) {
|
|
|
|
if (n > 0 && n <= mb->len) {
|
|
|
|
memmove(mb->buf, mb->buf + n, mb->len - n);
|
|
|
|
mb->len -= n;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
#endif /* EXCLUDE_COMMON */
|
|
|
|
#ifdef NS_MODULE_LINES
|
2015-12-15 02:45:18 +08:00
|
|
|
#line 1 "./src/../../common/sha1.c"
|
2015-11-01 00:26:19 +08:00
|
|
|
/**/
|
|
|
|
#endif
|
|
|
|
/* Copyright(c) By Steve Reid <steve@edmweb.com> */
|
|
|
|
/* 100% Public Domain */
|
|
|
|
|
|
|
|
#if !defined(DISABLE_SHA1) && !defined(EXCLUDE_COMMON)
|
|
|
|
|
2015-12-12 22:05:54 +08:00
|
|
|
/* Amalgamated: #include "common/sha1.h" */
|
2015-11-01 00:26:19 +08:00
|
|
|
|
|
|
|
#define SHA1HANDSOFF
|
|
|
|
#if defined(__sun)
|
2015-12-12 22:05:54 +08:00
|
|
|
/* Amalgamated: #include "common/solarisfixes.h" */
|
2015-11-01 00:26:19 +08:00
|
|
|
#endif
|
|
|
|
|
|
|
|
union char64long16 {
|
|
|
|
unsigned char c[64];
|
|
|
|
uint32_t l[16];
|
|
|
|
};
|
|
|
|
|
|
|
|
#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);
|
|
|
|
|
|
|
|
void cs_sha1_transform(uint32_t state[5], const unsigned char buffer[64]) {
|
|
|
|
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
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
void cs_sha1_init(cs_sha1_ctx *context) {
|
|
|
|
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
|
|
|
}
|
|
|
|
|
2015-11-12 23:18:51 +08:00
|
|
|
void cs_sha1_update(cs_sha1_ctx *context, const unsigned char *data,
|
|
|
|
uint32_t len) {
|
2015-11-01 00:26:19 +08:00
|
|
|
uint32_t i, j;
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
j = context->count[0];
|
|
|
|
if ((context->count[0] += len << 3) < j) context->count[1]++;
|
|
|
|
context->count[1] += (len >> 29);
|
|
|
|
j = (j >> 3) & 63;
|
|
|
|
if ((j + len) > 63) {
|
|
|
|
memcpy(&context->buffer[j], data, (i = 64 - j));
|
|
|
|
cs_sha1_transform(context->state, context->buffer);
|
|
|
|
for (; i + 63 < len; i += 64) {
|
|
|
|
cs_sha1_transform(context->state, &data[i]);
|
|
|
|
}
|
|
|
|
j = 0;
|
|
|
|
} else
|
|
|
|
i = 0;
|
|
|
|
memcpy(&context->buffer[j], &data[i], len - i);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
void cs_sha1_final(unsigned char digest[20], cs_sha1_ctx *context) {
|
|
|
|
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;
|
|
|
|
cs_sha1_update(context, &c, 1);
|
|
|
|
while ((context->count[0] & 504) != 448) {
|
|
|
|
c = 0000;
|
|
|
|
cs_sha1_update(context, &c, 1);
|
|
|
|
}
|
|
|
|
cs_sha1_update(context, finalcount, 8);
|
|
|
|
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
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
void cs_hmac_sha1(const unsigned char *key, size_t keylen,
|
|
|
|
const unsigned char *data, size_t datalen,
|
|
|
|
unsigned char out[20]) {
|
|
|
|
cs_sha1_ctx ctx;
|
|
|
|
unsigned char buf1[64], buf2[64], tmp_key[20], i;
|
|
|
|
|
|
|
|
if (keylen > sizeof(buf1)) {
|
|
|
|
cs_sha1_init(&ctx);
|
|
|
|
cs_sha1_update(&ctx, key, keylen);
|
|
|
|
cs_sha1_final(tmp_key, &ctx);
|
|
|
|
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
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
cs_sha1_init(&ctx);
|
|
|
|
cs_sha1_update(&ctx, buf1, sizeof(buf1));
|
|
|
|
cs_sha1_update(&ctx, data, datalen);
|
|
|
|
cs_sha1_final(out, &ctx);
|
|
|
|
|
|
|
|
cs_sha1_init(&ctx);
|
|
|
|
cs_sha1_update(&ctx, buf2, sizeof(buf2));
|
|
|
|
cs_sha1_update(&ctx, out, 20);
|
|
|
|
cs_sha1_final(out, &ctx);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
#endif /* EXCLUDE_COMMON */
|
|
|
|
#ifdef NS_MODULE_LINES
|
2015-12-15 02:45:18 +08:00
|
|
|
#line 1 "./src/../../common/str_util.c"
|
2015-11-01 00:26:19 +08:00
|
|
|
/**/
|
|
|
|
#endif
|
|
|
|
/*
|
|
|
|
* Copyright (c) 2015 Cesanta Software Limited
|
|
|
|
* All rights reserved
|
|
|
|
*/
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
#ifndef EXCLUDE_COMMON
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-12-12 22:05:54 +08:00
|
|
|
/* Amalgamated: #include "common/osdep.h" */
|
|
|
|
/* Amalgamated: #include "common/str_util.h" */
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-24 06:28:09 +08:00
|
|
|
#ifdef _MG_PROVIDE_STRNLEN
|
2015-11-01 00:26:19 +08:00
|
|
|
size_t strnlen(const char *s, size_t maxlen) {
|
|
|
|
size_t l = 0;
|
|
|
|
for (; l < maxlen && s[l] != '\0'; l++) {
|
|
|
|
}
|
|
|
|
return l;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-11-01 00:26:19 +08:00
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
#define C_SNPRINTF_APPEND_CHAR(ch) \
|
|
|
|
do { \
|
|
|
|
if (i < (int) buf_size) buf[i] = ch; \
|
|
|
|
i++; \
|
|
|
|
} while (0)
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
#define C_SNPRINTF_FLAG_ZERO 1
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
#ifdef C_DISABLE_BUILTIN_SNPRINTF
|
|
|
|
int c_vsnprintf(char *buf, size_t buf_size, const char *fmt, va_list ap) {
|
|
|
|
return vsnprintf(buf, buf_size, fmt, ap);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-11-01 00:26:19 +08:00
|
|
|
#else
|
|
|
|
static int c_itoa(char *buf, size_t buf_size, int64_t num, int base, int flags,
|
|
|
|
int field_width) {
|
|
|
|
char tmp[40];
|
|
|
|
int i = 0, k = 0, neg = 0;
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
if (num < 0) {
|
|
|
|
neg++;
|
|
|
|
num = -num;
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
/* Print into temporary buffer - in reverse order */
|
|
|
|
do {
|
|
|
|
int rem = num % base;
|
|
|
|
if (rem < 10) {
|
|
|
|
tmp[k++] = '0' + rem;
|
|
|
|
} else {
|
|
|
|
tmp[k++] = 'a' + (rem - 10);
|
|
|
|
}
|
|
|
|
num /= base;
|
|
|
|
} while (num > 0);
|
|
|
|
|
|
|
|
/* Zero padding */
|
|
|
|
if (flags && C_SNPRINTF_FLAG_ZERO) {
|
|
|
|
while (k < field_width && k < (int) sizeof(tmp) - 1) {
|
|
|
|
tmp[k++] = '0';
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
/* And sign */
|
|
|
|
if (neg) {
|
|
|
|
tmp[k++] = '-';
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
/* Now output */
|
|
|
|
while (--k >= 0) {
|
|
|
|
C_SNPRINTF_APPEND_CHAR(tmp[k]);
|
|
|
|
}
|
|
|
|
|
|
|
|
return i;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
int c_vsnprintf(char *buf, size_t buf_size, const char *fmt, va_list ap) {
|
|
|
|
int ch, i = 0, len_mod, flags, precision, field_width;
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
while ((ch = *fmt++) != '\0') {
|
|
|
|
if (ch != '%') {
|
|
|
|
C_SNPRINTF_APPEND_CHAR(ch);
|
|
|
|
} else {
|
|
|
|
/*
|
|
|
|
* Conversion specification:
|
|
|
|
* zero or more flags (one of: # 0 - <space> + ')
|
|
|
|
* an optional minimum field width (digits)
|
|
|
|
* an optional precision (. followed by digits, or *)
|
|
|
|
* an optional length modifier (one of: hh h l ll L q j z t)
|
|
|
|
* conversion specifier (one of: d i o u x X e E f F g G a A c s p n)
|
|
|
|
*/
|
|
|
|
flags = field_width = precision = len_mod = 0;
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
/* Flags. only zero-pad flag is supported. */
|
|
|
|
if (*fmt == '0') {
|
|
|
|
flags |= C_SNPRINTF_FLAG_ZERO;
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
/* Field width */
|
|
|
|
while (*fmt >= '0' && *fmt <= '9') {
|
|
|
|
field_width *= 10;
|
|
|
|
field_width += *fmt++ - '0';
|
|
|
|
}
|
|
|
|
/* Dynamic field width */
|
|
|
|
if (*fmt == '*') {
|
|
|
|
field_width = va_arg(ap, int);
|
|
|
|
fmt++;
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
/* Precision */
|
|
|
|
if (*fmt == '.') {
|
|
|
|
fmt++;
|
|
|
|
if (*fmt == '*') {
|
|
|
|
precision = va_arg(ap, int);
|
|
|
|
fmt++;
|
|
|
|
} else {
|
|
|
|
while (*fmt >= '0' && *fmt <= '9') {
|
|
|
|
precision *= 10;
|
|
|
|
precision += *fmt++ - '0';
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
/* Length modifier */
|
|
|
|
switch (*fmt) {
|
|
|
|
case 'h':
|
|
|
|
case 'l':
|
|
|
|
case 'L':
|
|
|
|
case 'I':
|
|
|
|
case 'q':
|
|
|
|
case 'j':
|
|
|
|
case 'z':
|
|
|
|
case 't':
|
|
|
|
len_mod = *fmt++;
|
|
|
|
if (*fmt == 'h') {
|
|
|
|
len_mod = 'H';
|
|
|
|
fmt++;
|
|
|
|
}
|
|
|
|
if (*fmt == 'l') {
|
|
|
|
len_mod = 'q';
|
|
|
|
fmt++;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
ch = *fmt++;
|
|
|
|
if (ch == 's') {
|
|
|
|
const char *s = va_arg(ap, const char *); /* Always fetch parameter */
|
|
|
|
int j;
|
|
|
|
int pad = field_width - (precision >= 0 ? strnlen(s, precision) : 0);
|
|
|
|
for (j = 0; j < pad; j++) {
|
|
|
|
C_SNPRINTF_APPEND_CHAR(' ');
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-11-01 00:26:19 +08:00
|
|
|
|
|
|
|
/* Ignore negative and 0 precisions */
|
|
|
|
for (j = 0; (precision <= 0 || j < precision) && s[j] != '\0'; j++) {
|
|
|
|
C_SNPRINTF_APPEND_CHAR(s[j]);
|
|
|
|
}
|
|
|
|
} else if (ch == 'c') {
|
|
|
|
ch = va_arg(ap, int); /* Always fetch parameter */
|
|
|
|
C_SNPRINTF_APPEND_CHAR(ch);
|
|
|
|
} else if (ch == 'd' && len_mod == 0) {
|
|
|
|
i += c_itoa(buf + i, buf_size - i, va_arg(ap, int), 10, flags,
|
|
|
|
field_width);
|
|
|
|
} else if (ch == 'd' && len_mod == 'l') {
|
|
|
|
i += c_itoa(buf + i, buf_size - i, va_arg(ap, long), 10, flags,
|
|
|
|
field_width);
|
2015-12-09 21:35:15 +08:00
|
|
|
} else if (ch == 'd' && len_mod == 'q') {
|
|
|
|
i += c_itoa(buf + i, buf_size - i, va_arg(ap, int64_t), 10, flags,
|
|
|
|
field_width);
|
2015-11-01 00:26:19 +08:00
|
|
|
} else if ((ch == 'x' || ch == 'u') && len_mod == 0) {
|
|
|
|
i += c_itoa(buf + i, buf_size - i, va_arg(ap, unsigned),
|
|
|
|
ch == 'x' ? 16 : 10, flags, field_width);
|
|
|
|
} else if ((ch == 'x' || ch == 'u') && len_mod == 'l') {
|
|
|
|
i += c_itoa(buf + i, buf_size - i, va_arg(ap, unsigned long),
|
|
|
|
ch == 'x' ? 16 : 10, flags, field_width);
|
|
|
|
} else if (ch == 'p') {
|
|
|
|
unsigned long num = (unsigned long) va_arg(ap, void *);
|
|
|
|
C_SNPRINTF_APPEND_CHAR('0');
|
|
|
|
C_SNPRINTF_APPEND_CHAR('x');
|
|
|
|
i += c_itoa(buf + i, buf_size - i, num, 16, flags, 0);
|
|
|
|
} else {
|
|
|
|
#ifndef NO_LIBC
|
|
|
|
/*
|
|
|
|
* TODO(lsm): abort is not nice in a library, remove it
|
|
|
|
* Also, ESP8266 SDK doesn't have it
|
|
|
|
*/
|
|
|
|
abort();
|
|
|
|
#endif
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
/* Zero-terminate the result */
|
|
|
|
if (buf_size > 0) {
|
|
|
|
buf[i < (int) buf_size ? i : (int) buf_size - 1] = '\0';
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
return i;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-11-01 00:26:19 +08:00
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
int c_snprintf(char *buf, size_t buf_size, const char *fmt, ...) {
|
|
|
|
int result;
|
2015-09-08 19:49:03 +08:00
|
|
|
va_list ap;
|
|
|
|
va_start(ap, fmt);
|
2015-11-01 00:26:19 +08:00
|
|
|
result = c_vsnprintf(buf, buf_size, fmt, ap);
|
2015-09-08 19:49:03 +08:00
|
|
|
va_end(ap);
|
2015-11-01 00:26:19 +08:00
|
|
|
return result;
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-01 00:26:19 +08:00
|
|
|
#ifdef _WIN32
|
|
|
|
void to_wchar(const char *path, wchar_t *wbuf, size_t wbuf_len) {
|
|
|
|
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';
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Convert to Unicode and back. If doubly-converted string does not
|
|
|
|
* match the original, something is fishy, reject.
|
|
|
|
*/
|
|
|
|
memset(wbuf, 0, wbuf_len * sizeof(wchar_t));
|
|
|
|
MultiByteToWideChar(CP_UTF8, 0, buf, -1, wbuf, (int) wbuf_len);
|
|
|
|
WideCharToMultiByte(CP_UTF8, 0, wbuf, (int) wbuf_len, buf2, sizeof(buf2),
|
|
|
|
NULL, NULL);
|
|
|
|
if (strcmp(buf, buf2) != 0) {
|
|
|
|
wbuf[0] = L'\0';
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-11-01 00:26:19 +08:00
|
|
|
#endif /* _WIN32 */
|
|
|
|
|
|
|
|
#endif /* EXCLUDE_COMMON */
|
2015-09-08 19:49:03 +08:00
|
|
|
#ifdef NS_MODULE_LINES
|
2015-12-15 02:45:18 +08:00
|
|
|
#line 1 "./src/net.c"
|
2015-09-08 19:49:03 +08:00
|
|
|
/**/
|
|
|
|
#endif
|
|
|
|
/*
|
|
|
|
* Copyright (c) 2014 Cesanta Software Limited
|
|
|
|
* 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>.
|
|
|
|
*/
|
|
|
|
|
2015-12-12 22:05:54 +08:00
|
|
|
/* Amalgamated: #include "mongoose/src/internal.h" */
|
2015-12-12 22:16:48 +08:00
|
|
|
/* Amalgamated: #include "mongoose/src/util.h" */
|
|
|
|
/* Amalgamated: #include "mongoose/src/dns.h" */
|
|
|
|
/* Amalgamated: #include "mongoose/src/resolv.h" */
|
2015-12-19 17:04:45 +08:00
|
|
|
/* Amalgamated: #include "common/cs_time.h" */
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#if MG_MGR_EV_MGR == 1 /* epoll() */
|
2015-09-08 19:49:03 +08:00
|
|
|
#include <sys/epoll.h>
|
|
|
|
#endif
|
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#define MG_MAX_HOST_LEN 200
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#define MG_COPY_COMMON_CONNECTION_OPTIONS(dst, src) \
|
2015-09-08 19:49:03 +08:00
|
|
|
memcpy(dst, src, sizeof(*dst));
|
|
|
|
|
|
|
|
/* Which flags can be pre-set by the user at connection creation time. */
|
2015-09-19 19:38:46 +08:00
|
|
|
#define _MG_ALLOWED_CONNECT_FLAGS_MASK \
|
|
|
|
(MG_F_USER_1 | MG_F_USER_2 | MG_F_USER_3 | MG_F_USER_4 | MG_F_USER_5 | \
|
|
|
|
MG_F_USER_6 | MG_F_WEBSOCKET_NO_DEFRAG)
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Which flags should be modifiable by user's callbacks. */
|
2015-09-19 19:38:46 +08:00
|
|
|
#define _MG_CALLBACK_MODIFIABLE_FLAGS_MASK \
|
|
|
|
(MG_F_USER_1 | MG_F_USER_2 | MG_F_USER_3 | MG_F_USER_4 | MG_F_USER_5 | \
|
|
|
|
MG_F_USER_6 | MG_F_WEBSOCKET_NO_DEFRAG | MG_F_SEND_AND_CLOSE | \
|
2015-11-18 15:07:48 +08:00
|
|
|
MG_F_CLOSE_IMMEDIATELY | MG_F_IS_WEBSOCKET | MG_F_DELETE_CHUNK)
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
#ifndef intptr_t
|
|
|
|
#define intptr_t long
|
|
|
|
#endif
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
int mg_is_error(int n);
|
|
|
|
void mg_set_non_blocking_mode(sock_t sock);
|
|
|
|
|
2015-10-24 07:05:05 +08:00
|
|
|
extern void mg_ev_mgr_init(struct mg_mgr *mgr);
|
|
|
|
extern void mg_ev_mgr_free(struct mg_mgr *mgr);
|
|
|
|
extern void mg_ev_mgr_add_conn(struct mg_connection *nc);
|
|
|
|
extern void mg_ev_mgr_remove_conn(struct mg_connection *nc);
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
MG_INTERNAL void mg_add_conn(struct mg_mgr *mgr, struct mg_connection *c) {
|
2015-10-22 20:04:35 +08:00
|
|
|
DBG(("%p %p", mgr, c));
|
2015-09-08 19:49:03 +08:00
|
|
|
c->mgr = mgr;
|
|
|
|
c->next = mgr->active_connections;
|
|
|
|
mgr->active_connections = c;
|
|
|
|
c->prev = NULL;
|
|
|
|
if (c->next != NULL) c->next->prev = c;
|
|
|
|
mg_ev_mgr_add_conn(c);
|
|
|
|
}
|
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
MG_INTERNAL void mg_remove_conn(struct mg_connection *conn) {
|
2015-09-08 19:49:03 +08:00
|
|
|
if (conn->prev == NULL) conn->mgr->active_connections = conn->next;
|
|
|
|
if (conn->prev) conn->prev->next = conn->next;
|
|
|
|
if (conn->next) conn->next->prev = conn->prev;
|
|
|
|
mg_ev_mgr_remove_conn(conn);
|
|
|
|
}
|
|
|
|
|
2015-10-29 06:12:55 +08:00
|
|
|
MG_INTERNAL void mg_call(struct mg_connection *nc,
|
|
|
|
mg_event_handler_t ev_handler, int ev, void *ev_data) {
|
2015-09-08 19:49:03 +08:00
|
|
|
unsigned long flags_before;
|
2015-11-18 15:07:48 +08:00
|
|
|
if (ev_handler == NULL) {
|
|
|
|
/*
|
|
|
|
* If protocol handler is specified, call it. Otherwise, call user-specified
|
|
|
|
* event handler.
|
|
|
|
*/
|
|
|
|
ev_handler = nc->proto_handler ? nc->proto_handler : nc->handler;
|
|
|
|
}
|
|
|
|
DBG(("%p %s ev=%d ev_data=%p flags=%lu rmbl=%d smbl=%d", nc,
|
|
|
|
ev_handler == nc->handler ? "user" : "proto", ev, ev_data, nc->flags,
|
|
|
|
(int) nc->recv_mbuf.len, (int) nc->send_mbuf.len));
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-27 18:32:36 +08:00
|
|
|
#if !defined(NO_LIBC) && !defined(MG_DISABLE_HEXDUMP)
|
2015-09-08 19:49:03 +08:00
|
|
|
/* LCOV_EXCL_START */
|
2015-09-19 19:38:46 +08:00
|
|
|
if (nc->mgr->hexdump_file != NULL && ev != MG_EV_POLL &&
|
|
|
|
ev != MG_EV_SEND /* handled separately */) {
|
2015-11-27 18:32:36 +08:00
|
|
|
if (ev == MG_EV_RECV) {
|
|
|
|
mg_hexdump_connection(nc, nc->mgr->hexdump_file, nc->recv_mbuf.buf,
|
|
|
|
*(int *) ev_data, ev);
|
|
|
|
} else {
|
|
|
|
mg_hexdump_connection(nc, nc->mgr->hexdump_file, NULL, 0, ev);
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
/* LCOV_EXCL_STOP */
|
|
|
|
#endif
|
|
|
|
if (ev_handler != NULL) {
|
|
|
|
flags_before = nc->flags;
|
|
|
|
ev_handler(nc, ev, ev_data);
|
2015-11-18 15:07:48 +08:00
|
|
|
/* Prevent user handler from fiddling with system flags. */
|
|
|
|
if (ev_handler == nc->handler && nc->flags != flags_before) {
|
2015-09-19 19:38:46 +08:00
|
|
|
nc->flags = (flags_before & ~_MG_CALLBACK_MODIFIABLE_FLAGS_MASK) |
|
|
|
|
(nc->flags & _MG_CALLBACK_MODIFIABLE_FLAGS_MASK);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
}
|
2015-11-18 15:07:48 +08:00
|
|
|
DBG(("%p after %s flags=%lu rmbl=%d smbl=%d", nc,
|
|
|
|
ev_handler == nc->handler ? "user" : "proto", nc->flags,
|
2015-10-22 20:04:35 +08:00
|
|
|
(int) nc->recv_mbuf.len, (int) nc->send_mbuf.len));
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-12-16 08:52:37 +08:00
|
|
|
void mg_if_timer(struct mg_connection *c, double now) {
|
2015-12-08 01:55:28 +08:00
|
|
|
if (c->ev_timer_time > 0 && now >= c->ev_timer_time) {
|
2015-12-16 08:52:37 +08:00
|
|
|
double old_value = c->ev_timer_time;
|
|
|
|
mg_call(c, NULL, MG_EV_TIMER, &now);
|
2015-12-08 01:55:28 +08:00
|
|
|
/*
|
|
|
|
* To prevent timer firing all the time, reset the timer after delivery.
|
|
|
|
* However, in case user sets it to new value, do not reset.
|
|
|
|
*/
|
|
|
|
if (c->ev_timer_time == old_value) {
|
|
|
|
c->ev_timer_time = 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-10-26 16:15:58 +08:00
|
|
|
void mg_if_poll(struct mg_connection *nc, time_t now) {
|
2015-10-29 06:12:55 +08:00
|
|
|
mg_call(nc, NULL, MG_EV_POLL, &now);
|
2015-10-26 16:15:58 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void mg_destroy_conn(struct mg_connection *conn) {
|
2015-10-22 20:04:35 +08:00
|
|
|
mg_if_destroy_conn(conn);
|
2015-12-09 17:43:41 +08:00
|
|
|
#ifdef MG_ENABLE_SSL
|
|
|
|
if (conn->ssl != NULL) SSL_free(conn->ssl);
|
|
|
|
if (conn->ssl_ctx != NULL) SSL_CTX_free(conn->ssl_ctx);
|
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
mbuf_free(&conn->recv_mbuf);
|
|
|
|
mbuf_free(&conn->send_mbuf);
|
2015-12-09 17:43:41 +08:00
|
|
|
memset(conn, 0, sizeof(*conn));
|
2015-09-19 19:38:46 +08:00
|
|
|
MG_FREE(conn);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-10-24 07:05:05 +08:00
|
|
|
void mg_close_conn(struct mg_connection *conn) {
|
2015-09-08 19:49:03 +08:00
|
|
|
DBG(("%p %lu", conn, conn->flags));
|
2015-11-30 20:37:47 +08:00
|
|
|
mg_call(conn, NULL, MG_EV_CLOSE, NULL);
|
2015-09-08 19:49:03 +08:00
|
|
|
mg_remove_conn(conn);
|
|
|
|
mg_destroy_conn(conn);
|
|
|
|
}
|
|
|
|
|
2015-09-22 00:58:45 +08:00
|
|
|
void mg_mgr_init(struct mg_mgr *m, void *user_data) {
|
|
|
|
memset(m, 0, sizeof(*m));
|
2015-12-08 20:13:03 +08:00
|
|
|
#ifndef MG_DISABLE_SOCKETPAIR
|
2015-09-22 00:58:45 +08:00
|
|
|
m->ctl[0] = m->ctl[1] = INVALID_SOCKET;
|
2015-12-08 20:13:03 +08:00
|
|
|
#endif
|
2015-09-22 00:58:45 +08:00
|
|
|
m->user_data = user_data;
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
#ifdef _WIN32
|
|
|
|
{
|
|
|
|
WSADATA data;
|
|
|
|
WSAStartup(MAKEWORD(2, 2), &data);
|
|
|
|
}
|
2015-09-19 19:38:46 +08:00
|
|
|
#elif !defined(AVR_LIBC) && !defined(MG_ESP8266)
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Ignore SIGPIPE signal, so if client cancels the request, it
|
|
|
|
* won't kill the whole process. */
|
|
|
|
signal(SIGPIPE, SIG_IGN);
|
|
|
|
#endif
|
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifdef MG_ENABLE_SSL
|
2015-09-08 19:49:03 +08:00
|
|
|
{
|
|
|
|
static int init_done;
|
|
|
|
if (!init_done) {
|
|
|
|
SSL_library_init();
|
|
|
|
init_done++;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|
2015-09-22 00:58:45 +08:00
|
|
|
|
|
|
|
mg_ev_mgr_init(m);
|
2015-09-08 19:49:03 +08:00
|
|
|
DBG(("=================================="));
|
2015-09-22 00:58:45 +08:00
|
|
|
DBG(("init mgr=%p", m));
|
|
|
|
}
|
|
|
|
|
|
|
|
#ifdef MG_ENABLE_JAVASCRIPT
|
2015-12-28 08:16:53 +08:00
|
|
|
static enum v7_err mg_send_js(struct v7 *v7, v7_val_t *res) {
|
2015-10-07 00:47:54 +08:00
|
|
|
v7_val_t arg0 = v7_arg(v7, 0);
|
|
|
|
v7_val_t arg1 = v7_arg(v7, 1);
|
2015-09-22 00:58:45 +08:00
|
|
|
struct mg_connection *c = (struct mg_connection *) v7_to_foreign(arg0);
|
|
|
|
size_t len = 0;
|
|
|
|
|
|
|
|
if (v7_is_string(arg1)) {
|
2015-12-08 08:22:22 +08:00
|
|
|
const char *data = v7_get_string_data(v7, &arg1, &len);
|
2015-09-22 00:58:45 +08:00
|
|
|
mg_send(c, data, len);
|
|
|
|
}
|
|
|
|
|
2016-01-12 18:38:26 +08:00
|
|
|
*res = v7_mk_number(len);
|
2015-12-28 08:16:53 +08:00
|
|
|
|
|
|
|
return V7_OK;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-09-22 00:58:45 +08:00
|
|
|
enum v7_err mg_enable_javascript(struct mg_mgr *m, struct v7 *v7,
|
|
|
|
const char *init_file_name) {
|
|
|
|
v7_val_t v;
|
|
|
|
m->v7 = v7;
|
2015-10-04 05:21:06 +08:00
|
|
|
v7_set_method(v7, v7_get_global(v7), "mg_send", mg_send_js);
|
2015-09-22 00:58:45 +08:00
|
|
|
return v7_exec_file(v7, init_file_name, &v);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
void mg_mgr_free(struct mg_mgr *m) {
|
2015-09-08 19:49:03 +08:00
|
|
|
struct mg_connection *conn, *tmp_conn;
|
|
|
|
|
2015-09-22 00:58:45 +08:00
|
|
|
DBG(("%p", m));
|
|
|
|
if (m == NULL) return;
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Do one last poll, see https://github.com/cesanta/mongoose/issues/286 */
|
2015-09-22 00:58:45 +08:00
|
|
|
mg_mgr_poll(m, 0);
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-24 07:05:05 +08:00
|
|
|
#ifndef MG_DISABLE_SOCKETPAIR
|
2015-09-22 00:58:45 +08:00
|
|
|
if (m->ctl[0] != INVALID_SOCKET) closesocket(m->ctl[0]);
|
|
|
|
if (m->ctl[1] != INVALID_SOCKET) closesocket(m->ctl[1]);
|
|
|
|
m->ctl[0] = m->ctl[1] = INVALID_SOCKET;
|
2015-12-08 20:13:03 +08:00
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-09-22 00:58:45 +08:00
|
|
|
for (conn = m->active_connections; conn != NULL; conn = tmp_conn) {
|
2015-09-08 19:49:03 +08:00
|
|
|
tmp_conn = conn->next;
|
|
|
|
mg_close_conn(conn);
|
|
|
|
}
|
|
|
|
|
2015-09-22 00:58:45 +08:00
|
|
|
mg_ev_mgr_free(m);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
int mg_vprintf(struct mg_connection *nc, const char *fmt, va_list ap) {
|
2015-09-19 19:38:46 +08:00
|
|
|
char mem[MG_VPRINTF_BUFFER_SIZE], *buf = mem;
|
2015-09-08 19:49:03 +08:00
|
|
|
int len;
|
|
|
|
|
|
|
|
if ((len = mg_avprintf(&buf, sizeof(mem), fmt, ap)) > 0) {
|
2015-10-22 20:04:35 +08:00
|
|
|
mg_send(nc, buf, len);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
if (buf != mem && buf != NULL) {
|
2015-09-19 19:38:46 +08:00
|
|
|
MG_FREE(buf); /* LCOV_EXCL_LINE */
|
2015-09-08 19:49:03 +08:00
|
|
|
} /* LCOV_EXCL_LINE */
|
|
|
|
|
|
|
|
return len;
|
|
|
|
}
|
|
|
|
|
|
|
|
int mg_printf(struct mg_connection *conn, const char *fmt, ...) {
|
|
|
|
int len;
|
|
|
|
va_list ap;
|
|
|
|
va_start(ap, fmt);
|
|
|
|
len = mg_vprintf(conn, fmt, ap);
|
|
|
|
va_end(ap);
|
|
|
|
return len;
|
|
|
|
}
|
|
|
|
|
2015-10-24 07:05:05 +08:00
|
|
|
#ifndef MG_DISABLE_SYNC_RESOLVER
|
2015-09-08 19:49:03 +08:00
|
|
|
/* TODO(lsm): use non-blocking resolver */
|
|
|
|
static int mg_resolve2(const char *host, struct in_addr *ina) {
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifdef MG_ENABLE_GETADDRINFO
|
2015-09-08 19:49:03 +08:00
|
|
|
int rv = 0;
|
|
|
|
struct addrinfo hints, *servinfo, *p;
|
|
|
|
struct sockaddr_in *h = NULL;
|
|
|
|
memset(&hints, 0, sizeof hints);
|
|
|
|
hints.ai_family = AF_INET;
|
|
|
|
hints.ai_socktype = SOCK_STREAM;
|
|
|
|
if ((rv = getaddrinfo(host, NULL, NULL, &servinfo)) != 0) {
|
|
|
|
DBG(("getaddrinfo(%s) failed: %s", host, strerror(errno)));
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
for (p = servinfo; p != NULL; p = p->ai_next) {
|
|
|
|
memcpy(&h, &p->ai_addr, sizeof(struct sockaddr_in *));
|
|
|
|
memcpy(ina, &h->sin_addr, sizeof(ina));
|
|
|
|
}
|
|
|
|
freeaddrinfo(servinfo);
|
|
|
|
return 1;
|
|
|
|
#else
|
|
|
|
struct hostent *he;
|
|
|
|
if ((he = gethostbyname(host)) == NULL) {
|
|
|
|
DBG(("gethostbyname(%s) failed: %s", host, strerror(errno)));
|
|
|
|
} else {
|
|
|
|
memcpy(ina, he->h_addr_list[0], sizeof(*ina));
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
return 0;
|
2015-09-19 19:38:46 +08:00
|
|
|
#endif /* MG_ENABLE_GETADDRINFO */
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
int mg_resolve(const char *host, char *buf, size_t n) {
|
|
|
|
struct in_addr ad;
|
|
|
|
return mg_resolve2(host, &ad) ? snprintf(buf, n, "%s", inet_ntoa(ad)) : 0;
|
|
|
|
}
|
2015-10-24 07:05:05 +08:00
|
|
|
#endif /* MG_DISABLE_SYNC_RESOLVER */
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
MG_INTERNAL struct mg_connection *mg_create_connection(
|
2015-09-08 19:49:03 +08:00
|
|
|
struct mg_mgr *mgr, mg_event_handler_t callback,
|
|
|
|
struct mg_add_sock_opts opts) {
|
|
|
|
struct mg_connection *conn;
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
if ((conn = (struct mg_connection *) MG_CALLOC(1, sizeof(*conn))) != NULL) {
|
2015-09-08 19:49:03 +08:00
|
|
|
conn->sock = INVALID_SOCKET;
|
|
|
|
conn->handler = callback;
|
|
|
|
conn->mgr = mgr;
|
2016-01-08 21:14:55 +08:00
|
|
|
conn->last_io_time = mg_time();
|
2015-09-19 19:38:46 +08:00
|
|
|
conn->flags = opts.flags & _MG_ALLOWED_CONNECT_FLAGS_MASK;
|
2015-09-08 19:49:03 +08:00
|
|
|
conn->user_data = opts.user_data;
|
|
|
|
/*
|
|
|
|
* SIZE_MAX is defined as a long long constant in
|
|
|
|
* system headers on some platforms and so it
|
|
|
|
* doesn't compile with pedantic ansi flags.
|
|
|
|
*/
|
|
|
|
conn->recv_mbuf_limit = ~0;
|
2015-12-09 17:43:41 +08:00
|
|
|
if (!mg_if_create_conn(conn)) {
|
|
|
|
MG_FREE(conn);
|
|
|
|
conn = NULL;
|
|
|
|
MG_SET_PTRPTR(opts.error_string, "failed init connection");
|
|
|
|
}
|
2015-10-22 20:04:35 +08:00
|
|
|
} else {
|
|
|
|
MG_SET_PTRPTR(opts.error_string, "failed create connection");
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
return conn;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Address format: [PROTO://][HOST]:PORT
|
|
|
|
*
|
|
|
|
* HOST could be IPv4/IPv6 address or a host name.
|
|
|
|
* `host` is a destination buffer to hold parsed HOST part. Shoud be at least
|
2015-09-19 19:38:46 +08:00
|
|
|
* MG_MAX_HOST_LEN bytes long.
|
2015-09-08 19:49:03 +08:00
|
|
|
* `proto` is a returned socket type, either SOCK_STREAM or SOCK_DGRAM
|
|
|
|
*
|
|
|
|
* Return:
|
|
|
|
* -1 on parse error
|
|
|
|
* 0 if HOST needs DNS lookup
|
|
|
|
* >0 length of the address string
|
|
|
|
*/
|
2015-09-19 19:38:46 +08:00
|
|
|
MG_INTERNAL int mg_parse_address(const char *str, union socket_address *sa,
|
2015-09-08 19:49:03 +08:00
|
|
|
int *proto, char *host, size_t host_len) {
|
|
|
|
unsigned int a, b, c, d, port = 0;
|
2015-11-19 06:47:17 +08:00
|
|
|
int ch, len = 0;
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifdef MG_ENABLE_IPV6
|
2015-09-08 19:49:03 +08:00
|
|
|
char buf[100];
|
|
|
|
#endif
|
|
|
|
|
|
|
|
/*
|
|
|
|
* MacOS needs that. If we do not zero it, subsequent bind() will fail.
|
|
|
|
* Also, all-zeroes in the socket address means binding to all addresses
|
|
|
|
* for both IPv4 and IPv6 (INADDR_ANY and IN6ADDR_ANY_INIT).
|
|
|
|
*/
|
|
|
|
memset(sa, 0, sizeof(*sa));
|
|
|
|
sa->sin.sin_family = AF_INET;
|
|
|
|
|
|
|
|
*proto = SOCK_STREAM;
|
|
|
|
|
|
|
|
if (strncmp(str, "udp://", 6) == 0) {
|
|
|
|
str += 6;
|
|
|
|
*proto = SOCK_DGRAM;
|
|
|
|
} else if (strncmp(str, "tcp://", 6) == 0) {
|
|
|
|
str += 6;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (sscanf(str, "%u.%u.%u.%u:%u%n", &a, &b, &c, &d, &port, &len) == 5) {
|
|
|
|
/* Bind to a specific IPv4 address, e.g. 192.168.1.5:8080 */
|
|
|
|
sa->sin.sin_addr.s_addr =
|
|
|
|
htonl(((uint32_t) a << 24) | ((uint32_t) b << 16) | c << 8 | d);
|
|
|
|
sa->sin.sin_port = htons((uint16_t) port);
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifdef MG_ENABLE_IPV6
|
2015-09-08 19:49:03 +08:00
|
|
|
} else if (sscanf(str, "[%99[^]]]:%u%n", buf, &port, &len) == 2 &&
|
|
|
|
inet_pton(AF_INET6, buf, &sa->sin6.sin6_addr)) {
|
|
|
|
/* IPv6 address, e.g. [3ffe:2a00:100:7031::1]:8080 */
|
|
|
|
sa->sin6.sin6_family = AF_INET6;
|
|
|
|
sa->sin.sin_port = htons((uint16_t) port);
|
|
|
|
#endif
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_DISABLE_RESOLVER
|
2015-09-08 19:49:03 +08:00
|
|
|
} else if (strlen(str) < host_len &&
|
|
|
|
sscanf(str, "%[^ :]:%u%n", host, &port, &len) == 2) {
|
|
|
|
sa->sin.sin_port = htons((uint16_t) port);
|
|
|
|
if (mg_resolve_from_hosts_file(host, sa) != 0) {
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
} else if (sscanf(str, ":%u%n", &port, &len) == 1 ||
|
|
|
|
sscanf(str, "%u%n", &port, &len) == 1) {
|
|
|
|
/* If only port is specified, bind to IPv4, INADDR_ANY */
|
|
|
|
sa->sin.sin_port = htons((uint16_t) port);
|
|
|
|
} else {
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2015-11-19 06:47:17 +08:00
|
|
|
ch = str[len]; /* Character that follows the address */
|
|
|
|
return port < 0xffffUL && (ch == '\0' || ch == ',' || isspace(ch)) ? len : -1;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifdef MG_ENABLE_SSL
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Certificate generation script is at
|
|
|
|
* https://github.com/cesanta/mongoose/blob/master/scripts/generate_ssl_certificates.sh
|
|
|
|
*/
|
|
|
|
|
2015-11-19 06:47:17 +08:00
|
|
|
#ifndef MG_DISABLE_PFS
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Cipher suite options used for TLS negotiation.
|
|
|
|
* https://wiki.mozilla.org/Security/Server_Side_TLS#Recommended_configurations
|
|
|
|
*/
|
|
|
|
static const char mg_s_cipher_list[] =
|
2015-09-19 19:38:46 +08:00
|
|
|
#if defined(MG_SSL_CRYPTO_MODERN)
|
2015-09-08 19:49:03 +08:00
|
|
|
"ECDHE-RSA-AES128-GCM-SHA256:ECDHE-ECDSA-AES128-GCM-SHA256:"
|
|
|
|
"ECDHE-RSA-AES256-GCM-SHA384:ECDHE-ECDSA-AES256-GCM-SHA384:"
|
|
|
|
"DHE-RSA-AES128-GCM-SHA256:DHE-DSS-AES128-GCM-SHA256:kEDH+AESGCM:"
|
|
|
|
"ECDHE-RSA-AES128-SHA256:ECDHE-ECDSA-AES128-SHA256:ECDHE-RSA-AES128-SHA:"
|
|
|
|
"ECDHE-ECDSA-AES128-SHA:ECDHE-RSA-AES256-SHA384:ECDHE-ECDSA-AES256-SHA384:"
|
|
|
|
"ECDHE-RSA-AES256-SHA:ECDHE-ECDSA-AES256-SHA:DHE-RSA-AES128-SHA256:"
|
|
|
|
"DHE-RSA-AES128-SHA:DHE-DSS-AES128-SHA256:DHE-RSA-AES256-SHA256:"
|
|
|
|
"DHE-DSS-AES256-SHA:DHE-RSA-AES256-SHA:"
|
|
|
|
"!aNULL:!eNULL:!EXPORT:!DES:!RC4:!3DES:!MD5:!PSK"
|
2015-09-19 19:38:46 +08:00
|
|
|
#elif defined(MG_SSL_CRYPTO_OLD)
|
2015-09-08 19:49:03 +08:00
|
|
|
"ECDHE-RSA-AES128-GCM-SHA256:ECDHE-ECDSA-AES128-GCM-SHA256:"
|
|
|
|
"ECDHE-RSA-AES256-GCM-SHA384:ECDHE-ECDSA-AES256-GCM-SHA384:"
|
|
|
|
"DHE-RSA-AES128-GCM-SHA256:DHE-DSS-AES128-GCM-SHA256:kEDH+AESGCM:"
|
|
|
|
"ECDHE-RSA-AES128-SHA256:ECDHE-ECDSA-AES128-SHA256:ECDHE-RSA-AES128-SHA:"
|
|
|
|
"ECDHE-ECDSA-AES128-SHA:ECDHE-RSA-AES256-SHA384:ECDHE-ECDSA-AES256-SHA384:"
|
|
|
|
"ECDHE-RSA-AES256-SHA:ECDHE-ECDSA-AES256-SHA:DHE-RSA-AES128-SHA256:"
|
|
|
|
"DHE-RSA-AES128-SHA:DHE-DSS-AES128-SHA256:DHE-RSA-AES256-SHA256:"
|
|
|
|
"DHE-DSS-AES256-SHA:DHE-RSA-AES256-SHA:ECDHE-RSA-DES-CBC3-SHA:"
|
|
|
|
"ECDHE-ECDSA-DES-CBC3-SHA:AES128-GCM-SHA256:AES256-GCM-SHA384:"
|
|
|
|
"AES128-SHA256:AES256-SHA256:AES128-SHA:AES256-SHA:AES:DES-CBC3-SHA:"
|
|
|
|
"HIGH:!aNULL:!eNULL:!EXPORT:!DES:!RC4:!MD5:!PSK:!aECDH:"
|
|
|
|
"!EDH-DSS-DES-CBC3-SHA:!EDH-RSA-DES-CBC3-SHA:!KRB5-DES-CBC3-SHA"
|
|
|
|
#else /* Default - intermediate. */
|
|
|
|
"ECDHE-RSA-AES128-GCM-SHA256:ECDHE-ECDSA-AES128-GCM-SHA256:"
|
|
|
|
"ECDHE-RSA-AES256-GCM-SHA384:ECDHE-ECDSA-AES256-GCM-SHA384:"
|
|
|
|
"DHE-RSA-AES128-GCM-SHA256:DHE-DSS-AES128-GCM-SHA256:kEDH+AESGCM:"
|
|
|
|
"ECDHE-RSA-AES128-SHA256:ECDHE-ECDSA-AES128-SHA256:ECDHE-RSA-AES128-SHA:"
|
|
|
|
"ECDHE-ECDSA-AES128-SHA:ECDHE-RSA-AES256-SHA384:ECDHE-ECDSA-AES256-SHA384:"
|
|
|
|
"ECDHE-RSA-AES256-SHA:ECDHE-ECDSA-AES256-SHA:DHE-RSA-AES128-SHA256:"
|
|
|
|
"DHE-RSA-AES128-SHA:DHE-DSS-AES128-SHA256:DHE-RSA-AES256-SHA256:"
|
|
|
|
"DHE-DSS-AES256-SHA:DHE-RSA-AES256-SHA:AES128-GCM-SHA256:AES256-GCM-SHA384:"
|
|
|
|
"AES128-SHA256:AES256-SHA256:AES128-SHA:AES256-SHA:AES:CAMELLIA:"
|
|
|
|
"DES-CBC3-SHA:!aNULL:!eNULL:!EXPORT:!DES:!RC4:!MD5:!PSK:!aECDH:"
|
|
|
|
"!EDH-DSS-DES-CBC3-SHA:!EDH-RSA-DES-CBC3-SHA:!KRB5-DES-CBC3-SHA"
|
|
|
|
#endif
|
|
|
|
;
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Default DH params for PFS cipher negotiation. This is a 2048-bit group.
|
|
|
|
* Will be used if none are provided by the user in the certificate file.
|
|
|
|
*/
|
|
|
|
static const char mg_s_default_dh_params[] =
|
|
|
|
"\
|
|
|
|
-----BEGIN DH PARAMETERS-----\n\
|
|
|
|
MIIBCAKCAQEAlvbgD/qh9znWIlGFcV0zdltD7rq8FeShIqIhkQ0C7hYFThrBvF2E\n\
|
|
|
|
Z9bmgaP+sfQwGpVlv9mtaWjvERbu6mEG7JTkgmVUJrUt/wiRzwTaCXBqZkdUO8Tq\n\
|
|
|
|
+E6VOEQAilstG90ikN1Tfo+K6+X68XkRUIlgawBTKuvKVwBhuvlqTGerOtnXWnrt\n\
|
|
|
|
ym//hd3cd5PBYGBix0i7oR4xdghvfR2WLVu0LgdThTBb6XP7gLd19cQ1JuBtAajZ\n\
|
|
|
|
wMuPn7qlUkEFDIkAZy59/Hue/H2Q2vU/JsvVhHWCQBL4F1ofEAt50il6ZxR1QfFK\n\
|
|
|
|
9VGKDC4oOgm9DlxwwBoC2FjqmvQlqVV3kwIBAg==\n\
|
|
|
|
-----END DH PARAMETERS-----\n";
|
|
|
|
#endif
|
|
|
|
|
|
|
|
static int mg_use_ca_cert(SSL_CTX *ctx, const char *cert) {
|
|
|
|
if (ctx == NULL) {
|
|
|
|
return -1;
|
|
|
|
} else if (cert == NULL || cert[0] == '\0') {
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
SSL_CTX_set_verify(ctx, SSL_VERIFY_PEER | SSL_VERIFY_FAIL_IF_NO_PEER_CERT, 0);
|
|
|
|
return SSL_CTX_load_verify_locations(ctx, cert, NULL) == 1 ? 0 : -2;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int mg_use_cert(SSL_CTX *ctx, const char *pem_file) {
|
|
|
|
if (ctx == NULL) {
|
|
|
|
return -1;
|
|
|
|
} else if (pem_file == NULL || pem_file[0] == '\0') {
|
|
|
|
return 0;
|
|
|
|
} else if (SSL_CTX_use_certificate_file(ctx, pem_file, 1) == 0 ||
|
|
|
|
SSL_CTX_use_PrivateKey_file(ctx, pem_file, 1) == 0) {
|
|
|
|
return -2;
|
|
|
|
} else {
|
2015-11-19 06:47:17 +08:00
|
|
|
#ifndef MG_DISABLE_PFS
|
2015-09-08 19:49:03 +08:00
|
|
|
BIO *bio = NULL;
|
|
|
|
DH *dh = NULL;
|
|
|
|
|
|
|
|
/* Try to read DH parameters from the cert/key file. */
|
|
|
|
bio = BIO_new_file(pem_file, "r");
|
|
|
|
if (bio != NULL) {
|
|
|
|
dh = PEM_read_bio_DHparams(bio, NULL, NULL, NULL);
|
|
|
|
BIO_free(bio);
|
|
|
|
}
|
|
|
|
/*
|
|
|
|
* If there are no DH params in the file, fall back to hard-coded ones.
|
|
|
|
* Not ideal, but better than nothing.
|
|
|
|
*/
|
|
|
|
if (dh == NULL) {
|
|
|
|
bio = BIO_new_mem_buf((void *) mg_s_default_dh_params, -1);
|
|
|
|
dh = PEM_read_bio_DHparams(bio, NULL, NULL, NULL);
|
|
|
|
BIO_free(bio);
|
|
|
|
}
|
|
|
|
if (dh != NULL) {
|
|
|
|
SSL_CTX_set_tmp_dh(ctx, dh);
|
|
|
|
SSL_CTX_set_options(ctx, SSL_OP_SINGLE_DH_USE);
|
|
|
|
DH_free(dh);
|
|
|
|
}
|
2015-11-19 06:47:17 +08:00
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
SSL_CTX_set_mode(ctx, SSL_MODE_ACCEPT_MOVING_WRITE_BUFFER);
|
|
|
|
SSL_CTX_use_certificate_chain_file(ctx, pem_file);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Turn the connection into SSL mode.
|
|
|
|
* `cert` is the certificate file in PEM format. For listening connections,
|
|
|
|
* certificate file must contain private key and server certificate,
|
|
|
|
* concatenated. It may also contain DH params - these will be used for more
|
|
|
|
* secure key exchange. `ca_cert` is a certificate authority (CA) PEM file, and
|
|
|
|
* it is optional (can be set to NULL). If `ca_cert` is non-NULL, then
|
|
|
|
* the connection is so-called two-way-SSL: other peer's certificate is
|
|
|
|
* checked against the `ca_cert`.
|
|
|
|
*
|
|
|
|
* Handy OpenSSL command to generate test self-signed certificate:
|
|
|
|
*
|
|
|
|
* openssl req -x509 -newkey rsa:2048 -keyout key.pem -out cert.pem -days 999
|
|
|
|
*
|
|
|
|
* Return NULL on success, or error message on failure.
|
|
|
|
*/
|
|
|
|
const char *mg_set_ssl(struct mg_connection *nc, const char *cert,
|
|
|
|
const char *ca_cert) {
|
|
|
|
const char *result = NULL;
|
2015-12-09 17:43:41 +08:00
|
|
|
DBG(("%p %s %s", nc, (cert ? cert : ""), (ca_cert ? ca_cert : "")));
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-12-23 23:54:33 +08:00
|
|
|
if (nc->ssl != NULL) {
|
|
|
|
SSL_free(nc->ssl);
|
|
|
|
nc->ssl = NULL;
|
|
|
|
}
|
|
|
|
if (nc->ssl_ctx != NULL) {
|
|
|
|
SSL_CTX_free(nc->ssl_ctx);
|
|
|
|
nc->ssl_ctx = NULL;
|
|
|
|
}
|
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
if ((nc->flags & MG_F_LISTENING) &&
|
2015-09-08 19:49:03 +08:00
|
|
|
(nc->ssl_ctx = SSL_CTX_new(SSLv23_server_method())) == NULL) {
|
|
|
|
result = "SSL_CTX_new() failed";
|
2015-09-19 19:38:46 +08:00
|
|
|
} else if (!(nc->flags & MG_F_LISTENING) &&
|
2015-09-08 19:49:03 +08:00
|
|
|
(nc->ssl_ctx = SSL_CTX_new(SSLv23_client_method())) == NULL) {
|
|
|
|
result = "SSL_CTX_new() failed";
|
|
|
|
} else if (mg_use_cert(nc->ssl_ctx, cert) != 0) {
|
|
|
|
result = "Invalid ssl cert";
|
|
|
|
} else if (mg_use_ca_cert(nc->ssl_ctx, ca_cert) != 0) {
|
|
|
|
result = "Invalid CA cert";
|
2015-09-19 19:38:46 +08:00
|
|
|
} else if (!(nc->flags & MG_F_LISTENING) &&
|
2015-09-08 19:49:03 +08:00
|
|
|
(nc->ssl = SSL_new(nc->ssl_ctx)) == NULL) {
|
|
|
|
result = "SSL_new() failed";
|
2015-09-19 19:38:46 +08:00
|
|
|
} else if (!(nc->flags & MG_F_LISTENING) && nc->sock != INVALID_SOCKET) {
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Socket is open here only if we are connecting to IP address
|
|
|
|
* and does not open if we are connecting using async DNS resolver
|
|
|
|
*/
|
|
|
|
SSL_set_fd(nc->ssl, nc->sock);
|
|
|
|
}
|
|
|
|
|
|
|
|
/* TODO(rojer): remove when krypton exposes this function, even a dummy one */
|
|
|
|
#ifdef OPENSSL_VERSION_NUMBER
|
|
|
|
SSL_CTX_set_cipher_list(nc->ssl_ctx, mg_s_cipher_list);
|
|
|
|
#endif
|
|
|
|
return result;
|
|
|
|
}
|
2015-09-19 19:38:46 +08:00
|
|
|
#endif /* MG_ENABLE_SSL */
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-27 04:31:45 +08:00
|
|
|
struct mg_connection *mg_if_accept_tcp_cb(struct mg_connection *lc,
|
|
|
|
union socket_address *sa,
|
|
|
|
size_t sa_len) {
|
2015-10-22 20:04:35 +08:00
|
|
|
struct mg_add_sock_opts opts;
|
|
|
|
struct mg_connection *nc;
|
|
|
|
(void) sa_len;
|
|
|
|
memset(&opts, 0, sizeof(opts));
|
|
|
|
nc = mg_create_connection(lc->mgr, lc->handler, opts);
|
2015-10-27 04:31:45 +08:00
|
|
|
if (nc == NULL) return NULL;
|
2015-10-22 20:04:35 +08:00
|
|
|
nc->listener = lc;
|
|
|
|
nc->proto_data = lc->proto_data;
|
|
|
|
nc->proto_handler = lc->proto_handler;
|
|
|
|
nc->user_data = lc->user_data;
|
|
|
|
nc->recv_mbuf_limit = lc->recv_mbuf_limit;
|
|
|
|
nc->sa = *sa;
|
|
|
|
mg_add_conn(nc->mgr, nc);
|
2015-12-07 23:56:13 +08:00
|
|
|
if (lc->ssl_ctx == NULL) {
|
2015-10-27 04:31:45 +08:00
|
|
|
/* For non-SSL connections deliver MG_EV_ACCEPT right away. */
|
2015-10-29 06:12:55 +08:00
|
|
|
mg_call(nc, NULL, MG_EV_ACCEPT, &nc->sa);
|
2015-10-22 20:04:35 +08:00
|
|
|
}
|
|
|
|
DBG(("%p %p %d %d, %p %p", lc, nc, nc->sock, (int) nc->flags, lc->ssl_ctx,
|
|
|
|
nc->ssl));
|
2015-10-27 04:31:45 +08:00
|
|
|
return nc;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-12-12 22:16:48 +08:00
|
|
|
MG_INTERNAL size_t recv_avail_size(struct mg_connection *conn, size_t max) {
|
2015-09-08 19:49:03 +08:00
|
|
|
size_t avail;
|
|
|
|
if (conn->recv_mbuf_limit < conn->recv_mbuf.len) return 0;
|
|
|
|
avail = conn->recv_mbuf_limit - conn->recv_mbuf.len;
|
|
|
|
return avail > max ? max : avail;
|
|
|
|
}
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
void mg_send(struct mg_connection *nc, const void *buf, int len) {
|
2016-01-08 21:14:55 +08:00
|
|
|
nc->last_io_time = mg_time();
|
2015-10-22 20:04:35 +08:00
|
|
|
if (nc->flags & MG_F_UDP) {
|
|
|
|
mg_if_udp_send(nc, buf, len);
|
|
|
|
} else {
|
|
|
|
mg_if_tcp_send(nc, buf, len);
|
|
|
|
}
|
2015-11-27 18:32:36 +08:00
|
|
|
#if !defined(NO_LIBC) && !defined(MG_DISABLE_HEXDUMP)
|
2015-10-22 20:04:35 +08:00
|
|
|
if (nc->mgr && nc->mgr->hexdump_file != NULL) {
|
2015-11-27 18:32:36 +08:00
|
|
|
mg_hexdump_connection(nc, nc->mgr->hexdump_file, buf, len, MG_EV_SEND);
|
2015-10-19 21:55:01 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
#endif
|
2015-10-22 20:04:35 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
void mg_if_sent_cb(struct mg_connection *nc, int num_sent) {
|
|
|
|
if (num_sent < 0) {
|
|
|
|
nc->flags |= MG_F_CLOSE_IMMEDIATELY;
|
|
|
|
}
|
2015-10-29 06:12:55 +08:00
|
|
|
mg_call(nc, NULL, MG_EV_SEND, &num_sent);
|
2015-10-22 20:04:35 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
static void mg_recv_common(struct mg_connection *nc, void *buf, int len) {
|
|
|
|
DBG(("%p %d %u", nc, len, (unsigned int) nc->recv_mbuf.len));
|
2015-11-18 15:07:48 +08:00
|
|
|
if (nc->flags & MG_F_CLOSE_IMMEDIATELY) {
|
|
|
|
DBG(("%p discarded %d bytes", nc, len));
|
|
|
|
/*
|
|
|
|
* This connection will not survive next poll. Do not deliver events,
|
|
|
|
* send data to /dev/null without acking.
|
|
|
|
*/
|
|
|
|
MG_FREE(buf);
|
|
|
|
return;
|
|
|
|
}
|
2016-01-08 21:14:55 +08:00
|
|
|
nc->last_io_time = mg_time();
|
2015-10-22 20:04:35 +08:00
|
|
|
if (nc->recv_mbuf.len == 0) {
|
|
|
|
/* Adopt buf as recv_mbuf's backing store. */
|
|
|
|
mbuf_free(&nc->recv_mbuf);
|
|
|
|
nc->recv_mbuf.buf = (char *) buf;
|
|
|
|
nc->recv_mbuf.size = nc->recv_mbuf.len = len;
|
2015-10-19 21:55:01 +08:00
|
|
|
} else {
|
2015-10-22 20:04:35 +08:00
|
|
|
size_t avail = recv_avail_size(nc, len);
|
|
|
|
len = avail;
|
|
|
|
mbuf_append(&nc->recv_mbuf, buf, len);
|
|
|
|
MG_FREE(buf);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-10-29 06:12:55 +08:00
|
|
|
mg_call(nc, NULL, MG_EV_RECV, &len);
|
2015-10-19 21:55:01 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
void mg_if_recv_tcp_cb(struct mg_connection *nc, void *buf, int len) {
|
|
|
|
mg_recv_common(nc, buf, len);
|
|
|
|
mg_if_recved(nc, len);
|
|
|
|
}
|
|
|
|
|
|
|
|
void mg_if_recv_udp_cb(struct mg_connection *nc, void *buf, int len,
|
|
|
|
union socket_address *sa, size_t sa_len) {
|
|
|
|
assert(nc->flags & MG_F_UDP);
|
|
|
|
DBG(("%p %u", nc, (unsigned int) len));
|
|
|
|
if (nc->flags & MG_F_LISTENING) {
|
|
|
|
struct mg_connection *lc = nc;
|
|
|
|
/*
|
|
|
|
* Do we have an existing connection for this source?
|
|
|
|
* This is very inefficient for long connection lists.
|
|
|
|
*/
|
|
|
|
for (nc = mg_next(lc->mgr, NULL); nc != NULL; nc = mg_next(lc->mgr, nc)) {
|
|
|
|
if (memcmp(&nc->sa.sa, &sa->sa, sa_len) == 0) break;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-10-22 20:04:35 +08:00
|
|
|
if (nc == NULL) {
|
|
|
|
struct mg_add_sock_opts opts;
|
|
|
|
memset(&opts, 0, sizeof(opts));
|
|
|
|
nc = mg_create_connection(lc->mgr, lc->handler, opts);
|
|
|
|
}
|
|
|
|
if (nc != NULL) {
|
|
|
|
nc->sock = lc->sock;
|
|
|
|
nc->listener = lc;
|
|
|
|
nc->sa = *sa;
|
|
|
|
nc->proto_data = lc->proto_data;
|
|
|
|
nc->proto_handler = lc->proto_handler;
|
|
|
|
nc->user_data = lc->user_data;
|
|
|
|
nc->recv_mbuf_limit = lc->recv_mbuf_limit;
|
|
|
|
nc->flags = MG_F_UDP;
|
|
|
|
mg_add_conn(lc->mgr, nc);
|
2015-10-29 06:12:55 +08:00
|
|
|
mg_call(nc, NULL, MG_EV_ACCEPT, &nc->sa);
|
2015-10-22 20:04:35 +08:00
|
|
|
} else {
|
|
|
|
DBG(("OOM"));
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
}
|
2015-10-22 20:04:35 +08:00
|
|
|
if (nc != NULL) {
|
|
|
|
mg_recv_common(nc, buf, len);
|
|
|
|
} else {
|
|
|
|
/* Drop on the floor. */
|
|
|
|
MG_FREE(buf);
|
|
|
|
}
|
|
|
|
mg_if_recved(nc, len);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
/*
|
|
|
|
* Schedules an async connect for a resolved address and proto.
|
|
|
|
* Called from two places: `mg_connect_opt()` and from async resolver.
|
|
|
|
* When called from the async resolver, it must trigger `MG_EV_CONNECT` event
|
|
|
|
* with a failure flag to indicate connection failure.
|
|
|
|
*/
|
|
|
|
MG_INTERNAL struct mg_connection *mg_do_connect(struct mg_connection *nc,
|
|
|
|
int proto,
|
|
|
|
union socket_address *sa) {
|
|
|
|
DBG(("%p %s://%s:%hu", nc, proto == SOCK_DGRAM ? "udp" : "tcp",
|
|
|
|
inet_ntoa(sa->sin.sin_addr), ntohs(sa->sin.sin_port)));
|
2015-09-22 21:16:40 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
nc->flags |= MG_F_CONNECTING;
|
|
|
|
if (proto == SOCK_DGRAM) {
|
|
|
|
mg_if_connect_udp(nc);
|
|
|
|
} else {
|
|
|
|
mg_if_connect_tcp(nc, sa);
|
|
|
|
}
|
|
|
|
mg_add_conn(nc->mgr, nc);
|
|
|
|
return nc;
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
void mg_if_connect_cb(struct mg_connection *nc, int err) {
|
|
|
|
DBG(("%p connect, err=%d", nc, err));
|
|
|
|
nc->flags &= ~MG_F_CONNECTING;
|
2015-12-07 23:56:13 +08:00
|
|
|
if (err != 0) {
|
2015-10-22 20:04:35 +08:00
|
|
|
nc->flags |= MG_F_CLOSE_IMMEDIATELY;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-10-29 06:12:55 +08:00
|
|
|
mg_call(nc, NULL, MG_EV_CONNECT, &err);
|
2015-10-22 20:04:35 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
#ifndef MG_DISABLE_RESOLVER
|
|
|
|
/*
|
|
|
|
* Callback for the async resolver on mg_connect_opt() call.
|
|
|
|
* Main task of this function is to trigger MG_EV_CONNECT event with
|
|
|
|
* either failure (and dealloc the connection)
|
|
|
|
* or success (and proceed with connect()
|
|
|
|
*/
|
2015-12-08 20:25:44 +08:00
|
|
|
static void resolve_cb(struct mg_dns_message *msg, void *data,
|
|
|
|
enum mg_resolve_err e) {
|
2015-10-22 20:04:35 +08:00
|
|
|
struct mg_connection *nc = (struct mg_connection *) data;
|
|
|
|
int i;
|
|
|
|
int failure = -1;
|
|
|
|
|
2015-12-08 20:25:44 +08:00
|
|
|
nc->flags &= ~MG_F_RESOLVING;
|
2015-10-22 20:04:35 +08:00
|
|
|
if (msg != NULL) {
|
|
|
|
/*
|
|
|
|
* Take the first DNS A answer and run...
|
|
|
|
*/
|
|
|
|
for (i = 0; i < msg->num_answers; i++) {
|
|
|
|
if (msg->answers[i].rtype == MG_DNS_A_RECORD) {
|
|
|
|
/*
|
|
|
|
* Async resolver guarantees that there is at least one answer.
|
|
|
|
* TODO(lsm): handle IPv6 answers too
|
|
|
|
*/
|
|
|
|
mg_dns_parse_record_data(msg, &msg->answers[i], &nc->sa.sin.sin_addr,
|
|
|
|
4);
|
|
|
|
mg_do_connect(nc, nc->flags & MG_F_UDP ? SOCK_DGRAM : SOCK_STREAM,
|
|
|
|
&nc->sa);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-12-08 20:25:44 +08:00
|
|
|
if (e == MG_RESOLVE_TIMEOUT) {
|
2016-01-08 21:14:55 +08:00
|
|
|
double now = mg_time();
|
2015-12-08 20:25:44 +08:00
|
|
|
mg_call(nc, NULL, MG_EV_TIMER, &now);
|
|
|
|
}
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
/*
|
|
|
|
* If we get there was no MG_DNS_A_RECORD in the answer
|
|
|
|
*/
|
2015-10-29 06:12:55 +08:00
|
|
|
mg_call(nc, NULL, MG_EV_CONNECT, &failure);
|
2015-11-30 20:37:47 +08:00
|
|
|
mg_call(nc, NULL, MG_EV_CLOSE, NULL);
|
2015-10-22 20:04:35 +08:00
|
|
|
mg_destroy_conn(nc);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-10-22 20:04:35 +08:00
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
struct mg_connection *mg_connect(struct mg_mgr *mgr, const char *address,
|
|
|
|
mg_event_handler_t callback) {
|
2015-10-24 07:31:10 +08:00
|
|
|
struct mg_connect_opts opts;
|
|
|
|
memset(&opts, 0, sizeof(opts));
|
2015-10-22 20:04:35 +08:00
|
|
|
return mg_connect_opt(mgr, address, callback, opts);
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
struct mg_connection *mg_connect_opt(struct mg_mgr *mgr, const char *address,
|
|
|
|
mg_event_handler_t callback,
|
|
|
|
struct mg_connect_opts opts) {
|
|
|
|
struct mg_connection *nc = NULL;
|
|
|
|
int proto, rc;
|
|
|
|
struct mg_add_sock_opts add_sock_opts;
|
|
|
|
char host[MG_MAX_HOST_LEN];
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
MG_COPY_COMMON_CONNECTION_OPTIONS(&add_sock_opts, &opts);
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
if ((nc = mg_create_connection(mgr, callback, add_sock_opts)) == NULL) {
|
|
|
|
return NULL;
|
|
|
|
} else if ((rc = mg_parse_address(address, &nc->sa, &proto, host,
|
|
|
|
sizeof(host))) < 0) {
|
|
|
|
/* Address is malformed */
|
|
|
|
MG_SET_PTRPTR(opts.error_string, "cannot parse address");
|
|
|
|
mg_destroy_conn(nc);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
nc->flags |= opts.flags & _MG_ALLOWED_CONNECT_FLAGS_MASK;
|
|
|
|
nc->flags |= (proto == SOCK_DGRAM) ? MG_F_UDP : 0;
|
|
|
|
nc->user_data = opts.user_data;
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
if (rc == 0) {
|
|
|
|
#ifndef MG_DISABLE_RESOLVER
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
2015-10-22 20:04:35 +08:00
|
|
|
* DNS resolution is required for host.
|
|
|
|
* mg_parse_address() fills port in nc->sa, which we pass to resolve_cb()
|
2015-09-08 19:49:03 +08:00
|
|
|
*/
|
2015-12-08 20:25:44 +08:00
|
|
|
struct mg_connection *dns_conn = NULL;
|
|
|
|
struct mg_resolve_async_opts o;
|
|
|
|
memset(&o, 0, sizeof(o));
|
|
|
|
o.dns_conn = &dns_conn;
|
|
|
|
if (mg_resolve_async_opt(nc->mgr, host, MG_DNS_A_RECORD, resolve_cb, nc,
|
|
|
|
o) != 0) {
|
2015-10-22 20:04:35 +08:00
|
|
|
MG_SET_PTRPTR(opts.error_string, "cannot schedule DNS lookup");
|
|
|
|
mg_destroy_conn(nc);
|
|
|
|
return NULL;
|
|
|
|
}
|
2015-12-08 20:25:44 +08:00
|
|
|
nc->priv_2 = dns_conn;
|
|
|
|
nc->flags |= MG_F_RESOLVING;
|
2015-10-22 20:04:35 +08:00
|
|
|
return nc;
|
|
|
|
#else
|
|
|
|
MG_SET_PTRPTR(opts.error_string, "Resolver is disabled");
|
|
|
|
mg_destroy_conn(nc);
|
|
|
|
return NULL;
|
|
|
|
#endif
|
|
|
|
} else {
|
|
|
|
/* Address is parsed and resolved to IP. proceed with connect() */
|
|
|
|
return mg_do_connect(nc, proto, &nc->sa);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
struct mg_connection *mg_bind(struct mg_mgr *srv, const char *address,
|
|
|
|
mg_event_handler_t event_handler) {
|
2015-10-24 07:31:10 +08:00
|
|
|
struct mg_bind_opts opts;
|
|
|
|
memset(&opts, 0, sizeof(opts));
|
2015-10-22 20:04:35 +08:00
|
|
|
return mg_bind_opt(srv, address, event_handler, opts);
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
struct mg_connection *mg_bind_opt(struct mg_mgr *mgr, const char *address,
|
|
|
|
mg_event_handler_t callback,
|
|
|
|
struct mg_bind_opts opts) {
|
|
|
|
union socket_address sa;
|
|
|
|
struct mg_connection *nc = NULL;
|
|
|
|
int proto, rc;
|
|
|
|
struct mg_add_sock_opts add_sock_opts;
|
|
|
|
char host[MG_MAX_HOST_LEN];
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
MG_COPY_COMMON_CONNECTION_OPTIONS(&add_sock_opts, &opts);
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
if (mg_parse_address(address, &sa, &proto, host, sizeof(host)) <= 0) {
|
|
|
|
MG_SET_PTRPTR(opts.error_string, "cannot parse address");
|
|
|
|
return NULL;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
nc = mg_create_connection(mgr, callback, add_sock_opts);
|
|
|
|
if (nc == NULL) {
|
|
|
|
return NULL;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
nc->sa = sa;
|
|
|
|
nc->flags |= MG_F_LISTENING;
|
|
|
|
if (proto == SOCK_DGRAM) {
|
|
|
|
nc->flags |= MG_F_UDP;
|
|
|
|
rc = mg_if_listen_udp(nc, &nc->sa);
|
|
|
|
} else {
|
|
|
|
rc = mg_if_listen_tcp(nc, &nc->sa);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-10-22 20:04:35 +08:00
|
|
|
if (rc != 0) {
|
|
|
|
DBG(("Failed to open listener: %d", rc));
|
|
|
|
MG_SET_PTRPTR(opts.error_string, "failed to open listener");
|
|
|
|
mg_destroy_conn(nc);
|
|
|
|
return NULL;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-10-22 20:04:35 +08:00
|
|
|
mg_add_conn(nc->mgr, nc);
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
return nc;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
struct mg_connection *mg_next(struct mg_mgr *s, struct mg_connection *conn) {
|
|
|
|
return conn == NULL ? s->active_connections : conn->next;
|
|
|
|
}
|
|
|
|
|
2015-10-24 07:05:05 +08:00
|
|
|
#ifndef MG_DISABLE_SOCKETPAIR
|
2015-10-22 20:04:35 +08:00
|
|
|
void mg_broadcast(struct mg_mgr *mgr, mg_event_handler_t cb, void *data,
|
|
|
|
size_t len) {
|
2015-09-08 19:49:03 +08:00
|
|
|
struct ctl_msg ctl_msg;
|
2015-10-22 04:40:58 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
/*
|
|
|
|
* Mongoose manager has a socketpair, `struct mg_mgr::ctl`,
|
|
|
|
* where `mg_broadcast()` pushes the message.
|
|
|
|
* `mg_mgr_poll()` wakes up, reads a message from the socket pair, and calls
|
|
|
|
* specified callback for each connection. Thus the callback function executes
|
|
|
|
* in event manager thread.
|
|
|
|
*/
|
|
|
|
if (mgr->ctl[0] != INVALID_SOCKET && data != NULL &&
|
|
|
|
len < sizeof(ctl_msg.message)) {
|
|
|
|
size_t dummy;
|
|
|
|
|
|
|
|
ctl_msg.callback = cb;
|
|
|
|
memcpy(ctl_msg.message, data, len);
|
|
|
|
dummy = MG_SEND_FUNC(mgr->ctl[0], (char *) &ctl_msg,
|
|
|
|
offsetof(struct ctl_msg, message) + len, 0);
|
|
|
|
dummy = MG_RECV_FUNC(mgr->ctl[0], (char *) &len, 1, 0);
|
|
|
|
(void) dummy; /* https://gcc.gnu.org/bugzilla/show_bug.cgi?id=25509 */
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
}
|
2015-10-24 07:05:05 +08:00
|
|
|
#endif /* MG_DISABLE_SOCKETPAIR */
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
static int isbyte(int n) {
|
|
|
|
return n >= 0 && n <= 255;
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
static int parse_net(const char *spec, uint32_t *net, uint32_t *mask) {
|
|
|
|
int n, a, b, c, d, slash = 32, len = 0;
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
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) | d;
|
|
|
|
*mask = slash ? 0xffffffffU << (32 - slash) : 0;
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
return len;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
int mg_check_ip_acl(const char *acl, uint32_t remote_ip) {
|
|
|
|
int allowed, flag;
|
|
|
|
uint32_t net, mask;
|
|
|
|
struct mg_str vec;
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
/* If any ACL is set, deny by default */
|
|
|
|
allowed = (acl == NULL || *acl == '\0') ? '+' : '-';
|
|
|
|
|
|
|
|
while ((acl = mg_next_comma_list_entry(acl, &vec, NULL)) != NULL) {
|
|
|
|
flag = vec.p[0];
|
|
|
|
if ((flag != '+' && flag != '-') ||
|
|
|
|
parse_net(&vec.p[1], &net, &mask) == 0) {
|
|
|
|
return -1;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
if (net == (remote_ip & mask)) {
|
|
|
|
allowed = flag;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
}
|
2015-10-22 20:04:35 +08:00
|
|
|
|
|
|
|
return allowed == '+';
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
/* Move data from one connection to another */
|
|
|
|
void mg_forward(struct mg_connection *from, struct mg_connection *to) {
|
|
|
|
mg_send(to, from->recv_mbuf.buf, from->recv_mbuf.len);
|
|
|
|
mbuf_remove(&from->recv_mbuf, from->recv_mbuf.len);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-12-08 20:25:44 +08:00
|
|
|
|
|
|
|
double mg_set_timer(struct mg_connection *c, double timestamp) {
|
|
|
|
double result = c->ev_timer_time;
|
|
|
|
c->ev_timer_time = timestamp;
|
|
|
|
/*
|
|
|
|
* If this connection is resolving, it's not in the list of active
|
|
|
|
* connections, so not processed yet. It has a DNS resolver connection
|
|
|
|
* linked to it. Set up a timer for the DNS connection.
|
|
|
|
*/
|
2015-12-08 21:34:32 +08:00
|
|
|
DBG(("%p %p %d -> %lu", c, c->priv_2, c->flags & MG_F_RESOLVING,
|
|
|
|
(unsigned long) timestamp));
|
2015-12-08 20:25:44 +08:00
|
|
|
if ((c->flags & MG_F_RESOLVING) && c->priv_2 != NULL) {
|
|
|
|
((struct mg_connection *) c->priv_2)->ev_timer_time = timestamp;
|
|
|
|
}
|
|
|
|
return result;
|
|
|
|
}
|
2016-01-08 21:14:55 +08:00
|
|
|
|
|
|
|
double mg_time() {
|
|
|
|
return cs_time();
|
|
|
|
}
|
2015-10-22 20:04:35 +08:00
|
|
|
#ifdef NS_MODULE_LINES
|
2015-12-15 02:45:18 +08:00
|
|
|
#line 1 "./src/net_if_socket.c"
|
2015-10-22 20:04:35 +08:00
|
|
|
/**/
|
|
|
|
#endif
|
2015-10-24 07:05:05 +08:00
|
|
|
#ifndef MG_DISABLE_SOCKET_IF
|
|
|
|
|
2015-12-12 22:05:54 +08:00
|
|
|
/* Amalgamated: #include "mongoose/src/internal.h" */
|
2015-12-12 22:16:48 +08:00
|
|
|
/* Amalgamated: #include "mongoose/src/util.h" */
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
#define MG_TCP_RECV_BUFFER_SIZE 1024
|
|
|
|
#define MG_UDP_RECV_BUFFER_SIZE 1500
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
static sock_t mg_open_listening_socket(union socket_address *sa, int proto);
|
2015-10-27 04:31:45 +08:00
|
|
|
static void mg_sock_set(struct mg_connection *nc, sock_t sock);
|
2015-12-07 23:56:13 +08:00
|
|
|
#ifdef MG_ENABLE_SSL
|
|
|
|
static void mg_ssl_begin(struct mg_connection *nc);
|
|
|
|
static int mg_ssl_err(struct mg_connection *conn, int res);
|
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
void mg_set_non_blocking_mode(sock_t sock) {
|
|
|
|
#ifdef _WIN32
|
|
|
|
unsigned long on = 1;
|
|
|
|
ioctlsocket(sock, FIONBIO, &on);
|
|
|
|
#elif defined(MG_CC3200)
|
|
|
|
cc3200_set_non_blocking_mode(sock);
|
|
|
|
#else
|
|
|
|
int flags = fcntl(sock, F_GETFL, 0);
|
|
|
|
fcntl(sock, F_SETFL, flags | O_NONBLOCK);
|
|
|
|
#endif
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
int mg_is_error(int n) {
|
|
|
|
#ifdef MG_CC3200
|
|
|
|
DBG(("n = %d, errno = %d", n, errno));
|
|
|
|
if (n < 0) errno = n;
|
|
|
|
#endif
|
|
|
|
return n == 0 || (n < 0 && errno != EINTR && errno != EINPROGRESS &&
|
|
|
|
errno != EAGAIN && errno != EWOULDBLOCK
|
|
|
|
#ifdef MG_CC3200
|
|
|
|
&& errno != SL_EALREADY
|
|
|
|
#endif
|
|
|
|
#ifdef _WIN32
|
|
|
|
&& WSAGetLastError() != WSAEINTR &&
|
|
|
|
WSAGetLastError() != WSAEWOULDBLOCK
|
|
|
|
#endif
|
|
|
|
);
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
void mg_if_connect_tcp(struct mg_connection *nc,
|
|
|
|
const union socket_address *sa) {
|
|
|
|
int rc;
|
|
|
|
nc->sock = socket(AF_INET, SOCK_STREAM, 0);
|
|
|
|
if (nc->sock < 0) {
|
|
|
|
nc->sock = INVALID_SOCKET;
|
|
|
|
nc->err = errno ? errno : 1;
|
|
|
|
return;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-10-22 20:04:35 +08:00
|
|
|
#if !defined(MG_CC3200) && !defined(MG_ESP8266)
|
|
|
|
mg_set_non_blocking_mode(nc->sock);
|
|
|
|
#endif
|
|
|
|
rc = connect(nc->sock, &sa->sa, sizeof(sa->sin));
|
2015-11-18 04:53:57 +08:00
|
|
|
nc->err = mg_is_error(rc) ? errno : 0;
|
2015-10-22 20:04:35 +08:00
|
|
|
DBG(("%p sock %d err %d", nc, nc->sock, nc->err));
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
void mg_if_connect_udp(struct mg_connection *nc) {
|
|
|
|
nc->sock = socket(AF_INET, SOCK_DGRAM, 0);
|
|
|
|
if (nc->sock < 0) {
|
|
|
|
nc->sock = INVALID_SOCKET;
|
|
|
|
nc->err = errno ? errno : 1;
|
|
|
|
return;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-10-22 20:04:35 +08:00
|
|
|
nc->err = 0;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
int mg_if_listen_tcp(struct mg_connection *nc, union socket_address *sa) {
|
|
|
|
sock_t sock = mg_open_listening_socket(sa, SOCK_STREAM);
|
|
|
|
if (sock < 0) return (errno ? errno : 1);
|
2015-10-27 04:31:45 +08:00
|
|
|
mg_sock_set(nc, sock);
|
2015-10-22 20:04:35 +08:00
|
|
|
return 0;
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
int mg_if_listen_udp(struct mg_connection *nc, union socket_address *sa) {
|
|
|
|
sock_t sock = mg_open_listening_socket(sa, SOCK_DGRAM);
|
|
|
|
if (sock < 0) return (errno ? errno : 1);
|
2015-10-27 04:31:45 +08:00
|
|
|
mg_sock_set(nc, sock);
|
2015-10-22 20:04:35 +08:00
|
|
|
return 0;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
void mg_if_tcp_send(struct mg_connection *nc, const void *buf, size_t len) {
|
|
|
|
mbuf_append(&nc->send_mbuf, buf, len);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
void mg_if_udp_send(struct mg_connection *nc, const void *buf, size_t len) {
|
|
|
|
DBG(("%p %d %d", nc, (int) len, (int) nc->send_mbuf.len));
|
|
|
|
mbuf_append(&nc->send_mbuf, buf, len);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
void mg_if_recved(struct mg_connection *nc, size_t len) {
|
2015-09-08 19:49:03 +08:00
|
|
|
(void) nc;
|
2015-10-22 20:04:35 +08:00
|
|
|
(void) len;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-12-09 17:43:41 +08:00
|
|
|
int mg_if_create_conn(struct mg_connection *nc) {
|
|
|
|
(void) nc;
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
void mg_if_destroy_conn(struct mg_connection *nc) {
|
|
|
|
if (nc->sock == INVALID_SOCKET) return;
|
|
|
|
if (!(nc->flags & MG_F_UDP)) {
|
|
|
|
closesocket(nc->sock);
|
|
|
|
} else {
|
|
|
|
/* Only close outgoing UDP sockets or listeners. */
|
|
|
|
if (nc->listener == NULL) closesocket(nc->sock);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-10-22 20:04:35 +08:00
|
|
|
/*
|
|
|
|
* avoid users accidentally double close a socket
|
|
|
|
* because it can lead to difficult to debug situations.
|
|
|
|
* It would happen only if reusing a destroyed mg_connection
|
|
|
|
* but it's not always possible to run the code through an
|
|
|
|
* address sanitizer.
|
|
|
|
*/
|
|
|
|
nc->sock = INVALID_SOCKET;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
static void mg_accept_conn(struct mg_connection *lc) {
|
2015-10-27 04:31:45 +08:00
|
|
|
struct mg_connection *nc;
|
2015-10-22 20:04:35 +08:00
|
|
|
union socket_address sa;
|
|
|
|
socklen_t sa_len = sizeof(sa);
|
|
|
|
/* NOTE(lsm): on Windows, sock is always > FD_SETSIZE */
|
|
|
|
sock_t sock = accept(lc->sock, &sa.sa, &sa_len);
|
|
|
|
if (sock < 0) {
|
|
|
|
DBG(("%p: failed to accept: %d", lc, errno));
|
|
|
|
return;
|
|
|
|
}
|
2015-10-27 04:31:45 +08:00
|
|
|
nc = mg_if_accept_tcp_cb(lc, &sa, sa_len);
|
|
|
|
if (nc == NULL) {
|
2015-10-24 07:05:05 +08:00
|
|
|
closesocket(sock);
|
2015-10-27 04:31:45 +08:00
|
|
|
return;
|
2015-10-24 07:05:05 +08:00
|
|
|
}
|
2015-10-27 04:31:45 +08:00
|
|
|
mg_sock_set(nc, sock);
|
|
|
|
#ifdef MG_ENABLE_SSL
|
|
|
|
if (lc->ssl_ctx != NULL) {
|
|
|
|
nc->ssl = SSL_new(lc->ssl_ctx);
|
|
|
|
if (nc->ssl == NULL || SSL_set_fd(nc->ssl, sock) != 1) {
|
|
|
|
DBG(("SSL error"));
|
|
|
|
mg_close_conn(nc);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|
2015-10-22 20:04:35 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
/* 'sa' must be an initialized address to bind to */
|
|
|
|
static sock_t mg_open_listening_socket(union socket_address *sa, int proto) {
|
|
|
|
socklen_t sa_len =
|
|
|
|
(sa->sa.sa_family == AF_INET) ? sizeof(sa->sin) : sizeof(sa->sin6);
|
|
|
|
sock_t sock = INVALID_SOCKET;
|
|
|
|
#if !defined(MG_CC3200) && !defined(MG_LWIP)
|
|
|
|
int on = 1;
|
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
if ((sock = socket(sa->sa.sa_family, proto, 0)) != INVALID_SOCKET &&
|
|
|
|
#if !defined(MG_CC3200) && \
|
|
|
|
!defined(MG_LWIP) /* CC3200 and LWIP don't support either */
|
|
|
|
#if defined(_WIN32) && defined(SO_EXCLUSIVEADDRUSE)
|
|
|
|
/* "Using SO_REUSEADDR and SO_EXCLUSIVEADDRUSE" http://goo.gl/RmrFTm */
|
|
|
|
!setsockopt(sock, SOL_SOCKET, SO_EXCLUSIVEADDRUSE, (void *) &on,
|
|
|
|
sizeof(on)) &&
|
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
#if !defined(_WIN32) || !defined(SO_EXCLUSIVEADDRUSE)
|
|
|
|
/*
|
|
|
|
* SO_RESUSEADDR is not enabled on Windows because the semantics of
|
|
|
|
* SO_REUSEADDR on UNIX and Windows is different. On Windows,
|
|
|
|
* 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.
|
|
|
|
*/
|
|
|
|
!setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (void *) &on, sizeof(on)) &&
|
|
|
|
#endif
|
|
|
|
#endif /* !MG_CC3200 && !MG_LWIP */
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
!bind(sock, &sa->sa, sa_len) &&
|
|
|
|
(proto == SOCK_DGRAM || listen(sock, SOMAXCONN) == 0)) {
|
|
|
|
#if !defined(MG_CC3200) && !defined(MG_LWIP) /* TODO(rojer): Fix this. */
|
|
|
|
mg_set_non_blocking_mode(sock);
|
|
|
|
/* In case port was set to 0, get the real port number */
|
|
|
|
(void) getsockname(sock, &sa->sa, &sa_len);
|
|
|
|
#endif
|
|
|
|
} else if (sock != INVALID_SOCKET) {
|
|
|
|
closesocket(sock);
|
|
|
|
sock = INVALID_SOCKET;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
return sock;
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
static void mg_write_to_socket(struct mg_connection *nc) {
|
|
|
|
struct mbuf *io = &nc->send_mbuf;
|
|
|
|
int n = 0;
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
#ifdef MG_LWIP
|
|
|
|
/* With LWIP we don't know if the socket is ready */
|
|
|
|
if (io->len == 0) return;
|
2015-09-11 23:38:21 +08:00
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
assert(io->len > 0);
|
|
|
|
|
|
|
|
if (nc->flags & MG_F_UDP) {
|
|
|
|
int n =
|
|
|
|
sendto(nc->sock, io->buf, io->len, 0, &nc->sa.sa, sizeof(nc->sa.sin));
|
|
|
|
DBG(("%p %d %d %d %s:%hu", nc, nc->sock, n, errno,
|
|
|
|
inet_ntoa(nc->sa.sin.sin_addr), ntohs(nc->sa.sin.sin_port)));
|
|
|
|
if (n > 0) {
|
|
|
|
mbuf_remove(io, n);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-10-22 20:04:35 +08:00
|
|
|
mg_if_sent_cb(nc, n);
|
|
|
|
return;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
#ifdef MG_ENABLE_SSL
|
|
|
|
if (nc->ssl != NULL) {
|
|
|
|
if (nc->flags & MG_F_SSL_HANDSHAKE_DONE) {
|
|
|
|
n = SSL_write(nc->ssl, io->buf, io->len);
|
|
|
|
if (n <= 0) {
|
|
|
|
int ssl_err = mg_ssl_err(nc, n);
|
|
|
|
if (ssl_err == SSL_ERROR_WANT_READ || ssl_err == SSL_ERROR_WANT_WRITE) {
|
|
|
|
return; /* Call us again */
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
/* Successful SSL operation, clear off SSL wait flags */
|
|
|
|
nc->flags &= ~(MG_F_WANT_READ | MG_F_WANT_WRITE);
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
mg_ssl_begin(nc);
|
|
|
|
return;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-10-22 20:04:35 +08:00
|
|
|
} else
|
|
|
|
#endif
|
|
|
|
{
|
|
|
|
n = (int) MG_SEND_FUNC(nc->sock, io->buf, io->len, 0);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
DBG(("%p %d bytes -> %d", nc, n, nc->sock));
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
if (n > 0) {
|
|
|
|
mbuf_remove(io, n);
|
|
|
|
}
|
|
|
|
mg_if_sent_cb(nc, n);
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
static void mg_read_from_socket(struct mg_connection *conn) {
|
|
|
|
int n = 0;
|
|
|
|
char *buf = (char *) MG_MALLOC(MG_TCP_RECV_BUFFER_SIZE);
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
if (buf == NULL) {
|
|
|
|
DBG(("OOM"));
|
|
|
|
return;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
#ifdef MG_ENABLE_SSL
|
|
|
|
if (conn->ssl != NULL) {
|
|
|
|
if (conn->flags & MG_F_SSL_HANDSHAKE_DONE) {
|
|
|
|
/* SSL library may have more bytes ready to read then we ask to read.
|
|
|
|
* Therefore, read in a loop until we read everything. Without the loop,
|
|
|
|
* we skip to the next select() cycle which can just timeout. */
|
|
|
|
while ((n = SSL_read(conn->ssl, buf, MG_TCP_RECV_BUFFER_SIZE)) > 0) {
|
|
|
|
DBG(("%p %d bytes <- %d (SSL)", conn, n, conn->sock));
|
|
|
|
mg_if_recv_tcp_cb(conn, buf, n);
|
|
|
|
buf = NULL;
|
|
|
|
if (conn->flags & MG_F_CLOSE_IMMEDIATELY) break;
|
|
|
|
/* buf has been freed, we need a new one. */
|
|
|
|
buf = (char *) MG_MALLOC(MG_TCP_RECV_BUFFER_SIZE);
|
|
|
|
if (buf == NULL) break;
|
|
|
|
}
|
|
|
|
MG_FREE(buf);
|
|
|
|
mg_ssl_err(conn, n);
|
|
|
|
} else {
|
|
|
|
MG_FREE(buf);
|
|
|
|
mg_ssl_begin(conn);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
} else
|
2015-09-08 19:49:03 +08:00
|
|
|
#endif
|
2015-10-22 20:04:35 +08:00
|
|
|
{
|
|
|
|
n = (int) MG_RECV_FUNC(conn->sock, buf,
|
|
|
|
recv_avail_size(conn, MG_TCP_RECV_BUFFER_SIZE), 0);
|
|
|
|
if (n > 0) {
|
|
|
|
DBG(("%p %d bytes (PLAIN) <- %d", conn, n, conn->sock));
|
|
|
|
mg_if_recv_tcp_cb(conn, buf, n);
|
|
|
|
} else {
|
|
|
|
MG_FREE(buf);
|
|
|
|
}
|
|
|
|
if (mg_is_error(n)) {
|
|
|
|
conn->flags |= MG_F_CLOSE_IMMEDIATELY;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
}
|
2015-10-22 20:04:35 +08:00
|
|
|
}
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
static int mg_recvfrom(struct mg_connection *nc, union socket_address *sa,
|
|
|
|
socklen_t *sa_len, char **buf) {
|
|
|
|
int n;
|
|
|
|
*buf = (char *) MG_MALLOC(MG_UDP_RECV_BUFFER_SIZE);
|
|
|
|
if (*buf == NULL) {
|
|
|
|
DBG(("Out of memory"));
|
|
|
|
return -ENOMEM;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-10-22 20:04:35 +08:00
|
|
|
n = recvfrom(nc->sock, *buf, MG_UDP_RECV_BUFFER_SIZE, 0, &sa->sa, sa_len);
|
|
|
|
if (n <= 0) {
|
|
|
|
DBG(("%p recvfrom: %s", nc, strerror(errno)));
|
|
|
|
MG_FREE(*buf);
|
|
|
|
}
|
|
|
|
return n;
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
static void mg_handle_udp_read(struct mg_connection *nc) {
|
|
|
|
char *buf = NULL;
|
|
|
|
union socket_address sa;
|
|
|
|
socklen_t sa_len = sizeof(sa);
|
|
|
|
int n = mg_recvfrom(nc, &sa, &sa_len, &buf);
|
|
|
|
DBG(("%p %d bytes from %s:%d", nc, n, inet_ntoa(nc->sa.sin.sin_addr),
|
|
|
|
ntohs(nc->sa.sin.sin_port)));
|
|
|
|
mg_if_recv_udp_cb(nc, buf, n, &sa, sa_len);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-12-07 23:56:13 +08:00
|
|
|
#ifdef MG_ENABLE_SSL
|
|
|
|
static int mg_ssl_err(struct mg_connection *conn, int res) {
|
|
|
|
int ssl_err = SSL_get_error(conn->ssl, res);
|
2015-12-11 17:11:44 +08:00
|
|
|
DBG(("%p %d -> %d", conn, res, ssl_err));
|
|
|
|
if (ssl_err == SSL_ERROR_WANT_READ) {
|
|
|
|
conn->flags |= MG_F_WANT_READ;
|
|
|
|
} else if (ssl_err == SSL_ERROR_WANT_WRITE) {
|
|
|
|
conn->flags |= MG_F_WANT_WRITE;
|
|
|
|
} else {
|
|
|
|
/* There could be an alert to deliver. Try our best. */
|
|
|
|
SSL_write(conn->ssl, "", 0);
|
|
|
|
conn->flags |= MG_F_CLOSE_IMMEDIATELY;
|
|
|
|
}
|
2015-12-07 23:56:13 +08:00
|
|
|
return ssl_err;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void mg_ssl_begin(struct mg_connection *nc) {
|
|
|
|
int server_side = nc->listener != NULL;
|
|
|
|
int res = server_side ? SSL_accept(nc->ssl) : SSL_connect(nc->ssl);
|
|
|
|
DBG(("%p %d res %d %d %d", nc, server_side, res, errno, mg_ssl_err(nc, res)));
|
|
|
|
|
|
|
|
if (res == 1) {
|
|
|
|
nc->flags |= MG_F_SSL_HANDSHAKE_DONE;
|
|
|
|
nc->flags &= ~(MG_F_WANT_READ | MG_F_WANT_WRITE);
|
|
|
|
|
|
|
|
if (server_side) {
|
|
|
|
union socket_address sa;
|
|
|
|
socklen_t sa_len = sizeof(sa);
|
|
|
|
/* In case port was set to 0, get the real port number */
|
|
|
|
(void) getsockname(nc->sock, &sa.sa, &sa_len);
|
|
|
|
mg_call(nc, NULL, MG_EV_ACCEPT, &sa);
|
|
|
|
} else {
|
|
|
|
mg_if_connect_cb(nc, 0);
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
int ssl_err = mg_ssl_err(nc, res);
|
|
|
|
if (ssl_err != SSL_ERROR_WANT_READ && ssl_err != SSL_ERROR_WANT_WRITE) {
|
|
|
|
if (!server_side) {
|
|
|
|
mg_if_connect_cb(nc, ssl_err);
|
|
|
|
}
|
|
|
|
nc->flags |= MG_F_CLOSE_IMMEDIATELY;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
#endif /* MG_ENABLE_SSL */
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
#define _MG_F_FD_CAN_READ 1
|
|
|
|
#define _MG_F_FD_CAN_WRITE 1 << 1
|
|
|
|
#define _MG_F_FD_ERROR 1 << 2
|
|
|
|
|
2015-12-16 08:52:37 +08:00
|
|
|
void mg_mgr_handle_conn(struct mg_connection *nc, int fd_flags, double now) {
|
2015-10-22 20:04:35 +08:00
|
|
|
DBG(("%p fd=%d fd_flags=%d nc_flags=%lu rmbl=%d smbl=%d", nc, nc->sock,
|
|
|
|
fd_flags, nc->flags, (int) nc->recv_mbuf.len, (int) nc->send_mbuf.len));
|
|
|
|
|
|
|
|
if (nc->flags & MG_F_CONNECTING) {
|
|
|
|
if (fd_flags != 0) {
|
|
|
|
int err = 0;
|
|
|
|
#if !defined(MG_CC3200) && !defined(MG_ESP8266)
|
|
|
|
if (!(nc->flags & MG_F_UDP)) {
|
|
|
|
socklen_t len = sizeof(err);
|
|
|
|
int ret =
|
|
|
|
getsockopt(nc->sock, SOL_SOCKET, SO_ERROR, (char *) &err, &len);
|
|
|
|
if (ret != 0) err = 1;
|
|
|
|
}
|
|
|
|
#else
|
|
|
|
/* On CC3200 and ESP8266 we use blocking connect. If we got as far as
|
|
|
|
* this, it means connect() was successful.
|
|
|
|
* TODO(rojer): Figure out why it fails where blocking succeeds.
|
2015-09-08 19:49:03 +08:00
|
|
|
*/
|
2015-10-22 20:04:35 +08:00
|
|
|
#endif
|
2015-12-07 23:56:13 +08:00
|
|
|
#ifdef MG_ENABLE_SSL
|
|
|
|
if (nc->ssl != NULL && err == 0) {
|
|
|
|
SSL_set_fd(nc->ssl, nc->sock);
|
|
|
|
mg_ssl_begin(nc);
|
|
|
|
} else {
|
|
|
|
mg_if_connect_cb(nc, err);
|
|
|
|
}
|
|
|
|
#else
|
2015-10-22 20:04:35 +08:00
|
|
|
mg_if_connect_cb(nc, err);
|
2015-12-07 23:56:13 +08:00
|
|
|
#endif
|
2015-10-22 20:04:35 +08:00
|
|
|
} else if (nc->err != 0) {
|
|
|
|
mg_if_connect_cb(nc, nc->err);
|
|
|
|
}
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
if (fd_flags & _MG_F_FD_CAN_READ) {
|
|
|
|
if (nc->flags & MG_F_UDP) {
|
|
|
|
mg_handle_udp_read(nc);
|
|
|
|
} else {
|
|
|
|
if (nc->flags & MG_F_LISTENING) {
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
2015-10-22 20:04:35 +08:00
|
|
|
* We're not looping here, and accepting just one connection at
|
|
|
|
* a time. The reason is that eCos does not respect non-blocking
|
|
|
|
* flag on a listening socket and hangs in a loop.
|
2015-09-08 19:49:03 +08:00
|
|
|
*/
|
2015-10-22 20:04:35 +08:00
|
|
|
if (fd_flags & _MG_F_FD_CAN_READ) mg_accept_conn(nc);
|
2015-09-08 19:49:03 +08:00
|
|
|
return;
|
2015-10-22 20:04:35 +08:00
|
|
|
} else {
|
|
|
|
mg_read_from_socket(nc);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
}
|
2015-10-22 20:04:35 +08:00
|
|
|
if (nc->flags & MG_F_CLOSE_IMMEDIATELY) return;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-11-16 04:34:40 +08:00
|
|
|
if ((fd_flags & _MG_F_FD_CAN_WRITE) && nc->send_mbuf.len > 0) {
|
2015-10-22 20:04:35 +08:00
|
|
|
mg_write_to_socket(nc);
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!(fd_flags & (_MG_F_FD_CAN_READ | _MG_F_FD_CAN_WRITE))) {
|
2015-10-26 16:15:58 +08:00
|
|
|
mg_if_poll(nc, now);
|
2015-10-22 20:04:35 +08:00
|
|
|
}
|
2015-12-08 01:55:28 +08:00
|
|
|
mg_if_timer(nc, now);
|
2015-10-22 20:04:35 +08:00
|
|
|
|
|
|
|
DBG(("%p after fd=%d nc_flags=%lu rmbl=%d smbl=%d", nc, nc->sock, nc->flags,
|
|
|
|
(int) nc->recv_mbuf.len, (int) nc->send_mbuf.len));
|
|
|
|
}
|
|
|
|
|
|
|
|
#ifndef MG_DISABLE_SOCKETPAIR
|
|
|
|
static void mg_mgr_handle_ctl_sock(struct mg_mgr *mgr) {
|
|
|
|
struct ctl_msg ctl_msg;
|
|
|
|
int len =
|
|
|
|
(int) MG_RECV_FUNC(mgr->ctl[1], (char *) &ctl_msg, sizeof(ctl_msg), 0);
|
|
|
|
size_t dummy = MG_SEND_FUNC(mgr->ctl[1], ctl_msg.message, 1, 0);
|
2015-12-08 21:34:32 +08:00
|
|
|
DBG(("read %d from ctl socket", len));
|
2015-10-22 20:04:35 +08:00
|
|
|
(void) dummy; /* https://gcc.gnu.org/bugzilla/show_bug.cgi?id=25509 */
|
|
|
|
if (len >= (int) sizeof(ctl_msg.callback) && ctl_msg.callback != NULL) {
|
|
|
|
struct mg_connection *nc;
|
|
|
|
for (nc = mg_next(mgr, NULL); nc != NULL; nc = mg_next(mgr, nc)) {
|
|
|
|
ctl_msg.callback(nc, MG_EV_POLL, ctl_msg.message);
|
|
|
|
}
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
struct mg_connection *mg_add_sock(struct mg_mgr *s, sock_t sock,
|
|
|
|
mg_event_handler_t callback) {
|
2015-10-24 07:31:10 +08:00
|
|
|
struct mg_add_sock_opts opts;
|
|
|
|
memset(&opts, 0, sizeof(opts));
|
2015-10-22 20:04:35 +08:00
|
|
|
return mg_add_sock_opt(s, sock, callback, opts);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
struct mg_connection *mg_add_sock_opt(struct mg_mgr *s, sock_t sock,
|
|
|
|
mg_event_handler_t callback,
|
|
|
|
struct mg_add_sock_opts opts) {
|
|
|
|
struct mg_connection *nc = mg_create_connection(s, callback, opts);
|
|
|
|
if (nc != NULL) {
|
2015-10-27 04:31:45 +08:00
|
|
|
mg_sock_set(nc, sock);
|
2015-10-22 20:04:35 +08:00
|
|
|
mg_add_conn(nc->mgr, nc);
|
|
|
|
}
|
|
|
|
return nc;
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
/* Associate a socket to a connection. */
|
2015-10-27 04:31:45 +08:00
|
|
|
static void mg_sock_set(struct mg_connection *nc, sock_t sock) {
|
2015-10-22 20:04:35 +08:00
|
|
|
mg_set_non_blocking_mode(sock);
|
|
|
|
mg_set_close_on_exec(sock);
|
|
|
|
nc->sock = sock;
|
|
|
|
DBG(("%p %d", nc, sock));
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
#if MG_MGR_EV_MGR == 1 /* epoll() */
|
|
|
|
|
|
|
|
#ifndef MG_EPOLL_MAX_EVENTS
|
|
|
|
#define MG_EPOLL_MAX_EVENTS 100
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#define _MG_EPF_EV_EPOLLIN (1 << 0)
|
|
|
|
#define _MG_EPF_EV_EPOLLOUT (1 << 1)
|
|
|
|
#define _MG_EPF_NO_POLL (1 << 2)
|
|
|
|
|
2015-10-24 07:05:05 +08:00
|
|
|
uint32_t mg_epf_to_evflags(unsigned int epf) {
|
2015-10-22 20:04:35 +08:00
|
|
|
uint32_t result = 0;
|
|
|
|
if (epf & _MG_EPF_EV_EPOLLIN) result |= EPOLLIN;
|
|
|
|
if (epf & _MG_EPF_EV_EPOLLOUT) result |= EPOLLOUT;
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
2015-10-24 07:05:05 +08:00
|
|
|
void mg_ev_mgr_epoll_set_flags(const struct mg_connection *nc,
|
|
|
|
struct epoll_event *ev) {
|
2015-10-22 20:04:35 +08:00
|
|
|
/* NOTE: EPOLLERR and EPOLLHUP are always enabled. */
|
|
|
|
ev->events = 0;
|
|
|
|
if ((nc->flags & MG_F_LISTENING) || nc->recv_mbuf.len < nc->recv_mbuf_limit) {
|
|
|
|
ev->events |= EPOLLIN;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-11-16 04:34:40 +08:00
|
|
|
if ((nc->flags & MG_F_CONNECTING) || (nc->send_mbuf.len > 0)) {
|
2015-10-22 20:04:35 +08:00
|
|
|
ev->events |= EPOLLOUT;
|
|
|
|
}
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-24 07:05:05 +08:00
|
|
|
void mg_ev_mgr_epoll_ctl(struct mg_connection *nc, int op) {
|
2015-10-22 20:04:35 +08:00
|
|
|
int epoll_fd = (intptr_t) nc->mgr->mgr_data;
|
|
|
|
struct epoll_event ev;
|
|
|
|
assert(op == EPOLL_CTL_ADD || op == EPOLL_CTL_MOD || EPOLL_CTL_DEL);
|
|
|
|
DBG(("%p %d %d", nc, nc->sock, op));
|
|
|
|
if (nc->sock == INVALID_SOCKET) return;
|
|
|
|
if (op != EPOLL_CTL_DEL) {
|
|
|
|
mg_ev_mgr_epoll_set_flags(nc, &ev);
|
|
|
|
if (op == EPOLL_CTL_MOD) {
|
|
|
|
uint32_t old_ev_flags = mg_epf_to_evflags((intptr_t) nc->mgr_data);
|
|
|
|
if (ev.events == old_ev_flags) return;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-10-22 20:04:35 +08:00
|
|
|
ev.data.ptr = nc;
|
|
|
|
}
|
|
|
|
if (epoll_ctl(epoll_fd, op, nc->sock, &ev) != 0) {
|
|
|
|
perror("epoll_ctl");
|
|
|
|
abort();
|
|
|
|
}
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-24 07:05:05 +08:00
|
|
|
void mg_ev_mgr_init(struct mg_mgr *mgr) {
|
2015-10-22 20:04:35 +08:00
|
|
|
int epoll_fd;
|
|
|
|
DBG(("%p using epoll()", mgr));
|
|
|
|
#ifndef MG_DISABLE_SOCKETPAIR
|
|
|
|
do {
|
|
|
|
mg_socketpair(mgr->ctl, SOCK_DGRAM);
|
|
|
|
} while (mgr->ctl[0] == INVALID_SOCKET);
|
2015-09-08 19:49:03 +08:00
|
|
|
#endif
|
2015-10-22 20:04:35 +08:00
|
|
|
epoll_fd = epoll_create(MG_EPOLL_MAX_EVENTS /* unused but required */);
|
|
|
|
if (epoll_fd < 0) {
|
|
|
|
perror("epoll_ctl");
|
|
|
|
abort();
|
|
|
|
}
|
|
|
|
mgr->mgr_data = (void *) ((intptr_t) epoll_fd);
|
|
|
|
if (mgr->ctl[1] != INVALID_SOCKET) {
|
|
|
|
struct epoll_event ev;
|
|
|
|
ev.events = EPOLLIN;
|
|
|
|
ev.data.ptr = NULL;
|
|
|
|
if (epoll_ctl(epoll_fd, EPOLL_CTL_ADD, mgr->ctl[1], &ev) != 0) {
|
|
|
|
perror("epoll_ctl");
|
|
|
|
abort();
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-10-24 07:05:05 +08:00
|
|
|
void mg_ev_mgr_free(struct mg_mgr *mgr) {
|
2015-10-22 20:04:35 +08:00
|
|
|
int epoll_fd = (intptr_t) mgr->mgr_data;
|
|
|
|
close(epoll_fd);
|
|
|
|
}
|
|
|
|
|
2015-10-24 07:05:05 +08:00
|
|
|
void mg_ev_mgr_add_conn(struct mg_connection *nc) {
|
2015-10-22 20:04:35 +08:00
|
|
|
if (!(nc->flags & MG_F_UDP) || nc->listener == NULL) {
|
|
|
|
mg_ev_mgr_epoll_ctl(nc, EPOLL_CTL_ADD);
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-10-24 07:05:05 +08:00
|
|
|
void mg_ev_mgr_remove_conn(struct mg_connection *nc) {
|
2015-10-22 20:04:35 +08:00
|
|
|
if (!(nc->flags & MG_F_UDP) || nc->listener == NULL) {
|
|
|
|
mg_ev_mgr_epoll_ctl(nc, EPOLL_CTL_DEL);
|
|
|
|
}
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
time_t mg_mgr_poll(struct mg_mgr *mgr, int timeout_ms) {
|
|
|
|
int epoll_fd = (intptr_t) mgr->mgr_data;
|
|
|
|
struct epoll_event events[MG_EPOLL_MAX_EVENTS];
|
|
|
|
struct mg_connection *nc, *next;
|
|
|
|
int num_ev, fd_flags;
|
2015-12-16 08:52:37 +08:00
|
|
|
double now;
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
num_ev = epoll_wait(epoll_fd, events, MG_EPOLL_MAX_EVENTS, timeout_ms);
|
2016-01-08 21:14:55 +08:00
|
|
|
now = mg_time();
|
2015-10-22 20:04:35 +08:00
|
|
|
DBG(("epoll_wait @ %ld num_ev=%d", (long) now, num_ev));
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
while (num_ev-- > 0) {
|
|
|
|
intptr_t epf;
|
|
|
|
struct epoll_event *ev = events + num_ev;
|
|
|
|
nc = (struct mg_connection *) ev->data.ptr;
|
|
|
|
if (nc == NULL) {
|
|
|
|
mg_mgr_handle_ctl_sock(mgr);
|
|
|
|
continue;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-10-22 20:04:35 +08:00
|
|
|
fd_flags = ((ev->events & (EPOLLIN | EPOLLHUP)) ? _MG_F_FD_CAN_READ : 0) |
|
|
|
|
((ev->events & (EPOLLOUT)) ? _MG_F_FD_CAN_WRITE : 0) |
|
|
|
|
((ev->events & (EPOLLERR)) ? _MG_F_FD_ERROR : 0);
|
|
|
|
mg_mgr_handle_conn(nc, fd_flags, now);
|
|
|
|
epf = (intptr_t) nc->mgr_data;
|
|
|
|
epf ^= _MG_EPF_NO_POLL;
|
|
|
|
nc->mgr_data = (void *) epf;
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
for (nc = mgr->active_connections; nc != NULL; nc = next) {
|
|
|
|
next = nc->next;
|
|
|
|
if (!(((intptr_t) nc->mgr_data) & _MG_EPF_NO_POLL)) {
|
|
|
|
mg_mgr_handle_conn(nc, 0, now);
|
|
|
|
} else {
|
|
|
|
intptr_t epf = (intptr_t) nc->mgr_data;
|
|
|
|
epf ^= _MG_EPF_NO_POLL;
|
|
|
|
nc->mgr_data = (void *) epf;
|
|
|
|
}
|
|
|
|
if ((nc->flags & MG_F_CLOSE_IMMEDIATELY) ||
|
|
|
|
(nc->send_mbuf.len == 0 && (nc->flags & MG_F_SEND_AND_CLOSE))) {
|
|
|
|
mg_close_conn(nc);
|
|
|
|
} else {
|
|
|
|
if (!(nc->flags & MG_F_UDP) || nc->listener == NULL) {
|
|
|
|
mg_ev_mgr_epoll_ctl(nc, EPOLL_CTL_MOD);
|
|
|
|
} else {
|
|
|
|
/* This is a kludge, but... */
|
|
|
|
if (nc->send_mbuf.len > 0) {
|
|
|
|
mg_mgr_handle_conn(nc, _MG_F_FD_CAN_WRITE, now);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
return now;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
#else /* select() */
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-24 07:05:05 +08:00
|
|
|
void mg_ev_mgr_init(struct mg_mgr *mgr) {
|
2015-10-22 20:04:35 +08:00
|
|
|
(void) mgr;
|
|
|
|
DBG(("%p using select()", mgr));
|
|
|
|
#ifndef MG_DISABLE_SOCKETPAIR
|
|
|
|
do {
|
|
|
|
mg_socketpair(mgr->ctl, SOCK_DGRAM);
|
|
|
|
} while (mgr->ctl[0] == INVALID_SOCKET);
|
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-10-24 07:05:05 +08:00
|
|
|
void mg_ev_mgr_free(struct mg_mgr *mgr) {
|
2015-10-22 20:04:35 +08:00
|
|
|
(void) mgr;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-10-24 07:05:05 +08:00
|
|
|
void mg_ev_mgr_add_conn(struct mg_connection *nc) {
|
2015-10-22 20:04:35 +08:00
|
|
|
(void) nc;
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-24 07:05:05 +08:00
|
|
|
void mg_ev_mgr_remove_conn(struct mg_connection *nc) {
|
2015-10-22 20:04:35 +08:00
|
|
|
(void) nc;
|
|
|
|
}
|
2015-10-22 04:40:58 +08:00
|
|
|
|
2015-10-24 07:05:05 +08:00
|
|
|
void mg_add_to_set(sock_t sock, fd_set *set, sock_t *max_fd) {
|
2015-10-22 20:04:35 +08:00
|
|
|
if (sock != INVALID_SOCKET) {
|
|
|
|
FD_SET(sock, set);
|
|
|
|
if (*max_fd == INVALID_SOCKET || sock > *max_fd) {
|
|
|
|
*max_fd = sock;
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
time_t mg_mgr_poll(struct mg_mgr *mgr, int milli) {
|
2016-01-08 21:14:55 +08:00
|
|
|
double now = mg_time();
|
2015-10-22 20:04:35 +08:00
|
|
|
struct mg_connection *nc, *tmp;
|
|
|
|
struct timeval tv;
|
|
|
|
fd_set read_set, write_set, err_set;
|
|
|
|
sock_t max_fd = INVALID_SOCKET;
|
|
|
|
int num_fds, num_selected;
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
FD_ZERO(&read_set);
|
|
|
|
FD_ZERO(&write_set);
|
|
|
|
FD_ZERO(&err_set);
|
2015-12-08 20:13:03 +08:00
|
|
|
#ifndef MG_DISABLE_SOCKETPAIR
|
2015-10-22 20:04:35 +08:00
|
|
|
mg_add_to_set(mgr->ctl[1], &read_set, &max_fd);
|
2015-12-08 20:13:03 +08:00
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
for (nc = mgr->active_connections, num_fds = 0; nc != NULL; nc = tmp) {
|
|
|
|
tmp = nc->next;
|
|
|
|
|
|
|
|
if (nc->sock == INVALID_SOCKET) {
|
|
|
|
mg_mgr_handle_conn(nc, 0, now);
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
num_fds++;
|
|
|
|
|
|
|
|
if (!(nc->flags & MG_F_WANT_WRITE) &&
|
|
|
|
nc->recv_mbuf.len < nc->recv_mbuf_limit &&
|
|
|
|
(!(nc->flags & MG_F_UDP) || nc->listener == NULL)) {
|
|
|
|
mg_add_to_set(nc->sock, &read_set, &max_fd);
|
|
|
|
}
|
|
|
|
|
|
|
|
if (((nc->flags & MG_F_CONNECTING) && !(nc->flags & MG_F_WANT_READ)) ||
|
2015-11-16 04:34:40 +08:00
|
|
|
(nc->send_mbuf.len > 0 && !(nc->flags & MG_F_CONNECTING))) {
|
2015-10-22 20:04:35 +08:00
|
|
|
mg_add_to_set(nc->sock, &write_set, &max_fd);
|
|
|
|
mg_add_to_set(nc->sock, &err_set, &max_fd);
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
tv.tv_sec = milli / 1000;
|
|
|
|
tv.tv_usec = (milli % 1000) * 1000;
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
num_selected = select((int) max_fd + 1, &read_set, &write_set, &err_set, &tv);
|
2016-01-08 21:14:55 +08:00
|
|
|
now = mg_time();
|
2015-10-22 20:04:35 +08:00
|
|
|
DBG(("select @ %ld num_ev=%d of %d", (long) now, num_selected, num_fds));
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
#ifndef MG_DISABLE_SOCKETPAIR
|
|
|
|
if (num_selected > 0 && mgr->ctl[1] != INVALID_SOCKET &&
|
|
|
|
FD_ISSET(mgr->ctl[1], &read_set)) {
|
|
|
|
mg_mgr_handle_ctl_sock(mgr);
|
|
|
|
}
|
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
for (nc = mgr->active_connections; nc != NULL; nc = tmp) {
|
|
|
|
int fd_flags = 0;
|
|
|
|
if (num_selected > 0) {
|
|
|
|
fd_flags = (FD_ISSET(nc->sock, &read_set) ? _MG_F_FD_CAN_READ : 0) |
|
|
|
|
(FD_ISSET(nc->sock, &write_set) ? _MG_F_FD_CAN_WRITE : 0) |
|
|
|
|
(FD_ISSET(nc->sock, &err_set) ? _MG_F_FD_ERROR : 0);
|
|
|
|
}
|
|
|
|
#ifdef MG_CC3200
|
|
|
|
// CC3200 does not report UDP sockets as writeable.
|
|
|
|
if (nc->flags & MG_F_UDP &&
|
|
|
|
(nc->send_mbuf.len > 0 || nc->flags & MG_F_CONNECTING)) {
|
|
|
|
fd_flags |= _MG_F_FD_CAN_WRITE;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-10-22 20:04:35 +08:00
|
|
|
#endif
|
|
|
|
#ifdef MG_LWIP
|
|
|
|
/* With LWIP socket emulation layer, we don't get write events */
|
|
|
|
fd_flags |= _MG_F_FD_CAN_WRITE;
|
|
|
|
#endif
|
|
|
|
tmp = nc->next;
|
|
|
|
mg_mgr_handle_conn(nc, fd_flags, now);
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
for (nc = mgr->active_connections; nc != NULL; nc = tmp) {
|
|
|
|
tmp = nc->next;
|
|
|
|
if ((nc->flags & MG_F_CLOSE_IMMEDIATELY) ||
|
|
|
|
(nc->send_mbuf.len == 0 && (nc->flags & MG_F_SEND_AND_CLOSE))) {
|
|
|
|
mg_close_conn(nc);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
return now;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
#endif
|
|
|
|
|
|
|
|
#ifndef MG_DISABLE_SOCKETPAIR
|
|
|
|
int mg_socketpair(sock_t sp[2], int sock_type) {
|
|
|
|
union socket_address sa;
|
|
|
|
sock_t sock;
|
|
|
|
socklen_t len = sizeof(sa.sin);
|
|
|
|
int ret = 0;
|
|
|
|
|
|
|
|
sock = sp[0] = sp[1] = INVALID_SOCKET;
|
|
|
|
|
|
|
|
(void) memset(&sa, 0, sizeof(sa));
|
|
|
|
sa.sin.sin_family = AF_INET;
|
|
|
|
sa.sin.sin_port = htons(0);
|
|
|
|
sa.sin.sin_addr.s_addr = htonl(0x7f000001); /* 127.0.0.1 */
|
|
|
|
|
|
|
|
if ((sock = socket(AF_INET, sock_type, 0)) == INVALID_SOCKET) {
|
|
|
|
} else if (bind(sock, &sa.sa, len) != 0) {
|
|
|
|
} else if (sock_type == SOCK_STREAM && listen(sock, 1) != 0) {
|
|
|
|
} else if (getsockname(sock, &sa.sa, &len) != 0) {
|
|
|
|
} else if ((sp[0] = socket(AF_INET, sock_type, 0)) == INVALID_SOCKET) {
|
|
|
|
} else if (connect(sp[0], &sa.sa, len) != 0) {
|
|
|
|
} else if (sock_type == SOCK_DGRAM &&
|
|
|
|
(getsockname(sp[0], &sa.sa, &len) != 0 ||
|
|
|
|
connect(sock, &sa.sa, len) != 0)) {
|
|
|
|
} else if ((sp[1] = (sock_type == SOCK_DGRAM ? sock
|
|
|
|
: accept(sock, &sa.sa, &len))) ==
|
|
|
|
INVALID_SOCKET) {
|
|
|
|
} else {
|
|
|
|
mg_set_close_on_exec(sp[0]);
|
|
|
|
mg_set_close_on_exec(sp[1]);
|
|
|
|
if (sock_type == SOCK_STREAM) closesocket(sock);
|
|
|
|
ret = 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!ret) {
|
|
|
|
if (sp[0] != INVALID_SOCKET) closesocket(sp[0]);
|
|
|
|
if (sp[1] != INVALID_SOCKET) closesocket(sp[1]);
|
|
|
|
if (sock != INVALID_SOCKET) closesocket(sock);
|
|
|
|
sock = sp[0] = sp[1] = INVALID_SOCKET;
|
|
|
|
}
|
|
|
|
|
|
|
|
return ret;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-10-22 20:04:35 +08:00
|
|
|
#endif /* MG_DISABLE_SOCKETPAIR */
|
2015-10-24 07:05:05 +08:00
|
|
|
|
2015-11-27 18:32:36 +08:00
|
|
|
static void mg_sock_get_addr(sock_t sock, int remote,
|
|
|
|
union socket_address *sa) {
|
2015-10-24 07:05:05 +08:00
|
|
|
#ifndef MG_CC3200
|
|
|
|
socklen_t slen = sizeof(sa);
|
2015-11-27 18:32:36 +08:00
|
|
|
memset(sa, 0, sizeof(*sa));
|
|
|
|
if (remote) {
|
|
|
|
getpeername(sock, &sa->sa, &slen);
|
2015-10-24 07:05:05 +08:00
|
|
|
} else {
|
2015-11-27 18:32:36 +08:00
|
|
|
getsockname(sock, &sa->sa, &slen);
|
2015-10-24 07:05:05 +08:00
|
|
|
}
|
2015-11-27 18:32:36 +08:00
|
|
|
#else
|
|
|
|
memset(sa, 0, sizeof(*sa));
|
2015-10-24 07:05:05 +08:00
|
|
|
#endif
|
2015-11-27 18:32:36 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
void mg_sock_to_str(sock_t sock, char *buf, size_t len, int flags) {
|
|
|
|
union socket_address sa;
|
|
|
|
mg_sock_get_addr(sock, flags & MG_SOCK_STRINGIFY_REMOTE, &sa);
|
2015-10-24 07:05:05 +08:00
|
|
|
mg_sock_addr_to_str(&sa, buf, len, flags);
|
|
|
|
}
|
|
|
|
|
2015-11-27 18:32:36 +08:00
|
|
|
void mg_if_get_conn_addr(struct mg_connection *nc, int remote,
|
|
|
|
union socket_address *sa) {
|
|
|
|
mg_sock_get_addr(nc->sock, remote, sa);
|
|
|
|
}
|
|
|
|
|
2015-10-24 07:05:05 +08:00
|
|
|
#endif /* !MG_DISABLE_SOCKET_IF */
|
2015-09-08 19:49:03 +08:00
|
|
|
#ifdef NS_MODULE_LINES
|
2015-12-15 02:45:18 +08:00
|
|
|
#line 1 "./src/multithreading.c"
|
2015-09-08 19:49:03 +08:00
|
|
|
/**/
|
|
|
|
#endif
|
|
|
|
/*
|
|
|
|
* Copyright (c) 2014 Cesanta Software Limited
|
|
|
|
* All rights reserved
|
|
|
|
*/
|
|
|
|
|
2015-12-12 22:05:54 +08:00
|
|
|
/* Amalgamated: #include "mongoose/src/internal.h" */
|
2015-12-12 22:16:48 +08:00
|
|
|
/* Amalgamated: #include "mongoose/src/util.h" */
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifdef MG_ENABLE_THREADS
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
static void multithreaded_ev_handler(struct mg_connection *c, int ev, void *p);
|
|
|
|
|
|
|
|
/*
|
|
|
|
* This thread function executes user event handler.
|
|
|
|
* It runs an event manager that has only one connection, until that
|
|
|
|
* connection is alive.
|
|
|
|
*/
|
|
|
|
static void *per_connection_thread_function(void *param) {
|
|
|
|
struct mg_connection *c = (struct mg_connection *) param;
|
|
|
|
struct mg_mgr m;
|
|
|
|
|
|
|
|
mg_mgr_init(&m, NULL);
|
|
|
|
mg_add_conn(&m, c);
|
|
|
|
while (m.active_connections != NULL) {
|
|
|
|
mg_mgr_poll(&m, 1000);
|
|
|
|
}
|
|
|
|
mg_mgr_free(&m);
|
|
|
|
|
|
|
|
return param;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void link_conns(struct mg_connection *c1, struct mg_connection *c2) {
|
|
|
|
c1->priv_2 = c2;
|
|
|
|
c2->priv_2 = c1;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void unlink_conns(struct mg_connection *c) {
|
|
|
|
struct mg_connection *peer = (struct mg_connection *) c->priv_2;
|
|
|
|
if (peer != NULL) {
|
2015-09-19 19:38:46 +08:00
|
|
|
peer->flags |= MG_F_SEND_AND_CLOSE;
|
2015-09-08 19:49:03 +08:00
|
|
|
peer->priv_2 = NULL;
|
|
|
|
}
|
|
|
|
c->priv_2 = NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void forwarder_ev_handler(struct mg_connection *c, int ev, void *p) {
|
|
|
|
(void) p;
|
2015-09-19 19:38:46 +08:00
|
|
|
if (ev == MG_EV_RECV && c->priv_2) {
|
2015-09-08 19:49:03 +08:00
|
|
|
mg_forward(c, (struct mg_connection *) c->priv_2);
|
2015-09-19 19:38:46 +08:00
|
|
|
} else if (ev == MG_EV_CLOSE) {
|
2015-09-08 19:49:03 +08:00
|
|
|
unlink_conns(c);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static void spawn_handling_thread(struct mg_connection *nc) {
|
|
|
|
struct mg_mgr dummy;
|
|
|
|
sock_t sp[2];
|
|
|
|
struct mg_connection *c[2];
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Create a socket pair, and wrap each socket into the connection with
|
|
|
|
* dummy event manager.
|
|
|
|
* c[0] stays in this thread, c[1] goes to another thread.
|
|
|
|
*/
|
|
|
|
mg_socketpair(sp, SOCK_STREAM);
|
|
|
|
memset(&dummy, 0, sizeof(dummy));
|
|
|
|
c[0] = mg_add_sock(&dummy, sp[0], forwarder_ev_handler);
|
|
|
|
c[1] = mg_add_sock(&dummy, sp[1], (mg_event_handler_t) nc->listener->priv_1);
|
|
|
|
|
|
|
|
/* Interlink client connection with c[0] */
|
|
|
|
link_conns(c[0], nc);
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Switch c[0] manager from the dummy one to the real one. c[1] manager
|
|
|
|
* will be set in another thread, allocated on stack of that thread.
|
|
|
|
*/
|
|
|
|
mg_add_conn(nc->mgr, c[0]);
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Dress c[1] as nc.
|
|
|
|
* TODO(lsm): code in accept_conn() looks similar. Refactor.
|
|
|
|
*/
|
|
|
|
c[1]->listener = nc->listener;
|
|
|
|
c[1]->proto_handler = nc->proto_handler;
|
|
|
|
c[1]->proto_data = nc->proto_data;
|
|
|
|
c[1]->user_data = nc->user_data;
|
|
|
|
|
|
|
|
mg_start_thread(per_connection_thread_function, c[1]);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void multithreaded_ev_handler(struct mg_connection *c, int ev, void *p) {
|
|
|
|
(void) p;
|
2015-09-19 19:38:46 +08:00
|
|
|
if (ev == MG_EV_ACCEPT) {
|
2015-09-08 19:49:03 +08:00
|
|
|
spawn_handling_thread(c);
|
|
|
|
c->handler = forwarder_ev_handler;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void mg_enable_multithreading(struct mg_connection *nc) {
|
|
|
|
/* Wrap user event handler into our multithreaded_ev_handler */
|
|
|
|
nc->priv_1 = (void *) nc->handler;
|
|
|
|
nc->handler = multithreaded_ev_handler;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
#ifdef NS_MODULE_LINES
|
2015-12-15 02:45:18 +08:00
|
|
|
#line 1 "./src/http.c"
|
2015-09-08 19:49:03 +08:00
|
|
|
/**/
|
|
|
|
#endif
|
|
|
|
/*
|
|
|
|
* Copyright (c) 2014 Cesanta Software Limited
|
|
|
|
* All rights reserved
|
|
|
|
*/
|
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_DISABLE_HTTP
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-12-12 22:05:54 +08:00
|
|
|
/* Amalgamated: #include "mongoose/src/internal.h" */
|
2015-12-12 22:16:48 +08:00
|
|
|
/* Amalgamated: #include "mongoose/src/util.h" */
|
|
|
|
/* Amalgamated: #include "common/sha1.h" */
|
|
|
|
/* Amalgamated: #include "common/md5.h" */
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
enum http_proto_data_type { DATA_NONE, DATA_FILE, DATA_PUT, DATA_CGI };
|
|
|
|
|
|
|
|
struct proto_data_http {
|
|
|
|
FILE *fp; /* Opened file. */
|
|
|
|
int64_t cl; /* Content-Length. How many bytes to send. */
|
|
|
|
int64_t sent; /* How many bytes have been already sent. */
|
|
|
|
int64_t body_len; /* How many bytes of chunked body was reassembled. */
|
|
|
|
struct mg_connection *cgi_nc;
|
|
|
|
enum http_proto_data_type type;
|
|
|
|
};
|
|
|
|
|
|
|
|
/*
|
|
|
|
* This structure helps to create an environment for the spawned CGI program.
|
|
|
|
* Environment is an array of "VARIABLE=VALUE\0" ASCIIZ strings,
|
|
|
|
* last element must be NULL.
|
|
|
|
* However, on Windows there is a requirement that all these VARIABLE=VALUE\0
|
|
|
|
* strings must reside in a contiguous buffer. The end of the buffer is
|
|
|
|
* marked by two '\0' characters.
|
|
|
|
* We satisfy both worlds: we create an envp array (which is vars), all
|
|
|
|
* entries are actually pointers inside buf.
|
|
|
|
*/
|
|
|
|
struct cgi_env_block {
|
|
|
|
struct mg_connection *nc;
|
2015-09-19 19:38:46 +08:00
|
|
|
char buf[MG_CGI_ENVIRONMENT_SIZE]; /* Environment buffer */
|
|
|
|
const char *vars[MG_MAX_CGI_ENVIR_VARS]; /* char *envp[] */
|
2015-09-08 19:49:03 +08:00
|
|
|
int len; /* Space taken */
|
|
|
|
int nvars; /* Number of variables in envp[] */
|
|
|
|
};
|
|
|
|
|
|
|
|
#define MIME_ENTRY(_ext, _type) \
|
|
|
|
{ _ext, sizeof(_ext) - 1, _type }
|
|
|
|
static const struct {
|
|
|
|
const char *extension;
|
|
|
|
size_t ext_len;
|
|
|
|
const char *mime_type;
|
|
|
|
} static_builtin_mime_types[] = {
|
|
|
|
MIME_ENTRY("html", "text/html"),
|
|
|
|
MIME_ENTRY("html", "text/html"),
|
|
|
|
MIME_ENTRY("htm", "text/html"),
|
|
|
|
MIME_ENTRY("shtm", "text/html"),
|
|
|
|
MIME_ENTRY("shtml", "text/html"),
|
|
|
|
MIME_ENTRY("css", "text/css"),
|
|
|
|
MIME_ENTRY("js", "application/x-javascript"),
|
|
|
|
MIME_ENTRY("ico", "image/x-icon"),
|
|
|
|
MIME_ENTRY("gif", "image/gif"),
|
|
|
|
MIME_ENTRY("jpg", "image/jpeg"),
|
|
|
|
MIME_ENTRY("jpeg", "image/jpeg"),
|
|
|
|
MIME_ENTRY("png", "image/png"),
|
|
|
|
MIME_ENTRY("svg", "image/svg+xml"),
|
|
|
|
MIME_ENTRY("txt", "text/plain"),
|
|
|
|
MIME_ENTRY("torrent", "application/x-bittorrent"),
|
|
|
|
MIME_ENTRY("wav", "audio/x-wav"),
|
|
|
|
MIME_ENTRY("mp3", "audio/x-mp3"),
|
|
|
|
MIME_ENTRY("mid", "audio/mid"),
|
|
|
|
MIME_ENTRY("m3u", "audio/x-mpegurl"),
|
|
|
|
MIME_ENTRY("ogg", "application/ogg"),
|
|
|
|
MIME_ENTRY("ram", "audio/x-pn-realaudio"),
|
|
|
|
MIME_ENTRY("xml", "text/xml"),
|
|
|
|
MIME_ENTRY("ttf", "application/x-font-ttf"),
|
|
|
|
MIME_ENTRY("json", "application/json"),
|
|
|
|
MIME_ENTRY("xslt", "application/xml"),
|
|
|
|
MIME_ENTRY("xsl", "application/xml"),
|
|
|
|
MIME_ENTRY("ra", "audio/x-pn-realaudio"),
|
|
|
|
MIME_ENTRY("doc", "application/msword"),
|
|
|
|
MIME_ENTRY("exe", "application/octet-stream"),
|
|
|
|
MIME_ENTRY("zip", "application/x-zip-compressed"),
|
|
|
|
MIME_ENTRY("xls", "application/excel"),
|
|
|
|
MIME_ENTRY("tgz", "application/x-tar-gz"),
|
|
|
|
MIME_ENTRY("tar", "application/x-tar"),
|
|
|
|
MIME_ENTRY("gz", "application/x-gunzip"),
|
|
|
|
MIME_ENTRY("arj", "application/x-arj-compressed"),
|
|
|
|
MIME_ENTRY("rar", "application/x-rar-compressed"),
|
|
|
|
MIME_ENTRY("rtf", "application/rtf"),
|
|
|
|
MIME_ENTRY("pdf", "application/pdf"),
|
|
|
|
MIME_ENTRY("swf", "application/x-shockwave-flash"),
|
|
|
|
MIME_ENTRY("mpg", "video/mpeg"),
|
|
|
|
MIME_ENTRY("webm", "video/webm"),
|
|
|
|
MIME_ENTRY("mpeg", "video/mpeg"),
|
|
|
|
MIME_ENTRY("mov", "video/quicktime"),
|
|
|
|
MIME_ENTRY("mp4", "video/mp4"),
|
|
|
|
MIME_ENTRY("m4v", "video/x-m4v"),
|
|
|
|
MIME_ENTRY("asf", "video/x-ms-asf"),
|
|
|
|
MIME_ENTRY("avi", "video/x-msvideo"),
|
|
|
|
MIME_ENTRY("bmp", "image/bmp"),
|
|
|
|
{NULL, 0, NULL}};
|
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_DISABLE_FILESYSTEM
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-09-22 02:20:05 +08:00
|
|
|
#ifndef MG_DISABLE_DAV
|
2015-09-08 19:49:03 +08:00
|
|
|
static int mg_mkdir(const char *path, uint32_t mode) {
|
|
|
|
#ifndef _WIN32
|
|
|
|
return mkdir(path, mode);
|
|
|
|
#else
|
|
|
|
(void) mode;
|
|
|
|
return _mkdir(path);
|
|
|
|
#endif
|
|
|
|
}
|
2015-09-22 02:20:05 +08:00
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
static struct mg_str get_mime_type(const char *path, const char *dflt,
|
|
|
|
const struct mg_serve_http_opts *opts) {
|
|
|
|
const char *ext, *overrides;
|
|
|
|
size_t i, path_len;
|
|
|
|
struct mg_str r, k, v;
|
|
|
|
|
|
|
|
path_len = strlen(path);
|
|
|
|
|
|
|
|
overrides = opts->custom_mime_types;
|
|
|
|
while ((overrides = mg_next_comma_list_entry(overrides, &k, &v)) != NULL) {
|
|
|
|
ext = path + (path_len - k.len);
|
|
|
|
if (path_len > k.len && mg_vcasecmp(&k, ext) == 0) {
|
|
|
|
return v;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
for (i = 0; static_builtin_mime_types[i].extension != NULL; i++) {
|
|
|
|
ext = path + (path_len - static_builtin_mime_types[i].ext_len);
|
|
|
|
if (path_len > static_builtin_mime_types[i].ext_len && ext[-1] == '.' &&
|
|
|
|
mg_casecmp(ext, static_builtin_mime_types[i].extension) == 0) {
|
|
|
|
r.p = static_builtin_mime_types[i].mime_type;
|
|
|
|
r.len = strlen(r.p);
|
|
|
|
return r;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
r.p = dflt;
|
|
|
|
r.len = strlen(r.p);
|
|
|
|
return r;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Check whether full request is buffered. Return:
|
|
|
|
* -1 if request is malformed
|
|
|
|
* 0 if request is not yet fully buffered
|
|
|
|
* >0 actual request length, including last \r\n\r\n
|
|
|
|
*/
|
|
|
|
static int get_request_len(const char *s, int buf_len) {
|
|
|
|
const unsigned char *buf = (unsigned char *) s;
|
|
|
|
int i;
|
|
|
|
|
|
|
|
for (i = 0; i < buf_len; i++) {
|
|
|
|
if (!isprint(buf[i]) && buf[i] != '\r' && buf[i] != '\n' && buf[i] < 128) {
|
|
|
|
return -1;
|
|
|
|
} else if (buf[i] == '\n' && i + 1 < buf_len && buf[i + 1] == '\n') {
|
|
|
|
return i + 2;
|
|
|
|
} else if (buf[i] == '\n' && i + 2 < buf_len && buf[i + 1] == '\r' &&
|
|
|
|
buf[i + 2] == '\n') {
|
|
|
|
return i + 3;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static const char *parse_http_headers(const char *s, const char *end, int len,
|
|
|
|
struct http_message *req) {
|
|
|
|
int i;
|
2015-10-20 17:31:23 +08:00
|
|
|
for (i = 0; i < (int) ARRAY_SIZE(req->header_names) - 1; i++) {
|
2015-09-08 19:49:03 +08:00
|
|
|
struct mg_str *k = &req->header_names[i], *v = &req->header_values[i];
|
|
|
|
|
|
|
|
s = mg_skip(s, end, ": ", k);
|
|
|
|
s = mg_skip(s, end, "\r\n", v);
|
|
|
|
|
|
|
|
while (v->len > 0 && v->p[v->len - 1] == ' ') {
|
|
|
|
v->len--; /* Trim trailing spaces in header value */
|
|
|
|
}
|
|
|
|
|
|
|
|
if (k->len == 0 || v->len == 0) {
|
|
|
|
k->p = v->p = NULL;
|
|
|
|
k->len = v->len = 0;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!mg_ncasecmp(k->p, "Content-Length", 14)) {
|
|
|
|
req->body.len = to64(v->p);
|
|
|
|
req->message.len = len + req->body.len;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return s;
|
|
|
|
}
|
|
|
|
|
|
|
|
int mg_parse_http(const char *s, int n, struct http_message *hm, int is_req) {
|
|
|
|
const char *end, *qs;
|
|
|
|
int len = get_request_len(s, n);
|
|
|
|
|
|
|
|
if (len <= 0) return len;
|
|
|
|
|
|
|
|
memset(hm, 0, sizeof(*hm));
|
|
|
|
hm->message.p = s;
|
|
|
|
hm->body.p = s + len;
|
|
|
|
hm->message.len = hm->body.len = (size_t) ~0;
|
|
|
|
end = s + len;
|
|
|
|
|
|
|
|
/* Request is fully buffered. Skip leading whitespaces. */
|
|
|
|
while (s < end && isspace(*(unsigned char *) s)) s++;
|
|
|
|
|
|
|
|
if (is_req) {
|
|
|
|
/* Parse request line: method, URI, proto */
|
|
|
|
s = mg_skip(s, end, " ", &hm->method);
|
|
|
|
s = mg_skip(s, end, " ", &hm->uri);
|
|
|
|
s = mg_skip(s, end, "\r\n", &hm->proto);
|
|
|
|
if (hm->uri.p <= hm->method.p || hm->proto.p <= hm->uri.p) return -1;
|
|
|
|
|
|
|
|
/* If URI contains '?' character, initialize query_string */
|
|
|
|
if ((qs = (char *) memchr(hm->uri.p, '?', hm->uri.len)) != NULL) {
|
|
|
|
hm->query_string.p = qs + 1;
|
|
|
|
hm->query_string.len = &hm->uri.p[hm->uri.len] - (qs + 1);
|
|
|
|
hm->uri.len = qs - hm->uri.p;
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
s = mg_skip(s, end, " ", &hm->proto);
|
|
|
|
if (end - s < 4 || s[3] != ' ') return -1;
|
|
|
|
hm->resp_code = atoi(s);
|
|
|
|
if (hm->resp_code < 100 || hm->resp_code >= 600) return -1;
|
|
|
|
s += 4;
|
|
|
|
s = mg_skip(s, end, "\r\n", &hm->resp_status_msg);
|
|
|
|
}
|
|
|
|
|
|
|
|
s = parse_http_headers(s, end, len, hm);
|
|
|
|
|
|
|
|
/*
|
|
|
|
* mg_parse_http() 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.
|
|
|
|
*/
|
|
|
|
if (hm->body.len == (size_t) ~0 && is_req &&
|
|
|
|
mg_vcasecmp(&hm->method, "PUT") != 0 &&
|
|
|
|
mg_vcasecmp(&hm->method, "POST") != 0) {
|
|
|
|
hm->body.len = 0;
|
|
|
|
hm->message.len = len;
|
|
|
|
}
|
|
|
|
|
|
|
|
return len;
|
|
|
|
}
|
|
|
|
|
|
|
|
struct mg_str *mg_get_http_header(struct http_message *hm, const char *name) {
|
|
|
|
size_t i, len = strlen(name);
|
|
|
|
|
2015-10-20 17:31:23 +08:00
|
|
|
for (i = 0; hm->header_names[i].len > 0; i++) {
|
2015-09-08 19:49:03 +08:00
|
|
|
struct mg_str *h = &hm->header_names[i], *v = &hm->header_values[i];
|
|
|
|
if (h->p != NULL && h->len == len && !mg_ncasecmp(h->p, name, len))
|
|
|
|
return v;
|
|
|
|
}
|
|
|
|
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_DISABLE_HTTP_WEBSOCKET
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
static int is_ws_fragment(unsigned char flags) {
|
|
|
|
return (flags & 0x80) == 0 || (flags & 0x0f) == 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int is_ws_first_fragment(unsigned char flags) {
|
|
|
|
return (flags & 0x80) == 0 && (flags & 0x0f) != 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void handle_incoming_websocket_frame(struct mg_connection *nc,
|
|
|
|
struct websocket_message *wsm) {
|
|
|
|
if (wsm->flags & 0x8) {
|
2015-10-29 06:12:55 +08:00
|
|
|
mg_call(nc, nc->handler, MG_EV_WEBSOCKET_CONTROL_FRAME, wsm);
|
2015-09-08 19:49:03 +08:00
|
|
|
} else {
|
2015-10-29 06:12:55 +08:00
|
|
|
mg_call(nc, nc->handler, MG_EV_WEBSOCKET_FRAME, wsm);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static int deliver_websocket_data(struct mg_connection *nc) {
|
|
|
|
/* Using unsigned char *, cause of integer arithmetic below */
|
|
|
|
uint64_t i, data_len = 0, frame_len = 0, buf_len = nc->recv_mbuf.len, len,
|
|
|
|
mask_len = 0, header_len = 0;
|
|
|
|
unsigned char *p = (unsigned char *) nc->recv_mbuf.buf, *buf = p,
|
|
|
|
*e = p + buf_len;
|
|
|
|
unsigned *sizep = (unsigned *) &p[1]; /* Size ptr for defragmented frames */
|
|
|
|
int ok, reass = buf_len > 0 && is_ws_fragment(p[0]) &&
|
2015-09-19 19:38:46 +08:00
|
|
|
!(nc->flags & MG_F_WEBSOCKET_NO_DEFRAG);
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
/* If that's a continuation frame that must be reassembled, handle it */
|
|
|
|
if (reass && !is_ws_first_fragment(p[0]) && buf_len >= 1 + sizeof(*sizep) &&
|
|
|
|
buf_len >= 1 + sizeof(*sizep) + *sizep) {
|
|
|
|
buf += 1 + sizeof(*sizep) + *sizep;
|
|
|
|
buf_len -= 1 + sizeof(*sizep) + *sizep;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (buf_len >= 2) {
|
|
|
|
len = buf[1] & 127;
|
|
|
|
mask_len = buf[1] & 128 ? 4 : 0;
|
|
|
|
if (len < 126 && buf_len >= mask_len) {
|
|
|
|
data_len = len;
|
|
|
|
header_len = 2 + mask_len;
|
|
|
|
} else if (len == 126 && buf_len >= 4 + mask_len) {
|
|
|
|
header_len = 4 + mask_len;
|
|
|
|
data_len = ntohs(*(uint16_t *) &buf[2]);
|
|
|
|
} else if (buf_len >= 10 + mask_len) {
|
|
|
|
header_len = 10 + mask_len;
|
|
|
|
data_len = (((uint64_t) ntohl(*(uint32_t *) &buf[2])) << 32) +
|
|
|
|
ntohl(*(uint32_t *) &buf[6]);
|
|
|
|
}
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
frame_len = header_len + data_len;
|
|
|
|
ok = frame_len > 0 && frame_len <= buf_len;
|
2014-04-24 21:09:54 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (ok) {
|
|
|
|
struct websocket_message wsm;
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
wsm.size = (size_t) data_len;
|
|
|
|
wsm.data = buf + header_len;
|
|
|
|
wsm.flags = buf[0];
|
2015-02-24 21:32:22 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Apply mask if necessary */
|
|
|
|
if (mask_len > 0) {
|
|
|
|
for (i = 0; i < data_len; i++) {
|
|
|
|
buf[i + header_len] ^= (buf + header_len - mask_len)[i % 4];
|
|
|
|
}
|
|
|
|
}
|
2015-01-28 05:55:13 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (reass) {
|
|
|
|
/* On first fragmented frame, nullify size */
|
|
|
|
if (is_ws_first_fragment(wsm.flags)) {
|
|
|
|
mbuf_resize(&nc->recv_mbuf, nc->recv_mbuf.size + sizeof(*sizep));
|
|
|
|
p[0] &= ~0x0f; /* Next frames will be treated as continuation */
|
|
|
|
buf = p + 1 + sizeof(*sizep);
|
|
|
|
*sizep = 0; /* TODO(lsm): fix. this can stomp over frame data */
|
|
|
|
}
|
2015-01-28 05:55:13 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Append this frame to the reassembled buffer */
|
|
|
|
memmove(buf, wsm.data, e - wsm.data);
|
|
|
|
(*sizep) += wsm.size;
|
|
|
|
nc->recv_mbuf.len -= wsm.data - buf;
|
|
|
|
|
|
|
|
/* On last fragmented frame - call user handler and remove data */
|
|
|
|
if (wsm.flags & 0x80) {
|
|
|
|
wsm.data = p + 1 + sizeof(*sizep);
|
|
|
|
wsm.size = *sizep;
|
|
|
|
handle_incoming_websocket_frame(nc, &wsm);
|
|
|
|
mbuf_remove(&nc->recv_mbuf, 1 + sizeof(*sizep) + *sizep);
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
/* TODO(lsm): properly handle OOB control frames during defragmentation */
|
|
|
|
handle_incoming_websocket_frame(nc, &wsm);
|
|
|
|
mbuf_remove(&nc->recv_mbuf, (size_t) frame_len); /* Cleanup frame */
|
|
|
|
}
|
|
|
|
|
|
|
|
/* If client closes, close too */
|
|
|
|
if ((buf[0] & 0x0f) == WEBSOCKET_OP_CLOSE) {
|
2015-09-19 19:38:46 +08:00
|
|
|
nc->flags |= MG_F_SEND_AND_CLOSE;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-01-28 05:55:13 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return ok;
|
|
|
|
}
|
|
|
|
|
|
|
|
struct ws_mask_ctx {
|
|
|
|
size_t pos; /* zero means unmasked */
|
|
|
|
uint32_t mask;
|
|
|
|
};
|
|
|
|
|
2015-10-06 04:21:58 +08:00
|
|
|
static uint32_t ws_random_mask(void) {
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* The spec requires WS client to generate hard to
|
|
|
|
* guess mask keys. From RFC6455, Section 5.3:
|
|
|
|
*
|
|
|
|
* The unpredictability of the masking key is essential to prevent
|
|
|
|
* authors of malicious applications from selecting the bytes that appear on
|
|
|
|
* the wire.
|
|
|
|
*
|
|
|
|
* Hence this feature is essential when the actual end user of this API
|
|
|
|
* is untrusted code that wouldn't have access to a lower level net API
|
|
|
|
* anyway (e.g. web browsers). Hence this feature is low prio for most
|
|
|
|
* mongoose use cases and thus can be disabled, e.g. when porting to a platform
|
|
|
|
* that lacks random().
|
|
|
|
*/
|
2015-10-06 04:21:58 +08:00
|
|
|
#ifdef MG_DISABLE_WS_RANDOM_MASK
|
2015-09-08 19:49:03 +08:00
|
|
|
return 0xefbeadde; /* generated with a random number generator, I swear */
|
2015-02-19 19:09:34 +08:00
|
|
|
#else
|
2015-09-08 19:49:03 +08:00
|
|
|
if (sizeof(long) >= 4) {
|
|
|
|
return (uint32_t) random();
|
|
|
|
} else if (sizeof(long) == 2) {
|
|
|
|
return (uint32_t) random() << 16 | (uint32_t) random();
|
2014-09-10 01:07:55 +08:00
|
|
|
}
|
2015-02-19 19:09:34 +08:00
|
|
|
#endif
|
2014-09-10 01:07:55 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void mg_send_ws_header(struct mg_connection *nc, int op, size_t len,
|
|
|
|
struct ws_mask_ctx *ctx) {
|
|
|
|
int header_len;
|
|
|
|
unsigned char header[10];
|
|
|
|
|
2015-09-14 19:58:08 +08:00
|
|
|
header[0] = (op & WEBSOCKET_DONT_FIN ? 0x0 : 0x80) + (op & 0x0f);
|
2015-09-08 19:49:03 +08:00
|
|
|
if (len < 126) {
|
|
|
|
header[1] = len;
|
|
|
|
header_len = 2;
|
|
|
|
} else if (len < 65535) {
|
|
|
|
uint16_t tmp = htons((uint16_t) len);
|
|
|
|
header[1] = 126;
|
|
|
|
memcpy(&header[2], &tmp, sizeof(tmp));
|
|
|
|
header_len = 4;
|
|
|
|
} else {
|
|
|
|
uint32_t tmp;
|
|
|
|
header[1] = 127;
|
|
|
|
tmp = htonl((uint32_t)((uint64_t) len >> 32));
|
|
|
|
memcpy(&header[2], &tmp, sizeof(tmp));
|
|
|
|
tmp = htonl((uint32_t)(len & 0xffffffff));
|
|
|
|
memcpy(&header[6], &tmp, sizeof(tmp));
|
|
|
|
header_len = 10;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* client connections enable masking */
|
|
|
|
if (nc->listener == NULL) {
|
|
|
|
header[1] |= 1 << 7; /* set masking flag */
|
|
|
|
mg_send(nc, header, header_len);
|
|
|
|
ctx->mask = ws_random_mask();
|
|
|
|
mg_send(nc, &ctx->mask, sizeof(ctx->mask));
|
|
|
|
ctx->pos = nc->send_mbuf.len;
|
|
|
|
} else {
|
|
|
|
mg_send(nc, header, header_len);
|
|
|
|
ctx->pos = 0;
|
|
|
|
}
|
2014-09-10 01:07:55 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void ws_mask_frame(struct mbuf *mbuf, struct ws_mask_ctx *ctx) {
|
|
|
|
size_t i;
|
|
|
|
if (ctx->pos == 0) return;
|
|
|
|
for (i = 0; i < (mbuf->len - ctx->pos); i++) {
|
|
|
|
mbuf->buf[ctx->pos + i] ^= ((char *) &ctx->mask)[i % 4];
|
|
|
|
}
|
|
|
|
}
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_send_websocket_frame(struct mg_connection *nc, int op, const void *data,
|
|
|
|
size_t len) {
|
|
|
|
struct ws_mask_ctx ctx;
|
|
|
|
mg_send_ws_header(nc, op, len, &ctx);
|
|
|
|
mg_send(nc, data, len);
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
ws_mask_frame(&nc->send_mbuf, &ctx);
|
2014-09-10 01:07:55 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (op == WEBSOCKET_OP_CLOSE) {
|
2015-09-19 19:38:46 +08:00
|
|
|
nc->flags |= MG_F_SEND_AND_CLOSE;
|
2014-09-10 01:07:55 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2014-09-10 01:07:55 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_send_websocket_framev(struct mg_connection *nc, int op,
|
|
|
|
const struct mg_str *strv, int strvcnt) {
|
|
|
|
struct ws_mask_ctx ctx;
|
|
|
|
int i;
|
|
|
|
int len = 0;
|
|
|
|
for (i = 0; i < strvcnt; i++) {
|
|
|
|
len += strv[i].len;
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
mg_send_ws_header(nc, op, len, &ctx);
|
2014-09-10 01:07:55 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
for (i = 0; i < strvcnt; i++) {
|
|
|
|
mg_send(nc, strv[i].p, strv[i].len);
|
|
|
|
}
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
ws_mask_frame(&nc->send_mbuf, &ctx);
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (op == WEBSOCKET_OP_CLOSE) {
|
2015-09-19 19:38:46 +08:00
|
|
|
nc->flags |= MG_F_SEND_AND_CLOSE;
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_printf_websocket_frame(struct mg_connection *nc, int op,
|
|
|
|
const char *fmt, ...) {
|
2015-12-06 00:46:53 +08:00
|
|
|
char mem[MG_VPRINTF_BUFFER_SIZE], *buf = mem;
|
2015-09-08 19:49:03 +08:00
|
|
|
va_list ap;
|
|
|
|
int len;
|
2014-09-10 01:07:55 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
va_start(ap, fmt);
|
|
|
|
if ((len = mg_avprintf(&buf, sizeof(mem), fmt, ap)) > 0) {
|
|
|
|
mg_send_websocket_frame(nc, op, buf, len);
|
2014-05-11 20:42:59 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
va_end(ap);
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (buf != mem && buf != NULL) {
|
2015-09-19 19:38:46 +08:00
|
|
|
MG_FREE(buf);
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
|
|
|
}
|
2014-09-10 01:07:55 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void websocket_handler(struct mg_connection *nc, int ev, void *ev_data) {
|
2015-11-18 15:07:48 +08:00
|
|
|
mg_call(nc, nc->handler, ev, ev_data);
|
2014-09-10 01:07:55 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
switch (ev) {
|
2015-09-19 19:38:46 +08:00
|
|
|
case MG_EV_RECV:
|
2015-09-08 19:49:03 +08:00
|
|
|
do {
|
|
|
|
} while (deliver_websocket_data(nc));
|
|
|
|
break;
|
2015-09-19 19:38:46 +08:00
|
|
|
case MG_EV_POLL:
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Ping idle websocket connections */
|
|
|
|
{
|
|
|
|
time_t now = *(time_t *) ev_data;
|
2015-09-19 19:38:46 +08:00
|
|
|
if (nc->flags & MG_F_IS_WEBSOCKET &&
|
|
|
|
now > nc->last_io_time + MG_WEBSOCKET_PING_INTERVAL_SECONDS) {
|
2015-09-08 19:49:03 +08:00
|
|
|
mg_send_websocket_frame(nc, WEBSOCKET_OP_PING, "", 0);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
2014-09-10 01:07:55 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void ws_handshake(struct mg_connection *nc, const struct mg_str *key) {
|
|
|
|
static const char *magic = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
|
2015-12-06 00:46:53 +08:00
|
|
|
char buf[MG_VPRINTF_BUFFER_SIZE], sha[20], b64_sha[sizeof(sha) * 2];
|
2015-09-08 19:49:03 +08:00
|
|
|
cs_sha1_ctx sha_ctx;
|
2014-09-10 01:07:55 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
snprintf(buf, sizeof(buf), "%.*s%s", (int) key->len, key->p, magic);
|
2014-09-10 01:07:55 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
cs_sha1_init(&sha_ctx);
|
|
|
|
cs_sha1_update(&sha_ctx, (unsigned char *) buf, strlen(buf));
|
|
|
|
cs_sha1_final((unsigned char *) sha, &sha_ctx);
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
mg_base64_encode((unsigned char *) sha, sizeof(sha), b64_sha);
|
|
|
|
mg_printf(nc, "%s%s%s",
|
|
|
|
"HTTP/1.1 101 Switching Protocols\r\n"
|
|
|
|
"Upgrade: websocket\r\n"
|
|
|
|
"Connection: Upgrade\r\n"
|
|
|
|
"Sec-WebSocket-Accept: ",
|
|
|
|
b64_sha, "\r\n\r\n");
|
2014-09-10 01:07:55 +08:00
|
|
|
}
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#endif /* MG_DISABLE_HTTP_WEBSOCKET */
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void free_http_proto_data(struct mg_connection *nc) {
|
|
|
|
struct proto_data_http *dp = (struct proto_data_http *) nc->proto_data;
|
|
|
|
if (dp != NULL) {
|
|
|
|
if (dp->fp != NULL) {
|
|
|
|
fclose(dp->fp);
|
|
|
|
}
|
|
|
|
if (dp->cgi_nc != NULL) {
|
2015-09-19 19:38:46 +08:00
|
|
|
dp->cgi_nc->flags |= MG_F_CLOSE_IMMEDIATELY;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-09-19 19:38:46 +08:00
|
|
|
MG_FREE(dp);
|
2015-09-08 19:49:03 +08:00
|
|
|
nc->proto_data = NULL;
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void transfer_file_data(struct mg_connection *nc) {
|
|
|
|
struct proto_data_http *dp = (struct proto_data_http *) nc->proto_data;
|
2015-09-19 19:38:46 +08:00
|
|
|
char buf[MG_MAX_HTTP_SEND_IOBUF];
|
2015-09-08 19:49:03 +08:00
|
|
|
int64_t left = dp->cl - dp->sent;
|
|
|
|
size_t n = 0, to_read = 0;
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (dp->type == DATA_FILE) {
|
|
|
|
struct mbuf *io = &nc->send_mbuf;
|
|
|
|
if (io->len < sizeof(buf)) {
|
|
|
|
to_read = sizeof(buf) - io->len;
|
|
|
|
}
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (left > 0 && to_read > (size_t) left) {
|
|
|
|
to_read = left;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (to_read == 0) {
|
|
|
|
/* Rate limiting. send_mbuf is too full, wait until it's drained. */
|
|
|
|
} else if (dp->sent < dp->cl && (n = fread(buf, 1, to_read, dp->fp)) > 0) {
|
|
|
|
mg_send(nc, buf, n);
|
|
|
|
dp->sent += n;
|
2014-03-31 05:50:10 +08:00
|
|
|
} else {
|
2015-09-08 19:49:03 +08:00
|
|
|
free_http_proto_data(nc);
|
2015-11-16 16:16:08 +08:00
|
|
|
#ifdef MG_DISABLE_HTTP_KEEP_ALIVE
|
|
|
|
nc->flags |= MG_F_SEND_AND_CLOSE;
|
|
|
|
#endif
|
2014-03-31 05:50:10 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
} else if (dp->type == DATA_PUT) {
|
|
|
|
struct mbuf *io = &nc->recv_mbuf;
|
|
|
|
size_t to_write =
|
|
|
|
left <= 0 ? 0 : left < (int64_t) io->len ? (size_t) left : io->len;
|
|
|
|
size_t n = fwrite(io->buf, 1, to_write, dp->fp);
|
|
|
|
if (n > 0) {
|
|
|
|
mbuf_remove(io, n);
|
|
|
|
dp->sent += n;
|
|
|
|
}
|
|
|
|
if (n == 0 || dp->sent >= dp->cl) {
|
|
|
|
free_http_proto_data(nc);
|
2015-11-16 16:16:08 +08:00
|
|
|
#ifdef MG_DISABLE_HTTP_KEEP_ALIVE
|
|
|
|
nc->flags |= MG_F_SEND_AND_CLOSE;
|
|
|
|
#endif
|
2014-03-31 05:50:10 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
} else if (dp->type == DATA_CGI) {
|
|
|
|
/* This is POST data that needs to be forwarded to the CGI process */
|
|
|
|
if (dp->cgi_nc != NULL) {
|
|
|
|
mg_forward(nc, dp->cgi_nc);
|
|
|
|
} else {
|
2015-09-19 19:38:46 +08:00
|
|
|
nc->flags |= MG_F_SEND_AND_CLOSE;
|
2014-03-26 20:20:02 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Parse chunked-encoded buffer. Return 0 if the buffer is not encoded, or
|
|
|
|
* if it's incomplete. If the chunk is fully buffered, return total number of
|
|
|
|
* bytes in a chunk, and store data in `data`, `data_len`.
|
|
|
|
*/
|
|
|
|
static size_t parse_chunk(char *buf, size_t len, char **chunk_data,
|
|
|
|
size_t *chunk_len) {
|
|
|
|
unsigned char *s = (unsigned char *) buf;
|
|
|
|
size_t n = 0; /* scanned chunk length */
|
|
|
|
size_t i = 0; /* index in s */
|
2014-03-26 20:20:02 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Scan chunk length. That should be a hexadecimal number. */
|
|
|
|
while (i < len && isxdigit(s[i])) {
|
|
|
|
n *= 16;
|
|
|
|
n += (s[i] >= '0' && s[i] <= '9') ? s[i] - '0' : tolower(s[i]) - 'a' + 10;
|
|
|
|
i++;
|
2014-03-26 20:20:02 +08:00
|
|
|
}
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Skip new line */
|
|
|
|
if (i == 0 || i + 2 > len || s[i] != '\r' || s[i + 1] != '\n') {
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
i += 2;
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Record where the data is */
|
|
|
|
*chunk_data = (char *) s + i;
|
|
|
|
*chunk_len = n;
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Skip data */
|
|
|
|
i += n;
|
|
|
|
|
|
|
|
/* Skip new line */
|
|
|
|
if (i == 0 || i + 2 > len || s[i] != '\r' || s[i + 1] != '\n') {
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
return i + 2;
|
2014-07-08 18:28:03 +08:00
|
|
|
}
|
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
MG_INTERNAL size_t mg_handle_chunked(struct mg_connection *nc,
|
2015-09-08 19:49:03 +08:00
|
|
|
struct http_message *hm, char *buf,
|
|
|
|
size_t blen) {
|
|
|
|
struct proto_data_http *dp;
|
|
|
|
char *data;
|
|
|
|
size_t i, n, data_len, body_len, zero_chunk_received = 0;
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* If not allocated, allocate proto_data to hold reassembled offset */
|
|
|
|
if (nc->proto_data == NULL &&
|
2015-09-19 19:38:46 +08:00
|
|
|
(nc->proto_data = MG_CALLOC(1, sizeof(*dp))) == NULL) {
|
|
|
|
nc->flags |= MG_F_CLOSE_IMMEDIATELY;
|
2015-09-08 19:49:03 +08:00
|
|
|
return 0;
|
|
|
|
}
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Find out piece of received data that is not yet reassembled */
|
|
|
|
dp = (struct proto_data_http *) nc->proto_data;
|
|
|
|
body_len = dp->body_len;
|
|
|
|
assert(blen >= body_len);
|
|
|
|
|
|
|
|
/* Traverse all fully buffered chunks */
|
|
|
|
for (i = body_len; (n = parse_chunk(buf + i, blen - i, &data, &data_len)) > 0;
|
|
|
|
i += n) {
|
|
|
|
/* Collapse chunk data to the rest of HTTP body */
|
|
|
|
memmove(buf + body_len, data, data_len);
|
|
|
|
body_len += data_len;
|
|
|
|
hm->body.len = body_len;
|
|
|
|
|
|
|
|
if (data_len == 0) {
|
|
|
|
zero_chunk_received = 1;
|
|
|
|
i += n;
|
|
|
|
break;
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (i > body_len) {
|
|
|
|
/* Shift unparsed content to the parsed body */
|
|
|
|
assert(i <= blen);
|
|
|
|
memmove(buf + body_len, buf + i, blen - i);
|
|
|
|
memset(buf + body_len + blen - i, 0, i - body_len);
|
|
|
|
nc->recv_mbuf.len -= i - body_len;
|
|
|
|
dp->body_len = body_len;
|
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
/* Send MG_EV_HTTP_CHUNK event */
|
|
|
|
nc->flags &= ~MG_F_DELETE_CHUNK;
|
2015-11-18 15:07:48 +08:00
|
|
|
mg_call(nc, nc->handler, MG_EV_HTTP_CHUNK, hm);
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
/* Delete processed data if user set MG_F_DELETE_CHUNK flag */
|
|
|
|
if (nc->flags & MG_F_DELETE_CHUNK) {
|
2015-09-08 19:49:03 +08:00
|
|
|
memset(buf, 0, body_len);
|
|
|
|
memmove(buf, buf + body_len, blen - i);
|
|
|
|
nc->recv_mbuf.len -= body_len;
|
|
|
|
hm->body.len = dp->body_len = 0;
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
if (zero_chunk_received) {
|
|
|
|
hm->message.len = dp->body_len + blen - i;
|
2014-06-25 18:43:07 +08:00
|
|
|
}
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
2014-05-21 23:18:21 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return body_len;
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
|
|
|
|
2015-10-27 02:20:57 +08:00
|
|
|
/*
|
|
|
|
* lx106 compiler has a bug (TODO(mkm) report and insert tracking bug here)
|
|
|
|
* If a big structure is declared in a big function, lx106 gcc will make it
|
|
|
|
* even bigger (round up to 4k, from 700 bytes of actual size).
|
|
|
|
*/
|
|
|
|
#ifdef MG_ESP8266
|
|
|
|
static void http_handler2(struct mg_connection *nc, int ev, void *ev_data,
|
2015-10-27 04:16:48 +08:00
|
|
|
struct http_message *hm) __attribute__((noinline));
|
2015-10-27 02:20:57 +08:00
|
|
|
|
|
|
|
void http_handler(struct mg_connection *nc, int ev, void *ev_data) {
|
2015-09-08 19:49:03 +08:00
|
|
|
struct http_message hm;
|
2015-10-27 02:20:57 +08:00
|
|
|
http_handler2(nc, ev, ev_data, &hm);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void http_handler2(struct mg_connection *nc, int ev, void *ev_data,
|
|
|
|
struct http_message *hm) {
|
|
|
|
#else
|
|
|
|
void http_handler(struct mg_connection *nc, int ev, void *ev_data) {
|
|
|
|
struct http_message shm;
|
|
|
|
struct http_message *hm = &shm;
|
|
|
|
#endif
|
|
|
|
struct mbuf *io = &nc->recv_mbuf;
|
2015-09-08 19:49:03 +08:00
|
|
|
int req_len;
|
|
|
|
const int is_req = (nc->listener != NULL);
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_DISABLE_HTTP_WEBSOCKET
|
2015-09-08 19:49:03 +08:00
|
|
|
struct mg_str *vec;
|
2014-02-20 00:38:53 +08:00
|
|
|
#endif
|
2015-09-25 19:10:39 +08:00
|
|
|
if (ev == MG_EV_CLOSE) {
|
|
|
|
/*
|
|
|
|
* For HTTP messages without Content-Length, always send HTTP message
|
|
|
|
* before MG_EV_CLOSE message.
|
|
|
|
*/
|
2015-10-27 02:20:57 +08:00
|
|
|
if (io->len > 0 && mg_parse_http(io->buf, io->len, hm, is_req) > 0) {
|
2015-11-18 15:07:48 +08:00
|
|
|
int ev2 = is_req ? MG_EV_HTTP_REQUEST : MG_EV_HTTP_REPLY;
|
2015-10-27 02:20:57 +08:00
|
|
|
hm->message.len = io->len;
|
|
|
|
hm->body.len = io->buf + io->len - hm->body.p;
|
2015-11-18 15:07:48 +08:00
|
|
|
mg_call(nc, nc->handler, ev2, hm);
|
2015-09-25 19:10:39 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
free_http_proto_data(nc);
|
|
|
|
}
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (nc->proto_data != NULL) {
|
|
|
|
transfer_file_data(nc);
|
|
|
|
}
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-11-18 15:07:48 +08:00
|
|
|
mg_call(nc, nc->handler, ev, ev_data);
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
if (ev == MG_EV_RECV) {
|
2015-09-08 19:49:03 +08:00
|
|
|
struct mg_str *s;
|
2015-10-27 02:20:57 +08:00
|
|
|
req_len = mg_parse_http(io->buf, io->len, hm, is_req);
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
if (req_len > 0 &&
|
2015-10-27 02:20:57 +08:00
|
|
|
(s = mg_get_http_header(hm, "Transfer-Encoding")) != NULL &&
|
2015-09-08 19:49:03 +08:00
|
|
|
mg_vcasecmp(s, "chunked") == 0) {
|
2015-10-27 02:20:57 +08:00
|
|
|
mg_handle_chunked(nc, hm, io->buf + req_len, io->len - req_len);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
if (req_len < 0 || (req_len == 0 && io->len >= MG_MAX_HTTP_REQUEST_SIZE)) {
|
2015-11-18 15:07:48 +08:00
|
|
|
DBG(("invalid request"));
|
2015-09-19 19:38:46 +08:00
|
|
|
nc->flags |= MG_F_CLOSE_IMMEDIATELY;
|
2015-09-08 19:49:03 +08:00
|
|
|
} else if (req_len == 0) {
|
|
|
|
/* Do nothing, request is not yet fully buffered */
|
|
|
|
}
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_DISABLE_HTTP_WEBSOCKET
|
2015-09-08 19:49:03 +08:00
|
|
|
else if (nc->listener == NULL &&
|
2015-10-27 02:20:57 +08:00
|
|
|
mg_get_http_header(hm, "Sec-WebSocket-Accept")) {
|
2015-09-08 19:49:03 +08:00
|
|
|
/* We're websocket client, got handshake response from server. */
|
|
|
|
/* TODO(lsm): check the validity of accept Sec-WebSocket-Accept */
|
|
|
|
mbuf_remove(io, req_len);
|
|
|
|
nc->proto_handler = websocket_handler;
|
2015-09-19 19:38:46 +08:00
|
|
|
nc->flags |= MG_F_IS_WEBSOCKET;
|
2015-11-18 15:07:48 +08:00
|
|
|
mg_call(nc, nc->handler, MG_EV_WEBSOCKET_HANDSHAKE_DONE, NULL);
|
2015-09-19 19:38:46 +08:00
|
|
|
websocket_handler(nc, MG_EV_RECV, ev_data);
|
2015-09-08 19:49:03 +08:00
|
|
|
} else if (nc->listener != NULL &&
|
2015-10-27 02:20:57 +08:00
|
|
|
(vec = mg_get_http_header(hm, "Sec-WebSocket-Key")) != NULL) {
|
2015-09-08 19:49:03 +08:00
|
|
|
/* This is a websocket request. Switch protocol handlers. */
|
|
|
|
mbuf_remove(io, req_len);
|
|
|
|
nc->proto_handler = websocket_handler;
|
2015-09-19 19:38:46 +08:00
|
|
|
nc->flags |= MG_F_IS_WEBSOCKET;
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
/* Send handshake */
|
2015-11-18 15:07:48 +08:00
|
|
|
mg_call(nc, nc->handler, MG_EV_WEBSOCKET_HANDSHAKE_REQUEST, hm);
|
2015-09-19 19:38:46 +08:00
|
|
|
if (!(nc->flags & MG_F_CLOSE_IMMEDIATELY)) {
|
2015-09-08 19:49:03 +08:00
|
|
|
if (nc->send_mbuf.len == 0) {
|
|
|
|
ws_handshake(nc, vec);
|
|
|
|
}
|
2015-11-18 15:07:48 +08:00
|
|
|
mg_call(nc, nc->handler, MG_EV_WEBSOCKET_HANDSHAKE_DONE, NULL);
|
2015-09-19 19:38:46 +08:00
|
|
|
websocket_handler(nc, MG_EV_RECV, ev_data);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
}
|
2015-09-19 19:38:46 +08:00
|
|
|
#endif /* MG_DISABLE_HTTP_WEBSOCKET */
|
2015-10-27 02:20:57 +08:00
|
|
|
else if (hm->message.len <= io->len) {
|
2015-09-22 00:58:45 +08:00
|
|
|
int trigger_ev = nc->listener ? MG_EV_HTTP_REQUEST : MG_EV_HTTP_REPLY;
|
|
|
|
|
|
|
|
/* Whole HTTP message is fully buffered, call event handler */
|
|
|
|
|
|
|
|
#ifdef MG_ENABLE_JAVASCRIPT
|
|
|
|
v7_val_t v1, v2, headers, req, args, res;
|
|
|
|
struct v7 *v7 = nc->mgr->v7;
|
|
|
|
const char *ev_name = trigger_ev == MG_EV_HTTP_REPLY ? "onsnd" : "onrcv";
|
|
|
|
int i, js_callback_handled_request = 0;
|
|
|
|
|
|
|
|
if (v7 != NULL) {
|
|
|
|
/* Lookup JS callback */
|
2015-10-04 05:21:06 +08:00
|
|
|
v1 = v7_get(v7, v7_get_global(v7), "Http", ~0);
|
2015-09-22 00:58:45 +08:00
|
|
|
v2 = v7_get(v7, v1, ev_name, ~0);
|
|
|
|
|
|
|
|
/* Create callback params. TODO(lsm): own/disown those */
|
2016-01-12 18:38:26 +08:00
|
|
|
args = v7_mk_array(v7);
|
|
|
|
req = v7_mk_object(v7);
|
|
|
|
headers = v7_mk_object(v7);
|
2015-09-22 00:58:45 +08:00
|
|
|
|
|
|
|
/* Populate request object */
|
2016-01-12 06:49:21 +08:00
|
|
|
v7_set(v7, req, "method", ~0,
|
2016-01-12 18:38:26 +08:00
|
|
|
v7_mk_string(v7, hm->method.p, hm->method.len, 1));
|
|
|
|
v7_set(v7, req, "uri", ~0, v7_mk_string(v7, hm->uri.p, hm->uri.len, 1));
|
2016-01-12 06:49:21 +08:00
|
|
|
v7_set(v7, req, "body", ~0,
|
2016-01-12 18:38:26 +08:00
|
|
|
v7_mk_string(v7, hm->body.p, hm->body.len, 1));
|
2016-01-12 06:49:21 +08:00
|
|
|
v7_set(v7, req, "headers", ~0, headers);
|
2015-10-27 02:20:57 +08:00
|
|
|
for (i = 0; hm->header_names[i].len > 0; i++) {
|
|
|
|
const struct mg_str *name = &hm->header_names[i];
|
|
|
|
const struct mg_str *value = &hm->header_values[i];
|
2016-01-12 06:49:21 +08:00
|
|
|
v7_set(v7, headers, name->p, name->len,
|
2016-01-12 18:38:26 +08:00
|
|
|
v7_mk_string(v7, value->p, value->len, 1));
|
2015-09-22 00:58:45 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
/* Invoke callback. TODO(lsm): report errors */
|
2016-01-12 18:38:26 +08:00
|
|
|
v7_array_push(v7, args, v7_mk_foreign(nc));
|
2015-09-22 00:58:45 +08:00
|
|
|
v7_array_push(v7, args, req);
|
2016-01-12 18:38:26 +08:00
|
|
|
if (v7_apply(v7, v2, v7_mk_undefined(), args, &res) == V7_OK &&
|
2016-01-13 08:06:08 +08:00
|
|
|
v7_is_truthy(v7, res)) {
|
2015-09-22 00:58:45 +08:00
|
|
|
js_callback_handled_request++;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* If JS callback returns true, stop request processing */
|
|
|
|
if (js_callback_handled_request) {
|
|
|
|
nc->flags |= MG_F_SEND_AND_CLOSE;
|
|
|
|
} else {
|
2015-11-18 15:07:48 +08:00
|
|
|
mg_call(nc, nc->handler, trigger_ev, hm);
|
2015-09-22 00:58:45 +08:00
|
|
|
}
|
|
|
|
#else
|
2015-11-18 15:07:48 +08:00
|
|
|
mg_call(nc, nc->handler, trigger_ev, hm);
|
2015-09-22 00:58:45 +08:00
|
|
|
#endif
|
2015-10-27 02:20:57 +08:00
|
|
|
mbuf_remove(io, hm->message.len);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_set_protocol_http_websocket(struct mg_connection *nc) {
|
|
|
|
nc->proto_handler = http_handler;
|
2014-09-10 01:07:55 +08:00
|
|
|
}
|
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_DISABLE_HTTP_WEBSOCKET
|
2014-09-10 01:07:55 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_send_websocket_handshake(struct mg_connection *nc, const char *uri,
|
|
|
|
const char *extra_headers) {
|
|
|
|
unsigned long random = (unsigned long) uri;
|
|
|
|
char key[sizeof(random) * 3];
|
|
|
|
|
|
|
|
mg_base64_encode((unsigned char *) &random, sizeof(random), key);
|
|
|
|
mg_printf(nc,
|
|
|
|
"GET %s HTTP/1.1\r\n"
|
|
|
|
"Upgrade: websocket\r\n"
|
|
|
|
"Connection: Upgrade\r\n"
|
|
|
|
"Sec-WebSocket-Version: 13\r\n"
|
|
|
|
"Sec-WebSocket-Key: %s\r\n"
|
|
|
|
"%s\r\n",
|
|
|
|
uri, key, extra_headers == NULL ? "" : extra_headers);
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#endif /* MG_DISABLE_HTTP_WEBSOCKET */
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-11-18 15:39:51 +08:00
|
|
|
void mg_send_response_line(struct mg_connection *nc, int status_code,
|
|
|
|
const char *extra_headers) {
|
|
|
|
const char *status_message = "OK";
|
|
|
|
switch (status_code) {
|
|
|
|
case 206:
|
|
|
|
status_message = "Partial Content";
|
|
|
|
break;
|
2015-11-27 05:37:20 +08:00
|
|
|
case 301:
|
|
|
|
status_message = "Moved";
|
|
|
|
break;
|
|
|
|
case 302:
|
|
|
|
status_message = "Found";
|
|
|
|
break;
|
2015-11-18 15:39:51 +08:00
|
|
|
case 416:
|
|
|
|
status_message = "Requested range not satisfiable";
|
|
|
|
break;
|
|
|
|
case 418:
|
|
|
|
status_message = "I'm a teapot";
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
mg_printf(nc, "HTTP/1.1 %d %s\r\n", status_code, status_message);
|
|
|
|
if (extra_headers != NULL) {
|
|
|
|
mg_printf(nc, "%s\r\n", extra_headers);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-11-21 22:35:14 +08:00
|
|
|
void mg_send_head(struct mg_connection *c, int status_code,
|
|
|
|
int64_t content_length, const char *extra_headers) {
|
|
|
|
mg_send_response_line(c, status_code, extra_headers);
|
|
|
|
if (content_length < 0) {
|
|
|
|
mg_printf(c, "%s", "Transfer-Encoding: chunked\r\n");
|
|
|
|
} else {
|
|
|
|
mg_printf(c, "Content-Length: %" INT64_FMT "\r\n", content_length);
|
|
|
|
}
|
|
|
|
mg_send(c, "\r\n", 2);
|
|
|
|
}
|
|
|
|
|
2015-12-14 03:56:33 +08:00
|
|
|
#ifdef MG_DISABLE_FILESYSTEM
|
|
|
|
void mg_serve_http(struct mg_connection *nc, struct http_message *hm,
|
|
|
|
struct mg_serve_http_opts opts) {
|
|
|
|
mg_send_head(nc, 501, 0, NULL);
|
|
|
|
}
|
|
|
|
#else
|
2015-09-08 19:49:03 +08:00
|
|
|
static void send_http_error(struct mg_connection *nc, int code,
|
|
|
|
const char *reason) {
|
2015-12-11 14:25:07 +08:00
|
|
|
(void) reason;
|
|
|
|
mg_send_head(nc, code, 0, NULL);
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_DISABLE_SSI
|
2015-09-08 19:49:03 +08:00
|
|
|
static void send_ssi_file(struct mg_connection *, const char *, FILE *, int,
|
|
|
|
const struct mg_serve_http_opts *);
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void send_file_data(struct mg_connection *nc, FILE *fp) {
|
|
|
|
char buf[BUFSIZ];
|
|
|
|
size_t n;
|
|
|
|
while ((n = fread(buf, 1, sizeof(buf), fp)) > 0) {
|
|
|
|
mg_send(nc, buf, n);
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void do_ssi_include(struct mg_connection *nc, const char *ssi, char *tag,
|
|
|
|
int include_level,
|
|
|
|
const struct mg_serve_http_opts *opts) {
|
|
|
|
char file_name[BUFSIZ], path[MAX_PATH_SIZE], *p;
|
|
|
|
FILE *fp;
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* sscanf() is safe here, since send_ssi_file() also uses buffer
|
|
|
|
* of size MG_BUF_LEN to get the tag. So strlen(tag) is always < MG_BUF_LEN.
|
|
|
|
*/
|
|
|
|
if (sscanf(tag, " virtual=\"%[^\"]\"", file_name) == 1) {
|
|
|
|
/* File name is relative to the webserver root */
|
|
|
|
snprintf(path, sizeof(path), "%s/%s", opts->document_root, file_name);
|
|
|
|
} else if (sscanf(tag, " abspath=\"%[^\"]\"", file_name) == 1) {
|
|
|
|
/*
|
|
|
|
* File name is relative to the webserver working directory
|
|
|
|
* or it is absolute system path
|
|
|
|
*/
|
|
|
|
snprintf(path, sizeof(path), "%s", file_name);
|
|
|
|
} else if (sscanf(tag, " file=\"%[^\"]\"", file_name) == 1 ||
|
|
|
|
sscanf(tag, " \"%[^\"]\"", file_name) == 1) {
|
|
|
|
/* File name is relative to the currect document */
|
|
|
|
snprintf(path, sizeof(path), "%s", ssi);
|
|
|
|
if ((p = strrchr(path, '/')) != NULL) {
|
|
|
|
p[1] = '\0';
|
2014-03-01 09:55:40 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
snprintf(path + strlen(path), sizeof(path) - strlen(path), "%s", file_name);
|
|
|
|
} else {
|
|
|
|
mg_printf(nc, "Bad SSI #include: [%s]", tag);
|
|
|
|
return;
|
|
|
|
}
|
2014-03-01 09:55:40 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if ((fp = fopen(path, "rb")) == NULL) {
|
|
|
|
mg_printf(nc, "SSI include error: fopen(%s): %s", path, strerror(errno));
|
|
|
|
} else {
|
|
|
|
mg_set_close_on_exec(fileno(fp));
|
|
|
|
if (mg_match_prefix(opts->ssi_pattern, strlen(opts->ssi_pattern), path) >
|
|
|
|
0) {
|
|
|
|
send_ssi_file(nc, path, fp, include_level + 1, opts);
|
|
|
|
} else {
|
|
|
|
send_file_data(nc, fp);
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
fclose(fp);
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_DISABLE_POPEN
|
2015-09-08 19:49:03 +08:00
|
|
|
static void do_ssi_exec(struct mg_connection *nc, char *tag) {
|
|
|
|
char cmd[BUFSIZ];
|
|
|
|
FILE *fp;
|
|
|
|
|
|
|
|
if (sscanf(tag, " \"%[^\"]\"", cmd) != 1) {
|
|
|
|
mg_printf(nc, "Bad SSI #exec: [%s]", tag);
|
|
|
|
} else if ((fp = popen(cmd, "r")) == NULL) {
|
|
|
|
mg_printf(nc, "Cannot SSI #exec: [%s]: %s", cmd, strerror(errno));
|
|
|
|
} else {
|
|
|
|
send_file_data(nc, fp);
|
|
|
|
pclose(fp);
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-09-19 19:38:46 +08:00
|
|
|
#endif /* !MG_DISABLE_POPEN */
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void do_ssi_call(struct mg_connection *nc, char *tag) {
|
2015-10-29 06:12:55 +08:00
|
|
|
mg_call(nc, NULL, MG_EV_SSI_CALL, tag);
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* SSI directive has the following format:
|
|
|
|
* <!--#directive parameter=value parameter=value -->
|
|
|
|
*/
|
|
|
|
static void send_ssi_file(struct mg_connection *nc, const char *path, FILE *fp,
|
|
|
|
int include_level,
|
|
|
|
const struct mg_serve_http_opts *opts) {
|
2015-09-19 19:38:46 +08:00
|
|
|
static const struct mg_str btag = MG_STR("<!--#");
|
|
|
|
static const struct mg_str d_include = MG_STR("include");
|
|
|
|
static const struct mg_str d_call = MG_STR("call");
|
2015-09-22 02:20:05 +08:00
|
|
|
#ifndef MG_DISABLE_POPEN
|
2015-09-19 19:38:46 +08:00
|
|
|
static const struct mg_str d_exec = MG_STR("exec");
|
2015-09-22 02:20:05 +08:00
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
char buf[BUFSIZ], *p = buf + btag.len; /* p points to SSI directive */
|
|
|
|
int ch, offset, len, in_ssi_tag;
|
2014-03-26 20:20:02 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (include_level > 10) {
|
|
|
|
mg_printf(nc, "SSI #include level is too deep (%s)", path);
|
|
|
|
return;
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
in_ssi_tag = len = offset = 0;
|
|
|
|
while ((ch = fgetc(fp)) != EOF) {
|
|
|
|
if (in_ssi_tag && ch == '>' && buf[len - 1] == '-' && buf[len - 2] == '-') {
|
|
|
|
size_t i = len - 2;
|
|
|
|
in_ssi_tag = 0;
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Trim closing --> */
|
|
|
|
buf[i--] = '\0';
|
|
|
|
while (i > 0 && buf[i] == ' ') {
|
|
|
|
buf[i--] = '\0';
|
|
|
|
}
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Handle known SSI directives */
|
|
|
|
if (memcmp(p, d_include.p, d_include.len) == 0) {
|
|
|
|
do_ssi_include(nc, path, p + d_include.len + 1, include_level, opts);
|
|
|
|
} else if (memcmp(p, d_call.p, d_call.len) == 0) {
|
|
|
|
do_ssi_call(nc, p + d_call.len + 1);
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_DISABLE_POPEN
|
2015-09-08 19:49:03 +08:00
|
|
|
} else if (memcmp(p, d_exec.p, d_exec.len) == 0) {
|
|
|
|
do_ssi_exec(nc, p + d_exec.len + 1);
|
|
|
|
#endif
|
|
|
|
} else {
|
|
|
|
/* Silently ignore unknown SSI directive. */
|
|
|
|
}
|
|
|
|
len = 0;
|
2015-10-16 19:42:22 +08:00
|
|
|
} else if (ch == '<') {
|
|
|
|
in_ssi_tag = 1;
|
|
|
|
if (len > 0) {
|
|
|
|
mg_send(nc, buf, (size_t) len);
|
|
|
|
}
|
|
|
|
len = 0;
|
|
|
|
buf[len++] = ch & 0xff;
|
2015-09-08 19:49:03 +08:00
|
|
|
} else if (in_ssi_tag) {
|
|
|
|
if (len == (int) btag.len && memcmp(buf, btag.p, btag.len) != 0) {
|
|
|
|
/* Not an SSI tag */
|
|
|
|
in_ssi_tag = 0;
|
|
|
|
} else if (len == (int) sizeof(buf) - 2) {
|
|
|
|
mg_printf(nc, "%s: SSI tag is too large", path);
|
|
|
|
len = 0;
|
|
|
|
}
|
|
|
|
buf[len++] = ch & 0xff;
|
2014-09-10 01:07:55 +08:00
|
|
|
} else {
|
2015-09-08 19:49:03 +08:00
|
|
|
buf[len++] = ch & 0xff;
|
|
|
|
if (len == (int) sizeof(buf)) {
|
|
|
|
mg_send(nc, buf, (size_t) len);
|
|
|
|
len = 0;
|
|
|
|
}
|
2014-09-10 01:07:55 +08:00
|
|
|
}
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Send the rest of buffered data */
|
|
|
|
if (len > 0) {
|
|
|
|
mg_send(nc, buf, (size_t) len);
|
|
|
|
}
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void handle_ssi_request(struct mg_connection *nc, const char *path,
|
|
|
|
const struct mg_serve_http_opts *opts) {
|
|
|
|
FILE *fp;
|
|
|
|
struct mg_str mime_type;
|
|
|
|
|
|
|
|
if ((fp = fopen(path, "rb")) == NULL) {
|
|
|
|
send_http_error(nc, 404, "Not Found");
|
|
|
|
} else {
|
|
|
|
mg_set_close_on_exec(fileno(fp));
|
|
|
|
|
|
|
|
mime_type = get_mime_type(path, "text/plain", opts);
|
2015-11-18 15:39:51 +08:00
|
|
|
mg_send_response_line(nc, 200, opts->extra_headers);
|
2015-09-08 19:49:03 +08:00
|
|
|
mg_printf(nc,
|
|
|
|
"Content-Type: %.*s\r\n"
|
|
|
|
"Connection: close\r\n\r\n",
|
|
|
|
(int) mime_type.len, mime_type.p);
|
|
|
|
send_ssi_file(nc, path, fp, 0, opts);
|
|
|
|
fclose(fp);
|
2015-09-19 19:38:46 +08:00
|
|
|
nc->flags |= MG_F_SEND_AND_CLOSE;
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
#else
|
|
|
|
static void handle_ssi_request(struct mg_connection *nc, const char *path,
|
|
|
|
const struct mg_serve_http_opts *opts) {
|
|
|
|
(void) path;
|
|
|
|
(void) opts;
|
|
|
|
send_http_error(nc, 500, "SSI disabled");
|
|
|
|
}
|
2015-09-19 19:38:46 +08:00
|
|
|
#endif /* MG_DISABLE_SSI */
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void construct_etag(char *buf, size_t buf_len, const cs_stat_t *st) {
|
|
|
|
snprintf(buf, buf_len, "\"%lx.%" INT64_FMT "\"", (unsigned long) st->st_mtime,
|
|
|
|
(int64_t) st->st_size);
|
|
|
|
}
|
|
|
|
static void gmt_time_string(char *buf, size_t buf_len, time_t *t) {
|
|
|
|
strftime(buf, buf_len, "%a, %d %b %Y %H:%M:%S GMT", gmtime(t));
|
2014-06-10 00:24:51 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static int parse_range_header(const struct mg_str *header, int64_t *a,
|
|
|
|
int64_t *b) {
|
|
|
|
/*
|
|
|
|
* There is no snscanf. Headers are not guaranteed to be NUL-terminated,
|
|
|
|
* so we have this. Ugh.
|
|
|
|
*/
|
|
|
|
int result;
|
2015-09-19 19:38:46 +08:00
|
|
|
char *p = (char *) MG_MALLOC(header->len + 1);
|
2015-09-08 19:49:03 +08:00
|
|
|
if (p == NULL) return 0;
|
|
|
|
memcpy(p, header->p, header->len);
|
|
|
|
p[header->len] = '\0';
|
|
|
|
result = sscanf(p, "bytes=%" INT64_FMT "-%" INT64_FMT, a, b);
|
2015-09-19 19:38:46 +08:00
|
|
|
MG_FREE(p);
|
2015-09-08 19:49:03 +08:00
|
|
|
return result;
|
2014-03-01 09:47:01 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void mg_send_http_file2(struct mg_connection *nc, const char *path,
|
|
|
|
cs_stat_t *st, struct http_message *hm,
|
|
|
|
struct mg_serve_http_opts *opts) {
|
|
|
|
struct proto_data_http *dp;
|
|
|
|
struct mg_str mime_type;
|
|
|
|
|
|
|
|
free_http_proto_data(nc);
|
2015-09-19 19:38:46 +08:00
|
|
|
if ((dp = (struct proto_data_http *) MG_CALLOC(1, sizeof(*dp))) == NULL) {
|
2015-09-08 19:49:03 +08:00
|
|
|
send_http_error(nc, 500, "Server Error"); /* LCOV_EXCL_LINE */
|
|
|
|
} else if ((dp->fp = fopen(path, "rb")) == NULL) {
|
2015-09-19 19:38:46 +08:00
|
|
|
MG_FREE(dp);
|
2015-09-08 19:49:03 +08:00
|
|
|
nc->proto_data = NULL;
|
|
|
|
send_http_error(nc, 500, "Server Error");
|
|
|
|
} else if (mg_match_prefix(opts->ssi_pattern, strlen(opts->ssi_pattern),
|
|
|
|
path) > 0) {
|
2015-10-21 17:05:13 +08:00
|
|
|
nc->proto_data = (void *) dp;
|
2015-09-08 19:49:03 +08:00
|
|
|
handle_ssi_request(nc, path, opts);
|
|
|
|
} else {
|
|
|
|
char etag[50], current_time[50], last_modified[50], range[50];
|
|
|
|
time_t t = time(NULL);
|
|
|
|
int64_t r1 = 0, r2 = 0, cl = st->st_size;
|
|
|
|
struct mg_str *range_hdr = mg_get_http_header(hm, "Range");
|
|
|
|
int n, status_code = 200;
|
|
|
|
|
|
|
|
/* Handle Range header */
|
|
|
|
range[0] = '\0';
|
|
|
|
if (range_hdr != NULL &&
|
|
|
|
(n = parse_range_header(range_hdr, &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_code = 416;
|
|
|
|
cl = 0;
|
|
|
|
snprintf(range, sizeof(range),
|
|
|
|
"Content-Range: bytes */%" INT64_FMT "\r\n",
|
|
|
|
(int64_t) st->st_size);
|
|
|
|
} else {
|
|
|
|
status_code = 206;
|
|
|
|
cl = r2 - r1 + 1;
|
|
|
|
snprintf(range, sizeof(range), "Content-Range: bytes %" INT64_FMT
|
|
|
|
"-%" INT64_FMT "/%" INT64_FMT "\r\n",
|
|
|
|
r1, r1 + cl - 1, (int64_t) st->st_size);
|
|
|
|
fseeko(dp->fp, r1, SEEK_SET);
|
|
|
|
}
|
|
|
|
}
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
construct_etag(etag, sizeof(etag), st);
|
|
|
|
gmt_time_string(current_time, sizeof(current_time), &t);
|
|
|
|
gmt_time_string(last_modified, sizeof(last_modified), &st->st_mtime);
|
|
|
|
mime_type = get_mime_type(path, "text/plain", opts);
|
2015-09-23 18:41:15 +08:00
|
|
|
/*
|
2015-09-24 01:33:25 +08:00
|
|
|
* Content length casted to size_t because:
|
|
|
|
* 1) that's the maximum buffer size anyway
|
|
|
|
* 2) ESP8266 RTOS SDK newlib vprintf cannot contain a 64bit arg at non-last
|
|
|
|
* position
|
|
|
|
* TODO(mkm): fix ESP8266 RTOS SDK
|
2015-09-23 18:41:15 +08:00
|
|
|
*/
|
2015-11-18 15:39:51 +08:00
|
|
|
mg_send_response_line(nc, status_code, opts->extra_headers);
|
2015-09-08 19:49:03 +08:00
|
|
|
mg_printf(nc,
|
|
|
|
"Date: %s\r\n"
|
|
|
|
"Last-Modified: %s\r\n"
|
|
|
|
"Accept-Ranges: bytes\r\n"
|
2015-09-24 01:33:25 +08:00
|
|
|
"Content-Type: %.*s\r\n"
|
2015-11-16 16:16:08 +08:00
|
|
|
#ifdef MG_DISABLE_HTTP_KEEP_ALIVE
|
|
|
|
"Connection: close\r\n"
|
|
|
|
#endif
|
2015-09-24 01:33:25 +08:00
|
|
|
"Content-Length: %" SIZE_T_FMT
|
|
|
|
"\r\n"
|
|
|
|
"%sEtag: %s\r\n\r\n",
|
2015-11-18 15:39:51 +08:00
|
|
|
current_time, last_modified, (int) mime_type.len, mime_type.p,
|
|
|
|
(size_t) cl, range, etag);
|
2015-09-23 18:41:15 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
nc->proto_data = (void *) dp;
|
|
|
|
dp->cl = cl;
|
|
|
|
dp->type = DATA_FILE;
|
|
|
|
transfer_file_data(nc);
|
|
|
|
}
|
|
|
|
}
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +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;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2015-06-04 18:27:31 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
*p = '\0';
|
|
|
|
}
|
2014-03-01 09:47:01 +08:00
|
|
|
|
2014-02-20 00:38:53 +08:00
|
|
|
#endif
|
|
|
|
|
2015-12-04 05:59:19 +08:00
|
|
|
int mg_url_decode(const char *src, int src_len, char *dst, int dst_len,
|
|
|
|
int is_form_url_encoded) {
|
2015-09-08 19:49:03 +08:00
|
|
|
int i, j, a, b;
|
|
|
|
#define HEXTOI(x) (isdigit(x) ? x - '0' : x - 'W')
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
for (i = j = 0; i < src_len && j < dst_len - 1; i++, j++) {
|
|
|
|
if (src[i] == '%') {
|
|
|
|
if (i < src_len - 2 && isxdigit(*(const unsigned char *) (src + i + 1)) &&
|
|
|
|
isxdigit(*(const unsigned char *) (src + i + 2))) {
|
|
|
|
a = tolower(*(const unsigned char *) (src + i + 1));
|
|
|
|
b = tolower(*(const unsigned char *) (src + i + 2));
|
|
|
|
dst[j] = (char) ((HEXTOI(a) << 4) | HEXTOI(b));
|
|
|
|
i += 2;
|
|
|
|
} else {
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
} else if (is_form_url_encoded && src[i] == '+') {
|
|
|
|
dst[j] = ' ';
|
|
|
|
} else {
|
|
|
|
dst[j] = src[i];
|
|
|
|
}
|
|
|
|
}
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
dst[j] = '\0'; /* Null-terminate the destination */
|
2014-02-20 00:38:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return i >= src_len ? j : -1;
|
2014-02-20 00:38:53 +08:00
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
int mg_get_http_var(const struct mg_str *buf, const char *name, char *dst,
|
|
|
|
size_t dst_len) {
|
|
|
|
const char *p, *e, *s;
|
|
|
|
size_t name_len;
|
|
|
|
int len;
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (dst == NULL || dst_len == 0) {
|
|
|
|
len = -2;
|
|
|
|
} else if (buf->p == NULL || name == NULL || buf->len == 0) {
|
|
|
|
len = -1;
|
|
|
|
dst[0] = '\0';
|
|
|
|
} else {
|
|
|
|
name_len = strlen(name);
|
|
|
|
e = buf->p + buf->len;
|
|
|
|
len = -1;
|
|
|
|
dst[0] = '\0';
|
2014-10-15 15:28:17 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
for (p = buf->p; p + name_len < e; p++) {
|
|
|
|
if ((p == buf->p || p[-1] == '&') && p[name_len] == '=' &&
|
|
|
|
!mg_ncasecmp(name, p, name_len)) {
|
|
|
|
p += name_len + 1;
|
|
|
|
s = (const char *) memchr(p, '&', (size_t)(e - p));
|
|
|
|
if (s == NULL) {
|
|
|
|
s = e;
|
|
|
|
}
|
|
|
|
len = mg_url_decode(p, (size_t)(s - p), dst, dst_len, 1);
|
|
|
|
if (len == -1) {
|
|
|
|
len = -2;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2014-10-15 15:28:17 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return len;
|
|
|
|
}
|
2013-12-21 23:50:11 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_send_http_chunk(struct mg_connection *nc, const char *buf, size_t len) {
|
|
|
|
char chunk_size[50];
|
|
|
|
int n;
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
n = snprintf(chunk_size, sizeof(chunk_size), "%lX\r\n", (unsigned long) len);
|
|
|
|
mg_send(nc, chunk_size, n);
|
|
|
|
mg_send(nc, buf, len);
|
|
|
|
mg_send(nc, "\r\n", 2);
|
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_printf_http_chunk(struct mg_connection *nc, const char *fmt, ...) {
|
2015-12-06 00:46:53 +08:00
|
|
|
char mem[MG_VPRINTF_BUFFER_SIZE], *buf = mem;
|
2015-09-08 19:49:03 +08:00
|
|
|
int len;
|
|
|
|
va_list ap;
|
2013-12-31 21:57:51 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
va_start(ap, fmt);
|
|
|
|
len = mg_avprintf(&buf, sizeof(mem), fmt, ap);
|
|
|
|
va_end(ap);
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (len >= 0) {
|
|
|
|
mg_send_http_chunk(nc, buf, len);
|
|
|
|
}
|
2014-01-16 21:23:23 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* LCOV_EXCL_START */
|
|
|
|
if (buf != mem && buf != NULL) {
|
2015-09-19 19:38:46 +08:00
|
|
|
MG_FREE(buf);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
/* LCOV_EXCL_STOP */
|
|
|
|
}
|
2014-02-04 17:59:53 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_printf_html_escape(struct mg_connection *nc, const char *fmt, ...) {
|
2015-12-06 00:46:53 +08:00
|
|
|
char mem[MG_VPRINTF_BUFFER_SIZE], *buf = mem;
|
2015-09-08 19:49:03 +08:00
|
|
|
int i, j, len;
|
|
|
|
va_list ap;
|
2014-01-11 17:48:27 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
va_start(ap, fmt);
|
|
|
|
len = mg_avprintf(&buf, sizeof(mem), fmt, ap);
|
|
|
|
va_end(ap);
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (len >= 0) {
|
|
|
|
for (i = j = 0; i < len; i++) {
|
|
|
|
if (buf[i] == '<' || buf[i] == '>') {
|
|
|
|
mg_send(nc, buf + j, i - j);
|
|
|
|
mg_send(nc, buf[i] == '<' ? "<" : ">", 4);
|
|
|
|
j = i + 1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
mg_send(nc, buf + j, i - j);
|
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* LCOV_EXCL_START */
|
|
|
|
if (buf != mem && buf != NULL) {
|
2015-09-19 19:38:46 +08:00
|
|
|
MG_FREE(buf);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
/* LCOV_EXCL_STOP */
|
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
int mg_http_parse_header(struct mg_str *hdr, const char *var_name, char *buf,
|
|
|
|
size_t buf_size) {
|
|
|
|
int ch = ' ', ch1 = ',', len = 0, n = strlen(var_name);
|
2015-10-20 17:31:23 +08:00
|
|
|
const char *p, *end = hdr ? hdr->p + hdr->len : NULL, *s = NULL;
|
2014-01-11 17:48:27 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (buf != NULL && buf_size > 0) buf[0] = '\0';
|
2015-10-20 17:31:23 +08:00
|
|
|
if (hdr == NULL) return 0;
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Find where variable starts */
|
|
|
|
for (s = hdr->p; s != NULL && s + n < end; s++) {
|
|
|
|
if ((s == hdr->p || s[-1] == ch || s[-1] == ch1) && s[n] == '=' &&
|
|
|
|
!memcmp(s, var_name, n))
|
|
|
|
break;
|
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (s != NULL && &s[n + 1] < end) {
|
|
|
|
s += n + 1;
|
|
|
|
if (*s == '"' || *s == '\'') {
|
|
|
|
ch = ch1 = *s++;
|
|
|
|
}
|
|
|
|
p = s;
|
|
|
|
while (p < end && p[0] != ch && p[0] != ch1 && len < (int) buf_size) {
|
|
|
|
if (ch != ' ' && p[0] == '\\' && p[1] == ch) p++;
|
|
|
|
buf[len++] = *p++;
|
|
|
|
}
|
|
|
|
if (len >= (int) buf_size || (ch != ' ' && *p != ch)) {
|
|
|
|
len = 0;
|
|
|
|
} else {
|
|
|
|
if (len > 0 && s[len - 1] == ',') len--;
|
|
|
|
if (len > 0 && s[len - 1] == ';') len--;
|
|
|
|
buf[len] = '\0';
|
|
|
|
}
|
|
|
|
}
|
2014-02-20 00:34:51 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return len;
|
|
|
|
}
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_DISABLE_FILESYSTEM
|
2015-09-08 19:49:03 +08:00
|
|
|
static int is_file_hidden(const char *path,
|
|
|
|
const struct mg_serve_http_opts *opts) {
|
|
|
|
const char *p1 = opts->per_directory_auth_file;
|
|
|
|
const char *p2 = opts->hidden_file_pattern;
|
2015-11-30 01:10:14 +08:00
|
|
|
|
|
|
|
/* Strip directory path from the file name */
|
|
|
|
const char *pdir = strrchr(path, DIRSEP);
|
|
|
|
if (pdir != NULL) {
|
|
|
|
path = pdir + 1;
|
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return !strcmp(path, ".") || !strcmp(path, "..") ||
|
|
|
|
(p1 != NULL && !strcmp(path, p1)) ||
|
|
|
|
(p2 != NULL && mg_match_prefix(p2, strlen(p2), path) > 0);
|
|
|
|
}
|
2014-02-20 00:34:51 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_DISABLE_HTTP_DIGEST_AUTH
|
2015-09-08 19:49:03 +08:00
|
|
|
static void mkmd5resp(const char *method, size_t method_len, const char *uri,
|
|
|
|
size_t uri_len, const char *ha1, size_t ha1_len,
|
|
|
|
const char *nonce, size_t nonce_len, const char *nc,
|
|
|
|
size_t nc_len, const char *cnonce, size_t cnonce_len,
|
|
|
|
const char *qop, size_t qop_len, char *resp) {
|
|
|
|
static const char colon[] = ":";
|
|
|
|
static const size_t one = 1;
|
|
|
|
char ha2[33];
|
|
|
|
|
2015-10-20 17:31:23 +08:00
|
|
|
cs_md5(ha2, method, method_len, colon, one, uri, uri_len, NULL);
|
|
|
|
cs_md5(resp, ha1, ha1_len, colon, one, nonce, nonce_len, colon, one, nc,
|
2015-09-08 19:49:03 +08:00
|
|
|
nc_len, colon, one, cnonce, cnonce_len, colon, one, qop, qop_len,
|
|
|
|
colon, one, ha2, sizeof(ha2) - 1, NULL);
|
|
|
|
}
|
|
|
|
|
|
|
|
int mg_http_create_digest_auth_header(char *buf, size_t buf_len,
|
|
|
|
const char *method, const char *uri,
|
|
|
|
const char *auth_domain, const char *user,
|
|
|
|
const char *passwd) {
|
|
|
|
static const char colon[] = ":", qop[] = "auth";
|
|
|
|
static const size_t one = 1;
|
|
|
|
char ha1[33], resp[33], cnonce[40];
|
|
|
|
|
|
|
|
snprintf(cnonce, sizeof(cnonce), "%x", (unsigned int) time(NULL));
|
2015-10-20 17:31:23 +08:00
|
|
|
cs_md5(ha1, user, (size_t) strlen(user), colon, one, auth_domain,
|
2015-09-08 19:49:03 +08:00
|
|
|
(size_t) strlen(auth_domain), colon, one, passwd,
|
|
|
|
(size_t) strlen(passwd), NULL);
|
|
|
|
mkmd5resp(method, strlen(method), uri, strlen(uri), ha1, sizeof(ha1) - 1,
|
|
|
|
cnonce, strlen(cnonce), "1", one, cnonce, strlen(cnonce), qop,
|
|
|
|
sizeof(qop) - 1, resp);
|
|
|
|
return snprintf(buf, buf_len,
|
|
|
|
"Authorization: Digest username=\"%s\","
|
|
|
|
"realm=\"%s\",uri=\"%s\",qop=%s,nc=1,cnonce=%s,"
|
|
|
|
"nonce=%s,response=%s\r\n",
|
|
|
|
user, auth_domain, uri, qop, cnonce, cnonce, resp);
|
2014-09-10 01:07:55 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Check for authentication timeout.
|
|
|
|
* Clients send time stamp encoded in nonce. Make sure it is not too old,
|
|
|
|
* to prevent replay attacks.
|
|
|
|
* Assumption: nonce is a hexadecimal number of seconds since 1970.
|
|
|
|
*/
|
|
|
|
static int check_nonce(const char *nonce) {
|
|
|
|
unsigned long now = (unsigned long) time(NULL);
|
|
|
|
unsigned long val = (unsigned long) strtoul(nonce, NULL, 16);
|
|
|
|
return 1 || now < val || now - val < 3600;
|
2014-09-10 01:07:55 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Authenticate HTTP request against opened passwords file.
|
|
|
|
* Returns 1 if authenticated, 0 otherwise.
|
|
|
|
*/
|
|
|
|
static int mg_http_check_digest_auth(struct http_message *hm,
|
|
|
|
const char *auth_domain, FILE *fp) {
|
|
|
|
struct mg_str *hdr;
|
|
|
|
char buf[128], f_user[sizeof(buf)], f_ha1[sizeof(buf)], f_domain[sizeof(buf)];
|
|
|
|
char user[50], cnonce[20], response[40], uri[200], qop[20], nc[20], nonce[30];
|
|
|
|
char expected_response[33];
|
|
|
|
|
|
|
|
/* Parse "Authorization:" header, fail fast on parse error */
|
|
|
|
if (hm == NULL || fp == NULL ||
|
|
|
|
(hdr = mg_get_http_header(hm, "Authorization")) == NULL ||
|
|
|
|
mg_http_parse_header(hdr, "username", user, sizeof(user)) == 0 ||
|
|
|
|
mg_http_parse_header(hdr, "cnonce", cnonce, sizeof(cnonce)) == 0 ||
|
|
|
|
mg_http_parse_header(hdr, "response", response, sizeof(response)) == 0 ||
|
|
|
|
mg_http_parse_header(hdr, "uri", uri, sizeof(uri)) == 0 ||
|
|
|
|
mg_http_parse_header(hdr, "qop", qop, sizeof(qop)) == 0 ||
|
|
|
|
mg_http_parse_header(hdr, "nc", nc, sizeof(nc)) == 0 ||
|
|
|
|
mg_http_parse_header(hdr, "nonce", nonce, sizeof(nonce)) == 0 ||
|
|
|
|
check_nonce(nonce) == 0) {
|
|
|
|
return 0;
|
|
|
|
}
|
2014-01-10 19:03:38 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Read passwords file line by line. If should have htdigest format,
|
|
|
|
* i.e. each line should be a colon-separated sequence:
|
|
|
|
* USER_NAME:DOMAIN_NAME:HA1_HASH_OF_USER_DOMAIN_AND_PASSWORD
|
|
|
|
*/
|
|
|
|
while (fgets(buf, sizeof(buf), fp) != NULL) {
|
|
|
|
if (sscanf(buf, "%[^:]:%[^:]:%s", f_user, f_domain, f_ha1) == 3 &&
|
|
|
|
strcmp(user, f_user) == 0 &&
|
|
|
|
/* NOTE(lsm): due to a bug in MSIE, we do not compare URIs */
|
|
|
|
strcmp(auth_domain, f_domain) == 0) {
|
|
|
|
/* User and domain matched, check the password */
|
|
|
|
mkmd5resp(hm->method.p, hm->method.len, hm->uri.p, hm->uri.len, f_ha1,
|
|
|
|
strlen(f_ha1), nonce, strlen(nonce), nc, strlen(nc), cnonce,
|
|
|
|
strlen(cnonce), qop, strlen(qop), expected_response);
|
|
|
|
return mg_casecmp(response, expected_response) == 0;
|
|
|
|
}
|
|
|
|
}
|
2014-01-11 05:59:37 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* None of the entries in the passwords file matched - return failure */
|
|
|
|
return 0;
|
|
|
|
}
|
2014-01-10 19:03:38 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static int is_authorized(struct http_message *hm, const char *path,
|
2015-10-20 17:31:23 +08:00
|
|
|
int is_directory, const char *domain,
|
|
|
|
const char *passwords_file, int is_global_pass_file) {
|
|
|
|
char buf[MAX_PATH_SIZE];
|
|
|
|
const char *p;
|
2015-09-08 19:49:03 +08:00
|
|
|
FILE *fp;
|
|
|
|
int authorized = 1;
|
|
|
|
|
2015-10-20 17:31:23 +08:00
|
|
|
if (domain != NULL && passwords_file != NULL) {
|
|
|
|
if (is_global_pass_file) {
|
|
|
|
fp = fopen(passwords_file, "r");
|
|
|
|
} else if (is_directory) {
|
|
|
|
snprintf(buf, sizeof(buf), "%s%c%s", path, DIRSEP, passwords_file);
|
|
|
|
fp = fopen(buf, "r");
|
|
|
|
} else {
|
|
|
|
if ((p = strrchr(path, '/')) == NULL &&
|
|
|
|
(p = strrchr(path, '\\')) == NULL) {
|
|
|
|
p = path;
|
|
|
|
}
|
|
|
|
snprintf(buf, sizeof(buf), "%.*s/%s", (int) (p - path), path,
|
|
|
|
passwords_file);
|
|
|
|
fp = fopen(buf, "r");
|
|
|
|
}
|
|
|
|
|
|
|
|
if (fp != NULL) {
|
|
|
|
authorized = mg_http_check_digest_auth(hm, domain, fp);
|
|
|
|
fclose(fp);
|
|
|
|
}
|
2014-01-10 19:03:38 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return authorized;
|
2014-01-10 19:03:38 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
#else
|
|
|
|
static int is_authorized(struct http_message *hm, const char *path,
|
2015-10-20 17:31:23 +08:00
|
|
|
int is_directory, const char *domain,
|
|
|
|
const char *passwords_file, int is_global_pass_file) {
|
2015-09-08 19:49:03 +08:00
|
|
|
(void) hm;
|
|
|
|
(void) path;
|
|
|
|
(void) is_directory;
|
2015-10-20 17:31:23 +08:00
|
|
|
(void) domain;
|
|
|
|
(void) passwords_file;
|
|
|
|
(void) is_global_pass_file;
|
2015-09-08 19:49:03 +08:00
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
#endif
|
2014-01-10 19:03:38 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_DISABLE_DIRECTORY_LISTING
|
2015-09-08 19:49:03 +08:00
|
|
|
static size_t mg_url_encode(const char *src, size_t s_len, char *dst,
|
|
|
|
size_t dst_len) {
|
2015-12-08 04:18:06 +08:00
|
|
|
static const char *dont_escape = "._-$,;~()/";
|
2015-09-08 19:49:03 +08:00
|
|
|
static const char *hex = "0123456789abcdef";
|
|
|
|
size_t i = 0, j = 0;
|
|
|
|
|
|
|
|
for (i = j = 0; dst_len > 0 && i < s_len && j + 2 < dst_len - 1; i++, j++) {
|
|
|
|
if (isalnum(*(const unsigned char *) (src + i)) ||
|
|
|
|
strchr(dont_escape, *(const unsigned char *) (src + i)) != NULL) {
|
|
|
|
dst[j] = src[i];
|
|
|
|
} else if (j + 3 < dst_len) {
|
|
|
|
dst[j] = '%';
|
|
|
|
dst[j + 1] = hex[(*(const unsigned char *) (src + i)) >> 4];
|
|
|
|
dst[j + 2] = hex[(*(const unsigned char *) (src + i)) & 0xf];
|
|
|
|
j += 2;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
dst[j] = '\0';
|
|
|
|
return j;
|
2014-01-10 19:03:38 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void escape(const char *src, char *dst, size_t dst_len) {
|
|
|
|
size_t n = 0;
|
|
|
|
while (*src != '\0' && n + 5 < dst_len) {
|
|
|
|
unsigned char ch = *(unsigned char *) src++;
|
|
|
|
if (ch == '<') {
|
|
|
|
n += snprintf(dst + n, dst_len - n, "%s", "<");
|
|
|
|
} else {
|
|
|
|
dst[n++] = ch;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
dst[n] = '\0';
|
2014-01-10 19:03:38 +08:00
|
|
|
}
|
2014-03-20 23:01:05 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void print_dir_entry(struct mg_connection *nc, const char *file_name,
|
|
|
|
cs_stat_t *stp) {
|
|
|
|
char size[64], mod[64], href[MAX_PATH_SIZE * 3], path[MAX_PATH_SIZE];
|
|
|
|
int64_t fsize = stp->st_size;
|
|
|
|
int is_dir = S_ISDIR(stp->st_mode);
|
|
|
|
const char *slash = is_dir ? "/" : "";
|
|
|
|
|
|
|
|
if (is_dir) {
|
|
|
|
snprintf(size, sizeof(size), "%s", "[DIRECTORY]");
|
2014-01-09 20:34:50 +08:00
|
|
|
} else {
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* We use (double) cast below because MSVC 6 compiler cannot
|
|
|
|
* convert unsigned __int64 to double.
|
|
|
|
*/
|
|
|
|
if (fsize < 1024) {
|
|
|
|
snprintf(size, sizeof(size), "%d", (int) fsize);
|
|
|
|
} else if (fsize < 0x100000) {
|
|
|
|
snprintf(size, sizeof(size), "%.1fk", (double) fsize / 1024.0);
|
|
|
|
} else if (fsize < 0x40000000) {
|
|
|
|
snprintf(size, sizeof(size), "%.1fM", (double) fsize / 1048576);
|
2014-01-09 20:34:50 +08:00
|
|
|
} else {
|
2015-09-08 19:49:03 +08:00
|
|
|
snprintf(size, sizeof(size), "%.1fG", (double) fsize / 1073741824);
|
2014-01-09 20:34:50 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
strftime(mod, sizeof(mod), "%d-%b-%Y %H:%M", localtime(&stp->st_mtime));
|
|
|
|
escape(file_name, path, sizeof(path));
|
|
|
|
mg_url_encode(file_name, strlen(file_name), href, sizeof(href));
|
|
|
|
mg_printf_http_chunk(nc,
|
|
|
|
"<tr><td><a href=\"%s%s\">%s%s</a></td>"
|
|
|
|
"<td>%s</td><td name=%" INT64_FMT ">%s</td></tr>\n",
|
|
|
|
href, slash, path, slash, mod, is_dir ? -1 : fsize,
|
|
|
|
size);
|
|
|
|
}
|
2014-01-09 20:34:50 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void scan_directory(struct mg_connection *nc, const char *dir,
|
|
|
|
const struct mg_serve_http_opts *opts,
|
|
|
|
void (*func)(struct mg_connection *, const char *,
|
|
|
|
cs_stat_t *)) {
|
|
|
|
char path[MAX_PATH_SIZE];
|
|
|
|
cs_stat_t st;
|
|
|
|
struct dirent *dp;
|
|
|
|
DIR *dirp;
|
|
|
|
|
|
|
|
if ((dirp = (opendir(dir))) != NULL) {
|
|
|
|
while ((dp = readdir(dirp)) != NULL) {
|
|
|
|
/* Do not show current dir and hidden files */
|
2015-12-08 22:16:56 +08:00
|
|
|
if (is_file_hidden((const char *) dp->d_name, opts)) {
|
2015-09-08 19:49:03 +08:00
|
|
|
continue;
|
|
|
|
}
|
|
|
|
snprintf(path, sizeof(path), "%s/%s", dir, dp->d_name);
|
|
|
|
if (mg_stat(path, &st) == 0) {
|
2015-12-08 22:16:56 +08:00
|
|
|
func(nc, (const char *) dp->d_name, &st);
|
2014-01-09 20:34:50 +08:00
|
|
|
}
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
closedir(dirp);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static void send_directory_listing(struct mg_connection *nc, const char *dir,
|
|
|
|
struct http_message *hm,
|
|
|
|
struct mg_serve_http_opts *opts) {
|
|
|
|
static const char *sort_js_code =
|
|
|
|
"<script>function srt(tb, col) {"
|
|
|
|
"var tr = Array.prototype.slice.call(tb.rows, 0),"
|
|
|
|
"tr = tr.sort(function (a, b) { var c1 = a.cells[col], c2 = b.cells[col],"
|
|
|
|
"n1 = c1.getAttribute('name'), n2 = c2.getAttribute('name'), "
|
|
|
|
"t1 = a.cells[2].getAttribute('name'), "
|
|
|
|
"t2 = b.cells[2].getAttribute('name'); "
|
|
|
|
"return 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]);}"
|
|
|
|
"window.onload = function() { "
|
|
|
|
"var tb = document.getElementById('tb');"
|
|
|
|
"document.onclick = function(ev){ "
|
|
|
|
"var c = ev.target.rel; if (c) srt(tb, c)}; srt(tb, 2); };</script>";
|
|
|
|
|
2015-11-18 15:39:51 +08:00
|
|
|
mg_send_response_line(nc, 200, opts->extra_headers);
|
|
|
|
mg_printf(nc, "%s: %s\r\n%s: %s\r\n\r\n", "Transfer-Encoding", "chunked",
|
|
|
|
"Content-Type", "text/html; charset=utf-8");
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
mg_printf_http_chunk(
|
|
|
|
nc,
|
|
|
|
"<html><head><title>Index of %.*s</title>%s%s"
|
|
|
|
"<style>th,td {text-align: left; padding-right: 1em; }</style></head>"
|
|
|
|
"<body><h1>Index of %.*s</h1><pre><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>",
|
|
|
|
(int) hm->uri.len, hm->uri.p, sort_js_code, sort_js_code2,
|
|
|
|
(int) hm->uri.len, hm->uri.p);
|
|
|
|
scan_directory(nc, dir, opts, print_dir_entry);
|
|
|
|
mg_printf_http_chunk(nc, "%s", "</tbody></body></html>");
|
|
|
|
mg_send_http_chunk(nc, "", 0);
|
|
|
|
/* TODO(rojer): Remove when cesanta/dev/issues/197 is fixed. */
|
2015-09-19 19:38:46 +08:00
|
|
|
nc->flags |= MG_F_SEND_AND_CLOSE;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-09-19 19:38:46 +08:00
|
|
|
#endif /* MG_DISABLE_DIRECTORY_LISTING */
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_DISABLE_DAV
|
2015-09-08 19:49:03 +08:00
|
|
|
static void print_props(struct mg_connection *nc, const char *name,
|
|
|
|
cs_stat_t *stp) {
|
|
|
|
char mtime[64], buf[MAX_PATH_SIZE * 3];
|
|
|
|
time_t t = stp->st_mtime; /* store in local variable for NDK compile */
|
|
|
|
gmt_time_string(mtime, sizeof(mtime), &t);
|
|
|
|
mg_url_encode(name, strlen(name), buf, sizeof(buf));
|
|
|
|
mg_printf(nc,
|
|
|
|
"<d:response>"
|
|
|
|
"<d:href>%s</d:href>"
|
|
|
|
"<d:propstat>"
|
|
|
|
"<d:prop>"
|
|
|
|
"<d:resourcetype>%s</d:resourcetype>"
|
|
|
|
"<d:getcontentlength>%" INT64_FMT
|
|
|
|
"</d:getcontentlength>"
|
|
|
|
"<d:getlastmodified>%s</d:getlastmodified>"
|
|
|
|
"</d:prop>"
|
|
|
|
"<d:status>HTTP/1.1 200 OK</d:status>"
|
|
|
|
"</d:propstat>"
|
|
|
|
"</d:response>\n",
|
|
|
|
buf, S_ISDIR(stp->st_mode) ? "<d:collection/>" : "",
|
|
|
|
(int64_t) stp->st_size, mtime);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void handle_propfind(struct mg_connection *nc, const char *path,
|
|
|
|
cs_stat_t *stp, struct http_message *hm,
|
|
|
|
struct mg_serve_http_opts *opts) {
|
|
|
|
static const char header[] =
|
|
|
|
"HTTP/1.1 207 Multi-Status\r\n"
|
|
|
|
"Connection: close\r\n"
|
|
|
|
"Content-Type: text/xml; charset=utf-8\r\n\r\n"
|
|
|
|
"<?xml version=\"1.0\" encoding=\"utf-8\"?>"
|
|
|
|
"<d:multistatus xmlns:d='DAV:'>\n";
|
|
|
|
static const char footer[] = "</d:multistatus>\n";
|
|
|
|
const struct mg_str *depth = mg_get_http_header(hm, "Depth");
|
|
|
|
|
|
|
|
/* Print properties for the requested resource itself */
|
|
|
|
if (S_ISDIR(stp->st_mode) &&
|
|
|
|
strcmp(opts->enable_directory_listing, "yes") != 0) {
|
|
|
|
mg_printf(nc, "%s", "HTTP/1.1 403 Directory Listing Denied\r\n\r\n");
|
|
|
|
} else {
|
|
|
|
char uri[MAX_PATH_SIZE];
|
|
|
|
mg_send(nc, header, sizeof(header) - 1);
|
|
|
|
snprintf(uri, sizeof(uri), "%.*s", (int) hm->uri.len, hm->uri.p);
|
|
|
|
print_props(nc, uri, stp);
|
|
|
|
if (S_ISDIR(stp->st_mode) && (depth == NULL || mg_vcmp(depth, "0") != 0)) {
|
|
|
|
scan_directory(nc, path, opts, print_props);
|
|
|
|
}
|
|
|
|
mg_send(nc, footer, sizeof(footer) - 1);
|
2015-09-19 19:38:46 +08:00
|
|
|
nc->flags |= MG_F_SEND_AND_CLOSE;
|
2014-01-09 20:34:50 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-01-11 18:52:35 +08:00
|
|
|
#ifdef MG_ENABLE_FAKE_DAVLOCK
|
|
|
|
/*
|
|
|
|
* Windows explorer (probably there are another WebDav clients like it)
|
|
|
|
* requires LOCK support in webdav. W/out this, it still works, but fails
|
|
|
|
* to save file: shows error message and offers "Save As".
|
|
|
|
* "Save as" works, but this message is very annoying.
|
|
|
|
* This is fake lock, which doesn't lock something, just returns LOCK token,
|
|
|
|
* UNLOCK always answers "OK".
|
|
|
|
* With this fake LOCK Windows Explorer looks happy and saves file.
|
|
|
|
* NOTE: that is not DAV LOCK imlementation, it is just a way to shut up
|
|
|
|
* Windows native DAV client. This is why FAKE LOCK is not enabed by default
|
|
|
|
*/
|
|
|
|
static void handle_lock(struct mg_connection *nc, const char *path) {
|
|
|
|
static const char *reply =
|
|
|
|
"HTTP/1.1 207 Multi-Status\r\n"
|
|
|
|
"Connection: close\r\n"
|
|
|
|
"Content-Type: text/xml; charset=utf-8\r\n\r\n"
|
|
|
|
"<?xml version=\"1.0\" encoding=\"utf-8\"?>"
|
|
|
|
"<d:multistatus xmlns:d='DAV:'>\n"
|
|
|
|
"<D:lockdiscovery>\n"
|
|
|
|
"<D:activelock>\n"
|
|
|
|
"<D:locktoken>\n"
|
|
|
|
"<D:href>\n"
|
|
|
|
"opaquelocktoken:%s%u"
|
|
|
|
"</D:href>"
|
|
|
|
"</D:locktoken>"
|
|
|
|
"</D:activelock>\n"
|
|
|
|
"</D:lockdiscovery>"
|
|
|
|
"</d:multistatus>\n";
|
|
|
|
mg_printf(nc, reply, path, (unsigned int) time(NULL));
|
|
|
|
nc->flags |= MG_F_SEND_AND_CLOSE;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void handle_mkcol(struct mg_connection *nc, const char *path,
|
|
|
|
struct http_message *hm) {
|
|
|
|
int status_code = 500;
|
2015-12-22 05:26:36 +08:00
|
|
|
if (hm->body.len != (size_t) ~0 && hm->body.len > 0) {
|
2015-09-08 19:49:03 +08:00
|
|
|
status_code = 415;
|
|
|
|
} else if (!mg_mkdir(path, 0755)) {
|
|
|
|
status_code = 201;
|
|
|
|
} else if (errno == EEXIST) {
|
|
|
|
status_code = 405;
|
|
|
|
} else if (errno == EACCES) {
|
|
|
|
status_code = 403;
|
|
|
|
} else if (errno == ENOENT) {
|
|
|
|
status_code = 409;
|
2015-12-22 05:26:36 +08:00
|
|
|
} else {
|
|
|
|
status_code = 500;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
send_http_error(nc, status_code, NULL);
|
2013-09-30 20:09:16 +08:00
|
|
|
}
|
|
|
|
|
2015-12-08 22:16:56 +08:00
|
|
|
static int remove_directory(const struct mg_serve_http_opts *opts,
|
|
|
|
const char *dir) {
|
2015-09-08 19:49:03 +08:00
|
|
|
char path[MAX_PATH_SIZE];
|
|
|
|
struct dirent *dp;
|
|
|
|
cs_stat_t st;
|
|
|
|
DIR *dirp;
|
2013-09-30 20:09:16 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if ((dirp = opendir(dir)) == NULL) return 0;
|
2014-01-10 01:50:57 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
while ((dp = readdir(dirp)) != NULL) {
|
2015-12-08 22:16:56 +08:00
|
|
|
if (is_file_hidden((const char *) dp->d_name, opts)) {
|
|
|
|
continue;
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
snprintf(path, sizeof(path), "%s%c%s", dir, '/', dp->d_name);
|
|
|
|
mg_stat(path, &st);
|
|
|
|
if (S_ISDIR(st.st_mode)) {
|
2015-12-08 22:16:56 +08:00
|
|
|
remove_directory(opts, path);
|
2015-09-08 19:49:03 +08:00
|
|
|
} else {
|
|
|
|
remove(path);
|
2014-01-10 01:50:57 +08:00
|
|
|
}
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
closedir(dirp);
|
|
|
|
rmdir(dir);
|
2014-01-10 01:50:57 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return 1;
|
2014-01-10 01:50:57 +08:00
|
|
|
}
|
|
|
|
|
2015-12-11 14:25:07 +08:00
|
|
|
static void handle_move(struct mg_connection *c,
|
2015-12-09 00:32:17 +08:00
|
|
|
const struct mg_serve_http_opts *opts, const char *path,
|
|
|
|
struct http_message *hm) {
|
2015-12-11 14:25:07 +08:00
|
|
|
const struct mg_str *dest = mg_get_http_header(hm, "Destination");
|
|
|
|
if (dest == NULL) {
|
|
|
|
send_http_error(c, 411, NULL);
|
|
|
|
} else {
|
|
|
|
const char *p = (char *) memchr(dest->p, '/', dest->len);
|
|
|
|
if (p != NULL && p[1] == '/' &&
|
|
|
|
(p = (char *) memchr(p + 2, '/', dest->p + dest->len - p)) != NULL) {
|
|
|
|
char buf[MAX_PATH_SIZE];
|
|
|
|
snprintf(buf, sizeof(buf), "%s%.*s", opts->dav_document_root,
|
|
|
|
(int) (dest->p + dest->len - p), p);
|
|
|
|
if (rename(path, buf) == 0) {
|
|
|
|
send_http_error(c, 200, NULL);
|
|
|
|
} else {
|
|
|
|
send_http_error(c, 418, NULL);
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
send_http_error(c, 500, NULL);
|
|
|
|
}
|
|
|
|
}
|
2015-12-09 00:32:17 +08:00
|
|
|
}
|
|
|
|
|
2015-12-08 22:16:56 +08:00
|
|
|
static void handle_delete(struct mg_connection *nc,
|
|
|
|
const struct mg_serve_http_opts *opts,
|
|
|
|
const char *path) {
|
2015-09-08 19:49:03 +08:00
|
|
|
cs_stat_t st;
|
|
|
|
if (mg_stat(path, &st) != 0) {
|
|
|
|
send_http_error(nc, 404, NULL);
|
|
|
|
} else if (S_ISDIR(st.st_mode)) {
|
2015-12-08 22:16:56 +08:00
|
|
|
remove_directory(opts, path);
|
2015-09-08 19:49:03 +08:00
|
|
|
send_http_error(nc, 204, NULL);
|
|
|
|
} else if (remove(path) == 0) {
|
|
|
|
send_http_error(nc, 204, NULL);
|
|
|
|
} else {
|
|
|
|
send_http_error(nc, 423, NULL);
|
|
|
|
}
|
|
|
|
}
|
2014-01-10 01:50:57 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Return -1 on error, 1 on success. */
|
|
|
|
static int create_itermediate_directories(const char *path) {
|
2015-12-11 14:25:07 +08:00
|
|
|
const char *s;
|
2014-01-10 01:50:57 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Create intermediate directories if they do not exist */
|
2015-12-11 14:25:07 +08:00
|
|
|
for (s = path + 1; *s != '\0'; s++) {
|
2015-09-08 19:49:03 +08:00
|
|
|
if (*s == '/') {
|
|
|
|
char buf[MAX_PATH_SIZE];
|
|
|
|
cs_stat_t st;
|
|
|
|
snprintf(buf, sizeof(buf), "%.*s", (int) (s - path), path);
|
|
|
|
buf[sizeof(buf) - 1] = '\0';
|
|
|
|
if (mg_stat(buf, &st) != 0 && mg_mkdir(buf, 0755) != 0) {
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
}
|
2014-01-10 01:50:57 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return 1;
|
2014-01-10 01:50:57 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void handle_put(struct mg_connection *nc, const char *path,
|
|
|
|
struct http_message *hm) {
|
|
|
|
cs_stat_t st;
|
|
|
|
const struct mg_str *cl_hdr = mg_get_http_header(hm, "Content-Length");
|
|
|
|
int rc, status_code = mg_stat(path, &st) == 0 ? 200 : 201;
|
|
|
|
struct proto_data_http *dp = (struct proto_data_http *) nc->proto_data;
|
2014-01-10 01:50:57 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
free_http_proto_data(nc);
|
|
|
|
if ((rc = create_itermediate_directories(path)) == 0) {
|
|
|
|
mg_printf(nc, "HTTP/1.1 %d OK\r\nContent-Length: 0\r\n\r\n", status_code);
|
|
|
|
} else if (rc == -1) {
|
|
|
|
send_http_error(nc, 500, NULL);
|
|
|
|
} else if (cl_hdr == NULL) {
|
|
|
|
send_http_error(nc, 411, NULL);
|
2015-09-19 19:38:46 +08:00
|
|
|
} else if ((dp = (struct proto_data_http *) MG_CALLOC(1, sizeof(*dp))) ==
|
2015-09-08 19:49:03 +08:00
|
|
|
NULL) {
|
|
|
|
send_http_error(nc, 500, NULL); /* LCOV_EXCL_LINE */
|
|
|
|
} else if ((dp->fp = fopen(path, "w+b")) == NULL) {
|
|
|
|
send_http_error(nc, 500, NULL);
|
|
|
|
free_http_proto_data(nc);
|
|
|
|
} else {
|
|
|
|
const struct mg_str *range_hdr = mg_get_http_header(hm, "Content-Range");
|
|
|
|
int64_t r1 = 0, r2 = 0;
|
|
|
|
dp->type = DATA_PUT;
|
|
|
|
mg_set_close_on_exec(fileno(dp->fp));
|
|
|
|
dp->cl = to64(cl_hdr->p);
|
|
|
|
if (range_hdr != NULL && parse_range_header(range_hdr, &r1, &r2) > 0) {
|
|
|
|
status_code = 206;
|
|
|
|
fseeko(dp->fp, r1, SEEK_SET);
|
|
|
|
dp->cl = r2 > r1 ? r2 - r1 + 1 : dp->cl - r1;
|
|
|
|
}
|
|
|
|
mg_printf(nc, "HTTP/1.1 %d OK\r\nContent-Length: 0\r\n\r\n", status_code);
|
|
|
|
nc->proto_data = dp;
|
|
|
|
/* Remove HTTP request from the mbuf, leave only payload */
|
|
|
|
mbuf_remove(&nc->recv_mbuf, hm->message.len - hm->body.len);
|
|
|
|
transfer_file_data(nc);
|
|
|
|
}
|
|
|
|
}
|
2015-09-19 19:38:46 +08:00
|
|
|
#endif /* MG_DISABLE_DAV */
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
static int is_dav_request(const struct mg_str *s) {
|
2016-01-11 18:52:35 +08:00
|
|
|
static const char *methods[] = {"PUT", "DELETE", "MKCOL", "PROPFIND", "MOVE"
|
|
|
|
#ifdef MG_ENABLE_FAKE_DAVLOCK
|
|
|
|
,
|
|
|
|
"LOCK", "UNLOCK"
|
|
|
|
#endif
|
|
|
|
};
|
|
|
|
size_t i;
|
|
|
|
|
|
|
|
for (i = 0; i < ARRAY_SIZE(methods); i++) {
|
|
|
|
if (mg_vcmp(s, methods[i]) == 0) {
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2014-01-09 20:36:25 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Given a directory path, find one of the files specified in the
|
|
|
|
* comma-separated list of index files `list`.
|
|
|
|
* First found index file wins. If an index file is found, then gets
|
|
|
|
* appended to the `path`, stat-ed, and result of `stat()` passed to `stp`.
|
|
|
|
* If index file is not found, then `path` and `stp` remain unchanged.
|
|
|
|
*/
|
2015-09-19 19:38:46 +08:00
|
|
|
MG_INTERNAL int find_index_file(char *path, size_t path_len, const char *list,
|
2015-09-08 19:49:03 +08:00
|
|
|
cs_stat_t *stp) {
|
|
|
|
cs_stat_t st;
|
|
|
|
size_t n = strlen(path);
|
|
|
|
struct mg_str vec;
|
|
|
|
int found = 0;
|
2014-01-13 20:17:47 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* The 'path' given to us points to the directory. Remove all trailing */
|
|
|
|
/* directory separator characters from the end of the path, and */
|
|
|
|
/* then append single directory separator character. */
|
|
|
|
while (n > 0 && (path[n - 1] == '/' || path[n - 1] == '\\')) {
|
|
|
|
n--;
|
2014-01-13 20:17:47 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Traverse index files list. For each entry, append it to the given */
|
|
|
|
/* path and see if the file exists. If it exists, break the loop */
|
|
|
|
while ((list = mg_next_comma_list_entry(list, &vec, NULL)) != NULL) {
|
|
|
|
/* Prepare full path to the index file */
|
|
|
|
snprintf(path + n, path_len - n, "/%.*s", (int) vec.len, vec.p);
|
|
|
|
path[path_len - 1] = '\0';
|
|
|
|
|
|
|
|
/* Does it exist? */
|
|
|
|
if (!mg_stat(path, &st)) {
|
|
|
|
/* Yes it does, break the loop */
|
|
|
|
*stp = st;
|
|
|
|
found = 1;
|
|
|
|
break;
|
2014-01-09 20:36:25 +08:00
|
|
|
}
|
|
|
|
}
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* If no index file exists, restore directory path, keep trailing slash. */
|
|
|
|
if (!found) {
|
|
|
|
path[n] = '\0';
|
|
|
|
strncat(path + n, "/", path_len - n);
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
2013-12-26 18:12:18 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return found;
|
2015-05-16 03:20:33 +08:00
|
|
|
}
|
|
|
|
|
2015-11-27 05:37:20 +08:00
|
|
|
static int send_port_based_redirect(struct mg_connection *c,
|
|
|
|
struct http_message *hm,
|
|
|
|
const struct mg_serve_http_opts *opts) {
|
|
|
|
const char *rewrites = opts->url_rewrites;
|
|
|
|
struct mg_str a, b;
|
|
|
|
char local_port[20] = {'%'};
|
|
|
|
|
|
|
|
#ifndef MG_ESP8266
|
|
|
|
mg_sock_to_str(c->sock, local_port + 1, sizeof(local_port) - 1,
|
|
|
|
MG_SOCK_STRINGIFY_PORT);
|
|
|
|
#else
|
|
|
|
/* TODO(lsm): remove when mg_sock_to_str() is implemented in LWIP codepath */
|
|
|
|
snprintf(local_port, sizeof(local_port), "%s", "%0");
|
|
|
|
#endif
|
|
|
|
|
|
|
|
while ((rewrites = mg_next_comma_list_entry(rewrites, &a, &b)) != NULL) {
|
|
|
|
if (mg_vcmp(&a, local_port) == 0) {
|
|
|
|
mg_send_response_line(c, 301, NULL);
|
|
|
|
mg_printf(c, "Content-Length: 0\r\nLocation: %.*s%.*s\r\n\r\n",
|
|
|
|
(int) b.len, b.p, (int) (hm->proto.p - hm->uri.p - 1),
|
|
|
|
hm->uri.p);
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void uri_to_path(struct http_message *hm, char *buf, size_t buf_len,
|
|
|
|
const struct mg_serve_http_opts *opts) {
|
2015-09-19 19:38:46 +08:00
|
|
|
char uri[MG_MAX_PATH];
|
2015-09-08 19:49:03 +08:00
|
|
|
struct mg_str a, b, *host_hdr = mg_get_http_header(hm, "Host");
|
|
|
|
const char *rewrites = opts->url_rewrites;
|
2015-05-16 03:20:33 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
mg_url_decode(hm->uri.p, hm->uri.len, uri, sizeof(uri), 0);
|
|
|
|
remove_double_dots(uri);
|
|
|
|
snprintf(buf, buf_len, "%s%s", opts->document_root, uri);
|
2015-05-16 03:20:33 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_DISABLE_DAV
|
2015-09-08 19:49:03 +08:00
|
|
|
if (is_dav_request(&hm->method) && opts->dav_document_root != NULL) {
|
|
|
|
snprintf(buf, buf_len, "%s%s", opts->dav_document_root, uri);
|
|
|
|
}
|
|
|
|
#endif
|
2014-01-20 16:55:34 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Handle URL rewrites */
|
|
|
|
while ((rewrites = mg_next_comma_list_entry(rewrites, &a, &b)) != NULL) {
|
|
|
|
if (a.len > 1 && a.p[0] == '@' && host_hdr != NULL &&
|
|
|
|
host_hdr->len == a.len - 1 &&
|
|
|
|
mg_ncasecmp(a.p + 1, host_hdr->p, a.len - 1) == 0) {
|
|
|
|
/* This is a virtual host rewrite: @domain.name=document_root_dir */
|
|
|
|
snprintf(buf, buf_len, "%.*s%s", (int) b.len, b.p, uri);
|
|
|
|
break;
|
|
|
|
} else {
|
|
|
|
/* This is a usual rewrite, URI=directory */
|
|
|
|
int match_len = mg_match_prefix(a.p, a.len, uri);
|
|
|
|
if (match_len > 0) {
|
|
|
|
snprintf(buf, buf_len, "%.*s%s", (int) b.len, b.p, uri + match_len);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2014-05-18 20:22:33 +08:00
|
|
|
}
|
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_DISABLE_CGI
|
2013-12-09 22:18:49 +08:00
|
|
|
#ifdef _WIN32
|
2014-01-11 05:59:37 +08:00
|
|
|
struct threadparam {
|
|
|
|
sock_t s;
|
|
|
|
HANDLE hPipe;
|
|
|
|
};
|
|
|
|
|
|
|
|
static int wait_until_ready(sock_t sock, int for_read) {
|
|
|
|
fd_set set;
|
|
|
|
FD_ZERO(&set);
|
|
|
|
FD_SET(sock, &set);
|
2015-09-08 19:49:03 +08:00
|
|
|
return select(sock + 1, for_read ? &set : 0, for_read ? 0 : &set, 0, 0) == 1;
|
2014-01-11 05:59:37 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
static void *push_to_stdin(void *arg) {
|
2015-09-08 19:49:03 +08:00
|
|
|
struct threadparam *tp = (struct threadparam *) arg;
|
2014-01-11 05:59:37 +08:00
|
|
|
int n, sent, stop = 0;
|
|
|
|
DWORD k;
|
2015-09-08 19:49:03 +08:00
|
|
|
char buf[BUFSIZ];
|
2014-01-11 05:59:37 +08:00
|
|
|
|
|
|
|
while (!stop && wait_until_ready(tp->s, 1) &&
|
|
|
|
(n = recv(tp->s, buf, sizeof(buf), 0)) > 0) {
|
|
|
|
if (n == -1 && GetLastError() == WSAEWOULDBLOCK) continue;
|
|
|
|
for (sent = 0; !stop && sent < n; sent += k) {
|
|
|
|
if (!WriteFile(tp->hPipe, buf + sent, n - sent, &k, 0)) stop = 1;
|
|
|
|
}
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
DBG(("%s", "FORWARED EVERYTHING TO CGI"));
|
2014-01-11 05:59:37 +08:00
|
|
|
CloseHandle(tp->hPipe);
|
2015-09-19 19:38:46 +08:00
|
|
|
MG_FREE(tp);
|
2014-01-11 05:59:37 +08:00
|
|
|
_endthread();
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void *pull_from_stdout(void *arg) {
|
2015-09-08 19:49:03 +08:00
|
|
|
struct threadparam *tp = (struct threadparam *) arg;
|
2014-06-16 17:44:56 +08:00
|
|
|
int k = 0, stop = 0;
|
2014-01-11 05:59:37 +08:00
|
|
|
DWORD n, sent;
|
2015-09-08 19:49:03 +08:00
|
|
|
char buf[BUFSIZ];
|
2014-01-11 05:59:37 +08:00
|
|
|
|
|
|
|
while (!stop && ReadFile(tp->hPipe, buf, sizeof(buf), &n, NULL)) {
|
|
|
|
for (sent = 0; !stop && sent < n; sent += k) {
|
|
|
|
if (wait_until_ready(tp->s, 0) &&
|
2015-09-08 19:49:03 +08:00
|
|
|
(k = send(tp->s, buf + sent, n - sent, 0)) <= 0)
|
|
|
|
stop = 1;
|
2014-01-11 05:59:37 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
DBG(("%s", "EOF FROM CGI"));
|
|
|
|
CloseHandle(tp->hPipe);
|
|
|
|
shutdown(tp->s, 2); // Without this, IO thread may get truncated data
|
|
|
|
closesocket(tp->s);
|
2015-09-19 19:38:46 +08:00
|
|
|
MG_FREE(tp);
|
2014-01-11 05:59:37 +08:00
|
|
|
_endthread();
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2014-01-15 15:10:09 +08:00
|
|
|
static void spawn_stdio_thread(sock_t sock, HANDLE hPipe,
|
|
|
|
void *(*func)(void *)) {
|
2015-09-19 19:38:46 +08:00
|
|
|
struct threadparam *tp = (struct threadparam *) MG_MALLOC(sizeof(*tp));
|
2014-01-11 05:59:37 +08:00
|
|
|
if (tp != NULL) {
|
|
|
|
tp->s = sock;
|
|
|
|
tp->hPipe = hPipe;
|
|
|
|
mg_start_thread(func, tp);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-01-12 07:17:17 +08:00
|
|
|
static void abs_path(const char *utf8_path, char *abs_path, size_t len) {
|
|
|
|
wchar_t buf[MAX_PATH_SIZE], buf2[MAX_PATH_SIZE];
|
|
|
|
to_wchar(utf8_path, buf, ARRAY_SIZE(buf));
|
|
|
|
GetFullPathNameW(buf, ARRAY_SIZE(buf2), buf2, NULL);
|
|
|
|
WideCharToMultiByte(CP_UTF8, 0, buf2, wcslen(buf2) + 1, abs_path, len, 0, 0);
|
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static pid_t start_process(const char *interp, const char *cmd, const char *env,
|
|
|
|
const char *envp[], const char *dir, sock_t sock) {
|
2014-08-11 21:27:23 +08:00
|
|
|
STARTUPINFOW si;
|
|
|
|
PROCESS_INFORMATION pi;
|
2014-01-11 05:59:37 +08:00
|
|
|
HANDLE a[2], b[2], me = GetCurrentProcess();
|
2014-01-12 07:17:17 +08:00
|
|
|
wchar_t wcmd[MAX_PATH_SIZE], full_dir[MAX_PATH_SIZE];
|
2015-09-08 19:49:03 +08:00
|
|
|
char buf[MAX_PATH_SIZE], buf2[MAX_PATH_SIZE], buf5[MAX_PATH_SIZE],
|
|
|
|
buf4[MAX_PATH_SIZE], cmdline[MAX_PATH_SIZE];
|
2014-01-11 05:59:37 +08:00
|
|
|
DWORD flags = DUPLICATE_CLOSE_SOURCE | DUPLICATE_SAME_ACCESS;
|
2013-12-09 22:18:49 +08:00
|
|
|
FILE *fp;
|
2013-11-23 15:46:11 +08:00
|
|
|
|
2014-08-11 21:27:23 +08:00
|
|
|
memset(&si, 0, sizeof(si));
|
|
|
|
memset(&pi, 0, sizeof(pi));
|
|
|
|
|
2013-12-09 22:18:49 +08:00
|
|
|
si.cb = sizeof(si);
|
|
|
|
si.dwFlags = STARTF_USESTDHANDLES | STARTF_USESHOWWINDOW;
|
|
|
|
si.wShowWindow = SW_HIDE;
|
|
|
|
si.hStdError = GetStdHandle(STD_ERROR_HANDLE);
|
2014-01-11 05:59:37 +08:00
|
|
|
|
|
|
|
CreatePipe(&a[0], &a[1], NULL, 0);
|
|
|
|
CreatePipe(&b[0], &b[1], NULL, 0);
|
|
|
|
DuplicateHandle(me, a[0], me, &si.hStdInput, 0, TRUE, flags);
|
|
|
|
DuplicateHandle(me, b[1], me, &si.hStdOutput, 0, TRUE, flags);
|
2013-11-23 18:24:25 +08:00
|
|
|
|
2013-12-09 22:18:49 +08:00
|
|
|
if (interp == NULL && (fp = fopen(cmd, "r")) != NULL) {
|
|
|
|
buf[0] = buf[1] = '\0';
|
|
|
|
fgets(buf, sizeof(buf), fp);
|
|
|
|
buf[sizeof(buf) - 1] = '\0';
|
|
|
|
if (buf[0] == '#' && buf[1] == '!') {
|
|
|
|
interp = buf + 2;
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Trim leading spaces: https://github.com/cesanta/mongoose/issues/489 */
|
|
|
|
while (*interp != '\0' && isspace(*(unsigned char *) interp)) {
|
|
|
|
interp++;
|
|
|
|
}
|
2013-11-23 18:24:25 +08:00
|
|
|
}
|
2013-12-09 22:18:49 +08:00
|
|
|
fclose(fp);
|
2013-11-23 18:24:25 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
snprintf(buf, sizeof(buf), "%s/%s", dir, cmd);
|
|
|
|
abs_path(buf, buf2, ARRAY_SIZE(buf2));
|
|
|
|
|
|
|
|
abs_path(dir, buf5, ARRAY_SIZE(buf5));
|
|
|
|
to_wchar(dir, full_dir, ARRAY_SIZE(full_dir));
|
|
|
|
|
2013-12-09 22:18:49 +08:00
|
|
|
if (interp != NULL) {
|
2014-01-12 07:17:17 +08:00
|
|
|
abs_path(interp, buf4, ARRAY_SIZE(buf4));
|
2015-09-08 19:49:03 +08:00
|
|
|
snprintf(cmdline, sizeof(cmdline), "%s \"%s\"", buf4, buf2);
|
|
|
|
} else {
|
|
|
|
snprintf(cmdline, sizeof(cmdline), "\"%s\"", buf2);
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
2014-01-12 07:17:17 +08:00
|
|
|
to_wchar(cmdline, wcmd, ARRAY_SIZE(wcmd));
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
#if 0
|
|
|
|
printf("[%ls] [%ls]\n", full_dir, wcmd);
|
|
|
|
#endif
|
|
|
|
|
2014-01-12 07:17:17 +08:00
|
|
|
if (CreateProcessW(NULL, wcmd, NULL, NULL, TRUE, CREATE_NEW_PROCESS_GROUP,
|
|
|
|
(void *) env, full_dir, &si, &pi) != 0) {
|
2014-01-11 05:59:37 +08:00
|
|
|
spawn_stdio_thread(sock, a[1], push_to_stdin);
|
|
|
|
spawn_stdio_thread(sock, b[0], pull_from_stdout);
|
|
|
|
} else {
|
|
|
|
CloseHandle(a[1]);
|
|
|
|
CloseHandle(b[0]);
|
|
|
|
closesocket(sock);
|
|
|
|
}
|
2014-01-12 07:17:17 +08:00
|
|
|
DBG(("CGI command: [%ls] -> %p", wcmd, pi.hProcess));
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Not closing a[0] and b[1] because we've used DUPLICATE_CLOSE_SOURCE */
|
2013-12-09 22:18:49 +08:00
|
|
|
CloseHandle(si.hStdOutput);
|
|
|
|
CloseHandle(si.hStdInput);
|
2015-09-08 19:49:03 +08:00
|
|
|
/* TODO(lsm): check if we need close process and thread handles too */
|
|
|
|
/* CloseHandle(pi.hThread); */
|
|
|
|
/* CloseHandle(pi.hProcess); */
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2014-03-28 17:18:38 +08:00
|
|
|
return pi.hProcess;
|
2013-11-23 18:24:25 +08:00
|
|
|
}
|
2013-12-09 22:18:49 +08:00
|
|
|
#else
|
2015-09-08 19:49:03 +08:00
|
|
|
static pid_t start_process(const char *interp, const char *cmd, const char *env,
|
|
|
|
const char *envp[], const char *dir, sock_t sock) {
|
2014-01-09 18:53:25 +08:00
|
|
|
char buf[500];
|
2015-09-08 19:49:03 +08:00
|
|
|
pid_t pid = fork();
|
2013-12-09 22:18:49 +08:00
|
|
|
(void) env;
|
|
|
|
|
|
|
|
if (pid == 0) {
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* In Linux `chdir` declared with `warn_unused_result` attribute
|
|
|
|
* To shutup compiler we have yo use result in some way
|
|
|
|
*/
|
|
|
|
int tmp = chdir(dir);
|
|
|
|
(void) tmp;
|
2014-01-25 17:45:47 +08:00
|
|
|
(void) dup2(sock, 0);
|
|
|
|
(void) dup2(sock, 1);
|
2013-12-09 22:18:49 +08:00
|
|
|
closesocket(sock);
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* After exec, all signal handlers are restored to their default values,
|
|
|
|
* with one exception of SIGCHLD. According to POSIX.1-2001 and Linux's
|
|
|
|
* implementation, SIGCHLD's handler will leave unchanged after exec
|
|
|
|
* if it was set to be ignored. Restore it to default action.
|
|
|
|
*/
|
2013-12-09 22:18:49 +08:00
|
|
|
signal(SIGCHLD, SIG_DFL);
|
2013-11-23 18:26:28 +08:00
|
|
|
|
2013-12-09 22:18:49 +08:00
|
|
|
if (interp == NULL) {
|
2015-09-08 19:49:03 +08:00
|
|
|
execle(cmd, cmd, (char *) 0, envp); /* (char *) 0 to squash warning */
|
2013-11-23 18:26:28 +08:00
|
|
|
} else {
|
2014-06-17 20:28:45 +08:00
|
|
|
execle(interp, interp, cmd, (char *) 0, envp);
|
2013-11-23 18:26:28 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
snprintf(buf, sizeof(buf),
|
|
|
|
"Status: 500\r\n\r\n"
|
|
|
|
"500 Server Error: %s%s%s: %s",
|
|
|
|
interp == NULL ? "" : interp, interp == NULL ? "" : " ", cmd,
|
|
|
|
strerror(errno));
|
2014-01-09 18:53:25 +08:00
|
|
|
send(1, buf, strlen(buf), 0);
|
2015-09-08 19:49:03 +08:00
|
|
|
exit(EXIT_FAILURE); /* exec call failed */
|
2013-11-23 18:26:28 +08:00
|
|
|
}
|
|
|
|
|
2013-12-09 22:18:49 +08:00
|
|
|
return pid;
|
2013-11-23 18:26:28 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
#endif /* _WIN32 */
|
2013-11-23 18:25:20 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Append VARIABLE=VALUE\0 string to the buffer, and add a respective
|
|
|
|
* pointer into the vars array.
|
|
|
|
*/
|
2013-12-09 22:18:49 +08:00
|
|
|
static char *addenv(struct cgi_env_block *block, const char *fmt, ...) {
|
|
|
|
int n, space;
|
2015-09-08 19:49:03 +08:00
|
|
|
char *added = block->buf + block->len;
|
2013-12-09 22:18:49 +08:00
|
|
|
va_list ap;
|
2013-11-23 18:25:20 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Calculate how much space is left in the buffer */
|
|
|
|
space = sizeof(block->buf) - (block->len + 2);
|
|
|
|
if (space > 0) {
|
|
|
|
/* Copy VARIABLE=VALUE\0 string into the free space */
|
|
|
|
va_start(ap, fmt);
|
|
|
|
n = vsnprintf(added, (size_t) space, fmt, ap);
|
|
|
|
va_end(ap);
|
2013-11-23 18:32:43 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Make sure we do not overflow buffer and the envp array */
|
|
|
|
if (n > 0 && n + 1 < space &&
|
|
|
|
block->nvars < (int) ARRAY_SIZE(block->vars) - 2) {
|
|
|
|
/* Append a pointer to the added string into the envp array */
|
|
|
|
block->vars[block->nvars++] = added;
|
|
|
|
/* Bump up used length counter. Include \0 terminator */
|
|
|
|
block->len += n + 1;
|
|
|
|
}
|
2013-11-23 18:32:43 +08:00
|
|
|
}
|
|
|
|
|
2013-12-09 22:18:49 +08:00
|
|
|
return added;
|
2013-10-03 06:21:13 +08:00
|
|
|
}
|
|
|
|
|
2013-12-09 22:18:49 +08:00
|
|
|
static void addenv2(struct cgi_env_block *blk, const char *name) {
|
|
|
|
const char *s;
|
|
|
|
if ((s = getenv(name)) != NULL) addenv(blk, "%s=%s", name, s);
|
2013-10-03 06:21:13 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void prepare_cgi_environment(struct mg_connection *nc, const char *prog,
|
|
|
|
const struct http_message *hm,
|
|
|
|
const struct mg_serve_http_opts *opts,
|
2013-12-09 22:18:49 +08:00
|
|
|
struct cgi_env_block *blk) {
|
|
|
|
const char *s, *slash;
|
2015-09-08 19:49:03 +08:00
|
|
|
struct mg_str *h;
|
|
|
|
char *p;
|
|
|
|
size_t i;
|
2013-10-01 21:54:26 +08:00
|
|
|
|
2013-12-09 22:18:49 +08:00
|
|
|
blk->len = blk->nvars = 0;
|
2015-09-08 19:49:03 +08:00
|
|
|
blk->nc = nc;
|
2013-10-01 21:54:26 +08:00
|
|
|
|
2014-02-04 01:54:31 +08:00
|
|
|
if ((s = getenv("SERVER_NAME")) != NULL) {
|
|
|
|
addenv(blk, "SERVER_NAME=%s", s);
|
|
|
|
} else {
|
2015-09-08 19:49:03 +08:00
|
|
|
char buf[100];
|
|
|
|
mg_sock_to_str(nc->sock, buf, sizeof(buf), 3);
|
|
|
|
addenv(blk, "SERVER_NAME=%s", buf);
|
2014-02-04 01:54:31 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
addenv(blk, "SERVER_ROOT=%s", opts->document_root);
|
|
|
|
addenv(blk, "DOCUMENT_ROOT=%s", opts->document_root);
|
2015-09-21 22:16:16 +08:00
|
|
|
addenv(blk, "SERVER_SOFTWARE=%s/%s", "Mongoose", MG_VERSION);
|
2013-10-01 21:54:26 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Prepare the environment block */
|
2013-12-09 22:18:49 +08:00
|
|
|
addenv(blk, "%s", "GATEWAY_INTERFACE=CGI/1.1");
|
|
|
|
addenv(blk, "%s", "SERVER_PROTOCOL=HTTP/1.1");
|
2015-09-08 19:49:03 +08:00
|
|
|
addenv(blk, "%s", "REDIRECT_STATUS=200"); /* For PHP */
|
2013-10-01 21:54:26 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* TODO(lsm): fix this for IPv6 case */
|
|
|
|
/*addenv(blk, "SERVER_PORT=%d", ri->remote_port); */
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
addenv(blk, "REQUEST_METHOD=%.*s", (int) hm->method.len, hm->method.p);
|
|
|
|
#if 0
|
2013-12-09 22:18:49 +08:00
|
|
|
addenv(blk, "REMOTE_ADDR=%s", ri->remote_ip);
|
|
|
|
addenv(blk, "REMOTE_PORT=%d", ri->remote_port);
|
2015-09-08 19:49:03 +08:00
|
|
|
#endif
|
|
|
|
addenv(blk, "REQUEST_URI=%.*s%s%.*s", (int) hm->uri.len, hm->uri.p,
|
|
|
|
hm->query_string.len == 0 ? "" : "?", (int) hm->query_string.len,
|
|
|
|
hm->query_string.p);
|
2013-10-01 21:54:26 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* SCRIPT_NAME */
|
|
|
|
#if 0
|
|
|
|
if (nc->path_info != NULL) {
|
2013-12-09 22:18:49 +08:00
|
|
|
addenv(blk, "SCRIPT_NAME=%.*s",
|
2015-09-08 19:49:03 +08:00
|
|
|
(int) (strlen(ri->uri) - strlen(nc->path_info)), ri->uri);
|
|
|
|
addenv(blk, "PATH_INFO=%s", nc->path_info);
|
2013-10-01 21:54:26 +08:00
|
|
|
} else {
|
2015-09-08 19:49:03 +08:00
|
|
|
#endif
|
|
|
|
s = strrchr(prog, '/');
|
|
|
|
slash = hm->uri.p + hm->uri.len;
|
|
|
|
while (slash > hm->uri.p && *slash != '/') {
|
|
|
|
slash--;
|
2013-09-30 20:11:24 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
addenv(blk, "SCRIPT_NAME=%.*s%s", (int) (slash - hm->uri.p), hm->uri.p,
|
|
|
|
s == NULL ? prog : s);
|
|
|
|
#if 0
|
|
|
|
}
|
|
|
|
#endif
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2013-12-09 22:18:49 +08:00
|
|
|
addenv(blk, "SCRIPT_FILENAME=%s", prog);
|
|
|
|
addenv(blk, "PATH_TRANSLATED=%s", prog);
|
2015-09-08 19:49:03 +08:00
|
|
|
addenv(blk, "HTTPS=%s", nc->ssl != NULL ? "on" : "off");
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if ((h = mg_get_http_header((struct http_message *) hm, "Content-Type")) !=
|
|
|
|
NULL) {
|
|
|
|
addenv(blk, "CONTENT_TYPE=%.*s", (int) h->len, h->p);
|
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (hm->query_string.len > 0) {
|
|
|
|
addenv(blk, "QUERY_STRING=%.*s", (int) hm->query_string.len,
|
|
|
|
hm->query_string.p);
|
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if ((h = mg_get_http_header((struct http_message *) hm, "Content-Length")) !=
|
|
|
|
NULL) {
|
|
|
|
addenv(blk, "CONTENT_LENGTH=%.*s", (int) h->len, h->p);
|
|
|
|
}
|
2013-10-01 21:54:26 +08:00
|
|
|
|
2013-12-09 22:18:49 +08:00
|
|
|
addenv2(blk, "PATH");
|
2014-02-04 01:54:31 +08:00
|
|
|
addenv2(blk, "TMP");
|
|
|
|
addenv2(blk, "TEMP");
|
2014-02-05 02:11:56 +08:00
|
|
|
addenv2(blk, "TMPDIR");
|
2013-12-09 22:18:49 +08:00
|
|
|
addenv2(blk, "PERLLIB");
|
2015-09-19 19:38:46 +08:00
|
|
|
addenv2(blk, MG_ENV_EXPORT_TO_CGI);
|
2013-10-01 21:54:26 +08:00
|
|
|
|
2013-12-09 22:18:49 +08:00
|
|
|
#if defined(_WIN32)
|
|
|
|
addenv2(blk, "COMSPEC");
|
|
|
|
addenv2(blk, "SYSTEMROOT");
|
|
|
|
addenv2(blk, "SystemDrive");
|
|
|
|
addenv2(blk, "ProgramFiles");
|
|
|
|
addenv2(blk, "ProgramFiles(x86)");
|
|
|
|
addenv2(blk, "CommonProgramFiles(x86)");
|
|
|
|
#else
|
|
|
|
addenv2(blk, "LD_LIBRARY_PATH");
|
2015-09-08 19:49:03 +08:00
|
|
|
#endif /* _WIN32 */
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Add all headers as HTTP_* variables */
|
|
|
|
for (i = 0; hm->header_names[i].len > 0; i++) {
|
|
|
|
p = addenv(blk, "HTTP_%.*s=%.*s", (int) hm->header_names[i].len,
|
|
|
|
hm->header_names[i].p, (int) hm->header_values[i].len,
|
|
|
|
hm->header_values[i].p);
|
2013-10-01 21:54:26 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Convert variable name into uppercase, and change - to _ */
|
2013-12-09 22:18:49 +08:00
|
|
|
for (; *p != '=' && *p != '\0'; p++) {
|
2015-09-08 19:49:03 +08:00
|
|
|
if (*p == '-') *p = '_';
|
|
|
|
*p = (char) toupper(*(unsigned char *) p);
|
2013-10-01 21:54:26 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-12-09 22:18:49 +08:00
|
|
|
blk->vars[blk->nvars++] = NULL;
|
|
|
|
blk->buf[blk->len++] = '\0';
|
2013-09-30 20:11:24 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void cgi_ev_handler(struct mg_connection *cgi_nc, int ev,
|
|
|
|
void *ev_data) {
|
|
|
|
struct mg_connection *nc = (struct mg_connection *) cgi_nc->user_data;
|
|
|
|
(void) ev_data;
|
|
|
|
|
|
|
|
if (nc == NULL) return;
|
|
|
|
|
|
|
|
switch (ev) {
|
2015-09-19 19:38:46 +08:00
|
|
|
case MG_EV_RECV:
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* CGI script does not output reply line, like "HTTP/1.1 CODE XXXXX\n"
|
|
|
|
* It outputs headers, then body. Headers might include "Status"
|
|
|
|
* header, which changes CODE, and it might include "Location" header
|
|
|
|
* which changes CODE to 302.
|
|
|
|
*
|
|
|
|
* Therefore we do not send the output from the CGI script to the user
|
2015-11-16 04:34:40 +08:00
|
|
|
* until all CGI headers are received.
|
2015-09-08 19:49:03 +08:00
|
|
|
*
|
|
|
|
* Here we parse the output from the CGI script, and if all headers has
|
2015-11-16 04:34:40 +08:00
|
|
|
* been received, send appropriate reply line, and forward all
|
|
|
|
* received headers to the client.
|
2015-09-08 19:49:03 +08:00
|
|
|
*/
|
2015-09-19 19:38:46 +08:00
|
|
|
if (nc->flags & MG_F_USER_1) {
|
2015-09-08 19:49:03 +08:00
|
|
|
struct mbuf *io = &cgi_nc->recv_mbuf;
|
|
|
|
int len = get_request_len(io->buf, io->len);
|
|
|
|
|
|
|
|
if (len == 0) break;
|
2015-09-19 19:38:46 +08:00
|
|
|
if (len < 0 || io->len > MG_MAX_HTTP_REQUEST_SIZE) {
|
|
|
|
cgi_nc->flags |= MG_F_CLOSE_IMMEDIATELY;
|
2015-09-08 19:49:03 +08:00
|
|
|
send_http_error(nc, 500, "Bad headers");
|
|
|
|
} else {
|
|
|
|
struct http_message hm;
|
|
|
|
struct mg_str *h;
|
|
|
|
parse_http_headers(io->buf, io->buf + io->len, io->len, &hm);
|
|
|
|
/*printf("=== %d [%.*s]\n", k, k, io->buf);*/
|
|
|
|
if (mg_get_http_header(&hm, "Location") != NULL) {
|
|
|
|
mg_printf(nc, "%s", "HTTP/1.1 302 Moved\r\n");
|
|
|
|
} else if ((h = mg_get_http_header(&hm, "Status")) != NULL) {
|
|
|
|
mg_printf(nc, "HTTP/1.1 %.*s\r\n", (int) h->len, h->p);
|
|
|
|
} else {
|
|
|
|
mg_printf(nc, "%s", "HTTP/1.1 200 OK\r\n");
|
|
|
|
}
|
|
|
|
}
|
2015-09-19 19:38:46 +08:00
|
|
|
nc->flags &= ~MG_F_USER_1;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-09-19 19:38:46 +08:00
|
|
|
if (!(nc->flags & MG_F_USER_1)) {
|
2015-09-08 19:49:03 +08:00
|
|
|
mg_forward(cgi_nc, nc);
|
|
|
|
}
|
|
|
|
break;
|
2015-09-19 19:38:46 +08:00
|
|
|
case MG_EV_CLOSE:
|
2015-10-21 17:05:13 +08:00
|
|
|
free_http_proto_data(cgi_nc);
|
2015-09-19 19:38:46 +08:00
|
|
|
nc->flags |= MG_F_SEND_AND_CLOSE;
|
2015-09-08 19:49:03 +08:00
|
|
|
nc->user_data = NULL;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
2014-01-11 05:59:37 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void handle_cgi(struct mg_connection *nc, const char *prog,
|
|
|
|
const struct http_message *hm,
|
|
|
|
const struct mg_serve_http_opts *opts) {
|
|
|
|
struct proto_data_http *dp;
|
2013-12-09 22:18:49 +08:00
|
|
|
struct cgi_env_block blk;
|
2014-02-20 00:44:01 +08:00
|
|
|
char dir[MAX_PATH_SIZE];
|
|
|
|
const char *p;
|
2013-12-09 22:18:49 +08:00
|
|
|
sock_t fds[2];
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
prepare_cgi_environment(nc, prog, hm, opts, &blk);
|
|
|
|
/*
|
|
|
|
* CGI must be executed in its own directory. 'dir' must point to the
|
|
|
|
* directory containing executable program, 'p' must point to the
|
|
|
|
* executable program name relative to 'dir'.
|
|
|
|
*/
|
2014-01-29 19:42:14 +08:00
|
|
|
if ((p = strrchr(prog, '/')) == NULL) {
|
2015-09-08 19:49:03 +08:00
|
|
|
snprintf(dir, sizeof(dir), "%s", ".");
|
2014-01-29 19:42:14 +08:00
|
|
|
} else {
|
2015-09-08 19:49:03 +08:00
|
|
|
snprintf(dir, sizeof(dir), "%.*s", (int) (p - prog), prog);
|
|
|
|
prog = p + 1;
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Try to create socketpair in a loop until success. mg_socketpair()
|
|
|
|
* can be interrupted by a signal and fail.
|
|
|
|
* TODO(lsm): use sigaction to restart interrupted syscall
|
|
|
|
*/
|
|
|
|
do {
|
|
|
|
mg_socketpair(fds, SOCK_STREAM);
|
|
|
|
} while (fds[0] == INVALID_SOCKET);
|
|
|
|
|
|
|
|
free_http_proto_data(nc);
|
2015-09-19 19:38:46 +08:00
|
|
|
if ((dp = (struct proto_data_http *) MG_CALLOC(1, sizeof(*dp))) == NULL) {
|
2015-09-08 19:49:03 +08:00
|
|
|
send_http_error(nc, 500, "OOM"); /* LCOV_EXCL_LINE */
|
|
|
|
} else if (start_process(opts->cgi_interpreter, prog, blk.buf, blk.vars, dir,
|
|
|
|
fds[1]) != 0) {
|
|
|
|
size_t n = nc->recv_mbuf.len - (hm->message.len - hm->body.len);
|
|
|
|
dp->type = DATA_CGI;
|
|
|
|
dp->cgi_nc = mg_add_sock(nc->mgr, fds[0], cgi_ev_handler);
|
|
|
|
dp->cgi_nc->user_data = nc;
|
2015-10-21 17:05:13 +08:00
|
|
|
dp->cgi_nc->proto_data = dp;
|
2015-09-19 19:38:46 +08:00
|
|
|
nc->flags |= MG_F_USER_1;
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Push POST data to the CGI */
|
|
|
|
if (n > 0 && n < nc->recv_mbuf.len) {
|
|
|
|
mg_send(dp->cgi_nc, hm->body.p, n);
|
|
|
|
}
|
|
|
|
mbuf_remove(&nc->recv_mbuf, nc->recv_mbuf.len);
|
2013-12-09 22:18:49 +08:00
|
|
|
} else {
|
|
|
|
closesocket(fds[0]);
|
2015-09-08 19:49:03 +08:00
|
|
|
send_http_error(nc, 500, "CGI failure");
|
2013-10-01 21:54:26 +08:00
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2014-01-11 05:59:37 +08:00
|
|
|
#ifndef _WIN32
|
2015-09-08 19:49:03 +08:00
|
|
|
closesocket(fds[1]); /* On Windows, CGI stdio thread closes that socket */
|
2014-01-11 05:59:37 +08:00
|
|
|
#endif
|
2013-09-30 20:11:24 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
#endif
|
2014-01-10 01:50:57 +08:00
|
|
|
|
2015-10-31 21:44:54 +08:00
|
|
|
static int mg_get_month_index(const char *s) {
|
2015-10-31 19:54:38 +08:00
|
|
|
static const char *month_names[] = {"Jan", "Feb", "Mar", "Apr", "May", "Jun",
|
|
|
|
"Jul", "Aug", "Sep", "Oct", "Nov", "Dec"};
|
|
|
|
size_t i;
|
|
|
|
|
|
|
|
for (i = 0; i < ARRAY_SIZE(month_names); i++)
|
|
|
|
if (!strcmp(s, month_names[i])) return (int) i;
|
|
|
|
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2015-10-31 21:44:54 +08:00
|
|
|
static int mg_num_leap_years(int year) {
|
2015-10-31 19:54:38 +08:00
|
|
|
return year / 4 - year / 100 + year / 400;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Parse UTC date-time string, and return the corresponding time_t value. */
|
2015-10-31 21:44:54 +08:00
|
|
|
MG_INTERNAL time_t mg_parse_date_string(const char *datetime) {
|
2015-10-31 19:54:38 +08:00
|
|
|
static const unsigned short days_before_month[] = {
|
|
|
|
0, 31, 59, 90, 120, 151, 181, 212, 243, 273, 304, 334};
|
|
|
|
char month_str[32];
|
|
|
|
int second, minute, hour, day, month, year, leap_days, days;
|
|
|
|
time_t result = (time_t) 0;
|
|
|
|
|
|
|
|
if (((sscanf(datetime, "%d/%3s/%d %d:%d:%d", &day, month_str, &year, &hour,
|
|
|
|
&minute, &second) == 6) ||
|
|
|
|
(sscanf(datetime, "%d %3s %d %d:%d:%d", &day, month_str, &year, &hour,
|
|
|
|
&minute, &second) == 6) ||
|
|
|
|
(sscanf(datetime, "%*3s, %d %3s %d %d:%d:%d", &day, month_str, &year,
|
|
|
|
&hour, &minute, &second) == 6) ||
|
|
|
|
(sscanf(datetime, "%d-%3s-%d %d:%d:%d", &day, month_str, &year, &hour,
|
|
|
|
&minute, &second) == 6)) &&
|
2015-10-31 21:44:54 +08:00
|
|
|
year > 1970 && (month = mg_get_month_index(month_str)) != -1) {
|
|
|
|
leap_days = mg_num_leap_years(year) - mg_num_leap_years(1970);
|
2015-10-31 19:54:38 +08:00
|
|
|
year -= 1970;
|
|
|
|
days = year * 365 + days_before_month[month] + (day - 1) + leap_days;
|
|
|
|
result = days * 24 * 3600 + hour * 3600 + minute * 60 + second;
|
|
|
|
}
|
|
|
|
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
2015-11-18 06:25:04 +08:00
|
|
|
MG_INTERNAL int mg_is_not_modified(struct http_message *hm, cs_stat_t *st) {
|
|
|
|
struct mg_str *hdr;
|
|
|
|
if ((hdr = mg_get_http_header(hm, "If-None-Match")) != NULL) {
|
|
|
|
char etag[64];
|
|
|
|
construct_etag(etag, sizeof(etag), st);
|
|
|
|
return mg_vcasecmp(hdr, etag) == 0;
|
|
|
|
} else if ((hdr = mg_get_http_header(hm, "If-Modified-Since")) != NULL) {
|
|
|
|
return st->st_mtime <= mg_parse_date_string(hdr->p);
|
|
|
|
} else {
|
|
|
|
return 0;
|
|
|
|
}
|
2015-10-31 19:54:38 +08:00
|
|
|
}
|
|
|
|
|
2015-10-20 17:31:23 +08:00
|
|
|
static void mg_send_digest_auth_request(struct mg_connection *c,
|
|
|
|
const char *domain) {
|
|
|
|
mg_printf(c,
|
|
|
|
"HTTP/1.1 401 Unauthorized\r\n"
|
|
|
|
"WWW-Authenticate: Digest qop=\"auth\", "
|
|
|
|
"realm=\"%s\", nonce=\"%lu\"\r\n"
|
|
|
|
"Content-Length: 0\r\n\r\n",
|
|
|
|
domain, (unsigned long) time(NULL));
|
|
|
|
}
|
|
|
|
|
2015-12-08 04:18:06 +08:00
|
|
|
static void send_options(struct mg_connection *nc) {
|
|
|
|
mg_printf(nc, "%s",
|
2015-12-11 00:39:09 +08:00
|
|
|
"HTTP/1.1 200 OK\r\nAllow: GET, POST, HEAD, CONNECT, OPTIONS"
|
2015-12-08 04:18:06 +08:00
|
|
|
#ifndef MG_DISABLE_DAV
|
2015-12-11 00:39:09 +08:00
|
|
|
", MKCOL, PUT, DELETE, PROPFIND, MOVE\r\nDAV: 1,2"
|
2015-12-08 04:18:06 +08:00
|
|
|
#endif
|
|
|
|
"\r\n\r\n");
|
|
|
|
nc->flags |= MG_F_SEND_AND_CLOSE;
|
|
|
|
}
|
|
|
|
|
2015-12-11 14:22:15 +08:00
|
|
|
static int is_creation_request(const struct http_message *hm) {
|
|
|
|
return mg_vcmp(&hm->method, "MKCOL") == 0 || mg_vcmp(&hm->method, "PUT") == 0;
|
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_send_http_file(struct mg_connection *nc, char *path,
|
|
|
|
size_t path_buf_len, struct http_message *hm,
|
|
|
|
struct mg_serve_http_opts *opts) {
|
|
|
|
int stat_result, is_directory, is_dav = is_dav_request(&hm->method);
|
|
|
|
uint32_t remote_ip = ntohl(*(uint32_t *) &nc->sa.sin.sin_addr);
|
|
|
|
cs_stat_t st;
|
2014-02-20 00:34:51 +08:00
|
|
|
|
2015-11-12 20:48:17 +08:00
|
|
|
DBG(("serving [%s]", path));
|
2015-09-08 19:49:03 +08:00
|
|
|
stat_result = mg_stat(path, &st);
|
|
|
|
is_directory = !stat_result && S_ISDIR(st.st_mode);
|
2014-02-20 00:34:51 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (mg_check_ip_acl(opts->ip_acl, remote_ip) != 1) {
|
|
|
|
/* Not allowed to connect */
|
2015-09-19 19:38:46 +08:00
|
|
|
nc->flags |= MG_F_CLOSE_IMMEDIATELY;
|
2015-09-08 19:49:03 +08:00
|
|
|
} else if (is_dav && opts->dav_document_root == NULL) {
|
|
|
|
send_http_error(nc, 501, NULL);
|
2015-10-20 17:31:23 +08:00
|
|
|
} else if (!is_authorized(hm, path, is_directory, opts->auth_domain,
|
|
|
|
opts->global_auth_file, 1) ||
|
|
|
|
!is_authorized(hm, path, is_directory, opts->auth_domain,
|
|
|
|
opts->per_directory_auth_file, 0)) {
|
|
|
|
mg_send_digest_auth_request(nc, opts->auth_domain);
|
2015-12-11 14:22:15 +08:00
|
|
|
} else if ((stat_result != 0 || is_file_hidden(path, opts)) &&
|
|
|
|
!is_creation_request(hm)) {
|
2015-09-08 19:49:03 +08:00
|
|
|
mg_printf(nc, "%s", "HTTP/1.1 404 Not Found\r\nContent-Length: 0\r\n\r\n");
|
|
|
|
} else if (is_directory && path[strlen(path) - 1] != '/' && !is_dav) {
|
|
|
|
mg_printf(nc,
|
|
|
|
"HTTP/1.1 301 Moved\r\nLocation: %.*s/\r\n"
|
|
|
|
"Content-Length: 0\r\n\r\n",
|
|
|
|
(int) hm->uri.len, hm->uri.p);
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_DISABLE_DAV
|
2015-09-08 19:49:03 +08:00
|
|
|
} else if (!mg_vcmp(&hm->method, "PROPFIND")) {
|
|
|
|
handle_propfind(nc, path, &st, hm, opts);
|
2015-10-20 17:31:23 +08:00
|
|
|
#ifndef MG_DISABLE_DAV_AUTH
|
|
|
|
} else if (is_dav &&
|
|
|
|
(opts->dav_auth_file == NULL ||
|
2015-12-11 14:25:07 +08:00
|
|
|
(strcmp(opts->dav_auth_file, "-") != 0 &&
|
|
|
|
!is_authorized(hm, path, is_directory, opts->auth_domain,
|
|
|
|
opts->dav_auth_file, 1)))) {
|
2015-10-20 17:31:23 +08:00
|
|
|
mg_send_digest_auth_request(nc, opts->auth_domain);
|
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
} else if (!mg_vcmp(&hm->method, "MKCOL")) {
|
|
|
|
handle_mkcol(nc, path, hm);
|
|
|
|
} else if (!mg_vcmp(&hm->method, "DELETE")) {
|
2015-12-08 22:16:56 +08:00
|
|
|
handle_delete(nc, opts, path);
|
2015-09-08 19:49:03 +08:00
|
|
|
} else if (!mg_vcmp(&hm->method, "PUT")) {
|
|
|
|
handle_put(nc, path, hm);
|
2015-12-09 00:32:17 +08:00
|
|
|
} else if (!mg_vcmp(&hm->method, "MOVE")) {
|
|
|
|
handle_move(nc, opts, path, hm);
|
2016-01-11 18:52:35 +08:00
|
|
|
#ifdef MG_ENABLE_FAKE_DAVLOCK
|
|
|
|
} else if (!mg_vcmp(&hm->method, "LOCK")) {
|
|
|
|
handle_lock(nc, path);
|
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
#endif
|
2015-12-08 04:18:06 +08:00
|
|
|
} else if (!mg_vcmp(&hm->method, "OPTIONS")) {
|
|
|
|
send_options(nc);
|
2015-09-08 19:49:03 +08:00
|
|
|
} else if (S_ISDIR(st.st_mode) &&
|
|
|
|
!find_index_file(path, path_buf_len, opts->index_files, &st)) {
|
|
|
|
if (strcmp(opts->enable_directory_listing, "yes") == 0) {
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_DISABLE_DIRECTORY_LISTING
|
2015-09-08 19:49:03 +08:00
|
|
|
send_directory_listing(nc, path, hm, opts);
|
|
|
|
#else
|
|
|
|
send_http_error(nc, 501, NULL);
|
|
|
|
#endif
|
2014-02-20 00:34:51 +08:00
|
|
|
} else {
|
2015-09-08 19:49:03 +08:00
|
|
|
send_http_error(nc, 403, NULL);
|
2014-01-10 01:50:57 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
} else if (mg_match_prefix(opts->cgi_file_pattern,
|
|
|
|
strlen(opts->cgi_file_pattern), path) > 0) {
|
2015-09-19 19:38:46 +08:00
|
|
|
#if !defined(MG_DISABLE_CGI)
|
2015-09-08 19:49:03 +08:00
|
|
|
handle_cgi(nc, path, hm, opts);
|
|
|
|
#else
|
|
|
|
send_http_error(nc, 501, NULL);
|
2015-09-19 19:38:46 +08:00
|
|
|
#endif /* MG_DISABLE_CGI */
|
2015-10-31 19:54:38 +08:00
|
|
|
} else if (mg_is_not_modified(hm, &st)) {
|
|
|
|
send_http_error(nc, 304, "Not Modified");
|
2015-09-08 19:49:03 +08:00
|
|
|
} else {
|
|
|
|
mg_send_http_file2(nc, path, &st, hm, opts);
|
2014-01-10 01:50:57 +08:00
|
|
|
}
|
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_serve_http(struct mg_connection *nc, struct http_message *hm,
|
|
|
|
struct mg_serve_http_opts opts) {
|
2015-09-19 19:38:46 +08:00
|
|
|
char path[MG_MAX_PATH];
|
2015-10-20 17:31:23 +08:00
|
|
|
struct mg_str *hdr;
|
2015-10-21 17:15:00 +08:00
|
|
|
|
2015-11-27 05:37:20 +08:00
|
|
|
if (send_port_based_redirect(nc, hm, &opts)) {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2015-10-21 17:15:00 +08:00
|
|
|
if (opts.document_root == NULL) {
|
|
|
|
opts.document_root = ".";
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
if (opts.per_directory_auth_file == NULL) {
|
|
|
|
opts.per_directory_auth_file = ".htpasswd";
|
2013-09-30 20:11:24 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
if (opts.enable_directory_listing == NULL) {
|
|
|
|
opts.enable_directory_listing = "yes";
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
if (opts.cgi_file_pattern == NULL) {
|
|
|
|
opts.cgi_file_pattern = "**.cgi$|**.php$";
|
2013-09-30 20:11:24 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
if (opts.ssi_pattern == NULL) {
|
|
|
|
opts.ssi_pattern = "**.shtml$|**.shtm$";
|
2013-09-30 20:11:24 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
if (opts.index_files == NULL) {
|
|
|
|
opts.index_files = "index.html,index.htm,index.shtml,index.cgi,index.php";
|
2013-09-30 20:11:24 +08:00
|
|
|
}
|
2015-10-21 17:15:00 +08:00
|
|
|
|
|
|
|
uri_to_path(hm, path, sizeof(path), &opts);
|
2015-09-08 19:49:03 +08:00
|
|
|
mg_send_http_file(nc, path, sizeof(path), hm, &opts);
|
2015-10-20 17:31:23 +08:00
|
|
|
|
|
|
|
/* Close connection for non-keep-alive requests */
|
|
|
|
if (mg_vcmp(&hm->proto, "HTTP/1.1") != 0 ||
|
|
|
|
((hdr = mg_get_http_header(hm, "Connection")) != NULL &&
|
|
|
|
mg_vcmp(hdr, "keep-alive") != 0)) {
|
|
|
|
#if 0
|
|
|
|
nc->flags |= MG_F_SEND_AND_CLOSE;
|
|
|
|
#endif
|
|
|
|
}
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#endif /* MG_DISABLE_FILESYSTEM */
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
struct mg_connection *mg_connect_http(struct mg_mgr *mgr,
|
|
|
|
mg_event_handler_t ev_handler,
|
|
|
|
const char *url,
|
|
|
|
const char *extra_headers,
|
|
|
|
const char *post_data) {
|
2015-10-24 06:31:54 +08:00
|
|
|
struct mg_connection *nc = NULL;
|
|
|
|
char *addr = NULL;
|
|
|
|
const char *path = NULL;
|
|
|
|
int use_ssl = 0, addr_len = 0, port_i = -1;
|
2013-12-31 17:02:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (memcmp(url, "http://", 7) == 0) {
|
|
|
|
url += 7;
|
|
|
|
} else if (memcmp(url, "https://", 8) == 0) {
|
|
|
|
url += 8;
|
|
|
|
use_ssl = 1;
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_ENABLE_SSL
|
2015-09-08 19:49:03 +08:00
|
|
|
return NULL; /* SSL is not enabled, cannot do HTTPS URLs */
|
|
|
|
#endif
|
|
|
|
}
|
2015-03-17 21:18:15 +08:00
|
|
|
|
2015-10-24 06:31:54 +08:00
|
|
|
while (*url != '\0') {
|
|
|
|
addr = (char *) MG_REALLOC(addr, addr_len + 5 /* space for port too. */);
|
|
|
|
if (addr == NULL) {
|
|
|
|
DBG(("OOM"));
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
if (*url == '/') {
|
|
|
|
url++;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
if (*url == ':') port_i = addr_len;
|
|
|
|
addr[addr_len++] = *url;
|
|
|
|
addr[addr_len] = '\0';
|
|
|
|
url++;
|
|
|
|
}
|
|
|
|
if (addr_len == 0) goto cleanup;
|
|
|
|
if (port_i < 0) {
|
|
|
|
port_i = addr_len;
|
|
|
|
strcpy(addr + port_i, use_ssl ? ":443" : ":80");
|
|
|
|
} else {
|
|
|
|
port_i = -1;
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
|
|
|
|
2015-10-24 06:31:54 +08:00
|
|
|
if (path == NULL) path = url;
|
|
|
|
|
|
|
|
DBG(("%s %s", addr, path));
|
2015-09-08 19:49:03 +08:00
|
|
|
if ((nc = mg_connect(mgr, addr, ev_handler)) != NULL) {
|
|
|
|
mg_set_protocol_http_websocket(nc);
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (use_ssl) {
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifdef MG_ENABLE_SSL
|
2015-09-08 19:49:03 +08:00
|
|
|
mg_set_ssl(nc, NULL, NULL);
|
|
|
|
#endif
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-10-24 06:31:54 +08:00
|
|
|
/* If the port was addred by us, restore the original host. */
|
|
|
|
if (port_i >= 0) addr[port_i] = '\0';
|
2015-09-24 01:33:25 +08:00
|
|
|
mg_printf(nc, "%s /%s HTTP/1.1\r\nHost: %s\r\nContent-Length: %" SIZE_T_FMT
|
|
|
|
"\r\n%s\r\n%s",
|
2015-09-08 19:49:03 +08:00
|
|
|
post_data == NULL ? "GET" : "POST", path, addr,
|
|
|
|
post_data == NULL ? 0 : strlen(post_data),
|
|
|
|
extra_headers == NULL ? "" : extra_headers,
|
|
|
|
post_data == NULL ? "" : post_data);
|
2013-09-30 20:11:24 +08:00
|
|
|
}
|
|
|
|
|
2015-10-24 06:31:54 +08:00
|
|
|
cleanup:
|
|
|
|
MG_FREE(addr);
|
2015-09-08 19:49:03 +08:00
|
|
|
return nc;
|
2014-01-11 17:48:27 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static size_t get_line_len(const char *buf, size_t buf_len) {
|
|
|
|
size_t len = 0;
|
|
|
|
while (len < buf_len && buf[len] != '\n') len++;
|
|
|
|
return buf[len] == '\n' ? len + 1 : 0;
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
2013-11-23 18:49:13 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
size_t mg_parse_multipart(const char *buf, size_t buf_len, char *var_name,
|
|
|
|
size_t var_name_len, char *file_name,
|
|
|
|
size_t file_name_len, const char **data,
|
|
|
|
size_t *data_len) {
|
|
|
|
static const char cd[] = "Content-Disposition: ";
|
|
|
|
size_t hl, bl, n, ll, pos, cdl = sizeof(cd) - 1;
|
2014-03-30 19:45:48 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (buf == NULL || buf_len <= 0) return 0;
|
|
|
|
if ((hl = get_request_len(buf, buf_len)) <= 0) return 0;
|
|
|
|
if (buf[0] != '-' || buf[1] != '-' || buf[2] == '\n') return 0;
|
2015-03-17 23:34:43 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Get boundary length */
|
|
|
|
bl = get_line_len(buf, buf_len);
|
2014-01-23 02:24:28 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Loop through headers, fetch variable name and file name */
|
|
|
|
var_name[0] = file_name[0] = '\0';
|
|
|
|
for (n = bl; (ll = get_line_len(buf + n, hl - n)) > 0; n += ll) {
|
|
|
|
if (mg_ncasecmp(cd, buf + n, cdl) == 0) {
|
|
|
|
struct mg_str header;
|
|
|
|
header.p = buf + n + cdl;
|
|
|
|
header.len = ll - (cdl + 2);
|
|
|
|
mg_http_parse_header(&header, "name", var_name, var_name_len);
|
|
|
|
mg_http_parse_header(&header, "filename", file_name, file_name_len);
|
2014-03-05 00:15:02 +08:00
|
|
|
}
|
|
|
|
}
|
2013-11-23 18:49:13 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Scan through the body, search for terminating boundary */
|
|
|
|
for (pos = hl; pos + (bl - 2) < buf_len; pos++) {
|
|
|
|
if (buf[pos] == '-' && !memcmp(buf, &buf[pos], bl - 2)) {
|
|
|
|
if (data_len != NULL) *data_len = (pos - 2) - hl;
|
|
|
|
if (data != NULL) *data = buf + hl;
|
|
|
|
return pos;
|
2013-11-23 18:49:13 +08:00
|
|
|
}
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
2013-11-23 18:49:13 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return 0;
|
|
|
|
}
|
2013-11-23 18:49:13 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#endif /* MG_DISABLE_HTTP */
|
2015-09-08 19:49:03 +08:00
|
|
|
#ifdef NS_MODULE_LINES
|
2015-12-15 02:45:18 +08:00
|
|
|
#line 1 "./src/util.c"
|
2015-09-08 19:49:03 +08:00
|
|
|
/**/
|
2013-12-09 22:18:49 +08:00
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Copyright (c) 2014 Cesanta Software Limited
|
|
|
|
* All rights reserved
|
|
|
|
*/
|
2013-11-23 18:49:13 +08:00
|
|
|
|
2015-12-12 22:16:48 +08:00
|
|
|
/* Amalgamated: #include "common/base64.h" */
|
2015-12-12 22:05:54 +08:00
|
|
|
/* Amalgamated: #include "mongoose/src/internal.h" */
|
2015-12-12 22:16:48 +08:00
|
|
|
/* Amalgamated: #include "mongoose/src/util.h" */
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
const char *mg_skip(const char *s, const char *end, const char *delims,
|
|
|
|
struct mg_str *v) {
|
|
|
|
v->p = s;
|
|
|
|
while (s < end && strchr(delims, *(unsigned char *) s) == NULL) s++;
|
|
|
|
v->len = s - v->p;
|
|
|
|
while (s < end && strchr(delims, *(unsigned char *) s) != NULL) s++;
|
|
|
|
return s;
|
2013-11-23 18:49:13 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static int lowercase(const char *s) {
|
|
|
|
return tolower(*(const unsigned char *) s);
|
2014-01-11 06:46:03 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
int mg_ncasecmp(const char *s1, const char *s2, size_t len) {
|
|
|
|
int diff = 0;
|
|
|
|
|
|
|
|
if (len > 0) do {
|
|
|
|
diff = lowercase(s1++) - lowercase(s2++);
|
|
|
|
} while (diff == 0 && s1[-1] != '\0' && --len > 0);
|
|
|
|
|
|
|
|
return diff;
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
2013-11-23 18:49:13 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
int mg_casecmp(const char *s1, const char *s2) {
|
|
|
|
return mg_ncasecmp(s1, s2, (size_t) ~0);
|
2014-01-04 19:31:55 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +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->p, str2, (n1 < n2) ? n1 : n2);
|
|
|
|
if (r == 0) {
|
|
|
|
return n1 - n2;
|
2014-01-04 19:31:55 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
return r;
|
2014-01-04 19:31:55 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
int mg_vcmp(const struct mg_str *str1, const char *str2) {
|
|
|
|
size_t n2 = strlen(str2), n1 = str1->len;
|
|
|
|
int r = memcmp(str1->p, str2, (n1 < n2) ? n1 : n2);
|
|
|
|
if (r == 0) {
|
|
|
|
return n1 - n2;
|
2014-01-04 19:31:55 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
return r;
|
2014-01-04 19:31:55 +08:00
|
|
|
}
|
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_DISABLE_FILESYSTEM
|
2015-09-08 19:49:03 +08:00
|
|
|
int mg_stat(const char *path, cs_stat_t *st) {
|
|
|
|
#ifdef _WIN32
|
|
|
|
wchar_t wpath[MAX_PATH_SIZE];
|
|
|
|
to_wchar(path, wpath, ARRAY_SIZE(wpath));
|
|
|
|
DBG(("[%ls] -> %d", wpath, _wstati64(wpath, st)));
|
|
|
|
return _wstati64(wpath, (struct _stati64 *) st);
|
|
|
|
#else
|
|
|
|
return stat(path, st);
|
|
|
|
#endif
|
2014-01-04 19:31:55 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
FILE *mg_fopen(const char *path, const char *mode) {
|
|
|
|
#ifdef _WIN32
|
|
|
|
wchar_t wpath[MAX_PATH_SIZE], wmode[10];
|
|
|
|
to_wchar(path, wpath, ARRAY_SIZE(wpath));
|
|
|
|
to_wchar(mode, wmode, ARRAY_SIZE(wmode));
|
|
|
|
return _wfopen(wpath, wmode);
|
|
|
|
#else
|
|
|
|
return fopen(path, mode);
|
|
|
|
#endif
|
|
|
|
}
|
2015-05-16 03:20:33 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
int mg_open(const char *path, int flag, int mode) { /* LCOV_EXCL_LINE */
|
|
|
|
#ifdef _WIN32
|
|
|
|
wchar_t wpath[MAX_PATH_SIZE];
|
|
|
|
to_wchar(path, wpath, ARRAY_SIZE(wpath));
|
|
|
|
return _wopen(wpath, flag, mode);
|
|
|
|
#else
|
|
|
|
return open(path, flag, mode); /* LCOV_EXCL_LINE */
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
#endif
|
2015-05-16 03:20:33 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_base64_encode(const unsigned char *src, int src_len, char *dst) {
|
|
|
|
cs_base64_encode(src, src_len, dst);
|
2015-05-16 03:20:33 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
int mg_base64_decode(const unsigned char *s, int len, char *dst) {
|
|
|
|
return cs_base64_decode(s, len, dst);
|
|
|
|
}
|
2014-01-04 19:31:55 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifdef MG_ENABLE_THREADS
|
2015-09-08 19:49:03 +08:00
|
|
|
void *mg_start_thread(void *(*f)(void *), void *p) {
|
|
|
|
#ifdef _WIN32
|
|
|
|
return (void *) _beginthread((void(__cdecl *) (void *) ) f, 0, p);
|
|
|
|
#else
|
|
|
|
pthread_t thread_id = (pthread_t) 0;
|
|
|
|
pthread_attr_t attr;
|
2014-01-04 19:31:55 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
(void) pthread_attr_init(&attr);
|
|
|
|
(void) pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED);
|
2014-02-23 00:11:47 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#if defined(MG_STACK_SIZE) && MG_STACK_SIZE > 1
|
|
|
|
(void) pthread_attr_setstacksize(&attr, MG_STACK_SIZE);
|
2015-09-08 19:49:03 +08:00
|
|
|
#endif
|
2014-01-04 19:31:55 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
pthread_create(&thread_id, &attr, f, p);
|
|
|
|
pthread_attr_destroy(&attr);
|
|
|
|
|
|
|
|
return (void *) thread_id;
|
2013-12-09 22:18:49 +08:00
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-09-19 19:38:46 +08:00
|
|
|
#endif /* MG_ENABLE_THREADS */
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Set close-on-exec bit for a given socket. */
|
|
|
|
void mg_set_close_on_exec(sock_t sock) {
|
|
|
|
#ifdef _WIN32
|
|
|
|
(void) SetHandleInformation((HANDLE) sock, HANDLE_FLAG_INHERIT, 0);
|
|
|
|
#else
|
|
|
|
fcntl(sock, F_SETFD, FD_CLOEXEC);
|
2013-11-23 18:47:35 +08:00
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_sock_addr_to_str(const union socket_address *sa, char *buf, size_t len,
|
|
|
|
int flags) {
|
|
|
|
int is_v6;
|
|
|
|
if (buf == NULL || len <= 0) return;
|
|
|
|
buf[0] = '\0';
|
2015-09-19 19:38:46 +08:00
|
|
|
#if defined(MG_ENABLE_IPV6)
|
2015-09-08 19:49:03 +08:00
|
|
|
is_v6 = sa->sa.sa_family == AF_INET6;
|
|
|
|
#else
|
|
|
|
is_v6 = 0;
|
|
|
|
#endif
|
2015-09-19 19:38:46 +08:00
|
|
|
if (flags & MG_SOCK_STRINGIFY_IP) {
|
|
|
|
#if defined(MG_ENABLE_IPV6)
|
2015-09-08 19:49:03 +08:00
|
|
|
const void *addr = NULL;
|
|
|
|
char *start = buf;
|
|
|
|
socklen_t capacity = len;
|
|
|
|
if (!is_v6) {
|
|
|
|
addr = &sa->sin.sin_addr;
|
|
|
|
} else {
|
|
|
|
addr = (void *) &sa->sin6.sin6_addr;
|
2015-09-19 19:38:46 +08:00
|
|
|
if (flags & MG_SOCK_STRINGIFY_PORT) {
|
2015-09-08 19:49:03 +08:00
|
|
|
*buf = '[';
|
|
|
|
start++;
|
|
|
|
capacity--;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (inet_ntop(sa->sa.sa_family, addr, start, capacity) == NULL) {
|
|
|
|
*buf = '\0';
|
|
|
|
}
|
2015-09-19 19:38:46 +08:00
|
|
|
#elif defined(_WIN32) || defined(MG_ESP8266)
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Only Windoze Vista (and newer) have inet_ntop() */
|
|
|
|
strncpy(buf, inet_ntoa(sa->sin.sin_addr), len);
|
|
|
|
#else
|
|
|
|
inet_ntop(AF_INET, (void *) &sa->sin.sin_addr, buf, len);
|
|
|
|
#endif
|
|
|
|
}
|
2015-09-19 19:38:46 +08:00
|
|
|
if (flags & MG_SOCK_STRINGIFY_PORT) {
|
2015-09-08 19:49:03 +08:00
|
|
|
int port = ntohs(sa->sin.sin_port);
|
2015-09-19 19:38:46 +08:00
|
|
|
if (flags & MG_SOCK_STRINGIFY_IP) {
|
2015-09-08 19:49:03 +08:00
|
|
|
snprintf(buf + strlen(buf), len - (strlen(buf) + 1), "%s:%d",
|
|
|
|
(is_v6 ? "]" : ""), port);
|
|
|
|
} else {
|
|
|
|
snprintf(buf, len, "%d", port);
|
|
|
|
}
|
2013-11-25 01:29:21 +08:00
|
|
|
}
|
|
|
|
}
|
2013-11-23 18:47:35 +08:00
|
|
|
|
2015-11-27 18:32:36 +08:00
|
|
|
void mg_conn_addr_to_str(struct mg_connection *nc, char *buf, size_t len,
|
|
|
|
int flags) {
|
|
|
|
union socket_address sa;
|
|
|
|
memset(&sa, 0, sizeof(sa));
|
|
|
|
mg_if_get_conn_addr(nc, flags & MG_SOCK_STRINGIFY_REMOTE, &sa);
|
|
|
|
mg_sock_addr_to_str(&sa, buf, len, flags);
|
|
|
|
}
|
|
|
|
|
|
|
|
#ifndef MG_DISABLE_HEXDUMP
|
2015-09-08 19:49:03 +08:00
|
|
|
int mg_hexdump(const void *buf, int len, char *dst, int dst_len) {
|
|
|
|
const unsigned char *p = (const unsigned char *) buf;
|
|
|
|
char ascii[17] = "";
|
|
|
|
int i, idx, n = 0;
|
2014-10-21 22:51:42 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
for (i = 0; i < len; i++) {
|
|
|
|
idx = i % 16;
|
|
|
|
if (idx == 0) {
|
|
|
|
if (i > 0) n += snprintf(dst + n, dst_len - n, " %s\n", ascii);
|
|
|
|
n += snprintf(dst + n, dst_len - n, "%04x ", i);
|
|
|
|
}
|
|
|
|
n += snprintf(dst + n, dst_len - n, " %02x", p[i]);
|
|
|
|
ascii[idx] = p[i] < 0x20 || p[i] > 0x7e ? '.' : p[i];
|
|
|
|
ascii[idx + 1] = '\0';
|
|
|
|
}
|
2013-11-23 18:47:35 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
while (i++ % 16) n += snprintf(dst + n, dst_len - n, "%s", " ");
|
|
|
|
n += snprintf(dst + n, dst_len - n, " %s\n\n", ascii);
|
2013-11-23 18:47:35 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return n;
|
2013-11-23 18:47:35 +08:00
|
|
|
}
|
2015-11-27 18:32:36 +08:00
|
|
|
#endif
|
2013-11-23 18:47:35 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
int mg_avprintf(char **buf, size_t size, const char *fmt, va_list ap) {
|
|
|
|
va_list ap_copy;
|
|
|
|
int len;
|
2013-11-23 18:47:35 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
va_copy(ap_copy, ap);
|
|
|
|
len = vsnprintf(*buf, size, fmt, ap_copy);
|
|
|
|
va_end(ap_copy);
|
|
|
|
|
|
|
|
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. */
|
|
|
|
*buf = NULL; /* LCOV_EXCL_START */
|
|
|
|
while (len < 0) {
|
2015-09-19 19:38:46 +08:00
|
|
|
MG_FREE(*buf);
|
2015-09-08 19:49:03 +08:00
|
|
|
size *= 2;
|
2015-09-19 19:38:46 +08:00
|
|
|
if ((*buf = (char *) MG_MALLOC(size)) == NULL) break;
|
2015-09-08 19:49:03 +08:00
|
|
|
va_copy(ap_copy, ap);
|
|
|
|
len = vsnprintf(*buf, size, fmt, ap_copy);
|
|
|
|
va_end(ap_copy);
|
|
|
|
}
|
|
|
|
/* LCOV_EXCL_STOP */
|
2015-10-24 09:22:00 +08:00
|
|
|
} else if (len >= (int) size) {
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Standard-compliant code path. Allocate a buffer that is large enough. */
|
2015-09-19 19:38:46 +08:00
|
|
|
if ((*buf = (char *) MG_MALLOC(len + 1)) == NULL) {
|
2015-09-08 19:49:03 +08:00
|
|
|
len = -1; /* LCOV_EXCL_LINE */
|
|
|
|
} else { /* LCOV_EXCL_LINE */
|
|
|
|
va_copy(ap_copy, ap);
|
|
|
|
len = vsnprintf(*buf, len + 1, fmt, ap_copy);
|
|
|
|
va_end(ap_copy);
|
2013-11-25 01:29:21 +08:00
|
|
|
}
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
return len;
|
2013-11-23 18:47:35 +08:00
|
|
|
}
|
|
|
|
|
2015-11-27 18:32:36 +08:00
|
|
|
#if !defined(NO_LIBC) && !defined(MG_DISABLE_HEXDUMP)
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_hexdump_connection(struct mg_connection *nc, const char *path,
|
2015-11-27 18:32:36 +08:00
|
|
|
const void *buf, int num_bytes, int ev) {
|
|
|
|
FILE *fp = NULL;
|
|
|
|
char *hexbuf, src[60], dst[60];
|
2015-09-08 19:49:03 +08:00
|
|
|
int buf_size = num_bytes * 5 + 100;
|
2013-11-23 18:47:35 +08:00
|
|
|
|
2015-11-27 18:32:36 +08:00
|
|
|
if (strcmp(path, "-") == 0) {
|
|
|
|
fp = stdout;
|
|
|
|
} else if (strcmp(path, "--") == 0) {
|
|
|
|
fp = stderr;
|
|
|
|
#ifndef MG_DISABLE_FILESYSTEM
|
|
|
|
} else {
|
|
|
|
fp = fopen(path, "a");
|
2015-10-27 17:54:10 +08:00
|
|
|
#endif
|
2013-11-23 18:47:35 +08:00
|
|
|
}
|
2015-11-27 18:32:36 +08:00
|
|
|
if (fp == NULL) return;
|
|
|
|
|
|
|
|
mg_conn_addr_to_str(nc, src, sizeof(src),
|
|
|
|
MG_SOCK_STRINGIFY_IP | MG_SOCK_STRINGIFY_PORT);
|
|
|
|
mg_conn_addr_to_str(nc, dst, sizeof(dst), MG_SOCK_STRINGIFY_IP |
|
|
|
|
MG_SOCK_STRINGIFY_PORT |
|
|
|
|
MG_SOCK_STRINGIFY_REMOTE);
|
|
|
|
fprintf(
|
|
|
|
fp, "%lu %p %s %s %s %d\n", (unsigned long) time(NULL), nc, src,
|
|
|
|
ev == MG_EV_RECV ? "<-" : ev == MG_EV_SEND
|
|
|
|
? "->"
|
|
|
|
: ev == MG_EV_ACCEPT
|
|
|
|
? "<A"
|
|
|
|
: ev == MG_EV_CONNECT ? "C>" : "XX",
|
|
|
|
dst, num_bytes);
|
|
|
|
if (num_bytes > 0 && (hexbuf = (char *) MG_MALLOC(buf_size)) != NULL) {
|
|
|
|
mg_hexdump(buf, num_bytes, hexbuf, buf_size);
|
|
|
|
fprintf(fp, "%s", hexbuf);
|
|
|
|
MG_FREE(hexbuf);
|
|
|
|
}
|
|
|
|
if (fp != stdin && fp != stdout) fclose(fp);
|
2013-11-23 18:47:35 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
#endif
|
2013-11-23 18:47:35 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
int mg_is_big_endian(void) {
|
|
|
|
static const int n = 1;
|
|
|
|
/* TODO(mkm) use compiletime check with 4-byte char literal */
|
|
|
|
return ((char *) &n)[0] == 0;
|
|
|
|
}
|
2013-11-23 18:47:35 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
const char *mg_next_comma_list_entry(const char *list, struct mg_str *val,
|
|
|
|
struct mg_str *eq_val) {
|
|
|
|
if (list == NULL || *list == '\0') {
|
|
|
|
/* End of the list */
|
|
|
|
list = NULL;
|
|
|
|
} else {
|
|
|
|
val->p = list;
|
|
|
|
if ((list = strchr(val->p, ',')) != NULL) {
|
|
|
|
/* Comma found. Store length and shift the list ptr */
|
|
|
|
val->len = list - val->p;
|
|
|
|
list++;
|
|
|
|
} else {
|
|
|
|
/* This value is the last one */
|
|
|
|
list = val->p + strlen(val->p);
|
|
|
|
val->len = list - val->p;
|
2013-11-25 01:29:21 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
if (eq_val != NULL) {
|
|
|
|
/* Value has form "x=y", adjust pointers and lengths */
|
|
|
|
/* so that val points to "x", and eq_val points to "y". */
|
|
|
|
eq_val->len = 0;
|
|
|
|
eq_val->p = (const char *) memchr(val->p, '=', val->len);
|
|
|
|
if (eq_val->p != NULL) {
|
|
|
|
eq_val->p++; /* Skip over '=' character */
|
|
|
|
eq_val->len = val->p + val->len - eq_val->p;
|
|
|
|
val->len = (eq_val->p - val->p) - 1;
|
|
|
|
}
|
2013-11-23 18:47:35 +08:00
|
|
|
}
|
|
|
|
}
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return list;
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
int mg_match_prefix(const char *pattern, int pattern_len, const char *str) {
|
|
|
|
const char *or_str;
|
|
|
|
int len, res, i = 0, j = 0;
|
2015-08-13 01:34:01 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if ((or_str = (const char *) memchr(pattern, '|', pattern_len)) != NULL) {
|
|
|
|
res = mg_match_prefix(pattern, or_str - pattern, str);
|
|
|
|
return res > 0 ? res : mg_match_prefix(
|
|
|
|
or_str + 1,
|
|
|
|
(pattern + pattern_len) - (or_str + 1), str);
|
|
|
|
}
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
for (; i < pattern_len; i++, j++) {
|
|
|
|
if (pattern[i] == '?' && str[j] != '\0') {
|
|
|
|
continue;
|
|
|
|
} else if (pattern[i] == '$') {
|
|
|
|
return str[j] == '\0' ? j : -1;
|
|
|
|
} else if (pattern[i] == '*') {
|
|
|
|
i++;
|
|
|
|
if (pattern[i] == '*') {
|
|
|
|
i++;
|
|
|
|
len = (int) strlen(str + j);
|
|
|
|
} else {
|
|
|
|
len = (int) strcspn(str + j, "/");
|
|
|
|
}
|
|
|
|
if (i == pattern_len) {
|
|
|
|
return j + len;
|
|
|
|
}
|
|
|
|
do {
|
|
|
|
res = mg_match_prefix(pattern + i, pattern_len - i, str + j + len);
|
|
|
|
} while (res == -1 && len-- > 0);
|
|
|
|
return res == -1 ? -1 : j + res + len;
|
|
|
|
} else if (lowercase(&pattern[i]) != lowercase(&str[j])) {
|
|
|
|
return -1;
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
return j;
|
|
|
|
}
|
|
|
|
#ifdef NS_MODULE_LINES
|
2015-12-15 02:45:18 +08:00
|
|
|
#line 1 "./src/json-rpc.c"
|
2015-09-08 19:49:03 +08:00
|
|
|
/**/
|
|
|
|
#endif
|
|
|
|
/* Copyright (c) 2014 Cesanta Software Limited */
|
|
|
|
/* All rights reserved */
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_DISABLE_JSON_RPC
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-12-12 22:05:54 +08:00
|
|
|
/* Amalgamated: #include "mongoose/src/internal.h" */
|
2015-12-12 22:16:48 +08:00
|
|
|
/* Amalgamated: #include "mongoose/src/json-rpc.h" */
|
|
|
|
/* Amalgamated: #include "mongoose/deps/frozen/frozen.h" */
|
2013-11-23 18:47:35 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
int mg_rpc_create_reply(char *buf, int len, const struct mg_rpc_request *req,
|
|
|
|
const char *result_fmt, ...) {
|
|
|
|
static const struct json_token null_tok = {"null", 4, 0, JSON_TYPE_NULL};
|
|
|
|
const struct json_token *id = req->id == NULL ? &null_tok : req->id;
|
|
|
|
va_list ap;
|
|
|
|
int n = 0;
|
2015-08-13 01:34:01 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
n += json_emit(buf + n, len - n, "{s:s,s:", "jsonrpc", "2.0", "id");
|
|
|
|
if (id->type == JSON_TYPE_STRING) {
|
|
|
|
n += json_emit_quoted_str(buf + n, len - n, id->ptr, id->len);
|
|
|
|
} else {
|
|
|
|
n += json_emit_unquoted_str(buf + n, len - n, id->ptr, id->len);
|
2013-11-23 18:47:35 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
n += json_emit(buf + n, len - n, ",s:", "result");
|
2013-11-23 18:47:35 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
va_start(ap, result_fmt);
|
|
|
|
n += json_emit_va(buf + n, len - n, result_fmt, ap);
|
|
|
|
va_end(ap);
|
2013-11-25 01:29:21 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
n += json_emit(buf + n, len - n, "}");
|
2015-02-13 03:12:37 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return n;
|
|
|
|
}
|
2013-11-25 01:29:21 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
int mg_rpc_create_request(char *buf, int len, const char *method,
|
|
|
|
const char *id, const char *params_fmt, ...) {
|
|
|
|
va_list ap;
|
|
|
|
int n = 0;
|
2013-11-25 01:29:21 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
n += json_emit(buf + n, len - n, "{s:s,s:s,s:s,s:", "jsonrpc", "2.0", "id",
|
|
|
|
id, "method", method, "params");
|
|
|
|
va_start(ap, params_fmt);
|
|
|
|
n += json_emit_va(buf + n, len - n, params_fmt, ap);
|
|
|
|
va_end(ap);
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
n += json_emit(buf + n, len - n, "}");
|
2014-06-12 03:37:35 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return n;
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
int mg_rpc_create_error(char *buf, int len, struct mg_rpc_request *req,
|
|
|
|
int code, const char *message, const char *fmt, ...) {
|
2014-04-24 21:09:54 +08:00
|
|
|
va_list ap;
|
2015-09-08 19:49:03 +08:00
|
|
|
int n = 0;
|
2014-04-24 21:09:54 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
n += json_emit(buf + n, len - n, "{s:s,s:V,s:{s:i,s:s,s:", "jsonrpc", "2.0",
|
|
|
|
"id", req->id == NULL ? "null" : req->id->ptr,
|
|
|
|
req->id == NULL ? 4 : req->id->len, "error", "code", code,
|
|
|
|
"message", message, "data");
|
2014-04-24 21:09:54 +08:00
|
|
|
va_start(ap, fmt);
|
2015-09-08 19:49:03 +08:00
|
|
|
n += json_emit_va(buf + n, len - n, fmt, ap);
|
2014-04-24 21:09:54 +08:00
|
|
|
va_end(ap);
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
n += json_emit(buf + n, len - n, "}}");
|
2014-04-24 21:09:54 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return n;
|
2014-04-24 21:09:54 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
int mg_rpc_create_std_error(char *buf, int len, struct mg_rpc_request *req,
|
|
|
|
int code) {
|
|
|
|
const char *message = NULL;
|
2013-12-18 17:35:36 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
switch (code) {
|
|
|
|
case JSON_RPC_PARSE_ERROR:
|
|
|
|
message = "parse error";
|
|
|
|
break;
|
|
|
|
case JSON_RPC_INVALID_REQUEST_ERROR:
|
|
|
|
message = "invalid request";
|
|
|
|
break;
|
|
|
|
case JSON_RPC_METHOD_NOT_FOUND_ERROR:
|
|
|
|
message = "method not found";
|
|
|
|
break;
|
|
|
|
case JSON_RPC_INVALID_PARAMS_ERROR:
|
|
|
|
message = "invalid parameters";
|
|
|
|
break;
|
|
|
|
case JSON_RPC_SERVER_ERROR:
|
|
|
|
message = "server error";
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
message = "unspecified error";
|
|
|
|
break;
|
2013-12-18 17:35:36 +08:00
|
|
|
}
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return mg_rpc_create_error(buf, len, req, code, message, "N");
|
2014-01-14 21:16:58 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
int mg_rpc_dispatch(const char *buf, int len, char *dst, int dst_len,
|
|
|
|
const char **methods, mg_rpc_handler_t *handlers) {
|
|
|
|
struct json_token tokens[200];
|
|
|
|
struct mg_rpc_request req;
|
|
|
|
int i, n;
|
2014-01-14 21:16:58 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
memset(&req, 0, sizeof(req));
|
|
|
|
n = parse_json(buf, len, tokens, sizeof(tokens) / sizeof(tokens[0]));
|
|
|
|
if (n <= 0) {
|
|
|
|
int err_code = (n == JSON_STRING_INVALID) ? JSON_RPC_PARSE_ERROR
|
|
|
|
: JSON_RPC_SERVER_ERROR;
|
|
|
|
return mg_rpc_create_std_error(dst, dst_len, &req, err_code);
|
|
|
|
}
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
req.message = tokens;
|
|
|
|
req.id = find_json_token(tokens, "id");
|
|
|
|
req.method = find_json_token(tokens, "method");
|
|
|
|
req.params = find_json_token(tokens, "params");
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (req.id == NULL || req.method == NULL) {
|
|
|
|
return mg_rpc_create_std_error(dst, dst_len, &req,
|
|
|
|
JSON_RPC_INVALID_REQUEST_ERROR);
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
for (i = 0; methods[i] != NULL; i++) {
|
|
|
|
int mlen = strlen(methods[i]);
|
|
|
|
if (mlen == req.method->len &&
|
|
|
|
memcmp(methods[i], req.method->ptr, mlen) == 0)
|
|
|
|
break;
|
|
|
|
}
|
2013-11-25 01:29:21 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (methods[i] == NULL) {
|
|
|
|
return mg_rpc_create_std_error(dst, dst_len, &req,
|
|
|
|
JSON_RPC_METHOD_NOT_FOUND_ERROR);
|
|
|
|
}
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return handlers[i](dst, dst_len, &req);
|
2013-11-25 01:33:05 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
int mg_rpc_parse_reply(const char *buf, int len, struct json_token *toks,
|
|
|
|
int max_toks, struct mg_rpc_reply *rep,
|
|
|
|
struct mg_rpc_error *er) {
|
|
|
|
int n = parse_json(buf, len, toks, max_toks);
|
2013-11-25 01:33:05 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
memset(rep, 0, sizeof(*rep));
|
|
|
|
memset(er, 0, sizeof(*er));
|
2013-11-25 01:33:05 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (n > 0) {
|
|
|
|
if ((rep->result = find_json_token(toks, "result")) != NULL) {
|
|
|
|
rep->message = toks;
|
|
|
|
rep->id = find_json_token(toks, "id");
|
|
|
|
} else {
|
|
|
|
er->message = toks;
|
|
|
|
er->id = find_json_token(toks, "id");
|
|
|
|
er->error_code = find_json_token(toks, "error.code");
|
|
|
|
er->error_message = find_json_token(toks, "error.message");
|
|
|
|
er->error_data = find_json_token(toks, "error.data");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return n;
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
2013-11-25 01:33:05 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#endif /* MG_DISABLE_JSON_RPC */
|
2015-09-08 19:49:03 +08:00
|
|
|
#ifdef NS_MODULE_LINES
|
2015-12-15 02:45:18 +08:00
|
|
|
#line 1 "./src/mqtt.c"
|
2015-09-08 19:49:03 +08:00
|
|
|
/**/
|
|
|
|
#endif
|
|
|
|
/*
|
|
|
|
* Copyright (c) 2014 Cesanta Software Limited
|
|
|
|
* All rights reserved
|
|
|
|
*/
|
2014-01-11 17:48:27 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_DISABLE_MQTT
|
2014-01-11 17:48:27 +08:00
|
|
|
|
2015-12-12 22:05:54 +08:00
|
|
|
/* Amalgamated: #include "mongoose/src/internal.h" */
|
2015-12-12 22:16:48 +08:00
|
|
|
/* Amalgamated: #include "mongoose/src/mqtt.h" */
|
2014-01-11 17:48:27 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static int parse_mqtt(struct mbuf *io, struct mg_mqtt_message *mm) {
|
|
|
|
uint8_t header;
|
|
|
|
int cmd;
|
|
|
|
size_t len = 0;
|
|
|
|
int var_len = 0;
|
|
|
|
char *vlen = &io->buf[1];
|
2014-01-11 17:48:27 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (io->len < 2) return -1;
|
2014-01-11 17:48:27 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
header = io->buf[0];
|
|
|
|
cmd = header >> 4;
|
2014-01-11 17:48:27 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* decode mqtt variable length */
|
|
|
|
do {
|
|
|
|
len += (*vlen & 127) << 7 * (vlen - &io->buf[1]);
|
|
|
|
} while ((*vlen++ & 128) != 0 && ((size_t)(vlen - io->buf) <= io->len));
|
2013-11-25 01:33:05 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (io->len < (size_t)(len - 1)) return -1;
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
mbuf_remove(io, 1 + (vlen - &io->buf[1]));
|
|
|
|
mm->cmd = cmd;
|
2015-09-19 19:38:46 +08:00
|
|
|
mm->qos = MG_MQTT_GET_QOS(header);
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
switch (cmd) {
|
2015-09-19 19:38:46 +08:00
|
|
|
case MG_MQTT_CMD_CONNECT:
|
2015-09-08 19:49:03 +08:00
|
|
|
/* TODO(mkm): parse keepalive and will */
|
|
|
|
break;
|
2015-09-19 19:38:46 +08:00
|
|
|
case MG_MQTT_CMD_CONNACK:
|
2015-09-08 19:49:03 +08:00
|
|
|
mm->connack_ret_code = io->buf[1];
|
|
|
|
var_len = 2;
|
|
|
|
break;
|
2015-09-19 19:38:46 +08:00
|
|
|
case MG_MQTT_CMD_PUBACK:
|
|
|
|
case MG_MQTT_CMD_PUBREC:
|
|
|
|
case MG_MQTT_CMD_PUBREL:
|
|
|
|
case MG_MQTT_CMD_PUBCOMP:
|
|
|
|
case MG_MQTT_CMD_SUBACK:
|
2015-09-08 19:49:03 +08:00
|
|
|
mm->message_id = ntohs(*(uint16_t *) io->buf);
|
|
|
|
var_len = 2;
|
|
|
|
break;
|
2015-09-19 19:38:46 +08:00
|
|
|
case MG_MQTT_CMD_PUBLISH: {
|
2015-09-08 19:49:03 +08:00
|
|
|
uint16_t topic_len = ntohs(*(uint16_t *) io->buf);
|
2015-09-19 19:38:46 +08:00
|
|
|
mm->topic = (char *) MG_MALLOC(topic_len + 1);
|
2015-09-08 19:49:03 +08:00
|
|
|
mm->topic[topic_len] = 0;
|
|
|
|
strncpy(mm->topic, io->buf + 2, topic_len);
|
|
|
|
var_len = topic_len + 2;
|
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
if (MG_MQTT_GET_QOS(header) > 0) {
|
2015-09-08 19:49:03 +08:00
|
|
|
mm->message_id = ntohs(*(uint16_t *) io->buf);
|
|
|
|
var_len += 2;
|
|
|
|
}
|
|
|
|
} break;
|
2015-09-19 19:38:46 +08:00
|
|
|
case MG_MQTT_CMD_SUBSCRIBE:
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* topic expressions are left in the payload and can be parsed with
|
|
|
|
* `mg_mqtt_next_subscribe_topic`
|
|
|
|
*/
|
|
|
|
mm->message_id = ntohs(*(uint16_t *) io->buf);
|
|
|
|
var_len = 2;
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
printf("TODO: UNHANDLED COMMAND %d\n", cmd);
|
|
|
|
break;
|
2013-11-25 01:29:21 +08:00
|
|
|
}
|
2013-11-23 18:47:35 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
mbuf_remove(io, var_len);
|
|
|
|
return len - var_len;
|
|
|
|
}
|
2013-12-05 18:01:11 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void mqtt_handler(struct mg_connection *nc, int ev, void *ev_data) {
|
|
|
|
int len;
|
|
|
|
struct mbuf *io = &nc->recv_mbuf;
|
|
|
|
struct mg_mqtt_message mm;
|
|
|
|
memset(&mm, 0, sizeof(mm));
|
2015-03-17 21:18:15 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
nc->handler(nc, ev, ev_data);
|
2013-12-05 18:01:11 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
switch (ev) {
|
2015-09-19 19:38:46 +08:00
|
|
|
case MG_EV_RECV:
|
2015-09-08 19:49:03 +08:00
|
|
|
len = parse_mqtt(io, &mm);
|
|
|
|
if (len == -1) break; /* not fully buffered */
|
|
|
|
mm.payload.p = io->buf;
|
|
|
|
mm.payload.len = len;
|
2013-12-05 18:01:11 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
nc->handler(nc, MG_MQTT_EVENT_BASE + mm.cmd, &mm);
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (mm.topic) {
|
2015-09-19 19:38:46 +08:00
|
|
|
MG_FREE(mm.topic);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
mbuf_remove(io, mm.payload.len);
|
2013-12-09 22:18:49 +08:00
|
|
|
break;
|
|
|
|
}
|
2013-12-05 18:01:11 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_set_protocol_mqtt(struct mg_connection *nc) {
|
|
|
|
nc->proto_handler = mqtt_handler;
|
2014-01-11 17:48:27 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_send_mqtt_handshake(struct mg_connection *nc, const char *client_id) {
|
|
|
|
static struct mg_send_mqtt_handshake_opts opts;
|
|
|
|
mg_send_mqtt_handshake_opt(nc, client_id, opts);
|
2014-01-11 17:48:27 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_send_mqtt_handshake_opt(struct mg_connection *nc, const char *client_id,
|
|
|
|
struct mg_send_mqtt_handshake_opts opts) {
|
2015-09-19 19:38:46 +08:00
|
|
|
uint8_t header = MG_MQTT_CMD_CONNECT << 4;
|
2015-09-08 19:49:03 +08:00
|
|
|
uint8_t rem_len;
|
|
|
|
uint16_t keep_alive;
|
|
|
|
uint16_t client_id_len;
|
2014-01-11 17:48:27 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* 9: version_header(len, magic_string, version_number), 1: flags, 2:
|
|
|
|
* keep-alive timer,
|
|
|
|
* 2: client_identifier_len, n: client_id
|
|
|
|
*/
|
|
|
|
rem_len = 9 + 1 + 2 + 2 + strlen(client_id);
|
|
|
|
|
|
|
|
mg_send(nc, &header, 1);
|
|
|
|
mg_send(nc, &rem_len, 1);
|
|
|
|
mg_send(nc, "\00\06MQIsdp\03", 9);
|
|
|
|
mg_send(nc, &opts.flags, 1);
|
|
|
|
|
|
|
|
if (opts.keep_alive == 0) {
|
|
|
|
opts.keep_alive = 60;
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
keep_alive = htons(opts.keep_alive);
|
|
|
|
mg_send(nc, &keep_alive, 2);
|
|
|
|
|
|
|
|
client_id_len = htons(strlen(client_id));
|
|
|
|
mg_send(nc, &client_id_len, 2);
|
|
|
|
mg_send(nc, client_id, strlen(client_id));
|
2013-11-23 18:47:35 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void mg_mqtt_prepend_header(struct mg_connection *nc, uint8_t cmd,
|
|
|
|
uint8_t flags, size_t len) {
|
|
|
|
size_t off = nc->send_mbuf.len - len;
|
|
|
|
uint8_t header = cmd << 4 | (uint8_t) flags;
|
2013-11-23 18:47:35 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
uint8_t buf[1 + sizeof(size_t)];
|
|
|
|
uint8_t *vlen = &buf[1];
|
2013-11-23 18:47:35 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
assert(nc->send_mbuf.len >= len);
|
2013-11-25 01:29:21 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
buf[0] = header;
|
2013-11-25 01:29:21 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* mqtt variable length encoding */
|
|
|
|
do {
|
|
|
|
*vlen = len % 0x80;
|
|
|
|
len /= 0x80;
|
|
|
|
if (len > 0) *vlen |= 0x80;
|
|
|
|
vlen++;
|
|
|
|
} while (len > 0);
|
2013-11-25 01:29:21 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
mbuf_insert(&nc->send_mbuf, off, buf, vlen - buf);
|
2013-11-25 01:29:21 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_mqtt_publish(struct mg_connection *nc, const char *topic,
|
|
|
|
uint16_t message_id, int flags, const void *data,
|
|
|
|
size_t len) {
|
|
|
|
size_t old_len = nc->send_mbuf.len;
|
2013-11-25 01:29:21 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
uint16_t topic_len = htons(strlen(topic));
|
|
|
|
uint16_t message_id_net = htons(message_id);
|
2013-11-25 01:29:21 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
mg_send(nc, &topic_len, 2);
|
|
|
|
mg_send(nc, topic, strlen(topic));
|
2015-09-19 19:38:46 +08:00
|
|
|
if (MG_MQTT_GET_QOS(flags) > 0) {
|
2015-09-08 19:49:03 +08:00
|
|
|
mg_send(nc, &message_id_net, 2);
|
2013-11-25 01:29:21 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
mg_send(nc, data, len);
|
2013-12-05 18:47:03 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
mg_mqtt_prepend_header(nc, MG_MQTT_CMD_PUBLISH, flags,
|
2015-09-08 19:49:03 +08:00
|
|
|
nc->send_mbuf.len - old_len);
|
2013-11-25 01:29:21 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_mqtt_subscribe(struct mg_connection *nc,
|
|
|
|
const struct mg_mqtt_topic_expression *topics,
|
|
|
|
size_t topics_len, uint16_t message_id) {
|
|
|
|
size_t old_len = nc->send_mbuf.len;
|
2013-11-25 01:29:21 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
uint16_t message_id_n = htons(message_id);
|
|
|
|
size_t i;
|
2013-11-25 01:29:21 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
mg_send(nc, (char *) &message_id_n, 2);
|
|
|
|
for (i = 0; i < topics_len; i++) {
|
|
|
|
uint16_t topic_len_n = htons(strlen(topics[i].topic));
|
|
|
|
mg_send(nc, &topic_len_n, 2);
|
|
|
|
mg_send(nc, topics[i].topic, strlen(topics[i].topic));
|
|
|
|
mg_send(nc, &topics[i].qos, 1);
|
2013-11-25 01:29:21 +08:00
|
|
|
}
|
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
mg_mqtt_prepend_header(nc, MG_MQTT_CMD_SUBSCRIBE, MG_MQTT_QOS(1),
|
2015-09-08 19:49:03 +08:00
|
|
|
nc->send_mbuf.len - old_len);
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
int mg_mqtt_next_subscribe_topic(struct mg_mqtt_message *msg,
|
|
|
|
struct mg_str *topic, uint8_t *qos, int pos) {
|
|
|
|
unsigned char *buf = (unsigned char *) msg->payload.p + pos;
|
|
|
|
if ((size_t) pos >= msg->payload.len) {
|
|
|
|
return -1;
|
|
|
|
}
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
topic->len = buf[0] << 8 | buf[1];
|
|
|
|
topic->p = (char *) buf + 2;
|
|
|
|
*qos = buf[2 + topic->len];
|
|
|
|
return pos + 2 + topic->len + 1;
|
|
|
|
}
|
2013-11-25 01:29:21 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_mqtt_unsubscribe(struct mg_connection *nc, char **topics,
|
|
|
|
size_t topics_len, uint16_t message_id) {
|
|
|
|
size_t old_len = nc->send_mbuf.len;
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
uint16_t message_id_n = htons(message_id);
|
|
|
|
size_t i;
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
mg_send(nc, (char *) &message_id_n, 2);
|
|
|
|
for (i = 0; i < topics_len; i++) {
|
|
|
|
uint16_t topic_len_n = htons(strlen(topics[i]));
|
|
|
|
mg_send(nc, &topic_len_n, 2);
|
|
|
|
mg_send(nc, topics[i], strlen(topics[i]));
|
2013-11-25 01:29:21 +08:00
|
|
|
}
|
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
mg_mqtt_prepend_header(nc, MG_MQTT_CMD_UNSUBSCRIBE, MG_MQTT_QOS(1),
|
2015-09-08 19:49:03 +08:00
|
|
|
nc->send_mbuf.len - old_len);
|
2013-11-25 01:29:21 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_mqtt_connack(struct mg_connection *nc, uint8_t return_code) {
|
|
|
|
uint8_t unused = 0;
|
|
|
|
mg_send(nc, &unused, 1);
|
|
|
|
mg_send(nc, &return_code, 1);
|
2015-09-19 19:38:46 +08:00
|
|
|
mg_mqtt_prepend_header(nc, MG_MQTT_CMD_CONNACK, 0, 2);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2014-04-26 01:04:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Sends a command which contains only a `message_id` and a QoS level of 1.
|
|
|
|
*
|
|
|
|
* Helper function.
|
|
|
|
*/
|
|
|
|
static void mg_send_mqtt_short_command(struct mg_connection *nc, uint8_t cmd,
|
|
|
|
uint16_t message_id) {
|
|
|
|
uint16_t message_id_net = htons(message_id);
|
|
|
|
mg_send(nc, &message_id_net, 2);
|
2015-09-19 19:38:46 +08:00
|
|
|
mg_mqtt_prepend_header(nc, cmd, MG_MQTT_QOS(1), 2);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_mqtt_puback(struct mg_connection *nc, uint16_t message_id) {
|
2015-09-19 19:38:46 +08:00
|
|
|
mg_send_mqtt_short_command(nc, MG_MQTT_CMD_PUBACK, message_id);
|
2013-11-25 01:29:21 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_mqtt_pubrec(struct mg_connection *nc, uint16_t message_id) {
|
2015-09-19 19:38:46 +08:00
|
|
|
mg_send_mqtt_short_command(nc, MG_MQTT_CMD_PUBREC, message_id);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2013-11-25 01:29:21 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_mqtt_pubrel(struct mg_connection *nc, uint16_t message_id) {
|
2015-09-19 19:38:46 +08:00
|
|
|
mg_send_mqtt_short_command(nc, MG_MQTT_CMD_PUBREL, message_id);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2013-11-23 18:47:35 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_mqtt_pubcomp(struct mg_connection *nc, uint16_t message_id) {
|
2015-09-19 19:38:46 +08:00
|
|
|
mg_send_mqtt_short_command(nc, MG_MQTT_CMD_PUBCOMP, message_id);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_mqtt_suback(struct mg_connection *nc, uint8_t *qoss, size_t qoss_len,
|
|
|
|
uint16_t message_id) {
|
|
|
|
size_t i;
|
|
|
|
uint16_t message_id_net = htons(message_id);
|
|
|
|
mg_send(nc, &message_id_net, 2);
|
|
|
|
for (i = 0; i < qoss_len; i++) {
|
|
|
|
mg_send(nc, &qoss[i], 1);
|
|
|
|
}
|
2015-09-19 19:38:46 +08:00
|
|
|
mg_mqtt_prepend_header(nc, MG_MQTT_CMD_SUBACK, MG_MQTT_QOS(1), 2 + qoss_len);
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_mqtt_unsuback(struct mg_connection *nc, uint16_t message_id) {
|
2015-09-19 19:38:46 +08:00
|
|
|
mg_send_mqtt_short_command(nc, MG_MQTT_CMD_UNSUBACK, message_id);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2013-11-23 18:47:35 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_mqtt_ping(struct mg_connection *nc) {
|
2015-09-19 19:38:46 +08:00
|
|
|
mg_mqtt_prepend_header(nc, MG_MQTT_CMD_PINGREQ, 0, 0);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2013-11-23 18:47:35 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_mqtt_pong(struct mg_connection *nc) {
|
2015-09-19 19:38:46 +08:00
|
|
|
mg_mqtt_prepend_header(nc, MG_MQTT_CMD_PINGRESP, 0, 0);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2013-11-23 18:47:35 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_mqtt_disconnect(struct mg_connection *nc) {
|
2015-09-19 19:38:46 +08:00
|
|
|
mg_mqtt_prepend_header(nc, MG_MQTT_CMD_DISCONNECT, 0, 0);
|
2013-11-23 18:47:35 +08:00
|
|
|
}
|
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#endif /* MG_DISABLE_MQTT */
|
2015-09-08 19:49:03 +08:00
|
|
|
#ifdef NS_MODULE_LINES
|
2015-12-15 02:45:18 +08:00
|
|
|
#line 1 "./src/mqtt-broker.c"
|
2015-09-08 19:49:03 +08:00
|
|
|
/**/
|
|
|
|
#endif
|
|
|
|
/*
|
|
|
|
* Copyright (c) 2014 Cesanta Software Limited
|
|
|
|
* All rights reserved
|
|
|
|
*/
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-12-12 22:05:54 +08:00
|
|
|
/* Amalgamated: #include "mongoose/src/internal.h" */
|
2015-12-12 22:16:48 +08:00
|
|
|
/* Amalgamated: #include "mongoose/src/mqtt-broker.h" */
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifdef MG_ENABLE_MQTT_BROKER
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
static void mg_mqtt_session_init(struct mg_mqtt_broker *brk,
|
|
|
|
struct mg_mqtt_session *s,
|
|
|
|
struct mg_connection *nc) {
|
|
|
|
s->brk = brk;
|
|
|
|
s->subscriptions = NULL;
|
|
|
|
s->num_subscriptions = 0;
|
|
|
|
s->nc = nc;
|
2013-09-30 20:11:24 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void mg_mqtt_add_session(struct mg_mqtt_session *s) {
|
|
|
|
s->next = s->brk->sessions;
|
|
|
|
s->brk->sessions = s;
|
|
|
|
s->prev = NULL;
|
|
|
|
if (s->next != NULL) s->next->prev = s;
|
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void mg_mqtt_remove_session(struct mg_mqtt_session *s) {
|
|
|
|
if (s->prev == NULL) s->brk->sessions = s->next;
|
|
|
|
if (s->prev) s->prev->next = s->next;
|
|
|
|
if (s->next) s->next->prev = s->prev;
|
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void mg_mqtt_destroy_session(struct mg_mqtt_session *s) {
|
|
|
|
size_t i;
|
|
|
|
for (i = 0; i < s->num_subscriptions; i++) {
|
2015-09-19 19:38:46 +08:00
|
|
|
MG_FREE((void *) s->subscriptions[i].topic);
|
2013-09-30 20:11:24 +08:00
|
|
|
}
|
2015-10-21 17:05:13 +08:00
|
|
|
MG_FREE(s->subscriptions);
|
2015-09-19 19:38:46 +08:00
|
|
|
MG_FREE(s);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void mg_mqtt_close_session(struct mg_mqtt_session *s) {
|
|
|
|
mg_mqtt_remove_session(s);
|
|
|
|
mg_mqtt_destroy_session(s);
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_mqtt_broker_init(struct mg_mqtt_broker *brk, void *user_data) {
|
|
|
|
brk->sessions = NULL;
|
|
|
|
brk->user_data = user_data;
|
|
|
|
}
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void mg_mqtt_broker_handle_connect(struct mg_mqtt_broker *brk,
|
|
|
|
struct mg_connection *nc) {
|
|
|
|
struct mg_mqtt_session *s = (struct mg_mqtt_session *) malloc(sizeof *s);
|
|
|
|
if (s == NULL) {
|
|
|
|
/* LCOV_EXCL_START */
|
2015-09-19 19:38:46 +08:00
|
|
|
mg_mqtt_connack(nc, MG_EV_MQTT_CONNACK_SERVER_UNAVAILABLE);
|
2015-09-08 19:49:03 +08:00
|
|
|
return;
|
|
|
|
/* LCOV_EXCL_STOP */
|
2013-09-30 20:11:24 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* TODO(mkm): check header (magic and version) */
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
mg_mqtt_session_init(brk, s, nc);
|
|
|
|
s->user_data = nc->user_data;
|
|
|
|
nc->user_data = s;
|
|
|
|
mg_mqtt_add_session(s);
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
mg_mqtt_connack(nc, MG_EV_MQTT_CONNACK_ACCEPTED);
|
2013-09-30 20:11:24 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void mg_mqtt_broker_handle_subscribe(struct mg_connection *nc,
|
|
|
|
struct mg_mqtt_message *msg) {
|
|
|
|
struct mg_mqtt_session *ss = (struct mg_mqtt_session *) nc->user_data;
|
|
|
|
uint8_t qoss[512];
|
|
|
|
size_t qoss_len = 0;
|
|
|
|
struct mg_str topic;
|
|
|
|
uint8_t qos;
|
|
|
|
int pos;
|
|
|
|
struct mg_mqtt_topic_expression *te;
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
for (pos = 0;
|
|
|
|
(pos = mg_mqtt_next_subscribe_topic(msg, &topic, &qos, pos)) != -1;) {
|
|
|
|
qoss[qoss_len++] = qos;
|
2015-06-22 18:31:08 +08:00
|
|
|
}
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
ss->subscriptions = (struct mg_mqtt_topic_expression *) realloc(
|
|
|
|
ss->subscriptions, sizeof(*ss->subscriptions) * qoss_len);
|
|
|
|
for (pos = 0;
|
|
|
|
(pos = mg_mqtt_next_subscribe_topic(msg, &topic, &qos, pos)) != -1;
|
|
|
|
ss->num_subscriptions++) {
|
|
|
|
te = &ss->subscriptions[ss->num_subscriptions];
|
|
|
|
te->topic = (char *) malloc(topic.len + 1);
|
|
|
|
te->qos = qos;
|
|
|
|
strncpy((char *) te->topic, topic.p, topic.len + 1);
|
2013-09-30 20:11:24 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
mg_mqtt_suback(nc, qoss, qoss_len, msg->message_id);
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2014-12-12 02:01:58 +08:00
|
|
|
/*
|
2015-09-08 19:49:03 +08:00
|
|
|
* Matches a topic against a topic expression
|
2014-12-12 02:01:58 +08:00
|
|
|
*
|
2015-09-08 19:49:03 +08:00
|
|
|
* See http://goo.gl/iWk21X
|
2014-12-12 02:01:58 +08:00
|
|
|
*
|
2015-09-08 19:49:03 +08:00
|
|
|
* Returns 1 if it matches; 0 otherwise.
|
2014-12-12 02:01:58 +08:00
|
|
|
*/
|
2015-09-08 19:49:03 +08:00
|
|
|
static int mg_mqtt_match_topic_expression(const char *exp, const char *topic) {
|
|
|
|
/* TODO(mkm): implement real matching */
|
|
|
|
int len = strlen(exp);
|
|
|
|
if (strchr(exp, '#')) {
|
|
|
|
len -= 2;
|
2013-09-30 20:11:24 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
return strncmp(exp, topic, len) == 0;
|
2013-09-30 20:11:24 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void mg_mqtt_broker_handle_publish(struct mg_mqtt_broker *brk,
|
|
|
|
struct mg_mqtt_message *msg) {
|
|
|
|
struct mg_mqtt_session *s;
|
|
|
|
size_t i;
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
for (s = mg_mqtt_next(brk, NULL); s != NULL; s = mg_mqtt_next(brk, s)) {
|
|
|
|
for (i = 0; i < s->num_subscriptions; i++) {
|
|
|
|
if (mg_mqtt_match_topic_expression(s->subscriptions[i].topic,
|
|
|
|
msg->topic)) {
|
|
|
|
mg_mqtt_publish(s->nc, msg->topic, 0, 0, msg->payload.p,
|
|
|
|
msg->payload.len);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_mqtt_broker(struct mg_connection *nc, int ev, void *data) {
|
|
|
|
struct mg_mqtt_message *msg = (struct mg_mqtt_message *) data;
|
|
|
|
struct mg_mqtt_broker *brk;
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (nc->listener) {
|
|
|
|
brk = (struct mg_mqtt_broker *) nc->listener->user_data;
|
|
|
|
} else {
|
|
|
|
brk = (struct mg_mqtt_broker *) nc->user_data;
|
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
switch (ev) {
|
2015-09-19 19:38:46 +08:00
|
|
|
case MG_EV_ACCEPT:
|
2015-09-08 19:49:03 +08:00
|
|
|
mg_set_protocol_mqtt(nc);
|
|
|
|
break;
|
2015-09-19 19:38:46 +08:00
|
|
|
case MG_EV_MQTT_CONNECT:
|
2015-09-08 19:49:03 +08:00
|
|
|
mg_mqtt_broker_handle_connect(brk, nc);
|
|
|
|
break;
|
2015-09-19 19:38:46 +08:00
|
|
|
case MG_EV_MQTT_SUBSCRIBE:
|
2015-09-08 19:49:03 +08:00
|
|
|
mg_mqtt_broker_handle_subscribe(nc, msg);
|
|
|
|
break;
|
2015-09-19 19:38:46 +08:00
|
|
|
case MG_EV_MQTT_PUBLISH:
|
2015-09-08 19:49:03 +08:00
|
|
|
mg_mqtt_broker_handle_publish(brk, msg);
|
|
|
|
break;
|
2015-09-19 19:38:46 +08:00
|
|
|
case MG_EV_CLOSE:
|
2015-09-08 19:49:03 +08:00
|
|
|
if (nc->listener) {
|
|
|
|
mg_mqtt_close_session((struct mg_mqtt_session *) nc->user_data);
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
struct mg_mqtt_session *mg_mqtt_next(struct mg_mqtt_broker *brk,
|
|
|
|
struct mg_mqtt_session *s) {
|
|
|
|
return s == NULL ? brk->sessions : s->next;
|
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#endif /* MG_ENABLE_MQTT_BROKER */
|
2015-09-08 19:49:03 +08:00
|
|
|
#ifdef NS_MODULE_LINES
|
2015-12-15 02:45:18 +08:00
|
|
|
#line 1 "./src/dns.c"
|
2015-09-08 19:49:03 +08:00
|
|
|
/**/
|
|
|
|
#endif
|
|
|
|
/*
|
|
|
|
* Copyright (c) 2014 Cesanta Software Limited
|
|
|
|
* All rights reserved
|
|
|
|
*/
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_DISABLE_DNS
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-12-12 22:05:54 +08:00
|
|
|
/* Amalgamated: #include "mongoose/src/internal.h" */
|
2015-12-12 22:16:48 +08:00
|
|
|
/* Amalgamated: #include "mongoose/src/dns.h" */
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-10-21 21:15:16 +08:00
|
|
|
static int mg_dns_tid = 0xa0;
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-10-21 21:15:16 +08:00
|
|
|
struct mg_dns_header {
|
2015-09-08 19:49:03 +08:00
|
|
|
uint16_t transaction_id;
|
|
|
|
uint16_t flags;
|
|
|
|
uint16_t num_questions;
|
|
|
|
uint16_t num_answers;
|
|
|
|
uint16_t num_authority_prs;
|
|
|
|
uint16_t num_other_prs;
|
|
|
|
};
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-10-21 21:15:16 +08:00
|
|
|
struct mg_dns_resource_record *mg_dns_next_record(
|
|
|
|
struct mg_dns_message *msg, int query,
|
|
|
|
struct mg_dns_resource_record *prev) {
|
|
|
|
struct mg_dns_resource_record *rr;
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
for (rr = (prev == NULL ? msg->answers : prev + 1);
|
|
|
|
rr - msg->answers < msg->num_answers; rr++) {
|
|
|
|
if (rr->rtype == query) {
|
|
|
|
return rr;
|
2013-09-30 20:11:24 +08:00
|
|
|
}
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
return NULL;
|
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-10-21 21:15:16 +08:00
|
|
|
int mg_dns_parse_record_data(struct mg_dns_message *msg,
|
|
|
|
struct mg_dns_resource_record *rr, void *data,
|
2015-09-08 19:49:03 +08:00
|
|
|
size_t data_len) {
|
|
|
|
switch (rr->rtype) {
|
2015-09-19 19:38:46 +08:00
|
|
|
case MG_DNS_A_RECORD:
|
2015-09-08 19:49:03 +08:00
|
|
|
if (data_len < sizeof(struct in_addr)) {
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
if (rr->rdata.p + data_len > msg->pkt.p + msg->pkt.len) {
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
memcpy(data, rr->rdata.p, data_len);
|
|
|
|
return 0;
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifdef MG_ENABLE_IPV6
|
|
|
|
case MG_DNS_AAAA_RECORD:
|
2015-09-08 19:49:03 +08:00
|
|
|
if (data_len < sizeof(struct in6_addr)) {
|
|
|
|
return -1; /* LCOV_EXCL_LINE */
|
|
|
|
}
|
|
|
|
memcpy(data, rr->rdata.p, data_len);
|
|
|
|
return 0;
|
|
|
|
#endif
|
2015-09-19 19:38:46 +08:00
|
|
|
case MG_DNS_CNAME_RECORD:
|
2015-10-21 21:15:16 +08:00
|
|
|
mg_dns_uncompress_name(msg, &rr->rdata, (char *) data, data_len);
|
2015-09-08 19:49:03 +08:00
|
|
|
return 0;
|
2013-09-30 20:11:24 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return -1;
|
2013-09-30 20:11:24 +08:00
|
|
|
}
|
|
|
|
|
2015-10-21 21:15:16 +08:00
|
|
|
int mg_dns_insert_header(struct mbuf *io, size_t pos,
|
|
|
|
struct mg_dns_message *msg) {
|
|
|
|
struct mg_dns_header header;
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
memset(&header, 0, sizeof(header));
|
|
|
|
header.transaction_id = msg->transaction_id;
|
|
|
|
header.flags = htons(msg->flags);
|
|
|
|
header.num_questions = htons(msg->num_questions);
|
|
|
|
header.num_answers = htons(msg->num_answers);
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return mbuf_insert(io, pos, &header, sizeof(header));
|
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-10-21 21:15:16 +08:00
|
|
|
int mg_dns_copy_body(struct mbuf *io, struct mg_dns_message *msg) {
|
|
|
|
return mbuf_append(io, msg->pkt.p + sizeof(struct mg_dns_header),
|
|
|
|
msg->pkt.len - sizeof(struct mg_dns_header));
|
2013-09-30 20:11:24 +08:00
|
|
|
}
|
|
|
|
|
2015-10-21 21:15:16 +08:00
|
|
|
static int mg_dns_encode_name(struct mbuf *io, const char *name, size_t len) {
|
2015-09-08 19:49:03 +08:00
|
|
|
const char *s;
|
|
|
|
unsigned char n;
|
|
|
|
size_t pos = io->len;
|
|
|
|
|
|
|
|
do {
|
|
|
|
if ((s = strchr(name, '.')) == NULL) {
|
|
|
|
s = name + len;
|
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (s - name > 127) {
|
|
|
|
return -1; /* TODO(mkm) cover */
|
|
|
|
}
|
|
|
|
n = s - name; /* chunk length */
|
|
|
|
mbuf_append(io, &n, 1); /* send length */
|
|
|
|
mbuf_append(io, name, n);
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (*s == '.') {
|
|
|
|
n++;
|
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
name += n;
|
|
|
|
len -= n;
|
|
|
|
} while (*s != '\0');
|
|
|
|
mbuf_append(io, "\0", 1); /* Mark end of host name */
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return io->len - pos;
|
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-10-21 21:15:16 +08:00
|
|
|
int mg_dns_encode_record(struct mbuf *io, struct mg_dns_resource_record *rr,
|
2015-09-08 19:49:03 +08:00
|
|
|
const char *name, size_t nlen, const void *rdata,
|
|
|
|
size_t rlen) {
|
|
|
|
size_t pos = io->len;
|
|
|
|
uint16_t u16;
|
|
|
|
uint32_t u32;
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
if (rr->kind == MG_DNS_INVALID_RECORD) {
|
2015-09-08 19:49:03 +08:00
|
|
|
return -1; /* LCOV_EXCL_LINE */
|
2013-09-30 20:11:24 +08:00
|
|
|
}
|
|
|
|
|
2015-10-21 21:15:16 +08:00
|
|
|
if (mg_dns_encode_name(io, name, nlen) == -1) {
|
2015-09-08 19:49:03 +08:00
|
|
|
return -1;
|
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
u16 = htons(rr->rtype);
|
|
|
|
mbuf_append(io, &u16, 2);
|
|
|
|
u16 = htons(rr->rclass);
|
|
|
|
mbuf_append(io, &u16, 2);
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
if (rr->kind == MG_DNS_ANSWER) {
|
2015-09-08 19:49:03 +08:00
|
|
|
u32 = htonl(rr->ttl);
|
|
|
|
mbuf_append(io, &u32, 4);
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
if (rr->rtype == MG_DNS_CNAME_RECORD) {
|
2015-09-08 19:49:03 +08:00
|
|
|
int clen;
|
|
|
|
/* fill size after encoding */
|
|
|
|
size_t off = io->len;
|
|
|
|
mbuf_append(io, &u16, 2);
|
2015-10-21 21:15:16 +08:00
|
|
|
if ((clen = mg_dns_encode_name(io, (const char *) rdata, rlen)) == -1) {
|
2015-09-08 19:49:03 +08:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
u16 = clen;
|
|
|
|
io->buf[off] = u16 >> 8;
|
|
|
|
io->buf[off + 1] = u16 & 0xff;
|
|
|
|
} else {
|
|
|
|
u16 = htons(rlen);
|
|
|
|
mbuf_append(io, &u16, 2);
|
|
|
|
mbuf_append(io, rdata, rlen);
|
|
|
|
}
|
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return io->len - pos;
|
2013-09-30 20:11:24 +08:00
|
|
|
}
|
|
|
|
|
2015-10-21 21:15:16 +08:00
|
|
|
void mg_send_dns_query(struct mg_connection *nc, const char *name,
|
2015-09-08 19:49:03 +08:00
|
|
|
int query_type) {
|
2015-10-21 21:15:16 +08:00
|
|
|
struct mg_dns_message *msg =
|
|
|
|
(struct mg_dns_message *) MG_CALLOC(1, sizeof(*msg));
|
2015-09-08 19:49:03 +08:00
|
|
|
struct mbuf pkt;
|
2015-10-21 21:15:16 +08:00
|
|
|
struct mg_dns_resource_record *rr = &msg->questions[0];
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
DBG(("%s %d", name, query_type));
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-10-26 16:15:58 +08:00
|
|
|
mbuf_init(&pkt, 64 /* Start small, it'll grow as needed. */);
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-10-21 21:15:16 +08:00
|
|
|
msg->transaction_id = ++mg_dns_tid;
|
2015-09-08 19:49:03 +08:00
|
|
|
msg->flags = 0x100;
|
|
|
|
msg->num_questions = 1;
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-10-21 21:15:16 +08:00
|
|
|
mg_dns_insert_header(&pkt, 0, msg);
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
rr->rtype = query_type;
|
|
|
|
rr->rclass = 1; /* Class: inet */
|
2015-09-19 19:38:46 +08:00
|
|
|
rr->kind = MG_DNS_QUESTION;
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-10-21 21:15:16 +08:00
|
|
|
if (mg_dns_encode_record(&pkt, rr, name, strlen(name), NULL, 0) == -1) {
|
2015-09-08 19:49:03 +08:00
|
|
|
/* TODO(mkm): return an error code */
|
|
|
|
goto cleanup; /* LCOV_EXCL_LINE */
|
2013-09-30 20:11:24 +08:00
|
|
|
}
|
2013-12-05 19:12:07 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* TCP DNS requires messages to be prefixed with len */
|
2015-09-19 19:38:46 +08:00
|
|
|
if (!(nc->flags & MG_F_UDP)) {
|
2015-09-08 19:49:03 +08:00
|
|
|
uint16_t len = htons(pkt.len);
|
|
|
|
mbuf_insert(&pkt, 0, &len, 2);
|
|
|
|
}
|
|
|
|
|
|
|
|
mg_send(nc, pkt.buf, pkt.len);
|
|
|
|
mbuf_free(&pkt);
|
|
|
|
|
|
|
|
cleanup:
|
2015-09-19 19:38:46 +08:00
|
|
|
MG_FREE(msg);
|
2013-09-30 20:11:24 +08:00
|
|
|
}
|
|
|
|
|
2015-10-21 21:15:16 +08:00
|
|
|
static unsigned char *mg_parse_dns_resource_record(
|
|
|
|
unsigned char *data, unsigned char *end, struct mg_dns_resource_record *rr,
|
2015-09-08 19:49:03 +08:00
|
|
|
int reply) {
|
|
|
|
unsigned char *name = data;
|
|
|
|
int chunk_len, data_len;
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
while (data < end && (chunk_len = *data)) {
|
|
|
|
if (((unsigned char *) data)[0] & 0xc0) {
|
|
|
|
data += 1;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
data += chunk_len + 1;
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
rr->name.p = (char *) name;
|
|
|
|
rr->name.len = data - name + 1;
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
data++;
|
|
|
|
if (data > end - 4) {
|
|
|
|
return data;
|
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
rr->rtype = data[0] << 8 | data[1];
|
|
|
|
data += 2;
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
rr->rclass = data[0] << 8 | data[1];
|
|
|
|
data += 2;
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
rr->kind = reply ? MG_DNS_ANSWER : MG_DNS_QUESTION;
|
2015-09-08 19:49:03 +08:00
|
|
|
if (reply) {
|
|
|
|
if (data >= end - 6) {
|
|
|
|
return data;
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
rr->ttl = (uint32_t) data[0] << 24 | (uint32_t) data[1] << 16 |
|
|
|
|
data[2] << 8 | data[3];
|
|
|
|
data += 4;
|
2014-01-14 00:29:12 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
data_len = *data << 8 | *(data + 1);
|
|
|
|
data += 2;
|
2014-03-03 02:27:51 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
rr->rdata.p = (char *) data;
|
|
|
|
rr->rdata.len = data_len;
|
|
|
|
data += data_len;
|
2014-03-03 02:27:51 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
return data;
|
2014-03-03 02:27:51 +08:00
|
|
|
}
|
|
|
|
|
2015-10-21 21:15:16 +08:00
|
|
|
int mg_parse_dns(const char *buf, int len, struct mg_dns_message *msg) {
|
|
|
|
struct mg_dns_header *header = (struct mg_dns_header *) buf;
|
2015-09-08 19:49:03 +08:00
|
|
|
unsigned char *data = (unsigned char *) buf + sizeof(*header);
|
|
|
|
unsigned char *end = (unsigned char *) buf + len;
|
|
|
|
int i;
|
|
|
|
msg->pkt.p = buf;
|
|
|
|
msg->pkt.len = len;
|
2014-03-03 02:27:51 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (len < (int) sizeof(*header)) {
|
|
|
|
return -1; /* LCOV_EXCL_LINE */
|
2014-03-03 02:27:51 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
msg->transaction_id = header->transaction_id;
|
|
|
|
msg->flags = ntohs(header->flags);
|
|
|
|
msg->num_questions = ntohs(header->num_questions);
|
|
|
|
msg->num_answers = ntohs(header->num_answers);
|
2014-03-03 02:27:51 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
for (i = 0; i < msg->num_questions && i < (int) ARRAY_SIZE(msg->questions);
|
|
|
|
i++) {
|
2015-10-21 21:15:16 +08:00
|
|
|
data = mg_parse_dns_resource_record(data, end, &msg->questions[i], 0);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2014-03-03 02:27:51 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
for (i = 0; i < msg->num_answers && i < (int) ARRAY_SIZE(msg->answers); i++) {
|
2015-10-21 21:15:16 +08:00
|
|
|
data = mg_parse_dns_resource_record(data, end, &msg->answers[i], 1);
|
2014-03-03 02:27:51 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
return 0;
|
2014-03-03 02:27:51 +08:00
|
|
|
}
|
|
|
|
|
2015-10-21 21:15:16 +08:00
|
|
|
size_t mg_dns_uncompress_name(struct mg_dns_message *msg, struct mg_str *name,
|
2015-09-08 19:49:03 +08:00
|
|
|
char *dst, int dst_len) {
|
|
|
|
int chunk_len;
|
|
|
|
char *old_dst = dst;
|
|
|
|
const unsigned char *data = (unsigned char *) name->p;
|
|
|
|
const unsigned char *end = (unsigned char *) msg->pkt.p + msg->pkt.len;
|
2014-03-03 02:27:51 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (data >= end) {
|
|
|
|
return 0;
|
2014-03-03 02:27:51 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
while ((chunk_len = *data++)) {
|
|
|
|
int leeway = dst_len - (dst - old_dst);
|
|
|
|
if (data >= end) {
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (chunk_len & 0xc0) {
|
|
|
|
uint16_t off = (data[-1] & (~0xc0)) << 8 | data[0];
|
|
|
|
if (off >= msg->pkt.len) {
|
|
|
|
return 0;
|
2014-03-03 02:27:51 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
data = (unsigned char *) msg->pkt.p + off;
|
|
|
|
continue;
|
2014-03-03 02:27:51 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
if (chunk_len > leeway) {
|
|
|
|
chunk_len = leeway;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (data + chunk_len >= end) {
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
memcpy(dst, data, chunk_len);
|
|
|
|
data += chunk_len;
|
|
|
|
dst += chunk_len;
|
|
|
|
leeway -= chunk_len;
|
|
|
|
if (leeway == 0) {
|
|
|
|
return dst - old_dst;
|
|
|
|
}
|
|
|
|
*dst++ = '.';
|
2014-03-03 02:27:51 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (dst != old_dst) {
|
|
|
|
*--dst = 0;
|
2014-03-03 02:27:51 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
return dst - old_dst;
|
2014-03-03 02:27:51 +08:00
|
|
|
}
|
|
|
|
|
2015-10-21 21:15:16 +08:00
|
|
|
static void dns_handler(struct mg_connection *nc, int ev, void *ev_data) {
|
2015-09-08 19:49:03 +08:00
|
|
|
struct mbuf *io = &nc->recv_mbuf;
|
2015-10-21 21:15:16 +08:00
|
|
|
struct mg_dns_message msg;
|
2014-03-03 02:27:51 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Pass low-level events to the user handler */
|
|
|
|
nc->handler(nc, ev, ev_data);
|
2014-03-03 02:27:51 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
switch (ev) {
|
2015-09-19 19:38:46 +08:00
|
|
|
case MG_EV_RECV:
|
|
|
|
if (!(nc->flags & MG_F_UDP)) {
|
2015-09-08 19:49:03 +08:00
|
|
|
mbuf_remove(&nc->recv_mbuf, 2);
|
|
|
|
}
|
|
|
|
if (mg_parse_dns(nc->recv_mbuf.buf, nc->recv_mbuf.len, &msg) == -1) {
|
|
|
|
/* reply + recursion allowed + format error */
|
|
|
|
memset(&msg, 0, sizeof(msg));
|
|
|
|
msg.flags = 0x8081;
|
2015-10-21 21:15:16 +08:00
|
|
|
mg_dns_insert_header(io, 0, &msg);
|
2015-09-19 19:38:46 +08:00
|
|
|
if (!(nc->flags & MG_F_UDP)) {
|
2015-09-08 19:49:03 +08:00
|
|
|
uint16_t len = htons(io->len);
|
|
|
|
mbuf_insert(io, 0, &len, 2);
|
|
|
|
}
|
|
|
|
mg_send(nc, io->buf, io->len);
|
|
|
|
} else {
|
|
|
|
/* Call user handler with parsed message */
|
2015-09-19 19:38:46 +08:00
|
|
|
nc->handler(nc, MG_DNS_MESSAGE, &msg);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
mbuf_remove(io, io->len);
|
|
|
|
break;
|
2014-05-30 00:04:13 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_set_protocol_dns(struct mg_connection *nc) {
|
2015-10-21 21:15:16 +08:00
|
|
|
nc->proto_handler = dns_handler;
|
2014-06-11 16:56:34 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#endif /* MG_DISABLE_DNS */
|
2015-09-08 19:49:03 +08:00
|
|
|
#ifdef NS_MODULE_LINES
|
2015-12-15 02:45:18 +08:00
|
|
|
#line 1 "./src/dns-server.c"
|
2015-09-08 19:49:03 +08:00
|
|
|
/**/
|
2014-06-11 16:56:34 +08:00
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Copyright (c) 2014 Cesanta Software Limited
|
|
|
|
* All rights reserved
|
|
|
|
*/
|
2014-06-11 16:56:34 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifdef MG_ENABLE_DNS_SERVER
|
2014-08-01 03:04:59 +08:00
|
|
|
|
2015-12-12 22:05:54 +08:00
|
|
|
/* Amalgamated: #include "mongoose/src/internal.h" */
|
2015-12-12 22:16:48 +08:00
|
|
|
/* Amalgamated: #include "mongoose/src/dns-server.h" */
|
2014-08-01 03:04:59 +08:00
|
|
|
|
2015-10-21 21:15:16 +08:00
|
|
|
struct mg_dns_reply mg_dns_create_reply(struct mbuf *io,
|
|
|
|
struct mg_dns_message *msg) {
|
|
|
|
struct mg_dns_reply rep;
|
2015-09-08 19:49:03 +08:00
|
|
|
rep.msg = msg;
|
|
|
|
rep.io = io;
|
|
|
|
rep.start = io->len;
|
2014-08-01 03:04:59 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* reply + recursion allowed */
|
|
|
|
msg->flags |= 0x8080;
|
2015-10-21 21:15:16 +08:00
|
|
|
mg_dns_copy_body(io, msg);
|
2014-08-01 03:04:59 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
msg->num_answers = 0;
|
|
|
|
return rep;
|
|
|
|
}
|
2014-05-18 20:22:33 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
void mg_dns_send_reply(struct mg_connection *nc, struct mg_dns_reply *r) {
|
2015-09-08 19:49:03 +08:00
|
|
|
size_t sent = r->io->len - r->start;
|
2015-10-21 21:15:16 +08:00
|
|
|
mg_dns_insert_header(r->io, r->start, r->msg);
|
2015-09-19 19:38:46 +08:00
|
|
|
if (!(nc->flags & MG_F_UDP)) {
|
2015-09-08 19:49:03 +08:00
|
|
|
uint16_t len = htons(sent);
|
|
|
|
mbuf_insert(r->io, r->start, &len, 2);
|
2014-06-04 19:32:47 +08:00
|
|
|
}
|
2014-05-21 23:18:21 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
if (&nc->send_mbuf != r->io) {
|
|
|
|
mg_send(nc, r->io->buf + r->start, r->io->len - r->start);
|
2015-09-08 19:49:03 +08:00
|
|
|
r->io->len = r->start;
|
2014-05-18 20:22:33 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-10-21 21:15:16 +08:00
|
|
|
int mg_dns_reply_record(struct mg_dns_reply *reply,
|
|
|
|
struct mg_dns_resource_record *question,
|
2015-09-08 19:49:03 +08:00
|
|
|
const char *name, int rtype, int ttl, const void *rdata,
|
|
|
|
size_t rdata_len) {
|
2015-10-21 21:15:16 +08:00
|
|
|
struct mg_dns_message *msg = (struct mg_dns_message *) reply->msg;
|
2015-09-08 19:49:03 +08:00
|
|
|
char rname[512];
|
2015-10-21 21:15:16 +08:00
|
|
|
struct mg_dns_resource_record *ans = &msg->answers[msg->num_answers];
|
2015-09-19 19:38:46 +08:00
|
|
|
if (msg->num_answers >= MG_MAX_DNS_ANSWERS) {
|
2015-09-08 19:49:03 +08:00
|
|
|
return -1; /* LCOV_EXCL_LINE */
|
|
|
|
}
|
2014-06-04 01:02:55 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (name == NULL) {
|
|
|
|
name = rname;
|
|
|
|
rname[511] = 0;
|
2015-10-21 21:15:16 +08:00
|
|
|
mg_dns_uncompress_name(msg, &question->name, rname, sizeof(rname) - 1);
|
2014-06-04 01:02:55 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
*ans = *question;
|
2015-09-19 19:38:46 +08:00
|
|
|
ans->kind = MG_DNS_ANSWER;
|
2015-09-08 19:49:03 +08:00
|
|
|
ans->rtype = rtype;
|
|
|
|
ans->ttl = ttl;
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-10-21 21:15:16 +08:00
|
|
|
if (mg_dns_encode_record(reply->io, ans, name, strlen(name), rdata,
|
2015-09-08 19:49:03 +08:00
|
|
|
rdata_len) == -1) {
|
|
|
|
return -1; /* LCOV_EXCL_LINE */
|
|
|
|
};
|
2014-03-10 16:35:41 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
msg->num_answers++;
|
|
|
|
return 0;
|
|
|
|
}
|
2014-01-25 04:49:57 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#endif /* MG_ENABLE_DNS_SERVER */
|
2015-09-08 19:49:03 +08:00
|
|
|
#ifdef NS_MODULE_LINES
|
2015-12-15 02:45:18 +08:00
|
|
|
#line 1 "./src/resolv.c"
|
2015-09-08 19:49:03 +08:00
|
|
|
/**/
|
2013-12-31 21:57:51 +08:00
|
|
|
#endif
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Copyright (c) 2014 Cesanta Software Limited
|
|
|
|
* All rights reserved
|
|
|
|
*/
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_DISABLE_RESOLVER
|
2014-08-04 01:56:19 +08:00
|
|
|
|
2015-12-12 22:05:54 +08:00
|
|
|
/* Amalgamated: #include "mongoose/src/internal.h" */
|
2015-12-12 22:16:48 +08:00
|
|
|
/* Amalgamated: #include "mongoose/src/resolv.h" */
|
2014-08-04 01:56:19 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_DEFAULT_NAMESERVER
|
|
|
|
#define MG_DEFAULT_NAMESERVER "8.8.8.8"
|
2013-09-30 20:11:24 +08:00
|
|
|
#endif
|
|
|
|
|
2015-10-21 21:15:16 +08:00
|
|
|
static const char *mg_default_dns_server = "udp://" MG_DEFAULT_NAMESERVER ":53";
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-10-21 21:15:16 +08:00
|
|
|
MG_INTERNAL char mg_dns_server[256];
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
struct mg_resolve_async_request {
|
|
|
|
char name[1024];
|
|
|
|
int query;
|
|
|
|
mg_resolve_callback_t callback;
|
|
|
|
void *data;
|
|
|
|
time_t timeout;
|
|
|
|
int max_retries;
|
2015-12-08 20:25:44 +08:00
|
|
|
enum mg_resolve_err err;
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* state */
|
|
|
|
time_t last_time;
|
|
|
|
int retries;
|
|
|
|
};
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Find what nameserver to use.
|
|
|
|
*
|
|
|
|
* Return 0 if OK, -1 if error
|
|
|
|
*/
|
|
|
|
static int mg_get_ip_address_of_nameserver(char *name, size_t name_len) {
|
|
|
|
int ret = -1;
|
|
|
|
|
|
|
|
#ifdef _WIN32
|
|
|
|
int i;
|
|
|
|
LONG err;
|
|
|
|
HKEY hKey, hSub;
|
|
|
|
char subkey[512], value[128],
|
|
|
|
*key = "SYSTEM\\ControlSet001\\Services\\Tcpip\\Parameters\\Interfaces";
|
|
|
|
|
|
|
|
if ((err = RegOpenKey(HKEY_LOCAL_MACHINE, key, &hKey)) != ERROR_SUCCESS) {
|
|
|
|
fprintf(stderr, "cannot open reg key %s: %d\n", key, err);
|
|
|
|
ret = -1;
|
|
|
|
} else {
|
|
|
|
for (ret = -1, i = 0;
|
|
|
|
RegEnumKey(hKey, i, subkey, sizeof(subkey)) == ERROR_SUCCESS; i++) {
|
|
|
|
DWORD type, len = sizeof(value);
|
|
|
|
if (RegOpenKey(hKey, subkey, &hSub) == ERROR_SUCCESS &&
|
|
|
|
(RegQueryValueEx(hSub, "NameServer", 0, &type, (void *) value,
|
|
|
|
&len) == ERROR_SUCCESS ||
|
|
|
|
RegQueryValueEx(hSub, "DhcpNameServer", 0, &type, (void *) value,
|
|
|
|
&len) == ERROR_SUCCESS)) {
|
|
|
|
/*
|
|
|
|
* See https://github.com/cesanta/mongoose/issues/176
|
|
|
|
* The value taken from the registry can be empty, a single
|
|
|
|
* IP address, or multiple IP addresses separated by comma.
|
|
|
|
* If it's empty, check the next interface.
|
|
|
|
* If it's multiple IP addresses, take the first one.
|
|
|
|
*/
|
|
|
|
char *comma = strchr(value, ',');
|
|
|
|
if (value[0] == '\0') {
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
if (comma != NULL) {
|
|
|
|
*comma = '\0';
|
|
|
|
}
|
|
|
|
snprintf(name, name_len, "udp://%s:53", value);
|
|
|
|
ret = 0;
|
|
|
|
RegCloseKey(hSub);
|
|
|
|
break;
|
|
|
|
}
|
2014-01-20 00:32:43 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
RegCloseKey(hKey);
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
2015-09-19 19:38:46 +08:00
|
|
|
#elif !defined(MG_DISABLE_FILESYSTEM)
|
2015-09-08 19:49:03 +08:00
|
|
|
FILE *fp;
|
|
|
|
char line[512];
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if ((fp = fopen("/etc/resolv.conf", "r")) == NULL) {
|
|
|
|
ret = -1;
|
2014-05-30 00:04:13 +08:00
|
|
|
} else {
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Try to figure out what nameserver to use */
|
|
|
|
for (ret = -1; fgets(line, sizeof(line), fp) != NULL;) {
|
|
|
|
char buf[256];
|
|
|
|
if (sscanf(line, "nameserver %255[^\n\t #]s", buf) == 1) {
|
|
|
|
snprintf(name, name_len, "udp://%s:53", buf);
|
|
|
|
ret = 0;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
(void) fclose(fp);
|
2014-05-30 00:04:13 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
#else
|
2015-10-21 21:15:16 +08:00
|
|
|
snprintf(name, name_len, "%s", mg_default_dns_server);
|
2015-09-08 19:49:03 +08:00
|
|
|
#endif /* _WIN32 */
|
|
|
|
|
|
|
|
return ret;
|
2014-05-22 14:31:19 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
int mg_resolve_from_hosts_file(const char *name, union socket_address *usa) {
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifndef MG_DISABLE_FILESYSTEM
|
2015-09-08 19:49:03 +08:00
|
|
|
/* TODO(mkm) cache /etc/hosts */
|
|
|
|
FILE *fp;
|
|
|
|
char line[1024];
|
|
|
|
char *p;
|
|
|
|
char alias[256];
|
|
|
|
unsigned int a, b, c, d;
|
|
|
|
int len = 0;
|
2014-01-20 00:32:43 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if ((fp = fopen("/etc/hosts", "r")) == NULL) {
|
|
|
|
return -1;
|
2014-05-30 00:04:13 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
for (; fgets(line, sizeof(line), fp) != NULL;) {
|
|
|
|
if (line[0] == '#') continue;
|
|
|
|
|
|
|
|
if (sscanf(line, "%u.%u.%u.%u%n", &a, &b, &c, &d, &len) == 0) {
|
|
|
|
/* TODO(mkm): handle ipv6 */
|
|
|
|
continue;
|
2015-06-22 18:31:08 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
for (p = line + len; sscanf(p, "%s%n", alias, &len) == 1; p += len) {
|
|
|
|
if (strcmp(alias, name) == 0) {
|
|
|
|
usa->sin.sin_addr.s_addr = htonl(a << 24 | b << 16 | c << 8 | d);
|
|
|
|
fclose(fp);
|
|
|
|
return 0;
|
|
|
|
}
|
2015-06-22 18:31:08 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
fclose(fp);
|
2013-12-09 22:18:49 +08:00
|
|
|
#endif
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void mg_resolve_async_eh(struct mg_connection *nc, int ev, void *data) {
|
|
|
|
time_t now = time(NULL);
|
|
|
|
struct mg_resolve_async_request *req;
|
2015-10-21 21:15:16 +08:00
|
|
|
struct mg_dns_message *msg;
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
DBG(("ev=%d", ev));
|
|
|
|
|
|
|
|
req = (struct mg_resolve_async_request *) nc->user_data;
|
|
|
|
|
|
|
|
switch (ev) {
|
2015-09-19 19:38:46 +08:00
|
|
|
case MG_EV_CONNECT:
|
|
|
|
case MG_EV_POLL:
|
2015-09-08 19:49:03 +08:00
|
|
|
if (req->retries > req->max_retries) {
|
2015-12-08 20:25:44 +08:00
|
|
|
req->err = MG_RESOLVE_EXCEEDED_RETRY_COUNT;
|
2015-09-19 19:38:46 +08:00
|
|
|
nc->flags |= MG_F_CLOSE_IMMEDIATELY;
|
2015-09-08 19:49:03 +08:00
|
|
|
break;
|
|
|
|
}
|
2015-10-26 16:15:58 +08:00
|
|
|
if (now - req->last_time >= req->timeout) {
|
2015-10-21 21:15:16 +08:00
|
|
|
mg_send_dns_query(nc, req->name, req->query);
|
2015-09-08 19:49:03 +08:00
|
|
|
req->last_time = now;
|
|
|
|
req->retries++;
|
|
|
|
}
|
|
|
|
break;
|
2015-09-19 19:38:46 +08:00
|
|
|
case MG_EV_RECV:
|
2015-10-21 21:15:16 +08:00
|
|
|
msg = (struct mg_dns_message *) MG_MALLOC(sizeof(*msg));
|
2015-09-08 19:49:03 +08:00
|
|
|
if (mg_parse_dns(nc->recv_mbuf.buf, *(int *) data, msg) == 0 &&
|
|
|
|
msg->num_answers > 0) {
|
2015-12-08 20:25:44 +08:00
|
|
|
req->callback(msg, req->data, MG_RESOLVE_OK);
|
2015-11-17 16:48:35 +08:00
|
|
|
nc->user_data = NULL;
|
|
|
|
MG_FREE(req);
|
2015-12-08 20:25:44 +08:00
|
|
|
} else {
|
|
|
|
req->err = MG_RESOLVE_NO_ANSWERS;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2015-09-19 19:38:46 +08:00
|
|
|
MG_FREE(msg);
|
2015-10-24 07:05:05 +08:00
|
|
|
nc->flags |= MG_F_CLOSE_IMMEDIATELY;
|
2015-09-08 19:49:03 +08:00
|
|
|
break;
|
2015-11-17 16:48:35 +08:00
|
|
|
case MG_EV_SEND:
|
|
|
|
/*
|
|
|
|
* If a send error occurs, prevent closing of the connection by the core.
|
|
|
|
* We will retry after timeout.
|
|
|
|
*/
|
|
|
|
nc->flags &= ~MG_F_CLOSE_IMMEDIATELY;
|
|
|
|
mbuf_remove(&nc->send_mbuf, nc->send_mbuf.len);
|
|
|
|
break;
|
2015-12-08 20:25:44 +08:00
|
|
|
case MG_EV_TIMER:
|
|
|
|
req->err = MG_RESOLVE_TIMEOUT;
|
|
|
|
nc->flags |= MG_F_CLOSE_IMMEDIATELY;
|
|
|
|
break;
|
2015-11-17 16:48:35 +08:00
|
|
|
case MG_EV_CLOSE:
|
|
|
|
/* If we got here with request still not done, fire an error callback. */
|
|
|
|
if (req != NULL) {
|
2015-12-08 20:25:44 +08:00
|
|
|
req->callback(NULL, req->data, req->err);
|
2015-11-17 16:48:35 +08:00
|
|
|
nc->user_data = NULL;
|
|
|
|
MG_FREE(req);
|
|
|
|
}
|
|
|
|
break;
|
2013-09-30 20:11:24 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
int mg_resolve_async(struct mg_mgr *mgr, const char *name, int query,
|
|
|
|
mg_resolve_callback_t cb, void *data) {
|
2015-10-24 07:31:10 +08:00
|
|
|
struct mg_resolve_async_opts opts;
|
|
|
|
memset(&opts, 0, sizeof(opts));
|
2015-09-08 19:49:03 +08:00
|
|
|
return mg_resolve_async_opt(mgr, name, query, cb, data, opts);
|
|
|
|
}
|
|
|
|
|
|
|
|
int mg_resolve_async_opt(struct mg_mgr *mgr, const char *name, int query,
|
|
|
|
mg_resolve_callback_t cb, void *data,
|
|
|
|
struct mg_resolve_async_opts opts) {
|
|
|
|
struct mg_resolve_async_request *req;
|
2015-10-21 21:15:16 +08:00
|
|
|
struct mg_connection *dns_nc;
|
2015-09-08 19:49:03 +08:00
|
|
|
const char *nameserver = opts.nameserver_url;
|
|
|
|
|
2015-12-08 20:25:44 +08:00
|
|
|
DBG(("%s %d %p", name, query, opts.dns_conn));
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
/* resolve with DNS */
|
2015-09-19 19:38:46 +08:00
|
|
|
req = (struct mg_resolve_async_request *) MG_CALLOC(1, sizeof(*req));
|
2015-09-08 19:49:03 +08:00
|
|
|
if (req == NULL) {
|
|
|
|
return -1;
|
2014-01-20 00:32:43 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
strncpy(req->name, name, sizeof(req->name));
|
|
|
|
req->query = query;
|
|
|
|
req->callback = cb;
|
|
|
|
req->data = data;
|
|
|
|
/* TODO(mkm): parse defaults out of resolve.conf */
|
|
|
|
req->max_retries = opts.max_retries ? opts.max_retries : 2;
|
|
|
|
req->timeout = opts.timeout ? opts.timeout : 5;
|
|
|
|
|
|
|
|
/* Lazily initialize dns server */
|
2015-10-21 21:15:16 +08:00
|
|
|
if (nameserver == NULL && mg_dns_server[0] == '\0' &&
|
|
|
|
mg_get_ip_address_of_nameserver(mg_dns_server, sizeof(mg_dns_server)) ==
|
2015-09-08 19:49:03 +08:00
|
|
|
-1) {
|
2015-10-21 21:15:16 +08:00
|
|
|
strncpy(mg_dns_server, mg_default_dns_server, sizeof(mg_dns_server));
|
2014-01-20 00:32:43 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (nameserver == NULL) {
|
2015-10-21 21:15:16 +08:00
|
|
|
nameserver = mg_dns_server;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2014-01-20 00:32:43 +08:00
|
|
|
|
2015-10-21 21:15:16 +08:00
|
|
|
dns_nc = mg_connect(mgr, nameserver, mg_resolve_async_eh);
|
|
|
|
if (dns_nc == NULL) {
|
2015-09-08 19:49:03 +08:00
|
|
|
free(req);
|
|
|
|
return -1;
|
2014-01-20 00:32:43 +08:00
|
|
|
}
|
2015-10-21 21:15:16 +08:00
|
|
|
dns_nc->user_data = req;
|
2015-12-08 20:25:44 +08:00
|
|
|
if (opts.dns_conn != NULL) {
|
|
|
|
*opts.dns_conn = dns_nc;
|
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
return 0;
|
2014-01-20 00:32:43 +08:00
|
|
|
}
|
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#endif /* MG_DISABLE_RESOLVE */
|
2015-09-08 19:49:03 +08:00
|
|
|
#ifdef NS_MODULE_LINES
|
2015-12-15 02:45:18 +08:00
|
|
|
#line 1 "./src/coap.c"
|
2015-09-08 19:49:03 +08:00
|
|
|
/**/
|
|
|
|
#endif
|
|
|
|
/*
|
|
|
|
* Copyright (c) 2015 Cesanta Software Limited
|
|
|
|
* 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>.
|
|
|
|
*/
|
|
|
|
|
2015-12-12 22:05:54 +08:00
|
|
|
/* Amalgamated: #include "mongoose/src/internal.h" */
|
2015-12-12 22:16:48 +08:00
|
|
|
/* Amalgamated: #include "mongoose/src/coap.h" */
|
2014-01-17 23:42:35 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#ifdef MG_ENABLE_COAP
|
2014-01-17 23:42:35 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
void mg_coap_free_options(struct mg_coap_message *cm) {
|
|
|
|
while (cm->options != NULL) {
|
|
|
|
struct mg_coap_option *next = cm->options->next;
|
2015-09-19 19:38:46 +08:00
|
|
|
MG_FREE(cm->options);
|
2015-09-08 19:49:03 +08:00
|
|
|
cm->options = next;
|
2014-01-17 23:42:35 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2014-01-17 23:42:35 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
struct mg_coap_option *mg_coap_add_option(struct mg_coap_message *cm,
|
|
|
|
uint32_t number, char *value,
|
|
|
|
size_t len) {
|
|
|
|
struct mg_coap_option *new_option =
|
2015-09-19 19:38:46 +08:00
|
|
|
(struct mg_coap_option *) MG_CALLOC(1, sizeof(*new_option));
|
2014-02-20 00:34:51 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
new_option->number = number;
|
|
|
|
new_option->value.p = value;
|
|
|
|
new_option->value.len = len;
|
2014-01-17 23:42:35 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (cm->options == NULL) {
|
|
|
|
cm->options = cm->optiomg_tail = new_option;
|
|
|
|
} else {
|
|
|
|
/*
|
|
|
|
* A very simple attention to help clients to compose options:
|
|
|
|
* CoAP wants to see options ASC ordered.
|
|
|
|
* Could be change by using sort in coap_compose
|
|
|
|
*/
|
|
|
|
if (cm->optiomg_tail->number <= new_option->number) {
|
|
|
|
/* if option is already ordered just add it */
|
|
|
|
cm->optiomg_tail = cm->optiomg_tail->next = new_option;
|
|
|
|
} else {
|
|
|
|
/* looking for appropriate position */
|
|
|
|
struct mg_coap_option *current_opt = cm->options;
|
|
|
|
struct mg_coap_option *prev_opt = 0;
|
2014-01-17 23:42:35 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
while (current_opt != NULL) {
|
|
|
|
if (current_opt->number > new_option->number) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
prev_opt = current_opt;
|
|
|
|
current_opt = current_opt->next;
|
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (prev_opt != NULL) {
|
|
|
|
prev_opt->next = new_option;
|
|
|
|
new_option->next = current_opt;
|
|
|
|
} else {
|
|
|
|
/* insert new_option to the beginning */
|
|
|
|
new_option->next = cm->options;
|
|
|
|
cm->options = new_option;
|
|
|
|
}
|
|
|
|
}
|
2013-09-30 20:11:24 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return new_option;
|
2013-10-01 22:16:09 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Fills CoAP header in mg_coap_message.
|
|
|
|
*
|
|
|
|
* Helper function.
|
|
|
|
*/
|
|
|
|
static char *coap_parse_header(char *ptr, struct mbuf *io,
|
|
|
|
struct mg_coap_message *cm) {
|
|
|
|
if (io->len < sizeof(uint32_t)) {
|
2015-09-19 19:38:46 +08:00
|
|
|
cm->flags |= MG_COAP_NOT_ENOUGH_DATA;
|
2015-09-08 19:49:03 +08:00
|
|
|
return NULL;
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
2013-10-01 22:16:09 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Version (Ver): 2-bit unsigned integer. Indicates the CoAP version
|
|
|
|
* number. Implementations of this specification MUST set this field
|
|
|
|
* to 1 (01 binary). Other values are reserved for future versions.
|
|
|
|
* Messages with unknown version numbers MUST be silently ignored.
|
|
|
|
*/
|
|
|
|
if (((uint8_t) *ptr >> 6) != 1) {
|
2015-09-19 19:38:46 +08:00
|
|
|
cm->flags |= MG_COAP_IGNORE;
|
2015-09-08 19:49:03 +08:00
|
|
|
return NULL;
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
2013-10-01 22:16:09 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Type (T): 2-bit unsigned integer. Indicates if this message is of
|
|
|
|
* type Confirmable (0), Non-confirmable (1), Acknowledgement (2), or
|
|
|
|
* Reset (3).
|
|
|
|
*/
|
|
|
|
cm->msg_type = ((uint8_t) *ptr & 0x30) >> 4;
|
2015-09-19 19:38:46 +08:00
|
|
|
cm->flags |= MG_COAP_MSG_TYPE_FIELD;
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
/*
|
|
|
|
* Token Length (TKL): 4-bit unsigned integer. Indicates the length of
|
|
|
|
* the variable-length Token field (0-8 bytes). Lengths 9-15 are
|
|
|
|
* reserved, MUST NOT be sent, and MUST be processed as a message
|
|
|
|
* format error.
|
|
|
|
*/
|
|
|
|
cm->token.len = *ptr & 0x0F;
|
|
|
|
if (cm->token.len > 8) {
|
2015-09-19 19:38:46 +08:00
|
|
|
cm->flags |= MG_COAP_FORMAT_ERROR;
|
2015-09-08 19:49:03 +08:00
|
|
|
return NULL;
|
2013-10-01 22:16:09 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
ptr++;
|
2014-06-04 00:00:52 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Code: 8-bit unsigned integer, split into a 3-bit class (most
|
|
|
|
* significant bits) and a 5-bit detail (least significant bits)
|
|
|
|
*/
|
|
|
|
cm->code_class = (uint8_t) *ptr >> 5;
|
|
|
|
cm->code_detail = *ptr & 0x1F;
|
2015-09-19 19:38:46 +08:00
|
|
|
cm->flags |= (MG_COAP_CODE_CLASS_FIELD | MG_COAP_CODE_DETAIL_FIELD);
|
2014-06-04 00:00:52 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
ptr++;
|
2013-10-01 22:16:09 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Message ID: 16-bit unsigned integer in network byte order. */
|
|
|
|
cm->msg_id = (uint8_t) *ptr << 8 | (uint8_t) * (ptr + 1);
|
2015-09-19 19:38:46 +08:00
|
|
|
cm->flags |= MG_COAP_MSG_ID_FIELD;
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
ptr += 2;
|
|
|
|
|
|
|
|
return ptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Fills token information in mg_coap_message.
|
|
|
|
*
|
|
|
|
* Helper function.
|
|
|
|
*/
|
|
|
|
static char *coap_get_token(char *ptr, struct mbuf *io,
|
|
|
|
struct mg_coap_message *cm) {
|
|
|
|
if (cm->token.len != 0) {
|
|
|
|
if (ptr + cm->token.len > io->buf + io->len) {
|
2015-09-19 19:38:46 +08:00
|
|
|
cm->flags |= MG_COAP_NOT_ENOUGH_DATA;
|
2015-09-08 19:49:03 +08:00
|
|
|
return NULL;
|
|
|
|
} else {
|
|
|
|
cm->token.p = ptr;
|
|
|
|
ptr += cm->token.len;
|
2015-09-19 19:38:46 +08:00
|
|
|
cm->flags |= MG_COAP_TOKEN_FIELD;
|
2013-10-01 22:16:09 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return ptr;
|
2013-10-01 22:16:09 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Returns Option Delta or Length.
|
|
|
|
*
|
|
|
|
* Helper function.
|
|
|
|
*/
|
|
|
|
static int coap_get_ext_opt(char *ptr, struct mbuf *io, uint16_t *opt_info) {
|
|
|
|
int ret = 0;
|
2014-02-20 00:34:51 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (*opt_info == 13) {
|
|
|
|
/*
|
|
|
|
* 13: An 8-bit unsigned integer follows the initial byte and
|
|
|
|
* indicates the Option Delta/Length minus 13.
|
|
|
|
*/
|
|
|
|
if (ptr < io->buf + io->len) {
|
|
|
|
*opt_info = (uint8_t) *ptr + 13;
|
|
|
|
ret = sizeof(uint8_t);
|
|
|
|
} else {
|
|
|
|
ret = -1; /* LCOV_EXCL_LINE */
|
|
|
|
}
|
|
|
|
} else if (*opt_info == 14) {
|
|
|
|
/*
|
|
|
|
* 14: A 16-bit unsigned integer in network byte order follows the
|
|
|
|
* initial byte and indicates the Option Delta/Length minus 269.
|
|
|
|
*/
|
|
|
|
if (ptr + sizeof(uint8_t) < io->buf + io->len) {
|
|
|
|
*opt_info = ((uint8_t) *ptr << 8 | (uint8_t) * (ptr + 1)) + 269;
|
|
|
|
ret = sizeof(uint16_t);
|
|
|
|
} else {
|
|
|
|
ret = -1; /* LCOV_EXCL_LINE */
|
2014-01-10 19:23:29 +08:00
|
|
|
}
|
2013-10-01 22:16:09 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
return ret;
|
2013-10-01 22:16:09 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Fills options in mg_coap_message.
|
|
|
|
*
|
|
|
|
* Helper function.
|
|
|
|
*
|
|
|
|
* General options format:
|
|
|
|
* +---------------+---------------+
|
|
|
|
* | Option Delta | Option Length | 1 byte
|
|
|
|
* +---------------+---------------+
|
|
|
|
* \ Option Delta (extended) \ 0-2 bytes
|
|
|
|
* +-------------------------------+
|
|
|
|
* / Option Length (extended) \ 0-2 bytes
|
|
|
|
* +-------------------------------+
|
|
|
|
* \ Option Value \ 0 or more bytes
|
|
|
|
* +-------------------------------+
|
|
|
|
*/
|
|
|
|
static char *coap_get_options(char *ptr, struct mbuf *io,
|
|
|
|
struct mg_coap_message *cm) {
|
|
|
|
uint16_t prev_opt = 0;
|
|
|
|
|
|
|
|
if (ptr == io->buf + io->len) {
|
|
|
|
/* end of packet, ok */
|
2014-09-10 01:07:55 +08:00
|
|
|
return NULL;
|
|
|
|
}
|
2013-10-01 22:16:09 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* 0xFF is payload marker */
|
|
|
|
while (ptr < io->buf + io->len && (uint8_t) *ptr != 0xFF) {
|
|
|
|
uint16_t option_delta, option_lenght;
|
|
|
|
int optinfo_len;
|
2013-10-01 22:16:09 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* Option Delta: 4-bit unsigned integer */
|
|
|
|
option_delta = ((uint8_t) *ptr & 0xF0) >> 4;
|
|
|
|
/* Option Length: 4-bit unsigned integer */
|
|
|
|
option_lenght = *ptr & 0x0F;
|
2013-10-01 22:16:09 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (option_delta == 15 || option_lenght == 15) {
|
|
|
|
/*
|
|
|
|
* 15: Reserved for future use. If the field is set to this value,
|
|
|
|
* it MUST be processed as a message format error
|
|
|
|
*/
|
2015-09-19 19:38:46 +08:00
|
|
|
cm->flags |= MG_COAP_FORMAT_ERROR;
|
2015-09-08 19:49:03 +08:00
|
|
|
break;
|
|
|
|
}
|
2013-10-01 22:16:09 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
ptr++;
|
2015-05-16 02:49:35 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* check for extended option delta */
|
|
|
|
optinfo_len = coap_get_ext_opt(ptr, io, &option_delta);
|
|
|
|
if (optinfo_len == -1) {
|
2015-09-19 19:38:46 +08:00
|
|
|
cm->flags |= MG_COAP_NOT_ENOUGH_DATA; /* LCOV_EXCL_LINE */
|
2015-09-08 19:49:03 +08:00
|
|
|
break; /* LCOV_EXCL_LINE */
|
|
|
|
}
|
2013-10-01 22:16:09 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
ptr += optinfo_len;
|
2013-10-01 22:16:09 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* check or extended option lenght */
|
|
|
|
optinfo_len = coap_get_ext_opt(ptr, io, &option_lenght);
|
|
|
|
if (optinfo_len == -1) {
|
2015-09-19 19:38:46 +08:00
|
|
|
cm->flags |= MG_COAP_NOT_ENOUGH_DATA; /* LCOV_EXCL_LINE */
|
2015-09-08 19:49:03 +08:00
|
|
|
break; /* LCOV_EXCL_LINE */
|
|
|
|
}
|
2013-10-01 22:16:09 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
ptr += optinfo_len;
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Instead of specifying the Option Number directly, the instances MUST
|
|
|
|
* appear in order of their Option Numbers and a delta encoding is used
|
|
|
|
* between them.
|
|
|
|
*/
|
|
|
|
option_delta += prev_opt;
|
|
|
|
|
|
|
|
mg_coap_add_option(cm, option_delta, ptr, option_lenght);
|
|
|
|
|
|
|
|
prev_opt = option_delta;
|
|
|
|
|
|
|
|
if (ptr + option_lenght > io->buf + io->len) {
|
2015-09-19 19:38:46 +08:00
|
|
|
cm->flags |= MG_COAP_NOT_ENOUGH_DATA; /* LCOV_EXCL_LINE */
|
2015-09-08 19:49:03 +08:00
|
|
|
break; /* LCOV_EXCL_LINE */
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
ptr += option_lenght;
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
2013-10-01 22:16:09 +08:00
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
if ((cm->flags & MG_COAP_ERROR) != 0) {
|
2015-09-08 19:49:03 +08:00
|
|
|
mg_coap_free_options(cm);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
cm->flags |= MG_COAP_OPTIOMG_FIELD;
|
2013-10-01 22:16:09 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (ptr == io->buf + io->len) {
|
|
|
|
/* end of packet, ok */
|
|
|
|
return NULL;
|
2013-10-01 22:16:09 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
ptr++;
|
2015-05-20 17:50:32 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return ptr;
|
2014-01-14 00:29:12 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
uint32_t mg_coap_parse(struct mbuf *io, struct mg_coap_message *cm) {
|
|
|
|
char *ptr;
|
2014-01-14 00:29:12 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
memset(cm, 0, sizeof(*cm));
|
2014-01-14 00:29:12 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if ((ptr = coap_parse_header(io->buf, io, cm)) == NULL) {
|
|
|
|
return cm->flags;
|
|
|
|
}
|
2014-01-14 00:29:12 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if ((ptr = coap_get_token(ptr, io, cm)) == NULL) {
|
|
|
|
return cm->flags;
|
2014-01-14 00:29:12 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if ((ptr = coap_get_options(ptr, io, cm)) == NULL) {
|
|
|
|
return cm->flags;
|
2014-01-14 00:29:12 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* the rest is payload */
|
|
|
|
cm->payload.len = io->len - (ptr - io->buf);
|
|
|
|
if (cm->payload.len != 0) {
|
|
|
|
cm->payload.p = ptr;
|
2015-09-19 19:38:46 +08:00
|
|
|
cm->flags |= MG_COAP_PAYLOAD_FIELD;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2014-01-14 00:29:12 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return cm->flags;
|
2013-10-01 22:16:09 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Calculates extended size of given Opt Number/Length in coap message.
|
|
|
|
*
|
|
|
|
* Helper function.
|
|
|
|
*/
|
|
|
|
static size_t coap_get_ext_opt_size(uint32_t value) {
|
|
|
|
int ret = 0;
|
2015-03-24 16:04:46 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (value >= 13 && value <= 0xFF + 13) {
|
|
|
|
ret = sizeof(uint8_t);
|
|
|
|
} else if (value > 0xFF + 13 && value <= 0xFFFF + 269) {
|
|
|
|
ret = sizeof(uint16_t);
|
2014-10-14 22:25:56 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
return ret;
|
2014-10-14 22:25:56 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Splits given Opt Number/Length into base and ext values.
|
|
|
|
*
|
|
|
|
* Helper function.
|
|
|
|
*/
|
|
|
|
static int coap_split_opt(uint32_t value, uint8_t *base, uint16_t *ext) {
|
|
|
|
int ret = 0;
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (value < 13) {
|
|
|
|
*base = value;
|
|
|
|
} else if (value >= 13 && value <= 0xFF + 13) {
|
|
|
|
*base = 13;
|
|
|
|
*ext = value - 13;
|
|
|
|
ret = sizeof(uint8_t);
|
|
|
|
} else if (value > 0xFF + 13 && value <= 0xFFFF + 269) {
|
|
|
|
*base = 14;
|
|
|
|
*ext = value - 269;
|
|
|
|
ret = sizeof(uint16_t);
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
|
|
|
|
return ret;
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Puts uint16_t (in network order) into given char stream.
|
|
|
|
*
|
|
|
|
* Helper function.
|
|
|
|
*/
|
|
|
|
static char *coap_add_uint16(char *ptr, uint16_t val) {
|
|
|
|
*ptr = val >> 8;
|
|
|
|
ptr++;
|
|
|
|
*ptr = val & 0x00FF;
|
|
|
|
ptr++;
|
|
|
|
return ptr;
|
|
|
|
}
|
2013-10-01 22:16:09 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Puts extended value of Opt Number/Length into given char stream.
|
|
|
|
*
|
|
|
|
* Helper function.
|
|
|
|
*/
|
|
|
|
static char *coap_add_opt_info(char *ptr, uint16_t val, size_t len) {
|
|
|
|
if (len == sizeof(uint8_t)) {
|
|
|
|
*ptr = val;
|
|
|
|
ptr++;
|
|
|
|
} else if (len == sizeof(uint16_t)) {
|
|
|
|
ptr = coap_add_uint16(ptr, val);
|
2013-10-01 22:16:09 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return ptr;
|
|
|
|
}
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Verifies given mg_coap_message and calculates message size for it.
|
|
|
|
*
|
|
|
|
* Helper function.
|
|
|
|
*/
|
|
|
|
static uint32_t coap_calculate_packet_size(struct mg_coap_message *cm,
|
|
|
|
size_t *len) {
|
|
|
|
struct mg_coap_option *opt;
|
|
|
|
uint32_t prev_opt_number;
|
2013-12-09 22:18:49 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
*len = 4; /* header */
|
2015-09-19 19:38:46 +08:00
|
|
|
if (cm->msg_type > MG_COAP_MSG_MAX) {
|
|
|
|
return MG_COAP_ERROR | MG_COAP_MSG_TYPE_FIELD;
|
2014-03-30 22:59:39 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
if (cm->token.len > 8) {
|
2015-09-19 19:38:46 +08:00
|
|
|
return MG_COAP_ERROR | MG_COAP_TOKEN_FIELD;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
if (cm->code_class > 7) {
|
2015-09-19 19:38:46 +08:00
|
|
|
return MG_COAP_ERROR | MG_COAP_CODE_CLASS_FIELD;
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
|
|
|
if (cm->code_detail > 31) {
|
2015-09-19 19:38:46 +08:00
|
|
|
return MG_COAP_ERROR | MG_COAP_CODE_DETAIL_FIELD;
|
2014-03-30 22:59:39 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
*len += cm->token.len;
|
|
|
|
if (cm->payload.len != 0) {
|
|
|
|
*len += cm->payload.len + 1; /* ... + 1; add payload marker */
|
|
|
|
}
|
2015-03-31 17:38:14 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
opt = cm->options;
|
|
|
|
prev_opt_number = 0;
|
|
|
|
while (opt != NULL) {
|
|
|
|
*len += 1; /* basic delta/length */
|
|
|
|
*len += coap_get_ext_opt_size(opt->number);
|
|
|
|
*len += coap_get_ext_opt_size((uint32_t) opt->value.len);
|
2015-03-18 17:37:52 +08:00
|
|
|
/*
|
2015-09-08 19:49:03 +08:00
|
|
|
* Current implementation performs check if
|
|
|
|
* option_number > previous option_number and produces an error
|
|
|
|
* TODO(alashkin): write design doc with limitations
|
|
|
|
* May be resorting is more suitable solution.
|
2015-03-18 17:37:52 +08:00
|
|
|
*/
|
2015-09-08 19:49:03 +08:00
|
|
|
if ((opt->next != NULL && opt->number > opt->next->number) ||
|
|
|
|
opt->value.len > 0xFFFF + 269 ||
|
|
|
|
opt->number - prev_opt_number > 0xFFFF + 269) {
|
2015-09-19 19:38:46 +08:00
|
|
|
return MG_COAP_ERROR | MG_COAP_OPTIOMG_FIELD;
|
2014-03-30 22:59:39 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
*len += opt->value.len;
|
|
|
|
opt = opt->next;
|
2013-10-01 22:16:09 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return 0;
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
uint32_t mg_coap_compose(struct mg_coap_message *cm, struct mbuf *io) {
|
|
|
|
struct mg_coap_option *opt;
|
|
|
|
uint32_t res, prev_opt_number;
|
|
|
|
size_t prev_io_len, packet_size;
|
|
|
|
char *ptr;
|
2014-03-31 05:50:10 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
res = coap_calculate_packet_size(cm, &packet_size);
|
|
|
|
if (res != 0) {
|
|
|
|
return res;
|
|
|
|
}
|
2014-03-31 05:50:10 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* saving previous lenght to handle non-empty mbuf */
|
|
|
|
prev_io_len = io->len;
|
|
|
|
mbuf_append(io, NULL, packet_size);
|
|
|
|
ptr = io->buf + prev_io_len;
|
2014-02-20 00:34:51 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* since cm is verified, it is possible to use bits shift operator
|
|
|
|
* without additional zeroing of unused bits
|
|
|
|
*/
|
2014-03-31 05:50:10 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* ver: 2 bits, msg_type: 2 bits, toklen: 4 bits */
|
|
|
|
*ptr = (1 << 6) | (cm->msg_type << 4) | (cm->token.len);
|
|
|
|
ptr++;
|
2014-02-20 00:34:51 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
/* code class: 3 bits, code detail: 5 bits */
|
|
|
|
*ptr = (cm->code_class << 5) | (cm->code_detail);
|
|
|
|
ptr++;
|
2014-09-11 22:52:42 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
ptr = coap_add_uint16(ptr, cm->msg_id);
|
2014-09-11 22:52:42 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (cm->token.len != 0) {
|
|
|
|
memcpy(ptr, cm->token.p, cm->token.len);
|
|
|
|
ptr += cm->token.len;
|
2014-04-26 01:04:24 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
opt = cm->options;
|
|
|
|
prev_opt_number = 0;
|
|
|
|
while (opt != NULL) {
|
|
|
|
uint8_t delta_base = 0, length_base = 0;
|
|
|
|
uint16_t delta_ext, length_ext;
|
2015-05-16 00:20:54 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
size_t opt_delta_len =
|
|
|
|
coap_split_opt(opt->number - prev_opt_number, &delta_base, &delta_ext);
|
|
|
|
size_t opt_lenght_len =
|
|
|
|
coap_split_opt((uint32_t) opt->value.len, &length_base, &length_ext);
|
2015-05-16 00:20:54 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
*ptr = (delta_base << 4) | length_base;
|
|
|
|
ptr++;
|
2014-02-20 00:34:51 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
ptr = coap_add_opt_info(ptr, delta_ext, opt_delta_len);
|
|
|
|
ptr = coap_add_opt_info(ptr, length_ext, opt_lenght_len);
|
2014-09-11 22:52:42 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (opt->value.len != 0) {
|
|
|
|
memcpy(ptr, opt->value.p, opt->value.len);
|
|
|
|
ptr += opt->value.len;
|
|
|
|
}
|
2014-02-20 00:34:51 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
prev_opt_number = opt->number;
|
|
|
|
opt = opt->next;
|
|
|
|
}
|
2014-02-20 00:34:51 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
if (cm->payload.len != 0) {
|
|
|
|
*ptr = 0xFF;
|
|
|
|
ptr++;
|
|
|
|
memcpy(ptr, cm->payload.p, cm->payload.len);
|
|
|
|
}
|
2014-02-20 00:34:51 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return 0;
|
|
|
|
}
|
2014-02-20 00:34:51 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
uint32_t mg_coap_send_message(struct mg_connection *nc,
|
|
|
|
struct mg_coap_message *cm) {
|
|
|
|
struct mbuf packet_out;
|
|
|
|
uint32_t compose_res;
|
2014-02-20 00:34:51 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
mbuf_init(&packet_out, 0);
|
|
|
|
compose_res = mg_coap_compose(cm, &packet_out);
|
|
|
|
if (compose_res != 0) {
|
|
|
|
return compose_res; /* LCOV_EXCL_LINE */
|
|
|
|
}
|
2014-03-05 04:22:46 +08:00
|
|
|
|
2015-10-22 20:04:35 +08:00
|
|
|
mg_send(nc, packet_out.buf, (int) packet_out.len);
|
2015-09-08 19:49:03 +08:00
|
|
|
mbuf_free(&packet_out);
|
2014-02-20 00:34:51 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return 0;
|
|
|
|
}
|
2014-02-20 00:34:51 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
uint32_t mg_coap_send_ack(struct mg_connection *nc, uint16_t msg_id) {
|
|
|
|
struct mg_coap_message cm;
|
|
|
|
memset(&cm, 0, sizeof(cm));
|
2015-09-19 19:38:46 +08:00
|
|
|
cm.msg_type = MG_COAP_MSG_ACK;
|
2015-09-08 19:49:03 +08:00
|
|
|
cm.msg_id = msg_id;
|
2014-02-20 00:34:51 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return mg_coap_send_message(nc, &cm);
|
2014-02-20 00:34:51 +08:00
|
|
|
}
|
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
static void coap_handler(struct mg_connection *nc, int ev, void *ev_data) {
|
|
|
|
struct mbuf *io = &nc->recv_mbuf;
|
|
|
|
struct mg_coap_message cm;
|
|
|
|
uint32_t parse_res;
|
2014-05-13 06:22:57 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
memset(&cm, 0, sizeof(cm));
|
2014-05-13 06:22:57 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
nc->handler(nc, ev, ev_data);
|
2014-05-13 06:22:57 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
switch (ev) {
|
2015-09-19 19:38:46 +08:00
|
|
|
case MG_EV_RECV:
|
2015-09-08 19:49:03 +08:00
|
|
|
parse_res = mg_coap_parse(io, &cm);
|
2015-09-19 19:38:46 +08:00
|
|
|
if ((parse_res & MG_COAP_IGNORE) == 0) {
|
|
|
|
if ((cm.flags & MG_COAP_NOT_ENOUGH_DATA) != 0) {
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Since we support UDP only
|
2015-09-19 19:38:46 +08:00
|
|
|
* MG_COAP_NOT_ENOUGH_DATA == MG_COAP_FORMAT_ERROR
|
2015-09-08 19:49:03 +08:00
|
|
|
*/
|
2015-09-19 19:38:46 +08:00
|
|
|
cm.flags |= MG_COAP_FORMAT_ERROR; /* LCOV_EXCL_LINE */
|
2015-09-08 19:49:03 +08:00
|
|
|
} /* LCOV_EXCL_LINE */
|
2015-09-19 19:38:46 +08:00
|
|
|
nc->handler(nc, MG_COAP_EVENT_BASE + cm.msg_type, &cm);
|
2015-09-08 19:49:03 +08:00
|
|
|
}
|
2014-05-13 06:22:57 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
mg_coap_free_options(&cm);
|
|
|
|
mbuf_remove(io, io->len);
|
|
|
|
break;
|
|
|
|
}
|
2014-05-13 06:22:57 +08:00
|
|
|
}
|
2015-09-08 19:49:03 +08:00
|
|
|
/*
|
|
|
|
* Attach built-in CoAP event handler to the given connection.
|
|
|
|
*
|
|
|
|
* The user-defined event handler will receive following extra events:
|
|
|
|
*
|
2015-09-19 19:38:46 +08:00
|
|
|
* - MG_EV_COAP_CON
|
|
|
|
* - MG_EV_COAP_NOC
|
|
|
|
* - MG_EV_COAP_ACK
|
|
|
|
* - MG_EV_COAP_RST
|
2015-09-08 19:49:03 +08:00
|
|
|
*/
|
|
|
|
int mg_set_protocol_coap(struct mg_connection *nc) {
|
|
|
|
/* supports UDP only */
|
2015-09-19 19:38:46 +08:00
|
|
|
if ((nc->flags & MG_F_UDP) == 0) {
|
2015-09-08 19:49:03 +08:00
|
|
|
return -1;
|
|
|
|
}
|
2014-05-13 06:22:57 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
nc->proto_handler = coap_handler;
|
2014-03-01 09:47:01 +08:00
|
|
|
|
2015-09-08 19:49:03 +08:00
|
|
|
return 0;
|
2013-12-09 22:18:49 +08:00
|
|
|
}
|
|
|
|
|
2015-09-19 19:38:46 +08:00
|
|
|
#endif /* MG_DISABLE_COAP */
|