add new function tvb_get_ephemeral_stringz()
[obnox/wireshark/wip.git] / file.c
1 /* file.c
2  * File I/O routines
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 #include <time.h>
34
35 #ifdef HAVE_IO_H
36 #include <io.h>
37 #endif
38
39 #include <stdlib.h>
40 #include <stdio.h>
41 #include <string.h>
42 #include <ctype.h>
43 #include <errno.h>
44 #include <signal.h>
45
46 #ifdef HAVE_SYS_STAT_H
47 #include <sys/stat.h>
48 #endif
49
50 #ifdef HAVE_FCNTL_H
51 #include <fcntl.h>
52 #endif
53
54 #ifdef NEED_SNPRINTF_H
55 # include "snprintf.h"
56 #endif
57
58 #ifdef NEED_STRERROR_H
59 #include "strerror.h"
60 #endif
61
62 #include <epan/epan.h>
63 #include <epan/filesystem.h>
64
65 #include "color.h"
66 #include "color_filters.h"
67 #include <epan/column.h>
68 #include <epan/packet.h>
69 #include "packet-range.h"
70 #include "print.h"
71 #include "file.h"
72 #include "fileset.h"
73 #include "util.h"
74 #include "merge.h"
75 #include "alert_box.h"
76 #include "simple_dialog.h"
77 #include "progress_dlg.h"
78 #include "ui_util.h"
79 #include <epan/prefs.h>
80 #include <epan/dfilter/dfilter.h>
81 #include <epan/conversation.h>
82 #include <epan/epan_dissect.h>
83 #include <epan/tap.h>
84 #include "tap_dfilter_dlg.h"
85 #include <epan/dissectors/packet-data.h>
86
87 /* Win32 needs the O_BINARY flag for open() */
88 #ifndef O_BINARY
89 #define O_BINARY        0
90 #endif
91
92 #ifdef HAVE_LIBPCAP
93 gboolean auto_scroll_live;
94 #endif
95
96 static guint32 firstsec, firstusec;
97 static guint32 prevsec, prevusec;
98 static guint32 cum_bytes = 0;
99
100 static void cf_reset_state(capture_file *cf);
101
102 static void read_packet(capture_file *cf, long offset);
103
104 static void rescan_packets(capture_file *cf, const char *action, const char *action_item,
105         gboolean refilter, gboolean redissect);
106
107 static gboolean match_protocol_tree(capture_file *cf, frame_data *fdata,
108         void *criterion);
109 static void match_subtree_text(proto_node *node, gpointer data);
110 static gboolean match_summary_line(capture_file *cf, frame_data *fdata,
111         void *criterion);
112 static gboolean match_ascii_and_unicode(capture_file *cf, frame_data *fdata,
113         void *criterion);
114 static gboolean match_ascii(capture_file *cf, frame_data *fdata,
115         void *criterion);
116 static gboolean match_unicode(capture_file *cf, frame_data *fdata,
117         void *criterion);
118 static gboolean match_binary(capture_file *cf, frame_data *fdata,
119         void *criterion);
120 static gboolean match_dfilter(capture_file *cf, frame_data *fdata,
121         void *criterion);
122 static gboolean find_packet(capture_file *cf,
123         gboolean (*match_function)(capture_file *, frame_data *, void *),
124         void *criterion);
125
126 static void cf_open_failure_alert_box(const char *filename, int err,
127                                       gchar *err_info, gboolean for_writing,
128                                       int file_type);
129 static const char *file_rename_error_message(int err);
130 static void cf_write_failure_alert_box(const char *filename, int err);
131 static void cf_close_failure_alert_box(const char *filename, int err);
132 static   gboolean copy_binary_file(const char *from_filename, const char *to_filename);
133
134 /* Update the progress bar this many times when reading a file. */
135 #define N_PROGBAR_UPDATES       100
136
137 /* Number of "frame_data" structures per memory chunk.
138    XXX - is this the right number? */
139 #define FRAME_DATA_CHUNK_SIZE   1024
140
141
142 /* one callback for now, we could have a list later */
143 static cf_callback_t cf_cb = NULL;
144 static gpointer cf_cb_user_data = NULL;
145
146 void
147 cf_callback_invoke(int event, gpointer data)
148 {
149     g_assert(cf_cb != NULL);
150     cf_cb(event, data, cf_cb_user_data);
151 }
152
153
154 void
155 cf_callback_add(cf_callback_t func, gpointer user_data)
156 {
157     /* More than one callback listener is currently not implemented,
158        but should be easy to do. */
159     g_assert(cf_cb == NULL);
160     cf_cb = func;
161     cf_cb_user_data = user_data;
162 }
163
164 void
165 cf_callback_remove(cf_callback_t func _U_)
166 {
167     g_assert(cf_cb != NULL);
168     cf_cb = NULL;
169     cf_cb_user_data = NULL;
170 }
171
172
173 cf_status_t
174 cf_open(capture_file *cf, const char *fname, gboolean is_tempfile, int *err)
175 {
176   wtap       *wth;
177   gchar       *err_info;
178   int         fd;
179   struct stat cf_stat;
180
181
182   wth = wtap_open_offline(fname, err, &err_info, TRUE);
183   if (wth == NULL)
184     goto fail;
185
186   /* Find the size of the file. */
187   fd = wtap_fd(wth);
188   if (fstat(fd, &cf_stat) < 0) {
189     *err = errno;
190     wtap_close(wth);
191     goto fail;
192   }
193
194   /* The open succeeded.  Close whatever capture file we had open,
195      and fill in the information for this file. */
196   cf_reset_state(cf);
197
198   /* Initialize all data structures used for dissection. */
199   init_dissection();
200
201   /* We're about to start reading the file. */
202   cf->state = FILE_READ_IN_PROGRESS;
203
204   cf->wth = wth;
205   cf->filed = fd;
206   cf->f_len = cf_stat.st_size;
207
208   /* Set the file name because we need it to set the follow stream filter.
209      XXX - is that still true?  We need it for other reasons, though,
210      in any case. */
211   cf->filename = g_strdup(fname);
212
213   /* Indicate whether it's a permanent or temporary file. */
214   cf->is_tempfile = is_tempfile;
215
216   /* If it's a temporary capture buffer file, mark it as not saved. */
217   cf->user_saved = !is_tempfile;
218
219   cf->cd_t      = wtap_file_type(cf->wth);
220   cf->count     = 0;
221   cf->displayed_count = 0;
222   cf->marked_count = 0;
223   cf->drops_known = FALSE;
224   cf->drops     = 0;
225   cf->esec      = 0;
226   cf->eusec     = 0;
227   cf->snap      = wtap_snapshot_length(cf->wth);
228   if (cf->snap == 0) {
229     /* Snapshot length not known. */
230     cf->has_snap = FALSE;
231     cf->snap = WTAP_MAX_PACKET_SIZE;
232   } else
233     cf->has_snap = TRUE;
234   firstsec = 0, firstusec = 0;
235   prevsec = 0, prevusec = 0;
236
237   cf->plist_chunk = g_mem_chunk_new("frame_data_chunk",
238         sizeof(frame_data),
239         FRAME_DATA_CHUNK_SIZE * sizeof(frame_data),
240         G_ALLOC_AND_FREE);
241   g_assert(cf->plist_chunk);
242
243   fileset_file_opened(fname);
244
245   return CF_OK;
246
247 fail:
248   cf_open_failure_alert_box(fname, *err, err_info, FALSE, 0);
249   return CF_ERROR;
250 }
251
252
253 /*
254  * Reset the state for the currently closed file, but don't do the
255  * UI callbacks; this is for use in "cf_open()", where we don't
256  * want the UI to go from "file open" to "file closed" back to
257  * "file open", we want it to go from "old file open" to "new file
258  * open and being read".
259  */
260 static void
261 cf_reset_state(capture_file *cf)
262 {
263   /* Die if we're in the middle of reading a file. */
264   g_assert(cf->state != FILE_READ_IN_PROGRESS);
265
266   if (cf->wth) {
267     wtap_close(cf->wth);
268     cf->wth = NULL;
269   }
270   /* We have no file open... */
271   if (cf->filename != NULL) {
272     /* If it's a temporary file, remove it. */
273     if (cf->is_tempfile)
274       unlink(cf->filename);
275     g_free(cf->filename);
276     cf->filename = NULL;
277   }
278   /* ...which means we have nothing to save. */
279   cf->user_saved = FALSE;
280
281   if (cf->plist_chunk != NULL) {
282     g_mem_chunk_destroy(cf->plist_chunk);
283     cf->plist_chunk = NULL;
284   }
285   if (cf->rfcode != NULL) {
286     dfilter_free(cf->rfcode);
287     cf->rfcode = NULL;
288   }
289   cf->plist = NULL;
290   cf->plist_end = NULL;
291   cf_unselect_packet(cf);       /* nothing to select */
292   cf->first_displayed = NULL;
293   cf->last_displayed = NULL;
294
295   /* No frame selected, no field in that frame selected. */
296   cf->current_frame = NULL;
297   cf->finfo_selected = NULL;
298
299   /* Clear the packet list. */
300   packet_list_freeze();
301   packet_list_clear();
302   packet_list_thaw();
303
304   cf->f_len = 0;
305   cf->count = 0;
306   cf->esec  = 0;
307   cf->eusec = 0;
308
309   reset_tap_listeners();
310
311   /* We have no file open. */
312   cf->state = FILE_CLOSED;
313
314   fileset_file_closed();
315 }
316
317 /* Reset everything to a pristine state */
318 void
319 cf_close(capture_file *cf)
320 {
321   cf_reset_state(cf);
322
323   cleanup_dissection();
324
325   cf_callback_invoke(cf_cb_file_closed, cf);
326 }
327
328 cf_read_status_t
329 cf_read(capture_file *cf)
330 {
331   int         err;
332   gchar       *err_info;
333   const gchar *name_ptr;
334   const char  *errmsg;
335   char         errmsg_errno[1024+1];
336   gchar        err_str[2048+1];
337   long         data_offset;
338   progdlg_t   *progbar = NULL;
339   gboolean     stop_flag;
340   /*
341    * XXX - should be "off_t", but Wiretap would need more work to handle
342    * the full size of "off_t" on platforms where it's more than a "long"
343    * as well.
344    */
345   long         file_pos;
346   float        prog_val;
347   int          fd;
348   struct stat  cf_stat;
349   GTimeVal     start_time;
350   gchar        status_str[100];
351   int          progbar_nextstep;
352   int          progbar_quantum;
353
354   cum_bytes=0;
355
356   reset_tap_listeners();
357   tap_dfilter_dlg_update();
358
359   cf_callback_invoke(cf_cb_file_read_start, cf);
360
361   name_ptr = get_basename(cf->filename);
362
363   /* Update the progress bar when it gets to this value. */
364   progbar_nextstep = 0;
365   /* When we reach the value that triggers a progress bar update,
366      bump that value by this amount. */
367   progbar_quantum = cf->f_len/N_PROGBAR_UPDATES;
368
369   packet_list_freeze();
370
371   stop_flag = FALSE;
372   g_get_current_time(&start_time);
373
374   while ((wtap_read(cf->wth, &err, &err_info, &data_offset))) {
375     /* Update the progress bar, but do it only N_PROGBAR_UPDATES times;
376        when we update it, we have to run the GTK+ main loop to get it
377        to repaint what's pending, and doing so may involve an "ioctl()"
378        to see if there's any pending input from an X server, and doing
379        that for every packet can be costly, especially on a big file. */
380     if (data_offset >= progbar_nextstep) {
381         file_pos = lseek(cf->filed, 0, SEEK_CUR);
382         prog_val = (gfloat) file_pos / (gfloat) cf->f_len;
383         if (prog_val > 1.0) {
384           /* The file probably grew while we were reading it.
385              Update "cf->f_len", and try again. */
386           fd = wtap_fd(cf->wth);
387           if (fstat(fd, &cf_stat) >= 0) {
388             cf->f_len = cf_stat.st_size;
389             prog_val = (gfloat) file_pos / (gfloat) cf->f_len;
390           }
391           /* If it's still > 1, either the "fstat()" failed (in which
392              case there's not much we can do about it), or the file
393              *shrank* (in which case there's not much we can do about
394              it); just clip the progress value at 1.0. */
395           if (prog_val > 1.0)
396             prog_val = 1.0;
397         }
398         if (progbar == NULL) {
399           /* Create the progress bar if necessary */
400           progbar = delayed_create_progress_dlg("Loading", name_ptr,
401             &stop_flag, &start_time, prog_val);
402         }
403         if (progbar != NULL) {
404           g_snprintf(status_str, sizeof(status_str),
405                      "%luKB of %luKB", file_pos / 1024, cf->f_len / 1024);
406           update_progress_dlg(progbar, prog_val, status_str);
407         }
408         progbar_nextstep += progbar_quantum;
409     }
410
411     if (stop_flag) {
412       /* Well, the user decided to abort the read.  Destroy the progress
413          bar, close the capture file, and return CF_READ_ABORTED so our caller
414          can do whatever is appropriate when that happens. */
415       destroy_progress_dlg(progbar);
416       cf->state = FILE_READ_ABORTED;    /* so that we're allowed to close it */
417       packet_list_thaw();               /* undo our freeze */
418       cf_close(cf);
419       return CF_READ_ABORTED;
420     }
421     read_packet(cf, data_offset);
422   }
423
424   /* We're done reading the file; destroy the progress bar if it was created. */
425   if (progbar != NULL)
426     destroy_progress_dlg(progbar);
427
428   /* We're done reading sequentially through the file. */
429   cf->state = FILE_READ_DONE;
430
431   /* Close the sequential I/O side, to free up memory it requires. */
432   wtap_sequential_close(cf->wth);
433
434   /* Allow the protocol dissectors to free up memory that they
435    * don't need after the sequential run-through of the packets. */
436   postseq_cleanup_all_protocols();
437
438   /* Set the file encapsulation type now; we don't know what it is until
439      we've looked at all the packets, as we don't know until then whether
440      there's more than one type (and thus whether it's
441      WTAP_ENCAP_PER_PACKET). */
442   cf->lnk_t = wtap_file_encap(cf->wth);
443
444   cf->current_frame = cf->first_displayed;
445   packet_list_thaw();
446
447   cf_callback_invoke(cf_cb_file_read_finished, cf);
448
449   /* If we have any displayed packets to select, select the first of those
450      packets by making the first row the selected row. */
451   if (cf->first_displayed != NULL)
452     packet_list_select_row(0);
453
454   if (err != 0) {
455     /* Put up a message box noting that the read failed somewhere along
456        the line.  Don't throw out the stuff we managed to read, though,
457        if any. */
458     switch (err) {
459
460     case WTAP_ERR_UNSUPPORTED_ENCAP:
461       snprintf(errmsg_errno, sizeof(errmsg_errno),
462                "The capture file has a packet with a network type that Ethereal doesn't support.\n(%s)",
463                err_info);
464       g_free(err_info);
465       errmsg = errmsg_errno;
466       break;
467
468     case WTAP_ERR_CANT_READ:
469       errmsg = "An attempt to read from the capture file failed for"
470                " some unknown reason.";
471       break;
472
473     case WTAP_ERR_SHORT_READ:
474       errmsg = "The capture file appears to have been cut short"
475                " in the middle of a packet.";
476       break;
477
478     case WTAP_ERR_BAD_RECORD:
479       snprintf(errmsg_errno, sizeof(errmsg_errno),
480                "The capture file appears to be damaged or corrupt.\n(%s)",
481                err_info);
482       g_free(err_info);
483       errmsg = errmsg_errno;
484       break;
485
486     default:
487       snprintf(errmsg_errno, sizeof(errmsg_errno),
488                "An error occurred while reading the"
489                " capture file: %s.", wtap_strerror(err));
490       errmsg = errmsg_errno;
491       break;
492     }
493     snprintf(err_str, sizeof err_str, errmsg);
494     simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK, err_str);
495     return CF_READ_ERROR;
496   } else
497     return CF_READ_OK;
498 }
499
500 #ifdef HAVE_LIBPCAP
501 cf_status_t
502 cf_start_tail(capture_file *cf, const char *fname, gboolean is_tempfile, int *err)
503 {
504   cf_status_t cf_status;
505
506   cf_status = cf_open(cf, fname, is_tempfile, err);
507   return cf_status;
508 }
509
510 cf_read_status_t
511 cf_continue_tail(capture_file *cf, int to_read, int *err)
512 {
513   long data_offset = 0;
514   gchar *err_info;
515
516   *err = 0;
517
518   packet_list_freeze();
519
520   while (to_read != 0 && (wtap_read(cf->wth, err, &err_info, &data_offset))) {
521     if (cf->state == FILE_READ_ABORTED) {
522       /* Well, the user decided to exit Ethereal.  Break out of the
523          loop, and let the code below (which is called even if there
524          aren't any packets left to read) exit. */
525       break;
526     }
527     read_packet(cf, data_offset);
528     to_read--;
529   }
530
531   packet_list_thaw();
532
533   /* XXX - this cheats and looks inside the packet list to find the final
534      row number. */
535   if (auto_scroll_live && cf->plist_end != NULL)
536     packet_list_moveto_end();
537
538   if (cf->state == FILE_READ_ABORTED) {
539     /* Well, the user decided to exit Ethereal.  Return CF_READ_ABORTED
540        so that our caller can kill off the capture child process;
541        this will cause an EOF on the pipe from the child, so
542        "cf_finish_tail()" will be called, and it will clean up
543        and exit. */
544     return CF_READ_ABORTED;
545   } else if (*err != 0) {
546     /* We got an error reading the capture file.
547        XXX - pop up a dialog box? */
548     return CF_READ_ERROR;
549   } else
550     return CF_READ_OK;
551 }
552
553 cf_read_status_t
554 cf_finish_tail(capture_file *cf, int *err)
555 {
556   gchar *err_info;
557   long data_offset;
558   int         fd;
559   struct stat cf_stat;
560
561
562   if(cf->wth == NULL) {
563     cf_close(cf);
564     return CF_READ_ERROR;
565   }
566
567   packet_list_freeze();
568
569   while ((wtap_read(cf->wth, err, &err_info, &data_offset))) {
570     if (cf->state == FILE_READ_ABORTED) {
571       /* Well, the user decided to abort the read.  Break out of the
572          loop, and let the code below (which is called even if there
573          aren't any packets left to read) exit. */
574       break;
575     }
576     read_packet(cf, data_offset);
577   }
578
579   if (cf->state == FILE_READ_ABORTED) {
580     /* Well, the user decided to abort the read.  We're only called
581        when the child capture process closes the pipe to us (meaning
582        it's probably exited), so we can just close the capture
583        file; we return CF_READ_ABORTED so our caller can do whatever
584        is appropriate when that happens. */
585     cf_close(cf);
586     return CF_READ_ABORTED;
587   }
588
589   packet_list_thaw();
590   if (auto_scroll_live && cf->plist_end != NULL)
591     /* XXX - this cheats and looks inside the packet list to find the final
592        row number. */
593     packet_list_moveto_end();
594
595   /* We're done reading sequentially through the file. */
596   cf->state = FILE_READ_DONE;
597
598   /* we have to update the f_len field */
599   /* Find the size of the file. */
600   fd = wtap_fd(cf->wth);
601   if (fstat(fd, &cf_stat) >= 0) {
602       cf->f_len = cf_stat.st_size;
603   }
604
605   /* We're done reading sequentially through the file; close the
606      sequential I/O side, to free up memory it requires. */
607   wtap_sequential_close(cf->wth);
608
609   /* Allow the protocol dissectors to free up memory that they
610    * don't need after the sequential run-through of the packets. */
611   postseq_cleanup_all_protocols();
612
613   /* Set the file encapsulation type now; we don't know what it is until
614      we've looked at all the packets, as we don't know until then whether
615      there's more than one type (and thus whether it's
616      WTAP_ENCAP_PER_PACKET). */
617   cf->lnk_t = wtap_file_encap(cf->wth);
618
619   if (*err != 0) {
620     /* We got an error reading the capture file.
621        XXX - pop up a dialog box? */
622     return CF_READ_ERROR;
623   } else {
624     return CF_READ_OK;
625   }
626 }
627 #endif /* HAVE_LIBPCAP */
628
629 const gchar *
630 cf_get_display_name(capture_file *cf)
631 {
632   const gchar *displayname;
633
634   /* Return a name to use in displays */
635   if (!cf->is_tempfile) {
636     /* Get the last component of the file name, and use that. */
637     if (cf->filename){
638       displayname = get_basename(cf->filename);
639     } else {
640       displayname="(No file)";
641     }
642   } else {
643     /* The file we read is a temporary file from a live capture;
644        we don't mention its name. */
645     displayname = "(Untitled)";
646   }
647   return displayname;
648 }
649
650 /* XXX - use a macro instead? */
651 int
652 cf_packet_count(capture_file *cf)
653 {
654     return cf->count;
655 }
656
657 /* XXX - use a macro instead? */
658 gboolean
659 cf_is_tempfile(capture_file *cf)
660 {
661     return cf->is_tempfile;
662 }
663
664 void cf_set_tempfile(capture_file *cf, gboolean is_tempfile)
665 {
666     cf->is_tempfile = is_tempfile;
667 }
668
669
670 /* XXX - use a macro instead? */
671 void cf_set_drops_known(capture_file *cf, gboolean drops_known)
672 {
673     cf->drops_known = drops_known;
674 }
675
676 /* XXX - use a macro instead? */
677 void cf_set_drops(capture_file *cf, guint32 drops)
678 {
679     cf->drops = drops;
680 }
681
682 /* XXX - use a macro instead? */
683 gboolean cf_get_drops_known(capture_file *cf)
684 {
685     return cf->drops_known;
686 }
687
688 /* XXX - use a macro instead? */
689 guint32 cf_get_drops(capture_file *cf)
690 {
691     return cf->drops;
692 }
693
694 void cf_set_rfcode(capture_file *cf, dfilter_t *rfcode)
695 {
696     cf->rfcode = rfcode;
697 }
698
699 static int
700 add_packet_to_packet_list(frame_data *fdata, capture_file *cf,
701         union wtap_pseudo_header *pseudo_header, const guchar *buf,
702         gboolean refilter)
703 {
704   gint          row;
705   gboolean      create_proto_tree = FALSE;
706   epan_dissect_t *edt;
707
708   /* just add some value here until we know if it is being displayed or not */
709   fdata->cum_bytes  = cum_bytes + fdata->pkt_len;
710
711   /* If we don't have the time stamp of the first packet in the
712      capture, it's because this is the first packet.  Save the time
713      stamp of this packet as the time stamp of the first packet. */
714   if (!firstsec && !firstusec) {
715     firstsec  = fdata->abs_secs;
716     firstusec = fdata->abs_usecs;
717   }
718   /* if this frames is marked as a reference time frame, reset
719      firstsec and firstusec to this frame */
720   if(fdata->flags.ref_time){
721     firstsec  = fdata->abs_secs;
722     firstusec = fdata->abs_usecs;
723   }
724
725   /* If we don't have the time stamp of the previous displayed packet,
726      it's because this is the first displayed packet.  Save the time
727      stamp of this packet as the time stamp of the previous displayed
728      packet. */
729   if (!prevsec && !prevusec) {
730     prevsec  = fdata->abs_secs;
731     prevusec = fdata->abs_usecs;
732   }
733
734   /* Get the time elapsed between the first packet and this packet. */
735   compute_timestamp_diff(&fdata->rel_secs, &fdata->rel_usecs,
736      fdata->abs_secs, fdata->abs_usecs, firstsec, firstusec);
737
738   /* If it's greater than the current elapsed time, set the elapsed time
739      to it (we check for "greater than" so as not to be confused by
740      time moving backwards). */
741   if ((gint32)cf->esec < fdata->rel_secs
742   || ((gint32)cf->esec == fdata->rel_secs && (gint32)cf->eusec < fdata->rel_usecs)) {
743     cf->esec = fdata->rel_secs;
744     cf->eusec = fdata->rel_usecs;
745   }
746
747   /* Get the time elapsed between the previous displayed packet and
748      this packet. */
749   compute_timestamp_diff(&fdata->del_secs, &fdata->del_usecs,
750         fdata->abs_secs, fdata->abs_usecs, prevsec, prevusec);
751
752   /* If either
753
754         we have a display filter and are re-applying it;
755
756         we have a list of color filters;
757
758         we have tap listeners;
759
760      allocate a protocol tree root node, so that we'll construct
761      a protocol tree against which a filter expression can be
762      evaluated. */
763   if ((cf->dfcode != NULL && refilter) || color_filters_used()
764         || num_tap_filters != 0)
765           create_proto_tree = TRUE;
766
767   /* Dissect the frame. */
768   edt = epan_dissect_new(create_proto_tree, FALSE);
769
770   if (cf->dfcode != NULL && refilter) {
771       epan_dissect_prime_dfilter(edt, cf->dfcode);
772   }
773   if (color_filters_used()) {
774       color_filters_prime_edt(edt);
775   }
776   tap_queue_init(edt);
777   epan_dissect_run(edt, pseudo_header, buf, fdata, &cf->cinfo);
778   tap_push_tapped_queue(edt);
779
780   /* If we have a display filter, apply it if we're refiltering, otherwise
781      leave the "passed_dfilter" flag alone.
782
783      If we don't have a display filter, set "passed_dfilter" to 1. */
784   if (cf->dfcode != NULL) {
785     if (refilter) {
786       fdata->flags.passed_dfilter = dfilter_apply_edt(cf->dfcode, edt) ? 1 : 0;
787     }
788   } else
789     fdata->flags.passed_dfilter = 1;
790
791   if( (fdata->flags.passed_dfilter) 
792    || (edt->pi.fd->flags.ref_time) ){
793     /* This frame either passed the display filter list or is marked as
794        a time reference frame.  All time reference frames are displayed
795        even if they dont pass the display filter */
796     /* if this was a TIME REF frame we should reset the cul bytes field */
797     if(edt->pi.fd->flags.ref_time){
798       cum_bytes = fdata->pkt_len;
799       fdata->cum_bytes  = cum_bytes;
800     }
801
802     /* increase cum_bytes with this packets length */
803     cum_bytes += fdata->pkt_len;
804
805     epan_dissect_fill_in_columns(edt);
806
807     /* If we haven't yet seen the first frame, this is it.
808
809        XXX - we must do this before we add the row to the display,
810        as, if the display's GtkCList's selection mode is
811        GTK_SELECTION_BROWSE, when the first entry is added to it,
812        "cf_select_packet()" will be called, and it will fetch the row
813        data for the 0th row, and will get a null pointer rather than
814        "fdata", as "gtk_clist_append()" won't yet have returned and
815        thus "gtk_clist_set_row_data()" won't yet have been called.
816
817        We thus need to leave behind bread crumbs so that
818        "cf_select_packet()" can find this frame.  See the comment
819        in "cf_select_packet()". */
820     if (cf->first_displayed == NULL)
821       cf->first_displayed = fdata;
822
823     /* This is the last frame we've seen so far. */
824     cf->last_displayed = fdata;
825
826     row = packet_list_append(cf->cinfo.col_data, fdata);
827
828     /* colorize packet: if packet is marked, use preferences, 
829        otherwise try to apply color filters */
830       if (fdata->flags.marked) {
831           fdata->color_filter = NULL;
832           packet_list_set_colors(row, &prefs.gui_marked_fg, &prefs.gui_marked_bg);
833       } else {
834           fdata->color_filter = color_filters_colorize_packet(row, edt);
835       }
836
837     /* Set the time of the previous displayed frame to the time of this
838        frame. */
839     prevsec = fdata->abs_secs;
840     prevusec = fdata->abs_usecs;
841
842     cf->displayed_count++;
843   } else {
844     /* This frame didn't pass the display filter, so it's not being added
845        to the clist, and thus has no row. */
846     row = -1;
847   }
848   epan_dissect_free(edt);
849   return row;
850 }
851
852 static void
853 read_packet(capture_file *cf, long offset)
854 {
855   const struct wtap_pkthdr *phdr = wtap_phdr(cf->wth);
856   union wtap_pseudo_header *pseudo_header = wtap_pseudoheader(cf->wth);
857   const guchar *buf = wtap_buf_ptr(cf->wth);
858   frame_data   *fdata;
859   int           passed;
860   frame_data   *plist_end;
861   epan_dissect_t *edt;
862
863   /* Allocate the next list entry, and add it to the list. */
864   fdata = g_mem_chunk_alloc(cf->plist_chunk);
865
866   fdata->next = NULL;
867   fdata->prev = NULL;
868   fdata->pfd  = NULL;
869   fdata->pkt_len  = phdr->len;
870   fdata->cap_len  = phdr->caplen;
871   fdata->file_off = offset;
872   fdata->lnk_t = phdr->pkt_encap;
873   fdata->abs_secs  = phdr->ts.tv_sec;
874   fdata->abs_usecs = phdr->ts.tv_usec;
875   fdata->flags.encoding = CHAR_ASCII;
876   fdata->flags.visited = 0;
877   fdata->flags.marked = 0;
878   fdata->flags.ref_time = 0;
879
880   passed = TRUE;
881   if (cf->rfcode) {
882     edt = epan_dissect_new(TRUE, FALSE);
883     epan_dissect_prime_dfilter(edt, cf->rfcode);
884     epan_dissect_run(edt, pseudo_header, buf, fdata, NULL);
885     passed = dfilter_apply_edt(cf->rfcode, edt);
886     epan_dissect_free(edt);
887   }
888   if (passed) {
889     plist_end = cf->plist_end;
890     fdata->prev = plist_end;
891     if (plist_end != NULL)
892       plist_end->next = fdata;
893     else
894       cf->plist = fdata;
895     cf->plist_end = fdata;
896
897     cf->count++;
898     cf->f_len = offset + phdr->caplen;
899     fdata->num = cf->count;
900     add_packet_to_packet_list(fdata, cf, pseudo_header, buf, TRUE);
901   } else {
902     /* XXX - if we didn't have read filters, or if we could avoid
903        allocating the "frame_data" structure until we knew whether
904        the frame passed the read filter, we could use a G_ALLOC_ONLY
905        memory chunk...
906
907        ...but, at least in one test I did, where I just made the chunk
908        a G_ALLOC_ONLY chunk and read in a huge capture file, it didn't
909        seem to save a noticeable amount of time or space. */
910     g_mem_chunk_free(cf->plist_chunk, fdata);
911   }
912 }
913
914 cf_status_t
915 cf_merge_files(char **out_filenamep, int in_file_count,
916                char *const *in_filenames, int file_type, gboolean do_append)
917 {
918   merge_in_file_t  *in_files;
919   wtap             *wth;
920   char             *out_filename;
921   char              tmpname[128+1];
922   int               out_fd;
923   wtap_dumper      *pdh;
924   int               open_err, read_err, write_err, close_err;
925   gchar            *err_info;
926   int               err_fileno; 
927   int               i;
928   char              errmsg_errno[1024+1];
929   gchar             err_str[2048+1];
930   const char       *errmsg;
931   gboolean          got_read_error = FALSE, got_write_error = FALSE;
932   long              data_offset;
933   progdlg_t        *progbar = NULL;
934   gboolean          stop_flag;
935   /*
936    * XXX - should be "off_t", but Wiretap would need more work to handle
937    * the full size of "off_t" on platforms where it's more than a "long"
938    * as well.
939    */
940   long        f_len, file_pos;
941   float       prog_val;
942   GTimeVal    start_time;
943   gchar       status_str[100];
944   int         progbar_nextstep;
945   int         progbar_quantum;
946
947   /* open the input files */
948   if (!merge_open_in_files(in_file_count, in_filenames, &in_files,
949                            &open_err, &err_info, &err_fileno)) {
950     free(in_files);
951     cf_open_failure_alert_box(in_filenames[err_fileno], open_err, err_info,
952                               FALSE, 0);
953     return CF_ERROR;
954   }
955
956   if (*out_filenamep != NULL) {
957     out_filename = *out_filenamep;
958     out_fd = open(out_filename, O_CREAT|O_TRUNC|O_BINARY, 0600);
959     if (out_fd == -1)
960       open_err = errno;
961   } else {
962     out_fd = create_tempfile(tmpname, sizeof tmpname, "ether");
963     if (out_fd == -1)
964       open_err = errno;
965     out_filename = g_strdup(tmpname);
966     *out_filenamep = out_filename;
967   }
968   if (out_fd == -1) {
969     err_info = NULL;
970     merge_close_in_files(in_file_count, in_files);
971     free(in_files);
972     cf_open_failure_alert_box(out_filename, open_err, NULL, TRUE, file_type);
973     return CF_ERROR;
974   }
975
976   pdh = wtap_dump_fdopen(out_fd, file_type,
977       merge_select_frame_type(in_file_count, in_files),
978       merge_max_snapshot_length(in_file_count, in_files), &open_err);
979   if (pdh == NULL) {
980     close(out_fd);
981     merge_close_in_files(in_file_count, in_files);
982     free(in_files);
983     cf_open_failure_alert_box(out_filename, open_err, err_info, TRUE,
984                               file_type);
985     return CF_ERROR;
986   }
987
988   /* Get the sum of the sizes of all the files. */
989   f_len = 0;
990   for (i = 0; i < in_file_count; i++)
991     f_len += in_files[i].size;
992
993   /* Update the progress bar when it gets to this value. */
994   progbar_nextstep = 0;
995   /* When we reach the value that triggers a progress bar update,
996      bump that value by this amount. */
997   progbar_quantum = f_len/N_PROGBAR_UPDATES;
998
999   stop_flag = FALSE;
1000   g_get_current_time(&start_time);
1001
1002   /* do the merge (or append) */
1003   for (;;) {
1004     if (do_append)
1005       wth = merge_append_read_packet(in_file_count, in_files, &read_err,
1006                                      &err_info);
1007     else
1008       wth = merge_read_packet(in_file_count, in_files, &read_err,
1009                               &err_info);
1010     if (wth == NULL) {
1011       if (read_err != 0)
1012         got_read_error = TRUE;
1013       break;
1014     }
1015
1016     /* Get the sum of the data offsets in all of the files. */
1017     data_offset = 0;
1018     for (i = 0; i < in_file_count; i++)
1019       data_offset += in_files[i].data_offset;
1020
1021     if (data_offset >= progbar_nextstep) {
1022         /* Get the sum of the seek positions in all of the files. */
1023         file_pos = 0;
1024         for (i = 0; i < in_file_count; i++)
1025           file_pos += lseek(wtap_fd(in_files[i].wth), 0, SEEK_CUR);
1026         prog_val = (gfloat) file_pos / (gfloat) f_len;
1027         if (prog_val > 1.0) {
1028           /* Some file probably grew while we were reading it.
1029              That "shouldn't happen", so we'll just clip the progress
1030              value at 1.0. */
1031           prog_val = 1.0;
1032         }
1033         if (progbar == NULL) {
1034           /* Create the progress bar if necessary */
1035           progbar = delayed_create_progress_dlg("Merging", "files",
1036             &stop_flag, &start_time, prog_val);
1037         }
1038         if (progbar != NULL) {
1039           g_snprintf(status_str, sizeof(status_str),
1040                      "%luKB of %luKB", file_pos / 1024, f_len / 1024);
1041           update_progress_dlg(progbar, prog_val, status_str);
1042         }
1043         progbar_nextstep += progbar_quantum;
1044     }
1045
1046     if (!wtap_dump(pdh, wtap_phdr(wth), wtap_pseudoheader(wth),
1047          wtap_buf_ptr(wth), &write_err)) {
1048       got_write_error = TRUE;
1049       break;
1050     }
1051   }
1052
1053   /* We're done merging the files; destroy the progress bar if it was created. */
1054   if (progbar != NULL)
1055     destroy_progress_dlg(progbar);
1056
1057   merge_close_in_files(in_file_count, in_files);
1058   if (!got_read_error && !got_write_error) {
1059     if (!wtap_dump_close(pdh, &write_err))
1060       got_write_error = TRUE;
1061   } else
1062     wtap_dump_close(pdh, &close_err);
1063
1064   if (got_read_error) {
1065     /*
1066      * Find the file on which we got the error, and report the error.
1067      */
1068     for (i = 0; i < in_file_count; i++) {
1069       if (in_files[i].state == GOT_ERROR) {
1070         /* Put up a message box noting that a read failed somewhere along
1071            the line. */
1072         switch (read_err) {
1073
1074         case WTAP_ERR_UNSUPPORTED_ENCAP:
1075           snprintf(errmsg_errno, sizeof(errmsg_errno),
1076                    "The capture file %%s has a packet with a network type that Ethereal doesn't support.\n(%s)",
1077                    err_info);
1078           g_free(err_info);
1079           errmsg = errmsg_errno;
1080           break;
1081
1082         case WTAP_ERR_CANT_READ:
1083           errmsg = "An attempt to read from the capture file %s failed for"
1084                    " some unknown reason.";
1085           break;
1086
1087         case WTAP_ERR_SHORT_READ:
1088           errmsg = "The capture file %s appears to have been cut short"
1089                    " in the middle of a packet.";
1090           break;
1091
1092         case WTAP_ERR_BAD_RECORD:
1093           snprintf(errmsg_errno, sizeof(errmsg_errno),
1094                    "The capture file %%s appears to be damaged or corrupt.\n(%s)",
1095                    err_info);
1096           g_free(err_info);
1097           errmsg = errmsg_errno;
1098           break;
1099
1100         default:
1101           snprintf(errmsg_errno, sizeof(errmsg_errno),
1102                    "An error occurred while reading the"
1103                    " capture file %%s: %s.", wtap_strerror(read_err));
1104           errmsg = errmsg_errno;
1105           break;
1106         }
1107         snprintf(err_str, sizeof err_str, errmsg, in_files[i].filename);
1108         simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK, err_str);
1109       }
1110     }
1111   }
1112
1113   if (got_write_error) {
1114     /* Put up an alert box for the write error. */
1115     cf_write_failure_alert_box(out_filename, write_err);
1116   }
1117
1118   return (!got_read_error && !got_write_error) ? CF_OK : CF_ERROR;
1119 }
1120
1121 cf_status_t
1122 cf_filter_packets(capture_file *cf, gchar *dftext, gboolean force)
1123 {
1124   dfilter_t  *dfcode;
1125   const char *filter_new = dftext ? dftext : "";
1126   const char *filter_old = cf->dfilter ? cf->dfilter : "";
1127
1128   /* if new filter equals old one, do nothing unless told to do so */
1129   if (!force && strcmp(filter_new, filter_old) == 0) {
1130     return CF_OK;
1131   }
1132
1133   if (dftext == NULL) {
1134     /* The new filter is an empty filter (i.e., display all packets). */
1135     dfcode = NULL;
1136   } else {
1137     /*
1138      * We have a filter; make a copy of it (as we'll be saving it),
1139      * and try to compile it.
1140      */
1141     dftext = g_strdup(dftext);
1142     if (!dfilter_compile(dftext, &dfcode)) {
1143       /* The attempt failed; report an error. */
1144       gchar *safe_dftext = simple_dialog_format_message(dftext);
1145       gchar *safe_dfilter_error_msg = simple_dialog_format_message(
1146           dfilter_error_msg);
1147       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK, 
1148           "%s%s%s\n"
1149           "\n"
1150           "The following display filter isn't a valid display filter:\n%s\n"
1151           "See the help for a description of the display filter syntax.",
1152           simple_dialog_primary_start(), safe_dfilter_error_msg,
1153           simple_dialog_primary_end(), safe_dftext);
1154       g_free(safe_dfilter_error_msg);
1155       g_free(safe_dftext);
1156       g_free(dftext);
1157       return CF_ERROR;
1158     }
1159
1160     /* Was it empty? */
1161     if (dfcode == NULL) {
1162       /* Yes - free the filter text, and set it to null. */
1163       g_free(dftext);
1164       dftext = NULL;
1165     }
1166   }
1167
1168   /* We have a valid filter.  Replace the current filter. */
1169   if (cf->dfilter != NULL)
1170     g_free(cf->dfilter);
1171   cf->dfilter = dftext;
1172   if (cf->dfcode != NULL)
1173     dfilter_free(cf->dfcode);
1174   cf->dfcode = dfcode;
1175
1176   /* Now rescan the packet list, applying the new filter, but not
1177      throwing away information constructed on a previous pass. */
1178   if (dftext == NULL) {
1179     rescan_packets(cf, "Resetting", "Filter", TRUE, FALSE);
1180   } else {
1181     rescan_packets(cf, "Filtering", dftext, TRUE, FALSE);
1182   }
1183   return CF_OK;
1184 }
1185
1186 void
1187 cf_colorize_packets(capture_file *cf)
1188 {
1189   rescan_packets(cf, "Colorizing", "all packets", FALSE, FALSE);
1190 }
1191
1192 void
1193 cf_reftime_packets(capture_file *cf)
1194 {
1195   rescan_packets(cf, "Updating Reftime", "all packets", FALSE, FALSE);
1196 }
1197
1198 void
1199 cf_redissect_packets(capture_file *cf)
1200 {
1201   rescan_packets(cf, "Reprocessing", "all packets", TRUE, TRUE);
1202 }
1203
1204 /* Rescan the list of packets, reconstructing the CList.
1205
1206    "action" describes why we're doing this; it's used in the progress
1207    dialog box.
1208
1209    "action_item" describes what we're doing; it's used in the progress
1210    dialog box.
1211
1212    "refilter" is TRUE if we need to re-evaluate the filter expression.
1213
1214    "redissect" is TRUE if we need to make the dissectors reconstruct
1215    any state information they have (because a preference that affects
1216    some dissector has changed, meaning some dissector might construct
1217    its state differently from the way it was constructed the last time). */
1218 static void
1219 rescan_packets(capture_file *cf, const char *action, const char *action_item,
1220                 gboolean refilter, gboolean redissect)
1221 {
1222   frame_data *fdata;
1223   progdlg_t  *progbar = NULL;
1224   gboolean    stop_flag;
1225   int         count;
1226   int         err;
1227   gchar      *err_info;
1228   frame_data *selected_frame, *preceding_frame, *following_frame, *prev_frame;
1229   int         selected_row, prev_row, preceding_row, following_row;
1230   gboolean    selected_frame_seen;
1231   int         row;
1232   float       prog_val;
1233   GTimeVal    start_time;
1234   gchar       status_str[100];
1235   int         progbar_nextstep;
1236   int         progbar_quantum;
1237
1238   cum_bytes=0;
1239   reset_tap_listeners();
1240   /* Which frame, if any, is the currently selected frame?
1241      XXX - should the selected frame or the focus frame be the "current"
1242      frame, that frame being the one from which "Find Frame" searches
1243      start? */
1244   selected_frame = cf->current_frame;
1245
1246   /* We don't yet know what row that frame will be on, if any, after we
1247      rebuild the clist, however. */
1248   selected_row = -1;
1249
1250   if (redissect) {
1251     /* We need to re-initialize all the state information that protocols
1252        keep, because some preference that controls a dissector has changed,
1253        which might cause the state information to be constructed differently
1254        by that dissector. */
1255
1256     /* Initialize all data structures used for dissection. */
1257     init_dissection();
1258   }
1259
1260   /* Freeze the packet list while we redo it, so we don't get any
1261      screen updates while it happens. */
1262   packet_list_freeze();
1263
1264   /* Clear it out. */
1265   packet_list_clear();
1266
1267   /* We don't yet know which will be the first and last frames displayed. */
1268   cf->first_displayed = NULL;
1269   cf->last_displayed = NULL;
1270
1271   /* We currently don't display any packets */
1272   cf->displayed_count = 0;
1273
1274   /* Iterate through the list of frames.  Call a routine for each frame
1275      to check whether it should be displayed and, if so, add it to
1276      the display list. */
1277   firstsec = 0;
1278   firstusec = 0;
1279   prevsec = 0;
1280   prevusec = 0;
1281
1282   /* Update the progress bar when it gets to this value. */
1283   progbar_nextstep = 0;
1284   /* When we reach the value that triggers a progress bar update,
1285      bump that value by this amount. */
1286   progbar_quantum = cf->count/N_PROGBAR_UPDATES;
1287   /* Count of packets at which we've looked. */
1288   count = 0;
1289
1290   stop_flag = FALSE;
1291   g_get_current_time(&start_time);
1292
1293   row = -1;             /* no previous row yet */
1294   prev_row = -1;
1295   prev_frame = NULL;
1296
1297   preceding_row = -1;
1298   preceding_frame = NULL;
1299   following_row = -1;
1300   following_frame = NULL;
1301
1302   selected_frame_seen = FALSE;
1303
1304   for (fdata = cf->plist; fdata != NULL; fdata = fdata->next) {
1305     /* Update the progress bar, but do it only N_PROGBAR_UPDATES times;
1306        when we update it, we have to run the GTK+ main loop to get it
1307        to repaint what's pending, and doing so may involve an "ioctl()"
1308        to see if there's any pending input from an X server, and doing
1309        that for every packet can be costly, especially on a big file. */
1310     if (count >= progbar_nextstep) {
1311       /* let's not divide by zero. I should never be started
1312        * with count == 0, so let's assert that
1313        */
1314       g_assert(cf->count > 0);
1315       prog_val = (gfloat) count / cf->count;
1316
1317       if (progbar == NULL)
1318         /* Create the progress bar if necessary */
1319         progbar = delayed_create_progress_dlg(action, action_item, &stop_flag,
1320           &start_time, prog_val);
1321
1322       if (progbar != NULL) {
1323         g_snprintf(status_str, sizeof(status_str),
1324                   "%4u of %u frames", count, cf->count);
1325         update_progress_dlg(progbar, prog_val, status_str);
1326       }
1327
1328       progbar_nextstep += progbar_quantum;
1329     }
1330
1331     if (stop_flag) {
1332       /* Well, the user decided to abort the filtering.  Just stop.
1333
1334          XXX - go back to the previous filter?  Users probably just
1335          want not to wait for a filtering operation to finish;
1336          unless we cancel by having no filter, reverting to the
1337          previous filter will probably be even more expensive than
1338          continuing the filtering, as it involves going back to the
1339          beginning and filtering, and even with no filter we currently
1340          have to re-generate the entire clist, which is also expensive.
1341
1342          I'm not sure what Network Monitor does, but it doesn't appear
1343          to give you an unfiltered display if you cancel. */
1344       break;
1345     }
1346
1347     count++;
1348
1349     if (redissect) {
1350       /* Since all state for the frame was destroyed, mark the frame
1351        * as not visited, free the GSList referring to the state
1352        * data (the per-frame data itself was freed by
1353        * "init_dissection()"), and null out the GSList pointer. */
1354       fdata->flags.visited = 0;
1355       if (fdata->pfd) {
1356         g_slist_free(fdata->pfd);
1357         fdata->pfd = NULL;
1358       }
1359     }
1360
1361     if (!wtap_seek_read (cf->wth, fdata->file_off, &cf->pseudo_header,
1362         cf->pd, fdata->cap_len, &err, &err_info)) {
1363         simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
1364                       cf_read_error_message(err, err_info), cf->filename);
1365         break;
1366     }
1367
1368     /* If the previous frame is displayed, and we haven't yet seen the
1369        selected frame, remember that frame - it's the closest one we've
1370        yet seen before the selected frame. */
1371     if (prev_row != -1 && !selected_frame_seen) {
1372       preceding_row = prev_row;
1373       preceding_frame = prev_frame;
1374     }
1375     row = add_packet_to_packet_list(fdata, cf, &cf->pseudo_header, cf->pd,
1376                                         refilter);
1377
1378     /* If this frame is displayed, and this is the first frame we've
1379        seen displayed after the selected frame, remember this frame -
1380        it's the closest one we've yet seen at or after the selected
1381        frame. */
1382     if (row != -1 && selected_frame_seen && following_row == -1) {
1383       following_row = row;
1384       following_frame = fdata;
1385     }
1386     if (fdata == selected_frame) {
1387       selected_row = row;
1388       selected_frame_seen = TRUE;
1389     }
1390
1391     /* Remember this row/frame - it'll be the previous row/frame
1392        on the next pass through the loop. */
1393     prev_row = row;
1394     prev_frame = fdata;
1395   }
1396
1397   if (redissect) {
1398     /* Clear out what remains of the visited flags and per-frame data
1399        pointers.
1400
1401        XXX - that may cause various forms of bogosity when dissecting
1402        these frames, as they won't have been seen by this sequential
1403        pass, but the only alternative I see is to keep scanning them
1404        even though the user requested that the scan stop, and that
1405        would leave the user stuck with an Ethereal grinding on
1406        until it finishes.  Should we just stick them with that? */
1407     for (; fdata != NULL; fdata = fdata->next) {
1408       fdata->flags.visited = 0;
1409       if (fdata->pfd) {
1410         g_slist_free(fdata->pfd);
1411         fdata->pfd = NULL;
1412       }
1413     }
1414   }
1415
1416   /* We're done filtering the packets; destroy the progress bar if it
1417      was created. */
1418   if (progbar != NULL)
1419     destroy_progress_dlg(progbar);
1420
1421   /* Unfreeze the packet list. */
1422   packet_list_thaw();
1423
1424   if (selected_row == -1) {
1425     /* The selected frame didn't pass the filter. */
1426     if (selected_frame == NULL) {
1427       /* That's because there *was* no selected frame.  Make the first
1428          displayed frame the current frame. */
1429       selected_row = 0;
1430     } else {
1431       /* Find the nearest displayed frame to the selected frame (whether
1432          it's before or after that frame) and make that the current frame.
1433          If the next and previous displayed frames are equidistant from the
1434          selected frame, choose the next one. */
1435       g_assert(following_frame == NULL ||
1436                following_frame->num >= selected_frame->num);
1437       g_assert(preceding_frame == NULL ||
1438                preceding_frame->num <= selected_frame->num);
1439       if (following_frame == NULL) {
1440         /* No frame after the selected frame passed the filter, so we
1441            have to select the last displayed frame before the selected
1442            frame. */
1443         selected_row = preceding_row;
1444       } else if (preceding_frame == NULL) {
1445         /* No frame before the selected frame passed the filter, so we
1446            have to select the first displayed frame after the selected
1447            frame. */
1448         selected_row = following_row;
1449       } else {
1450         /* Choose the closer of the last displayed frame before the
1451            selected frame and the first displayed frame after the
1452            selected frame; in case of a tie, choose the first displayed
1453            frame after the selected frame. */
1454         if (following_frame->num - selected_frame->num <=
1455             selected_frame->num - preceding_frame->num) {
1456           selected_row = following_row;
1457         } else {
1458           /* The previous frame is closer to the selected frame than the
1459              next frame. */
1460           selected_row = preceding_row;
1461         }
1462       }
1463     }
1464   }
1465
1466   if (selected_row == -1) {
1467     /* There are no frames displayed at all. */
1468     cf_unselect_packet(cf);
1469   } else {
1470     /* Either the frame that was selected passed the filter, or we've
1471        found the nearest displayed frame to that frame.  Select it, make
1472        it the focus row, and make it visible. */
1473     packet_list_set_selected_row(selected_row);
1474   }
1475 }
1476
1477 typedef enum {
1478   PSP_FINISHED,
1479   PSP_STOPPED,
1480   PSP_FAILED
1481 } psp_return_t;
1482
1483 psp_return_t
1484 process_specified_packets(capture_file *cf, packet_range_t *range,
1485     const char *string1, const char *string2,
1486     gboolean (*callback)(capture_file *, frame_data *,
1487                          union wtap_pseudo_header *, const guint8 *, void *),
1488     void *callback_args)
1489 {
1490   frame_data *fdata;
1491   int         err;
1492   gchar      *err_info;
1493   union wtap_pseudo_header pseudo_header;
1494   guint8      pd[WTAP_MAX_PACKET_SIZE+1];
1495   psp_return_t ret = PSP_FINISHED;
1496
1497   progdlg_t  *progbar = NULL;
1498   int         progbar_count;
1499   float       progbar_val;
1500   gboolean    progbar_stop_flag;
1501   GTimeVal    progbar_start_time;
1502   gchar       progbar_status_str[100];
1503   int         progbar_nextstep;
1504   int         progbar_quantum;
1505   range_process_e process_this;
1506
1507   /* Update the progress bar when it gets to this value. */
1508   progbar_nextstep = 0;
1509   /* When we reach the value that triggers a progress bar update,
1510      bump that value by this amount. */
1511   progbar_quantum = cf->count/N_PROGBAR_UPDATES;
1512   /* Count of packets at which we've looked. */
1513   progbar_count = 0;
1514
1515   progbar_stop_flag = FALSE;
1516   g_get_current_time(&progbar_start_time);
1517
1518   packet_range_process_init(range);
1519
1520   /* Iterate through the list of packets, printing the packets that
1521      were selected by the current display filter.  */
1522   for (fdata = cf->plist; fdata != NULL; fdata = fdata->next) {
1523     /* Update the progress bar, but do it only N_PROGBAR_UPDATES times;
1524        when we update it, we have to run the GTK+ main loop to get it
1525        to repaint what's pending, and doing so may involve an "ioctl()"
1526        to see if there's any pending input from an X server, and doing
1527        that for every packet can be costly, especially on a big file. */
1528     if (progbar_count >= progbar_nextstep) {
1529       /* let's not divide by zero. I should never be started
1530        * with count == 0, so let's assert that
1531        */
1532       g_assert(cf->count > 0);
1533       progbar_val = (gfloat) progbar_count / cf->count;
1534
1535       if (progbar == NULL)
1536         /* Create the progress bar if necessary */
1537         progbar = delayed_create_progress_dlg(string1, string2,
1538                                               &progbar_stop_flag,
1539                                               &progbar_start_time,
1540                                               progbar_val);
1541
1542       if (progbar != NULL) {
1543         g_snprintf(progbar_status_str, sizeof(progbar_status_str),
1544                    "%4u of %u packets", progbar_count, cf->count);
1545         update_progress_dlg(progbar, progbar_val, progbar_status_str);
1546       }
1547
1548       progbar_nextstep += progbar_quantum;
1549     }
1550
1551     if (progbar_stop_flag) {
1552       /* Well, the user decided to abort the operation.  Just stop,
1553          and arrange to return TRUE to our caller, so they know it
1554          was stopped explicitly. */
1555       ret = PSP_STOPPED;
1556       break;
1557     }
1558
1559     progbar_count++;
1560
1561     /* do we have to process this packet? */
1562     process_this = packet_range_process_packet(range, fdata);
1563     if (process_this == range_process_next) {
1564         /* this packet uninteresting, continue with next one */
1565         continue;
1566     } else if (process_this == range_processing_finished) {
1567         /* all interesting packets processed, stop the loop */
1568         break;
1569     }
1570
1571     /* Get the packet */
1572     if (!wtap_seek_read(cf->wth, fdata->file_off, &pseudo_header,
1573                         pd, fdata->cap_len, &err, &err_info)) {
1574       /* Attempt to get the packet failed. */
1575       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
1576                     cf_read_error_message(err, err_info), cf->filename);
1577       ret = PSP_FAILED;
1578       break;
1579     }
1580     /* Process the packet */
1581     if (!callback(cf, fdata, &pseudo_header, pd, callback_args)) {
1582       /* Callback failed.  We assume it reported the error appropriately. */
1583       ret = PSP_FAILED;
1584       break;
1585     }
1586   }
1587
1588   /* We're done printing the packets; destroy the progress bar if
1589      it was created. */
1590   if (progbar != NULL)
1591     destroy_progress_dlg(progbar);
1592
1593   return ret;
1594 }
1595
1596 static gboolean
1597 retap_packet(capture_file *cf _U_, frame_data *fdata,
1598              union wtap_pseudo_header *pseudo_header, const guint8 *pd,
1599              void *argsp _U_)
1600 {
1601   epan_dissect_t *edt;
1602
1603   /* If we have tap listeners, allocate a protocol tree root node, so that
1604      we'll construct a protocol tree against which a filter expression can
1605      be evaluated. */
1606   edt = epan_dissect_new(num_tap_filters != 0, FALSE);
1607   tap_queue_init(edt);
1608   epan_dissect_run(edt, pseudo_header, pd, fdata, NULL);
1609   tap_push_tapped_queue(edt);
1610   epan_dissect_free(edt);
1611
1612   return TRUE;
1613 }
1614
1615 cf_read_status_t
1616 cf_retap_packets(capture_file *cf)
1617 {
1618   packet_range_t range;
1619
1620   /* Reset the tap listeners. */
1621   reset_tap_listeners();
1622
1623   /* Iterate through the list of packets, dissecting all packets and
1624      re-running the taps. */
1625   packet_range_init(&range);
1626   packet_range_process_init(&range);
1627   switch (process_specified_packets(cf, &range, "Refiltering statistics on",
1628                                     "all packets", retap_packet,
1629                                     NULL)) {
1630   case PSP_FINISHED:
1631     /* Completed successfully. */
1632     return CF_OK;
1633
1634   case PSP_STOPPED:
1635     /* Well, the user decided to abort the refiltering.
1636        Return CF_READ_ABORTED so our caller knows they did that. */
1637     return CF_READ_ABORTED;
1638
1639   case PSP_FAILED:
1640     /* Error while retapping. */
1641     return CF_READ_ERROR;
1642   }
1643
1644   g_assert_not_reached();
1645   return CF_READ_OK;
1646 }
1647
1648 typedef struct {
1649   print_args_t *print_args;
1650   gboolean      print_header_line;
1651   char         *header_line_buf;
1652   int           header_line_buf_len;
1653   gboolean      print_formfeed;
1654   gboolean      print_separator;
1655   char         *line_buf;
1656   int           line_buf_len;
1657   gint         *col_widths;
1658 } print_callback_args_t;
1659
1660 static gboolean
1661 print_packet(capture_file *cf, frame_data *fdata,
1662              union wtap_pseudo_header *pseudo_header, const guint8 *pd,
1663              void *argsp)
1664 {
1665   print_callback_args_t *args = argsp;
1666   epan_dissect_t *edt;
1667   int             i;
1668   char           *cp;
1669   int             line_len;
1670   int             column_len;
1671   int             cp_off;
1672   gboolean        proto_tree_needed;
1673   char            bookmark_name[9+10+1];        /* "__frameNNNNNNNNNN__\0" */
1674   char            bookmark_title[6+10+1];       /* "Frame NNNNNNNNNN__\0" */
1675
1676   /* Create the protocol tree, and make it visible, if we're printing
1677      the dissection or the hex data.
1678      XXX - do we need it if we're just printing the hex data? */
1679   proto_tree_needed = 
1680       args->print_args->print_dissections != print_dissections_none || args->print_args->print_hex;
1681   edt = epan_dissect_new(proto_tree_needed, proto_tree_needed);
1682
1683   /* Fill in the column information if we're printing the summary
1684      information. */
1685   if (args->print_args->print_summary) {
1686     epan_dissect_run(edt, pseudo_header, pd, fdata, &cf->cinfo);
1687     epan_dissect_fill_in_columns(edt);
1688   } else
1689     epan_dissect_run(edt, pseudo_header, pd, fdata, NULL);
1690
1691   if (args->print_formfeed) {
1692     if (!new_page(args->print_args->stream))
1693       goto fail;
1694   } else {
1695       if (args->print_separator) {
1696         if (!print_line(args->print_args->stream, 0, ""))
1697           goto fail;
1698       }
1699   }
1700
1701   /*
1702    * We generate bookmarks, if the output format supports them.
1703    * The name is "__frameN__".
1704    */
1705   sprintf(bookmark_name, "__frame%u__", fdata->num);
1706
1707   if (args->print_args->print_summary) {
1708     if (args->print_header_line) {
1709       if (!print_line(args->print_args->stream, 0, args->header_line_buf))
1710         goto fail;
1711       args->print_header_line = FALSE;  /* we might not need to print any more */
1712     }
1713     cp = &args->line_buf[0];
1714     line_len = 0;
1715     for (i = 0; i < cf->cinfo.num_cols; i++) {
1716       /* Find the length of the string for this column. */
1717       column_len = strlen(cf->cinfo.col_data[i]);
1718       if (args->col_widths[i] > column_len)
1719          column_len = args->col_widths[i];
1720
1721       /* Make sure there's room in the line buffer for the column; if not,
1722          double its length. */
1723       line_len += column_len + 1;       /* "+1" for space */
1724       if (line_len > args->line_buf_len) {
1725         cp_off = cp - args->line_buf;
1726         args->line_buf_len = 2 * line_len;
1727         args->line_buf = g_realloc(args->line_buf, args->line_buf_len + 1);
1728         cp = args->line_buf + cp_off;
1729       }
1730
1731       /* Right-justify the packet number column. */
1732       if (cf->cinfo.col_fmt[i] == COL_NUMBER)
1733         sprintf(cp, "%*s", args->col_widths[i], cf->cinfo.col_data[i]);
1734       else
1735         sprintf(cp, "%-*s", args->col_widths[i], cf->cinfo.col_data[i]);
1736       cp += column_len;
1737       if (i != cf->cinfo.num_cols - 1)
1738         *cp++ = ' ';
1739     }
1740     *cp = '\0';
1741
1742     /*
1743      * Generate a bookmark, using the summary line as the title.
1744      */
1745     if (!print_bookmark(args->print_args->stream, bookmark_name,
1746                         args->line_buf))
1747       goto fail;
1748
1749     if (!print_line(args->print_args->stream, 0, args->line_buf))
1750       goto fail;
1751   } else {
1752     /*
1753      * Generate a bookmark, using "Frame N" as the title, as we're not
1754      * printing the summary line.
1755      */
1756     sprintf(bookmark_title, "Frame %u", fdata->num);
1757     if (!print_bookmark(args->print_args->stream, bookmark_name,
1758                         bookmark_title))
1759       goto fail;
1760   } /* if (print_summary) */
1761
1762   if (args->print_args->print_dissections != print_dissections_none) {
1763     if (args->print_args->print_summary) {
1764       /* Separate the summary line from the tree with a blank line. */
1765       if (!print_line(args->print_args->stream, 0, ""))
1766         goto fail;
1767     }
1768
1769     /* Print the information in that tree. */
1770     if (!proto_tree_print(args->print_args, edt, args->print_args->stream))
1771       goto fail;
1772
1773     /* Print a blank line if we print anything after this (aka more than one packet). */
1774     args->print_separator = TRUE;
1775
1776     /* Print a header line if we print any more packet summaries */
1777     args->print_header_line = TRUE;
1778   }
1779
1780   if (args->print_args->print_hex) {
1781     /* Print the full packet data as hex. */
1782     if (!print_hex_data(args->print_args->stream, edt))
1783       goto fail;
1784
1785     /* Print a blank line if we print anything after this (aka more than one packet). */
1786     args->print_separator = TRUE;
1787
1788     /* Print a header line if we print any more packet summaries */
1789     args->print_header_line = TRUE;
1790   } /* if (args->print_args->print_dissections != print_dissections_none) */
1791
1792   epan_dissect_free(edt);
1793
1794   /* do we want to have a formfeed between each packet from now on? */
1795   if(args->print_args->print_formfeed) {
1796     args->print_formfeed = TRUE;
1797   }
1798
1799   return TRUE;
1800
1801 fail:
1802   epan_dissect_free(edt);
1803   return FALSE;
1804 }
1805
1806 cf_print_status_t
1807 cf_print_packets(capture_file *cf, print_args_t *print_args)
1808 {
1809   int         i;
1810   print_callback_args_t callback_args;
1811   gint        data_width;
1812   char        *cp;
1813   int         cp_off;
1814   int         column_len;
1815   int         line_len;
1816   psp_return_t ret;
1817
1818   callback_args.print_args = print_args;
1819   callback_args.print_header_line = TRUE;
1820   callback_args.header_line_buf = NULL;
1821   callback_args.header_line_buf_len = 256;
1822   callback_args.print_formfeed = FALSE;
1823   callback_args.print_separator = FALSE;
1824   callback_args.line_buf = NULL;
1825   callback_args.line_buf_len = 256;
1826   callback_args.col_widths = NULL;
1827
1828   if (!print_preamble(print_args->stream, cf->filename)) {
1829     destroy_print_stream(print_args->stream);
1830     return CF_PRINT_WRITE_ERROR;
1831   }
1832
1833   if (print_args->print_summary) {
1834     /* We're printing packet summaries.  Allocate the header line buffer
1835        and get the column widths. */
1836     callback_args.header_line_buf = g_malloc(callback_args.header_line_buf_len + 1);
1837
1838     /* Find the widths for each of the columns - maximum of the
1839        width of the title and the width of the data - and construct
1840        a buffer with a line containing the column titles. */
1841     callback_args.col_widths = (gint *) g_malloc(sizeof(gint) * cf->cinfo.num_cols);
1842     cp = &callback_args.header_line_buf[0];
1843     line_len = 0;
1844     for (i = 0; i < cf->cinfo.num_cols; i++) {
1845       /* Don't pad the last column. */
1846       if (i == cf->cinfo.num_cols - 1)
1847         callback_args.col_widths[i] = 0;
1848       else {
1849         callback_args.col_widths[i] = strlen(cf->cinfo.col_title[i]);
1850         data_width = get_column_char_width(get_column_format(i));
1851         if (data_width > callback_args.col_widths[i])
1852           callback_args.col_widths[i] = data_width;
1853       }
1854
1855       /* Find the length of the string for this column. */
1856       column_len = strlen(cf->cinfo.col_title[i]);
1857       if (callback_args.col_widths[i] > column_len)
1858         column_len = callback_args.col_widths[i];
1859
1860       /* Make sure there's room in the line buffer for the column; if not,
1861          double its length. */
1862       line_len += column_len + 1;       /* "+1" for space */
1863       if (line_len > callback_args.header_line_buf_len) {
1864         cp_off = cp - callback_args.header_line_buf;
1865         callback_args.header_line_buf_len = 2 * line_len;
1866         callback_args.header_line_buf = g_realloc(callback_args.header_line_buf,
1867                                                   callback_args.header_line_buf_len + 1);
1868         cp = callback_args.header_line_buf + cp_off;
1869       }
1870
1871       /* Right-justify the packet number column. */
1872 /*      if (cf->cinfo.col_fmt[i] == COL_NUMBER)
1873         sprintf(cp, "%*s", callback_args.col_widths[i], cf->cinfo.col_title[i]);
1874       else*/
1875         sprintf(cp, "%-*s", callback_args.col_widths[i], cf->cinfo.col_title[i]);
1876       cp += column_len;
1877       if (i != cf->cinfo.num_cols - 1)
1878         *cp++ = ' ';
1879     }
1880     *cp = '\0';
1881
1882     /* Now start out the main line buffer with the same length as the
1883        header line buffer. */
1884     callback_args.line_buf_len = callback_args.header_line_buf_len;
1885     callback_args.line_buf = g_malloc(callback_args.line_buf_len + 1);
1886   } /* if (print_summary) */
1887
1888   /* Iterate through the list of packets, printing the packets we were
1889      told to print. */
1890   ret = process_specified_packets(cf, &print_args->range, "Printing",
1891                                   "selected packets", print_packet,
1892                                   &callback_args);
1893
1894   if (callback_args.header_line_buf != NULL)
1895     g_free(callback_args.header_line_buf);
1896   if (callback_args.line_buf != NULL)
1897     g_free(callback_args.line_buf);
1898   if (callback_args.col_widths != NULL)
1899     g_free(callback_args.col_widths);
1900
1901   switch (ret) {
1902
1903   case PSP_FINISHED:
1904     /* Completed successfully. */
1905     break;
1906
1907   case PSP_STOPPED:
1908     /* Well, the user decided to abort the printing.
1909
1910        XXX - note that what got generated before they did that
1911        will get printed if we're piping to a print program; we'd
1912        have to write to a file and then hand that to the print
1913        program to make it actually not print anything. */
1914     break;
1915
1916   case PSP_FAILED:
1917     /* Error while printing.
1918
1919        XXX - note that what got generated before they did that
1920        will get printed if we're piping to a print program; we'd
1921        have to write to a file and then hand that to the print
1922        program to make it actually not print anything. */
1923     destroy_print_stream(print_args->stream);
1924     return CF_PRINT_WRITE_ERROR;
1925   }
1926
1927   if (!print_finale(print_args->stream)) {
1928     destroy_print_stream(print_args->stream);
1929     return CF_PRINT_WRITE_ERROR;
1930   }
1931
1932   if (!destroy_print_stream(print_args->stream))
1933     return CF_PRINT_WRITE_ERROR;
1934
1935   return CF_PRINT_OK;
1936 }
1937
1938 static gboolean
1939 write_pdml_packet(capture_file *cf _U_, frame_data *fdata,
1940                   union wtap_pseudo_header *pseudo_header, const guint8 *pd,
1941                   void *argsp)
1942 {
1943   FILE *fh = argsp;
1944   epan_dissect_t *edt;
1945
1946   /* Create the protocol tree, but don't fill in the column information. */
1947   edt = epan_dissect_new(TRUE, TRUE);
1948   epan_dissect_run(edt, pseudo_header, pd, fdata, NULL);
1949
1950   /* Write out the information in that tree. */
1951   proto_tree_write_pdml(edt, fh);
1952
1953   epan_dissect_free(edt);
1954
1955   return !ferror(fh);
1956 }
1957
1958 cf_print_status_t
1959 cf_write_pdml_packets(capture_file *cf, print_args_t *print_args)
1960 {
1961   FILE        *fh;
1962   psp_return_t ret;
1963
1964   fh = fopen(print_args->file, "w");
1965   if (fh == NULL)
1966     return CF_PRINT_OPEN_ERROR; /* attempt to open destination failed */
1967
1968   write_pdml_preamble(fh);
1969   if (ferror(fh)) {
1970     fclose(fh);
1971     return CF_PRINT_WRITE_ERROR;
1972   }
1973
1974   /* Iterate through the list of packets, printing the packets we were
1975      told to print. */
1976   ret = process_specified_packets(cf, &print_args->range, "Writing PDML",
1977                                   "selected packets", write_pdml_packet,
1978                                   fh);
1979
1980   switch (ret) {
1981
1982   case PSP_FINISHED:
1983     /* Completed successfully. */
1984     break;
1985
1986   case PSP_STOPPED:
1987     /* Well, the user decided to abort the printing. */
1988     break;
1989
1990   case PSP_FAILED:
1991     /* Error while printing. */
1992     fclose(fh);
1993     return CF_PRINT_WRITE_ERROR;
1994   }
1995
1996   write_pdml_finale(fh);
1997   if (ferror(fh)) {
1998     fclose(fh);
1999     return CF_PRINT_WRITE_ERROR;
2000   }
2001
2002   /* XXX - check for an error */
2003   fclose(fh);
2004
2005   return CF_PRINT_OK;
2006 }
2007
2008 static gboolean
2009 write_psml_packet(capture_file *cf, frame_data *fdata,
2010                   union wtap_pseudo_header *pseudo_header, const guint8 *pd,
2011                   void *argsp)
2012 {
2013   FILE *fh = argsp;
2014   epan_dissect_t *edt;
2015
2016   /* Fill in the column information, but don't create the protocol tree. */
2017   edt = epan_dissect_new(FALSE, FALSE);
2018   epan_dissect_run(edt, pseudo_header, pd, fdata, &cf->cinfo);
2019   epan_dissect_fill_in_columns(edt);
2020
2021   /* Write out the information in that tree. */
2022   proto_tree_write_psml(edt, fh);
2023
2024   epan_dissect_free(edt);
2025
2026   return !ferror(fh);
2027 }
2028
2029 cf_print_status_t
2030 cf_write_psml_packets(capture_file *cf, print_args_t *print_args)
2031 {
2032   FILE        *fh;
2033   psp_return_t ret;
2034
2035   fh = fopen(print_args->file, "w");
2036   if (fh == NULL)
2037     return CF_PRINT_OPEN_ERROR; /* attempt to open destination failed */
2038
2039   write_psml_preamble(fh);
2040   if (ferror(fh)) {
2041     fclose(fh);
2042     return CF_PRINT_WRITE_ERROR;
2043   }
2044
2045   /* Iterate through the list of packets, printing the packets we were
2046      told to print. */
2047   ret = process_specified_packets(cf, &print_args->range, "Writing PSML",
2048                                   "selected packets", write_psml_packet,
2049                                   fh);
2050
2051   switch (ret) {
2052
2053   case PSP_FINISHED:
2054     /* Completed successfully. */
2055     break;
2056
2057   case PSP_STOPPED:
2058     /* Well, the user decided to abort the printing. */
2059     break;
2060
2061   case PSP_FAILED:
2062     /* Error while printing. */
2063     fclose(fh);
2064     return CF_PRINT_WRITE_ERROR;
2065   }
2066
2067   write_psml_finale(fh);
2068   if (ferror(fh)) {
2069     fclose(fh);
2070     return CF_PRINT_WRITE_ERROR;
2071   }
2072
2073   /* XXX - check for an error */
2074   fclose(fh);
2075
2076   return CF_PRINT_OK;
2077 }
2078
2079 static gboolean
2080 write_csv_packet(capture_file *cf, frame_data *fdata,
2081                  union wtap_pseudo_header *pseudo_header, const guint8 *pd,
2082                  void *argsp)
2083 {
2084   FILE *fh = argsp;
2085   epan_dissect_t *edt;
2086
2087   /* Fill in the column information, but don't create the protocol tree. */
2088   edt = epan_dissect_new(FALSE, FALSE);
2089   epan_dissect_run(edt, pseudo_header, pd, fdata, &cf->cinfo);
2090   epan_dissect_fill_in_columns(edt);
2091
2092   /* Write out the information in that tree. */
2093   proto_tree_write_csv(edt, fh);
2094
2095   epan_dissect_free(edt);
2096
2097   return !ferror(fh);
2098 }
2099
2100 cf_print_status_t
2101 cf_write_csv_packets(capture_file *cf, print_args_t *print_args)
2102 {
2103   FILE        *fh;
2104   psp_return_t ret;
2105
2106   fh = fopen(print_args->file, "w");
2107   if (fh == NULL)
2108     return CF_PRINT_OPEN_ERROR; /* attempt to open destination failed */
2109
2110   write_csv_preamble(fh);
2111   if (ferror(fh)) {
2112     fclose(fh);
2113     return CF_PRINT_WRITE_ERROR;
2114   }
2115
2116   /* Iterate through the list of packets, printing the packets we were
2117      told to print. */
2118   ret = process_specified_packets(cf, &print_args->range, "Writing CSV",
2119                                   "selected packets", write_csv_packet,
2120                                   fh);
2121
2122   switch (ret) {
2123
2124   case PSP_FINISHED:
2125     /* Completed successfully. */
2126     break;
2127
2128   case PSP_STOPPED:
2129     /* Well, the user decided to abort the printing. */
2130     break;
2131
2132   case PSP_FAILED:
2133     /* Error while printing. */
2134     fclose(fh);
2135     return CF_PRINT_WRITE_ERROR;
2136   }
2137
2138   write_csv_finale(fh);
2139   if (ferror(fh)) {
2140     fclose(fh);
2141     return CF_PRINT_WRITE_ERROR;
2142   }
2143
2144   /* XXX - check for an error */
2145   fclose(fh);
2146
2147   return CF_PRINT_OK;
2148 }
2149
2150 /* Scan through the packet list and change all columns that use the
2151    "command-line-specified" time stamp format to use the current
2152    value of that format. */
2153 void
2154 cf_change_time_formats(capture_file *cf)
2155 {
2156   frame_data *fdata;
2157   progdlg_t  *progbar = NULL;
2158   gboolean    stop_flag;
2159   int         count;
2160   int         row;
2161   int         i;
2162   float       prog_val;
2163   GTimeVal    start_time;
2164   gchar       status_str[100];
2165   int         progbar_nextstep;
2166   int         progbar_quantum;
2167   int         first, last;
2168   gboolean    sorted_by_frame_column;
2169
2170   /* Are there any columns with time stamps in the "command-line-specified"
2171      format?
2172
2173      XXX - we have to force the "column is writable" flag on, as it
2174      might be off from the last frame that was dissected. */
2175   col_set_writable(&cf->cinfo, TRUE);
2176   if (!check_col(&cf->cinfo, COL_CLS_TIME)) {
2177     /* No, there aren't any columns in that format, so we have no work
2178        to do. */
2179     return;
2180   }
2181   first = cf->cinfo.col_first[COL_CLS_TIME];
2182   g_assert(first >= 0);
2183   last = cf->cinfo.col_last[COL_CLS_TIME];
2184
2185   /* Freeze the packet list while we redo it, so we don't get any
2186      screen updates while it happens. */
2187   packet_list_freeze();
2188
2189   /* Update the progress bar when it gets to this value. */
2190   progbar_nextstep = 0;
2191   /* When we reach the value that triggers a progress bar update,
2192      bump that value by this amount. */
2193   progbar_quantum = cf->count/N_PROGBAR_UPDATES;
2194   /* Count of packets at which we've looked. */
2195   count = 0;
2196
2197   /*  If the rows are currently sorted by the frame column then we know
2198    *  the row number of each packet: it's the row number of the previously
2199    *  displayed packet + 1.
2200    *
2201    *  Otherwise, if the display is sorted by a different column then we have
2202    *  to use the O(N) packet_list_find_row_from_data() (thus making the job
2203    *  of changing the time display format O(N**2)).
2204    *
2205    *  (XXX - In fact it's still O(N**2) because gtk_clist_set_text() takes
2206    *  the row number and walks that many elements down the clist to find
2207    *  the appropriate element.)
2208    */
2209   sorted_by_frame_column = FALSE;
2210   for (i = 0; i < cf->cinfo.num_cols; i++) {
2211     if (cf->cinfo.col_fmt[i] == COL_NUMBER)
2212     {
2213       sorted_by_frame_column = (i == packet_list_get_sort_column());
2214       break;
2215     }
2216   }
2217
2218   stop_flag = FALSE;
2219   g_get_current_time(&start_time);
2220
2221   /* Iterate through the list of packets, checking whether the packet
2222      is in a row of the summary list and, if so, whether there are
2223      any columns that show the time in the "command-line-specified"
2224      format and, if so, update that row. */
2225   for (fdata = cf->plist, row = -1; fdata != NULL; fdata = fdata->next) {
2226     /* Update the progress bar, but do it only N_PROGBAR_UPDATES times;
2227        when we update it, we have to run the GTK+ main loop to get it
2228        to repaint what's pending, and doing so may involve an "ioctl()"
2229        to see if there's any pending input from an X server, and doing
2230        that for every packet can be costly, especially on a big file. */
2231     if (count >= progbar_nextstep) {
2232       /* let's not divide by zero. I should never be started
2233        * with count == 0, so let's assert that
2234        */
2235       g_assert(cf->count > 0);
2236
2237       prog_val = (gfloat) count / cf->count;
2238
2239       if (progbar == NULL)
2240         /* Create the progress bar if necessary */
2241         progbar = delayed_create_progress_dlg("Changing", "time display", 
2242           &stop_flag, &start_time, prog_val);
2243
2244       if (progbar != NULL) {
2245         g_snprintf(status_str, sizeof(status_str),
2246                    "%4u of %u packets", count, cf->count);
2247         update_progress_dlg(progbar, prog_val, status_str);
2248       }
2249
2250       progbar_nextstep += progbar_quantum;
2251     }
2252
2253     if (stop_flag) {
2254       /* Well, the user decided to abort the redisplay.  Just stop.
2255
2256          XXX - this leaves the time field in the old format in
2257          frames we haven't yet processed.  So it goes; should we
2258          simply not offer them the option of stopping? */
2259       break;
2260     }
2261
2262     count++;
2263
2264     /* Find what row this packet is in. */
2265     if (!sorted_by_frame_column) {
2266       /* This function is O(N), so we try to avoid using it... */
2267       row = packet_list_find_row_from_data(fdata);
2268     } else {
2269       /* ...which we do by maintaining a count of packets that are
2270          being displayed (i.e., that have passed the display filter),
2271          and using the current value of that count as the row number
2272          (which is why we can only do it when the display is sorted
2273          by the frame number). */
2274       if (fdata->flags.passed_dfilter)
2275         row++;
2276       else
2277         continue;
2278     }
2279
2280     if (row != -1) {
2281       /* This packet is in the summary list, on row "row". */
2282
2283       for (i = first; i <= last; i++) {
2284         if (cf->cinfo.fmt_matx[i][COL_CLS_TIME]) {
2285           /* This is one of the columns that shows the time in
2286              "command-line-specified" format; update it. */
2287           cf->cinfo.col_buf[i][0] = '\0';
2288           col_set_cls_time(fdata, &cf->cinfo, i);
2289           packet_list_set_text(row, i, cf->cinfo.col_data[i]);
2290         }
2291       }
2292     }
2293   }
2294
2295   /* We're done redisplaying the packets; destroy the progress bar if it
2296      was created. */
2297   if (progbar != NULL)
2298     destroy_progress_dlg(progbar);
2299
2300   /* Set the column widths of those columns that show the time in
2301      "command-line-specified" format. */
2302   for (i = first; i <= last; i++) {
2303     if (cf->cinfo.fmt_matx[i][COL_CLS_TIME]) {
2304       packet_list_set_cls_time_width(i);
2305     }
2306   }
2307
2308   /* Unfreeze the packet list. */
2309   packet_list_thaw();
2310 }
2311
2312 typedef struct {
2313         const char      *string;
2314         size_t          string_len;
2315         capture_file    *cf;
2316         gboolean        frame_matched;
2317 } match_data;
2318
2319 gboolean
2320 cf_find_packet_protocol_tree(capture_file *cf, const char *string)
2321 {
2322   match_data            mdata;
2323
2324   mdata.string = string;
2325   mdata.string_len = strlen(string);
2326   return find_packet(cf, match_protocol_tree, &mdata);
2327 }
2328
2329 static gboolean
2330 match_protocol_tree(capture_file *cf, frame_data *fdata, void *criterion)
2331 {
2332   match_data            *mdata = criterion;
2333   epan_dissect_t        *edt;
2334
2335   /* Construct the protocol tree, including the displayed text */
2336   edt = epan_dissect_new(TRUE, TRUE);
2337   /* We don't need the column information */
2338   epan_dissect_run(edt, &cf->pseudo_header, cf->pd, fdata, NULL);
2339
2340   /* Iterate through all the nodes, seeing if they have text that matches. */
2341   mdata->cf = cf;
2342   mdata->frame_matched = FALSE;
2343   proto_tree_children_foreach(edt->tree, match_subtree_text, mdata);
2344   epan_dissect_free(edt);
2345   return mdata->frame_matched;
2346 }
2347
2348 static void
2349 match_subtree_text(proto_node *node, gpointer data)
2350 {
2351   match_data    *mdata = (match_data*) data;
2352   const gchar   *string = mdata->string;
2353   size_t        string_len = mdata->string_len;
2354   capture_file  *cf = mdata->cf;
2355   field_info    *fi = PITEM_FINFO(node);
2356   gchar         label_str[ITEM_LABEL_LENGTH];
2357   gchar         *label_ptr;
2358   size_t        label_len;
2359   guint32       i;
2360   guint8        c_char;
2361   size_t        c_match = 0;
2362
2363   if (mdata->frame_matched) {
2364     /* We already had a match; don't bother doing any more work. */
2365     return;
2366   }
2367
2368   /* Don't match invisible entries. */
2369   if (PROTO_ITEM_IS_HIDDEN(node))
2370     return;
2371
2372   /* was a free format label produced? */
2373   if (fi->rep) {
2374     label_ptr = fi->rep->representation;
2375   } else {
2376     /* no, make a generic label */
2377     label_ptr = label_str;
2378     proto_item_fill_label(fi, label_str);
2379   }
2380     
2381   /* Does that label match? */
2382   label_len = strlen(label_ptr);
2383   for (i = 0; i < label_len; i++) {
2384     c_char = label_ptr[i];
2385     if (cf->case_type)
2386       c_char = toupper(c_char);
2387     if (c_char == string[c_match]) {
2388       c_match++;
2389       if (c_match == string_len) {
2390         /* No need to look further; we have a match */
2391         mdata->frame_matched = TRUE;
2392         return;
2393       }
2394     } else
2395       c_match = 0;
2396   }
2397   
2398   /* Recurse into the subtree, if it exists */
2399   if (node->first_child != NULL)
2400     proto_tree_children_foreach(node, match_subtree_text, mdata);
2401 }
2402
2403 gboolean
2404 cf_find_packet_summary_line(capture_file *cf, const char *string)
2405 {
2406   match_data            mdata;
2407
2408   mdata.string = string;
2409   mdata.string_len = strlen(string);
2410   return find_packet(cf, match_summary_line, &mdata);
2411 }
2412
2413 static gboolean
2414 match_summary_line(capture_file *cf, frame_data *fdata, void *criterion)
2415 {
2416   match_data            *mdata = criterion;
2417   const gchar           *string = mdata->string;
2418   size_t                string_len = mdata->string_len;
2419   epan_dissect_t        *edt;
2420   const char            *info_column;
2421   size_t                info_column_len;
2422   gboolean              frame_matched = FALSE;
2423   gint                  colx;
2424   guint32               i;
2425   guint8                c_char;
2426   size_t                c_match = 0;
2427
2428   /* Don't bother constructing the protocol tree */
2429   edt = epan_dissect_new(FALSE, FALSE);
2430   /* Get the column information */
2431   epan_dissect_run(edt, &cf->pseudo_header, cf->pd, fdata, &cf->cinfo);
2432
2433   /* Find the Info column */
2434   for (colx = 0; colx < cf->cinfo.num_cols; colx++) {
2435     if (cf->cinfo.fmt_matx[colx][COL_INFO]) {
2436       /* Found it.  See if we match. */
2437       info_column = edt->pi.cinfo->col_data[colx];
2438       info_column_len = strlen(info_column);
2439       for (i = 0; i < info_column_len; i++) {
2440         c_char = info_column[i];
2441         if (cf->case_type)
2442           c_char = toupper(c_char);
2443         if (c_char == string[c_match]) {
2444           c_match++;
2445           if (c_match == string_len) {
2446             frame_matched = TRUE;
2447             break;
2448           }
2449         } else
2450           c_match = 0;
2451       }
2452       break;
2453     }
2454   }
2455   epan_dissect_free(edt);
2456   return frame_matched;
2457 }
2458
2459 typedef struct {
2460         const guint8 *data;
2461         size_t data_len;
2462 } cbs_t;        /* "Counted byte string" */
2463
2464 gboolean
2465 cf_find_packet_data(capture_file *cf, const guint8 *string, size_t string_size)
2466 {
2467   cbs_t info;
2468
2469   info.data = string;
2470   info.data_len = string_size;
2471
2472   /* String or hex search? */
2473   if (cf->string) {
2474     /* String search - what type of string? */
2475     switch (cf->scs_type) {
2476
2477     case SCS_ASCII_AND_UNICODE:
2478       return find_packet(cf, match_ascii_and_unicode, &info);
2479
2480     case SCS_ASCII:
2481       return find_packet(cf, match_ascii, &info);
2482
2483     case SCS_UNICODE:
2484       return find_packet(cf, match_unicode, &info);
2485
2486     default:
2487       g_assert_not_reached();
2488       return FALSE;
2489     }
2490   } else
2491     return find_packet(cf, match_binary, &info);
2492 }
2493
2494 static gboolean
2495 match_ascii_and_unicode(capture_file *cf, frame_data *fdata, void *criterion)
2496 {
2497   cbs_t         *info = criterion;
2498   const char    *ascii_text = info->data;
2499   size_t        textlen = info->data_len;
2500   gboolean      frame_matched;
2501   guint32       buf_len;
2502   guint32       i;
2503   guint8        c_char;
2504   size_t        c_match = 0;
2505
2506   frame_matched = FALSE;
2507   buf_len = fdata->pkt_len;
2508   for (i = 0; i < buf_len; i++) {
2509     c_char = cf->pd[i];
2510     if (cf->case_type)
2511       c_char = toupper(c_char);
2512     if (c_char != 0) {
2513       if (c_char == ascii_text[c_match]) {
2514         c_match++;
2515         if (c_match == textlen) {
2516           frame_matched = TRUE;
2517           break;
2518         }
2519       } else
2520         c_match = 0;
2521     }
2522   }
2523   return frame_matched;
2524 }
2525
2526 static gboolean
2527 match_ascii(capture_file *cf, frame_data *fdata, void *criterion)
2528 {
2529   cbs_t         *info = criterion;
2530   const char    *ascii_text = info->data;
2531   size_t        textlen = info->data_len;
2532   gboolean      frame_matched;
2533   guint32       buf_len;
2534   guint32       i;
2535   guint8        c_char;
2536   size_t        c_match = 0;
2537
2538   frame_matched = FALSE;
2539   buf_len = fdata->pkt_len;
2540   for (i = 0; i < buf_len; i++) {
2541     c_char = cf->pd[i];
2542     if (cf->case_type)
2543       c_char = toupper(c_char);
2544     if (c_char == ascii_text[c_match]) {
2545       c_match++;
2546       if (c_match == textlen) {
2547         frame_matched = TRUE;
2548         break;
2549       }
2550     } else
2551       c_match = 0;
2552   }
2553   return frame_matched;
2554 }
2555
2556 static gboolean
2557 match_unicode(capture_file *cf, frame_data *fdata, void *criterion)
2558 {
2559   cbs_t         *info = criterion;
2560   const char    *ascii_text = info->data;
2561   size_t        textlen = info->data_len;
2562   gboolean      frame_matched;
2563   guint32       buf_len;
2564   guint32       i;
2565   guint8        c_char;
2566   size_t        c_match = 0;
2567
2568   frame_matched = FALSE;
2569   buf_len = fdata->pkt_len;
2570   for (i = 0; i < buf_len; i++) {
2571     c_char = cf->pd[i];
2572     if (cf->case_type)
2573       c_char = toupper(c_char);
2574     if (c_char == ascii_text[c_match]) {
2575       c_match++;
2576       i++;
2577       if (c_match == textlen) {
2578         frame_matched = TRUE;
2579         break;
2580       }
2581     } else
2582       c_match = 0;
2583   }
2584   return frame_matched;
2585 }
2586
2587 static gboolean
2588 match_binary(capture_file *cf, frame_data *fdata, void *criterion)
2589 {
2590   cbs_t         *info = criterion;
2591   const guint8  *binary_data = info->data;
2592   size_t        datalen = info->data_len;
2593   gboolean      frame_matched;
2594   guint32       buf_len;
2595   guint32       i;
2596   size_t        c_match = 0;
2597
2598   frame_matched = FALSE;
2599   buf_len = fdata->pkt_len;
2600   for (i = 0; i < buf_len; i++) {
2601     if (cf->pd[i] == binary_data[c_match]) {
2602       c_match++;
2603       if (c_match == datalen) {
2604         frame_matched = TRUE;
2605         break;
2606       }
2607     } else
2608       c_match = 0;
2609   }
2610   return frame_matched;
2611 }
2612
2613 gboolean
2614 cf_find_packet_dfilter(capture_file *cf, dfilter_t *sfcode)
2615 {
2616   return find_packet(cf, match_dfilter, sfcode);
2617 }
2618
2619 static gboolean
2620 match_dfilter(capture_file *cf, frame_data *fdata, void *criterion)
2621 {
2622   dfilter_t             *sfcode = criterion;
2623   epan_dissect_t        *edt;
2624   gboolean              frame_matched;
2625
2626   edt = epan_dissect_new(TRUE, FALSE);
2627   epan_dissect_prime_dfilter(edt, sfcode);
2628   epan_dissect_run(edt, &cf->pseudo_header, cf->pd, fdata, NULL);
2629   frame_matched = dfilter_apply_edt(sfcode, edt);
2630   epan_dissect_free(edt);
2631   return frame_matched;
2632 }
2633
2634 static gboolean
2635 find_packet(capture_file *cf,
2636             gboolean (*match_function)(capture_file *, frame_data *, void *),
2637             void *criterion)
2638 {
2639   frame_data *start_fd;
2640   frame_data *fdata;
2641   frame_data *new_fd = NULL;
2642   progdlg_t  *progbar = NULL;
2643   gboolean    stop_flag;
2644   int         count;
2645   int         err;
2646   gchar      *err_info;
2647   int         row;
2648   float       prog_val;
2649   GTimeVal    start_time;
2650   gchar       status_str[100];
2651   int         progbar_nextstep;
2652   int         progbar_quantum;
2653
2654   start_fd = cf->current_frame;
2655   if (start_fd != NULL)  {
2656     /* Iterate through the list of packets, starting at the packet we've
2657        picked, calling a routine to run the filter on the packet, see if
2658        it matches, and stop if so.  */
2659     count = 0;
2660     fdata = start_fd;
2661
2662     progbar_nextstep = 0;
2663     /* When we reach the value that triggers a progress bar update,
2664        bump that value by this amount. */
2665     progbar_quantum = cf->count/N_PROGBAR_UPDATES;
2666
2667     stop_flag = FALSE;
2668     g_get_current_time(&start_time);
2669
2670     fdata = start_fd;
2671     for (;;) {
2672       /* Update the progress bar, but do it only N_PROGBAR_UPDATES times;
2673          when we update it, we have to run the GTK+ main loop to get it
2674          to repaint what's pending, and doing so may involve an "ioctl()"
2675          to see if there's any pending input from an X server, and doing
2676          that for every packet can be costly, especially on a big file. */
2677       if (count >= progbar_nextstep) {
2678         /* let's not divide by zero. I should never be started
2679          * with count == 0, so let's assert that
2680          */
2681         g_assert(cf->count > 0);
2682
2683         prog_val = (gfloat) count / cf->count;
2684
2685         /* Create the progress bar if necessary */
2686         if (progbar == NULL)
2687            progbar = delayed_create_progress_dlg("Searching", cf->sfilter, 
2688              &stop_flag, &start_time, prog_val);
2689
2690         if (progbar != NULL) {
2691           g_snprintf(status_str, sizeof(status_str),
2692                      "%4u of %u packets", count, cf->count);
2693           update_progress_dlg(progbar, prog_val, status_str);
2694         }
2695
2696         progbar_nextstep += progbar_quantum;
2697       }
2698
2699       if (stop_flag) {
2700         /* Well, the user decided to abort the search.  Go back to the
2701            frame where we started. */
2702         new_fd = start_fd;
2703         break;
2704       }
2705
2706       /* Go past the current frame. */
2707       if (cf->sbackward) {
2708         /* Go on to the previous frame. */
2709         fdata = fdata->prev;
2710         if (fdata == NULL) {
2711           /*
2712            * XXX - other apps have a bit more of a detailed message
2713            * for this, and instead of offering "OK" and "Cancel",
2714            * they offer things such as "Continue" and "Cancel";
2715            * we need an API for popping up alert boxes with
2716            * {Verb} and "Cancel".
2717            */
2718
2719           if (prefs.gui_find_wrap)
2720           {
2721               simple_dialog(ESD_TYPE_INFO, ESD_BTN_OK,
2722                             "%sBeginning of capture exceeded!%s\n\n"
2723                             "Search is continued from the end of the capture.",
2724                             simple_dialog_primary_start(), simple_dialog_primary_end());
2725               fdata = cf->plist_end;    /* wrap around */
2726           }
2727           else
2728           {
2729               simple_dialog(ESD_TYPE_INFO, ESD_BTN_OK,
2730                             "%sBeginning of capture exceeded!%s\n\n"
2731                             "Try searching forwards.",
2732                             simple_dialog_primary_start(), simple_dialog_primary_end());
2733               fdata = start_fd;        /* stay on previous packet */
2734           }
2735         }
2736       } else {
2737         /* Go on to the next frame. */
2738         fdata = fdata->next;
2739         if (fdata == NULL) {
2740           if (prefs.gui_find_wrap)
2741           {
2742               simple_dialog(ESD_TYPE_INFO, ESD_BTN_OK,
2743                             "%sEnd of capture exceeded!%s\n\n"
2744                             "Search is continued from the start of the capture.",
2745                             simple_dialog_primary_start(), simple_dialog_primary_end());
2746               fdata = cf->plist;        /* wrap around */
2747           }
2748           else
2749           {
2750               simple_dialog(ESD_TYPE_INFO, ESD_BTN_OK,
2751                             "%sEnd of capture exceeded!%s\n\n"
2752                             "Try searching backwards.",
2753                             simple_dialog_primary_start(), simple_dialog_primary_end());
2754               fdata = start_fd;     /* stay on previous packet */
2755           }
2756         }
2757       }
2758
2759       count++;
2760
2761       /* Is this packet in the display? */
2762       if (fdata->flags.passed_dfilter) {
2763         /* Yes.  Load its data. */
2764         if (!wtap_seek_read(cf->wth, fdata->file_off, &cf->pseudo_header,
2765                         cf->pd, fdata->cap_len, &err, &err_info)) {
2766           /* Read error.  Report the error, and go back to the frame
2767              where we started. */
2768           simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
2769                         cf_read_error_message(err, err_info), cf->filename);
2770           new_fd = start_fd;
2771           break;
2772         }
2773
2774         /* Does it match the search criterion? */
2775         if ((*match_function)(cf, fdata, criterion)) {
2776           new_fd = fdata;
2777           break;        /* found it! */
2778         }
2779       }
2780
2781       if (fdata == start_fd) {
2782         /* We're back to the frame we were on originally, and that frame
2783            doesn't match the search filter.  The search failed. */
2784         break;
2785       }
2786     }
2787
2788     /* We're done scanning the packets; destroy the progress bar if it
2789        was created. */
2790     if (progbar != NULL)
2791       destroy_progress_dlg(progbar);
2792   }
2793
2794   if (new_fd != NULL) {
2795     /* We found a frame.  Find what row it's in. */
2796     row = packet_list_find_row_from_data(new_fd);
2797     g_assert(row != -1);
2798
2799     /* Select that row, make it the focus row, and make it visible. */
2800     packet_list_set_selected_row(row);
2801     return TRUE;        /* success */
2802   } else
2803     return FALSE;       /* failure */
2804 }
2805
2806 gboolean
2807 cf_goto_frame(capture_file *cf, guint fnumber)
2808 {
2809   frame_data *fdata;
2810   int row;
2811
2812   for (fdata = cf->plist; fdata != NULL && fdata->num < fnumber; fdata = fdata->next)
2813     ;
2814
2815   if (fdata == NULL) {
2816     /* we didn't find a packet with that packet number */
2817     simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
2818                   "There is no packet with the packet number %u.", fnumber);
2819     return FALSE;       /* we failed to go to that packet */
2820   }
2821   if (!fdata->flags.passed_dfilter) {
2822     /* that packet currently isn't displayed */
2823     /* XXX - add it to the set of displayed packets? */
2824     simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
2825                   "The packet number %u isn't currently being displayed.", fnumber);
2826     return FALSE;       /* we failed to go to that packet */
2827   }
2828
2829   /* We found that packet, and it's currently being displayed.
2830      Find what row it's in. */
2831   row = packet_list_find_row_from_data(fdata);
2832   g_assert(row != -1);
2833
2834   /* Select that row, make it the focus row, and make it visible. */
2835   packet_list_set_selected_row(row);
2836   return TRUE;  /* we got to that packet */
2837 }
2838
2839 gboolean
2840 cf_goto_top_frame(capture_file *cf)
2841 {
2842   frame_data *fdata;
2843   int row;
2844   frame_data *lowest_fdata = NULL;
2845
2846   for (fdata = cf->plist; fdata != NULL; fdata = fdata->next) {
2847     if (fdata->flags.passed_dfilter) {
2848         lowest_fdata = fdata;
2849         break;
2850     }
2851   }
2852
2853   if (lowest_fdata == NULL) {
2854       return FALSE;
2855   }
2856
2857   /* We found that packet, and it's currently being displayed.
2858      Find what row it's in. */
2859   row = packet_list_find_row_from_data(lowest_fdata);
2860   g_assert(row != -1);
2861
2862   /* Select that row, make it the focus row, and make it visible. */
2863   packet_list_set_selected_row(row);
2864   return TRUE;  /* we got to that packet */
2865 }
2866
2867 gboolean
2868 cf_goto_bottom_frame(capture_file *cf)
2869 {
2870   frame_data *fdata;
2871   int row;
2872   frame_data *highest_fdata = NULL;
2873
2874   for (fdata = cf->plist; fdata != NULL; fdata = fdata->next) {
2875     if (fdata->flags.passed_dfilter) {
2876         highest_fdata = fdata;
2877     }
2878   }
2879
2880   if (highest_fdata == NULL) {
2881       return FALSE;
2882   }
2883
2884   /* We found that packet, and it's currently being displayed.
2885      Find what row it's in. */
2886   row = packet_list_find_row_from_data(highest_fdata);
2887   g_assert(row != -1);
2888
2889   /* Select that row, make it the focus row, and make it visible. */
2890   packet_list_set_selected_row(row);
2891   return TRUE;  /* we got to that packet */
2892 }
2893
2894 /*
2895  * Go to frame specified by currently selected protocol tree item.
2896  */
2897 gboolean
2898 cf_goto_framenum(capture_file *cf)
2899 {
2900   header_field_info       *hfinfo;
2901   guint32                 framenum;
2902
2903   if (cf->finfo_selected) {
2904     hfinfo = cf->finfo_selected->hfinfo;
2905     g_assert(hfinfo);
2906     if (hfinfo->type == FT_FRAMENUM) {
2907       framenum = fvalue_get_integer(&cf->finfo_selected->value);
2908       if (framenum != 0)
2909         return cf_goto_frame(cf, framenum);
2910       }
2911   }
2912
2913   return FALSE;
2914 }
2915
2916 /* Select the packet on a given row. */
2917 void
2918 cf_select_packet(capture_file *cf, int row)
2919 {
2920   frame_data *fdata;
2921   int err;
2922   gchar *err_info;
2923
2924   /* Get the frame data struct pointer for this frame */
2925   fdata = (frame_data *)packet_list_get_row_data(row);
2926
2927   if (fdata == NULL) {
2928     /* XXX - if a GtkCList's selection mode is GTK_SELECTION_BROWSE, when
2929        the first entry is added to it by "real_insert_row()", that row
2930        is selected (see "real_insert_row()", in "gtk/gtkclist.c", in both
2931        our version and the vanilla GTK+ version).
2932
2933        This means that a "select-row" signal is emitted; this causes
2934        "packet_list_select_cb()" to be called, which causes "cf_select_packet()"
2935        to be called.
2936
2937        "cf_select_packet()" fetches, above, the data associated with the
2938        row that was selected; however, as "gtk_clist_append()", which
2939        called "real_insert_row()", hasn't yet returned, we haven't yet
2940        associated any data with that row, so we get back a null pointer.
2941
2942        We can't assume that there's only one frame in the frame list,
2943        either, as we may be filtering the display.
2944
2945        We therefore assume that, if "row" is 0, i.e. the first row
2946        is being selected, and "cf->first_displayed" equals
2947        "cf->last_displayed", i.e. there's only one frame being
2948        displayed, that frame is the frame we want.
2949
2950        This means we have to set "cf->first_displayed" and
2951        "cf->last_displayed" before adding the row to the
2952        GtkCList; see the comment in "add_packet_to_packet_list()". */
2953
2954        if (row == 0 && cf->first_displayed == cf->last_displayed)
2955          fdata = cf->first_displayed;
2956   }
2957
2958   /* Get the data in that frame. */
2959   if (!wtap_seek_read (cf->wth, fdata->file_off, &cf->pseudo_header,
2960                        cf->pd, fdata->cap_len, &err, &err_info)) {
2961     simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
2962                   cf_read_error_message(err, err_info), cf->filename);
2963     return;
2964   }
2965
2966   /* Record that this frame is the current frame. */
2967   cf->current_frame = fdata;
2968
2969   /* Create the logical protocol tree. */
2970   if (cf->edt != NULL) {
2971     epan_dissect_free(cf->edt);
2972     cf->edt = NULL;
2973   }
2974   /* We don't need the columns here. */
2975   cf->edt = epan_dissect_new(TRUE, TRUE);
2976   epan_dissect_run(cf->edt, &cf->pseudo_header, cf->pd, cf->current_frame,
2977           NULL);
2978
2979   cf_callback_invoke(cf_cb_packet_selected, cf);
2980 }
2981
2982 /* Unselect the selected packet, if any. */
2983 void
2984 cf_unselect_packet(capture_file *cf)
2985 {
2986   /* Destroy the epan_dissect_t for the unselected packet. */
2987   if (cf->edt != NULL) {
2988     epan_dissect_free(cf->edt);
2989     cf->edt = NULL;
2990   }
2991
2992   /* No packet is selected. */
2993   cf->current_frame = NULL;
2994
2995   cf_callback_invoke(cf_cb_packet_unselected, cf);
2996
2997   /* No protocol tree means no selected field. */
2998   cf_unselect_field(cf);
2999 }
3000
3001 /* Unset the selected protocol tree field, if any. */
3002 void
3003 cf_unselect_field(capture_file *cf)
3004 {
3005   cf->finfo_selected = NULL;
3006
3007   cf_callback_invoke(cf_cb_field_unselected, cf);
3008 }
3009
3010 /*
3011  * Mark a particular frame.
3012  */
3013 void
3014 cf_mark_frame(capture_file *cf, frame_data *frame)
3015 {
3016   if (! frame->flags.marked) {
3017     frame->flags.marked = TRUE;
3018     if (cf->count > cf->marked_count)
3019       cf->marked_count++;
3020   }
3021 }
3022
3023 /*
3024  * Unmark a particular frame.
3025  */
3026 void
3027 cf_unmark_frame(capture_file *cf, frame_data *frame)
3028 {
3029   if (frame->flags.marked) {
3030     frame->flags.marked = FALSE;
3031     if (cf->marked_count > 0)
3032       cf->marked_count--;
3033   }
3034 }
3035
3036 typedef struct {
3037   wtap_dumper *pdh;
3038   const char  *fname;
3039 } save_callback_args_t;
3040
3041 /*
3042  * Save a capture to a file, in a particular format, saving either
3043  * all packets, all currently-displayed packets, or all marked packets.
3044  *
3045  * Returns TRUE if it succeeds, FALSE otherwise; if it fails, it pops
3046  * up a message box for the failure.
3047  */
3048 static gboolean
3049 save_packet(capture_file *cf _U_, frame_data *fdata,
3050             union wtap_pseudo_header *pseudo_header, const guint8 *pd,
3051             void *argsp)
3052 {
3053   save_callback_args_t *args = argsp;
3054   struct wtap_pkthdr hdr;
3055   int           err;
3056
3057   /* init the wtap header for saving */
3058   hdr.ts.tv_sec  = fdata->abs_secs;
3059   hdr.ts.tv_usec = fdata->abs_usecs;
3060   hdr.caplen     = fdata->cap_len;
3061   hdr.len        = fdata->pkt_len;
3062   hdr.pkt_encap  = fdata->lnk_t;
3063
3064   /* and save the packet */
3065   if (!wtap_dump(args->pdh, &hdr, pseudo_header, pd, &err)) {
3066     cf_write_failure_alert_box(args->fname, err);
3067     return FALSE;
3068   }
3069   return TRUE;
3070 }
3071
3072 cf_status_t
3073 cf_save(capture_file *cf, const char *fname, packet_range_t *range, guint save_format)
3074 {
3075   gchar        *from_filename;
3076   int           err;
3077   gboolean      do_copy;
3078   wtap_dumper  *pdh;
3079   save_callback_args_t callback_args;
3080
3081   cf_callback_invoke(cf_cb_file_safe_started, fname);
3082
3083   /* don't write over an existing file. */
3084   /* this should've been already checked by our caller, just to be sure... */
3085   if (file_exists(fname)) {
3086     simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3087       "%sCapture file: \"%s\" already exists!%s\n\n"
3088       "Please choose a different filename.",
3089       simple_dialog_primary_start(), fname, simple_dialog_primary_end());
3090     goto fail;
3091   }
3092
3093   packet_range_process_init(range);
3094
3095   /* Used to be :
3096    * if (!save_filtered && !save_marked && !save_manual_range && 
3097    *     !save_marked_range && !save_curr && save_format == cf->cd_t) {
3098    */ 
3099         
3100   if (packet_range_process_all(range) && save_format == cf->cd_t) {
3101     /* We're not filtering packets, and we're saving it in the format
3102        it's already in, so we can just move or copy the raw data. */
3103
3104     if (cf->is_tempfile) {
3105       /* The file being saved is a temporary file from a live
3106          capture, so it doesn't need to stay around under that name;
3107          first, try renaming the capture buffer file to the new name. */
3108 #ifndef _WIN32
3109       if (rename(cf->filename, fname) == 0) {
3110         /* That succeeded - there's no need to copy the source file. */
3111         from_filename = NULL;
3112         do_copy = FALSE;
3113       } else {
3114         if (errno == EXDEV) {
3115           /* They're on different file systems, so we have to copy the
3116              file. */
3117           do_copy = TRUE;
3118           from_filename = cf->filename;
3119         } else {
3120           /* The rename failed, but not because they're on different
3121              file systems - put up an error message.  (Or should we
3122              just punt and try to copy?  The only reason why I'd
3123              expect the rename to fail and the copy to succeed would
3124              be if we didn't have permission to remove the file from
3125              the temporary directory, and that might be fixable - but
3126              is it worth requiring the user to go off and fix it?) */
3127           simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3128                                 file_rename_error_message(errno), fname);
3129           goto fail;
3130         }
3131       }
3132 #else
3133       do_copy = TRUE;
3134       from_filename = cf->filename;
3135 #endif
3136     } else {
3137       /* It's a permanent file, so we should copy it, and not remove the
3138          original. */
3139       do_copy = TRUE;
3140       from_filename = cf->filename;
3141     }
3142
3143     if (do_copy) {
3144       /* Copy the file, if we haven't moved it. */
3145       if (!copy_binary_file(from_filename, fname))
3146         goto fail;
3147     }
3148   } else {
3149     /* Either we're filtering packets, or we're saving in a different
3150        format; we can't do that by copying or moving the capture file,
3151        we have to do it by writing the packets out in Wiretap. */
3152     pdh = wtap_dump_open(fname, save_format, cf->lnk_t, cf->snap, &err);
3153     if (pdh == NULL) {
3154       cf_open_failure_alert_box(fname, err, NULL, TRUE, save_format);
3155       goto fail;
3156     }
3157
3158     /* XXX - we let the user save a subset of the packets.
3159
3160        If we do that, should we make that file the current file?  If so,
3161        it means we can no longer get at the other packets.  What does
3162        NetMon do? */
3163
3164     /* Iterate through the list of packets, processing the packets we were
3165        told to process.
3166
3167        XXX - we've already called "packet_range_process_init(range)", but
3168        "process_specified_packets()" will do it again.  Fortunately,
3169        that's harmless in this case, as we haven't done anything to
3170        "range" since we initialized it. */
3171     callback_args.pdh = pdh;
3172     callback_args.fname = fname;
3173     switch (process_specified_packets(cf, range, "Saving",
3174                                       "selected packets", save_packet,
3175                                       &callback_args)) {
3176
3177     case PSP_FINISHED:
3178       /* Completed successfully. */
3179       break;
3180
3181     case PSP_STOPPED:
3182       /* The user decided to abort the saving.
3183          XXX - remove the output file? */
3184       break;
3185
3186     case PSP_FAILED:
3187       /* Error while saving. */
3188       wtap_dump_close(pdh, &err);
3189       goto fail;
3190     }
3191
3192     if (!wtap_dump_close(pdh, &err)) {
3193       cf_close_failure_alert_box(fname, err);
3194       goto fail;
3195     }
3196   }
3197
3198   cf_callback_invoke(cf_cb_file_safe_finished, NULL);
3199
3200   if (packet_range_process_all(range)) {
3201     /* We saved the entire capture, not just some packets from it.
3202        Open and read the file we saved it to.
3203
3204        XXX - this is somewhat of a waste; we already have the
3205        packets, all this gets us is updated file type information
3206        (which we could just stuff into "cf"), and having the new
3207        file be the one we have opened and from which we're reading
3208        the data, and it means we have to spend time opening and
3209        reading the file, which could be a significant amount of
3210        time if the file is large. */
3211     cf->user_saved = TRUE;
3212
3213     if ((cf_open(cf, fname, FALSE, &err)) == CF_OK) {
3214       /* XXX - report errors if this fails?
3215          What should we return if it fails or is aborted? */
3216       switch (cf_read(cf)) {
3217
3218       case CF_READ_OK:
3219       case CF_READ_ERROR:
3220         /* Just because we got an error, that doesn't mean we were unable
3221            to read any of the file; we handle what we could get from the
3222            file. */
3223         break;
3224
3225       case CF_READ_ABORTED:
3226         /* The user bailed out of re-reading the capture file; the
3227            capture file has been closed - just return (without
3228            changing any menu settings; "cf_close()" set them
3229            correctly for the "no capture file open" state). */
3230         break;
3231       }
3232       cf_callback_invoke(cf_cb_file_safe_reload_finished, NULL);
3233     }
3234   }
3235   return CF_OK;
3236
3237 fail:
3238   cf_callback_invoke(cf_cb_file_safe_failed, NULL);
3239   return CF_ERROR;
3240 }
3241
3242 static void
3243 cf_open_failure_alert_box(const char *filename, int err, gchar *err_info,
3244                           gboolean for_writing, int file_type)
3245 {
3246   if (err < 0) {
3247     /* Wiretap error. */
3248     switch (err) {
3249
3250     case WTAP_ERR_NOT_REGULAR_FILE:
3251       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3252                     "The file \"%s\" is a \"special file\" or socket or other non-regular file.",
3253                     filename);
3254       break;
3255
3256     case WTAP_ERR_RANDOM_OPEN_PIPE:
3257       /* Seen only when opening a capture file for reading. */
3258       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3259                     "The file \"%s\" is a pipe or FIFO; Ethereal can't read pipe or FIFO files.",
3260                     filename);
3261       break;
3262
3263     case WTAP_ERR_FILE_UNKNOWN_FORMAT:
3264       /* Seen only when opening a capture file for reading. */
3265       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3266                     "The file \"%s\" isn't a capture file in a format Ethereal understands.",
3267                     filename);
3268       break;
3269
3270     case WTAP_ERR_UNSUPPORTED:
3271       /* Seen only when opening a capture file for reading. */
3272       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3273                     "The file \"%s\" isn't a capture file in a format Ethereal understands.\n"
3274                     "(%s)",
3275                     filename, err_info);
3276       g_free(err_info);
3277       break;
3278
3279     case WTAP_ERR_CANT_WRITE_TO_PIPE:
3280       /* Seen only when opening a capture file for writing. */
3281       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3282                     "The file \"%s\" is a pipe, and %s capture files can't be "
3283                     "written to a pipe.",
3284                     filename, wtap_file_type_string(file_type));
3285       break;
3286
3287     case WTAP_ERR_UNSUPPORTED_FILE_TYPE:
3288       /* Seen only when opening a capture file for writing. */
3289       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3290                     "Ethereal doesn't support writing capture files in that format.");
3291       break;
3292
3293     case WTAP_ERR_UNSUPPORTED_ENCAP:
3294       if (for_writing) {
3295         simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3296                       "Ethereal can't save this capture in that format.");
3297       } else {
3298         simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3299                       "The file \"%s\" is a capture for a network type that Ethereal doesn't support.\n"
3300                       "(%s)",
3301                       filename, err_info);
3302         g_free(err_info);
3303       }
3304       break;
3305
3306     case WTAP_ERR_ENCAP_PER_PACKET_UNSUPPORTED:
3307       if (for_writing) {
3308         simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3309                       "Ethereal can't save this capture in that format.");
3310       } else {
3311         simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3312                       "The file \"%s\" is a capture for a network type that Ethereal doesn't support.",
3313                       filename);
3314       }
3315       break;
3316
3317     case WTAP_ERR_BAD_RECORD:
3318       /* Seen only when opening a capture file for reading. */
3319       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3320                     "The file \"%s\" appears to be damaged or corrupt.\n"
3321                     "(%s)",
3322                     filename, err_info);
3323       g_free(err_info);
3324       break;
3325
3326     case WTAP_ERR_CANT_OPEN:
3327       if (for_writing) {
3328         simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3329                       "The file \"%s\" could not be created for some unknown reason.",
3330                       filename);
3331       } else {
3332         simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3333                       "The file \"%s\" could not be opened for some unknown reason.",
3334                       filename);
3335       }
3336       break;
3337
3338     case WTAP_ERR_SHORT_READ:
3339       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3340                     "The file \"%s\" appears to have been cut short"
3341                     " in the middle of a packet or other data.",
3342                     filename);
3343       break;
3344
3345     case WTAP_ERR_SHORT_WRITE:
3346       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3347                     "A full header couldn't be written to the file \"%s\".",
3348                     filename);
3349       break;
3350
3351     default:
3352       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3353                     "The file \"%s\" could not be %s: %s.",
3354                     filename,
3355                     for_writing ? "created" : "opened",
3356                     wtap_strerror(err));
3357       break;
3358     }
3359   } else {
3360     /* OS error. */
3361     open_failure_alert_box(filename, err, for_writing);
3362   }
3363 }
3364
3365 static const char *
3366 file_rename_error_message(int err)
3367 {
3368   const char *errmsg;
3369   static char errmsg_errno[1024+1];
3370
3371   switch (err) {
3372
3373   case ENOENT:
3374     errmsg = "The path to the file \"%s\" doesn't exist.";
3375     break;
3376
3377   case EACCES:
3378     errmsg = "You don't have permission to move the capture file to \"%s\".";
3379     break;
3380
3381   default:
3382     snprintf(errmsg_errno, sizeof(errmsg_errno),
3383                     "The file \"%%s\" could not be moved: %s.",
3384                                 wtap_strerror(err));
3385     errmsg = errmsg_errno;
3386     break;
3387   }
3388   return errmsg;
3389 }
3390
3391 char *
3392 cf_read_error_message(int err, const gchar *err_info)
3393 {
3394   static char errmsg_errno[1024+1];
3395
3396   switch (err) {
3397
3398   case WTAP_ERR_UNSUPPORTED_ENCAP:
3399       snprintf(errmsg_errno, sizeof(errmsg_errno),
3400                "The file \"%%s\" has a packet with a network type that Ethereal doesn't support.\n(%s)",
3401                err_info);
3402       break;
3403
3404   case WTAP_ERR_BAD_RECORD:
3405     snprintf(errmsg_errno, sizeof(errmsg_errno),
3406              "An error occurred while reading from the file \"%%s\": %s.\n(%s)",
3407              wtap_strerror(err), err_info);
3408     break;
3409
3410   default:
3411     snprintf(errmsg_errno, sizeof(errmsg_errno),
3412              "An error occurred while reading from the file \"%%s\": %s.",
3413              wtap_strerror(err));
3414     break;
3415   }
3416   return errmsg_errno;
3417 }
3418
3419 static void
3420 cf_write_failure_alert_box(const char *filename, int err)
3421 {
3422   if (err < 0) {
3423     /* Wiretap error. */
3424     simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3425                   "An error occurred while writing to the file \"%s\": %s.",
3426                   filename, wtap_strerror(err));
3427   } else {
3428     /* OS error. */
3429     write_failure_alert_box(filename, err);
3430   }
3431 }
3432
3433 /* Check for write errors - if the file is being written to an NFS server,
3434    a write error may not show up until the file is closed, as NFS clients
3435    might not send writes to the server until the "write()" call finishes,
3436    so that the write may fail on the server but the "write()" may succeed. */
3437 static void
3438 cf_close_failure_alert_box(const char *filename, int err)
3439 {
3440   if (err < 0) {
3441     /* Wiretap error. */
3442     switch (err) {
3443
3444     case WTAP_ERR_CANT_CLOSE:
3445       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3446                     "The file \"%s\" couldn't be closed for some unknown reason.",
3447                     filename);
3448       break;
3449
3450     case WTAP_ERR_SHORT_WRITE:
3451       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3452                     "Not all the packets could be written to the file \"%s\".",
3453                     filename);
3454       break;
3455
3456     default:
3457       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3458                     "An error occurred while closing the file \"%s\": %s.",
3459                     filename, wtap_strerror(err));
3460       break;
3461     }
3462   } else {
3463     /* OS error.
3464        We assume that a close error from the OS is really a write error. */
3465     write_failure_alert_box(filename, err);
3466   }
3467 }
3468
3469 /* Reload the current capture file. */
3470 void
3471 cf_reload(capture_file *cf) {
3472   gchar *filename;
3473   gboolean is_tempfile;
3474   int err;
3475
3476   /* If the file could be opened, "cf_open()" calls "cf_close()"
3477      to get rid of state for the old capture file before filling in state
3478      for the new capture file.  "cf_close()" will remove the file if
3479      it's a temporary file; we don't want that to happen (for one thing,
3480      it'd prevent subsequent reopens from working).  Remember whether it's
3481      a temporary file, mark it as not being a temporary file, and then
3482      reopen it as the type of file it was.
3483
3484      Also, "cf_close()" will free "cf->filename", so we must make
3485      a copy of it first. */
3486   filename = g_strdup(cf->filename);
3487   is_tempfile = cf->is_tempfile;
3488   cf->is_tempfile = FALSE;
3489   if (cf_open(cf, filename, is_tempfile, &err) == CF_OK) {
3490     switch (cf_read(cf)) {
3491
3492     case CF_READ_OK:
3493     case CF_READ_ERROR:
3494       /* Just because we got an error, that doesn't mean we were unable
3495          to read any of the file; we handle what we could get from the
3496          file. */
3497       break;
3498
3499     case CF_READ_ABORTED:
3500       /* The user bailed out of re-reading the capture file; the
3501          capture file has been closed - just free the capture file name
3502          string and return (without changing the last containing
3503          directory). */
3504       g_free(filename);
3505       return;
3506     }
3507   } else {
3508     /* The open failed, so "cf->is_tempfile" wasn't set to "is_tempfile".
3509        Instead, the file was left open, so we should restore "cf->is_tempfile"
3510        ourselves.
3511
3512        XXX - change the menu?  Presumably "cf_open()" will do that;
3513        make sure it does! */
3514     cf->is_tempfile = is_tempfile;
3515   }
3516   /* "cf_open()" made a copy of the file name we handed it, so
3517      we should free up our copy. */
3518   g_free(filename);
3519 }
3520
3521 /* Copies a file in binary mode, for those operating systems that care about
3522  * such things.
3523  * Returns TRUE on success, FALSE on failure. If a failure, it also
3524  * displays a simple dialog window with the error message.
3525  */
3526 static gboolean
3527 copy_binary_file(const char *from_filename, const char *to_filename)
3528 {
3529   int           from_fd, to_fd, nread, nwritten, err;
3530   guint8        pd[65536];
3531
3532   /* Copy the raw bytes of the file. */
3533   from_fd = open(from_filename, O_RDONLY | O_BINARY);
3534   if (from_fd < 0) {
3535     open_failure_alert_box(from_filename, errno, FALSE);
3536     goto done;
3537   }
3538
3539   /* Use open() instead of creat() so that we can pass the O_BINARY
3540      flag, which is relevant on Win32; it appears that "creat()"
3541      may open the file in text mode, not binary mode, but we want
3542      to copy the raw bytes of the file, so we need the output file
3543      to be open in binary mode. */
3544   to_fd = open(to_filename, O_WRONLY | O_CREAT | O_TRUNC | O_BINARY, 0644);
3545   if (to_fd < 0) {
3546     open_failure_alert_box(to_filename, errno, TRUE);
3547     close(from_fd);
3548     goto done;
3549   }
3550
3551   while ((nread = read(from_fd, pd, sizeof pd)) > 0) {
3552     nwritten = write(to_fd, pd, nread);
3553     if (nwritten < nread) {
3554       if (nwritten < 0)
3555         err = errno;
3556       else
3557         err = WTAP_ERR_SHORT_WRITE;
3558       write_failure_alert_box(to_filename, err);
3559       close(from_fd);
3560       close(to_fd);
3561       goto done;
3562     }
3563   }
3564   if (nread < 0) {
3565     err = errno;
3566     simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3567                   "An error occurred while reading from the file \"%s\": %s.",
3568                   from_filename, strerror(err));
3569     close(from_fd);
3570     close(to_fd);
3571     goto done;
3572   }
3573   close(from_fd);
3574   if (close(to_fd) < 0) {
3575     write_failure_alert_box(to_filename, errno);
3576     goto done;
3577   }
3578
3579   return TRUE;
3580
3581 done:
3582   return FALSE;
3583 }