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
|
/*
* Copyright (C) NGINX, Inc.
*/
#include <string.h>
#include <stdlib.h>
#include <nxt_unit.h>
#include <nxt_unit_request.h>
#include <nxt_clang.h>
#include <nxt_websocket.h>
#include <nxt_unit_websocket.h>
static void
ws_echo_request_handler(nxt_unit_request_info_t *req)
{
int rc;
const char *target;
rc = NXT_UNIT_OK;
target = nxt_unit_sptr_get(&req->request->target);
if (strcmp(target, "/") == 0) {
if (!nxt_unit_request_is_websocket_handshake(req)) {
goto notfound;
}
rc = nxt_unit_response_init(req, 101, 0, 0);
if (nxt_slow_path(rc != NXT_UNIT_OK)) {
goto fail;
}
nxt_unit_response_upgrade(req);
nxt_unit_response_send(req);
return;
}
notfound:
rc = nxt_unit_response_init(req, 404, 0, 0);
fail:
nxt_unit_request_done(req, rc);
}
static void
ws_echo_websocket_handler(nxt_unit_websocket_frame_t *ws)
{
uint8_t opcode;
ssize_t size;
nxt_unit_request_info_t *req;
static size_t buf_size = 0;
static uint8_t *buf = NULL;
if (buf_size < ws->content_length) {
buf = realloc(buf, ws->content_length);
buf_size = ws->content_length;
}
req = ws->req;
opcode = ws->header->opcode;
if (opcode == NXT_WEBSOCKET_OP_PONG) {
nxt_unit_websocket_done(ws);
return;
}
size = nxt_unit_websocket_read(ws, buf, ws->content_length);
nxt_unit_websocket_send(req, opcode, ws->header->fin, buf, size);
nxt_unit_websocket_done(ws);
if (opcode == NXT_WEBSOCKET_OP_CLOSE) {
nxt_unit_request_done(req, NXT_UNIT_OK);
}
}
int
main(void)
{
nxt_unit_ctx_t *ctx;
nxt_unit_init_t init;
memset(&init, 0, sizeof(nxt_unit_init_t));
init.callbacks.request_handler = ws_echo_request_handler;
init.callbacks.websocket_handler = ws_echo_websocket_handler;
ctx = nxt_unit_init(&init);
if (ctx == NULL) {
return 1;
}
nxt_unit_run(ctx);
nxt_unit_done(ctx);
return 0;
}
|
