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     ld->cap_pipe_fd = 0;
 803   }
 804
 805   /* if open, close the pcap "input file" */
 806   if(ld->pcap_h != NULL) {
 807     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_close_input: closing %p", ld->pcap_h);
 808     g_assert(!ld->from_cap_pipe);
 809     pcap_close(ld->pcap_h);
 810     ld->pcap_h = NULL;
 811   }
 812
 813   ld->go = FALSE;
 814
 815 #ifdef _WIN32
 816   /* Shut down windows sockets */
 817   WSACleanup();
 818 #endif
 819 }
 820
 821
 822 /* init the capture filter */
 823 initfilter_status_t capture_loop_init_filter(pcap_t *pcap_h, gboolean from_cap_pipe, const gchar * iface, gchar * cfilter) {
 824   bpf_u_int32 netnum, netmask;
 825   gchar       lookup_net_err_str[PCAP_ERRBUF_SIZE];
 826   struct bpf_program fcode;
 827
 828
 829   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_init_filter: %s", cfilter);
 830
 831   /* capture filters only work on real interfaces */
 832   if (cfilter && !from_cap_pipe) {
 833     /* A capture filter was specified; set it up. */
 834     if (pcap_lookupnet(iface, &netnum, &netmask, lookup_net_err_str) < 0) {
 835       /*
 836        * Well, we can't get the netmask for this interface; it's used
 837        * only for filters that check for broadcast IP addresses, so
 838        * we just punt and use 0.  It might be nice to warn the user,
 839        * but that's a pain in a GUI application, as it'd involve popping
 840        * up a message box, and it's not clear how often this would make
 841        * a difference (only filters that check for IP broadcast addresses
 842        * use the netmask).
 843        */
 844       /*cmdarg_err(
 845         "Warning:  Couldn't obtain netmask info (%s).", lookup_net_err_str);*/
 846       netmask = 0;
 847     }
 848     if (pcap_compile(pcap_h, &fcode, cfilter, 1, netmask) < 0) {
 849       /* Treat this specially - our caller might try to compile this
 850          as a display filter and, if that succeeds, warn the user that
 851          the display and capture filter syntaxes are different. */
 852       return INITFILTER_BAD_FILTER;
 853     }
 854     if (pcap_setfilter(pcap_h, &fcode) < 0) {
 855 #ifdef HAVE_PCAP_FREECODE
 856       pcap_freecode(&fcode);
 857 #endif
 858       return INITFILTER_OTHER_ERROR;
 859     }
 860 #ifdef HAVE_PCAP_FREECODE
 861     pcap_freecode(&fcode);
 862 #endif
 863   }
 864
 865   return INITFILTER_NO_ERROR;
 866 }
 867
 868
 869 /* set up to write to the already-opened capture output file/files */
 870 gboolean capture_loop_init_output(capture_options *capture_opts, int save_file_fd, loop_data *ld, char *errmsg, int errmsg_len) {
 871   int         file_snaplen;
 872   int         err;
 873
 874
 875   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_init_output");
 876
 877   /* get snaplen */
 878   if (ld->from_cap_pipe) {
 879     file_snaplen = ld->cap_pipe_hdr.snaplen;
 880   } else
 881   {
 882     file_snaplen = pcap_snapshot(ld->pcap_h);
 883   }
 884
 885   /* Set up to write to the capture file. */
 886   if (capture_opts->multi_files_on) {
 887     ld->pdh = ringbuf_init_libpcap_fdopen(ld->linktype, file_snaplen,
 888                                           &ld->bytes_written, &err);
 889   } else {
 890     ld->pdh = libpcap_fdopen(save_file_fd, ld->linktype, file_snaplen,
 891                              &ld->bytes_written, &err);
 892   }
 893
 894   if (ld->pdh == NULL) {
 895     /* We couldn't set up to write to the capture file. */
 896     /* XXX - use cf_open_error_message from tshark instead? */
 897     switch (err) {
 898
 899     case WTAP_ERR_CANT_OPEN:
 900       strcpy(errmsg, "The file to which the capture would be saved"
 901                " couldn't be created for some unknown reason.");
 902       break;
 903
 904     case WTAP_ERR_SHORT_WRITE:
 905       strcpy(errmsg, "A full header couldn't be written to the file"
 906                " to which the capture would be saved.");
 907       break;
 908
 909     default:
 910       if (err < 0) {
 911         g_snprintf(errmsg, errmsg_len,
 912                      "The file to which the capture would be"
 913                      " saved (\"%s\") could not be opened: Error %d.",
 914                         capture_opts->save_file, err);
 915       } else {
 916         g_snprintf(errmsg, errmsg_len,
 917                      "The file to which the capture would be"
 918                      " saved (\"%s\") could not be opened: %s.",
 919                         capture_opts->save_file, strerror(err));
 920       }
 921       break;
 922     }
 923
 924     return FALSE;
 925   }
 926
 927   return TRUE;
 928 }
 929
 930 gboolean capture_loop_close_output(capture_options *capture_opts, loop_data *ld, int *err_close) {
 931
 932   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_close_output");
 933
 934   if (capture_opts->multi_files_on) {
 935     return ringbuf_libpcap_dump_close(&capture_opts->save_file, err_close);
 936   } else {
 937     return libpcap_dump_close(ld->pdh, err_close);
 938   }
 939 }
 940
 941 /* dispatch incoming packets (pcap or capture pipe) */
 942 int
 943 capture_loop_dispatch(capture_options *capture_opts _U_, loop_data *ld,
 944                       char *errmsg, int errmsg_len) {
 945   int       inpkts;
 946   int         sel_ret;
 947   guchar pcap_data[WTAP_MAX_PACKET_SIZE];
 948
 949   if (ld->from_cap_pipe) {
 950     /* dispatch from capture pipe */
 951 #ifdef LOG_CAPTURE_VERBOSE
 952     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_dispatch: from capture pipe");
 953 #endif
 954     sel_ret = cap_pipe_select(ld->cap_pipe_fd, FALSE);
 955     if (sel_ret <= 0) {
 956       inpkts = 0;
 957       if (sel_ret < 0 && errno != EINTR) {
 958         g_snprintf(errmsg, errmsg_len,
 959           "Unexpected error from select: %s", strerror(errno));
 960         report_capture_error(errmsg, please_report);
 961         ld->go = FALSE;
 962       }
 963     } else {
 964       /*
 965        * "select()" says we can read from the pipe without blocking
 966        */
 967       inpkts = cap_pipe_dispatch(ld, pcap_data, errmsg, errmsg_len);
 968       if (inpkts < 0) {
 969         ld->go = FALSE;
 970       }
 971     }
 972   }
 973   else
 974   {
 975     /* dispatch from pcap */
 976 #ifdef MUST_DO_SELECT
 977     /*
 978      * If we have "pcap_get_selectable_fd()", we use it to get the
 979      * descriptor on which to select; if that's -1, it means there
 980      * is no descriptor on which you can do a "select()" (perhaps
 981      * because you're capturing on a special device, and that device's
 982      * driver unfortunately doesn't support "select()", in which case
 983      * we don't do the select - which means it might not be possible
 984      * to stop a capture until a packet arrives.  If that's unacceptable,
 985      * plead with whoever supplies the software for that device to add
 986      * "select()" support, or upgrade to libpcap 0.8.1 or later, and
 987      * rebuild Wireshark or get a version built with libpcap 0.8.1 or
 988      * later, so it can use pcap_breakloop().
 989      */
 990 #ifdef LOG_CAPTURE_VERBOSE
 991     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_dispatch: from pcap_dispatch with select");
 992 #endif
 993     if (ld->pcap_fd != -1) {
 994       sel_ret = cap_pipe_select(ld->pcap_fd, TRUE);
 995       if (sel_ret > 0) {
 996         /*
 997          * "select()" says we can read from it without blocking; go for
 998          * it.
 999          *
1000          * We don't have pcap_breakloop(), so we only process one packet
1001          * per pcap_dispatch() call, to allow a signal to stop the
1002          * processing immediately, rather than processing all packets
1003          * in a batch before quitting.
1004          */
1005         inpkts = pcap_dispatch(ld->pcap_h, 1, ld->packet_cb, (u_char *)ld);
1006         if (inpkts < 0) {
1007           ld->pcap_err = TRUE;
1008           ld->go = FALSE; /* error or pcap_breakloop() - stop capturing */
1009         }
1010       } else {
1011         inpkts = 0;
1012         if (sel_ret < 0 && errno != EINTR) {
1013           g_snprintf(errmsg, errmsg_len,
1014             "Unexpected error from select: %s", strerror(errno));
1015           report_capture_error(errmsg, please_report);
1016           ld->go = FALSE;
1017         }
1018       }
1019     }
1020     else
1021 #endif /* MUST_DO_SELECT */
1022     {
1023       /* dispatch from pcap without select */
1024 #if 1
1025 #ifdef LOG_CAPTURE_VERBOSE
1026       g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_dispatch: from pcap_dispatch");
1027 #endif
1028 #ifdef _WIN32
1029       /*
1030        * On Windows, we don't support asynchronously telling a process to
1031        * stop capturing; instead, we check for an indication on a pipe
1032        * after processing packets.  We therefore process only one packet
1033        * at a time, so that we can check the pipe after every packet.
1034        */
1035       inpkts = pcap_dispatch(ld->pcap_h, 1, ld->packet_cb, (u_char *) ld);
1036 #else
1037       inpkts = pcap_dispatch(ld->pcap_h, -1, ld->packet_cb, (u_char *) ld);
1038 #endif
1039       if (inpkts < 0) {
1040         if (inpkts == -1) {
1041           /* Error, rather than pcap_breakloop(). */
1042           ld->pcap_err = TRUE;
1043         }
1044         ld->go = FALSE; /* error or pcap_breakloop() - stop capturing */
1045       }
1046 #else /* pcap_next_ex */
1047 #ifdef LOG_CAPTURE_VERBOSE
1048       g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_dispatch: from pcap_next_ex");
1049 #endif
1050       /* XXX - this is currently unused, as there is some confusion with pcap_next_ex() vs. pcap_dispatch() */
1051
1052       /*
1053        * WinPcap's remote capturing feature doesn't work with pcap_dispatch(),
1054        * see http://wiki.wireshark.org/CaptureSetup_2fWinPcapRemote
1055        * This should be fixed in the WinPcap 4.0 alpha release.
1056        *
1057        * For reference, an example remote interface:
1058        * rpcap://[1.2.3.4]/\Device\NPF_{39993D68-7C9B-4439-A329-F2D888DA7C5C}
1059        */
1060
1061       /* emulate dispatch from pcap */
1062       {
1063         int in;
1064         struct pcap_pkthdr *pkt_header;
1065         u_char *pkt_data;
1066
1067         inpkts = 0;
1068         in = 0;
1069         while(ld->go &&
1070               (in = pcap_next_ex(ld->pcap_h, &pkt_header, &pkt_data)) == 1) {
1071           ld->packet_cb( (u_char *) ld, pkt_header, pkt_data);
1072           inpkts++;
1073         }
1074
1075         if(in < 0) {
1076           ld->pcap_err = TRUE;
1077           ld->go = FALSE;
1078           inpkts = in;
1079         }
1080       }
1081 #endif /* pcap_next_ex */
1082     }
1083   }
1084
1085 #ifdef LOG_CAPTURE_VERBOSE
1086   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_dispatch: %d new packet%s", inpkts, plurality(inpkts, "", "s"));
1087 #endif
1088
1089   return inpkts;
1090 }
1091
1092
1093 /* open the output file (temporary/specified name/ringbuffer/named pipe/stdout) */
1094 /* Returns TRUE if the file opened successfully, FALSE otherwise. */
1095 gboolean
1096 capture_loop_open_output(capture_options *capture_opts, int *save_file_fd,
1097                       char *errmsg, int errmsg_len) {
1098
1099   char tmpname[128+1];
1100   gchar *capfile_name;
1101   gboolean is_tempfile;
1102
1103
1104   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_open_output: %s",
1105       (capture_opts->save_file) ? capture_opts->save_file : "");
1106
1107   if (capture_opts->save_file != NULL) {
1108     /* We return to the caller while the capture is in progress.
1109      * Therefore we need to take a copy of save_file in
1110      * case the caller destroys it after we return.
1111      */
1112     capfile_name = g_strdup(capture_opts->save_file);
1113
1114     if (capture_opts->output_to_pipe == TRUE) { /* either "-" or named pipe */
1115       if (capture_opts->multi_files_on) {
1116         /* ringbuffer is enabled; that doesn't work with standard output or a named pipe */
1117         g_snprintf(errmsg, errmsg_len,
1118             "Ring buffer requested, but capture is being written to standard output or to a named pipe.");
1119         g_free(capfile_name);
1120         return FALSE;
1121       }
1122       if (strcmp(capfile_name, "-") == 0) {
1123         /* write to stdout */
1124         *save_file_fd = 1;
1125 #ifdef _WIN32
1126         /* set output pipe to binary mode to avoid Windows text-mode processing (eg: for CR/LF)  */
1127         _setmode(1, O_BINARY);
1128 #endif
1129       }
1130     } /* if (...output_to_pipe ... */
1131
1132     else {
1133       if (capture_opts->multi_files_on) {
1134         /* ringbuffer is enabled */
1135         *save_file_fd = ringbuf_init(capfile_name,
1136             (capture_opts->has_ring_num_files) ? capture_opts->ring_num_files : 0);
1137
1138         /* we need the ringbuf name */
1139         if(*save_file_fd != -1) {
1140             g_free(capfile_name);
1141             capfile_name = g_strdup(ringbuf_current_filename());
1142         }
1143       } else {
1144         /* Try to open/create the specified file for use as a capture buffer. */
1145         *save_file_fd = open(capfile_name, O_RDWR|O_BINARY|O_TRUNC|O_CREAT,
1146                              0600);
1147       }
1148     }
1149     is_tempfile = FALSE;
1150   } else {
1151     /* Choose a random name for the temporary capture buffer */
1152     *save_file_fd = create_tempfile(tmpname, sizeof tmpname, "ether");
1153     capfile_name = g_strdup(tmpname);
1154     is_tempfile = TRUE;
1155   }
1156
1157   /* did we fail to open the output file? */
1158   if (*save_file_fd == -1) {
1159     if (is_tempfile) {
1160       g_snprintf(errmsg, errmsg_len,
1161         "The temporary file to which the capture would be saved (\"%s\") "
1162         "could not be opened: %s.", capfile_name, strerror(errno));
1163     } else {
1164       if (capture_opts->multi_files_on) {
1165         ringbuf_error_cleanup();
1166       }
1167
1168       g_snprintf(errmsg, errmsg_len,
1169             "The file to which the capture would be saved (\"%s\") "
1170         "could not be opened: %s.", capfile_name,
1171         strerror(errno));
1172     }
1173     g_free(capfile_name);
1174     return FALSE;
1175   }
1176
1177   if(capture_opts->save_file != NULL) {
1178     g_free(capture_opts->save_file);
1179   }
1180   capture_opts->save_file = capfile_name;
1181   /* capture_opts.save_file is "g_free"ed later, which is equivalent to
1182      "g_free(capfile_name)". */
1183
1184   return TRUE;
1185 }
1186
1187
1188 static void
1189 capture_loop_stop_signal_handler(int signo _U_)
1190 {
1191   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO, "Signal: Stop capture");
1192   capture_loop_stop();
1193 }
1194
1195 #ifdef _WIN32
1196 #define TIME_GET() GetTickCount()
1197 #else
1198 #define TIME_GET() time(NULL)
1199 #endif
1200
1201 /* Do the low-level work of a capture.
1202    Returns TRUE if it succeeds, FALSE otherwise. */
1203 int
1204 capture_loop_start(capture_options *capture_opts, gboolean *stats_known, struct pcap_stat *stats)
1205 {
1206 #ifndef _WIN32
1207   struct sigaction act;
1208 #endif
1209   time_t      upd_time, cur_time;
1210   time_t      start_time;
1211   int         err_close;
1212   int         inpkts;
1213   gint        inpkts_to_sync_pipe = 0;     /* packets not already send out to the sync_pipe */
1214   condition  *cnd_file_duration = NULL;
1215   condition  *cnd_autostop_files = NULL;
1216   condition  *cnd_autostop_size = NULL;
1217   condition  *cnd_autostop_duration = NULL;
1218   guint32     autostop_files = 0;
1219   gboolean    write_ok;
1220   gboolean    close_ok;
1221   gboolean    cfilter_error = FALSE;
1222 #define MSG_MAX_LENGTH 4096
1223   char        errmsg[MSG_MAX_LENGTH+1];
1224   char        secondary_errmsg[MSG_MAX_LENGTH+1];
1225   int         save_file_fd = -1;
1226
1227   /* init the loop data */
1228   ld.go                 = TRUE;
1229   ld.packet_count       = 0;
1230   if (capture_opts->has_autostop_packets)
1231     ld.packet_max       = capture_opts->autostop_packets;
1232   else
1233     ld.packet_max       = 0;    /* no limit */
1234   ld.err                = 0;    /* no error seen yet */
1235   ld.wtap_linktype      = WTAP_ENCAP_UNKNOWN;
1236   ld.pcap_err           = FALSE;
1237   ld.from_cap_pipe      = FALSE;
1238   ld.pdh                = NULL;
1239   ld.cap_pipe_fd        = -1;
1240 #ifdef MUST_DO_SELECT
1241   ld.pcap_fd            = 0;
1242 #endif
1243   ld.packet_cb          = capture_loop_packet_cb;
1244
1245
1246   /* We haven't yet gotten the capture statistics. */
1247   *stats_known      = FALSE;
1248
1249 #ifndef _WIN32
1250   /*
1251    * Catch SIGUSR1, so that we exit cleanly if the parent process
1252    * kills us with it due to the user selecting "Capture->Stop".
1253    */
1254   act.sa_handler = capture_loop_stop_signal_handler;
1255   /*
1256    * Arrange that system calls not get restarted, because when
1257    * our signal handler returns we don't want to restart
1258    * a call that was waiting for packets to arrive.
1259    */
1260   act.sa_flags = 0;
1261   sigemptyset(&act.sa_mask);
1262   sigaction(SIGUSR1, &act, NULL);
1263 #endif /* _WIN32 */
1264
1265   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO, "Capture loop starting ...");
1266   capture_opts_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, capture_opts);
1267
1268   /* open the "input file" from network interface or capture pipe */
1269   if (!capture_loop_open_input(capture_opts, &ld, errmsg, sizeof(errmsg),
1270                                secondary_errmsg, sizeof(secondary_errmsg))) {
1271     goto error;
1272   }
1273
1274   /* init the input filter from the network interface (capture pipe will do nothing) */
1275   switch (capture_loop_init_filter(ld.pcap_h, ld.from_cap_pipe, capture_opts->iface, capture_opts->cfilter)) {
1276
1277   case INITFILTER_NO_ERROR:
1278     break;
1279
1280   case INITFILTER_BAD_FILTER:
1281     cfilter_error = TRUE;
1282     g_snprintf(errmsg, sizeof(errmsg), "%s", pcap_geterr(ld.pcap_h));
1283     *secondary_errmsg = '\0';
1284     goto error;
1285
1286   case INITFILTER_OTHER_ERROR:
1287     g_snprintf(errmsg, sizeof(errmsg), "Can't install filter (%s).",
1288                pcap_geterr(ld.pcap_h));
1289     g_snprintf(secondary_errmsg, sizeof(secondary_errmsg), "%s", please_report);
1290     goto error;
1291   }
1292
1293   /* If we're supposed to write to a capture file, open it for output
1294      (temporary/specified name/ringbuffer) */
1295   if (capture_opts->saving_to_file) {
1296     if (!capture_loop_open_output(capture_opts, &save_file_fd, errmsg, sizeof(errmsg))) {
1297       *secondary_errmsg = '\0';
1298       goto error;
1299     }
1300
1301     /* set up to write to the already-opened capture output file/files */
1302     if (!capture_loop_init_output(capture_opts, save_file_fd, &ld, errmsg, sizeof(errmsg))) {
1303       *secondary_errmsg = '\0';
1304       goto error;
1305     }
1306
1307   /* XXX - capture SIGTERM and close the capture, in case we're on a
1308      Linux 2.0[.x] system and you have to explicitly close the capture
1309      stream in order to turn promiscuous mode off?  We need to do that
1310      in other places as well - and I don't think that works all the
1311      time in any case, due to libpcap bugs. */
1312
1313     /* Well, we should be able to start capturing.
1314
1315        Sync out the capture file, so the header makes it to the file system,
1316        and send a "capture started successfully and capture file created"
1317        message to our parent so that they'll open the capture file and
1318        update its windows to indicate that we have a live capture in
1319        progress. */
1320     libpcap_dump_flush(ld.pdh, NULL);
1321     report_new_capture_file(capture_opts->save_file);
1322   }
1323
1324   /* initialize capture stop (and alike) conditions */
1325   init_capture_stop_conditions();
1326   /* create stop conditions */
1327   if (capture_opts->has_autostop_filesize)
1328     cnd_autostop_size =
1329         cnd_new(CND_CLASS_CAPTURESIZE,(long)capture_opts->autostop_filesize * 1024);
1330   if (capture_opts->has_autostop_duration)
1331     cnd_autostop_duration =
1332         cnd_new(CND_CLASS_TIMEOUT,(gint32)capture_opts->autostop_duration);
1333
1334   if (capture_opts->multi_files_on) {
1335       if (capture_opts->has_file_duration)
1336         cnd_file_duration =
1337             cnd_new(CND_CLASS_TIMEOUT, capture_opts->file_duration);
1338
1339       if (capture_opts->has_autostop_files)
1340         cnd_autostop_files =
1341             cnd_new(CND_CLASS_CAPTURESIZE, capture_opts->autostop_files);
1342   }
1343
1344   /* init the time values */
1345   start_time = TIME_GET();
1346   upd_time = TIME_GET();
1347
1348   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO, "Capture loop running!");
1349
1350   /* WOW, everything is prepared! */
1351   /* please fasten your seat belts, we will enter now the actual capture loop */
1352   while (ld.go) {
1353     /* dispatch incoming packets */
1354     inpkts = capture_loop_dispatch(capture_opts, &ld, errmsg, sizeof(errmsg));
1355
1356 #ifdef _WIN32
1357     /* any news from our parent (signal pipe)? -> just stop the capture */
1358     if (!signal_pipe_check_running()) {
1359       ld.go = FALSE;
1360     }
1361 #endif
1362
1363     if (inpkts > 0) {
1364       inpkts_to_sync_pipe += inpkts;
1365
1366       /* check capture size condition */
1367       if (cnd_autostop_size != NULL &&
1368           cnd_eval(cnd_autostop_size, (guint32)ld.bytes_written)){
1369         /* Capture size limit reached, do we have another file? */
1370         if (capture_opts->multi_files_on) {
1371           if (cnd_autostop_files != NULL &&
1372               cnd_eval(cnd_autostop_files, ++autostop_files)) {
1373              /* no files left: stop here */
1374             ld.go = FALSE;
1375             continue;
1376           }
1377
1378           /* Switch to the next ringbuffer file */
1379           if (ringbuf_switch_file(&ld.pdh, &capture_opts->save_file,
1380               &save_file_fd, &ld.bytes_written, &ld.err)) {
1381             /* File switch succeeded: reset the conditions */
1382             cnd_reset(cnd_autostop_size);
1383             if (cnd_file_duration) {
1384               cnd_reset(cnd_file_duration);
1385             }
1386             libpcap_dump_flush(ld.pdh, NULL);
1387             report_packet_count(inpkts_to_sync_pipe);
1388             inpkts_to_sync_pipe = 0;
1389             report_new_capture_file(capture_opts->save_file);
1390           } else {
1391             /* File switch failed: stop here */
1392             ld.go = FALSE;
1393             continue;
1394           }
1395         } else {
1396           /* single file, stop now */
1397           ld.go = FALSE;
1398           continue;
1399         }
1400       } /* cnd_autostop_size */
1401       if (capture_opts->output_to_pipe) {
1402         libpcap_dump_flush(ld.pdh, NULL);
1403       }
1404     } /* inpkts */
1405
1406     /* Only update once a second (Win32: 500ms) so as not to overload slow displays */
1407     cur_time = TIME_GET();
1408 #ifdef _WIN32
1409     if ( (cur_time - upd_time) > 500) {
1410 #else
1411     if (cur_time - upd_time > 0) {
1412 #endif
1413         upd_time = cur_time;
1414
1415       /*if (pcap_stats(pch, stats) >= 0) {
1416         *stats_known = TRUE;
1417       }*/
1418
1419       /* Let the parent process know. */
1420       if (inpkts_to_sync_pipe) {
1421         /* do sync here */
1422         libpcap_dump_flush(ld.pdh, NULL);
1423
1424         /* Send our parent a message saying we've written out "inpkts_to_sync_pipe"
1425            packets to the capture file. */
1426         report_packet_count(inpkts_to_sync_pipe);
1427
1428         inpkts_to_sync_pipe = 0;
1429       }
1430
1431       /* check capture duration condition */
1432       if (cnd_autostop_duration != NULL && cnd_eval(cnd_autostop_duration)) {
1433         /* The maximum capture time has elapsed; stop the capture. */
1434         ld.go = FALSE;
1435         continue;
1436       }
1437
1438       /* check capture file duration condition */
1439       if (cnd_file_duration != NULL && cnd_eval(cnd_file_duration)) {
1440         /* duration limit reached, do we have another file? */
1441         if (capture_opts->multi_files_on) {
1442           if (cnd_autostop_files != NULL &&
1443               cnd_eval(cnd_autostop_files, ++autostop_files)) {
1444             /* no files left: stop here */
1445             ld.go = FALSE;
1446             continue;
1447           }
1448
1449           /* Switch to the next ringbuffer file */
1450           if (ringbuf_switch_file(&ld.pdh, &capture_opts->save_file,
1451                                   &save_file_fd, &ld.bytes_written, &ld.err)) {
1452             /* file switch succeeded: reset the conditions */
1453             cnd_reset(cnd_file_duration);
1454             if(cnd_autostop_size)
1455               cnd_reset(cnd_autostop_size);
1456             libpcap_dump_flush(ld.pdh, NULL);
1457             report_packet_count(inpkts_to_sync_pipe);
1458             inpkts_to_sync_pipe = 0;
1459             report_new_capture_file(capture_opts->save_file);
1460           } else {
1461             /* File switch failed: stop here */
1462             ld.go = FALSE;
1463             continue;
1464           }
1465         } else {
1466           /* single file, stop now */
1467           ld.go = FALSE;
1468           continue;
1469         }
1470       } /* cnd_file_duration */
1471     }
1472
1473   } /* while (ld.go) */
1474
1475   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO, "Capture loop stopping ...");
1476
1477   /* delete stop conditions */
1478   if (cnd_file_duration != NULL)
1479     cnd_delete(cnd_file_duration);
1480   if (cnd_autostop_files != NULL)
1481     cnd_delete(cnd_autostop_files);
1482   if (cnd_autostop_size != NULL)
1483     cnd_delete(cnd_autostop_size);
1484   if (cnd_autostop_duration != NULL)
1485     cnd_delete(cnd_autostop_duration);
1486
1487   /* did we had a pcap (input) error? */
1488   if (ld.pcap_err) {
1489     g_snprintf(errmsg, sizeof(errmsg), "Error while capturing packets: %s",
1490       pcap_geterr(ld.pcap_h));
1491     report_capture_error(errmsg, please_report);
1492   }
1493     else if (ld.from_cap_pipe && ld.cap_pipe_err == PIPERR)
1494       report_capture_error(errmsg, "");
1495
1496   /* did we had an error while capturing? */
1497   if (ld.err == 0) {
1498     write_ok = TRUE;
1499   } else {
1500     capture_loop_get_errmsg(errmsg, sizeof(errmsg), capture_opts->save_file, ld.err,
1501                               FALSE);
1502     report_capture_error(errmsg, please_report);
1503     write_ok = FALSE;
1504   }
1505
1506   if (capture_opts->saving_to_file) {
1507     /* close the wiretap (output) file */
1508     close_ok = capture_loop_close_output(capture_opts, &ld, &err_close);
1509   } else
1510     close_ok = TRUE;
1511
1512   /* there might be packets not yet notified to the parent */
1513   /* (do this after closing the file, so all packets are already flushed) */
1514   if(inpkts_to_sync_pipe) {
1515     report_packet_count(inpkts_to_sync_pipe);
1516     inpkts_to_sync_pipe = 0;
1517   }
1518
1519   /* If we've displayed a message about a write error, there's no point
1520      in displaying another message about an error on close. */
1521   if (!close_ok && write_ok) {
1522     capture_loop_get_errmsg(errmsg, sizeof(errmsg), capture_opts->save_file, err_close,
1523                 TRUE);
1524     report_capture_error(errmsg, "");
1525   }
1526
1527   /*
1528    * XXX We exhibit different behaviour between normal mode and sync mode
1529    * when the pipe is stdin and not already at EOF.  If we're a child, the
1530    * parent's stdin isn't closed, so if the user starts another capture,
1531    * cap_pipe_open_live() will very likely not see the expected magic bytes and
1532    * will say "Unrecognized libpcap format".  On the other hand, in normal
1533    * mode, cap_pipe_open_live() will say "End of file on pipe during open".
1534    */
1535
1536   /* get packet drop statistics from pcap */
1537   if(ld.pcap_h != NULL) {
1538     g_assert(!ld.from_cap_pipe);
1539     /* Get the capture statistics, so we know how many packets were
1540        dropped. */
1541     if (pcap_stats(ld.pcap_h, stats) >= 0) {
1542       *stats_known = TRUE;
1543       /* Let the parent process know. */
1544       report_packet_drops(stats->ps_drop);
1545     } else {
1546       g_snprintf(errmsg, sizeof(errmsg),
1547                 "Can't get packet-drop statistics: %s",
1548                 pcap_geterr(ld.pcap_h));
1549       report_capture_error(errmsg, please_report);
1550     }
1551   }
1552
1553   /* close the input file (pcap or capture pipe) */
1554   capture_loop_close_input(&ld);
1555
1556   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO, "Capture loop stopped!");
1557
1558   /* ok, if the write and the close were successful. */
1559   return write_ok && close_ok;
1560
1561 error:
1562   if (capture_opts->multi_files_on) {
1563     /* cleanup ringbuffer */
1564     ringbuf_error_cleanup();
1565   } else {
1566     /* We can't use the save file, and we have no FILE * for the stream
1567        to close in order to close it, so close the FD directly. */
1568     if(save_file_fd != -1) {
1569       eth_close(save_file_fd);
1570     }
1571
1572     /* We couldn't even start the capture, so get rid of the capture
1573        file. */
1574     if(capture_opts->save_file != NULL) {
1575       eth_unlink(capture_opts->save_file);
1576       g_free(capture_opts->save_file);
1577     }
1578   }
1579   capture_opts->save_file = NULL;
1580   if (cfilter_error)
1581     report_cfilter_error(capture_opts->cfilter, errmsg);
1582   else
1583     report_capture_error(errmsg, secondary_errmsg);
1584
1585   /* close the input file (pcap or cap_pipe) */
1586   capture_loop_close_input(&ld);
1587
1588   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO, "Capture loop stopped with error");
1589
1590   return FALSE;
1591 }
1592
1593
1594 void capture_loop_stop(void)
1595 {
1596 #ifdef HAVE_PCAP_BREAKLOOP
1597   if(ld.pcap_h != NULL)
1598     pcap_breakloop(ld.pcap_h);
1599 #endif
1600   ld.go = FALSE;
1601 }
1602
1603
1604 static void
1605 capture_loop_get_errmsg(char *errmsg, int errmsglen, const char *fname,
1606                           int err, gboolean is_close)
1607 {
1608   switch (err) {
1609
1610   case ENOSPC:
1611     g_snprintf(errmsg, errmsglen,
1612                 "Not all the packets could be written to the file"
1613                 " to which the capture was being saved\n"
1614                 "(\"%s\") because there is no space left on the file system\n"
1615                 "on which that file resides.",
1616                 fname);
1617     break;
1618
1619 #ifdef EDQUOT
1620   case EDQUOT:
1621     g_snprintf(errmsg, errmsglen,
1622                 "Not all the packets could be written to the file"
1623                 " to which the capture was being saved\n"
1624                 "(\"%s\") because you are too close to, or over,"
1625                 " your disk quota\n"
1626                 "on the file system on which that file resides.",
1627                 fname);
1628   break;
1629 #endif
1630
1631   case WTAP_ERR_CANT_CLOSE:
1632     g_snprintf(errmsg, errmsglen,
1633                 "The file to which the capture was being saved"
1634                 " couldn't be closed for some unknown reason.");
1635     break;
1636
1637   case WTAP_ERR_SHORT_WRITE:
1638     g_snprintf(errmsg, errmsglen,
1639                 "Not all the packets could be written to the file"
1640                 " to which the capture was being saved\n"
1641                 "(\"%s\").",
1642                 fname);
1643     break;
1644
1645   default:
1646     if (is_close) {
1647       g_snprintf(errmsg, errmsglen,
1648                 "The file to which the capture was being saved\n"
1649                 "(\"%s\") could not be closed: %s.",
1650                 fname, wtap_strerror(err));
1651     } else {
1652       g_snprintf(errmsg, errmsglen,
1653                 "An error occurred while writing to the file"
1654                 " to which the capture was being saved\n"
1655                 "(\"%s\"): %s.",
1656                 fname, wtap_strerror(err));
1657     }
1658     break;
1659   }
1660 }
1661
1662
1663 /* one packet was captured, process it */
1664 static void
1665 capture_loop_packet_cb(u_char *user, const struct pcap_pkthdr *phdr,
1666   const u_char *pd)
1667 {
1668   loop_data *ld = (loop_data *) user;
1669   int err;
1670
1671   /* if the user told us to stop after x packets, do we have enough? */
1672   ld->packet_count++;
1673   if ((ld->packet_max > 0) && (ld->packet_count >= ld->packet_max))
1674   {
1675      ld->go = FALSE;
1676   }
1677
1678   if (ld->pdh) {
1679     /* We're supposed to write the packet to a file; do so.
1680        If this fails, set "ld->go" to FALSE, to stop the capture, and set
1681        "ld->err" to the error. */
1682     if (!libpcap_write_packet(ld->pdh, phdr, pd, &ld->bytes_written, &err)) {
1683       ld->go = FALSE;
1684       ld->err = err;
1685     }
1686   }
1687 }
1688
1689 #endif /* HAVE_LIBPCAP */