WiFiStation at %s

/* * WiFiStation * Copyright (c) 2021 joshua stein * * Permission to use, copy, modify, and distribute this software for any * purpose with or without fee is hereby granted, provided that the above * copyright notice and this permission notice appear in all copies. * * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. */ #include #include "wifistation.h" std::unique_ptr http = NULL; static const char html_wrapper[] PROGMEM = R"END( WiFiStation at %s

WiFiStation %s

%s )END"; static unsigned int upload_size = 0; static unsigned int delivered_bytes = 0; void http_process(void) { if (!settings->http_server) return; http->handleClient(); } void http_send_result(int status, bool include_home, const char *body, ...) { va_list arg; const char *accept = http->header("Accept").c_str(); int len; char *doc = NULL; /* expand body format */ va_start(arg, body); len = vasprintf(&doc, body, arg); va_end(arg); if (len == -1) goto fail; /* if Accept header starts with text/html, client prefers html */ if (accept && strncmp(accept, "text/html", 9) == 0) { char *tmp; /* append home link to body */ if (include_home) { len = asprintf(&tmp, "%s" "

" "Home", doc); if (len == -1) goto fail; free(doc); doc = tmp; } /* insert body into html template */ len = asprintf(&tmp, html_wrapper, WiFi.localIP().toString().c_str(), WIFISTATION_VERSION, doc); if (len == -1) goto fail; free(doc); doc = tmp; http->send(status, "text/html", doc); } else { /* append newline since this is probably going to a terminal */ doc = (char *)realloc(doc, len + 2); doc[len] = '\n'; doc[len + 1] = '\0'; http->send(status, "text/plain", doc); } free(doc); return; fail: if (doc != NULL) free(doc); http->send(500, "text/plain", "out of memory :("); return; } void http_setup(void) { if (settings->http_server) { if (http) return; http.reset(new ESP8266WebServer(80)); } else { if (http) http = NULL; return; } http->collectHeaders(F("Accept")); http->on("/", HTTP_GET, []() { http_send_result(200, false, R"END(

)END", MAX_UPLOAD_SIZE, MAX_UPLOAD_SIZE); }); /* * This measure step is because we don't get the total size of the * upload until we read the whole thing, but we need to send the file * size to the MailStation before sending any data. So to avoid * caching the entire upload, we process the upload and just count the * bytes being passed through, then send a 307 redirect to the actual * upload endpoint with the total size. Browsers follow a 307 with a * POST, so at /upload we'll have the actual size before it sends us * the same data. */ http->on("/upload", HTTP_POST, []() { char tmp[32]; if (upload_size == 0) { http_send_result(400, true, "Failed receiving file (size 0)."); } else { snprintf(tmp, sizeof(tmp), "/upload_measured?size=%d", upload_size); http->sendHeader("Location", tmp); http->send(307, "text/plain", "Ok, follow me!"); } }, []() { HTTPUpload& upload = http->upload(); switch (upload.status) { case UPLOAD_FILE_START: upload_size = 0; break; case UPLOAD_FILE_WRITE: upload_size += upload.currentSize; break; case UPLOAD_FILE_ABORTED: upload_size = 0; /* FALLTHROUGH */ case UPLOAD_FILE_END: if (upload_size == 0) { http_send_result(400, true, "Failed receiving file (size 0)."); return; } if (upload_size > MAX_UPLOAD_SIZE) { http_send_result(400, true, "File upload cannot be larger than %d " "bytes.", MAX_UPLOAD_SIZE); upload_size = 0; return; } break; } }); http->on("/upload_measured", HTTP_POST, []() { http_send_result(200, true, "Successfully uploaded %d byte%s to MailStation.", delivered_bytes, delivered_bytes == 1 ? "" : "s"); }, []() { HTTPUpload& upload = http->upload(); switch (upload.status) { case UPLOAD_FILE_START: delivered_bytes = 0; if (!(upload_size = atoi(http->arg("size").c_str()))) { http_send_result(400, true, "No size " "parameter passed. Perhaps your browser " "failed to follow the 307 redirect " "properly."); return; } if (upload_size > MAX_UPLOAD_SIZE) { http_send_result(400, true, "File upload cannot be larger than %d " "bytes.", MAX_UPLOAD_SIZE); return; } if (ms_write(upload_size & 0xff) != 0 || ms_write((upload_size >> 8) & 0xff) != 0) { http_send_result(400, true, "Failed sending size bytes " "(0x%x, 0x%x) to " "MailStation. Is the WSLoader program " "running?", upload_size & 0xff, (upload_size >> 8) & 0xff); return; } break; case UPLOAD_FILE_WRITE: for (int i = 0; i < (int)upload.currentSize; i++) { delivered_bytes++; if (ms_write(upload.buf[i]) == -1) { http_send_result(400, true, "Failed uploading to MailStation " "at byte %d/%d.", delivered_bytes, upload_size); return; } yield(); delayMicroseconds(500); } break; case UPLOAD_FILE_END: break; case UPLOAD_FILE_ABORTED: http_send_result(400, true, "Aborted upload to MailStation."); return; } }); http->onNotFound([]() { http_send_result(404, true, "404"); }); http->begin(); }