2 ** $Id: loslib.c,v 1.19 2006/04/26 18:19:49 roberto Exp $
3 ** Standard Operating System library
4 ** See Copyright Notice in lua.h
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 */
26 lua_pushboolean(L
, 1);
31 lua_pushfstring(L
, "%s: %s", filename
, strerror(en
));
32 lua_pushinteger(L
, en
);
38 static int os_execute (lua_State
*L
) {
39 lua_pushinteger(L
, system(luaL_optstring(L
, 1, NULL
)));
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
);
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
);
57 static int os_tmpname (lua_State
*L
) {
58 char buff
[LUA_TMPNAMBUFSIZE
];
60 lua_tmpnam(buff
, err
);
62 return luaL_error(L
, "unable to generate a unique filename");
63 lua_pushstring(L
, buff
);
68 static int os_getenv (lua_State
*L
) {
69 lua_pushstring(L
, getenv(luaL_checkstring(L
, 1))); /* if NULL push nil */
74 static int os_clock (lua_State
*L
) {
75 lua_pushnumber(L
, ((lua_Number
)clock())/(lua_Number
)CLOCKS_PER_SEC
);
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 ** =======================================================
88 static void setfield (lua_State
*L
, const char *key
, int value
) {
89 lua_pushinteger(L
, value
);
90 lua_setfield(L
, -2, key
);
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
);
100 static int getboolfield (lua_State
*L
, const char *key
) {
102 lua_getfield(L
, -1, key
);
103 res
= lua_isnil(L
, -1) ? -1 : lua_toboolean(L
, -1);
109 static int getfield (lua_State
*L
, const char *key
, int d
) {
111 lua_getfield(L
, -1, key
);
112 if (lua_isnumber(L
, -1))
113 res
= (int)lua_tointeger(L
, -1);
116 return luaL_error(L
, "field " LUA_QS
" missing in date table", key
);
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
));
128 if (*s
== '!') { /* UTC? */
134 if (stm
== NULL
) /* invalid date? */
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
);
150 if (strftime(b
, sizeof(b
), s
, stm
))
151 lua_pushstring(L
, b
);
153 return luaL_error(L
, LUA_QL("date") " format too long");
159 static int os_time (lua_State
*L
) {
161 if (lua_isnoneornil(L
, 1)) /* called without args? */
162 t
= time(NULL
); /* get current time */
165 luaL_checktype(L
, 1, LUA_TTABLE
);
166 lua_settop(L
, 1); /* make sure table is at the top */
167 ts
.tm_sec
= getfield(L
, "sec", 0);
168 ts
.tm_min
= getfield(L
, "min", 0);
169 ts
.tm_hour
= getfield(L
, "hour", 12);
170 ts
.tm_mday
= getfield(L
, "day", -1);
171 ts
.tm_mon
= getfield(L
, "month", -1) - 1;
172 ts
.tm_year
= getfield(L
, "year", -1) - 1900;
173 ts
.tm_isdst
= getboolfield(L
, "isdst");
176 if (t
== (time_t)(-1))
179 lua_pushnumber(L
, (lua_Number
)t
);
184 static int os_difftime (lua_State
*L
) {
185 lua_pushnumber(L
, difftime((time_t)(luaL_checknumber(L
, 1)),
186 (time_t)(luaL_optnumber(L
, 2, 0))));
190 /* }====================================================== */
193 static int os_setlocale (lua_State
*L
) {
194 static const int cat
[] = {LC_ALL
, LC_COLLATE
, LC_CTYPE
, LC_MONETARY
,
195 LC_NUMERIC
, LC_TIME
};
196 static const char *const catnames
[] = {"all", "collate", "ctype", "monetary",
197 "numeric", "time", NULL
};
198 const char *l
= luaL_optstring(L
, 1, NULL
);
199 int op
= luaL_checkoption(L
, 2, "all", catnames
);
200 lua_pushstring(L
, setlocale(cat
[op
], l
));
205 static int os_exit (lua_State
*L
) {
206 exit(luaL_optint(L
, 1, EXIT_SUCCESS
));
207 return 0; /* to avoid warnings */
210 static const luaL_Reg syslib
[] = {
213 {"difftime", os_difftime
},
214 {"execute", os_execute
},
216 {"getenv", os_getenv
},
217 {"remove", os_remove
},
218 {"rename", os_rename
},
219 {"setlocale", os_setlocale
},
221 {"tmpname", os_tmpname
},
225 /* }====================================================== */
229 LUALIB_API
int luaopen_os (lua_State
*L
) {
230 luaL_register(L
, LUA_OSLIBNAME
, syslib
);