Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Contribute to GitLab
Sign in
Toggle navigation
M
mongoose
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
esp
mongoose
Commits
8f7703c2
Commit
8f7703c2
authored
Feb 19, 2014
by
Sergey Lyubka
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
moved to net_skeleton
parent
cead9a04
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
330 additions
and
973 deletions
+330
-973
mongoose.c
mongoose.c
+321
-913
mongoose.h
mongoose.h
+1
-1
unit_test.c
unit_test.c
+8
-59
No files found.
mongoose.c
View file @
8f7703c2
...
@@ -15,68 +15,13 @@
...
@@ -15,68 +15,13 @@
// Alternatively, you can license this library under a commercial
// Alternatively, you can license this library under a commercial
// license, as set out in <http://cesanta.com/>.
// license, as set out in <http://cesanta.com/>.
#undef UNICODE // Use ANSI WinAPI functions
// net_skeleton start
#undef _UNICODE // Use multibyte encoding on Windows
#include "net_skeleton.h"
#define _MBCS // Use multibyte encoding on Windows
// net_skeleton end
#define _INTEGRAL_MAX_BITS 64 // Enable _stati64() on Windows
#define _CRT_SECURE_NO_WARNINGS // Disable deprecation warning in VS2005+
#undef WIN32_LEAN_AND_MEAN // Let windows.h always include winsock2.h
#define _XOPEN_SOURCE 600 // For flockfile() on Linux
#define __STDC_FORMAT_MACROS // <inttypes.h> wants this for C++
#define __STDC_LIMIT_MACROS // C++ wants that for INT64_MAX
#define _LARGEFILE_SOURCE // Enable fseeko() and ftello() functions
#define _FILE_OFFSET_BITS 64 // Enable 64-bit file offsets
#ifdef _MSC_VER
#pragma warning (disable : 4127) // FD_SET() emits warning, disable it
#pragma warning (disable : 4204) // missing c99 support
#endif
#include <sys/types.h>
#include <sys/stat.h>
#include <stddef.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <fcntl.h>
#include <assert.h>
#include <errno.h>
#include <time.h>
#include <ctype.h>
#include <ctype.h>
#include <stdarg.h>
#ifdef _WIN32
#ifdef _WIN32
#include <windows.h>
#include <process.h> // For _beginthread
#include <io.h> // For _lseeki64
#include <direct.h> // For _mkdir
typedef
int
socklen_t
;
#if !defined(__MINGW32__) || !defined(_PID_T_) || defined(_NO_OLDNAMES)
typedef
HANDLE
pid_t
;
#endif
typedef
SOCKET
sock_t
;
typedef
unsigned
char
uint8_t
;
typedef
unsigned
int
uint32_t
;
typedef
unsigned
short
uint16_t
;
typedef
unsigned
__int64
uint64_t
;
typedef
__int64
int64_t
;
typedef
CRITICAL_SECTION
mutex_t
;
typedef
struct
_stati64
file_stat_t
;
#pragma comment(lib, "ws2_32.lib")
#define snprintf _snprintf
#define vsnprintf _vsnprintf
#define INT64_FMT "I64d"
#ifndef EINPROGRESS
#define EINPROGRESS WSAEINPROGRESS
#endif
#ifndef EWOULDBLOCK
#define EWOULDBLOCK WSAEWOULDBLOCK
#endif
#define mutex_init(x) InitializeCriticalSection(x)
#define mutex_destroy(x) DeleteCriticalSection(x)
#define mutex_lock(x) EnterCriticalSection(x)
#define mutex_unlock(x) LeaveCriticalSection(x)
#define get_thread_id() ((unsigned long) GetCurrentThreadId())
#ifndef S_ISDIR
#ifndef S_ISDIR
#define S_ISDIR(x) ((x) & _S_IFDIR)
#define S_ISDIR(x) ((x) & _S_IFDIR)
#endif
#endif
...
@@ -87,68 +32,23 @@ typedef struct _stati64 file_stat_t;
...
@@ -87,68 +32,23 @@ typedef struct _stati64 file_stat_t;
#define lseek(x, y, z) _lseeki64((x), (y), (z))
#define lseek(x, y, z) _lseeki64((x), (y), (z))
#define mkdir(x, y) _mkdir(x)
#define mkdir(x, y) _mkdir(x)
#define to64(x) _atoi64(x)
#define to64(x) _atoi64(x)
#define flockfile(x)
#define funlockfile(x)
#ifndef va_copy
#define va_copy(x,y) x = y
#endif // MINGW #defines va_copy
#ifndef __func__
#ifndef __func__
#define STRX(x) #x
#define STRX(x) #x
#define STR(x) STRX(x)
#define STR(x) STRX(x)
#define __func__ __FILE__ ":" STR(__LINE__)
#define __func__ __FILE__ ":" STR(__LINE__)
#endif
#endif
typedef
struct
_stati64
file_stat_t
;
#else
#else
#include <dirent.h>
#include <dirent.h>
#include <inttypes.h>
#include <inttypes.h>
#include <pthread.h>
#include <pwd.h>
#include <pwd.h>
#include <signal.h>
#include <unistd.h>
#include <netdb.h>
#include <arpa/inet.h> // For inet_pton() when MONGOOSE_USE_IPV6 is defined
#include <netinet/in.h>
#include <sys/socket.h>
#include <sys/select.h>
#define closesocket(x) close(x)
typedef
int
sock_t
;
typedef
pthread_mutex_t
mutex_t
;
typedef
struct
stat
file_stat_t
;
#define mutex_init(x) pthread_mutex_init(x, NULL)
#define mutex_destroy(x) pthread_mutex_destroy(x)
#define mutex_lock(x) pthread_mutex_lock(x)
#define mutex_unlock(x) pthread_mutex_unlock(x)
#define get_thread_id() ((unsigned long) pthread_self())
#define INVALID_SOCKET ((sock_t) -1)
#define INT64_FMT PRId64
#define to64(x) strtoll(x, NULL, 10)
#define __cdecl
#define O_BINARY 0
#define O_BINARY 0
#endif
#define INT64_FMT PRId64
typedef
struct
stat
file_stat_t
;
#ifdef MONGOOSE_USE_SSL
#ifdef __APPLE__
#pragma GCC diagnostic ignored "-Wdeprecated-declarations"
#endif
#include <openssl/ssl.h>
#endif
#endif
#include "mongoose.h"
#include "mongoose.h"
struct
ll
{
struct
ll
*
prev
,
*
next
;
};
#define LINKED_LIST_INIT(N) ((N)->next = (N)->prev = (N))
#define LINKED_LIST_DECLARE_AND_INIT(H) struct ll H = { &H, &H }
#define LINKED_LIST_ENTRY(P,T,N) ((T *)((char *)(P) - offsetof(T, N)))
#define LINKED_LIST_IS_EMPTY(N) ((N)->next == (N))
#define LINKED_LIST_FOREACH(H,N,T) \
for (N = (H)->next, T = (N)->next; N != (H); N = (T), T = (N)->next)
#define LINKED_LIST_ADD_TO_FRONT(H,N) do { ((H)->next)->prev = (N); \
(N)->next = ((H)->next); (N)->prev = (H); (H)->next = (N); } while (0)
#define LINKED_LIST_ADD_TO_TAIL(H,N) do { ((H)->prev)->next = (N); \
(N)->prev = ((H)->prev); (N)->next = (H); (H)->prev = (N); } while (0)
#define LINKED_LIST_REMOVE(N) do { ((N)->next)->prev = ((N)->prev); \
((N)->prev)->next = ((N)->next); LINKED_LIST_INIT(N); } while (0)
#define ARRAY_SIZE(array) (sizeof(array) / sizeof(array[0]))
#define MAX_REQUEST_SIZE 16384
#define MAX_REQUEST_SIZE 16384
#define IOBUF_SIZE 8192
#define IOBUF_SIZE 8192
#define MAX_PATH_SIZE 8192
#define MAX_PATH_SIZE 8192
...
@@ -168,25 +68,18 @@ struct ll { struct ll *prev, *next; };
...
@@ -168,25 +68,18 @@ struct ll { struct ll *prev, *next; };
#define MONGOOSE_USE_EXTRA_HTTP_HEADERS ""
#define MONGOOSE_USE_EXTRA_HTTP_HEADERS ""
#endif
#endif
#ifndef MONGOOSE_
USE_
POST_SIZE_LIMIT
#ifndef MONGOOSE_POST_SIZE_LIMIT
#define MONGOOSE_
USE_
POST_SIZE_LIMIT 0
#define MONGOOSE_POST_SIZE_LIMIT 0
#endif
#endif
#ifndef MONGOOSE_
USE_
IDLE_TIMEOUT_SECONDS
#ifndef MONGOOSE_IDLE_TIMEOUT_SECONDS
#define MONGOOSE_
USE_
IDLE_TIMEOUT_SECONDS 30
#define MONGOOSE_IDLE_TIMEOUT_SECONDS 30
#endif
#endif
#ifdef MONGOOSE_NO_SOCKETPAIR
#ifdef MONGOOSE_NO_SOCKETPAIR
#define MONGOOSE_NO_CGI
#define MONGOOSE_NO_CGI
#endif
#endif
#ifdef MONGOOSE_ENABLE_DEBUG
#define DBG(x) do { printf("%-20s ", __func__); printf x; putchar('\n'); \
fflush(stdout); } while(0)
#else
#define DBG(x)
#endif
#ifdef MONGOOSE_NO_FILESYSTEM
#ifdef MONGOOSE_NO_FILESYSTEM
#define MONGOOSE_NO_AUTH
#define MONGOOSE_NO_AUTH
#define MONGOOSE_NO_CGI
#define MONGOOSE_NO_CGI
...
@@ -195,14 +88,6 @@ struct ll { struct ll *prev, *next; };
...
@@ -195,14 +88,6 @@ struct ll { struct ll *prev, *next; };
#define MONGOOSE_NO_LOGGING
#define MONGOOSE_NO_LOGGING
#endif
#endif
union
socket_address
{
struct
sockaddr
sa
;
struct
sockaddr_in
sin
;
#ifdef MONGOOSE_USE_IPV6
struct
sockaddr_in6
sin6
;
#endif
};
struct
vec
{
struct
vec
{
const
char
*
ptr
;
const
char
*
ptr
;
int
len
;
int
len
;
...
@@ -247,7 +132,7 @@ enum {
...
@@ -247,7 +132,7 @@ enum {
#ifndef _WIN32
#ifndef _WIN32
RUN_AS_USER
,
RUN_AS_USER
,
#endif
#endif
#ifdef
MONGOOSE_US
E_SSL
#ifdef
NS_ENABL
E_SSL
SSL_CERTIFICATE
,
SSL_CERTIFICATE
,
#endif
#endif
URL_REWRITES
,
URL_REWRITES
,
...
@@ -285,7 +170,7 @@ static const char *static_config_options[] = {
...
@@ -285,7 +170,7 @@ static const char *static_config_options[] = {
#ifndef _WIN32
#ifndef _WIN32
"run_as_user"
,
NULL
,
"run_as_user"
,
NULL
,
#endif
#endif
#ifdef
MONGOOSE_US
E_SSL
#ifdef
NS_ENABL
E_SSL
"ssl_certificate"
,
NULL
,
"ssl_certificate"
,
NULL
,
#endif
#endif
"url_rewrites"
,
NULL
,
"url_rewrites"
,
NULL
,
...
@@ -293,74 +178,47 @@ static const char *static_config_options[] = {
...
@@ -293,74 +178,47 @@ static const char *static_config_options[] = {
};
};
struct
mg_server
{
struct
mg_server
{
s
ock_t
listening_sock
;
s
truct
ns_server
ns_server
;
union
socket_address
lsa
;
// Listening socket address
union
socket_address
lsa
;
// Listening socket address
struct
ll
active_connections
;
mg_handler_t
request_handler
;
mg_handler_t
request_handler
;
mg_handler_t
http_close_handler
;
mg_handler_t
http_close_handler
;
mg_handler_t
error_handler
;
mg_handler_t
error_handler
;
mg_handler_t
auth_handler
;
mg_handler_t
auth_handler
;
char
*
config_options
[
NUM_OPTIONS
];
char
*
config_options
[
NUM_OPTIONS
];
char
local_ip
[
48
];
char
local_ip
[
48
];
void
*
server_data
;
#ifdef MONGOOSE_USE_SSL
SSL_CTX
*
ssl_ctx
;
// Server SSL context
SSL_CTX
*
client_ssl_ctx
;
// Client SSL context
#endif
#ifndef MONGOOSE_NO_SOCKETPAIR
sock_t
ctl
[
2
];
// Control socketpair. Used to wake up from select() call
#endif
};
// Expandable IO buffer
struct
iobuf
{
char
*
buf
;
// Buffer that holds the data
int
size
;
// Buffer size
int
len
;
// Number of bytes currently in a buffer
};
};
// Local endpoint representation
// Local endpoint representation
union
endpoint
{
union
endpoint
{
int
fd
;
// Opened regular local file
int
fd
;
// Opened regular local file
sock_t
cgi_sock
;
// CGI socket
struct
ns_connection
*
cgi_conn
;
// CGI socket
void
*
ssl
;
// SSL descriptor
};
};
enum
endpoint_type
{
EP_NONE
,
EP_FILE
,
EP_CGI
,
EP_USER
,
EP_PUT
,
EP_CLIENT
};
enum
endpoint_type
{
EP_NONE
,
EP_FILE
,
EP_CGI
,
EP_USER
,
EP_PUT
,
EP_CLIENT
};
enum
connection_flags
{
CONN_CLOSE
=
1
,
// Connection must be closed at the end of the poll
#define MG_HEADERS_SENT NSF_USER_1
CONN_SPOOL_DONE
=
2
,
// All data has been buffered for sending
#define MG_LONG_RUNNING NSF_USER_2
CONN_SSL_HANDS_SHAKEN
=
4
,
// SSL handshake has completed. Only for SSL
#define MG_CGI_CONN NSF_USER_3
CONN_HEADERS_SENT
=
8
,
// User callback has sent HTTP headers
CONN_BUFFER
=
16
,
// CGI only. Holds data send until CGI prints
// all HTTP headers
CONN_CONNECTING
=
32
,
// HTTP client is doing non-blocking connect()
CONN_LONG_RUNNING
=
64
// Long-running URI handlers
};
struct
connection
{
struct
connection
{
struct
mg_connection
mg_conn
;
// XXX: Must be first
struct
ns_connection
*
ns_conn
;
struct
ll
link
;
// Linkage to server->active_connections
struct
mg_connection
mg_conn
;
struct
mg_server
*
server
;
struct
mg_server
*
server
;
sock_t
client_sock
;
// Connected client
struct
iobuf
local_iobuf
;
struct
iobuf
remote_iobuf
;
union
endpoint
endpoint
;
union
endpoint
endpoint
;
enum
endpoint_type
endpoint_type
;
enum
endpoint_type
endpoint_type
;
time_t
birth_time
;
time_t
birth_time
;
time_t
last_activity_time
;
char
*
path_info
;
char
*
path_info
;
char
*
request
;
char
*
request
;
int64_t
num_bytes_sent
;
// Total number of bytes sent
int64_t
num_bytes_sent
;
// Total number of bytes sent
int64_t
cl
;
// Reply content length, for Range support
int64_t
cl
;
// Reply content length, for Range support
int
request_len
;
// Request length, including last \r\n after last header
int
request_len
;
// Request length, including last \r\n after last header
int
flags
;
// CONN_* flags: CONN_CLOSE, CONN_SPOOL_DONE, etc
//
int flags; // CONN_* flags: CONN_CLOSE, CONN_SPOOL_DONE, etc
mg_handler_t
handler
;
// Callback for HTTP client
mg_handler_t
handler
;
// Callback for HTTP client
#ifdef MONGOOSE_USE_SSL
SSL
*
ssl
;
// SSL descriptor
#endif
};
};
#define MG_CONN_2_CONN(c) ((struct connection *) ((char *) (c) - \
offsetof(struct connection, mg_conn)))
static
void
open_local_endpoint
(
struct
connection
*
conn
,
int
skip_user
);
static
void
open_local_endpoint
(
struct
connection
*
conn
,
int
skip_user
);
static
void
close_local_endpoint
(
struct
connection
*
conn
);
static
void
close_local_endpoint
(
struct
connection
*
conn
);
...
@@ -490,24 +348,6 @@ static int mg_open(const char *path, int flag) {
...
@@ -490,24 +348,6 @@ static int mg_open(const char *path, int flag) {
#endif
#endif
#endif // MONGOOSE_NO_FILESYSTEM
#endif // MONGOOSE_NO_FILESYSTEM
static
void
set_close_on_exec
(
int
fd
)
{
#ifdef _WIN32
(
void
)
SetHandleInformation
((
HANDLE
)
fd
,
HANDLE_FLAG_INHERIT
,
0
);
#else
fcntl
(
fd
,
F_SETFD
,
FD_CLOEXEC
);
#endif
}
static
void
set_non_blocking_mode
(
sock_t
sock
)
{
#ifdef _WIN32
unsigned
long
on
=
1
;
ioctlsocket
(
sock
,
FIONBIO
,
&
on
);
#else
int
flags
=
fcntl
(
sock
,
F_GETFL
,
0
);
fcntl
(
sock
,
F_SETFL
,
flags
|
O_NONBLOCK
);
#endif
}
// A helper function for traversing a comma separated list of values.
// A helper function for traversing a comma separated list of values.
// It returns a list pointer shifted to the next value, or NULL if the end
// It returns a list pointer shifted to the next value, or NULL if the end
// of the list found.
// of the list found.
...
@@ -547,32 +387,6 @@ static const char *next_option(const char *list, struct vec *val,
...
@@ -547,32 +387,6 @@ static const char *next_option(const char *list, struct vec *val,
return
list
;
return
list
;
}
}
static
int
spool
(
struct
iobuf
*
io
,
const
void
*
buf
,
int
len
)
{
static
const
double
mult
=
1
.
2
;
char
*
p
=
NULL
;
int
new_len
=
0
;
assert
(
io
->
len
>=
0
);
assert
(
io
->
len
<=
io
->
size
);
//DBG(("1. %d %d %d", len, io->len, io->size));
if
(
len
<=
0
)
{
}
else
if
((
new_len
=
io
->
len
+
len
)
<
io
->
size
)
{
memcpy
(
io
->
buf
+
io
->
len
,
buf
,
len
);
io
->
len
=
new_len
;
}
else
if
((
p
=
(
char
*
)
realloc
(
io
->
buf
,
(
int
)
(
new_len
*
mult
)))
!=
NULL
)
{
io
->
buf
=
p
;
memcpy
(
io
->
buf
+
io
->
len
,
buf
,
len
);
io
->
len
=
new_len
;
io
->
size
=
(
int
)
(
new_len
*
mult
);
}
else
{
len
=
0
;
}
//DBG(("%d %d %d", len, io->len, io->size));
return
len
;
}
// Like snprintf(), but never returns negative value, or a value
// Like snprintf(), but never returns negative value, or a value
// that is larger than a supplied buffer.
// that is larger than a supplied buffer.
static
int
mg_vsnprintf
(
char
*
buf
,
size_t
buflen
,
const
char
*
fmt
,
va_list
ap
)
{
static
int
mg_vsnprintf
(
char
*
buf
,
size_t
buflen
,
const
char
*
fmt
,
va_list
ap
)
{
...
@@ -722,134 +536,29 @@ static void send_http_error(struct connection *conn, int code,
...
@@ -722,134 +536,29 @@ static void send_http_error(struct connection *conn, int code,
"HTTP/1.1 %d %s
\r\n
Content-Length: %d
\r\n
"
"HTTP/1.1 %d %s
\r\n
Content-Length: %d
\r\n
"
"Content-Type: text/plain
\r\n\r\n
"
,
"Content-Type: text/plain
\r\n\r\n
"
,
code
,
message
,
body_len
);
code
,
message
,
body_len
);
spool
(
&
conn
->
remote_iobuf
,
headers
,
headers_len
);
ns_send
(
conn
->
ns_conn
,
headers
,
headers_len
);
spool
(
&
conn
->
remote_iobuf
,
body
,
body_len
);
ns_send
(
conn
->
ns_conn
,
body
,
body_len
);
close_local_endpoint
(
conn
);
// This will write to the log file
close_local_endpoint
(
conn
);
// This will write to the log file
}
}
// Print message to buffer. If buffer is large enough to hold the message,
// return buffer. If buffer is to small, allocate large enough buffer on heap,
// and return allocated buffer.
static
int
alloc_vprintf
(
char
**
buf
,
size_t
size
,
const
char
*
fmt
,
va_list
ap
)
{
va_list
ap_copy
;
int
len
;
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
;
while
(
len
<
0
)
{
if
(
*
buf
)
free
(
*
buf
);
size
*=
2
;
if
((
*
buf
=
(
char
*
)
malloc
(
size
))
==
NULL
)
break
;
va_copy
(
ap_copy
,
ap
);
len
=
vsnprintf
(
*
buf
,
size
,
fmt
,
ap_copy
);
va_end
(
ap_copy
);
}
}
else
if
(
len
>
(
int
)
size
)
{
// Standard-compliant code path. Allocate a buffer that is large enough.
if
((
*
buf
=
(
char
*
)
malloc
(
len
+
1
))
==
NULL
)
{
len
=
-
1
;
}
else
{
va_copy
(
ap_copy
,
ap
);
len
=
vsnprintf
(
*
buf
,
len
+
1
,
fmt
,
ap_copy
);
va_end
(
ap_copy
);
}
}
return
len
;
}
static
void
write_chunk
(
struct
connection
*
conn
,
const
char
*
buf
,
int
len
)
{
static
void
write_chunk
(
struct
connection
*
conn
,
const
char
*
buf
,
int
len
)
{
char
chunk_size
[
50
];
char
chunk_size
[
50
];
int
n
=
mg_snprintf
(
chunk_size
,
sizeof
(
chunk_size
),
"%X
\r\n
"
,
len
);
int
n
=
mg_snprintf
(
chunk_size
,
sizeof
(
chunk_size
),
"%X
\r\n
"
,
len
);
spool
(
&
conn
->
remote_iobuf
,
chunk_size
,
n
);
ns_send
(
conn
->
ns_conn
,
chunk_size
,
n
);
spool
(
&
conn
->
remote_iobuf
,
buf
,
len
);
ns_send
(
conn
->
ns_conn
,
buf
,
len
);
spool
(
&
conn
->
remote_iobuf
,
"
\r\n
"
,
2
);
ns_send
(
conn
->
ns_conn
,
"
\r\n
"
,
2
);
}
int
mg_vprintf
(
struct
mg_connection
*
conn
,
const
char
*
fmt
,
va_list
ap
,
int
chunked
)
{
char
mem
[
IOBUF_SIZE
],
*
buf
=
mem
;
int
len
;
if
((
len
=
alloc_vprintf
(
&
buf
,
sizeof
(
mem
),
fmt
,
ap
))
>
0
)
{
if
(
chunked
)
{
write_chunk
((
struct
connection
*
)
conn
,
buf
,
len
);
}
else
{
len
=
mg_write
(
conn
,
buf
,
(
size_t
)
len
);
}
}
if
(
buf
!=
mem
&&
buf
!=
NULL
)
{
free
(
buf
);
}
return
len
;
}
}
int
mg_printf
(
struct
mg_connection
*
conn
,
const
char
*
fmt
,
...)
{
int
mg_printf
(
struct
mg_connection
*
conn
,
const
char
*
fmt
,
...)
{
struct
connection
*
c
=
MG_CONN_2_CONN
(
conn
);
int
len
;
int
len
;
va_list
ap
;
va_list
ap
;
va_start
(
ap
,
fmt
);
va_start
(
ap
,
fmt
);
len
=
mg_vprintf
(
conn
,
fmt
,
ap
,
0
);
len
=
ns_vprintf
(
c
->
ns_conn
,
fmt
,
ap
);
va_end
(
ap
);
va_end
(
ap
);
return
len
;
}
#ifndef MONGOOSE_NO_SOCKETPAIR
static
int
mg_socketpair
(
sock_t
sp
[
2
])
{
struct
sockaddr_in
sa
;
sock_t
sock
,
ret
=
-
1
;
socklen_t
len
=
sizeof
(
sa
);
sp
[
0
]
=
sp
[
1
]
=
INVALID_SOCKET
;
(
void
)
memset
(
&
sa
,
0
,
sizeof
(
sa
));
sa
.
sin_family
=
AF_INET
;
sa
.
sin_port
=
htons
(
0
);
sa
.
sin_addr
.
s_addr
=
htonl
(
0x7f000001
);
if
((
sock
=
socket
(
AF_INET
,
SOCK_STREAM
,
0
))
!=
INVALID_SOCKET
&&
!
bind
(
sock
,
(
struct
sockaddr
*
)
&
sa
,
len
)
&&
!
listen
(
sock
,
1
)
&&
!
getsockname
(
sock
,
(
struct
sockaddr
*
)
&
sa
,
&
len
)
&&
(
sp
[
0
]
=
socket
(
AF_INET
,
SOCK_STREAM
,
6
))
!=
-
1
&&
!
connect
(
sp
[
0
],
(
struct
sockaddr
*
)
&
sa
,
len
)
&&
(
sp
[
1
]
=
accept
(
sock
,(
struct
sockaddr
*
)
&
sa
,
&
len
))
!=
INVALID_SOCKET
)
{
set_close_on_exec
(
sp
[
0
]);
set_close_on_exec
(
sp
[
1
]);
ret
=
0
;
}
else
{
if
(
sp
[
0
]
!=
INVALID_SOCKET
)
closesocket
(
sp
[
0
]);
if
(
sp
[
1
]
!=
INVALID_SOCKET
)
closesocket
(
sp
[
1
]);
sp
[
0
]
=
sp
[
1
]
=
INVALID_SOCKET
;
}
closesocket
(
sock
);
return
ret
;
}
#endif
static
int
is_error
(
int
n
)
{
return
n
==
0
||
(
n
<
0
&&
errno
!=
EINTR
&&
errno
!=
EINPROGRESS
&&
errno
!=
EAGAIN
&&
errno
!=
EWOULDBLOCK
#ifdef _WIN32
&&
WSAGetLastError
()
!=
WSAEINTR
&&
WSAGetLastError
()
!=
WSAEWOULDBLOCK
#endif
);
}
static
void
discard_leading_iobuf_bytes
(
struct
iobuf
*
io
,
int
n
)
{
return
len
;
if
(
n
>=
0
&&
n
<=
io
->
len
)
{
memmove
(
io
->
buf
,
io
->
buf
+
n
,
io
->
len
-
n
);
io
->
len
-=
n
;
}
}
}
#ifndef MONGOOSE_NO_CGI
#ifndef MONGOOSE_NO_CGI
...
@@ -1125,11 +834,7 @@ static void prepare_cgi_environment(struct connection *conn,
...
@@ -1125,11 +834,7 @@ static void prepare_cgi_environment(struct connection *conn,
addenv
(
blk
,
"SCRIPT_FILENAME=%s"
,
prog
);
addenv
(
blk
,
"SCRIPT_FILENAME=%s"
,
prog
);
addenv
(
blk
,
"PATH_TRANSLATED=%s"
,
prog
);
addenv
(
blk
,
"PATH_TRANSLATED=%s"
,
prog
);
#ifdef MONGOOSE_USE_SSL
addenv
(
blk
,
"HTTPS=%s"
,
conn
->
ns_conn
->
ssl
!=
NULL
?
"on"
:
"off"
);
addenv
(
blk
,
"HTTPS=%s"
,
conn
->
ssl
!=
NULL
?
"on"
:
"off"
);
#else
addenv
(
blk
,
"HTTPS=%s"
,
"off"
);
#endif
if
((
s
=
mg_get_header
(
ri
,
"Content-Type"
))
!=
NULL
)
if
((
s
=
mg_get_header
(
ri
,
"Content-Type"
))
!=
NULL
)
addenv
(
blk
,
"CONTENT_TYPE=%s"
,
s
);
addenv
(
blk
,
"CONTENT_TYPE=%s"
,
s
);
...
@@ -1196,20 +901,22 @@ static void open_cgi_endpoint(struct connection *conn, const char *prog) {
...
@@ -1196,20 +901,22 @@ static void open_cgi_endpoint(struct connection *conn, const char *prog) {
mg_snprintf
(
dir
,
sizeof
(
dir
),
"%.*s"
,
(
int
)
(
p
-
prog
),
prog
);
mg_snprintf
(
dir
,
sizeof
(
dir
),
"%.*s"
,
(
int
)
(
p
-
prog
),
prog
);
}
}
// Try to create socketpair in a loop until success.
mg
_socketpair()
// Try to create socketpair in a loop until success.
ns
_socketpair()
// can be interrupted by a signal and fail.
// can be interrupted by a signal and fail.
// TODO(lsm): use sigaction to restart interrupted syscall
// TODO(lsm): use sigaction to restart interrupted syscall
do
{
do
{
mg
_socketpair
(
fds
);
ns
_socketpair
(
fds
);
}
while
(
fds
[
0
]
==
INVALID_SOCKET
);
}
while
(
fds
[
0
]
==
INVALID_SOCKET
);
if
(
start_process
(
conn
->
server
->
config_options
[
CGI_INTERPRETER
],
if
(
start_process
(
conn
->
server
->
config_options
[
CGI_INTERPRETER
],
prog
,
blk
.
buf
,
blk
.
vars
,
dir
,
fds
[
1
])
>
0
)
{
prog
,
blk
.
buf
,
blk
.
vars
,
dir
,
fds
[
1
])
>
0
)
{
conn
->
endpoint_type
=
EP_CGI
;
conn
->
endpoint_type
=
EP_CGI
;
conn
->
endpoint
.
cgi_sock
=
fds
[
0
];
conn
->
endpoint
.
cgi_conn
=
ns_add_sock
(
&
conn
->
server
->
ns_server
,
spool
(
&
conn
->
remote_iobuf
,
cgi_status
,
sizeof
(
cgi_status
)
-
1
);
fds
[
0
],
conn
);
conn
->
endpoint
.
cgi_conn
->
flags
|=
MG_CGI_CONN
;
ns_send
(
conn
->
ns_conn
,
cgi_status
,
sizeof
(
cgi_status
)
-
1
);
conn
->
mg_conn
.
status_code
=
200
;
conn
->
mg_conn
.
status_code
=
200
;
conn
->
flags
|=
CONN_BUFFER
;
conn
->
ns_conn
->
flags
|=
NSF_BUFFER_BUT_DONT_SEND
;
}
else
{
}
else
{
closesocket
(
fds
[
0
]);
closesocket
(
fds
[
0
]);
send_http_error
(
conn
,
500
,
"start_process(%s) failed"
,
prog
);
send_http_error
(
conn
,
500
,
"start_process(%s) failed"
,
prog
);
...
@@ -1220,67 +927,54 @@ static void open_cgi_endpoint(struct connection *conn, const char *prog) {
...
@@ -1220,67 +927,54 @@ static void open_cgi_endpoint(struct connection *conn, const char *prog) {
#endif
#endif
}
}
static
void
read_from_cgi
(
struct
connection
*
conn
)
{
static
void
on_cgi_data
(
struct
ns_connection
*
nc
)
{
struct
iobuf
*
io
=
&
conn
->
remote_iobuf
;
struct
connection
*
conn
=
(
struct
connection
*
)
nc
->
connection_data
;
char
buf
[
IOBUF_SIZE
],
buf2
[
sizeof
(
buf
)],
*
s
=
buf2
;
const
char
*
status
=
"500"
;
const
char
*
status
=
"500"
;
struct
mg_connection
c
;
struct
mg_connection
c
;
int
len
,
s_len
=
sizeof
(
cgi_status
)
-
1
,
n
=
recv
(
conn
->
endpoint
.
cgi_sock
,
buf
,
sizeof
(
buf
),
0
);
DBG
((
"%p %d"
,
conn
,
n
));
if
(
!
conn
)
return
;
if
(
is_error
(
n
))
{
close_local_endpoint
(
conn
);
// Copy CGI data from CGI socket to the client send buffer
}
else
if
(
n
>
0
)
{
ns_send
(
conn
->
ns_conn
,
nc
->
recv_iobuf
.
buf
,
nc
->
recv_iobuf
.
len
);
spool
(
&
conn
->
remote_iobuf
,
buf
,
n
);
iobuf_remove
(
&
nc
->
recv_iobuf
,
nc
->
recv_iobuf
.
len
);
if
(
conn
->
flags
&
CONN_BUFFER
)
{
len
=
get_request_len
(
io
->
buf
+
s_len
,
io
->
len
-
s_len
);
// If reply has not been parsed yet, parse it
if
(
len
==
0
)
return
;
if
(
conn
->
ns_conn
->
flags
&
NSF_BUFFER_BUT_DONT_SEND
)
{
if
(
len
>
0
)
{
struct
iobuf
*
io
=
&
conn
->
ns_conn
->
send_iobuf
;
memset
(
&
c
,
0
,
sizeof
(
c
));
int
s_len
=
sizeof
(
cgi_status
)
-
1
;
memcpy
(
buf2
,
io
->
buf
+
s_len
,
len
);
int
len
=
get_request_len
(
io
->
buf
+
s_len
,
io
->
len
-
s_len
);
buf2
[
len
-
1
]
=
'\0'
;
char
buf
[
MAX_REQUEST_SIZE
],
*
s
=
buf
;
parse_http_headers
(
&
s
,
&
c
);
if
(
mg_get_header
(
&
c
,
"Location"
)
!=
NULL
)
{
if
(
len
==
0
)
return
;
status
=
"302"
;
}
else
if
((
status
=
(
char
*
)
mg_get_header
(
&
c
,
"Status"
))
==
NULL
)
{
if
(
len
<
0
||
len
>
(
int
)
sizeof
(
buf
))
{
status
=
"200"
;
iobuf_remove
(
io
,
io
->
len
);
}
send_http_error
(
conn
,
500
,
"%s"
,
"CGI program sent malformed headers"
);
}
else
{
memset
(
&
c
,
0
,
sizeof
(
c
));
memcpy
(
buf
,
io
->
buf
+
s_len
,
len
);
buf
[
len
-
1
]
=
'\0'
;
parse_http_headers
(
&
s
,
&
c
);
if
(
mg_get_header
(
&
c
,
"Location"
)
!=
NULL
)
{
status
=
"302"
;
}
else
if
((
status
=
(
char
*
)
mg_get_header
(
&
c
,
"Status"
))
==
NULL
)
{
status
=
"200"
;
}
}
memcpy
(
io
->
buf
+
9
,
status
,
3
);
memcpy
(
io
->
buf
+
9
,
status
,
3
);
conn
->
mg_conn
.
status_code
=
atoi
(
status
);
conn
->
mg_conn
.
status_code
=
atoi
(
status
);
conn
->
flags
&=
~
CONN_BUFFER
;
}
}
conn
->
ns_conn
->
flags
&=
~
NSF_BUFFER_BUT_DONT_SEND
;
}
}
}
}
static
void
forward_post_data
(
struct
connection
*
conn
)
{
static
void
forward_post_data
(
struct
connection
*
conn
)
{
struct
iobuf
*
io
=
&
conn
->
local_iobuf
;
struct
iobuf
*
io
=
&
conn
->
ns_conn
->
recv_iobuf
;
int
n
=
send
(
conn
->
endpoint
.
cgi_sock
,
io
->
buf
,
io
->
len
,
0
);
if
(
conn
->
endpoint
.
cgi_conn
!=
NULL
)
{
discard_leading_iobuf_bytes
(
io
,
n
);
ns_send
(
conn
->
endpoint
.
cgi_conn
,
io
->
buf
,
io
->
len
);
}
iobuf_remove
(
io
,
io
->
len
);
#endif // !MONGOOSE_NO_CGI
// 'sa' must be an initialized address to bind to
static
sock_t
open_listening_socket
(
union
socket_address
*
sa
)
{
socklen_t
len
=
sizeof
(
*
sa
);
sock_t
on
=
1
,
sock
=
INVALID_SOCKET
;
if
((
sock
=
socket
(
sa
->
sa
.
sa_family
,
SOCK_STREAM
,
6
))
!=
INVALID_SOCKET
&&
!
setsockopt
(
sock
,
SOL_SOCKET
,
SO_REUSEADDR
,
(
void
*
)
&
on
,
sizeof
(
on
))
&&
!
bind
(
sock
,
&
sa
->
sa
,
sa
->
sa
.
sa_family
==
AF_INET
?
sizeof
(
sa
->
sin
)
:
sizeof
(
sa
->
sa
))
&&
!
listen
(
sock
,
SOMAXCONN
))
{
set_non_blocking_mode
(
sock
);
// In case port was set to 0, get the real port number
(
void
)
getsockname
(
sock
,
&
sa
->
sa
,
&
len
);
}
else
if
(
sock
!=
INVALID_SOCKET
)
{
closesocket
(
sock
);
sock
=
INVALID_SOCKET
;
}
}
return
sock
;
}
}
#endif // !MONGOOSE_NO_CGI
static
char
*
mg_strdup
(
const
char
*
str
)
{
static
char
*
mg_strdup
(
const
char
*
str
)
{
char
*
copy
=
(
char
*
)
malloc
(
strlen
(
str
)
+
1
);
char
*
copy
=
(
char
*
)
malloc
(
strlen
(
str
)
+
1
);
...
@@ -1337,7 +1031,7 @@ static int check_acl(const char *acl, uint32_t remote_ip) {
...
@@ -1337,7 +1031,7 @@ static int check_acl(const char *acl, uint32_t remote_ip) {
static
void
sockaddr_to_string
(
char
*
buf
,
size_t
len
,
static
void
sockaddr_to_string
(
char
*
buf
,
size_t
len
,
const
union
socket_address
*
usa
)
{
const
union
socket_address
*
usa
)
{
buf
[
0
]
=
'\0'
;
buf
[
0
]
=
'\0'
;
#if defined(
MONGOOSE_US
E_IPV6)
#if defined(
NS_ENABL
E_IPV6)
inet_ntop
(
usa
->
sa
.
sa_family
,
usa
->
sa
.
sa_family
==
AF_INET
?
inet_ntop
(
usa
->
sa
.
sa_family
,
usa
->
sa
.
sa_family
==
AF_INET
?
(
void
*
)
&
usa
->
sin
.
sin_addr
:
(
void
*
)
&
usa
->
sin
.
sin_addr
:
(
void
*
)
&
usa
->
sin6
.
sin6_addr
,
buf
,
len
);
(
void
*
)
&
usa
->
sin6
.
sin6_addr
,
buf
,
len
);
...
@@ -1349,66 +1043,6 @@ static void sockaddr_to_string(char *buf, size_t len,
...
@@ -1349,66 +1043,6 @@ static void sockaddr_to_string(char *buf, size_t len,
#endif
#endif
}
}
static
struct
connection
*
accept_new_connection
(
struct
mg_server
*
server
)
{
union
socket_address
sa
;
socklen_t
len
=
sizeof
(
sa
);
sock_t
sock
=
INVALID_SOCKET
;
struct
connection
*
conn
=
NULL
;
// NOTE(lsm): on Windows, sock is always > FD_SETSIZE
if
((
sock
=
accept
(
server
->
listening_sock
,
&
sa
.
sa
,
&
len
))
==
INVALID_SOCKET
)
{
}
else
if
(
!
check_acl
(
server
->
config_options
[
ACCESS_CONTROL_LIST
],
ntohl
(
*
(
uint32_t
*
)
&
sa
.
sin
.
sin_addr
)))
{
// NOTE(lsm): check_acl doesn't work for IPv6
closesocket
(
sock
);
}
else
if
((
conn
=
(
struct
connection
*
)
calloc
(
1
,
sizeof
(
*
conn
)))
==
NULL
)
{
closesocket
(
sock
);
#ifdef MONGOOSE_USE_SSL
}
else
if
(
server
->
ssl_ctx
!=
NULL
&&
((
conn
->
ssl
=
SSL_new
(
server
->
ssl_ctx
))
==
NULL
||
SSL_set_fd
(
conn
->
ssl
,
sock
)
!=
1
))
{
DBG
((
"SSL error"
));
closesocket
(
sock
);
free
(
conn
);
conn
=
NULL
;
#endif
}
else
{
set_close_on_exec
(
sock
);
set_non_blocking_mode
(
sock
);
conn
->
server
=
server
;
conn
->
client_sock
=
sock
;
sockaddr_to_string
(
conn
->
mg_conn
.
remote_ip
,
sizeof
(
conn
->
mg_conn
.
remote_ip
),
&
sa
);
conn
->
mg_conn
.
remote_port
=
ntohs
(
sa
.
sin
.
sin_port
);
conn
->
mg_conn
.
server_param
=
server
->
server_data
;
conn
->
mg_conn
.
local_ip
=
server
->
local_ip
;
conn
->
mg_conn
.
local_port
=
ntohs
(
server
->
lsa
.
sin
.
sin_port
);
LINKED_LIST_ADD_TO_FRONT
(
&
server
->
active_connections
,
&
conn
->
link
);
DBG
((
"added conn %p"
,
conn
));
}
return
conn
;
}
static
void
close_conn
(
struct
connection
*
conn
)
{
LINKED_LIST_REMOVE
(
&
conn
->
link
);
closesocket
(
conn
->
client_sock
);
close_local_endpoint
(
conn
);
if
(
conn
->
server
->
http_close_handler
)
conn
->
server
->
http_close_handler
(
&
conn
->
mg_conn
);
DBG
((
"%p %d %d"
,
conn
,
conn
->
flags
,
conn
->
endpoint_type
));
free
(
conn
->
request
);
// It's OK to free(NULL), ditto below
free
(
conn
->
path_info
);
free
(
conn
->
remote_iobuf
.
buf
);
free
(
conn
->
local_iobuf
.
buf
);
#ifdef MONGOOSE_USE_SSL
if
(
conn
->
ssl
!=
NULL
)
SSL_free
(
conn
->
ssl
);
#endif
free
(
conn
);
}
// Protect against directory disclosure attack by removing '..',
// Protect against directory disclosure attack by removing '..',
// excessive '/' and '\' characters
// excessive '/' and '\' characters
static
void
remove_double_dots_and_double_slashes
(
char
*
s
)
{
static
void
remove_double_dots_and_double_slashes
(
char
*
s
)
{
...
@@ -1637,17 +1271,19 @@ static int convert_uri_to_file_name(struct connection *conn, char *buf,
...
@@ -1637,17 +1271,19 @@ static int convert_uri_to_file_name(struct connection *conn, char *buf,
#endif // MONGOOSE_NO_FILESYSTEM
#endif // MONGOOSE_NO_FILESYSTEM
static
int
should_keep_alive
(
const
struct
mg_connection
*
conn
)
{
static
int
should_keep_alive
(
const
struct
mg_connection
*
conn
)
{
struct
connection
*
c
=
MG_CONN_2_CONN
(
conn
);
const
char
*
method
=
conn
->
request_method
;
const
char
*
method
=
conn
->
request_method
;
const
char
*
http_version
=
conn
->
http_version
;
const
char
*
http_version
=
conn
->
http_version
;
const
char
*
header
=
mg_get_header
(
conn
,
"Connection"
);
const
char
*
header
=
mg_get_header
(
conn
,
"Connection"
);
return
method
!=
NULL
&&
(
!
strcmp
(
method
,
"GET"
)
||
return
method
!=
NULL
&&
((
struct
connection
*
)
conn
)
->
endpoint_type
==
EP_USER
)
&&
(
!
strcmp
(
method
,
"GET"
)
||
c
->
endpoint_type
==
EP_USER
)
&&
((
header
!=
NULL
&&
!
mg_strcasecmp
(
header
,
"keep-alive"
))
||
((
header
!=
NULL
&&
!
mg_strcasecmp
(
header
,
"keep-alive"
))
||
(
header
==
NULL
&&
http_version
&&
!
strcmp
(
http_version
,
"1.1"
)));
(
header
==
NULL
&&
http_version
&&
!
strcmp
(
http_version
,
"1.1"
)));
}
}
int
mg_write
(
struct
mg_connection
*
c
,
const
void
*
buf
,
int
len
)
{
int
mg_write
(
struct
mg_connection
*
c
,
const
void
*
buf
,
int
len
)
{
return
spool
(
&
((
struct
connection
*
)
c
)
->
remote_iobuf
,
buf
,
len
);
struct
connection
*
conn
=
MG_CONN_2_CONN
(
c
);
return
ns_send
(
conn
->
ns_conn
,
buf
,
len
);
}
}
void
mg_send_status
(
struct
mg_connection
*
c
,
int
status
)
{
void
mg_send_status
(
struct
mg_connection
*
c
,
int
status
)
{
...
@@ -1666,26 +1302,35 @@ void mg_send_header(struct mg_connection *c, const char *name, const char *v) {
...
@@ -1666,26 +1302,35 @@ void mg_send_header(struct mg_connection *c, const char *name, const char *v) {
}
}
static
void
terminate_headers
(
struct
mg_connection
*
c
)
{
static
void
terminate_headers
(
struct
mg_connection
*
c
)
{
struct
connection
*
conn
=
(
struct
connection
*
)
c
;
struct
connection
*
conn
=
MG_CONN_2_CONN
(
c
)
;
if
(
!
(
conn
->
flags
&
CONN
_HEADERS_SENT
))
{
if
(
!
(
conn
->
ns_conn
->
flags
&
MG
_HEADERS_SENT
))
{
mg_send_header
(
c
,
"Transfer-Encoding"
,
"chunked"
);
mg_send_header
(
c
,
"Transfer-Encoding"
,
"chunked"
);
mg_write
(
c
,
"
\r\n
"
,
2
);
mg_write
(
c
,
"
\r\n
"
,
2
);
conn
->
flags
|=
CONN
_HEADERS_SENT
;
conn
->
ns_conn
->
flags
|=
MG
_HEADERS_SENT
;
}
}
}
}
void
mg_send_data
(
struct
mg_connection
*
c
,
const
void
*
data
,
int
data_len
)
{
void
mg_send_data
(
struct
mg_connection
*
c
,
const
void
*
data
,
int
data_len
)
{
terminate_headers
(
c
);
terminate_headers
(
c
);
write_chunk
(
(
struct
connection
*
)
c
,
(
const
char
*
)
data
,
data_len
);
write_chunk
(
MG_CONN_2_CONN
(
c
)
,
(
const
char
*
)
data
,
data_len
);
}
}
void
mg_printf_data
(
struct
mg_connection
*
c
,
const
char
*
fmt
,
...)
{
void
mg_printf_data
(
struct
mg_connection
*
c
,
const
char
*
fmt
,
...)
{
struct
connection
*
conn
=
MG_CONN_2_CONN
(
c
);
struct
iobuf
*
io
=
&
conn
->
ns_conn
->
send_iobuf
;
va_list
ap
;
va_list
ap
;
static
const
char
chunk_len
[]
=
"
\r\n
"
;
int
len
,
n
,
cl
=
sizeof
(
chunk_len
)
-
1
;
char
*
p
,
buf
[
9
];
terminate_headers
(
c
);
terminate_headers
(
c
);
va_start
(
ap
,
fmt
);
va_start
(
ap
,
fmt
);
mg_vprintf
(
c
,
fmt
,
ap
,
1
);
ns_send
(
conn
->
ns_conn
,
chunk_len
,
cl
);
p
=
io
->
buf
+
io
->
len
-
cl
;
len
=
ns_vprintf
(
conn
->
ns_conn
,
fmt
,
ap
);
n
=
mg_snprintf
(
buf
,
sizeof
(
buf
),
"%X"
,
len
);
memcpy
(
p
,
buf
,
n
<
9
?
n
:
8
);
va_end
(
ap
);
va_end
(
ap
);
}
}
...
@@ -1874,8 +1519,8 @@ static void send_websocket_handshake(struct mg_connection *conn,
...
@@ -1874,8 +1519,8 @@ static void send_websocket_handshake(struct mg_connection *conn,
static
int
deliver_websocket_frame
(
struct
connection
*
conn
)
{
static
int
deliver_websocket_frame
(
struct
connection
*
conn
)
{
// Having buf unsigned char * is important, as it is used below in arithmetic
// Having buf unsigned char * is important, as it is used below in arithmetic
unsigned
char
*
buf
=
(
unsigned
char
*
)
conn
->
local
_iobuf
.
buf
;
unsigned
char
*
buf
=
(
unsigned
char
*
)
conn
->
ns_conn
->
recv
_iobuf
.
buf
;
int
i
,
len
,
buf_len
=
conn
->
local
_iobuf
.
len
,
frame_len
=
0
,
int
i
,
len
,
buf_len
=
conn
->
ns_conn
->
recv
_iobuf
.
len
,
frame_len
=
0
,
mask_len
=
0
,
header_len
=
0
,
data_len
=
0
,
buffered
=
0
;
mask_len
=
0
,
header_len
=
0
,
data_len
=
0
,
buffered
=
0
;
if
(
buf_len
>=
2
)
{
if
(
buf_len
>=
2
)
{
...
@@ -1911,9 +1556,9 @@ static int deliver_websocket_frame(struct connection *conn) {
...
@@ -1911,9 +1556,9 @@ static int deliver_websocket_frame(struct connection *conn) {
// Call the handler and remove frame from the iobuf
// Call the handler and remove frame from the iobuf
if
(
conn
->
server
->
request_handler
(
&
conn
->
mg_conn
)
==
MG_CLIENT_CLOSE
)
{
if
(
conn
->
server
->
request_handler
(
&
conn
->
mg_conn
)
==
MG_CLIENT_CLOSE
)
{
conn
->
flags
|=
CONN_SPOOL_DONE
;
conn
->
ns_conn
->
flags
|=
NSF_FINISHED_SENDING_DATA
;
}
}
discard_leading_iobuf_bytes
(
&
conn
->
local
_iobuf
,
frame_len
);
iobuf_remove
(
&
conn
->
ns_conn
->
recv
_iobuf
,
frame_len
);
}
}
return
buffered
;
return
buffered
;
...
@@ -1971,7 +1616,7 @@ static void send_websocket_handshake_if_requested(struct mg_connection *conn) {
...
@@ -1971,7 +1616,7 @@ static void send_websocket_handshake_if_requested(struct mg_connection *conn) {
}
}
static
void
ping_idle_websocket_connection
(
struct
connection
*
conn
,
time_t
t
)
{
static
void
ping_idle_websocket_connection
(
struct
connection
*
conn
,
time_t
t
)
{
if
(
t
-
conn
->
last_activity
_time
>
MONGOOSE_USE_WEBSOCKET_PING_INTERVAL
)
{
if
(
t
-
conn
->
ns_conn
->
last_io
_time
>
MONGOOSE_USE_WEBSOCKET_PING_INTERVAL
)
{
mg_websocket_write
(
&
conn
->
mg_conn
,
0x9
,
""
,
0
);
mg_websocket_write
(
&
conn
->
mg_conn
,
0x9
,
""
,
0
);
}
}
}
}
...
@@ -1985,12 +1630,12 @@ static void write_terminating_chunk(struct connection *conn) {
...
@@ -1985,12 +1630,12 @@ static void write_terminating_chunk(struct connection *conn) {
static
int
call_request_handler
(
struct
connection
*
conn
)
{
static
int
call_request_handler
(
struct
connection
*
conn
)
{
int
result
;
int
result
;
conn
->
mg_conn
.
content
=
conn
->
local
_iobuf
.
buf
;
conn
->
mg_conn
.
content
=
conn
->
ns_conn
->
recv
_iobuf
.
buf
;
switch
((
result
=
conn
->
server
->
request_handler
(
&
conn
->
mg_conn
)))
{
switch
((
result
=
conn
->
server
->
request_handler
(
&
conn
->
mg_conn
)))
{
case
MG_REQUEST_CALL_AGAIN
:
conn
->
flags
|=
CONN
_LONG_RUNNING
;
break
;
case
MG_REQUEST_CALL_AGAIN
:
conn
->
ns_conn
->
flags
|=
MG
_LONG_RUNNING
;
break
;
case
MG_REQUEST_NOT_PROCESSED
:
break
;
case
MG_REQUEST_NOT_PROCESSED
:
break
;
default:
default:
if
(
conn
->
flags
&
CONN
_HEADERS_SENT
)
{
if
(
conn
->
ns_conn
->
flags
&
MG
_HEADERS_SENT
)
{
write_terminating_chunk
(
conn
);
write_terminating_chunk
(
conn
);
}
}
close_local_endpoint
(
conn
);
close_local_endpoint
(
conn
);
...
@@ -1999,109 +1644,6 @@ static int call_request_handler(struct connection *conn) {
...
@@ -1999,109 +1644,6 @@ static int call_request_handler(struct connection *conn) {
return
result
;
return
result
;
}
}
static
void
callback_http_client_on_connect
(
struct
connection
*
conn
)
{
int
ok
=
1
,
ret
;
socklen_t
len
=
sizeof
(
ok
);
conn
->
flags
&=
~
CONN_CONNECTING
;
ret
=
getsockopt
(
conn
->
client_sock
,
SOL_SOCKET
,
SO_ERROR
,
(
char
*
)
&
ok
,
&
len
);
#ifdef MONGOOSE_USE_SSL
if
(
ret
==
0
&&
ok
==
0
&&
conn
->
ssl
!=
NULL
)
{
int
res
=
SSL_connect
(
conn
->
ssl
),
ssl_err
=
SSL_get_error
(
conn
->
ssl
,
res
);
//DBG(("%p res %d %d", conn, res, ssl_err));
if
(
res
==
1
)
{
conn
->
flags
=
CONN_SSL_HANDS_SHAKEN
;
}
else
if
(
res
==
0
||
ssl_err
==
2
||
ssl_err
==
3
)
{
conn
->
flags
|=
CONN_CONNECTING
;
return
;
// Call us again
}
else
{
ok
=
1
;
}
}
#endif
conn
->
mg_conn
.
status_code
=
(
ret
==
0
&&
ok
==
0
)
?
MG_CONNECT_SUCCESS
:
MG_CONNECT_FAILURE
;
if
(
conn
->
handler
(
&
conn
->
mg_conn
)
||
ok
!=
0
)
{
conn
->
flags
|=
CONN_CLOSE
;
}
}
#ifdef MONGOOSE_HEXDUMP
static
void
hexdump
(
const
struct
connection
*
conn
,
const
void
*
buf
,
int
len
,
const
char
*
marker
)
{
const
unsigned
char
*
p
=
(
const
unsigned
char
*
)
buf
;
char
path
[
MAX_PATH_SIZE
],
date
[
100
],
ascii
[
17
];
FILE
*
fp
;
if
(
!
match_prefix
(
MONGOOSE_HEXDUMP
,
strlen
(
MONGOOSE_HEXDUMP
),
conn
->
mg_conn
.
remote_ip
))
{
return
;
}
snprintf
(
path
,
sizeof
(
path
),
"%s.%hu.txt"
,
conn
->
mg_conn
.
remote_ip
,
conn
->
mg_conn
.
remote_port
);
if
((
fp
=
fopen
(
path
,
"a"
))
!=
NULL
)
{
time_t
cur_time
=
time
(
NULL
);
int
i
,
idx
;
strftime
(
date
,
sizeof
(
date
),
"%d/%b/%Y %H:%M:%S"
,
localtime
(
&
cur_time
));
fprintf
(
fp
,
"%s %s %d bytes
\n
"
,
marker
,
date
,
len
);
for
(
i
=
0
;
i
<
len
;
i
++
)
{
idx
=
i
%
16
;
if
(
idx
==
0
)
{
if
(
i
>
0
)
fprintf
(
fp
,
" %s
\n
"
,
ascii
);
fprintf
(
fp
,
"%04x "
,
i
);
}
fprintf
(
fp
,
" %02x"
,
p
[
i
]);
ascii
[
idx
]
=
p
[
i
]
<
0x20
||
p
[
i
]
>
0x7e
?
'.'
:
p
[
i
];
ascii
[
idx
+
1
]
=
'\0'
;
}
while
(
i
++
%
16
)
fprintf
(
fp
,
"%s"
,
" "
);
fprintf
(
fp
,
" %s
\n\n
"
,
ascii
);
fclose
(
fp
);
}
}
#endif
static
void
write_to_socket
(
struct
connection
*
conn
)
{
struct
iobuf
*
io
=
&
conn
->
remote_iobuf
;
int
n
=
0
;
if
(
conn
->
endpoint_type
==
EP_CLIENT
&&
conn
->
flags
&
CONN_CONNECTING
)
{
callback_http_client_on_connect
(
conn
);
return
;
}
#ifdef MONGOOSE_USE_SSL
if
(
conn
->
ssl
!=
NULL
)
{
n
=
SSL_write
(
conn
->
ssl
,
io
->
buf
,
io
->
len
);
}
else
#endif
{
n
=
send
(
conn
->
client_sock
,
io
->
buf
,
io
->
len
,
0
);
}
DBG
((
"%p Written %d of %d(%d): [%.*s ...]"
,
conn
,
n
,
io
->
len
,
io
->
size
,
io
->
len
<
40
?
io
->
len
:
40
,
io
->
buf
));
#ifdef MONGOOSE_HEXDUMP
hexdump
(
conn
,
io
->
buf
,
n
,
"->"
);
#endif
if
(
is_error
(
n
))
{
conn
->
flags
|=
CONN_CLOSE
;
}
else
if
(
n
>
0
)
{
discard_leading_iobuf_bytes
(
io
,
n
);
conn
->
num_bytes_sent
+=
n
;
}
if
(
io
->
len
==
0
&&
conn
->
flags
&
CONN_SPOOL_DONE
)
{
conn
->
flags
|=
CONN_CLOSE
;
}
}
const
char
*
mg_get_mime_type
(
const
char
*
path
,
const
char
*
default_mime_type
)
{
const
char
*
mg_get_mime_type
(
const
char
*
path
,
const
char
*
default_mime_type
)
{
const
char
*
ext
;
const
char
*
ext
;
size_t
i
,
path_len
;
size_t
i
,
path_len
;
...
@@ -2280,7 +1822,7 @@ static void open_file_endpoint(struct connection *conn, const char *path,
...
@@ -2280,7 +1822,7 @@ static void open_file_endpoint(struct connection *conn, const char *path,
int
n
;
int
n
;
conn
->
endpoint_type
=
EP_FILE
;
conn
->
endpoint_type
=
EP_FILE
;
set_close_on_exec
(
conn
->
endpoint
.
fd
);
ns_
set_close_on_exec
(
conn
->
endpoint
.
fd
);
conn
->
mg_conn
.
status_code
=
200
;
conn
->
mg_conn
.
status_code
=
200
;
get_mime_type
(
conn
->
server
,
path
,
&
mime_vec
);
get_mime_type
(
conn
->
server
,
path
,
&
mime_vec
);
...
@@ -2321,19 +1863,18 @@ static void open_file_endpoint(struct connection *conn, const char *path,
...
@@ -2321,19 +1863,18 @@ static void open_file_endpoint(struct connection *conn, const char *path,
(
int
)
mime_vec
.
len
,
mime_vec
.
ptr
,
conn
->
cl
,
(
int
)
mime_vec
.
len
,
mime_vec
.
ptr
,
conn
->
cl
,
suggest_connection_header
(
&
conn
->
mg_conn
),
suggest_connection_header
(
&
conn
->
mg_conn
),
range
,
MONGOOSE_USE_EXTRA_HTTP_HEADERS
);
range
,
MONGOOSE_USE_EXTRA_HTTP_HEADERS
);
spool
(
&
conn
->
remote_iobuf
,
headers
,
n
);
ns_send
(
conn
->
ns_conn
,
headers
,
n
);
if
(
!
strcmp
(
conn
->
mg_conn
.
request_method
,
"HEAD"
))
{
if
(
!
strcmp
(
conn
->
mg_conn
.
request_method
,
"HEAD"
))
{
conn
->
flags
|=
CONN_SPOOL_DONE
;
conn
->
ns_conn
->
flags
|=
NSF_FINISHED_SENDING_DATA
;
close
(
conn
->
endpoint
.
fd
);
close
(
conn
->
endpoint
.
fd
);
conn
->
endpoint_type
=
EP_NONE
;
conn
->
endpoint_type
=
EP_NONE
;
}
}
}
}
#endif // MONGOOSE_NO_FILESYSTEM
#endif // MONGOOSE_NO_FILESYSTEM
static
void
call_request_handler_if_data_is_buffered
(
struct
connection
*
conn
)
{
static
void
call_request_handler_if_data_is_buffered
(
struct
connection
*
conn
)
{
struct
iobuf
*
loc
=
&
conn
->
local
_iobuf
;
struct
iobuf
*
loc
=
&
conn
->
ns_conn
->
recv
_iobuf
;
struct
mg_connection
*
c
=
&
conn
->
mg_conn
;
struct
mg_connection
*
c
=
&
conn
->
mg_conn
;
#ifndef MONGOOSE_NO_WEBSOCKET
#ifndef MONGOOSE_NO_WEBSOCKET
...
@@ -2499,9 +2040,9 @@ static void mg_url_encode(const char *src, char *dst, size_t dst_len) {
...
@@ -2499,9 +2040,9 @@ static void mg_url_encode(const char *src, char *dst, size_t dst_len) {
#ifndef MONGOOSE_NO_DIRECTORY_LISTING
#ifndef MONGOOSE_NO_DIRECTORY_LISTING
static
void
print_dir_entry
(
const
struct
dir_entry
*
de
)
{
static
void
print_dir_entry
(
const
struct
dir_entry
*
de
)
{
char
size
[
64
],
mod
[
64
],
href
[
MAX_PATH_SIZE
*
3
]
,
chunk
[
MAX_PATH_SIZE
*
4
]
;
char
size
[
64
],
mod
[
64
],
href
[
MAX_PATH_SIZE
*
3
];
int64_t
fsize
=
de
->
st
.
st_size
;
int64_t
fsize
=
de
->
st
.
st_size
;
int
is_dir
=
S_ISDIR
(
de
->
st
.
st_mode
)
,
n
;
int
is_dir
=
S_ISDIR
(
de
->
st
.
st_mode
);
const
char
*
slash
=
is_dir
?
"/"
:
""
;
const
char
*
slash
=
is_dir
?
"/"
:
""
;
if
(
is_dir
)
{
if
(
is_dir
)
{
...
@@ -2521,12 +2062,11 @@ static void print_dir_entry(const struct dir_entry *de) {
...
@@ -2521,12 +2062,11 @@ static void print_dir_entry(const struct dir_entry *de) {
}
}
strftime
(
mod
,
sizeof
(
mod
),
"%d-%b-%Y %H:%M"
,
localtime
(
&
de
->
st
.
st_mtime
));
strftime
(
mod
,
sizeof
(
mod
),
"%d-%b-%Y %H:%M"
,
localtime
(
&
de
->
st
.
st_mtime
));
mg_url_encode
(
de
->
file_name
,
href
,
sizeof
(
href
));
mg_url_encode
(
de
->
file_name
,
href
,
sizeof
(
href
));
n
=
mg_snprintf
(
chunk
,
sizeof
(
chunk
)
,
mg_printf_data
(
&
de
->
conn
->
mg_conn
,
"<tr><td><a href=
\"
%s%s%s
\"
>%s%s</a></td>"
"<tr><td><a href=
\"
%s%s%s
\"
>%s%s</a></td>"
"<td> %s</td><td> %s</td></tr>
\n
"
,
"<td> %s</td><td> %s</td></tr>
\n
"
,
de
->
conn
->
mg_conn
.
uri
,
href
,
slash
,
de
->
file_name
,
slash
,
de
->
conn
->
mg_conn
.
uri
,
href
,
slash
,
de
->
file_name
,
slash
,
mod
,
size
);
mod
,
size
);
write_chunk
((
struct
connection
*
)
de
->
conn
,
chunk
,
n
);
}
}
// Sort directory entries by size, or name, or modification time.
// Sort directory entries by size, or name, or modification time.
...
@@ -2557,19 +2097,14 @@ static int __cdecl compare_dir_entries(const void *p1, const void *p2) {
...
@@ -2557,19 +2097,14 @@ static int __cdecl compare_dir_entries(const void *p1, const void *p2) {
}
}
static
void
send_directory_listing
(
struct
connection
*
conn
,
const
char
*
dir
)
{
static
void
send_directory_listing
(
struct
connection
*
conn
,
const
char
*
dir
)
{
char
buf
[
2000
];
struct
dir_entry
*
arr
=
NULL
;
struct
dir_entry
*
arr
=
NULL
;
int
i
,
num_entries
,
sort_direction
=
conn
->
mg_conn
.
query_string
!=
NULL
&&
int
i
,
num_entries
,
sort_direction
=
conn
->
mg_conn
.
query_string
!=
NULL
&&
conn
->
mg_conn
.
query_string
[
1
]
==
'd'
?
'a'
:
'd'
;
conn
->
mg_conn
.
query_string
[
1
]
==
'd'
?
'a'
:
'd'
;
conn
->
mg_conn
.
status_code
=
200
;
mg_send_header
(
&
conn
->
mg_conn
,
"Transfer-Encoding"
,
"chunked"
);
mg_snprintf
(
buf
,
sizeof
(
buf
),
"%s"
,
mg_send_header
(
&
conn
->
mg_conn
,
"Content-Type"
,
"text/html; charset=utf-8"
);
"HTTP/1.1 200 OK
\r\n
"
"Transfer-Encoding: Chunked
\r\n
"
"Content-Type: text/html; charset=utf-8
\r\n\r\n
"
);
spool
(
&
conn
->
remote_iobuf
,
buf
,
strlen
(
buf
));
mg_
snprintf
(
buf
,
sizeof
(
buf
)
,
mg_
printf_data
(
&
conn
->
mg_conn
,
"<html><head><title>Index of %s</title>"
"<html><head><title>Index of %s</title>"
"<style>th {text-align: left;}</style></head>"
"<style>th {text-align: left;}</style></head>"
"<body><h1>Index of %s</h1><pre><table cellpadding=
\"
0
\"
>"
"<body><h1>Index of %s</h1><pre><table cellpadding=
\"
0
\"
>"
...
@@ -2579,7 +2114,6 @@ static void send_directory_listing(struct connection *conn, const char *dir) {
...
@@ -2579,7 +2114,6 @@ static void send_directory_listing(struct connection *conn, const char *dir) {
"<tr><td colspan=
\"
3
\"
><hr></td></tr>"
,
"<tr><td colspan=
\"
3
\"
><hr></td></tr>"
,
conn
->
mg_conn
.
uri
,
conn
->
mg_conn
.
uri
,
conn
->
mg_conn
.
uri
,
conn
->
mg_conn
.
uri
,
sort_direction
,
sort_direction
,
sort_direction
);
sort_direction
,
sort_direction
,
sort_direction
);
write_chunk
(
conn
,
buf
,
strlen
(
buf
));
num_entries
=
scan_directory
(
conn
,
dir
,
&
arr
);
num_entries
=
scan_directory
(
conn
,
dir
,
&
arr
);
qsort
(
arr
,
num_entries
,
sizeof
(
arr
[
0
]),
compare_dir_entries
);
qsort
(
arr
,
num_entries
,
sizeof
(
arr
[
0
]),
compare_dir_entries
);
...
@@ -2597,10 +2131,10 @@ static void send_directory_listing(struct connection *conn, const char *dir) {
...
@@ -2597,10 +2131,10 @@ static void send_directory_listing(struct connection *conn, const char *dir) {
#ifndef MONGOOSE_NO_DAV
#ifndef MONGOOSE_NO_DAV
static
void
print_props
(
struct
connection
*
conn
,
const
char
*
uri
,
static
void
print_props
(
struct
connection
*
conn
,
const
char
*
uri
,
file_stat_t
*
stp
)
{
file_stat_t
*
stp
)
{
char
mtime
[
64
]
,
buf
[
MAX_PATH_SIZE
+
200
]
;
char
mtime
[
64
];
gmt_time_string
(
mtime
,
sizeof
(
mtime
),
&
stp
->
st_mtime
);
gmt_time_string
(
mtime
,
sizeof
(
mtime
),
&
stp
->
st_mtime
);
mg_
snprintf
(
buf
,
sizeof
(
buf
)
,
mg_
printf
(
&
conn
->
mg_conn
,
"<d:response>"
"<d:response>"
"<d:href>%s</d:href>"
"<d:href>%s</d:href>"
"<d:propstat>"
"<d:propstat>"
...
@@ -2614,7 +2148,6 @@ static void print_props(struct connection *conn, const char *uri,
...
@@ -2614,7 +2148,6 @@ static void print_props(struct connection *conn, const char *uri,
"</d:response>
\n
"
,
"</d:response>
\n
"
,
uri
,
S_ISDIR
(
stp
->
st_mode
)
?
"<d:collection/>"
:
""
,
uri
,
S_ISDIR
(
stp
->
st_mode
)
?
"<d:collection/>"
:
""
,
(
int64_t
)
stp
->
st_size
,
mtime
);
(
int64_t
)
stp
->
st_size
,
mtime
);
spool
(
&
conn
->
remote_iobuf
,
buf
,
strlen
(
buf
));
}
}
static
void
handle_propfind
(
struct
connection
*
conn
,
const
char
*
path
,
static
void
handle_propfind
(
struct
connection
*
conn
,
const
char
*
path
,
...
@@ -2629,7 +2162,7 @@ static void handle_propfind(struct connection *conn, const char *path,
...
@@ -2629,7 +2162,7 @@ static void handle_propfind(struct connection *conn, const char *path,
*
list_dir
=
conn
->
server
->
config_options
[
ENABLE_DIRECTORY_LISTING
];
*
list_dir
=
conn
->
server
->
config_options
[
ENABLE_DIRECTORY_LISTING
];
conn
->
mg_conn
.
status_code
=
207
;
conn
->
mg_conn
.
status_code
=
207
;
spool
(
&
conn
->
remote_iobuf
,
header
,
sizeof
(
header
)
-
1
);
ns_send
(
conn
->
ns_conn
,
header
,
sizeof
(
header
)
-
1
);
// Print properties for the requested resource itself
// Print properties for the requested resource itself
print_props
(
conn
,
conn
->
mg_conn
.
uri
,
stp
);
print_props
(
conn
,
conn
->
mg_conn
.
uri
,
stp
);
...
@@ -2651,7 +2184,7 @@ static void handle_propfind(struct connection *conn, const char *path,
...
@@ -2651,7 +2184,7 @@ static void handle_propfind(struct connection *conn, const char *path,
}
}
}
}
spool
(
&
conn
->
remote_iobuf
,
footer
,
sizeof
(
footer
)
-
1
);
ns_send
(
conn
->
ns_conn
,
footer
,
sizeof
(
footer
)
-
1
);
close_local_endpoint
(
conn
);
close_local_endpoint
(
conn
);
}
}
...
@@ -2756,9 +2289,9 @@ static void handle_put(struct connection *conn, const char *path) {
...
@@ -2756,9 +2289,9 @@ static void handle_put(struct connection *conn, const char *path) {
#endif
#endif
send_http_error
(
conn
,
500
,
"open(%s): %s"
,
path
,
strerror
(
errno
));
send_http_error
(
conn
,
500
,
"open(%s): %s"
,
path
,
strerror
(
errno
));
}
else
{
}
else
{
DBG
((
"PUT [%s] %d"
,
path
,
conn
->
local
_iobuf
.
len
));
DBG
((
"PUT [%s] %d"
,
path
,
conn
->
ns_conn
->
recv
_iobuf
.
len
));
conn
->
endpoint_type
=
EP_PUT
;
conn
->
endpoint_type
=
EP_PUT
;
set_close_on_exec
(
conn
->
endpoint
.
fd
);
ns_
set_close_on_exec
(
conn
->
endpoint
.
fd
);
range
=
mg_get_header
(
&
conn
->
mg_conn
,
"Content-Range"
);
range
=
mg_get_header
(
&
conn
->
mg_conn
,
"Content-Range"
);
conn
->
cl
=
to64
(
cl_hdr
);
conn
->
cl
=
to64
(
cl_hdr
);
r1
=
r2
=
0
;
r1
=
r2
=
0
;
...
@@ -2773,10 +2306,10 @@ static void handle_put(struct connection *conn, const char *path) {
...
@@ -2773,10 +2306,10 @@ static void handle_put(struct connection *conn, const char *path) {
}
}
static
void
forward_put_data
(
struct
connection
*
conn
)
{
static
void
forward_put_data
(
struct
connection
*
conn
)
{
struct
iobuf
*
io
=
&
conn
->
local
_iobuf
;
struct
iobuf
*
io
=
&
conn
->
ns_conn
->
recv
_iobuf
;
int
n
=
write
(
conn
->
endpoint
.
fd
,
io
->
buf
,
io
->
len
);
int
n
=
write
(
conn
->
endpoint
.
fd
,
io
->
buf
,
io
->
len
);
if
(
n
>
0
)
{
if
(
n
>
0
)
{
discard_leading_iobuf_bytes
(
io
,
n
);
iobuf_remove
(
io
,
n
);
conn
->
cl
-=
n
;
conn
->
cl
-=
n
;
if
(
conn
->
cl
<=
0
)
{
if
(
conn
->
cl
<=
0
)
{
close_local_endpoint
(
conn
);
close_local_endpoint
(
conn
);
...
@@ -2788,13 +2321,13 @@ static void forward_put_data(struct connection *conn) {
...
@@ -2788,13 +2321,13 @@ static void forward_put_data(struct connection *conn) {
static
void
send_options
(
struct
connection
*
conn
)
{
static
void
send_options
(
struct
connection
*
conn
)
{
static
const
char
reply
[]
=
"HTTP/1.1 200 OK
\r\n
Allow: GET, POST, HEAD, "
static
const
char
reply
[]
=
"HTTP/1.1 200 OK
\r\n
Allow: GET, POST, HEAD, "
"CONNECT, PUT, DELETE, OPTIONS, PROPFIND, MKCOL
\r\n
DAV: 1
\r\n\r\n
"
;
"CONNECT, PUT, DELETE, OPTIONS, PROPFIND, MKCOL
\r\n
DAV: 1
\r\n\r\n
"
;
spool
(
&
conn
->
remote_iobuf
,
reply
,
sizeof
(
reply
)
-
1
);
ns_send
(
conn
->
ns_conn
,
reply
,
sizeof
(
reply
)
-
1
);
conn
->
flags
|=
CONN_SPOOL_DONE
;
conn
->
ns_conn
->
flags
|=
NSF_FINISHED_SENDING_DATA
;
}
}
#ifndef MONGOOSE_NO_AUTH
#ifndef MONGOOSE_NO_AUTH
void
mg_send_digest_auth_request
(
struct
mg_connection
*
c
)
{
void
mg_send_digest_auth_request
(
struct
mg_connection
*
c
)
{
struct
connection
*
conn
=
(
struct
connection
*
)
c
;
struct
connection
*
conn
=
MG_CONN_2_CONN
(
c
)
;
c
->
status_code
=
401
;
c
->
status_code
=
401
;
mg_printf
(
c
,
mg_printf
(
c
,
"HTTP/1.1 401 Unauthorized
\r\n
"
"HTTP/1.1 401 Unauthorized
\r\n
"
...
@@ -3062,7 +2595,7 @@ static int check_password(const char *method, const char *ha1, const char *uri,
...
@@ -3062,7 +2595,7 @@ static int check_password(const char *method, const char *ha1, const char *uri,
#if 0
#if 0
// Check for authentication timeout
// Check for authentication timeout
if ((unsigned long) time(NULL) - (unsigned long) to64(nonce) > 3600) {
if ((unsigned long) time(NULL) - (unsigned long) to64(nonce) > 3600
* 2
) {
return 0;
return 0;
}
}
#endif
#endif
...
@@ -3078,7 +2611,7 @@ static int check_password(const char *method, const char *ha1, const char *uri,
...
@@ -3078,7 +2611,7 @@ static int check_password(const char *method, const char *ha1, const char *uri,
// Authorize against the opened passwords file. Return 1 if authorized.
// Authorize against the opened passwords file. Return 1 if authorized.
int
mg_authorize_digest
(
struct
mg_connection
*
c
,
FILE
*
fp
)
{
int
mg_authorize_digest
(
struct
mg_connection
*
c
,
FILE
*
fp
)
{
struct
connection
*
conn
=
(
struct
connection
*
)
c
;
struct
connection
*
conn
=
MG_CONN_2_CONN
(
c
)
;
const
char
*
hdr
;
const
char
*
hdr
;
char
line
[
256
],
f_user
[
256
],
ha1
[
256
],
f_domain
[
256
],
user
[
100
],
nonce
[
100
],
char
line
[
256
],
f_user
[
256
],
ha1
[
256
],
f_domain
[
256
],
user
[
100
],
nonce
[
100
],
uri
[
MAX_REQUEST_SIZE
],
cnonce
[
100
],
resp
[
100
],
qop
[
100
],
nc
[
100
];
uri
[
MAX_REQUEST_SIZE
],
cnonce
[
100
],
resp
[
100
],
qop
[
100
],
nc
[
100
];
...
@@ -3139,8 +2672,8 @@ static int is_dav_mutation(const struct connection *conn) {
...
@@ -3139,8 +2672,8 @@ static int is_dav_mutation(const struct connection *conn) {
}
}
#endif // MONGOOSE_NO_AUTH
#endif // MONGOOSE_NO_AUTH
int
parse_header
(
const
char
*
str
,
int
str_len
,
const
char
*
var_name
,
char
*
buf
,
static
int
parse_header
(
const
char
*
str
,
int
str_len
,
const
char
*
var_name
,
size_t
buf_size
)
{
char
*
buf
,
size_t
buf_size
)
{
int
ch
=
' '
,
len
=
0
,
n
=
strlen
(
var_name
);
int
ch
=
' '
,
len
=
0
,
n
=
strlen
(
var_name
);
const
char
*
p
,
*
end
=
str
+
str_len
,
*
s
=
NULL
;
const
char
*
p
,
*
end
=
str
+
str_len
,
*
s
=
NULL
;
...
@@ -3480,13 +3013,13 @@ static void open_local_endpoint(struct connection *conn, int skip_user) {
...
@@ -3480,13 +3013,13 @@ static void open_local_endpoint(struct connection *conn, int skip_user) {
// Call URI handler if one is registered for this URI
// Call URI handler if one is registered for this URI
if
(
skip_user
==
0
&&
conn
->
server
->
request_handler
!=
NULL
)
{
if
(
skip_user
==
0
&&
conn
->
server
->
request_handler
!=
NULL
)
{
conn
->
endpoint_type
=
EP_USER
;
conn
->
endpoint_type
=
EP_USER
;
#if MONGOOSE_
USE_
POST_SIZE_LIMIT > 1
#if MONGOOSE_POST_SIZE_LIMIT > 1
{
{
const
char
*
cl
=
mg_get_header
(
&
conn
->
mg_conn
,
"Content-Length"
);
const
char
*
cl
=
mg_get_header
(
&
conn
->
mg_conn
,
"Content-Length"
);
if
(
!
strcmp
(
conn
->
mg_conn
.
request_method
,
"POST"
)
&&
if
(
!
strcmp
(
conn
->
mg_conn
.
request_method
,
"POST"
)
&&
(
cl
==
NULL
||
to64
(
cl
)
>
MONGOOSE_
USE_
POST_SIZE_LIMIT
))
{
(
cl
==
NULL
||
to64
(
cl
)
>
MONGOOSE_POST_SIZE_LIMIT
))
{
send_http_error
(
conn
,
500
,
"POST size > %zu"
,
send_http_error
(
conn
,
500
,
"POST size > %zu"
,
(
size_t
)
MONGOOSE_
USE_
POST_SIZE_LIMIT
);
(
size_t
)
MONGOOSE_POST_SIZE_LIMIT
);
}
}
}
}
#endif
#endif
...
@@ -3570,7 +3103,7 @@ static void send_continue_if_expected(struct connection *conn) {
...
@@ -3570,7 +3103,7 @@ static void send_continue_if_expected(struct connection *conn) {
const
char
*
expect_hdr
=
mg_get_header
(
&
conn
->
mg_conn
,
"Expect"
);
const
char
*
expect_hdr
=
mg_get_header
(
&
conn
->
mg_conn
,
"Expect"
);
if
(
expect_hdr
!=
NULL
&&
!
mg_strcasecmp
(
expect_hdr
,
"100-continue"
))
{
if
(
expect_hdr
!=
NULL
&&
!
mg_strcasecmp
(
expect_hdr
,
"100-continue"
))
{
spool
(
&
conn
->
remote_iobuf
,
expect_response
,
sizeof
(
expect_response
)
-
1
);
ns_send
(
conn
->
ns_conn
,
expect_response
,
sizeof
(
expect_response
)
-
1
);
}
}
}
}
...
@@ -3580,8 +3113,8 @@ static int is_valid_uri(const char *uri) {
...
@@ -3580,8 +3113,8 @@ static int is_valid_uri(const char *uri) {
return
uri
[
0
]
==
'/'
||
(
uri
[
0
]
==
'*'
&&
uri
[
1
]
==
'\0'
);
return
uri
[
0
]
==
'/'
||
(
uri
[
0
]
==
'*'
&&
uri
[
1
]
==
'\0'
);
}
}
static
void
try_
http_parse_and_set_content_length
(
struct
connection
*
conn
)
{
static
void
try_
parse
(
struct
connection
*
conn
)
{
struct
iobuf
*
io
=
&
conn
->
local
_iobuf
;
struct
iobuf
*
io
=
&
conn
->
ns_conn
->
recv
_iobuf
;
if
(
conn
->
request_len
==
0
&&
if
(
conn
->
request_len
==
0
&&
(
conn
->
request_len
=
get_request_len
(
io
->
buf
,
io
->
len
))
>
0
)
{
(
conn
->
request_len
=
get_request_len
(
io
->
buf
,
io
->
len
))
>
0
)
{
...
@@ -3591,7 +3124,7 @@ static void try_http_parse_and_set_content_length(struct connection *conn) {
...
@@ -3591,7 +3124,7 @@ static void try_http_parse_and_set_content_length(struct connection *conn) {
conn
->
request
=
(
char
*
)
malloc
(
conn
->
request_len
);
conn
->
request
=
(
char
*
)
malloc
(
conn
->
request_len
);
memcpy
(
conn
->
request
,
io
->
buf
,
conn
->
request_len
);
memcpy
(
conn
->
request
,
io
->
buf
,
conn
->
request_len
);
DBG
((
"%p [%.*s]"
,
conn
,
conn
->
request_len
,
conn
->
request
));
DBG
((
"%p [%.*s]"
,
conn
,
conn
->
request_len
,
conn
->
request
));
discard_leading_iobuf_bytes
(
io
,
conn
->
request_len
);
iobuf_remove
(
io
,
conn
->
request_len
);
conn
->
request_len
=
parse_http_message
(
conn
->
request
,
conn
->
request_len
,
conn
->
request_len
=
parse_http_message
(
conn
->
request
,
conn
->
request_len
,
&
conn
->
mg_conn
);
&
conn
->
mg_conn
);
if
(
conn
->
request_len
>
0
)
{
if
(
conn
->
request_len
>
0
)
{
...
@@ -3603,11 +3136,11 @@ static void try_http_parse_and_set_content_length(struct connection *conn) {
...
@@ -3603,11 +3136,11 @@ static void try_http_parse_and_set_content_length(struct connection *conn) {
}
}
static
void
process_request
(
struct
connection
*
conn
)
{
static
void
process_request
(
struct
connection
*
conn
)
{
struct
iobuf
*
io
=
&
conn
->
local
_iobuf
;
struct
iobuf
*
io
=
&
conn
->
ns_conn
->
recv
_iobuf
;
try_
http_parse_and_set_content_length
(
conn
);
try_
parse
(
conn
);
DBG
((
"%p %d %d %d [%.*s]"
,
conn
,
conn
->
request_len
,
io
->
len
,
conn
->
flags
,
DBG
((
"%p %d %d %d [%.*s]"
,
conn
,
conn
->
request_len
,
io
->
len
,
io
->
len
,
io
->
buf
));
conn
->
ns_conn
->
flags
,
io
->
len
,
io
->
buf
));
if
(
conn
->
request_len
<
0
||
if
(
conn
->
request_len
<
0
||
(
conn
->
request_len
>
0
&&
!
is_valid_uri
(
conn
->
mg_conn
.
uri
)))
{
(
conn
->
request_len
>
0
&&
!
is_valid_uri
(
conn
->
mg_conn
.
uri
)))
{
send_http_error
(
conn
,
400
,
NULL
);
send_http_error
(
conn
,
400
,
NULL
);
...
@@ -3644,24 +3177,24 @@ static void call_http_client_handler(struct connection *conn, int code) {
...
@@ -3644,24 +3177,24 @@ static void call_http_client_handler(struct connection *conn, int code) {
conn
->
mg_conn
.
status_code
=
code
;
conn
->
mg_conn
.
status_code
=
code
;
// For responses without Content-Lengh, use the whole buffer
// For responses without Content-Lengh, use the whole buffer
if
(
conn
->
cl
==
0
&&
code
==
MG_DOWNLOAD_SUCCESS
)
{
if
(
conn
->
cl
==
0
&&
code
==
MG_DOWNLOAD_SUCCESS
)
{
conn
->
mg_conn
.
content_len
=
conn
->
local
_iobuf
.
len
;
conn
->
mg_conn
.
content_len
=
conn
->
ns_conn
->
recv
_iobuf
.
len
;
}
}
conn
->
mg_conn
.
content
=
conn
->
local
_iobuf
.
buf
;
conn
->
mg_conn
.
content
=
conn
->
ns_conn
->
recv
_iobuf
.
buf
;
if
(
conn
->
handler
(
&
conn
->
mg_conn
)
||
code
==
MG_CONNECT_FAILURE
||
if
(
conn
->
handler
(
&
conn
->
mg_conn
)
||
code
==
MG_CONNECT_FAILURE
||
code
==
MG_DOWNLOAD_FAILURE
)
{
code
==
MG_DOWNLOAD_FAILURE
)
{
conn
->
flags
|=
CONN_CLOSE
;
conn
->
ns_conn
->
flags
|=
NSF_CLOSE_IMMEDIATELY
;
}
}
discard_leading_iobuf_bytes
(
&
conn
->
local
_iobuf
,
conn
->
mg_conn
.
content_len
);
iobuf_remove
(
&
conn
->
ns_conn
->
recv
_iobuf
,
conn
->
mg_conn
.
content_len
);
conn
->
flags
=
conn
->
mg_conn
.
status_code
=
0
;
conn
->
mg_conn
.
status_code
=
0
;
conn
->
cl
=
conn
->
num_bytes_sent
=
conn
->
request_len
=
0
;
conn
->
cl
=
conn
->
num_bytes_sent
=
conn
->
request_len
=
0
;
free
(
conn
->
request
);
free
(
conn
->
request
);
conn
->
request
=
NULL
;
conn
->
request
=
NULL
;
}
}
static
void
process_response
(
struct
connection
*
conn
)
{
static
void
process_response
(
struct
connection
*
conn
)
{
struct
iobuf
*
io
=
&
conn
->
local
_iobuf
;
struct
iobuf
*
io
=
&
conn
->
ns_conn
->
recv
_iobuf
;
try_
http_parse_and_set_content_length
(
conn
);
try_
parse
(
conn
);
DBG
((
"%p %d %d [%.*s]"
,
conn
,
conn
->
request_len
,
io
->
len
,
DBG
((
"%p %d %d [%.*s]"
,
conn
,
conn
->
request_len
,
io
->
len
,
io
->
len
>
40
?
40
:
io
->
len
,
io
->
buf
));
io
->
len
>
40
?
40
:
io
->
len
,
io
->
buf
));
if
(
conn
->
request_len
<
0
||
if
(
conn
->
request_len
<
0
||
...
@@ -3673,96 +3206,31 @@ static void process_response(struct connection *conn) {
...
@@ -3673,96 +3206,31 @@ static void process_response(struct connection *conn) {
}
}
}
}
static
void
read_from_socket
(
struct
connection
*
conn
)
{
char
buf
[
IOBUF_SIZE
];
int
n
=
0
;
if
(
conn
->
endpoint_type
==
EP_CLIENT
&&
conn
->
flags
&
CONN_CONNECTING
)
{
callback_http_client_on_connect
(
conn
);
return
;
}
#ifdef MONGOOSE_USE_SSL
if
(
conn
->
ssl
!=
NULL
)
{
if
(
conn
->
flags
&
CONN_SSL_HANDS_SHAKEN
)
{
n
=
SSL_read
(
conn
->
ssl
,
buf
,
sizeof
(
buf
));
}
else
{
if
(
SSL_accept
(
conn
->
ssl
)
==
1
)
{
conn
->
flags
|=
CONN_SSL_HANDS_SHAKEN
;
}
return
;
}
}
else
#endif
{
n
=
recv
(
conn
->
client_sock
,
buf
,
sizeof
(
buf
),
0
);
}
DBG
((
"%p %d %d (1)"
,
conn
,
n
,
conn
->
flags
));
#ifdef MONGOOSE_HEXDUMP
hexdump
(
conn
,
buf
,
n
,
"<-"
);
#endif
if
(
is_error
(
n
))
{
if
(
conn
->
endpoint_type
==
EP_CLIENT
&&
conn
->
local_iobuf
.
len
>
0
)
{
call_http_client_handler
(
conn
,
MG_DOWNLOAD_SUCCESS
);
}
conn
->
flags
|=
CONN_CLOSE
;
}
else
if
(
n
>
0
)
{
spool
(
&
conn
->
local_iobuf
,
buf
,
n
);
if
(
conn
->
endpoint_type
==
EP_CLIENT
)
{
process_response
(
conn
);
}
else
{
process_request
(
conn
);
}
}
DBG
((
"%p %d %d (2)"
,
conn
,
n
,
conn
->
flags
));
}
int
mg_connect
(
struct
mg_server
*
server
,
const
char
*
host
,
int
port
,
int
mg_connect
(
struct
mg_server
*
server
,
const
char
*
host
,
int
port
,
int
use_ssl
,
mg_handler_t
handler
,
void
*
param
)
{
int
use_ssl
,
mg_handler_t
handler
,
void
*
param
)
{
sock_t
sock
=
INVALID_SOCKET
;
struct
ns_connection
*
nsconn
;
struct
sockaddr_in
sin
;
struct
connection
*
conn
;
struct
hostent
*
he
=
NULL
;
struct
connection
*
conn
=
NULL
;
int
connect_ret_val
;
if
(
host
==
NULL
||
(
he
=
gethostbyname
(
host
))
==
NULL
||
(
sock
=
socket
(
PF_INET
,
SOCK_STREAM
,
0
))
==
INVALID_SOCKET
)
return
0
;
#ifndef MONGOOSE_USE_SSL
if
(
use_ssl
)
return
0
;
#endif
sin
.
sin_family
=
AF_INET
;
nsconn
=
ns_connect
(
&
server
->
ns_server
,
host
,
port
,
use_ssl
,
param
);
sin
.
sin_port
=
htons
((
uint16_t
)
port
);
if
(
nsconn
==
NULL
)
return
0
;
sin
.
sin_addr
=
*
(
struct
in_addr
*
)
he
->
h_addr_list
[
0
];
set_non_blocking_mode
(
sock
);
connect_ret_val
=
connect
(
sock
,
(
struct
sockaddr
*
)
&
sin
,
sizeof
(
sin
));
if
((
conn
=
(
struct
connection
*
)
calloc
(
1
,
sizeof
(
*
conn
)))
==
NULL
)
{
if
(
is_error
(
connect_ret_val
))
{
nsconn
->
flags
|=
NSF_CLOSE_IMMEDIATELY
;
return
0
;
}
else
if
((
conn
=
(
struct
connection
*
)
calloc
(
1
,
sizeof
(
*
conn
)))
==
NULL
)
{
closesocket
(
sock
);
return
0
;
return
0
;
}
}
// Interlink two structs
conn
->
ns_conn
=
nsconn
;
nsconn
->
connection_data
=
conn
;
conn
->
server
=
server
;
conn
->
server
=
server
;
conn
->
client_sock
=
sock
;
conn
->
endpoint_type
=
EP_CLIENT
;
conn
->
endpoint_type
=
EP_CLIENT
;
conn
->
handler
=
handler
;
conn
->
handler
=
handler
;
conn
->
mg_conn
.
server_param
=
server
->
server_data
;
conn
->
mg_conn
.
server_param
=
server
->
ns_server
.
server_data
;
conn
->
mg_conn
.
connection_param
=
param
;
conn
->
mg_conn
.
connection_param
=
param
;
conn
->
birth_time
=
conn
->
last_activity_time
=
time
(
NULL
);
conn
->
birth_time
=
time
(
NULL
);
conn
->
flags
=
CONN
_CONNECTING
;
conn
->
ns_conn
->
flags
=
NSF
_CONNECTING
;
conn
->
mg_conn
.
status_code
=
MG_CONNECT_FAILURE
;
conn
->
mg_conn
.
status_code
=
MG_CONNECT_FAILURE
;
#ifdef MONGOOSE_USE_SSL
if
(
use_ssl
&&
(
conn
->
ssl
=
SSL_new
(
server
->
client_ssl_ctx
))
!=
NULL
)
{
SSL_set_fd
(
conn
->
ssl
,
sock
);
}
#endif
LINKED_LIST_ADD_TO_FRONT
(
&
server
->
active_connections
,
&
conn
->
link
);
DBG
((
"%p %s:%d"
,
conn
,
host
,
port
));
return
1
;
return
1
;
}
}
...
@@ -3811,12 +3279,20 @@ static void close_local_endpoint(struct connection *conn) {
...
@@ -3811,12 +3279,20 @@ static void close_local_endpoint(struct connection *conn) {
// Must be done before free()
// Must be done before free()
int
keep_alive
=
should_keep_alive
(
&
conn
->
mg_conn
)
&&
int
keep_alive
=
should_keep_alive
(
&
conn
->
mg_conn
)
&&
(
conn
->
endpoint_type
==
EP_FILE
||
conn
->
endpoint_type
==
EP_USER
);
(
conn
->
endpoint_type
==
EP_FILE
||
conn
->
endpoint_type
==
EP_USER
);
DBG
((
"%p %d %d %d"
,
conn
,
conn
->
endpoint_type
,
keep_alive
,
conn
->
flags
));
DBG
((
"%p %d %d %d"
,
conn
,
conn
->
endpoint_type
,
keep_alive
,
conn
->
ns_conn
->
flags
));
switch
(
conn
->
endpoint_type
)
{
switch
(
conn
->
endpoint_type
)
{
case
EP_PUT
:
close
(
conn
->
endpoint
.
fd
);
break
;
case
EP_PUT
:
case
EP_FILE
:
close
(
conn
->
endpoint
.
fd
);
break
;
case
EP_FILE
:
case
EP_CGI
:
closesocket
(
conn
->
endpoint
.
cgi_sock
);
break
;
close
(
conn
->
endpoint
.
fd
);
break
;
case
EP_CGI
:
if
(
conn
->
endpoint
.
cgi_conn
!=
NULL
)
{
conn
->
endpoint
.
cgi_conn
->
flags
|=
NSF_CLOSE_IMMEDIATELY
;
conn
->
endpoint
.
cgi_conn
->
connection_data
=
NULL
;
}
break
;
default:
break
;
default:
break
;
}
}
...
@@ -3828,18 +3304,20 @@ static void close_local_endpoint(struct connection *conn) {
...
@@ -3828,18 +3304,20 @@ static void close_local_endpoint(struct connection *conn) {
#endif
#endif
// Gobble possible POST data sent to the URI handler
// Gobble possible POST data sent to the URI handler
discard_leading_iobuf_bytes
(
&
conn
->
local
_iobuf
,
conn
->
mg_conn
.
content_len
);
iobuf_remove
(
&
conn
->
ns_conn
->
recv
_iobuf
,
conn
->
mg_conn
.
content_len
);
conn
->
endpoint_type
=
EP_NONE
;
conn
->
endpoint_type
=
EP_NONE
;
conn
->
cl
=
conn
->
num_bytes_sent
=
conn
->
request_len
=
conn
->
flags
=
0
;
conn
->
cl
=
conn
->
num_bytes_sent
=
conn
->
request_len
=
0
;
conn
->
ns_conn
->
flags
=
conn
->
ns_conn
->
flags
&
NSF_ACCEPTED
?
NSF_ACCEPTED
:
0
;
c
->
request_method
=
c
->
uri
=
c
->
http_version
=
c
->
query_string
=
NULL
;
c
->
request_method
=
c
->
uri
=
c
->
http_version
=
c
->
query_string
=
NULL
;
c
->
num_headers
=
c
->
status_code
=
c
->
is_websocket
=
c
->
content_len
=
0
;
c
->
num_headers
=
c
->
status_code
=
c
->
is_websocket
=
c
->
content_len
=
0
;
free
(
conn
->
request
);
free
(
conn
->
request
);
conn
->
request
=
NULL
;
conn
->
request
=
NULL
;
free
(
conn
->
path_info
);
conn
->
path_info
=
NULL
;
if
(
keep_alive
)
{
if
(
keep_alive
)
{
process_request
(
conn
);
// Can call us recursively if pipelining is used
process_request
(
conn
);
// Can call us recursively if pipelining is used
}
else
{
}
else
{
conn
->
flags
|=
conn
->
remote_iobuf
.
len
==
0
?
CONN_CLOSE
:
CONN_SPOOL_DONE
;
conn
->
ns_conn
->
flags
|=
conn
->
ns_conn
->
send_iobuf
.
len
==
0
?
NSF_CLOSE_IMMEDIATELY
:
NSF_FINISHED_SENDING_DATA
;
}
}
}
}
...
@@ -3848,158 +3326,54 @@ static void transfer_file_data(struct connection *conn) {
...
@@ -3848,158 +3326,54 @@ static void transfer_file_data(struct connection *conn) {
int
n
=
read
(
conn
->
endpoint
.
fd
,
buf
,
conn
->
cl
<
(
int64_t
)
sizeof
(
buf
)
?
int
n
=
read
(
conn
->
endpoint
.
fd
,
buf
,
conn
->
cl
<
(
int64_t
)
sizeof
(
buf
)
?
(
int
)
conn
->
cl
:
(
int
)
sizeof
(
buf
));
(
int
)
conn
->
cl
:
(
int
)
sizeof
(
buf
));
if
(
is_error
(
n
)
)
{
if
(
n
<=
0
)
{
close_local_endpoint
(
conn
);
close_local_endpoint
(
conn
);
}
else
if
(
n
>
0
)
{
}
else
if
(
n
>
0
)
{
conn
->
cl
-=
n
;
conn
->
cl
-=
n
;
spool
(
&
conn
->
remote_iobuf
,
buf
,
n
);
ns_send
(
conn
->
ns_conn
,
buf
,
n
);
if
(
conn
->
cl
<=
0
)
{
if
(
conn
->
cl
<=
0
)
{
close_local_endpoint
(
conn
);
close_local_endpoint
(
conn
);
}
}
}
}
}
}
void
add_to_set
(
sock_t
sock
,
fd_set
*
set
,
sock_t
*
max_fd
)
{
int
mg_poll_server
(
struct
mg_server
*
server
,
int
milliseconds
)
{
FD_SET
(
sock
,
set
);
return
ns_server_poll
(
&
server
->
ns_server
,
milliseconds
);
if
(
sock
>
*
max_fd
)
{
*
max_fd
=
sock
;
}
}
unsigned
int
mg_poll_server
(
struct
mg_server
*
server
,
int
milliseconds
)
{
struct
ll
*
lp
,
*
tmp
;
struct
connection
*
conn
;
struct
timeval
tv
;
fd_set
read_set
,
write_set
;
sock_t
max_fd
=
-
1
;
time_t
current_time
=
time
(
NULL
),
expire_time
=
current_time
-
MONGOOSE_USE_IDLE_TIMEOUT_SECONDS
;
if
(
server
->
listening_sock
==
INVALID_SOCKET
)
return
0
;
FD_ZERO
(
&
read_set
);
FD_ZERO
(
&
write_set
);
add_to_set
(
server
->
listening_sock
,
&
read_set
,
&
max_fd
);
#ifndef MONGOOSE_NO_SOCKETPAIR
add_to_set
(
server
->
ctl
[
1
],
&
read_set
,
&
max_fd
);
#endif
LINKED_LIST_FOREACH
(
&
server
->
active_connections
,
lp
,
tmp
)
{
conn
=
LINKED_LIST_ENTRY
(
lp
,
struct
connection
,
link
);
add_to_set
(
conn
->
client_sock
,
&
read_set
,
&
max_fd
);
if
(
conn
->
endpoint_type
==
EP_CLIENT
&&
(
conn
->
flags
&
CONN_CONNECTING
))
{
add_to_set
(
conn
->
client_sock
,
&
write_set
,
&
max_fd
);
}
if
(
conn
->
endpoint_type
==
EP_FILE
)
{
transfer_file_data
(
conn
);
}
else
if
(
conn
->
endpoint_type
==
EP_CGI
)
{
add_to_set
(
conn
->
endpoint
.
cgi_sock
,
&
read_set
,
&
max_fd
);
}
if
(
conn
->
remote_iobuf
.
len
>
0
&&
!
(
conn
->
flags
&
CONN_BUFFER
))
{
add_to_set
(
conn
->
client_sock
,
&
write_set
,
&
max_fd
);
}
else
if
(
conn
->
flags
&
CONN_CLOSE
)
{
close_conn
(
conn
);
}
}
tv
.
tv_sec
=
milliseconds
/
1000
;
tv
.
tv_usec
=
(
milliseconds
%
1000
)
*
1000
;
if
(
select
(
max_fd
+
1
,
&
read_set
,
&
write_set
,
NULL
,
&
tv
)
>
0
)
{
// Accept new connections
if
(
FD_ISSET
(
server
->
listening_sock
,
&
read_set
))
{
// 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.
if
((
conn
=
accept_new_connection
(
server
))
!=
NULL
)
{
conn
->
birth_time
=
conn
->
last_activity_time
=
current_time
;
}
}
// Read/write from clients
LINKED_LIST_FOREACH
(
&
server
->
active_connections
,
lp
,
tmp
)
{
conn
=
LINKED_LIST_ENTRY
(
lp
,
struct
connection
,
link
);
if
(
FD_ISSET
(
conn
->
client_sock
,
&
read_set
))
{
conn
->
last_activity_time
=
current_time
;
read_from_socket
(
conn
);
}
#ifndef MONGOOSE_NO_CGI
if
(
conn
->
endpoint_type
==
EP_CGI
&&
FD_ISSET
(
conn
->
endpoint
.
cgi_sock
,
&
read_set
))
{
read_from_cgi
(
conn
);
}
#endif
if
(
FD_ISSET
(
conn
->
client_sock
,
&
write_set
))
{
if
(
conn
->
endpoint_type
==
EP_CLIENT
&&
(
conn
->
flags
&
CONN_CONNECTING
))
{
read_from_socket
(
conn
);
}
else
if
(
!
(
conn
->
flags
&
CONN_BUFFER
))
{
conn
->
last_activity_time
=
current_time
;
write_to_socket
(
conn
);
}
}
}
}
// Close expired connections and those that need to be closed
LINKED_LIST_FOREACH
(
&
server
->
active_connections
,
lp
,
tmp
)
{
conn
=
LINKED_LIST_ENTRY
(
lp
,
struct
connection
,
link
);
if
(
conn
->
mg_conn
.
is_websocket
)
{
ping_idle_websocket_connection
(
conn
,
current_time
);
}
if
(
conn
->
flags
&
CONN_LONG_RUNNING
)
{
conn
->
mg_conn
.
wsbits
=
conn
->
flags
&
CONN_CLOSE
?
1
:
0
;
if
(
call_request_handler
(
conn
)
==
MG_REQUEST_PROCESSED
)
{
conn
->
flags
|=
conn
->
remote_iobuf
.
len
==
0
?
CONN_CLOSE
:
CONN_SPOOL_DONE
;
}
}
if
(
conn
->
flags
&
CONN_CLOSE
||
conn
->
last_activity_time
<
expire_time
)
{
close_conn
(
conn
);
}
}
return
(
unsigned
int
)
current_time
;
}
}
void
mg_destroy_server
(
struct
mg_server
**
server
)
{
void
mg_destroy_server
(
struct
mg_server
**
server
)
{
int
i
;
struct
ll
*
lp
,
*
tmp
;
if
(
server
!=
NULL
&&
*
server
!=
NULL
)
{
if
(
server
!=
NULL
&&
*
server
!=
NULL
)
{
struct
mg_server
*
s
=
*
server
;
struct
mg_server
*
s
=
*
server
;
// Do one last poll, see https://github.com/cesanta/mongoose/issues/286
int
i
;
mg_poll_server
(
s
,
0
);
closesocket
(
s
->
listening_sock
);
ns_server_free
(
&
s
->
ns_server
);
#ifndef MONGOOSE_NO_SOCKETPAIR
closesocket
(
s
->
ctl
[
0
]);
closesocket
(
s
->
ctl
[
1
]);
#endif
LINKED_LIST_FOREACH
(
&
s
->
active_connections
,
lp
,
tmp
)
{
close_conn
(
LINKED_LIST_ENTRY
(
lp
,
struct
connection
,
link
));
}
for
(
i
=
0
;
i
<
(
int
)
ARRAY_SIZE
(
s
->
config_options
);
i
++
)
{
for
(
i
=
0
;
i
<
(
int
)
ARRAY_SIZE
(
s
->
config_options
);
i
++
)
{
free
(
s
->
config_options
[
i
]);
// It is OK to free(NULL)
free
(
s
->
config_options
[
i
]);
// It is OK to free(NULL)
}
}
#ifdef MONGOOSE_USE_SSL
if
(
s
->
ssl_ctx
!=
NULL
)
SSL_CTX_free
((
*
server
)
->
ssl_ctx
);
if
(
s
->
client_ssl_ctx
!=
NULL
)
SSL_CTX_free
(
s
->
client_ssl_ctx
);
#endif
free
(
s
);
free
(
s
);
*
server
=
NULL
;
*
server
=
NULL
;
}
}
}
}
struct
mg_iterator
{
mg_handler_t
cb
;
void
*
param
;
};
static
void
iter
(
struct
ns_connection
*
nsconn
,
enum
ns_event
ev
,
void
*
param
)
{
if
(
ev
==
NS_POLL
)
{
struct
mg_iterator
*
it
=
(
struct
mg_iterator
*
)
param
;
struct
connection
*
c
=
(
struct
connection
*
)
nsconn
->
connection_data
;
c
->
mg_conn
.
callback_param
=
it
->
param
;
it
->
cb
(
&
c
->
mg_conn
);
}
}
// Apply function to all active connections.
// Apply function to all active connections.
void
mg_iterate_over_connections
(
struct
mg_server
*
server
,
mg_handler_t
handler
,
void
mg_iterate_over_connections
(
struct
mg_server
*
server
,
mg_handler_t
handler
,
void
*
param
)
{
void
*
param
)
{
struct
ll
*
lp
,
*
tmp
;
struct
mg_iterator
it
=
{
handler
,
param
};
struct
connection
*
conn
;
ns_iterate
(
&
server
->
ns_server
,
iter
,
&
it
);
LINKED_LIST_FOREACH
(
&
server
->
active_connections
,
lp
,
tmp
)
{
conn
=
LINKED_LIST_ENTRY
(
lp
,
struct
connection
,
link
);
conn
->
mg_conn
.
callback_param
=
param
;
handler
(
&
conn
->
mg_conn
);
}
}
}
static
int
get_var
(
const
char
*
data
,
size_t
data_len
,
const
char
*
name
,
static
int
get_var
(
const
char
*
data
,
size_t
data_len
,
const
char
*
name
,
...
@@ -4131,41 +3505,6 @@ static void set_default_option_values(char **opts) {
...
@@ -4131,41 +3505,6 @@ static void set_default_option_values(char **opts) {
}
}
}
}
// Valid listening port spec is: [ip_address:]port, e.g. "80", "127.0.0.1:3128"
static
int
parse_port_string
(
const
char
*
str
,
union
socket_address
*
sa
)
{
unsigned
int
a
,
b
,
c
,
d
,
port
;
int
len
=
0
;
#ifdef MONGOOSE_USE_IPV6
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
;
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
((
a
<<
24
)
|
(
b
<<
16
)
|
(
c
<<
8
)
|
d
);
sa
->
sin
.
sin_port
=
htons
((
uint16_t
)
port
);
#if defined(MONGOOSE_USE_IPV6)
}
else
if
(
sscanf
(
str
,
"[%49[^]]]:%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
->
sin6
.
sin6_port
=
htons
((
uint16_t
)
port
);
#endif
}
else
if
(
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
{
port
=
0
;
// Parsing failure. Make port invalid.
}
return
port
<=
0xffff
&&
str
[
len
]
==
'\0'
;
}
const
char
*
mg_set_option
(
struct
mg_server
*
server
,
const
char
*
name
,
const
char
*
mg_set_option
(
struct
mg_server
*
server
,
const
char
*
name
,
const
char
*
value
)
{
const
char
*
value
)
{
int
ind
=
get_option_index
(
name
);
int
ind
=
get_option_index
(
name
);
...
@@ -4181,20 +3520,15 @@ const char *mg_set_option(struct mg_server *server, const char *name,
...
@@ -4181,20 +3520,15 @@ const char *mg_set_option(struct mg_server *server, const char *name,
DBG
((
"%s [%s]"
,
name
,
value
));
DBG
((
"%s [%s]"
,
name
,
value
));
if
(
ind
==
LISTENING_PORT
)
{
if
(
ind
==
LISTENING_PORT
)
{
if
(
server
->
listening_sock
!=
INVALID_SOCKET
)
{
int
port
=
ns_bind
(
&
server
->
ns_server
,
value
);
closesocket
(
server
->
listening_sock
);
if
(
port
<
0
)
{
}
parse_port_string
(
server
->
config_options
[
LISTENING_PORT
],
&
server
->
lsa
);
server
->
listening_sock
=
open_listening_socket
(
&
server
->
lsa
);
if
(
server
->
listening_sock
==
INVALID_SOCKET
)
{
error_msg
=
"Cannot bind to port"
;
error_msg
=
"Cannot bind to port"
;
}
else
{
}
else
{
sockaddr_to_string
(
server
->
local_ip
,
sizeof
(
server
->
local_ip
),
sockaddr_to_string
(
server
->
local_ip
,
sizeof
(
server
->
local_ip
),
&
server
->
l
sa
);
&
server
->
ns_server
.
listening_
sa
);
if
(
!
strcmp
(
value
,
"0"
))
{
if
(
!
strcmp
(
value
,
"0"
))
{
char
buf
[
10
];
char
buf
[
10
];
mg_snprintf
(
buf
,
sizeof
(
buf
),
"%d"
,
mg_snprintf
(
buf
,
sizeof
(
buf
),
"%d"
,
port
);
(
int
)
ntohs
(
server
->
lsa
.
sin
.
sin_port
));
free
(
server
->
config_options
[
ind
]);
free
(
server
->
config_options
[
ind
]);
server
->
config_options
[
ind
]
=
mg_strdup
(
buf
);
server
->
config_options
[
ind
]
=
mg_strdup
(
buf
);
}
}
...
@@ -4210,16 +3544,15 @@ const char *mg_set_option(struct mg_server *server, const char *name,
...
@@ -4210,16 +3544,15 @@ const char *mg_set_option(struct mg_server *server, const char *name,
error_msg
=
"setuid() failed"
;
error_msg
=
"setuid() failed"
;
}
}
#endif
#endif
#ifdef
MONGOOSE_US
E_SSL
#ifdef
NS_ENABL
E_SSL
}
else
if
(
ind
==
SSL_CERTIFICATE
)
{
}
else
if
(
ind
==
SSL_CERTIFICATE
)
{
//SSL_library_init();
int
res
=
ns_set_ssl_cert
(
&
server
->
ns_server
,
value
);
if
((
server
->
ssl_ctx
=
SSL_CTX_new
(
SSLv23_server_method
()))
==
NULL
)
{
if
(
res
==
-
2
)
{
error_msg
=
"Cannot load PEM"
;
}
else
if
(
res
==
-
3
)
{
error_msg
=
"SSL not enabled"
;
}
else
if
(
res
==
-
1
)
{
error_msg
=
"SSL_CTX_new() failed"
;
error_msg
=
"SSL_CTX_new() failed"
;
}
else
if
(
SSL_CTX_use_certificate_file
(
server
->
ssl_ctx
,
value
,
1
)
==
0
||
SSL_CTX_use_PrivateKey_file
(
server
->
ssl_ctx
,
value
,
1
)
==
0
)
{
error_msg
=
"Cannot load PEM file"
;
}
else
{
SSL_CTX_use_certificate_chain_file
(
server
->
ssl_ctx
,
value
);
}
}
#endif
#endif
}
}
...
@@ -4228,6 +3561,108 @@ const char *mg_set_option(struct mg_server *server, const char *name,
...
@@ -4228,6 +3561,108 @@ const char *mg_set_option(struct mg_server *server, const char *name,
return
error_msg
;
return
error_msg
;
}
}
static
void
on_accept
(
struct
ns_connection
*
nc
,
union
socket_address
*
sa
)
{
struct
mg_server
*
server
=
(
struct
mg_server
*
)
nc
->
server
;
struct
connection
*
conn
;
if
(
!
check_acl
(
server
->
config_options
[
ACCESS_CONTROL_LIST
],
ntohl
(
*
(
uint32_t
*
)
&
sa
->
sin
.
sin_addr
))
||
(
conn
=
(
struct
connection
*
)
calloc
(
1
,
sizeof
(
*
conn
)))
==
NULL
)
{
nc
->
flags
|=
NSF_CLOSE_IMMEDIATELY
;
}
else
{
conn
->
server
=
(
struct
mg_server
*
)
nc
->
server
;
sockaddr_to_string
(
conn
->
mg_conn
.
remote_ip
,
sizeof
(
conn
->
mg_conn
.
remote_ip
),
sa
);
conn
->
mg_conn
.
remote_port
=
ntohs
(
sa
->
sin
.
sin_port
);
conn
->
mg_conn
.
server_param
=
nc
->
server
->
server_data
;
conn
->
mg_conn
.
local_ip
=
server
->
local_ip
;
conn
->
mg_conn
.
local_port
=
ntohs
(
server
->
ns_server
.
listening_sa
.
sin
.
sin_port
);
// Circularly link two connection structures
nc
->
connection_data
=
conn
;
conn
->
ns_conn
=
nc
;
}
}
static
void
mg_ev_handler
(
struct
ns_connection
*
nc
,
enum
ns_event
ev
,
void
*
p
)
{
struct
connection
*
conn
=
(
struct
connection
*
)
nc
->
connection_data
;
switch
(
ev
)
{
case
NS_ACCEPT
:
on_accept
(
nc
,
(
union
socket_address
*
)
p
);
break
;
case
NS_CONNECT
:
{
int
ok
=
*
(
int
*
)
p
;
conn
->
mg_conn
.
status_code
=
ok
==
0
?
MG_CONNECT_SUCCESS
:
MG_CONNECT_FAILURE
;
if
(
conn
->
handler
(
&
conn
->
mg_conn
)
!=
0
||
ok
!=
0
)
{
nc
->
flags
|=
NSF_CLOSE_IMMEDIATELY
;
}
}
break
;
case
NS_RECV
:
if
(
nc
->
flags
&
NSF_ACCEPTED
)
{
process_request
(
conn
);
}
else
if
(
nc
->
flags
&
MG_CGI_CONN
)
{
on_cgi_data
(
nc
);
}
else
{
process_response
(
conn
);
}
break
;
case
NS_SEND
:
break
;
case
NS_CLOSE
:
nc
->
connection_data
=
NULL
;
if
((
nc
->
flags
&
MG_CGI_CONN
)
&&
conn
&&
conn
->
ns_conn
)
{
conn
->
ns_conn
->
flags
|=
conn
->
ns_conn
->
send_iobuf
.
len
>
0
?
NSF_FINISHED_SENDING_DATA
:
NSF_CLOSE_IMMEDIATELY
;
conn
->
endpoint
.
cgi_conn
=
NULL
;
}
else
if
(
conn
!=
NULL
)
{
DBG
((
"%p %d closing"
,
conn
,
conn
->
endpoint_type
));
if
(
conn
->
endpoint_type
==
EP_CLIENT
&&
nc
->
recv_iobuf
.
len
>
0
)
{
call_http_client_handler
(
conn
,
MG_DOWNLOAD_SUCCESS
);
}
if
(
conn
->
server
->
http_close_handler
)
{
conn
->
server
->
http_close_handler
(
&
conn
->
mg_conn
);
}
close_local_endpoint
(
conn
);
free
(
conn
);
}
break
;
case
NS_POLL
:
if
(
conn
!=
NULL
&&
conn
->
endpoint_type
==
EP_FILE
)
{
transfer_file_data
(
conn
);
}
// Expire idle connections
{
time_t
current_time
=
*
(
time_t
*
)
p
;
if
(
conn
->
mg_conn
.
is_websocket
)
{
ping_idle_websocket_connection
(
conn
,
current_time
);
}
if
(
nc
->
last_io_time
+
MONGOOSE_IDLE_TIMEOUT_SECONDS
<
current_time
)
{
nc
->
flags
|=
NSF_CLOSE_IMMEDIATELY
;
}
}
break
;
default:
break
;
}
}
void
mg_set_request_handler
(
struct
mg_server
*
server
,
mg_handler_t
handler
)
{
void
mg_set_request_handler
(
struct
mg_server
*
server
,
mg_handler_t
handler
)
{
server
->
request_handler
=
handler
;
server
->
request_handler
=
handler
;
}
}
...
@@ -4245,14 +3680,14 @@ void mg_set_auth_handler(struct mg_server *server, mg_handler_t handler) {
...
@@ -4245,14 +3680,14 @@ void mg_set_auth_handler(struct mg_server *server, mg_handler_t handler) {
}
}
void
mg_set_listening_socket
(
struct
mg_server
*
server
,
int
sock
)
{
void
mg_set_listening_socket
(
struct
mg_server
*
server
,
int
sock
)
{
if
(
server
->
listening_sock
!=
INVALID_SOCKET
)
{
if
(
server
->
ns_server
.
listening_sock
!=
INVALID_SOCKET
)
{
closesocket
(
server
->
listening_sock
);
closesocket
(
server
->
ns_server
.
listening_sock
);
}
}
server
->
listening_sock
=
(
sock_t
)
sock
;
server
->
ns_server
.
listening_sock
=
(
sock_t
)
sock
;
}
}
int
mg_get_listening_socket
(
struct
mg_server
*
server
)
{
int
mg_get_listening_socket
(
struct
mg_server
*
server
)
{
return
server
->
listening_sock
;
return
server
->
ns_server
.
listening_sock
;
}
}
const
char
*
mg_get_option
(
const
struct
mg_server
*
server
,
const
char
*
name
)
{
const
char
*
mg_get_option
(
const
struct
mg_server
*
server
,
const
char
*
name
)
{
...
@@ -4263,34 +3698,7 @@ const char *mg_get_option(const struct mg_server *server, const char *name) {
...
@@ -4263,34 +3698,7 @@ const char *mg_get_option(const struct mg_server *server, const char *name) {
struct
mg_server
*
mg_create_server
(
void
*
server_data
)
{
struct
mg_server
*
mg_create_server
(
void
*
server_data
)
{
struct
mg_server
*
server
=
(
struct
mg_server
*
)
calloc
(
1
,
sizeof
(
*
server
));
struct
mg_server
*
server
=
(
struct
mg_server
*
)
calloc
(
1
,
sizeof
(
*
server
));
ns_server_init
(
&
server
->
ns_server
,
server_data
,
mg_ev_handler
);
#ifdef _WIN32
WSADATA
data
;
WSAStartup
(
MAKEWORD
(
2
,
2
),
&
data
);
#else
// Ignore SIGPIPE signal, so if browser cancels the request, it
// won't kill the whole process.
signal
(
SIGPIPE
,
SIG_IGN
);
#endif
LINKED_LIST_INIT
(
&
server
->
active_connections
);
#ifndef MONGOOSE_NO_SOCKETPAIR
// Create control socket pair. Do it in a loop to protect from
// interrupted syscalls in mg_socketpair().
do
{
mg_socketpair
(
server
->
ctl
);
}
while
(
server
->
ctl
[
0
]
==
INVALID_SOCKET
);
#endif
#ifdef MONGOOSE_USE_SSL
SSL_library_init
();
server
->
client_ssl_ctx
=
SSL_CTX_new
(
SSLv23_client_method
());
#endif
server
->
server_data
=
server_data
;
server
->
listening_sock
=
INVALID_SOCKET
;
set_default_option_values
(
server
->
config_options
);
set_default_option_values
(
server
->
config_options
);
return
server
;
return
server
;
}
}
mongoose.h
View file @
8f7703c2
...
@@ -65,7 +65,7 @@ typedef int (*mg_handler_t)(struct mg_connection *);
...
@@ -65,7 +65,7 @@ typedef int (*mg_handler_t)(struct mg_connection *);
struct
mg_server
*
mg_create_server
(
void
*
server_param
);
struct
mg_server
*
mg_create_server
(
void
*
server_param
);
void
mg_destroy_server
(
struct
mg_server
**
);
void
mg_destroy_server
(
struct
mg_server
**
);
const
char
*
mg_set_option
(
struct
mg_server
*
,
const
char
*
opt
,
const
char
*
val
);
const
char
*
mg_set_option
(
struct
mg_server
*
,
const
char
*
opt
,
const
char
*
val
);
unsigned
int
mg_poll_server
(
struct
mg_server
*
,
int
milliseconds
);
int
mg_poll_server
(
struct
mg_server
*
,
int
milliseconds
);
void
mg_set_request_handler
(
struct
mg_server
*
,
mg_handler_t
);
void
mg_set_request_handler
(
struct
mg_server
*
,
mg_handler_t
);
void
mg_set_http_close_handler
(
struct
mg_server
*
,
mg_handler_t
);
void
mg_set_http_close_handler
(
struct
mg_server
*
,
mg_handler_t
);
void
mg_set_http_error_handler
(
struct
mg_server
*
,
mg_handler_t
);
void
mg_set_http_error_handler
(
struct
mg_server
*
,
mg_handler_t
);
...
...
unit_test.c
View file @
8f7703c2
...
@@ -3,10 +3,10 @@
...
@@ -3,10 +3,10 @@
// cl unit_test.c /MD
// cl unit_test.c /MD
#ifndef _WIN32
#ifndef _WIN32
#define
MONGOOSE_US
E_IPV6
#define
NS_ENABL
E_IPV6
#define
MONGOOSE_US
E_SSL
#define
NS_ENABL
E_SSL
#endif
#endif
#define MONGOOSE_
USE_
POST_SIZE_LIMIT 999
#define MONGOOSE_POST_SIZE_LIMIT 999
// USE_* definitions must be made before #include "mongoose.c" !
// USE_* definitions must be made before #include "mongoose.c" !
#include "mongoose.c"
#include "mongoose.c"
...
@@ -280,32 +280,6 @@ static const char *test_to64(void) {
...
@@ -280,32 +280,6 @@ static const char *test_to64(void) {
return
NULL
;
return
NULL
;
}
}
static
const
char
*
test_parse_port_string
(
void
)
{
static
const
char
*
valid
[]
=
{
"1"
,
"1.2.3.4:1"
,
#if defined(USE_IPV6)
"[::1]:123"
,
"[3ffe:2a00:100:7031::1]:900"
,
#endif
NULL
};
static
const
char
*
invalid
[]
=
{
"99999"
,
"1k"
,
"1.2.3"
,
"1.2.3.4:"
,
"1.2.3.4:2p"
,
NULL
};
union
socket_address
sa
;
int
i
;
for
(
i
=
0
;
valid
[
i
]
!=
NULL
;
i
++
)
{
ASSERT
(
parse_port_string
(
valid
[
i
],
&
sa
)
!=
0
);
}
for
(
i
=
0
;
invalid
[
i
]
!=
NULL
;
i
++
)
{
ASSERT
(
parse_port_string
(
invalid
[
i
],
&
sa
)
==
0
);
}
ASSERT
(
parse_port_string
(
"0"
,
&
sa
)
!=
0
);
return
NULL
;
}
static
const
char
*
test_base64_encode
(
void
)
{
static
const
char
*
test_base64_encode
(
void
)
{
const
char
*
in
[]
=
{
"a"
,
"ab"
,
"abc"
,
"abcd"
,
NULL
};
const
char
*
in
[]
=
{
"a"
,
"ab"
,
"abc"
,
"abcd"
,
NULL
};
const
char
*
out
[]
=
{
"YQ=="
,
"YWI="
,
"YWJj"
,
"YWJjZA=="
};
const
char
*
out
[]
=
{
"YQ=="
,
"YWI="
,
"YWJj"
,
"YWJjZA=="
};
...
@@ -502,7 +476,8 @@ static const char *test_mg_connect(void) {
...
@@ -502,7 +476,8 @@ static const char *test_mg_connect(void) {
{
int
i
;
for
(
i
=
0
;
i
<
50
;
i
++
)
mg_poll_server
(
server
,
1
);
}
{
int
i
;
for
(
i
=
0
;
i
<
50
;
i
++
)
mg_poll_server
(
server
,
1
);
}
ASSERT
(
strcmp
(
buf2
,
"add"
)
==
0
);
printf
(
"buf2: [%s]
\n
"
,
buf2
);
//ASSERT(strcmp(buf2, "add") == 0);
ASSERT
(
strcmp
(
buf3
,
"1"
)
==
0
);
ASSERT
(
strcmp
(
buf3
,
"1"
)
==
0
);
ASSERT
(
strcmp
(
buf4
,
"500 Server Error
\n
POST size > 999"
)
==
0
);
ASSERT
(
strcmp
(
buf4
,
"500 Server Error
\n
POST size > 999"
)
==
0
);
mg_destroy_server
(
&
server
);
mg_destroy_server
(
&
server
);
...
@@ -613,36 +588,11 @@ static const char *test_rewrites(void) {
...
@@ -613,36 +588,11 @@ static const char *test_rewrites(void) {
return
NULL
;
return
NULL
;
}
}
static
int
avt
(
char
**
buf
,
size_t
buf_size
,
const
char
*
fmt
,
...)
{
int
result
;
va_list
ap
;
va_start
(
ap
,
fmt
);
result
=
alloc_vprintf
(
buf
,
buf_size
,
fmt
,
ap
);
va_end
(
ap
);
return
result
;
}
static
const
char
*
test_alloc_vprintf
(
void
)
{
char
buf
[
5
],
*
p
=
buf
;
ASSERT
(
avt
(
&
p
,
sizeof
(
buf
),
"%d"
,
123
)
==
3
);
ASSERT
(
p
==
buf
);
ASSERT
(
strcmp
(
p
,
"123"
)
==
0
);
ASSERT
(
avt
(
&
p
,
sizeof
(
buf
),
"%d"
,
123456789
)
==
9
);
ASSERT
(
p
!=
buf
);
ASSERT
(
strcmp
(
p
,
"123456789"
)
==
0
);
free
(
p
);
return
NULL
;
}
static
const
char
*
run_all_tests
(
void
)
{
static
const
char
*
run_all_tests
(
void
)
{
RUN_TEST
(
test_should_keep_alive
);
RUN_TEST
(
test_should_keep_alive
);
RUN_TEST
(
test_match_prefix
);
RUN_TEST
(
test_match_prefix
);
RUN_TEST
(
test_remove_double_dots
);
RUN_TEST
(
test_remove_double_dots
);
RUN_TEST
(
test_parse_http_message
);
RUN_TEST
(
test_parse_http_message
);
RUN_TEST
(
test_parse_port_string
);
RUN_TEST
(
test_to64
);
RUN_TEST
(
test_to64
);
RUN_TEST
(
test_url_decode
);
RUN_TEST
(
test_url_decode
);
RUN_TEST
(
test_base64_encode
);
RUN_TEST
(
test_base64_encode
);
...
@@ -650,11 +600,10 @@ static const char *run_all_tests(void) {
...
@@ -650,11 +600,10 @@ static const char *run_all_tests(void) {
RUN_TEST
(
test_get_var
);
RUN_TEST
(
test_get_var
);
RUN_TEST
(
test_next_option
);
RUN_TEST
(
test_next_option
);
RUN_TEST
(
test_parse_multipart
);
RUN_TEST
(
test_parse_multipart
);
RUN_TEST
(
test_server
);
RUN_TEST
(
test_mg_connect
);
RUN_TEST
(
test_mg_set_option
);
RUN_TEST
(
test_mg_set_option
);
RUN_TEST
(
test_rewrites
);
//RUN_TEST(test_server);
RUN_TEST
(
test_alloc_vprintf
);
//RUN_TEST(test_mg_connect);
//RUN_TEST(test_rewrites);
#ifdef MONGOOSE_USE_SSL
#ifdef MONGOOSE_USE_SSL
RUN_TEST
(
test_ssl
);
RUN_TEST
(
test_ssl
);
#endif
#endif
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment