Skip to content

Commit f555ed6

Browse files
fabledfabled
committed
initial commit
0 parents  commit f555ed6

File tree

6 files changed

+216
-0
lines changed

6 files changed

+216
-0
lines changed

‎.gitignore

+1
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
maxminddb.so

‎LICENSE

+19
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
Copyright (c) 2014 Timo Teräs
2+
3+
Permission is hereby granted, free of charge, to any person obtaining a copy
4+
of this software and associated documentation files (the "Software"), to deal
5+
in the Software without restriction, including without limitation the rights
6+
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
7+
copies of the Software, and to permit persons to whom the Software is
8+
furnished to do so, subject to the following conditions:
9+
10+
The above copyright notice and this permission notice shall be included in
11+
all copies or substantial portions of the Software.
12+
13+
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
14+
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
15+
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
16+
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
17+
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
18+
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
19+
THE SOFTWARE.

‎Makefile

+15
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
LUA_PKG ?= lua5.2
2+
LUA_CFLAGS ?= $(shell pkg-config --cflags $(LUA_PKG))
3+
LUA_CMOD ?= $(shell pkg-config --variable=INSTALL_CMOD $(LUA_PKG))
4+
5+
CFLAGS ?= -g
6+
7+
maxminddb.so: maxminddb.c
8+
$(CC) $(CFLAGS) -fPIC -std=c99 $(LUA_CFLAGS) -shared $< -o $@ $(LDFLAGS) -lmaxminddb
9+
10+
install:
11+
install -d $(DESTDIR)$(LUA_CMOD)
12+
install -c maxminddb.so $(DESTDIR)$(LUA_CMOD)
13+
14+
clean:
15+
rm maxminddb.so

‎README.md

Whitespace-only changes.

‎maxminddb.c

