libraries/WiFi: Refactor begin function.

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

Author: IFX-Anusha

libraries/WiFi/examples/WiFiUdpSendReceiveString/WiFiUdpSendReceiveString.ino

@@ -18,47 +18,38 @@ WiFiUDP::WiFiUDP() :
     _port(0) {
 }
 
-uint8_t WiFiUDP::begin(uint16_t port) {
-    // Initialize the socket for UDP
+uint8_t WiFiUDP::beginInternal(uint16_t port, IPAddress multicastIP) {
+
     socket.begin(SOCKET_PROTOCOL_UDP);
     if (socket.status() != SOCKET_STATUS_CREATED) {
         return 0;
     }
 
     socket.setReceiveOptCallback(receiveCallback, this);
 
-    // Bind the socket to the specified port
     socket.bind(port);
     if (socket.status() != SOCKET_STATUS_BOUND) {
         return 0;
-
-    }
-    return socket.status(); // Return the socket status
-}
-
-uint8_t WiFiUDP::beginMulticast(IPAddress ip, uint16_t port) {
-    // Initialize the socket for UDP
-    socket.begin(SOCKET_PROTOCOL_UDP);
-    if (socket.status() != SOCKET_STATUS_CREATED) {
-        return 0;
     }
 
-    socket.setReceiveOptCallback(receiveCallback, this);
-
-    // Bind the socket to the specified port
-    socket.bind(port);
-    if (socket.status() != SOCKET_STATUS_BOUND) {
-        return 0;
+    if (multicastIP != IPAddress(0, 0, 0, 0)) {
+        IPAddress local_ip = WiFi.localIP(); // Get the local IP address
 
+        // Bind the socket to the specified multicast address and port
+        cy_rslt_t result = socket.joinMulticastGroup(multicastIP, local_ip);
+        udp_assert_raise(result);
     }
 
-    IPAddress local_ip = WiFi.localIP(); // Get the local IP address
+    return socket.status(); // Return the socket status
+}
 
-    // Bind the socket to the specified multicast address and port
-    cy_rslt_t result = socket.joinMulticastGroup(ip, local_ip);
-    udp_assert_raise(result);
+uint8_t WiFiUDP::begin(uint16_t port) {
+    // Start the UDP socket on the specified port
+    return beginInternal(port, IPAddress(0, 0, 0, 0));
+}
 
-    return socket.status(); // Return the socket status
+uint8_t WiFiUDP::beginMulticast(IPAddress ip, uint16_t port) {
+    return beginInternal(port, ip);
 }
 
 void WiFiUDP::stop() {

libraries/WiFi/src/WiFiUdp.cpp

@@ -39,6 +39,7 @@ class WiFiUDP: public arduino::UDP {
 private:
     Socket socket;
 
+    uint8_t beginInternal(uint16_t port, IPAddress multicastIP = IPAddress());
     static cy_rslt_t receiveCallback(cy_socket_t socket_handle, void *arg);
 
     typedef struct {