libraries/WiFi: Write function.

Signed-off-by: IFX-Anusha <[email protected]>

Author: IFX-Anusha

libraries/WiFi/src/WiFiUdp.cpp

@@ -59,29 +59,26 @@ int WiFiUDP::beginPacket(const char *host, uint16_t port) {
 }
 
 int WiFiUDP::endPacket() {
+    return 0;
 }
 
-size_t WiFiUDP::write(uint8_t) {
+size_t WiFiUDP::write(uint8_t byte) {
     return write(&byte, 1);
 }
 
 size_t WiFiUDP::write(const uint8_t *buffer, size_t size) {
-    if (size == 0) {
-        return 0;
-    }
-    cy_socket_sockaddr_t addr = {
-        .port = _port,
-        .ip_address = {
-            .version = CY_SOCKET_IP_VER_V4,
-            .ip = { .v4 = (uint32_t)remote_ip }
+    size_t bytesStored = 0;
+
+    for (size_t i = 0; i < size; i++) {
+        if (!txBuffer.isFull()) { // Check if the buffer is not full
+            txBuffer.store_char(buffer[i]);
+            bytesStored++;
+        } else {
+            break; // Stop if the buffer is full
         }
-    };
-    size_t bytes_written = socket.send(buffer, size, &addr);
-    if (bytes_written != size) {
-        _status = SOCKET_STATUS_ERROR;
-        return 0;
     }
-    return bytes_written;
+
+    return bytesStored; // Return the number of bytes successfully stored
 }
 
 int WiFiUDP::parsePacket() {

libraries/WiFi/src/WiFiUdp.h

@@ -11,6 +11,9 @@
 
 class WiFiUDP: public arduino::UDP {
 public:
+
+    static const size_t WIFI_UDP_BUFFER_SIZE = 256;
+
     WiFiUDP();
     uint8_t begin(uint16_t);
 
@@ -19,9 +22,8 @@ class WiFiUDP: public arduino::UDP {
     int beginPacket(IPAddress ip, uint16_t port);
     int beginPacket(const char *host, uint16_t port);
     int endPacket();
-    size_t write(uint8_t);
+    size_t write(uint8_t byte);
     size_t write(const uint8_t *buffer, size_t size);
-
     using Print::write;
     int parsePacket();
     int available();
@@ -44,6 +46,8 @@ class WiFiUDP: public arduino::UDP {
     IPAddress remote_ip;
     uint16_t _port;
 
+    arduino::RingBufferN < WIFI_UDP_BUFFER_SIZE > txBuffer;
+
 };
 
 #endif /* WIFI_UDP_H */