#include <errno.h>
#include <locale.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <dirent.h>
#include "lua.h"
#include "lauxlib.h"
#include "lualib.h"

Include-Abhängigkeitsdiagramm für loslib.c:

gehe zum Quellcode dieser Datei

Datenstrukturen
struct	idir_udata_t

Makrodefinitionen
#define	loslib_c

#define	LUA_LIB

#define	IDIR_META "chdk_idir_meta"

Funktionen
static int	os_pushresult (lua_State L, int i, const char filename)

static int	os_remove (lua_State *L)

static int	os_rename (lua_State *L)

static void	setfield (lua_State L, const char key, int value)

static void	setboolfield (lua_State L, const char key, int value)

static int	getboolfield (lua_State L, const char key)

static int	getfield (lua_State L, const char key, int d)

static int	os_date (lua_State *L)

static int	os_time (lua_State *L)

static int	os_difftime (lua_State *L)

static int	os_mkdir (lua_State *L)

static int	get_table_optbool (lua_State L, int narg, const char fname, int d)

static int	os_listdir (lua_State *L)

static int	idir_iter (lua_State *L)

static int	idir_gc (lua_State *L)

static int	os_idir (lua_State *L)

static void	idir_register (lua_State *L)

static int	os_stat (lua_State *L)

static int	os_utime (lua_State *L)

LUALIB_API int	luaopen_os (lua_State *L)

Variablen
static const luaL_Reg	idir_meta_methods []

static const luaL_Reg	syslib []

Makro-Dokumentation

#define IDIR_META "chdk_idir_meta"

Definiert in Zeile 312 der Datei loslib.c.

#define loslib_c

Definiert in Zeile 18 der Datei loslib.c.

#define LUA_LIB

Definiert in Zeile 19 der Datei loslib.c.

Dokumentation der Funktionen

static int get_table_optbool	(	lua_State *	L,
		int	narg,
		const char *	fname,
		int	d
	)

static

Definiert in Zeile 257 der Datei loslib.c.

 {
     int r;
     lua_getfield(L, narg, fname);
     // not set - use default
     if(lua_isnil(L,-1)) {
         r=d;
     } else {// otherwise, treat as bool
         r=lua_toboolean(L,-1); 
     }
     lua_pop(L,1);
     return r;
 }

static int getboolfield	(	lua_State *	L,
		const char *	key
	)

static

Definiert in Zeile 111 der Datei loslib.c.

                                                         {
   int res;
   lua_getfield(L, -1, key);
   res = lua_isnil(L, -1) ? -1 : lua_toboolean(L, -1);
   lua_pop(L, 1);
   return res;
 }

static int getfield	(	lua_State *	L,
		const char *	key,
		int	d
	)

static

Definiert in Zeile 120 der Datei loslib.c.

                                                            {
   int res;
   lua_getfield(L, -1, key);
   if (lua_isnumber(L, -1))
     res = (int)lua_tointeger(L, -1);
   else {
     if (d < 0)
       return luaL_error(L, "field " LUA_QS " missing in date table", key);
     res = d;
   }
   lua_pop(L, 1);
   return res;
 }

static int idir_gc ( lua_State * L)

static

Definiert in Zeile 350 der Datei loslib.c.

                                  {
     idir_udata_t *ud = (idir_udata_t *)luaL_checkudata(L,1,IDIR_META);
     if(ud->dir) {
         closedir(ud->dir);
     }
     return 0;
 }

static int idir_iter ( lua_State * L)

static

Definiert in Zeile 319 der Datei loslib.c.

                                    {
     struct dirent *de;
     idir_udata_t *ud = (idir_udata_t *)luaL_checkudata(L,1,IDIR_META);
     // dir may be on first call if opendir failed, or previous explicit close
     if(!ud->dir) {
         return 0;
     }
     // allow explicit close by calling iterator(ud,false)
     // need to check type because first invocation is called with nil
     if(lua_type(L, 2) == LUA_TBOOLEAN && lua_toboolean(L,2) == 0) {
         closedir(ud->dir);
         ud->dir=NULL;
         return 0;
     }
     while((de = readdir(ud->dir))) {
         // if not all, skip over ignored items
         if(!ud->all && (de->d_name[0] == '\xE5' || (strcmp(de->d_name,".")==0) || (strcmp(de->d_name,"..")==0))) {
             continue;
         }
         break;
     }
     if(de) {
         lua_pushstring(L, de->d_name);
         return 1;
     } else { // on last, close immediately to avoid keeping the handle open until GC
         closedir(ud->dir);
         ud->dir=NULL;
         return 0;
     }
 }

