/* © 2021, Harald Barth. This file is part of CommandStation-EX This is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version. It is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with CommandStation. If not, see . */ #include "defines.h" #ifdef ESP_FAMILY #include #include #include #include #include "WifiESP.h" #include "DIAG.h" #include "RingStream.h" #include "CommandDistributor.h" #include static std::vector clients; // a list to hold all clients static AsyncServer *server; static RingStream *outboundRing = new RingStream(2048); static void handleError(void* arg, AsyncClient* client, int8_t error) { DIAG(F("connection error %s from client %s"), client->errorToString(error), client->remoteIP().toString().c_str()); } static void handleData(void* arg, AsyncClient* client, void *data, size_t len) { //DIAG(F("data received from client %s"), client->remoteIP().toString().c_str()); uint8_t clientId; for (clientId=0; clientIdmark(clientId); CommandDistributor::parse(clientId,cmd,outboundRing); outboundRing->commit(); } } //static AsyncClient *debugclient = NULL; bool sendData(AsyncClient *client, char* data, size_t count) { size_t willsend = 0; // reply to client if (client->canSend()) { while (count > 0) { if (client->connected()) willsend = client->add(data, count); // add checks for space() else willsend = 0; if (willsend < count) { DIAG(F("Willsend %d of count %d"), willsend, count); } if (client->connected() && client->send()) { count = count - willsend; data = data + willsend; } else { DIAG(F("Could not send promised %d"), count); return false; } } // Did send all bytes we wanted return true; } DIAG(F("Aborting: Busy or space=0")); return false; } static void deleteClient(AsyncClient* client) { uint8_t clientId; for (clientId=0; clientIdremoteIP().toString().c_str()); deleteClient(client); } static void handleNewClient(void* arg, AsyncClient* client) { DIAG(F("New client %s"), client->remoteIP().toString().c_str()); // add to list clients.push_back(client); // register events client->onData(&handleData, NULL); client->onError(&handleError, NULL); client->onDisconnect(&handleDisconnect, NULL); client->onTimeout(&handleTimeOut, NULL); } /* Things one _might_ want to do: Disable soft watchdog: ESP.wdtDisable() Enable soft watchdog: ESP.wdtEnable(X) ignores the value of X and enables it for fixed time at least in version 3.0.2 of the esp8266 package. Internet says: I manage to complety disable the hardware watchdog on ESP8266 in order to run the benchmark CoreMark. void hw_wdt_disable(){ *((volatile uint32_t*) 0x60000900) &= ~(1); // Hardware WDT OFF } void hw_wdt_enable(){ *((volatile uint32_t*) 0x60000900) |= 1; // Hardware WDT ON } */ bool WifiESP::setup(const char *SSid, const char *password, const char *hostname, int port, const byte channel) { bool havePassword = true; bool haveSSID = true; bool wifiUp = false; // We are server and should not sleep wifi_set_sleep_type(NONE_SLEEP_T); // connects to access point const char *yourNetwork = "Your network "; if (strncmp(yourNetwork, SSid, 13) == 0 || strncmp("", SSid, 13) == 0) haveSSID = false; if (strncmp(yourNetwork, password, 13) == 0 || strncmp("", password, 13) == 0) havePassword = false; if (haveSSID && havePassword) { WiFi.mode(WIFI_STA); WiFi.setAutoReconnect(true); WiFi.begin(SSid, password); while (WiFi.status() != WL_CONNECTED) { Serial.print('.'); delay(500); } if (WiFi.status() == WL_CONNECTED) { DIAG(F("Wifi STA IP %s"),WiFi.localIP().toString().c_str()); wifiUp = true; } } if (!haveSSID) { // prepare all strings String strSSID("DCC_"); String strPass("PASS_"); String strMac = WiFi.macAddress(); strMac.remove(0,9); strMac.replace(":",""); strMac.replace(":",""); strSSID.concat(strMac); strPass.concat(strMac); WiFi.mode(WIFI_AP); if (WiFi.softAP(strSSID.c_str(), havePassword ? password : strPass.c_str(), channel, false, 8)) { DIAG(F("Wifi AP SSID %s PASS %s"),strSSID.c_str(),havePassword ? password : strPass.c_str()); DIAG(F("Wifi AP IP %s"),WiFi.softAPIP().toString().c_str()); wifiUp = true; } } if (!wifiUp) { DIAG(F("Wifi all fail")); // no idea to go on return false; } server = new AsyncServer(port); // start listening on tcp port server->onClient(&handleNewClient, server); server->begin(); DIAG(F("Server up port %d"),port); return true; } void WifiESP::loop() { AsyncClient *client = NULL; // Do something with outboundRing // call sendData int clientId=outboundRing->peek(); if (clientId >= 0) { if (clientId > clients.size()) { // something is wrong with the ringbuffer position outboundRing->info(); client = NULL; } else { client = clients[clientId]; } // if (client != debugclient) { // DIAG(F("new client pointer = %x from id %d"), client, clientId); // debugclient = client; // } } else { client = NULL; } if (clientId>=0 && client && client->connected() && client->canSend()) { outboundRing->read(); int count=outboundRing->count(); //DIAG(F("Wifi reply client=%d, count=%d"), clientId,count); { char buffer[count+1]; for(int i=0;iread(); if (c >= 0) buffer[i] = (char)c; else { DIAG(F("Ringread fail at %d"),i); break; } } buffer[count]=0; //DIAG(F("SEND:%s COUNT:%d"),buffer,count); uint8_t tries = 3; while (! sendData(client, buffer, count)) { DIAG(F("senData fail")); yield(); if (tries == 0) break; } } } #ifdef ESP_DEBUG static unsigned long last = 0; if (millis() - last > 60000) { last = millis(); DIAG(F("+")); } #endif ESP.wdtFeed(); } #endif //ESP_FAMILY