summaryrefslogtreecommitdiff
path: root/addons/websocket/nx_websocket_client.h
blob: b06f9463b2333a6e9dd0e56d0108642a3b9f9899 (plain)
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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
/***************************************************************************
 * Copyright (c) 2024 Microsoft Corporation
 * Copyright (c) 2025-present Eclipse ThreadX Contributors
 *
 * This program and the accompanying materials are made available under the
 * terms of the MIT License which is available at
 * https://opensource.org/licenses/MIT.
 *
 * SPDX-License-Identifier: MIT
 **************************************************************************/


/**************************************************************************/
/**************************************************************************/
/**                                                                       */
/** NetX WebSocket Component                                              */
/**                                                                       */
/**   WebSocket Protocol                                                  */
/**                                                                       */
/**************************************************************************/
/**************************************************************************/


/**************************************************************************/
/*                                                                        */
/*  APPLICATION INTERFACE DEFINITION                       RELEASE        */
/*                                                                        */
/*    nx_websocket_client.h                               PORTABLE C      */
/*                                                           6.4.3        */
/*  AUTHOR                                                                */
/*                                                                        */
/*                                                                        */
/*                                                                        */
/*  DESCRIPTION                                                           */
/*                                                                        */
/*    This file defines the NetX WebSocket Protocol component, including  */
/*    all data types and external references.                             */
/*                                                                        */
/**************************************************************************/

#ifndef NX_WEBSOCKET_CLIENT_H
#define NX_WEBSOCKET_CLIENT_H

/* Determine if a C++ compiler is being used.  If so, ensure that standard
   C is used to process the API information.  */


#ifdef   __cplusplus

