#include "websock.h"
#define BLOCK_DATA_SIZE 4096
+#if !defined(WEBSOCKET_DEFAULT_MAXLENGTH)
+# define WEBSOCKET_DEFAULT_MAXLENGTH 1048500 /* 76 less than 1M, probably enougth for headers */
+#endif
#define FRAME_GET_FIN(BYTE) (((BYTE) >> 7) & 0x01)
#define FRAME_GET_RSV1(BYTE) (((BYTE) >> 6) & 0x01)
#define FRAME_SET_RSV3(BYTE) (((BYTE) & 0x01) << 4)
#define FRAME_SET_OPCODE(BYTE) ((BYTE) & 0x0F)
#define FRAME_SET_MASK(BYTE) (((BYTE) & 0x01) << 7)
-#define FRAME_SET_LENGTH(X64, IDX) (unsigned char)(((X64) >> ((IDX)*8)) & 0xFF)
+#define FRAME_SET_LENGTH(X64, IDX) (unsigned char)((sizeof(X64)) <= (IDX) ? 0 : (((X64) >> ((IDX)*8)) & 0xFF))
#define OPCODE_CONTINUATION 0x0
#define OPCODE_TEXT 0x1
#define STATE_LENGTH 2
#define STATE_DATA 3
+static size_t default_maxlength = WEBSOCKET_DEFAULT_MAXLENGTH;
+
struct websock {
int state;
uint64_t maxlength;
if (result) {
result->itf = itf;
result->closure = closure;
- result->maxlength = 65000;
+ result->maxlength = default_maxlength;
}
return result;
}
{
free(ws);
}
+
+void websock_set_default_max_length(size_t maxlen)
+{
+ default_maxlength = maxlen;
+}
+
+void websock_set_max_length(struct websock *ws, size_t maxlen)
+{
+ ws->maxlength = (uint64_t)maxlen;
+}