lua_c binding basic framework(Not tested yet...)

binding/clua/Ip2region.c

@@ -1,13 +0,0 @@
-/**
- * Ip2region lua c binding
- *
- * @author  chenxin<chenxin619315@gmail.com>
- * @date    2018/10/04
-*/
-
-#include <lua.h>
-#include <lualib.h>
-#include <luaxlib.h>
-#include "../c/ip2region.h"
-
-

binding/lua/Ip2region.lua

@@ -30,7 +30,7 @@ _M.__index = _M;
 _M.__tostring = function(table)
     local t = {
         "dbFile=" .. table.dbFile,
-        "dbFileHandler=" .. table.dbFileHandler,
+        -- "dbFileHandler=" .. table.dbFileHandler,
         "headerLen=" .. table.headerLen,
         "firstIndexPtr" .. table.firstIndexPtr,
         "lastIndexPtr" .. table.lastIndexPtr,

binding/lua_c/Ip2region.c

@@ -0,0 +1,218 @@
+/**
+ * Ip2region lua c binding
+ *
+ * @author  chenxin<chenxin619315@gmail.com>
+ * @date    2018/10/04
+*/
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <lua.h>
+#include <luaxlib.h>
+#include "../c/ip2region.h"
+
+#define L_METATABLE_NAME "ip2region"
+
+
+/** create a new ip2region object with a specified dbFile */
+static int lua_ip2region_new(lua_State *L)
+{
+    ip2region_entry *self;
+    char *dbFile;
+
+    /* Check the arguments are valid */
+    dbFile = luaL_checkstring(L, 1);
+    if ( dbFile == NULL ) {
+        luaL_error(L, "dbFile cannot be empty");
+    }
+
+    /* Create the user data and push it onto the stack */
+    self = (ip2region_entry *) lua_newuserdata(L, sizeof(ip2region_entry));
+    /* Push the metatable onto the stack */
+    luaL_getmetatable(L, L_METATABLE_NAME);
+    /* Set the metatable on the userdata */
+    lua_setmetatable(L, -2);
+
+    /* Initialize the entry */
+    ip2region_create(self, dbFile);
+
+    return 1;
+}
+
+/** destroy the specified ip2region instance */
+static int lua_ip2region_destroy(lua_State *L)
+{
+    ip2region_entry *self;
+    self = (ip2region_entry *) luaL_checkudata(L, 1, L_METATABLE_NAME);
+    ip2region_destroy(self);
+    &self = NULL;
+
+    return 0;
+}
+
+
+
+#define get_search_args(entry, ip) \
+do {    \
+    entry = (ip2region_entry *) luaL_checkudata(L, 1, L_METATABLE_NAME); \
+    ip = luaL_checkstring(L, 2);  \
+while (0);
+
+
+#define set_search_result(data, rptr) \
+do { \
+    rptr = (datablock_entry *) lua_newuserdata(L, sizeof(datablock_entry)); \
+    rptr->city_id = data.city_id;   \
+    memcpy(rptr->data.region, data.region, strlen(data.region)); \
+while (0); \
+
+
+/** ip2region_memory_search wrapper */
+static int lua_ip2region_memory_search(lua_State *L)
+{
+    ip2region_entry *self;
+    char *addr;
+    datablock_entry data, *rptr;
+
+    // self = (ip2region_entry *) luaL_checkudata(L, 1, L_METATABLE_NAME);
+    // addr = luaL_checkstring(L, 2);
+    /* Check and get the search parameters */
+    get_search_args(self, addr);
+
+    /* do the memory search */
+    if ( ip2region_memory_search(self, ip2long(addr), &data) == 0 ) {
+        lua_pushnil(L);
+        return 1;
+    }
+    
+    // rptr = (datablock_entry *) lua_newuserdata(L, sizeof(datablock_entry));
+    // rptr->city_id = data.city_id;
+    // memcpy(rptr->data.region, data.region, strlen(data.region));
+    /* Set the return value */
+    set_search_result(data, rptr);
+
+    return 1;
+}
+
+/** ip2region_binary_search wrapper */
+static int lua_ip2region_binary_search(lua_State *L)
+{
+    ip2region_destroy *self;
+    char *addr;
+    datablock_entry data, *rptr;
+
+    /* Check and get the search parameters */
+    get_search_args(self, addr);
+
+    /* Do the binary search */
+    if ( ip2region_binary_search(self, ip2long(addr), &data) == 0 ) {
+        lua_pushnil(L);
+        return 1;
+    }
+
+    /* Set the return value */
+    set_search_result(data, rptr);
+
+    return 1;
+}
+
+/** ip2region_btree_search wrapper */
+static int lua_ip2region_btree_search(lua_State *L)
+{
+    ip2region_destroy *self;
+    char *addr;
+    datablock_entry data, *rptr;
+
+    /* Check and get the search parameters */
+    get_search_args(self, addr);
+
+    /* Do the btree search */
+    if ( ip2region_btree_search(self, ip2long(addr), &data) == 0 ) {
+        lua_pushnil(L);
+        return 1;
+    }
+
+    /* Set the return value */
+    set_search_result(data, rptr);
+
+    return 1;
+}
+
+
+/** ip2long wrapper */
+static int lua_ip2long(lua_State *L)
+{
+    char *addr;
+    uint_t ipval;
+
+    addr = luaL_checkstring(L, 1);
+
+    if ( (ipval = ip2long(addr)) == 0 ) {
+        lua_pushnil(L);
+        return 1;
+    }
+
+    lua_pushinteger(L, (ipval & 0x7FFFFFFF));
+
+    return 1;
+}
+
+static int lua_ip2region_tostring(lua_State *L)
+{
+    ip2region_entry *self;
+    self = (ip2region_entry *) luaL_checkudata(L, 1, L_METATABLE_NAME);
+
+    /* Push the string to return to lua */
+    lua_pushfstring(L, 
+        "dbFile=%s, headerLen=%d, fristIndexPtr=%d, lastIndexPtr=%d, totalBlocks=%d"
+        self->dbFile, self->headerLen, self->firstIndexPtr,
+        self->lastIndexPtr, self->totalBlocks
+    );
+
+    return 1;
+}
+
+
+
+/** module method array */
+static const struct luaL_Reg ip2region_methods[] = {
+    { "ip2long",        lua_ip2long },
+    { "memorySearch",   lua_ip2region_memory_search },
+    { "binarySearch",   lua_ip2region_binary_search },
+    { "btreeSerach",    lua_ip2region_binary_search },
+    { "__gc",           lua_ip2region_destroy },
+    { "__tostring",     ip2region_tostring },
+    { NULL, NULL },
+};
+
+/** module function array */
+static const struct luaL_Reg ip2region_functions[] = {
+    { "new", lua_ip2region_new },
+    { NULL,  NULL }
+};
+
+
+/** module open function interface */
+int luaopen_ip2region(lua_State *L)
+{
+    /* Create a metatable and push it onto the stack */
+    luaL_newmetatable(L, L_METATABLE_NAME);
+    /* Duplicate the metatable on the stack */
+    lua_pushvalue(L, -1);
+
+    /* Pop the first metatable off the stack
+     * and assign it to the __index of the second one.
+     * so we set the metatable to the table itself.
+    */
+    lua_setfield(L, -2, "__index");
+
+
+    /* Set the methods fo the metatable that could and should be 
+     * access via object:func in lua block 
+     */
+    luaL_setfuncs(L, ip2region_methods, 0);
+
+    /* Finally register the object.func functions 
+    * into the table witch at the top of the stack */
+    luaL_newlib(L, ip2region_functions);
+}