diff --git a/Wifi_NINA.cpp b/Wifi_NINA.cpp index 8d54030..ea970f3 100644 --- a/Wifi_NINA.cpp +++ b/Wifi_NINA.cpp @@ -21,7 +21,7 @@ #include "defines.h" #ifdef WIFI_NINA -#include +//#include #include #ifndef ARDUINO_GIGA #include @@ -49,13 +49,13 @@ #else #warning "WiFiNINA has no SPI port or pin allocations for this archiecture yet!" #endif - -class NetworkClient { +#define MAX_CLIENTS 10 +/*class NetworkClient { public: NetworkClient(WiFiClient c) { wifi = c; }; - bool ok() { + bool ok() { return (inUse && wifi.connected()); }; bool recycle(WiFiClient c) { @@ -65,22 +65,21 @@ public: // return false here until we have // implemented a LRU timer // if (LRU too recent) return false; - return false; + //return false; wifi = c; inUse = true; return true; }; - WiFiClient wifi; + WiFiClient wifi; bool inUse = true; -}; +};*/ -static std::vector clients; // a list to hold all clients +//static std::vector clients; // a list to hold all clients static WiFiServer *server = NULL; static RingStream *outboundRing = new RingStream(10240); static bool APmode = false; static IPAddress ip; - // #ifdef WIFI_TASK_ON_CORE0 // void wifiLoop(void *){ // for(;;){ @@ -149,7 +148,7 @@ bool WifiNINA::setup(const char *SSid, } if (WiFi.status() == WL_CONNECTED) { // String ip_str = sprintf("%xl", WiFi.localIP()); - DIAG(F("Wifi STA IP %d.%d.%d.%d"), WiFi.localIP()[0], WiFi.localIP()[1],WiFi.localIP()[2],WiFi.localIP()[3],WiFi.localIP()[4],WiFi.localIP()[5]); + DIAG(F("Wifi STA IP %d.%d.%d.%d"), WiFi.localIP()[0], WiFi.localIP()[1],WiFi.localIP()[2],WiFi.localIP()[3]); wifiUp = true; } else { DIAG(F("Could not connect to Wifi SSID %s"),SSid); @@ -164,7 +163,7 @@ bool WifiNINA::setup(const char *SSid, } if (WiFi.status() == WL_CONNECTED) { ip = WiFi.localIP(); - DIAG(F("Wifi STA IP 2nd try %s"), ip); + DIAG(F("Wifi STA IP 2nd try %d.%d.%d.%d"), ip[0], ip[1], ip[2], ip[3]); wifiUp = true; } else { DIAG(F("Wifi STA mode FAIL. Will revert to AP mode")); @@ -184,7 +183,7 @@ bool WifiNINA::setup(const char *SSid, strMac += String(mac[i], HEX); } - DIAG(F("MAC address: %x:%x:%x:%x:%X;%x"), mac[0], mac[1], mac[2], mac[3], mac[4], mac[5]); + DIAG(F("MAC address: %x:%x:%x:%x:%x:%x"), mac[0], mac[1], mac[2], mac[3], mac[4], mac[5]); strMac.remove(0,9); strMac.replace(":",""); @@ -200,7 +199,7 @@ bool WifiNINA::setup(const char *SSid, channel) == WL_AP_LISTENING) { DIAG(F("Wifi AP SSID %s PASS %s"),strSSID.c_str(),havePassword ? password : strPass.c_str()); ip = WiFi.localIP(); - DIAG(F("Wifi AP IP %s"),ip); + DIAG(F("Wifi AP IP %d.%d.%d.%d"),ip[0], ip[1], ip[2], ip[3]); wifiUp = true; APmode = true; } else { @@ -248,6 +247,9 @@ bool WifiNINA::setup(const char *SSid, // #else DIAG(F("Server will be started on port %d"),port); // #endif + ip = WiFi.localIP(); + LCD(4,F("IP: %d.%d.%d.%d"), ip[0], ip[1], ip[2], ip[3]); + LCD(5,F("Port:%d"), port); return true; } @@ -261,7 +263,7 @@ const char *wlerror[] = { "WL_DISCONNECTED" }; -void WifiNINA::loop() { +/*void WifiNINA::loop() { int clientId; //tmp loop var // really no good way to check for LISTEN especially in AP mode? @@ -271,31 +273,33 @@ void WifiNINA::loop() { for (clientId=0; clientIdavailable()) { - WiFiClient client; - while (client = server->available()) { + WiFiClient client = server->available(); + if (client) { + ///while (client.available() == true) { for (clientId=0; clientId=clients.size()) { - NetworkClient nc(client); - clients.push_back(nc); + NetworkClient* nc=new NetworkClient(client); + clients.push_back(*nc); + //delete nc; ip = client.remoteIP(); - DIAG(F("New client %d, %s"), clientId, ip); + DIAG(F("New client %d, %d.%d.%d.%d"), clientId, ip[0], ip[1], ip[2], ip[3]); } - } + ///} } // loop over all connected clients for (clientId=0; clientIdavailable(); + if (!newClient) return; + for (byte clientId=0; clientIdconnected()) { + clients[clientId]->stop(); + DIAG(F("Remove client %d"), clientId); + CommandDistributor::forget(clientId); + //delete c; //TJF: this causes a crash when client drops.. commenting out for now. + clients[clientId]=nullptr; // TJF: what to do... what to do... + } + } +} + +void WifiNINA::checkForClientInput() { + // Find a client providing input + for (byte clientId=0; clientIdavailable(); + if (len) { + // read data from client + byte cmd[len+1]; + for(int i=0; iread(); + cmd[len]=0; + CommandDistributor::parse(clientId,cmd,outboundRing); + } + } + } +} + +void WifiNINA::checkForClientOutput() { + // something to write out? + auto clientId=outboundRing->read(); + if (clientId < 0) return; + auto replySize=outboundRing->count(); + if (replySize==0) return; // nothing to send + auto c=clients[clientId]; + if (!c) { + // client is gone, throw away msg + for (int i=0;iread(); + DIAG(F("gone, drop message.")); //TJF: only for diag + return; + } + // emit data to the client object + // This should work in theory, the + DIAG(F("send message")); //TJF: only for diag + //TJF: the old code had to add a 0x00 byte to the end to terminate the + //TJF: c string, before sending it. i take it this is not needed? + for (int i=0;iwrite(outboundRing->read()); +} + +void WifiNINA::loop() { + checkForLostClients(); // *** + checkForNewClient(); + checkForClientInput(); // *** + WiThrottle::loop(outboundRing); // allow withrottle to broadcast if needed + checkForClientOutput(); +} + #endif // WIFI_NINA \ No newline at end of file diff --git a/Wifi_NINA.h b/Wifi_NINA.h index f841b4f..2cf5fbd 100644 --- a/Wifi_NINA.h +++ b/Wifi_NINA.h @@ -35,8 +35,12 @@ public: const char *hostname, const int port, const byte channel, - const bool forceAP); + const bool forceAP); static void loop(); private: + static void checkForNewClient(); + static void checkForLostClients(); + static void checkForClientInput(); + static void checkForClientOutput(); }; #endif //WifiNINA_h