2 * lmo - Lua Machine Objects - Lua binding
4 * Copyright (C) 2009 Jo-Philipp Wich <xm@subsignal.org>
6 * Licensed under the Apache License, Version 2.0 (the "License");
7 * you may not use this file except in compliance with the License.
8 * You may obtain a copy of the License at
10 * http://www.apache.org/licenses/LICENSE-2.0
12 * Unless required by applicable law or agreed to in writing, software
13 * distributed under the License is distributed on an "AS IS" BASIS,
14 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 * See the License for the specific language governing permissions and
16 * limitations under the License.
19 #include "lmo_lualib.h"
21 extern char _lmo_error[1024];
24 static int lmo_L_open(lua_State *L) {
25 const char *filename = luaL_checklstring(L, 1, NULL);
26 lmo_archive_t *ar, **udata;
28 if( (ar = lmo_open(filename)) != NULL )
30 if( (udata = lua_newuserdata(L, sizeof(lmo_archive_t *))) != NULL )
33 luaL_getmetatable(L, LMO_ARCHIVE_META);
34 lua_setmetatable(L, -2);
39 lua_pushstring(L, "out of memory");
44 lua_pushstring(L, lmo_error());
48 static int lmo_L_hash(lua_State *L) {
49 const char *data = luaL_checkstring(L, 1);
50 uint32_t hash = sfh_hash(data, strlen(data));
51 lua_pushnumber(L, hash);
55 static int _lmo_lookup(lua_State *L, lmo_archive_t *ar, uint32_t hash) {
56 lmo_entry_t *e = ar->index;
60 if( e->key_id == hash )
62 lua_pushlstring(L, &ar->mmap[e->offset], e->length);
73 static int lmo_L_get(lua_State *L) {
74 lmo_archive_t **ar = luaL_checkudata(L, 1, LMO_ARCHIVE_META);
75 uint32_t hash = (uint32_t) luaL_checknumber(L, 2);
76 return _lmo_lookup(L, *ar, hash);
79 static int lmo_L_lookup(lua_State *L) {
80 lmo_archive_t **ar = luaL_checkudata(L, 1, LMO_ARCHIVE_META);
81 const char *key = luaL_checkstring(L, 2);
82 uint32_t hash = sfh_hash(key, strlen(key));
83 return _lmo_lookup(L, *ar, hash);
86 static int lmo_L_foreach(lua_State *L) {
87 lmo_archive_t **ar = luaL_checkudata(L, 1, LMO_ARCHIVE_META);
88 lmo_entry_t *e = (*ar)->index;
90 if( lua_isfunction(L, 2) )
95 lua_pushinteger(L, e->key_id);
96 lua_pushlstring(L, &(*ar)->mmap[e->offset], e->length);
97 lua_pcall(L, 2, 0, 0);
105 static int lmo_L__gc(lua_State *L) {
106 lmo_archive_t **ar = luaL_checkudata(L, 1, LMO_ARCHIVE_META);
116 static int lmo_L__tostring(lua_State *L) {
117 lmo_archive_t **ar = luaL_checkudata(L, 1, LMO_ARCHIVE_META);
118 lua_pushfstring(L, "LMO Archive (%d bytes)", (*ar)->length);
124 static const luaL_reg M[] = {
125 {"close", lmo_L__gc},
127 {"lookup", lmo_L_lookup},
128 {"foreach", lmo_L_foreach},
129 {"__tostring", lmo_L__tostring},
135 static const luaL_reg R[] = {
136 {"open", lmo_L_open},
137 {"hash", lmo_L_hash},
141 LUALIB_API int luaopen_lmo(lua_State *L) {
142 luaL_newmetatable(L, LMO_LUALIB_META);
143 luaL_register(L, NULL, R);
144 lua_pushvalue(L, -1);
145 lua_setfield(L, -2, "__index");
146 lua_setglobal(L, LMO_LUALIB_META);
148 luaL_newmetatable(L, LMO_ARCHIVE_META);
149 luaL_register(L, NULL, M);
150 lua_pushvalue(L, -1);
151 lua_setfield(L, -2, "__index");
152 lua_setglobal(L, LMO_ARCHIVE_META);