improve bytes

1 files changed, 10 insertions, 13 deletions

diff --git a/core/src/xmake/libc/dataptr.c b/core/src/xmake/libc/dataptr.c
index a15fc108b..d8d6c6f0d 100644
--- a/core/src/xmake/libc/dataptr.c
+++ b/core/src/xmake/libc/dataptr.c
@@ -38,18 +38,15 @@ tb_int_t xm_libc_dataptr(lua_State* lua)
     // check
     tb_assert_and_check_return_val(lua, 0);
 
-    if (xm_lua_ispointer(lua, 1))
-    {
-        tb_pointer_t data = (tb_pointer_t)xm_lua_topointer(lua, 1);
-        xm_lua_pushpointer(lua, data);
-        return 1;
-    }
-    else if (lua_isstring(lua, 1))
-    {
-        tb_char_t const* cstr = luaL_checkstring(lua, 1);
-        xm_lua_pushpointer(lua, (tb_pointer_t)cstr);
-        return 1;
-    }
-    xm_libc_return_error(lua, "libc.dataptr(invalid data)!");
+    tb_pointer_t data = tb_null;
+    if (lua_isstring(lua, 1))
+        data = (tb_pointer_t)luaL_checkstring(lua, 1);
+    else if (lua_isnumber(lua, 1))
+        data = (tb_pointer_t)(tb_size_t)lua_tointeger(lua, 1);
+    else if (xm_lua_ispointer(lua, 1))
+        data = (tb_pointer_t)xm_lua_topointer(lua, 1);
+    else xm_libc_return_error(lua, "libc.dataptr(invalid data)!");
+    lua_pushinteger(lua, (lua_Integer)data);
+    return 1;
 }