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