When dumpcap is run to get an interface list, interface capabilities, or
[obnox/wireshark/wip.git] / capture.c
index b0efea16724d09c573c34e46c109b24158f106fd..ad3bb3458ba347c1c39d3f68bc91ec6a98073eef 100644 (file)
--- a/capture.c
+++ b/capture.c
 #include <unistd.h>
 #endif
 
-#include <stdlib.h>
+#include <stdio.h>
 #include <string.h>
-#include <ctype.h>
-
-#ifdef HAVE_FCNTL_H
-#include <fcntl.h>
-#endif
-
-#ifdef HAVE_SYS_TYPES_H
-#include <sys/types.h>
-#endif
-
-#ifdef HAVE_NETINET_IN_H
-#include <netinet/in.h>
-#endif
-
-#ifdef HAVE_NETDB_H
-#include <netdb.h>
-#endif
-
-#ifdef HAVE_ARPA_INET_H
-#include <arpa/inet.h>
-#endif
-
-#ifdef HAVE_SYS_SOCKET_H
-#include <sys/socket.h>         /* needed to define AF_ values on UNIX */
-#endif
-
-#ifdef HAVE_WINSOCK2_H
-#include <winsock2.h>           /* needed to define AF_ values on Windows */
-#endif
-
-#ifdef NEED_INET_V6DEFS_H
-# include "inet_v6defs.h"
-#endif
-
-#include <signal.h>
-#include <errno.h>
 
 #include <glib.h>
 
 #include <epan/packet.h>
 #include <epan/dfilter/dfilter.h>
-#include <epan/ws_strsplit.h>
 #include "file.h"
 #include "capture.h"
+#include "capture_ifinfo.h"
 #include "capture_sync.h"
 #include "capture_info.h"
 #include "capture_ui_utils.h"
 #include "util.h"
 #include "capture-pcap-util.h"
-#include "alert_box.h"
 #include "simple_dialog.h"
 #include <epan/prefs.h>
 
@@ -91,7 +54,7 @@
 #include "capture-wpcap.h"
 #endif
 #include "ui_util.h"
-#include "file_util.h"
+#include "wsutil/file_util.h"
 #include "log.h"
 
 typedef struct if_stat_cache_item_s {
@@ -105,6 +68,62 @@ struct if_stat_cache_s {
     GList *cache_list;  /* List of if_stat_chache_entry_t */
 };
 
+/* this callback mechanism should possibly be replaced by the g_signal_...() stuff (if I only would know how :-) */
+typedef struct {
+    capture_callback_t cb_fct;
+    gpointer user_data;
+} capture_callback_data_t;
+
+static GList *capture_callbacks = NULL;
+
+static void
+capture_callback_invoke(int event, capture_options *capture_opts)
+{
+    capture_callback_data_t *cb;
+    GList *cb_item = capture_callbacks;
+
+    /* there should be at least one interested */
+    g_assert(cb_item != NULL);
+
+    while(cb_item != NULL) {
+        cb = cb_item->data;
+        cb->cb_fct(event, capture_opts, cb->user_data);
+        cb_item = g_list_next(cb_item);
+    }
+}
+
+
+void
+capture_callback_add(capture_callback_t func, gpointer user_data)
+{
+    capture_callback_data_t *cb;
+
+    cb = g_malloc(sizeof(capture_callback_data_t));
+    cb->cb_fct = func;
+    cb->user_data = user_data;
+
+    capture_callbacks = g_list_append(capture_callbacks, cb);
+}
+
+void
+capture_callback_remove(capture_callback_t func)
+{
+    capture_callback_data_t *cb;
+    GList *cb_item = capture_callbacks;
+
+    while(cb_item != NULL) {
+        cb = cb_item->data;
+        if(cb->cb_fct == func) {
+            capture_callbacks = g_list_remove(capture_callbacks, cb);
+            g_free(cb);
+            return;
+        }
+        cb_item = g_list_next(cb_item);
+    }
+
+    g_assert_not_reached();
+}
+
 /**
  * Start a capture.
  *
@@ -114,7 +133,7 @@ gboolean
 capture_start(capture_options *capture_opts)
 {
   gboolean ret;
-
+  GString *source = g_string_new("");
 
   /* close the currently loaded capture file */
   cf_close(capture_opts->cf);
@@ -124,6 +143,13 @@ capture_start(capture_options *capture_opts)
 
   g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_MESSAGE, "Capture Start ...");
 
