diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-10 20:34:10 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-10 20:34:10 +0000 |
commit | e4ba6dbc3f1e76890b22773807ea37fe8fa2b1bc (patch) | |
tree | 68cb5ef9081156392f1dd62a00c6ccc1451b93df /epan/disabled_protos.c | |
parent | Initial commit. (diff) | |
download | wireshark-e4ba6dbc3f1e76890b22773807ea37fe8fa2b1bc.tar.xz wireshark-e4ba6dbc3f1e76890b22773807ea37fe8fa2b1bc.zip |
Adding upstream version 4.2.2.upstream/4.2.2
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'epan/disabled_protos.c')
-rw-r--r-- | epan/disabled_protos.c | 1122 |
1 files changed, 1122 insertions, 0 deletions
diff --git a/epan/disabled_protos.c b/epan/disabled_protos.c new file mode 100644 index 00000000..fa0bec43 --- /dev/null +++ b/epan/disabled_protos.c @@ -0,0 +1,1122 @@ +/* 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 + * + * SPDX-License-Identifier: GPL-2.0-or-later + */ + +#include "config.h" +#define WS_LOG_DOMAIN LOG_DOMAIN_EPAN + +#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/report_message.h> +#include <wsutil/wslog.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; + +static gboolean unsaved_changes = FALSE; + +#define INIT_BUF_SIZE 128 + +gboolean +enabled_protos_unsaved_changes(void) +{ + return unsaved_changes; +} + +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 = ws_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_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_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 = g_new(protocol_def, 1); + 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 + */ +gboolean +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) { + unsaved_changes = TRUE; + proto_set_decoding(proto_id, FALSE); + } + } + return TRUE; + } + else if (!strcmp(name, "ALL")) { + unsaved_changes = TRUE; + proto_disable_all(); + return TRUE; + } + else { + return 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) + ************************************************************************/ + +gboolean +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(protocol) == FALSE)) { + if (proto_can_toggle_protocol(proto_id) == TRUE) { + unsaved_changes = TRUE; + proto_set_decoding(proto_id, TRUE); + } + } + return TRUE; + } + else if (!strcmp(name, "ALL")) { + unsaved_changes = TRUE; + proto_reenable_all(); + return TRUE; + } + else { + return FALSE; + } +} + +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_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_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 = g_new(heur_protocol_def, 1); + 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 = ws_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); +} + +static gboolean +proto_set_heuristic_by_name(const char *name, gboolean enable) +{ + heur_dtbl_entry_t* heur = find_heur_dissector_by_unique_short_name(name); + if (heur != NULL) { + unsaved_changes |= (heur->enabled != enable); + heur->enabled = enable; + return TRUE; + } else { + return FALSE; + } +} + +gboolean +proto_enable_heuristic_by_name(const char *name) +{ + return proto_set_heuristic_by_name(name, TRUE); +} + +gboolean +proto_disable_heuristic_by_name(const char *name) +{ + return proto_set_heuristic_by_name(name, 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(); + unsaved_changes = FALSE; +} + +/* + * 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; + gboolean ok = TRUE; + + /* 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); + ok = FALSE; + } + + 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); + ok = FALSE; + } + + 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); + ok = FALSE; + } + + if (ok) + unsaved_changes = FALSE; +} + +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 - https://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: + */ |