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