- add persconffile(opt_str)

- add datafile_path(opt_str)
- add Dir class -- a Directory Iterator

git-svn-id: http://anonsvn.wireshark.org/wireshark/trunk@19507 f5534014-38df-0310-8fa8-9805f1628bb7

diff --git a/epan/wslua/wslua.h b/epan/wslua/wslua.h
index c8dde86a9ad4beb91fcab1dfb37f22bf742f782d..6490fa492e97cebc7d8801022cd480e4489da324 100644 (file)
--- a/epan/wslua/wslua.h
+++ b/epan/wslua/wslua.h
@@ -121,12 +121,32 @@ struct _wslua_treeitem {
        proto_tree* tree;
 };
 
-#if 0
-struct _wslua_main {
-       capture_options capture_opts;
-};
+
+#if GLIB_MAJOR_VERSION < 2
+#  define DIRECTORY_T DIR
+#  define FILE_T struct dirent
+#  define OPENDIR_OP(name) opendir(name)
+#  define DIRGETNEXT_OP(dir) readdir(dir)
+#  define GETFNAME_OP(file) (gchar *)file->d_name
+#  define CLOSEDIR_OP(dir) closedir(dir)
+#else /* GLIB 2 */
+#  define DIRECTORY_T GDir
+#  define FILE_T gchar
+#  define OPENDIR_OP(name) g_dir_open(name, 0, dir->dummy)
+#  define DIRGETNEXT_OP(dir) g_dir_read_name(dir)
+#  define GETFNAME_OP(file) (file);
+#  define CLOSEDIR_OP(dir) g_dir_close(dir)
+#endif
+
+struct _wslua_dir {
+       DIRECTORY_T* dir;
+       char* ext;
+#if GLIB_MAJOR_VERSION >= 2
+       GError** dummy;
 #endif
 
+};
+
 typedef void (*tap_extractor_t)(lua_State*,const void*);
 
 typedef struct { const char* name; tap_extractor_t extractor; } tappable_t;
@@ -158,6 +178,7 @@ typedef tvbparse_wanted_t* Rule;
 typedef tvbparse_elem_t* Node;
 typedef tvbparse_action_t* Shortcut;
 typedef struct _wslua_main* WireShark;
