diff options
Diffstat (limited to 'libs/lua/src/loslib.c')
-rw-r--r-- | libs/lua/src/loslib.c | 243 |
1 files changed, 243 insertions, 0 deletions
diff --git a/libs/lua/src/loslib.c b/libs/lua/src/loslib.c new file mode 100644 index 0000000..da06a57 --- /dev/null +++ b/libs/lua/src/loslib.c | |||
@@ -0,0 +1,243 @@ | |||
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 | |||
23 | static 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 | |||
38 | static int os_execute (lua_State *L) { | ||
39 | lua_pushinteger(L, system(luaL_optstring(L, 1, NULL))); | ||
40 | return 1; | ||
41 | } | ||
42 | |||
43 | |||
44 | static 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 | |||
50 | static 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 | |||
57 | static 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 | |||
68 | static 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 | |||
74 | static 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 | |||
88 | static void setfield (lua_State *L, const char *key, int value) { | ||
89 | lua_pushinteger(L, value); | ||
90 | lua_setfield(L, -2, key); | ||
91 | } | ||
92 | |||
93 | static 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 | |||
100 | static 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 | |||
109 | static 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 | |||
124 | static 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 | |||
170 | static 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 | |||
195 | static 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 | |||
204 | static 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 | |||
216 | static int os_exit (lua_State *L) { | ||
217 | exit(luaL_optint(L, 1, EXIT_SUCCESS)); | ||
218 | } | ||
219 | |||
220 | static 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 | |||
239 | LUALIB_API int luaopen_os (lua_State *L) { | ||
240 | luaL_register(L, LUA_OSLIBNAME, syslib); | ||
241 | return 1; | ||
242 | } | ||
243 | |||