libraries/WiFi: UDP write added and modified socket.write.

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

Author: IFX-Anusha

libraries/WiFi/src/SecSocket.cpp

@@ -150,16 +150,25 @@ bool Socket::accept(Socket & client_socket) {
     return true;
 }
 
-uint32_t Socket::send(const void *data, uint32_t len) {
+uint32_t Socket::send(const void *data, uint32_t len, cy_socket_sockaddr_t *addr) {
     uint32_t bytes_sent = 0;
-    _last_error = cy_socket_send(socket, data, len,
-        CY_SOCKET_FLAGS_NONE, &bytes_sent);
+    switch (_protocol) {
+        case SOCKET_PROTOCOL_TCP:
+            _last_error = cy_socket_send(socket, data, len,
+                CY_SOCKET_FLAGS_NONE, &bytes_sent);
+            break;
+        case SOCKET_PROTOCOL_UDP:
+            _last_error = cy_socket_sendto(socket, data, len,
+                CY_SOCKET_FLAGS_NONE, addr,
+                sizeof(cy_socket_sockaddr_t), &bytes_sent);
+            break;
+    }
     if (_last_error != CY_RSLT_SUCCESS) {
         _status = SOCKET_STATUS_ERROR;
     }
-
     return bytes_sent;
 }
+
 uint32_t Socket::available() {
     return rx_buf.available();
 }

libraries/WiFi/src/SecSocket.h

@@ -41,7 +41,7 @@ class Socket {
 
     void listen(int max_connections);
     bool accept(Socket & client_socket);
-    uint32_t send(const void *data, uint32_t len);
+    uint32_t send(const void *data, uint32_t len, cy_socket_sockaddr_t *addr = NULL);
     uint32_t available();
     int peek();
     uint32_t receive(uint8_t *data, uint32_t len);

libraries/WiFi/src/WiFiUdp.cpp

@@ -62,11 +62,26 @@ int WiFiUDP::endPacket() {
 }
 
 size_t WiFiUDP::write(uint8_t) {
-    return 0;
+    return write(&byte, 1);
 }
 
 size_t WiFiUDP::write(const uint8_t *buffer, size_t size) {
-    return 0;
+    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 bytes_written = socket.send(buffer, size, &addr);
+    if (bytes_written != size) {
+        _status = SOCKET_STATUS_ERROR;
+        return 0;
+    }
+    return bytes_written;
 }
 
 int WiFiUDP::parsePacket() {