extcap: Move extcap handling to base

Also add the extcap-version parameter to enable parsing of
version and helppage separately

Change-Id: I35ba5aa992940ffbb0cd9ebea8b7c3a1e8629d74
Reviewed-on: https://code.wireshark.org/review/14094
Petri-Dish: Roland Knall <rknall@gmail.com>
Tested-by: Petri Dish Buildbot <buildbot-no-reply@wireshark.org>
Reviewed-by: Dario Lombardo <lomato@gmail.com>
Reviewed-by: Roland Knall <rknall@gmail.com>

1 files changed, 191 insertions, 0 deletions

diff --git a/extcap/extcap-base.c b/extcap/extcap-base.c
index 51d2061469..e449b085e5 100644
--- a/extcap/extcap-base.c
+++ b/extcap/extcap-base.c
@@ -22,8 +22,36 @@
  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  */
 
+#include "config.h"
+
 #include "extcap-base.h"
 
+#include <stdlib.h>
+#include <stdint.h>
+#include <string.h>
+
+#ifdef HAVE_GETOPT_H
+	#include <getopt.h>
+#endif
+
+#ifndef HAVE_GETOPT_LONG
+	#include "wsutil/wsgetopt.h"
+#endif
+
+enum extcap_options {
+	EXTCAP_BASE_OPTIONS_ENUM
+};
+
+typedef struct _extcap_interface
+{
+	char * interface;
+	char * description;
+
+	uint16_t dlt;
+	char * dltname;
+	char * dltdescription;
+} extcap_interface;
+
 #ifdef _WIN32
 BOOLEAN IsHandleRedirected(DWORD handle)
 {
@@ -71,6 +99,169 @@ void attach_parent_console()
 }
 #endif
 
+void extcap_base_register_interface(extcap_parameters * extcap, const char * interface, const char * ifdescription, uint16_t dlt, const char * dltdescription )
+{
+	extcap_base_register_interface_ext(extcap, interface, ifdescription, dlt, NULL, dltdescription );
+}
+
+void extcap_base_register_interface_ext(extcap_parameters * extcap,
+		const char * interface, const char * ifdescription,
+		uint16_t dlt, const char * dltname, const char * dltdescription )
+{
+	extcap_interface * iface = g_new0(extcap_interface, 1);
+
+	if (interface == NULL)
+		return;
+
+	iface->interface = g_strdup(interface);
+	iface->description = g_strdup(ifdescription);
+	iface->dlt = dlt;
+	iface->dltname = g_strdup(dltname);
+	iface->dltdescription = g_strdup(dltdescription);
+
+	extcap->interfaces = g_list_append(extcap->interfaces, (gpointer) iface);
+}
+
+void extcap_base_set_util_info(extcap_parameters * extcap, const char * major, const char * minor, const char * release, const char * helppage)
+{
+	g_assert(major);
+	extcap->version = g_strdup_printf("%s%s%s%s%s",
+			major,
+			minor ? "." : "",
+			minor ? minor : "",
+			release ? "." : "",
+			release ? release : "");
+	extcap->helppage = g_strdup(helppage);
+}
+
+uint8_t extcap_base_parse_options(extcap_parameters * extcap, int result, char * optargument )
+{
+	switch (result) {
+	case EXTCAP_OPT_LIST_INTERFACES:
+		extcap->do_list_interfaces = 1;
+		break;
+	case EXTCAP_OPT_VERSION:
+		extcap->do_version = 1;
+		break;
+	case EXTCAP_OPT_LIST_DLTS:
+		extcap->do_list_dlts = 1;
+		break;
+	case EXTCAP_OPT_INTERFACE:
+		extcap->interface = g_strdup(optargument);
+		break;
+	case EXTCAP_OPT_CONFIG:
+		extcap->show_config = 1;
+		break;
+	case EXTCAP_OPT_CAPTURE:
+		extcap->capture = 1;
+		break;
+	case EXTCAP_OPT_CAPTURE_FILTER:
+		extcap->capture_filter = g_strdup(optargument);
+		break;
+	case EXTCAP_OPT_FIFO:
+		extcap->fifo = g_strdup(optargument);
+		break;
+	}
+
+	return 1;
+}
+
+static void extcap_iface_print(gpointer data, gpointer userdata _U_)
+{
+	extcap_interface * iface = (extcap_interface *)data;
+
+	printf("interface {value=%s}", iface->interface);
+	if (iface->description != NULL)
+		printf ("{display=%s}\n", iface->description);
+	else
+		printf ("\n");
+}
+
+static gint extcap_iface_compare(gconstpointer  a, gconstpointer  b)
+{
+	extcap_interface * iface_a = (extcap_interface *)a;
+
+	return (g_strcmp0(iface_a->interface, (char *) b));
+}
+
+static void extcap_print_version(extcap_parameters * extcap)
+{
+	printf("extcap {version=%s}", extcap->version != NULL ? extcap->version : "unknown");
+	if (extcap->helppage != NULL)
+		printf("{help=%s}", extcap->helppage);
+	printf("\n");
+}
+
+static gint extcap_iface_listall(extcap_parameters * extcap, uint8_t list_ifs)
+{
+	if (list_ifs) {
+		extcap_print_version(extcap);
+		g_list_foreach(extcap->interfaces, extcap_iface_print, extcap);
+	} else {
+		if (extcap->do_version) {
+			extcap_print_version(extcap);
+		} else {
+			GList * element = NULL;
+			extcap_interface * iface = NULL;
+			if ((element = g_list_find_custom(extcap->interfaces, extcap->interface, extcap_iface_compare)) == NULL)
+				return 0;
+
+			iface = (extcap_interface *) element->data;
+			printf("dlt {number=%d}{name=%s}", iface->dlt, iface->dltname != NULL ? iface->dltname : iface->interface);
+			if (iface->description != NULL)
+				printf ("{display=%s}\n", iface->dltdescription);
+			else
+				printf ("\n");
+		}
+	}
+
+	return 1;
+}
+
+uint8_t extcap_base_handle_interface(extcap_parameters * extcap)
+{
+	/* A fifo must be provided for capture */
+	if (extcap->capture && (extcap->fifo == NULL || strlen(extcap->fifo) <= 0)) {
+		extcap->capture = 0;
+		errmsg_print("Extcap Error: No FIFO pipe provided");
+		return 0;
+	}
+
+	if (extcap->do_list_interfaces) {
+		return extcap_iface_listall(extcap, 1);
+	} else if (extcap->do_version || extcap->do_list_dlts) {
+		return extcap_iface_listall(extcap, 0);
+	} else {
+		/* An interface must exist */
+		if (g_list_find_custom(extcap->interfaces, extcap->interface, extcap_iface_compare) == NULL) {
+			errmsg_print("Extcap Error: No interface [%s] provided", extcap->interface);
+			return 0;
+		}
+	}
+
+	return 0;
+}
+
+static void extcap_iface_free(gpointer data)
+{
+	extcap_interface * iface = (extcap_interface *)data;
+	g_free(iface->interface);
+	g_free(iface->description);
+	g_free(iface->dltname);
+	g_free(iface->dltdescription);
+}
+
+void extcap_base_cleanup(extcap_parameters ** extcap)
+{
+	g_list_free_full((*extcap)->interfaces, extcap_iface_free);
+	g_free((*extcap)->fifo);
+	g_free((*extcap)->interface);
+	g_free((*extcap)->version);
+	g_free((*extcap)->helppage);
+	g_free(*extcap);
+	*extcap = NULL;
+}
+
 /*
  * Editor modelines  -  https://www.wireshark.org/tools/modelines.html
  *