diff options
| -rw-r--r-- | libs/lmo/src/lmo_lualib.c | 45 |
1 files changed, 38 insertions, 7 deletions
diff --git a/libs/lmo/src/lmo_lualib.c b/libs/lmo/src/lmo_lualib.c index 3f9170e88..878ca02b2 100644 --- a/libs/lmo/src/lmo_lualib.c +++ b/libs/lmo/src/lmo_lualib.c @@ -1,5 +1,5 @@ /* - * lmo - Lua Machine Objects - Lookup utility + * lmo - Lua Machine Objects - Lua binding * * Copyright (C) 2009 Jo-Philipp Wich <xm@subsignal.org> * @@ -52,17 +52,14 @@ static int lmo_L_hash(lua_State *L) { return 1; } -static int lmo_L_lookup(lua_State *L) { - lmo_archive_t **ar = luaL_checkudata(L, 1, LMO_ARCHIVE_META); - lmo_entry_t *e = (*ar)->index; - const char *key = luaL_checkstring(L, 2); - uint32_t hash = sfh_hash(key, strlen(key)); +static int _lmo_lookup(lua_State *L, lmo_archive_t *ar, uint32_t hash) { + lmo_entry_t *e = ar->index; while( e != NULL ) { if( e->key_id == hash ) { - lua_pushlstring(L, &(*ar)->mmap[e->offset], e->length); + lua_pushlstring(L, &ar->mmap[e->offset], e->length); return 1; } @@ -73,6 +70,38 @@ static int lmo_L_lookup(lua_State *L) { return 1; } +static int lmo_L_get(lua_State *L) { + lmo_archive_t **ar = luaL_checkudata(L, 1, LMO_ARCHIVE_META); + uint32_t hash = (uint32_t) luaL_checknumber(L, 2); + return _lmo_lookup(L, *ar, hash); +} + +static int lmo_L_lookup(lua_State *L) { + lmo_archive_t **ar = luaL_checkudata(L, 1, LMO_ARCHIVE_META); + const char *key = luaL_checkstring(L, 2); + uint32_t hash = sfh_hash(key, strlen(key)); + return _lmo_lookup(L, *ar, hash); +} + +static int lmo_L_foreach(lua_State *L) { + lmo_archive_t **ar = luaL_checkudata(L, 1, LMO_ARCHIVE_META); + lmo_entry_t *e = (*ar)->index; + + if( lua_isfunction(L, 2) ) + { + while( e != NULL ) + { + lua_pushvalue(L, 2); + lua_pushinteger(L, e->key_id); + lua_pushlstring(L, &(*ar)->mmap[e->offset], e->length); + lua_pcall(L, 2, 0, 0); + e = e->next; + } + } + + return 0; +} + static int lmo_L__gc(lua_State *L) { lmo_archive_t **ar = luaL_checkudata(L, 1, LMO_ARCHIVE_META); @@ -94,7 +123,9 @@ static int lmo_L__tostring(lua_State *L) { /* method table */ static const luaL_reg M[] = { {"close", lmo_L__gc}, + {"get", lmo_L_get}, {"lookup", lmo_L_lookup}, + {"foreach", lmo_L_foreach}, {"__tostring", lmo_L__tostring}, {"__gc", lmo_L__gc}, {NULL, NULL} |