aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorGuy Harris <guy@alum.mit.edu>2005-08-19 19:40:00 +0000
committerGuy Harris <guy@alum.mit.edu>2005-08-19 19:40:00 +0000
commit38ec1644e68721d6b5afb84dce1684b943b1aee0 (patch)
tree8ffe4f642463b43885354f4dd755e59b4daf3823
parent06823cdce8a0e63cf888c449a1b37071d622f1c3 (diff)
Add APIs to Wiretap to return the file of the size as supplied by the OS
(so if the file's gzipped, it's *NOT* the size of the file after uncompressing), and an approximation of the amount of that data read sequentially so far. Use those for various progress bars and the like. Make the fstat() in the Ascend trace reader directly use wth->fd, as it's inside Wiretap; that gets rid of the last caller of wtap_fd() (as we're no longer directly using fstat() or lseek() in Ethereal), so get rid of wtap_fd(). svn path=/trunk/; revision=15437
-rw-r--r--capinfos.c19
-rw-r--r--cfile.h3
-rw-r--r--file.c71
-rw-r--r--gtk/file_dlg.c16
-rw-r--r--gtk/fileset_dlg.c9
-rw-r--r--merge.c12
-rw-r--r--merge.h2
-rw-r--r--ringbuffer.c6
-rw-r--r--tethereal.c1
-rw-r--r--wiretap/ascend.c2
-rw-r--r--wiretap/wtap.c51
-rw-r--r--wiretap/wtap.def3
-rw-r--r--wiretap/wtap.h8
13 files changed, 103 insertions, 100 deletions
diff --git a/capinfos.c b/capinfos.c
index fc398bf41e..60c73de3b3 100644
--- a/capinfos.c
+++ b/capinfos.c
@@ -41,10 +41,6 @@
#include <sys/time.h>
#endif
-#ifdef HAVE_SYS_STAT_H
-#include <sys/stat.h>
-#endif
-
#include <glib.h>
#include <epan/packet.h>
@@ -70,7 +66,7 @@ static gboolean cap_packet_size = FALSE;
typedef struct _capture_info {
const char *filename;
guint16 file_type;
- guint64 filesize;
+ gint64 filesize;
guint64 packet_bytes;
double start_time;
double stop_time;
@@ -106,7 +102,7 @@ print_stats(capture_info *cf_info)
if (cap_file_type) printf("File type: %s\n", file_type_string);
if (cap_packet_count) printf("Number of packets: %u \n", cf_info->packet_count);
- if (cap_file_size) printf("File size: %" PRIu64 " bytes\n", cf_info->filesize);
+ if (cap_file_size) printf("File size: %" PRId64 " bytes\n", cf_info->filesize);
if (cap_data_size) printf("Data size: %" PRIu64 " bytes\n", cf_info->packet_bytes);
if (cap_duration) printf("Capture duration: %f seconds\n", cf_info->duration);
if (cap_start_time) printf("Start time: %s", ctime (&start_time_t));
@@ -121,7 +117,7 @@ process_cap_file(wtap *wth, const char *filename)
{
int err;
gchar *err_info;
- struct stat cf_stat;
+ gint64 size;
long data_offset;
guint32 packet = 0;
@@ -166,14 +162,15 @@ process_cap_file(wtap *wth, const char *filename)
}
/* File size */
- if (fstat(wtap_fd(wth), &cf_stat) < 0) {
+ size = wtap_file_size(wth, &err);
+ if (size == -1) {
fprintf(stderr,
- "capinfos: Can't fstat \"%s\": %s.\n",
- filename, strerror(errno));
+ "capinfos: Can't get size of \"%s\": %s.\n",
+ filename, strerror(err));
return 1;
}
- cf_info.filesize = cf_stat.st_size;
+ cf_info.filesize = size;
/* File Type */
cf_info.file_type = wtap_file_type(wth);
diff --git a/cfile.h b/cfile.h
index d78461c72d..59f60bb51f 100644
--- a/cfile.h
+++ b/cfile.h
@@ -43,12 +43,11 @@ typedef enum {
typedef struct _capture_file {
file_state state; /* Current state of capture file */
- int filed; /* File descriptor of capture file */
gchar *filename; /* Name of capture file */
gboolean is_tempfile; /* Is capture file a temporary file? */
gboolean user_saved;/* If capture file is temporary, has it been saved by user yet? */
long f_datalen; /* Size of capture file data (uncompressed) */
- long f_len; /* Length of capture file (compressed if file is) */
+ gint64 f_len; /* Length of capture file (compressed if file is) */
guint16 cd_t; /* File type of capture file */
int lnk_t; /* Link-layer type with which to save capture */
guint32 vers; /* Version. For tcpdump minor is appended to major */
diff --git a/file.c b/file.c
index d14ce5cde6..8425b6c70e 100644
--- a/file.c
+++ b/file.c
@@ -43,10 +43,6 @@
#include <errno.h>
#include <signal.h>
-#ifdef HAVE_SYS_STAT_H
-#include <sys/stat.h>
-#endif
-
#ifdef HAVE_FCNTL_H
#include <fcntl.h>
#endif
@@ -171,18 +167,15 @@ cf_open(capture_file *cf, const char *fname, gboolean is_tempfile, int *err)
{
wtap *wth;
gchar *err_info;
- int fd;
- struct stat cf_stat;
-
+ gint64 size;
wth = wtap_open_offline(fname, err, &err_info, TRUE);
if (wth == NULL)
goto fail;
/* Find the size of the file. */
- fd = wtap_fd(wth);
- if (fstat(fd, &cf_stat) < 0) {
- *err = errno;
+ size = wtap_file_size(wth, err);
+ if (size == -1) {
wtap_close(wth);
goto fail;
}
@@ -198,9 +191,8 @@ cf_open(capture_file *cf, const char *fname, gboolean is_tempfile, int *err)
cf->state = FILE_READ_IN_PROGRESS;
cf->wth = wth;
- cf->filed = fd;
cf->f_datalen = 0;
- cf->f_len = cf_stat.st_size;
+ cf->f_len = size;
/* Set the file name because we need it to set the follow stream filter.
XXX - is that still true? We need it for other reasons, though,
@@ -335,15 +327,8 @@ cf_read(capture_file *cf)
long data_offset;
progdlg_t *progbar = NULL;
gboolean stop_flag;
- /*
- * XXX - should be "off_t", but Wiretap would need more work to handle
- * the full size of "off_t" on platforms where it's more than a "long"
- * as well.
- */
- long file_pos;
+ gint64 size, file_pos;
float prog_val;
- int fd;
- struct stat cf_stat;
GTimeVal start_time;
gchar status_str[100];
int progbar_nextstep;
@@ -376,17 +361,17 @@ cf_read(capture_file *cf)
to see if there's any pending input from an X server, and doing
that for every packet can be costly, especially on a big file. */
if (data_offset >= progbar_nextstep) {
- file_pos = lseek(cf->filed, 0, SEEK_CUR);
+ file_pos = wtap_read_so_far(cf->wth, NULL);
prog_val = (gfloat) file_pos / (gfloat) cf->f_len;
if (prog_val > 1.0) {
/* The file probably grew while we were reading it.
Update "cf->f_len", and try again. */
- fd = wtap_fd(cf->wth);
- if (fstat(fd, &cf_stat) >= 0) {
- cf->f_len = cf_stat.st_size;
+ size = wtap_file_size(cf->wth, NULL);
+ if (size != -1) {
+ cf->f_len = size;
prog_val = (gfloat) file_pos / (gfloat) cf->f_len;
}
- /* If it's still > 1, either the "fstat()" failed (in which
+ /* If it's still > 1, either "wtap_file_size()" failed (in which
case there's not much we can do about it), or the file
*shrank* (in which case there's not much we can do about
it); just clip the progress value at 1.0. */
@@ -400,7 +385,8 @@ cf_read(capture_file *cf)
}
if (progbar != NULL) {
g_snprintf(status_str, sizeof(status_str),
- "%luKB of %luKB", file_pos / 1024, cf->f_len / 1024);
+ "%" PRId64 "KB of %" PRId64 "KB",
+ file_pos / 1024, cf->f_len / 1024);
update_progress_dlg(progbar, prog_val, status_str);
}
progbar_nextstep += progbar_quantum;
@@ -553,9 +539,7 @@ cf_finish_tail(capture_file *cf, int *err)
{
gchar *err_info;
long data_offset;
- int fd;
- struct stat cf_stat;
-
+ gint64 size;
if(cf->wth == NULL) {
cf_close(cf);
@@ -595,10 +579,9 @@ cf_finish_tail(capture_file *cf, int *err)
/* we have to update the f_len field */
/* Find the size of the file. */
- fd = wtap_fd(cf->wth);
- if (fstat(fd, &cf_stat) >= 0) {
- cf->f_len = cf_stat.st_size;
- }
+ size = wtap_file_size(cf->wth, NULL);
+ if (size != -1)
+ cf->f_len = size;
/* We're done reading sequentially through the file; close the
sequential I/O side, to free up memory it requires. */
@@ -930,17 +913,12 @@ cf_merge_files(char **out_filenamep, int in_file_count,
long data_offset;
progdlg_t *progbar = NULL;
gboolean stop_flag;
- /*
- * XXX - should be "off_t", but Wiretap would need more work to handle
- * the full size of "off_t" on platforms where it's more than a "long"
- * as well.
- */
- long f_len, file_pos;
- float prog_val;
- GTimeVal start_time;
- gchar status_str[100];
- int progbar_nextstep;
- int progbar_quantum;
+ gint64 f_len, file_pos;
+ float prog_val;
+ GTimeVal start_time;
+ gchar status_str[100];
+ int progbar_nextstep;
+ int progbar_quantum;
/* open the input files */
if (!merge_open_in_files(in_file_count, in_filenames, &in_files,
@@ -1020,7 +998,7 @@ cf_merge_files(char **out_filenamep, int in_file_count,
/* Get the sum of the seek positions in all of the files. */
file_pos = 0;
for (i = 0; i < in_file_count; i++)
- file_pos += lseek(wtap_fd(in_files[i].wth), 0, SEEK_CUR);
+ file_pos += wtap_read_so_far(in_files[i].wth, NULL);
prog_val = (gfloat) file_pos / (gfloat) f_len;
if (prog_val > 1.0) {
/* Some file probably grew while we were reading it.
@@ -1035,7 +1013,8 @@ cf_merge_files(char **out_filenamep, int in_file_count,
}
if (progbar != NULL) {
g_snprintf(status_str, sizeof(status_str),
- "%luKB of %luKB", file_pos / 1024, f_len / 1024);
+ "%" PRId64 "KB of %" PRId64 "KB",
+ file_pos / 1024, f_len / 1024);
update_progress_dlg(progbar, prog_val, status_str);
}
progbar_nextstep += progbar_quantum;
diff --git a/gtk/file_dlg.c b/gtk/file_dlg.c
index 6493738dea..ac4af2fa1c 100644
--- a/gtk/file_dlg.c
+++ b/gtk/file_dlg.c
@@ -116,10 +116,6 @@ static GtkWidget *cfmark_cb;
static GtkWidget *ft_om;
static GtkWidget *range_tb;
-#ifdef HAVE_SYS_STAT_H
-#include <sys/stat.h>
-#endif
-
#define PREVIEW_STR_MAX 200
static double
@@ -137,9 +133,8 @@ preview_set_filename(GtkWidget *prev, const gchar *cf_name)
wtap *wth;
int err = 0;
gchar *err_info;
- struct stat cf_stat;
gchar string_buff[PREVIEW_STR_MAX];
- guint64 filesize;
+ gint64 filesize;
/* init preview labels */
@@ -181,14 +176,13 @@ preview_set_filename(GtkWidget *prev, const gchar *cf_name)
}
/* Find the size of the file. */
- if (fstat(wtap_fd(wth), &cf_stat) < 0) {
+ filesize = wtap_file_size(wth, &err);
+ if (filesize == -1) {
+ gtk_label_set_text(GTK_LABEL(label), "error getting file size");
wtap_close(wth);
return NULL;
}
-
- /* size */
- filesize = cf_stat.st_size;
- g_snprintf(string_buff, PREVIEW_STR_MAX, "%" PRIu64 " bytes", filesize);
+ g_snprintf(string_buff, PREVIEW_STR_MAX, "%" PRId64 " bytes", filesize);
label = OBJECT_GET_DATA(prev, PREVIEW_SIZE_KEY);
gtk_label_set_text(GTK_LABEL(label), string_buff);
diff --git a/gtk/fileset_dlg.c b/gtk/fileset_dlg.c
index ef3910d3a4..941ab8cc19 100644
--- a/gtk/fileset_dlg.c
+++ b/gtk/fileset_dlg.c
@@ -36,15 +36,6 @@
#include <fcntl.h>
#endif
-#ifdef HAVE_SYS_STAT_H
-# include <sys/stat.h>
-#endif
-
-#ifdef HAVE_SYS_WAIT_H
-# include <sys/wait.h>
-#endif
-
-
#include <gtk/gtk.h>
#include "globals.h"
diff --git a/merge.c b/merge.c
index d82c14136b..f2fb294aa2 100644
--- a/merge.c
+++ b/merge.c
@@ -24,10 +24,6 @@
#include <sys/time.h>
#endif
-#ifdef HAVE_SYS_STAT_H
-#include <sys/stat.h>
-#endif
-
#include <string.h>
#include "wtap.h"
#include "merge.h"
@@ -43,7 +39,7 @@ merge_open_in_files(int in_file_count, char *const *in_file_names,
int i, j;
int files_size = in_file_count * sizeof(merge_in_file_t);
merge_in_file_t *files;
- struct stat statb;
+ gint64 size;
files = g_malloc(files_size);
*in_files = files;
@@ -60,14 +56,14 @@ merge_open_in_files(int in_file_count, char *const *in_file_names,
*err_fileno = i;
return FALSE;
}
- if (fstat(wtap_fd(files[i].wth), &statb) < 0) {
- *err = errno;
+ size = wtap_file_size(files[i].wth, err);
+ if (size == -1) {
for (j = 0; j <= i; j++)
wtap_close(files[j].wth);
*err_fileno = i;
return FALSE;
}
- files[i].size = statb.st_size;
+ files[i].size = size;
}
return TRUE;
}
diff --git a/merge.h b/merge.h
index 887e1b866c..209f78bc05 100644
--- a/merge.h
+++ b/merge.h
@@ -44,7 +44,7 @@ typedef struct merge_in_file_s {
wtap *wth;
long data_offset;
in_file_state_e state;
- long size; /* file size */
+ gint64 size; /* file size */
} merge_in_file_t;
/** Open a number of input files to merge.
diff --git a/ringbuffer.c b/ringbuffer.c
index 02c66138a2..1400420a68 100644
--- a/ringbuffer.c
+++ b/ringbuffer.c
@@ -56,10 +56,6 @@
#include <fcntl.h>
#endif
-#ifdef HAVE_SYS_STAT_H
-#include <sys/stat.h>
-#endif
-
#ifdef HAVE_UNISTD_H
#include <unistd.h>
#endif
@@ -69,7 +65,7 @@
#include <time.h>
#include <errno.h>
-#include "wiretap/wtap.h"
+#include <wiretap/wtap.h>
#include "ringbuffer.h"
/* Win32 needs the O_BINARY flag for open() */
diff --git a/tethereal.c b/tethereal.c
index 13f7ce0820..75d67923cf 100644
--- a/tethereal.c
+++ b/tethereal.c
@@ -2970,7 +2970,6 @@ cf_open(capture_file *cf, const char *fname, gboolean is_tempfile, int *err)
init_dissection();
cf->wth = wth;
- cf->filed = -1; /* not used, but set it anyway */
cf->f_datalen = 0; /* not used, but set it anyway */
cf->f_len = 0; /* not used, but set it anyway */
diff --git a/wiretap/ascend.c b/wiretap/ascend.c
index 3b239226ce..5c57279905 100644
--- a/wiretap/ascend.c
+++ b/wiretap/ascend.c
@@ -218,7 +218,7 @@ int ascend_open(wtap *wth, int *err, gchar **err_info _U_)
packet's timestamp from the capture file's ctime, which gives us an
offset that we can apply to each packet.
*/
- if (fstat(wtap_fd(wth), &statbuf) == -1) {
+ if (fstat(wth->fd, &statbuf) == -1) {
*err = errno;
g_free(wth->capture.ascend);
return -1;
diff --git a/wiretap/wtap.c b/wiretap/wtap.c
index 8876b9c060..727c77d286 100644
--- a/wiretap/wtap.c
+++ b/wiretap/wtap.c
@@ -27,14 +27,41 @@
#include <string.h>
#include <errno.h>
+#ifdef HAVE_SYS_TYPES_H
+#include <sys/types.h>
+#endif
+
+#ifdef HAVE_SYS_STAT_H
+#include <sys/stat.h>
+#endif
+
+#ifdef HAVE_UNISTD_H
+#include <unistd.h>
+#endif
+
+#ifdef HAVE_IO_H
+#include <io.h>
+#endif
+
#include "wtap-int.h"
#include "file_wrappers.h"
#include "buffer.h"
-int
-wtap_fd(wtap *wth)
+/*
+ * Return the size of the file, as reported by the OS.
+ * (gint64, in case that's 64 bits.)
+ */
+gint64
+wtap_file_size(wtap *wth, int *err)
{
- return wth->fd;
+ struct stat statb;
+
+ if (fstat(wth->fd, &statb) == -1) {
+ if (err != NULL)
+ *err = errno;
+ return -1;
+ }
+ return statb.st_size;
}
int
@@ -465,6 +492,24 @@ wtap_read(wtap *wth, int *err, gchar **err_info, long *data_offset)
return TRUE; /* success */
}
+/*
+ * Return an approximation of the amount of data we've read sequentially
+ * from the file so far. (gint64, in case that's 64 bits.)
+ */
+gint64
+wtap_read_so_far(wtap *wth, int *err)
+{
+ off_t file_pos;
+
+ file_pos = lseek(wth->fd, 0, SEEK_CUR);
+ if (file_pos == -1) {
+ if (err != NULL)
+ *err = errno;
+ return -1;
+ }
+ return file_pos;
+}
+
struct wtap_pkthdr*
wtap_phdr(wtap *wth)
{
diff --git a/wiretap/wtap.def b/wiretap/wtap.def
index 1554ebefc4..1e314db475 100644
--- a/wiretap/wtap.def
+++ b/wiretap/wtap.def
@@ -10,8 +10,8 @@ wtap_dump_file
wtap_dump_open
wtap_encap_short_string
wtap_encap_string
-wtap_fd
wtap_file_encap
+wtap_file_size
wtap_file_type
wtap_file_type_short_string
wtap_file_type_string
@@ -23,6 +23,7 @@ wtap_phdr
wtap_process_pcap_packet
wtap_pseudoheader
wtap_read
+wtap_read_so_far
wtap_seek_read
wtap_sequential_close
wtap_short_string_to_encap
diff --git a/wiretap/wtap.h b/wiretap/wtap.h
index 8008dde650..b828a7e4b6 100644
--- a/wiretap/wtap.h
+++ b/wiretap/wtap.h
@@ -523,11 +523,17 @@ struct wtap* wtap_open_offline(const char *filename, int *err,
gboolean wtap_read(wtap *wth, int *err, gchar **err_info,
long *data_offset);
+/*
+ * Return an approximation of the amount of data we've read sequentially
+ * from the file so far. (gint64, in case that's 64 bits.)
+ */
+gint64 wtap_read_so_far(wtap *wth, int *err);
+
struct wtap_pkthdr *wtap_phdr(wtap *wth);
union wtap_pseudo_header *wtap_pseudoheader(wtap *wth);
guint8 *wtap_buf_ptr(wtap *wth);
-int wtap_fd(wtap *wth);
+gint64 wtap_file_size(wtap *wth, int *err);
int wtap_snapshot_length(wtap *wth); /* per file */
int wtap_file_type(wtap *wth);
int wtap_file_encap(wtap *wth);