+typedef struct _wslua_dir* Dir;
 
 /*
  * toXxx(L,idx) gets a Xxx from an index (Lua Error if fails)

diff --git a/epan/wslua/wslua_util.c b/epan/wslua/wslua_util.c
index 85a876c6f3321aeabc7d3a28ea881dc182bff659..7b426fd150ce92f108a5a6047b38ef3724295360 100644 (file)
--- a/epan/wslua/wslua_util.c
+++ b/epan/wslua/wslua_util.c
@@ -149,11 +149,9 @@ WSLUA_FUNCTION wslua_loadfile(lua_State* L) {
        const char *given_fname = luaL_checkstring(L, WSLUA_ARG_loadfile_FILENAME);
        const char* filename;
        
-       if (!given_fname) WSLUA_ARG_ERROR(loadfile,FILENAME,"must be a string");
-       
        filename = get_actual_filename(given_fname);
        
-       if (!filename)  WSLUA_ARG_ERROR(loadfile,FILENAME,"file does not exist");
+       if (!filename) WSLUA_ARG_ERROR(loadfile,FILENAME,"file does not exist");
        
        if (luaL_loadfile(L, filename) == 0) {
                return 1;
@@ -181,3 +179,151 @@ WSLUA_FUNCTION wslua_dofile(lua_State* L) {
        lua_call(L, 0, LUA_MULTRET);
        return lua_gettop(L) - n;
 }
+
+
+WSLUA_FUNCTION wslua_persconffile_path(lua_State* L) {
+#define WSLUA_OPTARG_persconffile_path_FILENAME 1
+       const char *fname = luaL_optstring(L, WSLUA_OPTARG_persconffile_path_FILENAME,"");
+       const char* filename = get_persconffile_path(fname,FALSE);
+       
+       lua_pushstring(L,filename);
+       return 1;
+}
+
+WSLUA_FUNCTION wslua_datafile_path(lua_State* L) {
+#define WSLUA_OPTARG_datafile_path_FILENAME 1
+       const char *fname = luaL_optstring(L, WSLUA_OPTARG_datafile_path_FILENAME,"");
+       const char* filename = get_datafile_path(fname);
+
+       lua_pushstring(L,filename);
+       return 1;
+}
+
+
+WSLUA_CLASS_DEFINE(Dir,NOP,NOP); /* A Directory */
+
+WSLUA_CONSTRUCTOR wslua_Dir_open(lua_State* L) {
+#define WSLUA_ARG_Dir_open_PATHNAME 1
+#define WSLUA_OPTARG_Dir_open_EXTENSION 2
+       
+       const char* dirname = luaL_checkstring(L,WSLUA_ARG_Dir_open_PATHNAME);
+       const char* extension = luaL_optstring(L,WSLUA_OPTARG_Dir_open_EXTENSION,NULL);
+       Dir dir;
+
+       if (!dirname) WSLUA_ARG_ERROR(Dir_open,PATHNAME,"must be a string");
+       if (!test_for_directory(dirname))  WSLUA_ARG_ERROR(Dir_open,PATHNAME,"must be a directory");
+
+       dir = g_malloc(sizeof(struct _wslua_dir));
+       dir->dir = OPENDIR_OP(dirname);
+       dir->ext = extension ? g_strdup(extension) : NULL;
+#if GLIB_MAJOR_VERSION >= 2
+       dir->dummy = g_malloc(sizeof(GError *));
+       *(dir->dummy) = NULL;
+#endif
+       
+       if (dir->dir == NULL) {
+#if GLIB_MAJOR_VERSION >= 2
+               g_free(dir->dummy);
+#endif
+               g_free(dir);
+
+               WSLUA_ARG_ERROR(Dir_open,PATHNAME,"could not open directory");
+               return 0;
+       }
+       
+       pushDir(L,dir);
+       return 1;
+}
+
+WSLUA_METAMETHOD wslua_Dir__call(lua_State* L) {
+#define WSLUA_ARG_Dir__call_DIR 1
+       Dir dir = checkDir(L,1);
+       const FILE_T* file;
+       const gchar* filename;
+       const char* ext;
+       
+       if (!dir) 
+               WSLUA_ARG_ERROR(Dir__call,DIR,"must be a Dir");
+
+       if (!dir->dir) {
+               return 0;
+       }
+       
+       if ( ! ( file = DIRGETNEXT_OP(dir->dir ) )) {
+               CLOSEDIR_OP(dir->dir);
+               dir->dir = NULL;
+               return 0;
+       }
+
+
+       if ( ! dir->ext ) {
+               filename = GETFNAME_OP(file);
+               lua_pushstring(L,filename);
+               return 1;
+       }
+       
+       do {
+               filename = GETFNAME_OP(file);
+
+               /* XXX strstr returns ptr to first match,
+                       this fails ext=".xxx" filename="aaa.xxxz.xxx"  */
+               if ( ( ext = strstr(filename,dir->ext)) && g_str_equal(ext,dir->ext) ) {
+                       lua_pushstring(L,filename);
+                       return 1;
+               }
+       } while(( file = DIRGETNEXT_OP(dir->dir) ));
+               
+       CLOSEDIR_OP(dir->dir);
+       dir->dir = NULL;
+       return 0;
+}
+
+WSLUA_METHOD wslua_Dir_close(lua_State* L) {
+#define WSLUA_ARG_Dir_close_DIR 1
+       Dir dir = checkDir(L,1);
+
+       if (dir->dir) {
+               CLOSEDIR_OP(dir->dir);
+               dir->dir = NULL;
+       }
+
+       return 0;
+}
+
+WSLUA_METAMETHOD wslua_Dir__gc(lua_State* L) {
+#define WSLUA_ARG_Dir__gc_DIR 1
+       Dir dir = checkDir(L,1);
+       
+       if (dir->dir) {
+               CLOSEDIR_OP(dir->dir);
+       }
+       
+#if GLIB_MAJOR_VERSION >= 2
+       g_free(dir->dummy);
+#endif
+       
+       if (dir->ext) g_free(dir->ext);
+
+       g_free(dir);
+       
+       return 0;
+}
+
+static const luaL_reg Dir_methods[] = {
+    {"open", wslua_Dir_open},
+    {"close", wslua_Dir_close},
+    {0, 0}
+};
+
+static const luaL_reg Dir_meta[] = {
+    {"__call", wslua_Dir__call},
+    {"__gc", wslua_Dir__gc},
+    {0, 0}
+};
+
+int Dir_register(lua_State* L) {
+       
+    WSLUA_REGISTER_CLASS(Dir);
+       
+    return 1;
+}