Joshua Moerman
12 years ago
commit
f544faf78d
4 changed files with 265 additions and 0 deletions
@ -0,0 +1,3 @@ |
|||
[submodule "contrib/libwebsockets"] |
|||
path = contrib/libwebsockets |
|||
url = git://github.com/warmcat/libwebsockets.git |
@ -0,0 +1,17 @@ |
|||
{ |
|||
"folders": |
|||
[ |
|||
{ |
|||
"path": "/Users/joshua/Documents/Code/OnlineBeats" |
|||
} |
|||
], |
|||
|
|||
"settings": |
|||
{ |
|||
"sublimeclang_options": |
|||
[ |
|||
"-isystem${folder:${project_path:OnlineBeats.sublime-project}}/contrib/libwebsockets/", |
|||
"-Wall" |
|||
] |
|||
} |
|||
} |
@ -0,0 +1,244 @@ |
|||
/*
|
|||
* libwebsockets-test-echo - libwebsockets echo test implementation |
|||
* |
|||
* This implements both the client and server sides. It defaults to |
|||
* serving, use --client <remote address> to connect as client. |
|||
* |
|||
* Copyright (C) 2010-2013 Andy Green <andy@warmcat.com> |
|||
* |
|||
* This library is free software; you can redistribute it and/or |
|||
* modify it under the terms of the GNU Lesser General Public |
|||
* License as published by the Free Software Foundation: |
|||
* version 2.1 of the License. |
|||
* |
|||
* This library is distributed in the hope that it will be useful, |
|||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
|||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
|||
* Lesser General Public License for more details. |
|||
* |
|||
* You should have received a copy of the GNU Lesser General Public |
|||
* License along with this library; if not, write to the Free Software |
|||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, |
|||
* MA 02110-1301 USA |
|||
*/ |
|||
|
|||
#include <stdio.h> |
|||
#include <stdlib.h> |
|||
#include <unistd.h> |
|||
#include <getopt.h> |
|||
#include <string.h> |
|||
#include <sys/time.h> |
|||
#include <assert.h> |
|||
#include <syslog.h> |
|||
#include <signal.h> |
|||
|
|||
#ifdef CMAKE_BUILD |
|||
#include "lws_config.h" |
|||
#endif |
|||
|
|||
#include "lib/libwebsockets.h" |
|||
|
|||
int force_exit = 0; |
|||
|
|||
#define MAX_ECHO_PAYLOAD 1400 |
|||
|
|||
struct per_session_data__echo { |
|||
unsigned char buf[LWS_SEND_BUFFER_PRE_PADDING + MAX_ECHO_PAYLOAD + LWS_SEND_BUFFER_POST_PADDING]; |
|||
unsigned int len; |
|||
unsigned int index; |
|||
}; |
|||
|
|||
static int |
|||
callback_echo(struct libwebsocket_context *context, |
|||
struct libwebsocket *wsi, |
|||
enum libwebsocket_callback_reasons reason, void *user, |
|||
void *in, size_t len){ |
|||
struct per_session_data__echo *pss = (struct per_session_data__echo *)user; |
|||
int n; |
|||
|
|||
switch (reason) { |
|||
case LWS_CALLBACK_SERVER_WRITEABLE: |
|||
n = libwebsocket_write(wsi, &pss->buf[LWS_SEND_BUFFER_PRE_PADDING], pss->len, LWS_WRITE_TEXT); |
|||
if (n < 0) { |
|||
lwsl_err("ERROR %d writing to socket, hanging up\n", n); |
|||
return 1; |
|||
} |
|||
if (n < pss->len) { |
|||
lwsl_err("Partial write\n"); |
|||
return -1; |
|||
} |
|||
break; |
|||
|
|||
case LWS_CALLBACK_RECEIVE: |
|||
if (len > MAX_ECHO_PAYLOAD) { |
|||
lwsl_err("Server received packet bigger than %u, hanging up\n", MAX_ECHO_PAYLOAD); |
|||
return 1; |
|||
} |
|||
memcpy(&pss->buf[LWS_SEND_BUFFER_PRE_PADDING], in, len); |
|||
pss->len = len; |
|||
libwebsocket_callback_on_writable(context, wsi); |
|||
break; |
|||
case LWS_CALLBACK_CLIENT_ESTABLISHED: |
|||
case LWS_CALLBACK_CLIENT_RECEIVE: |
|||
case LWS_CALLBACK_CLIENT_WRITEABLE: |
|||
default: |
|||
break; |
|||
} |
|||
|
|||
return 0; |
|||
} |
|||
|
|||
static struct libwebsocket_protocols protocols[] = { |
|||
/* first protocol must always be HTTP handler */ |
|||
|
|||
{ |
|||
"default", /* name */ |
|||
callback_echo, /* callback */ |
|||
sizeof(struct per_session_data__echo) /* per_session_data_size */ |
|||
}, |
|||
{ |
|||
NULL, NULL, 0 /* End of list */ |
|||
} |
|||
}; |
|||
|
|||
void sighandler(int sig) |
|||
{ |
|||
force_exit = 1; |
|||
} |
|||
|
|||
static struct option options[] = { |
|||
{ "help", no_argument, NULL, 'h' }, |
|||
{ "debug", required_argument, NULL, 'd' }, |
|||
{ "port", required_argument, NULL, 'p' }, |
|||
{ "ssl", no_argument, NULL, 's' }, |
|||
{ "interface", required_argument, NULL, 'i' }, |
|||
#ifndef LWS_NO_DAEMONIZE |
|||
{ "daemonize", no_argument, NULL, 'D' }, |
|||
#endif |
|||
{ NULL, 0, 0, 0 } |
|||
}; |
|||
|
|||
int main(int argc, char **argv){ |
|||
int n = 0; |
|||
int port = 7681; |
|||
int use_ssl = 0; |
|||
struct libwebsocket_context *context; |
|||
int opts = 0; |
|||
char interface_name[128] = ""; |
|||
const char *interface = NULL; |
|||
int syslog_options = LOG_PID | LOG_PERROR; |
|||
int client = 0; |
|||
int listen_port; |
|||
struct lws_context_creation_info info; |
|||
|
|||
int debug_level = 7; |
|||
#ifndef LWS_NO_DAEMONIZE |
|||
int daemonize = 0; |
|||
#endif |
|||
|
|||
memset(&info, 0, sizeof info); |
|||
|
|||
#ifndef LWS_NO_SERVER |
|||
lwsl_notice("Built to support server operations\n"); |
|||
#endif |
|||
|
|||
while (n >= 0) { |
|||
n = getopt_long(argc, argv, "i:hsp:d:D" |
|||
, options, NULL); |
|||
if (n < 0) |
|||
continue; |
|||
switch (n) { |
|||
#ifndef LWS_NO_DAEMONIZE |
|||
case 'D': |
|||
daemonize = 1; |
|||
syslog_options &= ~LOG_PERROR; |
|||
break; |
|||
#endif |
|||
case 'd': |
|||
debug_level = atoi(optarg); |
|||
break; |
|||
case 's': |
|||
use_ssl = 1; /* 1 = take care about cert verification, 2 = allow anything */ |
|||
break; |
|||
case 'p': |
|||
port = atoi(optarg); |
|||
break; |
|||
case 'i': |
|||
strncpy(interface_name, optarg, sizeof interface_name); |
|||
interface_name[(sizeof interface_name) - 1] = '\0'; |
|||
interface = interface_name; |
|||
break; |
|||
case '?': |
|||
case 'h': |
|||
fprintf(stderr, "Usage: libwebsockets-test-echo " |
|||
"[--ssl] " |
|||
"[--port=<p>] " |
|||
"[-d <log bitfield>]\n"); |
|||
exit(1); |
|||
} |
|||
} |
|||
|
|||
#ifndef LWS_NO_DAEMONIZE |
|||
/*
|
|||
* normally lock path would be /var/lock/lwsts or similar, to |
|||
* simplify getting started without having to take care about |
|||
* permissions or running as root, set to /tmp/.lwsts-lock |
|||
*/ |
|||
if (!client && daemonize && lws_daemonize("/tmp/.lwstecho-lock")) { |
|||
fprintf(stderr, "Failed to daemonize\n"); |
|||
return 1; |
|||
} |
|||
#endif |
|||
|
|||
/* we will only try to log things according to our debug_level */ |
|||
setlogmask(LOG_UPTO (LOG_DEBUG)); |
|||
openlog("lwsts", syslog_options, LOG_DAEMON); |
|||
|
|||
/* tell the library what debug level to emit and to send it to syslog */ |
|||
lws_set_log_level(debug_level, lwsl_emit_syslog); |
|||
|
|||
lwsl_notice("libwebsockets echo test - " |
|||
"(C) Copyright 2010-2013 Andy Green <andy@warmcat.com> - " |
|||
"licensed under LGPL2.1\n"); |
|||
|
|||
#ifndef LWS_NO_SERVER |
|||
lwsl_notice("Running in server mode\n"); |
|||
listen_port = port; |
|||
#endif |
|||
|
|||
info.port = listen_port; |
|||
info.iface = interface; |
|||
info.protocols = protocols; |
|||
#ifndef LWS_NO_EXTENSIONS |
|||
info.extensions = libwebsocket_get_internal_extensions(); |
|||
#endif |
|||
if (use_ssl && !client) { |
|||
info.ssl_cert_filepath = "libwebsockets-test-server.pem"; |
|||
info.ssl_private_key_filepath = "libwebsockets-test-server.key.pem"; |
|||
} |
|||
info.gid = -1; |
|||
info.uid = -1; |
|||
info.options = opts; |
|||
|
|||
context = libwebsocket_create_context(&info); |
|||
|
|||
if (context == NULL) { |
|||
lwsl_err("libwebsocket init failed\n"); |
|||
return -1; |
|||
} |
|||
|
|||
signal(SIGINT, sighandler); |
|||
|
|||
n = 0; |
|||
while (n >= 0 && !force_exit) { |
|||
n = libwebsocket_service(context, 10); |
|||
} |
|||
|
|||
libwebsocket_context_destroy(context); |
|||
|
|||
lwsl_notice("libwebsockets-test-echo exited cleanly\n"); |
|||
|
|||
closelog(); |
|||
|
|||
return 0; |
|||
} |
Reference in new issue