stv0g/libwebsockets
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

test client bit more cleaning

Browse source 
Signed-off-by: Andy Green <andy.green@linaro.org>
This commit is contained in:
Andy Green 2015-12-09 08:07:38 +08:00
parent 8266450ba2
commit cb0cbfaec1
1 changed files with 42 additions and 51 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

93
test-server/test-client.c
View file

@ -36,13 +36,10 @@
#include "../lib/libwebsockets.h"
static unsigned int opts;
static int deny_deflate;
static int deny_mux;
static int deny_deflate, deny_mux, longlived, mirror_lifetime;
static struct lws *wsi_dumb, *wsi_mirror;
static int mirror_lifetime = 0;
static volatile int force_exit = 0;
static int longlived = 0;
static volatile int force_exit;
static unsigned int opts;
/*
* This demo shows how to connect multiple websockets simultaneously to a
@ -67,7 +64,12 @@ enum demo_protocols {
};
/* dumb_increment protocol */
/*
* dumb_increment protocol
*
* since this also happens to be protocols[0], some callbacks that are not
* bound to a specific protocol also turn up here.
*/
static int
callback_dumb_increment(struct lws_context *this,
@ -78,12 +80,21 @@ callback_dumb_increment(struct lws_context *this,
switch (reason) {
case LWS_CALLBACK_CLIENT_ESTABLISHED:
fprintf(stderr, "dumb: LWS_CALLBACK_CLIENT_ESTABLISHED\n");
lwsl_info("dumb: LWS_CALLBACK_CLIENT_ESTABLISHED\n");
break;
/* this callback comes only at protocols[0] handler
* since at this time, no protocol had been negotiated
*/
case LWS_CALLBACK_CLOSED:
lwsl_notice("dumb: LWS_CALLBACK_CLOSED\n");
wsi_dumb = NULL;
break;
case LWS_CALLBACK_CLIENT_RECEIVE:
((char *)in)[len] = '\0';
lwsl_info("rx %d '%s'\n", (int)len, (char *)in);
break;
/* because we are protocols[0] ... */
case LWS_CALLBACK_CLIENT_CONNECTION_ERROR:
if (wsi == wsi_dumb) {
lwsl_err("dumb: LWS_CALLBACK_CLIENT_CONNECTION_ERROR\n");
@ -95,32 +106,19 @@ callback_dumb_increment(struct lws_context *this,
}
break;
case LWS_CALLBACK_CLOSED:
lwsl_notice("dumb: LWS_CALLBACK_CLOSED\n");
wsi_dumb = NULL;
break;
case LWS_CALLBACK_CLIENT_RECEIVE:
((char *)in)[len] = '\0';
fprintf(stderr, "rx %d '%s'\n", (int)len, (char *)in);
break;
/* because we are protocols[0] ... */
case LWS_CALLBACK_CLIENT_CONFIRM_EXTENSION_SUPPORTED:
if ((strcmp(in, "deflate-stream") == 0) && deny_deflate) {
fprintf(stderr, "denied deflate-stream extension\n");
lwsl_notice("denied deflate-stream extension\n");
return 1;
}
if ((strcmp(in, "deflate-frame") == 0) && deny_deflate) {
fprintf(stderr, "denied deflate-frame extension\n");
lwsl_notice("denied deflate-frame extension\n");
return 1;
}
if ((strcmp(in, "x-google-mux") == 0) && deny_mux) {
fprintf(stderr, "denied x-google-mux extension\n");
lwsl_notice("denied x-google-mux extension\n");
return 1;
}
break;
default:
@ -136,24 +134,23 @@ callback_dumb_increment(struct lws_context *this,
static int
callback_lws_mirror(struct lws_context *context,
struct lws *wsi,
enum lws_callback_reasons reason,
void *user, void *in, size_t len)
struct lws *wsi,
enum lws_callback_reasons reason,
void *user, void *in, size_t len)
{
unsigned char buf[LWS_SEND_BUFFER_PRE_PADDING + 4096 +
LWS_SEND_BUFFER_POST_PADDING];
LWS_SEND_BUFFER_POST_PADDING];
unsigned int rands[4];
int l = 0;
int n;
unsigned int rands[4];
switch (reason) {
case LWS_CALLBACK_CLIENT_ESTABLISHED:
lwsl_notice("mirror: LWS_CALLBACK_CLIENT_ESTABLISHED\n");
lws_get_random(context, rands, sizeof(rands[0]));
mirror_lifetime = 65535 + (rands[0] & 65535);
mirror_lifetime = 16384 + (rands[0] & 65535);
/* useful to test single connection stability */
if (longlived)
mirror_lifetime += 500000;
@ -166,13 +163,10 @@ callback_lws_mirror(struct lws_context *context,
* zero the connection is closed in the send callback.
* When the close callback comes, wsi_mirror is set to NULL
* so a new connection will be opened
*/
/*
*
* start the ball rolling,
* LWS_CALLBACK_CLIENT_WRITEABLE will come next service
*/
lws_callback_on_writable(context, wsi);
break;
@ -181,12 +175,7 @@ callback_lws_mirror(struct lws_context *context,
wsi_mirror = NULL;
break;
case LWS_CALLBACK_CLIENT_RECEIVE:
/* fprintf(stderr, "rx %d '%s'\n", (int)len, (char *)in); */
break;
case LWS_CALLBACK_CLIENT_WRITEABLE:
for (n = 0; n < 1; n++) {
lws_get_random(context, rands, sizeof(rands));
l += sprintf((char *)&buf[LWS_SEND_BUFFER_PRE_PADDING + l],
@ -197,9 +186,8 @@ callback_lws_mirror(struct lws_context *context,
(int)rands[3] % 24);
}
n = lws_write(wsi,
&buf[LWS_SEND_BUFFER_PRE_PADDING], l, opts | LWS_WRITE_TEXT);
n = lws_write(wsi, &buf[LWS_SEND_BUFFER_PRE_PADDING], l,
opts | LWS_WRITE_TEXT);
if (n < 0)
return -1;
if (n < l) {
@ -209,11 +197,11 @@ callback_lws_mirror(struct lws_context *context,
mirror_lifetime--;
if (!mirror_lifetime) {
fprintf(stderr, "closing mirror session\n");
lwsl_info("closing mirror session\n");
return -1;
} else
/* get notified as soon as we can write again */
lws_callback_on_writable(context, wsi);
}
/* get notified as soon as we can write again */
lws_callback_on_writable(context, wsi);
break;
default:
@ -353,8 +341,11 @@ int main(int argc, char **argv)
/*
* sit there servicing the websocket context to handle incoming
* packets, and drawing random circles on the mirror protocol websocket
*
* nothing happens until the client websocket connection is
* asynchronously established
* asynchronously established... calling lws_client_connect() only
* instantiates the connection logically, lws_service() progresses it
* asynchronously.
*/
while (!force_exit) {