capture_loop.c

   1 /* capture_loop.c
   2  * The actual capturing loop, getting packets and storing it
   3  *
   4  * $Id$
   5  *
   6  * Wireshark - Network traffic analyzer
   7  * By Gerald Combs <gerald@wireshark.org>
   8  * Copyright 1998 Gerald Combs
   9  *
  10  * This program is free software; you can redistribute it and/or
  11  * modify it under the terms of the GNU General Public License
  12  * as published by the Free Software Foundation; either version 2
  13  * of the License, or (at your option) any later version.
  14  *
  15  * This program is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  * GNU General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with this program; if not, write to the Free Software
  22  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  23  */
  24
  25
  26 /** @file
  27  *
  28  * Capture loop (internal interface).
  29  *
  30  * It will open the input and output files, capture the packets,
  31  * change ringbuffer output files while capturing and close all files again.
  32  *
  33  * The input file can be a network interface or capture pipe (unix only).
  34  * The output file can be a single or a ringbuffer file handled by wiretap.
  35  *
  36  */
  37
  38 #ifdef HAVE_CONFIG_H
  39 # include "config.h"
  40 #endif
  41
  42 #ifdef HAVE_LIBPCAP
  43
  44 #include <string.h>
  45
  46 #ifdef HAVE_FCNTL_H
  47 #include <fcntl.h>
  48 #endif
  49
  50 #ifdef HAVE_UNISTD_H
  51 #include <unistd.h>
  52 #endif
  53
  54 #ifdef HAVE_SYS_TYPES_H
  55 # include <sys/types.h>
  56 #endif
  57
  58 #ifdef HAVE_SYS_STAT_H
  59 # include <sys/stat.h>
  60 #endif
  61
  62 #include <signal.h>
  63 #include <errno.h>
  64 #include <setjmp.h>
  65
  66
  67 #include <glib.h>
  68
  69 #include <pcap.h>
  70
  71 #include "pcapio.h"
  72
  73 #include "capture-pcap-util.h"
  74
  75 #include "capture.h"
  76 #include "capture_sync.h"
  77
  78 #include "conditions.h"
  79 #include "capture_stop_conditions.h"
  80 #include "ringbuffer.h"
  81
  82 #include "simple_dialog.h"
  83 #include "tempfile.h"
  84 #include "log.h"
  85 #include "file_util.h"
  86
  87 #include "epan/unicode-utils.h"
  88
  89 #include "capture_loop.h"
  90
  91 /*
  92  * Standard secondary message for unexpected errors.
  93  */
  94 static const char please_report[] =
  95     "Please report this to the Wireshark developers.  (This is not a crash;\n"
  96     "please do not report it as such.)";
  97
  98 /*
  99  * This needs to be static, so that the SIGUSR1 handler can clear the "go"
 100  * flag.
 101  */
 102 static loop_data   ld;
 103
 104
 105 /*
 106  * Timeout, in milliseconds, for reads from the stream of captured packets.
 107  */
 108 #define CAP_READ_TIMEOUT        250
 109 static char *cap_pipe_err_str;
 110
 111 static void capture_loop_packet_cb(u_char *user, const struct pcap_pkthdr *phdr,
 112   const u_char *pd);
 113 static void capture_loop_get_errmsg(char *errmsg, int errmsglen, const char *fname,
 114                           int err, gboolean is_close);
 115
 116
 117
 118 /* Take care of byte order in the libpcap headers read from pipes.
 119  * (function taken from wiretap/libpcap.c) */
 120 static void
 121 cap_pipe_adjust_header(gboolean byte_swapped, struct pcap_hdr *hdr, struct pcaprec_hdr *rechdr)
 122 {
 123   if (byte_swapped) {
 124     /* Byte-swap the record header fields. */
 125     rechdr->ts_sec = BSWAP32(rechdr->ts_sec);
 126     rechdr->ts_usec = BSWAP32(rechdr->ts_usec);
 127     rechdr->incl_len = BSWAP32(rechdr->incl_len);
 128     rechdr->orig_len = BSWAP32(rechdr->orig_len);
 129   }
 130
 131   /* In file format version 2.3, the "incl_len" and "orig_len" fields were
 132      swapped, in order to match the BPF header layout.
 133
 134      Unfortunately, some files were, according to a comment in the "libpcap"
 135      source, written with version 2.3 in their headers but without the
 136      interchanged fields, so if "incl_len" is greater than "orig_len" - which
 137      would make no sense - we assume that we need to swap them.  */
 138   if (hdr->version_major == 2 &&
 139       (hdr->version_minor < 3 ||
 140        (hdr->version_minor == 3 && rechdr->incl_len > rechdr->orig_len))) {
 141     guint32 temp;
 142
 143     temp = rechdr->orig_len;
 144     rechdr->orig_len = rechdr->incl_len;
 145     rechdr->incl_len = temp;
 146   }
 147 }
 148
 149 /* Provide select() functionality for a single file descriptor
 150  * on both UNIX/POSIX and Windows.
 151  *
 152  * The Windows version calls WaitForSingleObject instead of
 153  * select().
 154  *
 155  * Returns the same values as select.  If an error is returned,
 156  * the string cap_pipe_err_str should be used instead of errno.
 157  */
 158 static int
 159 cap_pipe_select(int pipe_fd, gboolean wait_forever) {
 160 #ifndef _WIN32
 161   fd_set      rfds;
 162   struct timeval timeout, *pto;
 163   int sel_ret;
 164
 165   cap_pipe_err_str = "Unknown error";
 166
 167   FD_ZERO(&rfds);
 168   FD_SET(pipe_fd, &rfds);
 169   if (wait_forever) {
 170     pto = NULL;
 171   } else {
 172     timeout.tv_sec = 0;
 173     timeout.tv_usec = CAP_READ_TIMEOUT * 1000;
 174     pto = &timeout;
 175   }
 176   sel_ret = select(pipe_fd+1, &rfds, NULL, NULL, pto);
 177   if (sel_ret < 0)
 178     cap_pipe_err_str = strerror(errno);
 179   return sel_ret;
 180 }
 181 #else
 182   /* XXX - Should we just use file handles exclusively under Windows?
 183    * Otherwise we have to convert between file handles and file descriptors
 184    * here and when we open a named pipe.
 185    */
 186   HANDLE hPipe = (HANDLE) _get_osfhandle(pipe_fd);
 187   wchar_t *err_str;
 188   DWORD timeout = wait_forever ? INFINITE : CAP_READ_TIMEOUT * 1000;
 189   DWORD wait_ret;
 190
 191   if (hPipe == INVALID_HANDLE_VALUE) {
 192     cap_pipe_err_str = "Could not open standard input";
 193     return -1;
 194   }
 195
 196   cap_pipe_err_str = "Unknown error";
 197
 198   wait_ret = WaitForSingleObject(hPipe, timeout);
 199   switch (wait_ret) {
 200     /* XXX - This probably isn't correct */
 201     case WAIT_ABANDONED:
 202       errno = EINTR;
 203       return -1;
 204     case WAIT_OBJECT_0:
 205       return 1;
 206     case WAIT_TIMEOUT:
 207       return 0;
 208     case WAIT_FAILED:
 209       FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_ALLOCATE_BUFFER,
 210         NULL, GetLastError(), 0, (LPTSTR) &err_str, 0, NULL);
 211       cap_pipe_err_str = utf_16to8(err_str);
 212       LocalFree(err_str);
 213       return -1;
 214     default:
 215       g_assert_not_reached();
 216       return -1;
 217   }
 218 }
 219 #endif
 220
 221
 222 /* Mimic pcap_open_live() for pipe captures
 223  * We check if "pipename" is "-" (stdin) or a FIFO, open it, and read the
 224  * header.
 225  * N.B. : we can't read the libpcap formats used in RedHat 6.1 or SuSE 6.3
 226  * because we can't seek on pipes (see wiretap/libpcap.c for details) */
 227 static int
 228 cap_pipe_open_live(char *pipename, struct pcap_hdr *hdr, loop_data *ld,
 229                  char *errmsg, int errmsgl)
 230 {
 231 #ifndef _WIN32
 232   struct stat pipe_stat;
 233 #else
 234 #if 0
 235   char *pncopy, *pos;
 236   wchar_t *err_str;
 237 #endif
 238   HANDLE hPipe = NULL;
 239 #endif
 240   int          sel_ret;
 241   int          fd;
 242   int          b;
 243   guint32       magic;
 244   unsigned int bytes_read;
 245
 246   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "cap_pipe_open_live: %s", pipename);
 247
 248   /*
 249    * XXX (T)Wireshark blocks until we return
 250    */
 251   if (strcmp(pipename, "-") == 0) {
 252     fd = 0; /* read from stdin */
 253 #ifdef _WIN32
 254     /*
 255      * This is needed to set the stdin pipe into binary mode, otherwise
 256      * CR/LF are mangled...
 257      */
 258     _setmode(0, _O_BINARY);
 259 #endif  /* _WIN32 */
 260   } else {
 261 #ifndef _WIN32
 262     if (eth_stat(pipename, &pipe_stat) < 0) {
 263       if (errno == ENOENT || errno == ENOTDIR)
 264         ld->cap_pipe_err = PIPNEXIST;
 265       else {
 266         g_snprintf(errmsg, errmsgl,
 267           "The capture session could not be initiated "
 268           "due to error on pipe: %s", strerror(errno));
 269         ld->cap_pipe_err = PIPERR;
 270       }
 271       return -1;
 272     }
 273     if (! S_ISFIFO(pipe_stat.st_mode)) {
 274       if (S_ISCHR(pipe_stat.st_mode)) {
 275         /*
 276          * Assume the user specified an interface on a system where
 277          * interfaces are in /dev.  Pretend we haven't seen it.
 278          */
 279          ld->cap_pipe_err = PIPNEXIST;
 280       } else
 281       {
 282         g_snprintf(errmsg, errmsgl,
 283             "The capture session could not be initiated because\n"
 284             "\"%s\" is neither an interface nor a pipe", pipename);
 285         ld->cap_pipe_err = PIPERR;
 286       }
 287       return -1;
 288     }
 289     fd = eth_open(pipename, O_RDONLY | O_NONBLOCK, 0000 /* no creation so don't matter */);
 290     if (fd == -1) {
 291       g_snprintf(errmsg, errmsgl,
 292           "The capture session could not be initiated "
 293           "due to error on pipe open: %s", strerror(errno));
 294       ld->cap_pipe_err = PIPERR;
 295       return -1;
 296     }
 297 #else /* _WIN32 */
 298 #if 0 /* Enable/disable Windows named pipes */
 299 #define PIPE_STR "\\pipe\\"
 300     /* Under Windows, named pipes _must_ have the form
 301      * "\\<server>\pipe\<pipename>".  <server> may be "." for localhost.
 302      */
 303     pncopy = g_strdup(pipename);
 304     if (strstr(pncopy, "\\\\") == pncopy) {
 305       pos = strchr(pncopy + 3, '\\');
 306       if (pos && g_strncasecmp(pos, PIPE_STR, strlen(PIPE_STR)) != 0)
 307         pos = NULL;
 308     }
 309
 310     g_free(pncopy);
 311
 312     if (!pos) {
 313       g_snprintf(errmsg, errmsgl,
 314           "The capture session could not be initiated because\n"
 315           "\"%s\" is neither an interface nor a pipe", pipename);
 316       ld->cap_pipe_err = PIPNEXIST;
 317       return -1;
 318     }
 319
 320     /* Wait for the pipe to appear */
 321     while (1) {
 322       hPipe = CreateFile(utf_8to16(pipename), GENERIC_READ, 0, NULL,
 323           OPEN_EXISTING, 0, NULL);
 324
 325       if (hPipe != INVALID_HANDLE_VALUE)
 326         break;
 327
 328       if (GetLastError() != ERROR_PIPE_BUSY) {
 329         FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_ALLOCATE_BUFFER,
 330           NULL, GetLastError(), 0, (LPTSTR) &err_str, 0, NULL);
 331         g_snprintf(errmsg, errmsgl,
 332             "The capture session on \"%s\" could not be initiated "
 333             "due to error on pipe open: pipe busy: %s (error %d)",
 334             pipename, utf_16to8(err_str), GetLastError());
 335         LocalFree(err_str);
 336         ld->cap_pipe_err = PIPERR;
 337         return -1;
 338       }
 339
 340       if (!WaitNamedPipe(utf_8to16(pipename), 30 * 1000)) {
 341         FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_ALLOCATE_BUFFER,
 342           NULL, GetLastError(), 0, (LPTSTR) &err_str, 0, NULL);
 343         g_snprintf(errmsg, errmsgl,
 344             "The capture session could not be initiated "
 345             "due to error on pipe open: %s (error %d)",
 346             utf_16to8(err_str), GetLastError());
 347         LocalFree(err_str);
 348         ld->cap_pipe_err = PIPERR;
 349         return -1;
 350       }
 351     }
 352
 353     fd = _open_osfhandle((long) hPipe, _O_RDONLY);
 354     if (fd == -1) {
 355       g_snprintf(errmsg, errmsgl,
 356           "The capture session could not be initiated "
 357           "due to error on pipe open: %s", strerror(errno));
 358       ld->cap_pipe_err = PIPERR;
 359       return -1;
 360     }
 361 #else /* Enable/disable Windows named pipes */
 362     /* On Windows, we don't support capturing on pipes, so we give up. */
 363
 364     g_snprintf(errmsg, errmsgl,
 365 "The capture session could not be initiated.  Unable to open interface.");
 366     return -1;
 367 #endif /* Enable/disable Windows named pipes */
 368 #endif /* _WIN32 */
 369   }
 370
 371   ld->from_cap_pipe = TRUE;
 372
 373   /* read the pcap header */
 374   bytes_read = 0;
 375   while (bytes_read < sizeof magic) {
 376     sel_ret = cap_pipe_select(fd, FALSE);
 377     if (sel_ret < 0) {
 378       g_snprintf(errmsg, errmsgl,
 379         "Unexpected error from select: %s", strerror(errno));
 380       goto error;
 381     } else if (sel_ret > 0) {
 382       b = read(fd, ((char *)&magic)+bytes_read, sizeof magic-bytes_read);
 383       if (b <= 0) {
 384         if (b == 0)
 385           g_snprintf(errmsg, errmsgl, "End of file on pipe during open");
 386         else
 387           g_snprintf(errmsg, errmsgl, "Error on pipe during open: %s",
 388             strerror(errno));
 389         goto error;
 390       }
 391       bytes_read += b;
 392     }
 393   }
 394
 395   switch (magic) {
 396   case PCAP_MAGIC:
 397     /* Host that wrote it has our byte order, and was running
 398        a program using either standard or ss990417 libpcap. */
 399     ld->cap_pipe_byte_swapped = FALSE;
 400     ld->cap_pipe_modified = FALSE;
 401     break;
 402   case PCAP_MODIFIED_MAGIC:
 403     /* Host that wrote it has our byte order, but was running
 404        a program using either ss990915 or ss991029 libpcap. */
 405     ld->cap_pipe_byte_swapped = FALSE;
 406     ld->cap_pipe_modified = TRUE;
 407     break;
 408   case PCAP_SWAPPED_MAGIC:
 409     /* Host that wrote it has a byte order opposite to ours,
 410        and was running a program using either standard or
 411        ss990417 libpcap. */
 412     ld->cap_pipe_byte_swapped = TRUE;
 413     ld->cap_pipe_modified = FALSE;
 414     break;
 415   case PCAP_SWAPPED_MODIFIED_MAGIC:
 416     /* Host that wrote it out has a byte order opposite to
 417        ours, and was running a program using either ss990915
 418        or ss991029 libpcap. */
 419     ld->cap_pipe_byte_swapped = TRUE;
 420     ld->cap_pipe_modified = TRUE;
 421     break;
 422   default:
 423     /* Not a "libpcap" type we know about. */
 424     g_snprintf(errmsg, errmsgl, "Unrecognized libpcap format");
 425     goto error;
 426   }
 427
 428   /* Read the rest of the header */
 429   bytes_read = 0;
 430   while (bytes_read < sizeof(struct pcap_hdr)) {
 431     sel_ret = cap_pipe_select(fd, FALSE);
 432     if (sel_ret < 0) {
 433       g_snprintf(errmsg, errmsgl,
 434         "Unexpected error from select: %s", strerror(errno));
 435       goto error;
 436     } else if (sel_ret > 0) {
 437       b = read(fd, ((char *)hdr)+bytes_read,
 438             sizeof(struct pcap_hdr) - bytes_read);
 439       if (b <= 0) {
 440         if (b == 0)
 441           g_snprintf(errmsg, errmsgl, "End of file on pipe during open");
 442         else
 443           g_snprintf(errmsg, errmsgl, "Error on pipe during open: %s",
 444             strerror(errno));
 445         goto error;
 446       }
 447       bytes_read += b;
 448     }
 449   }
 450
 451   if (ld->cap_pipe_byte_swapped) {
 452     /* Byte-swap the header fields about which we care. */
 453     hdr->version_major = BSWAP16(hdr->version_major);
 454     hdr->version_minor = BSWAP16(hdr->version_minor);
 455     hdr->snaplen = BSWAP32(hdr->snaplen);
 456     hdr->network = BSWAP32(hdr->network);
 457   }
 458   ld->linktype = hdr->network;
 459
 460   if (hdr->version_major < 2) {
 461     g_snprintf(errmsg, errmsgl, "Unable to read old libpcap format");
 462     goto error;
 463   }
 464
 465   ld->cap_pipe_state = STATE_EXPECT_REC_HDR;
 466   ld->cap_pipe_err = PIPOK;
 467   return fd;
 468
 469 error:
 470   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "cap_pipe_open_live: error %s", errmsg);
 471   ld->cap_pipe_err = PIPERR;
 472   eth_close(fd);
 473   return -1;
 474
 475 }
 476
 477
 478 /* We read one record from the pipe, take care of byte order in the record
 479  * header, write the record to the capture file, and update capture statistics. */
 480 static int
 481 cap_pipe_dispatch(loop_data *ld, guchar *data, char *errmsg, int errmsgl)
 482 {
 483   struct pcap_pkthdr phdr;
 484   int b;
 485   enum { PD_REC_HDR_READ, PD_DATA_READ, PD_PIPE_EOF, PD_PIPE_ERR,
 486           PD_ERR } result;
 487
 488
 489 #ifdef LOG_CAPTURE_VERBOSE
 490   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "cap_pipe_dispatch");
 491 #endif
 492
 493   switch (ld->cap_pipe_state) {
 494
 495   case STATE_EXPECT_REC_HDR:
 496     ld->cap_pipe_bytes_to_read = ld->cap_pipe_modified ?
 497       sizeof(struct pcaprec_modified_hdr) : sizeof(struct pcaprec_hdr);
 498     ld->cap_pipe_bytes_read = 0;
 499     ld->cap_pipe_state = STATE_READ_REC_HDR;
 500     /* Fall through */
 501
 502   case STATE_READ_REC_HDR:
 503     b = read(ld->cap_pipe_fd, ((char *)&ld->cap_pipe_rechdr)+ld->cap_pipe_bytes_read,
 504              ld->cap_pipe_bytes_to_read - ld->cap_pipe_bytes_read);
 505     if (b <= 0) {
 506       if (b == 0)
 507         result = PD_PIPE_EOF;
 508       else
 509         result = PD_PIPE_ERR;
 510       break;
 511     }
 512     if ((ld->cap_pipe_bytes_read += b) < ld->cap_pipe_bytes_to_read)
 513         return 0;
 514     result = PD_REC_HDR_READ;
 515     break;
 516
 517   case STATE_EXPECT_DATA:
 518     ld->cap_pipe_bytes_read = 0;
 519     ld->cap_pipe_state = STATE_READ_DATA;
 520     /* Fall through */
 521
 522   case STATE_READ_DATA:
 523     b = read(ld->cap_pipe_fd, data+ld->cap_pipe_bytes_read,
 524              ld->cap_pipe_rechdr.hdr.incl_len - ld->cap_pipe_bytes_read);
 525     if (b <= 0) {
 526       if (b == 0)
 527         result = PD_PIPE_EOF;
 528       else
 529         result = PD_PIPE_ERR;
 530       break;
 531     }
 532     if ((ld->cap_pipe_bytes_read += b) < ld->cap_pipe_rechdr.hdr.incl_len)
 533       return 0;
 534     result = PD_DATA_READ;
 535     break;
 536
 537   default:
 538     g_snprintf(errmsg, errmsgl, "cap_pipe_dispatch: invalid state");
 539     result = PD_ERR;
 540
 541   } /* switch (ld->cap_pipe_state) */
 542
 543   /*
 544    * We've now read as much data as we were expecting, so process it.
 545    */
 546   switch (result) {
 547
 548   case PD_REC_HDR_READ:
 549     /* We've read the header. Take care of byte order. */
 550     cap_pipe_adjust_header(ld->cap_pipe_byte_swapped, &ld->cap_pipe_hdr,
 551                            &ld->cap_pipe_rechdr.hdr);
 552     if (ld->cap_pipe_rechdr.hdr.incl_len > WTAP_MAX_PACKET_SIZE) {
 553       g_snprintf(errmsg, errmsgl, "Frame %u too long (%d bytes)",
 554         ld->packet_count+1, ld->cap_pipe_rechdr.hdr.incl_len);
 555       break;
 556     }
 557     ld->cap_pipe_state = STATE_EXPECT_DATA;
 558     return 0;
 559
 560   case PD_DATA_READ:
 561     /* Fill in a "struct pcap_pkthdr", and process the packet. */
 562     phdr.ts.tv_sec = ld->cap_pipe_rechdr.hdr.ts_sec;
 563     phdr.ts.tv_usec = ld->cap_pipe_rechdr.hdr.ts_usec;
 564     phdr.caplen = ld->cap_pipe_rechdr.hdr.incl_len;
 565     phdr.len = ld->cap_pipe_rechdr.hdr.orig_len;
 566
 567     ld->packet_cb((u_char *)ld, &phdr, data);
 568
 569     ld->cap_pipe_state = STATE_EXPECT_REC_HDR;
 570     return 1;
 571
 572   case PD_PIPE_EOF:
 573     ld->cap_pipe_err = PIPEOF;
 574     return -1;
 575
 576   case PD_PIPE_ERR:
 577     g_snprintf(errmsg, errmsgl, "Error reading from pipe: %s",
 578       strerror(errno));
 579     /* Fall through */
 580   case PD_ERR:
 581     break;
 582   }
 583
 584   ld->cap_pipe_err = PIPERR;
 585   /* Return here rather than inside the switch to prevent GCC warning */
 586   return -1;
 587 }
 588
 589
 590 /* open the capture input file (pcap or capture pipe) */
 591 gboolean
 592 capture_loop_open_input(capture_options *capture_opts, loop_data *ld,
 593                         char *errmsg, size_t errmsg_len,
 594                         char *secondary_errmsg, size_t secondary_errmsg_len)
 595 {
 596   gchar       open_err_str[PCAP_ERRBUF_SIZE];
 597   gchar      *sync_msg_str;
 598   static const char ppamsg[] = "can't find PPA for ";
 599   const char *set_linktype_err_str;
 600   const char  *libpcap_warn;
 601 #ifdef _WIN32
 602   gchar      *sync_secondary_msg_str;
 603   int         err;
 604   WORD        wVersionRequested;
 605   WSADATA     wsaData;
 606 #endif
 607
 608
 609   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_open_input : %s", capture_opts->iface);
 610
 611
 612 /* XXX - opening Winsock on tshark? */
 613
 614   /* Initialize Windows Socket if we are in a WIN32 OS
 615      This needs to be done before querying the interface for network/netmask */
 616 #ifdef _WIN32
 617   /* XXX - do we really require 1.1 or earlier?
 618      Are there any versions that support only 2.0 or higher? */
 619   wVersionRequested = MAKEWORD(1, 1);
 620   err = WSAStartup(wVersionRequested, &wsaData);
 621   if (err != 0) {
 622     switch (err) {
 623
 624     case WSASYSNOTREADY:
 625       g_snprintf(errmsg, errmsg_len,
 626         "Couldn't initialize Windows Sockets: Network system not ready for network communication");
 627       break;
 628
 629     case WSAVERNOTSUPPORTED:
 630       g_snprintf(errmsg, errmsg_len,
 631         "Couldn't initialize Windows Sockets: Windows Sockets version %u.%u not supported",
 632         LOBYTE(wVersionRequested), HIBYTE(wVersionRequested));
 633       break;
 634
 635     case WSAEINPROGRESS:
 636       g_snprintf(errmsg, errmsg_len,
 637         "Couldn't initialize Windows Sockets: Blocking operation is in progress");
 638       break;
 639
 640     case WSAEPROCLIM:
 641       g_snprintf(errmsg, errmsg_len,
 642         "Couldn't initialize Windows Sockets: Limit on the number of tasks supported by this WinSock implementation has been reached");
 643       break;
 644
 645     case WSAEFAULT:
 646       g_snprintf(errmsg, errmsg_len,
 647         "Couldn't initialize Windows Sockets: Bad pointer passed to WSAStartup");
 648       break;
 649
 650     default:
 651       g_snprintf(errmsg, errmsg_len,
 652         "Couldn't initialize Windows Sockets: error %d", err);
 653       break;
 654     }
 655     g_snprintf(secondary_errmsg, secondary_errmsg_len, please_report);
 656     return FALSE;
 657   }
 658 #endif
 659
 660   /* Open the network interface to capture from it.
 661      Some versions of libpcap may put warnings into the error buffer
 662      if they succeed; to tell if that's happened, we have to clear
 663      the error buffer, and check if it's still a null string.  */
 664   open_err_str[0] = '\0';
 665   ld->pcap_h = pcap_open_live(capture_opts->iface,
 666                        capture_opts->has_snaplen ? capture_opts->snaplen :
 667                                                   WTAP_MAX_PACKET_SIZE,
 668                        capture_opts->promisc_mode, CAP_READ_TIMEOUT,
 669                        open_err_str);
 670
 671   if (ld->pcap_h != NULL) {
 672     /* we've opened "iface" as a network device */
 673 #ifdef _WIN32
 674     /* try to set the capture buffer size */
 675     if (pcap_setbuff(ld->pcap_h, capture_opts->buffer_size * 1024 * 1024) != 0) {
 676         sync_secondary_msg_str = g_strdup_printf(
 677           "The capture buffer size of %luMB seems to be too high for your machine,\n"
 678           "the default of 1MB will be used.\n"
 679           "\n"
 680           "Nonetheless, the capture is started.\n",
 681           capture_opts->buffer_size);
 682         report_capture_error("Couldn't set the capture buffer size!",
 683                                    sync_secondary_msg_str);
 684         g_free(sync_secondary_msg_str);
 685     }
 686 #endif
 687
 688     /* setting the data link type only works on real interfaces */
 689     if (capture_opts->linktype != -1) {
 690       set_linktype_err_str = set_pcap_linktype(ld->pcap_h, capture_opts->iface,
 691         capture_opts->linktype);
 692       if (set_linktype_err_str != NULL) {
 693         g_snprintf(errmsg, errmsg_len, "Unable to set data link type (%s).",
 694                    set_linktype_err_str);
 695         g_snprintf(secondary_errmsg, secondary_errmsg_len, please_report);
 696         return FALSE;
 697       }
 698     }
 699     ld->linktype = get_pcap_linktype(ld->pcap_h, capture_opts->iface);
 700   } else {
 701     /* We couldn't open "iface" as a network device. */
 702     /* Try to open it as a pipe */
 703     ld->cap_pipe_fd = cap_pipe_open_live(capture_opts->iface, &ld->cap_pipe_hdr, ld, errmsg, errmsg_len);
 704
 705     if (ld->cap_pipe_fd == -1) {
 706
 707       if (ld->cap_pipe_err == PIPNEXIST) {
 708         /* Pipe doesn't exist, so output message for interface */
 709
 710         /* If we got a "can't find PPA for X" message, warn the user (who
 711            is running (T)Wireshark on HP-UX) that they don't have a version
 712            of libpcap that properly handles HP-UX (libpcap 0.6.x and later
 713            versions, which properly handle HP-UX, say "can't find /dev/dlpi
 714            PPA for X" rather than "can't find PPA for X"). */
 715         if (strncmp(open_err_str, ppamsg, sizeof ppamsg - 1) == 0)
 716           libpcap_warn =
 717             "\n\n"
 718             "You are running (T)Wireshark with a version of the libpcap library\n"
 719             "that doesn't handle HP-UX network devices well; this means that\n"
 720             "(T)Wireshark may not be able to capture packets.\n"
 721             "\n"
 722             "To fix this, you should install libpcap 0.6.2, or a later version\n"
 723             "of libpcap, rather than libpcap 0.4 or 0.5.x.  It is available in\n"
 724             "packaged binary form from the Software Porting And Archive Centre\n"
 725             "for HP-UX; the Centre is at http://hpux.connect.org.uk/ - the page\n"
 726             "at the URL lists a number of mirror sites.";
 727         else
 728           libpcap_warn = "";
 729         g_snprintf(errmsg, errmsg_len,
 730           "The capture session could not be initiated (%s).", open_err_str);
 731 #ifndef _WIN32
 732         g_snprintf(secondary_errmsg, secondary_errmsg_len,
 733 "Please check to make sure you have sufficient permissions, and that you have\n"
 734 "the proper interface or pipe specified.%s", libpcap_warn);
 735 #else
 736     g_snprintf(secondary_errmsg, secondary_errmsg_len,
 737 "\n"
 738 "Please check that \"%s\" is the proper interface.\n"
 739 "\n"
 740 "\n"
 741 "Help can be found at:\n"
 742 "\n"
 743 "       http://wiki.wireshark.org/CaptureSetup\n"
 744 "\n"
 745 "64-bit Windows:\n"
 746 "WinPcap does not support 64-bit Windows; you will have to use some other\n"
 747 "tool to capture traffic, such as netcap.\n"
 748 "For netcap details see: http://support.microsoft.com/?id=310875\n"
 749 "\n"
 750 "Modem (PPP/WAN):\n"
 751 "Note that version 3.0 of WinPcap, and earlier versions of WinPcap, don't\n"
 752 "support capturing on PPP/WAN interfaces on Windows NT 4.0 / 2000 / XP /\n"
 753 "Server 2003.\n"
 754 "WinPcap 3.1 has support for it on Windows 2000 / XP / Server 2003, but has no\n"
 755 "support for it on Windows NT 4.0 or Windows Vista (Beta 1).",
 756     capture_opts->iface);
 757 #endif /* _WIN32 */
 758       }
 759       /*
 760        * Else pipe (or file) does exist and cap_pipe_open_live() has
 761        * filled in errmsg
 762        */
 763       return FALSE;
 764     } else
 765       /* cap_pipe_open_live() succeeded; don't want
 766          error message from pcap_open_live() */
 767       open_err_str[0] = '\0';
 768   }
 769
 770 /* XXX - will this work for tshark? */
 771 #ifdef MUST_DO_SELECT
 772   if (!ld->from_cap_pipe) {
 773 #ifdef HAVE_PCAP_GET_SELECTABLE_FD
 774     ld->pcap_fd = pcap_get_selectable_fd(ld->pcap_h);
 775 #else
 776     ld->pcap_fd = pcap_fileno(ld->pcap_h);
 777 #endif
 778   }
 779 #endif
 780
 781   /* Does "open_err_str" contain a non-empty string?  If so, "pcap_open_live()"
 782      returned a warning; print it, but keep capturing. */
 783   if (open_err_str[0] != '\0') {
 784     sync_msg_str = g_strdup_printf("%s.", open_err_str);
 785     report_capture_error(sync_msg_str, "");
 786     g_free(sync_msg_str);
 787   }
 788
 789   return TRUE;
 790 }
 791
 792
 793 /* close the capture input file (pcap or capture pipe) */
 794 static void capture_loop_close_input(loop_data *ld) {
 795
 796   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_close_input");
 797
 798   /* if open, close the capture pipe "input file" */
 799   if (ld->cap_pipe_fd >= 0) {
 800     g_assert(ld->from_cap_pipe);
 801     eth_close(ld->cap_pipe_fd);
 802   }
 803
 804   /* if open, close the pcap "input file" */
 805   if(ld->pcap_h != NULL) {
 806     g_assert(!ld->from_cap_pipe);
 807     pcap_close(ld->pcap_h);
 808   }
 809
 810 #ifdef _WIN32
 811   /* Shut down windows sockets */
 812   WSACleanup();
 813 #endif
 814 }
 815
 816
 817 /* init the capture filter */
 818 initfilter_status_t capture_loop_init_filter(pcap_t *pcap_h, gboolean from_cap_pipe, const gchar * iface, gchar * cfilter) {
 819   bpf_u_int32 netnum, netmask;
 820   gchar       lookup_net_err_str[PCAP_ERRBUF_SIZE];
 821   struct bpf_program fcode;
 822
 823
 824   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_init_filter: %s", cfilter);
 825
 826   /* capture filters only work on real interfaces */
 827   if (cfilter && !from_cap_pipe) {
 828     /* A capture filter was specified; set it up. */
 829     if (pcap_lookupnet(iface, &netnum, &netmask, lookup_net_err_str) < 0) {
 830       /*
 831        * Well, we can't get the netmask for this interface; it's used
 832        * only for filters that check for broadcast IP addresses, so
 833        * we just punt and use 0.  It might be nice to warn the user,
 834        * but that's a pain in a GUI application, as it'd involve popping
 835        * up a message box, and it's not clear how often this would make
 836        * a difference (only filters that check for IP broadcast addresses
 837        * use the netmask).
 838        */
 839       /*cmdarg_err(
 840         "Warning:  Couldn't obtain netmask info (%s).", lookup_net_err_str);*/
 841       netmask = 0;
 842     }
 843     if (pcap_compile(pcap_h, &fcode, cfilter, 1, netmask) < 0) {
 844       /* Treat this specially - our caller might try to compile this
 845          as a display filter and, if that succeeds, warn the user that
 846          the display and capture filter syntaxes are different. */
 847       return INITFILTER_BAD_FILTER;
 848     }
 849     if (pcap_setfilter(pcap_h, &fcode) < 0) {
 850 #ifdef HAVE_PCAP_FREECODE
 851       pcap_freecode(&fcode);
 852 #endif
 853       return INITFILTER_OTHER_ERROR;
 854     }
 855 #ifdef HAVE_PCAP_FREECODE
 856     pcap_freecode(&fcode);
 857 #endif
 858   }
 859
 860   return INITFILTER_NO_ERROR;
 861 }
 862
 863
 864 /* set up to write to the already-opened capture output file/files */
 865 gboolean capture_loop_init_output(capture_options *capture_opts, int save_file_fd, loop_data *ld, char *errmsg, int errmsg_len) {
 866   int         file_snaplen;
 867   int         err;
 868
 869
 870   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_init_output");
 871
 872   /* get snaplen */
 873   if (ld->from_cap_pipe) {
 874     file_snaplen = ld->cap_pipe_hdr.snaplen;
 875   } else
 876   {
 877     file_snaplen = pcap_snapshot(ld->pcap_h);
 878   }
 879
 880   /* Set up to write to the capture file. */
 881   if (capture_opts->multi_files_on) {
 882     ld->pdh = ringbuf_init_libpcap_fdopen(ld->linktype, file_snaplen,
 883                                           &ld->bytes_written, &err);
 884   } else {
 885     ld->pdh = libpcap_fdopen(save_file_fd, ld->linktype, file_snaplen,
 886                              &ld->bytes_written, &err);
 887   }
 888
 889   if (ld->pdh == NULL) {
 890     /* We couldn't set up to write to the capture file. */
 891     /* XXX - use cf_open_error_message from tshark instead? */
 892     switch (err) {
 893
 894     case WTAP_ERR_CANT_OPEN:
 895       strcpy(errmsg, "The file to which the capture would be saved"
 896                " couldn't be created for some unknown reason.");
 897       break;
 898
 899     case WTAP_ERR_SHORT_WRITE:
 900       strcpy(errmsg, "A full header couldn't be written to the file"
 901                " to which the capture would be saved.");
 902       break;
 903
 904     default:
 905       if (err < 0) {
 906         g_snprintf(errmsg, errmsg_len,
 907                      "The file to which the capture would be"
 908                      " saved (\"%s\") could not be opened: Error %d.",
 909                         capture_opts->save_file, err);
 910       } else {
 911         g_snprintf(errmsg, errmsg_len,
 912                      "The file to which the capture would be"
 913                      " saved (\"%s\") could not be opened: %s.",
 914                         capture_opts->save_file, strerror(err));
 915       }
 916       break;
 917     }
 918
 919     return FALSE;
 920   }
 921
 922   return TRUE;
 923 }
 924
 925 gboolean capture_loop_close_output(capture_options *capture_opts, loop_data *ld, int *err_close) {
 926
 927   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_close_output");
 928
 929   if (capture_opts->multi_files_on) {
 930     return ringbuf_libpcap_dump_close(&capture_opts->save_file, err_close);
 931   } else {
 932     return libpcap_dump_close(ld->pdh, err_close);
 933   }
 934 }
 935
 936 /* dispatch incoming packets (pcap or capture pipe) */
 937 int
 938 capture_loop_dispatch(capture_options *capture_opts _U_, loop_data *ld,
 939                       char *errmsg, int errmsg_len) {
 940   int       inpkts;
 941   int         sel_ret;
 942   guchar pcap_data[WTAP_MAX_PACKET_SIZE];
 943
 944   if (ld->from_cap_pipe) {
 945     /* dispatch from capture pipe */
 946 #ifdef LOG_CAPTURE_VERBOSE
 947     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_dispatch: from capture pipe");
 948 #endif
 949     sel_ret = cap_pipe_select(ld->cap_pipe_fd, FALSE);
 950     if (sel_ret <= 0) {
 951       inpkts = 0;
 952       if (sel_ret < 0 && errno != EINTR) {
 953         g_snprintf(errmsg, errmsg_len,
 954           "Unexpected error from select: %s", strerror(errno));
 955         report_capture_error(errmsg, please_report);
 956         ld->go = FALSE;
 957       }
 958     } else {
 959       /*
 960        * "select()" says we can read from the pipe without blocking
 961        */
 962       inpkts = cap_pipe_dispatch(ld, pcap_data, errmsg, errmsg_len);
 963       if (inpkts < 0) {
 964         ld->go = FALSE;
 965       }
 966     }
 967   }
 968   else
 969   {
 970     /* dispatch from pcap */
 971 #ifdef MUST_DO_SELECT
 972     /*
 973      * If we have "pcap_get_selectable_fd()", we use it to get the
 974      * descriptor on which to select; if that's -1, it means there
 975      * is no descriptor on which you can do a "select()" (perhaps
 976      * because you're capturing on a special device, and that device's
 977      * driver unfortunately doesn't support "select()", in which case
 978      * we don't do the select - which means it might not be possible
 979      * to stop a capture until a packet arrives.  If that's unacceptable,
 980      * plead with whoever supplies the software for that device to add
 981      * "select()" support, or upgrade to libpcap 0.8.1 or later, and
 982      * rebuild Wireshark or get a version built with libpcap 0.8.1 or
 983      * later, so it can use pcap_breakloop().
 984      */
 985 #ifdef LOG_CAPTURE_VERBOSE
 986     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_dispatch: from pcap_dispatch with select");
 987 #endif
 988     if (ld->pcap_fd != -1) {
 989       sel_ret = cap_pipe_select(ld->pcap_fd, TRUE);
 990       if (sel_ret > 0) {
 991         /*
 992          * "select()" says we can read from it without blocking; go for
 993          * it.
 994          *
 995          * We don't have pcap_breakloop(), so we only process one packet
 996          * per pcap_dispatch() call, to allow a signal to stop the
 997          * processing immediately, rather than processing all packets
 998          * in a batch before quitting.
 999          */
1000         inpkts = pcap_dispatch(ld->pcap_h, 1, ld->packet_cb, (u_char *)ld);
1001         if (inpkts < 0) {
1002           ld->pcap_err = TRUE;
1003           ld->go = FALSE; /* error or pcap_breakloop() - stop capturing */
1004         }
1005       } else {
1006         inpkts = 0;
1007         if (sel_ret < 0 && errno != EINTR) {
1008           g_snprintf(errmsg, errmsg_len,
1009             "Unexpected error from select: %s", strerror(errno));
1010           report_capture_error(errmsg, please_report);
1011           ld->go = FALSE;
1012         }
1013       }
1014     }
1015     else
1016 #endif /* MUST_DO_SELECT */
1017     {
1018       /* dispatch from pcap without select */
1019 #if 1
1020 #ifdef LOG_CAPTURE_VERBOSE
1021       g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_dispatch: from pcap_dispatch");
1022 #endif
1023 #ifdef _WIN32
1024       /*
1025        * On Windows, we don't support asynchronously telling a process to
1026        * stop capturing; instead, we check for an indication on a pipe
1027        * after processing packets.  We therefore process only one packet
1028        * at a time, so that we can check the pipe after every packet.
1029        */
1030       inpkts = pcap_dispatch(ld->pcap_h, 1, ld->packet_cb, (u_char *) ld);
1031 #else
1032       inpkts = pcap_dispatch(ld->pcap_h, -1, ld->packet_cb, (u_char *) ld);
1033 #endif
1034       if (inpkts < 0) {
1035         if (inpkts == -1) {
1036           /* Error, rather than pcap_breakloop(). */
1037           ld->pcap_err = TRUE;
1038         }
1039         ld->go = FALSE; /* error or pcap_breakloop() - stop capturing */
1040       }
1041 #else /* pcap_next_ex */
1042 #ifdef LOG_CAPTURE_VERBOSE
1043       g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_dispatch: from pcap_next_ex");
1044 #endif
1045       /* XXX - this is currently unused, as there is some confusion with pcap_next_ex() vs. pcap_dispatch() */
1046
1047       /*
1048        * WinPcap's remote capturing feature doesn't work with pcap_dispatch(),
1049        * see http://wiki.wireshark.org/CaptureSetup_2fWinPcapRemote
1050        * This should be fixed in the WinPcap 4.0 alpha release.
1051        *
1052        * For reference, an example remote interface:
1053        * rpcap://[1.2.3.4]/\Device\NPF_{39993D68-7C9B-4439-A329-F2D888DA7C5C}
1054        */
1055
1056       /* emulate dispatch from pcap */
1057       {
1058         int in;
1059         struct pcap_pkthdr *pkt_header;
1060         u_char *pkt_data;
1061
1062         inpkts = 0;
1063         in = 0;
1064         while(ld->go &&
1065               (in = pcap_next_ex(ld->pcap_h, &pkt_header, &pkt_data)) == 1) {
1066           ld->packet_cb( (u_char *) ld, pkt_header, pkt_data);
1067           inpkts++;
1068         }
1069
1070         if(in < 0) {
1071           ld->pcap_err = TRUE;
1072           ld->go = FALSE;
1073           inpkts = in;
1074         }
1075       }
1076 #endif /* pcap_next_ex */
1077     }
1078   }
1079
1080 #ifdef LOG_CAPTURE_VERBOSE
1081   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_dispatch: %d new packet%s", inpkts, plurality(inpkts, "", "s"));
1082 #endif
1083
1084   return inpkts;
1085 }
1086
1087
1088 /* open the output file (temporary/specified name/ringbuffer/named pipe/stdout) */
1089 /* Returns TRUE if the file opened successfully, FALSE otherwise. */
1090 gboolean
1091 capture_loop_open_output(capture_options *capture_opts, int *save_file_fd,
1092                       char *errmsg, int errmsg_len) {
1093
1094   char tmpname[128+1];
1095   gchar *capfile_name;
1096   gboolean is_tempfile;
1097
1098
1099   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_open_output: %s",
1100       (capture_opts->save_file) ? capture_opts->save_file : "");
1101
1102   if (capture_opts->save_file != NULL) {
1103     /* We return to the caller while the capture is in progress.
1104      * Therefore we need to take a copy of save_file in
1105      * case the caller destroys it after we return.
1106      */
1107     capfile_name = g_strdup(capture_opts->save_file);
1108     if (strcmp(capfile_name, "-") == 0) {
1109       /* Write to the standard output. */
1110       if (capture_opts->multi_files_on) {
1111         /* ringbuffer is enabled; that doesn't work with standard output */
1112         g_snprintf(errmsg, errmsg_len,
1113             "Ring buffer requested, but capture is being written to the standard output.");
1114         g_free(capfile_name);
1115         return FALSE;
1116       } else {
1117         *save_file_fd = 1;
1118       }
1119     } else {
1120       if (capture_opts->multi_files_on) {
1121         /* ringbuffer is enabled */
1122         *save_file_fd = ringbuf_init(capfile_name,
1123             (capture_opts->has_ring_num_files) ? capture_opts->ring_num_files : 0);
1124
1125         /* we need the ringbuf name */
1126         if(*save_file_fd != -1) {
1127             g_free(capfile_name);
1128             capfile_name = g_strdup(ringbuf_current_filename());
1129         }
1130       } else {
1131         /* Try to open/create the specified file for use as a capture buffer. */
1132         *save_file_fd = open(capfile_name, O_RDWR|O_BINARY|O_TRUNC|O_CREAT,
1133                              0600);
1134       }
1135     }
1136     is_tempfile = FALSE;
1137   } else {
1138     /* Choose a random name for the temporary capture buffer */
1139     *save_file_fd = create_tempfile(tmpname, sizeof tmpname, "ether");
1140     capfile_name = g_strdup(tmpname);
1141     is_tempfile = TRUE;
1142   }
1143
1144   /* did we fail to open the output file? */
1145   if (*save_file_fd == -1) {
1146     if (is_tempfile) {
1147       g_snprintf(errmsg, errmsg_len,
1148         "The temporary file to which the capture would be saved (\"%s\") "
1149         "could not be opened: %s.", capfile_name, strerror(errno));
1150     } else {
1151       if (capture_opts->multi_files_on) {
1152         ringbuf_error_cleanup();
1153       }
1154
1155       g_snprintf(errmsg, errmsg_len,
1156             "The file to which the capture would be saved (\"%s\") "
1157         "could not be opened: %s.", capfile_name,
1158         strerror(errno));
1159     }
1160     g_free(capfile_name);
1161     return FALSE;
1162   }
1163
1164   if(capture_opts->save_file != NULL) {
1165     g_free(capture_opts->save_file);
1166   }
1167   capture_opts->save_file = capfile_name;
1168   /* capture_opts.save_file is "g_free"ed later, which is equivalent to
1169      "g_free(capfile_name)". */
1170
1171   return TRUE;
1172 }
1173
1174
1175 static void
1176 capture_loop_stop_signal_handler(int signo _U_)
1177 {
1178   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO, "Signal: Stop capture");
1179   capture_loop_stop();
1180 }
1181
1182 #ifdef _WIN32
1183 #define TIME_GET() GetTickCount()
1184 #else
1185 #define TIME_GET() time(NULL)
1186 #endif
1187
1188 /* Do the low-level work of a capture.
1189    Returns TRUE if it succeeds, FALSE otherwise. */
1190 int
1191 capture_loop_start(capture_options *capture_opts, gboolean *stats_known, struct pcap_stat *stats)
1192 {
1193 #ifndef _WIN32
1194   struct sigaction act;
1195 #endif
1196   time_t      upd_time, cur_time;
1197   time_t      start_time;
1198   int         err_close;
1199   int         inpkts;
1200   gint        inpkts_to_sync_pipe = 0;     /* packets not already send out to the sync_pipe */
1201   condition  *cnd_file_duration = NULL;
1202   condition  *cnd_autostop_files = NULL;
1203   condition  *cnd_autostop_size = NULL;
1204   condition  *cnd_autostop_duration = NULL;
1205   guint32     autostop_files = 0;
1206   gboolean    write_ok;
1207   gboolean    close_ok;
1208   gboolean    cfilter_error = FALSE;
1209   char        errmsg[4096+1];
1210   char        secondary_errmsg[4096+1];
1211   int         save_file_fd = -1;
1212
1213
1214   /* init the loop data */
1215   ld.go                 = TRUE;
1216   ld.packet_count       = 0;
1217   if (capture_opts->has_autostop_packets)
1218     ld.packet_max       = capture_opts->autostop_packets;
1219   else
1220     ld.packet_max       = 0;    /* no limit */
1221   ld.err                = 0;    /* no error seen yet */
1222   ld.wtap_linktype      = WTAP_ENCAP_UNKNOWN;
1223   ld.pcap_err           = FALSE;
1224   ld.from_cap_pipe      = FALSE;
1225   ld.pdh                = NULL;
1226   ld.cap_pipe_fd        = -1;
1227 #ifdef MUST_DO_SELECT
1228   ld.pcap_fd            = 0;
1229 #endif
1230   ld.packet_cb          = capture_loop_packet_cb;
1231
1232
1233   /* We haven't yet gotten the capture statistics. */
1234   *stats_known      = FALSE;
1235
1236 #ifndef _WIN32
1237   /*
1238    * Catch SIGUSR1, so that we exit cleanly if the parent process
1239    * kills us with it due to the user selecting "Capture->Stop".
1240    */
1241   act.sa_handler = capture_loop_stop_signal_handler;
1242   /*
1243    * Arrange that system calls not get restarted, because when
1244    * our signal handler returns we don't want to restart
1245    * a call that was waiting for packets to arrive.
1246    */
1247   act.sa_flags = 0;
1248   sigemptyset(&act.sa_mask);
1249   sigaction(SIGUSR1, &act, NULL);
1250 #endif /* _WIN32 */
1251
1252   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO, "Capture loop starting ...");
1253   capture_opts_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, capture_opts);
1254
1255   /* open the "input file" from network interface or capture pipe */
1256   if (!capture_loop_open_input(capture_opts, &ld, errmsg, sizeof(errmsg),
1257                                secondary_errmsg, sizeof(secondary_errmsg))) {
1258     goto error;
1259   }
1260
1261   /* init the input filter from the network interface (capture pipe will do nothing) */
1262   switch (capture_loop_init_filter(ld.pcap_h, ld.from_cap_pipe, capture_opts->iface, capture_opts->cfilter)) {
1263
1264   case INITFILTER_NO_ERROR:
1265     break;
1266
1267   case INITFILTER_BAD_FILTER:
1268     cfilter_error = TRUE;
1269     g_snprintf(errmsg, sizeof(errmsg), "%s", pcap_geterr(ld.pcap_h));
1270     *secondary_errmsg = '\0';
1271     goto error;
1272
1273   case INITFILTER_OTHER_ERROR:
1274     g_snprintf(errmsg, sizeof(errmsg), "Can't install filter (%s).",
1275                pcap_geterr(ld.pcap_h));
1276     g_snprintf(secondary_errmsg, sizeof(secondary_errmsg), "%s", please_report);
1277     goto error;
1278   }
1279
1280   /* If we're supposed to write to a capture file, open it for output
1281      (temporary/specified name/ringbuffer) */
1282   if (capture_opts->saving_to_file) {
1283     if (!capture_loop_open_output(capture_opts, &save_file_fd, errmsg, sizeof(errmsg))) {
1284       *secondary_errmsg = '\0';
1285       goto error;
1286     }
1287
1288     /* set up to write to the already-opened capture output file/files */
1289     if (!capture_loop_init_output(capture_opts, save_file_fd, &ld, errmsg, sizeof(errmsg))) {
1290       *secondary_errmsg = '\0';
1291       goto error;
1292     }
1293
1294   /* XXX - capture SIGTERM and close the capture, in case we're on a
1295      Linux 2.0[.x] system and you have to explicitly close the capture
1296      stream in order to turn promiscuous mode off?  We need to do that
1297      in other places as well - and I don't think that works all the
1298      time in any case, due to libpcap bugs. */
1299
1300     /* Well, we should be able to start capturing.
1301
1302        Sync out the capture file, so the header makes it to the file system,
1303        and send a "capture started successfully and capture file created"
1304        message to our parent so that they'll open the capture file and
1305        update its windows to indicate that we have a live capture in
1306        progress. */
1307     libpcap_dump_flush(ld.pdh, NULL);
1308     report_new_capture_file(capture_opts->save_file);
1309   }
1310
1311   /* initialize capture stop (and alike) conditions */
1312   init_capture_stop_conditions();
1313   /* create stop conditions */
1314   if (capture_opts->has_autostop_filesize)
1315     cnd_autostop_size =
1316         cnd_new(CND_CLASS_CAPTURESIZE,(long)capture_opts->autostop_filesize * 1024);
1317   if (capture_opts->has_autostop_duration)
1318     cnd_autostop_duration =
1319         cnd_new(CND_CLASS_TIMEOUT,(gint32)capture_opts->autostop_duration);
1320
1321   if (capture_opts->multi_files_on) {
1322       if (capture_opts->has_file_duration)
1323         cnd_file_duration =
1324             cnd_new(CND_CLASS_TIMEOUT, capture_opts->file_duration);
1325
1326       if (capture_opts->has_autostop_files)
1327         cnd_autostop_files =
1328             cnd_new(CND_CLASS_CAPTURESIZE, capture_opts->autostop_files);
1329   }
1330
1331   /* init the time values */
1332   start_time = TIME_GET();
1333   upd_time = TIME_GET();
1334
1335   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO, "Capture loop running!");
1336
1337   /* WOW, everything is prepared! */
1338   /* please fasten your seat belts, we will enter now the actual capture loop */
1339   while (ld.go) {
1340     /* dispatch incoming packets */
1341     inpkts = capture_loop_dispatch(capture_opts, &ld, errmsg, sizeof(errmsg));
1342
1343 #ifdef _WIN32
1344     /* any news from our parent (signal pipe)? -> just stop the capture */
1345     if (!signal_pipe_check_running()) {
1346       ld.go = FALSE;
1347     }
1348 #endif
1349
1350     if (inpkts > 0) {
1351       inpkts_to_sync_pipe += inpkts;
1352
1353       /* check capture size condition */
1354       if (cnd_autostop_size != NULL &&
1355           cnd_eval(cnd_autostop_size, (guint32)ld.bytes_written)){
1356         /* Capture size limit reached, do we have another file? */
1357         if (capture_opts->multi_files_on) {
1358           if (cnd_autostop_files != NULL &&
1359               cnd_eval(cnd_autostop_files, ++autostop_files)) {
1360              /* no files left: stop here */
1361             ld.go = FALSE;
1362             continue;
1363           }
1364
1365           /* Switch to the next ringbuffer file */
1366           if (ringbuf_switch_file(&ld.pdh, &capture_opts->save_file,
1367               &save_file_fd, &ld.bytes_written, &ld.err)) {
1368             /* File switch succeeded: reset the conditions */
1369             cnd_reset(cnd_autostop_size);
1370             if (cnd_file_duration) {
1371               cnd_reset(cnd_file_duration);
1372             }
1373             libpcap_dump_flush(ld.pdh, NULL);
1374             report_packet_count(inpkts_to_sync_pipe);
1375             inpkts_to_sync_pipe = 0;
1376             report_new_capture_file(capture_opts->save_file);
1377           } else {
1378             /* File switch failed: stop here */
1379             ld.go = FALSE;
1380             continue;
1381           }
1382         } else {
1383           /* single file, stop now */
1384           ld.go = FALSE;
1385           continue;
1386         }
1387       } /* cnd_autostop_size */
1388       if (capture_opts->output_to_pipe) {
1389         libpcap_dump_flush(ld.pdh, NULL);
1390       }
1391     } /* inpkts */
1392
1393     /* Only update once a second (Win32: 500ms) so as not to overload slow displays */
1394     cur_time = TIME_GET();
1395 #ifdef _WIN32
1396     if ( (cur_time - upd_time) > 500) {
1397 #else
1398     if (cur_time - upd_time > 0) {
1399 #endif
1400         upd_time = cur_time;
1401
1402       /*if (pcap_stats(pch, stats) >= 0) {
1403         *stats_known = TRUE;
1404       }*/
1405
1406       /* Let the parent process know. */
1407       if (inpkts_to_sync_pipe) {
1408         /* do sync here */
1409         libpcap_dump_flush(ld.pdh, NULL);
1410
1411         /* Send our parent a message saying we've written out "inpkts_to_sync_pipe"
1412            packets to the capture file. */
1413         report_packet_count(inpkts_to_sync_pipe);
1414
1415         inpkts_to_sync_pipe = 0;
1416       }
1417
1418       /* check capture duration condition */
1419       if (cnd_autostop_duration != NULL && cnd_eval(cnd_autostop_duration)) {
1420         /* The maximum capture time has elapsed; stop the capture. */
1421         ld.go = FALSE;
1422         continue;
1423       }
1424
1425       /* check capture file duration condition */
1426       if (cnd_file_duration != NULL && cnd_eval(cnd_file_duration)) {
1427         /* duration limit reached, do we have another file? */
1428         if (capture_opts->multi_files_on) {
1429           if (cnd_autostop_files != NULL &&
1430               cnd_eval(cnd_autostop_files, ++autostop_files)) {
1431             /* no files left: stop here */
1432             ld.go = FALSE;
1433             continue;
1434           }
1435
1436           /* Switch to the next ringbuffer file */
1437           if (ringbuf_switch_file(&ld.pdh, &capture_opts->save_file,
1438                                   &save_file_fd, &ld.bytes_written, &ld.err)) {
1439             /* file switch succeeded: reset the conditions */
1440             cnd_reset(cnd_file_duration);
1441             if(cnd_autostop_size)
1442               cnd_reset(cnd_autostop_size);
1443             libpcap_dump_flush(ld.pdh, NULL);
1444             report_packet_count(inpkts_to_sync_pipe);
1445             inpkts_to_sync_pipe = 0;
1446             report_new_capture_file(capture_opts->save_file);
1447           } else {
1448             /* File switch failed: stop here */
1449             ld.go = FALSE;
1450             continue;
1451           }
1452         } else {
1453           /* single file, stop now */
1454           ld.go = FALSE;
1455           continue;
1456         }
1457       } /* cnd_file_duration */
1458     }
1459
1460   } /* while (ld.go) */
1461
1462   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO, "Capture loop stopping ...");
1463
1464   /* delete stop conditions */
1465   if (cnd_file_duration != NULL)
1466     cnd_delete(cnd_file_duration);
1467   if (cnd_autostop_files != NULL)
1468     cnd_delete(cnd_autostop_files);
1469   if (cnd_autostop_size != NULL)
1470     cnd_delete(cnd_autostop_size);
1471   if (cnd_autostop_duration != NULL)
1472     cnd_delete(cnd_autostop_duration);
1473
1474   /* did we had a pcap (input) error? */
1475   if (ld.pcap_err) {
1476     g_snprintf(errmsg, sizeof(errmsg), "Error while capturing packets: %s",
1477       pcap_geterr(ld.pcap_h));
1478     report_capture_error(errmsg, please_report);
1479   }
1480     else if (ld.from_cap_pipe && ld.cap_pipe_err == PIPERR)
1481       report_capture_error(errmsg, "");
1482
1483   /* did we had an error while capturing? */
1484   if (ld.err == 0) {
1485     write_ok = TRUE;
1486   } else {
1487     capture_loop_get_errmsg(errmsg, sizeof(errmsg), capture_opts->save_file, ld.err,
1488                               FALSE);
1489     report_capture_error(errmsg, please_report);
1490     write_ok = FALSE;
1491   }
1492
1493   if (capture_opts->saving_to_file) {
1494     /* close the wiretap (output) file */
1495     close_ok = capture_loop_close_output(capture_opts, &ld, &err_close);
1496   } else
1497     close_ok = TRUE;
1498
1499   /* there might be packets not yet notified to the parent */
1500   /* (do this after closing the file, so all packets are already flushed) */
1501   if(inpkts_to_sync_pipe) {
1502     report_packet_count(inpkts_to_sync_pipe);
1503     inpkts_to_sync_pipe = 0;
1504   }
1505
1506   /* If we've displayed a message about a write error, there's no point
1507      in displaying another message about an error on close. */
1508   if (!close_ok && write_ok) {
1509     capture_loop_get_errmsg(errmsg, sizeof(errmsg), capture_opts->save_file, err_close,
1510                 TRUE);
1511     report_capture_error(errmsg, "");
1512   }
1513
1514   /*
1515    * XXX We exhibit different behaviour between normal mode and sync mode
1516    * when the pipe is stdin and not already at EOF.  If we're a child, the
1517    * parent's stdin isn't closed, so if the user starts another capture,
1518    * cap_pipe_open_live() will very likely not see the expected magic bytes and
1519    * will say "Unrecognized libpcap format".  On the other hand, in normal
1520    * mode, cap_pipe_open_live() will say "End of file on pipe during open".
1521    */
1522
1523   /* get packet drop statistics from pcap */
1524   if(ld.pcap_h != NULL) {
1525     g_assert(!ld.from_cap_pipe);
1526     /* Get the capture statistics, so we know how many packets were
1527        dropped. */
1528     if (pcap_stats(ld.pcap_h, stats) >= 0) {
1529       *stats_known = TRUE;
1530       /* Let the parent process know. */
1531       report_packet_drops(stats->ps_drop);
1532     } else {
1533       g_snprintf(errmsg, sizeof(errmsg),
1534                 "Can't get packet-drop statistics: %s",
1535                 pcap_geterr(ld.pcap_h));
1536       report_capture_error(errmsg, please_report);
1537     }
1538   }
1539
1540   /* close the input file (pcap or capture pipe) */
1541   capture_loop_close_input(&ld);
1542
1543   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO, "Capture loop stopped!");
1544
1545   /* ok, if the write and the close were successful. */
1546   return write_ok && close_ok;
1547
1548 error:
1549   if (capture_opts->multi_files_on) {
1550     /* cleanup ringbuffer */
1551     ringbuf_error_cleanup();
1552   } else {
1553     /* We can't use the save file, and we have no FILE * for the stream
1554        to close in order to close it, so close the FD directly. */
1555     if(save_file_fd != -1) {
1556       eth_close(save_file_fd);
1557     }
1558
1559     /* We couldn't even start the capture, so get rid of the capture
1560        file. */
1561     if(capture_opts->save_file != NULL) {
1562       eth_unlink(capture_opts->save_file);
1563       g_free(capture_opts->save_file);
1564     }
1565   }
1566   capture_opts->save_file = NULL;
1567   if (cfilter_error)
1568     report_cfilter_error(capture_opts->cfilter, errmsg);
1569   else
1570     report_capture_error(errmsg, secondary_errmsg);
1571
1572   /* close the input file (pcap or cap_pipe) */
1573   capture_loop_close_input(&ld);
1574
1575   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO, "Capture loop stopped with error");
1576
1577   return FALSE;
1578 }
1579
1580
1581 void capture_loop_stop(void)
1582 {
1583 #ifdef HAVE_PCAP_BREAKLOOP
1584   if(ld.pcap_h != NULL)
1585     pcap_breakloop(ld.pcap_h);
1586 #endif
1587   ld.go = FALSE;
1588 }
1589
1590
1591 static void
1592 capture_loop_get_errmsg(char *errmsg, int errmsglen, const char *fname,
1593                           int err, gboolean is_close)
1594 {
1595   switch (err) {
1596
1597   case ENOSPC:
1598     g_snprintf(errmsg, errmsglen,
1599                 "Not all the packets could be written to the file"
1600                 " to which the capture was being saved\n"
1601                 "(\"%s\") because there is no space left on the file system\n"
1602                 "on which that file resides.",
1603                 fname);
1604     break;
1605
1606 #ifdef EDQUOT
1607   case EDQUOT:
1608     g_snprintf(errmsg, errmsglen,
1609                 "Not all the packets could be written to the file"
1610                 " to which the capture was being saved\n"
1611                 "(\"%s\") because you are too close to, or over,"
1612                 " your disk quota\n"
1613                 "on the file system on which that file resides.",
1614                 fname);
1615   break;
1616 #endif
1617
1618   case WTAP_ERR_CANT_CLOSE:
1619     g_snprintf(errmsg, errmsglen,
1620                 "The file to which the capture was being saved"
1621                 " couldn't be closed for some unknown reason.");
1622     break;
1623
1624   case WTAP_ERR_SHORT_WRITE:
1625     g_snprintf(errmsg, errmsglen,
1626                 "Not all the packets could be written to the file"
1627                 " to which the capture was being saved\n"
1628                 "(\"%s\").",
1629                 fname);
1630     break;
1631
1632   default:
1633     if (is_close) {
1634       g_snprintf(errmsg, errmsglen,
1635                 "The file to which the capture was being saved\n"
1636                 "(\"%s\") could not be closed: %s.",
1637                 fname, wtap_strerror(err));
1638     } else {
1639       g_snprintf(errmsg, errmsglen,
1640                 "An error occurred while writing to the file"
1641                 " to which the capture was being saved\n"
1642                 "(\"%s\"): %s.",
1643                 fname, wtap_strerror(err));
1644     }
1645     break;
1646   }
1647 }
1648
1649
1650 /* one packet was captured, process it */
1651 static void
1652 capture_loop_packet_cb(u_char *user, const struct pcap_pkthdr *phdr,
1653   const u_char *pd)
1654 {
1655   loop_data *ld = (loop_data *) user;
1656   int err;
1657
1658   /* if the user told us to stop after x packets, do we have enough? */
1659   ld->packet_count++;
1660   if ((ld->packet_max > 0) && (ld->packet_count >= ld->packet_max))
1661   {
1662      ld->go = FALSE;
1663   }
1664
1665   if (ld->pdh) {
1666     /* We're supposed to write the packet to a file; do so.
1667        If this fails, set "ld->go" to FALSE, to stop the capture, and set
1668        "ld->err" to the error. */
1669     if (!libpcap_write_packet(ld->pdh, phdr, pd, &ld->bytes_written, &err)) {
1670       ld->go = FALSE;
1671       ld->err = err;
1672     }
1673   }
1674 }
1675
1676 #endif /* HAVE_LIBPCAP */