Make the "Save only marked frames" button in the "Save As..." dialog box
[obnox/wireshark/wip.git] / capture.c
index 85b8aada0e6926225542fd3e382d968388e8591a..413b1750edcdf9554fe0ea2d8eb7e960a13a5fda 100644 (file)
--- a/capture.c
+++ b/capture.c
@@ -1,12 +1,11 @@
 /* capture.c
  * Routines for packet capture windows
  *
- * $Id: capture.c,v 1.152 2001/06/15 01:36:46 guy Exp $
+ * $Id: capture.c,v 1.163 2001/12/04 08:25:55 guy Exp $
  *
  * Ethereal - Network traffic analyzer
- * By Gerald Combs <gerald@zing.org>
+ * By Gerald Combs <gerald@ethereal.com>
  * Copyright 1998 Gerald Combs
- *
  * 
  * This program is free software; you can redistribute it and/or
  * modify it under the terms of the GNU General Public License
 #include "file.h"
 #include "capture.h"
 #include "util.h"
+#include "pcap-util.h"
 #include "simple_dialog.h"
 #include "prefs.h"
 #include "globals.h"
+#include "conditions.h"
+#include "capture_stop_conditions.h"
+#include "ringbuffer.h"
 
 #include "wiretap/libpcap.h"
 #include "wiretap/wtap.h"
 
+#include "packet-atalk.h"
 #include "packet-clip.h"
 #include "packet-eth.h"
 #include "packet-fddi.h"
 #include "packet-tr.h"
 #include "packet-ieee80211.h"
 #include "packet-chdlc.h"
+#include "packet-prism.h"
+
+#ifdef WIN32
+#include "capture-wpcap.h"
+#endif
 
 static int sync_pipe[2]; /* used to sync father */
 enum PIPES { READ, WRITE }; /* Constants 0 and 1 for READ and WRITE */
@@ -171,9 +180,9 @@ static guint cap_input_id;
  * Indications sent out on the sync pipe.
  */
 #define SP_CAPSTART    ';'     /* capture start message */
-#define SP_PACKET_COUNT        '*'     /* count of packets captured since last message */
-#define SP_ERROR_MSG   '!'     /* length of error message that follows */
-#define SP_DROPS       '#'     /* count of packets dropped in capture */
+#define SP_PACKET_COUNT        '*'     /* followed by count of packets captured since last message */
+#define SP_ERROR_MSG   '!'     /* followed by length of error message that follows */
+#define SP_DROPS       '#'     /* followed by count of packets dropped in capture */
 
 #ifdef _WIN32
 static guint cap_timer_id;
@@ -279,8 +288,13 @@ do_capture(char *capfile_name)
   struct pcap_stat stats;
 
   if (capfile_name != NULL) {
-    /* Try to open/create the specified file for use as a capture buffer. */
-    cfile.save_file_fd = open(capfile_name, O_RDWR|O_BINARY|O_TRUNC|O_CREAT, 0600);
+    if (cfile.ringbuffer_on) {
+      /* ringbuffer is enabled */
+      cfile.save_file_fd = ringbuf_init(capfile_name, cfile.ringbuffer_num_files);
+    } else {
+      /* Try to open/create the specified file for use as a capture buffer. */
+      cfile.save_file_fd = open(capfile_name, O_RDWR|O_BINARY|O_TRUNC|O_CREAT, 0600);
+    }
     is_tempfile = FALSE;
   } else {
     /* Choose a random name for the capture buffer */
@@ -294,6 +308,9 @@ do_capture(char *capfile_name)
        "The temporary file to which the capture would be saved (\"%s\")"
        "could not be opened: %s.", capfile_name, strerror(errno));
     } else {
+      if (cfile.ringbuffer_on) {
+        ringbuf_error_cleanup();
+      }
       simple_dialog(ESD_TYPE_CRIT, NULL,
        file_open_error_message(errno, TRUE), capfile_name);
     }