static void idir_register ( lua_State * L)

static

Definiert in Zeile 419 der Datei loslib.c.

                                         {
     luaL_newmetatable(L,IDIR_META);
     luaL_register(L, NULL, idir_meta_methods);  
 }

LUALIB_API int luaopen_os ( lua_State * L)

Definiert in Zeile 545 der Datei loslib.c.

                                          {
   idir_register(L);
   luaL_register(L, LUA_OSLIBNAME, syslib);
   return 1;
 }

static int os_date ( lua_State * L)

static

Definiert in Zeile 135 der Datei loslib.c.

                                   {
   const char *s = luaL_optstring(L, 1, "%c");
   time_t t = luaL_opt(L, (time_t)luaL_checknumber, 2, time(NULL));
   struct tm *stm;
   if (*s == '!') {  /* UTC? */
   #if 0
   // reyalp - we have no idea about timezones, so just eat the !
   // and use local time
   // TODO some cams may be timezone/dst aware ?
     stm = gmtime(&t);
   #endif
     stm = localtime(&t);
     s++;  /* skip `!' */
   }
   else
     stm = localtime(&t);
   if (stm == NULL)  /* invalid date? */
     lua_pushnil(L);
   else if (strcmp(s, "*t") == 0) {
     lua_createtable(L, 0, 9);  /* 9 = number of fields */
     setfield(L, "sec", stm->tm_sec);
     setfield(L, "min", stm->tm_min);
     setfield(L, "hour", stm->tm_hour);
     setfield(L, "day", stm->tm_mday);
     setfield(L, "month", stm->tm_mon+1);
     setfield(L, "year", stm->tm_year+1900);
     setfield(L, "wday", stm->tm_wday+1);
     setfield(L, "yday", stm->tm_yday+1);
     setboolfield(L, "isdst", stm->tm_isdst);
   }
   else {
     char cc[3];
     luaL_Buffer b;
     cc[0] = '%'; cc[2] = '\0';
     luaL_buffinit(L, &b);
     for (; *s; s++) {
       if (*s != '%' || *(s + 1) == '\0')  /* no conversion specifier? */
         luaL_addchar(&b, *s);
       else {
         size_t reslen;
         char buff[200];  /* should be big enough for any conversion result */
         cc[1] = *(++s);
         reslen = strftime(buff, sizeof(buff), cc, stm);
         luaL_addlstring(&b, buff, reslen);
       }
     }
     luaL_pushresult(&b);
   }
   return 1;
 }

static int os_difftime ( lua_State * L)

static

Definiert in Zeile 219 der Datei loslib.c.

                                       {
   lua_pushnumber(L, (time_t)(luaL_checknumber(L, 1) - (time_t)(luaL_optnumber(L, 2, 0))));
   return 1;
 }

static int os_idir ( lua_State * L)

static

Definiert in Zeile 387 der Datei loslib.c.

                                   {
 #ifdef HOST_LUA
   // TODO: Not implemented for 'hostlua'
   (void)L;
   return 0;
 #else
   const char *dirname = luaL_checkstring(L, 1);
   int all=0,od_flags=OPENDIR_FL_CHDK_LFN;
   if(lua_istable(L,2)) {
     all=get_table_optbool(L,2,"showall",0);
     od_flags=(get_table_optbool(L,2,"chdklfn",1))?OPENDIR_FL_CHDK_LFN:OPENDIR_FL_NONE;
   } else {
     all=lua_toboolean(L, 2);
   }
 
   lua_pushcfunction(L, idir_iter);
 
   idir_udata_t *ud = lua_newuserdata(L,sizeof(idir_udata_t));
   ud->dir = opendir_chdk(dirname,od_flags); // may be null, in which case iterator will stop on first iteration
                                 // no obvious way to return error status
   ud->all = all;
 
   luaL_getmetatable(L, IDIR_META);
   lua_setmetatable(L, -2);
   return 2;
 #endif
 }

static int os_listdir ( lua_State * L)

static

