mirror of
https://github.com/lammertb/libhttp.git
synced 2025-07-29 21:01:13 +03:00
Apply revised clang-format (#207) to all actively maintained source files
This commit is contained in:
bel
@ -8,7 +8,7 @@
|
||||
// Simple example program on how to use websocket client embedded C interface.
|
||||
#ifdef _WIN32
|
||||
#include <Windows.h>
|
||||
#define sleep(x) Sleep(1000*(x))
|
||||
#define sleep(x) Sleep(1000 * (x))
|
||||
#else
|
||||
#include <unistd.h>
|
||||
#endif
|
||||
@ -21,318 +21,393 @@
|
||||
#define PORT "8888"
|
||||
#define SSL_CERT "./ssl/server.pem"
|
||||
|
||||
const char * websocket_welcome_msg = "websocket welcome\n";
|
||||
const size_t websocket_welcome_msg_len = 18 /* strlen(websocket_welcome_msg) */ ;
|
||||
const char * websocket_acknowledge_msg = "websocket msg ok\n";
|
||||
const size_t websocket_acknowledge_msg_len = 17 /* strlen(websocket_acknowledge_msg) */ ;
|
||||
const char * websocket_goodbye_msg = "websocket bye\n";
|
||||
const size_t websocket_goodbye_msg_len = 14 /* strlen(websocket_goodbye_msg) */ ;
|
||||
const char *websocket_welcome_msg = "websocket welcome\n";
|
||||
const size_t websocket_welcome_msg_len = 18 /* strlen(websocket_welcome_msg) */;
|
||||
const char *websocket_acknowledge_msg = "websocket msg ok\n";
|
||||
const size_t websocket_acknowledge_msg_len =
|
||||
17 /* strlen(websocket_acknowledge_msg) */;
|
||||
const char *websocket_goodbye_msg = "websocket bye\n";
|
||||
const size_t websocket_goodbye_msg_len = 14 /* strlen(websocket_goodbye_msg) */;
|
||||
|
||||
|
||||
/*************************************************************************************/
|
||||
/* WEBSOCKET SERVER */
|
||||
/* WEBSOCKET SERVER */
|
||||
/*************************************************************************************/
|
||||
#if defined(MG_LEGACY_INTERFACE)
|
||||
int websock_server_connect(const struct mg_connection * conn)
|
||||
int
|
||||
websock_server_connect(const struct mg_connection *conn)
|
||||
#else
|
||||
int websocket_server_connect(const struct mg_connection * conn, void * _ignored)
|
||||
int
|
||||
websocket_server_connect(const struct mg_connection *conn, void *_ignored)
|
||||
#endif
|
||||
{
|
||||
printf("Server: Websocket connected\n");
|
||||
return 0; /* return 0 to accept every connection */
|
||||
printf("Server: Websocket connected\n");
|
||||
return 0; /* return 0 to accept every connection */
|
||||
}
|
||||
|
||||
|
||||
#if defined(MG_LEGACY_INTERFACE)
|
||||
void websocket_server_ready(struct mg_connection * conn)
|
||||
void
|
||||
websocket_server_ready(struct mg_connection *conn)
|
||||
#else
|
||||
void websocket_server_ready(struct mg_connection * conn, void * _ignored)
|
||||
void
|
||||
websocket_server_ready(struct mg_connection *conn, void *_ignored)
|
||||
#endif
|
||||
{
|
||||
printf("Server: Websocket ready\n");
|
||||
printf("Server: Websocket ready\n");
|
||||
|
||||
/* Send websocket welcome message */
|
||||
mg_lock_connection(conn);
|
||||
mg_websocket_write(conn, WEBSOCKET_OPCODE_TEXT, websocket_welcome_msg, websocket_welcome_msg_len);
|
||||
mg_unlock_connection(conn);
|
||||
/* Send websocket welcome message */
|
||||
mg_lock_connection(conn);
|
||||
mg_websocket_write(conn,
|
||||
WEBSOCKET_OPCODE_TEXT,
|
||||
websocket_welcome_msg,
|
||||
websocket_welcome_msg_len);
|
||||
mg_unlock_connection(conn);
|
||||
}
|
||||
|
||||
|
||||
#if defined(MG_LEGACY_INTERFACE)
|
||||
int websocket_server_data(struct mg_connection * conn, int bits, char *data, size_t data_len)
|
||||
int
|
||||
websocket_server_data(struct mg_connection *conn,
|
||||
int bits,
|
||||
char *data,
|
||||
size_t data_len)
|
||||
#else
|
||||
int websocket_server_data(struct mg_connection * conn, int bits, char *data, size_t data_len, void *_ignored)
|
||||
int
|
||||
websocket_server_data(struct mg_connection *conn,
|
||||
int bits,
|
||||
char *data,
|
||||
size_t data_len,
|
||||
void *_ignored)
|
||||
#endif
|
||||
{
|
||||
printf("Server: Got %u bytes from the client\n", data_len);
|
||||
printf("Server: Got %u bytes from the client\n", data_len);
|
||||
|
||||
if (data_len<3 || 0!=memcmp(data, "bye", 3)) {
|
||||
/* Send websocket acknowledge message */
|
||||
mg_lock_connection(conn);
|
||||
mg_websocket_write(conn, WEBSOCKET_OPCODE_TEXT, websocket_acknowledge_msg, websocket_acknowledge_msg_len);
|
||||
mg_unlock_connection(conn);
|
||||
} else {
|
||||
/* Send websocket acknowledge message */
|
||||
mg_lock_connection(conn);
|
||||
mg_websocket_write(conn, WEBSOCKET_OPCODE_TEXT, websocket_goodbye_msg, websocket_goodbye_msg_len);
|
||||
mg_unlock_connection(conn);
|
||||
}
|
||||
if (data_len < 3 || 0 != memcmp(data, "bye", 3)) {
|
||||
/* Send websocket acknowledge message */
|
||||
mg_lock_connection(conn);
|
||||
mg_websocket_write(conn,
|
||||
WEBSOCKET_OPCODE_TEXT,
|
||||
websocket_acknowledge_msg,
|
||||
websocket_acknowledge_msg_len);
|
||||
mg_unlock_connection(conn);
|
||||
} else {
|
||||
/* Send websocket acknowledge message */
|
||||
mg_lock_connection(conn);
|
||||
mg_websocket_write(conn,
|
||||
WEBSOCKET_OPCODE_TEXT,
|
||||
websocket_goodbye_msg,
|
||||
websocket_goodbye_msg_len);
|
||||
mg_unlock_connection(conn);
|
||||
}
|
||||
|
||||
return 1; /* return 1 to keep the connetion open */
|
||||
return 1; /* return 1 to keep the connetion open */
|
||||
}
|
||||
|
||||
|
||||
#if defined(MG_LEGACY_INTERFACE)
|
||||
void websocket_server_connection_close(const struct mg_connection * conn)
|
||||
void
|
||||
websocket_server_connection_close(const struct mg_connection *conn)
|
||||
#else
|
||||
void websocket_server_connection_close(const struct mg_connection * conn, void *_ignored)
|
||||
void
|
||||
websocket_server_connection_close(const struct mg_connection *conn,
|
||||
void *_ignored)
|
||||
#endif
|
||||
{
|
||||
printf("Server: Close connection\n");
|
||||
printf("Server: Close connection\n");
|
||||
|
||||
/* Can not send a websocket goodbye message here - the connection is already closed */
|
||||
/* Can not send a websocket goodbye message here - the connection is already
|
||||
* closed */
|
||||
}
|
||||
|
||||
|
||||
struct mg_context * start_websocket_server()
|
||||
struct mg_context *
|
||||
start_websocket_server()
|
||||
{
|
||||
const char * options[] = { "document_root", DOCUMENT_ROOT,
|
||||
"ssl_certificate", SSL_CERT,
|
||||
"listening_ports", PORT,
|
||||
"request_timeout_ms", "5000",
|
||||
0
|
||||
};
|
||||
struct mg_callbacks callbacks;
|
||||
struct mg_context *ctx;
|
||||
const char *options[] = {"document_root",
|
||||
DOCUMENT_ROOT,
|
||||
"ssl_certificate",
|
||||
SSL_CERT,
|
||||
"listening_ports",
|
||||
PORT,
|
||||
"request_timeout_ms",
|
||||
"5000",
|
||||
0};
|
||||
struct mg_callbacks callbacks;
|
||||
struct mg_context *ctx;
|
||||
|
||||
memset(&callbacks, 0, sizeof(callbacks));
|
||||
memset(&callbacks, 0, sizeof(callbacks));
|
||||
|
||||
#if defined(MG_LEGACY_INTERFACE)
|
||||
/* Obsolete: */
|
||||
callbacks.websocket_connect = websock_server_connect;
|
||||
callbacks.websocket_ready = websocket_server_ready;
|
||||
callbacks.websocket_data = websocket_server_data;
|
||||
callbacks.connection_close = websocket_server_connection_close;
|
||||
/* Obsolete: */
|
||||
callbacks.websocket_connect = websock_server_connect;
|
||||
callbacks.websocket_ready = websocket_server_ready;
|
||||
callbacks.websocket_data = websocket_server_data;
|
||||
callbacks.connection_close = websocket_server_connection_close;
|
||||
|
||||
ctx = mg_start(&callbacks, 0, options);
|
||||
ctx = mg_start(&callbacks, 0, options);
|
||||
#else
|
||||
/* New interface: */
|
||||
ctx = mg_start(&callbacks, 0, options);
|
||||
/* New interface: */
|
||||
ctx = mg_start(&callbacks, 0, options);
|
||||
|
||||
mg_set_websocket_handler(ctx, "/websocket",
|
||||
websocket_server_connect,
|
||||
websocket_server_ready,
|
||||
websocket_server_data,
|
||||
websocket_server_connection_close,
|
||||
NULL);
|
||||
mg_set_websocket_handler(ctx,
|
||||
"/websocket",
|
||||
websocket_server_connect,
|
||||
websocket_server_ready,
|
||||
websocket_server_data,
|
||||
websocket_server_connection_close,
|
||||
NULL);
|
||||
#endif
|
||||
|
||||
return ctx;
|
||||
return ctx;
|
||||
}
|
||||
|
||||
|
||||
/*************************************************************************************/
|
||||
/* WEBSOCKET CLIENT */
|
||||
/* WEBSOCKET CLIENT */
|
||||
/*************************************************************************************/
|
||||
struct tclient_data {
|
||||
void * data;
|
||||
size_t len;
|
||||
int closed;
|
||||
void *data;
|
||||
size_t len;
|
||||
int closed;
|
||||
};
|
||||
|
||||
static int websocket_client_data_handler(struct mg_connection *conn, int flags, char *data, size_t data_len, void * user_data)
|
||||
static int
|
||||
websocket_client_data_handler(struct mg_connection *conn,
|
||||
int flags,
|
||||
char *data,
|
||||
size_t data_len,
|
||||
void *user_data)
|
||||
{
|
||||
struct mg_context *ctx = mg_get_context(conn);
|
||||
struct tclient_data *pclient_data = (struct tclient_data *) mg_get_user_data(ctx);
|
||||
struct mg_context *ctx = mg_get_context(conn);
|
||||
struct tclient_data *pclient_data =
|
||||
(struct tclient_data *)mg_get_user_data(ctx);
|
||||
|
||||
printf("Client received data from server: ");
|
||||
fwrite(data, 1, data_len, stdout);
|
||||
printf("\n");
|
||||
printf("Client received data from server: ");
|
||||
fwrite(data, 1, data_len, stdout);
|
||||
printf("\n");
|
||||
|
||||
pclient_data->data = malloc(data_len);
|
||||
assert(pclient_data->data != NULL);
|
||||
memcpy(pclient_data->data, data, data_len);
|
||||
pclient_data->len = data_len;
|
||||
pclient_data->data = malloc(data_len);
|
||||
assert(pclient_data->data != NULL);
|
||||
memcpy(pclient_data->data, data, data_len);
|
||||
pclient_data->len = data_len;
|
||||
|
||||
return 1;
|
||||
return 1;
|
||||
}
|
||||
|
||||
static void websocket_client_close_handler(const struct mg_connection *conn, void * user_data)
|
||||
static void
|
||||
websocket_client_close_handler(const struct mg_connection *conn,
|
||||
void *user_data)
|
||||
{
|
||||
struct mg_context *ctx = mg_get_context(conn);
|
||||
struct tclient_data *pclient_data = (struct tclient_data *) mg_get_user_data(ctx);
|
||||
struct mg_context *ctx = mg_get_context(conn);
|
||||
struct tclient_data *pclient_data =
|
||||
(struct tclient_data *)mg_get_user_data(ctx);
|
||||
|
||||
printf("Client: Close handler\n");
|
||||
pclient_data->closed++;
|
||||
printf("Client: Close handler\n");
|
||||
pclient_data->closed++;
|
||||
}
|
||||
|
||||
|
||||
int main(int argc, char *argv[])
|
||||
int
|
||||
main(int argc, char *argv[])
|
||||
{
|
||||
struct mg_context *ctx = NULL;
|
||||
struct tclient_data client1_data = {NULL, 0, 0};
|
||||
struct tclient_data client2_data = {NULL, 0, 0};
|
||||
struct tclient_data client3_data = {NULL, 0, 0};
|
||||
struct mg_connection* newconn1 = NULL;
|
||||
struct mg_connection* newconn2 = NULL;
|
||||
struct mg_connection* newconn3 = NULL;
|
||||
char ebuf[100] = {0};
|
||||
struct mg_context *ctx = NULL;
|
||||
struct tclient_data client1_data = {NULL, 0, 0};
|
||||
struct tclient_data client2_data = {NULL, 0, 0};
|
||||
struct tclient_data client3_data = {NULL, 0, 0};
|
||||
struct mg_connection *newconn1 = NULL;
|
||||
struct mg_connection *newconn2 = NULL;
|
||||
struct mg_connection *newconn3 = NULL;
|
||||
char ebuf[100] = {0};
|
||||
|
||||
assert(websocket_welcome_msg_len == strlen(websocket_welcome_msg));
|
||||
assert(websocket_welcome_msg_len == strlen(websocket_welcome_msg));
|
||||
|
||||
/* First set up a websocket server */
|
||||
ctx = start_websocket_server();
|
||||
assert(ctx != NULL);
|
||||
printf("Server init\n\n");
|
||||
/* First set up a websocket server */
|
||||
ctx = start_websocket_server();
|
||||
assert(ctx != NULL);
|
||||
printf("Server init\n\n");
|
||||
|
||||
/* Then connect a first client */
|
||||
newconn1 = mg_connect_websocket_client("localhost", atoi(PORT), 0, ebuf, sizeof(ebuf),
|
||||
"/websocket", NULL, websocket_client_data_handler, websocket_client_close_handler,
|
||||
&client1_data);
|
||||
/* Then connect a first client */
|
||||
newconn1 = mg_connect_websocket_client("localhost",
|
||||
atoi(PORT),
|
||||
0,
|
||||
ebuf,
|
||||
sizeof(ebuf),
|
||||
"/websocket",
|
||||
NULL,
|
||||
websocket_client_data_handler,
|
||||
websocket_client_close_handler,
|
||||
&client1_data);
|
||||
|
||||
if (newconn1 == NULL)
|
||||
{
|
||||
printf("Error: %s", ebuf);
|
||||
return 1;
|
||||
}
|
||||
if (newconn1 == NULL) {
|
||||
printf("Error: %s", ebuf);
|
||||
return 1;
|
||||
}
|
||||
|
||||
sleep(1); /* Should get the websocket welcome message */
|
||||
assert(client1_data.closed == 0);
|
||||
assert(client2_data.closed == 0);
|
||||
assert(client2_data.data == NULL);
|
||||
assert(client2_data.len == 0);
|
||||
assert(client1_data.data != NULL);
|
||||
assert(client1_data.len == websocket_welcome_msg_len);
|
||||
assert(!memcmp(client1_data.data, websocket_welcome_msg, websocket_welcome_msg_len));
|
||||
free(client1_data.data);
|
||||
client1_data.data = NULL;
|
||||
client1_data.len = 0;
|
||||
sleep(1); /* Should get the websocket welcome message */
|
||||
assert(client1_data.closed == 0);
|
||||
assert(client2_data.closed == 0);
|
||||
assert(client2_data.data == NULL);
|
||||
assert(client2_data.len == 0);
|
||||
assert(client1_data.data != NULL);
|
||||
assert(client1_data.len == websocket_welcome_msg_len);
|
||||
assert(!memcmp(client1_data.data,
|
||||
websocket_welcome_msg,
|
||||
websocket_welcome_msg_len));
|
||||
free(client1_data.data);
|
||||
client1_data.data = NULL;
|
||||
client1_data.len = 0;
|
||||
|
||||
mg_websocket_write(newconn1, WEBSOCKET_OPCODE_TEXT, "data1", 5);
|
||||
mg_websocket_write(newconn1, WEBSOCKET_OPCODE_TEXT, "data1", 5);
|
||||
|
||||
sleep(1); /* Should get the acknowledge message */
|
||||
assert(client1_data.closed == 0);
|
||||
assert(client2_data.closed == 0);
|
||||
assert(client2_data.data == NULL);
|
||||
assert(client2_data.len == 0);
|
||||
assert(client1_data.data != NULL);
|
||||
assert(client1_data.len == websocket_acknowledge_msg_len);
|
||||
assert(!memcmp(client1_data.data, websocket_acknowledge_msg, websocket_acknowledge_msg_len));
|
||||
free(client1_data.data);
|
||||
client1_data.data = NULL;
|
||||
client1_data.len = 0;
|
||||
sleep(1); /* Should get the acknowledge message */
|
||||
assert(client1_data.closed == 0);
|
||||
assert(client2_data.closed == 0);
|
||||
assert(client2_data.data == NULL);
|
||||
assert(client2_data.len == 0);
|
||||
assert(client1_data.data != NULL);
|
||||
assert(client1_data.len == websocket_acknowledge_msg_len);
|
||||
assert(!memcmp(client1_data.data,
|
||||
websocket_acknowledge_msg,
|
||||
websocket_acknowledge_msg_len));
|
||||
free(client1_data.data);
|
||||
client1_data.data = NULL;
|
||||
client1_data.len = 0;
|
||||
|
||||
/* Now connect a second client */
|
||||
newconn2 = mg_connect_websocket_client("localhost", atoi(PORT), 0, ebuf, sizeof(ebuf),
|
||||
"/websocket", NULL, websocket_client_data_handler, websocket_client_close_handler,
|
||||
&client2_data);
|
||||
/* Now connect a second client */
|
||||
newconn2 = mg_connect_websocket_client("localhost",
|
||||
atoi(PORT),
|
||||
0,
|
||||
ebuf,
|
||||
sizeof(ebuf),
|
||||
"/websocket",
|
||||
NULL,
|
||||
websocket_client_data_handler,
|
||||
websocket_client_close_handler,
|
||||
&client2_data);
|
||||
|
||||
if (newconn2 == NULL)
|
||||
{
|
||||
printf("Error: %s", ebuf);
|
||||
return 1;
|
||||
}
|
||||
if (newconn2 == NULL) {
|
||||
printf("Error: %s", ebuf);
|
||||
return 1;
|
||||
}
|
||||
|
||||
sleep(1); /* Client 2 should get the websocket welcome message */
|
||||
assert(client1_data.closed == 0);
|
||||
assert(client2_data.closed == 0);
|
||||
assert(client1_data.data == NULL);
|
||||
assert(client1_data.len == 0);
|
||||
assert(client2_data.data != NULL);
|
||||
assert(client2_data.len == websocket_welcome_msg_len);
|
||||
assert(!memcmp(client2_data.data, websocket_welcome_msg, websocket_welcome_msg_len));
|
||||
free(client2_data.data);
|
||||
client2_data.data = NULL;
|
||||
client2_data.len = 0;
|
||||
sleep(1); /* Client 2 should get the websocket welcome message */
|
||||
assert(client1_data.closed == 0);
|
||||
assert(client2_data.closed == 0);
|
||||
assert(client1_data.data == NULL);
|
||||
assert(client1_data.len == 0);
|
||||
assert(client2_data.data != NULL);
|
||||
assert(client2_data.len == websocket_welcome_msg_len);
|
||||
assert(!memcmp(client2_data.data,
|
||||
websocket_welcome_msg,
|
||||
websocket_welcome_msg_len));
|
||||
free(client2_data.data);
|
||||
client2_data.data = NULL;
|
||||
client2_data.len = 0;
|
||||
|
||||
mg_websocket_write(newconn1, WEBSOCKET_OPCODE_TEXT, "data2", 5);
|
||||
mg_websocket_write(newconn1, WEBSOCKET_OPCODE_TEXT, "data2", 5);
|
||||
|
||||
sleep(1); /* Should get the acknowledge message */
|
||||
assert(client1_data.closed == 0);
|
||||
assert(client2_data.closed == 0);
|
||||
assert(client2_data.data == NULL);
|
||||
assert(client2_data.len == 0);
|
||||
assert(client1_data.data != NULL);
|
||||
assert(client1_data.len == websocket_acknowledge_msg_len);
|
||||
assert(!memcmp(client1_data.data, websocket_acknowledge_msg, websocket_acknowledge_msg_len));
|
||||
free(client1_data.data);
|
||||
client1_data.data = NULL;
|
||||
client1_data.len = 0;
|
||||
sleep(1); /* Should get the acknowledge message */
|
||||
assert(client1_data.closed == 0);
|
||||
assert(client2_data.closed == 0);
|
||||
assert(client2_data.data == NULL);
|
||||
assert(client2_data.len == 0);
|
||||
assert(client1_data.data != NULL);
|
||||
assert(client1_data.len == websocket_acknowledge_msg_len);
|
||||
assert(!memcmp(client1_data.data,
|
||||
websocket_acknowledge_msg,
|
||||
websocket_acknowledge_msg_len));
|
||||
free(client1_data.data);
|
||||
client1_data.data = NULL;
|
||||
client1_data.len = 0;
|
||||
|
||||
mg_websocket_write(newconn1, WEBSOCKET_OPCODE_TEXT, "bye", 3);
|
||||
mg_websocket_write(newconn1, WEBSOCKET_OPCODE_TEXT, "bye", 3);
|
||||
|
||||
sleep(1); /* Should get the goodbye message */
|
||||
assert(client1_data.closed == 0);
|
||||
assert(client2_data.closed == 0);
|
||||
assert(client2_data.data == NULL);
|
||||
assert(client2_data.len == 0);
|
||||
assert(client1_data.data != NULL);
|
||||
assert(client1_data.len == websocket_goodbye_msg_len);
|
||||
assert(!memcmp(client1_data.data, websocket_goodbye_msg, websocket_goodbye_msg_len));
|
||||
free(client1_data.data);
|
||||
client1_data.data = NULL;
|
||||
client1_data.len = 0;
|
||||
sleep(1); /* Should get the goodbye message */
|
||||
assert(client1_data.closed == 0);
|
||||
assert(client2_data.closed == 0);
|
||||
assert(client2_data.data == NULL);
|
||||
assert(client2_data.len == 0);
|
||||
assert(client1_data.data != NULL);
|
||||
assert(client1_data.len == websocket_goodbye_msg_len);
|
||||
assert(!memcmp(client1_data.data,
|
||||
websocket_goodbye_msg,
|
||||
websocket_goodbye_msg_len));
|
||||
free(client1_data.data);
|
||||
client1_data.data = NULL;
|
||||
client1_data.len = 0;
|
||||
|
||||
mg_close_connection(newconn1);
|
||||
mg_close_connection(newconn1);
|
||||
|
||||
sleep(1); /* Won't get any message */
|
||||
assert(client1_data.closed == 1);
|
||||
assert(client2_data.closed == 0);
|
||||
assert(client1_data.data == NULL);
|
||||
assert(client1_data.len == 0);
|
||||
assert(client2_data.data == NULL);
|
||||
assert(client2_data.len == 0);
|
||||
sleep(1); /* Won't get any message */
|
||||
assert(client1_data.closed == 1);
|
||||
assert(client2_data.closed == 0);
|
||||
assert(client1_data.data == NULL);
|
||||
assert(client1_data.len == 0);
|
||||
assert(client2_data.data == NULL);
|
||||
assert(client2_data.len == 0);
|
||||
|
||||
mg_websocket_write(newconn2, WEBSOCKET_OPCODE_TEXT, "bye", 3);
|
||||
mg_websocket_write(newconn2, WEBSOCKET_OPCODE_TEXT, "bye", 3);
|
||||
|
||||
sleep(1); /* Should get the goodbye message */
|
||||
assert(client1_data.closed == 1);
|
||||
assert(client2_data.closed == 0);
|
||||
assert(client1_data.data == NULL);
|
||||
assert(client1_data.len == 0);
|
||||
assert(client2_data.data != NULL);
|
||||
assert(client2_data.len == websocket_goodbye_msg_len);
|
||||
assert(!memcmp(client2_data.data, websocket_goodbye_msg, websocket_goodbye_msg_len));
|
||||
free(client2_data.data);
|
||||
client2_data.data = NULL;
|
||||
client2_data.len = 0;
|
||||
sleep(1); /* Should get the goodbye message */
|
||||
assert(client1_data.closed == 1);
|
||||
assert(client2_data.closed == 0);
|
||||
assert(client1_data.data == NULL);
|
||||
assert(client1_data.len == 0);
|
||||
assert(client2_data.data != NULL);
|
||||
assert(client2_data.len == websocket_goodbye_msg_len);
|
||||
assert(!memcmp(client2_data.data,
|
||||
websocket_goodbye_msg,
|
||||
websocket_goodbye_msg_len));
|
||||
free(client2_data.data);
|
||||
client2_data.data = NULL;
|
||||
client2_data.len = 0;
|
||||
|
||||
mg_close_connection(newconn2);
|
||||
mg_close_connection(newconn2);
|
||||
|
||||
sleep(1); /* Won't get any message */
|
||||
assert(client1_data.closed == 1);
|
||||
assert(client2_data.closed == 1);
|
||||
assert(client1_data.data == NULL);
|
||||
assert(client1_data.len == 0);
|
||||
assert(client2_data.data == NULL);
|
||||
assert(client2_data.len == 0);
|
||||
sleep(1); /* Won't get any message */
|
||||
assert(client1_data.closed == 1);
|
||||
assert(client2_data.closed == 1);
|
||||
assert(client1_data.data == NULL);
|
||||
assert(client1_data.len == 0);
|
||||
assert(client2_data.data == NULL);
|
||||
assert(client2_data.len == 0);
|
||||
|
||||
/* Connect client 3 */
|
||||
newconn3 = mg_connect_websocket_client("localhost", atoi(PORT), 0, ebuf, sizeof(ebuf),
|
||||
"/websocket", NULL, websocket_client_data_handler, websocket_client_close_handler,
|
||||
&client3_data);
|
||||
/* Connect client 3 */
|
||||
newconn3 = mg_connect_websocket_client("localhost",
|
||||
atoi(PORT),
|
||||
0,
|
||||
ebuf,
|
||||
sizeof(ebuf),
|
||||
"/websocket",
|
||||
NULL,
|
||||
websocket_client_data_handler,
|
||||
websocket_client_close_handler,
|
||||
&client3_data);
|
||||
|
||||
sleep(1); /* Client 3 should get the websocket welcome message */
|
||||
assert(client1_data.closed == 1);
|
||||
assert(client2_data.closed == 1);
|
||||
assert(client3_data.closed == 0);
|
||||
assert(client1_data.data == NULL);
|
||||
assert(client1_data.len == 0);
|
||||
assert(client2_data.data == NULL);
|
||||
assert(client2_data.len == 0);
|
||||
assert(client3_data.data != NULL);
|
||||
assert(client3_data.len == websocket_welcome_msg_len);
|
||||
assert(!memcmp(client3_data.data, websocket_welcome_msg, websocket_welcome_msg_len));
|
||||
free(client3_data.data);
|
||||
client3_data.data = NULL;
|
||||
client3_data.len = 0;
|
||||
sleep(1); /* Client 3 should get the websocket welcome message */
|
||||
assert(client1_data.closed == 1);
|
||||
assert(client2_data.closed == 1);
|
||||
assert(client3_data.closed == 0);
|
||||
assert(client1_data.data == NULL);
|
||||
assert(client1_data.len == 0);
|
||||
assert(client2_data.data == NULL);
|
||||
assert(client2_data.len == 0);
|
||||
assert(client3_data.data != NULL);
|
||||
assert(client3_data.len == websocket_welcome_msg_len);
|
||||
assert(!memcmp(client3_data.data,
|
||||
websocket_welcome_msg,
|
||||
websocket_welcome_msg_len));
|
||||
free(client3_data.data);
|
||||
client3_data.data = NULL;
|
||||
client3_data.len = 0;
|
||||
|
||||
mg_stop(ctx);
|
||||
printf("Server shutdown\n");
|
||||
mg_stop(ctx);
|
||||
printf("Server shutdown\n");
|
||||
|
||||
sleep(10);
|
||||
sleep(10);
|
||||
|
||||
assert(client3_data.closed == 1);
|
||||
assert(client3_data.closed == 1);
|
||||
|
||||
return 0;
|
||||
return 0;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user