diff --git a/README.md b/README.md index 8edd8d5..dc27cfc 100644 --- a/README.md +++ b/README.md @@ -274,8 +274,9 @@ In a browser navigate to the IP address assigned to the ESP32. #### Credentials -- User: Pick a username to enable HTTP Basic authentication for the Web Configuration, Set to "#" to disable authentication. -- Password/Retype password: Pick a password to enable HTTP Basic authentication for the Web Configuration. +- User: Pick a username to enable HTTP authentication for the Web Configuration, Set to "#" to disable authentication. +- Password/Retype password: Pick a password to enable HTTP authentication for the Web Configuration. +- Use Digest Authentication (more secure): Enable to use HTTP Digest Authentication instead of HTTP Basic Authentication. Digest authentication is more secure, especially over unencrypted (HTTP) connections. #### Nuki Lock PIN / Nuki Opener PIN diff --git a/src/PreferencesKeys.h b/src/PreferencesKeys.h index f1b961a..e172907 100644 --- a/src/PreferencesKeys.h +++ b/src/PreferencesKeys.h @@ -67,6 +67,7 @@ #define preference_debug_hex_data (char*)"dbgHexData" #define preference_debug_command (char*)"dbgCommand" #define preference_connect_mode (char*)"nukiConnMode" +#define preference_http_auth_type (char*)"httpdAuthType" // CHANGE DOES NOT REQUIRE REBOOT TO TAKE EFFECT #define preference_find_best_rssi (char*)"nwbestrssi" @@ -222,8 +223,9 @@ inline void initPreferences(Preferences* preferences) preferences->putBool(preference_debug_hex_data, false); preferences->putBool(preference_debug_command, false); preferences->putBool(preference_connect_mode, true); - + preferences->putBool(preference_http_auth_type, false); preferences->putBool(preference_retain_gpio, false); + #ifndef CONFIG_IDF_TARGET_ESP32H2 WiFi.begin(); @@ -370,7 +372,7 @@ private: preference_opener_continuous_mode, preference_lock_max_keypad_code_count, preference_opener_max_keypad_code_count, preference_lock_max_timecontrol_entry_count, preference_opener_max_timecontrol_entry_count, preference_enable_bootloop_reset, preference_mqtt_ca, preference_mqtt_crt, preference_mqtt_key, preference_mqtt_hass_discovery, preference_mqtt_hass_cu_url, preference_buffer_size, preference_ip_dhcp_enabled, preference_ip_address, - preference_ip_subnet, preference_ip_gateway, preference_ip_dns_server, preference_network_hardware, + preference_ip_subnet, preference_ip_gateway, preference_ip_dns_server, preference_network_hardware, preference_http_auth_type, preference_rssi_publish_interval, preference_hostname, preference_network_timeout, preference_restart_on_disconnect, preference_restart_ble_beacon_lost, preference_query_interval_lockstate, preference_timecontrol_topic_per_entry, preference_keypad_topic_per_entry, preference_query_interval_configuration, preference_query_interval_battery, preference_query_interval_keypad, preference_keypad_control_enabled, @@ -402,7 +404,7 @@ private: preference_publish_authdata, preference_publish_debug_info, preference_official_hybrid_enabled, preference_mqtt_hass_enabled, preference_retain_gpio, preference_official_hybrid_actions, preference_official_hybrid_retry, preference_conf_info_enabled, preference_disable_non_json, preference_update_from_mqtt, preference_auth_control_enabled, preference_auth_topic_per_entry, preference_auth_info_enabled, preference_webserial_enabled, preference_hass_device_discovery, - preference_ntw_reconfigure, preference_keypad_check_code_enabled, preference_disable_network_not_connected, preference_find_best_rssi, + preference_ntw_reconfigure, preference_keypad_check_code_enabled, preference_disable_network_not_connected, preference_find_best_rssi, preference_http_auth_type, preference_debug_connect, preference_debug_communication, preference_debug_readable_data, preference_debug_hex_data, preference_debug_command, preference_connect_mode, preference_lock_force_id, preference_lock_force_doorsensor, preference_lock_force_keypad, preference_opener_force_id, preference_opener_force_keypad }; diff --git a/src/WebCfgServer.cpp b/src/WebCfgServer.cpp index ebf4d31..ef6ad57 100644 --- a/src/WebCfgServer.cpp +++ b/src/WebCfgServer.cpp @@ -75,64 +75,80 @@ WebCfgServer::WebCfgServer(NukiNetwork* network, Preferences* preferences, bool void WebCfgServer::initialize() { - _psychicServer->on("/", HTTP_GET, [&](PsychicRequest *request) + _psychicServer->onOpen([&](PsychicClient* client) { Log->printf("[http] connection #%u connected from %s\n", client->socket(), client->localIP().toString().c_str()); }); + _psychicServer->onClose([&](PsychicClient* client) { Log->printf("[http] connection #%u closed from %s\n", client->socket(), client->localIP().toString().c_str()); }); + + HTTPAuthMethod auth_type = BASIC_AUTH; + if (_preferences->getBool(preference_http_auth_type, false)) + { + auth_type = DIGEST_AUTH; + } + + _psychicServer->on("/", HTTP_GET, [&](PsychicRequest *request, PsychicResponse* resp) { if(strlen(_credUser) > 0 && strlen(_credPassword) > 0 && !request->authenticate(_credUser, _credPassword)) { - return request->requestAuthentication(BASIC_AUTH, "Nuki Hub", "You must log in."); + return request->requestAuthentication(auth_type, "Nuki Hub", "You must log in."); } + +#ifndef CONFIG_IDF_TARGET_ESP32H2 if(!_network->isApOpen()) { -#ifndef NUKI_HUB_UPDATER - return buildHtml(request); -#else - return buildOtaHtml(request); #endif - } +#ifndef NUKI_HUB_UPDATER + return buildHtml(request, resp); +#else + return buildOtaHtml(request, resp); +#endif #ifndef CONFIG_IDF_TARGET_ESP32H2 + } else { - return buildWifiConnectHtml(request); + return buildWifiConnectHtml(request, resp); } #endif }); - _psychicServer->on("/style.css", HTTP_GET, [&](PsychicRequest *request) + _psychicServer->on("/style.css", HTTP_GET, [&](PsychicRequest *request, PsychicResponse* resp) { if(strlen(_credUser) > 0 && strlen(_credPassword) > 0 && !request->authenticate(_credUser, _credPassword)) { - return request->requestAuthentication(BASIC_AUTH, "Nuki Hub", "You must log in."); + return request->requestAuthentication(auth_type, "Nuki Hub", "You must log in."); } - return sendCss(request); + + return sendCss(request, resp); }); - _psychicServer->on("/favicon.ico", HTTP_GET, [&](PsychicRequest *request) + _psychicServer->on("/favicon.ico", HTTP_GET, [&](PsychicRequest *request, PsychicResponse* resp) { if(strlen(_credUser) > 0 && strlen(_credPassword) > 0 && !request->authenticate(_credUser, _credPassword)) { - return request->requestAuthentication(BASIC_AUTH, "Nuki Hub", "You must log in."); + return request->requestAuthentication(auth_type, "Nuki Hub", "You must log in."); } - return sendFavicon(request); + + return sendFavicon(request, resp); }); if(_network->isApOpen()) { #ifndef CONFIG_IDF_TARGET_ESP32H2 - _psychicServer->on("/ssidlist", HTTP_GET, [&](PsychicRequest *request) + _psychicServer->on("/ssidlist", HTTP_GET, [&](PsychicRequest *request, PsychicResponse* resp) { if(strlen(_credUser) > 0 && strlen(_credPassword) > 0 && !request->authenticate(_credUser, _credPassword)) { - return request->requestAuthentication(BASIC_AUTH, "Nuki Hub", "You must log in."); + return request->requestAuthentication(auth_type, "Nuki Hub", "You must log in."); } - return buildSSIDListHtml(request); + + return buildSSIDListHtml(request, resp); }); - _psychicServer->on("/savewifi", HTTP_POST, [&](PsychicRequest *request) + _psychicServer->on("/savewifi", HTTP_POST, [&](PsychicRequest *request, PsychicResponse* resp) { if(strlen(_credUser) > 0 && strlen(_credPassword) > 0 && !request->authenticate(_credUser, _credPassword)) { - return request->requestAuthentication(BASIC_AUTH, "Nuki Hub", "You must log in."); + return request->requestAuthentication(auth_type, "Nuki Hub", "You must log in."); } + String message = ""; - bool connected = processWiFi(request, message); - esp_err_t res = buildConfirmHtml(request, message, 10, true); + bool connected = processWiFi(request, resp, message); + esp_err_t res = buildConfirmHtml(request, resp, message, 10, true); if(connected) { @@ -142,11 +158,11 @@ void WebCfgServer::initialize() } return res; }); - _psychicServer->on("/reboot", HTTP_GET, [&](PsychicRequest *request) + _psychicServer->on("/reboot", HTTP_GET, [&](PsychicRequest *request, PsychicResponse* resp) { if(strlen(_credUser) > 0 && strlen(_credPassword) > 0 && !request->authenticate(_credUser, _credPassword)) { - return request->requestAuthentication(BASIC_AUTH, "Nuki Hub", "You must log in."); + return request->requestAuthentication(auth_type, "Nuki Hub", "You must log in."); } String value = ""; @@ -160,14 +176,14 @@ void WebCfgServer::initialize() } else { - return buildConfirmHtml(request, "No confirm code set.", 3, true); + return buildConfirmHtml(request, resp, "No confirm code set.", 3, true); } if(value != _confirmCode) { - return request->redirect("/"); + return resp->redirect("/"); } - esp_err_t res = buildConfirmHtml(request, "Rebooting...", 2, true); + esp_err_t res = buildConfirmHtml(request, resp, "Rebooting...", 2, true); waitAndProcess(true, 1000); restartEsp(RestartReason::RequestedViaWebServer); return res; @@ -176,11 +192,11 @@ void WebCfgServer::initialize() } else { - _psychicServer->on("/get", HTTP_GET, [&](PsychicRequest *request) + _psychicServer->on("/get", HTTP_GET, [&](PsychicRequest *request, PsychicResponse* resp) { if(strlen(_credUser) > 0 && strlen(_credPassword) > 0 && !request->authenticate(_credUser, _credPassword)) { - return request->requestAuthentication(BASIC_AUTH, "Nuki Hub", "You must log in."); + return request->requestAuthentication(auth_type, "Nuki Hub", "You must log in."); } String value = ""; @@ -206,14 +222,14 @@ void WebCfgServer::initialize() } else { - return buildConfirmHtml(request, "No confirm code set.", 3, true); + return buildConfirmHtml(request, resp, "No confirm code set.", 3, true); } if(value != _confirmCode) { - return request->redirect("/"); + return resp->redirect("/"); } - esp_err_t res = buildConfirmHtml(request, "Rebooting...", 2, true); + esp_err_t res = buildConfirmHtml(request, resp, "Rebooting...", 2, true); waitAndProcess(true, 1000); restartEsp(RestartReason::RequestedViaWebServer); return res; @@ -221,78 +237,78 @@ void WebCfgServer::initialize() #ifndef NUKI_HUB_UPDATER else if (value == "info") { - return buildInfoHtml(request); + return buildInfoHtml(request, resp); } else if (value == "debugon") { _preferences->putBool(preference_publish_debug_info, true); - return buildConfirmHtml(request, "Debug On", 3, true); + return buildConfirmHtml(request, resp, "Debug On", 3, true); } else if (value == "debugoff") { _preferences->putBool(preference_publish_debug_info, false); - return buildConfirmHtml(request, "Debug Off", 3, true); + return buildConfirmHtml(request, resp, "Debug Off", 3, true); } else if (value == "export") { - return sendSettings(request); + return sendSettings(request, resp); } else if (value == "impexpcfg") { - return buildImportExportHtml(request); + return buildImportExportHtml(request, resp); } else if (value == "status") { - return buildStatusHtml(request); + return buildStatusHtml(request, resp); } else if (value == "acclvl") { - return buildAccLvlHtml(request); + return buildAccLvlHtml(request, resp); } else if (value == "custntw") { - return buildCustomNetworkConfigHtml(request); + return buildCustomNetworkConfigHtml(request, resp); } else if (value == "advanced") { - return buildAdvancedConfigHtml(request); + return buildAdvancedConfigHtml(request, resp); } else if (value == "cred") { - return buildCredHtml(request); + return buildCredHtml(request, resp); } else if (value == "ntwconfig") { - return buildNetworkConfigHtml(request); + return buildNetworkConfigHtml(request, resp); } else if (value == "mqttconfig") { - return buildMqttConfigHtml(request); + return buildMqttConfigHtml(request, resp); } else if (value == "mqttcaconfig") { - return buildMqttSSLConfigHtml(request, 0); + return buildMqttSSLConfigHtml(request, resp, 0); } else if (value == "mqttcrtconfig") { - return buildMqttSSLConfigHtml(request, 1); + return buildMqttSSLConfigHtml(request, resp, 1); } else if (value == "mqttkeyconfig") { - return buildMqttSSLConfigHtml(request, 2); + return buildMqttSSLConfigHtml(request, resp, 2); } else if (value == "nukicfg") { - return buildNukiConfigHtml(request); + return buildNukiConfigHtml(request, resp); } else if (value == "gpiocfg") { - return buildGpioConfigHtml(request); + return buildGpioConfigHtml(request, resp); } #ifndef CONFIG_IDF_TARGET_ESP32H2 else if (value == "wifi") { - return buildConfigureWifiHtml(request); + return buildConfigureWifiHtml(request, resp); } else if (value == "wifimanager") { @@ -307,17 +323,17 @@ void WebCfgServer::initialize() } else { - return buildConfirmHtml(request, "No confirm code set.", 3, true); + return buildConfirmHtml(request, resp, "No confirm code set.", 3, true); } if(value != _confirmCode) { - return request->redirect("/"); + return resp->redirect("/"); } if(!_allowRestartToPortal) { - return buildConfirmHtml(request, "Can't reset WiFi when network device is Ethernet", 3, true); + return buildConfirmHtml(request, resp, "Can't reset WiFi when network device is Ethernet", 3, true); } - esp_err_t res = buildConfirmHtml(request, "Restarting. Connect to ESP access point (\"NukiHub\" with password \"NukiHubESP32\") to reconfigure Wi-Fi.", 0); + esp_err_t res = buildConfirmHtml(request, resp, "Restarting. Connect to ESP access point (\"NukiHub\" with password \"NukiHubESP32\") to reconfigure Wi-Fi.", 0); waitAndProcess(false, 1000); _network->reconfigureDevice(); return res; @@ -326,11 +342,11 @@ void WebCfgServer::initialize() #endif else if (value == "ota") { - return buildOtaHtml(request); + return buildOtaHtml(request, resp); } else if (value == "otadebug") { - return buildOtaHtml(request, true); + return buildOtaHtml(request, resp, true); } else if (value == "reboottoota") { @@ -345,14 +361,14 @@ void WebCfgServer::initialize() } else { - return buildConfirmHtml(request, "No confirm code set.", 3, true); + return buildConfirmHtml(request, resp, "No confirm code set.", 3, true); } if(value != _confirmCode) { - return request->redirect("/"); + return resp->redirect("/"); } - esp_err_t res = buildConfirmHtml(request, "Rebooting to other partition...", 2, true); + esp_err_t res = buildConfirmHtml(request, resp, "Rebooting to other partition...", 2, true); waitAndProcess(true, 1000); esp_ota_set_boot_partition(esp_ota_get_next_update_partition(NULL)); restartEsp(RestartReason::OTAReboot); @@ -361,34 +377,36 @@ void WebCfgServer::initialize() else if (value == "autoupdate") { #ifndef NUKI_HUB_UPDATER - return processUpdate(request); + return processUpdate(request, resp); #else - return request->redirect("/"); + return resp->redirect("/"); #endif } else { + #ifndef CONFIG_IDF_TARGET_ESP32H2 if(!_network->isApOpen()) { + #endif #ifndef NUKI_HUB_UPDATER - return buildHtml(request); + return buildHtml(request, resp); #else - return buildOtaHtml(request); + return buildOtaHtml(request, resp); #endif - } #ifndef CONFIG_IDF_TARGET_ESP32H2 + } else { - return buildWifiConnectHtml(request); + return buildWifiConnectHtml(request, resp); } #endif } }); - _psychicServer->on("/post", HTTP_POST, [&](PsychicRequest *request) + _psychicServer->on("/post", HTTP_POST, [&](PsychicRequest *request, PsychicResponse* resp) { if(strlen(_credUser) > 0 && strlen(_credPassword) > 0 && !request->authenticate(_credUser, _credPassword)) { - return request->requestAuthentication(BASIC_AUTH, "Nuki Hub", "You must log in."); + return request->requestAuthentication(auth_type, "Nuki Hub", "You must log in."); } String value = ""; @@ -405,20 +423,20 @@ void WebCfgServer::initialize() if (value == "savecfg") { String message = ""; - bool restart = processArgs(request, message); + bool restart = processArgs(request, resp, message); if(request->hasParam("mqttssl")) { - return buildConfirmHtml(request, message, 3, true, "/get?page=mqttconfig"); + return buildConfirmHtml(request, resp, message, 3, true, "/get?page=mqttconfig"); } else { - return buildConfirmHtml(request, message, 3, true); + return buildConfirmHtml(request, resp, message, 3, true); } } else if (value == "savegpiocfg") { - processGpioArgs(request); - esp_err_t res = buildConfirmHtml(request, "Saving GPIO configuration. Restarting.", 3, true); + processGpioArgs(request, resp); + esp_err_t res = buildConfirmHtml(request, resp, "Saving GPIO configuration. Restarting.", 3, true); Log->println(F("Restarting")); waitAndProcess(true, 1000); restartEsp(RestartReason::GpioConfigurationUpdated); @@ -426,69 +444,74 @@ void WebCfgServer::initialize() } else if (value == "unpairlock") { - return processUnpair(request, false); + return processUnpair(request, resp, false); } else if (value == "unpairopener") { - return processUnpair(request, true); + return processUnpair(request, resp, true); } else if (value == "factoryreset") { - return processFactoryReset(request); + return processFactoryReset(request, resp); } else if (value == "import") { String message = ""; - bool restart = processImport(request, message); - return buildConfirmHtml(request, message, 3, true); + bool restart = processImport(request, resp, message); + return buildConfirmHtml(request, resp, message, 3, true); } else #else if (1 == 1) #endif { + #ifndef CONFIG_IDF_TARGET_ESP32H2 if(!_network->isApOpen()) { + #endif #ifndef NUKI_HUB_UPDATER - return buildHtml(request); + return buildHtml(request, resp); #else - return buildOtaHtml(request); + return buildOtaHtml(request, resp); #endif - } #ifndef CONFIG_IDF_TARGET_ESP32H2 + } else { - return buildWifiConnectHtml(request); + return buildWifiConnectHtml(request, resp); } #endif } }); PsychicUploadHandler *updateHandler = new PsychicUploadHandler(); - updateHandler->onUpload([&](PsychicRequest *request, const String& filename, uint64_t index, uint8_t *data, size_t len, bool final) + updateHandler->onUpload([&](PsychicRequest *request, const String& filename, uint64_t index, uint8_t *data, size_t len, bool last) { if(strlen(_credUser) > 0 && strlen(_credPassword) > 0 && !request->authenticate(_credUser, _credPassword)) { - return request->requestAuthentication(BASIC_AUTH, "Nuki Hub", "You must log in."); + return request->requestAuthentication(auth_type, "Nuki Hub", "You must log in."); } - return handleOtaUpload(request, filename, index, data, len, final); - } - ); + + return handleOtaUpload(request, filename, index, data, len, last); + }); - updateHandler->onRequest([&](PsychicRequest *request) + updateHandler->onRequest([&](PsychicRequest* request, PsychicResponse* resp) { if(strlen(_credUser) > 0 && strlen(_credPassword) > 0 && !request->authenticate(_credUser, _credPassword)) { - return request->requestAuthentication(BASIC_AUTH, "Nuki Hub", "You must log in."); + return request->requestAuthentication(auth_type, "Nuki Hub", "You must log in."); } - + String result; if (!Update.hasError()) { Log->print("Update code or data OK Update.errorString() "); Log->println(Update.errorString()); result = "Update OK."; - esp_err_t res = request->reply(200,"text/html",result.c_str()); + resp->setCode(200); + resp->setContentType("text/html"); + resp->setContent(result.c_str()); + esp_err_t res = resp->send(); restartEsp(RestartReason::OTACompleted); return res; } @@ -497,7 +520,10 @@ void WebCfgServer::initialize() result = " Update.errorString() " + String(Update.errorString()); Log->print("ERROR : error "); Log->println(result.c_str()); - esp_err_t res = request->reply(500, "text/html", result.c_str()); + resp->setCode(500); + resp->setContentType("text/html"); + resp->setContent(result.c_str()); + esp_err_t res = resp->send(); restartEsp(RestartReason::OTAAborted); return res; } @@ -531,12 +557,12 @@ void WebCfgServer::printCheckBox(PsychicStreamResponse *response, const char *to } #ifndef CONFIG_IDF_TARGET_ESP32H2 -esp_err_t WebCfgServer::buildSSIDListHtml(PsychicRequest *request) +esp_err_t WebCfgServer::buildSSIDListHtml(PsychicRequest *request, PsychicResponse* resp) { _network->scan(true, false); createSsidList(); - PsychicStreamResponse response(request, "text/html"); + PsychicStreamResponse response(resp, "text/html"); response.beginSend(); for (int i = 0; i < _ssidList.size(); i++) @@ -585,10 +611,10 @@ void WebCfgServer::createSsidList() } } -esp_err_t WebCfgServer::buildWifiConnectHtml(PsychicRequest *request) +esp_err_t WebCfgServer::buildWifiConnectHtml(PsychicRequest *request, PsychicResponse* resp) { String header = ""; - PsychicStreamResponse response(request, "text/html"); + PsychicStreamResponse response(resp, "text/html"); response.beginSend(); buildHtmlHeader(&response, header); response.print("

