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
|
#include "tcp_server.h"
#include <ctype.h>
#include <stdlib.h>
#include "freertos/semphr.h"
#include "lwip/sockets.h"
#include "param.h"
#include "sockbuf.h"
#include "station.h"
static int strcmp_(const char* s1, const char* s2)
{
if (!s1) {
return !s2;
}
if (!s2) {
return !s1;
}
return strcmp(s1, s2);
}
static void handle_set_cmd(
sockbuf_t* sockbuf, ws_params_t* ws_params, const char* key)
{
ws_state_t snapshot;
bool should_save = false;
if (xSemaphoreTake(ws_params->state_lock, portMAX_DELAY) != pdTRUE) {
sockbuf_write(sockbuf, "Failed to lock state\n");
return;
}
bool handled = false;
if (false) {
}
#define STATE_PARAM(ty, attr, ...) \
else if (!strcmp_(key, #attr)) \
{ \
cmd_handle(ty)(#attr, sockbuf, &ws_params->state.attr); \
handled = true; \
}
#include "state_params.i"
#undef STATE_PARAM
if (!handled) {
sockbuf_write(sockbuf, "Unknown attribute: '");
sockbuf_write(sockbuf, key);
sockbuf_write(sockbuf, "'\n");
} else {
snapshot = ws_params->state;
should_save = true;
}
xSemaphoreGive(ws_params->state_lock);
if (should_save) {
ws_state_save(&snapshot);
}
}
static void handle_print_cmd(sockbuf_t* sockbuf, ws_params_t* ws_params)
{
ws_state_t snapshot;
if (xSemaphoreTake(ws_params->state_lock, portMAX_DELAY) != pdTRUE) {
sockbuf_write(sockbuf, "Failed to lock state\n");
return;
}
snapshot = ws_params->state;
xSemaphoreGive(ws_params->state_lock);
#define STATE_PARAM(ty, attr, ...) print(ty)(sockbuf, #attr, snapshot.attr);
#include "state_params.i"
#undef STATE_PARAM
}
static void run_sockbuf(sockbuf_t* sockbuf, ws_params_t* ws_params)
{
char buf[128];
while (true) {
sockbuf_write(sockbuf, "\x1b[1;32mws2812b>\x1b[0m ");
char* cmd = read_token(sockbuf, buf, sizeof(buf));
if (!cmd) {
break;
}
if (!strcmp_(cmd, "set")) {
char* key = read_token(sockbuf, buf, sizeof(buf));
if (key) {
handle_set_cmd(sockbuf, ws_params, key);
}
} else if (!strcmp_(cmd, "print")) {
handle_print_cmd(sockbuf, ws_params);
} else if (!strcmp_(cmd, "end")) {
close(sockbuf->socket);
return;
} else {
sockbuf_write(sockbuf, "Unknown command: '");
sockbuf_write(sockbuf, cmd);
sockbuf_write(sockbuf, "'\n");
}
}
}
portTASK_FUNCTION(tcp_server, params)
{
ws_params_t* ws_params = (ws_params_t*)params;
int s = socket(AF_INET, SOCK_STREAM, 0);
struct sockaddr_in addr;
memset(&addr, 0, sizeof(addr));
addr.sin_port = htons(1234);
addr.sin_family = AF_INET;
addr.sin_addr.s_addr = INADDR_ANY;
int err = bind(s, (struct sockaddr*)&addr, sizeof(addr));
if (err) {
printf("Error binding to sockaddr: %d\n", err);
return;
}
printf("Listening ...\n");
listen(s, 1);
while (true) {
printf("Waiting for connection... \n");
struct sockaddr_in client_addr;
socklen_t size;
int sock = accept(s, (struct sockaddr*)&client_addr, &size);
printf("Accepted connection\n");
sockbuf_t sockbuf;
init_sockbuf(&sockbuf, sock);
run_sockbuf(&sockbuf, ws_params);
}
}
|
