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