From 3d46fe67bf0bfc848741a76e945b16b2e45e74cb Mon Sep 17 00:00:00 2001 From: iabdalkader Date: Wed, 29 Mar 2023 20:42:44 +0200 Subject: [PATCH] extmod/network_ninaw10: Check socket types when creating new sockets. The NINA socket types have the same values as modnetwork, but that may change in the future. So check the socket types passed to socket() and convert them (if needed) to their respective Nina socket types. Also remove the unnecessary socket type check code from bind(), as pointed out by @robert-hh. --- extmod/network_ninaw10.c | 32 +++++++++++++++++--------------- 1 file changed, 17 insertions(+), 15 deletions(-) diff --git a/extmod/network_ninaw10.c b/extmod/network_ninaw10.c index 8e806cb423..1f2e72707f 100644 --- a/extmod/network_ninaw10.c +++ b/extmod/network_ninaw10.c @@ -486,13 +486,29 @@ STATIC int network_ninaw10_socket_listening(mod_network_socket_obj_t *socket, in STATIC int network_ninaw10_socket_socket(mod_network_socket_obj_t *socket, int *_errno) { debug_printf("socket_socket(%d %d %d)\n", socket->domain, socket->type, socket->proto); + uint8_t socket_type; + + switch (socket->type) { + case MOD_NETWORK_SOCK_STREAM: + socket_type = NINA_SOCKET_TYPE_TCP; + break; + + case MOD_NETWORK_SOCK_DGRAM: + socket_type = NINA_SOCKET_TYPE_UDP; + break; + + default: + *_errno = MP_EINVAL; + return -1; + } + if (socket->domain != MOD_NETWORK_AF_INET) { *_errno = MP_EAFNOSUPPORT; return -1; } // open socket - int fd = nina_socket_socket(socket->type, socket->proto); + int fd = nina_socket_socket(socket_type, socket->proto); if (fd < 0) { nina_socket_errno(_errno); debug_printf("socket_socket() -> errno %d\n", *_errno); @@ -522,20 +538,6 @@ STATIC void network_ninaw10_socket_close(mod_network_socket_obj_t *socket) { STATIC int network_ninaw10_socket_bind(mod_network_socket_obj_t *socket, byte *ip, mp_uint_t port, int *_errno) { debug_printf("socket_bind(%d, %d)\n", socket->fileno, port); - uint8_t type; - switch (socket->type) { - case MOD_NETWORK_SOCK_STREAM: - type = NINA_SOCKET_TYPE_TCP; - break; - - case MOD_NETWORK_SOCK_DGRAM: - type = NINA_SOCKET_TYPE_UDP; - break; - - default: - *_errno = MP_EINVAL; - return -1; - } int ret = nina_socket_bind(socket->fileno, ip, port); if (ret < 0) {