/* Yes, C++ compiler is present.  Use standard C.  */
extern   "C" {

#endif

#include "nx_api.h"
#include "nx_sha1.h"

#ifdef NX_SECURE_ENABLE
#include "nx_secure_tls_api.h"
#endif /* NX_SECURE_ENABLE */

#ifdef NX_DISABLE_PACKET_CHAIN
#error "NX_DISABLE_PACKET_CHAIN must not be defined"
#endif /* NX_DISABLE_PACKET_CHAIN */

/* Define the WebSocket ID.  */
#define NX_WEBSOCKET_CLIENT_ID                  0x57454253UL

/* Define the GUID size.  */
#define NX_WEBSOCKET_CLIENT_GUID_SIZE           16

/* Define the WebSocket Key size.  */
#define NX_WEBSOCKET_CLIENT_KEY_SIZE            26

/* WebSocket Header Format:
     0                   1                   2                   3
     0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
    +-+-+-+-+-------+-+-------------+-------------------------------+
    |F|R|R|R| opcode|M| Payload len |    Extended payload length    |
    |I|S|S|S|  (4)  |A|     (7)     |             (16)              |
    |N|V|V|V|       |S|             |                               |
    | |1|2|3|       |K|             |                               |
    +-+-+-+-+-------+-+-------------+ - - - - - - - - - - - - - - - +
    | Masking-key , if MASK set to 1                                |
    +-------------------------------- - - - - - - - - - - - - - - - +
    | Payload Data...                                               |
    +-------------------------------- - - - - - - - - - - - - - - - +
*/

/* First two bytes are required.  */

/* Define FIN.  */
#define NX_WEBSOCKET_FIN                            0x80
#define NX_WEBSOCKET_FIN_MASK                       0x80

/* Define the opcode.  */
#define NX_WEBSOCKET_OPCODE_CONTINUATION_FRAME      0x00
#define NX_WEBSOCKET_OPCODE_TEXT_FRAME              0x01
#define NX_WEBSOCKET_OPCODE_BINARY_FRAME            0x02
#define NX_WEBSOCKET_OPCODE_CONNECTION_CLOSE        0x08
#define NX_WEBSOCKET_OPCODE_PING                    0x09
#define NX_WEBSOCKET_OPCODE_PONG                    0x0A
#define NX_WEBSOCKET_OPCODE_MASK                    0x0F

/* Define the mask bit.  */
#define NX_WEBSOCKET_MASK                           0x80
#define NX_WEBSOCKET_MASK_MASK                      0x80

/* Define the payload length, always using 7bits + 16bits for payload length.  */
#define NX_WEBSOCKET_PAYLOAD_LEN_MASK               0x7F
#define NX_WEBSOCKET_PAYLOAD_LEN_16BITS             0x7E /* Payload length: 126, the following 2 bytes interpreted as a 16-bits unsigned integer are the payload length.  */
#define NX_WEBSOCKET_PAYLOAD_LEN_64BITS             0x7F /* Payload length: 127, the following 4 bytes interpreted as a 64-bits unsigned integer are the payload length.  */
#define NX_WEBSOCKET_EXTENDED_PAYLOAD_16BITS_SIZE   2
#define NX_WEBSOCKET_EXTENDED_PAYLOAD_64BITS_SIZE   8

/* Define the masking key size.  */
#define NX_WEBSOCKET_MASKING_KEY_SIZE               4

/* Define the basic header size (2 bytes normal header  + 4 bytes masking key)*/
#define NX_WEBSOCKET_HEADER_NORMAL_SIZE             6

/* Define the header size (2 bytes normal header  + 2 bytes extended payload length + 4 bytes masking key). 8 bytes extended payload length is not supported yet.  */
#define NX_WEBSOCKET_HEADER_SIZE                    8

/* Define the state.  */
#define NX_WEBSOCKET_CLIENT_STATE_INITIALIZE           0
#define NX_WEBSOCKET_CLIENT_STATE_IDLE                 1
#define NX_WEBSOCKET_CLIENT_STATE_CONNECTING           2
#define NX_WEBSOCKET_CLIENT_STATE_CONNECTED            3
#define NX_WEBSOCKET_CLIENT_STATE_DISCONNECT_SENT      4
#define NX_WEBSOCKET_CLIENT_STATE_DISCONNECT_RECEIVED  5

/* Define the return value.  */
#define NX_WEBSOCKET_SUCCESS                        0x0
#define NX_WEBSOCKET_ERROR                          0x30001
#define NX_WEBSOCKET_CONNECTING                     0x30002
#define NX_WEBSOCKET_ALREADY_CONNECTED              0x30003
#define NX_WEBSOCKET_NOT_CONNECTED                  0x30004
#define NX_WEBSOCKET_DATA_APPEND_FAILURE            0x30005
#define NX_WEBSOCKET_INVALID_STATE                  0x30006
#define NX_WEBSOCKET_INVALID_PACKET                 0x30007
#define NX_WEBSOCKET_INVALID_STATUS_CODE            0x30008
#define NX_WEBSOCKET_DISCONNECTED                   0x30009

/* Define the websocket Client data structure.  */
typedef struct NX_WEBSOCKET_CLIENT_STRUCT
{

    /* WebSocket Client ID.  */
    ULONG                       nx_websocket_client_id;

    /* Name of this WebSocket Client.  */
    UCHAR                      *nx_websocket_client_name;

    /* Pointer to associated IP structure.  */
    NX_IP                      *nx_websocket_client_ip_ptr;

    /* Pointer to WebSocket Client packet pool.  */
    NX_PACKET_POOL             *nx_websocket_client_packet_pool_ptr;

    /* State.  */
    UINT                        nx_websocket_client_state;

    /* Pointer to WebSocket Client TCP socket.  */
    NX_TCP_SOCKET              *nx_websocket_client_socket_ptr;

#ifdef NX_SECURE_ENABLE
    UINT                        nx_websocket_client_use_tls;
    NX_SECURE_TLS_SESSION      *nx_websocket_client_tls_session_ptr;
#endif

    /* Pointer to the received packet to be processed. This packet may be composited by more than one tcp/tls packet. */
    NX_PACKET                  *nx_websocket_client_processing_packet;

    /* Globally Unique Identifier.  */
    UCHAR                       nx_websocket_client_guid[NX_WEBSOCKET_CLIENT_GUID_SIZE];

    /* Protocol Name and length */
    UCHAR                      *nx_websocket_client_subprotocol;
    UINT                        nx_websocket_client_subprotocol_length;

    /* WebSocket-Key.  */
    UINT                        nx_websocket_client_key_size;
    UCHAR                       nx_websocket_client_key[NX_WEBSOCKET_CLIENT_KEY_SIZE];

    /* Websocket frame header parse context */
    UCHAR                       nx_websocket_client_frame_header_found;
    UCHAR                       nx_websocket_client_frame_fragmented;
    UCHAR                       nx_websocket_client_frame_opcode;
    UCHAR                       nx_websocket_client_frame_masked;
    UCHAR                       nx_websocket_client_frame_masking_key[4];
    ULONG                       nx_websocket_client_frame_data_length;
    ULONG                       nx_websocket_client_frame_data_received;

    /* SHA1 in connect response calculation for Sec-Protocol-Accept field */
    NX_SHA1                     nx_websocket_client_sha1;

    /* Connection status callback function.  */
    VOID                        (*nx_websocket_client_connection_status_callback)(struct NX_WEBSOCKET_CLIENT_STRUCT *, VOID *, UINT);

    /* Pointer to an argument passed to connection status callback.  */
    VOID                       *nx_websocket_client_connection_context;

    /* Define the websocket protect purpose mutex */
    TX_MUTEX                    nx_websocket_client_mutex;

} NX_WEBSOCKET_CLIENT;



#ifndef NX_WEBSOCKET_CLIENT_SOURCE_CODE

/* Application caller is present, perform API mapping.  */

/* Determine if error checking is desired.  If so, map API functions
   to the appropriate error checking front-ends.  Otherwise, map API
   functions to the core functions that actually perform the work.
   Note: error checking is enabled by default.  */

#ifdef NX_DISABLE_ERROR_CHECKING

/* Services without error checking.  */

#define nx_websocket_client_create                          _nx_websocket_client_create
#define nx_websocket_client_delete                          _nx_websocket_client_delete
#define nx_websocket_client_connect                         _nx_websocket_client_connect
#ifdef NX_SECURE_ENABLE
#define nx_websocket_client_secure_connect                  _nx_websocket_client_secure_connect
#endif /* NX_SECURE_ENABLE */
#define nx_websocket_client_disconnect                      _nx_websocket_client_disconnect
#define nx_websocket_client_packet_allocate                 _nx_websocket_client_packet_allocate
#define nx_websocket_client_send                            _nx_websocket_client_send
#define nx_websocket_client_receive                         _nx_websocket_client_receive
#define nx_websocket_client_connection_status_callback_set  _nx_websocket_client_connection_status_callback_set

#else

/* Services with error checking.  */

#define nx_websocket_client_create                          _nxe_websocket_client_create
#define nx_websocket_client_delete                          _nxe_websocket_client_delete
#define nx_websocket_client_connect                         _nxe_websocket_client_connect
#ifdef NX_SECURE_ENABLE
#define nx_websocket_client_secure_connect                  _nxe_websocket_client_secure_connect
#endif /* NX_SECURE_ENABLE */
#define nx_websocket_client_disconnect                      _nxe_websocket_client_disconnect
#define nx_websocket_client_packet_allocate                 _nxe_websocket_client_packet_allocate
#define nx_websocket_client_send                            _nxe_websocket_client_send
#define nx_websocket_client_receive                         _nxe_websocket_client_receive
#define nx_websocket_client_connection_status_callback_set  _nxe_websocket_client_connection_status_callback_set

#endif  /* NX_DISABLE_ERROR_CHECKING */

/* Define the prototypes accessible to the application software.  */

UINT  nx_websocket_client_create(NX_WEBSOCKET_CLIENT *client_ptr, UCHAR *client_name, NX_IP *ip_ptr, NX_PACKET_POOL *pool_ptr);
UINT  nx_websocket_client_delete(NX_WEBSOCKET_CLIENT *client_ptr);
UINT  nx_websocket_client_packet_allocate(NX_WEBSOCKET_CLIENT *client_ptr, NX_PACKET **packet_ptr, ULONG wait_option);
UINT  nx_websocket_client_connect(NX_WEBSOCKET_CLIENT *client_ptr, NX_TCP_SOCKET *socket_ptr,
                                  UCHAR *host, UINT host_length,
                                  UCHAR *uri_path, UINT uri_path_length,
                                  UCHAR *protocol, UINT protocol_length,
                                  UCHAR *bearer, UINT bearer_length, UINT wait_option);
#ifdef NX_SECURE_ENABLE
UINT  nx_websocket_client_secure_connect(NX_WEBSOCKET_CLIENT *client_ptr, NX_SECURE_TLS_SESSION *tls_session,
                                         UCHAR *host, UINT host_length,
                                         UCHAR *uri_path, UINT uri_path_length,
                                         UCHAR *protocol, UINT protocol_length,
                                         UCHAR *bearer, UINT bearer_length, UINT wait_option);
#endif /* NX_SECURE_ENABLE */
UINT  nx_websocket_client_disconnect(NX_WEBSOCKET_CLIENT *client_ptr, UINT wait_option);
UINT  nx_websocket_client_send(NX_WEBSOCKET_CLIENT *client_ptr, NX_PACKET *packet_ptr, UINT code, UINT is_final, UINT wait_option);
UINT  nx_websocket_client_receive(NX_WEBSOCKET_CLIENT *client_ptr, NX_PACKET **packet_ptr, UINT *code, UINT wait_option);
UINT  nx_websocket_client_connection_status_callback_set(NX_WEBSOCKET_CLIENT *client_ptr, VOID *context,
                                                         VOID (*connection_status_callback)(NX_WEBSOCKET_CLIENT *, VOID *, UINT));

#else

/* Websocket source code is being compiled, do not perform any API mapping.  */

UINT  _nxe_websocket_client_create(NX_WEBSOCKET_CLIENT *client_ptr, UCHAR *client_name, NX_IP *ip_ptr, NX_PACKET_POOL *pool_ptr);
UINT  _nx_websocket_client_create(NX_WEBSOCKET_CLIENT *client_ptr, UCHAR *client_name, NX_IP *ip_ptr, NX_PACKET_POOL *pool_ptr);
UINT  _nxe_websocket_client_delete(NX_WEBSOCKET_CLIENT *client_ptr);
UINT  _nx_websocket_client_delete(NX_WEBSOCKET_CLIENT *client_ptr);
UINT  _nxe_websocket_client_packet_allocate(NX_WEBSOCKET_CLIENT *client_ptr, NX_PACKET **packet_ptr, ULONG wait_option);
UINT  _nx_websocket_client_packet_allocate(NX_WEBSOCKET_CLIENT *client_ptr, NX_PACKET **packet_ptr, ULONG wait_option);
UINT  _nxe_websocket_client_connect(NX_WEBSOCKET_CLIENT *client_ptr, NX_TCP_SOCKET *socket_ptr,
                                    UCHAR *host, UINT host_length,
                                    UCHAR *uri_path, UINT uri_path_length,
                                    UCHAR *protocol, UINT protocol_length,
                                    UCHAR *bearer, UINT bearer_length, UINT wait_option);
UINT  _nx_websocket_client_connect(NX_WEBSOCKET_CLIENT *client_ptr, NX_TCP_SOCKET *socket_ptr,
                                   UCHAR *host, UINT host_length,
                                   UCHAR *uri_path, UINT uri_path_length,
                                   UCHAR *protocol, UINT protocol_length,
                                   UCHAR *bearer, UINT bearer_length, UINT wait_option);
#ifdef NX_SECURE_ENABLE
UINT  _nxe_websocket_client_secure_connect(NX_WEBSOCKET_CLIENT *client_ptr, NX_SECURE_TLS_SESSION *tls_session,
                                           UCHAR *host, UINT host_length,
                                           UCHAR *uri_path, UINT uri_path_length,
                                           UCHAR *protocol, UINT protocol_length,
                                           UCHAR *bearer, UINT bearer_length, UINT wait_option);
UINT  _nx_websocket_client_secure_connect(NX_WEBSOCKET_CLIENT *client_ptr, NX_SECURE_TLS_SESSION *tls_session,
                                          UCHAR *host, UINT host_length,
                                          UCHAR *uri_path, UINT uri_path_length,
                                          UCHAR *protocol, UINT protocol_length,
                                          UCHAR *bearer, UINT bearer_length, UINT wait_option);
#endif /* NX_SECURE_ENABLE */
UINT  _nxe_websocket_client_disconnect(NX_WEBSOCKET_CLIENT *client_ptr, UINT wait_option);
UINT  _nx_websocket_client_disconnect(NX_WEBSOCKET_CLIENT *client_ptr, UINT wait_option);
UINT  _nxe_websocket_client_send(NX_WEBSOCKET_CLIENT *client_ptr, NX_PACKET *packet_ptr, UINT code, UINT is_final, UINT wait_option);
UINT  _nx_websocket_client_send(NX_WEBSOCKET_CLIENT *client_ptr, NX_PACKET *packet_ptr, UINT code, UINT is_final, UINT wait_option);
UINT  _nxe_websocket_client_receive(NX_WEBSOCKET_CLIENT *client_ptr, NX_PACKET **packet_ptr, UINT *code, UINT wait_option);
UINT  _nx_websocket_client_receive(NX_WEBSOCKET_CLIENT *client_ptr, NX_PACKET **packet_ptr, UINT *code, UINT wait_option);
UINT  _nxe_websocket_client_connection_status_callback_set(NX_WEBSOCKET_CLIENT *client_ptr, VOID * context,
                                                           VOID (*connection_status_callback)(NX_WEBSOCKET_CLIENT *, VOID *, UINT));
UINT  _nx_websocket_client_connection_status_callback_set(NX_WEBSOCKET_CLIENT *client_ptr, VOID * context,
                                                          VOID (*connection_status_callback)(NX_WEBSOCKET_CLIENT *, VOID *, UINT));

#endif  /* NX_WEBSOCKET_CLIENT_SOURCE_CODE */

/* Define internal websocket functions.  */
UINT  _nx_websocket_client_connect_internal(NX_WEBSOCKET_CLIENT *client_ptr,
                                            UCHAR *host, UINT host_length,
                                            UCHAR *uri_path, UINT uri_path_length,
                                            UCHAR *protocol, UINT protocol_length,
                                            UCHAR *bearer, UINT bearer_length, UINT wait_option);
UINT  _nx_websocket_client_name_compare(UCHAR *src, ULONG src_length, UCHAR *dest, ULONG dest_length);
UINT  _nx_websocket_client_connect_response_process(NX_WEBSOCKET_CLIENT *client_ptr, NX_PACKET *packet_ptr);
UINT  _nx_websocket_client_packet_trim(NX_WEBSOCKET_CLIENT *client_ptr, NX_PACKET **packet_ptr, ULONG trim_size);
UINT  _nx_websocket_client_packet_send(NX_WEBSOCKET_CLIENT *client_ptr, NX_PACKET *packet_ptr, ULONG wait_option);
UINT  _nx_websocket_client_packet_receive(NX_WEBSOCKET_CLIENT *client_ptr, NX_PACKET **packet_ptr, ULONG wait_option);
UINT  _nx_websocket_client_data_process(NX_WEBSOCKET_CLIENT *client_ptr, NX_PACKET **packet_ptr, UINT *code);
UINT  _nx_websocket_client_connect_response_check(NX_WEBSOCKET_CLIENT *client_ptr, NX_PACKET *packet_ptr, UINT wait_option);
void  _nx_websocket_client_cleanup(NX_WEBSOCKET_CLIENT *client_ptr);

/* Determine if a C++ compiler is being used.  If so, complete the standard
   C conditional started above.  */
#ifdef   __cplusplus
        }
#endif

#endif  /* NX_WEBSOCKET_CLIENT_H */