forked from osmocom/wireshark
ed6f23406b
Change-Id: I3c1e4080c28d0d14eaf7ecfb5540c53f8736703c Reviewed-on: https://code.wireshark.org/review/20971 Reviewed-by: Guy Harris <guy@alum.mit.edu>
1086 lines
32 KiB
C
1086 lines
32 KiB
C
/* disabled_protos.c
|
|
* Declarations of routines for reading and writing protocols file that determine
|
|
* enabling and disabling of protocols.
|
|
*
|
|
* 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 <stdio.h>
|
|
#include <string.h>
|
|
#include <errno.h>
|
|
|
|
#include <glib.h>
|
|
|
|
#include <wsutil/filesystem.h>
|
|
#include <epan/proto.h>
|
|
#include <epan/packet.h>
|
|
|
|
#include "disabled_protos.h"
|
|
#include <wsutil/file_util.h>
|
|
#include <wsutil/ws_printf.h> /* ws_g_warning */
|
|
#include <wsutil/report_message.h>
|
|
|
|
#define ENABLED_PROTOCOLS_FILE_NAME "enabled_protos"
|
|
#define DISABLED_PROTOCOLS_FILE_NAME "disabled_protos"
|
|
#define HEURISTICS_FILE_NAME "heuristic_protos"
|
|
|
|
/*
|
|
* Item in a list of disabled protocols.
|
|
*/
|
|
typedef struct {
|
|
char *name; /* protocol name */
|
|
} protocol_def;
|
|
|
|
/*
|
|
* Item in a list of heuristic dissectors and their enabled state.
|
|
*/
|
|
typedef struct {
|
|
char *name; /* heuristic short name */
|
|
gboolean enabled; /* heuristc enabled */
|
|
} heur_protocol_def;
|
|
|
|
/*
|
|
* List of disabled protocols
|
|
*/
|
|
static GList *global_disabled_protos = NULL;
|
|
static GList *disabled_protos = NULL;
|
|
/*
|
|
* List of enabled protocols (that are disabled by default)
|
|
*/
|
|
static GList *global_enabled_protos = NULL;
|
|
static GList *enabled_protos = NULL;
|
|
/*
|
|
* List of disabled heuristics
|
|
*/
|
|
static GList *global_disabled_heuristics = NULL;
|
|
static GList *disabled_heuristics = NULL;
|
|
|
|
#define INIT_BUF_SIZE 128
|
|
|
|
static void
|
|
discard_existing_list (GList **flp)
|
|
{
|
|
GList *fl_ent;
|
|
protocol_def *prot;
|
|
|
|
if (*flp != NULL) {
|
|
fl_ent = g_list_first(*flp);
|
|
while (fl_ent != NULL) {
|
|
prot = (protocol_def *) fl_ent->data;
|
|
g_free(prot->name);
|
|
g_free(prot);
|
|
fl_ent = fl_ent->next;
|
|
}
|
|
g_list_free(*flp);
|
|
*flp = NULL;
|
|
}
|
|
}
|
|
|
|
static void
|
|
heur_discard_existing_list (GList **flp)
|
|
{
|
|
GList *fl_ent;
|
|
heur_protocol_def *prot;
|
|
|
|
if (*flp != NULL) {
|
|
fl_ent = g_list_first(*flp);
|
|
while (fl_ent != NULL) {
|
|
prot = (heur_protocol_def *) fl_ent->data;
|
|
g_free(prot->name);
|
|
g_free(prot);
|
|
fl_ent = fl_ent->next;
|
|
}
|
|
g_list_free(*flp);
|
|
*flp = NULL;
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Enable/Disable protocols as per the stored configuration
|
|
*/
|
|
static void
|
|
set_protos_list(GList *protos_list, GList *global_protos_list, gboolean enable)
|
|
{
|
|
gint i;
|
|
GList *fl_ent;
|
|
protocol_def *prot;
|
|
|
|
/*
|
|
* Assume no protocols disabled by default wants to be enabled
|
|
*/
|
|
if (protos_list == NULL)
|
|
goto skip;
|
|
|
|
fl_ent = g_list_first(protos_list);
|
|
|
|
while (fl_ent != NULL) {
|
|
prot = (protocol_def *) fl_ent->data;
|
|
i = proto_get_id_by_filter_name(prot->name);
|
|
if (i == -1) {
|
|
/* XXX - complain here? */
|
|
} else {
|
|
if (proto_can_toggle_protocol(i))
|
|
proto_set_decoding(i, enable);
|
|
}
|
|
|
|
fl_ent = fl_ent->next;
|
|
}
|
|
|
|
skip:
|
|
if (global_protos_list == NULL)
|
|
return;
|
|
|
|
fl_ent = g_list_first(global_protos_list);
|
|
|
|
while (fl_ent != NULL) {
|
|
prot = (protocol_def *) fl_ent->data;
|
|
i = proto_get_id_by_filter_name(prot->name);
|
|
if (i == -1) {
|
|
/* XXX - complain here? */
|
|
} else {
|
|
if (proto_can_toggle_protocol(i)) {
|
|
proto_set_decoding(i, enable);
|
|
proto_set_cant_toggle(i);
|
|
}
|
|
}
|
|
|
|
fl_ent = fl_ent->next;
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Write out a list of protocols based on condition
|
|
*
|
|
* On success, "*pref_path_return" is set to NULL.
|
|
* On error, "*pref_path_return" is set to point to the pathname of
|
|
* the file we tried to read - it should be freed by our caller -
|
|
* and "*errno_return" is set to the error.
|
|
*/
|
|
static void
|
|
save_protos_list(char **pref_path_return, int *errno_return, const char* filename,
|
|
const char* header_comment, gboolean (*protocol_check)(protocol_t *protocol))
|
|
{
|
|
gchar *ff_path, *ff_path_new;
|
|
FILE *ff;
|
|
gint i;
|
|
protocol_t *protocol;
|
|
void *cookie;
|
|
gboolean first = TRUE;
|
|
|
|
*pref_path_return = NULL; /* assume no error */
|
|
|
|
ff_path = get_persconffile_path(filename, TRUE);
|
|
|
|
/* Write to "XXX.new", and rename if that succeeds.
|
|
That means we don't trash the file if we fail to write it out
|
|
completely. */
|
|
ff_path_new = g_strdup_printf("%s.new", ff_path);
|
|
|
|
if ((ff = ws_fopen(ff_path_new, "w")) == NULL) {
|
|
*pref_path_return = ff_path;
|
|
*errno_return = errno;
|
|
g_free(ff_path_new);
|
|
return;
|
|
}
|
|
|
|
/* Iterate over all the protocols */
|
|
for (i = proto_get_first_protocol(&cookie); i != -1;
|
|
i = proto_get_next_protocol(&cookie)) {
|
|
|
|
if (!proto_can_toggle_protocol(i)) {
|
|
continue;
|
|
}
|
|
|
|
protocol = find_protocol_by_id(i);
|
|
if (protocol_check(protocol) == FALSE)
|
|
continue;
|
|
|
|
if (first) {
|
|
if (header_comment != NULL) {
|
|
/* Write out a comment explaining what the file is */
|
|
fprintf(ff, "%s\n", header_comment);
|
|
}
|
|
first = FALSE;
|
|
}
|
|
|
|
/* Write out the protocol name. */
|
|
fprintf(ff, "%s\n", proto_get_protocol_filter_name(i));
|
|
}
|
|
|
|
if (fclose(ff) == EOF) {
|
|
*pref_path_return = ff_path;
|
|
*errno_return = errno;
|
|
ws_unlink(ff_path_new);
|
|
g_free(ff_path_new);
|
|
return;
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
/* ANSI C doesn't say whether "rename()" removes the target if it
|
|
exists; the Win32 call to rename files doesn't do so, which I
|
|
infer is the reason why the MSVC++ "rename()" doesn't do so.
|
|
We must therefore remove the target file first, on Windows.
|
|
|
|
XXX - ws_rename() should be ws_stdio_rename() on Windows,
|
|
and ws_stdio_rename() uses MoveFileEx() with MOVEFILE_REPLACE_EXISTING,
|
|
so it should remove the target if it exists, so this stuff
|
|
shouldn't be necessary. Perhaps it dates back to when we were
|
|
calling rename(), with that being a wrapper around Microsoft's
|
|
_rename(), which didn't remove the target. */
|
|
if (ws_remove(ff_path) < 0 && errno != ENOENT) {
|
|
/* It failed for some reason other than "it's not there"; if
|
|
it's not there, we don't need to remove it, so we just
|
|
drive on. */
|
|
*pref_path_return = ff_path;
|
|
*errno_return = errno;
|
|
ws_unlink(ff_path_new);
|
|
g_free(ff_path_new);
|
|
return;
|
|
}
|
|
#endif
|
|
|
|
if (ws_rename(ff_path_new, ff_path) < 0) {
|
|
*pref_path_return = ff_path;
|
|
*errno_return = errno;
|
|
ws_unlink(ff_path_new);
|
|
g_free(ff_path_new);
|
|
return;
|
|
}
|
|
g_free(ff_path_new);
|
|
g_free(ff_path);
|
|
}
|
|
|
|
static int
|
|
read_protos_list_file(const char *ff_path, FILE *ff, GList **flp)
|
|
{
|
|
protocol_def *prot;
|
|
int c;
|
|
char *prot_name;
|
|
int prot_name_len;
|
|
int prot_name_index;
|
|
int line = 1;
|
|
gboolean in_comment = FALSE;
|
|
|
|
|
|
/* Allocate the protocol name buffer. */
|
|
prot_name_len = INIT_BUF_SIZE;
|
|
prot_name = (char *)g_malloc(prot_name_len + 1);
|
|
|
|
for (line = 1; ; line++) {
|
|
/* Lines in a protocol file contain the "filter name" of a protocol
|
|
to be enabled or disabled. */
|
|
|
|
/* Skip over leading white space, if any. */
|
|
while ((c = ws_getc_unlocked(ff)) != EOF && g_ascii_isspace(c)) {
|
|
if (c == '\n') {
|
|
/* Blank line. */
|
|
continue;
|
|
}
|
|
}
|
|
|
|
if (c == EOF) {
|
|
if (ferror(ff))
|
|
goto error; /* I/O error */
|
|
else
|
|
break; /* Nothing more to read */
|
|
}
|
|
ungetc(c, ff); /* Unread the non-white-space character. */
|
|
|
|
/* Get the name of the protocol. */
|
|
prot_name_index = 0;
|
|
for (;;) {
|
|
c = ws_getc_unlocked(ff);
|
|
if (c == EOF)
|
|
break; /* End of file, or I/O error */
|
|
if (g_ascii_isspace(c))
|
|
break; /* Trailing white space, or end of line. */
|
|
if (c == '#') {
|
|
in_comment = TRUE;
|
|
break; /* Start of comment, running to end of line. */
|
|
}
|
|
/* Add this character to the protocol name string. */
|
|
if (prot_name_index >= prot_name_len) {
|
|
/* protocol name buffer isn't long enough; double its length. */
|
|
prot_name_len *= 2;
|
|
prot_name = (char *)g_realloc(prot_name, prot_name_len + 1);
|
|
}
|
|
prot_name[prot_name_index] = c;
|
|
prot_name_index++;
|
|
}
|
|
|
|
if (g_ascii_isspace(c) && c != '\n') {
|
|
/* Skip over trailing white space. */
|
|
while ((c = ws_getc_unlocked(ff)) != EOF && c != '\n' && g_ascii_isspace(c))
|
|
;
|
|
if (c != EOF && c != '\n' && c != '#') {
|
|
/* Non-white-space after the protocol name; warn about it,
|
|
in case we come up with a reason to use it. */
|
|
ws_g_warning("'%s' line %d has extra stuff after the protocol name.",
|
|
ff_path, line);
|
|
}
|
|
}
|
|
if (c != EOF && c != '\n' && in_comment == TRUE) {
|
|
/* Skip to end of line. */
|
|
while ((c = ws_getc_unlocked(ff)) != EOF && c != '\n')
|
|
;
|
|
}
|
|
|
|
if (c == EOF) {
|
|
if (ferror(ff))
|
|
goto error; /* I/O error */
|
|
else {
|
|
/* EOF, not error; no newline seen before EOF */
|
|
ws_g_warning("'%s' line %d doesn't have a newline.", ff_path,
|
|
line);
|
|
}
|
|
break; /* nothing more to read */
|
|
}
|
|
|
|
if (in_comment) {
|
|
in_comment = FALSE;
|
|
continue;
|
|
}
|
|
|
|
/* Null-terminate the protocol name. */
|
|
if (prot_name_index >= prot_name_len) {
|
|
/* protocol name buffer isn't long enough; double its length. */
|
|
prot_name_len *= 2;
|
|
prot_name = (char *)g_realloc(prot_name, prot_name_len + 1);
|
|
}
|
|
prot_name[prot_name_index] = '\0';
|
|
|
|
/* Add the new protocol to the list of disabled protocols */
|
|
prot = (protocol_def *) g_malloc(sizeof(protocol_def));
|
|
prot->name = g_strdup(prot_name);
|
|
*flp = g_list_append(*flp, prot);
|
|
}
|
|
g_free(prot_name);
|
|
return 0;
|
|
|
|
error:
|
|
g_free(prot_name);
|
|
return errno;
|
|
}
|
|
|
|
/*
|
|
* Read in global and personal versions of a list of protocols.
|
|
*
|
|
* If we can open and read the global version, *gpath_return is set to
|
|
* NULL. Otherwise, *gpath_return is set to point to the pathname of
|
|
* the file we tried to read - it should be freed by our caller - and
|
|
* *gopen_errno_return is set to the error if an open failed or
|
|
* *gread_errno_return is set to the error if a read failed.
|
|
*
|
|
* If we can open and read the personal version, *path_return is set to
|
|
* NULL. Otherwise, *path_return is set to point to the pathname of
|
|
* the file we tried to read - it should be freed by our caller - and
|
|
* *open_errno_return is set to the error if an open failed or
|
|
* *read_errno_return is set to the error if a read failed.
|
|
*/
|
|
static void
|
|
read_protos_list(char **gpath_return, int *gopen_errno_return,
|
|
int *gread_errno_return,
|
|
char **path_return, int *open_errno_return,
|
|
int *read_errno_return,
|
|
const char* filename,
|
|
GList **global_protos_list, GList **protos_list)
|
|
{
|
|
int err;
|
|
char *gff_path, *ff_path;
|
|
FILE *ff;
|
|
|
|
/* Construct the pathname of the global disabled protocols file. */
|
|
gff_path = get_datafile_path(filename);
|
|
|
|
/* If we already have a list of protocols, discard it. */
|
|
discard_existing_list (global_protos_list);
|
|
|
|
/* Read the global disabled protocols file, if it exists. */
|
|
*gpath_return = NULL;
|
|
if ((ff = ws_fopen(gff_path, "r")) != NULL) {
|
|
/* We succeeded in opening it; read it. */
|
|
err = read_protos_list_file(gff_path, ff, global_protos_list);
|
|
if (err != 0) {
|
|
/* We had an error reading the file; return the errno and the
|
|
pathname, so our caller can report the error. */
|
|
*gopen_errno_return = 0;
|
|
*gread_errno_return = err;
|
|
*gpath_return = gff_path;
|
|
} else
|
|
g_free(gff_path);
|
|
fclose(ff);
|
|
} else {
|
|
/* We failed to open it. If we failed for some reason other than
|
|
"it doesn't exist", return the errno and the pathname, so our
|
|
caller can report the error. */
|
|
if (errno != ENOENT) {
|
|
*gopen_errno_return = errno;
|
|
*gread_errno_return = 0;
|
|
*gpath_return = gff_path;
|
|
} else
|
|
g_free(gff_path);
|
|
}
|
|
|
|
/* Construct the pathname of the user's disabled protocols file. */
|
|
ff_path = get_persconffile_path(filename, TRUE);
|
|
|
|
/* If we already have a list of protocols, discard it. */
|
|
discard_existing_list (protos_list);
|
|
|
|
/* Read the user's disabled protocols file, if it exists. */
|
|
*path_return = NULL;
|
|
if ((ff = ws_fopen(ff_path, "r")) != NULL) {
|
|
/* We succeeded in opening it; read it. */
|
|
err = read_protos_list_file(ff_path, ff, protos_list);
|
|
if (err != 0) {
|
|
/* We had an error reading the file; return the errno and the
|
|
pathname, so our caller can report the error. */
|
|
*open_errno_return = 0;
|
|
*read_errno_return = err;
|
|
*path_return = ff_path;
|
|
} else
|
|
g_free(ff_path);
|
|
fclose(ff);
|
|
} else {
|
|
/* We failed to open it. If we failed for some reason other than
|
|
"it doesn't exist", return the errno and the pathname, so our
|
|
caller can report the error. */
|
|
if (errno != ENOENT) {
|
|
*open_errno_return = errno;
|
|
*read_errno_return = 0;
|
|
*path_return = ff_path;
|
|
} else
|
|
g_free(ff_path);
|
|
}
|
|
}
|
|
|
|
/************************************************************************
|
|
* Disabling dissectors
|
|
************************************************************************/
|
|
|
|
/*
|
|
* Disable a particular protocol by name
|
|
*/
|
|
void
|
|
proto_disable_proto_by_name(const char *name)
|
|
{
|
|
protocol_t *protocol;
|
|
int proto_id;
|
|
|
|
proto_id = proto_get_id_by_filter_name(name);
|
|
if (proto_id >= 0 ) {
|
|
protocol = find_protocol_by_id(proto_id);
|
|
if (proto_is_protocol_enabled(protocol) == TRUE) {
|
|
if (proto_can_toggle_protocol(proto_id) == TRUE) {
|
|
proto_set_decoding(proto_id, FALSE);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
static gboolean disable_proto_list_check(protocol_t *protocol)
|
|
{
|
|
if (proto_is_protocol_enabled(protocol) == FALSE)
|
|
return TRUE;
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
/************************************************************************
|
|
* Enabling dissectors (that are disabled by default)
|
|
************************************************************************/
|
|
|
|
WS_DLL_PUBLIC void
|
|
proto_enable_proto_by_name(const char *name)
|
|
{
|
|
protocol_t *protocol;
|
|
int proto_id;
|
|
|
|
proto_id = proto_get_id_by_filter_name(name);
|
|
if (proto_id >= 0 ) {
|
|
protocol = find_protocol_by_id(proto_id);
|
|
if ((proto_is_protocol_enabled_by_default(protocol) == FALSE) &&
|
|
(proto_is_protocol_enabled(protocol) == FALSE)) {
|
|
if (proto_can_toggle_protocol(proto_id) == TRUE) {
|
|
proto_set_decoding(proto_id, TRUE);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
static gboolean enable_proto_list_check(protocol_t *protocol)
|
|
{
|
|
if ((proto_is_protocol_enabled_by_default(protocol) == FALSE) &&
|
|
(proto_is_protocol_enabled(protocol) == TRUE))
|
|
return TRUE;
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
/************************************************************************
|
|
* Heuristic dissectors
|
|
************************************************************************/
|
|
|
|
|
|
static void
|
|
set_disabled_heur_dissector_list(void)
|
|
{
|
|
GList *fl_ent;
|
|
heur_protocol_def *heur;
|
|
heur_dtbl_entry_t* h;
|
|
|
|
if (disabled_heuristics == NULL)
|
|
goto skip;
|
|
|
|
fl_ent = g_list_first(disabled_heuristics);
|
|
|
|
while (fl_ent != NULL) {
|
|
heur = (heur_protocol_def *) fl_ent->data;
|
|
h = find_heur_dissector_by_unique_short_name(heur->name);
|
|
if (h != NULL) {
|
|
h->enabled = heur->enabled;
|
|
}
|
|
|
|
fl_ent = fl_ent->next;
|
|
}
|
|
|
|
skip:
|
|
if (global_disabled_heuristics == NULL)
|
|
return;
|
|
|
|
fl_ent = g_list_first(global_disabled_heuristics);
|
|
|
|
while (fl_ent != NULL) {
|
|
heur = (heur_protocol_def *) fl_ent->data;
|
|
|
|
h = find_heur_dissector_by_unique_short_name(heur->name);
|
|
if (h != NULL) {
|
|
h->enabled = heur->enabled;
|
|
}
|
|
|
|
fl_ent = fl_ent->next;
|
|
}
|
|
}
|
|
|
|
static int
|
|
read_heur_dissector_list_file(const char *ff_path, FILE *ff, GList **flp)
|
|
{
|
|
heur_protocol_def *heur;
|
|
int c;
|
|
char *heuristic_name;
|
|
int heuristic_name_len;
|
|
int name_index;
|
|
gboolean parse_enabled;
|
|
gboolean enabled;
|
|
int line = 1;
|
|
|
|
|
|
/* Allocate the protocol name buffer. */
|
|
heuristic_name_len = INIT_BUF_SIZE;
|
|
heuristic_name = (char *)g_malloc(heuristic_name_len + 1);
|
|
|
|
for (line = 1; ; line++) {
|
|
/* Lines in a disabled protocol file contain the "filter name" of
|
|
a protocol to be disabled. */
|
|
|
|
/* Skip over leading white space, if any. */
|
|
while ((c = ws_getc_unlocked(ff)) != EOF && g_ascii_isspace(c)) {
|
|
if (c == '\n') {
|
|
/* Blank line. */
|
|
continue;
|
|
}
|
|
}
|
|
|
|
if (c == EOF) {
|
|
if (ferror(ff))
|
|
goto error; /* I/O error */
|
|
else
|
|
break; /* Nothing more to read */
|
|
}
|
|
ungetc(c, ff); /* Unread the non-white-space character. */
|
|
|
|
/* Get the name of the protocol. */
|
|
name_index = 0;
|
|
enabled = FALSE;
|
|
parse_enabled = FALSE;
|
|
for (;;) {
|
|
c = ws_getc_unlocked(ff);
|
|
if (c == EOF)
|
|
break; /* End of file, or I/O error */
|
|
if (g_ascii_isspace(c))
|
|
break; /* Trailing white space, or end of line. */
|
|
if (c == ',') {/* Separator for enable/disable */
|
|
parse_enabled = TRUE;
|
|
continue;
|
|
}
|
|
if (c == '#')
|
|
break; /* Start of comment, running to end of line. */
|
|
if (parse_enabled) {
|
|
enabled = ((c == '1') ? TRUE : FALSE);
|
|
break;
|
|
}
|
|
/* Add this character to the protocol name string. */
|
|
if (name_index >= heuristic_name_len) {
|
|
/* protocol name buffer isn't long enough; double its length. */
|
|
heuristic_name_len *= 2;
|
|
heuristic_name = (char *)g_realloc(heuristic_name, heuristic_name_len + 1);
|
|
}
|
|
heuristic_name[name_index] = c;
|
|
name_index++;
|
|
}
|
|
|
|
if (g_ascii_isspace(c) && c != '\n') {
|
|
/* Skip over trailing white space. */
|
|
while ((c = ws_getc_unlocked(ff)) != EOF && c != '\n' && g_ascii_isspace(c))
|
|
;
|
|
if (c != EOF && c != '\n' && c != '#') {
|
|
/* Non-white-space after the protocol name; warn about it,
|
|
in case we come up with a reason to use it. */
|
|
ws_g_warning("'%s' line %d has extra stuff after the protocol name.",
|
|
ff_path, line);
|
|
}
|
|
}
|
|
if (c != EOF && c != '\n') {
|
|
/* Skip to end of line. */
|
|
while ((c = ws_getc_unlocked(ff)) != EOF && c != '\n')
|
|
;
|
|
}
|
|
|
|
if (c == EOF) {
|
|
if (ferror(ff))
|
|
goto error; /* I/O error */
|
|
else {
|
|
/* EOF, not error; no newline seen before EOF */
|
|
ws_g_warning("'%s' line %d doesn't have a newline.", ff_path,
|
|
line);
|
|
}
|
|
break; /* nothing more to read */
|
|
}
|
|
|
|
/* Null-terminate the protocol name. */
|
|
if (name_index >= heuristic_name_len) {
|
|
/* protocol name buffer isn't long enough; double its length. */
|
|
heuristic_name_len *= 2;
|
|
heuristic_name = (char *)g_realloc(heuristic_name, heuristic_name_len + 1);
|
|
}
|
|
heuristic_name[name_index] = '\0';
|
|
|
|
/* Add the new protocol to the list of protocols */
|
|
heur = (heur_protocol_def *) g_malloc(sizeof(heur_protocol_def));
|
|
heur->name = g_strdup(heuristic_name);
|
|
heur->enabled = enabled;
|
|
*flp = g_list_append(*flp, heur);
|
|
}
|
|
g_free(heuristic_name);
|
|
return 0;
|
|
|
|
error:
|
|
g_free(heuristic_name);
|
|
return errno;
|
|
}
|
|
|
|
static void
|
|
read_heur_dissector_list(char **gpath_return, int *gopen_errno_return,
|
|
int *gread_errno_return,
|
|
char **path_return, int *open_errno_return,
|
|
int *read_errno_return)
|
|
{
|
|
int err;
|
|
char *gff_path, *ff_path;
|
|
FILE *ff;
|
|
|
|
/* If we already have a list of protocols, discard it. */
|
|
heur_discard_existing_list(&global_disabled_heuristics);
|
|
|
|
/* Construct the pathname of the global disabled heuristic dissectors file. */
|
|
gff_path = get_datafile_path(HEURISTICS_FILE_NAME);
|
|
|
|
/* Read the global disabled protocols file, if it exists. */
|
|
*gpath_return = NULL;
|
|
if ((ff = ws_fopen(gff_path, "r")) != NULL) {
|
|
/* We succeeded in opening it; read it. */
|
|
err = read_heur_dissector_list_file(gff_path, ff,
|
|
&global_disabled_heuristics);
|
|
if (err != 0) {
|
|
/* We had an error reading the file; return the errno and the
|
|
pathname, so our caller can report the error. */
|
|
*gopen_errno_return = 0;
|
|
*gread_errno_return = err;
|
|
*gpath_return = gff_path;
|
|
} else
|
|
g_free(gff_path);
|
|
fclose(ff);
|
|
} else {
|
|
/* We failed to open it. If we failed for some reason other than
|
|
"it doesn't exist", return the errno and the pathname, so our
|
|
caller can report the error. */
|
|
if (errno != ENOENT) {
|
|
*gopen_errno_return = errno;
|
|
*gread_errno_return = 0;
|
|
*gpath_return = gff_path;
|
|
} else
|
|
g_free(gff_path);
|
|
}
|
|
|
|
/* Construct the pathname of the user's disabled protocols file. */
|
|
ff_path = get_persconffile_path(HEURISTICS_FILE_NAME, TRUE);
|
|
|
|
/* If we already have a list of protocols, discard it. */
|
|
heur_discard_existing_list (&disabled_heuristics);
|
|
|
|
/* Read the user's disabled protocols file, if it exists. */
|
|
*path_return = NULL;
|
|
if ((ff = ws_fopen(ff_path, "r")) != NULL) {
|
|
/* We succeeded in opening it; read it. */
|
|
err = read_heur_dissector_list_file(ff_path, ff, &disabled_heuristics);
|
|
if (err != 0) {
|
|
/* We had an error reading the file; return the errno and the
|
|
pathname, so our caller can report the error. */
|
|
*open_errno_return = 0;
|
|
*read_errno_return = err;
|
|
*path_return = ff_path;
|
|
} else
|
|
g_free(ff_path);
|
|
fclose(ff);
|
|
} else {
|
|
/* We failed to open it. If we failed for some reason other than
|
|
"it doesn't exist", return the errno and the pathname, so our
|
|
caller can report the error. */
|
|
if (errno != ENOENT) {
|
|
*open_errno_return = errno;
|
|
*read_errno_return = 0;
|
|
*path_return = ff_path;
|
|
} else
|
|
g_free(ff_path);
|
|
}
|
|
}
|
|
|
|
static gint
|
|
heur_compare(gconstpointer a, gconstpointer b)
|
|
{
|
|
return strcmp(((const heur_dtbl_entry_t *)a)->short_name,
|
|
((const heur_dtbl_entry_t *)b)->short_name);
|
|
}
|
|
|
|
static void
|
|
write_heur_dissector(gpointer data, gpointer user_data)
|
|
{
|
|
heur_dtbl_entry_t* dtbl_entry = (heur_dtbl_entry_t*)data;
|
|
FILE *ff = (FILE*)user_data;
|
|
|
|
/* Write out the heuristic short name and its enabled state */
|
|
fprintf(ff, "%s,%d\n", dtbl_entry->short_name, dtbl_entry->enabled ? 1 : 0);
|
|
}
|
|
|
|
static void
|
|
sort_dissector_table_entries(const char *table_name _U_,
|
|
heur_dtbl_entry_t *dtbl_entry, gpointer user_data)
|
|
{
|
|
GSList **list = (GSList**)user_data;
|
|
*list = g_slist_insert_sorted(*list, dtbl_entry, heur_compare);
|
|
}
|
|
|
|
static void
|
|
sort_heur_dissector_tables(const char *table_name, struct heur_dissector_list *list, gpointer w)
|
|
{
|
|
if (list) {
|
|
heur_dissector_table_foreach(table_name, sort_dissector_table_entries, w);
|
|
}
|
|
}
|
|
|
|
static void
|
|
save_disabled_heur_dissector_list(char **pref_path_return, int *errno_return)
|
|
{
|
|
gchar *ff_path, *ff_path_new;
|
|
GSList *sorted_heur_list = NULL;
|
|
FILE *ff;
|
|
|
|
*pref_path_return = NULL; /* assume no error */
|
|
|
|
ff_path = get_persconffile_path(HEURISTICS_FILE_NAME, TRUE);
|
|
|
|
/* Write to "XXX.new", and rename if that succeeds.
|
|
That means we don't trash the file if we fail to write it out
|
|
completely. */
|
|
ff_path_new = g_strdup_printf("%s.new", ff_path);
|
|
|
|
if ((ff = ws_fopen(ff_path_new, "w")) == NULL) {
|
|
*pref_path_return = ff_path;
|
|
*errno_return = errno;
|
|
g_free(ff_path_new);
|
|
return;
|
|
}
|
|
|
|
/* Iterate over all the heuristic dissectors to sort them in alphabetical order by short name */
|
|
dissector_all_heur_tables_foreach_table(sort_heur_dissector_tables, &sorted_heur_list, NULL);
|
|
|
|
/* Write the list */
|
|
g_slist_foreach(sorted_heur_list, write_heur_dissector, ff);
|
|
g_slist_free(sorted_heur_list);
|
|
|
|
if (fclose(ff) == EOF) {
|
|
*pref_path_return = ff_path;
|
|
*errno_return = errno;
|
|
ws_unlink(ff_path_new);
|
|
g_free(ff_path_new);
|
|
return;
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
/* ANSI C doesn't say whether "rename()" removes the target if it
|
|
exists; the Win32 call to rename files doesn't do so, which I
|
|
infer is the reason why the MSVC++ "rename()" doesn't do so.
|
|
We must therefore remove the target file first, on Windows.
|
|
|
|
XXX - ws_rename() should be ws_stdio_rename() on Windows,
|
|
and ws_stdio_rename() uses MoveFileEx() with MOVEFILE_REPLACE_EXISTING,
|
|
so it should remove the target if it exists, so this stuff
|
|
shouldn't be necessary. Perhaps it dates back to when we were
|
|
calling rename(), with that being a wrapper around Microsoft's
|
|
_rename(), which didn't remove the target. */
|
|
if (ws_remove(ff_path) < 0 && errno != ENOENT) {
|
|
/* It failed for some reason other than "it's not there"; if
|
|
it's not there, we don't need to remove it, so we just
|
|
drive on. */
|
|
*pref_path_return = ff_path;
|
|
*errno_return = errno;
|
|
ws_unlink(ff_path_new);
|
|
g_free(ff_path_new);
|
|
return;
|
|
}
|
|
#endif
|
|
|
|
if (ws_rename(ff_path_new, ff_path) < 0) {
|
|
*pref_path_return = ff_path;
|
|
*errno_return = errno;
|
|
ws_unlink(ff_path_new);
|
|
g_free(ff_path_new);
|
|
return;
|
|
}
|
|
g_free(ff_path_new);
|
|
g_free(ff_path);
|
|
}
|
|
|
|
gboolean
|
|
proto_enable_heuristic_by_name(const char *name, gboolean enable)
|
|
{
|
|
heur_dtbl_entry_t* heur = find_heur_dissector_by_unique_short_name(name);
|
|
if (heur != NULL) {
|
|
heur->enabled = enable;
|
|
return TRUE;
|
|
} else {
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
static void
|
|
disabled_protos_free(gpointer p, gpointer user_data _U_)
|
|
{
|
|
protocol_def* pd = (protocol_def*)p;
|
|
g_free(pd->name);
|
|
g_free(p);
|
|
}
|
|
|
|
/*
|
|
* Read the files that enable and disable protocols and heuristic
|
|
* dissectors. Report errors through the UI.
|
|
*/
|
|
void
|
|
read_enabled_and_disabled_lists(void)
|
|
{
|
|
char *gpath, *path;
|
|
int gopen_errno, gread_errno;
|
|
int open_errno, read_errno;
|
|
|
|
/*
|
|
* Read the global and personal disabled protocols files.
|
|
*/
|
|
read_protos_list(&gpath, &gopen_errno, &gread_errno,
|
|
&path, &open_errno, &read_errno,
|
|
DISABLED_PROTOCOLS_FILE_NAME,
|
|
&global_disabled_protos, &disabled_protos);
|
|
if (gpath != NULL) {
|
|
if (gopen_errno != 0) {
|
|
report_warning("Could not open global disabled protocols file\n\"%s\": %s.",
|
|
gpath, g_strerror(gopen_errno));
|
|
}
|
|
if (gread_errno != 0) {
|
|
report_warning("I/O error reading global disabled protocols file\n\"%s\": %s.",
|
|
gpath, g_strerror(gread_errno));
|
|
}
|
|
g_free(gpath);
|
|
gpath = NULL;
|
|
}
|
|
if (path != NULL) {
|
|
if (open_errno != 0) {
|
|
report_warning("Could not open your disabled protocols file\n\"%s\": %s.",
|
|
path, g_strerror(open_errno));
|
|
}
|
|
if (read_errno != 0) {
|
|
report_warning("I/O error reading your disabled protocols file\n\"%s\": %s.",
|
|
path, g_strerror(read_errno));
|
|
}
|
|
g_free(path);
|
|
path = NULL;
|
|
}
|
|
|
|
/*
|
|
* Read the global and personal enabled protocols files.
|
|
*/
|
|
read_protos_list(&gpath, &gopen_errno, &gread_errno,
|
|
&path, &open_errno, &read_errno,
|
|
ENABLED_PROTOCOLS_FILE_NAME,
|
|
&global_enabled_protos, &enabled_protos);
|
|
if (gpath != NULL) {
|
|
if (gopen_errno != 0) {
|
|
report_warning("Could not open global enabled protocols file\n\"%s\": %s.",
|
|
gpath, g_strerror(gopen_errno));
|
|
}
|
|
if (gread_errno != 0) {
|
|
report_warning("I/O error reading global enabled protocols file\n\"%s\": %s.",
|
|
gpath, g_strerror(gread_errno));
|
|
}
|
|
g_free(gpath);
|
|
gpath = NULL;
|
|
}
|
|
if (path != NULL) {
|
|
if (open_errno != 0) {
|
|
report_warning("Could not open your enabled protocols file\n\"%s\": %s.",
|
|
path, g_strerror(open_errno));
|
|
}
|
|
if (read_errno != 0) {
|
|
report_warning("I/O error reading your enabled protocols file\n\"%s\": %s.",
|
|
path, g_strerror(read_errno));
|
|
}
|
|
g_free(path);
|
|
path = NULL;
|
|
}
|
|
|
|
/*
|
|
* Read the global and personal heuristic dissector list files.
|
|
*/
|
|
read_heur_dissector_list(&gpath, &gopen_errno, &gread_errno,
|
|
&path, &open_errno, &read_errno);
|
|
if (gpath != NULL) {
|
|
if (gopen_errno != 0) {
|
|
report_warning("Could not open global heuristic dissectors file\n\"%s\": %s.",
|
|
gpath, g_strerror(gopen_errno));
|
|
}
|
|
if (gread_errno != 0) {
|
|
report_warning("I/O error reading global heuristic dissectors file\n\"%s\": %s.",
|
|
gpath, g_strerror(gread_errno));
|
|
}
|
|
g_free(gpath);
|
|
gpath = NULL;
|
|
}
|
|
if (path != NULL) {
|
|
if (open_errno != 0) {
|
|
report_warning("Could not open your heuristic dissectors file\n\"%s\": %s.",
|
|
path, g_strerror(open_errno));
|
|
}
|
|
if (read_errno != 0) {
|
|
report_warning("I/O error reading your heuristic dissectors file\n\"%s\": %s.",
|
|
path, g_strerror(read_errno));
|
|
}
|
|
g_free(path);
|
|
path = NULL;
|
|
}
|
|
|
|
/*
|
|
* Enable/disable protocols and heuristic dissectors as per the
|
|
* contents of the files we just read.
|
|
*/
|
|
set_protos_list(disabled_protos, global_disabled_protos, FALSE);
|
|
set_protos_list(enabled_protos, global_enabled_protos, TRUE);
|
|
set_disabled_heur_dissector_list();
|
|
}
|
|
|
|
/*
|
|
* Write out the lists of enabled and disabled protocols and heuristic
|
|
* dissectors to the corresponding files. Report errors through the UI.
|
|
*/
|
|
void
|
|
save_enabled_and_disabled_lists(void)
|
|
{
|
|
char *pf_dir_path;
|
|
char *pf_path;
|
|
int pf_save_errno;
|
|
|
|
/* Create the directory that holds personal configuration files, if
|
|
necessary. */
|
|
if (create_persconffile_dir(&pf_dir_path) == -1) {
|
|
report_failure("Can't create directory\n\"%s\"\nfor disabled protocols file: %s.",
|
|
pf_dir_path, g_strerror(errno));
|
|
g_free(pf_dir_path);
|
|
return;
|
|
}
|
|
|
|
save_protos_list(&pf_path, &pf_save_errno, DISABLED_PROTOCOLS_FILE_NAME,
|
|
NULL, disable_proto_list_check);
|
|
if (pf_path != NULL) {
|
|
report_failure("Could not save to your disabled protocols file\n\"%s\": %s.",
|
|
pf_path, g_strerror(pf_save_errno));
|
|
g_free(pf_path);
|
|
}
|
|
|
|
save_protos_list(&pf_path, &pf_save_errno, ENABLED_PROTOCOLS_FILE_NAME,
|
|
"#This file is for enabling protocols that are disabled by default",
|
|
enable_proto_list_check);
|
|
if (pf_path != NULL) {
|
|
report_failure("Could not save to your enabled protocols file\n\"%s\": %s.",
|
|
pf_path, g_strerror(pf_save_errno));
|
|
g_free(pf_path);
|
|
}
|
|
|
|
save_disabled_heur_dissector_list(&pf_path, &pf_save_errno);
|
|
if (pf_path != NULL) {
|
|
report_failure("Could not save to your disabled heuristic protocol file\n\"%s\": %s.",
|
|
pf_path, g_strerror(pf_save_errno));
|
|
g_free(pf_path);
|
|
}
|
|
}
|
|
|
|
void
|
|
cleanup_enabled_and_disabled_lists(void)
|
|
{
|
|
g_list_foreach(global_disabled_heuristics, disabled_protos_free, NULL);
|
|
g_list_free(global_disabled_heuristics);
|
|
g_list_foreach(disabled_heuristics, disabled_protos_free, NULL);
|
|
g_list_free(disabled_heuristics);
|
|
g_list_foreach(global_disabled_protos, disabled_protos_free, NULL);
|
|
g_list_free(global_disabled_protos);
|
|
g_list_foreach(disabled_protos, disabled_protos_free, NULL);
|
|
g_list_free(disabled_protos);
|
|
}
|
|
|
|
/*
|
|
* Editor modelines - http://www.wireshark.org/tools/modelines.html
|
|
*
|
|
* Local Variables:
|
|
* c-basic-offset: 2
|
|
* tab-width: 8
|
|
* indent-tabs-mode: nil
|
|
* End:
|
|
*
|
|
* ex: set shiftwidth=2 tabstop=8 expandtab:
|
|
* :indentSize=2:tabSize=8:noTabs=true:
|
|
*/
|