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

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

Author: IFX-Anusha

libraries/WiFi/src/SecSocket.cpp

@@ -152,14 +152,24 @@ bool Socket::accept(Socket & client_socket) {
 
 uint32_t Socket::send(const void *data, uint32_t len) {
     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, &address,
+                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/WiFiUdp.cpp

@@ -53,11 +53,19 @@ 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;
+    }
+    size_t bytes_written = socket.send(buffer, size);
+    if (bytes_written != size) {
+        _status = SOCKET_STATUS_ERROR;
+        return 0;
+    }
+    return bytes_written;
 }
 
 int WiFiUDP::parsePacket() {