90 lines
2.3 KiB
C
90 lines
2.3 KiB
C
|
#ifndef HV_WS_DEF_H_
|
||
|
|
#define HV_WS_DEF_H_
|
||
|
|
|
||
|
|
#include "hexport.h"
|
||
|
|
|
||
|
|
#include <stdbool.h>
|
||
|
|
#include <stdlib.h> // import rand
|
||
|
|
|
||
|
|
#define SEC_WEBSOCKET_VERSION "Sec-WebSocket-Version"
|
||
|
|
#define SEC_WEBSOCKET_KEY "Sec-WebSocket-Key"
|
||
|
|
#define SEC_WEBSOCKET_ACCEPT "Sec-WebSocket-Accept"
|
||
|
|
#define SEC_WEBSOCKET_PROTOCOL "Sec-WebSocket-Protocol"
|
||
|
|
#define SEC_WEBSOCKET_EXTENSIONS "Sec-WebSocket-Extensions"
|
||
|
|
|
||
|
|
#define WS_SERVER_MIN_FRAME_SIZE 2
|
||
|
|
// 1000 1001 0000 0000
|
||
|
|
#define WS_SERVER_PING_FRAME "\211\0"
|
||
|
|
// 1000 1010 0000 0000
|
||
|
|
#define WS_SERVER_PONG_FRAME "\212\0"
|
||
|
|
|
||
|
|
#define WS_CLIENT_MIN_FRAME_SIZE 6
|
||
|
|
// 1000 1001 1000 0000
|
||
|
|
#define WS_CLIENT_PING_FRAME "\211\200WSWS"
|
||
|
|
// 1000 1010 1000 0000
|
||
|
|
#define WS_CLIENT_PONG_FRAME "\212\200WSWS"
|
||
|
|
|
||
|
|
enum ws_session_type {
|
||
|
|
WS_CLIENT,
|
||
|
|
WS_SERVER,
|
||
|
|
};
|
||
|
|
|
||
|
|
enum ws_opcode {
|
||
|
|
WS_OPCODE_CONTINUE = 0x0,
|
||
|
|
WS_OPCODE_TEXT = 0x1,
|
||
|
|
WS_OPCODE_BINARY = 0x2,
|
||
|
|
WS_OPCODE_CLOSE = 0x8,
|
||
|
|
WS_OPCODE_PING = 0x9,
|
||
|
|
WS_OPCODE_PONG = 0xA,
|
||
|
|
};
|
||
|
|
|
||
|
|
BEGIN_EXTERN_C
|
||
|
|
|
||
|
|
// Sec-WebSocket-Key => Sec-WebSocket-Accept
|
||
|
|
HV_EXPORT void ws_encode_key(const char* key, char accept[]);
|
||
|
|
|
||
|
|
// fix-header[2] + var-length[2/8] + mask[4] + data[data_len]
|
||
|
|
HV_EXPORT int ws_calc_frame_size(int data_len, bool has_mask DEFAULT(false));
|
||
|
|
|
||
|
|
HV_EXPORT int ws_build_frame(
|
||
|
|
char* out,
|
||
|
|
const char* data,
|
||
|
|
int data_len,
|
||
|
|
const char mask[4],
|
||
|
|
bool has_mask DEFAULT(false),
|
||
|
|
enum ws_opcode opcode DEFAULT(WS_OPCODE_TEXT),
|
||
|
|
bool fin DEFAULT(true));
|
||
|
|
|
||
|
|
HV_INLINE int ws_client_build_frame(
|
||
|
|
char* out,
|
||
|
|
const char* data,
|
||
|
|
int data_len,
|
||
|
|
/* const char mask[4] */
|
||
|
|
/* bool has_mask = true */
|
||
|
|
enum ws_opcode opcode DEFAULT(WS_OPCODE_TEXT),
|
||
|
|
bool fin DEFAULT(true)) {
|
||
|
|
char mask[4] = {0};
|
||
|
|
int i = 0;
|
||
|
|
int imask = rand();
|
||
|
|
for (i = 0; i < 4; i++) {
|
||
|
|
mask[i] = (imask >> (8 * i)) & 0xff;
|
||
|
|
}
|
||
|
|
return ws_build_frame(out, data, data_len, mask, true, opcode, fin);
|
||
|
|
}
|
||
|
|
|
||
|
|
HV_INLINE int ws_server_build_frame(
|
||
|
|
char* out,
|
||
|
|
const char* data,
|
||
|
|
int data_len,
|
||
|
|
/* const char mask[4] */
|
||
|
|
/* bool has_mask = false */
|
||
|
|
enum ws_opcode opcode DEFAULT(WS_OPCODE_TEXT),
|
||
|
|
bool fin DEFAULT(true)) {
|
||
|
|
char mask[4] = {0};
|
||
|
|
return ws_build_frame(out, data, data_len, mask, false, opcode, fin);
|
||
|
|
}
|
||
|
|
|
||
|
|
END_EXTERN_C
|
||
|
|
|
||
|
|
#endif // HV_WS_DEF_H_
|