osmith
/
wireshark
forked from osmocom/wireshark
1
0
Fork 0
Code Pull Requests Releases Wiki Activity
wireshark/fileset.c

436 lines
10 KiB
C
Raw Blame History

/* fileset.c
* Routines for handling file sets
*
* Wireshark - Network traffic analyzer
* By Gerald Combs <gerald@wireshark.org>
* Copyright 1998 Gerald Combs
*
* SPDX-License-Identifier: GPL-2.0-or-later
*/
#include <config.h>
#ifdef HAVE_SYS_WAIT_H
# include <sys/wait.h>
#endif
#include <stdio.h>
#include <string.h>
#include <glib.h>
#include <wsutil/file_util.h>
#include <wsutil/filesystem.h>
#include <wsutil/ws_assert.h>
#include <epan/strutil.h>
#include "fileset.h"
typedef struct _fileset {
GList *entries;
char *dirname;
} fileset;
/*
* This is the fileset's global data.
*
* XXX This should probably be per-main-window instead of global.
*/
static fileset set = { NULL, NULL};
/*
* Given a stat structure, get the creation time of the file if available,
* or 0 if not.
*/
#ifdef _WIN32
/* Microsoft's documentation says this is the creation time */
#define ST_CREATE_TIME(statb) ((statb).st_ctime)
#else /* _WIN32 */
/* UN*X - do we have a creation time? */
#if defined(HAVE_STRUCT_STAT_ST_BIRTHTIME)
#define ST_CREATE_TIME(statb) ((statb).st_birthtime)
#elif defined(HAVE_STRUCT_STAT___ST_BIRTHTIME)
#define ST_CREATE_TIME(statb) ((statb).__st_birthtime)
#else /* nothing */
#define ST_CREATE_TIME(statb) (0)
#endif /* creation time on UN*X */
#endif /* _WIN32 */
/* is this a probable file of a file set (does the naming pattern match)? */
gboolean
fileset_filename_match_pattern(const char *fname)
{
char *pfx;
size_t baselen;
size_t minlen = strlen("_00001_20050418010750");
char *filename;
/* d:\dir1\test_00001_20050418010750.cap */
filename = g_strdup(get_basename(fname));
/* test_00001_20050418010750.cap */
pfx = strrchr(filename, '.');
if(pfx == NULL) { /* suffix is optional */
pfx = filename + strlen(filename);
}
/* test_00001_20050418010750 */
*pfx = '\0';
/* filename long enough? */
baselen = strlen(filename);
if(baselen < minlen) {
g_free(filename);
return FALSE;
}
/* there must be two underscores at special places */
if(filename[baselen-minlen] != '_' || filename[baselen-minlen+6] != '_') {
g_free(filename);
return FALSE;
}
/* replace the two underscores by digits */
filename[baselen-minlen] = '0';
filename[baselen-minlen+6] = '0';
/* we should have only digits now */
while(minlen--) {
baselen--;
if(!g_ascii_isdigit( filename[baselen])) {
g_free(filename);
return FALSE;
}
}
g_free(filename);
/* ok, seems to be good */
return TRUE;
}
/* test, if both files could be in the same file set */
/* (the filenames must already be in correct shape) */
static gboolean
fileset_is_file_in_set(const char *fname1, const char *fname2)
{
char *pfx1;
char *pfx2;
char *dup_f1;
char *dup_f2;
size_t minlen = strlen("_00001_20050418010750");
/* just to be sure ... */
ws_assert(fileset_filename_match_pattern(fname1));
ws_assert(fileset_filename_match_pattern(fname2));
dup_f1 = g_strdup(fname1);
dup_f2 = g_strdup(fname2);
pfx1 = strrchr(dup_f1, '.');
pfx2 = strrchr(dup_f2, '.');
/* suffix is optional */
if (!pfx1) pfx1 = dup_f1 + strlen(dup_f1);
if (!pfx2) pfx2 = dup_f2 + strlen(dup_f2);
/* the optional suffix (file extension) must be equal */
if(strcmp(pfx1, pfx2) != 0) {
g_free(dup_f1);
g_free(dup_f2);
return FALSE;
}
*(pfx1-minlen) = '\0';
*(pfx2-minlen) = '\0';
if(strcmp(dup_f1, dup_f2) != 0) {
g_free(dup_f1);
g_free(dup_f2);
return FALSE;
}
g_free(dup_f1);
g_free(dup_f2);
return TRUE;
}
/* GCompareFunc helper for g_list_find_custom() */
static gint
fileset_find_by_path(gconstpointer a, gconstpointer b)
{
const fileset_entry *entry;
const char *path;
entry = (const fileset_entry *) a;
path = (const char *) b;
return g_strcmp0(entry->fullname, path);
}
/* update the time and size of this file in the list */
void
fileset_update_file(const char *path)
{
int fh, result;
ws_statb64 buf;
fileset_entry *entry = NULL;
GList *entry_list;
fh = ws_open( path, O_RDONLY, 0000 /* no creation so don't matter */);
if(fh != -1) {
/* Get statistics */
result = ws_fstat64( fh, &buf );
/* Show statistics if they are valid */
if( result == 0 ) {
entry_list = g_list_find_custom(set.entries, path,
fileset_find_by_path);
if (entry_list) {
entry = (fileset_entry *) entry_list->data;
entry->ctime = ST_CREATE_TIME(buf);
entry->mtime = buf.st_mtime;
entry->size = buf.st_size;
}
}
ws_close(fh);
}
}
/* we know this file is part of the set, so add it */
static fileset_entry *
fileset_add_file(const char *dirname, const char *fname, gboolean current)
{
int fh, result;
ws_statb64 buf;
char *path;
fileset_entry *entry = NULL;
path = ws_strdup_printf("%s%s", dirname, fname);
fh = ws_open( path, O_RDONLY, 0000 /* no creation so don't matter */);
if(fh != -1) {
/* Get statistics */
result = ws_fstat64( fh, &buf );
/* Show statistics if they are valid */
if( result == 0 ) {
entry = g_new(fileset_entry, 1);
entry->fullname = g_strdup(path);
entry->name = g_strdup(fname);
entry->ctime = ST_CREATE_TIME(buf);
entry->mtime = buf.st_mtime;
entry->size = buf.st_size;
entry->current = current;
set.entries = g_list_append(set.entries, entry);
}
ws_close(fh);
}
g_free(path);
return entry;
}
/* compare two list entries by creation date/time (through filename) */
static gint
fileset_sort_compare(gconstpointer a, gconstpointer b)
{
const fileset_entry *entry_a = (const fileset_entry *)a;
const fileset_entry *entry_b = (const fileset_entry *)b;
return strcmp(entry_a->name, entry_b->name);
}
/* add all file set entries to the dialog */
void fileset_update_dlg(void *window)
{
GList *le;
/* Add all entries to the dialog. */
fileset_dlg_begin_add_file(window);
le = g_list_first(set.entries);
while(le) {
fileset_dlg_add_file((fileset_entry *)le->data, window);
le = g_list_next(le);
}
fileset_dlg_end_add_file(window);
}
/* walk through the directory of the loaded file and add every file matching the current file */
void
fileset_add_dir(const char *fname, void *window)
{
WS_DIR *dir; /* scanned directory */
WS_DIRENT *file; /* current file */
const char *name;
GString *dirname;
gchar *fname_dup;
/* get (convert) directory name, but don't touch the given string */
fname_dup = g_strdup(fname);
dirname = g_string_new(get_dirname(fname_dup));
g_free(fname_dup);
set.dirname = g_strdup(dirname->str);
dirname = g_string_append_c(dirname, G_DIR_SEPARATOR);
/* is the current file probably a part of any fileset? */
if(fileset_filename_match_pattern(fname)) {
/* yes, go through the files in the directory and check if the file in question is part of the current file set */
if ((dir = ws_dir_open(dirname->str, 0, NULL)) != NULL) {
while ((file = ws_dir_read_name(dir)) != NULL) {
name = ws_dir_get_name(file);
if(fileset_filename_match_pattern(name) && fileset_is_file_in_set(name, get_basename(fname))) {
fileset_add_file(dirname->str, name, strcmp(name, get_basename(fname))== 0 /* current */);
}
} /* while */
ws_dir_close(dir);
} /* if */
} else {
/* no, this is a "standalone file", just add this one */
fileset_add_file(dirname->str, get_basename(fname), TRUE /* current */);
/* don't add the file to the dialog here, this will be done in fileset_update_dlg() below */
}
g_string_free(dirname, TRUE /* free_segment */);
/* sort entries by creation time */
set.entries = g_list_sort(set.entries, fileset_sort_compare);
fileset_update_dlg(window);
}
/* get current directory name */
const char *
fileset_get_dirname(void)
{
return set.dirname;
}
/* get the current list entry, or NULL */
static GList *
fileset_get_current(void)
{
GList *le;
fileset_entry *entry;
/* add all entries to the dialog */
le = g_list_first(set.entries);
while(le) {
entry = (fileset_entry *)le->data;
if(entry->current) {
return le;
}
le = g_list_next(le);
}
return NULL;
}
/* get the file set entry after the current one, or NULL */
fileset_entry *
fileset_get_next(void)
{
GList *le;
le = fileset_get_current();
if(le == NULL) {
return NULL;
}
le = g_list_next(le);
if(le == NULL) {
return NULL;
}
return (fileset_entry *)le->data;
}
/* get the file set entry before the current one, or NULL */
fileset_entry *
fileset_get_previous(void)
{
GList *le;
le = fileset_get_current();
if(le == NULL) {
return NULL;
}
le = g_list_previous(le);
if(le == NULL) {
return NULL;
}
return (fileset_entry *)le->data;
}
/* delete a single entry */
static void fileset_entry_delete(gpointer data, gpointer user_data _U_)
{
fileset_entry *entry = (fileset_entry *)data;
g_free( (gpointer) entry->fullname);
entry->fullname = NULL;
g_free( (gpointer) entry->name);
entry->name = NULL;
g_free(entry);
}
/* delete the whole file set */
void fileset_delete(void)
{
/* free the entry list */
if(set.entries) {
g_list_foreach(set.entries, fileset_entry_delete, NULL);
g_list_free(set.entries);
set.entries = NULL;
}
/* free the rest */
if(set.dirname) {
g_free( (gpointer) set.dirname);
set.dirname = NULL;
}
}
/*
* Editor modelines - https://www.wireshark.org/tools/modelines.html
*
* Local variables:
* c-basic-offset: 4
* tab-width: 8
* indent-tabs-mode: nil
* End:
*
* vi: set shiftwidth=4 tabstop=8 expandtab:
* :indentSize=4:tabSize=8:noTabs=true:
*/
View Git Blame Copy Permalink