Available WiFi networks

"); @@ -622,7 +648,7 @@ esp_err_t WebCfgServer::buildWifiConnectHtml(PsychicRequest *request) return response.endSend(); } -bool WebCfgServer::processWiFi(PsychicRequest *request, String& message) +bool WebCfgServer::processWiFi(PsychicRequest *request, PsychicResponse* resp, String& message) { bool res = false; int params = request->params(); @@ -752,9 +778,9 @@ bool WebCfgServer::processWiFi(PsychicRequest *request, String& message) } #endif -esp_err_t WebCfgServer::buildOtaHtml(PsychicRequest *request, bool debug) +esp_err_t WebCfgServer::buildOtaHtml(PsychicRequest *request, PsychicResponse* resp, bool debug) { - PsychicStreamResponse response(request, "text/html"); + PsychicStreamResponse response(resp, "text/html"); response.beginSend(); buildHtmlHeader(&response); @@ -1117,9 +1143,9 @@ esp_err_t WebCfgServer::handleOtaUpload(PsychicRequest *request, const String& f } } -esp_err_t WebCfgServer::buildConfirmHtml(PsychicRequest *request, const String &message, uint32_t redirectDelay, bool redirect, String redirectTo) +esp_err_t WebCfgServer::buildConfirmHtml(PsychicRequest *request, PsychicResponse* resp, const String &message, uint32_t redirectDelay, bool redirect, String redirectTo) { - PsychicStreamResponse response(request, "text/html"); + PsychicStreamResponse response(resp, "text/html"); response.beginSend(); String header; @@ -1139,25 +1165,23 @@ esp_err_t WebCfgServer::buildConfirmHtml(PsychicRequest *request, const String & return response.endSend(); } -esp_err_t WebCfgServer::sendCss(PsychicRequest *request) +esp_err_t WebCfgServer::sendCss(PsychicRequest* request, PsychicResponse* resp) { // escaped by https://www.cescaper.com/ - PsychicResponse response(request); - response.addHeader("Cache-Control", "public, max-age=3600"); - response.setCode(200); - response.setContentType("text/css"); - response.setContent(stylecss); - return response.send(); + resp->addHeader("Cache-Control", "public, max-age=3600"); + resp->setCode(200); + resp->setContentType("text/css"); + resp->setContent(stylecss); + return resp->send(); } -esp_err_t WebCfgServer::sendFavicon(PsychicRequest *request) +esp_err_t WebCfgServer::sendFavicon(PsychicRequest* request, PsychicResponse* resp) { - PsychicResponse response(request); - response.addHeader("Cache-Control", "public, max-age=604800"); - response.setCode(200); - response.setContentType("image/png"); - response.setContent((const uint8_t *)favicon_32x32, sizeof(favicon_32x32)); - return response.send(); + resp->addHeader("Cache-Control", "public, max-age=604800"); + resp->setCode(200); + resp->setContentType("image/png"); + resp->setContent((const uint8_t *)favicon_32x32, sizeof(favicon_32x32)); + return resp->send(); } String WebCfgServer::generateConfirmCode() @@ -1166,7 +1190,7 @@ String WebCfgServer::generateConfirmCode() return String(code); } -void WebCfgServer::printInputField(PsychicStreamResponse *response, +void WebCfgServer::printInputField(PsychicStreamResponse* response, const char *token, const char *description, const char *value, @@ -1223,7 +1247,7 @@ void WebCfgServer::printInputField(PsychicStreamResponse *response, } #ifndef NUKI_HUB_UPDATER -esp_err_t WebCfgServer::sendSettings(PsychicRequest *request) +esp_err_t WebCfgServer::sendSettings(PsychicRequest *request, PsychicResponse* resp) { bool redacted = false; bool pairing = false; @@ -1425,11 +1449,13 @@ esp_err_t WebCfgServer::sendSettings(PsychicRequest *request) } serializeJsonPretty(json, jsonPretty); - - return request->reply(200, "application/json", jsonPretty.c_str()); + resp->setCode(200); + resp->setContentType("application/json"); + resp->setContent(jsonPretty.c_str()); + return resp->send(); } -bool WebCfgServer::processArgs(PsychicRequest *request, String& message) +bool WebCfgServer::processArgs(PsychicRequest *request, PsychicResponse* resp, String& message) { bool configChanged = false; bool aclLvlChanged = false; @@ -2426,6 +2452,16 @@ bool WebCfgServer::processArgs(PsychicRequest *request, String& message) //configChanged = true; } } + else if(key == "CREDDIGEST") + { + if(_preferences->getBool(preference_http_auth_type, false) != (value == "1")) + { + _preferences->putBool(preference_http_auth_type, (value == "1")); + Log->print(F("Setting changed: ")); + Log->println(key); + configChanged = true; + } + } else if(key == "ACLLCKLCK") { aclPrefs[0] = ((value == "1") ? 1 : 0); @@ -2996,7 +3032,7 @@ bool WebCfgServer::processArgs(PsychicRequest *request, String& message) configChanged = true; } - if(pass1 != "" && pass1 == pass2) + if(pass1 != "" && pass1 != "*" && pass1 == pass2) { if(_preferences->getString(preference_cred_password, "") != pass1) { @@ -3137,7 +3173,7 @@ bool WebCfgServer::processArgs(PsychicRequest *request, String& message) return configChanged; } -bool WebCfgServer::processImport(PsychicRequest *request, String& message) +bool WebCfgServer::processImport(PsychicRequest *request, PsychicResponse* resp, String& message) { bool configChanged = false; unsigned char currentBleAddress[6]; @@ -3361,7 +3397,7 @@ bool WebCfgServer::processImport(PsychicRequest *request, String& message) return configChanged; } -void WebCfgServer::processGpioArgs(PsychicRequest *request) +void WebCfgServer::processGpioArgs(PsychicRequest *request, PsychicResponse* resp) { int params = request->params(); std::vector pinConfiguration; @@ -3393,9 +3429,9 @@ void WebCfgServer::processGpioArgs(PsychicRequest *request) _gpio->savePinConfiguration(pinConfiguration); } -esp_err_t WebCfgServer::buildImportExportHtml(PsychicRequest *request) +esp_err_t WebCfgServer::buildImportExportHtml(PsychicRequest *request, PsychicResponse* resp) { - PsychicStreamResponse response(request, "text/html"); + PsychicStreamResponse response(resp, "text/html"); response.beginSend(); buildHtmlHeader(&response); response.print("

