implement to open socket

1 files changed, 38 insertions, 2 deletions

diff --git a/core/src/xmake/io/socket_open.c b/core/src/xmake/io/socket_open.c
index 1c50c9d37..647cd247e 100644
--- a/core/src/xmake/io/socket_open.c
+++ b/core/src/xmake/io/socket_open.c
@@ -35,15 +35,51 @@
  */
 
 /*
- * io.socket_open()
+ * io.socket_open(socktype, family)
  */
 tb_int_t xm_io_socket_open(lua_State* lua)
 {
     // check
     tb_assert_and_check_return_val(lua, 0);
 
+    // get socket type 
+    tb_char_t const* socktype = luaL_checkstring(lua, 1);
+    tb_assert_and_check_return_val(socktype, 0);
+
+    // get address family 
+    tb_char_t const* family = luaL_checkstring(lua, 2);
+    tb_assert_and_check_return_val(family, 0);
+
+    // map socket type
+    tb_size_t t = TB_SOCKET_TYPE_NONE;
+    if (!tb_strcmp(socktype, "tcp"))
+        t = TB_SOCKET_TYPE_TCP;
+    else if (!tb_strcmp(socktype, "udp"))
+        t = TB_SOCKET_TYPE_UDP;
+    else if (!tb_strcmp(socktype, "icmp"))
+        t = TB_SOCKET_TYPE_ICMP;
+    else
+    {
+        lua_pushnil(lua);
+        lua_pushliteral(lua, "invalid socket type!");
+        return 2;
+    }
+
+    // map address family
+    tb_size_t f = TB_IPADDR_FAMILY_NONE;
+    if (!tb_strcmp(family, "ipv4"))
+        f = TB_IPADDR_FAMILY_IPV4;
+    else if (!tb_strcmp(family, "ipv6"))
+        f = TB_IPADDR_FAMILY_IPV6;
+    else
+    {
+        lua_pushnil(lua);
+        lua_pushliteral(lua, "invalid address family!");
+        return 2;
+    }
+
     // init socket
-    tb_socket_ref_t sock = tb_socket_init(TB_SOCKET_TYPE_TCP, TB_IPADDR_FAMILY_IPV4);
+    tb_socket_ref_t sock = tb_socket_init(t, f);
     if (sock) lua_pushlightuserdata(lua, (tb_pointer_t)sock);
     else lua_pushnil(lua);
     return 1;