+176
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,176 @@
1+
#include <netdb.h>
2+
#include <string.h>
3+
4+
#include <lua.h>
5+
#include <lauxlib.h>
6+
#include <maxminddb.h>
7+
8+
#if LUA_VERSION_NUM >= 502
9+
10+
#define luaL_register(L,n,f) luaL_newlib(L,f)
11+
12+
#else
13+
14+
static void luaL_setmetatable(lua_State *L, const char *tname)
15+
{
16+
luaL_getmetatable(L, tname);
17+
lua_setmetatable(L, -2);
18+
}
19+
20+
static int lua_absindex(lua_State *L, int idx)
21+
{
22+
return (idx > 0 || idx <= LUA_REGISTRYINDEX)? idx : lua_gettop(L) + idx + 1;
23+
}
24+
25+
static void luaL_setfuncs(lua_State *L, const luaL_Reg *l, int nup)
26+
{
27+
int i, t = lua_absindex(L, -1 - nup);
28+
29+
for (; l->name; l++) {
30+
for (i = 0; i < nup; i++)
31+
lua_pushvalue(L, -nup);
32+
lua_pushcclosure(L, l->func, nup);
33+
lua_setfield(L, t, l->name);
34+
}
35+
lua_pop(L, nup);
36+
}
37+
38+
#endif
39+
40+
static void addclass(lua_State *L, const char *name, const luaL_Reg *methods, const luaL_Reg *metamethods)
41+
{
42+
if (!luaL_newmetatable(L, name)) return;
43+
if (metamethods)
44+
luaL_setfuncs(L, metamethods, 0);
45+
if (methods) {
46+
lua_newtable(L);
47+
luaL_setfuncs(L, methods, 0);
48+
lua_setfield(L, -2, "__index");
49+
}
50+
lua_pop(L, 1);
51+
}
52+
53+
static void *allocudata(lua_State *L, size_t size, const char *tname)
54+
{
55+
void *p = memset(lua_newuserdata(L, size), 0, size);
56+
luaL_setmetatable(L, tname);
57+
return p;
58+
}
59+
60+
static int mm_error(lua_State *L, int err)
61+
{
62+
lua_pushstring(L, MMDB_strerror(err));
63+
return lua_error(L);
64+
}
65+
66+
#define MMDB_RESULT_CLASS "MMDB_lookup_result_s"
67+
68+
static int mmdbres_get(lua_State *L)
69+
{
70+
MMDB_entry_data_s data;
71+
MMDB_lookup_result_s *res = luaL_checkudata(L, 1, MMDB_RESULT_CLASS);
72+
const char *keys[16];
73+
int r, i;
74+
75+
for (i = 2; i <= lua_gettop(L) && i < 15; i++)
76+
keys[i-2] = luaL_checkstring(L, i);
77+
keys[i-2] = 0;
78+
79+
r = MMDB_aget_value(&res->entry, &data, keys);
80+
if (!data.has_data) return mm_error(L, r);
81+
82+
switch (data.type) {
83+
case MMDB_DATA_TYPE_BYTES:
84+
lua_pushlstring(L, data.bytes, data.data_size);
85+
return 1;
86+
return 1;
87+
case MMDB_DATA_TYPE_UTF8_STRING:
88+
lua_pushlstring(L, data.utf8_string, data.data_size);
89+
return 1;
90+
case MMDB_DATA_TYPE_DOUBLE:
91+
lua_pushnumber(L, data.double_value);
92+
return 1;
93+
case MMDB_DATA_TYPE_FLOAT:
94+
lua_pushnumber(L, data.float_value);
95+
return 1;
96+
case MMDB_DATA_TYPE_UINT16:
97+
lua_pushinteger(L, data.uint16);
98+
return 1;
99+
case MMDB_DATA_TYPE_UINT32:
100+
lua_pushinteger(L, data.uint32);
101+
return 1;
102+
case MMDB_DATA_TYPE_INT32:
103+
lua_pushinteger(L, data.int32);
104+
return 1;
105+
}
106+
107+
lua_pushstring(L, "Unsupported maxmind data type");
108+
return lua_error(L);
109+
}
110+
111+
static const luaL_Reg mmdbres_methods[] = {
112+
{ "get", mmdbres_get },
113+
{ 0, 0 },
114+
};
115+
116+
117+
#define MMDB_CLASS "MMDB_s"
118+
119+
static int mmdb_gc(lua_State *L)
120+
{
121+
MMDB_s *mmdb = luaL_checkudata(L, 1, MMDB_CLASS);
122+
MMDB_close(mmdb);
123+
return 0;
124+
}
125+
126+
static const luaL_Reg mmdb_metatable[] = {
127+
{ "__gc", mmdb_gc },
128+
{ 0, 0 },
129+
};
130+
131+
static int mmdb_lookup(lua_State *L)
132+
{
133+
int gaierr, mmerr;
134+
MMDB_s *mmdb = luaL_checkudata(L, 1, MMDB_CLASS);
135+
const char *addr = luaL_checkstring(L, 2);
136+
MMDB_lookup_result_s *res = allocudata(L, sizeof(MMDB_lookup_result_s), MMDB_RESULT_CLASS);
137+
138+
*res = MMDB_lookup_string(mmdb, addr, &gaierr, &mmerr);
139+
if (gaierr != 0) {
140+
lua_pushstring(L, gai_strerror(gaierr));
141+
return lua_error(L);
142+
}
143+
if (!res->found_entry) return mm_error(L, mmerr);
144+
145+
return 1;
146+
}
147+
148+
static const luaL_Reg mmdb_methods[] = {
149+
{ "lookup", mmdb_lookup },
150+
{ 0, 0 },
151+
};
152+
153+
static int mm_open(lua_State *L)
154+
{
155+
const char *filename = luaL_checkstring(L, 1);
156+
MMDB_s *mmdb = allocudata(L, sizeof(MMDB_s), MMDB_CLASS);
157+
int r;
158+
159+
r = MMDB_open(filename, 0, mmdb);
160+
if (r != MMDB_SUCCESS) return mm_error(L, r);
161+
162+
return 1;
163+
}
164+
165+
LUALIB_API int luaopen_maxminddb(lua_State * L)
166+
{
167+
static const luaL_Reg api[] = {
168+
{ "open", mm_open },
169+
{ 0, 0 },
170+
};
171+
172+
luaL_register(L, "maxminddb", api);
173+
addclass(L, MMDB_CLASS, mmdb_methods, mmdb_metatable);
174+
addclass(L, MMDB_RESULT_CLASS, mmdbres_methods, NULL);
175+
return 1;
176+
}

‎test.lua

+5
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
local mm = require 'maxminddb'
2+
local db = mm.open('/var/lib/libmaxminddb/GeoLite2-City.mmdb')
3+
4+
local res = db:lookup('8.8.8.8')
5+
print(res:get("country", "names", "en"), res:get("location", "longitude"), res:get("location", "latitude"))

0 commit comments

Comments
 (0)