+  g_string_printf(source, "%s", get_iface_description(capture_opts));
+  if(capture_opts->cfilter && capture_opts->cfilter[0]) {
+    g_string_append_printf(source, " (%s)", capture_opts->cfilter);
+  }
+  cf_set_tempfile_source(capture_opts->cf, source->str);
+  g_string_free(source, TRUE);
+
   /* try to start the capture child process */
   ret = sync_pipe_start(capture_opts);
   if(!ret) {
@@ -136,14 +162,14 @@ capture_start(capture_options *capture_opts)
       capture_opts->state = CAPTURE_STOPPED;
   } else {
       /* the capture child might not respond shortly after bringing it up */
-      /* (especially it will block, if no input coming from an input capture pipe (e.g. mkfifo) is coming in) */
+      /* (for example: it will block if no input arrives from an input capture pipe (e.g. mkfifo)) */
 
-      /* to prevent problems, bring the main GUI into "capture mode" right after successfully */
-      /* spawn/exec the capture child, without waiting for any response from it */
-      cf_callback_invoke(cf_cb_live_capture_prepared, capture_opts);
+      /* to prevent problems, bring the main GUI into "capture mode" right after a successful */
+      /* spawn/exec of the capture child, without waiting for any response from it */
+      capture_callback_invoke(capture_cb_capture_prepared, capture_opts);
 
       if(capture_opts->show_info)
-        capture_info_open(capture_opts->iface);
+        capture_info_open(capture_opts);
   }
 
   return ret;
@@ -155,7 +181,7 @@ capture_stop(capture_options *capture_opts)
 {
   g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_MESSAGE, "Capture Stop ...");
 
-  cf_callback_invoke(cf_cb_live_capture_stopping, capture_opts);
+  capture_callback_invoke(capture_cb_capture_stopping, capture_opts);
 
   /* stop the capture child gracefully */
   sync_pipe_stop(capture_opts);
@@ -183,23 +209,21 @@ capture_kill_child(capture_options *capture_opts)
 
 
 
-/* We've succeeded a (non real-time) capture, try to read it into a new capture file */
+/* We've succeeded in doing a (non real-time) capture; try to read it into a new capture file */
 static gboolean
 capture_input_read_all(capture_options *capture_opts, gboolean is_tempfile, gboolean drops_known,
 guint32 drops)
 {
   int err;
 
-
   /* Capture succeeded; attempt to open the capture file. */
   if (cf_open(capture_opts->cf, capture_opts->save_file, is_tempfile, &err) != CF_OK) {
-    /* We're not doing a capture any more, so we don't have a save
-       file. */
+    /* We're not doing a capture any more, so we don't have a save file. */
     return FALSE;
   }
-
+  
   /* Set the read filter to NULL. */
-  /* XXX - this is odd here, try to put it somewhere, where it fits better */
+  /* XXX - this is odd here; try to put it somewhere where it fits better */
   cf_set_rfcode(capture_opts->cf, NULL);
 
   /* Get the packet-drop statistics.
@@ -235,7 +259,7 @@ guint32 drops)
   }
 
   /* read in the packet data */
-  switch (cf_read(capture_opts->cf)) {
+  switch (cf_read(capture_opts->cf, FALSE)) {
 
   case CF_READ_OK:
   case CF_READ_ERROR:
@@ -251,7 +275,7 @@ guint32 drops)
     return FALSE;
   }
 
-  /* if we didn't captured even a single packet, close the file again */
+  /* if we didn't capture even a single packet, close the file again */
   if(cf_get_packet_count(capture_opts->cf) == 0 && !capture_opts->restart) {
     simple_dialog(ESD_TYPE_INFO, ESD_BTN_OK,
 "%sNo packets captured!%s\n"
@@ -283,7 +307,6 @@ capture_input_new_file(capture_options *capture_opts, gchar *new_file)
   gboolean is_tempfile;
   int  err;
 
-
   if(capture_opts->state == CAPTURE_PREPARING) {
     g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_MESSAGE, "Capture started!");
   }
@@ -293,10 +316,10 @@ capture_input_new_file(capture_options *capture_opts, gchar *new_file)
 
   /* free the old filename */
   if(capture_opts->save_file != NULL) {
-    /* we start a new capture file, close the old one (if we had one before) */
+    /* we start a new capture file, close the old one (if we had one before). */
     /* (we can only have an open capture file in real_time_mode!) */
     if( ((capture_file *) capture_opts->cf)->state != FILE_CLOSED) {
-        cf_callback_invoke(cf_cb_live_capture_update_finished, capture_opts->cf);
+        capture_callback_invoke(capture_cb_capture_update_finished, capture_opts);
         cf_finish_tail(capture_opts->cf, &err);
         cf_close(capture_opts->cf);
     }
@@ -304,7 +327,7 @@ capture_input_new_file(capture_options *capture_opts, gchar *new_file)
     is_tempfile = FALSE;
     cf_set_tempfile(capture_opts->cf, FALSE);
   } else {
-    /* we didn't had a save_file before, must be a tempfile */
+    /* we didn't have a save_file before; must be a tempfile */
     is_tempfile = TRUE;
     cf_set_tempfile(capture_opts->cf, TRUE);
   }
@@ -324,7 +347,6 @@ capture_input_new_file(capture_options *capture_opts, gchar *new_file)
       g_free(capture_opts->save_file);
       capture_opts->save_file = NULL;
       return FALSE;
-      break;
     }
   }
 