@@ -305,7 +322,9 @@ do_capture(char *capfile_name)
 
   if (prefs.capture_real_time) {       /* do the capture in a child process */
     char ssnap[24];
-    char scount[24];   /* need a constant for len of numbers */
+    char scount[24];                   /* need a constant for len of numbers */
+    char sautostop_filesize[24];       /* need a constant for len of numbers */
+    char sautostop_duration[24];       /* need a constant for len of numbers */
     char save_file_fd[24];
     char errmsg[1024+1];
     int error;
@@ -344,6 +363,14 @@ do_capture(char *capfile_name)
     sprintf(ssnap,"%d",cfile.snap);
     argv = add_arg(argv, &argc, ssnap);
 
+    argv = add_arg(argv, &argc, "-a");
+    sprintf(sautostop_filesize,"filesize:%d",cfile.autostop_filesize);
+    argv = add_arg(argv, &argc, sautostop_filesize);
+
+    argv = add_arg(argv, &argc, "-a");
+    sprintf(sautostop_duration,"duration:%d",cfile.autostop_duration);
+    argv = add_arg(argv, &argc, sautostop_duration);
+
     if (!prefs.capture_prom_mode)
       argv = add_arg(argv, &argc, "-p");
 
@@ -634,7 +661,11 @@ do_capture(char *capfile_name)
     }
     /* We're not doing a capture any more, so we don't have a save
        file. */
-    g_free(cfile.save_file);
+    if (cfile.ringbuffer_on) {
+      ringbuf_free();
+    } else {
+      g_free(cfile.save_file);
+    }
     cfile.save_file = NULL;
   }
 }
@@ -1059,7 +1090,7 @@ pipe_open_live(char *pipename, struct pcap_hdr *hdr, loop_data *ld, char *ebuf)
     close(fd);
     return -1;
   }
