Make dissection of AVP: 3GPP-User-Location-Info(22) l=15 f=V-- vnd=TGPP val=303231...
[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 <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     ws_statb64 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 = ws_fstat64( 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