db1e0bcf65
Change-Id: I5669e2442582f899643fae4a9f86ab6d505dde07 Reviewed-on: https://code.wireshark.org/review/16505 Petri-Dish: Michael Mann <mmann78@netscape.net> Tested-by: Petri Dish Buildbot <buildbot-no-reply@wireshark.org> Reviewed-by: Michael Mann <mmann78@netscape.net>
278 lines
7.3 KiB
C
278 lines
7.3 KiB
C
/* extcap-base.c
|
|
* Base function for extcaps
|
|
*
|
|
* Copyright 2015, Dario Lombardo
|
|
*
|
|
* Wireshark - Network traffic analyzer
|
|
* By Gerald Combs <gerald@wireshark.org>
|
|
* 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
|
|
* as published by the Free Software Foundation; either version 2
|
|
* of the License, or (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* 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)
|
|
{
|
|
HANDLE h = GetStdHandle(handle);
|
|
if (h) {
|
|
BY_HANDLE_FILE_INFORMATION fi;
|
|
if (GetFileInformationByHandle(h, &fi)) {
|
|
return TRUE;
|
|
}
|
|
}
|
|
return FALSE;
|
|
}
|
|
|
|
void attach_parent_console()
|
|
{
|
|
BOOL outRedirected, errRedirected;
|
|
|
|
outRedirected = IsHandleRedirected(STD_OUTPUT_HANDLE);
|
|
errRedirected = IsHandleRedirected(STD_ERROR_HANDLE);
|
|
|
|
if (outRedirected && errRedirected) {
|
|
/* Both standard output and error handles are redirected.
|
|
* There is no point in attaching to parent process console.
|
|
*/
|
|
return;
|
|
}
|
|
|
|
if (AttachConsole(ATTACH_PARENT_PROCESS) == 0) {
|
|
/* Console attach failed. */
|
|
return;
|
|
}
|
|
|
|
/* Console attach succeeded */
|
|
if (outRedirected == FALSE) {
|
|
if (!freopen("CONOUT$", "w", stdout)) {
|
|
errmsg_print("WARNING: Cannot redirect to stdout.");
|
|
}
|
|
}
|
|
|
|
if (errRedirected == FALSE) {
|
|
if (!freopen("CONOUT$", "w", stderr)) {
|
|
errmsg_print("WARNING: Cannot redirect to strerr.");
|
|
}
|
|
}
|
|
}
|
|
#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;
|
|
|
|
if (interface == NULL)
|
|
return;
|
|
|
|
iface = g_new0(extcap_interface, 1);
|
|
|
|
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)
|
|
{
|
|
const extcap_interface * iface_a = (const extcap_interface *)a;
|
|
|
|
return (g_strcmp0(iface_a->interface, (const 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) {
|
|
if (g_list_length(extcap->interfaces) > 0) {
|
|
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=%u}{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);
|
|
}
|
|
|
|
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);
|
|
g_free(iface);
|
|
}
|
|
|
|
void extcap_base_cleanup(extcap_parameters ** extcap)
|
|
{
|
|
/* g_list_free_full() only exists since 2.28. g_list_free_full((*extcap)->interfaces, extcap_iface_free);*/
|
|
g_list_foreach((*extcap)->interfaces, (GFunc)extcap_iface_free, NULL);
|
|
g_list_free((*extcap)->interfaces);
|
|
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
|
|
*
|
|
* Local variables:
|
|
* c-basic-offset: 4
|
|
* tab-width: 8
|
|
* indent-tabs-mode: t
|
|
* End:
|
|
*
|
|
* vi: set shiftwidth=4 tabstop=8 noexpandtab:
|
|
* :indentSize=4:tabSize=8:noTabs=false:
|
|
*/
|