Import configuration

"); @@ -3411,10 +3447,10 @@ esp_err_t WebCfgServer::buildImportExportHtml(PsychicRequest *request) return response.endSend(); } -esp_err_t WebCfgServer::buildCustomNetworkConfigHtml(PsychicRequest *request) +esp_err_t WebCfgServer::buildCustomNetworkConfigHtml(PsychicRequest *request, PsychicResponse* resp) { String header = ""; - PsychicStreamResponse response(request, "text/html"); + PsychicStreamResponse response(resp, "text/html"); response.beginSend(); buildHtmlHeader(&response, header); response.print("
"); @@ -3442,10 +3478,10 @@ esp_err_t WebCfgServer::buildCustomNetworkConfigHtml(PsychicRequest *request) return response.endSend(); } -esp_err_t WebCfgServer::buildHtml(PsychicRequest *request) +esp_err_t WebCfgServer::buildHtml(PsychicRequest *request, PsychicResponse* resp) { String header = ""; - PsychicStreamResponse response(request, "text/html"); + PsychicStreamResponse response(resp, "text/html"); response.beginSend(); buildHtmlHeader(&response, header); if(_rebootRequired) @@ -3541,9 +3577,9 @@ esp_err_t WebCfgServer::buildHtml(PsychicRequest *request) return response.endSend(); } -esp_err_t WebCfgServer::buildCredHtml(PsychicRequest *request) +esp_err_t WebCfgServer::buildCredHtml(PsychicRequest *request, PsychicResponse* resp) { - PsychicStreamResponse response(request, "text/html"); + PsychicStreamResponse response(resp, "text/html"); response.beginSend(); buildHtmlHeader(&response); response.print(""); @@ -3553,6 +3589,7 @@ esp_err_t WebCfgServer::buildCredHtml(PsychicRequest *request) printInputField(&response, "CREDUSER", "User (# to clear)", _preferences->getString(preference_cred_user).c_str(), 30, "id=\"inputuser\"", false, true); printInputField(&response, "CREDPASS", "Password", "*", 30, "id=\"inputpass\"", true, true); printInputField(&response, "CREDPASSRE", "Retype password", "*", 30, "id=\"inputpass2\"", true); + printCheckBox(&response, "CREDDIGEST", "Use Digest Authentication (more secure)", _preferences->getBool(preference_http_auth_type, false), ""); response.print(""); response.print("
"); response.print("
"); @@ -3626,9 +3663,9 @@ esp_err_t WebCfgServer::buildCredHtml(PsychicRequest *request) return response.endSend(); } -esp_err_t WebCfgServer::buildNetworkConfigHtml(PsychicRequest *request) +esp_err_t WebCfgServer::buildNetworkConfigHtml(PsychicRequest *request, PsychicResponse* resp) { - PsychicStreamResponse response(request, "text/html"); + PsychicStreamResponse response(resp, "text/html"); response.beginSend(); buildHtmlHeader(&response); response.print("
"); @@ -3659,9 +3696,9 @@ esp_err_t WebCfgServer::buildNetworkConfigHtml(PsychicRequest *request) return response.endSend(); } -esp_err_t WebCfgServer::buildMqttConfigHtml(PsychicRequest *request) +esp_err_t WebCfgServer::buildMqttConfigHtml(PsychicRequest *request, PsychicResponse* resp) { - PsychicStreamResponse response(request, "text/html"); + PsychicStreamResponse response(resp, "text/html"); response.beginSend(); buildHtmlHeader(&response); response.print(""); @@ -3705,9 +3742,9 @@ esp_err_t WebCfgServer::buildMqttConfigHtml(PsychicRequest *request) return response.endSend(); } -esp_err_t WebCfgServer::buildMqttSSLConfigHtml(PsychicRequest *request, int type) +esp_err_t WebCfgServer::buildMqttSSLConfigHtml(PsychicRequest *request, PsychicResponse* resp, int type) { - PsychicStreamResponse response(request, "text/html"); + PsychicStreamResponse response(resp, "text/html"); response.beginSend(); buildHtmlHeader(&response); response.print(""); @@ -3736,9 +3773,9 @@ esp_err_t WebCfgServer::buildMqttSSLConfigHtml(PsychicRequest *request, int type return response.endSend(); } -esp_err_t WebCfgServer::buildAdvancedConfigHtml(PsychicRequest *request) +esp_err_t WebCfgServer::buildAdvancedConfigHtml(PsychicRequest *request, PsychicResponse* resp) { - PsychicStreamResponse response(request, "text/html"); + PsychicStreamResponse response(resp, "text/html"); response.beginSend(); buildHtmlHeader(&response); response.print(""); @@ -3815,7 +3852,7 @@ esp_err_t WebCfgServer::buildAdvancedConfigHtml(PsychicRequest *request) return response.endSend(); } -esp_err_t WebCfgServer::buildStatusHtml(PsychicRequest *request) +esp_err_t WebCfgServer::buildStatusHtml(PsychicRequest *request, PsychicResponse* resp) { JsonDocument json; String jsonStr; @@ -3913,7 +3950,10 @@ esp_err_t WebCfgServer::buildStatusHtml(PsychicRequest *request) } serializeJson(json, jsonStr); - return request->reply(200, "application/json", jsonStr.c_str()); + resp->setCode(200); + resp->setContentType("application/json"); + resp->setContent(jsonStr.c_str()); + return resp->send(); } String WebCfgServer::pinStateToString(uint8_t value) @@ -3931,9 +3971,9 @@ String WebCfgServer::pinStateToString(uint8_t value) } } -esp_err_t WebCfgServer::buildAccLvlHtml(PsychicRequest *request) +esp_err_t WebCfgServer::buildAccLvlHtml(PsychicRequest *request, PsychicResponse* resp) { - PsychicStreamResponse response(request, "text/html"); + PsychicStreamResponse response(resp, "text/html"); response.beginSend(); buildHtmlHeader(&response); @@ -4115,9 +4155,9 @@ esp_err_t WebCfgServer::buildAccLvlHtml(PsychicRequest *request) return response.endSend(); } -esp_err_t WebCfgServer::buildNukiConfigHtml(PsychicRequest *request) +esp_err_t WebCfgServer::buildNukiConfigHtml(PsychicRequest *request, PsychicResponse* resp) { - PsychicStreamResponse response(request, "text/html"); + PsychicStreamResponse response(resp, "text/html"); response.beginSend(); buildHtmlHeader(&response); response.print(""); @@ -4158,9 +4198,9 @@ esp_err_t WebCfgServer::buildNukiConfigHtml(PsychicRequest *request) return response.endSend(); } -esp_err_t WebCfgServer::buildGpioConfigHtml(PsychicRequest *request) +esp_err_t WebCfgServer::buildGpioConfigHtml(PsychicRequest *request, PsychicResponse* resp) { - PsychicStreamResponse response(request, "text/html"); + PsychicStreamResponse response(resp, "text/html"); response.beginSend(); buildHtmlHeader(&response); response.print(""); @@ -4213,9 +4253,9 @@ esp_err_t WebCfgServer::buildGpioConfigHtml(PsychicRequest *request) } #ifndef CONFIG_IDF_TARGET_ESP32H2 -esp_err_t WebCfgServer::buildConfigureWifiHtml(PsychicRequest *request) +esp_err_t WebCfgServer::buildConfigureWifiHtml(PsychicRequest *request, PsychicResponse* resp) { - PsychicStreamResponse response(request, "text/html"); + PsychicStreamResponse response(resp, "text/html"); response.beginSend(); buildHtmlHeader(&response); response.print(""); @@ -4229,11 +4269,11 @@ esp_err_t WebCfgServer::buildConfigureWifiHtml(PsychicRequest *request) } #endif -esp_err_t WebCfgServer::buildInfoHtml(PsychicRequest *request) +esp_err_t WebCfgServer::buildInfoHtml(PsychicRequest *request, PsychicResponse* resp) { uint32_t aclPrefs[17]; _preferences->getBytes(preference_acl, &aclPrefs, sizeof(aclPrefs)); - PsychicStreamResponse response(request, "text/html"); + PsychicStreamResponse response(resp, "text/html"); response.beginSend(); buildHtmlHeader(&response); response.print("

System Information

");
@@ -4836,7 +4876,7 @@ esp_err_t WebCfgServer::buildInfoHtml(PsychicRequest *request)
     return response.endSend();
 }
 
