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