@@ -334,9 +356,9 @@ capture_input_new_file(capture_options *capture_opts, gchar *new_file)
   }
 
   if(capture_opts->real_time_mode) {
-    cf_callback_invoke(cf_cb_live_capture_update_started, capture_opts);
+    capture_callback_invoke(capture_cb_capture_update_started, capture_opts);
   } else {
-    cf_callback_invoke(cf_cb_live_capture_fixed_started, capture_opts);
+    capture_callback_invoke(capture_cb_capture_fixed_started, capture_opts);
   }
   capture_opts->state = CAPTURE_RUNNING;
 
@@ -364,7 +386,7 @@ capture_input_new_packets(capture_options *capture_opts, int to_read)
          file.
 
          XXX - abort on a read error? */
-         cf_callback_invoke(cf_cb_live_capture_update_continue, capture_opts->cf);
+         capture_callback_invoke(capture_cb_capture_update_continue, capture_opts);
       break;
 
     case CF_READ_ABORTED:
@@ -374,14 +396,14 @@ capture_input_new_packets(capture_options *capture_opts, int to_read)
       break;
     }
   } else {
-    /* increase capture file packet counter by the number or incoming packets */
+    /* increase the capture file packet counter by the number of incoming packets */
     cf_set_packet_count(capture_opts->cf,
         cf_get_packet_count(capture_opts->cf) + to_read);
 
-    cf_callback_invoke(cf_cb_live_capture_fixed_continue, capture_opts->cf);
+    capture_callback_invoke(capture_cb_capture_fixed_continue, capture_opts);
   }
 
-  /* update the main window, so we get events (e.g. from the stop toolbar button) */
+  /* update the main window so we get events (e.g. from the stop toolbar button) */
   main_window_update();
 
   if(capture_opts->show_info)
@@ -392,9 +414,9 @@ capture_input_new_packets(capture_options *capture_opts, int to_read)
 /* Capture child told us how many dropped packets it counted.
  */
 void
-capture_input_drops(capture_options *capture_opts, int dropped)
+capture_input_drops(capture_options *capture_opts, guint32 dropped)
 {
-  g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_INFO, "%d packet%s dropped", dropped, plurality(dropped, "", "s"));
+  g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_INFO, "%u packet%s dropped", dropped, plurality(dropped, "", "s"));
 
   g_assert(capture_opts->state == CAPTURE_RUNNING);
 
@@ -496,13 +518,13 @@ capture_input_closed(capture_options *capture_opts)
     g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_MESSAGE, "Capture stopped!");
     g_assert(capture_opts->state == CAPTURE_PREPARING || capture_opts->state == CAPTURE_RUNNING);
 
