/src/middleware/lua/loslib.c
C | 243 lines | 186 code | 43 blank | 14 comment | 27 complexity | e516ae0ffc3d215ea581246517e85c7f MD5 | raw file
1/* 2** $Id: loslib.c,v 1.19.1.3 2008/01/18 16:38:18 roberto Exp $ 3** Standard Operating System library 4** See Copyright Notice in lua.h 5*/ 6 7 8#include <errno.h> 9#include <locale.h> 10#include <stdlib.h> 11#include <string.h> 12#include <time.h> 13 14#define loslib_c 15#define LUA_LIB 16 17#include "lua.h" 18 19#include "lauxlib.h" 20#include "lualib.h" 21 22 23static int os_pushresult (lua_State *L, int i, const char *filename) { 24 int en = errno; /* calls to Lua API may change this value */ 25 if (i) { 26 lua_pushboolean(L, 1); 27 return 1; 28 } 29 else { 30 lua_pushnil(L); 31 lua_pushfstring(L, "%s: %s", filename, strerror(en)); 32 lua_pushinteger(L, en); 33 return 3; 34 } 35} 36 37 38static int os_execute (lua_State *L) { 39 lua_pushinteger(L, system(luaL_optstring(L, 1, NULL))); 40 return 1; 41} 42 43 44static int os_remove (lua_State *L) { 45 const char *filename = luaL_checkstring(L, 1); 46 return os_pushresult(L, remove(filename) == 0, filename); 47} 48 49 50static int os_rename (lua_State *L) { 51 const char *fromname = luaL_checkstring(L, 1); 52 const char *toname = luaL_checkstring(L, 2); 53 return os_pushresult(L, rename(fromname, toname) == 0, fromname); 54} 55 56 57static int os_tmpname (lua_State *L) { 58 char buff[LUA_TMPNAMBUFSIZE]; 59 int err; 60 lua_tmpnam(buff, err); 61 if (err) 62 return luaL_error(L, "unable to generate a unique filename"); 63 lua_pushstring(L, buff); 64 return 1; 65} 66 67 68static int os_getenv (lua_State *L) { 69 lua_pushstring(L, getenv(luaL_checkstring(L, 1))); /* if NULL push nil */ 70 return 1; 71} 72 73 74static int os_clock (lua_State *L) { 75 lua_pushnumber(L, ((lua_Number)clock())/(lua_Number)CLOCKS_PER_SEC); 76 return 1; 77} 78 79 80/* 81** {====================================================== 82** Time/Date operations 83** { year=%Y, month=%m, day=%d, hour=%H, min=%M, sec=%S, 84** wday=%w+1, yday=%j, isdst=? } 85** ======================================================= 86*/ 87 88static void setfield (lua_State *L, const char *key, int value) { 89 lua_pushinteger(L, value); 90 lua_setfield(L, -2, key); 91} 92 93static void setboolfield (lua_State *L, const char *key, int value) { 94 if (value < 0) /* undefined? */ 95 return; /* does not set field */ 96 lua_pushboolean(L, value); 97 lua_setfield(L, -2, key); 98} 99 100static int getboolfield (lua_State *L, const char *key) { 101 int res; 102 lua_getfield(L, -1, key); 103 res = lua_isnil(L, -1) ? -1 : lua_toboolean(L, -1); 104 lua_pop(L, 1); 105 return res; 106} 107 108 109static int getfield (lua_State *L, const char *key, int d) { 110 int res; 111 lua_getfield(L, -1, key); 112 if (lua_isnumber(L, -1)) 113 res = (int)lua_tointeger(L, -1); 114 else { 115 if (d < 0) 116 return luaL_error(L, "field " LUA_QS " missing in date table", key); 117 res = d; 118 } 119 lua_pop(L, 1); 120 return res; 121} 122 123 124static int os_date (lua_State *L) { 125 const char *s = luaL_optstring(L, 1, "%c"); 126 time_t t = luaL_opt(L, (time_t)luaL_checknumber, 2, time(NULL)); 127 struct tm *stm; 128 if (*s == '!') { /* UTC? */ 129 stm = gmtime(&t); 130 s++; /* skip `!' */ 131 } 132 else 133 stm = localtime(&t); 134 if (stm == NULL) /* invalid date? */ 135 lua_pushnil(L); 136 else if (strcmp(s, "*t") == 0) { 137 lua_createtable(L, 0, 9); /* 9 = number of fields */ 138 setfield(L, "sec", stm->tm_sec); 139 setfield(L, "min", stm->tm_min); 140 setfield(L, "hour", stm->tm_hour); 141 setfield(L, "day", stm->tm_mday); 142 setfield(L, "month", stm->tm_mon+1); 143 setfield(L, "year", stm->tm_year+1900); 144 setfield(L, "wday", stm->tm_wday+1); 145 setfield(L, "yday", stm->tm_yday+1); 146 setboolfield(L, "isdst", stm->tm_isdst); 147 } 148 else { 149 char cc[3]; 150 luaL_Buffer b; 151 cc[0] = '%'; cc[2] = '\0'; 152 luaL_buffinit(L, &b); 153 for (; *s; s++) { 154 if (*s != '%' || *(s + 1) == '\0') /* no conversion specifier? */ 155 luaL_addchar(&b, *s); 156 else { 157 size_t reslen; 158 char buff[200]; /* should be big enough for any conversion result */ 159 cc[1] = *(++s); 160 reslen = strftime(buff, sizeof(buff), cc, stm); 161 luaL_addlstring(&b, buff, reslen); 162 } 163 } 164 luaL_pushresult(&b); 165 } 166 return 1; 167} 168 169 170static int os_time (lua_State *L) { 171 time_t t; 172 if (lua_isnoneornil(L, 1)) /* called without args? */ 173 t = time(NULL); /* get current time */ 174 else { 175 struct tm ts; 176 luaL_checktype(L, 1, LUA_TTABLE); 177 lua_settop(L, 1); /* make sure table is at the top */ 178 ts.tm_sec = getfield(L, "sec", 0); 179 ts.tm_min = getfield(L, "min", 0); 180 ts.tm_hour = getfield(L, "hour", 12); 181 ts.tm_mday = getfield(L, "day", -1); 182 ts.tm_mon = getfield(L, "month", -1) - 1; 183 ts.tm_year = getfield(L, "year", -1) - 1900; 184 ts.tm_isdst = getboolfield(L, "isdst"); 185 t = mktime(&ts); 186 } 187 if (t == (time_t)(-1)) 188 lua_pushnil(L); 189 else 190 lua_pushnumber(L, (lua_Number)t); 191 return 1; 192} 193 194 195static int os_difftime (lua_State *L) { 196 lua_pushnumber(L, difftime((time_t)(luaL_checknumber(L, 1)), 197 (time_t)(luaL_optnumber(L, 2, 0)))); 198 return 1; 199} 200 201/* }====================================================== */ 202 203 204static int os_setlocale (lua_State *L) { 205 static const int cat[] = {LC_ALL, LC_COLLATE, LC_CTYPE, LC_MONETARY, 206 LC_NUMERIC, LC_TIME}; 207 static const char *const catnames[] = {"all", "collate", "ctype", "monetary", 208 "numeric", "time", NULL}; 209 const char *l = luaL_optstring(L, 1, NULL); 210 int op = luaL_checkoption(L, 2, "all", catnames); 211 lua_pushstring(L, setlocale(cat[op], l)); 212 return 1; 213} 214 215 216static int os_exit (lua_State *L) { 217 exit(luaL_optint(L, 1, EXIT_SUCCESS)); 218} 219 220static const luaL_Reg syslib[] = { 221 {"clock", os_clock}, 222 {"date", os_date}, 223 {"difftime", os_difftime}, 224 {"execute", os_execute}, 225 {"exit", os_exit}, 226 {"getenv", os_getenv}, 227 {"remove", os_remove}, 228 {"rename", os_rename}, 229 {"setlocale", os_setlocale}, 230 {"time", os_time}, 231 {"tmpname", os_tmpname}, 232 {NULL, NULL} 233}; 234 235/* }====================================================== */ 236 237 238 239LUALIB_API int luaopen_os (lua_State *L) { 240 luaL_register(L, LUA_OSLIBNAME, syslib); 241 return 1; 242} 243