set the connected status of a socket after UDP & raw connect

jepler · jepler · commit 8acfc0c70ce1 · 2024-05-15T13:05:14.000-05:00
diff --git 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

Original file line number	Diff line number	Diff line change
`@@ -1014,11 +1014,17 @@ void common_hal_socketpool_socket_connect(socketpool_socket_obj_t *socket,`
`1014`	`1014`	`}`
`1015`	`1015`	`case MOD_NETWORK_SOCK_DGRAM: {`
`1016`	`1016`	`err = udp_connect(socket->pcb.udp, &dest, port);`
	`1017`	`+ if (err == ERR_OK) {`
	`1018`	`+ socket->state = STATE_CONNECTED;`
	`1019`	`+ }`
`1017`	`1020`	`break;`
`1018`	`1021`	`}`
`1019`	`1022`	`#if MICROPY_PY_LWIP_SOCK_RAW`
`1020`	`1023`	`case MOD_NETWORK_SOCK_RAW: {`
`1021`	`1024`	`err = raw_connect(socket->pcb.raw, &dest);`
	`1025`	`+ if (err == ERR_OK) {`
	`1026`	`+ socket->state = STATE_CONNECTED;`
	`1027`	`+ }`
`1022`	`1028`	`break;`
`1023`	`1029`	`}`
`1024`	`1030`	`#endif`