Add a file name field to the "Capture/Start" dialog box; if it's blank,
[obnox/wireshark/wip.git] / gtk / capture_dlg.c
1 /* capture_dlg.c
2  * Routines for packet capture windows
3  *
4  * $Id: capture_dlg.c,v 1.7 1999/10/02 07:13:20 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 #include <stdlib.h>
32 #include <stdio.h>
33 #include <string.h>
34 #include <gtk/gtk.h>
35
36 #ifdef HAVE_UNISTD_H
37 #include <unistd.h>
38 #endif
39
40 #include <time.h>
41
42 #ifdef HAVE_SYS_SOCKET_H
43 #include <sys/socket.h>
44 #endif
45
46 #ifdef HAVE_SYS_IOCTL_H
47 #include <sys/ioctl.h>
48 #endif
49
50 #ifdef HAVE_NET_IF_H
51 #include <net/if.h>
52 #endif
53
54 #include <signal.h>
55 #include <errno.h>
56
57 #ifdef NEED_SNPRINTF_H
58 # ifdef HAVE_STDARG_H
59 #  include <stdarg.h>
60 # else
61 #  include <varargs.h>
62 # endif
63 # include "snprintf.h"
64 #endif
65
66 #ifdef HAVE_SYS_SOCKIO_H
67 # include <sys/sockio.h>
68 #endif
69
70 #include <wiretap/wtap.h>
71 #include "capture.h"
72 #include "globals.h"
73 #include "main.h"
74 #include "capture_dlg.h"
75 #include "prefs_dlg.h"
76 #include "util.h"
77
78 /* Capture callback data keys */
79 #define E_CAP_IFACE_KEY    "cap_iface"
80 #define E_CAP_FILT_KEY     "cap_filter_te"
81 #define E_CAP_FILE_TE_KEY  "cap_file_te"
82 #define E_CAP_COUNT_KEY    "cap_count"
83 #define E_CAP_SNAP_KEY     "cap_snap"
84 #define E_CAP_SYNC_KEY     "cap_sync"
85 #define E_CAP_RESOLVE_KEY  "cap_resolve"
86
87 /* Capture filter key */
88 #define E_CAP_FILT_TE_KEY "cap_filt_te"
89
90 static GList*
91 get_interface_list();
92
93 static void
94 capture_prep_file_cb(GtkWidget *w, gpointer te);
95
96 static void
97 cap_prep_fs_ok_cb(GtkWidget *w, gpointer data);
98
99 static void
100 cap_prep_fs_cancel_cb(GtkWidget *w, gpointer data);
101
102 static void
103 capture_prep_ok_cb(GtkWidget *ok_bt, gpointer parent_w);
104
105 static void
106 capture_prep_close_cb(GtkWidget *close_bt, gpointer parent_w);
107
108 static void
109 search_for_if_cb(gpointer data, gpointer user_data);
110
111 static void
112 free_if_cb(gpointer data, gpointer user_data);
113
114 void
115 capture_prep_cb(GtkWidget *w, gpointer d)
116 {
117   GtkWidget     *cap_open_w, *if_cb, *if_lb,
118                 *count_lb, *count_cb, *main_vb, *if_hb, *count_hb,
119                 *filter_hb, *filter_bt, *filter_te,
120                 *file_hb, *file_bt, *file_te,
121                 *caplen_hb,
122                 *bbox, *ok_bt, *cancel_bt, *snap_lb,
123                 *snap_sb, *sync_cb, *resolv_cb;
124   GtkAdjustment *adj;
125   GList         *if_list, *count_list = NULL;
126   gchar         *count_item1 = "0 (Infinite)", count_item2[16];
127
128   if_list = get_interface_list();
129   if (if_list == NULL)
130     return;
131   
132   cap_open_w = gtk_window_new(GTK_WINDOW_TOPLEVEL);
133   gtk_window_set_title(GTK_WINDOW(cap_open_w), "Ethereal: Capture Preferences");
134   
135   /* Container for each row of widgets */
136   main_vb = gtk_vbox_new(FALSE, 3);
137   gtk_container_border_width(GTK_CONTAINER(main_vb), 5);
138   gtk_container_add(GTK_CONTAINER(cap_open_w), main_vb);
139   gtk_widget_show(main_vb);
140   
141   /* Interface row */
142   if_hb = gtk_hbox_new(FALSE, 3);
143   gtk_container_add(GTK_CONTAINER(main_vb), if_hb);
144   gtk_widget_show(if_hb);
145   
146   if_lb = gtk_label_new("Interface:");
147   gtk_box_pack_start(GTK_BOX(if_hb), if_lb, FALSE, FALSE, 0);
148   gtk_widget_show(if_lb);
149   
150   if_cb = gtk_combo_new();
151   gtk_combo_set_popdown_strings(GTK_COMBO(if_cb), if_list);
152   if (cf.iface)
153     gtk_entry_set_text(GTK_ENTRY(GTK_COMBO(if_cb)->entry), cf.iface);
154   else if (if_list)
155     gtk_entry_set_text(GTK_ENTRY(GTK_COMBO(if_cb)->entry), if_list->data);
156   gtk_box_pack_start(GTK_BOX(if_hb), if_cb, FALSE, FALSE, 0);
157   gtk_widget_show(if_cb);
158   
159   while (if_list) {
160     g_free(if_list->data);
161     if_list = g_list_remove_link(if_list, if_list);
162   }
163
164   /* Count row */
165   count_hb = gtk_hbox_new(FALSE, 3);
166   gtk_container_add(GTK_CONTAINER(main_vb), count_hb);
167   gtk_widget_show(count_hb);
168   
169   count_lb = gtk_label_new("Count:");
170   gtk_box_pack_start(GTK_BOX(count_hb), count_lb, FALSE, FALSE, 0);
171   gtk_widget_show(count_lb);
172   
173   count_list = g_list_append(count_list, count_item1);
174   if (cf.count) {
175     snprintf(count_item2, 15, "%d", cf.count);
176     count_list = g_list_append(count_list, count_item2);
177   }
178
179   count_cb = gtk_combo_new();
180   gtk_combo_set_popdown_strings(GTK_COMBO(count_cb), count_list);
181   gtk_box_pack_start(GTK_BOX(count_hb), count_cb, FALSE, FALSE, 0);
182   gtk_widget_show(count_cb);
183
184   while (count_list)
185     count_list = g_list_remove_link(count_list, count_list);
186
187   /* Filter row */
188   filter_hb = gtk_hbox_new(FALSE, 3);
189   gtk_container_add(GTK_CONTAINER(main_vb), filter_hb);
190   gtk_widget_show(filter_hb);
191   
192   filter_bt = gtk_button_new_with_label("Filter:");
193   gtk_signal_connect(GTK_OBJECT(filter_bt), "clicked",
194     GTK_SIGNAL_FUNC(prefs_cb), (gpointer) E_PR_PG_FILTER);
195   gtk_box_pack_start(GTK_BOX(filter_hb), filter_bt, FALSE, TRUE, 0);
196   gtk_widget_show(filter_bt);
197   
198   filter_te = gtk_entry_new();
199   if (cf.cfilter) gtk_entry_set_text(GTK_ENTRY(filter_te), cf.cfilter);
200   gtk_object_set_data(GTK_OBJECT(filter_bt), E_FILT_TE_PTR_KEY, filter_te);
201   gtk_box_pack_start(GTK_BOX(filter_hb), filter_te, TRUE, TRUE, 0);
202   gtk_widget_show(filter_te);
203   
204   /* File row */
205   file_hb = gtk_hbox_new(FALSE, 1);
206   gtk_container_add(GTK_CONTAINER(main_vb), file_hb);
207   gtk_widget_show(file_hb);
208   
209   file_bt = gtk_button_new_with_label("File:");
210   gtk_box_pack_start(GTK_BOX(file_hb), file_bt, FALSE, FALSE, 3);
211   gtk_widget_show(file_bt);
212   
213   file_te = gtk_entry_new();
214   gtk_box_pack_start(GTK_BOX(file_hb), file_te, TRUE, TRUE, 3);
215   gtk_widget_show(file_te);
216
217   gtk_signal_connect(GTK_OBJECT(file_bt), "clicked",
218     GTK_SIGNAL_FUNC(capture_prep_file_cb), GTK_OBJECT(file_te));
219
220   /* Misc row: Capture file checkbox and snap spinbutton */
221   caplen_hb = gtk_hbox_new(FALSE, 3);
222   gtk_container_add(GTK_CONTAINER(main_vb), caplen_hb);
223   gtk_widget_show(caplen_hb);
224
225   snap_lb = gtk_label_new("Capture length");
226   gtk_misc_set_alignment(GTK_MISC(snap_lb), 0, 0.5);
227   gtk_box_pack_start(GTK_BOX(caplen_hb), snap_lb, FALSE, FALSE, 6);
228   gtk_widget_show(snap_lb);
229
230   adj = (GtkAdjustment *) gtk_adjustment_new((float) cf.snap,
231     MIN_PACKET_SIZE, WTAP_MAX_PACKET_SIZE, 1.0, 10.0, 0.0);
232   snap_sb = gtk_spin_button_new (adj, 0, 0);
233   gtk_spin_button_set_wrap (GTK_SPIN_BUTTON (snap_sb), TRUE);
234   gtk_widget_set_usize (snap_sb, 80, 0);
235   gtk_box_pack_start (GTK_BOX(caplen_hb), snap_sb, FALSE, FALSE, 3); 
236   gtk_widget_show(snap_sb);
237   
238   sync_cb = gtk_check_button_new_with_label("Update list of packets in real time");
239   gtk_toggle_button_set_state(GTK_TOGGLE_BUTTON(sync_cb), sync_mode);
240   gtk_container_add(GTK_CONTAINER(main_vb), sync_cb);
241   gtk_widget_show(sync_cb);
242
243   resolv_cb = gtk_check_button_new_with_label("Enable name resolution");
244   gtk_toggle_button_set_state(GTK_TOGGLE_BUTTON(resolv_cb), g_resolving_actif);
245   gtk_container_add(GTK_CONTAINER(main_vb), resolv_cb);
246   gtk_widget_show(resolv_cb);
247   
248   /* Button row: OK and cancel buttons */
249   bbox = gtk_hbutton_box_new();
250   gtk_button_box_set_layout (GTK_BUTTON_BOX (bbox), GTK_BUTTONBOX_END);
251   gtk_button_box_set_spacing(GTK_BUTTON_BOX(bbox), 5);
252   gtk_container_add(GTK_CONTAINER(main_vb), bbox);
253   gtk_widget_show(bbox);
254
255   ok_bt = gtk_button_new_with_label ("OK");
256   gtk_signal_connect(GTK_OBJECT(ok_bt), "clicked",
257     GTK_SIGNAL_FUNC(capture_prep_ok_cb), GTK_OBJECT(cap_open_w));
258   GTK_WIDGET_SET_FLAGS(ok_bt, GTK_CAN_DEFAULT);
259   gtk_box_pack_start (GTK_BOX (bbox), ok_bt, TRUE, TRUE, 0);
260   gtk_widget_grab_default(ok_bt);
261   gtk_widget_show(ok_bt);
262
263   cancel_bt = gtk_button_new_with_label ("Cancel");
264   gtk_signal_connect(GTK_OBJECT(cancel_bt), "clicked",
265     GTK_SIGNAL_FUNC(capture_prep_close_cb), GTK_OBJECT(cap_open_w));
266   GTK_WIDGET_SET_FLAGS(cancel_bt, GTK_CAN_DEFAULT);
267   gtk_box_pack_start (GTK_BOX (bbox), cancel_bt, TRUE, TRUE, 0);
268   gtk_widget_show(cancel_bt);
269
270   /* Attach pointers to needed widgets to the capture prefs window/object */
271   gtk_object_set_data(GTK_OBJECT(cap_open_w), E_CAP_IFACE_KEY, if_cb);
272   gtk_object_set_data(GTK_OBJECT(cap_open_w), E_CAP_FILT_KEY,  filter_te);
273   gtk_object_set_data(GTK_OBJECT(cap_open_w), E_CAP_FILE_TE_KEY,  file_te);
274   gtk_object_set_data(GTK_OBJECT(cap_open_w), E_CAP_COUNT_KEY, count_cb);
275   gtk_object_set_data(GTK_OBJECT(cap_open_w), E_CAP_SNAP_KEY,  snap_sb);
276   gtk_object_set_data(GTK_OBJECT(cap_open_w), E_CAP_SYNC_KEY,  sync_cb);
277   gtk_object_set_data(GTK_OBJECT(cap_open_w), E_CAP_RESOLVE_KEY,  resolv_cb);
278
279   gtk_widget_show(cap_open_w);
280 }
281
282 static void
283 capture_prep_file_cb(GtkWidget *w, gpointer file_te)
284 {
285   GtkWidget *fs;
286
287   fs = gtk_file_selection_new ("Ethereal: Capture File");
288
289   gtk_object_set_data(GTK_OBJECT(fs), E_CAP_FILE_TE_KEY, file_te);
290
291   gtk_signal_connect (GTK_OBJECT (GTK_FILE_SELECTION(fs)->ok_button),
292     "clicked", (GtkSignalFunc) cap_prep_fs_ok_cb, fs);
293
294   /* Connect the cancel_button to destroy the widget */
295   gtk_signal_connect (GTK_OBJECT (GTK_FILE_SELECTION(fs)->cancel_button),
296     "clicked", (GtkSignalFunc) cap_prep_fs_cancel_cb, fs);
297   
298   gtk_widget_show(fs);
299 }
300
301 static void
302 cap_prep_fs_ok_cb(GtkWidget *w, gpointer data)
303 {
304   gtk_entry_set_text(GTK_ENTRY(gtk_object_get_data(GTK_OBJECT(data),
305       E_CAP_FILE_TE_KEY)),
306       gtk_file_selection_get_filename (GTK_FILE_SELECTION(data)));
307   gtk_widget_destroy(GTK_WIDGET(data));
308 }
309
310 static void
311 cap_prep_fs_cancel_cb(GtkWidget *w, gpointer data)
312 {
313   gtk_widget_destroy(GTK_WIDGET(data));
314 }  
315
316 static void
317 capture_prep_ok_cb(GtkWidget *ok_bt, gpointer parent_w) {
318   GtkWidget *if_cb, *filter_te, *file_te, *count_cb, *snap_sb, *sync_cb,
319             *resolv_cb;
320   gchar *filter_text;
321   gchar *save_file;
322
323   if_cb     = (GtkWidget *) gtk_object_get_data(GTK_OBJECT(parent_w), E_CAP_IFACE_KEY);
324   filter_te = (GtkWidget *) gtk_object_get_data(GTK_OBJECT(parent_w), E_CAP_FILT_KEY);
325   file_te   = (GtkWidget *) gtk_object_get_data(GTK_OBJECT(parent_w), E_CAP_FILE_TE_KEY);
326   count_cb  = (GtkWidget *) gtk_object_get_data(GTK_OBJECT(parent_w), E_CAP_COUNT_KEY);
327   snap_sb   = (GtkWidget *) gtk_object_get_data(GTK_OBJECT(parent_w), E_CAP_SNAP_KEY);
328   sync_cb   = (GtkWidget *) gtk_object_get_data(GTK_OBJECT(parent_w), E_CAP_SYNC_KEY);
329   resolv_cb = (GtkWidget *) gtk_object_get_data(GTK_OBJECT(parent_w), E_CAP_RESOLVE_KEY);
330
331   if (cf.iface)
332     g_free(cf.iface);
333   cf.iface =
334     g_strdup(gtk_entry_get_text(GTK_ENTRY(GTK_COMBO(if_cb)->entry)));
335
336   filter_text = gtk_entry_get_text(GTK_ENTRY(filter_te));
337   if (cf.cfilter)
338     g_free(cf.cfilter);
339   cf.cfilter = NULL; /* ead 06/16/99 */
340   if (filter_text && filter_text[0]) {
341     cf.cfilter = g_strdup(filter_text); 
342   }
343
344   save_file = gtk_entry_get_text(GTK_ENTRY(file_te));
345   if (save_file && save_file[0]) {
346     /* User specified a file to which the capture should be written. */
347     save_file = g_strdup(save_file);
348   } else {
349     /* User didn't specify a file; save to a temporary file. */
350     save_file = NULL;
351   }
352
353   cf.count = atoi(gtk_entry_get_text(GTK_ENTRY(GTK_COMBO(count_cb)->entry)));
354
355   cf.snap = gtk_spin_button_get_value_as_int(GTK_SPIN_BUTTON(snap_sb));
356   if (cf.snap < 1)
357     cf.snap = WTAP_MAX_PACKET_SIZE;
358   else if (cf.snap < MIN_PACKET_SIZE)
359     cf.snap = MIN_PACKET_SIZE;
360
361   sync_mode = GTK_TOGGLE_BUTTON (sync_cb)->active;
362
363   g_resolving_actif = GTK_TOGGLE_BUTTON (resolv_cb)->active;
364
365   gtk_widget_destroy(GTK_WIDGET(parent_w));
366
367   do_capture(save_file);
368 }
369
370 static void
371 capture_prep_close_cb(GtkWidget *close_bt, gpointer parent_w)
372 {
373   gtk_grab_remove(GTK_WIDGET(parent_w));
374   gtk_widget_destroy(GTK_WIDGET(parent_w));
375 }
376
377 struct search_user_data {
378         char    *name;
379         int     found;
380 };
381
382 static GList *
383 get_interface_list() {
384   GList  *il = NULL;
385   gint    nonloopback_pos = 0;
386   struct  ifreq *ifr, *last;
387   struct  ifconf ifc;
388   struct  ifreq ifrflags;
389   int     sock = socket(AF_INET, SOCK_DGRAM, 0);
390   struct search_user_data user_data;
391   pcap_t *pch;
392   gchar   err_str[PCAP_ERRBUF_SIZE];
393
394   if (sock < 0)
395   {
396     simple_dialog(ESD_TYPE_WARN, NULL,
397       "Can't list interfaces: error opening socket.");
398     return NULL;
399   }
400
401   /* Since we have to grab the interface list all at once, we'll make
402      plenty of room */
403   ifc.ifc_len = 1024 * sizeof(struct ifreq);
404   ifc.ifc_buf = malloc(ifc.ifc_len);
405
406   if (ioctl(sock, SIOCGIFCONF, &ifc) < 0 ||
407     ifc.ifc_len < sizeof(struct ifreq))
408   {
409     simple_dialog(ESD_TYPE_WARN, NULL,
410       "Can't list interfaces: SIOCGIFCONF error: %s", strerror(errno));
411     goto fail;
412   }
413
414   ifr  = (struct ifreq *) ifc.ifc_req;
415   last = (struct ifreq *) ((char *) ifr + ifc.ifc_len);
416   while (ifr < last)
417   {
418     /*
419      * Skip addresses that begin with "dummy", or that include a ":"
420      * (the latter are Solaris virtuals).
421      */
422     if (strncmp(ifr->ifr_name, "dummy", 5) == 0 ||
423         strchr(ifr->ifr_name, ':') != NULL)
424       goto next;
425
426     /*
427      * If we already have this interface name on the list, don't add
428      * it (SIOCGIFCONF returns, at least on BSD-flavored systems, one
429      * entry per interface *address*; if an interface has multiple
430      * addresses, we get multiple entries for it).
431      */
432     user_data.name = ifr->ifr_name;
433     user_data.found = FALSE;
434     g_list_foreach(il, search_for_if_cb, &user_data);
435     if (user_data.found)
436       goto next;
437
438     /*
439      * Get the interface flags.
440      */
441     memset(&ifrflags, 0, sizeof ifrflags);
442     strncpy(ifrflags.ifr_name, ifr->ifr_name, sizeof ifrflags.ifr_name);
443     if (ioctl(sock, SIOCGIFFLAGS, (char *)&ifrflags) < 0) {
444       if (errno == ENXIO)
445         goto next;
446       simple_dialog(ESD_TYPE_WARN, NULL,
447         "Can't list interfaces: SIOCGIFFLAGS error on %s: %s",
448         ifr->ifr_name, strerror(errno));
449       goto fail;
450     }
451
452     /*
453      * Skip interfaces that aren't up.
454      */
455     if (!(ifrflags.ifr_flags & IFF_UP))
456       goto next;
457
458     /*
459      * Skip interfaces that we can't open with "libpcap".
460      */
461     pch = pcap_open_live(ifr->ifr_name, WTAP_MAX_PACKET_SIZE, 0, 0, err_str);
462     if (pch == NULL)
463       goto next;
464     pcap_close(pch);
465
466     /*
467      * If it's a loopback interface, add it at the end of the list,
468      * otherwise add it after the last non-loopback interface,
469      * so all loopback interfaces go at the end - we don't want a
470      * loopback interface to be the default capture device unless there
471      * are no non-loopback devices.
472      */
473     if ((ifrflags.ifr_flags & IFF_LOOPBACK) ||
474         strncmp(ifr->ifr_name, "lo", 2) == 0)
475       il = g_list_insert(il, g_strdup(ifr->ifr_name), -1);
476     else {
477       il = g_list_insert(il, g_strdup(ifr->ifr_name), nonloopback_pos);
478       /* Insert the next non-loopback interface after this one. */
479       nonloopback_pos++;
480     }
481
482 next:
483 #ifdef HAVE_SA_LEN
484     ifr = (struct ifreq *) ((char *) ifr + ifr->ifr_addr.sa_len + IFNAMSIZ);
485 #else
486     ifr = (struct ifreq *) ((char *) ifr + sizeof(struct ifreq));
487 #endif
488   }
489
490   free(ifc.ifc_buf);
491   close(sock);
492
493   if (il == NULL) {
494     simple_dialog(ESD_TYPE_WARN, NULL,
495       "There are no network interfaces that can be opened.\n"
496       "Please check to make sure you have sufficient permission\n"
497       "to capture packets.");
498     return NULL;
499   }
500
501   return il;
502
503 fail:
504   if (il != NULL) {
505     g_list_foreach(il, free_if_cb, NULL);
506     g_list_free(il);
507   }
508   free(ifc.ifc_buf);
509   close(sock);
510   return NULL;
511 }
512
513 static void
514 search_for_if_cb(gpointer data, gpointer user_data)
515 {
516         struct search_user_data *search_user_data = user_data;
517
518         if (strcmp((char *)data, search_user_data->name) == 0)
519                 search_user_data->found = TRUE;
520 }
521
522 static void
523 free_if_cb(gpointer data, gpointer user_data)
524 {
525         g_free(data);
526 }
527
528