fileset.c

   1 /* fileset.c
   2  * Routines for handling file sets
   3  *
   4  * $Id$
   5  *
   6  * Wireshark - Network traffic analyzer
   7  * By Gerald Combs <gerald@wireshark.org>
   8  * Copyright 1998 Gerald Combs
   9  *
  10  * This program is free software; you can redistribute it and/or
  11  * modify it under the terms of the GNU General Public License
  12  * as published by the Free Software Foundation; either version 2
  13  * of the License, or (at your option) any later version.
  14  *
  15  * This program is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  * GNU General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with this program; if not, write to the Free Software
  22  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  23  */
  24
  25 #ifdef HAVE_CONFIG_H
  26 # include "config.h"
  27 #endif
  28
  29 #ifdef HAVE_UNISTD_H
  30 #include <unistd.h>
  31 #endif
  32
  33 #ifdef HAVE_FCNTL_H
  34 #include <fcntl.h>
  35 #endif
  36
  37 #ifdef HAVE_DIRENT_H
  38 #include <dirent.h>
  39 #endif
  40
  41 #ifdef HAVE_SYS_STAT_H
  42 # include <sys/stat.h>
  43 #endif
  44
  45 #ifdef HAVE_SYS_WAIT_H
  46 # include <sys/wait.h>
  47 #endif
  48
  49 #include <stdio.h>
  50 #include <string.h>
  51 #include <ctype.h>
  52
  53 #include <glib.h>
  54
  55 #include <wsutil/file_util.h>
  56 #include "globals.h"
  57
  58 #include <epan/filesystem.h>
  59
  60 #include "fileset.h"
  61
  62
  63
  64 typedef struct _fileset {
  65     GList   *entries;
  66     char    *dirname;
  67 } fileset;
  68
  69 /* this is the fileset's global data */
  70 static fileset set = { NULL, NULL};
  71
  72
  73 /* is this a probable file of a file set (does the naming pattern match)? */
  74 gboolean
  75 fileset_filename_match_pattern(const char *fname)
  76 {
  77     char        *pfx;
  78     size_t       baselen;
  79     size_t      minlen = strlen("_00001_20050418010750");
  80     char        *filename;
  81
  82
  83     /* d:\dir1\test_00001_20050418010750.cap */
  84     filename = g_strdup(get_basename(fname));
  85
  86     /* test_00001_20050418010750.cap */
  87     pfx = strrchr(filename, '.');
  88     if(pfx == NULL) {  /* suffix is optional */
  89         pfx = filename + strlen(filename);
  90     }
  91     /* test_00001_20050418010750 */
  92     *pfx = '\0';
  93
  94     /* filename long enough? */
  95     baselen = strlen(filename);
  96     if(baselen < minlen) {
  97         g_free(filename);
  98         return FALSE;
  99     }
 100
 101     /* there must be two underscores at special places */
 102     if(filename[baselen-minlen] != '_' || filename[baselen-minlen+6] != '_') {
 103         g_free(filename);
 104         return FALSE;
 105     }
 106
 107     /* replace the two underscores by digits */
 108     filename[baselen-minlen] = '0';
 109     filename[baselen-minlen+6] = '0';
 110
 111     /* we should have only digits now */
 112     while(minlen--) {
 113         baselen--;
 114
 115         if(!isdigit( (guchar) filename[baselen])) {
 116             g_free(filename);
 117             return FALSE;
 118         }
 119     }
 120
 121     g_free(filename);
 122
 123     /* ok, seems to be good */
 124     return TRUE;
 125 }
 126
 127
 128 /* test, if both files could be in the same file set */
 129 /* (the filenames must already be in correct shape) */
 130 gboolean
 131 fileset_is_file_in_set(const char *fname1, const char *fname2)
 132 {
 133     char        *pfx1;
 134     char        *pfx2;
 135     char        *dup_f1;
 136     char        *dup_f2;
 137     size_t       minlen = strlen("_00001_20050418010750");
 138
 139
 140     /* just to be sure ... */
 141     g_assert(fileset_filename_match_pattern(fname1));
 142     g_assert(fileset_filename_match_pattern(fname2));
 143
 144     dup_f1 = g_strdup(fname1);
 145     dup_f2 = g_strdup(fname2);
 146
 147     pfx1 = strrchr(dup_f1, '.');
 148     pfx2 = strrchr(dup_f2, '.');
 149     /* suffix is optional */
 150     if (!pfx1) pfx1 = dup_f1 + strlen(dup_f1);
 151     if (!pfx2) pfx2 = dup_f2 + strlen(dup_f2);
 152
 153     /* the optional suffix (file extension) must be equal */
 154     if(strcmp(pfx1, pfx2) != 0) {
 155         g_free(dup_f1);
 156         g_free(dup_f2);
 157         return FALSE;
 158     }
 159
 160     *(pfx1-minlen) = '\0';
 161     *(pfx2-minlen) = '\0';
 162
 163     if(strcmp(dup_f1, dup_f2) != 0) {
 164         g_free(dup_f1);
 165         g_free(dup_f2);
 166         return FALSE;
 167     }
 168
 169     g_free(dup_f1);
 170     g_free(dup_f2);
 171     return TRUE;
 172 }
 173
 174
 175 /* we know this file is part of the set, so add it */
 176 static fileset_entry *
 177 fileset_add_file(const char *dirname, const char *fname, gboolean current)
 178 {
 179     int fh, result;
 180     struct stat buf;
 181     char *path;
 182     fileset_entry *entry = NULL;
 183
 184
 185     path = g_strdup_printf("%s%s", dirname, fname);
 186
 187     fh = ws_open( path, O_RDONLY, 0000 /* no creation so don't matter */);
 188     if(fh !=  -1) {
 189
 190         /* Get statistics */
 191         result = fstat( fh, &buf );
 192
 193         /* Show statistics if they are valid */
 194         if( result == 0 ) {
 195             entry = (fileset_entry *)g_malloc(sizeof(fileset_entry));
 196
 197             entry->fullname = g_strdup(path);
 198             entry->name     = g_strdup(fname);
 199             entry->ctime    = buf.st_ctime;
 200             entry->mtime    = buf.st_mtime;
 201             entry->size     = buf.st_size;
 202             entry->current  = current;
 203
 204             set.entries = g_list_append(set.entries, entry);
 205         }
 206
 207         ws_close(fh);
 208     }
 209
 210     g_free(path);
 211
 212     return entry;
 213 }
 214
 215
 216 /* compare two list entries by creation date/time (through filename) */
 217 static gint
 218 fileset_sort_compare(gconstpointer a, gconstpointer b)
 219 {
 220     const fileset_entry *entry_a = (const fileset_entry *)a;
 221     const fileset_entry *entry_b = (const fileset_entry *)b;
 222
 223     return strcmp(entry_a->name, entry_b->name);
 224 }
 225
 226
 227 /* add all file set entries to the dialog */
 228 void fileset_update_dlg(void)
 229 {
 230     GList         *le;
 231
 232
 233     /* add all entires to the dialog */
 234     le = g_list_first(set.entries);
 235     while(le) {
 236         fileset_dlg_add_file((fileset_entry *)le->data);
 237         le = g_list_next(le);
 238     }
 239 }
 240
 241
 242 /* walk through the directory of the loaded file and add every file matching the current file */
 243 void
 244 fileset_add_dir(const char *fname)
 245 {
 246     WS_DIR        *dir;             /* scanned directory */
 247     WS_DIRENT     *file;            /* current file */
 248     const char    *name;
 249     GString       *dirname;
 250     gchar         *fname_dup;
 251
 252
 253     /* get (convert) directory name, but don't touch the given string */
 254     fname_dup = get_dirname(g_strdup(fname));
 255     dirname = g_string_new(fname_dup);
 256     g_free(fname_dup);
 257
 258     set.dirname = g_strdup(dirname->str);
 259
 260     dirname = g_string_append_c(dirname, G_DIR_SEPARATOR);
 261
 262     /* is the current file probably a part of any fileset? */
 263     if(fileset_filename_match_pattern(fname)) {
 264         /* yes, go through the files in the directory and check if the file in question is part of the current file set */
 265         if ((dir = ws_dir_open(dirname->str, 0, NULL)) != NULL) {
 266             while ((file = ws_dir_read_name(dir)) != NULL) {
 267                 name = ws_dir_get_name(file);
 268                 if(fileset_filename_match_pattern(name) && fileset_is_file_in_set(name, get_basename(fname))) {
 269                     fileset_add_file(dirname->str, name, strcmp(name, get_basename(fname))== 0 /* current */);
 270                 }
 271             } /* while */
 272
 273             ws_dir_close(dir);
 274         } /* if */
 275     } else {
 276         /* no, this is a "standalone file", just add this one */
 277         fileset_add_file(dirname->str, get_basename(fname), TRUE /* current */);
 278         /* don't add the file to the dialog here, this will be done in fileset_update_dlg() below */
 279     }
 280
 281     g_string_free(dirname, TRUE /* free_segment */);
 282
 283     /* sort entries by creation time */
 284     set.entries = g_list_sort(set.entries, fileset_sort_compare);
 285
 286     fileset_update_dlg();
 287 }
 288
 289
 290 /* get current directory name */
 291 const char *
 292 fileset_get_dirname(void)
 293 {
 294     return set.dirname;
 295 }
 296
 297
 298 /* get the current list entry, or NULL */
 299 static GList *
 300 fileset_get_current(void)
 301 {
 302     GList         *le;
 303     fileset_entry *entry;
 304
 305
 306     /* add all entires to the dialog */
 307     le = g_list_first(set.entries);
 308     while(le) {
 309         entry = (fileset_entry *)le->data;
 310         if(entry->current) {
 311             return le;
 312         }
 313         le = g_list_next(le);
 314     }
 315
 316     return NULL;
 317 }
 318
 319
 320 /* get the file set entry after the current one, or NULL */
 321 fileset_entry *
 322 fileset_get_next(void)
 323 {
 324     GList         *le;
 325
 326
 327     le = fileset_get_current();
 328     if(le == NULL) {
 329         return NULL;
 330     }
 331
 332     le = g_list_next(le);
 333     if(le == NULL) {
 334         return NULL;
 335     }
 336
 337     return (fileset_entry *)le->data;
 338 }
 339
 340
 341 /* get the file set entry before the current one, or NULL */
 342 fileset_entry *
 343 fileset_get_previous(void)
 344 {
 345     GList         *le;
 346
 347
 348     le = fileset_get_current();
 349     if(le == NULL) {
 350         return NULL;
 351     }
 352
 353     le = g_list_previous(le);
 354     if(le == NULL) {
 355         return NULL;
 356     }
 357
 358     return (fileset_entry *)le->data;
 359 }
 360
 361
 362 /* delete a single entry */
 363 static void fileset_entry_delete(gpointer data, gpointer user_data _U_)
 364 {
 365     fileset_entry *entry = (fileset_entry *)data;
 366
 367     g_free( (gpointer) entry->fullname);
 368     entry->fullname = NULL;
 369     g_free( (gpointer) entry->name);
 370     entry->name = NULL;
 371     g_free(entry);
 372 }
 373
 374
 375 /* delete the whole file set */
 376 void fileset_delete(void)
 377 {
 378     /* free the entry list */
 379     if(set.entries) {
 380         g_list_foreach(set.entries, fileset_entry_delete, NULL);
 381         g_list_free(set.entries);
 382         set.entries = NULL;
 383     }
 384
 385     /* free the rest */
 386     if(set.dirname) {
 387         g_free( (gpointer) set.dirname);
 388         set.dirname = NULL;
 389     }
 390 }
 391
 392