From Jim Young via bug 3196: Add preference for displaying expert infos LEDs.
[metze/wireshark/wip.git] / dumpcap.c
1 /* dumpcap.c
2  *
3  * $Id$
4  *
5  * Wireshark - Network traffic analyzer
6  * By Gerald Combs <gerald@wireshark.org>
7  * Copyright 1998 Gerald Combs
8  *
9  * This program is free software; you can redistribute it and/or
10  * modify it under the terms of the GNU General Public License
11  * as published by the Free Software Foundation; either version 2
12  * of the License, or (at your option) any later version.
13  *
14  * This program is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17  * GNU General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write to the Free Software
21  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
22  */
23
24 #ifdef HAVE_CONFIG_H
25 # include "config.h"
26 #endif
27
28 #include <stdio.h>
29 #include <stdlib.h> /* for exit() */
30 #include <glib.h>
31
32 #include <string.h>
33 #include <ctype.h>
34
35 #ifdef HAVE_SYS_TYPES_H
36 # include <sys/types.h>
37 #endif
38
39 #ifdef HAVE_SYS_STAT_H
40 # include <sys/stat.h>
41 #endif
42
43 #ifdef HAVE_FCNTL_H
44 #include <fcntl.h>
45 #endif
46
47 #ifdef HAVE_UNISTD_H
48 #include <unistd.h>
49 #endif
50
51 #ifdef HAVE_ARPA_INET_H
52 #include <arpa/inet.h>
53 #endif
54
55 #if defined(__APPLE__) && defined(__LP64__)
56 #include <sys/utsname.h>
57 #endif
58
59 #include <signal.h>
60 #include <errno.h>
61
62 #ifdef HAVE_GETOPT_H
63 #include <getopt.h>
64 #else
65 #include "wsutil/wsgetopt.h"
66 #endif
67
68 #ifdef HAVE_NETDB_H
69 #include <netdb.h>
70 #endif
71
72 #ifdef HAVE_LIBCAP
73 # include <sys/prctl.h>
74 # include <sys/capability.h>
75 #endif
76
77 #include "ringbuffer.h"
78 #include "clopts_common.h"
79 #include "console_io.h"
80 #include "cmdarg_err.h"
81 #include "version_info.h"
82
83 #include "capture-pcap-util.h"
84
85 #include "pcapio.h"
86
87 #ifdef _WIN32
88 #include "capture-wpcap.h"
89 #include <wsutil/unicode-utils.h>
90 #endif
91
92 #ifndef _WIN32
93 #include <sys/socket.h>
94 #include <sys/un.h>
95 #endif
96
97 #ifdef NEED_INET_V6DEFS_H
98 # include "wsutil/inet_v6defs.h"
99 #endif
100
101 #include <wsutil/privileges.h>
102
103 #include "sync_pipe.h"
104
105 #include "capture_opts.h"
106 #include "capture_ifinfo.h"
107 #include "capture_sync.h"
108
109 #include "conditions.h"
110 #include "capture_stop_conditions.h"
111
112 #include "tempfile.h"
113 #include "log.h"
114 #include "wsutil/file_util.h"
115
116 /*
117  * Get information about libpcap format from "wiretap/libpcap.h".
118  * XXX - can we just use pcap_open_offline() to read the pipe?
119  */
120 #include "wiretap/libpcap.h"
121
122 /**#define DEBUG_DUMPCAP**/
123 /**#define DEBUG_CHILD_DUMPCAP**/
124
125 #ifdef _WIN32
126 #ifdef DEBUG_DUMPCAP
127 #include <conio.h>          /* _getch() */
128 #endif
129 #endif
130
131 #ifdef DEBUG_CHILD_DUMPCAP
132 FILE *debug_log;   /* for logging debug messages to  */
133                    /*  a file if DEBUG_CHILD_DUMPCAP */
134                    /*  is defined                    */
135 #endif
136
137 static GAsyncQueue *pcap_queue;
138 static gint64 pcap_queue_bytes;
139 static gint64 pcap_queue_packets;
140 static gint64 pcap_queue_byte_limit = 1024 * 1024;
141 static gint64 pcap_queue_packet_limit = 1000;
142
143 static gboolean capture_child = FALSE; /* FALSE: standalone call, TRUE: this is an Wireshark capture child */
144 #ifdef _WIN32
145 static gchar *sig_pipe_name = NULL;
146 static HANDLE sig_pipe_handle = NULL;
147 static gboolean signal_pipe_check_running(void);
148 #endif
149
150 #ifdef SIGINFO
151 static gboolean infodelay;      /* if TRUE, don't print capture info in SIGINFO handler */
152 static gboolean infoprint;      /* if TRUE, print capture info after clearing infodelay */
153 #endif /* SIGINFO */
154
155 /** Stop a low-level capture (stops the capture child). */
156 static void capture_loop_stop(void);
157
158 #if !defined (__linux__)
159 #ifndef HAVE_PCAP_BREAKLOOP
160 /*
161  * We don't have pcap_breakloop(), which is the only way to ensure that
162  * pcap_dispatch(), pcap_loop(), or even pcap_next() or pcap_next_ex()
163  * won't, if the call to read the next packet or batch of packets is
164  * is interrupted by a signal on UN*X, just go back and try again to
165  * read again.
166  *
167  * On UN*X, we catch SIGINT as a "stop capturing" signal, and, in
168  * the signal handler, set a flag to stop capturing; however, without
169  * a guarantee of that sort, we can't guarantee that we'll stop capturing
170  * if the read will be retried and won't time out if no packets arrive.
171  *
172  * Therefore, on at least some platforms, we work around the lack of
173  * pcap_breakloop() by doing a select() on the pcap_t's file descriptor
174  * to wait for packets to arrive, so that we're probably going to be
175  * blocked in the select() when the signal arrives, and can just bail
176  * out of the loop at that point.
177  *
178  * However, we don't want to do that on BSD (because "select()" doesn't work
179  * correctly on BPF devices on at least some releases of some flavors of
180  * BSD), and we don't want to do it on Windows (because "select()" is
181  * something for sockets, not for arbitrary handles).  (Note that "Windows"
182  * here includes Cygwin; even in its pretend-it's-UNIX environment, we're
183  * using WinPcap, not a UNIX libpcap.)
184  *
185  * Fortunately, we don't need to do it on BSD, because the libpcap timeout
186  * on BSD times out even if no packets have arrived, so we'll eventually
187  * exit pcap_dispatch() with an indication that no packets have arrived,
188  * and will break out of the capture loop at that point.
189  *
190  * On Windows, we can't send a SIGINT to stop capturing, so none of this
191  * applies in any case.
192  *
193  * XXX - the various BSDs appear to define BSD in <sys/param.h>; we don't
194  * want to include it if it's not present on this platform, however.
195  */
196 # if !defined(__FreeBSD__) && !defined(__NetBSD__) && !defined(__OpenBSD__) && \
197     !defined(__bsdi__) && !defined(__APPLE__) && !defined(_WIN32) && \
198     !defined(__CYGWIN__)
199 #  define MUST_DO_SELECT
200 # endif /* avoid select */
201 #endif /* HAVE_PCAP_BREAKLOOP */
202 #else /* linux */
203 /* whatever the deal with pcap_breakloop, linux doesn't support timeouts
204  * in pcap_dispatch(); on the other hand, select() works just fine there.
205  * Hence we use a select for that come what may.
206  */
207 #define MUST_DO_SELECT
208 #endif
209
210 /** init the capture filter */
211 typedef enum {
212     INITFILTER_NO_ERROR,
213     INITFILTER_BAD_FILTER,
214     INITFILTER_OTHER_ERROR
215 } initfilter_status_t;
216
217 typedef struct _pcap_options {
218     guint32        received;
219     guint32        dropped;
220     pcap_t         *pcap_h;
221 #ifdef MUST_DO_SELECT
222     int            pcap_fd;               /* pcap file descriptor */
223 #endif
224     gboolean       pcap_err;
225     guint          interface_id;
226     GThread        *tid;
227     int            snaplen;
228     int            linktype;
229     /* capture pipe (unix only "input file") */
230     gboolean       from_cap_pipe;         /* TRUE if we are capturing data from a capture pipe */
231     struct pcap_hdr cap_pipe_hdr;         /* Pcap header when capturing from a pipe */
232     struct pcaprec_modified_hdr cap_pipe_rechdr;  /* Pcap record header when capturing from a pipe */
233 #ifdef _WIN32
234     HANDLE         cap_pipe_h;            /* The handle of the capture pipe */
235 #else
236     int            cap_pipe_fd;           /* the file descriptor of the capture pipe */
237 #endif
238     gboolean       cap_pipe_modified;     /* TRUE if data in the pipe uses modified pcap headers */
239     gboolean       cap_pipe_byte_swapped; /* TRUE if data in the pipe is byte swapped */
240 #if defined(USE_THREADS) && defined(_WIN32)
241     char *         cap_pipe_buf;          /* Pointer to the data buffer we read into */
242 #endif
243     int            cap_pipe_bytes_to_read;/* Used by cap_pipe_dispatch */
244     int            cap_pipe_bytes_read;   /* Used by cap_pipe_dispatch */
245     enum {
246         STATE_EXPECT_REC_HDR,
247         STATE_READ_REC_HDR,
248         STATE_EXPECT_DATA,
249         STATE_READ_DATA
250     } cap_pipe_state;
251     enum { PIPOK, PIPEOF, PIPERR, PIPNEXIST } cap_pipe_err;
252 #if defined(USE_THREADS) && defined(_WIN32)
253     GMutex *cap_pipe_read_mtx;
254     GAsyncQueue *cap_pipe_pending_q, *cap_pipe_done_q;
255 #endif
256 } pcap_options;
257
258 typedef struct _loop_data {
259     /* common */
260     gboolean       go;                    /* TRUE as long as we're supposed to keep capturing */
261     int            err;                   /* if non-zero, error seen while capturing */
262     gint           packet_count;          /* Number of packets we have already captured */
263     gint           packet_max;            /* Number of packets we're supposed to capture - 0 means infinite */
264     gint           inpkts_to_sync_pipe;   /* Packets not already send out to the sync_pipe */
265 #ifdef SIGINFO
266     gboolean       report_packet_count;   /* Set by SIGINFO handler; print packet count */
267 #endif
268     GArray         *pcaps;
269     /* output file(s) */
270     FILE          *pdh;
271     int            save_file_fd;
272     long           bytes_written;
273     guint32        autostop_files;
274 } loop_data;
275
276 typedef struct _pcap_queue_element {
277     pcap_options       *pcap_opts;
278     struct pcap_pkthdr phdr;
279     u_char             *pd;
280 } pcap_queue_element;
281
282 /*
283  * Standard secondary message for unexpected errors.
284  */
285 static const char please_report[] =
286     "Please report this to the Wireshark developers.\n"
287     "(This is not a crash; please do not report it as such.)";
288
289 /*
290  * This needs to be static, so that the SIGINT handler can clear the "go"
291  * flag.
292  */
293 static loop_data   global_ld;
294
295
296 /*
297  * Timeout, in milliseconds, for reads from the stream of captured packets
298  * from a capture device.
299  *
300  * A bug in Mac OS X 10.6 and 10.6.1 causes calls to pcap_open_live(), in
301  * 64-bit applications, with sub-second timeouts not to work.  The bug is
302  * fixed in 10.6.2, re-broken in 10.6.3, and again fixed in 10.6.5.
303  */
304 #if defined(__APPLE__) && defined(__LP64__)
305 static gboolean need_timeout_workaround;
306
307 #define CAP_READ_TIMEOUT        (need_timeout_workaround ? 1000 : 250)
308 #else
309 #define CAP_READ_TIMEOUT        250
310 #endif
311
312 /*
313  * Timeout, in microseconds, for reads from the stream of captured packets
314  * from a pipe.  Pipes don't have the same problem that BPF devices do
315  * in OS X 10.6, 10.6.1, 10.6.3, and 10.6.4, so we always use a timeout
316  * of 250ms, i.e. the same value as CAP_READ_TIMEOUT when not on one
317  * of the offending versions of Snow Leopard.
318  *
319  * On Windows this value is converted to milliseconds and passed to
320  * WaitForSingleObject. If it's less than 1000 WaitForSingleObject
321  * will return immediately.
322  */
323 #if defined(USE_THREADS) && defined(_WIN32)
324 #define PIPE_READ_TIMEOUT   100000
325 #else
326 #define PIPE_READ_TIMEOUT   250000
327 #endif
328
329 #define WRITER_THREAD_TIMEOUT 100000 /* usecs */
330
331 static void
332 console_log_handler(const char *log_domain, GLogLevelFlags log_level,
333                     const char *message, gpointer user_data _U_);
334
335 /* capture related options */
336 static capture_options global_capture_opts;
337 static gboolean quiet = FALSE;
338 static gboolean use_threads = FALSE;
339
340 static void capture_loop_write_packet_cb(u_char *pcap_opts_p, const struct pcap_pkthdr *phdr,
341                                          const u_char *pd);
342 static void capture_loop_queue_packet_cb(u_char *pcap_opts_p, const struct pcap_pkthdr *phdr,
343                                          const u_char *pd);
344 static void capture_loop_get_errmsg(char *errmsg, int errmsglen, const char *fname,
345                                     int err, gboolean is_close);
346
347 static void WS_MSVC_NORETURN exit_main(int err) G_GNUC_NORETURN;
348
349 static void report_new_capture_file(const char *filename);
350 static void report_packet_count(int packet_count);
351 static void report_packet_drops(guint32 received, guint32 drops, gchar *name);
352 static void report_capture_error(const char *error_msg, const char *secondary_error_msg);
353 static void report_cfilter_error(capture_options *capture_opts, guint i, const char *errmsg);
354
355 #define MSG_MAX_LENGTH 4096
356
357 static void
358 print_usage(gboolean print_ver)
359 {
360     FILE *output;
361
362     if (print_ver) {
363         output = stdout;
364         fprintf(output,
365                 "Dumpcap " VERSION "%s\n"
366                 "Capture network packets and dump them into a libpcap file.\n"
367                 "See http://www.wireshark.org for more information.\n",
368                 wireshark_svnversion);
369     } else {
370         output = stderr;
371     }
372     fprintf(output, "\nUsage: dumpcap [options] ...\n");
373     fprintf(output, "\n");
374     fprintf(output, "Capture interface:\n");
375     fprintf(output, "  -i <interface>           name or idx of interface (def: first non-loopback)\n");
376     fprintf(output, "  -f <capture filter>      packet filter in libpcap filter syntax\n");
377     fprintf(output, "  -s <snaplen>             packet snapshot length (def: 65535)\n");
378     fprintf(output, "  -p                       don't capture in promiscuous mode\n");
379 #ifdef HAVE_PCAP_CREATE
380     fprintf(output, "  -I                       capture in monitor mode, if available\n");
381 #endif
382 #if defined(_WIN32) || defined(HAVE_PCAP_CREATE)
383     fprintf(output, "  -B <buffer size>         size of kernel buffer (def: 1MB)\n");
384 #endif
385     fprintf(output, "  -y <link type>           link layer type (def: first appropriate)\n");
386     fprintf(output, "  -D                       print list of interfaces and exit\n");
387     fprintf(output, "  -L                       print list of link-layer types of iface and exit\n");
388 #ifdef HAVE_BPF_IMAGE
389     fprintf(output, "  -d                       print generated BPF code for capture filter\n");
390 #endif
391     fprintf(output, "  -S                       print statistics for each interface once every second\n");
392     fprintf(output, "  -M                       for -D, -L, and -S, produce machine-readable output\n");
393     fprintf(output, "\n");
394 #ifdef HAVE_PCAP_REMOTE
395     fprintf(output, "\nRPCAP options:\n");
396     fprintf(output, "  -r                       don't ignore own RPCAP traffic in capture\n");
397     fprintf(output, "  -u                       use UDP for RPCAP data transfer\n");
398     fprintf(output, "  -A <user>:<password>     use RPCAP password authentication\n");
399 #ifdef HAVE_PCAP_SETSAMPLING
400     fprintf(output, "  -m <sampling type>       use packet sampling\n");
401     fprintf(output, "                           count:NUM - capture one packet of every NUM\n");
402     fprintf(output, "                           timer:NUM - capture no more than 1 packet in NUM ms\n");
403 #endif
404 #endif
405     fprintf(output, "Stop conditions:\n");
406     fprintf(output, "  -c <packet count>        stop after n packets (def: infinite)\n");
407     fprintf(output, "  -a <autostop cond.> ...  duration:NUM - stop after NUM seconds\n");
408     fprintf(output, "                           filesize:NUM - stop this file after NUM KB\n");
409     fprintf(output, "                              files:NUM - stop after NUM files\n");
410     /*fprintf(output, "\n");*/
411     fprintf(output, "Output (files):\n");
412     fprintf(output, "  -w <filename>            name of file to save (def: tempfile)\n");
413     fprintf(output, "  -g                       enable group read access on the output file(s)\n");
414     fprintf(output, "  -b <ringbuffer opt.> ... duration:NUM - switch to next file after NUM secs\n");
415     fprintf(output, "                           filesize:NUM - switch to next file after NUM KB\n");
416     fprintf(output, "                              files:NUM - ringbuffer: replace after NUM files\n");
417     fprintf(output, "  -n                       use pcapng format instead of pcap\n");
418     /*fprintf(output, "\n");*/
419     fprintf(output, "Miscellaneous:\n");
420 #ifdef USE_THREADS
421     fprintf(output, "  -t                       use a separate thread per interface\n");
422 #endif
423     fprintf(output, "  -q                       don't report packet capture counts\n");
424     fprintf(output, "  -v                       print version information and exit\n");
425     fprintf(output, "  -h                       display this help and exit\n");
426     fprintf(output, "\n");
427     fprintf(output, "Example: dumpcap -i eth0 -a duration:60 -w output.pcap\n");
428     fprintf(output, "\"Capture network packets from interface eth0 until 60s passed into output.pcap\"\n");
429     fprintf(output, "\n");
430     fprintf(output, "Use Ctrl-C to stop capturing at any time.\n");
431 }
432
433 static void
434 show_version(GString *comp_info_str, GString *runtime_info_str)
435 {
436     printf(
437         "Dumpcap " VERSION "%s\n"
438         "\n"
439         "%s\n"
440         "%s\n"
441         "%s\n"
442         "See http://www.wireshark.org for more information.\n",
443         wireshark_svnversion, get_copyright_info() ,comp_info_str->str, runtime_info_str->str);
444 }
445
446 /*
447  * Print to the standard error.  This is a command-line tool, so there's
448  * no need to pop up a console.
449  */
450 void
451 vfprintf_stderr(const char *fmt, va_list ap)
452 {
453     vfprintf(stderr, fmt, ap);
454 }
455
456 void
457 fprintf_stderr(const char *fmt, ...)
458 {
459     va_list ap;
460
461     va_start(ap, fmt);
462     vfprintf_stderr(fmt, ap);
463     va_end(ap);
464 }
465
466 /*
467  * Report an error in command-line arguments.
468  */
469 void
470 cmdarg_err(const char *fmt, ...)
471 {
472     va_list ap;
473
474     if(capture_child) {
475         gchar *msg;
476         /* Generate a 'special format' message back to parent */
477         va_start(ap, fmt);
478         msg = g_strdup_vprintf(fmt, ap);
479         sync_pipe_errmsg_to_parent(2, msg, "");
480         g_free(msg);
481         va_end(ap);
482     } else {
483         va_start(ap, fmt);
484         fprintf(stderr, "dumpcap: ");
485         vfprintf(stderr, fmt, ap);
486         fprintf(stderr, "\n");
487         va_end(ap);
488     }
489 }
490
491 /*
492  * Report additional information for an error in command-line arguments.
493  */
494 void
495 cmdarg_err_cont(const char *fmt, ...)
496 {
497     va_list ap;
498
499     if(capture_child) {
500         gchar *msg;
501         va_start(ap, fmt);
502         msg = g_strdup_vprintf(fmt, ap);
503         sync_pipe_errmsg_to_parent(2, msg, "");
504         g_free(msg);
505         va_end(ap);
506     } else {
507         va_start(ap, fmt);
508         vfprintf(stderr, fmt, ap);
509         fprintf(stderr, "\n");
510         va_end(ap);
511     }
512 }
513
514 #ifdef HAVE_LIBCAP
515 static void
516 #if 0 /* Set to enable capability debugging */
517 /* see 'man cap_to_text()' for explanation of output                         */
518 /* '='   means 'all= '  ie: no capabilities                                  */
519 /* '=ip' means 'all=ip' ie: all capabilities are permissible and inheritable */
520 /* ....                                                                      */
521 print_caps(const char *pfx) {
522     cap_t caps = cap_get_proc();
523     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG,
524           "%s: EUID: %d  Capabilities: %s", pfx,
525           geteuid(), cap_to_text(caps, NULL));
526     cap_free(caps);
527 #else
528 print_caps(const char *pfx _U_) {
529 #endif
530 }
531
532 static void
533 relinquish_all_capabilities(void)
534 {
535     /* Drop any and all capabilities this process may have.            */
536     /* Allowed whether or not process has any privileges.              */
537     cap_t caps = cap_init();    /* all capabilities initialized to off */
538     print_caps("Pre-clear");
539     if (cap_set_proc(caps)) {
540         cmdarg_err("cap_set_proc() fail return: %s", g_strerror(errno));
541     }
542     print_caps("Post-clear");
543     cap_free(caps);
544 }
545 #endif
546
547 static pcap_t *
548 open_capture_device(interface_options *interface_opts,
549                     char (*open_err_str)[PCAP_ERRBUF_SIZE])
550 {
551     pcap_t *pcap_h;
552 #ifdef HAVE_PCAP_CREATE
553     int         err;
554 #endif
555 #if defined(HAVE_PCAP_OPEN) && defined(HAVE_PCAP_REMOTE)
556     struct pcap_rmtauth auth;
557 #endif
558
559     /* Open the network interface to capture from it.
560        Some versions of libpcap may put warnings into the error buffer
561        if they succeed; to tell if that's happened, we have to clear
562        the error buffer, and check if it's still a null string.  */
563     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "Entering open_capture_device().");
564     (*open_err_str)[0] = '\0';
565 #if defined(HAVE_PCAP_OPEN) && defined(HAVE_PCAP_REMOTE)
566     /*
567      * If we're opening a remote device, use pcap_open(); that's currently
568      * the only open routine that supports remote devices.
569      */
570     if (strncmp (interface_opts->name, "rpcap://", 8) == 0) {
571         auth.type = interface_opts->auth_type == CAPTURE_AUTH_PWD ?
572             RPCAP_RMTAUTH_PWD : RPCAP_RMTAUTH_NULL;
573         auth.username = interface_opts->auth_username;
574         auth.password = interface_opts->auth_password;
575
576         g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG,
577               "Calling pcap_open() using name %s, snaplen %d, promisc_mode %d, datatx_udp %d, nocap_rpcap %d.",
578               interface_opts->name, interface_opts->snaplen, interface_opts->promisc_mode,
579               interface_opts->datatx_udp, interface_opts->nocap_rpcap);
580         pcap_h = pcap_open(interface_opts->name, interface_opts->snaplen,
581                            /* flags */
582                            (interface_opts->promisc_mode ? PCAP_OPENFLAG_PROMISCUOUS : 0) |
583                            (interface_opts->datatx_udp ? PCAP_OPENFLAG_DATATX_UDP : 0) |
584                            (interface_opts->nocap_rpcap ? PCAP_OPENFLAG_NOCAPTURE_RPCAP : 0),
585                            CAP_READ_TIMEOUT, &auth, *open_err_str);
586         g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG,
587               "pcap_open() returned %p.", (void *)pcap_h);
588     } else
589 #endif
590     {
591         /*
592          * If we're not opening a remote device, use pcap_create() and
593          * pcap_activate() if we have them, so that we can set the buffer
594          * size, otherwise use pcap_open_live().
595          */
596 #ifdef HAVE_PCAP_CREATE
597         g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG,
598               "Calling pcap_create() using %s.", interface_opts->name);
599         pcap_h = pcap_create(interface_opts->name, *open_err_str);
600         g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG,
601               "pcap_create() returned %p.", (void *)pcap_h);
602         if (pcap_h != NULL) {
603             g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG,
604                   "Calling pcap_set_snaplen() with snaplen %d.", interface_opts->snaplen);
605             pcap_set_snaplen(pcap_h, interface_opts->snaplen);
606             g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG,
607                   "Calling pcap_set_snaplen() with promisc_mode %d.", interface_opts->promisc_mode);
608             pcap_set_promisc(pcap_h, interface_opts->promisc_mode);
609             pcap_set_timeout(pcap_h, CAP_READ_TIMEOUT);
610
611             g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG,
612                   "buffersize %d.", interface_opts->buffer_size);
613             if (interface_opts->buffer_size > 1) {
614                 pcap_set_buffer_size(pcap_h, interface_opts->buffer_size * 1024 * 1024);
615             }
616             g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG,
617                   "monitor_mode %d.", interface_opts->monitor_mode);
618             if (interface_opts->monitor_mode)
619                 pcap_set_rfmon(pcap_h, 1);
620             err = pcap_activate(pcap_h);
621             g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG,
622                   "pcap_activate() returned %d.", err);
623             if (err < 0) {
624                 /* Failed to activate, set to NULL */
625                 if (err == PCAP_ERROR)
626                     g_strlcpy(*open_err_str, pcap_geterr(pcap_h), sizeof *open_err_str);
627                 else
628                     g_strlcpy(*open_err_str, pcap_statustostr(err), sizeof *open_err_str);
629                 pcap_close(pcap_h);
630                 pcap_h = NULL;
631             }
632         }
633 #else
634         g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG,
635               "pcap_open_live() calling using name %s, snaplen %d, promisc_mode %d.",
636               interface_opts->name, interface_opts->snaplen, interface_opts->promisc_mode);
637         pcap_h = pcap_open_live(interface_opts->name, interface_opts->snaplen,
638                                 interface_opts->promisc_mode, CAP_READ_TIMEOUT,
639                                 *open_err_str);
640         g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG,
641               "pcap_open_live() returned %p.", (void *)pcap_h);
642 #endif
643     }
644     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "open_capture_device %s : %s", pcap_h ? "SUCCESS" : "FAILURE", interface_opts->name);
645     return pcap_h;
646 }
647
648 static void
649 get_capture_device_open_failure_messages(const char *open_err_str,
650                                          const char *iface
651 #ifndef _WIN32
652                                                            _U_
653 #endif
654                                          ,
655                                          char *errmsg, size_t errmsg_len,
656                                          char *secondary_errmsg,
657                                          size_t secondary_errmsg_len)
658 {
659     const char *libpcap_warn;
660     static const char ppamsg[] = "can't find PPA for ";
661
662     /* If we got a "can't find PPA for X" message, warn the user (who
663        is running dumcap on HP-UX) that they don't have a version of
664        libpcap that properly handles HP-UX (libpcap 0.6.x and later
665        versions, which properly handle HP-UX, say "can't find /dev/dlpi
666        PPA for X" rather than "can't find PPA for X"). */
667     if (strncmp(open_err_str, ppamsg, sizeof ppamsg - 1) == 0)
668         libpcap_warn =
669             "\n\n"
670             "You are running (T)Wireshark with a version of the libpcap library\n"
671             "that doesn't handle HP-UX network devices well; this means that\n"
672             "(T)Wireshark may not be able to capture packets.\n"
673             "\n"
674             "To fix this, you should install libpcap 0.6.2, or a later version\n"
675             "of libpcap, rather than libpcap 0.4 or 0.5.x.  It is available in\n"
676             "packaged binary form from the Software Porting And Archive Centre\n"
677             "for HP-UX; the Centre is at http://hpux.connect.org.uk/ - the page\n"
678             "at the URL lists a number of mirror sites.";
679     else
680         libpcap_warn = "";
681     g_snprintf(errmsg, (gulong) errmsg_len,
682                "The capture session could not be initiated (%s).", open_err_str);
683 #ifndef _WIN32
684     g_snprintf(secondary_errmsg, (gulong) secondary_errmsg_len,
685                "Please check to make sure you have sufficient permissions, and that you have "
686                "the proper interface or pipe specified.%s", libpcap_warn);
687 #else
688     g_snprintf(secondary_errmsg, (gulong) secondary_errmsg_len,
689                "\n"
690                "Please check that \"%s\" is the proper interface.\n"
691                "\n"
692                "\n"
693                "Help can be found at:\n"
694                "\n"
695                "       http://wiki.wireshark.org/WinPcap\n"
696                "       http://wiki.wireshark.org/CaptureSetup\n",
697                iface);
698 #endif /* _WIN32 */
699 }
700
701 /* Set the data link type on a pcap. */
702 static gboolean
703 set_pcap_linktype(pcap_t *pcap_h, int linktype,
704 #ifdef HAVE_PCAP_SET_DATALINK
705                   char *name _U_,
706 #else
707                   char *name,
708 #endif
709                   char *errmsg, size_t errmsg_len,
710                   char *secondary_errmsg, size_t secondary_errmsg_len)
711 {
712     char *set_linktype_err_str;
713
714     if (linktype == -1)
715         return TRUE; /* just use the default */
716 #ifdef HAVE_PCAP_SET_DATALINK
717     if (pcap_set_datalink(pcap_h, linktype) == 0)
718         return TRUE; /* no error */
719     set_linktype_err_str = pcap_geterr(pcap_h);
720 #else
721     /* Let them set it to the type it is; reject any other request. */
722     if (get_pcap_linktype(pcap_h, name) == linktype)
723         return TRUE; /* no error */
724     set_linktype_err_str =
725         "That DLT isn't one of the DLTs supported by this device";
726 #endif
727     g_snprintf(errmsg, (gulong) errmsg_len, "Unable to set data link type (%s).",
728                set_linktype_err_str);
729     /*
730      * If the error isn't "XXX is not one of the DLTs supported by this device",
731      * tell the user to tell the Wireshark developers about it.
732      */
733     if (strstr(set_linktype_err_str, "is not one of the DLTs supported by this device") == NULL)
734         g_snprintf(secondary_errmsg, (gulong) secondary_errmsg_len, please_report);
735     else
736         secondary_errmsg[0] = '\0';
737     return FALSE;
738 }
739
740 static gboolean
741 compile_capture_filter(const char *iface, pcap_t *pcap_h,
742                        struct bpf_program *fcode, const char *cfilter)
743 {
744     bpf_u_int32 netnum, netmask;
745     gchar       lookup_net_err_str[PCAP_ERRBUF_SIZE];
746
747     if (pcap_lookupnet(iface, &netnum, &netmask, lookup_net_err_str) < 0) {
748         /*
749          * Well, we can't get the netmask for this interface; it's used
750          * only for filters that check for broadcast IP addresses, so
751          * we just punt and use 0.  It might be nice to warn the user,
752          * but that's a pain in a GUI application, as it'd involve popping
753          * up a message box, and it's not clear how often this would make
754          * a difference (only filters that check for IP broadcast addresses
755          * use the netmask).
756          */
757         /*cmdarg_err(
758           "Warning:  Couldn't obtain netmask info (%s).", lookup_net_err_str);*/
759         netmask = 0;
760     }
761
762     /*
763      * Sigh.  Older versions of libpcap don't properly declare the
764      * third argument to pcap_compile() as a const pointer.  Cast
765      * away the warning.
766      */
767     if (pcap_compile(pcap_h, fcode, (char *)cfilter, 1, netmask) < 0)
768         return FALSE;
769     return TRUE;
770 }
771
772 #ifdef HAVE_BPF_IMAGE
773 static gboolean
774 show_filter_code(capture_options *capture_opts)
775 {
776     interface_options interface_opts;
777     pcap_t *pcap_h;
778     gchar open_err_str[PCAP_ERRBUF_SIZE];
779     char errmsg[MSG_MAX_LENGTH+1];
780     char secondary_errmsg[MSG_MAX_LENGTH+1];
781     struct bpf_program fcode;
782     struct bpf_insn *insn;
783     u_int i;
784     guint j;
785
786     for (j = 0; j < capture_opts->ifaces->len; j++) {
787         interface_opts = g_array_index(capture_opts->ifaces, interface_options, j);
788         pcap_h = open_capture_device(&interface_opts, &open_err_str);
789         if (pcap_h == NULL) {
790             /* Open failed; get messages */
791             get_capture_device_open_failure_messages(open_err_str,
792                                                      interface_opts.name,
793                                                      errmsg, sizeof errmsg,
794                                                      secondary_errmsg,
795                                                      sizeof secondary_errmsg);
796             /* And report them */
797             report_capture_error(errmsg, secondary_errmsg);
798             return FALSE;
799         }
800
801         /* Set the link-layer type. */
802         if (!set_pcap_linktype(pcap_h, interface_opts.linktype, interface_opts.name,
803                                errmsg, sizeof errmsg,
804                                secondary_errmsg, sizeof secondary_errmsg)) {
805             pcap_close(pcap_h);
806             report_capture_error(errmsg, secondary_errmsg);
807             return FALSE;
808         }
809
810         /* OK, try to compile the capture filter. */
811         if (!compile_capture_filter(interface_opts.name, pcap_h, &fcode,
812                                     interface_opts.cfilter)) {
813             pcap_close(pcap_h);
814             report_cfilter_error(capture_opts, j, errmsg);
815             return FALSE;
816         }
817         pcap_close(pcap_h);
818
819         /* Now print the filter code. */
820         insn = fcode.bf_insns;
821
822         for (i = 0; i < fcode.bf_len; insn++, i++)
823             printf("%s\n", bpf_image(insn, i));
824     }
825     /* If not using libcap: we now can now set euid/egid to ruid/rgid         */
826     /*  to remove any suid privileges.                                        */
827     /* If using libcap: we can now remove NET_RAW and NET_ADMIN capabilities  */
828     /*  (euid/egid have already previously been set to ruid/rgid.             */
829     /* (See comment in main() for details)                                    */
830 #ifndef HAVE_LIBCAP
831     relinquish_special_privs_perm();
832 #else
833     relinquish_all_capabilities();
834 #endif
835     if (capture_child) {
836         /* Let our parent know we succeeded. */
837         pipe_write_block(2, SP_SUCCESS, NULL);
838     }
839     return TRUE;
840 }
841 #endif
842
843 /*
844  * capture_interface_list() is expected to do the right thing to get
845  * a list of interfaces.
846  *
847  * In most of the programs in the Wireshark suite, "the right thing"
848  * is to run dumpcap and ask it for the list, because dumpcap may
849  * be the only program in the suite with enough privileges to get
850  * the list.
851  *
852  * In dumpcap itself, however, we obviously can't run dumpcap to
853  * ask for the list.  Therefore, our capture_interface_list() should
854  * just call get_interface_list().
855  */
856 GList *
857 capture_interface_list(int *err, char **err_str)
858 {
859     return get_interface_list(err, err_str);
860 }
861
862 /*
863  * Get the data-link type for a libpcap device.
864  * This works around AIX 5.x's non-standard and incompatible-with-the-
865  * rest-of-the-universe libpcap.
866  */
867 static int
868 get_pcap_linktype(pcap_t *pch, const char *devname
869 #ifndef _AIX
870         _U_
871 #endif
872 )
873 {
874     int linktype;
875 #ifdef _AIX
876     const char *ifacename;
877 #endif
878
879     linktype = pcap_datalink(pch);
880 #ifdef _AIX
881
882     /*
883      * The libpcap that comes with AIX 5.x uses RFC 1573 ifType values
884      * rather than DLT_ values for link-layer types; the ifType values
885      * for LAN devices are:
886      *
887      *  Ethernet        6
888      *  802.3           7
889      *  Token Ring      9
890      *  FDDI            15
891      *
892      * and the ifType value for a loopback device is 24.
893      *
894      * The AIX names for LAN devices begin with:
895      *
896      *  Ethernet                en
897      *  802.3                   et
898      *  Token Ring              tr
899      *  FDDI                    fi
900      *
901      * and the AIX names for loopback devices begin with "lo".
902      *
903      * (The difference between "Ethernet" and "802.3" is presumably
904      * whether packets have an Ethernet header, with a packet type,
905      * or an 802.3 header, with a packet length, followed by an 802.2
906      * header and possibly a SNAP header.)
907      *
908      * If the device name matches "linktype" interpreted as an ifType
909      * value, rather than as a DLT_ value, we will assume this is AIX's
910      * non-standard, incompatible libpcap, rather than a standard libpcap,
911      * and will map the link-layer type to the standard DLT_ value for
912      * that link-layer type, as that's what the rest of Wireshark expects.
913      *
914      * (This means the capture files won't be readable by a tcpdump
915      * linked with AIX's non-standard libpcap, but so it goes.  They
916      * *will* be readable by standard versions of tcpdump, Wireshark,
917      * and so on.)
918      *
919      * XXX - if we conclude we're using AIX libpcap, should we also
920      * set a flag to cause us to assume the time stamps are in
921      * seconds-and-nanoseconds form, and to convert them to
922      * seconds-and-microseconds form before processing them and
923      * writing them out?
924      */
925
926     /*
927      * Find the last component of the device name, which is the
928      * interface name.
929      */
930     ifacename = strchr(devname, '/');
931     if (ifacename == NULL)
932         ifacename = devname;
933
934     /* See if it matches any of the LAN device names. */
935     if (strncmp(ifacename, "en", 2) == 0) {
936         if (linktype == 6) {
937             /*
938              * That's the RFC 1573 value for Ethernet; map it to DLT_EN10MB.
939              */
940             linktype = 1;
941         }
942     } else if (strncmp(ifacename, "et", 2) == 0) {
943         if (linktype == 7) {
944             /*
945              * That's the RFC 1573 value for 802.3; map it to DLT_EN10MB.
946              * (libpcap, tcpdump, Wireshark, etc. don't care if it's Ethernet
947              * or 802.3.)
948              */
949             linktype = 1;
950         }
951     } else if (strncmp(ifacename, "tr", 2) == 0) {
952         if (linktype == 9) {
953             /*
954              * That's the RFC 1573 value for 802.5 (Token Ring); map it to
955              * DLT_IEEE802, which is what's used for Token Ring.
956              */
957             linktype = 6;
958         }
959     } else if (strncmp(ifacename, "fi", 2) == 0) {
960         if (linktype == 15) {
961             /*
962              * That's the RFC 1573 value for FDDI; map it to DLT_FDDI.
963              */
964             linktype = 10;
965         }
966     } else if (strncmp(ifacename, "lo", 2) == 0) {
967         if (linktype == 24) {
968             /*
969              * That's the RFC 1573 value for "software loopback" devices; map it
970              * to DLT_NULL, which is what's used for loopback devices on BSD.
971              */
972             linktype = 0;
973         }
974     }
975 #endif
976
977     return linktype;
978 }
979
980 static data_link_info_t *
981 create_data_link_info(int dlt)
982 {
983     data_link_info_t *data_link_info;
984     const char *text;
985
986     data_link_info = (data_link_info_t *)g_malloc(sizeof (data_link_info_t));
987     data_link_info->dlt = dlt;
988     text = pcap_datalink_val_to_name(dlt);
989     if (text != NULL)
990         data_link_info->name = g_strdup(text);
991     else
992         data_link_info->name = g_strdup_printf("DLT %d", dlt);
993     text = pcap_datalink_val_to_description(dlt);
994     if (text != NULL)
995         data_link_info->description = g_strdup(text);
996     else
997         data_link_info->description = NULL;
998     return data_link_info;
999 }
1000
1001 /*
1002  * Get the capabilities of a network device.
1003  */
1004 static if_capabilities_t *
1005 get_if_capabilities(const char *devname, gboolean monitor_mode
1006 #ifndef HAVE_PCAP_CREATE
1007         _U_
1008 #endif
1009 , char **err_str)
1010 {
1011     if_capabilities_t *caps;
1012     char errbuf[PCAP_ERRBUF_SIZE];
1013     pcap_t *pch;
1014 #ifdef HAVE_PCAP_CREATE
1015     int status;
1016 #endif
1017     int deflt;
1018 #ifdef HAVE_PCAP_LIST_DATALINKS
1019     int *linktypes;
1020     int i, nlt;
1021 #endif
1022     data_link_info_t *data_link_info;
1023
1024     /*
1025      * Allocate the interface capabilities structure.
1026      */
1027     caps = g_malloc(sizeof *caps);
1028
1029 #ifdef HAVE_PCAP_OPEN
1030     pch = pcap_open(devname, MIN_PACKET_SIZE, 0, 0, NULL, errbuf);
1031     caps->can_set_rfmon = FALSE;
1032     if (pch == NULL) {
1033         if (err_str != NULL)
1034             *err_str = g_strdup(errbuf);
1035         g_free(caps);
1036         return NULL;
1037     }
1038 #elif defined(HAVE_PCAP_CREATE)
1039     pch = pcap_create(devname, errbuf);
1040     if (pch == NULL) {
1041         if (err_str != NULL)
1042             *err_str = g_strdup(errbuf);
1043         g_free(caps);
1044         return NULL;
1045     }
1046     status = pcap_can_set_rfmon(pch);
1047     if (status < 0) {
1048         /* Error. */
1049         if (status == PCAP_ERROR)
1050             *err_str = g_strdup_printf("pcap_can_set_rfmon() failed: %s",
1051                                        pcap_geterr(pch));
1052         else
1053             *err_str = g_strdup(pcap_statustostr(status));
1054         pcap_close(pch);
1055         g_free(caps);
1056         return NULL;
1057     }
1058     if (status == 0)
1059         caps->can_set_rfmon = FALSE;
1060     else if (status == 1) {
1061         caps->can_set_rfmon = TRUE;
1062         if (monitor_mode)
1063             pcap_set_rfmon(pch, 1);
1064     } else {
1065         if (err_str != NULL) {
1066             *err_str = g_strdup_printf("pcap_can_set_rfmon() returned %d",
1067                                        status);
1068         }
1069         pcap_close(pch);
1070         g_free(caps);
1071         return NULL;
1072     }
1073
1074     status = pcap_activate(pch);
1075     if (status < 0) {
1076         /* Error.  We ignore warnings (status > 0). */
1077         if (err_str != NULL) {
1078             if (status == PCAP_ERROR)
1079                 *err_str = g_strdup_printf("pcap_activate() failed: %s",
1080                                            pcap_geterr(pch));
1081             else
1082                 *err_str = g_strdup(pcap_statustostr(status));
1083         }
1084         pcap_close(pch);
1085         g_free(caps);
1086         return NULL;
1087     }
1088 #else
1089     pch = pcap_open_live(devname, MIN_PACKET_SIZE, 0, 0, errbuf);
1090     caps->can_set_rfmon = FALSE;
1091     if (pch == NULL) {
1092         if (err_str != NULL)
1093             *err_str = g_strdup(errbuf);
1094         g_free(caps);
1095         return NULL;
1096     }
1097 #endif
1098     deflt = get_pcap_linktype(pch, devname);
1099 #ifdef HAVE_PCAP_LIST_DATALINKS
1100     nlt = pcap_list_datalinks(pch, &linktypes);
1101     if (nlt == 0 || linktypes == NULL) {
1102         pcap_close(pch);
1103         if (err_str != NULL)
1104             *err_str = NULL; /* an empty list doesn't mean an error */
1105         return NULL;
1106     }
1107     caps->data_link_types = NULL;
1108     for (i = 0; i < nlt; i++) {
1109         data_link_info = create_data_link_info(linktypes[i]);
1110
1111         /*
1112          * XXX - for 802.11, make the most detailed 802.11
1113          * version the default, rather than the one the
1114          * device has as the default?
1115          */
1116         if (linktypes[i] == deflt)
1117             caps->data_link_types = g_list_prepend(caps->data_link_types,
1118                                                    data_link_info);
1119         else
1120             caps->data_link_types = g_list_append(caps->data_link_types,
1121                                                   data_link_info);
1122     }
1123 #ifdef HAVE_PCAP_FREE_DATALINKS
1124     pcap_free_datalinks(linktypes);
1125 #else
1126     /*
1127      * In Windows, there's no guarantee that if you have a library
1128      * built with one version of the MSVC++ run-time library, and
1129      * it returns a pointer to allocated data, you can free that
1130      * data from a program linked with another version of the
1131      * MSVC++ run-time library.
1132      *
1133      * This is not an issue on UN*X.
1134      *
1135      * See the mail threads starting at
1136      *
1137      *    http://www.winpcap.org/pipermail/winpcap-users/2006-September/001421.html
1138      *
1139      * and
1140      *
1141      *    http://www.winpcap.org/pipermail/winpcap-users/2008-May/002498.html
1142      */
1143 #ifndef _WIN32
1144 #define xx_free free  /* hack so checkAPIs doesn't complain */
1145     xx_free(linktypes);
1146 #endif /* _WIN32 */
1147 #endif /* HAVE_PCAP_FREE_DATALINKS */
1148 #else /* HAVE_PCAP_LIST_DATALINKS */
1149
1150     data_link_info = create_data_link_info(deflt);
1151     caps->data_link_types = g_list_append(caps->data_link_types,
1152                                           data_link_info);
1153 #endif /* HAVE_PCAP_LIST_DATALINKS */
1154
1155     pcap_close(pch);
1156
1157     if (err_str != NULL)
1158         *err_str = NULL;
1159     return caps;
1160 }
1161
1162 #define ADDRSTRLEN 46 /* Covers IPv4 & IPv6 */
1163 static void
1164 print_machine_readable_interfaces(GList *if_list)
1165 {
1166     int         i;
1167     GList       *if_entry;
1168     if_info_t   *if_info;
1169     GSList      *addr;
1170     if_addr_t   *if_addr;
1171     char        addr_str[ADDRSTRLEN];
1172
1173     if (capture_child) {
1174         /* Let our parent know we succeeded. */
1175         pipe_write_block(2, SP_SUCCESS, NULL);
1176     }
1177
1178     i = 1;  /* Interface id number */
1179     for (if_entry = g_list_first(if_list); if_entry != NULL;
1180          if_entry = g_list_next(if_entry)) {
1181         if_info = (if_info_t *)if_entry->data;
1182         printf("%d. %s", i++, if_info->name);
1183
1184         /*
1185          * Print the contents of the if_entry struct in a parseable format.
1186          * Each if_entry element is tab-separated.  Addresses are comma-
1187          * separated.
1188          */
1189         /* XXX - Make sure our description doesn't contain a tab */
1190         if (if_info->description != NULL)
1191             printf("\t%s\t", if_info->description);
1192         else
1193             printf("\t\t");
1194
1195         for(addr = g_slist_nth(if_info->addrs, 0); addr != NULL;
1196                     addr = g_slist_next(addr)) {
1197             if (addr != g_slist_nth(if_info->addrs, 0))
1198                 printf(",");
1199
1200             if_addr = (if_addr_t *)addr->data;
1201             switch(if_addr->ifat_type) {
1202             case IF_AT_IPv4:
1203                 if (inet_ntop(AF_INET, &if_addr->addr.ip4_addr, addr_str,
1204                               ADDRSTRLEN)) {
1205                     printf("%s", addr_str);
1206                 } else {
1207                     printf("<unknown IPv4>");
1208                 }
1209                 break;
1210             case IF_AT_IPv6:
1211                 if (inet_ntop(AF_INET6, &if_addr->addr.ip6_addr,
1212                               addr_str, ADDRSTRLEN)) {
1213                     printf("%s", addr_str);
1214                 } else {
1215                     printf("<unknown IPv6>");
1216                 }
1217                 break;
1218             default:
1219                 printf("<type unknown %u>", if_addr->ifat_type);
1220             }
1221         }
1222
1223         if (if_info->loopback)
1224             printf("\tloopback");
1225         else
1226             printf("\tnetwork");
1227
1228         printf("\n");
1229     }
1230 }
1231
1232 /*
1233  * If you change the machine-readable output format of this function,
1234  * you MUST update capture_ifinfo.c:capture_get_if_capabilities() accordingly!
1235  */
1236 static void
1237 print_machine_readable_if_capabilities(if_capabilities_t *caps)
1238 {
1239     GList *lt_entry;
1240     data_link_info_t *data_link_info;
1241     const gchar *desc_str;
1242
1243     if (capture_child) {
1244         /* Let our parent know we succeeded. */
1245         pipe_write_block(2, SP_SUCCESS, NULL);
1246     }
1247
1248     if (caps->can_set_rfmon)
1249         printf("1\n");
1250     else
1251         printf("0\n");
1252     for (lt_entry = caps->data_link_types; lt_entry != NULL;
1253          lt_entry = g_list_next(lt_entry)) {
1254       data_link_info = (data_link_info_t *)lt_entry->data;
1255       if (data_link_info->description != NULL)
1256         desc_str = data_link_info->description;
1257       else
1258         desc_str = "(not supported)";
1259       printf("%d\t%s\t%s\n", data_link_info->dlt, data_link_info->name,
1260              desc_str);
1261     }
1262 }
1263
1264 typedef struct {
1265     char *name;
1266     pcap_t *pch;
1267 } if_stat_t;
1268
1269 /* Print the number of packets captured for each interface until we're killed. */
1270 static int
1271 print_statistics_loop(gboolean machine_readable)
1272 {
1273     GList       *if_list, *if_entry, *stat_list = NULL, *stat_entry;
1274     if_info_t   *if_info;
1275     if_stat_t   *if_stat;
1276     int         err;
1277     gchar       *err_str;
1278     pcap_t      *pch;
1279     char        errbuf[PCAP_ERRBUF_SIZE];
1280     struct pcap_stat ps;
1281
1282     if_list = get_interface_list(&err, &err_str);
1283     if (if_list == NULL) {
1284         switch (err) {
1285         case CANT_GET_INTERFACE_LIST:
1286             cmdarg_err("%s", err_str);
1287             g_free(err_str);
1288             break;
1289
1290         case NO_INTERFACES_FOUND:
1291             cmdarg_err("There are no interfaces on which a capture can be done");
1292             break;
1293         }
1294         return err;
1295     }
1296
1297     for (if_entry = g_list_first(if_list); if_entry != NULL; if_entry = g_list_next(if_entry)) {
1298         if_info = (if_info_t *)if_entry->data;
1299 #ifdef HAVE_PCAP_OPEN
1300         pch = pcap_open(if_info->name, MIN_PACKET_SIZE, 0, 0, NULL, errbuf);
1301 #else
1302         pch = pcap_open_live(if_info->name, MIN_PACKET_SIZE, 0, 0, errbuf);
1303 #endif
1304
1305         if (pch) {
1306             if_stat = (if_stat_t *)g_malloc(sizeof(if_stat_t));
1307             if_stat->name = g_strdup(if_info->name);
1308             if_stat->pch = pch;
1309             stat_list = g_list_append(stat_list, if_stat);
1310         }
1311     }
1312
1313     if (!stat_list) {
1314         cmdarg_err("There are no interfaces on which a capture can be done");
1315         return 2;
1316     }
1317
1318     if (capture_child) {
1319         /* Let our parent know we succeeded. */
1320         pipe_write_block(2, SP_SUCCESS, NULL);
1321     }
1322
1323     if (!machine_readable) {
1324         printf("%-15s  %10s  %10s\n", "Interface", "Received",
1325             "Dropped");
1326     }
1327
1328     global_ld.go = TRUE;
1329     while (global_ld.go) {
1330         for (stat_entry = g_list_first(stat_list); stat_entry != NULL; stat_entry = g_list_next(stat_entry)) {
1331             if_stat = (if_stat_t *)stat_entry->data;
1332             pcap_stats(if_stat->pch, &ps);
1333
1334             if (!machine_readable) {
1335                 printf("%-15s  %10u  %10u\n", if_stat->name,
1336                     ps.ps_recv, ps.ps_drop);
1337             } else {
1338                 printf("%s\t%u\t%u\n", if_stat->name,
1339                     ps.ps_recv, ps.ps_drop);
1340                 fflush(stdout);
1341             }
1342         }
1343 #ifdef _WIN32
1344         Sleep(1 * 1000);
1345 #else
1346         sleep(1);
1347 #endif
1348     }
1349
1350     /* XXX - Not reached.  Should we look for 'q' in stdin? */
1351     for (stat_entry = g_list_first(stat_list); stat_entry != NULL; stat_entry = g_list_next(stat_entry)) {
1352         if_stat = (if_stat_t *)stat_entry->data;
1353         pcap_close(if_stat->pch);
1354         g_free(if_stat->name);
1355         g_free(if_stat);
1356     }
1357     g_list_free(stat_list);
1358     free_interface_list(if_list);
1359
1360     return 0;
1361 }
1362
1363
1364 #ifdef _WIN32
1365 static BOOL WINAPI
1366 capture_cleanup_handler(DWORD dwCtrlType)
1367 {
1368     /* CTRL_C_EVENT is sort of like SIGINT, CTRL_BREAK_EVENT is unique to
1369        Windows, CTRL_CLOSE_EVENT is sort of like SIGHUP, CTRL_LOGOFF_EVENT
1370        is also sort of like SIGHUP, and CTRL_SHUTDOWN_EVENT is sort of
1371        like SIGTERM at least when the machine's shutting down.
1372
1373        For now, if we're running as a command rather than a capture child,
1374        we handle all but CTRL_LOGOFF_EVENT as indications that we should
1375        clean up and quit, just as we handle SIGINT, SIGHUP, and SIGTERM
1376        in that way on UN*X.
1377
1378        If we're not running as a capture child, we might be running as
1379        a service; ignore CTRL_LOGOFF_EVENT, so we keep running after the
1380        user logs out.  (XXX - can we explicitly check whether we're
1381        running as a service?) */
1382
1383     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO,
1384         "Console: Control signal");
1385     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG,
1386         "Console: Control signal, CtrlType: %u", dwCtrlType);
1387
1388     /* Keep capture running if we're a service and a user logs off */
1389     if (capture_child || (dwCtrlType != CTRL_LOGOFF_EVENT)) {
1390         capture_loop_stop();
1391         return TRUE;
1392     } else {
1393         return FALSE;
1394     }
1395 }
1396 #else
1397 static void
1398 capture_cleanup_handler(int signum _U_)
1399 {
1400     /* On UN*X, we cleanly shut down the capture on SIGINT, SIGHUP, and
1401        SIGTERM.  We assume that if the user wanted it to keep running
1402        after they logged out, they'd have nohupped it. */
1403
1404     /* Note: don't call g_log() in the signal handler: if we happened to be in
1405      * g_log() in process context when the signal came in, g_log will detect
1406      * the "recursion" and abort.
1407      */
1408
1409     capture_loop_stop();
1410 }
1411 #endif
1412
1413
1414 static void
1415 report_capture_count(gboolean reportit)
1416 {
1417     /* Don't print this if we're a capture child. */
1418     if (!capture_child && reportit) {
1419         fprintf(stderr, "\rPackets captured: %u\n", global_ld.packet_count);
1420         /* stderr could be line buffered */
1421         fflush(stderr);
1422     }
1423 }
1424
1425
1426 #ifdef SIGINFO
1427 static void
1428 report_counts_for_siginfo(void)
1429 {
1430     report_capture_count(quiet);
1431     infoprint = FALSE; /* we just reported it */
1432 }
1433
1434 static void
1435 report_counts_siginfo(int signum _U_)
1436 {
1437     int sav_errno = errno;
1438
1439     /* If we've been told to delay printing, just set a flag asking
1440        that we print counts (if we're supposed to), otherwise print
1441        the count of packets captured (if we're supposed to). */
1442     if (infodelay)
1443         infoprint = TRUE;
1444     else
1445         report_counts_for_siginfo();
1446     errno = sav_errno;
1447 }
1448 #endif /* SIGINFO */
1449
1450 static void
1451 exit_main(int status)
1452 {
1453 #ifdef _WIN32
1454     /* Shutdown windows sockets */
1455     WSACleanup();
1456
1457     /* can be helpful for debugging */
1458 #ifdef DEBUG_DUMPCAP
1459     printf("Press any key\n");
1460     _getch();
1461 #endif
1462
1463 #endif /* _WIN32 */
1464
1465     exit(status);
1466 }
1467
1468 #ifdef HAVE_LIBCAP
1469 /*
1470  * If we were linked with libcap (not libpcap), make sure we have
1471  * CAP_NET_ADMIN and CAP_NET_RAW, then relinquish our permissions.
1472  * (See comment in main() for details)
1473  */
1474 static void
1475 relinquish_privs_except_capture(void)
1476 {
1477     /* If 'started_with_special_privs' (ie: suid) then enable for
1478      *  ourself the  NET_ADMIN and NET_RAW capabilities and then
1479      *  drop our suid privileges.
1480      *
1481      * CAP_NET_ADMIN: Promiscuous mode and a truckload of other
1482      *                stuff we don't need (and shouldn't have).
1483      * CAP_NET_RAW:   Packet capture (raw sockets).
1484      */
1485
1486     if (started_with_special_privs()) {
1487         cap_value_t cap_list[2] = { CAP_NET_ADMIN, CAP_NET_RAW };
1488         int cl_len = sizeof(cap_list) / sizeof(cap_value_t);
1489
1490         cap_t caps = cap_init();    /* all capabilities initialized to off */
1491
1492         print_caps("Pre drop, pre set");
1493
1494         if (prctl(PR_SET_KEEPCAPS, 1, 0, 0, 0) == -1) {
1495             cmdarg_err("prctl() fail return: %s", g_strerror(errno));
1496         }
1497
1498         cap_set_flag(caps, CAP_PERMITTED,   cl_len, cap_list, CAP_SET);
1499         cap_set_flag(caps, CAP_INHERITABLE, cl_len, cap_list, CAP_SET);
1500
1501         if (cap_set_proc(caps)) {
1502             cmdarg_err("cap_set_proc() fail return: %s", g_strerror(errno));
1503         }
1504         print_caps("Pre drop, post set");
1505
1506         relinquish_special_privs_perm();
1507
1508         print_caps("Post drop, pre set");
1509         cap_set_flag(caps, CAP_EFFECTIVE,   cl_len, cap_list, CAP_SET);
1510         if (cap_set_proc(caps)) {
1511             cmdarg_err("cap_set_proc() fail return: %s", g_strerror(errno));
1512         }
1513         print_caps("Post drop, post set");
1514
1515         cap_free(caps);
1516     }
1517 }
1518
1519 #endif /* HAVE_LIBCAP */
1520
1521 /* Take care of byte order in the libpcap headers read from pipes.
1522  * (function taken from wiretap/libpcap.c) */
1523 static void
1524 cap_pipe_adjust_header(gboolean byte_swapped, struct pcap_hdr *hdr, struct pcaprec_hdr *rechdr)
1525 {
1526     if (byte_swapped) {
1527         /* Byte-swap the record header fields. */
1528         rechdr->ts_sec = BSWAP32(rechdr->ts_sec);
1529         rechdr->ts_usec = BSWAP32(rechdr->ts_usec);
1530         rechdr->incl_len = BSWAP32(rechdr->incl_len);
1531         rechdr->orig_len = BSWAP32(rechdr->orig_len);
1532     }
1533
1534     /* In file format version 2.3, the "incl_len" and "orig_len" fields were
1535        swapped, in order to match the BPF header layout.
1536
1537        Unfortunately, some files were, according to a comment in the "libpcap"
1538        source, written with version 2.3 in their headers but without the
1539        interchanged fields, so if "incl_len" is greater than "orig_len" - which
1540        would make no sense - we assume that we need to swap them.  */
1541     if (hdr->version_major == 2 &&
1542         (hdr->version_minor < 3 ||
1543          (hdr->version_minor == 3 && rechdr->incl_len > rechdr->orig_len))) {
1544         guint32 temp;
1545
1546         temp = rechdr->orig_len;
1547         rechdr->orig_len = rechdr->incl_len;
1548         rechdr->incl_len = temp;
1549     }
1550 }
1551
1552 #if defined(USE_THREADS) && defined(_WIN32)
1553 /*
1554  * Thread function that reads from a pipe and pushes the data
1555  * to the main application thread.
1556  */
1557 /*
1558  * XXX Right now we use async queues for basic signaling. The main thread
1559  * sets cap_pipe_buf and cap_bytes_to_read, then pushes an item onto
1560  * cap_pipe_pending_q which triggers a read in the cap_pipe_read thread.
1561  * Iff the read is successful cap_pipe_read pushes an item onto
1562  * cap_pipe_done_q, otherwise an error is signaled. No data is passed in
1563  * the queues themselves (yet).
1564  *
1565  * We might want to move some of the cap_pipe_dispatch logic here so that
1566  * we can let cap_pipe_read run independently, queuing up multiple reads
1567  * for the main thread (and possibly get rid of cap_pipe_read_mtx).
1568  */
1569 static void *cap_pipe_read(void *arg)
1570 {
1571     pcap_options *pcap_opts;
1572     int bytes_read;
1573 #ifdef _WIN32
1574     BOOL res;
1575     DWORD b, last_err;
1576 #else /* _WIN32 */
1577     int b;
1578 #endif /* _WIN32 */
1579
1580     pcap_opts = (pcap_options *)arg;
1581     while (pcap_opts->cap_pipe_err == PIPOK) {
1582         g_async_queue_pop(pcap_opts->cap_pipe_pending_q); /* Wait for our cue (ahem) from the main thread */
1583         g_mutex_lock(pcap_opts->cap_pipe_read_mtx);
1584         bytes_read = 0;
1585         while (bytes_read < (int) pcap_opts->cap_pipe_bytes_to_read) {
1586 #ifdef _WIN32
1587             /* If we try to use read() on a named pipe on Windows with partial
1588              * data it appears to return EOF.
1589              */
1590             res = ReadFile(pcap_opts->cap_pipe_h, pcap_opts->cap_pipe_buf+bytes_read,
1591                            pcap_opts->cap_pipe_bytes_to_read - bytes_read,
1592                            &b, NULL);
1593
1594             bytes_read += b;
1595             if (!res) {
1596                 last_err = GetLastError();
1597                 if (last_err == ERROR_MORE_DATA) {
1598                     continue;
1599                 } else if (last_err == ERROR_HANDLE_EOF || last_err == ERROR_BROKEN_PIPE || last_err == ERROR_PIPE_NOT_CONNECTED) {
1600                     pcap_opts->cap_pipe_err = PIPEOF;
1601                     bytes_read = 0;
1602                     break;
1603                 }
1604                 pcap_opts->cap_pipe_err = PIPERR;
1605                 bytes_read = -1;
1606                 break;
1607             } else if (b == 0 && pcap_opts->cap_pipe_bytes_to_read > 0) {
1608                 pcap_opts->cap_pipe_err = PIPEOF;
1609                 bytes_read = 0;
1610                 break;
1611             }
1612 #else /* _WIN32 */
1613             b = read(pcap_opts->cap_pipe_fd, pcap_opts->cap_pipe_buf+bytes_read,
1614                      pcap_opts->cap_pipe_bytes_to_read - bytes_read);
1615             if (b <= 0) {
1616                 if (b == 0) {
1617                     pcap_opts->cap_pipe_err = PIPEOF;
1618                     bytes_read = 0;
1619                     break;
1620                 } else {
1621                     pcap_opts->cap_pipe_err = PIPERR;
1622                     bytes_read = -1;
1623                     break;
1624                 }
1625             } else {
1626                 bytes_read += b;
1627             }
1628 #endif /*_WIN32 */
1629         }
1630         pcap_opts->cap_pipe_bytes_read = bytes_read;
1631         if (pcap_opts->cap_pipe_bytes_read >= pcap_opts->cap_pipe_bytes_to_read) {
1632             g_async_queue_push(pcap_opts->cap_pipe_done_q, pcap_opts->cap_pipe_buf); /* Any non-NULL value will do */
1633         }
1634         g_mutex_unlock(pcap_opts->cap_pipe_read_mtx);
1635     }
1636     return NULL;
1637 }
1638 #endif
1639
1640 #if (!(defined(USE_THREADS) && defined(_WIN32))) || defined(MUST_DO_SELECT)
1641 /* Provide select() functionality for a single file descriptor
1642  * on UNIX/POSIX. Windows uses cap_pipe_read via a thread.
1643  *
1644  * Returns the same values as select.
1645  */
1646 static int
1647 cap_pipe_select(int pipe_fd)
1648 {
1649     fd_set      rfds;
1650     struct timeval timeout;
1651
1652     FD_ZERO(&rfds);
1653     FD_SET(pipe_fd, &rfds);
1654
1655     timeout.tv_sec = PIPE_READ_TIMEOUT / 1000000;
1656     timeout.tv_usec = PIPE_READ_TIMEOUT % 1000000;
1657
1658     return select(pipe_fd+1, &rfds, NULL, NULL, &timeout);
1659 }
1660 #endif
1661
1662
1663 /* Mimic pcap_open_live() for pipe captures
1664
1665  * We check if "pipename" is "-" (stdin), a AF_UNIX socket, or a FIFO,
1666  * open it, and read the header.
1667  *
1668  * N.B. : we can't read the libpcap formats used in RedHat 6.1 or SuSE 6.3
1669  * because we can't seek on pipes (see wiretap/libpcap.c for details) */
1670 static void
1671 cap_pipe_open_live(char *pipename,
1672                    pcap_options *pcap_opts,
1673                    struct pcap_hdr *hdr,
1674                    char *errmsg, int errmsgl)
1675 {
1676 #ifndef _WIN32
1677     ws_statb64   pipe_stat;
1678     struct sockaddr_un sa;
1679     int          b;
1680     int          fd;
1681 #else /* _WIN32 */
1682 #if 1
1683     char *pncopy, *pos;
1684     wchar_t *err_str;
1685 #endif
1686 #endif
1687 #if !(defined(USE_THREADS) && defined(_WIN32))
1688     int          sel_ret;
1689     unsigned int bytes_read;
1690 #endif
1691     guint32       magic = 0;
1692
1693 #ifndef _WIN32
1694     pcap_opts->cap_pipe_fd = -1;
1695 #else
1696     pcap_opts->cap_pipe_h = INVALID_HANDLE_VALUE;
1697 #endif
1698     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "cap_pipe_open_live: %s", pipename);
1699
1700     /*
1701      * XXX - this blocks until a pcap per-file header has been written to
1702      * the pipe, so it could block indefinitely.
1703      */
1704     if (strcmp(pipename, "-") == 0) {
1705 #ifndef _WIN32
1706         fd = 0; /* read from stdin */
1707 #else /* _WIN32 */
1708         pcap_opts->cap_pipe_h = GetStdHandle(STD_INPUT_HANDLE);
1709 #endif  /* _WIN32 */
1710     } else {
1711 #ifndef _WIN32
1712         if (ws_stat64(pipename, &pipe_stat) < 0) {
1713             if (errno == ENOENT || errno == ENOTDIR)
1714                 pcap_opts->cap_pipe_err = PIPNEXIST;
1715             else {
1716                 g_snprintf(errmsg, errmsgl,
1717                            "The capture session could not be initiated "
1718                            "due to error getting information on pipe/socket: %s", g_strerror(errno));
1719                 pcap_opts->cap_pipe_err = PIPERR;
1720             }
1721             return;
1722         }
1723         if (S_ISFIFO(pipe_stat.st_mode)) {
1724             fd = ws_open(pipename, O_RDONLY | O_NONBLOCK, 0000 /* no creation so don't matter */);
1725             if (fd == -1) {
1726                 g_snprintf(errmsg, errmsgl,
1727                            "The capture session could not be initiated "
1728                            "due to error on pipe open: %s", g_strerror(errno));
1729                 pcap_opts->cap_pipe_err = PIPERR;
1730                 return;
1731             }
1732         } else if (S_ISSOCK(pipe_stat.st_mode)) {
1733             fd = socket(AF_UNIX, SOCK_STREAM, 0);
1734             if (fd == -1) {
1735                 g_snprintf(errmsg, errmsgl,
1736                            "The capture session could not be initiated "
1737                            "due to error on socket create: %s", g_strerror(errno));
1738                 pcap_opts->cap_pipe_err = PIPERR;
1739                 return;
1740             }
1741             sa.sun_family = AF_UNIX;
1742             /*
1743              * The Single UNIX Specification says:
1744              *
1745              *   The size of sun_path has intentionally been left undefined.
1746              *   This is because different implementations use different sizes.
1747              *   For example, 4.3 BSD uses a size of 108, and 4.4 BSD uses a size
1748              *   of 104. Since most implementations originate from BSD versions,
1749              *   the size is typically in the range 92 to 108.
1750              *
1751              *   Applications should not assume a particular length for sun_path
1752              *   or assume that it can hold {_POSIX_PATH_MAX} bytes (256).
1753              *
1754              * It also says
1755              *
1756              *   The <sys/un.h> header shall define the sockaddr_un structure,
1757              *   which shall include at least the following members:
1758              *
1759              *   sa_family_t  sun_family  Address family.
1760              *   char         sun_path[]  Socket pathname.
1761              *
1762              * so we assume that it's an array, with a specified size,
1763              * and that the size reflects the maximum path length.
1764              */
1765             if (g_strlcpy(sa.sun_path, pipename, sizeof sa.sun_path) > sizeof sa.sun_path) {
1766                 /* Path name too long */
1767                 g_snprintf(errmsg, errmsgl,
1768                            "The capture session coud not be initiated "
1769                            "due to error on socket connect: Path name too long");
1770                 pcap_opts->cap_pipe_err = PIPERR;
1771                 return;
1772             }
1773             b = connect(fd, (struct sockaddr *)&sa, sizeof sa);
1774             if (b == -1) {
1775                 g_snprintf(errmsg, errmsgl,
1776                            "The capture session coud not be initiated "
1777                            "due to error on socket connect: %s", g_strerror(errno));
1778                 pcap_opts->cap_pipe_err = PIPERR;
1779                 return;
1780             }
1781         } else {
1782             if (S_ISCHR(pipe_stat.st_mode)) {
1783                 /*
1784                  * Assume the user specified an interface on a system where
1785                  * interfaces are in /dev.  Pretend we haven't seen it.
1786                  */
1787                 pcap_opts->cap_pipe_err = PIPNEXIST;
1788             } else
1789             {
1790                 g_snprintf(errmsg, errmsgl,
1791                            "The capture session could not be initiated because\n"
1792                            "\"%s\" is neither an interface nor a socket nor a pipe", pipename);
1793                 pcap_opts->cap_pipe_err = PIPERR;
1794             }
1795             return;
1796         }
1797 #else /* _WIN32 */
1798 #define PIPE_STR "\\pipe\\"
1799         /* Under Windows, named pipes _must_ have the form
1800          * "\\<server>\pipe\<pipename>".  <server> may be "." for localhost.
1801          */
1802         pncopy = g_strdup(pipename);
1803         if ( (pos=strstr(pncopy, "\\\\")) == pncopy) {
1804             pos = strchr(pncopy + 3, '\\');
1805             if (pos && g_ascii_strncasecmp(pos, PIPE_STR, strlen(PIPE_STR)) != 0)
1806                 pos = NULL;
1807         }
1808
1809         g_free(pncopy);
1810
1811         if (!pos) {
1812             g_snprintf(errmsg, errmsgl,
1813                        "The capture session could not be initiated because\n"
1814                        "\"%s\" is neither an interface nor a pipe", pipename);
1815             pcap_opts->cap_pipe_err = PIPNEXIST;
1816             return;
1817         }
1818
1819         /* Wait for the pipe to appear */
1820         while (1) {
1821             pcap_opts->cap_pipe_h = CreateFile(utf_8to16(pipename), GENERIC_READ, 0, NULL,
1822                                                OPEN_EXISTING, 0, NULL);
1823
1824             if (pcap_opts->cap_pipe_h != INVALID_HANDLE_VALUE)
1825                 break;
1826
1827             if (GetLastError() != ERROR_PIPE_BUSY) {
1828                 FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_ALLOCATE_BUFFER,
1829                               NULL, GetLastError(), 0, (LPTSTR) &err_str, 0, NULL);
1830                 g_snprintf(errmsg, errmsgl,
1831                            "The capture session on \"%s\" could not be started "
1832                            "due to error on pipe open: %s (error %d)",
1833                            pipename, utf_16to8(err_str), GetLastError());
1834                 LocalFree(err_str);
1835                 pcap_opts->cap_pipe_err = PIPERR;
1836                 return;
1837             }
1838
1839             if (!WaitNamedPipe(utf_8to16(pipename), 30 * 1000)) {
1840                 FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_ALLOCATE_BUFFER,
1841                               NULL, GetLastError(), 0, (LPTSTR) &err_str, 0, NULL);
1842                 g_snprintf(errmsg, errmsgl,
1843                            "The capture session on \"%s\" timed out during "
1844                            "pipe open: %s (error %d)",
1845                            pipename, utf_16to8(err_str), GetLastError());
1846                 LocalFree(err_str);
1847                 pcap_opts->cap_pipe_err = PIPERR;
1848                 return;
1849             }
1850         }
1851 #endif /* _WIN32 */
1852     }
1853
1854     pcap_opts->from_cap_pipe = TRUE;
1855
1856 #if !(defined(USE_THREADS) && defined(_WIN32))
1857     /* read the pcap header */
1858     bytes_read = 0;
1859     while (bytes_read < sizeof magic) {
1860         sel_ret = cap_pipe_select(fd);
1861         if (sel_ret < 0) {
1862             g_snprintf(errmsg, errmsgl,
1863                        "Unexpected error from select: %s", g_strerror(errno));
1864             goto error;
1865         } else if (sel_ret > 0) {
1866             b = read(fd, ((char *)&magic)+bytes_read, sizeof magic-bytes_read);
1867             if (b <= 0) {
1868                 if (b == 0)
1869                     g_snprintf(errmsg, errmsgl, "End of file on pipe magic during open");
1870                 else
1871                     g_snprintf(errmsg, errmsgl, "Error on pipe magic during open: %s",
1872                                g_strerror(errno));
1873                 goto error;
1874             }
1875             bytes_read += b;
1876         }
1877     }
1878 #else
1879     g_thread_create(&cap_pipe_read, pcap_opts, FALSE, NULL);
1880
1881     pcap_opts->cap_pipe_buf = (char *) &magic;
1882     pcap_opts->cap_pipe_bytes_read = 0;
1883     pcap_opts->cap_pipe_bytes_to_read = sizeof(magic);
1884     /* We don't have to worry about cap_pipe_read_mtx here */
1885     g_async_queue_push(pcap_opts->cap_pipe_pending_q, pcap_opts->cap_pipe_buf);
1886     g_async_queue_pop(pcap_opts->cap_pipe_done_q);
1887     if (pcap_opts->cap_pipe_bytes_read <= 0) {
1888         if (pcap_opts->cap_pipe_bytes_read == 0)
1889             g_snprintf(errmsg, errmsgl, "End of file on pipe magic during open");
1890         else
1891             g_snprintf(errmsg, errmsgl, "Error on pipe magic during open: %s",
1892                        g_strerror(errno));
1893         goto error;
1894     }
1895
1896 #endif
1897
1898     switch (magic) {
1899     case PCAP_MAGIC:
1900         /* Host that wrote it has our byte order, and was running
1901            a program using either standard or ss990417 libpcap. */
1902         pcap_opts->cap_pipe_byte_swapped = FALSE;
1903         pcap_opts->cap_pipe_modified = FALSE;
1904         break;
1905     case PCAP_MODIFIED_MAGIC:
1906         /* Host that wrote it has our byte order, but was running
1907            a program using either ss990915 or ss991029 libpcap. */
1908         pcap_opts->cap_pipe_byte_swapped = FALSE;
1909         pcap_opts->cap_pipe_modified = TRUE;
1910         break;
1911     case PCAP_SWAPPED_MAGIC:
1912         /* Host that wrote it has a byte order opposite to ours,
1913            and was running a program using either standard or
1914            ss990417 libpcap. */
1915         pcap_opts->cap_pipe_byte_swapped = TRUE;
1916         pcap_opts->cap_pipe_modified = FALSE;
1917         break;
1918     case PCAP_SWAPPED_MODIFIED_MAGIC:
1919         /* Host that wrote it out has a byte order opposite to
1920            ours, and was running a program using either ss990915
1921            or ss991029 libpcap. */
1922         pcap_opts->cap_pipe_byte_swapped = TRUE;
1923         pcap_opts->cap_pipe_modified = TRUE;
1924         break;
1925     default:
1926         /* Not a "libpcap" type we know about. */
1927         g_snprintf(errmsg, errmsgl, "Unrecognized libpcap format");
1928         goto error;
1929     }
1930
1931 #if !(defined(USE_THREADS) && defined(_WIN32))
1932     /* Read the rest of the header */
1933     bytes_read = 0;
1934     while (bytes_read < sizeof(struct pcap_hdr)) {
1935         sel_ret = cap_pipe_select(fd);
1936         if (sel_ret < 0) {
1937             g_snprintf(errmsg, errmsgl,
1938                        "Unexpected error from select: %s", g_strerror(errno));
1939             goto error;
1940         } else if (sel_ret > 0) {
1941             b = read(fd, ((char *)hdr)+bytes_read,
1942                      sizeof(struct pcap_hdr) - bytes_read);
1943             if (b <= 0) {
1944                 if (b == 0)
1945                     g_snprintf(errmsg, errmsgl, "End of file on pipe header during open");
1946                 else
1947                     g_snprintf(errmsg, errmsgl, "Error on pipe header during open: %s",
1948                                g_strerror(errno));
1949                 goto error;
1950             }
1951             bytes_read += b;
1952         }
1953     }
1954 #else
1955     pcap_opts->cap_pipe_buf = (char *) hdr;
1956     pcap_opts->cap_pipe_bytes_read = 0;
1957     pcap_opts->cap_pipe_bytes_to_read = sizeof(struct pcap_hdr);
1958     g_async_queue_push(pcap_opts->cap_pipe_pending_q, pcap_opts->cap_pipe_buf);
1959     g_async_queue_pop(pcap_opts->cap_pipe_done_q);
1960     if (pcap_opts->cap_pipe_bytes_read <= 0) {
1961         if (pcap_opts->cap_pipe_bytes_read == 0)
1962             g_snprintf(errmsg, errmsgl, "End of file on pipe header during open");
1963         else
1964             g_snprintf(errmsg, errmsgl, "Error on pipe header header during open: %s",
1965                        g_strerror(errno));
1966         goto error;
1967     }
1968 #endif
1969
1970     if (pcap_opts->cap_pipe_byte_swapped) {
1971         /* Byte-swap the header fields about which we care. */
1972         hdr->version_major = BSWAP16(hdr->version_major);
1973         hdr->version_minor = BSWAP16(hdr->version_minor);
1974         hdr->snaplen = BSWAP32(hdr->snaplen);
1975         hdr->network = BSWAP32(hdr->network);
1976     }
1977     pcap_opts->linktype = hdr->network;
1978
1979     if (hdr->version_major < 2) {
1980         g_snprintf(errmsg, errmsgl, "Unable to read old libpcap format");
1981         goto error;
1982     }
1983
1984     pcap_opts->cap_pipe_state = STATE_EXPECT_REC_HDR;
1985     pcap_opts->cap_pipe_err = PIPOK;
1986 #ifndef _WIN32
1987     pcap_opts->cap_pipe_fd = fd;
1988 #endif
1989     return;
1990
1991 error:
1992     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "cap_pipe_open_live: error %s", errmsg);
1993     pcap_opts->cap_pipe_err = PIPERR;
1994 #ifndef _WIN32
1995     ws_close(fd);
1996     pcap_opts->cap_pipe_fd = -1;
1997 #endif
1998     return;
1999
2000 }
2001
2002
2003 /* We read one record from the pipe, take care of byte order in the record
2004  * header, write the record to the capture file, and update capture statistics. */
2005 static int
2006 cap_pipe_dispatch(loop_data *ld, pcap_options *pcap_opts, guchar *data, char *errmsg, int errmsgl)
2007 {
2008     struct pcap_pkthdr phdr;
2009     enum { PD_REC_HDR_READ, PD_DATA_READ, PD_PIPE_EOF, PD_PIPE_ERR,
2010            PD_ERR } result;
2011 #if defined(USE_THREADS) && defined(_WIN32)
2012     GTimeVal wait_time;
2013     gpointer q_status;
2014 #else
2015     int b;
2016 #endif
2017 #ifdef _WIN32
2018     wchar_t *err_str;
2019 #endif
2020
2021 #ifdef LOG_CAPTURE_VERBOSE
2022     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "cap_pipe_dispatch");
2023 #endif
2024
2025     switch (pcap_opts->cap_pipe_state) {
2026
2027     case STATE_EXPECT_REC_HDR:
2028 #if defined(USE_THREADS) && defined(_WIN32)
2029         if (g_mutex_trylock(pcap_opts->cap_pipe_read_mtx)) {
2030 #endif
2031
2032             pcap_opts->cap_pipe_state = STATE_READ_REC_HDR;
2033             pcap_opts->cap_pipe_bytes_to_read = pcap_opts->cap_pipe_modified ?
2034                 sizeof(struct pcaprec_modified_hdr) : sizeof(struct pcaprec_hdr);
2035             pcap_opts->cap_pipe_bytes_read = 0;
2036
2037 #if defined(USE_THREADS) && defined(_WIN32)
2038             pcap_opts->cap_pipe_buf = (char *) &pcap_opts->cap_pipe_rechdr;
2039             g_async_queue_push(pcap_opts->cap_pipe_pending_q, pcap_opts->cap_pipe_buf);
2040             g_mutex_unlock(pcap_opts->cap_pipe_read_mtx);
2041         }
2042 #endif
2043         /* Fall through */
2044
2045     case STATE_READ_REC_HDR:
2046 #if !(defined(USE_THREADS) && defined(_WIN32))
2047         b = read(pcap_opts->cap_pipe_fd, ((char *)&pcap_opts->cap_pipe_rechdr)+pcap_opts->cap_pipe_bytes_read,
2048                  pcap_opts->cap_pipe_bytes_to_read - pcap_opts->cap_pipe_bytes_read);
2049         if (b <= 0) {
2050             if (b == 0)
2051                 result = PD_PIPE_EOF;
2052             else
2053                 result = PD_PIPE_ERR;
2054             break;
2055         }
2056         pcap_opts->cap_pipe_bytes_read += b;
2057 #else
2058         g_get_current_time(&wait_time);
2059         g_time_val_add(&wait_time, PIPE_READ_TIMEOUT);
2060         q_status = g_async_queue_timed_pop(pcap_opts->cap_pipe_done_q, &wait_time);
2061         if (pcap_opts->cap_pipe_err == PIPEOF) {
2062             result = PD_PIPE_EOF;
2063             break;
2064         } else if (pcap_opts->cap_pipe_err == PIPERR) {
2065             result = PD_PIPE_ERR;
2066             break;
2067         }
2068         if (!q_status) {
2069             return 0;
2070         }
2071 #endif
2072         if ((pcap_opts->cap_pipe_bytes_read) < pcap_opts->cap_pipe_bytes_to_read)
2073             return 0;
2074         result = PD_REC_HDR_READ;
2075         break;
2076
2077     case STATE_EXPECT_DATA:
2078 #if defined(USE_THREADS) && defined(_WIN32)
2079         if (g_mutex_trylock(pcap_opts->cap_pipe_read_mtx)) {
2080 #endif
2081
2082             pcap_opts->cap_pipe_state = STATE_READ_DATA;
2083             pcap_opts->cap_pipe_bytes_to_read = pcap_opts->cap_pipe_rechdr.hdr.incl_len;
2084             pcap_opts->cap_pipe_bytes_read = 0;
2085
2086 #if defined(USE_THREADS) && defined(_WIN32)
2087             pcap_opts->cap_pipe_buf = (char *) data;
2088             g_async_queue_push(pcap_opts->cap_pipe_pending_q, pcap_opts->cap_pipe_buf);
2089             g_mutex_unlock(pcap_opts->cap_pipe_read_mtx);
2090         }
2091 #endif
2092         /* Fall through */
2093
2094     case STATE_READ_DATA:
2095 #if !(defined(USE_THREADS) && defined(_WIN32))
2096         b = read(pcap_opts->cap_pipe_fd, data+pcap_opts->cap_pipe_bytes_read,
2097                  pcap_opts->cap_pipe_bytes_to_read - pcap_opts->cap_pipe_bytes_read);
2098         if (b <= 0) {
2099             if (b == 0)
2100                 result = PD_PIPE_EOF;
2101             else
2102                 result = PD_PIPE_ERR;
2103             break;
2104         }
2105         pcap_opts->cap_pipe_bytes_read += b;
2106 #else
2107         g_get_current_time(&wait_time);
2108         g_time_val_add(&wait_time, PIPE_READ_TIMEOUT);
2109         q_status = g_async_queue_timed_pop(pcap_opts->cap_pipe_done_q, &wait_time);
2110         if (pcap_opts->cap_pipe_err == PIPEOF) {
2111             result = PD_PIPE_EOF;
2112             break;
2113         } else if (pcap_opts->cap_pipe_err == PIPERR) {
2114             result = PD_PIPE_ERR;
2115             break;
2116         }
2117         if (!q_status) {
2118             return 0;
2119         }
2120 #endif
2121         if ((pcap_opts->cap_pipe_bytes_read) < pcap_opts->cap_pipe_bytes_to_read)
2122             return 0;
2123         result = PD_DATA_READ;
2124         break;
2125
2126     default:
2127         g_snprintf(errmsg, errmsgl, "cap_pipe_dispatch: invalid state");
2128         result = PD_ERR;
2129
2130     } /* switch (ld->cap_pipe_state) */
2131
2132     /*
2133      * We've now read as much data as we were expecting, so process it.
2134      */
2135     switch (result) {
2136
2137     case PD_REC_HDR_READ:
2138         /* We've read the header. Take care of byte order. */
2139         cap_pipe_adjust_header(pcap_opts->cap_pipe_byte_swapped, &pcap_opts->cap_pipe_hdr,
2140                                &pcap_opts->cap_pipe_rechdr.hdr);
2141         if (pcap_opts->cap_pipe_rechdr.hdr.incl_len > WTAP_MAX_PACKET_SIZE) {
2142             g_snprintf(errmsg, errmsgl, "Frame %u too long (%d bytes)",
2143                        ld->packet_count+1, pcap_opts->cap_pipe_rechdr.hdr.incl_len);
2144             break;
2145         }
2146
2147         if (pcap_opts->cap_pipe_rechdr.hdr.incl_len) {
2148             pcap_opts->cap_pipe_state = STATE_EXPECT_DATA;
2149             return 0;
2150         }
2151         /* no data to read? fall through */
2152
2153     case PD_DATA_READ:
2154         /* Fill in a "struct pcap_pkthdr", and process the packet. */
2155         phdr.ts.tv_sec = pcap_opts->cap_pipe_rechdr.hdr.ts_sec;
2156         phdr.ts.tv_usec = pcap_opts->cap_pipe_rechdr.hdr.ts_usec;
2157         phdr.caplen = pcap_opts->cap_pipe_rechdr.hdr.incl_len;
2158         phdr.len = pcap_opts->cap_pipe_rechdr.hdr.orig_len;
2159
2160         if (use_threads) {
2161             capture_loop_queue_packet_cb((u_char *)pcap_opts, &phdr, data);
2162         } else {
2163             capture_loop_write_packet_cb((u_char *)pcap_opts, &phdr, data);
2164         }
2165         pcap_opts->cap_pipe_state = STATE_EXPECT_REC_HDR;
2166         return 1;
2167
2168     case PD_PIPE_EOF:
2169         pcap_opts->cap_pipe_err = PIPEOF;
2170         return -1;
2171
2172     case PD_PIPE_ERR:
2173 #ifdef _WIN32
2174         FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_ALLOCATE_BUFFER,
2175                       NULL, GetLastError(), 0, (LPTSTR) &err_str, 0, NULL);
2176         g_snprintf(errmsg, errmsgl,
2177                    "Error reading from pipe: %s (error %d)",
2178                    utf_16to8(err_str), GetLastError());
2179         LocalFree(err_str);
2180 #else
2181         g_snprintf(errmsg, errmsgl, "Error reading from pipe: %s",
2182                    g_strerror(errno));
2183 #endif
2184         /* Fall through */
2185     case PD_ERR:
2186         break;
2187     }
2188
2189     pcap_opts->cap_pipe_err = PIPERR;
2190     /* Return here rather than inside the switch to prevent GCC warning */
2191     return -1;
2192 }
2193
2194
2195 /** Open the capture input file (pcap or capture pipe).
2196  *  Returns TRUE if it succeeds, FALSE otherwise. */
2197 static gboolean
2198 capture_loop_open_input(capture_options *capture_opts, loop_data *ld,
2199                         char *errmsg, size_t errmsg_len,
2200                         char *secondary_errmsg, size_t secondary_errmsg_len)
2201 {
2202     gchar             open_err_str[PCAP_ERRBUF_SIZE];
2203     gchar             *sync_msg_str;
2204     interface_options interface_opts;
2205     pcap_options      *pcap_opts;
2206     guint             i;
2207 #ifdef _WIN32
2208     int         err;
2209     gchar      *sync_secondary_msg_str;
2210     WORD        wVersionRequested;
2211     WSADATA     wsaData;
2212 #endif
2213
2214 /* XXX - opening Winsock on tshark? */
2215
2216     /* Initialize Windows Socket if we are in a WIN32 OS
2217        This needs to be done before querying the interface for network/netmask */
2218 #ifdef _WIN32
2219     /* XXX - do we really require 1.1 or earlier?
2220        Are there any versions that support only 2.0 or higher? */
2221     wVersionRequested = MAKEWORD(1, 1);
2222     err = WSAStartup(wVersionRequested, &wsaData);
2223     if (err != 0) {
2224         switch (err) {
2225
2226         case WSASYSNOTREADY:
2227             g_snprintf(errmsg, (gulong) errmsg_len,
2228                        "Couldn't initialize Windows Sockets: Network system not ready for network communication");
2229             break;
2230
2231         case WSAVERNOTSUPPORTED:
2232             g_snprintf(errmsg, (gulong) errmsg_len,
2233                        "Couldn't initialize Windows Sockets: Windows Sockets version %u.%u not supported",
2234                        LOBYTE(wVersionRequested), HIBYTE(wVersionRequested));
2235             break;
2236
2237         case WSAEINPROGRESS:
2238             g_snprintf(errmsg, (gulong) errmsg_len,
2239                        "Couldn't initialize Windows Sockets: Blocking operation is in progress");
2240             break;
2241
2242         case WSAEPROCLIM:
2243             g_snprintf(errmsg, (gulong) errmsg_len,
2244                        "Couldn't initialize Windows Sockets: Limit on the number of tasks supported by this WinSock implementation has been reached");
2245             break;
2246
2247         case WSAEFAULT:
2248             g_snprintf(errmsg, (gulong) errmsg_len,
2249                        "Couldn't initialize Windows Sockets: Bad pointer passed to WSAStartup");
2250             break;
2251
2252         default:
2253             g_snprintf(errmsg, (gulong) errmsg_len,
2254                        "Couldn't initialize Windows Sockets: error %d", err);
2255             break;
2256         }
2257         g_snprintf(secondary_errmsg, (gulong) secondary_errmsg_len, please_report);
2258         return FALSE;
2259     }
2260 #endif
2261     if ((use_threads == FALSE) &&
2262         (capture_opts->ifaces->len > 1)) {
2263         g_snprintf(errmsg, (gulong) errmsg_len,
2264                    "Using threads is required for capturing on mulitple interfaces!");
2265         return FALSE;
2266     }
2267
2268     for (i = 0; i < capture_opts->ifaces->len; i++) {
2269         interface_opts = g_array_index(capture_opts->ifaces, interface_options, i);
2270         pcap_opts = (pcap_options *)g_malloc(sizeof (pcap_options));
2271         if (pcap_opts == NULL) {
2272             g_snprintf(errmsg, (gulong) errmsg_len,
2273                    "Could not allocate memory.");
2274             return FALSE;
2275         }
2276         pcap_opts->received = 0;
2277         pcap_opts->dropped = 0;
2278         pcap_opts->pcap_h = NULL;
2279 #ifdef MUST_DO_SELECT
2280         pcap_opts->pcap_fd = -1;
2281 #endif
2282         pcap_opts->pcap_err = FALSE;
2283         pcap_opts->interface_id = i;
2284         pcap_opts->tid = NULL;
2285         pcap_opts->snaplen = 0;
2286         pcap_opts->linktype = -1;
2287         pcap_opts->from_cap_pipe = FALSE;
2288         memset(&pcap_opts->cap_pipe_hdr, 0, sizeof(struct pcap_hdr));
2289         memset(&pcap_opts->cap_pipe_rechdr, 0, sizeof(struct pcaprec_modified_hdr));
2290 #ifdef _WIN32
2291         pcap_opts->cap_pipe_h = INVALID_HANDLE_VALUE;
2292 #else
2293         pcap_opts->cap_pipe_fd = -1;
2294 #endif
2295         pcap_opts->cap_pipe_modified = FALSE;
2296         pcap_opts->cap_pipe_byte_swapped = FALSE;
2297 #if defined(USE_THREADS) && defined(_WIN32)
2298         pcap_opts->cap_pipe_buf = NULL;
2299 #endif
2300         pcap_opts->cap_pipe_bytes_to_read = 0;
2301         pcap_opts->cap_pipe_bytes_read = 0;
2302         pcap_opts->cap_pipe_state = 0;
2303         pcap_opts->cap_pipe_err = PIPOK;
2304 #if defined(USE_THREADS) && defined(_WIN32)
2305         pcap_opts->cap_pipe_read_mtx = g_mutex_new();
2306         pcap_opts->cap_pipe_pending_q = g_async_queue_new();
2307         pcap_opts->cap_pipe_done_q = g_async_queue_new();
2308 #endif
2309         g_array_append_val(ld->pcaps, pcap_opts);
2310
2311         g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_open_input : %s", interface_opts.name);
2312         pcap_opts->pcap_h = open_capture_device(&interface_opts, &open_err_str);
2313
2314         if (pcap_opts->pcap_h != NULL) {
2315             /* we've opened "iface" as a network device */
2316 #ifdef _WIN32
2317             /* try to set the capture buffer size */
2318             if (interface_opts.buffer_size > 1 &&
2319                 pcap_setbuff(pcap_opts->pcap_h, interface_opts.buffer_size * 1024 * 1024) != 0) {
2320                 sync_secondary_msg_str = g_strdup_printf(
2321                     "The capture buffer size of %dMB seems to be too high for your machine,\n"
2322                     "the default of 1MB will be used.\n"
2323                     "\n"
2324                     "Nonetheless, the capture is started.\n",
2325                     interface_opts.buffer_size);
2326                 report_capture_error("Couldn't set the capture buffer size!",
2327                                      sync_secondary_msg_str);
2328                 g_free(sync_secondary_msg_str);
2329             }
2330 #endif
2331
2332 #if defined(HAVE_PCAP_SETSAMPLING)
2333             if (interface_opts.sampling_method != CAPTURE_SAMP_NONE) {
2334                 struct pcap_samp *samp;
2335
2336                 if ((samp = pcap_setsampling(pcap_opts->pcap_h)) != NULL) {
2337                     switch (interface_opts.sampling_method) {
2338                     case CAPTURE_SAMP_BY_COUNT:
2339                         samp->method = PCAP_SAMP_1_EVERY_N;
2340                         break;
2341
2342                     case CAPTURE_SAMP_BY_TIMER:
2343                         samp->method = PCAP_SAMP_FIRST_AFTER_N_MS;
2344                         break;
2345
2346                     default:
2347                         sync_msg_str = g_strdup_printf(
2348                             "Unknown sampling method %d specified,\n"
2349                             "continue without packet sampling",
2350                             interface_opts.sampling_method);
2351                         report_capture_error("Couldn't set the capture "
2352                                              "sampling", sync_msg_str);
2353                         g_free(sync_msg_str);
2354                     }
2355                     samp->value = interface_opts.sampling_param;
2356                 } else {
2357                     report_capture_error("Couldn't set the capture sampling",
2358                                          "Cannot get packet sampling data structure");
2359                 }
2360             }
2361 #endif
2362
2363             /* setting the data link type only works on real interfaces */
2364             if (!set_pcap_linktype(pcap_opts->pcap_h, interface_opts.linktype, interface_opts.name,
2365                                    errmsg, errmsg_len,
2366                                    secondary_errmsg, secondary_errmsg_len)) {
2367                 return FALSE;
2368             }
2369             pcap_opts->linktype = get_pcap_linktype(pcap_opts->pcap_h, interface_opts.name);
2370         } else {
2371             /* We couldn't open "iface" as a network device. */
2372             /* Try to open it as a pipe */
2373             cap_pipe_open_live(interface_opts.name, pcap_opts, &pcap_opts->cap_pipe_hdr, errmsg, (int) errmsg_len);
2374
2375 #ifndef _WIN32
2376             if (pcap_opts->cap_pipe_fd == -1) {
2377 #else
2378             if (pcap_opts->cap_pipe_h == INVALID_HANDLE_VALUE) {
2379 #endif
2380                 if (pcap_opts->cap_pipe_err == PIPNEXIST) {
2381                     /* Pipe doesn't exist, so output message for interface */
2382                     get_capture_device_open_failure_messages(open_err_str,
2383                                                              interface_opts.name,
2384                                                              errmsg,
2385                                                              errmsg_len,
2386                                                              secondary_errmsg,
2387                                                              secondary_errmsg_len);
2388                 }
2389                 /*
2390                  * Else pipe (or file) does exist and cap_pipe_open_live() has
2391                  * filled in errmsg
2392                  */
2393                 return FALSE;
2394             } else {
2395                 /* cap_pipe_open_live() succeeded; don't want
2396                    error message from pcap_open_live() */
2397                 open_err_str[0] = '\0';
2398             }
2399         }
2400
2401 /* XXX - will this work for tshark? */
2402 #ifdef MUST_DO_SELECT
2403         if (!pcap_opts->from_cap_pipe) {
2404 #ifdef HAVE_PCAP_GET_SELECTABLE_FD
2405             pcap_opts->pcap_fd = pcap_get_selectable_fd(pcap_opts->pcap_h);
2406 #else
2407             pcap_opts->pcap_fd = pcap_fileno(pcap_opts->pcap_h);
2408 #endif
2409         }
2410 #endif
2411
2412         /* Does "open_err_str" contain a non-empty string?  If so, "pcap_open_live()"
2413            returned a warning; print it, but keep capturing. */
2414         if (open_err_str[0] != '\0') {
2415             sync_msg_str = g_strdup_printf("%s.", open_err_str);
2416             report_capture_error(sync_msg_str, "");
2417             g_free(sync_msg_str);
2418         }
2419         capture_opts->ifaces = g_array_remove_index(capture_opts->ifaces, i);
2420         g_array_insert_val(capture_opts->ifaces, i, interface_opts);
2421     }
2422
2423     /* If not using libcap: we now can now set euid/egid to ruid/rgid         */
2424     /*  to remove any suid privileges.                                        */
2425     /* If using libcap: we can now remove NET_RAW and NET_ADMIN capabilities  */
2426     /*  (euid/egid have already previously been set to ruid/rgid.             */
2427     /* (See comment in main() for details)                                    */
2428 #ifndef HAVE_LIBCAP
2429     relinquish_special_privs_perm();
2430 #else
2431     relinquish_all_capabilities();
2432 #endif
2433     return TRUE;
2434 }
2435
2436 /* close the capture input file (pcap or capture pipe) */
2437 static void capture_loop_close_input(loop_data *ld)
2438 {
2439     guint i;
2440     pcap_options *pcap_opts;
2441
2442     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_close_input");
2443
2444     for (i = 0; i < ld->pcaps->len; i++) {
2445         pcap_opts = g_array_index(ld->pcaps, pcap_options *, i);
2446         /* if open, close the capture pipe "input file" */
2447 #ifndef _WIN32
2448         if (pcap_opts->cap_pipe_fd >= 0) {
2449             g_assert(pcap_opts->from_cap_pipe);
2450             ws_close(pcap_opts->cap_pipe_fd);
2451             pcap_opts->cap_pipe_fd = -1;
2452         }
2453 #else
2454         if (pcap_opts->cap_pipe_h != INVALID_HANDLE_VALUE) {
2455             CloseHandle(pcap_opts->cap_pipe_h);
2456             pcap_opts->cap_pipe_h = INVALID_HANDLE_VALUE;
2457         }
2458 #endif
2459         /* if open, close the pcap "input file" */
2460         if (pcap_opts->pcap_h != NULL) {
2461             g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_close_input: closing %p", (void *)pcap_opts->pcap_h);
2462             pcap_close(pcap_opts->pcap_h);
2463             pcap_opts->pcap_h = NULL;
2464         }
2465     }
2466
2467     ld->go = FALSE;
2468
2469 #ifdef _WIN32
2470     /* Shut down windows sockets */
2471     WSACleanup();
2472 #endif
2473 }
2474
2475
2476 /* init the capture filter */
2477 static initfilter_status_t
2478 capture_loop_init_filter(pcap_t *pcap_h, gboolean from_cap_pipe,
2479                          const gchar * name, const gchar * cfilter)
2480 {
2481     struct bpf_program fcode;
2482
2483     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_init_filter: %s", cfilter);
2484
2485     /* capture filters only work on real interfaces */
2486     if (cfilter && !from_cap_pipe) {
2487         /* A capture filter was specified; set it up. */
2488         if (!compile_capture_filter(name, pcap_h, &fcode, cfilter)) {
2489             /* Treat this specially - our caller might try to compile this
2490                as a display filter and, if that succeeds, warn the user that
2491                the display and capture filter syntaxes are different. */
2492             return INITFILTER_BAD_FILTER;
2493         }
2494         if (pcap_setfilter(pcap_h, &fcode) < 0) {
2495 #ifdef HAVE_PCAP_FREECODE
2496             pcap_freecode(&fcode);
2497 #endif
2498             return INITFILTER_OTHER_ERROR;
2499         }
2500 #ifdef HAVE_PCAP_FREECODE
2501         pcap_freecode(&fcode);
2502 #endif
2503     }
2504
2505     return INITFILTER_NO_ERROR;
2506 }
2507
2508
2509 /* set up to write to the already-opened capture output file/files */
2510 static gboolean
2511 capture_loop_init_output(capture_options *capture_opts, loop_data *ld, char *errmsg, int errmsg_len)
2512 {
2513     int err;
2514     guint i;
2515     pcap_options *pcap_opts;
2516     interface_options interface_opts;
2517     gboolean successful;
2518
2519     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_init_output");
2520
2521     if ((capture_opts->use_pcapng == FALSE) &&
2522         (capture_opts->ifaces->len > 1)) {
2523         g_snprintf(errmsg, errmsg_len,
2524                    "Using PCAPNG is required for capturing on mulitple interfaces! Use the -n option.");
2525         return FALSE;
2526     }
2527
2528     /* Set up to write to the capture file. */
2529     if (capture_opts->multi_files_on) {
2530         ld->pdh = ringbuf_init_libpcap_fdopen(&err);
2531     } else {
2532         ld->pdh = libpcap_fdopen(ld->save_file_fd, &err);
2533     }
2534     if (ld->pdh) {
2535         if (capture_opts->use_pcapng) {
2536             char appname[100];
2537
2538             g_snprintf(appname, sizeof(appname), "Dumpcap " VERSION "%s", wireshark_svnversion);
2539             successful = libpcap_write_session_header_block(ld->pdh, appname, &ld->bytes_written, &err);
2540             for (i = 0; successful && (i < capture_opts->ifaces->len); i++) {
2541                 interface_opts = g_array_index(capture_opts->ifaces, interface_options, i);
2542                 pcap_opts = g_array_index(ld->pcaps, pcap_options *, i);
2543                 if (pcap_opts->from_cap_pipe) {
2544                     pcap_opts->snaplen = pcap_opts->cap_pipe_hdr.snaplen;
2545                 } else {
2546                     pcap_opts->snaplen = pcap_snapshot(pcap_opts->pcap_h);
2547                 }
2548                 successful = libpcap_write_interface_description_block(ld->pdh,
2549                                                                        interface_opts.name,
2550                                                                        interface_opts.cfilter?interface_opts.cfilter:"",
2551                                                                        pcap_opts->linktype,
2552                                                                        pcap_opts->snaplen,
2553                                                                        &ld->bytes_written,
2554                                                                        &err);
2555             }
2556         } else {
2557             interface_opts = g_array_index(capture_opts->ifaces, interface_options, 0);
2558             pcap_opts = g_array_index(ld->pcaps, pcap_options *, 0);
2559             if (pcap_opts->from_cap_pipe) {
2560                 pcap_opts->snaplen = pcap_opts->cap_pipe_hdr.snaplen;
2561             } else {
2562                 pcap_opts->snaplen = pcap_snapshot(pcap_opts->pcap_h);
2563             }
2564             successful = libpcap_write_file_header(ld->pdh, pcap_opts->linktype, pcap_opts->snaplen,
2565                                                    &ld->bytes_written, &err);
2566         }
2567         if (!successful) {
2568             fclose(ld->pdh);
2569             ld->pdh = NULL;
2570         }
2571     }
2572
2573     if (ld->pdh == NULL) {
2574         /* We couldn't set up to write to the capture file. */
2575         /* XXX - use cf_open_error_message from tshark instead? */
2576         switch (err) {
2577
2578         default:
2579             if (err < 0) {
2580                 g_snprintf(errmsg, errmsg_len,
2581                            "The file to which the capture would be"
2582                            " saved (\"%s\") could not be opened: Error %d.",
2583                            capture_opts->save_file, err);
2584             } else {
2585                 g_snprintf(errmsg, errmsg_len,
2586                            "The file to which the capture would be"
2587                            " saved (\"%s\") could not be opened: %s.",
2588                            capture_opts->save_file, g_strerror(err));
2589             }
2590             break;
2591         }
2592
2593         return FALSE;
2594     }
2595
2596     return TRUE;
2597 }
2598
2599 static gboolean
2600 capture_loop_close_output(capture_options *capture_opts, loop_data *ld, int *err_close)
2601 {
2602
2603     unsigned int i;
2604     pcap_options *pcap_opts;
2605
2606     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_close_output");
2607
2608     if (capture_opts->multi_files_on) {
2609         return ringbuf_libpcap_dump_close(&capture_opts->save_file, err_close);
2610     } else {
2611         if (capture_opts->use_pcapng) {
2612             for (i = 0; i < global_ld.pcaps->len; i++) {
2613                 pcap_opts = g_array_index(global_ld.pcaps, pcap_options *, i);
2614                 if (!pcap_opts->from_cap_pipe) {
2615                     libpcap_write_interface_statistics_block(ld->pdh, i, pcap_opts->pcap_h, &ld->bytes_written, err_close);
2616                 }
2617             }
2618         }
2619         return libpcap_dump_close(ld->pdh, err_close);
2620     }
2621 }
2622
2623 /* dispatch incoming packets (pcap or capture pipe)
2624  *
2625  * Waits for incoming packets to be available, and calls pcap_dispatch()
2626  * to cause them to be processed.
2627  *
2628  * Returns the number of packets which were processed.
2629  *
2630  * Times out (returning zero) after CAP_READ_TIMEOUT ms; this ensures that the
2631  * packet-batching behaviour does not cause packets to get held back
2632  * indefinitely.
2633  */
2634 static int
2635 capture_loop_dispatch(loop_data *ld,
2636                       char *errmsg, int errmsg_len, pcap_options *pcap_opts)
2637 {
2638     int       inpkts;
2639     gint      packet_count_before;
2640     guchar    pcap_data[WTAP_MAX_PACKET_SIZE];
2641 #if !(defined(USE_THREADS) && defined(_WIN32))
2642     int       sel_ret;
2643 #endif
2644
2645     packet_count_before = ld->packet_count;
2646     if (pcap_opts->from_cap_pipe) {
2647         /* dispatch from capture pipe */
2648 #ifdef LOG_CAPTURE_VERBOSE
2649         g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_dispatch: from capture pipe");
2650 #endif
2651 #if !(defined(USE_THREADS) && defined(_WIN32))
2652         sel_ret = cap_pipe_select(pcap_opts->cap_pipe_fd);
2653         if (sel_ret <= 0) {
2654             if (sel_ret < 0 && errno != EINTR) {
2655                 g_snprintf(errmsg, errmsg_len,
2656                            "Unexpected error from select: %s", g_strerror(errno));
2657                 report_capture_error(errmsg, please_report);
2658                 ld->go = FALSE;
2659             }
2660         } else {
2661             /*
2662              * "select()" says we can read from the pipe without blocking
2663              */
2664 #endif
2665             inpkts = cap_pipe_dispatch(ld, pcap_opts, pcap_data, errmsg, errmsg_len);
2666             if (inpkts < 0) {
2667                 ld->go = FALSE;
2668             }
2669 #if !(defined(USE_THREADS) && defined(_WIN32))
2670         }
2671 #endif
2672     }
2673     else
2674     {
2675         /* dispatch from pcap */
2676 #ifdef MUST_DO_SELECT
2677         /*
2678          * If we have "pcap_get_selectable_fd()", we use it to get the
2679          * descriptor on which to select; if that's -1, it means there
2680          * is no descriptor on which you can do a "select()" (perhaps
2681          * because you're capturing on a special device, and that device's
2682          * driver unfortunately doesn't support "select()", in which case
2683          * we don't do the select - which means it might not be possible
2684          * to stop a capture until a packet arrives.  If that's unacceptable,
2685          * plead with whoever supplies the software for that device to add
2686          * "select()" support, or upgrade to libpcap 0.8.1 or later, and
2687          * rebuild Wireshark or get a version built with libpcap 0.8.1 or
2688          * later, so it can use pcap_breakloop().
2689          */
2690 #ifdef LOG_CAPTURE_VERBOSE
2691         g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_dispatch: from pcap_dispatch with select");
2692 #endif
2693         if (pcap_opts->pcap_fd != -1) {
2694             sel_ret = cap_pipe_select(pcap_opts->pcap_fd);
2695             if (sel_ret > 0) {
2696                 /*
2697                  * "select()" says we can read from it without blocking; go for
2698                  * it.
2699                  *
2700                  * We don't have pcap_breakloop(), so we only process one packet
2701                  * per pcap_dispatch() call, to allow a signal to stop the
2702                  * processing immediately, rather than processing all packets
2703                  * in a batch before quitting.
2704                  */
2705                 if (use_threads) {
2706                     inpkts = pcap_dispatch(pcap_opts->pcap_h, 1, capture_loop_queue_packet_cb, (u_char *)pcap_opts);
2707                 } else {
2708                     inpkts = pcap_dispatch(pcap_opts->pcap_h, 1, capture_loop_write_packet_cb, (u_char *)pcap_opts);
2709                 }
2710                 if (inpkts < 0) {
2711                     if (inpkts == -1) {
2712                         /* Error, rather than pcap_breakloop(). */
2713                         pcap_opts->pcap_err = TRUE;
2714                     }
2715                     ld->go = FALSE; /* error or pcap_breakloop() - stop capturing */
2716                 }
2717             } else {
2718                 if (sel_ret < 0 && errno != EINTR) {
2719                     g_snprintf(errmsg, errmsg_len,
2720                                "Unexpected error from select: %s", g_strerror(errno));
2721                     report_capture_error(errmsg, please_report);
2722                     ld->go = FALSE;
2723                 }
2724             }
2725         }
2726         else
2727 #endif /* MUST_DO_SELECT */
2728         {
2729             /* dispatch from pcap without select */
2730 #if 1
2731 #ifdef LOG_CAPTURE_VERBOSE
2732             g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_dispatch: from pcap_dispatch");
2733 #endif
2734 #ifdef _WIN32
2735             /*
2736              * On Windows, we don't support asynchronously telling a process to
2737              * stop capturing; instead, we check for an indication on a pipe
2738              * after processing packets.  We therefore process only one packet
2739              * at a time, so that we can check the pipe after every packet.
2740              */
2741             if (use_threads) {
2742                 inpkts = pcap_dispatch(pcap_opts->pcap_h, 1, capture_loop_queue_packet_cb, (u_char *)pcap_opts);
2743             } else {
2744                 inpkts = pcap_dispatch(pcap_opts->pcap_h, 1, capture_loop_write_packet_cb, (u_char *)pcap_opts);
2745             }
2746 #else
2747             if (use_threads) {
2748                 inpkts = pcap_dispatch(pcap_opts->pcap_h, -1, capture_loop_queue_packet_cb, (u_char *)pcap_opts);
2749             } else {
2750                 inpkts = pcap_dispatch(pcap_opts->pcap_h, -1, capture_loop_write_packet_cb, (u_char *)pcap_opts);
2751             }
2752 #endif
2753             if (inpkts < 0) {
2754                 if (inpkts == -1) {
2755                     /* Error, rather than pcap_breakloop(). */
2756                     pcap_opts->pcap_err = TRUE;
2757                 }
2758                 ld->go = FALSE; /* error or pcap_breakloop() - stop capturing */
2759             }
2760 #else /* pcap_next_ex */
2761 #ifdef LOG_CAPTURE_VERBOSE
2762             g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_dispatch: from pcap_next_ex");
2763 #endif
2764             /* XXX - this is currently unused, as there is some confusion with pcap_next_ex() vs. pcap_dispatch() */
2765
2766             /*
2767              * WinPcap's remote capturing feature doesn't work with pcap_dispatch(),
2768              * see http://wiki.wireshark.org/CaptureSetup_2fWinPcapRemote
2769              * This should be fixed in the WinPcap 4.0 alpha release.
2770              *
2771              * For reference, an example remote interface:
2772              * rpcap://[1.2.3.4]/\Device\NPF_{39993D68-7C9B-4439-A329-F2D888DA7C5C}
2773              */
2774
2775             /* emulate dispatch from pcap */
2776             {
2777                 int in;
2778                 struct pcap_pkthdr *pkt_header;
2779                 u_char *pkt_data;
2780
2781                 in = 0;
2782                 while(ld->go &&
2783                       (in = pcap_next_ex(pcap_opts->pcap_h, &pkt_header, &pkt_data)) == 1) {
2784                     if (use_threads) {
2785                         capture_loop_queue_packet_cb((u_char *)pcap_opts, pkt_header, pkt_data);
2786                     } else {
2787                         capture_loop_write_packet_cb((u_char *)pcap_opts, pkt_header, pkt_data);
2788                     }
2789                 }
2790
2791                 if(in < 0) {
2792                     pcap_opts->pcap_err = TRUE;
2793                     ld->go = FALSE;
2794                 }
2795             }
2796 #endif /* pcap_next_ex */
2797         }
2798     }
2799
2800 #ifdef LOG_CAPTURE_VERBOSE
2801     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_dispatch: %d new packet%s", inpkts, plurality(inpkts, "", "s"));
2802 #endif
2803
2804     return ld->packet_count - packet_count_before;
2805 }
2806
2807 #ifdef _WIN32
2808 /* Isolate the Universally Unique Identifier from the interface.  Basically, we
2809  * want to grab only the characters between the '{' and '}' delimiters.
2810  *
2811  * Returns a GString that must be freed with g_string_free(). */
2812 static GString *
2813 isolate_uuid(const char *iface)
2814 {
2815     gchar *ptr;
2816     GString *gstr;
2817
2818     ptr = strchr(iface, '{');
2819     if (ptr == NULL)
2820         return g_string_new(iface);
2821     gstr = g_string_new(ptr + 1);
2822
2823     ptr = strchr(gstr->str, '}');
2824     if (ptr == NULL)
2825         return gstr;
2826
2827     gstr = g_string_truncate(gstr, ptr - gstr->str);
2828     return gstr;
2829 }
2830 #endif
2831
2832 /* open the output file (temporary/specified name/ringbuffer/named pipe/stdout) */
2833 /* Returns TRUE if the file opened successfully, FALSE otherwise. */
2834 static gboolean
2835 capture_loop_open_output(capture_options *capture_opts, int *save_file_fd,
2836                          char *errmsg, int errmsg_len)
2837 {
2838     char *tmpname;
2839     gchar *capfile_name;
2840     gchar *prefix;
2841     gboolean is_tempfile;
2842
2843     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_open_output: %s",
2844           (capture_opts->save_file) ? capture_opts->save_file : "(not specified)");
2845
2846     if (capture_opts->save_file != NULL) {
2847         /* We return to the caller while the capture is in progress.
2848          * Therefore we need to take a copy of save_file in
2849          * case the caller destroys it after we return.
2850          */
2851         capfile_name = g_strdup(capture_opts->save_file);
2852
2853         if (capture_opts->output_to_pipe == TRUE) { /* either "-" or named pipe */
2854             if (capture_opts->multi_files_on) {
2855                 /* ringbuffer is enabled; that doesn't work with standard output or a named pipe */
2856                 g_snprintf(errmsg, errmsg_len,
2857                            "Ring buffer requested, but capture is being written to standard output or to a named pipe.");
2858                 g_free(capfile_name);
2859                 return FALSE;
2860             }
2861             if (strcmp(capfile_name, "-") == 0) {
2862                 /* write to stdout */
2863                 *save_file_fd = 1;
2864 #ifdef _WIN32
2865                 /* set output pipe to binary mode to avoid Windows text-mode processing (eg: for CR/LF)  */
2866                 _setmode(1, O_BINARY);
2867 #endif
2868             }
2869         } /* if (...output_to_pipe ... */
2870
2871         else {
2872             if (capture_opts->multi_files_on) {
2873                 /* ringbuffer is enabled */
2874                 *save_file_fd = ringbuf_init(capfile_name,
2875                                              (capture_opts->has_ring_num_files) ? capture_opts->ring_num_files : 0,
2876                                              capture_opts->group_read_access);
2877
2878                 /* we need the ringbuf name */
2879                 if(*save_file_fd != -1) {
2880                     g_free(capfile_name);
2881                     capfile_name = g_strdup(ringbuf_current_filename());
2882                 }
2883             } else {
2884                 /* Try to open/create the specified file for use as a capture buffer. */
2885                 *save_file_fd = ws_open(capfile_name, O_RDWR|O_BINARY|O_TRUNC|O_CREAT,
2886                                         (capture_opts->group_read_access) ? 0640 : 0600);
2887             }
2888         }
2889         is_tempfile = FALSE;
2890     } else {
2891         /* Choose a random name for the temporary capture buffer */
2892         if (global_capture_opts.ifaces->len > 1) {
2893             prefix = g_strdup_printf("wireshark_%d_interfaces", global_capture_opts.ifaces->len);
2894         } else {
2895             gchar *basename;
2896 #ifdef _WIN32
2897             GString *iface;
2898             iface = isolate_uuid(g_array_index(global_capture_opts.ifaces, interface_options, 0).name);
2899             basename = g_path_get_basename(iface->str);
2900             g_string_free(iface, TRUE);
2901 #else
2902             basename = g_path_get_basename(g_array_index(global_capture_opts.ifaces, interface_options, 0).name);
2903 #endif
2904             prefix = g_strconcat("wireshark_", basename, NULL);
2905             g_free(basename);
2906         }
2907         *save_file_fd = create_tempfile(&tmpname, prefix);
2908         g_free(prefix);
2909         capfile_name = g_strdup(tmpname);
2910         is_tempfile = TRUE;
2911     }
2912
2913     /* did we fail to open the output file? */
2914     if (*save_file_fd == -1) {
2915         if (is_tempfile) {
2916             g_snprintf(errmsg, errmsg_len,
2917                        "The temporary file to which the capture would be saved (\"%s\") "
2918                        "could not be opened: %s.", capfile_name, g_strerror(errno));
2919         } else {
2920             if (capture_opts->multi_files_on) {
2921                 ringbuf_error_cleanup();
2922             }
2923
2924             g_snprintf(errmsg, errmsg_len,
2925                        "The file to which the capture would be saved (\"%s\") "
2926                        "could not be opened: %s.", capfile_name,
2927                        g_strerror(errno));
2928         }
2929         g_free(capfile_name);
2930         return FALSE;
2931     }
2932
2933     if(capture_opts->save_file != NULL) {
2934         g_free(capture_opts->save_file);
2935     }
2936     capture_opts->save_file = capfile_name;
2937     /* capture_opts.save_file is "g_free"ed later, which is equivalent to
2938        "g_free(capfile_name)". */
2939
2940     return TRUE;
2941 }
2942
2943
2944 /* Do the work of handling either the file size or file duration capture
2945    conditions being reached, and switching files or stopping. */
2946 static gboolean
2947 do_file_switch_or_stop(capture_options *capture_opts,
2948                        condition *cnd_autostop_files,
2949                        condition *cnd_autostop_size,
2950                        condition *cnd_file_duration)
2951 {
2952     guint i;
2953     pcap_options *pcap_opts;
2954     interface_options interface_opts;
2955     gboolean successful;
2956
2957     if (capture_opts->multi_files_on) {
2958         if (cnd_autostop_files != NULL &&
2959             cnd_eval(cnd_autostop_files, ++global_ld.autostop_files)) {
2960             /* no files left: stop here */
2961             global_ld.go = FALSE;
2962             return FALSE;
2963         }
2964
2965         /* Switch to the next ringbuffer file */
2966         if (ringbuf_switch_file(&global_ld.pdh, &capture_opts->save_file,
2967                                 &global_ld.save_file_fd, &global_ld.err)) {
2968
2969             /* File switch succeeded: reset the conditions */
2970             global_ld.bytes_written = 0;
2971             if (capture_opts->use_pcapng) {
2972                 char appname[100];
2973
2974                 g_snprintf(appname, sizeof(appname), "Dumpcap " VERSION "%s", wireshark_svnversion);
2975                 successful = libpcap_write_session_header_block(global_ld.pdh, appname, &(global_ld.bytes_written), &global_ld.err);
2976                 for (i = 0; successful && (i < capture_opts->ifaces->len); i++) {
2977                     interface_opts = g_array_index(capture_opts->ifaces, interface_options, i);
2978                     pcap_opts = g_array_index(global_ld.pcaps, pcap_options *, i);
2979                     successful = libpcap_write_interface_description_block(global_ld.pdh,
2980                                                                            interface_opts.name,
2981                                                                            interface_opts.cfilter?interface_opts.cfilter:"",
2982                                                                            pcap_opts->linktype,
2983                                                                            pcap_opts->snaplen,
2984                                                                            &(global_ld.bytes_written),
2985                                                                            &global_ld.err);
2986                 }
2987             } else {
2988                 interface_opts = g_array_index(capture_opts->ifaces, interface_options, 0);
2989                 pcap_opts = g_array_index(global_ld.pcaps, pcap_options *, 0);
2990                 successful = libpcap_write_file_header(global_ld.pdh, pcap_opts->linktype, pcap_opts->snaplen,
2991                                                        &global_ld.bytes_written, &global_ld.err);
2992             }
2993             if (!successful) {
2994                 fclose(global_ld.pdh);
2995                 global_ld.pdh = NULL;
2996                 global_ld.go = FALSE;
2997                 return FALSE;
2998             }
2999             if (cnd_autostop_size)
3000                 cnd_reset(cnd_autostop_size);
3001             if (cnd_file_duration)
3002                 cnd_reset(cnd_file_duration);
3003             libpcap_dump_flush(global_ld.pdh, NULL);
3004             if (!quiet)
3005                 report_packet_count(global_ld.inpkts_to_sync_pipe);
3006             global_ld.inpkts_to_sync_pipe = 0;
3007             report_new_capture_file(capture_opts->save_file);
3008         } else {
3009             /* File switch failed: stop here */
3010             global_ld.go = FALSE;
3011             return FALSE;
3012         }
3013     } else {
3014         /* single file, stop now */
3015         global_ld.go = FALSE;
3016         return FALSE;
3017     }
3018     return TRUE;
3019 }
3020
3021 static void *
3022 pcap_read_handler(void* arg)
3023 {
3024     pcap_options *pcap_opts;
3025     char errmsg[MSG_MAX_LENGTH+1];
3026
3027     pcap_opts = (pcap_options *)arg;
3028
3029     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO, "Started thread for interface %d.",
3030           pcap_opts->interface_id);
3031
3032     while (global_ld.go) {
3033         /* dispatch incoming packets */
3034         capture_loop_dispatch(&global_ld, errmsg, sizeof(errmsg), pcap_opts);
3035     }
3036     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO, "Stopped thread for interface %d.",
3037           pcap_opts->interface_id);
3038     g_thread_exit(NULL);
3039     return (NULL);
3040 }
3041
3042 /* Do the low-level work of a capture.
3043    Returns TRUE if it succeeds, FALSE otherwise. */
3044 static gboolean
3045 capture_loop_start(capture_options *capture_opts, gboolean *stats_known, struct pcap_stat *stats)
3046 {
3047 #ifdef WIN32
3048     time_t upd_time, cur_time;
3049 #else
3050     struct timeval upd_time, cur_time;
3051 #endif
3052     int         err_close;
3053     int         inpkts;
3054     condition  *cnd_file_duration = NULL;
3055     condition  *cnd_autostop_files = NULL;
3056     condition  *cnd_autostop_size = NULL;
3057     condition  *cnd_autostop_duration = NULL;
3058     gboolean    write_ok;
3059     gboolean    close_ok;
3060     gboolean    cfilter_error = FALSE;
3061     char        errmsg[MSG_MAX_LENGTH+1];
3062     char        secondary_errmsg[MSG_MAX_LENGTH+1];
3063     pcap_options *pcap_opts;
3064     interface_options interface_opts;
3065     guint i, error_index = 0;
3066
3067     interface_opts = capture_opts->default_options;
3068     *errmsg           = '\0';
3069     *secondary_errmsg = '\0';
3070
3071     /* init the loop data */
3072     global_ld.go                  = TRUE;
3073     global_ld.packet_count        = 0;
3074 #ifdef SIGINFO
3075     global_ld.report_packet_count = FALSE;
3076 #endif
3077     if (capture_opts->has_autostop_packets)
3078         global_ld.packet_max      = capture_opts->autostop_packets;
3079     else
3080         global_ld.packet_max      = 0;        /* no limit */
3081     global_ld.inpkts_to_sync_pipe = 0;
3082     global_ld.err                 = 0;  /* no error seen yet */
3083     global_ld.pdh                 = NULL;
3084     global_ld.autostop_files      = 0;
3085     global_ld.save_file_fd        = -1;
3086
3087     /* We haven't yet gotten the capture statistics. */
3088     *stats_known      = FALSE;
3089
3090     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO, "Capture loop starting ...");
3091     capture_opts_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, capture_opts);
3092
3093     /* open the "input file" from network interface or capture pipe */
3094     if (!capture_loop_open_input(capture_opts, &global_ld, errmsg, sizeof(errmsg),
3095                                  secondary_errmsg, sizeof(secondary_errmsg))) {
3096         goto error;
3097     }
3098     for (i = 0; i < capture_opts->ifaces->len; i++) {
3099         pcap_opts = g_array_index(global_ld.pcaps, pcap_options *, i);
3100         interface_opts = g_array_index(capture_opts->ifaces, interface_options, i);
3101         /* init the input filter from the network interface (capture pipe will do nothing) */
3102         /*
3103          * When remote capturing WinPCap crashes when the capture filter
3104          * is NULL. This might be a bug in WPCap. Therefore we provide an emtpy
3105          * string.
3106          */
3107         switch (capture_loop_init_filter(pcap_opts->pcap_h, pcap_opts->from_cap_pipe,
3108                                          interface_opts.name,
3109                                          interface_opts.cfilter?interface_opts.cfilter:"")) {
3110
3111         case INITFILTER_NO_ERROR:
3112             break;
3113
3114         case INITFILTER_BAD_FILTER:
3115             cfilter_error = TRUE;
3116             error_index = i;
3117             g_snprintf(errmsg, sizeof(errmsg), "%s", pcap_geterr(pcap_opts->pcap_h));
3118             goto error;
3119
3120         case INITFILTER_OTHER_ERROR:
3121             g_snprintf(errmsg, sizeof(errmsg), "Can't install filter (%s).",
3122                        pcap_geterr(pcap_opts->pcap_h));
3123             g_snprintf(secondary_errmsg, sizeof(secondary_errmsg), "%s", please_report);
3124             goto error;
3125         }
3126     }
3127
3128     /* If we're supposed to write to a capture file, open it for output
3129        (temporary/specified name/ringbuffer) */
3130     if (capture_opts->saving_to_file) {
3131         if (!capture_loop_open_output(capture_opts, &global_ld.save_file_fd,
3132                                       errmsg, sizeof(errmsg))) {
3133             goto error;
3134         }
3135
3136         /* set up to write to the already-opened capture output file/files */
3137         if (!capture_loop_init_output(capture_opts, &global_ld, errmsg,
3138                                       sizeof(errmsg))) {
3139             goto error;
3140         }
3141
3142         /* XXX - capture SIGTERM and close the capture, in case we're on a
3143            Linux 2.0[.x] system and you have to explicitly close the capture
3144            stream in order to turn promiscuous mode off?  We need to do that
3145            in other places as well - and I don't think that works all the
3146            time in any case, due to libpcap bugs. */
3147
3148         /* Well, we should be able to start capturing.
3149
3150            Sync out the capture file, so the header makes it to the file system,
3151            and send a "capture started successfully and capture file created"
3152            message to our parent so that they'll open the capture file and
3153            update its windows to indicate that we have a live capture in
3154            progress. */
3155         libpcap_dump_flush(global_ld.pdh, NULL);
3156         report_new_capture_file(capture_opts->save_file);
3157     }
3158
3159     /* initialize capture stop (and alike) conditions */
3160     init_capture_stop_conditions();
3161     /* create stop conditions */
3162     if (capture_opts->has_autostop_filesize)
3163         cnd_autostop_size =
3164             cnd_new(CND_CLASS_CAPTURESIZE,(long)capture_opts->autostop_filesize * 1024);
3165     if (capture_opts->has_autostop_duration)
3166         cnd_autostop_duration =
3167             cnd_new(CND_CLASS_TIMEOUT,(gint32)capture_opts->autostop_duration);
3168
3169     if (capture_opts->multi_files_on) {
3170         if (capture_opts->has_file_duration)
3171             cnd_file_duration =
3172                 cnd_new(CND_CLASS_TIMEOUT, capture_opts->file_duration);
3173
3174         if (capture_opts->has_autostop_files)
3175             cnd_autostop_files =
3176                 cnd_new(CND_CLASS_CAPTURESIZE, capture_opts->autostop_files);
3177     }
3178
3179     /* init the time values */
3180 #ifdef WIN32
3181     upd_time = GetTickCount();
3182 #else
3183     gettimeofday(&upd_time, NULL);
3184 #endif
3185
3186     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO, "Capture loop running!");
3187
3188     /* WOW, everything is prepared! */
3189     /* please fasten your seat belts, we will enter now the actual capture loop */
3190     if (use_threads) {
3191         pcap_queue = g_async_queue_new();
3192         pcap_queue_bytes = 0;
3193         pcap_queue_packets = 0;
3194         for (i = 0; i < global_ld.pcaps->len; i++) {
3195             pcap_opts = g_array_index(global_ld.pcaps, pcap_options *, i);
3196             pcap_opts->tid = g_thread_create(pcap_read_handler, pcap_opts, TRUE, NULL);
3197         }
3198     }
3199     while (global_ld.go) {
3200         /* dispatch incoming packets */
3201         if (use_threads) {
3202             GTimeVal write_thread_time;
3203             pcap_queue_element *queue_element;
3204
3205             g_get_current_time(&write_thread_time);
3206             g_time_val_add(&write_thread_time, WRITER_THREAD_TIMEOUT);
3207             g_async_queue_lock(pcap_queue);
3208             queue_element = g_async_queue_timed_pop_unlocked(pcap_queue, &write_thread_time);
3209             if (queue_element) {
3210                 pcap_queue_bytes -= queue_element->phdr.caplen;
3211                 pcap_queue_packets -= 1;
3212             }
3213             g_async_queue_unlock(pcap_queue);
3214             if (queue_element) {
3215                 g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO,
3216                       "Dequeued a packet of length %d captured on interface %d.",
3217                       queue_element->phdr.caplen, queue_element->pcap_opts->interface_id);
3218
3219                 capture_loop_write_packet_cb((u_char *) queue_element->pcap_opts,
3220                                              &queue_element->phdr,
3221                                              queue_element->pd);
3222                 g_free(queue_element->pd);
3223                 g_free(queue_element);
3224                 inpkts = 1;
3225             } else {
3226                 inpkts = 0;
3227             }
3228         } else {
3229             pcap_opts = g_array_index(global_ld.pcaps, pcap_options *, 0);
3230             inpkts = capture_loop_dispatch(&global_ld, errmsg,
3231                                            sizeof(errmsg), pcap_opts);
3232         }
3233 #ifdef SIGINFO
3234         /* Were we asked to print packet counts by the SIGINFO handler? */
3235         if (global_ld.report_packet_count) {
3236             fprintf(stderr, "%u packet%s captured\n", global_ld.packet_count,
3237                     plurality(global_ld.packet_count, "", "s"));
3238             global_ld.report_packet_count = FALSE;
3239         }
3240 #endif
3241
3242 #ifdef _WIN32
3243         /* any news from our parent (signal pipe)? -> just stop the capture */
3244         if (!signal_pipe_check_running()) {
3245             global_ld.go = FALSE;
3246         }
3247 #endif
3248
3249         if (inpkts > 0) {
3250             global_ld.inpkts_to_sync_pipe += inpkts;
3251
3252             /* check capture size condition */
3253             if (cnd_autostop_size != NULL &&
3254                 cnd_eval(cnd_autostop_size, (guint32)global_ld.bytes_written)) {
3255                 /* Capture size limit reached, do we have another file? */
3256                 if (!do_file_switch_or_stop(capture_opts, cnd_autostop_files,
3257                                             cnd_autostop_size, cnd_file_duration))
3258                     continue;
3259             } /* cnd_autostop_size */
3260             if (capture_opts->output_to_pipe) {
3261                 libpcap_dump_flush(global_ld.pdh, NULL);
3262             }
3263         } /* inpkts */
3264
3265         /* Only update once every 500ms so as not to overload slow displays.
3266          * This also prevents too much context-switching between the dumpcap
3267          * and wireshark processes.
3268          */
3269 #define DUMPCAP_UPD_TIME 500
3270
3271 #ifdef WIN32
3272         cur_time = GetTickCount();
3273         if ( (cur_time - upd_time) > DUMPCAP_UPD_TIME) {
3274 #else
3275         gettimeofday(&cur_time, NULL);
3276         if ((cur_time.tv_sec * 1000000 + cur_time.tv_usec) >
3277             (upd_time.tv_sec * 1000000 + upd_time.tv_usec + DUMPCAP_UPD_TIME*1000)) {
3278 #endif
3279
3280             upd_time = cur_time;
3281
3282 #if 0
3283             if (pcap_stats(pch, stats) >= 0) {
3284                 *stats_known = TRUE;
3285             }
3286 #endif
3287             /* Let the parent process know. */
3288             if (global_ld.inpkts_to_sync_pipe) {
3289                 /* do sync here */
3290                 libpcap_dump_flush(global_ld.pdh, NULL);
3291
3292                 /* Send our parent a message saying we've written out
3293                    "global_ld.inpkts_to_sync_pipe" packets to the capture file. */
3294                 if (!quiet)
3295                     report_packet_count(global_ld.inpkts_to_sync_pipe);
3296
3297                 global_ld.inpkts_to_sync_pipe = 0;
3298             }
3299
3300             /* check capture duration condition */
3301             if (cnd_autostop_duration != NULL && cnd_eval(cnd_autostop_duration)) {
3302                 /* The maximum capture time has elapsed; stop the capture. */
3303                 global_ld.go = FALSE;
3304                 continue;
3305             }
3306
3307             /* check capture file duration condition */
3308             if (cnd_file_duration != NULL && cnd_eval(cnd_file_duration)) {
3309                 /* duration limit reached, do we have another file? */
3310                 if (!do_file_switch_or_stop(capture_opts, cnd_autostop_files,
3311                                             cnd_autostop_size, cnd_file_duration))
3312                     continue;
3313             } /* cnd_file_duration */
3314         }
3315     }
3316
3317     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO, "Capture loop stopping ...");
3318     if (use_threads) {
3319         pcap_queue_element *queue_element;
3320
3321         for (i = 0; i < global_ld.pcaps->len; i++) {
3322             pcap_opts = g_array_index(global_ld.pcaps, pcap_options *, i);
3323             g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO, "Waiting for thread of interface %u...",
3324                   pcap_opts->interface_id);
3325             g_thread_join(pcap_opts->tid);
3326             g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO, "Thread of interface %u terminated.",
3327                   pcap_opts->interface_id);
3328         }
3329         while (1) {
3330             g_async_queue_lock(pcap_queue);
3331             queue_element = g_async_queue_try_pop_unlocked(pcap_queue);
3332             if (queue_element) {
3333                 pcap_queue_bytes -= queue_element->phdr.caplen;
3334                 pcap_queue_packets -= 1;
3335             }
3336             g_async_queue_unlock(pcap_queue);
3337             if (queue_element == NULL) {
3338                 break;
3339             }
3340             g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO,
3341                   "Dequeued a packet of length %d captured on interface %d.",
3342                   queue_element->phdr.caplen, queue_element->pcap_opts->interface_id);
3343             capture_loop_write_packet_cb((u_char *)queue_element->pcap_opts,
3344                                          &queue_element->phdr,
3345                                          queue_element->pd);
3346             g_free(queue_element->pd);
3347             g_free(queue_element);
3348             global_ld.inpkts_to_sync_pipe += 1;
3349             if (capture_opts->output_to_pipe) {
3350                 libpcap_dump_flush(global_ld.pdh, NULL);
3351             }
3352         }
3353     }
3354
3355
3356     /* delete stop conditions */
3357     if (cnd_file_duration != NULL)
3358         cnd_delete(cnd_file_duration);
3359     if (cnd_autostop_files != NULL)
3360         cnd_delete(cnd_autostop_files);
3361     if (cnd_autostop_size != NULL)
3362         cnd_delete(cnd_autostop_size);
3363     if (cnd_autostop_duration != NULL)
3364         cnd_delete(cnd_autostop_duration);
3365
3366     /* did we have a pcap (input) error? */
3367     for (i = 0; i < capture_opts->ifaces->len; i++) {
3368         pcap_opts = g_array_index(global_ld.pcaps, pcap_options *, i);
3369         if (pcap_opts->pcap_err) {
3370             /* On Linux, if an interface goes down while you're capturing on it,
3371                you'll get a "recvfrom: Network is down" or
3372                "The interface went down" error (ENETDOWN).
3373                (At least you will if g_strerror() doesn't show a local translation
3374                of the error.)
3375
3376                On FreeBSD and OS X, if a network adapter disappears while
3377                you're capturing on it, you'll get a "read: Device not configured"
3378                error (ENXIO).  (See previous parenthetical note.)
3379
3380                On OpenBSD, you get "read: I/O error" (EIO) in the same case.
3381
3382                These should *not* be reported to the Wireshark developers. */
3383             char *cap_err_str;
3384
3385             cap_err_str = pcap_geterr(pcap_opts->pcap_h);
3386             if (strcmp(cap_err_str, "recvfrom: Network is down") == 0 ||
3387                 strcmp(cap_err_str, "The interface went down") == 0 ||
3388                 strcmp(cap_err_str, "read: Device not configured") == 0 ||
3389                 strcmp(cap_err_str, "read: I/O error") == 0 ||
3390                 strcmp(cap_err_str, "read error: PacketReceivePacket failed") == 0) {
3391                 report_capture_error("The network adapter on which the capture was being done "
3392                                      "is no longer running; the capture has stopped.",
3393                                      "");
3394             } else {
3395                 g_snprintf(errmsg, sizeof(errmsg), "Error while capturing packets: %s",
3396                            cap_err_str);
3397                 report_capture_error(errmsg, please_report);
3398             }
3399             break;
3400         } else if (pcap_opts->from_cap_pipe && pcap_opts->cap_pipe_err == PIPERR) {
3401             report_capture_error(errmsg, "");
3402             break;
3403         }
3404     }
3405     /* did we have an output error while capturing? */
3406     if (global_ld.err == 0) {
3407         write_ok = TRUE;
3408     } else {
3409         capture_loop_get_errmsg(errmsg, sizeof(errmsg), capture_opts->save_file,
3410                                 global_ld.err, FALSE);
3411         report_capture_error(errmsg, please_report);
3412         write_ok = FALSE;
3413     }
3414
3415     if (capture_opts->saving_to_file) {
3416         /* close the output file */
3417         close_ok = capture_loop_close_output(capture_opts, &global_ld, &err_close);
3418     } else
3419         close_ok = TRUE;
3420
3421     /* there might be packets not yet notified to the parent */
3422     /* (do this after closing the file, so all packets are already flushed) */
3423     if(global_ld.inpkts_to_sync_pipe) {
3424         if (!quiet)
3425             report_packet_count(global_ld.inpkts_to_sync_pipe);
3426         global_ld.inpkts_to_sync_pipe = 0;
3427     }
3428
3429     /* If we've displayed a message about a write error, there's no point
3430        in displaying another message about an error on close. */
3431     if (!close_ok && write_ok) {
3432         capture_loop_get_errmsg(errmsg, sizeof(errmsg), capture_opts->save_file, err_close,
3433                                 TRUE);
3434         report_capture_error(errmsg, "");
3435     }
3436
3437     /*
3438      * XXX We exhibit different behaviour between normal mode and sync mode
3439      * when the pipe is stdin and not already at EOF.  If we're a child, the
3440      * parent's stdin isn't closed, so if the user starts another capture,
3441      * cap_pipe_open_live() will very likely not see the expected magic bytes and
3442      * will say "Unrecognized libpcap format".  On the other hand, in normal
3443      * mode, cap_pipe_open_live() will say "End of file on pipe during open".
3444      */
3445
3446     report_capture_count(TRUE);
3447
3448     /* get packet drop statistics from pcap */
3449     for (i = 0; i < capture_opts->ifaces->len; i++) {
3450         guint32 received;
3451         guint32 dropped;
3452
3453         pcap_opts = g_array_index(global_ld.pcaps, pcap_options *, i);
3454         interface_opts = g_array_index(capture_opts->ifaces, interface_options, i);
3455         received = pcap_opts->received;
3456         dropped = pcap_opts->dropped;
3457         if (pcap_opts->pcap_h != NULL) {
3458             g_assert(!pcap_opts->from_cap_pipe);
3459             /* Get the capture statistics, so we know how many packets were dropped. */
3460             if (pcap_stats(pcap_opts->pcap_h, stats) >= 0) {
3461                 *stats_known = TRUE;
3462                 /* Let the parent process know. */
3463                 dropped += stats->ps_drop;
3464             } else {
3465                 g_snprintf(errmsg, sizeof(errmsg),
3466                            "Can't get packet-drop statistics: %s",
3467                            pcap_geterr(pcap_opts->pcap_h));
3468                 report_capture_error(errmsg, please_report);
3469             }
3470         }
3471         report_packet_drops(received, dropped, interface_opts.name);
3472     }
3473
3474     /* close the input file (pcap or capture pipe) */
3475     capture_loop_close_input(&global_ld);
3476
3477     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO, "Capture loop stopped!");
3478
3479     /* ok, if the write and the close were successful. */
3480     return write_ok && close_ok;
3481
3482 error:
3483     if (capture_opts->multi_files_on) {
3484         /* cleanup ringbuffer */
3485         ringbuf_error_cleanup();
3486     } else {
3487         /* We can't use the save file, and we have no FILE * for the stream
3488            to close in order to close it, so close the FD directly. */
3489         if (global_ld.save_file_fd != -1) {
3490             ws_close(global_ld.save_file_fd);
3491         }
3492
3493         /* We couldn't even start the capture, so get rid of the capture
3494            file. */
3495         if (capture_opts->save_file != NULL) {
3496             ws_unlink(capture_opts->save_file);
3497             g_free(capture_opts->save_file);
3498         }
3499     }
3500     capture_opts->save_file = NULL;
3501     if (cfilter_error)
3502         report_cfilter_error(capture_opts, error_index, errmsg);
3503     else
3504         report_capture_error(errmsg, secondary_errmsg);
3505
3506     /* close the input file (pcap or cap_pipe) */
3507     capture_loop_close_input(&global_ld);
3508
3509     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO, "Capture loop stopped with error");
3510
3511     return FALSE;
3512 }
3513
3514
3515 static void capture_loop_stop(void)
3516 {
3517 #ifdef HAVE_PCAP_BREAKLOOP
3518     guint i;
3519     pcap_options *pcap_opts;
3520
3521     for (i = 0; i < global_ld.pcaps->len; i++) {
3522         pcap_opts = g_array_index(global_ld.pcaps, pcap_options *, i);
3523         if (pcap_opts->pcap_h != NULL)
3524             pcap_breakloop(pcap_opts->pcap_h);
3525     }
3526 #endif
3527     global_ld.go = FALSE;
3528 }
3529
3530
3531 static void
3532 capture_loop_get_errmsg(char *errmsg, int errmsglen, const char *fname,
3533                         int err, gboolean is_close)
3534 {
3535     switch (err) {
3536
3537     case ENOSPC:
3538         g_snprintf(errmsg, errmsglen,
3539                    "Not all the packets could be written to the file"
3540                    " to which the capture was being saved\n"
3541                    "(\"%s\") because there is no space left on the file system\n"
3542                    "on which that file resides.",
3543                    fname);
3544         break;
3545
3546 #ifdef EDQUOT
3547     case EDQUOT:
3548         g_snprintf(errmsg, errmsglen,
3549                    "Not all the packets could be written to the file"
3550                    " to which the capture was being saved\n"
3551                    "(\"%s\") because you are too close to, or over,"
3552                    " your disk quota\n"
3553                    "on the file system on which that file resides.",
3554                    fname);
3555         break;
3556 #endif
3557
3558     default:
3559         if (is_close) {
3560             g_snprintf(errmsg, errmsglen,
3561                        "The file to which the capture was being saved\n"
3562                        "(\"%s\") could not be closed: %s.",
3563                        fname, g_strerror(err));
3564         } else {
3565             g_snprintf(errmsg, errmsglen,
3566                        "An error occurred while writing to the file"
3567                        " to which the capture was being saved\n"
3568                        "(\"%s\"): %s.",
3569                        fname, g_strerror(err));
3570         }
3571         break;
3572     }
3573 }
3574
3575
3576 /* one packet was captured, process it */
3577 static void
3578 capture_loop_write_packet_cb(u_char *pcap_opts_p, const struct pcap_pkthdr *phdr,
3579                              const u_char *pd)
3580 {
3581     pcap_options *pcap_opts = (pcap_options *) (void *) pcap_opts_p;
3582     int err;
3583
3584     /* We may be called multiple times from pcap_dispatch(); if we've set
3585        the "stop capturing" flag, ignore this packet, as we're not
3586        supposed to be saving any more packets. */
3587     if (!global_ld.go)
3588         return;
3589
3590     if (global_ld.pdh) {
3591         gboolean successful;
3592
3593         /* We're supposed to write the packet to a file; do so.
3594            If this fails, set "ld->go" to FALSE, to stop the capture, and set
3595            "ld->err" to the error. */
3596         if (global_capture_opts.use_pcapng) {
3597             successful = libpcap_write_enhanced_packet_block(global_ld.pdh, phdr, pcap_opts->interface_id, pd, &global_ld.bytes_written, &err);
3598         } else {
3599             successful = libpcap_write_packet(global_ld.pdh, phdr, pd, &global_ld.bytes_written, &err);
3600         }
3601         if (!successful) {
3602             global_ld.go = FALSE;
3603             global_ld.err = err;
3604         } else {
3605             g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO,
3606                   "Wrote a packet of length %d captured on interface %u.",
3607                    phdr->caplen, pcap_opts->interface_id);
3608             global_ld.packet_count++;
3609             pcap_opts->received++;
3610             /* if the user told us to stop after x packets, do we already have enough? */
3611             if ((global_ld.packet_max > 0) && (global_ld.packet_count >= global_ld.packet_max)) {
3612                 global_ld.go = FALSE;
3613             }
3614         }
3615     }
3616 }
3617
3618 /* one packet was captured, queue it */
3619 static void
3620 capture_loop_queue_packet_cb(u_char *pcap_opts_p, const struct pcap_pkthdr *phdr,
3621                              const u_char *pd)
3622 {
3623     pcap_options *pcap_opts = (pcap_options *) (void *) pcap_opts_p;
3624     pcap_queue_element *queue_element;
3625     gboolean limit_reached;
3626
3627     /* We may be called multiple times from pcap_dispatch(); if we've set
3628        the "stop capturing" flag, ignore this packet, as we're not
3629        supposed to be saving any more packets. */
3630     if (!global_ld.go)
3631         return;
3632
3633     queue_element = (pcap_queue_element *)g_malloc(sizeof(pcap_queue_element));
3634     if (queue_element == NULL) {
3635        pcap_opts->dropped++;
3636        return;
3637     }
3638     queue_element->pcap_opts = pcap_opts;
3639     queue_element->phdr = *phdr;
3640     queue_element->pd = (u_char *)g_malloc(phdr->caplen);
3641     if (queue_element->pd == NULL) {
3642         pcap_opts->dropped++;
3643         g_free(queue_element);
3644         return;
3645     }
3646     memcpy(queue_element->pd, pd, phdr->caplen);
3647     g_async_queue_lock(pcap_queue);
3648     if (((pcap_queue_byte_limit > 0) && (pcap_queue_bytes < pcap_queue_byte_limit)) &&
3649         ((pcap_queue_packet_limit > 0) && (pcap_queue_packets < pcap_queue_packet_limit))) {
3650         limit_reached = FALSE;
3651         g_async_queue_push_unlocked(pcap_queue, queue_element);
3652         pcap_queue_bytes += phdr->caplen;
3653         pcap_queue_packets += 1;
3654     } else {
3655         limit_reached = TRUE;
3656     }
3657     g_async_queue_unlock(pcap_queue);
3658     if (limit_reached) {
3659         pcap_opts->dropped++;
3660         g_free(queue_element->pd);
3661         g_free(queue_element);
3662         g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO,
3663               "Dropped a packet of length %d captured on interface %u.",
3664               phdr->caplen, pcap_opts->interface_id);
3665     } else {
3666         pcap_opts->received++;
3667         g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO,
3668               "Queued a packet of length %d captured on interface %u.",
3669               phdr->caplen, pcap_opts->interface_id);
3670     }
3671     /* I don't want to hold the mutex over the debug output. So the
3672        output may be wrong */
3673     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO,
3674           "Queue size is now %" G_GINT64_MODIFIER "d bytes (%" G_GINT64_MODIFIER "d packets)",
3675           pcap_queue_bytes, pcap_queue_packets);
3676 }
3677
3678 /* And now our feature presentation... [ fade to music ] */
3679 int
3680 main(int argc, char *argv[])
3681 {
3682     int                  opt;
3683     gboolean             arg_error = FALSE;
3684
3685 #ifdef _WIN32
3686     WSADATA              wsaData;
3687 #else
3688     struct sigaction action, oldaction;
3689 #endif
3690
3691     gboolean             start_capture = TRUE;
3692     gboolean             stats_known;
3693     struct pcap_stat     stats;
3694     GLogLevelFlags       log_flags;
3695     gboolean             list_interfaces = FALSE;
3696     gboolean             list_link_layer_types = FALSE;
3697 #ifdef HAVE_BPF_IMAGE
3698     gboolean             print_bpf_code = FALSE;
3699 #endif
3700     gboolean             machine_readable = FALSE;
3701     gboolean             print_statistics = FALSE;
3702     int                  status, run_once_args = 0;
3703     gint                 i;
3704     guint                j;
3705 #if defined(__APPLE__) && defined(__LP64__)
3706     struct utsname       osinfo;
3707 #endif
3708
3709 #ifdef _WIN32
3710     arg_list_utf_16to8(argc, argv);
3711 #endif /* _WIN32 */
3712
3713 #ifdef _WIN32
3714     /*
3715      * Initialize our DLL search path. MUST be called before LoadLibrary
3716      * or g_module_open.
3717      */
3718     ws_init_dll_search_path();
3719 #endif
3720
3721 #ifdef HAVE_PCAP_REMOTE
3722 #define OPTSTRING_A "A:"
3723 #define OPTSTRING_r "r"
3724 #define OPTSTRING_u "u"
3725 #else
3726 #define OPTSTRING_A ""
3727 #define OPTSTRING_r ""
3728 #define OPTSTRING_u ""
3729 #endif
3730
3731 #ifdef HAVE_PCAP_SETSAMPLING
3732 #define OPTSTRING_m "m:"
3733 #else
3734 #define OPTSTRING_m ""
3735 #endif
3736
3737 #if defined(_WIN32) || defined(HAVE_PCAP_CREATE)
3738 #define OPTSTRING_B "B:"
3739 #else
3740 #define OPTSTRING_B ""
3741 #endif  /* _WIN32 or HAVE_PCAP_CREATE */
3742
3743 #ifdef HAVE_PCAP_CREATE
3744 #define OPTSTRING_I "I"
3745 #else
3746 #define OPTSTRING_I ""
3747 #endif
3748
3749 #ifdef HAVE_BPF_IMAGE
3750 #define OPTSTRING_d "d"
3751 #else
3752 #define OPTSTRING_d ""
3753 #endif
3754
3755 #ifdef USE_THREADS
3756 #define OPTSTRING_t "t"
3757 #else
3758 #define OPTSTRING_t ""
3759 #endif
3760 #define OPTSTRING "a:" OPTSTRING_A "b:" OPTSTRING_B "c:" OPTSTRING_d "Df:ghi:" OPTSTRING_I "L" OPTSTRING_m "MnpPq" OPTSTRING_r "Ss:" OPTSTRING_t OPTSTRING_u "vw:y:Z:"
3761
3762 #ifdef DEBUG_CHILD_DUMPCAP
3763     if ((debug_log = ws_fopen("dumpcap_debug_log.tmp","w")) == NULL) {
3764         fprintf (stderr, "Unable to open debug log file !\n");
3765         exit (1);
3766     }
3767 #endif
3768
3769 #if defined(__APPLE__) && defined(__LP64__)
3770     /*
3771      * Is this Mac OS X 10.6.0, 10.6.1, 10.6.3, or 10.6.4?  If so, we need
3772      * a bug workaround - timeouts less than 1 second don't work with libpcap
3773      * in 64-bit code.  (The bug was introduced in 10.6, fixed in 10.6.2,
3774      * re-introduced in 10.6.3, not fixed in 10.6.4, and fixed in 10.6.5.
3775      * The problem is extremely unlikely to be reintroduced in a future
3776      * release.)
3777      */
3778     if (uname(&osinfo) == 0) {
3779         /*
3780          * Mac OS X 10.x uses Darwin {x+4}.0.0.  Mac OS X 10.x.y uses Darwin
3781          * {x+4}.y.0 (except that 10.6.1 appears to have a uname version
3782          * number of 10.0.0, not 10.1.0 - go figure).
3783          */
3784         if (strcmp(osinfo.release, "10.0.0") == 0 ||    /* 10.6, 10.6.1 */
3785             strcmp(osinfo.release, "10.3.0") == 0 ||    /* 10.6.3 */
3786             strcmp(osinfo.release, "10.4.0") == 0)              /* 10.6.4 */
3787             need_timeout_workaround = TRUE;
3788     }
3789 #endif
3790
3791     /*
3792      * Determine if dumpcap is being requested to run in a special
3793      * capture_child mode by going thru the command line args to see if
3794      * a -Z is present. (-Z is a hidden option).
3795      *
3796      * The primary result of running in capture_child mode is that
3797      * all messages sent out on stderr are in a special type/len/string
3798      * format to allow message processing by type.  These messages include
3799      * error messages if dumpcap fails to start the operation it was
3800      * requested to do, as well as various "status" messages which are sent
3801      * when an actual capture is in progress, and a "success" message sent
3802      * if dumpcap was requested to perform an operation other than a
3803      * capture.
3804      *
3805      * Capture_child mode would normally be requested by a parent process
3806      * which invokes dumpcap and obtains dumpcap stderr output via a pipe
3807      * to which dumpcap stderr has been redirected.  It might also have
3808      * another pipe to obtain dumpcap stdout output; for operations other
3809      * than a capture, that information is formatted specially for easier
3810      * parsing by the parent process.
3811      *
3812      * Capture_child mode needs to be determined immediately upon
3813      * startup so that any messages generated by dumpcap in this mode
3814      * (eg: during initialization) will be formatted properly.
3815      */
3816
3817     for (i=1; i<argc; i++) {
3818         if (strcmp("-Z", argv[i]) == 0) {
3819             capture_child = TRUE;
3820             machine_readable = TRUE;  /* request machine-readable output */
3821 #ifdef _WIN32
3822             /* set output pipe to binary mode, to avoid ugly text conversions */
3823             _setmode(2, O_BINARY);
3824 #endif
3825         }
3826     }
3827
3828     /* The default_log_handler will use stdout, which makes trouble in   */
3829     /* capture child mode, as it uses stdout for it's sync_pipe.         */
3830     /* So: the filtering is done in the console_log_handler and not here.*/
3831     /* We set the log handlers right up front to make sure that any log  */
3832     /* messages when running as child will be sent back to the parent    */
3833     /* with the correct format.                                          */
3834
3835     log_flags =
3836         G_LOG_LEVEL_ERROR|
3837         G_LOG_LEVEL_CRITICAL|
3838         G_LOG_LEVEL_WARNING|
3839         G_LOG_LEVEL_MESSAGE|
3840         G_LOG_LEVEL_INFO|
3841         G_LOG_LEVEL_DEBUG|
3842         G_LOG_FLAG_FATAL|G_LOG_FLAG_RECURSION;
3843
3844     g_log_set_handler(NULL,
3845                       log_flags,
3846                       console_log_handler, NULL /* user_data */);
3847     g_log_set_handler(LOG_DOMAIN_MAIN,
3848                       log_flags,
3849                       console_log_handler, NULL /* user_data */);
3850     g_log_set_handler(LOG_DOMAIN_CAPTURE,
3851                       log_flags,
3852                       console_log_handler, NULL /* user_data */);
3853     g_log_set_handler(LOG_DOMAIN_CAPTURE_CHILD,
3854                       log_flags,
3855                       console_log_handler, NULL /* user_data */);
3856
3857     /* Initialize the pcaps list */
3858     global_ld.pcaps = g_array_new(FALSE, FALSE, sizeof(pcap_options *));
3859
3860     /* Initialize the thread system */
3861 #ifdef USE_THREADS
3862     if (!g_thread_supported())
3863         g_thread_init(NULL);
3864 #endif
3865 #ifdef _WIN32
3866     /* Load wpcap if possible. Do this before collecting the run-time version information */
3867     load_wpcap();
3868
3869     /* ... and also load the packet.dll from wpcap */
3870     /* XXX - currently not required, may change later. */
3871     /*wpcap_packet_load();*/
3872
3873     /* Start windows sockets */
3874     WSAStartup( MAKEWORD( 1, 1 ), &wsaData );
3875
3876     /* Set handler for Ctrl+C key */
3877     SetConsoleCtrlHandler(capture_cleanup_handler, TRUE);
3878 #else
3879     /* Catch SIGINT and SIGTERM and, if we get either of them, clean up
3880        and exit. */
3881     action.sa_handler = capture_cleanup_handler;
3882     /*
3883      * Arrange that system calls not get restarted, because when
3884      * our signal handler returns we don't want to restart
3885      * a call that was waiting for packets to arrive.
3886      */
3887     action.sa_flags = 0;
3888     sigemptyset(&action.sa_mask);
3889     sigaction(SIGTERM, &action, NULL);
3890     sigaction(SIGINT, &action, NULL);
3891     sigaction(SIGPIPE, &action, NULL);
3892     sigaction(SIGHUP, NULL, &oldaction);
3893     if (oldaction.sa_handler == SIG_DFL)
3894         sigaction(SIGHUP, &action, NULL);
3895
3896 #ifdef SIGINFO
3897     /* Catch SIGINFO and, if we get it and we're capturing in
3898        quiet mode, report the number of packets we've captured. */
3899     action.sa_handler = report_counts_siginfo;
3900     action.sa_flags = SA_RESTART;
3901     sigemptyset(&action.sa_mask);
3902     sigaction(SIGINFO, &action, NULL);
3903 #endif /* SIGINFO */
3904 #endif  /* _WIN32 */
3905
3906     /* ----------------------------------------------------------------- */
3907     /* Privilege and capability handling                                 */
3908     /* Cases:                                                            */
3909     /* 1. Running not as root or suid root; no special capabilities.     */
3910     /*    Action: none                                                   */
3911     /*                                                                   */
3912     /* 2. Running logged in as root (euid=0; ruid=0); Not using libcap.  */
3913     /*    Action: none                                                   */
3914     /*                                                                   */
3915     /* 3. Running logged in as root (euid=0; ruid=0). Using libcap.      */
3916     /*    Action:                                                        */
3917     /*      - Near start of program: Enable NET_RAW and NET_ADMIN        */
3918     /*        capabilities; Drop all other capabilities;                 */
3919     /*      - If not -w  (ie: doing -S or -D, etc) run to completion;    */
3920     /*        else: after  pcap_open_live() in capture_loop_open_input() */
3921     /*         drop all capabilities (NET_RAW and NET_ADMIN);            */
3922     /*         (Note: this means that the process, although logged in    */
3923     /*          as root, does not have various permissions such as the   */
3924     /*          ability to bypass file access permissions).              */
3925     /*      XXX: Should we just leave capabilities alone in this case    */
3926     /*          so that user gets expected effect that root can do       */
3927     /*          anything ??                                              */
3928     /*                                                                   */
3929     /* 4. Running as suid root (euid=0, ruid=n); Not using libcap.       */
3930     /*    Action:                                                        */
3931     /*      - If not -w  (ie: doing -S or -D, etc) run to completion;    */
3932     /*        else: after  pcap_open_live() in capture_loop_open_input() */
3933     /*         drop suid root (set euid=ruid).(ie: keep suid until after */
3934     /*         pcap_open_live).                                          */
3935     /*                                                                   */
3936     /* 5. Running as suid root (euid=0, ruid=n); Using libcap.           */
3937     /*    Action:                                                        */
3938     /*      - Near start of program: Enable NET_RAW and NET_ADMIN        */
3939     /*        capabilities; Drop all other capabilities;                 */
3940     /*        Drop suid privileges (euid=ruid);                          */
3941     /*      - If not -w  (ie: doing -S or -D, etc) run to completion;    */
3942     /*        else: after  pcap_open_live() in capture_loop_open_input() */
3943     /*         drop all capabilities (NET_RAW and NET_ADMIN).            */
3944     /*                                                                   */
3945     /*      XXX: For some Linux versions/distros with capabilities       */
3946     /*        a 'normal' process with any capabilities cannot be         */
3947     /*        'killed' (signaled) from another (same uid) non-privileged */
3948     /*        process.                                                   */
3949     /*        For example: If (non-suid) Wireshark forks a               */
3950     /*        child suid dumpcap which acts as described here (case 5),  */
3951     /*        Wireshark will be unable to kill (signal) the child        */
3952     /*        dumpcap process until the capabilities have been dropped   */
3953     /*        (after pcap_open_live()).                                  */
3954     /*        This behaviour will apparently be changed in the kernel    */
3955     /*        to allow the kill (signal) in this case.                   */
3956     /*        See the following for details:                             */
3957     /*           http://www.mail-archive.com/  [wrapped]                 */
3958     /*             linux-security-module@vger.kernel.org/msg02913.html   */
3959     /*                                                                   */
3960     /*        It is therefore conceivable that if dumpcap somehow hangs  */
3961     /*        in pcap_open_live or before that wireshark will not        */
3962     /*        be able to stop dumpcap using a signal (INT, TERM, etc).   */
3963     /*        In this case, exiting wireshark will kill the child        */
3964     /*        dumpcap process.                                           */
3965     /*                                                                   */
3966     /* 6. Not root or suid root; Running with NET_RAW & NET_ADMIN        */
3967     /*     capabilities; Using libcap.  Note: capset cmd (which see)     */
3968     /*     used to assign capabilities to file.                          */
3969     /*    Action:                                                        */
3970     /*      - If not -w  (ie: doing -S or -D, etc) run to completion;    */
3971     /*        else: after  pcap_open_live() in capture_loop_open_input() */
3972     /*         drop all capabilities (NET_RAW and NET_ADMIN)             */
3973     /*                                                                   */
3974     /* ToDo: -S (stats) should drop privileges/capabilities when no      */
3975     /*       longer required (similar to capture).                       */
3976     /*                                                                   */
3977     /* ----------------------------------------------------------------- */
3978
3979     init_process_policies();
3980
3981 #ifdef HAVE_LIBCAP
3982     /* If 'started with special privileges' (and using libcap)  */
3983     /*   Set to keep only NET_RAW and NET_ADMIN capabilities;   */
3984     /*   Set euid/egid = ruid/rgid to remove suid privileges    */
3985     relinquish_privs_except_capture();
3986 #endif
3987
3988     /* Set the initial values in the capture options. This might be overwritten
3989        by the command line parameters. */
3990     capture_opts_init(&global_capture_opts, NULL);
3991
3992     /* We always save to a file - if no file was specified, we save to a
3993        temporary file. */
3994     global_capture_opts.saving_to_file      = TRUE;
3995     global_capture_opts.has_ring_num_files  = TRUE;
3996
3997     /* Now get our args */
3998     while ((opt = getopt(argc, argv, OPTSTRING)) != -1) {
3999         switch (opt) {
4000         case 'h':        /* Print help and exit */
4001             print_usage(TRUE);
4002             exit_main(0);
4003             break;
4004         case 'v':        /* Show version and exit */
4005         {
4006             GString             *comp_info_str;
4007             GString             *runtime_info_str;
4008             /* Assemble the compile-time version information string */
4009             comp_info_str = g_string_new("Compiled ");
4010             get_compiled_version_info(comp_info_str, NULL, NULL);
4011
4012             /* Assemble the run-time version information string */
4013             runtime_info_str = g_string_new("Running ");
4014             get_runtime_version_info(runtime_info_str, NULL);
4015             show_version(comp_info_str, runtime_info_str);
4016             g_string_free(comp_info_str, TRUE);
4017             g_string_free(runtime_info_str, TRUE);
4018             exit_main(0);
4019             break;
4020         }
4021         /*** capture option specific ***/
4022         case 'a':        /* autostop criteria */
4023         case 'b':        /* Ringbuffer option */
4024         case 'c':        /* Capture x packets */
4025         case 'f':        /* capture filter */
4026         case 'i':        /* Use interface x */
4027         case 'n':        /* Use pcapng format */
4028         case 'p':        /* Don't capture in promiscuous mode */
4029         case 'P':        /* Use pcap format */
4030         case 's':        /* Set the snapshot (capture) length */
4031         case 'w':        /* Write to capture file x */
4032         case 'g':        /* enable group read accesson file(s) */
4033         case 'y':        /* Set the pcap data link type */
4034 #ifdef HAVE_PCAP_REMOTE
4035         case 'u':        /* Use UDP for data transfer */
4036         case 'r':        /* Capture own RPCAP traffic too */
4037         case 'A':        /* Authentication */
4038 #endif
4039 #ifdef HAVE_PCAP_SETSAMPLING
4040         case 'm':        /* Sampling */
4041 #endif
4042 #if defined(_WIN32) || defined(HAVE_PCAP_CREATE)
4043         case 'B':        /* Buffer size */
4044 #endif /* _WIN32 or HAVE_PCAP_CREATE */
4045 #ifdef HAVE_PCAP_CREATE
4046         case 'I':        /* Monitor mode */
4047 #endif
4048             status = capture_opts_add_opt(&global_capture_opts, opt, optarg, &start_capture);
4049             if(status != 0) {
4050                 exit_main(status);
4051             }
4052             break;
4053             /*** hidden option: Wireshark child mode (using binary output messages) ***/
4054         case 'Z':
4055             capture_child = TRUE;
4056 #ifdef _WIN32
4057             /* set output pipe to binary mode, to avoid ugly text conversions */
4058             _setmode(2, O_BINARY);
4059             /*
4060              * optarg = the control ID, aka the PPID, currently used for the
4061              * signal pipe name.
4062              */
4063             if (strcmp(optarg, SIGNAL_PIPE_CTRL_ID_NONE) != 0) {
4064                 sig_pipe_name = g_strdup_printf(SIGNAL_PIPE_FORMAT, optarg);
4065                 sig_pipe_handle = CreateFile(utf_8to16(sig_pipe_name),
4066                                              GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, NULL);
4067
4068                 if (sig_pipe_handle == INVALID_HANDLE_VALUE) {
4069                     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO,
4070                           "Signal pipe: Unable to open %s.  Dead parent?",
4071                           sig_pipe_name);
4072                     exit_main(1);
4073                 }
4074             }
4075 #endif
4076             break;
4077
4078         case 'q':        /* Quiet */
4079             quiet = TRUE;
4080             break;
4081 #ifdef USE_THREADS
4082         case 't':
4083             use_threads = TRUE;
4084             break;
4085 #endif
4086             /*** all non capture option specific ***/
4087         case 'D':        /* Print a list of capture devices and exit */
4088             list_interfaces = TRUE;
4089             run_once_args++;
4090             break;
4091         case 'L':        /* Print list of link-layer types and exit */
4092             list_link_layer_types = TRUE;
4093             run_once_args++;
4094             break;
4095 #ifdef HAVE_BPF_IMAGE
4096         case 'd':        /* Print BPF code for capture filter and exit */
4097             print_bpf_code = TRUE;
4098             run_once_args++;
4099             break;
4100 #endif
4101         case 'S':        /* Print interface statistics once a second */
4102             print_statistics = TRUE;
4103             run_once_args++;
4104             break;
4105         case 'M':        /* For -D, -L, and -S, print machine-readable output */
4106             machine_readable = TRUE;
4107             break;
4108         default:
4109             cmdarg_err("Invalid Option: %s", argv[optind-1]);
4110         case '?':        /* Bad flag - print usage message */
4111             arg_error = TRUE;
4112             break;
4113         }
4114     }
4115     if (!arg_error) {
4116         argc -= optind;
4117         argv += optind;
4118         if (argc >= 1) {
4119             /* user specified file name as regular command-line argument */
4120             /* XXX - use it as the capture file name (or something else)? */
4121             argc--;
4122             argv++;
4123         }
4124         if (argc != 0) {
4125             /*
4126              * Extra command line arguments were specified; complain.
4127              * XXX - interpret as capture filter, as tcpdump and tshark do?
4128              */
4129             cmdarg_err("Invalid argument: %s", argv[0]);
4130             arg_error = TRUE;
4131         }
4132     }
4133
4134     if (arg_error) {
4135         print_usage(FALSE);
4136         exit_main(1);
4137     }
4138
4139     if (run_once_args > 1) {
4140         cmdarg_err("Only one of -D, -L, or -S may be supplied.");
4141         exit_main(1);
4142     } else if (run_once_args == 1) {
4143         /* We're supposed to print some information, rather than
4144            to capture traffic; did they specify a ring buffer option? */
4145         if (global_capture_opts.multi_files_on) {
4146             cmdarg_err("Ring buffer requested, but a capture isn't being done.");
4147             exit_main(1);
4148         }
4149     } else {
4150         /* We're supposed to capture traffic; */
4151         /* Are we capturing on multiple interface? If so, use threads and pcapng. */
4152 #ifdef USE_THREADS
4153         if (global_capture_opts.ifaces->len > 1) {
4154             use_threads = TRUE;
4155             global_capture_opts.use_pcapng = TRUE;
4156         }
4157 #endif
4158         /* Was the ring buffer option specified and, if so, does it make sense? */
4159         if (global_capture_opts.multi_files_on) {
4160             /* Ring buffer works only under certain conditions:
4161                a) ring buffer does not work with temporary files;
4162                b) it makes no sense to enable the ring buffer if the maximum
4163                file size is set to "infinite". */
4164             if (global_capture_opts.save_file == NULL) {
4165                 cmdarg_err("Ring buffer requested, but capture isn't being saved to a permanent file.");
4166                 global_capture_opts.multi_files_on = FALSE;
4167             }
4168             if (!global_capture_opts.has_autostop_filesize && !global_capture_opts.has_file_duration) {
4169                 cmdarg_err("Ring buffer requested, but no maximum capture file size or duration were specified.");
4170 #if 0
4171                 /* XXX - this must be redesigned as the conditions changed */
4172                 global_capture_opts.multi_files_on = FALSE;
4173 #endif
4174             }
4175         }
4176     }
4177
4178     /*
4179      * "-D" requires no interface to be selected; it's supposed to list
4180      * all interfaces.
4181      */
4182     if (list_interfaces) {
4183         /* Get the list of interfaces */
4184         GList       *if_list;
4185         int         err;
4186         gchar       *err_str;
4187
4188         if_list = capture_interface_list(&err, &err_str);
4189         if (if_list == NULL) {
4190             switch (err) {
4191             case CANT_GET_INTERFACE_LIST:
4192                 cmdarg_err("%s", err_str);
4193                 g_free(err_str);
4194                 exit_main(2);
4195                 break;
4196
4197             case NO_INTERFACES_FOUND:
4198                 /*
4199                  * If we're being run by another program, just give them
4200                  * an empty list of interfaces, don't report this as
4201                  * an error; that lets them decide whether to report
4202                  * this as an error or not.
4203                  */
4204                 if (!machine_readable) {
4205                     cmdarg_err("There are no interfaces on which a capture can be done");
4206                     exit_main(2);
4207                 }
4208                 break;
4209             }
4210         }
4211
4212         if (machine_readable)      /* tab-separated values to stdout */
4213             print_machine_readable_interfaces(if_list);
4214         else
4215             capture_opts_print_interfaces(if_list);
4216         free_interface_list(if_list);
4217         exit_main(0);
4218     }
4219
4220     /*
4221      * "-S" requires no interface to be selected; it gives statistics
4222      * for all interfaces.
4223      */
4224     if (print_statistics) {
4225         status = print_statistics_loop(machine_readable);
4226         exit_main(status);
4227     }
4228
4229     /*
4230      * "-L", "-d", and capturing act on a particular interface, so we have to
4231      * have an interface; if none was specified, pick a default.
4232      */
4233     if (capture_opts_trim_iface(&global_capture_opts, NULL) == FALSE) {
4234         /* cmdarg_err() already called .... */
4235         exit_main(1);
4236     }
4237
4238     /* Let the user know what interfaces were chosen. */
4239     /* get_interface_descriptive_name() is not available! */
4240     for (j = 0; j < global_capture_opts.ifaces->len; j++) {
4241         interface_options interface_opts;
4242
4243         interface_opts = g_array_index(global_capture_opts.ifaces, interface_options, j);
4244         g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "Interface: %s", interface_opts.name);
4245     }
4246
4247     if (list_link_layer_types) {
4248         /* Get the list of link-layer types for the capture device. */
4249         if_capabilities_t *caps;
4250         gchar *err_str;
4251         guint i;
4252
4253         for (i = 0; i < global_capture_opts.ifaces->len; i++) {
4254             interface_options interface_opts;
4255
4256             interface_opts = g_array_index(global_capture_opts.ifaces, interface_options, i);
4257             caps = get_if_capabilities(interface_opts.name,
4258                                        interface_opts.monitor_mode, &err_str);
4259             if (caps == NULL) {
4260                 cmdarg_err("The capabilities of the capture device \"%s\" could not be obtained (%s).\n"
4261                            "Please check to make sure you have sufficient permissions, and that\n"
4262                            "you have the proper interface or pipe specified.", interface_opts.name, err_str);
4263                 g_free(err_str);
4264                 exit_main(2);
4265             }
4266             if (caps->data_link_types == NULL) {
4267                 cmdarg_err("The capture device \"%s\" has no data link types.", interface_opts.name);
4268                 exit_main(2);
4269             }
4270             if (machine_readable)      /* tab-separated values to stdout */
4271                 /* XXX: We need to change the format and adopt consumers */
4272                 print_machine_readable_if_capabilities(caps);
4273             else
4274                 /* XXX: We might want to print also the interface name */
4275                 capture_opts_print_if_capabilities(caps, interface_opts.name,
4276                                                    interface_opts.monitor_mode);
4277             free_if_capabilities(caps);
4278         }
4279         exit_main(0);
4280     }
4281
4282     /* We're supposed to do a capture, or print the BPF code for a filter.
4283        Process the snapshot length, as that affects the generated BPF code. */
4284     capture_opts_trim_snaplen(&global_capture_opts, MIN_PACKET_SIZE);
4285
4286 #ifdef HAVE_BPF_IMAGE
4287     if (print_bpf_code) {
4288         show_filter_code(&global_capture_opts);
4289         exit_main(0);
4290     }
4291 #endif
4292
4293     /* We're supposed to do a capture.  Process the ring buffer arguments. */
4294     capture_opts_trim_ring_num_files(&global_capture_opts);
4295
4296     /* Now start the capture. */
4297
4298     if(capture_loop_start(&global_capture_opts, &stats_known, &stats) == TRUE) {
4299         /* capture ok */
4300         exit_main(0);
4301     } else {
4302         /* capture failed */
4303         exit_main(1);
4304     }
4305     return 0; /* never here, make compiler happy */
4306 }
4307
4308
4309 static void
4310 console_log_handler(const char *log_domain, GLogLevelFlags log_level,
4311                     const char *message, gpointer user_data _U_)
4312 {
4313     time_t curr;
4314     struct tm  *today;
4315     const char *level;
4316     gchar      *msg;
4317
4318     /* ignore log message, if log_level isn't interesting */
4319     if( !(log_level & G_LOG_LEVEL_MASK & ~(G_LOG_LEVEL_DEBUG|G_LOG_LEVEL_INFO))) {
4320 #if !defined(DEBUG_DUMPCAP) && !defined(DEBUG_CHILD_DUMPCAP)
4321         return;
4322 #endif
4323     }
4324
4325     /* create a "timestamp" */
4326     time(&curr);
4327     today = localtime(&curr);
4328
4329     switch(log_level & G_LOG_LEVEL_MASK) {
4330     case G_LOG_LEVEL_ERROR:
4331         level = "Err ";
4332         break;
4333     case G_LOG_LEVEL_CRITICAL:
4334         level = "Crit";
4335         break;
4336     case G_LOG_LEVEL_WARNING:
4337         level = "Warn";
4338         break;
4339     case G_LOG_LEVEL_MESSAGE:
4340         level = "Msg ";
4341         break;
4342     case G_LOG_LEVEL_INFO:
4343         level = "Info";
4344         break;
4345     case G_LOG_LEVEL_DEBUG:
4346         level = "Dbg ";
4347         break;
4348     default:
4349         fprintf(stderr, "unknown log_level %u\n", log_level);
4350         level = NULL;
4351         g_assert_not_reached();
4352     }
4353
4354     /* Generate the output message                                  */
4355     if(log_level & G_LOG_LEVEL_MESSAGE) {
4356         /* normal user messages without additional infos */
4357         msg =  g_strdup_printf("%s\n", message);
4358     } else {
4359         /* info/debug messages with additional infos */
4360         msg = g_strdup_printf("%02u:%02u:%02u %8s %s %s\n",
4361                               today->tm_hour, today->tm_min, today->tm_sec,
4362                               log_domain != NULL ? log_domain : "",
4363                               level, message);
4364     }
4365
4366     /* DEBUG & INFO msgs (if we're debugging today)                 */
4367 #if defined(DEBUG_DUMPCAP) || defined(DEBUG_CHILD_DUMPCAP)
4368     if( !(log_level & G_LOG_LEVEL_MASK & ~(G_LOG_LEVEL_DEBUG|G_LOG_LEVEL_INFO))) {
4369 #ifdef DEBUG_DUMPCAP
4370         fprintf(stderr, "%s", msg);
4371         fflush(stderr);
4372 #endif
4373 #ifdef DEBUG_CHILD_DUMPCAP
4374         fprintf(debug_log, "%s", msg);
4375         fflush(debug_log);
4376 #endif
4377         g_free(msg);
4378         return;
4379     }
4380 #endif
4381
4382     /* ERROR, CRITICAL, WARNING, MESSAGE messages goto stderr or    */
4383     /*  to parent especially formatted if dumpcap running as child. */
4384     if (capture_child) {
4385         sync_pipe_errmsg_to_parent(2, msg, "");
4386     } else {
4387         fprintf(stderr, "%s", msg);
4388         fflush(stderr);
4389     }
4390     g_free(msg);
4391 }
4392
4393
4394 /****************************************************************************************************************/
4395 /* indication report routines */
4396
4397
4398 static void
4399 report_packet_count(int packet_count)
4400 {
4401     char tmp[SP_DECISIZE+1+1];
4402     static int count = 0;
4403
4404     if(capture_child) {
4405         g_snprintf(tmp, sizeof(tmp), "%d", packet_count);
4406         g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "Packets: %s", tmp);
4407         pipe_write_block(2, SP_PACKET_COUNT, tmp);
4408     } else {
4409         count += packet_count;
4410         fprintf(stderr, "\rPackets: %u ", count);
4411         /* stderr could be line buffered */
4412         fflush(stderr);
4413     }
4414 }
4415
4416 static void
4417 report_new_capture_file(const char *filename)
4418 {
4419     if(capture_child) {
4420         g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "File: %s", filename);
4421         pipe_write_block(2, SP_FILE, filename);
4422     } else {
4423 #ifdef SIGINFO
4424         /*
4425          * Prevent a SIGINFO handler from writing to the standard error
4426          * while we're doing so; instead, have it just set a flag telling
4427          * us to print that information when we're done.
4428          */
4429         infodelay = TRUE;
4430 #endif /* SIGINFO */
4431         fprintf(stderr, "File: %s\n", filename);
4432         /* stderr could be line buffered */
4433         fflush(stderr);
4434
4435 #ifdef SIGINFO
4436         /*
4437          * Allow SIGINFO handlers to write.
4438          */
4439         infodelay = FALSE;
4440
4441         /*
4442          * If a SIGINFO handler asked us to write out capture counts, do so.
4443          */
4444         if (infoprint)
4445           report_counts_for_siginfo();
4446 #endif /* SIGINFO */
4447     }
4448 }
4449
4450 static void
4451 report_cfilter_error(capture_options *capture_opts, guint i, const char *errmsg)
4452 {
4453     interface_options interface_opts;
4454     char tmp[MSG_MAX_LENGTH+1+6];
4455
4456     if (i < capture_opts->ifaces->len) {
4457         if (capture_child) {
4458             g_snprintf(tmp, sizeof(tmp), "%u:%s", i, errmsg);
4459             g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "Capture filter error: %s", errmsg);
4460             pipe_write_block(2, SP_BAD_FILTER, tmp);
4461         } else {
4462             /*
4463              * clopts_step_invalid_capfilter in test/suite-clopts.sh MUST match
4464              * the error message below.
4465              */
4466             interface_opts = g_array_index(capture_opts->ifaces, interface_options, i);
4467             fprintf(stderr,
4468               "Invalid capture filter \"%s\" for interface %s!\n"
4469               "\n"
4470               "That string isn't a valid capture filter (%s).\n"
4471               "See the User's Guide for a description of the capture filter syntax.\n",
4472               interface_opts.cfilter, interface_opts.name, errmsg);
4473         }
4474     }
4475 }
4476
4477 static void
4478 report_capture_error(const char *error_msg, const char *secondary_error_msg)
4479 {
4480     if(capture_child) {
4481         g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG,
4482             "Primary Error: %s", error_msg);
4483         g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG,
4484             "Secondary Error: %s", secondary_error_msg);
4485         sync_pipe_errmsg_to_parent(2, error_msg, secondary_error_msg);
4486     } else {
4487         fprintf(stderr, "%s\n", error_msg);
4488         if (secondary_error_msg[0] != '\0')
4489           fprintf(stderr, "%s\n", secondary_error_msg);
4490     }
4491 }
4492
4493 static void
4494 report_packet_drops(guint32 received, guint32 drops, gchar *name)
4495 {
4496     char tmp[SP_DECISIZE+1+1];
4497
4498     g_snprintf(tmp, sizeof(tmp), "%u", drops);
4499
4500     if(capture_child) {
4501         g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG,
4502             "Packets received/dropped on interface %s: %u/%u",
4503             name, received, drops);
4504         /* XXX: Need to provide interface id, changes to consumers required. */
4505         pipe_write_block(2, SP_DROPS, tmp);
4506     } else {
4507         fprintf(stderr,
4508             "Packets received/dropped on interface %s: %u/%u (%.1f%%)\n",
4509             name, received, drops,
4510             received ? 100.0 * received / (received + drops) : 0.0);
4511         /* stderr could be line buffered */
4512         fflush(stderr);
4513     }
4514 }
4515
4516
4517 /****************************************************************************************************************/
4518 /* signal_pipe handling */
4519
4520
4521 #ifdef _WIN32
4522 static gboolean
4523 signal_pipe_check_running(void)
4524 {
4525     /* any news from our parent? -> just stop the capture */
4526     DWORD avail = 0;
4527     gboolean result;
4528
4529     /* if we are running standalone, no check required */
4530     if(!capture_child) {
4531         return TRUE;
4532     }
4533
4534     if(!sig_pipe_name || !sig_pipe_handle) {
4535         /* This shouldn't happen */
4536         g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO,
4537             "Signal pipe: No name or handle");
4538         return FALSE;
4539     }
4540
4541     /*
4542      * XXX - We should have the process ID of the parent (from the "-Z" flag)
4543      * at this point.  Should we check to see if the parent is still alive,
4544      * e.g. by using OpenProcess?
4545      */
4546
4547     result = PeekNamedPipe(sig_pipe_handle, NULL, 0, NULL, &avail, NULL);
4548
4549     if(!result || avail > 0) {
4550         /* peek failed or some bytes really available */
4551         /* (if not piping from stdin this would fail) */
4552         g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO,
4553             "Signal pipe: Stop capture: %s", sig_pipe_name);
4554         g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG,
4555             "Signal pipe: %s (%p) result: %u avail: %u", sig_pipe_name,
4556             sig_pipe_handle, result, avail);
4557         return FALSE;
4558     } else {
4559         /* pipe ok and no bytes available */
4560         return TRUE;
4561     }
4562 }
4563 #endif
4564
4565 /*
4566  * Editor modelines  -  http://www.wireshark.org/tools/modelines.html
4567  *
4568  * Local variables:
4569  * c-basic-offset: 4
4570  * tab-width: 8
4571  * indent-tabs-mode: nil
4572  * End:
4573  *
4574  * vi: set shiftwidth=4 tabstop=8 expandtab
4575  * :indentSize=4:tabSize=8:noTabs=true:
4576  */