-esp_err_t WebCfgServer::processUnpair(PsychicRequest *request, bool opener)
+esp_err_t WebCfgServer::processUnpair(PsychicRequest *request, PsychicResponse* resp, bool opener)
 {
     String value = "";
     if(request->hasParam("CONFIRMTOKEN"))
@@ -4850,10 +4890,10 @@ esp_err_t WebCfgServer::processUnpair(PsychicRequest *request, bool opener)
 
     if(value != _confirmCode)
     {
-        return buildConfirmHtml(request, "Confirm code is invalid.", 3, true);
+        return buildConfirmHtml(request, resp, "Confirm code is invalid.", 3, true);
     }
 
-    esp_err_t res = buildConfirmHtml(request, opener ? "Unpairing Nuki Opener and restarting." : "Unpairing Nuki Lock and restarting.", 3, true);
+    esp_err_t res = buildConfirmHtml(request, resp, opener ? "Unpairing Nuki Opener and restarting." : "Unpairing Nuki Lock and restarting.", 3, true);
 
     if(!opener && _nuki != nullptr)
     {
@@ -4872,7 +4912,7 @@ esp_err_t WebCfgServer::processUnpair(PsychicRequest *request, bool opener)
     return res;
 }
 
-esp_err_t WebCfgServer::processUpdate(PsychicRequest *request)
+esp_err_t WebCfgServer::processUpdate(PsychicRequest *request, PsychicResponse* resp)
 {
     esp_err_t res;
     String value = "";
@@ -4887,20 +4927,20 @@ esp_err_t WebCfgServer::processUpdate(PsychicRequest *request)
 
     if(value != _confirmCode)
     {
-        return buildConfirmHtml(request, "Confirm code is invalid.", 3, true);
+        return buildConfirmHtml(request, resp, "Confirm code is invalid.", 3, true);
     }
 
     if(request->hasParam("beta"))
     {
         if(request->hasParam("debug"))
         {
-            res = buildConfirmHtml(request, "Rebooting to update Nuki Hub and Nuki Hub updater
Updating to latest DEBUG BETA version", 2, true);
+            res = buildConfirmHtml(request, resp, "Rebooting to update Nuki Hub and Nuki Hub updater
Updating to latest DEBUG BETA version", 2, true);
             _preferences->putString(preference_ota_updater_url, GITHUB_BETA_UPDATER_BINARY_URL_DBG);
             _preferences->putString(preference_ota_main_url, GITHUB_BETA_RELEASE_BINARY_URL_DBG);
         }
         else
         {
-            res = buildConfirmHtml(request, "Rebooting to update Nuki Hub and Nuki Hub updater
Updating to latest BETA version", 2, true);
+            res = buildConfirmHtml(request, resp, "Rebooting to update Nuki Hub and Nuki Hub updater
Updating to latest BETA version", 2, true);
             _preferences->putString(preference_ota_updater_url, GITHUB_BETA_UPDATER_BINARY_URL);
             _preferences->putString(preference_ota_main_url, GITHUB_BETA_RELEASE_BINARY_URL);
         }
@@ -4909,13 +4949,13 @@ esp_err_t WebCfgServer::processUpdate(PsychicRequest *request)
     {
         if(request->hasParam("debug"))
         {
-            res = buildConfirmHtml(request, "Rebooting to update Nuki Hub and Nuki Hub updater
Updating to latest DEBUG DEVELOPMENT version", 2, true);
+            res = buildConfirmHtml(request, resp, "Rebooting to update Nuki Hub and Nuki Hub updater
Updating to latest DEBUG DEVELOPMENT version", 2, true);
             _preferences->putString(preference_ota_updater_url, GITHUB_MASTER_UPDATER_BINARY_URL_DBG);
             _preferences->putString(preference_ota_main_url, GITHUB_MASTER_RELEASE_BINARY_URL_DBG);
         }
         else
         {
-            res = buildConfirmHtml(request, "Rebooting to update Nuki Hub and Nuki Hub updater
Updating to latest DEVELOPMENT version", 2, true);
+            res = buildConfirmHtml(request, resp, "Rebooting to update Nuki Hub and Nuki Hub updater
Updating to latest DEVELOPMENT version", 2, true);
             _preferences->putString(preference_ota_updater_url, GITHUB_MASTER_UPDATER_BINARY_URL);
             _preferences->putString(preference_ota_main_url, GITHUB_MASTER_RELEASE_BINARY_URL);
         }
@@ -4924,13 +4964,13 @@ esp_err_t WebCfgServer::processUpdate(PsychicRequest *request)
     {
         if(request->hasParam("debug"))
         {
-            res = buildConfirmHtml(request, "Rebooting to update Nuki Hub and Nuki Hub updater
Updating to latest DEBUG RELEASE version", 2, true);
+            res = buildConfirmHtml(request, resp, "Rebooting to update Nuki Hub and Nuki Hub updater
Updating to latest DEBUG RELEASE version", 2, true);
             _preferences->putString(preference_ota_updater_url, GITHUB_LATEST_UPDATER_BINARY_URL_DBG);
             _preferences->putString(preference_ota_main_url, GITHUB_LATEST_UPDATER_BINARY_URL_DBG);
         }
         else
         {
-            res = buildConfirmHtml(request, "Rebooting to update Nuki Hub and Nuki Hub updater
Updating to latest RELEASE version", 2, true);
+            res = buildConfirmHtml(request, resp, "Rebooting to update Nuki Hub and Nuki Hub updater
Updating to latest RELEASE version", 2, true);
             _preferences->putString(preference_ota_updater_url, GITHUB_LATEST_UPDATER_BINARY_URL);
             _preferences->putString(preference_ota_main_url, GITHUB_LATEST_RELEASE_BINARY_URL);
         }
@@ -4940,7 +4980,7 @@ esp_err_t WebCfgServer::processUpdate(PsychicRequest *request)
     return res;
 }
 
-esp_err_t WebCfgServer::processFactoryReset(PsychicRequest *request)
+esp_err_t WebCfgServer::processFactoryReset(PsychicRequest *request, PsychicResponse* resp)
 {
     esp_err_t res;
     String value = "";
@@ -4956,7 +4996,7 @@ esp_err_t WebCfgServer::processFactoryReset(PsychicRequest *request)
     bool resetWifi = false;
     if(value.length() == 0 || value != _confirmCode)
     {
-        return buildConfirmHtml(request, "Confirm code is invalid.", 3, true);
+        return buildConfirmHtml(request, resp, "Confirm code is invalid.", 3, true);
     }
     else
     {
@@ -4973,11 +5013,11 @@ esp_err_t WebCfgServer::processFactoryReset(PsychicRequest *request)
         if(value2 == "1")
         {
             resetWifi = true;
-            res = buildConfirmHtml(request, "Factory resetting Nuki Hub, unpairing Nuki Lock and Nuki Opener and resetting WiFi.", 3, true);
+            res = buildConfirmHtml(request, resp, "Factory resetting Nuki Hub, unpairing Nuki Lock and Nuki Opener and resetting WiFi.", 3, true);
         }
         else
         {
-            res = buildConfirmHtml(request, "Factory resetting Nuki Hub, unpairing Nuki Lock and Nuki Opener.", 3, true);
+            res = buildConfirmHtml(request, resp, "Factory resetting Nuki Hub, unpairing Nuki Lock and Nuki Opener.", 3, true);
         }
     }
 
@@ -4991,7 +5031,7 @@ esp_err_t WebCfgServer::processFactoryReset(PsychicRequest *request)
     {
         _nukiOpener->unpair();
     }
-    
+
     String ssid = _preferences->getString(preference_wifi_ssid, "");
     String pass = _preferences->getString(preference_wifi_pass, "");
 
diff --git a/src/WebCfgServer.h b/src/WebCfgServer.h
index 68359e4..3886f08 100644
--- a/src/WebCfgServer.h
+++ b/src/WebCfgServer.h
@@ -48,29 +48,29 @@ public:
 
 private:
     #ifndef NUKI_HUB_UPDATER
-    esp_err_t sendSettings(PsychicRequest *request);
-    bool processArgs(PsychicRequest *request, String& message);
-    bool processImport(PsychicRequest *request, String& message);
-    void processGpioArgs(PsychicRequest *request);
-    esp_err_t buildHtml(PsychicRequest *request);
-    esp_err_t buildAccLvlHtml(PsychicRequest *request);
-    esp_err_t buildCredHtml(PsychicRequest *request);
-    esp_err_t buildImportExportHtml(PsychicRequest *request);
-    esp_err_t buildNetworkConfigHtml(PsychicRequest *request);
-    esp_err_t buildMqttConfigHtml(PsychicRequest *request);
-    esp_err_t buildMqttSSLConfigHtml(PsychicRequest *request, int type=0);
-    esp_err_t buildStatusHtml(PsychicRequest *request);    
-    esp_err_t buildAdvancedConfigHtml(PsychicRequest *request);
-    esp_err_t buildNukiConfigHtml(PsychicRequest *request);
-    esp_err_t buildGpioConfigHtml(PsychicRequest *request);
+    esp_err_t sendSettings(PsychicRequest *request, PsychicResponse* resp);
+    bool processArgs(PsychicRequest *request, PsychicResponse* resp, String& message);
+    bool processImport(PsychicRequest *request, PsychicResponse* resp, String& message);
+    void processGpioArgs(PsychicRequest *request, PsychicResponse* resp);
+    esp_err_t buildHtml(PsychicRequest *request, PsychicResponse* resp);
+    esp_err_t buildAccLvlHtml(PsychicRequest *request, PsychicResponse* resp);
+    esp_err_t buildCredHtml(PsychicRequest *request, PsychicResponse* resp);
+    esp_err_t buildImportExportHtml(PsychicRequest *request, PsychicResponse* resp);
+    esp_err_t buildNetworkConfigHtml(PsychicRequest *request, PsychicResponse* resp);
+    esp_err_t buildMqttConfigHtml(PsychicRequest *request, PsychicResponse* resp);
+    esp_err_t buildMqttSSLConfigHtml(PsychicRequest *request, PsychicResponse* resp, int type=0);
+    esp_err_t buildStatusHtml(PsychicRequest *request, PsychicResponse* resp);    
+    esp_err_t buildAdvancedConfigHtml(PsychicRequest *request, PsychicResponse* resp);
+    esp_err_t buildNukiConfigHtml(PsychicRequest *request, PsychicResponse* resp);
+    esp_err_t buildGpioConfigHtml(PsychicRequest *request, PsychicResponse* resp);
     #ifndef CONFIG_IDF_TARGET_ESP32H2
-    esp_err_t buildConfigureWifiHtml(PsychicRequest *request);
+    esp_err_t buildConfigureWifiHtml(PsychicRequest *request, PsychicResponse* resp);
     #endif
-    esp_err_t buildInfoHtml(PsychicRequest *request);
-    esp_err_t buildCustomNetworkConfigHtml(PsychicRequest *request);
-    esp_err_t processUnpair(PsychicRequest *request, bool opener);
-    esp_err_t processUpdate(PsychicRequest *request);
-    esp_err_t processFactoryReset(PsychicRequest *request);
+    esp_err_t buildInfoHtml(PsychicRequest *request, PsychicResponse* resp);
+    esp_err_t buildCustomNetworkConfigHtml(PsychicRequest *request, PsychicResponse* resp);
+    esp_err_t processUnpair(PsychicRequest *request, PsychicResponse* resp, bool opener);
+    esp_err_t processUpdate(PsychicRequest *request, PsychicResponse* resp);
+    esp_err_t processFactoryReset(PsychicRequest *request, PsychicResponse* resp);
     void printTextarea(PsychicStreamResponse *response, const char *token, const char *description, const char *value, const size_t& maxLength, const bool& enabled = true, const bool& showLengthRestriction = false);
     void printDropDown(PsychicStreamResponse *response, const char *token, const char *description, const String preselectedValue, std::vector> options, const String className);
     void buildNavigationMenuEntry(PsychicStreamResponse *response, const char *title, const char *targetPath, const char* warningMessage = "");
@@ -100,19 +100,19 @@ private:
     String generateConfirmCode();
     String _confirmCode = "----";
     
-    esp_err_t buildSSIDListHtml(PsychicRequest *request);
-    esp_err_t buildConfirmHtml(PsychicRequest *request, const String &message, uint32_t redirectDelay = 5, bool redirect = false, String redirectTo = "/");
-    esp_err_t buildOtaHtml(PsychicRequest *request, bool debug = false);
-    esp_err_t sendCss(PsychicRequest *request);
-    esp_err_t sendFavicon(PsychicRequest *request);
+    esp_err_t buildSSIDListHtml(PsychicRequest *request, PsychicResponse* resp);
+    esp_err_t buildConfirmHtml(PsychicRequest *request, PsychicResponse* resp, const String &message, uint32_t redirectDelay = 5, bool redirect = false, String redirectTo = "/");
+    esp_err_t buildOtaHtml(PsychicRequest *request, PsychicResponse* resp, bool debug = false);
+    esp_err_t sendCss(PsychicRequest *request, PsychicResponse* resp);
+    esp_err_t sendFavicon(PsychicRequest *request, PsychicResponse* resp);
     void createSsidList();
     void buildHtmlHeader(PsychicStreamResponse *response, String additionalHeader = "");
     void waitAndProcess(const bool blocking, const uint32_t duration);
     esp_err_t handleOtaUpload(PsychicRequest *request, const String& filename, uint64_t index, uint8_t *data, size_t len, bool final);
     void printCheckBox(PsychicStreamResponse *response, const char* token, const char* description, const bool value, const char* htmlClass);
     #ifndef CONFIG_IDF_TARGET_ESP32H2
-    esp_err_t buildWifiConnectHtml(PsychicRequest *request);
-    bool processWiFi(PsychicRequest *request, String& message);
+    esp_err_t buildWifiConnectHtml(PsychicRequest *request, PsychicResponse* resp);
+    bool processWiFi(PsychicRequest *request, PsychicResponse* resp, String& message);
     
     #endif
     void printInputField(PsychicStreamResponse *response, const char* token, const char* description, const char* value, const size_t& maxLength, const char* args, const bool& isPassword = false, const bool& showLengthRestriction = false);
diff --git a/src/main.cpp b/src/main.cpp
index cbe3832..18a5317 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -80,6 +80,7 @@ RTC_NOINIT_ATTR bool disableNetwork;
 RTC_NOINIT_ATTR bool wifiFallback;
 RTC_NOINIT_ATTR bool ethCriticalFailure;
 
+bool doOta = false;
 bool restartReason_isValid;
 RestartReason currentRestartReason = RestartReason::NotApplicable;
 
@@ -132,7 +133,7 @@ void setReroute()
     esp_log_level_set("httpd_parse", ESP_LOG_ERROR);
     esp_log_level_set("httpd_txrx", ESP_LOG_ERROR);
     esp_log_level_set("httpd_uri", ESP_LOG_ERROR);
-    esp_log_level_set("event", ESP_LOG_ERROR);  
+    esp_log_level_set("event", ESP_LOG_ERROR);
     esp_log_level_set("psychic", ESP_LOG_ERROR);
     esp_log_level_set("ARDUINO", ESP_LOG_DEBUG);
     esp_log_level_set("nvs", ESP_LOG_ERROR);
@@ -494,7 +495,6 @@ void setup()
     preferences = new Preferences();
     preferences->begin("nukihub", false);
     initPreferences(preferences);
-    bool doOta = false;
     uint8_t partitionType = checkPartition();
 
     initializeRestartReason();
@@ -540,15 +540,13 @@ void setup()
     if(!doOta)
     {
         psychicServer = new PsychicHttpServer;
-        psychicServer->config.max_uri_handlers = 10;
         psychicServer->config.stack_size = HTTPD_TASK_SIZE;
-        psychicServer->listen(80);
         webCfgServer = new WebCfgServer(network, preferences, network->networkDeviceType() == NetworkDeviceType::WiFi, partitionType, psychicServer);
         webCfgServer->initialize();
-        psychicServer->onNotFound([](PsychicRequest* request)
-        {
-            return request->redirect("/");
+        psychicServer->onNotFound([](PsychicRequest* request, PsychicResponse* response) {
+            return response->redirect("/");
         });
+        psychicServer->begin();
     }
 #else
     if(preferences->getBool(preference_enable_bootloop_reset, false))
@@ -636,18 +634,16 @@ void setup()
     if(!doOta && !disableNetwork && (forceEnableWebServer || preferences->getBool(preference_webserver_enabled, true) || preferences->getBool(preference_webserial_enabled, false)))
     {
         psychicServer = new PsychicHttpServer;
-        psychicServer->config.max_uri_handlers = 10;
         psychicServer->config.stack_size = HTTPD_TASK_SIZE;
-        psychicServer->listen(80);
 
         if(forceEnableWebServer || preferences->getBool(preference_webserver_enabled, true))
         {
             webCfgServer = new WebCfgServer(nuki, nukiOpener, network, gpio, preferences, network->networkDeviceType() == NetworkDeviceType::WiFi, partitionType, psychicServer);
             webCfgServer->initialize();
-            psychicServer->onNotFound([](PsychicRequest* request)
-            {
-                return request->redirect("/");
+            psychicServer->onNotFound([](PsychicRequest* request, PsychicResponse* response) {
+                return response->redirect("/");
             });
+            psychicServer->begin();
         }
         /*
 #ifdef DEBUG_NUKIHUB
@@ -656,7 +652,7 @@ void setup()
         if(preferences->getBool(preference_webserial_enabled, false))
         {
           WebSerial.setAuthentication(preferences->getString(preference_cred_user), preferences->getString(preference_cred_password));
-          WebSerial.begin(asyncServer);
+          WebSerial.begin(psychicServer);
           WebSerial.setBuffer(1024);
         }
 #endif