1
0
mirror of https://github.com/lammertb/libhttp.git synced 2025-07-29 21:01:13 +03:00

Update websocket example - use new API functions

This commit is contained in:
bel
2014-08-10 23:49:49 +02:00
parent 64c97d56e4
commit 56389ad023
5 changed files with 49 additions and 65 deletions

View File

@ -86,7 +86,7 @@
</PrecompiledHeader> </PrecompiledHeader>
<WarningLevel>Level3</WarningLevel> <WarningLevel>Level3</WarningLevel>
<Optimization>Disabled</Optimization> <Optimization>Disabled</Optimization>
<PreprocessorDefinitions>USE_WEBSOCKET;WIN32;_DEBUG;_CONSOLE;%(PreprocessorDefinitions)</PreprocessorDefinitions> <PreprocessorDefinitions>USE_WEBSOCKET;WIN32;_DEBUG;_CONSOLE;_CRT_SECURE_NO_WARNINGS;%(PreprocessorDefinitions)</PreprocessorDefinitions>
<AdditionalIncludeDirectories>$(ProjectDir)..\..\include;%(AdditionalIncludeDirectories)</AdditionalIncludeDirectories> <AdditionalIncludeDirectories>$(ProjectDir)..\..\include;%(AdditionalIncludeDirectories)</AdditionalIncludeDirectories>
</ClCompile> </ClCompile>
<Link> <Link>
@ -145,6 +145,7 @@
</Link> </Link>
</ItemDefinitionGroup> </ItemDefinitionGroup>
<ItemGroup> <ItemGroup>
<ClInclude Include="..\..\examples\websocket\WebSockCallbacks.h" />
<ClInclude Include="..\..\include\civetweb.h" /> <ClInclude Include="..\..\include\civetweb.h" />
</ItemGroup> </ItemGroup>
<ItemGroup> <ItemGroup>

View File

@ -18,6 +18,9 @@
<ClInclude Include="..\..\include\civetweb.h"> <ClInclude Include="..\..\include\civetweb.h">
<Filter>Header Files</Filter> <Filter>Header Files</Filter>
</ClInclude> </ClInclude>
<ClInclude Include="..\..\examples\websocket\WebSockCallbacks.h">
<Filter>Header Files</Filter>
</ClInclude>
</ItemGroup> </ItemGroup>
<ItemGroup> <ItemGroup>
<ClCompile Include="..\..\examples\websocket\websocket.c"> <ClCompile Include="..\..\examples\websocket\websocket.c">
@ -26,5 +29,8 @@
<ClCompile Include="..\..\src\civetweb.c"> <ClCompile Include="..\..\src\civetweb.c">
<Filter>Source Files</Filter> <Filter>Source Files</Filter>
</ClCompile> </ClCompile>
<ClCompile Include="..\..\examples\websocket\WebSockCallbacks.c">
<Filter>Source Files</Filter>
</ClCompile>
</ItemGroup> </ItemGroup>
</Project> </Project>

View File

