mirror of
https://github.com/lammertb/libhttp.git
synced 2025-07-31 08:24:23 +03:00
replace Windows CRLF with Unix newline
Signed-off-by: Danny Al-Gaaf <danny.al-gaaf@bisect.de>
This commit is contained in:
@ -1,187 +1,187 @@
|
||||
#include <assert.h>
|
||||
#include <stdlib.h>
|
||||
#include <time.h>
|
||||
#include <string.h>
|
||||
#include "WebSockCallbacks.h"
|
||||
|
||||
#ifdef _WIN32
|
||||
#include <Windows.h>
|
||||
#define mg_sleep(x) Sleep(x)
|
||||
#else
|
||||
#include <unistd.h>
|
||||
#include <pthread.h>
|
||||
#define mg_sleep(x) usleep((x) * 1000)
|
||||
#endif
|
||||
|
||||
|
||||
static void send_to_all_websockets(struct mg_context *ctx, const char * data, int data_len) {
|
||||
|
||||
int i;
|
||||
tWebSockContext *ws_ctx = (tWebSockContext*) mg_get_user_data(ctx);
|
||||
|
||||
mg_lock_context(ctx);
|
||||
for (i=0;i<MAX_NUM_OF_WEBSOCKS;i++) {
|
||||
if (ws_ctx->socketList[i] && (ws_ctx->socketList[i]->webSockState==2)) {
|
||||
mg_websocket_write(ws_ctx->socketList[i]->conn, WEBSOCKET_OPCODE_TEXT, data, data_len);
|
||||
}
|
||||
}
|
||||
mg_unlock_context(ctx);
|
||||
}
|
||||
|
||||
|
||||
void websocket_ready_handler(struct mg_connection *conn) {
|
||||
|
||||
int i;
|
||||
struct mg_request_info * rq = mg_get_request_info(conn);
|
||||
struct mg_context * ctx = mg_get_context(conn);
|
||||
tWebSockContext *ws_ctx = (tWebSockContext*) mg_get_user_data(ctx);
|
||||
tWebSockInfo * wsock = malloc(sizeof(tWebSockInfo));
|
||||
assert(wsock);
|
||||
wsock->webSockState = 0;
|
||||
rq->conn_data = wsock;
|
||||
|
||||
mg_lock_context(ctx);
|
||||
for (i=0;i<MAX_NUM_OF_WEBSOCKS;i++) {
|
||||
if (0==ws_ctx->socketList[i]) {
|
||||
ws_ctx->socketList[i] = wsock;
|
||||
wsock->conn = conn;
|
||||
wsock->webSockState = 1;
|
||||
break;
|
||||
}
|
||||
}
|
||||
printf("\nNew websocket attached: %s:%u\n", rq->remote_addr, rq->remote_port);
|
||||
mg_unlock_context(ctx);
|
||||
}
|
||||
|
||||
|
||||
static void websocket_done(tWebSockContext *ws_ctx, tWebSockInfo * wsock) {
|
||||
|
||||
int i;
|
||||
|
||||
if (wsock) {
|
||||
wsock->webSockState = 99;
|
||||
for (i=0;i<MAX_NUM_OF_WEBSOCKS;i++) {
|
||||
if (wsock==ws_ctx->socketList[i]) {
|
||||
ws_ctx->socketList[i] = 0;
|
||||
break;
|
||||
}
|
||||
}
|
||||
printf("\nClose websocket attached: %s:%u\n", mg_get_request_info(wsock->conn)->remote_addr, mg_get_request_info(wsock->conn)->remote_port);
|
||||
free(wsock);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
int websocket_data_handler(struct mg_connection *conn, int flags, char *data, size_t data_len) {
|
||||
|
||||
struct mg_request_info * rq = mg_get_request_info(conn);
|
||||
tWebSockInfo * wsock = (tWebSockInfo*)rq->conn_data;
|
||||
struct mg_context * ctx = mg_get_context(conn);
|
||||
tWebSockContext *ws_ctx = (tWebSockContext*) mg_get_user_data(ctx);
|
||||
char msg[128];
|
||||
|
||||
mg_lock_context(ctx);
|
||||
if (flags==136) {
|
||||
// close websock
|
||||
websocket_done(ws_ctx, wsock);
|
||||
rq->conn_data = 0;
|
||||
mg_unlock_context(ctx);
|
||||
return 1;
|
||||
}
|
||||
if (((data_len>=5) && (data_len<100) && (flags==129)) || (flags==130)) {
|
||||
|
||||
// init command
|
||||
if ((wsock->webSockState==1) && (!memcmp(data,"init ",5))) {
|
||||
char * chk;
|
||||
unsigned long gid;
|
||||
memcpy(msg,data+5,data_len-5);
|
||||
msg[data_len-5]=0;
|
||||
gid = strtoul(msg,&chk,10);
|
||||
wsock->initId = gid;
|
||||
if (gid>0 && chk!=NULL && *chk==0) {
|
||||
wsock->webSockState = 2;
|
||||
}
|
||||
mg_unlock_context(ctx);
|
||||
return 1;
|
||||
}
|
||||
|
||||
// chat message
|
||||
if ((wsock->webSockState==2) && (!memcmp(data,"msg ",4))) {
|
||||
send_to_all_websockets(ctx, data, data_len);
|
||||
mg_unlock_context(ctx);
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
|
||||
// keep alive
|
||||
if ((data_len==4) && !memcmp(data,"ping",4)) {
|
||||
mg_unlock_context(ctx);
|
||||
return 1;
|
||||
}
|
||||
|
||||
mg_unlock_context(ctx);
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
void connection_close_handler(struct mg_connection *conn) {
|
||||
|
||||
struct mg_request_info * rq = mg_get_request_info(conn);
|
||||
tWebSockInfo * wsock = (tWebSockInfo*)rq->conn_data;
|
||||
struct mg_context * ctx = mg_get_context(conn);
|
||||
tWebSockContext *ws_ctx = (tWebSockContext*) mg_get_user_data(ctx);
|
||||
|
||||
mg_lock_context(ctx);
|
||||
websocket_done(ws_ctx, wsock);
|
||||
rq->conn_data = 0;
|
||||
mg_unlock_context(ctx);
|
||||
}
|
||||
|
||||
|
||||
static void * eventMain(void * arg) {
|
||||
|
||||
char msg[256];
|
||||
struct mg_context *ctx = (struct mg_context *)arg;
|
||||
tWebSockContext *ws_ctx = (tWebSockContext*) mg_get_user_data(ctx);
|
||||
|
||||
ws_ctx->runLoop = 1;
|
||||
while (ws_ctx->runLoop) {
|
||||
time_t t = time(0);
|
||||
struct tm * timestr = localtime(&t);
|
||||
sprintf(msg,"title %s",asctime(timestr));
|
||||
|
||||
send_to_all_websockets(ctx, msg, strlen(msg));
|
||||
|
||||
mg_sleep(1000);
|
||||
}
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
void websock_send_broadcast(struct mg_context *ctx, const char * data, int data_len) {
|
||||
|
||||
char buffer[260];
|
||||
|
||||
if (data_len<=256) {
|
||||
strcpy(buffer, "msg ");
|
||||
memcpy(buffer+4, data, data_len);
|
||||
|
||||
send_to_all_websockets(ctx, buffer, data_len+4);
|
||||
}
|
||||
}
|
||||
|
||||
void websock_init_lib(struct mg_context *ctx) {
|
||||
|
||||
tWebSockContext *ws_ctx = (tWebSockContext*) mg_get_user_data(ctx);
|
||||
memset(ws_ctx,0,sizeof(*ws_ctx));
|
||||
/* todo: use mg_start_thread_id instead of mg_start_thread */
|
||||
mg_start_thread(eventMain, ctx);
|
||||
}
|
||||
|
||||
void websock_exit_lib(struct mg_context *ctx) {
|
||||
|
||||
tWebSockContext *ws_ctx = (tWebSockContext*) mg_get_user_data(ctx);
|
||||
ws_ctx->runLoop = 0;
|
||||
/* todo: wait for the thread instead of a timeout */
|
||||
mg_sleep(2000);
|
||||
}
|
||||
#include <assert.h>
|
||||
#include <stdlib.h>
|
||||
#include <time.h>
|
||||
#include <string.h>
|
||||
#include "WebSockCallbacks.h"
|
||||
|
||||
#ifdef _WIN32
|
||||
#include <Windows.h>
|
||||
#define mg_sleep(x) Sleep(x)
|
||||
#else
|
||||
#include <unistd.h>
|
||||
#include <pthread.h>
|
||||
#define mg_sleep(x) usleep((x) * 1000)
|
||||
#endif
|
||||
|
||||
|
||||
static void send_to_all_websockets(struct mg_context *ctx, const char * data, int data_len) {
|
||||
|
||||
int i;
|
||||
tWebSockContext *ws_ctx = (tWebSockContext*) mg_get_user_data(ctx);
|
||||
|
||||
mg_lock_context(ctx);
|
||||
for (i=0;i<MAX_NUM_OF_WEBSOCKS;i++) {
|
||||
if (ws_ctx->socketList[i] && (ws_ctx->socketList[i]->webSockState==2)) {
|
||||
mg_websocket_write(ws_ctx->socketList[i]->conn, WEBSOCKET_OPCODE_TEXT, data, data_len);
|
||||
}
|
||||
}
|
||||
mg_unlock_context(ctx);
|
||||
}
|
||||
|
||||
|
||||
void websocket_ready_handler(struct mg_connection *conn) {
|
||||
|
||||
int i;
|
||||
struct mg_request_info * rq = mg_get_request_info(conn);
|
||||
struct mg_context * ctx = mg_get_context(conn);
|
||||
tWebSockContext *ws_ctx = (tWebSockContext*) mg_get_user_data(ctx);
|
||||
tWebSockInfo * wsock = malloc(sizeof(tWebSockInfo));
|
||||
assert(wsock);
|
||||
wsock->webSockState = 0;
|
||||
rq->conn_data = wsock;
|
||||
|
||||
mg_lock_context(ctx);
|
||||
for (i=0;i<MAX_NUM_OF_WEBSOCKS;i++) {
|
||||
if (0==ws_ctx->socketList[i]) {
|
||||
ws_ctx->socketList[i] = wsock;
|
||||
wsock->conn = conn;
|
||||
wsock->webSockState = 1;
|
||||
break;
|
||||
}
|
||||
}
|
||||
printf("\nNew websocket attached: %s:%u\n", rq->remote_addr, rq->remote_port);
|
||||
mg_unlock_context(ctx);
|
||||
}
|
||||
|
||||
|
||||
static void websocket_done(tWebSockContext *ws_ctx, tWebSockInfo * wsock) {
|
||||
|
||||
int i;
|
||||
|
||||
if (wsock) {
|
||||
wsock->webSockState = 99;
|
||||
for (i=0;i<MAX_NUM_OF_WEBSOCKS;i++) {
|
||||
if (wsock==ws_ctx->socketList[i]) {
|
||||
ws_ctx->socketList[i] = 0;
|
||||
break;
|
||||
}
|
||||
}
|
||||
printf("\nClose websocket attached: %s:%u\n", mg_get_request_info(wsock->conn)->remote_addr, mg_get_request_info(wsock->conn)->remote_port);
|
||||
free(wsock);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
int websocket_data_handler(struct mg_connection *conn, int flags, char *data, size_t data_len) {
|
||||
|
||||
struct mg_request_info * rq = mg_get_request_info(conn);
|
||||
tWebSockInfo * wsock = (tWebSockInfo*)rq->conn_data;
|
||||
struct mg_context * ctx = mg_get_context(conn);
|
||||
tWebSockContext *ws_ctx = (tWebSockContext*) mg_get_user_data(ctx);
|
||||
char msg[128];
|
||||
|
||||
mg_lock_context(ctx);
|
||||
if (flags==136) {
|
||||
// close websock
|
||||
websocket_done(ws_ctx, wsock);
|
||||
rq->conn_data = 0;
|
||||
mg_unlock_context(ctx);
|
||||
return 1;
|
||||
}
|
||||
if (((data_len>=5) && (data_len<100) && (flags==129)) || (flags==130)) {
|
||||
|
||||
// init command
|
||||
if ((wsock->webSockState==1) && (!memcmp(data,"init ",5))) {
|
||||
char * chk;
|
||||
unsigned long gid;
|
||||
memcpy(msg,data+5,data_len-5);
|
||||
msg[data_len-5]=0;
|
||||
gid = strtoul(msg,&chk,10);
|
||||
wsock->initId = gid;
|
||||
if (gid>0 && chk!=NULL && *chk==0) {
|
||||
wsock->webSockState = 2;
|
||||
}
|
||||
mg_unlock_context(ctx);
|
||||
return 1;
|
||||
}
|
||||
|
||||
// chat message
|
||||
if ((wsock->webSockState==2) && (!memcmp(data,"msg ",4))) {
|
||||
send_to_all_websockets(ctx, data, data_len);
|
||||
mg_unlock_context(ctx);
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
|
||||
// keep alive
|
||||
if ((data_len==4) && !memcmp(data,"ping",4)) {
|
||||
mg_unlock_context(ctx);
|
||||
return 1;
|
||||
}
|
||||
|
||||
mg_unlock_context(ctx);
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
void connection_close_handler(struct mg_connection *conn) {
|
||||
|
||||
struct mg_request_info * rq = mg_get_request_info(conn);
|
||||
tWebSockInfo * wsock = (tWebSockInfo*)rq->conn_data;
|
||||
struct mg_context * ctx = mg_get_context(conn);
|
||||
tWebSockContext *ws_ctx = (tWebSockContext*) mg_get_user_data(ctx);
|
||||
|
||||
mg_lock_context(ctx);
|
||||
websocket_done(ws_ctx, wsock);
|
||||
rq->conn_data = 0;
|
||||
mg_unlock_context(ctx);
|
||||
}
|
||||
|
||||
|
||||
static void * eventMain(void * arg) {
|
||||
|
||||
char msg[256];
|
||||
struct mg_context *ctx = (struct mg_context *)arg;
|
||||
tWebSockContext *ws_ctx = (tWebSockContext*) mg_get_user_data(ctx);
|
||||
|
||||
ws_ctx->runLoop = 1;
|
||||
while (ws_ctx->runLoop) {
|
||||
time_t t = time(0);
|
||||
struct tm * timestr = localtime(&t);
|
||||
sprintf(msg,"title %s",asctime(timestr));
|
||||
|
||||
send_to_all_websockets(ctx, msg, strlen(msg));
|
||||
|
||||
mg_sleep(1000);
|
||||
}
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
void websock_send_broadcast(struct mg_context *ctx, const char * data, int data_len) {
|
||||
|
||||
char buffer[260];
|
||||
|
||||
if (data_len<=256) {
|
||||
strcpy(buffer, "msg ");
|
||||
memcpy(buffer+4, data, data_len);
|
||||
|
||||
send_to_all_websockets(ctx, buffer, data_len+4);
|
||||
}
|
||||
}
|
||||
|
||||
void websock_init_lib(struct mg_context *ctx) {
|
||||
|
||||
tWebSockContext *ws_ctx = (tWebSockContext*) mg_get_user_data(ctx);
|
||||
memset(ws_ctx,0,sizeof(*ws_ctx));
|
||||
/* todo: use mg_start_thread_id instead of mg_start_thread */
|
||||
mg_start_thread(eventMain, ctx);
|
||||
}
|
||||
|
||||
void websock_exit_lib(struct mg_context *ctx) {
|
||||
|
||||
tWebSockContext *ws_ctx = (tWebSockContext*) mg_get_user_data(ctx);
|
||||
ws_ctx->runLoop = 0;
|
||||
/* todo: wait for the thread instead of a timeout */
|
||||
mg_sleep(2000);
|
||||
}
|
||||
|
@ -1,39 +1,39 @@
|
||||
|
||||
#ifndef WEBSOCKCALLBACKS_H_INCLUDED
|
||||
#define WEBSOCKCALLBACKS_H_INCLUDED
|
||||
|
||||
#include "civetweb.h"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
typedef struct tWebSockInfo {
|
||||
int webSockState;
|
||||
unsigned long initId;
|
||||
struct mg_connection *conn;
|
||||
} tWebSockInfo;
|
||||
|
||||
#define MAX_NUM_OF_WEBSOCKS (256)
|
||||
typedef struct tWebSockContext {
|
||||
int runLoop;
|
||||
void * thread_id;
|
||||
tWebSockInfo *socketList[MAX_NUM_OF_WEBSOCKS];
|
||||
} tWebSockContext;
|
||||
|
||||
|
||||
void websock_init_lib(struct mg_context *ctx);
|
||||
void websock_exit_lib(struct mg_context *ctx);
|
||||
|
||||
void websock_send_broadcast(struct mg_context *ctx, const char * data, int data_len);
|
||||
|
||||
void websocket_ready_handler(struct mg_connection *conn);
|
||||
int websocket_data_handler(struct mg_connection *conn, int flags, char *data, size_t data_len);
|
||||
void connection_close_handler(struct mg_connection *conn);
|
||||
|
||||
|
||||
#ifdef __cplusplus
|
||||
}
|
||||
#endif
|
||||
|
||||
|
||||
#ifndef WEBSOCKCALLBACKS_H_INCLUDED
|
||||
#define WEBSOCKCALLBACKS_H_INCLUDED
|
||||
|
||||
#include "civetweb.h"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
typedef struct tWebSockInfo {
|
||||
int webSockState;
|
||||
unsigned long initId;
|
||||
struct mg_connection *conn;
|
||||
} tWebSockInfo;
|
||||
|
||||
#define MAX_NUM_OF_WEBSOCKS (256)
|
||||
typedef struct tWebSockContext {
|
||||
int runLoop;
|
||||
void * thread_id;
|
||||
tWebSockInfo *socketList[MAX_NUM_OF_WEBSOCKS];
|
||||
} tWebSockContext;
|
||||
|
||||
|
||||
void websock_init_lib(struct mg_context *ctx);
|
||||
void websock_exit_lib(struct mg_context *ctx);
|
||||
|
||||
void websock_send_broadcast(struct mg_context *ctx, const char * data, int data_len);
|
||||
|
||||
void websocket_ready_handler(struct mg_connection *conn);
|
||||
int websocket_data_handler(struct mg_connection *conn, int flags, char *data, size_t data_len);
|
||||
void connection_close_handler(struct mg_connection *conn);
|
||||
|
||||
|
||||
#ifdef __cplusplus
|
||||
}
|
||||
#endif
|
||||
|
||||
#endif
|
34
src/main.c
34
src/main.c
@ -1510,25 +1510,25 @@ static int MakeConsole() {
|
||||
if (!ok) {
|
||||
if (!AttachConsole(ATTACH_PARENT_PROCESS)) {
|
||||
FreeConsole();
|
||||
if (!AllocConsole()) {
|
||||
err = GetLastError();
|
||||
if (err==ERROR_ACCESS_DENIED) {
|
||||
MessageBox(NULL, "Insufficient rights to create a console window", "Error", MB_ICONERROR);
|
||||
}
|
||||
}
|
||||
AttachConsole(GetCurrentProcessId());
|
||||
}
|
||||
if (!AllocConsole()) {
|
||||
err = GetLastError();
|
||||
if (err==ERROR_ACCESS_DENIED) {
|
||||
MessageBox(NULL, "Insufficient rights to create a console window", "Error", MB_ICONERROR);
|
||||
}
|
||||
}
|
||||
AttachConsole(GetCurrentProcessId());
|
||||
}
|
||||
|
||||
ok = (GetConsoleWindow() != NULL);
|
||||
if (ok) {
|
||||
freopen("CONIN$", "r", stdin);
|
||||
freopen("CONOUT$", "w", stdout);
|
||||
freopen("CONOUT$", "w", stderr);
|
||||
}
|
||||
}
|
||||
|
||||
if (ok) {
|
||||
SetConsoleTitle(server_name);
|
||||
if (ok) {
|
||||
freopen("CONIN$", "r", stdin);
|
||||
freopen("CONOUT$", "w", stdout);
|
||||
freopen("CONOUT$", "w", stderr);
|
||||
}
|
||||
}
|
||||
|
||||
if (ok) {
|
||||
SetConsoleTitle(server_name);
|
||||
}
|
||||
|
||||
return ok;
|
||||
|
1790
src/third_party/lfs.c
vendored
1790
src/third_party/lfs.c
vendored
File diff suppressed because it is too large
Load Diff
34
src/third_party/lfs.h
vendored
34
src/third_party/lfs.h
vendored
@ -1,17 +1,17 @@
|
||||
/*
|
||||
** LuaFileSystem
|
||||
** Copyright Kepler Project 2003 (http://www.keplerproject.org/luafilesystem)
|
||||
**
|
||||
** $Id: lfs.h,v 1.5 2008/02/19 20:08:23 mascarenhas Exp $
|
||||
*/
|
||||
|
||||
/* Define 'chdir' for systems that do not implement it */
|
||||
#ifdef NO_CHDIR
|
||||
#define chdir(p) (-1)
|
||||
#define chdir_error "Function 'chdir' not provided by system"
|
||||
#else
|
||||
#define chdir_error strerror(errno)
|
||||
#endif
|
||||
|
||||
|
||||
int luaopen_lfs (lua_State *L);
|
||||
/*
|
||||
** LuaFileSystem
|
||||
** Copyright Kepler Project 2003 (http://www.keplerproject.org/luafilesystem)
|
||||
**
|
||||
** $Id: lfs.h,v 1.5 2008/02/19 20:08:23 mascarenhas Exp $
|
||||
*/
|
||||
|
||||
/* Define 'chdir' for systems that do not implement it */
|
||||
#ifdef NO_CHDIR
|
||||
#define chdir(p) (-1)
|
||||
#define chdir_error "Function 'chdir' not provided by system"
|
||||
#else
|
||||
#define chdir_error strerror(errno)
|
||||
#endif
|
||||
|
||||
|
||||
int luaopen_lfs (lua_State *L);
|
||||
|
364
test/embed.c
364
test/embed.c
@ -1,182 +1,182 @@
|
||||
// Copyright (c) 2004-2009 Sergey Lyubka
|
||||
//
|
||||
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
// of this software and associated documentation files (the "Software"), to deal
|
||||
// in the Software without restriction, including without limitation the rights
|
||||
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
// copies of the Software, and to permit persons to whom the Software is
|
||||
// furnished to do so, subject to the following conditions:
|
||||
//
|
||||
// The above copyright notice and this permission notice shall be included in
|
||||
// all copies or substantial portions of the Software.
|
||||
//
|
||||
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||
// THE SOFTWARE.
|
||||
//
|
||||
// Unit test for the civetweb web server. Tests embedded API.
|
||||
|
||||
|
||||
#include <stdlib.h>
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
#ifndef _WIN32
|
||||
#include <unistd.h>
|
||||
#endif
|
||||
|
||||
#include "civetweb.h"
|
||||
|
||||
#if !defined(LISTENING_PORT)
|
||||
#define LISTENING_PORT "23456"
|
||||
#endif
|
||||
|
||||
static const char *standard_reply = "HTTP/1.1 200 OK\r\n"
|
||||
"Content-Type: text/plain\r\n"
|
||||
"Connection: close\r\n\r\n";
|
||||
|
||||
static void test_get_var(struct mg_connection *conn,
|
||||
const struct mg_request_info *ri) {
|
||||
char *var, *buf;
|
||||
size_t buf_len;
|
||||
const char *cl;
|
||||
int var_len;
|
||||
|
||||
mg_printf(conn, "%s", standard_reply);
|
||||
|
||||
buf_len = 0;
|
||||
var = buf = NULL;
|
||||
cl = mg_get_header(conn, "Content-Length");
|
||||
mg_printf(conn, "cl: %p\n", cl);
|
||||
if ((!strcmp(ri->request_method, "POST") ||
|
||||
!strcmp(ri->request_method, "PUT"))
|
||||
&& cl != NULL) {
|
||||
buf_len = atoi(cl);
|
||||
buf = malloc(buf_len);
|
||||
/* Read in two pieces, to test continuation */
|
||||
if (buf_len > 2) {
|
||||
mg_read(conn, buf, 2);
|
||||
mg_read(conn, buf + 2, buf_len - 2);
|
||||
} else {
|
||||
mg_read(conn, buf, buf_len);
|
||||
}
|
||||
} else if (ri->query_string != NULL) {
|
||||
buf_len = strlen(ri->query_string);
|
||||
buf = malloc(buf_len + 1);
|
||||
strcpy(buf, ri->query_string);
|
||||
}
|
||||
var = malloc(buf_len + 1);
|
||||
var_len = mg_get_var(buf, buf_len, "my_var", var, buf_len + 1);
|
||||
mg_printf(conn, "Value: [%s]\n", var);
|
||||
mg_printf(conn, "Value size: [%d]\n", var_len);
|
||||
free(buf);
|
||||
free(var);
|
||||
}
|
||||
|
||||
static void test_get_header(struct mg_connection *conn,
|
||||
const struct mg_request_info *ri) {
|
||||
const char *value;
|
||||
int i;
|
||||
|
||||
mg_printf(conn, "%s", standard_reply);
|
||||
printf("HTTP headers: %d\n", ri->num_headers);
|
||||
for (i = 0; i < ri->num_headers; i++) {
|
||||
printf("[%s]: [%s]\n", ri->http_headers[i].name, ri->http_headers[i].value);
|
||||
}
|
||||
|
||||
value = mg_get_header(conn, "Host");
|
||||
if (value != NULL) {
|
||||
mg_printf(conn, "Value: [%s]", value);
|
||||
}
|
||||
}
|
||||
|
||||
static void test_get_request_info(struct mg_connection *conn,
|
||||
const struct mg_request_info *ri) {
|
||||
int i;
|
||||
|
||||
mg_printf(conn, "%s", standard_reply);
|
||||
|
||||
mg_printf(conn, "Method: [%s]\n", ri->request_method);
|
||||
mg_printf(conn, "URI: [%s]\n", ri->uri);
|
||||
mg_printf(conn, "HTTP version: [%s]\n", ri->http_version);
|
||||
|
||||
for (i = 0; i < ri->num_headers; i++) {
|
||||
mg_printf(conn, "HTTP header [%s]: [%s]\n",
|
||||
ri->http_headers[i].name,
|
||||
ri->http_headers[i].value);
|
||||
}
|
||||
|
||||
mg_printf(conn, "Query string: [%s]\n",
|
||||
ri->query_string ? ri->query_string: "");
|
||||
mg_printf(conn, "Remote IP: [%lu]\n", ri->remote_ip);
|
||||
mg_printf(conn, "Remote port: [%d]\n", ri->remote_port);
|
||||
mg_printf(conn, "Remote user: [%s]\n",
|
||||
ri->remote_user ? ri->remote_user : "");
|
||||
}
|
||||
|
||||
static void test_error(struct mg_connection *conn,
|
||||
const struct mg_request_info *ri) {
|
||||
int status = (int) ri->ev_data;
|
||||
mg_printf(conn, "HTTP/1.1 %d XX\r\n"
|
||||
"Conntection: close\r\n\r\n", status);
|
||||
mg_printf(conn, "Error: [%d]", status);
|
||||
}
|
||||
|
||||
static void test_post(struct mg_connection *conn,
|
||||
const struct mg_request_info *ri) {
|
||||
const char *cl;
|
||||
char *buf;
|
||||
int len;
|
||||
|
||||
mg_printf(conn, "%s", standard_reply);
|
||||
if (strcmp(ri->request_method, "POST") == 0 &&
|
||||
(cl = mg_get_header(conn, "Content-Length")) != NULL) {
|
||||
len = atoi(cl);
|
||||
if ((buf = malloc(len)) != NULL) {
|
||||
mg_write(conn, buf, len);
|
||||
free(buf);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static const struct test_config {
|
||||
enum mg_event event;
|
||||
const char *uri;
|
||||
void (*func)(struct mg_connection *, const struct mg_request_info *);
|
||||
} test_config[] = {
|
||||
{MG_NEW_REQUEST, "/test_get_header", &test_get_header},
|
||||
{MG_NEW_REQUEST, "/test_get_var", &test_get_var},
|
||||
{MG_NEW_REQUEST, "/test_get_request_info", &test_get_request_info},
|
||||
{MG_NEW_REQUEST, "/test_post", &test_post},
|
||||
{MG_HTTP_ERROR, "", &test_error},
|
||||
{0, NULL, NULL}
|
||||
};
|
||||
|
||||
static void *callback(enum mg_event event,
|
||||
struct mg_connection *conn) {
|
||||
const struct mg_request_info *request_info = mg_get_request_info(conn);
|
||||
int i;
|
||||
|
||||
for (i = 0; test_config[i].uri != NULL; i++) {
|
||||
if (event == test_config[i].event &&
|
||||
(event == MG_HTTP_ERROR ||
|
||||
!strcmp(request_info->uri, test_config[i].uri))) {
|
||||
test_config[i].func(conn, request_info);
|
||||
return "processed";
|
||||
}
|
||||
}
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
int main(void) {
|
||||
struct mg_context *ctx;
|
||||
const char *options[] = {"listening_ports", LISTENING_PORT, NULL};
|
||||
|
||||
ctx = mg_start(callback, NULL, options);
|
||||
pause();
|
||||
return 0;
|
||||
}
|
||||
// Copyright (c) 2004-2009 Sergey Lyubka
|
||||
//
|
||||
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
// of this software and associated documentation files (the "Software"), to deal
|
||||
// in the Software without restriction, including without limitation the rights
|
||||
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
// copies of the Software, and to permit persons to whom the Software is
|
||||
// furnished to do so, subject to the following conditions:
|
||||
//
|
||||
// The above copyright notice and this permission notice shall be included in
|
||||
// all copies or substantial portions of the Software.
|
||||
//
|
||||
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||
// THE SOFTWARE.
|
||||
//
|
||||
// Unit test for the civetweb web server. Tests embedded API.
|
||||
|
||||
|
||||
#include <stdlib.h>
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
#ifndef _WIN32
|
||||
#include <unistd.h>
|
||||
#endif
|
||||
|
||||
#include "civetweb.h"
|
||||
|
||||
#if !defined(LISTENING_PORT)
|
||||
#define LISTENING_PORT "23456"
|
||||
#endif
|
||||
|
||||
static const char *standard_reply = "HTTP/1.1 200 OK\r\n"
|
||||
"Content-Type: text/plain\r\n"
|
||||
"Connection: close\r\n\r\n";
|
||||
|
||||
static void test_get_var(struct mg_connection *conn,
|
||||
const struct mg_request_info *ri) {
|
||||
char *var, *buf;
|
||||
size_t buf_len;
|
||||
const char *cl;
|
||||
int var_len;
|
||||
|
||||
mg_printf(conn, "%s", standard_reply);
|
||||
|
||||
buf_len = 0;
|
||||
var = buf = NULL;
|
||||
cl = mg_get_header(conn, "Content-Length");
|
||||
mg_printf(conn, "cl: %p\n", cl);
|
||||
if ((!strcmp(ri->request_method, "POST") ||
|
||||
!strcmp(ri->request_method, "PUT"))
|
||||
&& cl != NULL) {
|
||||
buf_len = atoi(cl);
|
||||
buf = malloc(buf_len);
|
||||
/* Read in two pieces, to test continuation */
|
||||
if (buf_len > 2) {
|
||||
mg_read(conn, buf, 2);
|
||||
mg_read(conn, buf + 2, buf_len - 2);
|
||||
} else {
|
||||
mg_read(conn, buf, buf_len);
|
||||
}
|
||||
} else if (ri->query_string != NULL) {
|
||||
buf_len = strlen(ri->query_string);
|
||||
buf = malloc(buf_len + 1);
|
||||
strcpy(buf, ri->query_string);
|
||||
}
|
||||
var = malloc(buf_len + 1);
|
||||
var_len = mg_get_var(buf, buf_len, "my_var", var, buf_len + 1);
|
||||
mg_printf(conn, "Value: [%s]\n", var);
|
||||
mg_printf(conn, "Value size: [%d]\n", var_len);
|
||||
free(buf);
|
||||
free(var);
|
||||
}
|
||||
|
||||
static void test_get_header(struct mg_connection *conn,
|
||||
const struct mg_request_info *ri) {
|
||||
const char *value;
|
||||
int i;
|
||||
|
||||
mg_printf(conn, "%s", standard_reply);
|
||||
printf("HTTP headers: %d\n", ri->num_headers);
|
||||
for (i = 0; i < ri->num_headers; i++) {
|
||||
printf("[%s]: [%s]\n", ri->http_headers[i].name, ri->http_headers[i].value);
|
||||
}
|
||||
|
||||
value = mg_get_header(conn, "Host");
|
||||
if (value != NULL) {
|
||||
mg_printf(conn, "Value: [%s]", value);
|
||||
}
|
||||
}
|
||||
|
||||
static void test_get_request_info(struct mg_connection *conn,
|
||||
const struct mg_request_info *ri) {
|
||||
int i;
|
||||
|
||||
mg_printf(conn, "%s", standard_reply);
|
||||
|
||||
mg_printf(conn, "Method: [%s]\n", ri->request_method);
|
||||
mg_printf(conn, "URI: [%s]\n", ri->uri);
|
||||
mg_printf(conn, "HTTP version: [%s]\n", ri->http_version);
|
||||
|
||||
for (i = 0; i < ri->num_headers; i++) {
|
||||
mg_printf(conn, "HTTP header [%s]: [%s]\n",
|
||||
ri->http_headers[i].name,
|
||||
ri->http_headers[i].value);
|
||||
}
|
||||
|
||||
mg_printf(conn, "Query string: [%s]\n",
|
||||
ri->query_string ? ri->query_string: "");
|
||||
mg_printf(conn, "Remote IP: [%lu]\n", ri->remote_ip);
|
||||
mg_printf(conn, "Remote port: [%d]\n", ri->remote_port);
|
||||
mg_printf(conn, "Remote user: [%s]\n",
|
||||
ri->remote_user ? ri->remote_user : "");
|
||||
}
|
||||
|
||||
static void test_error(struct mg_connection *conn,
|
||||
const struct mg_request_info *ri) {
|
||||
int status = (int) ri->ev_data;
|
||||
mg_printf(conn, "HTTP/1.1 %d XX\r\n"
|
||||
"Conntection: close\r\n\r\n", status);
|
||||
mg_printf(conn, "Error: [%d]", status);
|
||||
}
|
||||
|
||||
static void test_post(struct mg_connection *conn,
|
||||
const struct mg_request_info *ri) {
|
||||
const char *cl;
|
||||
char *buf;
|
||||
int len;
|
||||
|
||||
mg_printf(conn, "%s", standard_reply);
|
||||
if (strcmp(ri->request_method, "POST") == 0 &&
|
||||
(cl = mg_get_header(conn, "Content-Length")) != NULL) {
|
||||
len = atoi(cl);
|
||||
if ((buf = malloc(len)) != NULL) {
|
||||
mg_write(conn, buf, len);
|
||||
free(buf);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static const struct test_config {
|
||||
enum mg_event event;
|
||||
const char *uri;
|
||||
void (*func)(struct mg_connection *, const struct mg_request_info *);
|
||||
} test_config[] = {
|
||||
{MG_NEW_REQUEST, "/test_get_header", &test_get_header},
|
||||
{MG_NEW_REQUEST, "/test_get_var", &test_get_var},
|
||||
{MG_NEW_REQUEST, "/test_get_request_info", &test_get_request_info},
|
||||
{MG_NEW_REQUEST, "/test_post", &test_post},
|
||||
{MG_HTTP_ERROR, "", &test_error},
|
||||
{0, NULL, NULL}
|
||||
};
|
||||
|
||||
static void *callback(enum mg_event event,
|
||||
struct mg_connection *conn) {
|
||||
const struct mg_request_info *request_info = mg_get_request_info(conn);
|
||||
int i;
|
||||
|
||||
for (i = 0; test_config[i].uri != NULL; i++) {
|
||||
if (event == test_config[i].event &&
|
||||
(event == MG_HTTP_ERROR ||
|
||||
!strcmp(request_info->uri, test_config[i].uri))) {
|
||||
test_config[i].func(conn, request_info);
|
||||
return "processed";
|
||||
}
|
||||
}
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
int main(void) {
|
||||
struct mg_context *ctx;
|
||||
const char *options[] = {"listening_ports", LISTENING_PORT, NULL};
|
||||
|
||||
ctx = mg_start(callback, NULL, options);
|
||||
pause();
|
||||
return 0;
|
||||
}
|
||||
|
Reference in New Issue
Block a user