Definiert in Zeile 278 der Datei loslib.c.

                                      {
 #ifdef HOST_LUA
   // TODO: Not implemented for 'hostlua'
   (void)L;
   return 0;
 #else
   DIR *dir;
   struct dirent *de;
   const char *dirname = luaL_checkstring(L, 1);
   int all=0,od_flags=OPENDIR_FL_CHDK_LFN;
   if(lua_istable(L,2)) {
     all=get_table_optbool(L,2,"showall",0);
     od_flags=(get_table_optbool(L,2,"chdklfn",1))?OPENDIR_FL_CHDK_LFN:OPENDIR_FL_NONE;
   } else {
     all=lua_toboolean(L, 2);
   }
   int i=1;
   dir = opendir_chdk(dirname,od_flags);
   if(!dir) 
     return os_pushresult(L, 0 , dirname);
   lua_newtable(L); 
   while((de = readdir(dir))) {
     if(!all && (de->d_name[0] == '\xE5' || (strcmp(de->d_name,".")==0) || (strcmp(de->d_name,"..")==0)))
       continue;
     lua_pushinteger(L, i);
     lua_pushstring(L, de->d_name);
     lua_settable(L,-3);
     ++i;
   }
   closedir(dir);
   return 1;
 #endif
 }

static int os_mkdir ( lua_State * L)

static

Definiert in Zeile 247 der Datei loslib.c.

                                    {
   const char *dirname = luaL_checkstring(L, 1);
 #if defined(HOST_LUA) && !defined(_WIN32)
   return os_pushresult(L, mkdir(dirname,0777) == 0, dirname);
 #else
   return os_pushresult(L, mkdir(dirname) == 0, dirname);
 #endif
 }

static int os_pushresult	(	lua_State *	L,
		int	i,
		const char *	filename
	)

static

Definiert in Zeile 27 der Datei loslib.c.

                                                                      {
   int en = errno;  /* calls to Lua API may change this value */
   if (i) {
     lua_pushboolean(L, 1);
     return 1;
   }
   else {
     lua_pushnil(L);
     lua_pushfstring(L, "%s: %s", filename, strerror(en));
     lua_pushinteger(L, en);
     return 3;
   }
 }

static int os_remove ( lua_State * L)

static

Definiert in Zeile 50 der Datei loslib.c.

                                     {
   const char *filename = luaL_checkstring(L, 1);
   return os_pushresult(L, remove(filename) == 0, filename);
 }

static int os_rename ( lua_State * L)

static

Definiert in Zeile 56 der Datei loslib.c.

                                     {
   const char *fromname = luaL_checkstring(L, 1);
   const char *toname = luaL_checkstring(L, 2);
   return os_pushresult(L, rename(fromname, toname) == 0, fromname);
 }

static int os_stat ( lua_State * L)

static

Definiert in Zeile 426 der Datei loslib.c.

                                   {
   struct stat st;
   const char *name = luaL_checkstring(L, 1);
   int result = stat(name,&st);
   if (result==0) {
     lua_createtable(L, 0, 6);  /* = number of fields */
     // don't expose the fields that aren't useful
     // but leave them commented out for reference
 //#ifndef CAM_DRYOS_2_3_R39
 //    setfield(L,"dev",st.st_dev);      /* device ID number */
 //    setfield(L,"ino",st.st_ino);      /* no inodes in fat, always -1 */
 //    setfield(L,"mode",st.st_mode);    /* file mode (see below) */
 //#endif
 //    setfield(L,"nlink",st.st_nlink);  /* dryos 0, vxworks 1 */
 //    setfield(L,"uid",st.st_uid);      /* no users or groups on fat */
 //    setfield(L,"gid",st.st_gid);      /* " */
 //#if !CAM_DRYOS
 // doesn't appear useful, I wasn't able to stat any special files
 //    setfield(L,"rdev",st.st_rdev);    /* device ID, only if special file */
 //#endif
     setfield(L,"size",st.st_size);  /* size of file, in bytes */
 //#ifndef CAM_DRYOS_2_3_R39
 //    setfield(L,"atime",st.st_atime);  /* time of last access */
 //#endif
     setfield(L,"mtime",st.st_mtime);    /* time of last modification */
     setfield(L,"ctime",st.st_ctime);    /* time of last change of file status */
 #ifdef HOST_LUA
 // fill in some sane values if we aren't running on the camera
 // from chdk stdlib
 #define DOS_ATTR_DIRECTORY      0x10            /* entry is a sub-directory */
 #ifndef CAM_DRYOS_2_3_R39
     setfield(L,"blksize",512); 
     setfield(L,"blocks",(st.st_size/512) + (st.st_size%512)?1:0); 
 #endif
     if ( S_ISDIR(st.st_mode) ) {
       setfield(L,"attrib",DOS_ATTR_DIRECTORY);
       setboolfield(L,"is_dir",1);
       setboolfield(L,"is_file",0);
     }
     else {
       setboolfield(L,"is_dir",0);
       setfield(L,"attrib",0);
       if S_ISREG(st.st_mode) {
         setboolfield(L,"is_file",1);
       }
     }
 #else
 //#ifndef CAM_DRYOS_2_3_R39
 //    setfield(L,"blksize",st.st_blksize); /* This is NOT the dos sector size. Appears to be 512 on all I've tested! */
 //    setfield(L,"blocks",st.st_blocks);   /* number of blocks required to store file. May not be the same as size on disk, per above*/
 //#endif
     setfield(L,"attrib",st.st_attrib);  /* file attribute byte (dosFs only) */
     // for convenience
     // note volume labels are neither file nor directory
     setboolfield(L,"is_dir",st.st_attrib & DOS_ATTR_DIRECTORY);
     setboolfield(L,"is_file",!(st.st_attrib & (DOS_ATTR_DIRECTORY | DOS_ATTR_VOL_LABEL)));
 #endif
 #if 0
     setfield(L,"reserved1",st.reserved1);
     setfield(L,"reserved2",st.reserved2);
     setfield(L,"reserved3",st.reserved3);
     setfield(L,"reserved4",st.reserved4);
     setfield(L,"reserved5",st.reserved5);
     setfield(L,"reserved6",st.reserved6);
 #endif
     return 1;
   }
   else {
     int en = errno;
     lua_pushnil(L);
     lua_pushfstring(L, "%s: %s", name, strerror(en));
     lua_pushinteger(L, en);
     return 3;
   }
 }

