(hopefully) removed various gcc warnings
[obnox/wireshark/wip.git] / fileset.c
1 /* fileset.c
2  * Routines for handling file sets
3  *
4  * $Id$
5  *
6  * Ethereal - Network traffic analyzer
7  * By Gerald Combs <gerald@ethereal.com>
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_IO_H
34 #include <io.h>
35 #endif
36
37 #ifdef HAVE_FCNTL_H
38 #include <fcntl.h>
39 #endif
40
41 #ifdef HAVE_DIRENT_H
42 #include <dirent.h>
43 #endif
44
45 #ifdef HAVE_SYS_STAT_H
46 # include <sys/stat.h>
47 #endif
48
49 #ifdef HAVE_SYS_WAIT_H
50 # include <sys/wait.h>
51 #endif
52
53 #include <string.h>
54 #include <ctype.h>
55
56 #include <glib.h>
57
58 #include "globals.h"
59
60 #include <epan/filesystem.h>
61
62 #include "fileset.h"
63
64
65
66 typedef struct _fileset {
67   GList         *entries;
68   const char    *dirname;
69 } fileset;
70
71 /* this is the fileset's global data */
72 fileset set = { NULL, NULL};
73
74
75 /* is this a probable file of a file set (does the naming pattern match)? */
76 gboolean
77 fileset_filename_match_pattern(const char *fname)
78 {
79     char        *pfx;
80     int         baselen;
81     int         minlen = strlen("_00001_20050418010750");
82     char        *filename;
83   
84
85     /* d:\dir1\test_00001_20050418010750.cap */
86     filename = g_strdup(get_basename(fname));
87
88     /* test_00001_20050418010750.cap */
89     pfx = strrchr(filename, '.');
90     if(pfx == NULL) {
91         return FALSE;
92     }
93     /* test_00001_20050418010750 */
94     *pfx = '\0';
95
96     /* filename long enough? */
97     baselen = strlen(filename);
98     if(baselen < minlen) {
99         g_free(filename);
100         return FALSE;
101     }
102
103     /* there must be two underscores at special places */
104     if(filename[baselen-minlen] != '_' || filename[baselen-minlen+6] != '_') {
105         g_free(filename);
106         return FALSE;
107     }
108
109     /* replace the two underscores by digits */
110     filename[baselen-minlen] = '0';
111     filename[baselen-minlen+6] = '0';
112
113     /* we should have only digits now */
114     while(minlen--) {
115         baselen--;
116
117         if(!isdigit( (int) filename[baselen])) {
118             g_free(filename);
119             return FALSE;
120         }
121     }
122
123     g_free(filename);
124
125     /* ok, seems to be good */
126     return TRUE;
127 }
128
129
130 /* test, if both files could be in the same file set */
131 gboolean
132 fileset_is_file_in_set(const char *fname1, const char *fname2)
133 {
134     char        *pfx1;
135     char        *pfx2;
136     char        *dup_f1;
137     char        *dup_f2;
138     int         minlen = strlen("_00001_20050418010750");
139
140
141     dup_f1 = g_strdup(fname1);
142     dup_f2 = g_strdup(fname2);
143
144     pfx1 = strrchr(dup_f1, '.');
145     pfx2 = strrchr(dup_f2, '.');
146
147     if(strcmp(pfx1, pfx2) != 0) {
148         g_free(dup_f1);
149         g_free(dup_f2);
150         return FALSE;
151     }
152
153     *(pfx1-minlen) = '\0';
154     *(pfx2-minlen) = '\0';
155
156     if(strcmp(dup_f1, dup_f2) != 0) {
157         g_free(dup_f1);
158         g_free(dup_f2);
159         return FALSE;
160     }
161
162     g_free(dup_f1);
163     g_free(dup_f2);
164     return TRUE;
165 }
166
167
168 /* we know this file is part of the set, so add it */
169 fileset_entry *
170 fileset_add_file(const char *dirname, const char *fname, gboolean current)
171 {
172     int fh, result;
173     struct stat buf;
174     char *path;
175     fileset_entry *entry = NULL;
176
177
178     path = g_strdup_printf("%s%s", dirname, fname);
179
180     fh = open( path, O_RDONLY );
181     if(fh !=  -1) {
182
183         /* Get statistics */
184         result = fstat( fh, &buf );
185
186         /* Show statistics if they are valid */
187         if( result == 0 ) {
188             entry = g_malloc(sizeof(fileset_entry));
189
190             entry->fullname = g_strdup(path);
191             entry->name     = g_strdup(fname);
192             entry->ctime    = buf.st_ctime;
193             entry->mtime    = buf.st_mtime;
194             entry->size     = buf.st_size;
195             entry->current  = current;
196
197             set.entries = g_list_append(set.entries, entry);
198         }
199
200         close(fh);
201     }
202
203     g_free(path);
204
205     return entry;
206 }
207
208
209 /* compare two list entries by creation date/time */
210 gint
211 fileset_compare_creation(gconstpointer a, gconstpointer b)
212 {
213     const fileset_entry *entry_a = a;
214     const fileset_entry *entry_b = b;
215
216     return entry_a->ctime - entry_b->ctime;
217 }
218
219
220 /* add all file set entries to the dialog */
221 void fileset_update_dlg(void)
222 {
223     GList         *le;
224
225
226     /* add all entires to the dialog */
227     le = g_list_first(set.entries);
228     while(le) {
229         fileset_dlg_add_file(le->data);
230         le = g_list_next(le);
231     }
232 }
233
234
235 /* walk through the directory of the loaded file and add every file matching the current file */
236 void
237 fileset_add_dir(const char *fname)
238 {
239 #if GLIB_MAJOR_VERSION < 2
240     DIR           *dir;             /* scanned directory */
241     struct dirent *file;            /* current file */
242     gchar         *name;
243 #else /* GLIB 2 */
244     GDir          *dir;             /* scanned directory */
245     GError        **dummy;
246     const char    *name;
247 #endif
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     /* if the current file can't be part of any fileset, do nothing */
263     if(!fileset_filename_match_pattern(fname)) {
264         entry = fileset_add_file(dirname->str, get_basename(fname), TRUE /* current */);
265         if(entry) {
266             fileset_dlg_add_file(entry);
267         }
268     }
269
270     /* go through the files in the directory and check if it's part of the current file set */
271 #if GLIB_MAJOR_VERSION < 2
272     if ((dir = opendir(dirname->str)) != NULL) {
273             while ((file = readdir(dir)) != NULL) {
274                 name = (gchar *)file->d_name;
275 #else
276     dummy = g_malloc(sizeof(GError *));
277     *dummy = NULL;
278
279     if ((dir = g_dir_open(dirname->str, 0, dummy)) != NULL) {
280         while ((name = g_dir_read_name(dir)) != NULL) {
281 #endif
282             if(fileset_filename_match_pattern(name) && fileset_is_file_in_set(name, get_basename(fname))) {
283                 fileset_add_file(dirname->str, name, strcmp(name, get_basename(fname))== 0 /* current */);
284             }
285         }
286     }
287
288 #if GLIB_MAJOR_VERSION >= 2
289     g_free(dummy);
290 #endif
291     g_string_free(dirname, TRUE /* free_segment */);
292
293     /* sort entries by creation time */
294     set.entries = g_list_sort(set.entries, fileset_compare_creation);
295
296     fileset_update_dlg();
297 }
298
299
300 /* get current directory name */
301 const char *
302 fileset_get_dirname(void)
303 {
304     return set.dirname;
305 }
306
307
308 /* get the current list entry, or NULL */
309 GList *
310 fileset_get_current(void)
311 {
312     GList         *le;
313     fileset_entry *entry;
314
315
316     /* add all entires to the dialog */
317     le = g_list_first(set.entries);
318     while(le) {
319         entry = le->data;
320         if(entry->current) {
321             return le;
322         }
323         le = g_list_next(le);
324     }
325
326     return NULL;
327 }
328
329
330 /* get the file set entry after the current one, or NULL */
331 fileset_entry *
332 fileset_get_next(void)
333 {
334     GList         *le;
335
336
337     le = fileset_get_current();
338     if(le == NULL) {
339         return NULL;
340     }
341
342     le = g_list_next(le);
343     if(le == NULL) {
344         return NULL;
345     }
346
347     return le->data;
348 }
349
350
351 /* get the file set entry before the current one, or NULL */
352 fileset_entry *
353 fileset_get_previous(void)
354 {
355     GList         *le;
356
357
358     le = fileset_get_current();
359     if(le == NULL) {
360         return NULL;
361     }
362
363     le = g_list_previous(le);
364     if(le == NULL) {
365         return NULL;
366     }
367
368     return le->data;
369 }
370
371
372 /* delete a single entry */
373 void fileset_entry_delete(gpointer data, gpointer user_data _U_)
374 {
375     fileset_entry *entry = data;
376
377     g_free( (gpointer) entry->fullname);
378     entry->fullname = NULL;
379     g_free( (gpointer) entry->name);
380     entry->name = NULL;
381 }
382
383
384 /* delete the whole file set */
385 void fileset_delete(void)
386 {
387     /* free the entry list */
388     if(set.entries) {
389         g_list_foreach(set.entries, fileset_entry_delete, NULL);
390         g_list_free(set.entries);
391         set.entries = NULL;
392     }
393
394     /* free the rest */
395     if(set.dirname) {
396         g_free( (gpointer) set.dirname);
397         set.dirname = NULL;
398     }
399 }
400
401