Wifi UDP class and Begin function

libraries/WiFi/src/SecSocket.cpp

@@ -14,16 +14,30 @@ Socket::Socket():
     _status(SOCKET_STATUS_UNINITED),
     _last_error(CY_RSLT_SUCCESS),
     remote_ip(0, 0, 0, 0),
-    _port(0) {
+    _port(0),
+    _protocol(0) {
 
 }
 
-void Socket::begin() {
+void Socket::begin(socket_protocol_t protocol) {
     _last_error = Socket::global_sockets_init();
     socket_assert(_last_error);
+    int socket_type = 0, socket_proto = 0;
+
+    _protocol = protocol;
+    switch (_protocol) {
+        case SOCKET_PROTOCOL_TCP:
+            socket_type = CY_SOCKET_TYPE_STREAM;
+            socket_proto = CY_SOCKET_IPPROTO_TCP;
+            break;
+        case SOCKET_PROTOCOL_UDP:
+            socket_type = CY_SOCKET_TYPE_DGRAM;
+            socket_proto = CY_SOCKET_IPPROTO_UDP;
+            break;
+    }
 
-    _last_error = cy_socket_create(CY_SOCKET_DOMAIN_AF_INET, CY_SOCKET_TYPE_STREAM,
-        CY_SOCKET_IPPROTO_TCP, &socket);
+    _last_error = cy_socket_create(CY_SOCKET_DOMAIN_AF_INET, socket_type,
+        socket_proto, &socket);
     socket_assert(_last_error);
 
     _status = SOCKET_STATUS_CREATED;

libraries/WiFi/src/SecSocket.h

@@ -16,13 +16,18 @@ typedef enum {
     SOCKET_STATUS_ERROR = -1
 } socket_status_t;
 
+typedef enum {
+    SOCKET_PROTOCOL_TCP = 0,
+    SOCKET_PROTOCOL_UDP,
+} socket_protocol_t;
+
 class Socket {
 
 public:
 
     Socket();
 
-    void begin();
+    void begin(socket_protocol_t protocol);
     void end();
 
     void setTimeout(uint32_t timeout);
@@ -58,6 +63,7 @@ class Socket {
 
     IPAddress remote_ip;
     uint16_t _port;
+    socket_protocol_t _protocol;
 
     void setOptCallback(int optname, cy_socket_callback_t cback, void *arg);
 

libraries/WiFi/src/WiFiClient.cpp

@@ -14,7 +14,7 @@ WiFiClient::WiFiClient() :
 }
 
 int WiFiClient::connect(IPAddress ip, uint16_t port) {
-    socket->begin();
+    socket->begin(SOCKET_PROTOCOL_TCP);
 
     socket->setReceiveOptCallback(receiveCallback, this);
     socket->setDisconnectOptCallback(disconnectionCallback, this);
@@ -23,7 +23,7 @@ int WiFiClient::connect(IPAddress ip, uint16_t port) {
 }
 
 int WiFiClient::connect(const char *host, uint16_t port) {
-    socket->begin();
+    socket->begin(SOCKET_PROTOCOL_TCP);
 
     socket->setReceiveOptCallback(receiveCallback, this);
     socket->setDisconnectOptCallback(disconnectionCallback, this);

libraries/WiFi/src/WiFiServer.cpp

@@ -9,7 +9,7 @@ WiFiServer::WiFiServer() {
 }
 
 void WiFiServer::begin(uint16_t port) {
-    socket.begin();
+    socket.begin(SOCKET_PROTOCOL_TCP);
 
     socket.setTimeout(SERVER_RECV_TIMEOUT_MS);
     socket.setConnectOptCallback(connectionCallback, this);

libraries/WiFi/src/WiFiUdp.cpp

@@ -0,0 +1,97 @@
+#include "WiFiUdp.h"
+#include <Arduino.h>
+
+#define udp_assert(cy_ret)   if (cy_ret != CY_RSLT_SUCCESS) { \
+            _status = SOCKET_STATUS_ERROR; \
+            return; \
+}
+
+#define udp_assert_raise(cy_ret)   if (cy_ret != CY_RSLT_SUCCESS) { \
+            return cy_ret; \
+}
+
+WiFiUDP::WiFiUDP() :
+    client_handle(nullptr),
+    _status(SOCKET_STATUS_UNINITED),
+    _last_error(CY_RSLT_SUCCESS),
+    remote_ip(0, 0, 0, 0),
+    _port(0) {
+}
+
+uint8_t WiFiUDP::begin(uint16_t port) {
+    _port = port;
+
+    // Initialize the socket for UDP
+    socket.begin(SOCKET_PROTOCOL_UDP);
+    if (socket.status() != SOCKET_STATUS_CREATED) {
+        return 0; // Return 0 if socket creation fails
+    }
+    // Bind the socket to the specified port
+    socket.bind(port);
+    if (socket.status() != SOCKET_STATUS_BOUND) {
+        return 0; // Return 0 if binding fail
+
+    }
+    return socket.status(); // Return the socket status
+}
+
+
+void WiFiUDP::stop() {
+
+}
+
+int WiFiUDP::beginPacket(IPAddress ip, uint16_t port) {
+    return 0;
+}
+
+int WiFiUDP::beginPacket(const char *host, uint16_t port) {
+    return 0;
+}
+
+int WiFiUDP::endPacket() {
+    return 0;
+}
+
+size_t WiFiUDP::write(uint8_t) {
+    return 0;
+}
+
+size_t WiFiUDP::write(const uint8_t *buffer, size_t size) {
+    return 0;
+}
+
+int WiFiUDP::parsePacket() {
+    return 0;
+}
+
+int WiFiUDP::available() {
+    return 0;
+}
+
+int WiFiUDP::read() {
+    return 0;
+}
+
+int WiFiUDP::read(unsigned char *buffer, size_t len) {
+    return 0;
+}
+
+int WiFiUDP::read(char *buffer, size_t len) {
+    return 0;
+}
+
+int WiFiUDP::peek() {
+    return 0;
+}
+
+void WiFiUDP::flush() {
+    return;
+}
+
+IPAddress WiFiUDP::remoteIP() {
+    return IPAddress();
+}
+
+uint16_t WiFiUDP::remotePort() {
+    return 0;
+}

libraries/WiFi/src/WiFiUdp.h

@@ -0,0 +1,49 @@
+#ifndef WIFI_UDP_H
+#define WIFI_UDP_H
+
+#include <stdint.h>
+#include <string.h>
+#include <api/Udp.h>
+#include "WiFiClient.h"
+#include "WiFi.h"
+#include "WiFiServer.h"
+#include "SecSocket.h"
+
+class WiFiUDP: public arduino::UDP {
+public:
+    WiFiUDP();
+    uint8_t begin(uint16_t);
+
+    // yet to implement these functions
+    void stop();
+    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(const uint8_t *buffer, size_t size);
+
+    using Print::write;
+    int parsePacket();
+    int available();
+    int read();
+    int read(unsigned char *buffer, size_t len);
+    int read(char *buffer, size_t len);
+    int peek();
+    void flush();
+    IPAddress remoteIP();
+    uint16_t remotePort();
+
+
+
+private:
+    Socket socket;
+
+    cy_socket_t client_handle;
+    socket_status_t _status;
+    cy_rslt_t _last_error;
+    IPAddress remote_ip;
+    uint16_t _port;
+
+};
+
+#endif /* WIFI_UDP_H */