From 8acfc0c70ce14f989fcecc0e160ed3501988ad9a Mon Sep 17 00:00:00 2001
From: Jeff Epler <jepler@gmail.com>
Date: Wed, 15 May 2024 13:05:14 -0500
Subject: [PATCH] set the connected status of a socket after UDP & raw connect

---
 ports/raspberrypi/common-hal/socketpool/Socket.c | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/ports/raspberrypi/common-hal/socketpool/Socket.c b/ports/raspberrypi/common-hal/socketpool/Socket.c
index d305f63c7882..4c290acfbc73 100644
--- a/ports/raspberrypi/common-hal/socketpool/Socket.c
+++ b/ports/raspberrypi/common-hal/socketpool/Socket.c
@@ -1014,11 +1014,17 @@ void common_hal_socketpool_socket_connect(socketpool_socket_obj_t *socket,
         }
         case MOD_NETWORK_SOCK_DGRAM: {
             err = udp_connect(socket->pcb.udp, &dest, port);
+            if (err == ERR_OK) {
+                socket->state = STATE_CONNECTED;
+            }
             break;
         }
         #if MICROPY_PY_LWIP_SOCK_RAW
         case MOD_NETWORK_SOCK_RAW: {
             err = raw_connect(socket->pcb.raw, &dest);
+            if (err == ERR_OK) {
+                socket->state = STATE_CONNECTED;
+            }
             break;
         }
         #endif