-    /* if we didn't started the capture, do a fake start */
-    /* (happens if we got an error message - we won't get a filename then) */
+    /* if we didn't start the capture, do a fake start. */
+    /* (happens if we got an error message - we won't get a filename then). */
     if(capture_opts->state == CAPTURE_PREPARING) {
         if(capture_opts->real_time_mode) {
-            cf_callback_invoke(cf_cb_live_capture_update_started, capture_opts);
+            capture_callback_invoke(capture_cb_capture_update_started, capture_opts);
         } else {
-            cf_callback_invoke(cf_cb_live_capture_fixed_started, capture_opts);
+            capture_callback_invoke(capture_cb_capture_fixed_started, capture_opts);
         }
     }
 
@@ -514,10 +536,10 @@ capture_input_closed(capture_options *capture_opts)
 
         /* XXX: If -Q (quit-after-cap) then cf->count clr'd below so save it first */
        packet_count_save = cf_get_packet_count(capture_opts->cf);
-        /* Tell the GUI, we are not doing a capture any more.
-                  Must be done after the cf_finish_tail(), so file lengths are displayed
-                  correct. */
-        cf_callback_invoke(cf_cb_live_capture_update_finished, capture_opts->cf);
+        /* Tell the GUI we are not doing a capture any more.
+                  Must be done after the cf_finish_tail(), so file lengths are 
+                   correctly displayed */
+        capture_callback_invoke(capture_cb_capture_update_finished, capture_opts);
 
         /* Finish the capture. */
         switch (status) {
@@ -558,7 +580,7 @@ capture_input_closed(capture_options *capture_opts)
 
     } else {
         /* first of all, we are not doing a capture any more */
-        cf_callback_invoke(cf_cb_live_capture_fixed_finished, capture_opts->cf);
+        capture_callback_invoke(capture_cb_capture_fixed_finished, capture_opts);
 
         /* this is a normal mode capture and if no error happened, read in the capture file data */
         if(capture_opts->save_file != NULL) {
@@ -582,7 +604,7 @@ capture_input_closed(capture_options *capture_opts)
     if(capture_opts->restart) {
         capture_opts->restart = FALSE;
 
-        eth_unlink(capture_opts->save_file);
+        ws_unlink(capture_opts->save_file);
 
         /* if it was a tempfile, throw away the old filename (so it will become a tempfile again) */
         if(cf_is_tempfile(capture_opts->cf)) {
@@ -599,157 +621,6 @@ capture_input_closed(capture_options *capture_opts)
     }
 }
 
-/**
- * Fetch the interface list from a child process (dumpcap).
- *
- * @return A GList containing if_info_t structs if successful, NULL otherwise.
- */
-
-/* XXX - We parse simple text output to get our interface list.  Should
- * we use "real" data serialization instead, e.g. via XML? */
-GList *
-capture_interface_list(int *err, char **err_str)
-{
-    GList     *if_list = NULL;
-    int        i, j;
-    gchar     *msg;
-    gchar    **raw_list, **if_parts, **addr_parts;
-    gchar     *name;
-    if_info_t *if_info;
-    if_addr_t *if_addr;
-
-    g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_MESSAGE, "Capture Interface List ...");
-
-    /* Try to get our interface list */
-    *err = sync_interface_list_open(&msg);
-    if (*err != 0) {
-        g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_MESSAGE, "Capture Interface List failed!");
-        if (err_str) {
-            if (*err_str)
-                *err_str = msg;
-            else
-                g_free(msg);
-        } else {
-            g_free(msg);
-        }
-        return NULL;
-    }
-
-    /* Split our lines */
-    raw_list = g_strsplit(msg, "\n", 0);
-    g_free(msg);
-
-    for (i = 0; raw_list[i] != NULL; i++) {
-        if_parts = g_strsplit(raw_list[i], "\t", 4);
-        if (if_parts[0] == NULL || if_parts[1] == NULL || if_parts[2] == NULL ||
-                if_parts[3] == NULL) {
-            g_strfreev(if_parts);
-            continue;
-        }
-
-        /* Number followed by the name, e.g "1. eth0" */
-        name = strchr(if_parts[0], ' ');
-        if (name) {
-            name++;
-        } else {
-            g_strfreev(if_parts);
-            continue;
-        }
-
-        if_info = g_malloc0(sizeof(if_info_t));
-        if_info->name = g_strdup(name);
-        if (strlen(if_parts[1]) > 0)
-            if_info->description = g_strdup(if_parts[1]);
-        addr_parts = g_strsplit(if_parts[2], ",", 0);
-        for (j = 0; addr_parts[j] != NULL; j++) {
-            if_addr = g_malloc0(sizeof(if_addr_t));
-            if (inet_pton(AF_INET, addr_parts[j], &if_addr->ip_addr.ip4_addr)) {
-                if_addr->type = AT_IPv4;
-            } else if (inet_pton(AF_INET6, addr_parts[j],
-                    &if_addr->ip_addr.ip6_addr)) {
-                if_addr->type = AT_IPv6;
-            } else {
-                g_free(if_addr);
-                if_addr = NULL;
-            }
-            if (if_addr) {
-                if_info->ip_addr = g_slist_append(if_info->ip_addr, if_addr);
-            }
-        }
-        if (strcmp(if_parts[3], "loopback") == 0)
-            if_info->loopback = TRUE;
-        g_strfreev(if_parts);
-        g_strfreev(addr_parts);
-        if_list = g_list_append(if_list, if_info);
-    }
-    g_strfreev(raw_list);
-
-    /* Check to see if we built a list */
-    if (if_list == NULL) {
-        if (err_str && *err_str)
-            *err_str = g_strdup("No interfaces found");
-        *err = NO_INTERFACES_FOUND;
-    }
-    return if_list;
-}
-
-/* XXX - We parse simple text output to get our interface list.  Should
- * we use "real" data serialization instead, e.g. via XML? */
-GList *
-capture_pcap_linktype_list(gchar *ifname, char **err_str)
-{
-    GList     *linktype_list = NULL;
-    int        err, i;
-    gchar     *msg;
-    gchar    **raw_list, **lt_parts;
-    data_link_info_t *data_link_info;
-
-    g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_MESSAGE, "Capture Interface List ...");
-
-    /* Try to get our interface list */
-    err = sync_linktype_list_open(ifname, &msg);
-    if (err != 0) {
-        g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_MESSAGE, "Capture Interface List failed!");
-        if (err_str) {
-            *err_str = msg;
-        } else {
-            g_free(msg);
-        }
-        return NULL;
-    }
-
-    /* Split our lines */
-    raw_list = g_strsplit(msg, "\n", 0);
-    g_free(msg);
-
-    for (i = 0; raw_list[i] != NULL; i++) {
-        /* ...and what if the interface name has a tab in it, Mr. Clever Programmer? */
-        lt_parts = g_strsplit(raw_list[i], "\t", 3);
-        if (lt_parts[0] == NULL || lt_parts[1] == NULL || lt_parts[2] == NULL) {
-            g_strfreev(lt_parts);
-            continue;
-        }
-
-        data_link_info = g_malloc(sizeof (data_link_info_t));
-        data_link_info->dlt = (int) strtol(lt_parts[0], NULL, 10);
-        data_link_info->name = g_strdup(lt_parts[1]);
-        if (strcmp(lt_parts[2], "(not supported)") != 0)
-            data_link_info->description = g_strdup(lt_parts[2]);
-        else
-            data_link_info->description = NULL;
-
-        linktype_list = g_list_append(linktype_list, data_link_info);
-    }
-    g_strfreev(raw_list);
-
-    /* Check to see if we built a list */
-    if (linktype_list == NULL) {
-        if (err_str)
-            *err_str = NULL;
-    }
-    return linktype_list;
-}
-
 if_stat_cache_t *
 capture_stat_start(GList *if_list) {
     int stat_fd, fork_child;
@@ -850,12 +721,17 @@ void
 capture_stat_stop(if_stat_cache_t *sc) {
     GList *sc_entry;
     if_stat_cache_item_t *sc_item;
+    int ret;
     gchar *msg;
 
     if (!sc)
         return;
 
-    sync_interface_stats_close(&sc->stat_fd, &sc->fork_child, &msg);
+    ret = sync_interface_stats_close(&sc->stat_fd, &sc->fork_child, &msg);
+    if (ret == -1) {
+       /* XXX - report failure? */
+        g_free(msg);
+    }
 
     for (sc_entry = sc->cache_list; sc_entry != NULL; sc_entry = g_list_next(sc_entry)) {
         sc_item = sc_entry->data;