/* ** $Id: lbaselib.c,v 1.44 2001/10/17 21:12:57 roberto Exp $ ** Basic library ** See Copyright Notice in lua.h */ #include #include #include #include #define LUA_PRIVATE #include "lua.h" #include "lauxlib.h" #include "luadebug.h" #include "lualib.h" static void aux_setn (lua_State *L, int t, int n) { lua_pushliteral(L, l_s("n")); lua_pushnumber(L, n); lua_settable(L, t); } /* ** If your system does not support `stderr', redefine this function, or ** redefine _ERRORMESSAGE so that it won't need _ALERT. */ static int luaB__ALERT (lua_State *L) { fputs(luaL_check_string(L, 1), stderr); return 0; } /* ** Basic implementation of _ERRORMESSAGE. ** The library `liolib' redefines _ERRORMESSAGE for better error information. */ static int luaB__ERRORMESSAGE (lua_State *L) { luaL_check_rawtype(L, 1, LUA_TSTRING); lua_getglobal(L, l_s(LUA_ALERT)); if (lua_isfunction(L, -1)) { /* avoid error loop if _ALERT is not defined */ lua_Debug ar; lua_pushliteral(L, l_s("error: ")); lua_pushvalue(L, 1); if (lua_getstack(L, 1, &ar)) { lua_getinfo(L, l_s("Sl"), &ar); if (ar.source && ar.currentline > 0) { l_char buff[100]; sprintf(buff, l_s("\n <%.70s: line %d>"), ar.short_src, ar.currentline); lua_pushstring(L, buff); lua_concat(L, 2); } } lua_pushliteral(L, l_s("\n")); lua_concat(L, 3); lua_rawcall(L, 1, 0); } return 0; } /* ** If your system does not support `stdout', you can just remove this function. ** If you need, you can define your own `print' function, following this ** model but changing `fputs' to put the strings at a proper place ** (a console window or a log file, for instance). */ static int luaB_print (lua_State *L) { int n = lua_gettop(L); /* number of arguments */ int i; lua_getglobal(L, l_s("tostring")); for (i=1; i<=n; i++) { const l_char *s; lua_pushvalue(L, -1); /* function to be called */ lua_pushvalue(L, i); /* value to print */ lua_rawcall(L, 1, 1); s = lua_tostring(L, -1); /* get result */ if (s == NULL) lua_error(L, l_s("`tostring' must return a string to `print'")); if (i>1) fputs(l_s("\t"), stdout); fputs(s, stdout); lua_pop(L, 1); /* pop result */ } fputs(l_s("\n"), stdout); return 0; } static int luaB_tonumber (lua_State *L) { int base = luaL_opt_int(L, 2, 10); if (base == 10) { /* standard conversion */ luaL_check_any(L, 1); if (lua_isnumber(L, 1)) { lua_pushnumber(L, lua_tonumber(L, 1)); return 1; } } else { const l_char *s1 = luaL_check_string(L, 1); l_char *s2; unsigned long n; luaL_arg_check(L, 2 <= base && base <= 36, 2, l_s("base out of range")); n = strtoul(s1, &s2, base); if (s1 != s2) { /* at least one valid digit? */ while (isspace(uchar(*s2))) s2++; /* skip trailing spaces */ if (*s2 == l_c('\0')) { /* no invalid trailing characters? */ lua_pushnumber(L, n); return 1; } } } lua_pushnil(L); /* else not a number */ return 1; } static int luaB_error (lua_State *L) { lua_error(L, luaL_opt_string(L, 1, NULL)); return 0; /* to avoid warnings */ } static int luaB_setglobal (lua_State *L) { luaL_check_any(L, 2); lua_setglobal(L, luaL_check_string(L, 1)); return 0; } static int luaB_getglobal (lua_State *L) { lua_getglobal(L, luaL_check_string(L, 1)); return 1; } /* auxiliary function to get `tags' */ static int gettag (lua_State *L, int narg) { switch (lua_rawtag(L, narg)) { case LUA_TNUMBER: return (int)(lua_tonumber(L, narg)); case LUA_TSTRING: { const l_char *name = lua_tostring(L, narg); int tag = lua_name2tag(L, name); if (tag == LUA_TNONE) luaL_verror(L, l_s("'%.30s' is not a valid type name"), name); return tag; } default: luaL_argerror(L, narg, l_s("tag or type name expected")); return 0; /* to avoid warnings */ } } static int luaB_tag (lua_State *L) { luaL_check_any(L, 1); lua_pushnumber(L, lua_tag(L, 1)); return 1; } static int luaB_settype (lua_State *L) { luaL_check_rawtype(L, 1, LUA_TTABLE); lua_pushvalue(L, 1); /* push table */ lua_settag(L, gettag(L, 2)); return 1; /* return table */ } static int luaB_weakmode (lua_State *L) { const l_char *mode = luaL_check_string(L, 2); luaL_check_rawtype(L, 1, LUA_TTABLE); if (*mode == l_c('?')) { l_char buff[3]; l_char *s = buff; int imode = lua_getweakmode(L, 1); if (imode & LUA_WEAK_KEY) *s++ = 'k'; if (imode & LUA_WEAK_VALUE) *s++ = 'v'; *s = '\0'; lua_pushstring(L, buff); return 1; } else { int imode = 0; if (strchr(mode, l_c('k'))) imode |= LUA_WEAK_KEY; if (strchr(mode, l_c('v'))) imode |= LUA_WEAK_VALUE; lua_pushvalue(L, 1); /* push table */ lua_setweakmode(L, imode); return 1; /* return the table */ } } static int luaB_newtype (lua_State *L) { const l_char *name = luaL_opt_string(L, 1, NULL); lua_pushnumber(L, lua_newtype(L, name, LUA_TTABLE)); return 1; } static int luaB_globals (lua_State *L) { lua_getglobals(L); /* value to be returned */ if (!lua_isnull(L, 1)) { luaL_check_rawtype(L, 1, LUA_TTABLE); lua_pushvalue(L, 1); /* new table of globals */ lua_setglobals(L); } return 1; } static int luaB_rawget (lua_State *L) { luaL_check_rawtype(L, 1, LUA_TTABLE); luaL_check_any(L, 2); lua_rawget(L, -2); return 1; } static int luaB_rawset (lua_State *L) { luaL_check_rawtype(L, 1, LUA_TTABLE); luaL_check_any(L, 2); luaL_check_any(L, 3); lua_rawset(L, -3); return 1; } static int luaB_settagmethod (lua_State *L) { int tag = gettag(L, 1); const l_char *event = luaL_check_string(L, 2); luaL_arg_check(L, lua_isfunction(L, 3) || lua_isnil(L, 3), 3, l_s("function or nil expected")); if (strcmp(event, l_s("gc")) == 0) lua_error(L, l_s("cannot set `gc' tag method from Lua")); lua_gettagmethod(L, tag, event); lua_pushvalue(L, 3); lua_settagmethod(L, tag, event); return 1; } static int luaB_gettagmethod (lua_State *L) { int tag = gettag(L, 1); const l_char *event = luaL_check_string(L, 2); if (strcmp(event, l_s("gc")) == 0) lua_error(L, l_s("cannot get `gc' tag method from Lua")); lua_gettagmethod(L, tag, event); return 1; } static int luaB_gcinfo (lua_State *L) { lua_pushnumber(L, lua_getgccount(L)); lua_pushnumber(L, lua_getgcthreshold(L)); return 2; } static int luaB_collectgarbage (lua_State *L) { lua_setgcthreshold(L, luaL_opt_int(L, 1, 0)); return 0; } static int luaB_type (lua_State *L) { luaL_check_any(L, 1); lua_pushstring(L, lua_type(L, 1)); return 1; } static int luaB_rawtype (lua_State *L) { luaL_check_any(L, 1); lua_pushstring(L, lua_tag2name(L, lua_rawtag(L, 1))); return 1; } static int luaB_next (lua_State *L) { luaL_check_rawtype(L, 1, LUA_TTABLE); lua_settop(L, 2); /* create a 2nd argument if there isn't one */ if (lua_next(L, 1)) return 2; else { lua_pushnil(L); return 1; } } static int passresults (lua_State *L, int status, int oldtop) { static const l_char *const errornames[] = {l_s("ok"), l_s("run-time error"), l_s("file error"), l_s("syntax error"), l_s("memory error"), l_s("error in error handling")}; if (status == 0) { int nresults = lua_gettop(L) - oldtop; if (nresults > 0) return nresults; /* results are already on the stack */ else { lua_newuserdatabox(L, NULL); /* at least one result to signal no errors */ return 1; } } else { /* error */ lua_pushnil(L); lua_pushstring(L, errornames[status]); /* error code */ return 2; } } static int luaB_dostring (lua_State *L) { int oldtop = lua_gettop(L); size_t l; const l_char *s = luaL_check_lstr(L, 1, &l); const l_char *chunkname = luaL_opt_string(L, 2, s); return passresults(L, lua_dobuffer(L, s, l, chunkname), oldtop); } static int luaB_loadstring (lua_State *L) { int oldtop = lua_gettop(L); size_t l; const l_char *s = luaL_check_lstr(L, 1, &l); const l_char *chunkname = luaL_opt_string(L, 2, s); return passresults(L, lua_loadbuffer(L, s, l, chunkname), oldtop); } static int luaB_dofile (lua_State *L) { int oldtop = lua_gettop(L); const l_char *fname = luaL_opt_string(L, 1, NULL); return passresults(L, lua_dofile(L, fname), oldtop); } static int luaB_loadfile (lua_State *L) { int oldtop = lua_gettop(L); const l_char *fname = luaL_opt_string(L, 1, NULL); return passresults(L, lua_loadfile(L, fname), oldtop); } #define LUA_PATH l_s("LUA_PATH") #define LUA_PATH_SEP l_s(";") #ifndef LUA_PATH_DEFAULT #define LUA_PATH_DEFAULT l_s("./") #endif static int luaB_require (lua_State *L) { const l_char *path; luaL_check_string(L, 1); lua_settop(L, 1); lua_getglobal(L, LUA_PATH); /* get path */ if (lua_isstring(L, 2)) /* is LUA_PATH defined? */ path = lua_tostring(L, 2); else { /* LUA_PATH not defined */ lua_pop(L, 1); /* pop old global value */ path = getenv(LUA_PATH); /* try environment variable */ if (path == NULL) path = LUA_PATH_DEFAULT; /* else use default */ lua_pushstring(L, path); lua_pushvalue(L, -1); /* duplicate to leave a copy on stack */ lua_setglobal(L, LUA_PATH); } lua_pushvalue(L, 1); /* check package's name in book-keeping table */ lua_gettable(L, lua_upvalueindex(1)); if (!lua_isnil(L, -1)) /* is it there? */ return 0; /* package is already loaded */ else { /* must load it */ for (;;) { /* traverse path */ int res; int l = strcspn(path, LUA_PATH_SEP); /* find separator */ lua_pushlstring(L, path, l); /* directory name */ lua_pushvalue(L, 1); /* package name */ lua_concat(L, 2); /* concat directory with package name */ res = lua_dofile(L, lua_tostring(L, -1)); /* try to load it */ lua_settop(L, 2); /* pop string and eventual results from dofile */ if (res == 0) break; /* ok; file done */ else if (res != LUA_ERRFILE) lua_error(L, NULL); /* error running package; propagate it */ if (*(path+l) == l_c('\0')) /* no more directories? */ luaL_verror(L, l_s("could not load package `%.20s' from path `%.200s'"), lua_tostring(L, 1), lua_tostring(L, 2)); path += l+1; /* try next directory */ } } lua_pushvalue(L, 1); lua_pushnumber(L, 1); lua_settable(L, lua_upvalueindex(1)); /* mark it as loaded */ return 0; } static int aux_unpack (lua_State *L, int arg) { int n, i; luaL_check_rawtype(L, arg, LUA_TTABLE); n = lua_getn(L, arg); luaL_check_stack(L, n, l_s("table too big to unpack")); for (i=1; i<=n; i++) /* push arg[1...n] */ lua_rawgeti(L, arg, i); return n; } static int luaB_unpack (lua_State *L) { return aux_unpack(L, 1); } static int luaB_call (lua_State *L) { int oldtop; const l_char *options = luaL_opt_string(L, 3, l_s("")); int err = 0; /* index of old error method */ int status; int n; if (!lua_isnull(L, 4)) { /* set new error method */ lua_getglobal(L, l_s(LUA_ERRORMESSAGE)); err = lua_gettop(L); /* get index */ lua_pushvalue(L, 4); lua_setglobal(L, l_s(LUA_ERRORMESSAGE)); } oldtop = lua_gettop(L); /* top before function-call preparation */ /* push function */ lua_pushvalue(L, 1); n = aux_unpack(L, 2); /* push arg[1...n] */ status = lua_call(L, n, LUA_MULTRET); if (err != 0) { /* restore old error method */ lua_pushvalue(L, err); lua_setglobal(L, l_s(LUA_ERRORMESSAGE)); } if (status != 0) { /* error in call? */ if (strchr(options, l_c('x'))) lua_pushnil(L); /* return nil to signal the error */ else lua_error(L, NULL); /* propagate error without additional messages */ return 1; } if (strchr(options, l_c('p'))) /* pack results? */ lua_error(L, l_s("obsolete option `p' in `call'")); return lua_gettop(L) - oldtop; /* results are already on the stack */ } static int luaB_tostring (lua_State *L) { l_char buff[64]; switch (lua_rawtag(L, 1)) { case LUA_TNUMBER: lua_pushstring(L, lua_tostring(L, 1)); return 1; case LUA_TSTRING: lua_pushvalue(L, 1); return 1; case LUA_TTABLE: sprintf(buff, l_s("%.40s: %p"), lua_type(L, 1), lua_topointer(L, 1)); break; case LUA_TFUNCTION: sprintf(buff, l_s("function: %p"), lua_topointer(L, 1)); break; case LUA_TUSERDATA: { const l_char *t = lua_type(L, 1); if (strcmp(t, l_s("userdata")) == 0) sprintf(buff, l_s("userdata(%d): %p"), lua_tag(L, 1), lua_touserdata(L, 1)); else sprintf(buff, l_s("%.40s: %p"), t, lua_touserdata(L, 1)); break; } case LUA_TNIL: lua_pushliteral(L, l_s("nil")); return 1; default: luaL_argerror(L, 1, l_s("value expected")); } lua_pushstring(L, buff); return 1; } static int luaB_foreachi (lua_State *L) { int n, i; luaL_check_rawtype(L, 1, LUA_TTABLE); luaL_check_rawtype(L, 2, LUA_TFUNCTION); n = lua_getn(L, 1); for (i=1; i<=n; i++) { lua_pushvalue(L, 2); /* function */ lua_pushnumber(L, i); /* 1st argument */ lua_rawgeti(L, 1, i); /* 2nd argument */ lua_rawcall(L, 2, 1); if (!lua_isnil(L, -1)) return 1; lua_pop(L, 1); /* remove nil result */ } return 0; } static int luaB_foreach (lua_State *L) { luaL_check_rawtype(L, 1, LUA_TTABLE); luaL_check_rawtype(L, 2, LUA_TFUNCTION); lua_pushnil(L); /* first index */ for (;;) { if (lua_next(L, 1) == 0) return 0; lua_pushvalue(L, 2); /* function */ lua_pushvalue(L, -3); /* key */ lua_pushvalue(L, -3); /* value */ lua_rawcall(L, 2, 1); if (!lua_isnil(L, -1)) return 1; lua_pop(L, 2); /* remove value and result */ } } static int luaB_assert (lua_State *L) { luaL_check_any(L, 1); if (lua_isnil(L, 1)) luaL_verror(L, l_s("assertion failed! %.90s"), luaL_opt_string(L, 2, l_s(""))); lua_settop(L, 1); return 1; } static int luaB_getn (lua_State *L) { luaL_check_rawtype(L, 1, LUA_TTABLE); lua_pushnumber(L, lua_getn(L, 1)); return 1; } static int luaB_tinsert (lua_State *L) { int v = lua_gettop(L); /* number of arguments */ int n, pos; luaL_check_rawtype(L, 1, LUA_TTABLE); n = lua_getn(L, 1); if (v == 2) /* called with only 2 arguments */ pos = n+1; else { v = 3; /* function may be called with more than 3 args */ pos = luaL_check_int(L, 2); /* 2nd argument is the position */ } aux_setn(L, 1, n+1); /* t.n = n+1 */ for (; n>=pos; n--) { lua_rawgeti(L, 1, n); lua_rawseti(L, 1, n+1); /* t[n+1] = t[n] */ } lua_pushvalue(L, v); lua_rawseti(L, 1, pos); /* t[pos] = v */ return 0; } static int luaB_tremove (lua_State *L) { int pos, n; luaL_check_rawtype(L, 1, LUA_TTABLE); n = lua_getn(L, 1); pos = luaL_opt_int(L, 2, n); if (n <= 0) return 0; /* table is `empty' */ aux_setn(L, 1, n-1); /* t.n = n-1 */ lua_rawgeti(L, 1, pos); /* result = t[pos] */ for ( ;pos= P */ while (lua_rawgeti(L, 1, ++i), sort_comp(L, -1, -2)) { if (i>u) lua_error(L, l_s("invalid order function for sorting")); lua_pop(L, 1); /* remove a[i] */ } /* repeat --j until a[j] <= P */ while (lua_rawgeti(L, 1, --j), sort_comp(L, -3, -1)) { if (j