mirror of
https://github.com/cesanta/mongoose.git
synced 2025-06-07 17:42:30 +08:00
67 lines
2.4 KiB
C
67 lines
2.4 KiB
C
// Copyright (c) 2020 Cesanta Software Limited
|
|
// All rights reserved
|
|
//
|
|
// Example Websocket client. Usage:
|
|
// 1. Start websocket server: cd ../websocket-server && make
|
|
// 2. In another terminal, start this client: make
|
|
|
|
#include "mongoose.h"
|
|
|
|
static const char *s_url = "ws://localhost:8000/websocket";
|
|
static const char *s_ca_path = "ca.pem";
|
|
|
|
// Print websocket response and signal that we're done
|
|
static void fn(struct mg_connection *c, int ev, void *ev_data) {
|
|
if (ev == MG_EV_OPEN) {
|
|
c->is_hexdumping = 1;
|
|
} else if (c->is_tls && ev == MG_EV_CONNECT) {
|
|
struct mg_str ca = mg_file_read(&mg_fs_posix, s_ca_path);
|
|
struct mg_tls_opts opts = {.ca = ca, .name = mg_url_host(s_url)};
|
|
mg_tls_init(c, &opts);
|
|
} else if (ev == MG_EV_ERROR) {
|
|
// On error, log error message
|
|
MG_ERROR(("%p %s", c->fd, (char *) ev_data));
|
|
} else if (ev == MG_EV_WS_OPEN) {
|
|
// When websocket handshake is successful, send message
|
|
mg_ws_send(c, "hello", 5, WEBSOCKET_OP_TEXT);
|
|
} else if (ev == MG_EV_WS_MSG) {
|
|
// When we get echo response, print it
|
|
struct mg_ws_message *wm = (struct mg_ws_message *) ev_data;
|
|
printf("GOT ECHO REPLY: [%.*s]\n", (int) wm->data.len, wm->data.buf);
|
|
}
|
|
|
|
if (ev == MG_EV_ERROR || ev == MG_EV_CLOSE || ev == MG_EV_WS_MSG) {
|
|
*(bool *) c->fn_data = true; // Signal that we're done
|
|
}
|
|
}
|
|
|
|
int main(int argc, char *argv[]) {
|
|
struct mg_mgr mgr; // Event manager
|
|
bool done = false; // Event handler flips it to true
|
|
struct mg_connection *c; // Client connection
|
|
int i;
|
|
|
|
// Parse command-line flags
|
|
for (i = 1; i < argc; i++) {
|
|
if (strcmp(argv[i], "-url") == 0 && argv[i + 1] != NULL) {
|
|
s_url = argv[++i];
|
|
} else if (strcmp(argv[i], "-ca") == 0 && argv[i + 1] != NULL) {
|
|
s_ca_path = argv[++i];
|
|
} else {
|
|
printf(
|
|
"Usage: %s OPTIONS\n"
|
|
" -ca PATH - Path to the CA file, default: '%s'\n"
|
|
" -url URL - Destination URL, default: '%s'\n",
|
|
argv[0], s_ca_path, s_url);
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
mg_mgr_init(&mgr); // Initialise event manager
|
|
mg_log_set(MG_LL_DEBUG); // Set log level
|
|
c = mg_ws_connect(&mgr, s_url, fn, &done, NULL); // Create client
|
|
while (c && done == false) mg_mgr_poll(&mgr, 1000); // Wait for echo
|
|
mg_mgr_free(&mgr); // Deallocate resources
|
|
return 0;
|
|
}
|