@ -1,32 +1,11 @@
#include <assert.h> #include <assert.h>
#include <stdlib.h> #include <stdlib.h>
#include <time.h> #include <time.h>
#include "WebSockCallbacks.h"
#ifdef __APPLE__
#include <string.h> #include <string.h>
#endif #include "WebSockCallbacks.h"
#ifdef _WIN32 #ifdef _WIN32
#include <Windows.h> #include <Windows.h>
typedef HANDLE pthread_mutex_t;
static int pthread_mutex_init(pthread_mutex_t *mutex, void *unused) {
unused = NULL;
*mutex = CreateMutex(NULL, FALSE, NULL);
return *mutex == NULL ? -1 : 0;
}
static int pthread_mutex_destroy(pthread_mutex_t *mutex) {
return CloseHandle(*mutex) == 0 ? -1 : 0;
}
static int pthread_mutex_lock(pthread_mutex_t *mutex) {
return WaitForSingleObject(*mutex, INFINITE) == WAIT_OBJECT_0? 0 : -1;
}
static int pthread_mutex_unlock(pthread_mutex_t *mutex) {
return ReleaseMutex(*mutex) == 0 ? -1 : 0;
}
#define mg_sleep(x) Sleep(x) #define mg_sleep(x) Sleep(x)
#else #else
#include <unistd.h> #include <unistd.h>
@ -41,21 +20,22 @@ typedef struct tWebSockInfo {
struct mg_connection *conn; struct mg_connection *conn;
} tWebSockInfo; } tWebSockInfo;
static pthread_mutex_t sMutex;
#define MAX_NUM_OF_WEBSOCKS (256) #define MAX_NUM_OF_WEBSOCKS (256)
static tWebSockInfo *socketList[MAX_NUM_OF_WEBSOCKS]; static tWebSockInfo *socketList[MAX_NUM_OF_WEBSOCKS];
static void send_to_all_websockets(const char * data, int data_len) { static void send_to_all_websockets(struct mg_context *ctx, const char * data, int data_len) {
int i; int i;
mg_lock_context(ctx);
for (i=0;i<MAX_NUM_OF_WEBSOCKS;i++) { for (i=0;i<MAX_NUM_OF_WEBSOCKS;i++) {
if (socketList[i] && (socketList[i]->webSockState==2)) { if (socketList[i] && (socketList[i]->webSockState==2)) {
mg_websocket_write(socketList[i]->conn, WEBSOCKET_OPCODE_TEXT, data, data_len); mg_websocket_write(socketList[i]->conn, WEBSOCKET_OPCODE_TEXT, data, data_len);
} }
} }
mg_unlock_context(ctx);
} }
@ -63,12 +43,13 @@ void websocket_ready_handler(struct mg_connection *conn) {
int i; int i;
struct mg_request_info * rq = mg_get_request_info(conn); struct mg_request_info * rq = mg_get_request_info(conn);
struct mg_context * ctx = mg_get_context(conn);
tWebSockInfo * wsock = malloc(sizeof(tWebSockInfo)); tWebSockInfo * wsock = malloc(sizeof(tWebSockInfo));
assert(wsock); assert(wsock);
wsock->webSockState = 0; wsock->webSockState = 0;
rq->conn_data = wsock; rq->conn_data = wsock;
pthread_mutex_lock(&sMutex); mg_lock_context(ctx);
for (i=0;i<MAX_NUM_OF_WEBSOCKS;i++) { for (i=0;i<MAX_NUM_OF_WEBSOCKS;i++) {
if (0==socketList[i]) { if (0==socketList[i]) {
socketList[i] = wsock; socketList[i] = wsock;
@ -78,11 +59,12 @@ void websocket_ready_handler(struct mg_connection *conn) {
} }
} }
printf("\nNew websocket attached: %08lx:%u\n", rq->remote_ip, rq->remote_port); printf("\nNew websocket attached: %08lx:%u\n", rq->remote_ip, rq->remote_port);
pthread_mutex_unlock(&sMutex); mg_unlock_context(ctx);
} }
static void websocket_done(tWebSockInfo * wsock) { static void websocket_done(tWebSockInfo * wsock) {
int i; int i;
if (wsock) { if (wsock) {
wsock->webSockState = 99; wsock->webSockState = 99;
@ -99,16 +81,18 @@ static void websocket_done(tWebSockInfo * wsock) {
int websocket_data_handler(struct mg_connection *conn, int flags, char *data, size_t data_len) { 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); struct mg_request_info * rq = mg_get_request_info(conn);
struct mg_context * ctx = mg_get_context(conn);
tWebSockInfo * wsock = (tWebSockInfo*)rq->conn_data; tWebSockInfo * wsock = (tWebSockInfo*)rq->conn_data;
char msg[128]; char msg[128];
pthread_mutex_lock(&sMutex); mg_lock_context(ctx);
if (flags==136) { if (flags==136) {
// close websock // close websock
websocket_done(wsock); websocket_done(wsock);
rq->conn_data = 0; rq->conn_data = 0;
pthread_mutex_unlock(&sMutex); mg_unlock_context(ctx);
return 1; return 1;
} }
if (((data_len>=5) && (data_len<100) && (flags==129)) || (flags==130)) { if (((data_len>=5) && (data_len<100) && (flags==129)) || (flags==130)) {
@ -124,45 +108,48 @@ int websocket_data_handler(struct mg_connection *conn, int flags, char *data, si
if (gid>0 && chk!=NULL && *chk==0) { if (gid>0 && chk!=NULL && *chk==0) {
wsock->webSockState = 2; wsock->webSockState = 2;
} }
pthread_mutex_unlock(&sMutex); mg_unlock_context(ctx);
return 1; return 1;
} }
// chat message // chat message
if ((wsock->webSockState==2) && (!memcmp(data,"msg ",4))) { if ((wsock->webSockState==2) && (!memcmp(data,"msg ",4))) {
send_to_all_websockets(data, data_len); send_to_all_websockets(ctx, data, data_len);
pthread_mutex_unlock(&sMutex); mg_unlock_context(ctx);
return 1; return 1;
} }
} }
// keep alive // keep alive
if ((data_len==4) && !memcmp(data,"ping",4)) { if ((data_len==4) && !memcmp(data,"ping",4)) {
pthread_mutex_unlock(&sMutex); mg_unlock_context(ctx);
return 1; return 1;
} }
pthread_mutex_unlock(&sMutex); mg_unlock_context(ctx);
return 0; return 0;
} }
void connection_close_handler(struct mg_connection *conn) { void connection_close_handler(struct mg_connection *conn) {
struct mg_request_info * rq = mg_get_request_info(conn); struct mg_request_info * rq = mg_get_request_info(conn);
struct mg_context * ctx = mg_get_context(conn);
tWebSockInfo * wsock = (tWebSockInfo*)rq->conn_data; tWebSockInfo * wsock = (tWebSockInfo*)rq->conn_data;
pthread_mutex_lock(&sMutex); mg_lock_context(ctx);
websocket_done(wsock); websocket_done(wsock);
rq->conn_data = 0; rq->conn_data = 0;
pthread_mutex_unlock(&sMutex); mg_unlock_context(ctx);
} }
static int runLoop = 0; static int runLoop = 0;
static void * eventMain(void * _ignored) { static void * eventMain(void * arg) {
int i;
char msg[256]; char msg[256];
struct mg_context *ctx = (struct mg_context *)arg;
runLoop = 1; runLoop = 1;
while (runLoop) { while (runLoop) {
@ -170,21 +157,15 @@ static void * eventMain(void * _ignored) {
struct tm * timestr = localtime(&t); struct tm * timestr = localtime(&t);
sprintf(msg,"title %s",asctime(timestr)); sprintf(msg,"title %s",asctime(timestr));
pthread_mutex_lock(&sMutex); send_to_all_websockets(ctx, msg, strlen(msg));
for (i=0;i<MAX_NUM_OF_WEBSOCKS;i++) {
if (socketList[i] && (socketList[i]->webSockState==2)) {
mg_websocket_write(socketList[i]->conn, WEBSOCKET_OPCODE_TEXT, msg, strlen(msg));
}
}
pthread_mutex_unlock(&sMutex);
mg_sleep(1000); mg_sleep(1000);
} }
return _ignored; return NULL;
} }
void websock_send_broadcast(const char * data, int data_len) { void websock_send_broadcast(struct mg_context *ctx, const char * data, int data_len) {
char buffer[260]; char buffer[260];
@ -192,24 +173,20 @@ void websock_send_broadcast(const char * data, int data_len) {
strcpy(buffer, "msg "); strcpy(buffer, "msg ");
memcpy(buffer+4, data, data_len); memcpy(buffer+4, data, data_len);
pthread_mutex_lock(&sMutex); send_to_all_websockets(ctx, buffer, data_len+4);
send_to_all_websockets(buffer, data_len+4);
pthread_mutex_unlock(&sMutex);
} }
} }
void websock_init_lib(void) { void websock_init_lib(struct mg_context *ctx) {
int ret;
ret = pthread_mutex_init(&sMutex, 0);
assert(ret==0);
/* todo: use variable in the ctx instead of static ones */
memset(socketList,0,sizeof(socketList)); memset(socketList,0,sizeof(socketList));
mg_start_thread(eventMain, ctx);
mg_start_thread(eventMain, 0);
} }
void websock_exit_lib(void) { void websock_exit_lib(struct mg_context *ctx) {
runLoop = 0; runLoop = 0;
/* todo: wait for the thread instead of a timeout */
mg_sleep(2000);
} }

View File

@ -8,10 +8,10 @@
extern "C" { extern "C" {
#endif #endif
void websock_init_lib(void); void websock_init_lib(struct mg_context *ctx);
void websock_exit_lib(void); void websock_exit_lib(struct mg_context *ctx);
void websock_send_broadcast(const char * data, int data_len); void websock_send_broadcast(struct mg_context *ctx, const char * data, int data_len);
void websocket_ready_handler(struct mg_connection *conn); void websocket_ready_handler(struct mg_connection *conn);
int websocket_data_handler(struct mg_connection *conn, int flags, char *data, size_t data_len); int websocket_data_handler(struct mg_connection *conn, int flags, char *data, size_t data_len);

View File

@ -22,12 +22,13 @@ int main(void)
NULL NULL
}; };
websock_init_lib(); callback_funcs.init_context = websock_init_lib;
callback_funcs.exit_context = websock_exit_lib;
callback_funcs.websocket_ready = websocket_ready_handler; callback_funcs.websocket_ready = websocket_ready_handler;
callback_funcs.websocket_data = websocket_data_handler; callback_funcs.websocket_data = websocket_data_handler;
callback_funcs.connection_close = connection_close_handler; callback_funcs.connection_close = connection_close_handler;
ctx = mg_start(&callback_funcs, NULL, server_options); ctx = mg_start(&callback_funcs, NULL, server_options);
printf("Connect to localhost:%s/websock.htm\n", mg_get_option(ctx, "listening_ports"));
puts("Enter an (ASCII) character or * to exit:"); puts("Enter an (ASCII) character or * to exit:");
for (;;) { for (;;) {
@ -37,11 +38,10 @@ int main(void)
break; break;
} }
inbuf[0] = toupper(inbuf[0]); inbuf[0] = toupper(inbuf[0]);
websock_send_broadcast(inbuf, 1); websock_send_broadcast(ctx, inbuf, 1);
} }
mg_stop(ctx); mg_stop(ctx);
websock_exit_lib();
return 0; return 0;
} }