Merge branch 'main' into dev

src/helpers/esp32/SerialWifiInterface.cpp

@@ -0,0 +1,93 @@
+#include "SerialWifiInterface.h"
+#include <WiFi.h>
+
+void SerialWifiInterface::begin(int port) {
+  // wifi setup is handled outside of this class, only starts the server
+  server.begin(port);
+}
+
+// ---------- public methods
+void SerialWifiInterface::enable() { 
+  if (_isEnabled) return;
+
+  _isEnabled = true;
+  clearBuffers();
+}
+
+void SerialWifiInterface::disable() {
+  _isEnabled = false;
+}
+
+size_t SerialWifiInterface::writeFrame(const uint8_t src[], size_t len) {
+  if (len > MAX_FRAME_SIZE) {
+    WIFI_DEBUG_PRINTLN("writeFrame(), frame too big, len=%d\n", len);
+    return 0;
+  }
+
+  if (deviceConnected && len > 0) {
+    if (send_queue_len >= FRAME_QUEUE_SIZE) {
+      WIFI_DEBUG_PRINTLN("writeFrame(), send_queue is full!");
+      return 0;
+    }
+
+    send_queue[send_queue_len].len = len;  // add to send queue
+    memcpy(send_queue[send_queue_len].buf, src, len);
+    send_queue_len++;
+
+    return len;
+  }
+  return 0;
+}
+
+bool SerialWifiInterface::isWriteBusy() const {
+  return false;
+}
+
+size_t SerialWifiInterface::checkRecvFrame(uint8_t dest[]) {
+  if (!client) client = server.available();
+
+  if (client.connected()) {
+    if (!deviceConnected) {
+      WIFI_DEBUG_PRINTLN("Got connection");
+      deviceConnected = true;
+    }
+  } else {
+    if (deviceConnected) {
+      deviceConnected = false;
+      WIFI_DEBUG_PRINTLN("Disconnected");
+    }
+  }
+
+  if (deviceConnected) {
+    if (send_queue_len > 0) {   // first, check send queue
+      
+      _last_write = millis();
+      int len = send_queue[0].len;
+
+      uint8_t pkt[3+len]; // use same header as serial interface so client can delimit frames
+      pkt[0] = '>';
+      pkt[1] = (len & 0xFF);  // LSB
+      pkt[2] = (len >> 8);    // MSB
+      memcpy(&pkt[3], send_queue[0].buf, send_queue[0].len);
+      client.write(pkt, 3 + len);
+      send_queue_len--;
+      for (int i = 0; i < send_queue_len; i++) {   // delete top item from queue
+        send_queue[i] = send_queue[i + 1];
+      }
+    } else {
+      int len = client.available();
+      if (len > 0) {
+        uint8_t buf[MAX_FRAME_SIZE + 4];
+        client.readBytes(buf, len);
+        memcpy(dest, buf+3, len-3); // remove header (don't even check ... problems are on the other dir)
+        return len-3;
+      }
+    }
+  }
+
+  return 0;
+}
+
+bool SerialWifiInterface::isConnected() const {
+  return deviceConnected;  //pServer != NULL && pServer->getConnectedCount() > 0;
+}

src/helpers/esp32/SerialWifiInterface.h

@@ -0,0 +1,59 @@
+#pragma once
+
+#include "../BaseSerialInterface.h"
+#include <WiFi.h>
+
+class SerialWifiInterface : public BaseSerialInterface {
+  bool deviceConnected;
+  bool _isEnabled;
+  unsigned long _last_write;
+  unsigned long adv_restart_time;
+
+  WiFiServer server;
+  WiFiClient client;
+
+  struct Frame {
+    uint8_t len;
+    uint8_t buf[MAX_FRAME_SIZE];
+  };
+
+  #define FRAME_QUEUE_SIZE  4
+  int recv_queue_len;
+  Frame recv_queue[FRAME_QUEUE_SIZE];
+  int send_queue_len;
+  Frame send_queue[FRAME_QUEUE_SIZE];
+
+  void clearBuffers() { recv_queue_len = 0; send_queue_len = 0; }
+
+protected:
+
+public:
+  SerialWifiInterface() : server(WiFiServer()), client(WiFiClient()) {
+    deviceConnected = false;
+    _isEnabled = false;
+    _last_write = 0;
+    send_queue_len = recv_queue_len = 0;
+  }
+
+  void begin(int port);
+
+  // BaseSerialInterface methods
+  void enable() override;
+  void disable() override;
+  bool isEnabled() const override { return _isEnabled; }
+
+  bool isConnected() const override;
+  bool isWriteBusy() const override;
+
+  size_t writeFrame(const uint8_t src[], size_t len) override;
+  size_t checkRecvFrame(uint8_t dest[]) override;
+};
+
+#if WIFI_DEBUG_LOGGING && ARDUINO
+  #include <Arduino.h>
+  #define WIFI_DEBUG_PRINT(F, ...) Serial.printf("WiFi: " F, ##__VA_ARGS__)
+  #define WIFI_DEBUG_PRINTLN(F, ...) Serial.printf("WiFi: " F "\n", ##__VA_ARGS__)
+#else
+  #define WIFI_DEBUG_PRINT(...) {}
+  #define WIFI_DEBUG_PRINTLN(...) {}
+#endif