774fccf5108a6e70cdbb59f48a5b2c59a0a7ac7d
[obnox/wireshark/wip.git] / capture.c
1 /* capture.c
2  * Routines for packet capture windows
3  *
4  * $Id: capture.c,v 1.17 1999/02/02 02:53:24 guy Exp $
5  *
6  * Ethereal - Network traffic analyzer
7  * By Gerald Combs <gerald@zing.org>
8  * Copyright 1998 Gerald Combs
9  *
10  * 
11  * This program is free software; you can redistribute it and/or
12  * modify it under the terms of the GNU General Public License
13  * as published by the Free Software Foundation; either version 2
14  * of the License, or (at your option) any later version.
15  * 
16  * This program is distributed in the hope that it will be useful,
17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
19  * GNU General Public License for more details.
20  * 
21  * You should have received a copy of the GNU General Public License
22  * along with this program; if not, write to the Free Software
23  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
24  */
25
26
27 #ifdef HAVE_CONFIG_H
28 # include "config.h"
29 #endif
30
31 #ifdef HAVE_SYS_TYPES_H
32 # include <sys/types.h>
33 #endif
34
35 #include <gtk/gtk.h>
36 #include <pcap.h>
37
38 #include <stdlib.h>
39 #include <string.h>
40 #include <unistd.h>
41 #include <time.h>
42 #include <sys/socket.h>
43 #include <sys/ioctl.h>
44 #include <net/if.h>
45
46 #ifdef NEED_SNPRINTF_H
47 # ifdef HAVE_STDARG_H
48 #  include <stdarg.h>
49 # else
50 #  include <varargs.h>
51 # endif
52 # include "snprintf.h"
53 #endif
54
55 #ifdef HAVE_SYS_SOCKIO_H
56 # include <sys/sockio.h>
57 #endif
58
59 #include "ethereal.h"
60 #include "packet.h"
61 #include "file.h"
62 #include "capture.h"
63 #include "etypes.h"
64 #include "util.h"
65 #include "prefs.h"
66
67 extern capture_file  cf;
68 extern GtkWidget    *info_bar;
69 extern guint         file_ctx;
70
71 /* File selection data keys */
72 #define E_CAP_PREP_FS_KEY "cap_prep_fs"
73 #define E_CAP_PREP_TE_KEY "cap_prep_te"
74
75 /* Capture callback data keys */
76 #define E_CAP_IFACE_KEY "cap_iface"
77 #define E_CAP_FILT_KEY  "cap_filter"
78 #define E_CAP_FILE_KEY  "cap_file"
79 #define E_CAP_COUNT_KEY "cap_count"
80 #define E_CAP_OPEN_KEY  "cap_open"
81 #define E_CAP_SNAP_KEY  "cap_snap"
82
83 /* Capture filter key */
84 #define E_CAP_FILT_TE_KEY "cap_filt_te"
85
86 GList *
87 get_interface_list() {
88   GList  *il = NULL;
89   struct  ifreq *ifr, *last;
90   struct  ifconf ifc;
91   int     sock = socket(AF_INET, SOCK_DGRAM, 0);
92
93   if (sock < 0)
94   {
95     simple_dialog(ESD_TYPE_WARN, NULL,
96       "Can't list interfaces: error opening socket.");
97     return NULL;
98   }
99
100   /* Since we have to grab the interface list all at once, we'll make
101      plenty of room */
102   ifc.ifc_len = 1024 * sizeof(struct ifreq);
103   ifc.ifc_buf = malloc(ifc.ifc_len);
104
105   if (ioctl(sock, SIOCGIFCONF, &ifc) < 0 ||
106     ifc.ifc_len < sizeof(struct ifreq))
107   {
108     simple_dialog(ESD_TYPE_WARN, NULL,
109       "Can't list interfaces: ioctl error.");
110     return NULL;
111   }
112
113   ifr  = (struct ifreq *) ifc.ifc_req;
114   last = (struct ifreq *) ((char *) ifr + ifc.ifc_len);
115   while (ifr < last)
116   {
117     /*
118      * What we want:
119      * - Interfaces that are up, and not loopback
120      * - IP interfaces (do we really need this?)
121      * - Anything that doesn't begin with "lo" (loopback again) or "dummy"
122      * - Anything that doesn't include a ":" (Solaris virtuals)
123      */
124     if (! (ifr->ifr_flags & (IFF_UP | IFF_LOOPBACK)) &&
125         (ifr->ifr_addr.sa_family == AF_INET) &&
126         strncmp(ifr->ifr_name, "lo", 2) &&
127         strncmp(ifr->ifr_name, "dummy", 5) &&
128         ! strchr(ifr->ifr_name, ':')) {
129       il = g_list_append(il, g_strdup(ifr->ifr_name));
130     }
131 #ifdef HAVE_SA_LEN
132     ifr = (struct ifreq *) ((char *) ifr + ifr->ifr_addr.sa_len + IFNAMSIZ);
133 #else
134     ifr = (struct ifreq *) ((char *) ifr + sizeof(struct ifreq));
135 #endif
136   }
137
138   free(ifc.ifc_buf);
139   return il;
140 }
141
142 void
143 capture_prep_cb(GtkWidget *w, gpointer d) {
144   GtkWidget     *cap_open_w, *if_cb, *if_lb, *file_te, *file_bt,
145                 *count_lb, *count_cb, *main_vb, *if_hb, *count_hb,
146                 *filter_hb, *filter_bt, *filter_te, *file_hb, *caplen_hb,
147                 *bbox, *ok_bt, *cancel_bt, *capfile_ck, *snap_lb,
148                 *snap_sb;
149   GtkAdjustment *adj;
150   GList         *if_list, *count_list = NULL;
151   gchar         *count_item1 = "0 (Infinite)", count_item2[16];
152
153   cap_open_w = gtk_window_new(GTK_WINDOW_TOPLEVEL);
154   gtk_window_set_title(GTK_WINDOW(cap_open_w), "Ethereal: Capture Preferences");
155   
156   /* Container for each row of widgets */
157   main_vb = gtk_vbox_new(FALSE, 3);
158   gtk_container_border_width(GTK_CONTAINER(main_vb), 5);
159   gtk_container_add(GTK_CONTAINER(cap_open_w), main_vb);
160   gtk_widget_show(main_vb);
161   
162   /* Interface row */
163   if_hb = gtk_hbox_new(FALSE, 3);
164   gtk_container_add(GTK_CONTAINER(main_vb), if_hb);
165   gtk_widget_show(if_hb);
166   
167   if_lb = gtk_label_new("Interface:");
168   gtk_box_pack_start(GTK_BOX(if_hb), if_lb, FALSE, FALSE, 0);
169   gtk_widget_show(if_lb);
170   
171   if_list = get_interface_list();
172   
173   if_cb = gtk_combo_new();
174   gtk_combo_set_popdown_strings(GTK_COMBO(if_cb), if_list);
175   if (cf.iface)
176     gtk_entry_set_text(GTK_ENTRY(GTK_COMBO(if_cb)->entry), cf.iface);
177   else if (if_list)
178     gtk_entry_set_text(GTK_ENTRY(GTK_COMBO(if_cb)->entry), if_list->data);
179   gtk_box_pack_start(GTK_BOX(if_hb), if_cb, FALSE, FALSE, 0);
180   gtk_widget_show(if_cb);
181   
182   while (if_list) {
183     g_free(if_list->data);
184     if_list = g_list_remove_link(if_list, if_list);
185   }
186
187   /* Count row */
188   count_hb = gtk_hbox_new(FALSE, 3);
189   gtk_container_add(GTK_CONTAINER(main_vb), count_hb);
190   gtk_widget_show(count_hb);
191   
192   count_lb = gtk_label_new("Count:");
193   gtk_box_pack_start(GTK_BOX(count_hb), count_lb, FALSE, FALSE, 0);
194   gtk_widget_show(count_lb);
195   
196   if (cf.count) {
197     snprintf(count_item2, 15, "%d", cf.count);
198     count_list = g_list_append(count_list, count_item2);
199   }
200   count_list = g_list_append(count_list, count_item1);
201
202   count_cb = gtk_combo_new();
203   gtk_combo_set_popdown_strings(GTK_COMBO(count_cb), count_list);
204   gtk_box_pack_start(GTK_BOX(count_hb), count_cb, FALSE, FALSE, 0);
205   gtk_widget_show(count_cb);
206
207   while (count_list)
208     count_list = g_list_remove_link(count_list, count_list);
209
210   /* Filter row */
211   filter_hb = gtk_hbox_new(FALSE, 3);
212   gtk_container_add(GTK_CONTAINER(main_vb), filter_hb);
213   gtk_widget_show(filter_hb);
214   
215   filter_bt = gtk_button_new_with_label("Filter:");
216   gtk_signal_connect(GTK_OBJECT(filter_bt), "clicked",
217     GTK_SIGNAL_FUNC(prefs_cb), (gpointer) E_PR_PG_FILTER);
218   gtk_box_pack_start(GTK_BOX(filter_hb), filter_bt, FALSE, TRUE, 0);
219   gtk_widget_show(filter_bt);
220   
221   filter_te = gtk_entry_new();
222   if (cf.cfilter) gtk_entry_set_text(GTK_ENTRY(filter_te), cf.cfilter);
223   gtk_object_set_data(GTK_OBJECT(filter_bt), E_FILT_TE_PTR_KEY, filter_te);
224   gtk_box_pack_start(GTK_BOX(filter_hb), filter_te, TRUE, TRUE, 0);
225   gtk_widget_show(filter_te);
226   
227   /* File row: File: button and text entry */
228   file_hb = gtk_hbox_new(FALSE, 3);
229   gtk_container_add(GTK_CONTAINER(main_vb), file_hb);
230   gtk_widget_show(file_hb);
231   
232   file_bt = gtk_button_new_with_label("File:");
233   gtk_box_pack_start(GTK_BOX(file_hb), file_bt, FALSE, FALSE, 0);
234   gtk_widget_show(file_bt);
235   
236   file_te = gtk_entry_new();
237   if (cf.save_file)
238     gtk_entry_set_text(GTK_ENTRY(file_te), cf.save_file);
239   gtk_box_pack_start(GTK_BOX(file_hb), file_te, TRUE, TRUE, 0);
240   gtk_widget_show(file_te);
241
242   gtk_signal_connect_object(GTK_OBJECT(file_bt), "clicked",
243     GTK_SIGNAL_FUNC(capture_prep_file_cb), GTK_OBJECT(file_te));
244
245   /* Misc row: Capture file checkbox and snap spinbutton */
246   caplen_hb = gtk_hbox_new(FALSE, 3);
247   gtk_container_add(GTK_CONTAINER(main_vb), caplen_hb);
248   gtk_widget_show(caplen_hb);
249
250   capfile_ck = gtk_check_button_new_with_label("Open file after capture");
251   gtk_toggle_button_set_state(GTK_TOGGLE_BUTTON(capfile_ck), TRUE);
252   gtk_box_pack_start(GTK_BOX(caplen_hb), capfile_ck, FALSE, FALSE, 3);
253   gtk_widget_show(capfile_ck);
254
255   snap_lb = gtk_label_new("Capture length");
256   gtk_misc_set_alignment(GTK_MISC(snap_lb), 0, 0.5);
257   gtk_box_pack_start(GTK_BOX(caplen_hb), snap_lb, FALSE, FALSE, 6);
258   gtk_widget_show(snap_lb);
259
260   adj = (GtkAdjustment *) gtk_adjustment_new((float) cf.snap, 1.0, 4096.0,
261     1.0, 10.0, 0.0);
262   snap_sb = gtk_spin_button_new (adj, 0, 0);
263   gtk_spin_button_set_wrap (GTK_SPIN_BUTTON (snap_sb), TRUE);
264   gtk_widget_set_usize (snap_sb, 80, 0);
265   gtk_box_pack_start (GTK_BOX(caplen_hb), snap_sb, FALSE, FALSE, 3); 
266   gtk_widget_show(snap_sb);
267   
268   /* Button row: OK and cancel buttons */
269   bbox = gtk_hbutton_box_new();
270   gtk_button_box_set_layout (GTK_BUTTON_BOX (bbox), GTK_BUTTONBOX_END);
271   gtk_button_box_set_spacing(GTK_BUTTON_BOX(bbox), 5);
272   gtk_container_add(GTK_CONTAINER(main_vb), bbox);
273   gtk_widget_show(bbox);
274
275   ok_bt = gtk_button_new_with_label ("OK");
276   gtk_signal_connect_object(GTK_OBJECT(ok_bt), "clicked",
277     GTK_SIGNAL_FUNC(capture_prep_ok_cb), GTK_OBJECT(cap_open_w));
278   GTK_WIDGET_SET_FLAGS(ok_bt, GTK_CAN_DEFAULT);
279   gtk_box_pack_start (GTK_BOX (bbox), ok_bt, TRUE, TRUE, 0);
280   gtk_widget_grab_default(ok_bt);
281   gtk_widget_show(ok_bt);
282
283   cancel_bt = gtk_button_new_with_label ("Cancel");
284   gtk_signal_connect_object(GTK_OBJECT(cancel_bt), "clicked",
285     GTK_SIGNAL_FUNC(capture_prep_close_cb), GTK_OBJECT(cap_open_w));
286   GTK_WIDGET_SET_FLAGS(cancel_bt, GTK_CAN_DEFAULT);
287   gtk_box_pack_start (GTK_BOX (bbox), cancel_bt, TRUE, TRUE, 0);
288   gtk_widget_show(cancel_bt);
289
290   /* Attach pointers to needed widgets to the capture prefs window/object */
291   gtk_object_set_data(GTK_OBJECT(cap_open_w), E_CAP_IFACE_KEY, if_cb);
292   gtk_object_set_data(GTK_OBJECT(cap_open_w), E_CAP_FILT_KEY,  filter_te);
293   gtk_object_set_data(GTK_OBJECT(cap_open_w), E_CAP_FILE_KEY,  file_te);
294   gtk_object_set_data(GTK_OBJECT(cap_open_w), E_CAP_COUNT_KEY, count_cb);
295   gtk_object_set_data(GTK_OBJECT(cap_open_w), E_CAP_OPEN_KEY,  capfile_ck);
296   gtk_object_set_data(GTK_OBJECT(cap_open_w), E_CAP_SNAP_KEY,  snap_sb);
297
298   gtk_widget_show(cap_open_w);
299 }
300
301 void
302 capture_prep_file_cb(GtkWidget *w, gpointer te) {
303   GtkWidget *fs;
304
305   fs = gtk_file_selection_new ("Ethereal: Open Save File");
306
307   gtk_object_set_data(GTK_OBJECT(w), E_CAP_PREP_FS_KEY, fs);
308   gtk_object_set_data(GTK_OBJECT(w), E_CAP_PREP_TE_KEY, (GtkWidget *) te);
309   
310   gtk_signal_connect (GTK_OBJECT (GTK_FILE_SELECTION(fs)->ok_button),
311     "clicked", (GtkSignalFunc) cap_prep_fs_ok_cb, w);
312
313   /* Connect the cancel_button to destroy the widget */
314   gtk_signal_connect (GTK_OBJECT (GTK_FILE_SELECTION(fs)->cancel_button),
315     "clicked", (GtkSignalFunc) cap_prep_fs_cancel_cb, w);
316   
317   gtk_widget_show(fs);
318 }
319
320 void
321 cap_prep_fs_ok_cb(GtkWidget *w, gpointer data) {
322   GtkWidget *fs, *te;
323   
324   fs = (GtkWidget *) gtk_object_get_data(GTK_OBJECT(data), E_CAP_PREP_FS_KEY);
325   te = (GtkWidget *) gtk_object_get_data(GTK_OBJECT(data), E_CAP_PREP_TE_KEY);
326
327   gtk_entry_set_text(GTK_ENTRY(te),
328     gtk_file_selection_get_filename (GTK_FILE_SELECTION(fs)));
329   cap_prep_fs_cancel_cb(w, data);
330 }
331
332 void
333 cap_prep_fs_cancel_cb(GtkWidget *w, gpointer data) {
334   GtkWidget *fs;
335   
336   fs = (GtkWidget *) gtk_object_get_data(GTK_OBJECT(data), E_CAP_PREP_FS_KEY);
337
338   gtk_widget_destroy(fs);
339 }  
340
341 void
342 capture_prep_ok_cb(GtkWidget *w, gpointer data) {
343   GtkWidget *if_cb, *filter_te, *file_te, *count_cb, *open_ck, *snap_sb;
344   gint     open;
345
346 #ifdef GTK_HAVE_FEATURES_1_1_0
347   data = w;
348 #endif
349   if_cb     = (GtkWidget *) gtk_object_get_data(GTK_OBJECT(data), E_CAP_IFACE_KEY);
350   filter_te = (GtkWidget *) gtk_object_get_data(GTK_OBJECT(data), E_CAP_FILT_KEY);
351   file_te   = (GtkWidget *) gtk_object_get_data(GTK_OBJECT(data), E_CAP_FILE_KEY);
352   count_cb  = (GtkWidget *) gtk_object_get_data(GTK_OBJECT(data), E_CAP_COUNT_KEY);
353   open_ck   = (GtkWidget *) gtk_object_get_data(GTK_OBJECT(data), E_CAP_OPEN_KEY);
354   snap_sb   = (GtkWidget *) gtk_object_get_data(GTK_OBJECT(data), E_CAP_SNAP_KEY);
355
356   if (cf.iface) g_free(cf.iface);
357   cf.iface =
358     g_strdup(gtk_entry_get_text(GTK_ENTRY(GTK_COMBO(if_cb)->entry)));
359   if (cf.cfilter) g_free(cf.cfilter);
360   cf.cfilter = g_strdup(gtk_entry_get_text(GTK_ENTRY(filter_te)));
361   if (cf.save_file) g_free(cf.save_file);
362   cf.save_file = g_strdup(gtk_entry_get_text(GTK_ENTRY(file_te)));
363   cf.count =
364     atoi(g_strdup(gtk_entry_get_text(GTK_ENTRY(GTK_COMBO(count_cb)->entry))));
365   open = GTK_TOGGLE_BUTTON(open_ck)->active;
366   cf.snap = gtk_spin_button_get_value_as_int(GTK_SPIN_BUTTON(snap_sb));
367   if (cf.snap < 1)
368     cf.snap = 4096;
369   else if (cf.snap < 68)
370     cf.snap = 68;
371
372   gtk_widget_destroy(GTK_WIDGET(data));
373   
374   capture(open);
375 }
376
377 void
378 capture_prep_close_cb(GtkWidget *w, gpointer win) {
379
380 #ifdef GTK_HAVE_FEATURES_1_1_0
381   win = w;
382 #endif
383   gtk_grab_remove(GTK_WIDGET(win));
384   gtk_widget_destroy(GTK_WIDGET(win));
385 }
386
387 void
388 capture(gint open) {
389   GtkWidget  *cap_w, *main_vb, *count_lb, *tcp_lb, *udp_lb, 
390              *ospf_lb, *other_lb, *stop_bt;
391   pcap_t     *pch;
392   gchar       err_str[PCAP_ERRBUF_SIZE], label_str[32];
393   loop_data   ld;
394   bpf_u_int32 netnum, netmask;
395   time_t      upd_time, cur_time;
396   
397   ld.go    = TRUE;
398   ld.count = 0;
399   ld.max   = cf.count;
400   ld.tcp   = 0;
401   ld.udp   = 0;
402   ld.ospf  = 0;
403   ld.other = 0;
404   ld.pdh   = NULL;
405
406   close_cap_file(&cf, info_bar, file_ctx);
407
408   pch = pcap_open_live(cf.iface, cf.snap, 1, 250, err_str);
409
410   if (pch) {
411     if (cf.save_file[0]) {
412       ld.pdh = pcap_dump_open(pch, cf.save_file);
413       if (ld.pdh == NULL) {  /* We have an error */
414         snprintf(err_str, PCAP_ERRBUF_SIZE, "Error trying to open dump "
415           "file:\n%s", pcap_geterr(pch));
416         simple_dialog(ESD_TYPE_WARN, NULL, err_str);
417         g_free(cf.save_file);
418         cf.save_file = NULL;
419         pcap_close(pch);
420         return;
421       }
422     }
423
424     if (cf.cfilter) {
425       if (pcap_lookupnet (cf.iface, &netnum, &netmask, err_str) < 0) {
426         simple_dialog(ESD_TYPE_WARN, NULL,
427           "Can't use filter:  Couldn't obtain netmask info.");
428         return;
429       } else if (pcap_compile(pch, &cf.fcode, cf.cfilter, 1, netmask) < 0) {
430         simple_dialog(ESD_TYPE_WARN, NULL, "Unable to parse filter string.");
431         return;
432       } else if (pcap_setfilter(pch, &cf.fcode) < 0) {
433         simple_dialog(ESD_TYPE_WARN, NULL, "Can't install filter.");
434         return;
435       }
436     }
437
438     cap_w = gtk_window_new(GTK_WINDOW_TOPLEVEL);
439     gtk_window_set_title(GTK_WINDOW(cap_w), "Ethereal: Capture / Playback");
440
441     /* Container for capture display widgets */
442     main_vb = gtk_vbox_new(FALSE, 1);
443     gtk_container_border_width(GTK_CONTAINER(main_vb), 5);
444     gtk_container_add(GTK_CONTAINER(cap_w), main_vb);
445     gtk_widget_show(main_vb);
446
447     count_lb = gtk_label_new("Count: 0");
448     gtk_box_pack_start(GTK_BOX(main_vb), count_lb, FALSE, FALSE, 3);
449     gtk_widget_show(count_lb);
450
451     tcp_lb = gtk_label_new("TCP: 0 (0.0%)");
452     gtk_box_pack_start(GTK_BOX(main_vb), tcp_lb, FALSE, FALSE, 3);
453     gtk_widget_show(tcp_lb);
454
455     udp_lb = gtk_label_new("UDP: 0 (0.0%)");
456     gtk_box_pack_start(GTK_BOX(main_vb), udp_lb, FALSE, FALSE, 3);
457     gtk_widget_show(udp_lb);
458
459     ospf_lb = gtk_label_new("OSPF: 0 (0.0%)");
460     gtk_box_pack_start(GTK_BOX(main_vb), ospf_lb, FALSE, FALSE, 3);
461     gtk_widget_show(ospf_lb);
462
463     other_lb = gtk_label_new("Other: 0 (0.0%)");
464     gtk_box_pack_start(GTK_BOX(main_vb), other_lb, FALSE, FALSE, 3);
465     gtk_widget_show(other_lb);
466
467     stop_bt = gtk_button_new_with_label ("Stop");
468     gtk_signal_connect(GTK_OBJECT(stop_bt), "clicked",
469       GTK_SIGNAL_FUNC(capture_stop_cb), (gpointer) &ld);
470     gtk_box_pack_end(GTK_BOX(main_vb), stop_bt, FALSE, FALSE, 3);
471     GTK_WIDGET_SET_FLAGS(stop_bt, GTK_CAN_DEFAULT);
472     gtk_widget_grab_default(stop_bt);
473     GTK_WIDGET_SET_FLAGS(stop_bt, GTK_CAN_DEFAULT);
474     gtk_widget_grab_default(stop_bt);
475     gtk_widget_show(stop_bt);
476
477     gtk_widget_show(cap_w);
478     gtk_grab_add(cap_w);
479
480     upd_time = time(NULL);
481     while (ld.go) {
482       while (gtk_events_pending()) gtk_main_iteration();
483       pcap_dispatch(pch, 1, capture_pcap_cb, (u_char *) &ld);
484
485       /* Only update once a second so as not to overload slow displays */
486       cur_time = time(NULL);
487       if (cur_time > upd_time) {
488
489         upd_time = cur_time;
490
491         sprintf(label_str, "Count: %d", ld.count);
492         gtk_label_set(GTK_LABEL(count_lb), label_str);
493
494         sprintf(label_str, "TCP: %d (%.1f%%)", ld.tcp, pct(ld.tcp, ld.count));
495         gtk_label_set(GTK_LABEL(tcp_lb), label_str);
496
497         sprintf(label_str, "UDP: %d (%.1f%%)", ld.udp, pct(ld.udp, ld.count));
498         gtk_label_set(GTK_LABEL(udp_lb), label_str);
499
500         sprintf(label_str, "OSPF: %d (%.1f%%)", ld.ospf, pct(ld.ospf, ld.count));
501         gtk_label_set(GTK_LABEL(ospf_lb), label_str);
502
503         sprintf(label_str, "Other: %d (%.1f%%)", ld.other,
504           pct(ld.other, ld.count));
505         gtk_label_set(GTK_LABEL(other_lb), label_str);
506       }
507     }
508     
509     if (ld.pdh) pcap_dump_close(ld.pdh);
510     pcap_close(pch);
511
512     gtk_grab_remove(GTK_WIDGET(cap_w));
513     gtk_widget_destroy(GTK_WIDGET(cap_w));
514   } else {
515     while (gtk_events_pending()) gtk_main_iteration();
516     simple_dialog(ESD_TYPE_WARN, NULL,
517       "The capture session could not be initiated.  Please\n"
518       "check to make sure you have sufficient permissions, and\n"
519       "that you have the proper interface specified.");
520     g_free(cf.save_file);
521     cf.save_file = NULL;
522   }
523
524   if (cf.save_file && open) load_cap_file(cf.save_file, &cf);
525 }
526
527 float
528 pct(gint num, gint denom) {
529   if (denom) {
530     return (float) num * 100.0 / (float) denom;
531   } else {
532     return 0.0;
533   }
534 }
535
536 void
537 capture_stop_cb(GtkWidget *w, gpointer data) {
538   loop_data *ld = (loop_data *) data;
539   
540   ld->go = FALSE;
541 }
542
543 void
544 capture_pcap_cb(u_char *user, const struct pcap_pkthdr *phdr,
545   const u_char *pd) {
546   
547   guint16 etype;
548   guint8  iptype = 0;
549   gint    offset = 14;
550   
551   loop_data *ld = (loop_data *) user;
552   
553   if ((++ld->count >= ld->max) && (ld->max > 0)) 
554   {
555      ld->go = FALSE;
556   }
557   /* Currently, pcap_dumper_t is a FILE *.  Let's hope that doesn't change. */
558   if (ld->pdh) pcap_dump((u_char *) ld->pdh, phdr, pd);
559   
560   etype = etype = (pd[12] << 8) | pd[13];
561   if (etype <= IEEE_802_3_MAX_LEN) {
562     etype = (pd[20] << 8) | pd[21];
563     offset = 22;
564   }
565   
566   switch(etype){ 
567     case ETHERTYPE_IP:
568       iptype = pd[offset + 9];
569       switch (iptype) {
570         case IP_PROTO_TCP:
571           ld->tcp++;
572           break;
573         case IP_PROTO_UDP:
574           ld->udp++;
575           break;
576         case IP_PROTO_OSPF:
577           ld->ospf++;
578           break;
579         default:
580           ld->other++;
581         }
582         break;
583       case ETHERTYPE_IPX:
584       case ETHERTYPE_IPv6:
585       case ETHERTYPE_ATALK:
586       case ETHERTYPE_VINES:
587       case ETHERTYPE_ARP:
588       default:
589         ld->other++;
590   }
591 }