86725f9b4c
- add dup2 functionality. This requires adding a ref cnt to the _WS_connections structure so that we only free the structure once all dup'd referenced are closed. Also, refactor malloc and free of connection structure into _WS_alloc and _WS_free. - allow select to accept a NULL timeout value which means sleep forever instead of segfaulting. - fix some compile warnings related to ppoll definition. - move some WebSockets related html test pages into utils and symlink them from tests.
68 lines
1.4 KiB
C
68 lines
1.4 KiB
C
/*
|
|
* wswrap/wswrapper: Add WebSockets support to any service.
|
|
* Copyright 2010 Joel Martin
|
|
* Licensed under LGPL version 3 (see docs/LICENSE.LGPL-3)
|
|
*
|
|
* wswrapper.so is meant to be LD preloaded. Use wswrap to run a program using
|
|
* wswrapper.so.
|
|
*/
|
|
|
|
#ifdef DO_MSG
|
|
#define MSG(...) \
|
|
fprintf(stderr, "wswrapper: "); \
|
|
fprintf(stderr, __VA_ARGS__);
|
|
#else
|
|
#define MSG(...)
|
|
#endif
|
|
|
|
#ifdef DO_DEBUG
|
|
#define DEBUG(...) \
|
|
if (DO_DEBUG) { \
|
|
fprintf(stderr, "wswrapper: "); \
|
|
fprintf(stderr, __VA_ARGS__); \
|
|
}
|
|
#else
|
|
#define DEBUG(...)
|
|
#endif
|
|
|
|
#ifdef DO_TRACE
|
|
#define TRACE(...) \
|
|
if (DO_TRACE) { \
|
|
fprintf(stderr, "wswrapper: "); \
|
|
fprintf(stderr, __VA_ARGS__); \
|
|
}
|
|
#else
|
|
#define TRACE(...)
|
|
#endif
|
|
|
|
#define RET_ERROR(eno, ...) \
|
|
fprintf(stderr, "wswrapper error: "); \
|
|
fprintf(stderr, __VA_ARGS__); \
|
|
errno = eno; \
|
|
return -1;
|
|
|
|
|
|
const char _WS_response[] = "\
|
|
HTTP/1.1 101 Web Socket Protocol Handshake\r\n\
|
|
Upgrade: WebSocket\r\n\
|
|
Connection: Upgrade\r\n\
|
|
%sWebSocket-Origin: %s\r\n\
|
|
%sWebSocket-Location: %s://%s%s\r\n\
|
|
%sWebSocket-Protocol: sample\r\n\
|
|
\r\n%s";
|
|
|
|
#define WS_BUFSIZE 65536
|
|
#define WS_MAX_FDS 1024
|
|
|
|
/* Buffers and state for each wrapped WebSocket connection */
|
|
typedef struct {
|
|
char rbuf[WS_BUFSIZE];
|
|
char sbuf[WS_BUFSIZE];
|
|
int rcarry_cnt;
|
|
char rcarry[3];
|
|
int newframe;
|
|
int refcnt;
|
|
} _WS_connection;
|
|
|
|
|