forked from swoole/swoole-src
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathwebsocket.h
110 lines (97 loc) · 3.71 KB
/
websocket.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
/*
+----------------------------------------------------------------------+
| Swoole |
+----------------------------------------------------------------------+
| This source file is subject to version 2.0 of the Apache license, |
| that is bundled with this package in the file LICENSE, and is |
| available through the world-wide-web at the following url: |
| http://www.apache.org/licenses/LICENSE-2.0.html |
| If you did not receive a copy of the Apache2.0 license and are unable|
| to obtain it through the world-wide-web, please send a note to |
| [email protected] so we can mail you a copy immediately. |
+----------------------------------------------------------------------+
| Author: Tianfeng Han <[email protected]> |
+----------------------------------------------------------------------+
*/
#ifndef SW_WEBSOCKET_H_
#define SW_WEBSOCKET_H_
#ifdef __cplusplus
extern "C"
{
#endif
#include "http.h"
#define SW_WEBSOCKET_GUID "258EAFA5-E914-47DA-95CA-C5AB0DC85B11"
#define SW_WEBSOCKET_HEADER_LEN 2
#define SW_WEBSOCKET_MASK_LEN 4
#define SW_WEBSOCKET_MASK_DATA "258E"
#define SW_WEBSOCKET_EXT16_LENGTH 0x7E
#define SW_WEBSOCKET_EXT16_MAX_LEN 0xFFFF
#define SW_WEBSOCKET_EXT64_LENGTH 0x7F
#define SW_WEBSOCKET_MASKED(frm) (frm->header.MASK)
#define SW_WEBSOCKET_CLOSE_CODE_LEN 2
#define SW_WEBSOCKET_CLOSE_REASON_MAX_LEN 125
#define SW_WEBSOCKET_OPCODE_MAX WEBSOCKET_OPCODE_PONG
#define FRAME_SET_FIN(BYTE) (((BYTE) & 0x01) << 7)
#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)
enum swWebsocketStatus
{
WEBSOCKET_STATUS_NONE = 0,
WEBSOCKET_STATUS_CONNECTION = 1,
WEBSOCKET_STATUS_HANDSHAKE = 2,
WEBSOCKET_STATUS_ACTIVE = 3,
WEBSOCKET_STATUS_CLOSING = 4,
};
typedef struct
{
/**
* fin:1 rsv1:1 rsv2:1 rsv3:1 opcode:4
*/
struct
{
uchar OPCODE :4;
uchar RSV3 :1;
uchar RSV2 :1;
uchar RSV1 :1;
uchar FIN :1;
uchar LENGTH :7;
uchar MASK :1;
} header;
char mask_key[SW_WEBSOCKET_MASK_LEN];
uint16_t header_length;
size_t payload_length;
char *payload;
} swWebSocket_frame;
enum swWebsocketCode
{
WEBSOCKET_OPCODE_CONTINUATION = 0x0,
WEBSOCKET_OPCODE_TEXT = 0x1,
WEBSOCKET_OPCODE_BINARY = 0x2,
WEBSOCKET_OPCODE_CLOSE = 0x8,
WEBSOCKET_OPCODE_PING = 0x9,
WEBSOCKET_OPCODE_PONG = 0xa,
WEBSOCKET_CLOSE_NORMAL = 1000,
WEBSOCKET_CLOSE_GOING_AWAY = 1001,
WEBSOCKET_CLOSE_PROTOCOL_ERROR = 1002,
WEBSOCKET_CLOSE_DATA_ERROR = 1003,
WEBSOCKET_CLOSE_STATUS_ERROR = 1005,
WEBSOCKET_CLOSE_ABNORMAL = 1006,
WEBSOCKET_CLOSE_MESSAGE_ERROR = 1007,
WEBSOCKET_CLOSE_POLICY_ERROR = 1008,
WEBSOCKET_CLOSE_MESSAGE_TOO_BIG = 1009,
WEBSOCKET_CLOSE_EXTENSION_MISSING = 1010,
WEBSOCKET_CLOSE_SERVER_ERROR = 1011,
WEBSOCKET_CLOSE_TLS = 1015,
WEBSOCKET_VERSION = 13,
};
void swWebSocket_encode(swString *buffer, const char *data, size_t length, char opcode, uint8_t finish, uint8_t mask);
void swWebSocket_decode(swWebSocket_frame *frame, swString *data);
int swWebSocket_pack_close_frame(swString *buffer, int code, char* reason, size_t length, uint8_t mask);
void swWebSocket_print_frame(swWebSocket_frame *frame);
ssize_t swWebSocket_get_package_length(swProtocol *protocol, swConnection *conn, char *data, uint32_t length);
int swWebSocket_dispatch_frame(swProtocol *protocol, swConnection *conn, char *data, uint32_t length);
#ifdef __cplusplus
}
#endif
#endif /* SW_WEBSOCKET_H_ */