diff options
| -rw-r--r-- | unix/modsocket.c | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/unix/modsocket.c b/unix/modsocket.c index 738581a0af..3d9a6ddd3a 100644 --- a/unix/modsocket.c +++ b/unix/modsocket.c @@ -357,7 +357,7 @@ STATIC MP_DEFINE_CONST_FUN_OBJ_1(mod_socket_gethostbyname_obj, mod_socket_gethos STATIC mp_obj_t mod_socket_getaddrinfo(mp_uint_t n_args, const mp_obj_t *args) { // TODO: Implement all args - assert(n_args == 2); + assert(n_args >= 2 && n_args <= 4); assert(MP_OBJ_IS_STR(args[0])); const char *host = mp_obj_str_get_str(args[0]); @@ -390,6 +390,13 @@ STATIC mp_obj_t mod_socket_getaddrinfo(mp_uint_t n_args, const mp_obj_t *args) { serv = mp_obj_str_get_str(args[1]); } + if (n_args > 2) { + hints.ai_family = MP_OBJ_SMALL_INT_VALUE(args[2]); + if (n_args > 3) { + hints.ai_socktype = MP_OBJ_SMALL_INT_VALUE(args[3]); + } + } + struct addrinfo *addr_list; int res = getaddrinfo(host, serv, &hints, &addr_list); |