static int os_time ( lua_State * L)

static

Definiert in Zeile 187 der Datei loslib.c.

                                   {
   time_t t;
   if (lua_isnoneornil(L, 1))  /* called without args? */
     t = time(NULL);  /* get current time */
   else {
     struct tm ts;
     luaL_checktype(L, 1, LUA_TTABLE);
     lua_settop(L, 1);  /* make sure table is at the top */
     ts.tm_sec = getfield(L, "sec", 0);
     ts.tm_min = getfield(L, "min", 0);
     ts.tm_hour = getfield(L, "hour", 12);
     ts.tm_mday = getfield(L, "day", -1);
     ts.tm_mon = getfield(L, "month", -1) - 1;
     ts.tm_year = getfield(L, "year", -1) - 1900;
     ts.tm_isdst = getboolfield(L, "isdst");
     t = mktime(&ts);
   }
   if (t == (time_t)(-1))
     lua_pushnil(L);
   else
     lua_pushnumber(L, (lua_Number)t);
   return 1;
 }

static int os_utime ( lua_State * L)

static

Definiert in Zeile 505 der Datei loslib.c.

                                    {
   const char *name = luaL_checkstring(L, 1);
   struct utimbuf t;
   t.modtime = luaL_optnumber(L, 2, time(NULL));
   t.actime = luaL_optnumber(L, 3, time(NULL));
   return os_pushresult(L, utime(name,&t) == 0, name);
 }

static void setboolfield	(	lua_State *	L,
		const char *	key,
		int	value
	)

static

Definiert in Zeile 104 der Datei loslib.c.

                                                                     {
   if (value < 0)  /* undefined? */
     return;  /* does not set field */
   lua_pushboolean(L, value);
   lua_setfield(L, -2, key);
 }

static void setfield	(	lua_State *	L,
		const char *	key,
		int	value
	)

static

Definiert in Zeile 99 der Datei loslib.c.

                                                                 {
   lua_pushinteger(L, value);
   lua_setfield(L, -2, key);
 }

Variablen-Dokumentation

const luaL_Reg idir_meta_methods[]

static

Initialisierung:

= {
  {"__gc", idir_gc},
  {NULL, NULL}
}

Definiert in Zeile 415 der Datei loslib.c.

const luaL_Reg syslib[]

static

Initialisierung:

= {
  {"date",      os_date},
  {"difftime",  os_difftime},
  {"mkdir",     os_mkdir}, 
  {"listdir",   os_listdir}, 
  {"idir",      os_idir}, 
  {"stat",      os_stat}, 
  {"utime",     os_utime}, 
  {"remove",    os_remove},
  {"rename",    os_rename},
  {"time",      os_time},
  {NULL, NULL}
}

Definiert in Zeile 513 der Datei loslib.c.

Datenstrukturen

Makrodefinitionen

Funktionen

Variablen

Makro-Dokumentation

Dokumentation der Funktionen

Variablen-Dokumentation