loslib.c 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407
  1. /*
  2. ** $Id: loslib.c,v 1.65 2016/07/18 17:58:58 roberto Exp $
  3. ** Standard Operating System library
  4. ** See Copyright Notice in lua.h
  5. */
  6. #define loslib_c
  7. #define LUA_LIB
  8. #include "lprefix.h"
  9. #include <errno.h>
  10. #include <locale.h>
  11. #include <stdlib.h>
  12. #include <string.h>
  13. #include <time.h>
  14. #include "lua.h"
  15. #include "lauxlib.h"
  16. #include "lualib.h"
  17. /*
  18. ** {==================================================================
  19. ** List of valid conversion specifiers for the 'strftime' function;
  20. ** options are grouped by length; group of length 2 start with '||'.
  21. ** ===================================================================
  22. */
  23. #if !defined(LUA_STRFTIMEOPTIONS) /* { */
  24. /* options for ANSI C 89 (only 1-char options) */
  25. #define L_STRFTIMEC89 "aAbBcdHIjmMpSUwWxXyYZ%"
  26. /* options for ISO C 99 and POSIX */
  27. #define L_STRFTIMEC99 "aAbBcCdDeFgGhHIjmMnprRStTuUVwWxXyYzZ%" \
  28. "||" "EcECExEXEyEY" "OdOeOHOIOmOMOSOuOUOVOwOWOy" /* two-char options */
  29. /* options for Windows */
  30. #define L_STRFTIMEWIN "aAbBcdHIjmMpSUwWxXyYzZ%" \
  31. "||" "#c#x#d#H#I#j#m#M#S#U#w#W#y#Y" /* two-char options */
  32. #if defined(LUA_USE_WINDOWS)
  33. #define LUA_STRFTIMEOPTIONS L_STRFTIMEWIN
  34. #elif defined(LUA_USE_C89)
  35. #define LUA_STRFTIMEOPTIONS L_STRFTIMEC89
  36. #else /* C99 specification */
  37. #define LUA_STRFTIMEOPTIONS L_STRFTIMEC99
  38. #endif
  39. #endif /* } */
  40. /* }================================================================== */
  41. /*
  42. ** {==================================================================
  43. ** Configuration for time-related stuff
  44. ** ===================================================================
  45. */
  46. #if !defined(l_time_t) /* { */
  47. /*
  48. ** type to represent time_t in Lua
  49. */
  50. #define l_timet lua_Integer
  51. #define l_pushtime(L,t) lua_pushinteger(L,(lua_Integer)(t))
  52. static time_t l_checktime (lua_State *L, int arg) {
  53. lua_Integer t = luaL_checkinteger(L, arg);
  54. luaL_argcheck(L, (time_t)t == t, arg, "time out-of-bounds");
  55. return (time_t)t;
  56. }
  57. #endif /* } */
  58. #if !defined(l_gmtime) /* { */
  59. /*
  60. ** By default, Lua uses gmtime/localtime, except when POSIX is available,
  61. ** where it uses gmtime_r/localtime_r
  62. */
  63. #if defined(LUA_USE_POSIX) /* { */
  64. #define l_gmtime(t,r) gmtime_r(t,r)
  65. #define l_localtime(t,r) localtime_r(t,r)
  66. #else /* }{ */
  67. /* ISO C definitions */
  68. #define l_gmtime(t,r) ((void)(r)->tm_sec, gmtime(t))
  69. #define l_localtime(t,r) ((void)(r)->tm_sec, localtime(t))
  70. #endif /* } */
  71. #endif /* } */
  72. /* }================================================================== */
  73. /*
  74. ** {==================================================================
  75. ** Configuration for 'tmpnam':
  76. ** By default, Lua uses tmpnam except when POSIX is available, where
  77. ** it uses mkstemp.
  78. ** ===================================================================
  79. */
  80. #if !defined(lua_tmpnam) /* { */
  81. #if defined(LUA_USE_POSIX) /* { */
  82. #include <unistd.h>
  83. #define LUA_TMPNAMBUFSIZE 32
  84. #if !defined(LUA_TMPNAMTEMPLATE)
  85. #define LUA_TMPNAMTEMPLATE "/tmp/lua_XXXXXX"
  86. #endif
  87. #define lua_tmpnam(b,e) { \
  88. strcpy(b, LUA_TMPNAMTEMPLATE); \
  89. e = mkstemp(b); \
  90. if (e != -1) close(e); \
  91. e = (e == -1); }
  92. #else /* }{ */
  93. /* ISO C definitions */
  94. #define LUA_TMPNAMBUFSIZE L_tmpnam
  95. #define lua_tmpnam(b,e) { e = (tmpnam(b) == NULL); }
  96. #endif /* } */
  97. #endif /* } */
  98. /* }================================================================== */
  99. static int os_execute (lua_State *L) {
  100. const char *cmd = luaL_optstring(L, 1, NULL);
  101. int stat = system(cmd);
  102. if (cmd != NULL)
  103. return luaL_execresult(L, stat);
  104. else {
  105. lua_pushboolean(L, stat); /* true if there is a shell */
  106. return 1;
  107. }
  108. }
  109. static int os_remove (lua_State *L) {
  110. const char *filename = luaL_checkstring(L, 1);
  111. return luaL_fileresult(L, remove(filename) == 0, filename);
  112. }
  113. static int os_rename (lua_State *L) {
  114. const char *fromname = luaL_checkstring(L, 1);
  115. const char *toname = luaL_checkstring(L, 2);
  116. return luaL_fileresult(L, rename(fromname, toname) == 0, NULL);
  117. }
  118. static int os_tmpname (lua_State *L) {
  119. char buff[LUA_TMPNAMBUFSIZE];
  120. int err;
  121. lua_tmpnam(buff, err);
  122. if (err)
  123. return luaL_error(L, "unable to generate a unique filename");
  124. lua_pushstring(L, buff);
  125. return 1;
  126. }
  127. static int os_getenv (lua_State *L) {
  128. lua_pushstring(L, getenv(luaL_checkstring(L, 1))); /* if NULL push nil */
  129. return 1;
  130. }
  131. static int os_clock (lua_State *L) {
  132. lua_pushnumber(L, ((lua_Number)clock())/(lua_Number)CLOCKS_PER_SEC);
  133. return 1;
  134. }
  135. /*
  136. ** {======================================================
  137. ** Time/Date operations
  138. ** { year=%Y, month=%m, day=%d, hour=%H, min=%M, sec=%S,
  139. ** wday=%w+1, yday=%j, isdst=? }
  140. ** =======================================================
  141. */
  142. static void setfield (lua_State *L, const char *key, int value) {
  143. lua_pushinteger(L, value);
  144. lua_setfield(L, -2, key);
  145. }
  146. static void setboolfield (lua_State *L, const char *key, int value) {
  147. if (value < 0) /* undefined? */
  148. return; /* does not set field */
  149. lua_pushboolean(L, value);
  150. lua_setfield(L, -2, key);
  151. }
  152. /*
  153. ** Set all fields from structure 'tm' in the table on top of the stack
  154. */
  155. static void setallfields (lua_State *L, struct tm *stm) {
  156. setfield(L, "sec", stm->tm_sec);
  157. setfield(L, "min", stm->tm_min);
  158. setfield(L, "hour", stm->tm_hour);
  159. setfield(L, "day", stm->tm_mday);
  160. setfield(L, "month", stm->tm_mon + 1);
  161. setfield(L, "year", stm->tm_year + 1900);
  162. setfield(L, "wday", stm->tm_wday + 1);
  163. setfield(L, "yday", stm->tm_yday + 1);
  164. setboolfield(L, "isdst", stm->tm_isdst);
  165. }
  166. static int getboolfield (lua_State *L, const char *key) {
  167. int res;
  168. res = (lua_getfield(L, -1, key) == LUA_TNIL) ? -1 : lua_toboolean(L, -1);
  169. lua_pop(L, 1);
  170. return res;
  171. }
  172. /* maximum value for date fields (to avoid arithmetic overflows with 'int') */
  173. #if !defined(L_MAXDATEFIELD)
  174. #define L_MAXDATEFIELD (INT_MAX / 2)
  175. #endif
  176. static int getfield (lua_State *L, const char *key, int d, int delta) {
  177. int isnum;
  178. int t = lua_getfield(L, -1, key); /* get field and its type */
  179. lua_Integer res = lua_tointegerx(L, -1, &isnum);
  180. if (!isnum) { /* field is not an integer? */
  181. if (t != LUA_TNIL) /* some other value? */
  182. return luaL_error(L, "field '%s' is not an integer", key);
  183. else if (d < 0) /* absent field; no default? */
  184. return luaL_error(L, "field '%s' missing in date table", key);
  185. res = d;
  186. }
  187. else {
  188. if (!(-L_MAXDATEFIELD <= res && res <= L_MAXDATEFIELD))
  189. return luaL_error(L, "field '%s' is out-of-bound", key);
  190. res -= delta;
  191. }
  192. lua_pop(L, 1);
  193. return (int)res;
  194. }
  195. static const char *checkoption (lua_State *L, const char *conv,
  196. ptrdiff_t convlen, char *buff) {
  197. const char *option = LUA_STRFTIMEOPTIONS;
  198. int oplen = 1; /* length of options being checked */
  199. for (; *option != '\0' && oplen <= convlen; option += oplen) {
  200. if (*option == '|') /* next block? */
  201. oplen++; /* will check options with next length (+1) */
  202. else if (memcmp(conv, option, oplen) == 0) { /* match? */
  203. memcpy(buff, conv, oplen); /* copy valid option to buffer */
  204. buff[oplen] = '\0';
  205. return conv + oplen; /* return next item */
  206. }
  207. }
  208. luaL_argerror(L, 1,
  209. lua_pushfstring(L, "invalid conversion specifier '%%%s'", conv));
  210. return conv; /* to avoid warnings */
  211. }
  212. /* maximum size for an individual 'strftime' item */
  213. #define SIZETIMEFMT 250
  214. static int os_date (lua_State *L) {
  215. size_t slen;
  216. const char *s = luaL_optlstring(L, 1, "%c", &slen);
  217. time_t t = luaL_opt(L, l_checktime, 2, time(NULL));
  218. const char *se = s + slen; /* 's' end */
  219. struct tm tmr, *stm;
  220. if (*s == '!') { /* UTC? */
  221. stm = l_gmtime(&t, &tmr);
  222. s++; /* skip '!' */
  223. }
  224. else
  225. stm = l_localtime(&t, &tmr);
  226. if (stm == NULL) /* invalid date? */
  227. luaL_error(L, "time result cannot be represented in this installation");
  228. if (strcmp(s, "*t") == 0) {
  229. lua_createtable(L, 0, 9); /* 9 = number of fields */
  230. setallfields(L, stm);
  231. }
  232. else {
  233. char cc[4]; /* buffer for individual conversion specifiers */
  234. luaL_Buffer b;
  235. cc[0] = '%';
  236. luaL_buffinit(L, &b);
  237. while (s < se) {
  238. if (*s != '%') /* not a conversion specifier? */
  239. luaL_addchar(&b, *s++);
  240. else {
  241. size_t reslen;
  242. char *buff = luaL_prepbuffsize(&b, SIZETIMEFMT);
  243. s++; /* skip '%' */
  244. s = checkoption(L, s, se - s, cc + 1); /* copy specifier to 'cc' */
  245. reslen = strftime(buff, SIZETIMEFMT, cc, stm);
  246. luaL_addsize(&b, reslen);
  247. }
  248. }
  249. luaL_pushresult(&b);
  250. }
  251. return 1;
  252. }
  253. static int os_time (lua_State *L) {
  254. time_t t;
  255. if (lua_isnoneornil(L, 1)) /* called without args? */
  256. t = time(NULL); /* get current time */
  257. else {
  258. struct tm ts;
  259. luaL_checktype(L, 1, LUA_TTABLE);
  260. lua_settop(L, 1); /* make sure table is at the top */
  261. ts.tm_sec = getfield(L, "sec", 0, 0);
  262. ts.tm_min = getfield(L, "min", 0, 0);
  263. ts.tm_hour = getfield(L, "hour", 12, 0);
  264. ts.tm_mday = getfield(L, "day", -1, 0);
  265. ts.tm_mon = getfield(L, "month", -1, 1);
  266. ts.tm_year = getfield(L, "year", -1, 1900);
  267. ts.tm_isdst = getboolfield(L, "isdst");
  268. t = mktime(&ts);
  269. setallfields(L, &ts); /* update fields with normalized values */
  270. }
  271. if (t != (time_t)(l_timet)t || t == (time_t)(-1))
  272. luaL_error(L, "time result cannot be represented in this installation");
  273. l_pushtime(L, t);
  274. return 1;
  275. }
  276. static int os_difftime (lua_State *L) {
  277. time_t t1 = l_checktime(L, 1);
  278. time_t t2 = l_checktime(L, 2);
  279. lua_pushnumber(L, (lua_Number)difftime(t1, t2));
  280. return 1;
  281. }
  282. /* }====================================================== */
  283. static int os_setlocale (lua_State *L) {
  284. static const int cat[] = {LC_ALL, LC_COLLATE, LC_CTYPE, LC_MONETARY,
  285. LC_NUMERIC, LC_TIME};
  286. static const char *const catnames[] = {"all", "collate", "ctype", "monetary",
  287. "numeric", "time", NULL};
  288. const char *l = luaL_optstring(L, 1, NULL);
  289. int op = luaL_checkoption(L, 2, "all", catnames);
  290. lua_pushstring(L, setlocale(cat[op], l));
  291. return 1;
  292. }
  293. static int os_exit (lua_State *L) {
  294. int status;
  295. if (lua_isboolean(L, 1))
  296. status = (lua_toboolean(L, 1) ? EXIT_SUCCESS : EXIT_FAILURE);
  297. else
  298. status = (int)luaL_optinteger(L, 1, EXIT_SUCCESS);
  299. if (lua_toboolean(L, 2))
  300. lua_close(L);
  301. if (L) exit(status); /* 'if' to avoid warnings for unreachable 'return' */
  302. return 0;
  303. }
  304. static const luaL_Reg syslib[] = {
  305. {"clock", os_clock},
  306. {"date", os_date},
  307. {"difftime", os_difftime},
  308. {"execute", os_execute},
  309. {"exit", os_exit},
  310. {"getenv", os_getenv},
  311. {"remove", os_remove},
  312. {"rename", os_rename},
  313. {"setlocale", os_setlocale},
  314. {"time", os_time},
  315. {"tmpname", os_tmpname},
  316. {NULL, NULL}
  317. };
  318. /* }====================================================== */
  319. LUAMOD_API int luaopen_os (lua_State *L) {
  320. luaL_newlib(L, syslib);
  321. return 1;
  322. }