compat-5.1.c (3081B)
1 /* 2 ** Compat-5.1 3 ** Copyright Kepler Project 2004-2006 (http://www.keplerproject.org/compat) 4 ** $Id: compat-5.1.c,v 1.13 2006/02/20 21:12:47 carregal Exp $ 5 */ 6 7 #include <stdio.h> 8 #include <string.h> 9 #include "lua.h" 10 #include "lauxlib.h" 11 #include "compat-5.1.h" 12 13 static void getfield(lua_State *L, int idx, const char *name) { 14 const char *end = strchr(name, '.'); 15 lua_pushvalue(L, idx); 16 while (end) { 17 lua_pushlstring(L, name, end - name); 18 lua_gettable(L, -2); 19 lua_remove(L, -2); 20 if (lua_isnil(L, -1)) return; 21 name = end+1; 22 end = strchr(name, '.'); 23 } 24 lua_pushstring(L, name); 25 lua_gettable(L, -2); 26 lua_remove(L, -2); 27 } 28 29 static void setfield(lua_State *L, int idx, const char *name) { 30 const char *end = strchr(name, '.'); 31 lua_pushvalue(L, idx); 32 while (end) { 33 lua_pushlstring(L, name, end - name); 34 lua_gettable(L, -2); 35 /* create table if not found */ 36 if (lua_isnil(L, -1)) { 37 lua_pop(L, 1); 38 lua_newtable(L); 39 lua_pushlstring(L, name, end - name); 40 lua_pushvalue(L, -2); 41 lua_settable(L, -4); 42 } 43 lua_remove(L, -2); 44 name = end+1; 45 end = strchr(name, '.'); 46 } 47 lua_pushstring(L, name); 48 lua_pushvalue(L, -3); 49 lua_settable(L, -3); 50 lua_pop(L, 2); 51 } 52 53 LUALIB_API void luaL_module(lua_State *L, const char *libname, 54 const luaL_reg *l, int nup) { 55 if (libname) { 56 getfield(L, LUA_GLOBALSINDEX, libname); /* check whether lib already exists */ 57 if (lua_isnil(L, -1)) { 58 int env, ns; 59 lua_pop(L, 1); /* get rid of nil */ 60 lua_pushliteral(L, "require"); 61 lua_gettable(L, LUA_GLOBALSINDEX); /* look for require */ 62 lua_getfenv(L, -1); /* getfenv(require) */ 63 lua_remove(L, -2); /* remove function require */ 64 env = lua_gettop(L); 65 66 lua_newtable(L); /* create namespace for lib */ 67 ns = lua_gettop(L); 68 getfield(L, env, "package.loaded"); /* get package.loaded table */ 69 if (lua_isnil(L, -1)) { /* create package.loaded table */ 70 lua_pop(L, 1); /* remove previous result */ 71 lua_newtable(L); 72 lua_pushvalue(L, -1); 73 setfield(L, env, "package.loaded"); 74 } 75 else if (!lua_istable(L, -1)) 76 luaL_error(L, "name conflict for library `%s'", libname); 77 lua_pushstring(L, libname); 78 lua_pushvalue(L, ns); 79 lua_settable(L, -3); /* package.loaded[libname] = ns */ 80 lua_pop(L, 1); /* get rid of package.loaded table */ 81 lua_pushvalue(L, ns); /* copy namespace */ 82 setfield(L, LUA_GLOBALSINDEX, libname); 83 lua_remove (L, env); /* remove env */ 84 } 85 lua_insert(L, -(nup+1)); /* move library table to below upvalues */ 86 } 87 for (; l->name; l++) { 88 int i; 89 lua_pushstring(L, l->name); 90 for (i=0; i<nup; i++) /* copy upvalues to the top */ 91 lua_pushvalue(L, -(nup+1)); 92 lua_pushcclosure(L, l->func, nup); 93 lua_settable(L, -(nup+3)); 94 } 95 lua_pop(L, nup); /* remove upvalues */ 96 } 97