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