-  while (bytes_read < sizeof(struct pcap_hdr))
+  while ((unsigned) bytes_read < sizeof(struct pcap_hdr))
   {
     b = read(fd, ((char *)&hdr)+bytes_read, sizeof(struct pcap_hdr) - bytes_read);
     if (b <= 0) {
@@ -1126,7 +1157,7 @@ pipe_dispatch(int fd, loop_data *ld, struct pcap_hdr *hdr)
     ld->go = FALSE;
     return 0;
   }
-  while (bytes_read < rechdr.hdr.incl_len)
+  while ((unsigned) bytes_read < rechdr.hdr.incl_len)
   {
     b = read(fd, pd+bytes_read, rechdr.hdr.incl_len - bytes_read);
     if (b <= 0) {
@@ -1144,41 +1175,41 @@ pipe_dispatch(int fd, loop_data *ld, struct pcap_hdr *hdr)
   whdr.pkt_encap = ld->linktype;
   wtap_dump(ld->pdh, &whdr, NULL, pd, &err);
 
-  /* Set the initial payload to the packet length, and the initial
-     captured payload to the capture length (other protocols may
-     reduce them if their headers say they're less). */
-  pi.len = whdr.len;
-  pi.captured_len = whdr.caplen;
-    
   /* update capture statistics */
   switch (ld->linktype) {
     case WTAP_ENCAP_ETHERNET:
-      capture_eth(pd, 0, &ld->counts);
+      capture_eth(pd, 0, whdr.caplen, &ld->counts);
       break;
     case WTAP_ENCAP_FDDI:
     case WTAP_ENCAP_FDDI_BITSWAPPED:
-      capture_fddi(pd, &ld->counts);
+      capture_fddi(pd, whdr.caplen, &ld->counts);
+      break;
+    case WTAP_ENCAP_PRISM_HEADER:
+      capture_prism(pd, 0, whdr.caplen, &ld->counts);
       break;
     case WTAP_ENCAP_TOKEN_RING:
-      capture_tr(pd, 0, &ld->counts);
+      capture_tr(pd, 0, whdr.caplen, &ld->counts);
       break;
     case WTAP_ENCAP_NULL:
-      capture_null(pd, &ld->counts);
+      capture_null(pd, whdr.caplen, &ld->counts);
       break;
     case WTAP_ENCAP_PPP:
-      capture_ppp_hdlc(pd, 0, &ld->counts);
+      capture_ppp_hdlc(pd, 0, whdr.caplen, &ld->counts);
       break;
     case WTAP_ENCAP_RAW_IP:
-      capture_raw(pd, &ld->counts);
+      capture_raw(pd, whdr.caplen, &ld->counts);
       break;
     case WTAP_ENCAP_LINUX_ATM_CLIP:
-      capture_clip(pd, &ld->counts);
+      capture_clip(pd, whdr.caplen, &ld->counts);
       break;
     case WTAP_ENCAP_IEEE_802_11:
-      capture_ieee80211(pd, 0, &ld->counts);
+      capture_ieee80211(pd, 0, whdr.caplen, &ld->counts);
       break;
     case WTAP_ENCAP_CHDLC:
-      capture_chdlc(pd, 0, &ld->counts);
+      capture_chdlc(pd, 0, whdr.caplen, &ld->counts);
+      break;
+    case WTAP_ENCAP_LOCALTALK:
+      capture_llap(pd, whdr.caplen, &ld->counts);
       break;
     /* XXX - FreeBSD may append 4-byte ATM pseudo-header to DLT_ATM_RFC1483,
        with LLC header following; we should implement it at some
@@ -1204,13 +1235,19 @@ capture(gboolean *stats_known, struct pcap_stat *stats)
   pcap_t     *pch;
   int         pcap_encap;
   int         snaplen;
-  gchar       err_str[PCAP_ERRBUF_SIZE], label_str[64];
+  gchar       open_err_str[PCAP_ERRBUF_SIZE];
+  gchar       lookup_net_err_str[PCAP_ERRBUF_SIZE];
+  gchar       label_str[64];
   bpf_u_int32 netnum, netmask;
   struct bpf_program fcode;
   time_t      upd_time, cur_time;
-  int         err, inpkts, i;
+  int         err, inpkts;
+  condition *cnd_stop_capturesize;
+  condition *cnd_stop_timeout;
+  unsigned int i;
   static const char capstart_msg = SP_CAPSTART;
   char        errmsg[4096+1];
+  gboolean    dump_ok;
 #ifndef _WIN32
   static const char ppamsg[] = "can't find PPA for ";
   char       *libpcap_warn;
@@ -1283,9 +1320,13 @@ capture(gboolean *stats_known, struct pcap_stat *stats)
   /* We haven't yet gotten the capture statistics. */
   *stats_known      = FALSE;
 
-  /* Open the network interface to capture from it. */
+  /* Open the network interface to capture from it.
+     Some versions of libpcap may put warnings into the error buffer
+     if they succeed; to tell if that's happened, we have to clear
+     the error buffer, and check if it's still a null string.  */
+  open_err_str[0] = '\0';
   pch = pcap_open_live(cfile.iface, cfile.snap, prefs.capture_prom_mode,
-                       CAP_READ_TIMEOUT, err_str);
+                      CAP_READ_TIMEOUT, open_err_str);
 
   if (pch == NULL) {
 #ifdef _WIN32
@@ -1308,11 +1349,11 @@ capture(gboolean *stats_known, struct pcap_stat *stats)
        "\n"
        "Note that the driver Ethereal uses for packet capture on Windows\n"
        "doesn't support capturing on PPP/WAN interfaces in Windows NT/2000.\n",
-       err_str);
+       open_err_str);
     goto error;
 #else
     /* try to open cfile.iface as a pipe */
-    pipe_fd = pipe_open_live(cfile.iface, &hdr, &ld, err_str);
+    pipe_fd = pipe_open_live(cfile.iface, &hdr, &ld, open_err_str);
 
     if (pipe_fd == -1) {
       /* Well, we couldn't start the capture.
@@ -1329,7 +1370,7 @@ capture(gboolean *stats_known, struct pcap_stat *stats)
         of libpcap that properly handles HP-UX (libpcap 0.6.x and later
         versions, which properly handle HP-UX, say "can't find /dev/dlpi
         PPA for XXX" rather than "can't find PPA for XXX"). */
-      if (strncmp(err_str, ppamsg, sizeof ppamsg - 1) == 0)
+      if (strncmp(open_err_str, ppamsg, sizeof ppamsg - 1) == 0)
        libpcap_warn =
          "\n\n"
          "You are running Ethereal with a version of the libpcap library\n"
@@ -1346,7 +1387,7 @@ capture(gboolean *stats_known, struct pcap_stat *stats)
       snprintf(errmsg, sizeof errmsg,
          "The capture session could not be initiated (%s).\n"
          "Please check to make sure you have sufficient permissions, and that\n"
-         "you have the proper interface or pipe specified.%s", err_str,
+         "you have the proper interface or pipe specified.%s", open_err_str,
          libpcap_warn);
       goto error;
     }
@@ -1356,7 +1397,7 @@ capture(gboolean *stats_known, struct pcap_stat *stats)
   /* capture filters only work on real interfaces */
   if (cfile.cfilter && !ld.from_pipe) {
     /* A capture filter was specified; set it up. */
-    if (pcap_lookupnet (cfile.iface, &netnum, &netmask, err_str) < 0) {
+    if (pcap_lookupnet(cfile.iface, &netnum, &netmask, lookup_net_err_str) < 0) {
       /*
        * Well, we can't get the netmask for this interface; it's used
        * only for filters that check for broadcast IP addresses, so
@@ -1388,7 +1429,7 @@ capture(gboolean *stats_known, struct pcap_stat *stats)
   } else
 #endif
   {
-    pcap_encap = pcap_datalink(pch);
+    pcap_encap = get_pcap_linktype(pch, cfile.iface);
     snaplen = pcap_snapshot(pch);
   }
   ld.linktype = wtap_pcap_encap_to_wtap_encap(pcap_encap);
@@ -1398,8 +1439,13 @@ capture(gboolean *stats_known, struct pcap_stat *stats)
        " that Ethereal doesn't support (data link type %d).", pcap_encap);
     goto error;
   }
-  ld.pdh = wtap_dump_fdopen(cfile.save_file_fd, WTAP_FILE_PCAP,
+  if (cfile.ringbuffer_on) {
+    ld.pdh = ringbuf_init_wtap_dump_fdopen(WTAP_FILE_PCAP, ld.linktype, 
+      snaplen, &err);
+  } else {
+    ld.pdh = wtap_dump_fdopen(cfile.save_file_fd, WTAP_FILE_PCAP,
       ld.linktype, snaplen, &err);
+  }
 
   if (ld.pdh == NULL) {
     /* We couldn't set up to write to the capture file. */
@@ -1432,6 +1478,11 @@ capture(gboolean *stats_known, struct pcap_stat *stats)
     goto error;
   }
 
+  /* Does "open_err_str" contain a non-empty string?  If so, "pcap_open_live()"
+     returned a warning; print it, but keep capturing. */
+  if (open_err_str[0] != '\0')
+    g_warning("%s.", open_err_str);
+
   /* XXX - capture SIGTERM and close the capture, in case we're on a
      Linux 2.0[.x] system and you have to explicitly close the capture
      stream in order to turn promiscuous mode off?  We need to do that
@@ -1519,6 +1570,14 @@ capture(gboolean *stats_known, struct pcap_stat *stats)
    */
   signal(SIGUSR1, stop_capture);
 #endif
+  /* initialize capture stop conditions */ 
+  init_capture_stop_conditions();
+  /* create stop conditions */
+  cnd_stop_capturesize = 
+    cnd_new(CND_CLASS_CAPTURESIZE,(long)cfile.autostop_filesize * 1000); 
+  cnd_stop_timeout = 
+    cnd_new(CND_CLASS_TIMEOUT,(gint32)cfile.autostop_duration);
+
   while (ld.go) {
     while (gtk_events_pending()) gtk_main_iteration();
 
@@ -1588,6 +1647,28 @@ capture(gboolean *stats_known, struct pcap_stat *stats)
     }
     if (inpkts > 0)
       ld.sync_packets += inpkts;
+    /* check capture stop conditons */
+    if (cnd_eval(cnd_stop_timeout) == TRUE) {
+      /* The specified capture time has elapsed; stop the capture. */
+      ld.go = FALSE;
+    } else if ((cnd_eval(cnd_stop_capturesize, 
+                  (guint32)wtap_get_bytes_dumped(ld.pdh))) == TRUE){
+      /* Capture file reached its maximum size. */
+      if (cfile.ringbuffer_on) {
+        /* Switch to the next ringbuffer file */
+        if (ringbuf_switch_file(&cfile, &ld.pdh, &ld.err)) {
+          /* File switch succeeded: reset the condition */
+          cnd_reset(cnd_stop_capturesize);
+        } else {
+          /* File switch failed: stop here */
+          ld.go = FALSE;
+          continue;
+        }
+      } else {
+        /* no ringbuffer - just stop */
+        ld.go = FALSE;
+      }
+    }
     /* Only update once a second so as not to overload slow displays */
     cur_time = time(NULL);
     if (cur_time > upd_time) {
@@ -1619,6 +1700,10 @@ capture(gboolean *stats_known, struct pcap_stat *stats)
     }
   }
     
+  /* delete stop conditions */
+  cnd_delete(cnd_stop_capturesize);
+  cnd_delete(cnd_stop_timeout);
+
   if (ld.err != 0) {
     get_capture_file_io_error(errmsg, sizeof(errmsg), cfile.save_file, ld.err,
                              FALSE);
@@ -1635,9 +1720,18 @@ capture(gboolean *stats_known, struct pcap_stat *stats)
     /* A write failed, so we've already told the user there's a problem;
        if the close fails, there's no point in telling them about that
        as well. */
-    wtap_dump_close(ld.pdh, &err);
-  } else {
-    if (!wtap_dump_close(ld.pdh, &err)) {
+    if (cfile.ringbuffer_on) {
+      ringbuf_wtap_dump_close(&cfile, &err);
+    } else {
+      wtap_dump_close(ld.pdh, &err);
+    }
+   } else {
+    if (cfile.ringbuffer_on) {
+      dump_ok = ringbuf_wtap_dump_close(&cfile, &err);
+    } else {
+      dump_ok = wtap_dump_close(ld.pdh, &err);
+    }
+    if (!dump_ok) {
       get_capture_file_io_error(errmsg, sizeof(errmsg), cfile.save_file, err,
                                TRUE);
       if (capture_child) {
@@ -1695,14 +1789,19 @@ capture(gboolean *stats_known, struct pcap_stat *stats)
   return TRUE;
 
 error:
-  /* We can't use the save file, and we have no wtap_dump stream
-     to close in order to close it, so close the FD directly. */
-  close(cfile.save_file_fd);
-
-  /* We couldn't even start the capture, so get rid of the capture
-     file. */
-  unlink(cfile.save_file); /* silently ignore error */
-  g_free(cfile.save_file);
+  if (cfile.ringbuffer_on) {
+    /* cleanup ringbuffer */
+    ringbuf_error_cleanup();
+  } else {
+    /* We can't use the save file, and we have no wtap_dump stream
+       to close in order to close it, so close the FD directly. */
+    close(cfile.save_file_fd);
+
+    /* We couldn't even start the capture, so get rid of the capture
+       file. */
+    unlink(cfile.save_file); /* silently ignore error */
+    g_free(cfile.save_file);
+  }
   cfile.save_file = NULL;
   if (capture_child) {
     /* This is the child process for a sync mode capture.
@@ -1857,37 +1956,37 @@ capture_pcap_cb(u_char *user, const struct pcap_pkthdr *phdr,
      }
   }
 
-  /* Set the initial payload to the packet length, and the initial
-     captured payload to the capture length (other protocols may
-     reduce them if their headers say they're less). */
-  pi.len = phdr->len;
-  pi.captured_len = phdr->caplen;
-    
   switch (ld->linktype) {
     case WTAP_ENCAP_ETHERNET:
-      capture_eth(pd, 0, &ld->counts);
+      capture_eth(pd, 0, phdr->len, &ld->counts);
       break;
     case WTAP_ENCAP_FDDI:
     case WTAP_ENCAP_FDDI_BITSWAPPED:
-      capture_fddi(pd, &ld->counts);
+      capture_fddi(pd, phdr->len, &ld->counts);
+      break;
+    case WTAP_ENCAP_PRISM_HEADER:
+      capture_prism(pd, 0, phdr->len, &ld->counts);
       break;
     case WTAP_ENCAP_TOKEN_RING:
-      capture_tr(pd, 0, &ld->counts);
+      capture_tr(pd, 0, phdr->len, &ld->counts);
       break;
     case WTAP_ENCAP_NULL:
-      capture_null(pd, &ld->counts);
+      capture_null(pd, phdr->len, &ld->counts);
       break;
     case WTAP_ENCAP_PPP:
-      capture_ppp_hdlc(pd, 0, &ld->counts);
+      capture_ppp_hdlc(pd, 0, phdr->len, &ld->counts);
       break;
     case WTAP_ENCAP_RAW_IP:
-      capture_raw(pd, &ld->counts);
+      capture_raw(pd, phdr->len, &ld->counts);
       break;
     case WTAP_ENCAP_SLL:
-      capture_sll(pd, &ld->counts);
+      capture_sll(pd, phdr->len, &ld->counts);
       break;
     case WTAP_ENCAP_LINUX_ATM_CLIP:
-      capture_clip(pd, &ld->counts);
+      capture_clip(pd, phdr->len, &ld->counts);
+      break;
+    case WTAP_ENCAP_LOCALTALK:
+      capture_llap(pd, phdr->len, &ld->counts);
       break;
     /* XXX - FreeBSD may append 4-byte ATM pseudo-header to DLT_ATM_RFC1483,
        with LLC header following; we should implement it at some