unix: socket.getaddrinfo: Accept family & socktype arguments.

pfalcon · pfalcon · commit 8eb802a55b1e · 2015-07-10T01:50:23.000+03:00
This usually allows to get just a single address entry.
diff --git 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);
 
