/
sdcard_http.cpp
319 lines (261 loc) · 9.98 KB
/
sdcard_http.cpp
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
#include <stdio.h>
#include "pico/stdlib.h"
#include "hardware/uart.h"
#include "hardware/gpio.h"
#include "hardware/spi.h"
#include "pico_wireless.hpp"
#include "secrets.h"
#include "ff.h"
#include <vector>
#include <algorithm>
#define HTTP_PORT 80
#define HTTP_REQUEST_BUF_SIZE 2048
#define HTTP_RESPONSE_BUF_SIZE 1024 * 10
#define HTTP_RESPONSE_CHUNK_SIZE 1024u
#define DNS_CLOUDFLARE IPAddress(1, 1, 1, 1)
#define DNS_GOOGLE IPAddress(8, 8, 8, 8)
#define USE_DNS DNS_CLOUDFLARE
using namespace pimoroni;
FATFS fs;
FIL fil;
FRESULT fr;
PicoWireless wireless;
uint8_t r, g, b;
uint8_t request_buf[HTTP_REQUEST_BUF_SIZE];
uint8_t response_buf[HTTP_RESPONSE_BUF_SIZE];
const std::string response_501 = "HTTP/1.1 501 Not Implemented\nContent-Length: 19\n\n501 Not Implemented";
enum http_request_method_t {
GET,
POST
};
enum http_content_type_t {
TEXT_HTML,
IMAGE_SVG,
};
struct http_response_t {
std::string_view response_body;
int response_code;
http_content_type_t content_type = TEXT_HTML;
};
typedef http_response_t(*http_request_handler)(http_request_method_t method, std::string_view path, std::vector<std::string_view> request_head, std::vector<std::string_view> request_body);
std::vector<std::string_view> split(std::string_view str, std::string delim="\r\n") {
std::vector<std::string_view> result;
size_t offset = 0;
while (offset < str.size()) {
const auto pos = str.find_first_of(delim, offset);
// Emit an empty view even if two adjacent delimiters are found
// this ensurs the HTTP "blank line" start of content is found
result.emplace_back(str.substr(offset, pos - offset));
if (pos == std::string_view::npos) break;
offset = pos + delim.length();
}
return result;
}
uint32_t millis() {
return to_us_since_boot(get_absolute_time()) / 1000;
}
bool wifi_connect(std::string network, std::string password, IPAddress dns_server, uint32_t timeout=10000) {
printf("Connecting to %s...\n", network.c_str());
wireless.wifi_set_passphrase(network, password);
uint32_t t_start = millis();
while(millis() - t_start < timeout) {
if(wireless.get_connection_status() == WL_CONNECTED) {
printf("Connected!\n");
wireless.set_dns(1, dns_server, 0);
return true;
}
wireless.set_led(255, 0, 0);
sleep_ms(500);
wireless.set_led(0, 0, 0);
sleep_ms(500);
printf("...\n");
}
return false;
}
int socket_accept(int8_t server_sock) {
uint8_t new_sock = wireless.avail_server(server_sock);
if(new_sock != server_sock
&& new_sock != 255) {
return new_sock;
}
return -1;
}
bool has_suffix(std::string_view path, std::string_view suffix) {
return path.length() >= suffix.length() && path.compare(path.length() - suffix.length(), suffix.length(), suffix) == 0;
}
uint8_t start_server(uint16_t http_port) {
printf("Starting server...\n");
// Get a socket for our server
uint8_t server_sock = wireless.get_socket();
wireless.start_server(http_port, server_sock);
while(1){
uint8_t state = wireless.get_server_state(server_sock);
if(state == LISTEN) break;
}
IPAddress ip;
wireless.get_ip_address(ip);
printf("Server listening on %s:%i\n", ip.to_string().c_str(), http_port);
return server_sock;
}
bool poll_http_requests(uint8_t server_sock, http_request_handler request_handler, uint32_t timeout = 1000) {
uint16_t request_length = 0;
uint16_t avail_length = 0;
uint32_t t_start = millis();
// Clear the request buffer
memset(request_buf, 0, HTTP_REQUEST_BUF_SIZE);
// Try to accept an incoming connection
int client_sock = socket_accept(server_sock);
if(client_sock == -1) return false;
printf("Got client: %d\n", client_sock);
// Any data? This *usually* seems to be available immediately,
// so we can dispense with any ongoing polling and hope for the best.
avail_length = wireless.avail_data(client_sock);
if(avail_length == 0) return false;
printf("Got request: %i bytes\n", avail_length);
// Read the full response
// Sometimes the bytes read is less than the bytes we request, so loop until we get the data we expect
while(request_length < avail_length) {
uint16_t read_length = avail_length; // Request the full buffer
wireless.get_data_buf(client_sock, request_buf + request_length, &read_length);
request_length += read_length; // Increment the response_length by the amount we actually read
// Also check for timeouts here, too
if(millis() - t_start >= timeout) break;
}
// Bail if we timed out.
if(millis() - t_start >= timeout) {
wireless.stop_client(client_sock);
};
if(request_length > 0) {
std::vector<std::string_view> request = split(std::string_view((char *)request_buf, request_length));
std::vector<std::string_view> request_body;
std::vector<std::string_view> request_detail = split(request[0], " ");
http_request_method_t method;
// Bail early on an invalid HTTP request
if(request_detail[0].compare("POST") == 0) {
method = POST;
} else if (request_detail[0].compare("GET") == 0) {
method = GET;
} else { // PUT, DELETE, HEAD?
// Return a 501 unimplemented
wireless.send_data(client_sock, (const uint8_t *)response_501.data(), response_501.length());
wireless.stop_client(client_sock);
return false;
}
// Get the request path so we can handle routes
std::string_view path = request_detail[1];
// Scan for the blank line indicating content start for form post data
auto body_start = std::find(request.begin(), request.end(), "");
// Split the body from the head (ow!)
if(body_start != request.end()) {
request_body = std::vector<std::string_view>(body_start + 1, request.end());
request = std::vector<std::string_view>(request.begin(), body_start);
}
std::string response_head;
std::string content_type = "text/html";
http_response_t response = request_handler(method, path, request, request_body);
switch(response.content_type) {
case TEXT_HTML:
content_type = "text/html";
break;
case IMAGE_SVG:
content_type = "image/svg+xml";
break;
}
if(response.response_code == 200) {
response_head = "HTTP/1.1 200 OK\nContent-Length: " + std::to_string(response.response_body.length()) + "\nContent-Type: " + content_type + "\n\n";
} else { // Assume 404
response_head = "HTTP/1.1 404 File Not Found\nContent-Length: 22\nContent-Type: " + content_type + "\n\n";
}
wireless.send_data(client_sock, (const uint8_t *)response_head.data(), response_head.length());
uint16_t chunks = (response.response_body.length() / HTTP_RESPONSE_CHUNK_SIZE) + 1;
for(auto chunk = 0u; chunk < chunks; chunk++) {
uint16_t offset = chunk * HTTP_RESPONSE_CHUNK_SIZE;
uint16_t length = std::min(response.response_body.length() - offset, HTTP_RESPONSE_CHUNK_SIZE);
if(length == 0) break;
wireless.send_data(client_sock, (const uint8_t *)response.response_body.data() + offset, length);
sleep_ms(1);
}
wireless.stop_client(client_sock);
return true;
}
return false;
}
int main() {
stdio_init_all();
fr = f_mount(&fs, "", 1);
if (fr != FR_OK) {
printf("Failed to mount SD card, error: %d\n", fr);
return 0;
}
FILINFO file;
auto dir = new DIR();
printf("Listing /\n");
f_opendir(dir, "/");
while(f_readdir(dir, &file) == FR_OK && file.fname[0]) {
printf("%s %lld\n", file.fname, file.fsize);
}
f_closedir(dir);
wireless.init();
sleep_ms(500);
printf("Firmware version Nina %s\n", wireless.get_fw_version());
if(!wifi_connect(NETWORK, PASSWORD, USE_DNS)) {
return 0;
}
uint8_t server_sock = start_server(HTTP_PORT);
g = 255;
wireless.set_led(r, g, b);
while(1) {
// Handle any incoming HTTP requests
poll_http_requests(server_sock, [](
http_request_method_t request_method,
std::string_view request_path,
std::vector<std::string_view> request_head,
std::vector<std::string_view> request_body) -> http_response_t {
http_response_t response;
response.response_body = "Nothing to see here...";
response.response_code = 404;
printf("Request path: \"%s\"\n", std::string(request_path).c_str());
// Only serve HTML files!
if (has_suffix(request_path, ".html")) {
fr = f_open(&fil, std::string(request_path).c_str(), FA_READ);
if(fr == FR_OK) {
size_t read = 0;
f_read(&fil, response_buf, HTTP_RESPONSE_BUF_SIZE, &read);
f_close(&fil);
response.response_body = std::string_view((char*)response_buf, read);
response.response_code = 200;
}
// And, maybe SVG!?
} else if (has_suffix(request_path, ".svg")) {
fr = f_open(&fil, std::string(request_path).c_str(), FA_READ);
if(fr == FR_OK) {
size_t read = 0;
f_read(&fil, response_buf, HTTP_RESPONSE_BUF_SIZE, &read);
f_close(&fil);
response.response_body = std::string_view((char*)response_buf, read);
response.response_code = 200;
response.content_type = IMAGE_SVG;
}
/// Oooh! Directory listings...
} else if (has_suffix(request_path, "/")) {
FILINFO file;
auto dir = new DIR();
printf("Listing directory: %s\n", std::string(request_path).c_str());
f_opendir(dir, std::string(request_path).c_str());
uint16_t offset = 0u;
while(f_readdir(dir, &file) == FR_OK && file.fname[0]) {
std::string filename = std::string(file.fname);
if(!has_suffix(filename, ".svg") && !has_suffix(filename, ".html")) continue;
snprintf((char *)response_buf + offset, (size_t)HTTP_RESPONSE_BUF_SIZE - offset, "<li><a href=\"%s%s\">%s</a>", std::string(request_path).c_str(), file.fname, file.fname);
offset += filename.length() * 2 + 20 + request_path.length(); // Ugh
}
f_closedir(dir);
response.response_body = std::string_view((char*)response_buf, offset);
response.response_code = 200;
}
return response;
});
sleep_ms(10);
}
return 0;
}