libraries/WiFi: Write function.

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

Author: IFX-Anusha

libraries/WiFi/src/WiFiUdp.cpp

@@ -59,14 +59,26 @@ int WiFiUDP::beginPacket(const char *host, uint16_t port) {
 }
 
 int WiFiUDP::endPacket() {
+    return 0;
 }
 
-size_t WiFiUDP::write(uint8_t) {
-    return 0;
+size_t WiFiUDP::write(uint8_t byte) {
+    return write(&byte, 1);
 }
 
 size_t WiFiUDP::write(const uint8_t *buffer, size_t size) {
-    return 0;
+    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
+        }
+    }
+
+    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 */