Fix yet more casts of ctype.h macro arguments - and fix some cases where
[obnox/wireshark/wip.git] / 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 <string.h>
50 #include <ctype.h>
51
52 #include <glib.h>
53
54 #include <wiretap/file_util.h>
55 #include "globals.h"
56
57 #include <epan/filesystem.h>
58
59 #include "fileset.h"
60
61
62
63 typedef struct _fileset {
64   GList         *entries;
65   const char    *dirname;
66 } fileset;
67
68 /* this is the fileset's global data */
69 fileset set = { NULL, NULL};
70
71
72 /* is this a probable file of a file set (does the naming pattern match)? */
73 gboolean
74 fileset_filename_match_pattern(const char *fname)
75 {
76     char        *pfx;
77     int         baselen;
78     int         minlen = strlen("_00001_20050418010750");
79     char        *filename;
80   
81
82     /* d:\dir1\test_00001_20050418010750.cap */
83     filename = g_strdup(get_basename(fname));
84
85     /* test_00001_20050418010750.cap */
86     pfx = strrchr(filename, '.');
87     if(pfx == NULL) {  /* suffix is optional */
88         pfx = filename + strlen(filename);
89     }
90     /* test_00001_20050418010750 */
91     *pfx = '\0';
92
93     /* filename long enough? */
94     baselen = strlen(filename);
95     if(baselen < minlen) {
96         g_free(filename);
97         return FALSE;
98     }
99
100     /* there must be two underscores at special places */
101     if(filename[baselen-minlen] != '_' || filename[baselen-minlen+6] != '_') {
102         g_free(filename);
103         return FALSE;
104     }
105
106     /* replace the two underscores by digits */
107     filename[baselen-minlen] = '0';
108     filename[baselen-minlen+6] = '0';
109
110     /* we should have only digits now */
111     while(minlen--) {
112         baselen--;
113
114         if(!isdigit( (guchar) filename[baselen])) {
115             g_free(filename);
116             return FALSE;
117         }
118     }
119
120     g_free(filename);
121
122     /* ok, seems to be good */
123     return TRUE;
124 }
125
126
127 /* test, if both files could be in the same file set */
128 /* (the filenames must already be in correct shape) */
129 gboolean
130 fileset_is_file_in_set(const char *fname1, const char *fname2)
131 {
132     char        *pfx1;
133     char        *pfx2;
134     char        *dup_f1;
135     char        *dup_f2;
136     int         minlen = strlen("_00001_20050418010750");
137
138
139     /* just to be sure ... */
140     g_assert(fileset_filename_match_pattern(fname1));
141     g_assert(fileset_filename_match_pattern(fname2));
142
143     dup_f1 = g_strdup(fname1);
144     dup_f2 = g_strdup(fname2);
145
146     pfx1 = strrchr(dup_f1, '.');
147     pfx2 = strrchr(dup_f2, '.');
148     /* suffix is optional */
149     if (!pfx1) pfx1 = dup_f1 + strlen(dup_f1);
150     if (!pfx2) pfx2 = dup_f2 + strlen(dup_f2);
151
152     /* the optional suffix (file extension) must be equal */
153     if(strcmp(pfx1, pfx2) != 0) {
154         g_free(dup_f1);
155         g_free(dup_f2);
156         return FALSE;
157     }
158
159     *(pfx1-minlen) = '\0';
160     *(pfx2-minlen) = '\0';
161
162     if(strcmp(dup_f1, dup_f2) != 0) {
163         g_free(dup_f1);
164         g_free(dup_f2);
165         return FALSE;
166     }
167
168     g_free(dup_f1);
169     g_free(dup_f2);
170     return TRUE;
171 }
172
173
174 /* we know this file is part of the set, so add it */
175 static fileset_entry *
176 fileset_add_file(const char *dirname, const char *fname, gboolean current)
177 {
178     int fh, result;
179     struct stat buf;
180     char *path;
181     fileset_entry *entry = NULL;
182
183
184     path = g_strdup_printf("%s%s", dirname, fname);
185
186     fh = eth_open( path, O_RDONLY, 0000 /* no creation so don't matter */);
187     if(fh !=  -1) {
188
189         /* Get statistics */
190         result = fstat( fh, &buf );
191
192         /* Show statistics if they are valid */
193         if( result == 0 ) {
194             entry = g_malloc(sizeof(fileset_entry));
195
196             entry->fullname = g_strdup(path);
197             entry->name     = g_strdup(fname);
198             entry->ctime    = buf.st_ctime;
199             entry->mtime    = buf.st_mtime;
200             entry->size     = buf.st_size;
201             entry->current  = current;
202
203             set.entries = g_list_append(set.entries, entry);
204         }
205
206         eth_close(fh);
207     }
208
209     g_free(path);
210
211     return entry;
212 }
213
214
215 /* compare two list entries by creation date/time (through filename) */
216 static gint
217 fileset_sort_compare(gconstpointer a, gconstpointer b)
218 {
219     const fileset_entry *entry_a = a;
220     const fileset_entry *entry_b = b;
221
222         return strcmp(entry_a->name, entry_b->name);
223 }
224
225
226 /* add all file set entries to the dialog */
227 void fileset_update_dlg(void)
228 {
229     GList         *le;
230
231
232     /* add all entires to the dialog */
233     le = g_list_first(set.entries);
234     while(le) {
235         fileset_dlg_add_file(le->data);
236         le = g_list_next(le);
237     }
238 }
239
240
241 /* walk through the directory of the loaded file and add every file matching the current file */
242 void
243 fileset_add_dir(const char *fname)
244 {
245     ETH_DIR       *dir;             /* scanned directory */
246     ETH_DIRENT    *file;            /* current file */
247     const char    *name;
248     fileset_entry *entry;
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 = eth_dir_open(dirname->str, 0, NULL)) != NULL) {
266                 while ((file = eth_dir_read_name(dir)) != NULL) {
267                     name = eth_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             eth_dir_close(dir);
274         } /* if */
275     } else {
276         /* no, this is a "standalone file", just add this one */
277         entry = 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 = 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 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 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 = data;
366
367     g_free( (gpointer) entry->fullname);
368     entry->fullname = NULL;
369     g_free( (gpointer) entry->name);
370     entry->name = NULL;
371 }
372
373
374 /* delete the whole file set */
375 void fileset_delete(void)
376 {
377     /* free the entry list */
378     if(set.entries) {
379         g_list_foreach(set.entries, fileset_entry_delete, NULL);
380         g_list_free(set.entries);
381         set.entries = NULL;
382     }
383
384     /* free the rest */
385     if(set.dirname) {
386         g_free( (gpointer) set.dirname);
387         set.dirname = NULL;
388     }
389 }
390
391