Fix Coverity CID's 1339 and 1340: UNUSED_VALUE.
[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_FILE:
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_FILE:
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_FILE:
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,
2712              const guint8 *pd, void *argsp)
2713 {
2714   FILE *fh = argsp;
2715   epan_dissect_t edt;
2716
2717   epan_dissect_init(&edt, TRUE, TRUE);
2718   epan_dissect_run(&edt, pseudo_header, pd, fdata, NULL);
2719   proto_tree_write_carrays(fdata->num, fh, &edt);
2720   epan_dissect_cleanup(&edt);
2721
2722   return !ferror(fh);
2723 }
2724
2725 cf_print_status_t
2726 cf_write_carrays_packets(capture_file *cf, print_args_t *print_args)
2727 {
2728   FILE        *fh;
2729   psp_return_t ret;
2730
2731   fh = ws_fopen(print_args->file, "w");
2732
2733   if (fh == NULL)
2734     return CF_PRINT_OPEN_ERROR; /* attempt to open destination failed */
2735
2736   write_carrays_preamble(fh);
2737
2738   if (ferror(fh)) {
2739     fclose(fh);
2740     return CF_PRINT_WRITE_ERROR;
2741   }
2742
2743   /* Iterate through the list of packets, printing the packets we were
2744      told to print. */
2745   ret = process_specified_packets(cf, &print_args->range,
2746                   "Writing C Arrays",
2747                   "selected packets", TRUE,
2748                                   write_carrays_packet, fh);
2749   switch (ret) {
2750   case PSP_FINISHED:
2751     /* Completed successfully. */
2752     break;
2753   case PSP_STOPPED:
2754     /* Well, the user decided to abort the printing. */
2755     break;
2756   case PSP_FAILED:
2757     /* Error while printing. */
2758     fclose(fh);
2759     return CF_PRINT_WRITE_ERROR;
2760   }
2761
2762   write_carrays_finale(fh);
2763
2764   if (ferror(fh)) {
2765     fclose(fh);
2766     return CF_PRINT_WRITE_ERROR;
2767   }
2768
2769   fclose(fh);
2770   return CF_PRINT_OK;
2771 }
2772
2773 gboolean
2774 cf_find_packet_protocol_tree(capture_file *cf, const char *string,
2775                              search_direction dir)
2776 {
2777   match_data        mdata;
2778
2779   mdata.string = string;
2780   mdata.string_len = strlen(string);
2781   return find_packet(cf, match_protocol_tree, &mdata, dir);
2782 }
2783
2784 gboolean
2785 cf_find_string_protocol_tree(capture_file *cf, proto_tree *tree,  match_data *mdata)
2786 {
2787   mdata->frame_matched = FALSE;
2788   mdata->string = convert_string_case(cf->sfilter, cf->case_type);
2789   mdata->string_len = strlen(mdata->string);
2790   mdata->cf = cf;
2791   /* Iterate through all the nodes looking for matching text */
2792   proto_tree_children_foreach(tree, match_subtree_text, mdata);
2793   return mdata->frame_matched ? MR_MATCHED : MR_NOTMATCHED;
2794 }
2795
2796 static match_result
2797 match_protocol_tree(capture_file *cf, frame_data *fdata, void *criterion)
2798 {
2799   match_data        *mdata = criterion;
2800   epan_dissect_t    edt;
2801
2802   /* Load the frame's data. */
2803   if (!cf_read_frame(cf, fdata)) {
2804     /* Attempt to get the packet failed. */
2805     return MR_ERROR;
2806   }
2807
2808   /* Construct the protocol tree, including the displayed text */
2809   epan_dissect_init(&edt, TRUE, TRUE);
2810   /* We don't need the column information */
2811   epan_dissect_run(&edt, &cf->pseudo_header, cf->pd, fdata, NULL);
2812
2813   /* Iterate through all the nodes, seeing if they have text that matches. */
2814   mdata->cf = cf;
2815   mdata->frame_matched = FALSE;
2816   proto_tree_children_foreach(edt.tree, match_subtree_text, mdata);
2817   epan_dissect_cleanup(&edt);
2818   return mdata->frame_matched ? MR_MATCHED : MR_NOTMATCHED;
2819 }
2820
2821 static void
2822 match_subtree_text(proto_node *node, gpointer data)
2823 {
2824   match_data    *mdata = (match_data*) data;
2825   const gchar   *string = mdata->string;
2826   size_t        string_len = mdata->string_len;
2827   capture_file  *cf = mdata->cf;
2828   field_info    *fi = PNODE_FINFO(node);
2829   gchar         label_str[ITEM_LABEL_LENGTH];
2830   gchar         *label_ptr;
2831   size_t        label_len;
2832   guint32       i;
2833   guint8        c_char;
2834   size_t        c_match = 0;
2835
2836   g_assert(fi && "dissection with an invisible proto tree?");
2837
2838   if (mdata->frame_matched) {
2839     /* We already had a match; don't bother doing any more work. */
2840     return;
2841   }
2842
2843   /* Don't match invisible entries. */
2844   if (PROTO_ITEM_IS_HIDDEN(node))
2845     return;
2846
2847   /* was a free format label produced? */
2848   if (fi->rep) {
2849     label_ptr = fi->rep->representation;
2850   } else {
2851     /* no, make a generic label */
2852     label_ptr = label_str;
2853     proto_item_fill_label(fi, label_str);
2854   }
2855
2856   /* Does that label match? */
2857   label_len = strlen(label_ptr);
2858   for (i = 0; i < label_len; i++) {
2859     c_char = label_ptr[i];
2860     if (cf->case_type)
2861       c_char = toupper(c_char);
2862     if (c_char == string[c_match]) {
2863       c_match++;
2864       if (c_match == string_len) {
2865         /* No need to look further; we have a match */
2866         mdata->frame_matched = TRUE;
2867         mdata->finfo = fi;
2868         return;
2869       }
2870     } else
2871       c_match = 0;
2872   }
2873
2874   /* Recurse into the subtree, if it exists */
2875   if (node->first_child != NULL)
2876     proto_tree_children_foreach(node, match_subtree_text, mdata);
2877 }
2878
2879 gboolean
2880 cf_find_packet_summary_line(capture_file *cf, const char *string,
2881                             search_direction dir)
2882 {
2883   match_data        mdata;
2884
2885   mdata.string = string;
2886   mdata.string_len = strlen(string);
2887   return find_packet(cf, match_summary_line, &mdata, dir);
2888 }
2889
2890 static match_result
2891 match_summary_line(capture_file *cf, frame_data *fdata, void *criterion)
2892 {
2893   match_data        *mdata = criterion;
2894   const gchar       *string = mdata->string;
2895   size_t            string_len = mdata->string_len;
2896   epan_dissect_t    edt;
2897   const char        *info_column;
2898   size_t            info_column_len;
2899   match_result      result = MR_NOTMATCHED;
2900   gint              colx;
2901   guint32           i;
2902   guint8            c_char;
2903   size_t            c_match = 0;
2904
2905   /* Load the frame's data. */
2906   if (!cf_read_frame(cf, fdata)) {
2907     /* Attempt to get the packet failed. */
2908     return MR_ERROR;
2909   }
2910
2911   /* Don't bother constructing the protocol tree */
2912   epan_dissect_init(&edt, FALSE, FALSE);
2913   /* Get the column information */
2914   epan_dissect_run(&edt, &cf->pseudo_header, cf->pd, fdata, &cf->cinfo);
2915
2916   /* Find the Info column */
2917   for (colx = 0; colx < cf->cinfo.num_cols; colx++) {
2918     if (cf->cinfo.fmt_matx[colx][COL_INFO]) {
2919       /* Found it.  See if we match. */
2920       info_column = edt.pi.cinfo->col_data[colx];
2921       info_column_len = strlen(info_column);
2922       for (i = 0; i < info_column_len; i++) {
2923         c_char = info_column[i];
2924         if (cf->case_type)
2925           c_char = toupper(c_char);
2926         if (c_char == string[c_match]) {
2927           c_match++;
2928           if (c_match == string_len) {
2929             result = MR_MATCHED;
2930             break;
2931           }
2932         } else
2933           c_match = 0;
2934       }
2935       break;
2936     }
2937   }
2938   epan_dissect_cleanup(&edt);
2939   return result;
2940 }
2941
2942 typedef struct {
2943     const guint8 *data;
2944     size_t data_len;
2945 } cbs_t;    /* "Counted byte string" */
2946
2947 gboolean
2948 cf_find_packet_data(capture_file *cf, const guint8 *string, size_t string_size,
2949                     search_direction dir)
2950 {
2951   cbs_t info;
2952
2953   info.data = string;
2954   info.data_len = string_size;
2955
2956   /* String or hex search? */
2957   if (cf->string) {
2958     /* String search - what type of string? */
2959     switch (cf->scs_type) {
2960
2961     case SCS_ASCII_AND_UNICODE:
2962       return find_packet(cf, match_ascii_and_unicode, &info, dir);
2963
2964     case SCS_ASCII:
2965       return find_packet(cf, match_ascii, &info, dir);
2966
2967     case SCS_UNICODE:
2968       return find_packet(cf, match_unicode, &info, dir);
2969
2970     default:
2971       g_assert_not_reached();
2972       return FALSE;
2973     }
2974   } else
2975     return find_packet(cf, match_binary, &info, dir);
2976 }
2977
2978 static match_result
2979 match_ascii_and_unicode(capture_file *cf, frame_data *fdata, void *criterion)
2980 {
2981   cbs_t        *info = criterion;
2982   const guint8 *ascii_text = info->data;
2983   size_t       textlen = info->data_len;
2984   match_result result;
2985   guint32      buf_len;
2986   guint32      i;
2987   guint8       c_char;
2988   size_t       c_match = 0;
2989
2990   /* Load the frame's data. */
2991   if (!cf_read_frame(cf, fdata)) {
2992     /* Attempt to get the packet failed. */
2993     return MR_ERROR;
2994   }
2995
2996   result = MR_NOTMATCHED;
2997   buf_len = fdata->pkt_len;
2998   for (i = 0; i < buf_len; i++) {
2999     c_char = cf->pd[i];
3000     if (cf->case_type)
3001       c_char = toupper(c_char);
3002     if (c_char != 0) {
3003       if (c_char == ascii_text[c_match]) {
3004         c_match++;
3005         if (c_match == textlen) {
3006           result = MR_MATCHED;
3007           cf->search_pos = i; /* Save the position of the last character
3008                                  for highlighting the field. */
3009           break;
3010         }
3011       } else
3012         c_match = 0;
3013     }
3014   }
3015   return result;
3016 }
3017
3018 static match_result
3019 match_ascii(capture_file *cf, frame_data *fdata, void *criterion)
3020 {
3021   cbs_t        *info = criterion;
3022   const guint8 *ascii_text = info->data;
3023   size_t       textlen = info->data_len;
3024   match_result result;
3025   guint32      buf_len;
3026   guint32      i;
3027   guint8       c_char;
3028   size_t       c_match = 0;
3029
3030   /* Load the frame's data. */
3031   if (!cf_read_frame(cf, fdata)) {
3032     /* Attempt to get the packet failed. */
3033     return MR_ERROR;
3034   }
3035
3036   result = MR_NOTMATCHED;
3037   buf_len = fdata->pkt_len;
3038   for (i = 0; i < buf_len; i++) {
3039     c_char = cf->pd[i];
3040     if (cf->case_type)
3041       c_char = toupper(c_char);
3042     if (c_char == ascii_text[c_match]) {
3043       c_match++;
3044       if (c_match == textlen) {
3045         result = MR_MATCHED;
3046         cf->search_pos = i; /* Save the position of the last character
3047                                for highlighting the field. */
3048         break;
3049       }
3050     } else
3051       c_match = 0;
3052   }
3053   return result;
3054 }
3055
3056 static match_result
3057 match_unicode(capture_file *cf, frame_data *fdata, void *criterion)
3058 {
3059   cbs_t        *info = criterion;
3060   const guint8 *ascii_text = info->data;
3061   size_t       textlen = info->data_len;
3062   match_result result;
3063   guint32      buf_len;
3064   guint32      i;
3065   guint8       c_char;
3066   size_t       c_match = 0;
3067
3068   /* Load the frame's data. */
3069   if (!cf_read_frame(cf, fdata)) {
3070     /* Attempt to get the packet failed. */
3071     return MR_ERROR;
3072   }
3073
3074   result = MR_NOTMATCHED;
3075   buf_len = fdata->pkt_len;
3076   for (i = 0; i < buf_len; i++) {
3077     c_char = cf->pd[i];
3078     if (cf->case_type)
3079       c_char = toupper(c_char);
3080     if (c_char == ascii_text[c_match]) {
3081       c_match++;
3082       i++;
3083       if (c_match == textlen) {
3084         result = MR_MATCHED;
3085         cf->search_pos = i; /* Save the position of the last character
3086                                for highlighting the field. */
3087         break;
3088       }
3089     } else
3090       c_match = 0;
3091   }
3092   return result;
3093 }
3094
3095 static match_result
3096 match_binary(capture_file *cf, frame_data *fdata, void *criterion)
3097 {
3098   cbs_t        *info = criterion;
3099   const guint8 *binary_data = info->data;
3100   size_t       datalen = info->data_len;
3101   match_result result;
3102   guint32      buf_len;
3103   guint32      i;
3104   size_t       c_match = 0;
3105
3106   /* Load the frame's data. */
3107   if (!cf_read_frame(cf, fdata)) {
3108     /* Attempt to get the packet failed. */
3109     return MR_ERROR;
3110   }
3111
3112   result = MR_NOTMATCHED;
3113   buf_len = fdata->pkt_len;
3114   for (i = 0; i < buf_len; i++) {
3115     if (cf->pd[i] == binary_data[c_match]) {
3116       c_match++;
3117       if (c_match == datalen) {
3118         result = MR_MATCHED;
3119         cf->search_pos = i; /* Save the position of the last character
3120                                for highlighting the field. */
3121         break;
3122       }
3123     } else
3124       c_match = 0;
3125   }
3126   return result;
3127 }
3128
3129 gboolean
3130 cf_find_packet_dfilter(capture_file *cf, dfilter_t *sfcode,
3131                        search_direction dir)
3132 {
3133   return find_packet(cf, match_dfilter, sfcode, dir);
3134 }
3135
3136 gboolean
3137 cf_find_packet_dfilter_string(capture_file *cf, const char *filter,
3138                               search_direction dir)
3139 {
3140   dfilter_t *sfcode;
3141   gboolean result;
3142
3143   if (!dfilter_compile(filter, &sfcode)) {
3144      /*
3145       * XXX - this shouldn't happen, as the filter string is machine
3146       * generated
3147       */
3148     return FALSE;
3149   }
3150   if (sfcode == NULL) {
3151     /*
3152      * XXX - this shouldn't happen, as the filter string is machine
3153      * generated.
3154      */
3155     return FALSE;
3156   }
3157   result = find_packet(cf, match_dfilter, sfcode, dir);
3158   dfilter_free(sfcode);
3159   return result;
3160 }
3161
3162 static match_result
3163 match_dfilter(capture_file *cf, frame_data *fdata, void *criterion)
3164 {
3165   dfilter_t      *sfcode = criterion;
3166   epan_dissect_t edt;
3167   match_result   result;
3168
3169   /* Load the frame's data. */
3170   if (!cf_read_frame(cf, fdata)) {
3171     /* Attempt to get the packet failed. */
3172     return MR_ERROR;
3173   }
3174
3175   epan_dissect_init(&edt, TRUE, FALSE);
3176   epan_dissect_prime_dfilter(&edt, sfcode);
3177   epan_dissect_run(&edt, &cf->pseudo_header, cf->pd, fdata, NULL);
3178   result = dfilter_apply_edt(sfcode, &edt) ? MR_MATCHED : MR_NOTMATCHED;
3179   epan_dissect_cleanup(&edt);
3180   return result;
3181 }
3182
3183 gboolean
3184 cf_find_packet_marked(capture_file *cf, search_direction dir)
3185 {
3186   return find_packet(cf, match_marked, NULL, dir);
3187 }
3188
3189 static match_result
3190 match_marked(capture_file *cf _U_, frame_data *fdata, void *criterion _U_)
3191 {
3192   return fdata->flags.marked ? MR_MATCHED : MR_NOTMATCHED;
3193 }
3194
3195 gboolean
3196 cf_find_packet_time_reference(capture_file *cf, search_direction dir)
3197 {
3198   return find_packet(cf, match_time_reference, NULL, dir);
3199 }
3200
3201 static match_result
3202 match_time_reference(capture_file *cf _U_, frame_data *fdata, void *criterion _U_)
3203 {
3204   return fdata->flags.ref_time ? MR_MATCHED : MR_NOTMATCHED;
3205 }
3206
3207 static gboolean
3208 find_packet(capture_file *cf,
3209             match_result (*match_function)(capture_file *, frame_data *, void *),
3210             void *criterion, search_direction dir)
3211 {
3212   frame_data  *start_fd;
3213   guint32      framenum;
3214   frame_data  *fdata;
3215   frame_data  *new_fd = NULL;
3216   progdlg_t   *progbar = NULL;
3217   gboolean     stop_flag;
3218   int          count;
3219   gboolean     found;
3220   float        progbar_val;
3221   GTimeVal     start_time;
3222   gchar        status_str[100];
3223   int          progbar_nextstep;
3224   int          progbar_quantum;
3225   const char  *title;
3226   match_result result;
3227
3228   start_fd = cf->current_frame;
3229   if (start_fd != NULL)  {
3230     /* Iterate through the list of packets, starting at the packet we've
3231        picked, calling a routine to run the filter on the packet, see if
3232        it matches, and stop if so.  */
3233     count = 0;
3234     framenum = start_fd->num;
3235
3236     /* Update the progress bar when it gets to this value. */
3237     progbar_nextstep = 0;
3238     /* When we reach the value that triggers a progress bar update,
3239        bump that value by this amount. */
3240     progbar_quantum = cf->count/N_PROGBAR_UPDATES;
3241     /* Progress so far. */
3242     progbar_val = 0.0f;
3243
3244     stop_flag = FALSE;
3245     g_get_current_time(&start_time);
3246
3247     title = cf->sfilter?cf->sfilter:"";
3248     for (;;) {
3249       /* Create the progress bar if necessary.
3250          We check on every iteration of the loop, so that it takes no
3251          longer than the standard time to create it (otherwise, for a
3252          large file, we might take considerably longer than that standard
3253          time in order to get to the next progress bar step). */
3254       if (progbar == NULL)
3255          progbar = delayed_create_progress_dlg("Searching", title,
3256            FALSE, &stop_flag, &start_time, progbar_val);
3257
3258       /* Update the progress bar, but do it only N_PROGBAR_UPDATES times;
3259          when we update it, we have to run the GTK+ main loop to get it
3260          to repaint what's pending, and doing so may involve an "ioctl()"
3261          to see if there's any pending input from an X server, and doing
3262          that for every packet can be costly, especially on a big file. */
3263       if (count >= progbar_nextstep) {
3264         /* let's not divide by zero. I should never be started
3265          * with count == 0, so let's assert that
3266          */
3267         g_assert(cf->count > 0);
3268
3269         progbar_val = (gfloat) count / cf->count;
3270
3271         if (progbar != NULL) {
3272           g_snprintf(status_str, sizeof(status_str),
3273                      "%4u of %u packets", count, cf->count);
3274           update_progress_dlg(progbar, progbar_val, status_str);
3275         }
3276
3277         progbar_nextstep += progbar_quantum;
3278       }
3279
3280       if (stop_flag) {
3281         /* Well, the user decided to abort the search.  Go back to the
3282            frame where we started. */
3283         new_fd = start_fd;
3284         break;
3285       }
3286
3287       /* Go past the current frame. */
3288       if (dir == SD_BACKWARD) {
3289         /* Go on to the previous frame. */
3290         if (framenum == 1) {
3291           /*
3292            * XXX - other apps have a bit more of a detailed message
3293            * for this, and instead of offering "OK" and "Cancel",
3294            * they offer things such as "Continue" and "Cancel";
3295            * we need an API for popping up alert boxes with
3296            * {Verb} and "Cancel".
3297            */
3298
3299           if (prefs.gui_find_wrap)
3300           {
3301               statusbar_push_temporary_msg("Search reached the beginning. Continuing at end.");
3302               framenum = cf->count;     /* wrap around */
3303           }
3304           else
3305           {
3306               statusbar_push_temporary_msg("Search reached the beginning.");
3307               framenum = start_fd->num; /* stay on previous packet */
3308           }
3309         } else
3310           framenum--;
3311       } else {
3312         /* Go on to the next frame. */
3313         if (framenum == cf->count) {
3314           if (prefs.gui_find_wrap)
3315           {
3316               statusbar_push_temporary_msg("Search reached the end. Continuing at beginning.");
3317               framenum = 1;             /* wrap around */
3318           }
3319           else
3320           {
3321               statusbar_push_temporary_msg("Search reached the end.");
3322               framenum = start_fd->num; /* stay on previous packet */
3323           }
3324         } else
3325           framenum++;
3326       }
3327       fdata = frame_data_sequence_find(cf->frames, framenum);
3328
3329       count++;
3330
3331       /* Is this packet in the display? */
3332       if (fdata->flags.passed_dfilter) {
3333         /* Yes.  Does it match the search criterion? */
3334         result = (*match_function)(cf, fdata, criterion);
3335         if (result == MR_ERROR) {
3336           /* Error; our caller has reported the error.  Go back to the frame
3337              where we started. */
3338           new_fd = start_fd;
3339           break;
3340         } else if (result == MR_MATCHED) {
3341           /* Yes.  Go to the new frame. */
3342           new_fd = fdata;
3343           break;
3344         }
3345       }
3346
3347       if (fdata == start_fd) {
3348         /* We're back to the frame we were on originally, and that frame
3349            doesn't match the search filter.  The search failed. */
3350         break;
3351       }
3352     }
3353
3354     /* We're done scanning the packets; destroy the progress bar if it
3355        was created. */
3356     if (progbar != NULL)
3357       destroy_progress_dlg(progbar);
3358   }
3359
3360   if (new_fd != NULL) {
3361     /* Find and select */
3362     cf->search_in_progress = TRUE;
3363     found = new_packet_list_select_row_from_data(new_fd);
3364     cf->search_in_progress = FALSE;
3365     cf->search_pos = 0; /* Reset the position */
3366     if (!found) {
3367       /* We didn't find a row corresponding to this frame.
3368          This means that the frame isn't being displayed currently,
3369          so we can't select it. */
3370       simple_dialog(ESD_TYPE_INFO, ESD_BTN_OK,
3371                     "%sEnd of capture exceeded!%s\n\n"
3372                     "The capture file is probably not fully dissected.",
3373                     simple_dialog_primary_start(), simple_dialog_primary_end());
3374       return FALSE;
3375     }
3376     return TRUE;    /* success */
3377   } else
3378     return FALSE;   /* failure */
3379 }
3380
3381 gboolean
3382 cf_goto_frame(capture_file *cf, guint fnumber)
3383 {
3384   frame_data *fdata;
3385
3386   fdata = frame_data_sequence_find(cf->frames, fnumber);
3387
3388   if (fdata == NULL) {
3389     /* we didn't find a packet with that packet number */
3390     statusbar_push_temporary_msg("There is no packet number %u.", fnumber);
3391     return FALSE;   /* we failed to go to that packet */
3392   }
3393   if (!fdata->flags.passed_dfilter) {
3394     /* that packet currently isn't displayed */
3395     /* XXX - add it to the set of displayed packets? */
3396     statusbar_push_temporary_msg("Packet number %u isn't displayed.", fnumber);
3397     return FALSE;   /* we failed to go to that packet */
3398   }
3399
3400   if (!new_packet_list_select_row_from_data(fdata)) {
3401     /* We didn't find a row corresponding to this frame.
3402        This means that the frame isn't being displayed currently,
3403        so we can't select it. */
3404     simple_dialog(ESD_TYPE_INFO, ESD_BTN_OK,
3405                   "%sEnd of capture exceeded!%s\n\n"
3406                   "The capture file is probably not fully dissected.",
3407                   simple_dialog_primary_start(), simple_dialog_primary_end());
3408     return FALSE;
3409   }
3410   return TRUE;  /* we got to that packet */
3411 }
3412
3413 gboolean
3414 cf_goto_top_frame(void)
3415 {
3416   /* Find and select */
3417   new_packet_list_select_first_row();
3418   return TRUE;  /* we got to that packet */
3419 }
3420
3421 gboolean
3422 cf_goto_bottom_frame(void)
3423 {
3424   /* Find and select */
3425   new_packet_list_select_last_row();
3426   return TRUE;  /* we got to that packet */
3427 }
3428
3429 /*
3430  * Go to frame specified by currently selected protocol tree item.
3431  */
3432 gboolean
3433 cf_goto_framenum(capture_file *cf)
3434 {
3435   header_field_info       *hfinfo;
3436   guint32                 framenum;
3437
3438   if (cf->finfo_selected) {
3439     hfinfo = cf->finfo_selected->hfinfo;
3440     g_assert(hfinfo);
3441     if (hfinfo->type == FT_FRAMENUM) {
3442       framenum = fvalue_get_uinteger(&cf->finfo_selected->value);
3443       if (framenum != 0)
3444         return cf_goto_frame(cf, framenum);
3445       }
3446   }
3447
3448   return FALSE;
3449 }
3450
3451 /* Select the packet on a given row. */
3452 void
3453 cf_select_packet(capture_file *cf, int row)
3454 {
3455   frame_data *fdata;
3456
3457   /* Get the frame data struct pointer for this frame */
3458   fdata = new_packet_list_get_row_data(row);
3459
3460   if (fdata == NULL) {
3461     /* XXX - if a GtkCList's selection mode is GTK_SELECTION_BROWSE, when
3462        the first entry is added to it by "real_insert_row()", that row
3463        is selected (see "real_insert_row()", in "gtk/gtkclist.c", in both
3464        our version and the vanilla GTK+ version).
3465
3466        This means that a "select-row" signal is emitted; this causes
3467        "packet_list_select_cb()" to be called, which causes "cf_select_packet()"
3468        to be called.
3469
3470        "cf_select_packet()" fetches, above, the data associated with the
3471        row that was selected; however, as "gtk_clist_append()", which
3472        called "real_insert_row()", hasn't yet returned, we haven't yet
3473        associated any data with that row, so we get back a null pointer.
3474
3475        We can't assume that there's only one frame in the frame list,
3476        either, as we may be filtering the display.
3477
3478        We therefore assume that, if "row" is 0, i.e. the first row
3479        is being selected, and "cf->first_displayed" equals
3480        "cf->last_displayed", i.e. there's only one frame being
3481        displayed, that frame is the frame we want.
3482
3483        This means we have to set "cf->first_displayed" and
3484        "cf->last_displayed" before adding the row to the
3485        GtkCList; see the comment in "add_packet_to_packet_list()". */
3486
3487        if (row == 0 && cf->first_displayed == cf->last_displayed)
3488          fdata = frame_data_sequence_find(cf->frames, cf->first_displayed);
3489   }
3490
3491   /* If fdata _still_ isn't set simply give up. */
3492   if (fdata == NULL) {
3493     return;
3494   }
3495
3496   /* Get the data in that frame. */
3497   if (!cf_read_frame (cf, fdata)) {
3498     return;
3499   }
3500
3501   /* Record that this frame is the current frame. */
3502   cf->current_frame = fdata;
3503   cf->current_row = row;
3504
3505   /* Create the logical protocol tree. */
3506   if (cf->edt != NULL)
3507     epan_dissect_free(cf->edt);
3508
3509   /* We don't need the columns here. */
3510   cf->edt = epan_dissect_new(TRUE, TRUE);
3511
3512   tap_build_interesting(cf->edt);
3513   epan_dissect_run(cf->edt, &cf->pseudo_header, cf->pd, cf->current_frame,
3514           NULL);
3515
3516   dfilter_macro_build_ftv_cache(cf->edt->tree);
3517
3518   cf_callback_invoke(cf_cb_packet_selected, cf);
3519 }
3520
3521 /* Unselect the selected packet, if any. */
3522 void
3523 cf_unselect_packet(capture_file *cf)
3524 {
3525   /* Destroy the epan_dissect_t for the unselected packet. */
3526   if (cf->edt != NULL) {
3527     epan_dissect_free(cf->edt);
3528     cf->edt = NULL;
3529   }
3530
3531   /* No packet is selected. */
3532   cf->current_frame = NULL;
3533   cf->current_row = 0;
3534
3535   cf_callback_invoke(cf_cb_packet_unselected, cf);
3536
3537   /* No protocol tree means no selected field. */
3538   cf_unselect_field(cf);
3539 }
3540
3541 /* Unset the selected protocol tree field, if any. */
3542 void
3543 cf_unselect_field(capture_file *cf)
3544 {
3545   cf->finfo_selected = NULL;
3546
3547   cf_callback_invoke(cf_cb_field_unselected, cf);
3548 }
3549
3550 /*
3551  * Mark a particular frame.
3552  */
3553 void
3554 cf_mark_frame(capture_file *cf, frame_data *frame)
3555 {
3556   if (! frame->flags.marked) {
3557     frame->flags.marked = TRUE;
3558     if (cf->count > cf->marked_count)
3559       cf->marked_count++;
3560   }
3561 }
3562
3563 /*
3564  * Unmark a particular frame.
3565  */
3566 void
3567 cf_unmark_frame(capture_file *cf, frame_data *frame)
3568 {
3569   if (frame->flags.marked) {
3570     frame->flags.marked = FALSE;
3571     if (cf->marked_count > 0)
3572       cf->marked_count--;
3573   }
3574 }
3575
3576 /*
3577  * Ignore a particular frame.
3578  */
3579 void
3580 cf_ignore_frame(capture_file *cf, frame_data *frame)
3581 {
3582   if (! frame->flags.ignored) {
3583     frame->flags.ignored = TRUE;
3584     if (cf->count > cf->ignored_count)
3585       cf->ignored_count++;
3586   }
3587 }
3588
3589 /*
3590  * Un-ignore a particular frame.
3591  */
3592 void
3593 cf_unignore_frame(capture_file *cf, frame_data *frame)
3594 {
3595   if (frame->flags.ignored) {
3596     frame->flags.ignored = FALSE;
3597     if (cf->ignored_count > 0)
3598       cf->ignored_count--;
3599   }
3600 }
3601
3602 typedef struct {
3603   wtap_dumper *pdh;
3604   const char  *fname;
3605   int          file_type;
3606 } save_callback_args_t;
3607
3608 /*
3609  * Save a capture to a file, in a particular format, saving either
3610  * all packets, all currently-displayed packets, or all marked packets.
3611  *
3612  * Returns TRUE if it succeeds, FALSE otherwise; if it fails, it pops
3613  * up a message box for the failure.
3614  */
3615 static gboolean
3616 save_packet(capture_file *cf _U_, frame_data *fdata,
3617             union wtap_pseudo_header *pseudo_header, const guint8 *pd,
3618             void *argsp)
3619 {
3620   save_callback_args_t *args = argsp;
3621   struct wtap_pkthdr hdr;
3622   int           err;
3623
3624   /* init the wtap header for saving */
3625   hdr.ts.secs    = fdata->abs_ts.secs;
3626   hdr.ts.nsecs   = fdata->abs_ts.nsecs;
3627   hdr.caplen     = fdata->cap_len;
3628   hdr.len        = fdata->pkt_len;
3629   hdr.pkt_encap  = fdata->lnk_t;
3630
3631   /* and save the packet */
3632   if (!wtap_dump(args->pdh, &hdr, pseudo_header, pd, &err)) {
3633     if (err < 0) {
3634       /* Wiretap error. */
3635       switch (err) {
3636
3637       case WTAP_ERR_UNSUPPORTED_ENCAP:
3638         /*
3639          * This is a problem with the particular frame we're writing;
3640          * note that, and give the frame number.
3641          */
3642         simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3643                       "Frame %u has a network type that can't be saved in a \"%s\" file.",
3644                       fdata->num, wtap_file_type_string(args->file_type));
3645         break;
3646
3647       default:
3648         simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3649                       "An error occurred while writing to the file \"%s\": %s.",
3650                       args->fname, wtap_strerror(err));
3651         break;
3652       }
3653     } else {
3654       /* OS error. */
3655       write_failure_alert_box(args->fname, err);
3656     }
3657     return FALSE;
3658   }
3659   return TRUE;
3660 }
3661
3662 /*
3663  * Can this capture file be saved in any format except by copying the raw data?
3664  */
3665 gboolean
3666 cf_can_save_as(capture_file *cf)
3667 {
3668   int ft;
3669
3670   for (ft = 0; ft < WTAP_NUM_FILE_TYPES; ft++) {
3671     /* To save a file with Wiretap, Wiretap has to handle that format,
3672        and its code to handle that format must be able to write a file
3673        with this file's encapsulation type. */
3674     if (wtap_dump_can_open(ft) && wtap_dump_can_write_encap(ft, cf->lnk_t)) {
3675       /* OK, we can write it out in this type. */
3676       return TRUE;
3677     }
3678   }
3679
3680   /* No, we couldn't save it in any format. */
3681   return FALSE;
3682 }
3683
3684 cf_status_t
3685 cf_save(capture_file *cf, const char *fname, packet_range_t *range, guint save_format, gboolean compressed)
3686 {
3687   gchar        *from_filename;
3688   int           err;
3689   gboolean      do_copy;
3690   wtap_dumper  *pdh;
3691   save_callback_args_t callback_args;
3692
3693   cf_callback_invoke(cf_cb_file_save_started, (gpointer)fname);
3694
3695   /* don't write over an existing file. */
3696   /* this should've been already checked by our caller, just to be sure... */
3697   if (file_exists(fname)) {
3698     simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3699       "%sCapture file: \"%s\" already exists!%s\n\n"
3700       "Please choose a different filename.",
3701       simple_dialog_primary_start(), fname, simple_dialog_primary_end());
3702     goto fail;
3703   }
3704
3705   packet_range_process_init(range);
3706
3707   if (packet_range_process_all(range) && save_format == cf->cd_t) {
3708     /* We're not filtering packets, and we're saving it in the format
3709        it's already in, so we can just move or copy the raw data. */
3710
3711     if (cf->is_tempfile) {
3712       /* The file being saved is a temporary file from a live
3713          capture, so it doesn't need to stay around under that name;
3714          first, try renaming the capture buffer file to the new name. */
3715 #ifndef _WIN32
3716       if (ws_rename(cf->filename, fname) == 0) {
3717         /* That succeeded - there's no need to copy the source file. */
3718         from_filename = NULL;
3719     do_copy = FALSE;
3720       } else {
3721         if (errno == EXDEV) {
3722           /* They're on different file systems, so we have to copy the
3723              file. */
3724           do_copy = TRUE;
3725           from_filename = cf->filename;
3726         } else {
3727           /* The rename failed, but not because they're on different
3728              file systems - put up an error message.  (Or should we
3729              just punt and try to copy?  The only reason why I'd
3730              expect the rename to fail and the copy to succeed would
3731              be if we didn't have permission to remove the file from
3732              the temporary directory, and that might be fixable - but
3733              is it worth requiring the user to go off and fix it?) */
3734           simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3735                         file_rename_error_message(errno), fname);
3736           goto fail;
3737         }
3738       }
3739 #else
3740       do_copy = TRUE;
3741       from_filename = cf->filename;
3742 #endif
3743     } else {
3744       /* It's a permanent file, so we should copy it, and not remove the
3745          original. */
3746       do_copy = TRUE;
3747       from_filename = cf->filename;
3748     }
3749
3750     if (do_copy) {
3751       /* Copy the file, if we haven't moved it. */
3752       if (!copy_file_binary_mode(from_filename, fname))
3753         goto fail;
3754     }
3755   } else {
3756     /* Either we're filtering packets, or we're saving in a different
3757        format; we can't do that by copying or moving the capture file,
3758        we have to do it by writing the packets out in Wiretap. */
3759     pdh = wtap_dump_open(fname, save_format, cf->lnk_t, cf->snap,
3760         compressed, &err);
3761     if (pdh == NULL) {
3762       cf_open_failure_alert_box(fname, err, NULL, TRUE, save_format);
3763       goto fail;
3764     }
3765
3766     /* XXX - we let the user save a subset of the packets.
3767
3768        If we do that, should we make that file the current file?  If so,
3769        it means we can no longer get at the other packets.  What does
3770        NetMon do? */
3771
3772     /* Iterate through the list of packets, processing the packets we were
3773        told to process.
3774
3775        XXX - we've already called "packet_range_process_init(range)", but
3776        "process_specified_packets()" will do it again.  Fortunately,
3777        that's harmless in this case, as we haven't done anything to
3778        "range" since we initialized it. */
3779     callback_args.pdh = pdh;
3780     callback_args.fname = fname;
3781     callback_args.file_type = save_format;
3782     switch (process_specified_packets(cf, range, "Saving", "selected packets",
3783                                       TRUE, save_packet, &callback_args)) {
3784
3785     case PSP_FINISHED:
3786       /* Completed successfully. */
3787       break;
3788
3789     case PSP_STOPPED:
3790       /* The user decided to abort the saving.
3791          XXX - remove the output file? */
3792       break;
3793
3794     case PSP_FAILED:
3795       /* Error while saving. */
3796       wtap_dump_close(pdh, &err);
3797       goto fail;
3798     }
3799
3800     if (!wtap_dump_close(pdh, &err)) {
3801       cf_close_failure_alert_box(fname, err);
3802       goto fail;
3803     }
3804   }
3805
3806   cf_callback_invoke(cf_cb_file_save_finished, NULL);
3807
3808   if (packet_range_process_all(range)) {
3809     /* We saved the entire capture, not just some packets from it.
3810        Open and read the file we saved it to.
3811
3812        XXX - this is somewhat of a waste; we already have the
3813        packets, all this gets us is updated file type information
3814        (which we could just stuff into "cf"), and having the new
3815        file be the one we have opened and from which we're reading
3816        the data, and it means we have to spend time opening and
3817        reading the file, which could be a significant amount of
3818        time if the file is large. */
3819     cf->user_saved = TRUE;
3820
3821     if ((cf_open(cf, fname, FALSE, &err)) == CF_OK) {
3822       /* XXX - report errors if this fails?
3823          What should we return if it fails or is aborted? */
3824
3825       switch (cf_read(cf, TRUE)) {
3826
3827       case CF_READ_OK:
3828       case CF_READ_ERROR:
3829     /* Just because we got an error, that doesn't mean we were unable
3830        to read any of the file; we handle what we could get from the
3831        file. */
3832     break;
3833
3834       case CF_READ_ABORTED:
3835     /* The user bailed out of re-reading the capture file; the
3836        capture file has been closed - just return (without
3837        changing any menu settings; "cf_close()" set them
3838        correctly for the "no capture file open" state). */
3839     break;
3840       }
3841       cf_callback_invoke(cf_cb_file_save_reload_finished, cf);
3842     }
3843   }
3844   return CF_OK;
3845
3846 fail:
3847   cf_callback_invoke(cf_cb_file_save_failed, NULL);
3848   return CF_ERROR;
3849 }
3850
3851 static void
3852 cf_open_failure_alert_box(const char *filename, int err, gchar *err_info,
3853                           gboolean for_writing, int file_type)
3854 {
3855   if (err < 0) {
3856     /* Wiretap error. */
3857     switch (err) {
3858
3859     case WTAP_ERR_NOT_REGULAR_FILE:
3860       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3861             "The file \"%s\" is a \"special file\" or socket or other non-regular file.",
3862             filename);
3863       break;
3864
3865     case WTAP_ERR_RANDOM_OPEN_PIPE:
3866       /* Seen only when opening a capture file for reading. */
3867       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3868             "The file \"%s\" is a pipe or FIFO; Wireshark can't read pipe or FIFO files.",
3869             filename);
3870       break;
3871
3872     case WTAP_ERR_FILE_UNKNOWN_FORMAT:
3873       /* Seen only when opening a capture file for reading. */
3874       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3875             "The file \"%s\" isn't a capture file in a format Wireshark understands.",
3876             filename);
3877       break;
3878
3879     case WTAP_ERR_UNSUPPORTED:
3880       /* Seen only when opening a capture file for reading. */
3881       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3882             "The file \"%s\" isn't a capture file in a format Wireshark understands.\n"
3883             "(%s)",
3884             filename, err_info);
3885       g_free(err_info);
3886       break;
3887
3888     case WTAP_ERR_CANT_WRITE_TO_PIPE:
3889       /* Seen only when opening a capture file for writing. */
3890       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3891             "The file \"%s\" is a pipe, and %s capture files can't be "
3892             "written to a pipe.",
3893             filename, wtap_file_type_string(file_type));
3894       break;
3895
3896     case WTAP_ERR_UNSUPPORTED_FILE_TYPE:
3897       /* Seen only when opening a capture file for writing. */
3898       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3899             "Wireshark doesn't support writing capture files in that format.");
3900       break;
3901
3902     case WTAP_ERR_UNSUPPORTED_ENCAP:
3903       if (for_writing) {
3904         simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3905               "Wireshark can't save this capture in that format.");
3906       } else {
3907         simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3908               "The file \"%s\" is a capture for a network type that Wireshark doesn't support.\n"
3909               "(%s)",
3910               filename, err_info);
3911         g_free(err_info);
3912       }
3913       break;
3914
3915     case WTAP_ERR_ENCAP_PER_PACKET_UNSUPPORTED:
3916       if (for_writing) {
3917         simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3918               "Wireshark can't save this capture in that format.");
3919       } else {
3920         simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3921               "The file \"%s\" is a capture for a network type that Wireshark doesn't support.",
3922               filename);
3923       }
3924       break;
3925
3926     case WTAP_ERR_BAD_FILE:
3927       /* Seen only when opening a capture file for reading. */
3928       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3929             "The file \"%s\" appears to be damaged or corrupt.\n"
3930             "(%s)",
3931             filename, err_info);
3932       g_free(err_info);
3933       break;
3934
3935     case WTAP_ERR_CANT_OPEN:
3936       if (for_writing) {
3937         simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3938               "The file \"%s\" could not be created for some unknown reason.",
3939               filename);
3940       } else {
3941         simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3942               "The file \"%s\" could not be opened for some unknown reason.",
3943               filename);
3944       }
3945       break;
3946
3947     case WTAP_ERR_SHORT_READ:
3948       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3949             "The file \"%s\" appears to have been cut short"
3950             " in the middle of a packet or other data.",
3951             filename);
3952       break;
3953
3954     case WTAP_ERR_SHORT_WRITE:
3955       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3956             "A full header couldn't be written to the file \"%s\".",
3957             filename);
3958       break;
3959
3960     case WTAP_ERR_COMPRESSION_NOT_SUPPORTED:
3961       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3962             "Gzip compression not supported by this file type.");
3963       break;
3964
3965     case WTAP_ERR_DECOMPRESS:
3966       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3967             "The compressed file \"%s\" appears to be damaged or corrupt.\n"
3968             "(%s)", filename, err_info);
3969       g_free(err_info);
3970       break;
3971
3972     default:
3973       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
3974             "The file \"%s\" could not be %s: %s.",
3975             filename,
3976             for_writing ? "created" : "opened",
3977             wtap_strerror(err));
3978       break;
3979     }
3980   } else {
3981     /* OS error. */
3982     open_failure_alert_box(filename, err, for_writing);
3983   }
3984 }
3985
3986 static const char *
3987 file_rename_error_message(int err)
3988 {
3989   const char *errmsg;
3990   static char errmsg_errno[1024+1];
3991
3992   switch (err) {
3993
3994   case ENOENT:
3995     errmsg = "The path to the file \"%s\" doesn't exist.";
3996     break;
3997
3998   case EACCES:
3999     errmsg = "You don't have permission to move the capture file to \"%s\".";
4000     break;
4001
4002   default:
4003     g_snprintf(errmsg_errno, sizeof(errmsg_errno),
4004             "The file \"%%s\" could not be moved: %s.",
4005                 wtap_strerror(err));
4006     errmsg = errmsg_errno;
4007     break;
4008   }
4009   return errmsg;
4010 }
4011
4012 /* Check for write errors - if the file is being written to an NFS server,
4013    a write error may not show up until the file is closed, as NFS clients
4014    might not send writes to the server until the "write()" call finishes,
4015    so that the write may fail on the server but the "write()" may succeed. */
4016 static void
4017 cf_close_failure_alert_box(const char *filename, int err)
4018 {
4019   if (err < 0) {
4020     /* Wiretap error. */
4021     switch (err) {
4022
4023     case WTAP_ERR_CANT_CLOSE:
4024       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
4025             "The file \"%s\" couldn't be closed for some unknown reason.",
4026             filename);
4027       break;
4028
4029     case WTAP_ERR_SHORT_WRITE:
4030       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
4031             "Not all the packets could be written to the file \"%s\".",
4032                     filename);
4033       break;
4034
4035     default:
4036       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
4037             "An error occurred while closing the file \"%s\": %s.",
4038             filename, wtap_strerror(err));
4039       break;
4040     }
4041   } else {
4042     /* OS error.
4043        We assume that a close error from the OS is really a write error. */
4044     write_failure_alert_box(filename, err);
4045   }
4046 }
4047
4048 /* Reload the current capture file. */
4049 void
4050 cf_reload(capture_file *cf) {
4051   gchar *filename;
4052   gboolean is_tempfile;
4053   int err;
4054
4055   /* If the file could be opened, "cf_open()" calls "cf_close()"
4056      to get rid of state for the old capture file before filling in state
4057      for the new capture file.  "cf_close()" will remove the file if
4058      it's a temporary file; we don't want that to happen (for one thing,
4059      it'd prevent subsequent reopens from working).  Remember whether it's
4060      a temporary file, mark it as not being a temporary file, and then
4061      reopen it as the type of file it was.
4062
4063      Also, "cf_close()" will free "cf->filename", so we must make
4064      a copy of it first. */
4065   filename = g_strdup(cf->filename);
4066   is_tempfile = cf->is_tempfile;
4067   cf->is_tempfile = FALSE;
4068   if (cf_open(cf, filename, is_tempfile, &err) == CF_OK) {
4069     switch (cf_read(cf, FALSE)) {
4070
4071     case CF_READ_OK:
4072     case CF_READ_ERROR:
4073       /* Just because we got an error, that doesn't mean we were unable
4074          to read any of the file; we handle what we could get from the
4075          file. */
4076       break;
4077
4078     case CF_READ_ABORTED:
4079       /* The user bailed out of re-reading the capture file; the
4080          capture file has been closed - just free the capture file name
4081          string and return (without changing the last containing
4082          directory). */
4083       g_free(filename);
4084       return;
4085     }
4086   } else {
4087     /* The open failed, so "cf->is_tempfile" wasn't set to "is_tempfile".
4088        Instead, the file was left open, so we should restore "cf->is_tempfile"
4089        ourselves.
4090
4091        XXX - change the menu?  Presumably "cf_open()" will do that;
4092        make sure it does! */
4093     cf->is_tempfile = is_tempfile;
4094   }
4095   /* "cf_open()" made a copy of the file name we handed it, so
4096      we should free up our copy. */
4097   g_free(filename);
4098 }
4099
4100 /*
4101  * Editor modelines
4102  *
4103  * Local Variables:
4104  * c-basic-offset: 2
4105  * tab-width: 8
4106  * indent-tabs-mode: nil
4107  * End:
4108  *
4109  * ex: set shiftwidth=2 tabstop=8 expandtab:
4110  * :indentSize=2:tabSize=8:noTabs=true:
4111  */