/* disabled_protos.c
- * Code for reading and writing the disabled protocols file.
+ * 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.
+ * SPDX-License-Identifier: GPL-2.0-or-later
*/
#include "config.h"
#include <stdio.h>
#include <string.h>
-#include <ctype.h>
#include <errno.h>
-#ifdef HAVE_UNISTD_H
-#include <unistd.h>
-#endif
-
#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/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;
-#define GLOBAL_PROTOCOLS_FILE_NAME "disabled_protos"
-#define PROTOCOLS_FILE_NAME "disabled_protos"
+/*
+ * 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
+#define INIT_BUF_SIZE 128
static void
discard_existing_list (GList **flp)
}
}
+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;
+ }
+}
+
/*
- * Read in a list of disabled protocols.
+ * 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 "*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.
+ * 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;
-static int read_disabled_protos_list_file(const char *ff_path, FILE *ff,
- GList **flp);
+ *pref_path_return = NULL; /* assume no error */
-void
-read_disabled_protos_list(char **gpath_return, int *gopen_errno_return,
- int *gread_errno_return,
- char **path_return, int *open_errno_return,
- int *read_errno_return)
+ 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. */
+ 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 */
+ 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(GLOBAL_PROTOCOLS_FILE_NAME);
+ gff_path = get_datafile_path(filename);
/* If we already have a list of protocols, discard it. */
- discard_existing_list (&global_disabled_protos);
+ 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_disabled_protos_list_file(gff_path, ff,
- &global_disabled_protos);
+ 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. */
}
/* Construct the pathname of the user's disabled protocols file. */
- ff_path = get_persconffile_path(PROTOCOLS_FILE_NAME, TRUE);
+ ff_path = get_persconffile_path(filename, TRUE);
/* If we already have a list of protocols, discard it. */
- discard_existing_list (&disabled_protos);
+ 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_disabled_protos_list_file(ff_path, ff, &disabled_protos);
+ 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. */
}
}
+/************************************************************************
+ * 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_disabled_protos_list_file(const char *ff_path, FILE *ff,
- GList **flp)
+read_heur_dissector_list_file(const char *ff_path, FILE *ff, GList **flp)
{
- protocol_def *prot;
+ heur_protocol_def *heur;
int c;
- char *prot_name;
- int prot_name_len;
- int prot_name_index;
+ char *heuristic_name;
+ int heuristic_name_len;
+ int name_index;
+ gboolean parse_enabled;
+ gboolean enabled;
int line = 1;
/* Allocate the protocol name buffer. */
- prot_name_len = INIT_BUF_SIZE;
- prot_name = (char *)g_malloc(prot_name_len + 1);
+ 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 = getc(ff)) != EOF && isspace(c)) {
+ while ((c = ws_getc_unlocked(ff)) != EOF && g_ascii_isspace(c)) {
if (c == '\n') {
- /* Blank line. */
- continue;
+ /* Blank line. */
+ continue;
}
}
if (c == EOF) {
if (ferror(ff))
- goto error; /* I/O error */
+ goto error; /* I/O error */
else
- break; /* Nothing more to read */
+ break; /* Nothing more to read */
}
- ungetc(c, ff); /* Unread the non-white-space character. */
+ ungetc(c, ff); /* Unread the non-white-space character. */
/* Get the name of the protocol. */
- prot_name_index = 0;
+ name_index = 0;
+ enabled = FALSE;
+ parse_enabled = FALSE;
for (;;) {
- c = getc(ff);
+ c = ws_getc_unlocked(ff);
if (c == EOF)
- break; /* End of file, or I/O error */
- if (isspace(c))
- break; /* Trailing white space, or end of line. */
+ 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. */
+ 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 (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);
+ 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);
}
- prot_name[prot_name_index] = c;
- prot_name_index++;
+ heuristic_name[name_index] = c;
+ name_index++;
}
- if (isspace(c) && c != '\n') {
+ if (g_ascii_isspace(c) && c != '\n') {
/* Skip over trailing white space. */
- while ((c = getc(ff)) != EOF && c != '\n' && isspace(c))
+ 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. */
- g_warning("'%s' line %d has extra stuff after the protocol name.",
- ff_path, line);
+ /* Non-white-space after the protocol name; warn about it,
+ in case we come up with a reason to use it. */
+ 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 = getc(ff)) != EOF && c != '\n')
+ while ((c = ws_getc_unlocked(ff)) != EOF && c != '\n')
;
}
if (c == EOF) {
if (ferror(ff))
- goto error; /* I/O error */
+ goto error; /* I/O error */
else {
- /* EOF, not error; no newline seen before EOF */
- g_warning("'%s' line %d doesn't have a newline.", ff_path,
- line);
+ /* EOF, not error; no newline seen before EOF */
+ g_warning("'%s' line %d doesn't have a newline.", ff_path,
+ line);
}
- break; /* nothing more to read */
+ break; /* nothing more to read */
}
/* Null-terminate the protocol name. */
- if (prot_name_index >= prot_name_len) {
+ if (name_index >= heuristic_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);
+ heuristic_name_len *= 2;
+ heuristic_name = (char *)g_realloc(heuristic_name, heuristic_name_len + 1);
}
- prot_name[prot_name_index] = '\0';
+ heuristic_name[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);
+ /* 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(prot_name);
+ g_free(heuristic_name);
return 0;
error:
+ g_free(heuristic_name);
return errno;
}
-/*
- * Disable protocols as per the stored configuration
- */
-void
-set_disabled_protos_list(void)
+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)
{
- gint i;
- GList *fl_ent;
- protocol_def *prot;
+ int err;
+ char *gff_path, *ff_path;
+ FILE *ff;
- /*
- * assume all protocols are enabled by default
- */
- if (disabled_protos == NULL)
- goto skip;
+ /* If we already have a list of protocols, discard it. */
+ heur_discard_existing_list(&global_disabled_heuristics);
- fl_ent = g_list_first(disabled_protos);
+ /* Construct the pathname of the global disabled heuristic dissectors file. */
+ gff_path = get_datafile_path(HEURISTICS_FILE_NAME);
- 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, FALSE);
- }
+ /* 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);
+ }
- fl_ent = fl_ent->next;
+ /* 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);
}
+}
-skip:
- if (global_disabled_protos == NULL)
- return;
+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);
+}
- fl_ent = g_list_first(global_disabled_protos);
+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;
- 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, FALSE);
- proto_set_cant_toggle(i);
- }
- }
+ /* Write out the heuristic short name and its enabled state */
+ fprintf(ff, "%s,%d\n", dtbl_entry->short_name, dtbl_entry->enabled ? 1 : 0);
+}
- fl_ent = fl_ent->next;
+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);
}
}
-/*
- * Write out a list of disabled protocols.
- *
- * 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.
- */
-void
-save_disabled_protos_list(char **pref_path_return, int *errno_return)
+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;
- gint i;
- protocol_t *protocol;
- void *cookie;
- *pref_path_return = NULL; /* assume no error */
+ *pref_path_return = NULL; /* assume no error */
- ff_path = get_persconffile_path(PROTOCOLS_FILE_NAME, TRUE);
+ 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
return;
}
- /* Iterate over all the protocols */
+ /* 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);
- 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 (proto_is_protocol_enabled(protocol)) {
- continue;
- }
-
- /* Write out the protocol name. */
- fprintf(ff, "%s\n", proto_get_protocol_filter_name(i));
- }
+ /* 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;
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:
+ */
git.samba.org / metze / wireshark / wip.git / blobdiff