HACK: pinfo->private_data points to smb_info again

[wireshark-wip.git] / epan / packet.c

/* packet.c
 * Routines for packet disassembly
 *
 * $Id$
 *
 * 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 <glib.h>

#include <stdio.h>
#include <stdlib.h>

#ifdef HAVE_STDARG_H
#include <stdarg.h>
#endif

#include <string.h>
#include <ctype.h>
#include <time.h>

#include "packet.h"
#include "timestamp.h"

#include "atalk-utils.h"
#include "sna-utils.h"
#include "osi-utils.h"
#include "to_str.h"

#include "addr_resolv.h"
#include "tvbuff.h"
#include "plugins.h"
#include "epan_dissect.h"

#include "emem.h"
#include "wmem/wmem.h"

#include <epan/exceptions.h>
#include <epan/reassemble.h>
#include <epan/stream.h>
#include <epan/expert.h>

static gint proto_malformed = -1;
static dissector_handle_t frame_handle = NULL;
static dissector_handle_t data_handle = NULL;

/**
 * A data source.
 * Has a tvbuff and a name.
 */
struct data_source {
        tvbuff_t *tvb;
        char *name;
};

/*
 * A dissector table.
 *
 * "hash_table" is a hash table, indexed by port number, supplying
 * a "struct dtbl_entry"; it records what dissector is assigned to
 * that uint or string value in that table.
 *
 * "dissector_handles" is a list of all dissectors that *could* be
 * used in that table; not all of them are necessarily in the table,
 * as they may be for protocols that don't have a fixed uint value,
 * e.g. for TCP or UDP port number tables and protocols with no fixed
 * port number.
 *
 * "ui_name" is the name the dissector table has in the user interface.
 *
 * "type" is a field type giving the width of the uint value for that
 * dissector table, if it's a uint dissector table.
 *
 * "base" is the base in which to display the uint value for that
 * dissector table, if it's a uint dissector table.
 */
struct dissector_table {
        GHashTable      *hash_table;
        GSList          *dissector_handles;
        const char      *ui_name;
        ftenum_t        type;
        int             base;
};

static GHashTable *dissector_tables = NULL;

/*
 * List of registered dissectors.
 */
static GHashTable *registered_dissectors = NULL;

static GHashTable *heur_dissector_lists = NULL;

static void
destroy_heuristic_dissector_entry(gpointer data, gpointer user_data _U_)
{
        g_slice_free(heur_dtbl_entry_t, data);
}

static void
destroy_heuristic_dissector_list(void *data)
{
        GSList **list = (GSList**)data;

        g_slist_foreach(*list, destroy_heuristic_dissector_entry, NULL);
        g_slist_free(*list);
        *list = NULL;
}

static void
destroy_dissector_table(void *data)
{
        struct dissector_table *table = (struct dissector_table *)data;

        g_hash_table_destroy(table->hash_table);
        g_slist_free(table->dissector_handles);
        g_slice_free(struct dissector_table, data);
}

void
packet_init(void)
{
        dissector_tables = g_hash_table_new_full(g_str_hash, g_str_equal,
                        NULL, destroy_dissector_table);

        registered_dissectors = g_hash_table_new_full(g_str_hash, g_str_equal,
                        NULL, NULL);

        heur_dissector_lists = g_hash_table_new_full(g_str_hash, g_str_equal,
                        NULL, destroy_heuristic_dissector_list);
}

void
packet_cache_proto_handles(void)
{
        frame_handle = find_dissector("frame");
        g_assert(frame_handle != NULL);

        data_handle = find_dissector("data");
        g_assert(data_handle != NULL);

        proto_malformed = proto_get_id_by_filter_name("_ws.malformed");
        g_assert(proto_malformed != -1);
}

void
packet_cleanup(void)
{
        g_hash_table_destroy(dissector_tables);
        g_hash_table_destroy(registered_dissectors);
        g_hash_table_destroy(heur_dissector_lists);
}

/*
 * Given a tvbuff, and a length from a packet header, adjust the length
 * of the tvbuff to reflect the specified length.
 */
void
set_actual_length(tvbuff_t *tvb, const guint specified_len)
{
        if (specified_len < tvb_reported_length(tvb)) {
                /* Adjust the length of this tvbuff to include only the specified
                   payload length.

                   The dissector above the one calling us (the dissector above is
                   probably us) may use that to determine how much of its packet
                   was padding. */
                tvb_set_reported_length(tvb, specified_len);
        }
}

/* Allow protocols to register "init" routines, which are called before
   we make a pass through a capture file and dissect all its packets
   (e.g., when we read in a new capture file, or run a "filter packets"
   or "colorize packets" pass over the current capture file). */
static GSList *init_routines;

void
register_init_routine(void (*func)(void))
{
        init_routines = g_slist_prepend(init_routines, (gpointer)func);
}

typedef void (*void_func_t)(void);

/* Initialize all data structures used for dissection. */
static void
call_init_routine(gpointer routine, gpointer dummy _U_)
{
        void_func_t func = (void_func_t)routine;
        (*func)();
}

/*
 * XXX - for now, these are the same; the "init" routines free whatever
 * stuff is left over from any previous dissection, and then initialize
 * their tables.
 *
 * We should probably split that into "init" and "cleanup" routines, for
 * cleanliness' sake.
 */
void
init_dissection(void)
{
        /* Reclaim and reinitialize all memory of seasonal scope */
        se_free_all();

        wmem_enter_file_scope();

        /*
         * Reinitialize resolution information. We do initialization here in
         * case we need to resolve between captures.
         */
        host_name_lookup_init();

        /* Initialize the table of conversations. */
        epan_conversation_init();

        /* Initialize the table of circuits. */
        epan_circuit_init();

        /* Initialize protocol-specific variables. */
        g_slist_foreach(init_routines, &call_init_routine, NULL);

        /* Initialize the stream-handling tables */
        stream_init();

        /* Initialize the expert infos */
        expert_packet_init();
}

void
cleanup_dissection(void)
{
        /* Cleanup the table of conversations. Do this before freeing seasonal
         * memory (at least until conversation's use of g_slist is changed).
         */
        epan_conversation_cleanup();

        /* Reclaim all memory of seasonal scope */
        se_free_all();

        /* Cleanup the table of circuits. */
        epan_circuit_cleanup();

        /* TODO: Introduce cleanup_routines */
        /* Cleanup protocol-specific variables. */
        g_slist_foreach(init_routines, &call_init_routine, NULL);

        /* Cleanup the stream-handling tables */
        stream_cleanup();

        /* Initialize the expert infos */
        expert_packet_cleanup();

        wmem_leave_file_scope();

        /*
         * Reinitialize resolution information. We do initialization here in
         * case we need to resolve between captures.
         */
        host_name_lookup_cleanup();
}

/* Allow protocols to register a "cleanup" routine to be
 * run after the initial sequential run through the packets.
 * Note that the file can still be open after this; this is not
 * the final cleanup. */
static GSList *postseq_cleanup_routines;

void
register_postseq_cleanup_routine(void_func_t func)
{
        postseq_cleanup_routines = g_slist_prepend(postseq_cleanup_routines,
                        (gpointer)func);
}

/* Call all the registered "postseq_cleanup" routines. */
static void
call_postseq_cleanup_routine(gpointer routine, gpointer dummy _U_)
{
        void_func_t func = (void_func_t)routine;
        (*func)();
}

void
postseq_cleanup_all_protocols(void)
{
        g_slist_foreach(postseq_cleanup_routines,
                        &call_postseq_cleanup_routine, NULL);
}

/*
 * Add a new data source to the list of data sources for a frame, given
 * the tvbuff for the data source and its name.
 */
void
add_new_data_source(packet_info *pinfo, tvbuff_t *tvb, const char *name)
{
        struct data_source *src;

        src = g_slice_new(struct data_source);
        src->tvb = tvb;
        src->name = g_strdup(name);
        pinfo->data_src = g_slist_append(pinfo->data_src, src);
}

const char*
get_data_source_name(const struct data_source *src)
{
        guint length = tvb_length(src->tvb);

        return ep_strdup_printf("%s (%u byte%s)", src->name, length,
                                plurality(length, "", "s"));
}

tvbuff_t *
get_data_source_tvb(const struct data_source *src)
{
        return src->tvb;
}

/*
 * Free up a frame's list of data sources.
 */
void
free_data_sources(packet_info *pinfo)
{
        if (pinfo->data_src) {
                GSList *l;

                for (l = pinfo->data_src; l; l = l->next) {
                        struct data_source *src = (struct data_source *)l->data;

                        g_free(src->name);
                        g_slice_free(struct data_source, src);
                }
                g_slist_free(pinfo->data_src);
                pinfo->data_src = NULL;
        }
}

void
mark_frame_as_depended_upon(packet_info *pinfo, guint32 frame_num)
{
        /* Don't mark a frame as dependent on itself */
        if (frame_num != PINFO_FD_NUM(pinfo)) {
                pinfo->dependent_frames = g_slist_prepend(pinfo->dependent_frames, GUINT_TO_POINTER(frame_num));
        }
}

/* Allow dissectors to register a "final_registration" routine
 * that is run like the proto_register_XXX() routine, but at the
 * end of the epan_init() function; that is, *after* all other
 * subsystems, like dfilters, have finished initializing. This is
 * useful for dissector registration routines which need to compile
 * display filters. dfilters can't initialize itself until all protocols
 * have registered themselves. */
static GSList *final_registration_routines;

void
register_final_registration_routine(void (*func)(void))
{
        final_registration_routines = g_slist_prepend(final_registration_routines,
                        (gpointer)func);
}

/* Call all the registered "final_registration" routines. */
static void
call_final_registration_routine(gpointer routine, gpointer dummy _U_)
{
        void_func_t func = (void_func_t)routine;

        (*func)();
}

void
final_registration_all_protocols(void)
{
        g_slist_foreach(final_registration_routines,
                        &call_final_registration_routine, NULL);
}


/* Creates the top-most tvbuff and calls dissect_frame() */
void
dissect_packet(epan_dissect_t *edt, struct wtap_pkthdr *phdr,
               tvbuff_t *tvb, frame_data *fd, column_info *cinfo)
{
        if (cinfo != NULL)
                col_init(cinfo, edt->session);
        edt->pi.epan = edt->session;
        /* edt->pi.pool created in epan_dissect_init() */
        edt->pi.current_proto = "<Missing Protocol Name>";
        edt->pi.cinfo = cinfo;
        edt->pi.fd    = fd;
        edt->pi.phdr  = phdr;
        edt->pi.pseudo_header = &phdr->pseudo_header;
        edt->pi.dl_src.type   = AT_NONE;
        edt->pi.dl_dst.type   = AT_NONE;
        edt->pi.net_src.type  = AT_NONE;
        edt->pi.net_dst.type  = AT_NONE;
        edt->pi.src.type = AT_NONE;
        edt->pi.dst.type = AT_NONE;
        edt->pi.ctype = CT_NONE;
        edt->pi.noreassembly_reason = "";
        edt->pi.ptype = PT_NONE;
        edt->pi.p2p_dir = P2P_DIR_UNKNOWN;
        edt->pi.dcetransporttype = -1;
        edt->pi.annex_a_used = MTP2_ANNEX_A_USED_UNKNOWN;
        edt->pi.dcerpc_procedure_name="";
        edt->pi.link_dir = LINK_DIR_UNKNOWN;
        edt->pi.layers = wmem_list_new(edt->pi.pool);
        edt->tvb = tvb;


        frame_delta_abs_time(edt->session, fd, fd->frame_ref_num, &edt->pi.rel_ts);

        /* pkt comment use first user, later from phdr */
        if (fd->flags.has_user_comment) 
                edt->pi.pkt_comment = epan_get_user_comment(edt->session, fd);
        else if (fd->flags.has_phdr_comment)
                edt->pi.pkt_comment = phdr->opt_comment;

        /* to enable decode as for ethertype=0x0000 (fix for bug 4721) */
        edt->pi.ethertype = G_MAXINT;

        EP_CHECK_CANARY(("before dissecting frame %d",fd->num));

        TRY {
                /* Add this tvbuffer into the data_src list */
                add_new_data_source(&edt->pi, edt->tvb, "Frame");

                /* Even though dissect_frame() catches all the exceptions a
                 * sub-dissector can throw, dissect_frame() itself may throw
                 * a ReportedBoundsError in bizarre cases. Thus, we catch the exception
                 * in this function. */
                call_dissector(frame_handle, edt->tvb, &edt->pi, edt->tree);

        }
        CATCH(BoundsError) {
                g_assert_not_reached();
        }
        CATCH2(FragmentBoundsError, ReportedBoundsError) {
                proto_tree_add_protocol_format(edt->tree, proto_malformed, edt->tvb, 0, 0,
                                               "[Malformed Frame: Packet Length]" );
        }
        ENDTRY;

        EP_CHECK_CANARY(("after dissecting frame %d",fd->num));

        fd->flags.visited = 1;
}

/*********************** code added for sub-dissector lookup *********************/

/*
 * A dissector handle.
 */
struct dissector_handle {
        const char      *name;          /* dissector name */
        gboolean        is_new;         /* TRUE if new-style dissector */
        union {
                dissector_t     old;
                new_dissector_t new_d;
        } dissector;
        protocol_t      *protocol;
};

/* This function will return
 * old style dissector :
 *   length of the payload or 1 of the payload is empty
 * new dissector :
 *   >0  this protocol was successfully dissected and this was this protocol.
 *   0   this packet did not match this protocol.
 *
 * The only time this function will return 0 is if it is a new style dissector
 * and if the dissector rejected the packet.
 */
static int
call_dissector_through_handle(dissector_handle_t handle, tvbuff_t *tvb,
                              packet_info *pinfo, proto_tree *tree, void *data)
{
        const char *saved_proto;
        int         ret;

        saved_proto = pinfo->current_proto;

        if (handle->protocol != NULL) {
                pinfo->current_proto =
                        proto_get_protocol_short_name(handle->protocol);
        }

        if (handle->is_new) {
                EP_CHECK_CANARY(("before calling handle->dissector.new_d for %s",handle->name));
                ret = (*handle->dissector.new_d)(tvb, pinfo, tree, data);
                EP_CHECK_CANARY(("after calling handle->dissector.new_d for %s",handle->name));
        } else {
                EP_CHECK_CANARY(("before calling handle->dissector.old for %s",handle->name));
                (*handle->dissector.old)(tvb, pinfo, tree);
                EP_CHECK_CANARY(("after calling handle->dissector.old for %s",handle->name));
                ret = tvb_length(tvb);
                if (ret == 0) {
                        /*
                         * XXX - a tvbuff can have 0 bytes of data in
                         * it, so we have to make sure we don't return
                         * 0.
                         */
                        ret = 1;
                }
        }

        pinfo->current_proto = saved_proto;

        return ret;
}

/*
 * Call a dissector through a handle.
 * If the protocol for that handle isn't enabled, return 0 without
 * calling the dissector.
 * Otherwise, if the handle refers to a new-style dissector, call the
 * dissector and return its return value, otherwise call it and return
 * the length of the tvbuff pointed to by the argument.
 */

static int
call_dissector_work_error(dissector_handle_t handle, tvbuff_t *tvb,
                          packet_info *pinfo_arg, proto_tree *tree, void *);

static int
call_dissector_work(dissector_handle_t handle, tvbuff_t *tvb, packet_info *pinfo_arg,
                    proto_tree *tree, gboolean add_proto_name, void *data)
{
        packet_info *pinfo = pinfo_arg;
        const char  *saved_proto;
        guint16      saved_can_desegment;
        int          ret;
        guint        saved_layers_len = 0;

        if (handle->protocol != NULL &&
            !proto_is_protocol_enabled(handle->protocol)) {
                /*
                 * The protocol isn't enabled.
                 */
                return 0;
        }

        saved_proto = pinfo->current_proto;
        saved_can_desegment = pinfo->can_desegment;
        saved_layers_len = wmem_list_count(pinfo->layers);

        /*
         * can_desegment is set to 2 by anyone which offers the
         * desegmentation api/service.
         * Then everytime a subdissector is called it is decremented
         * by one.
         * Thus only the subdissector immediately on top of whoever
         * offers this service can use it.
         * We save the current value of "can_desegment" for the
         * benefit of TCP proxying dissectors such as SOCKS, so they
         * can restore it and allow the dissectors they call to use
         * the desegmentation service.
         */
        pinfo->saved_can_desegment = saved_can_desegment;
        pinfo->can_desegment = saved_can_desegment-(saved_can_desegment>0);
        if (handle->protocol != NULL) {
                pinfo->current_proto =
                        proto_get_protocol_short_name(handle->protocol);

                /*
                 * Add the protocol name to the layers
                 * if not told not to. Asn2wrs generated dissectors may be added multiple times otherwise.
                 */
                if (add_proto_name) {
                        pinfo->curr_layer_num++;
                        wmem_list_append(pinfo->layers, GINT_TO_POINTER(proto_get_id(handle->protocol)));
                }
        }

        if (pinfo->flags.in_error_pkt) {
                ret = call_dissector_work_error(handle, tvb, pinfo, tree, data);
        } else {
                /*
                 * Just call the subdissector.
                 */
                ret = call_dissector_through_handle(handle, tvb, pinfo, tree, data);
        }
        if (ret == 0) {
                /*
                 * That dissector didn't accept the packet, so
                 * remove its protocol's name from the list
                 * of protocols.
                 */
                while (wmem_list_count(pinfo->layers) > saved_layers_len) {
                        wmem_list_remove_frame(pinfo->layers, wmem_list_tail(pinfo->layers));
                }
        }
        pinfo->current_proto = saved_proto;
        pinfo->can_desegment = saved_can_desegment;
        return ret;
}


static int
call_dissector_work_error(dissector_handle_t handle, tvbuff_t *tvb,
                          packet_info *pinfo_arg, proto_tree *tree, void *data)
{
        packet_info  *pinfo = pinfo_arg;
        const char   *saved_proto;
        guint16       saved_can_desegment;
        volatile int  ret = 0;
        gboolean      save_writable;
        address       save_dl_src;
        address       save_dl_dst;
        address       save_net_src;
        address       save_net_dst;
        address       save_src;
        address       save_dst;

        /*
        * This isn't a packet being transported inside
        * the protocol whose dissector is calling us,
        * it's a copy of a packet that caused an error
        * in some protocol included in a packet that
        * reports the error (e.g., an ICMP Unreachable
        * packet).
        */

        /*
        * Save the current state of the writability of
        * the columns, and restore them after the
        * dissector returns, so that the columns
        * don't reflect the packet that got the error,
        * they reflect the packet that reported the
        * error.
        */
        saved_proto = pinfo->current_proto;
        saved_can_desegment = pinfo->can_desegment;

        save_writable = col_get_writable(pinfo->cinfo);
        col_set_writable(pinfo->cinfo, FALSE);
        save_dl_src   = pinfo->dl_src;
        save_dl_dst   = pinfo->dl_dst;
        save_net_src  = pinfo->net_src;
        save_net_dst  = pinfo->net_dst;
        save_src      = pinfo->src;
        save_dst      = pinfo->dst;

        /* Dissect the contained packet. */
        TRY {
                ret = call_dissector_through_handle(handle, tvb,pinfo, tree, data);
        }
        CATCH(BoundsError) {
                /*
                * Restore the column writability and addresses.
                */
                col_set_writable(pinfo->cinfo, save_writable);
                pinfo->dl_src  = save_dl_src;
                pinfo->dl_dst  = save_dl_dst;
                pinfo->net_src = save_net_src;
                pinfo->net_dst = save_net_dst;
                pinfo->src     = save_src;
                pinfo->dst     = save_dst;

                /*
                * Restore the current protocol, so any
                * "Short Frame" indication reflects that
                * protocol, not the protocol for the
                * packet that got the error.
                */
                pinfo->current_proto = saved_proto;

                /*
                * Restore the desegmentability state.
                */
                pinfo->can_desegment = saved_can_desegment;

                /*
                * Rethrow the exception, so this will be
                * reported as a short frame.
                */
                RETHROW;
        }
        CATCH2(FragmentBoundsError, ReportedBoundsError) {
                /*
                * "ret" wasn't set because an exception was thrown
                * before "call_dissector_through_handle()" returned.
                * As it called something, at least one dissector
                * accepted the packet, and, as an exception was
                * thrown, not only was all the tvbuff dissected,
                * a dissector tried dissecting past the end of
                * the data in some tvbuff, so we'll assume that
                * the entire tvbuff was dissected.
                */
                ret = tvb_length(tvb);
        }
        ENDTRY;

        col_set_writable(pinfo->cinfo, save_writable);
        pinfo->dl_src  = save_dl_src;
        pinfo->dl_dst  = save_dl_dst;
        pinfo->net_src = save_net_src;
        pinfo->net_dst = save_net_dst;
        pinfo->src     = save_src;
        pinfo->dst     = save_dst;
        pinfo->want_pdu_tracking = 0;
        return ret;
}

/*
 * An entry in the hash table portion of a dissector table.
 */
struct dtbl_entry {
        dissector_handle_t initial;
        dissector_handle_t current;
};

/* Finds a dissector table by table name. */
dissector_table_t
find_dissector_table(const char *name)
{
        return (dissector_table_t)g_hash_table_lookup( dissector_tables, name );
}

/* Find an entry in a uint dissector table. */
static dtbl_entry_t *
find_uint_dtbl_entry(dissector_table_t sub_dissectors, const guint32 pattern)
{
        switch (sub_dissectors->type) {

        case FT_UINT8:
        case FT_UINT16:
        case FT_UINT24:
        case FT_UINT32:
                /*
                 * You can do a uint lookup in these tables.
                 */
                break;

        default:
                /*
                 * But you can't do a uint lookup in any other types
                 * of tables.
                 */
                g_assert_not_reached();
        }

        /*
         * Find the entry.
         */
        return (dtbl_entry_t *)g_hash_table_lookup(sub_dissectors->hash_table,
                                   GUINT_TO_POINTER(pattern));
}

#if 0
static void
dissector_add_uint_sanity_check(const char *name, guint32 pattern, dissector_handle_t handle, dissector_table_t sub_dissectors)
{
        dtbl_entry_t *dtbl_entry;

        if (pattern == 0) {
                g_warning("%s: %s registering using a pattern of 0",
                          name, proto_get_protocol_filter_name(proto_get_id(handle->protocol)));
        }

        dtbl_entry = g_hash_table_lookup(sub_dissectors->hash_table, GUINT_TO_POINTER(pattern));
        if (dtbl_entry != NULL) {
                g_warning("%s: %s registering using pattern %d already registered by %s",
                          name, proto_get_protocol_filter_name(proto_get_id(handle->protocol)),
                          pattern, proto_get_protocol_filter_name(proto_get_id(dtbl_entry->initial->protocol)));
        }
}
#endif

/* Add an entry to a uint dissector table. */
void
dissector_add_uint(const char *name, const guint32 pattern, dissector_handle_t handle)
{
        dissector_table_t  sub_dissectors;
        dtbl_entry_t      *dtbl_entry;

        sub_dissectors = find_dissector_table(name);

        /*
         * Make sure the dissector table exists.
         */
        if (sub_dissectors == NULL) {
                fprintf(stderr, "OOPS: dissector table \"%s\" doesn't exist\n",
                    name);
                fprintf(stderr, "Protocol being registered is \"%s\"\n",
                    proto_get_protocol_long_name(handle->protocol));
                if (getenv("WIRESHARK_ABORT_ON_DISSECTOR_BUG") != NULL)
                        abort();
                return;
        }

        /* sanity checks */
        g_assert(handle!=NULL);
        switch (sub_dissectors->type) {

        case FT_UINT8:
        case FT_UINT16:
        case FT_UINT24:
        case FT_UINT32:
                /*
                 * You can do a uint lookup in these tables.
                 */
                break;

        default:
                /*
                 * But you can't do a uint lookup in any other types
                 * of tables.
                 */
                g_assert_not_reached();
        }

#if 0
        dissector_add_uint_sanity_check(name, pattern, handle, sub_dissectors);
#endif

        dtbl_entry = (dtbl_entry_t *)g_malloc(sizeof (dtbl_entry_t));
        dtbl_entry->current = handle;
        dtbl_entry->initial = dtbl_entry->current;

        /* do the table insertion */
        g_hash_table_insert( sub_dissectors->hash_table,
                             GUINT_TO_POINTER( pattern), (gpointer)dtbl_entry);

        /*
         * Now add it to the list of handles that could be used with this
         * table, because it *is* being used with this table.
         */
        dissector_add_handle(name, handle);
}



void dissector_add_uint_range(const char *abbrev, range_t *range,
                              dissector_handle_t handle)
{
        guint32 i, j;

        if (range) {
                for (i = 0; i < range->nranges; i++) {
                        for (j = range->ranges[i].low; j <= range->ranges[i].high; j++)
                                dissector_add_uint(abbrev, j, handle);
                }
        }
}

/* Delete the entry for a dissector in a uint dissector table
   with a particular pattern. */

/* NOTE: this doesn't use the dissector call variable. It is included to */
/*      be consistant with the dissector_add_uint and more importantly to be used */
/*      if the technique of adding a temporary dissector is implemented.  */
/*      If temporary dissectors are deleted, then the original dissector must */
/*      be available. */
void
dissector_delete_uint(const char *name, const guint32 pattern,
        dissector_handle_t handle _U_)
{
        dissector_table_t sub_dissectors = find_dissector_table( name);
        dtbl_entry_t *dtbl_entry;

        /* sanity check */
        g_assert( sub_dissectors);

        /*
         * Find the entry.
         */
        dtbl_entry = find_uint_dtbl_entry(sub_dissectors, pattern);

        if (dtbl_entry != NULL) {
                /*
                 * Found - remove it.
                 */
                g_hash_table_remove(sub_dissectors->hash_table,
                                    GUINT_TO_POINTER(pattern));
        }
}

void dissector_delete_uint_range(const char *abbrev, range_t *range,
                                 dissector_handle_t handle)
{
        guint32 i, j;

        if (range) {
                for (i = 0; i < range->nranges; i++) {
                        for (j = range->ranges[i].low; j <= range->ranges[i].high; j++)
                                dissector_delete_uint(abbrev, j, handle);
                }
        }
}

static gboolean
dissector_delete_all_check (gpointer key _U_, gpointer value, gpointer user_data)
{
        dtbl_entry_t *dtbl_entry = (dtbl_entry_t *) value;
        dissector_handle_t handle = (dissector_handle_t) user_data;

        return (proto_get_id (dtbl_entry->current->protocol) == proto_get_id (handle->protocol));
}

/* Delete all entries from a dissector table. */
void dissector_delete_all(const char *name, dissector_handle_t handle)
{
        dissector_table_t sub_dissectors = find_dissector_table(name);
        g_assert (sub_dissectors);

        g_hash_table_foreach_remove (sub_dissectors->hash_table, dissector_delete_all_check, handle);
}

/* Change the entry for a dissector in a uint dissector table
   with a particular pattern to use a new dissector handle. */
void
dissector_change_uint(const char *name, const guint32 pattern, dissector_handle_t handle)
{
        dissector_table_t sub_dissectors = find_dissector_table( name);
        dtbl_entry_t *dtbl_entry;

        /* sanity check */
        g_assert( sub_dissectors);

        /*
         * See if the entry already exists. If so, reuse it.
         */
        dtbl_entry = find_uint_dtbl_entry(sub_dissectors, pattern);
        if (dtbl_entry != NULL) {
                dtbl_entry->current = handle;
                return;
        }

        /*
         * Don't create an entry if there is no dissector handle - I.E. the
         * user said not to decode something that wasn't being decoded
         * in the first place.
         */
        if (handle == NULL)
                return;

        dtbl_entry = (dtbl_entry_t *)g_malloc(sizeof (dtbl_entry_t));
        dtbl_entry->initial = NULL;
        dtbl_entry->current = handle;

        /* do the table insertion */
        g_hash_table_insert( sub_dissectors->hash_table,
                             GUINT_TO_POINTER( pattern), (gpointer)dtbl_entry);
}

/* Reset an entry in a uint dissector table to its initial value. */
void
dissector_reset_uint(const char *name, const guint32 pattern)
{
        dissector_table_t  sub_dissectors = find_dissector_table( name);
        dtbl_entry_t      *dtbl_entry;

        /* sanity check */
        g_assert( sub_dissectors);

        /*
         * Find the entry.
         */
        dtbl_entry = find_uint_dtbl_entry(sub_dissectors, pattern);

        if (dtbl_entry == NULL)
                return;

        /*
         * Found - is there an initial value?
         */
        if (dtbl_entry->initial != NULL) {
                dtbl_entry->current = dtbl_entry->initial;
        } else {
                g_hash_table_remove(sub_dissectors->hash_table,
                                    GUINT_TO_POINTER(pattern));
        }
}

/* Look for a given value in a given uint dissector table and, if found,
   call the dissector with the arguments supplied, and return TRUE,
   otherwise return FALSE. */

gboolean
dissector_try_uint_new(dissector_table_t sub_dissectors, const guint32 uint_val,
                       tvbuff_t *tvb, packet_info *pinfo, proto_tree *tree,
                       const gboolean add_proto_name, void *data)
{
        dtbl_entry_t            *dtbl_entry;
        struct dissector_handle *handle;
        guint32                  saved_match_uint;
        int ret;

        dtbl_entry = find_uint_dtbl_entry(sub_dissectors, uint_val);
        if (dtbl_entry != NULL) {
                /*
                 * Is there currently a dissector handle for this entry?
                 */
                handle = dtbl_entry->current;
                if (handle == NULL) {
                        /*
                         * No - pretend this dissector didn't exist,
                         * so that other dissectors might have a chance
                         * to dissect this packet.
                         */
                        return FALSE;
                }

                /*
                 * Save the current value of "pinfo->match_uint",
                 * set it to the uint_val that matched, call the
                 * dissector, and restore "pinfo->match_uint".
                 */
                saved_match_uint  = pinfo->match_uint;
                pinfo->match_uint = uint_val;
                ret = call_dissector_work(handle, tvb, pinfo, tree, add_proto_name, data);
                pinfo->match_uint = saved_match_uint;

                /*
                 * If a new-style dissector returned 0, it means that
                 * it didn't think this tvbuff represented a packet for
                 * its protocol, and didn't dissect anything.
                 *
                 * Old-style dissectors can't reject the packet.
                 *
                 * 0 is also returned if the protocol wasn't enabled.
                 *
                 * If the packet was rejected, we return FALSE, so that
                 * other dissectors might have a chance to dissect this
                 * packet, otherwise we return TRUE.
                 */
                return ret != 0;
        }
        return FALSE;
}

gboolean
dissector_try_uint(dissector_table_t sub_dissectors, const guint32 uint_val,
                   tvbuff_t *tvb, packet_info *pinfo, proto_tree *tree)
{

        return dissector_try_uint_new(sub_dissectors, uint_val, tvb, pinfo, tree, TRUE, NULL);
}

/* Look for a given value in a given uint dissector table and, if found,
   return the dissector handle for that value. */
dissector_handle_t
dissector_get_uint_handle(dissector_table_t const sub_dissectors, const guint32 uint_val)
{
        dtbl_entry_t *dtbl_entry;

        dtbl_entry = find_uint_dtbl_entry(sub_dissectors, uint_val);
        if (dtbl_entry != NULL)
                return dtbl_entry->current;
        else
                return NULL;
}

/* Find an entry in a string dissector table. */
static dtbl_entry_t *
find_string_dtbl_entry(dissector_table_t const sub_dissectors, const gchar *pattern)
{
        switch (sub_dissectors->type) {

        case FT_STRING:
        case FT_STRINGZ:
                /*
                 * You can do a string lookup in these tables.
                 */
                break;

        default:
                /*
                 * But you can't do a string lookup in any other types
                 * of tables.
                 */
                g_assert_not_reached();
        }

        /*
         * Find the entry.
         */
        return (dtbl_entry_t *)g_hash_table_lookup(sub_dissectors->hash_table, pattern);
}

/* Add an entry to a string dissector table. */
void
dissector_add_string(const char *name, const gchar *pattern,
                     dissector_handle_t handle)
{
        dissector_table_t  sub_dissectors = find_dissector_table( name);
        dtbl_entry_t      *dtbl_entry;

        /*
         * Make sure the dissector table exists.
         */
        if (sub_dissectors == NULL) {
                fprintf(stderr, "OOPS: dissector table \"%s\" doesn't exist\n",
                    name);
                fprintf(stderr, "Protocol being registered is \"%s\"\n",
                    proto_get_protocol_long_name(handle->protocol));
                if (getenv("WIRESHARK_ABORT_ON_DISSECTOR_BUG") != NULL)
                        abort();
                return;
        }

        /* sanity checks */
        g_assert(handle!=NULL);
        switch (sub_dissectors->type) {

        case FT_STRING:
        case FT_STRINGZ:
                /*
                 * You can do a string lookup in these tables.
                 */
                break;

        default:
                /*
                 * But you can't do a string lookup in any other types
                 * of tables.
                 */
                g_assert_not_reached();
        }

        dtbl_entry = (dtbl_entry_t *)g_malloc(sizeof (dtbl_entry_t));
        dtbl_entry->current = handle;
        dtbl_entry->initial = dtbl_entry->current;

        /* do the table insertion */
        g_hash_table_insert( sub_dissectors->hash_table, (gpointer)g_strdup(pattern),
                             (gpointer)dtbl_entry);

        /*
         * Now add it to the list of handles that could be used with this
         * table, because it *is* being used with this table.
         */
        dissector_add_handle(name, handle);
}

/* Delete the entry for a dissector in a string dissector table
   with a particular pattern. */

/* NOTE: this doesn't use the dissector call variable. It is included to */
/*      be consistant with the dissector_add_string and more importantly to */
/*      be used if the technique of adding a temporary dissector is */
/*      implemented.  */
/*      If temporary dissectors are deleted, then the original dissector must */
/*      be available. */
void
dissector_delete_string(const char *name, const gchar *pattern,
        dissector_handle_t handle _U_)
{
        dissector_table_t  sub_dissectors = find_dissector_table( name);
        dtbl_entry_t      *dtbl_entry;

        /* sanity check */
        g_assert( sub_dissectors);

        /*
         * Find the entry.
         */
        dtbl_entry = find_string_dtbl_entry(sub_dissectors, pattern);

        if (dtbl_entry != NULL) {
                /*
                 * Found - remove it.
                 */
                g_hash_table_remove(sub_dissectors->hash_table, pattern);
        }
}

/* Change the entry for a dissector in a string dissector table
   with a particular pattern to use a new dissector handle. */
void
dissector_change_string(const char *name, const gchar *pattern,
                        dissector_handle_t handle)
{
        dissector_table_t  sub_dissectors = find_dissector_table( name);
        dtbl_entry_t      *dtbl_entry;

        /* sanity check */
        g_assert( sub_dissectors);

        /*
         * See if the entry already exists. If so, reuse it.
         */
        dtbl_entry = find_string_dtbl_entry(sub_dissectors, pattern);
        if (dtbl_entry != NULL) {
                dtbl_entry->current = handle;
                return;
        }

        /*
         * Don't create an entry if there is no dissector handle - I.E. the
         * user said not to decode something that wasn't being decoded
         * in the first place.
         */
        if (handle == NULL)
                return;

        dtbl_entry = (dtbl_entry_t *)g_malloc(sizeof (dtbl_entry_t));
        dtbl_entry->initial = NULL;
        dtbl_entry->current = handle;

        /* do the table insertion */
        g_hash_table_insert( sub_dissectors->hash_table, (gpointer)g_strdup(pattern),
                             (gpointer)dtbl_entry);
}

/* Reset an entry in a string sub-dissector table to its initial value. */
void
dissector_reset_string(const char *name, const gchar *pattern)
{
        dissector_table_t  sub_dissectors = find_dissector_table( name);
        dtbl_entry_t      *dtbl_entry;

        /* sanity check */
        g_assert( sub_dissectors);

        /*
         * Find the entry.
         */
        dtbl_entry = find_string_dtbl_entry(sub_dissectors, pattern);

        if (dtbl_entry == NULL)
                return;

        /*
         * Found - is there an initial value?
         */
        if (dtbl_entry->initial != NULL) {
                dtbl_entry->current = dtbl_entry->initial;
        } else {
                g_hash_table_remove(sub_dissectors->hash_table, pattern);
        }
}

/* Look for a given string in a given dissector table and, if found, call
   the dissector with the arguments supplied, and return TRUE, otherwise
   return FALSE. */
gboolean
dissector_try_string(dissector_table_t sub_dissectors, const gchar *string,
                     tvbuff_t *tvb, packet_info *pinfo, proto_tree *tree, void *data)
{
        dtbl_entry_t            *dtbl_entry;
        struct dissector_handle *handle;
        int                      ret;
        const gchar             *saved_match_string;

        /* XXX ASSERT instead ? */
        if (!string) return FALSE;
        dtbl_entry = find_string_dtbl_entry(sub_dissectors, string);
        if (dtbl_entry != NULL) {
                /*
                 * Is there currently a dissector handle for this entry?
                 */
                handle = dtbl_entry->current;
                if (handle == NULL) {
                        /*
                         * No - pretend this dissector didn't exist,
                         * so that other dissectors might have a chance
                         * to dissect this packet.
                         */
                        return FALSE;
                }

                /*
                 * Save the current value of "pinfo->match_string",
                 * set it to the string that matched, call the
                 * dissector, and restore "pinfo->match_string".
                 */
                saved_match_string = pinfo->match_string;
                pinfo->match_string = string;
                ret = call_dissector_work(handle, tvb, pinfo, tree, TRUE, data);
                pinfo->match_string = saved_match_string;

                /*
                 * If a new-style dissector returned 0, it means that
                 * it didn't think this tvbuff represented a packet for
                 * its protocol, and didn't dissect anything.
                 *
                 * Old-style dissectors can't reject the packet.
                 *
                 * 0 is also returned if the protocol wasn't enabled.
                 *
                 * If the packet was rejected, we return FALSE, so that
                 * other dissectors might have a chance to dissect this
                 * packet, otherwise we return TRUE.
                 */
                return ret != 0;
        }
        return FALSE;
}

/* Look for a given value in a given string dissector table and, if found,
   return the dissector handle for that value. */
dissector_handle_t
dissector_get_string_handle(dissector_table_t sub_dissectors,
                            const gchar *string)
{
        dtbl_entry_t *dtbl_entry;

        dtbl_entry = find_string_dtbl_entry(sub_dissectors, string);
        if (dtbl_entry != NULL)
                return dtbl_entry->current;
        else
                return NULL;
}

dissector_handle_t
dtbl_entry_get_handle (dtbl_entry_t *dtbl_entry)
{
        return dtbl_entry->current;
}

static gint
dissector_compare_filter_name(gconstpointer dissector_a, gconstpointer dissector_b)
{
        const dissector_handle_t a = (const dissector_handle_t)dissector_a;
        const dissector_handle_t b = (const dissector_handle_t)dissector_b;
        const char *a_name, *b_name;
        gint ret;

        if (a->protocol == NULL)
                a_name = "";
        else
                a_name = proto_get_protocol_filter_name(proto_get_id(a->protocol));

        if (b->protocol == NULL)
                b_name = "";
        else
                b_name = proto_get_protocol_filter_name(proto_get_id(b->protocol));

        ret = strcmp(a_name, b_name);
        return ret;
}

/* Add a handle to the list of handles that *could* be used with this
   table.  That list is used by code in the UI. */
void
dissector_add_handle(const char *name, dissector_handle_t handle)
{
        dissector_table_t  sub_dissectors = find_dissector_table( name);
        GSList            *entry;

        /*
         * Make sure the dissector table exists.
         */
        if (sub_dissectors == NULL) {
                fprintf(stderr, "OOPS: dissector table \"%s\" doesn't exist\n",
                    name);
                fprintf(stderr, "Protocol being registered is \"%s\"\n",
                    proto_get_protocol_long_name(handle->protocol));
                if (getenv("WIRESHARK_ABORT_ON_DISSECTOR_BUG") != NULL)
                        abort();
                return;
        }

        /* Is it already in this list? */
        entry = g_slist_find(sub_dissectors->dissector_handles, (gpointer)handle);
        if (entry != NULL) {
                /*
                 * Yes - don't insert it again.
                 */
                return;
        }

        /* Add it to the list. */
        sub_dissectors->dissector_handles =
                g_slist_insert_sorted(sub_dissectors->dissector_handles, (gpointer)handle, (GCompareFunc)dissector_compare_filter_name);
}

dissector_handle_t
dtbl_entry_get_initial_handle (dtbl_entry_t *dtbl_entry)
{
        return dtbl_entry->initial;
}

/**************************************************/
/*                                                */
/*       Routines to walk dissector tables        */
/*                                                */
/**************************************************/

typedef struct dissector_foreach_info {
        gpointer      caller_data;
        DATFunc       caller_func;
        GHFunc        next_func;
        const gchar  *table_name;
        ftenum_t      selector_type;
} dissector_foreach_info_t;

/*
 * Called for each entry in a dissector table.
 */
static void
dissector_table_foreach_func (gpointer key, gpointer value, gpointer user_data)
{
        dissector_foreach_info_t *info;
        dtbl_entry_t             *dtbl_entry;

        g_assert(value);
        g_assert(user_data);

        dtbl_entry = (dtbl_entry_t *)value;
        if (dtbl_entry->current == NULL ||
            dtbl_entry->current->protocol == NULL) {
                /*
                 * Either there is no dissector for this entry, or
                 * the dissector doesn't have a protocol associated
                 * with it.
                 *
                 * XXX - should the latter check be done?
                 */
                return;
        }

        info = (dissector_foreach_info_t *)user_data;
        info->caller_func(info->table_name, info->selector_type, key, value,
                          info->caller_data);
}

/*
 * Called for each entry in the table of all dissector tables.
 */
static void
dissector_all_tables_foreach_func (gpointer key, gpointer value, gpointer user_data)
{
        dissector_table_t         sub_dissectors;
        dissector_foreach_info_t *info;

        g_assert(value);
        g_assert(user_data);

        sub_dissectors = (dissector_table_t)value;
        info = (dissector_foreach_info_t *)user_data;
        info->table_name = (gchar*) key;
        info->selector_type = get_dissector_table_selector_type(info->table_name);
        g_hash_table_foreach(sub_dissectors->hash_table, info->next_func, info);
}

/*
 * Walk all dissector tables calling a user supplied function on each
 * entry.
 */
static void
dissector_all_tables_foreach (DATFunc func,
                              gpointer user_data)
{
        dissector_foreach_info_t info;

        info.caller_data = user_data;
        info.caller_func = func;
        info.next_func   = dissector_table_foreach_func;
        g_hash_table_foreach(dissector_tables, dissector_all_tables_foreach_func, &info);
}

/*
 * Walk one dissector table's hash table calling a user supplied function
 * on each entry.
 */
void
dissector_table_foreach (const char *name,
                         DATFunc     func,
                         gpointer    user_data)
{
        dissector_foreach_info_t info;
        dissector_table_t        sub_dissectors = find_dissector_table( name);

        info.table_name    = name;
        info.selector_type = sub_dissectors->type;
        info.caller_func   = func;
        info.caller_data   = user_data;
        g_hash_table_foreach(sub_dissectors->hash_table, dissector_table_foreach_func, &info);
}

/*
 * Walk one dissector table's list of handles calling a user supplied
 * function on each entry.
 */
void
dissector_table_foreach_handle(const char     *name,
                               DATFunc_handle  func,
                               gpointer        user_data)
{
        dissector_table_t sub_dissectors = find_dissector_table( name);
        GSList *tmp;

        for (tmp = sub_dissectors->dissector_handles; tmp != NULL;
             tmp = g_slist_next(tmp))
                func(name, tmp->data, user_data);
}

/*
 * Called for each entry in a dissector table.
 */
static void
dissector_table_foreach_changed_func (gpointer key, gpointer value, gpointer user_data)
{
        dtbl_entry_t             *dtbl_entry;
        dissector_foreach_info_t *info;

        g_assert(value);
        g_assert(user_data);

        dtbl_entry = (dtbl_entry_t *)value;
        if (dtbl_entry->initial == dtbl_entry->current) {
                /*
                 * Entry hasn't changed - don't call the function.
                 */
                return;
        }

        info = (dissector_foreach_info_t *)user_data;
        info->caller_func(info->table_name, info->selector_type, key, value,
                          info->caller_data);
}

/*
 * Walk all dissector tables calling a user supplied function only on
 * any entry that has been changed from its original state.
 */
void
dissector_all_tables_foreach_changed (DATFunc  func,
                                      gpointer user_data)
{
        dissector_foreach_info_t info;

        info.caller_data = user_data;
        info.caller_func = func;
        info.next_func   = dissector_table_foreach_changed_func;
        g_hash_table_foreach(dissector_tables, dissector_all_tables_foreach_func, &info);
}

/*
 * Walk one dissector table calling a user supplied function only on
 * any entry that has been changed from its original state.
 */
void
dissector_table_foreach_changed (const char *name,
                                 DATFunc     func,
                                 gpointer    user_data)
{
        dissector_foreach_info_t info;
        dissector_table_t sub_dissectors = find_dissector_table( name);

        info.table_name    = name;
        info.selector_type = sub_dissectors->type;
        info.caller_func   = func;
        info.caller_data   = user_data;
        g_hash_table_foreach(sub_dissectors->hash_table,
                             dissector_table_foreach_changed_func, &info);
}

typedef struct dissector_foreach_table_info {
        gpointer      caller_data;
        DATFunc_table caller_func;
} dissector_foreach_table_info_t;

/*
 * Called for each entry in the table of all dissector tables.
 */
static void
dissector_all_tables_foreach_table_func (gpointer key, const gpointer value, const gpointer user_data)
{
        dissector_table_t               table;
        dissector_foreach_table_info_t *info;

        table = (dissector_table_t)value;
        info  = (dissector_foreach_table_info_t *)user_data;
        (*info->caller_func)((gchar*)key, table->ui_name, info->caller_data);
}

/*
 * Called for each key in the table of all dissector tables.
 */
static void
dissector_all_tables_foreach_list_func (gpointer key, gpointer user_data)
{
        dissector_table_t               table;
        dissector_foreach_table_info_t *info;

        table = (dissector_table_t)g_hash_table_lookup( dissector_tables, key );
        info  = (dissector_foreach_table_info_t *)user_data;
        (*info->caller_func)((gchar*)key, table->ui_name, info->caller_data);
}

/*
 * Walk all dissector tables calling a user supplied function on each
 * table.
 */
void
dissector_all_tables_foreach_table (DATFunc_table func,
                                        gpointer      user_data,
                                        GCompareFunc compare_key_func)
{
        dissector_foreach_table_info_t info;
        GList *list;

        info.caller_data = user_data;
        info.caller_func = func;
        if (compare_key_func != NULL)
        {
                list = g_hash_table_get_keys(dissector_tables);
                list = g_list_sort(list, compare_key_func);
                g_list_foreach(list, dissector_all_tables_foreach_list_func, &info);
                g_list_free(list);
        }
        else
        {
                g_hash_table_foreach(dissector_tables, dissector_all_tables_foreach_table_func, &info);
        }
}

dissector_table_t
register_dissector_table(const char *name, const char *ui_name, const ftenum_t type,
                         const int base)
{
        dissector_table_t       sub_dissectors;

        /* Make sure the registration is unique */
        if(g_hash_table_lookup( dissector_tables, name )) {
                g_error("The filter name %s (%s) is already registered - do you use a buggy plugin?", name, ui_name);
        }

        /* Create and register the dissector table for this name; returns */
        /* a pointer to the dissector table. */
        sub_dissectors = g_slice_new(struct dissector_table);
        switch (type) {

        case FT_UINT8:
        case FT_UINT16:
        case FT_UINT24:
        case FT_UINT32:
                /*
                 * XXX - there's no "g_uint_hash()" or "g_uint_equal()",
                 * so we use "g_direct_hash()" and "g_direct_equal()".
                 */
                sub_dissectors->hash_table = g_hash_table_new_full( g_direct_hash,
                                                               g_direct_equal,
                                                               NULL,
                                                               &g_free );
                break;

        case FT_STRING:
        case FT_STRINGZ:
                sub_dissectors->hash_table = g_hash_table_new_full( g_str_hash,
                                                               g_str_equal,
                                                               &g_free,
                                                               &g_free );
                break;

        default:
                g_assert_not_reached();
        }
        sub_dissectors->dissector_handles = NULL;
        sub_dissectors->ui_name = ui_name;
        sub_dissectors->type    = type;
        sub_dissectors->base    = base;
        g_hash_table_insert( dissector_tables, (gpointer)name, (gpointer) sub_dissectors );
        return sub_dissectors;
}

const char *
get_dissector_table_ui_name(const char *name)
{
        dissector_table_t sub_dissectors = find_dissector_table( name);

        return sub_dissectors->ui_name;
}

ftenum_t
get_dissector_table_selector_type(const char *name)
{
        dissector_table_t sub_dissectors = find_dissector_table( name);

        return sub_dissectors->type;
}

int
get_dissector_table_base(const char *name)
{
        dissector_table_t sub_dissectors = find_dissector_table( name);

        return sub_dissectors->base;
}

/* Finds a heuristic dissector table by table name. */
static heur_dissector_list_t *
find_heur_dissector_list(const char *name)
{
        return (heur_dissector_list_t *)g_hash_table_lookup(heur_dissector_lists, name);
}

void
heur_dissector_add(const char *name, heur_dissector_t dissector, const int proto)
{
        heur_dissector_list_t *sub_dissectors = find_heur_dissector_list(name);
        const char            *proto_name;
        heur_dtbl_entry_t     *hdtbl_entry;

        /*
         * Make sure the dissector table exists.
         */
        if (sub_dissectors == NULL) {
                fprintf(stderr, "OOPS: dissector table \"%s\" doesn't exist\n",
                    name);
                proto_name = proto_get_protocol_name(proto);
                if (proto_name != NULL) {
                        fprintf(stderr, "Protocol being registered is \"%s\"\n",
                            proto_name);
                }
                if (getenv("WIRESHARK_ABORT_ON_DISSECTOR_BUG") != NULL)
                        abort();
                return;
        }

        /* XXX: Should verify that sub-dissector is not already in the list ? */

        hdtbl_entry = g_slice_new(heur_dtbl_entry_t);
        hdtbl_entry->dissector = dissector;
        hdtbl_entry->protocol  = find_protocol_by_id(proto);
        hdtbl_entry->enabled   = TRUE;

        /* do the table insertion */
        *sub_dissectors = g_slist_prepend(*sub_dissectors, (gpointer)hdtbl_entry);
}



static int
find_matching_heur_dissector( gconstpointer a, gconstpointer b) {
        const heur_dtbl_entry_t *hdtbl_entry_a = (const heur_dtbl_entry_t *) a;
        const heur_dtbl_entry_t *hdtbl_entry_b = (const heur_dtbl_entry_t *) b;

        return (hdtbl_entry_a->dissector == hdtbl_entry_b->dissector) &&
                (hdtbl_entry_a->protocol == hdtbl_entry_b->protocol) ? 0 : 1;
}

void
heur_dissector_delete(const char *name, heur_dissector_t dissector, const int proto) {
        heur_dissector_list_t *sub_dissectors = find_heur_dissector_list(name);
        heur_dtbl_entry_t      hdtbl_entry;
        GSList                *found_entry;

        /* sanity check */
        g_assert(sub_dissectors != NULL);

        hdtbl_entry.dissector = dissector;

        hdtbl_entry.protocol  = find_protocol_by_id(proto);

        found_entry = g_slist_find_custom(*sub_dissectors, (gpointer) &hdtbl_entry, find_matching_heur_dissector);

        if (found_entry) {
                g_slice_free(heur_dtbl_entry_t, found_entry->data);
                *sub_dissectors = g_slist_delete_link(*sub_dissectors, found_entry);
        }
}

void
heur_dissector_set_enabled(const char *name, heur_dissector_t dissector, const int proto, const gboolean enabled) {
        heur_dissector_list_t *sub_dissectors = find_heur_dissector_list(name);
        GSList                *found_entry;
        heur_dtbl_entry_t      hdtbl_entry;

        /* sanity check */
        g_assert(sub_dissectors != NULL);

        hdtbl_entry.dissector = dissector;

        hdtbl_entry.protocol  = find_protocol_by_id(proto);

        found_entry = g_slist_find_custom(*sub_dissectors, (gpointer) &hdtbl_entry, find_matching_heur_dissector);

        if (found_entry) {
                heur_dtbl_entry_t *hdtbl_entry_p;
                hdtbl_entry_p = (heur_dtbl_entry_t *)found_entry->data;
                hdtbl_entry_p->enabled = enabled;
        }
}

gboolean
dissector_try_heuristic(heur_dissector_list_t sub_dissectors, tvbuff_t *tvb,
                        packet_info *pinfo, proto_tree *tree, void *data)
{
        gboolean           status;
        const char        *saved_proto;
        GSList            *entry;
        heur_dtbl_entry_t *hdtbl_entry;
        guint16            saved_can_desegment;
        guint              saved_layers_len = 0;

        /* can_desegment is set to 2 by anyone which offers this api/service.
           then everytime a subdissector is called it is decremented by one.
           thus only the subdissector immediately ontop of whoever offers this
           service can use it.
           We save the current value of "can_desegment" for the
           benefit of TCP proxying dissectors such as SOCKS, so they
           can restore it and allow the dissectors they call to use
           the desegmentation service.
        */
        saved_can_desegment        = pinfo->can_desegment;
        pinfo->saved_can_desegment = saved_can_desegment;
        pinfo->can_desegment       = saved_can_desegment-(saved_can_desegment>0);

        status      = FALSE;
        saved_proto = pinfo->current_proto;

        saved_layers_len = wmem_list_count(pinfo->layers);

        for (entry = sub_dissectors; entry != NULL; entry = g_slist_next(entry)) {
                /* XXX - why set this now and above? */
                pinfo->can_desegment = saved_can_desegment-(saved_can_desegment>0);
                hdtbl_entry = (heur_dtbl_entry_t *)entry->data;

                if (hdtbl_entry->protocol != NULL &&
                    (!proto_is_protocol_enabled(hdtbl_entry->protocol)||(hdtbl_entry->enabled==FALSE))) {
                        /*
                         * No - don't try this dissector.
                         */
                        continue;
                }

                if (hdtbl_entry->protocol != NULL) {
                        pinfo->current_proto =
                                proto_get_protocol_short_name(hdtbl_entry->protocol);

                        /*
                         * Add the protocol name to the layers; we'll remove it
                         * if the dissector fails.
                         */
                        wmem_list_append(pinfo->layers, GINT_TO_POINTER(proto_get_id(hdtbl_entry->protocol)));
                }
                EP_CHECK_CANARY(("before calling heuristic dissector for protocol: %s",
                                 proto_get_protocol_filter_name(proto_get_id(hdtbl_entry->protocol))));
                if ((*hdtbl_entry->dissector)(tvb, pinfo, tree, data)) {
                        EP_CHECK_CANARY(("after heuristic dissector for protocol: %s has accepted and dissected packet",
                                         proto_get_protocol_filter_name(proto_get_id(hdtbl_entry->protocol))));
                        status = TRUE;
                        break;
                } else {
                        EP_CHECK_CANARY(("after heuristic dissector for protocol: %s has returned false",
                                         proto_get_protocol_filter_name(proto_get_id(hdtbl_entry->protocol))));

                        /*
                         * That dissector didn't accept the packet, so
                         * remove its protocol's name from the list
                         * of protocols.
                         */
                        while (wmem_list_count(pinfo->layers) > saved_layers_len) {
                                wmem_list_remove_frame(pinfo->layers, wmem_list_tail(pinfo->layers));
                        }
                }
        }
        pinfo->current_proto = saved_proto;
        pinfo->can_desegment=saved_can_desegment;
        return status;
}

/*
 * Called for each entry in the table of all heuristic dissector tables.
 */
typedef struct heur_dissector_foreach_table_info {
        gpointer           caller_data;
        DATFunc_heur_table caller_func;
} heur_dissector_foreach_table_info_t;


static void
dissector_dump_heur_decodes_display(const gchar *table_name, const gpointer value, const gpointer user_data _U_)
{
        heur_dissector_list_t  sub_dissectors = *(heur_dissector_list_t *)value;
        GSList                *entry;
        heur_dtbl_entry_t     *hdtbl_entry;

        for (entry = sub_dissectors; entry != NULL; entry = g_slist_next(entry)) {
                hdtbl_entry = (heur_dtbl_entry_t *)entry->data;
                if (hdtbl_entry->protocol != NULL) {
                        printf("%s\t%s\t%c\n",
                               table_name,
                               proto_get_protocol_filter_name(proto_get_id(hdtbl_entry->protocol)),
                               (proto_is_protocol_enabled(hdtbl_entry->protocol) && hdtbl_entry->enabled) ? 'T' : 'F');
                }
        }
}


static void
dissector_all_heur_tables_foreach_table_func (gpointer key, const gpointer value, const gpointer user_data)
{
        heur_dissector_foreach_table_info_t *info;

        info = (heur_dissector_foreach_table_info_t *)user_data;
        (*info->caller_func)((gchar*)key, value, info->caller_data);
}

/*
 * Walk all heuristic dissector tables calling a user supplied function on each
 * table.
 */
void
dissector_all_heur_tables_foreach_table (DATFunc_heur_table func,
                                         gpointer user_data)
{
        heur_dissector_foreach_table_info_t info;

        info.caller_data = user_data;
        info.caller_func = func;
        g_hash_table_foreach(heur_dissector_lists, dissector_all_heur_tables_foreach_table_func, &info);
}

/*
 * For each heuristic dissector table, dump list of dissectors (filter_names) for that table
 */
void
dissector_dump_heur_decodes(void)
{
        dissector_all_heur_tables_foreach_table(dissector_dump_heur_decodes_display, NULL);
}


void
register_heur_dissector_list(const char *name, heur_dissector_list_t *sub_dissectors)
{
        /* Make sure the registration is unique */
        g_assert(g_hash_table_lookup(heur_dissector_lists, name) == NULL);

        *sub_dissectors = NULL; /* initially empty */
        g_hash_table_insert(heur_dissector_lists, (gpointer)name,
                            (gpointer) sub_dissectors);
}

/*
 * Register dissectors by name; used if one dissector always calls a
 * particular dissector, or if it bases the decision of which dissector
 * to call on something other than a numerical value or on "try a bunch
 * of dissectors until one likes the packet".
 */

/* Get the long name of the protocol for a dissector handle, if it has
   a protocol. */
const char *
dissector_handle_get_long_name(const dissector_handle_t handle)
{
        if (handle == NULL || handle->protocol == NULL) {
                return NULL;
        }
        return proto_get_protocol_long_name(handle->protocol);
}

/* Get the short name of the protocol for a dissector handle, if it has
   a protocol. */
const char *
dissector_handle_get_short_name(const dissector_handle_t handle)
{
        if (handle->protocol == NULL) {
                /*
                 * No protocol (see, for example, the handle for
                 * dissecting the set of protocols where the first
                 * octet of the payload is an OSI network layer protocol
                 * ID).
                 */
                return NULL;
        }
        return proto_get_protocol_short_name(handle->protocol);
}

/* Get the index of the protocol for a dissector handle, if it has
   a protocol. */
int
dissector_handle_get_protocol_index(const dissector_handle_t handle)
{
        if (handle->protocol == NULL) {
                /*
                 * No protocol (see, for example, the handle for
                 * dissecting the set of protocols where the first
                 * octet of the payload is an OSI network layer protocol
                 * ID).
                 */
                return -1;
        }
        return proto_get_id(handle->protocol);
}

/* Find a registered dissector by name. */
dissector_handle_t
find_dissector(const char *name)
{
        return (dissector_handle_t)g_hash_table_lookup(registered_dissectors, name);
}

/* Get a dissector name from handle. */
const char *
dissector_handle_get_dissector_name(const dissector_handle_t handle)
{
        if (handle == NULL) {
                return NULL;
        }
        return handle->name;
}

/* Create an anonymous handle for a dissector. */
dissector_handle_t
create_dissector_handle(dissector_t dissector, const int proto)
{
        struct dissector_handle *handle;

        handle                = wmem_new(wmem_epan_scope(), struct dissector_handle);
        handle->name          = NULL;
        handle->is_new        = FALSE;
        handle->dissector.old = dissector;
        handle->protocol      = find_protocol_by_id(proto);

        return handle;
}

dissector_handle_t
new_create_dissector_handle(new_dissector_t dissector, const int proto)
{
        struct dissector_handle *handle;

        handle                  = wmem_new(wmem_epan_scope(), struct dissector_handle);
        handle->name            = NULL;
        handle->is_new          = TRUE;
        handle->dissector.new_d = dissector;
        handle->protocol        = find_protocol_by_id(proto);

        return handle;
}

/* Register a dissector by name. */
dissector_handle_t
register_dissector(const char *name, dissector_t dissector, const int proto)
{
        struct dissector_handle *handle;

        /* Make sure the registration is unique */
        g_assert(g_hash_table_lookup(registered_dissectors, name) == NULL);

        handle                = wmem_new(wmem_epan_scope(), struct dissector_handle);
        handle->name          = name;
        handle->is_new        = FALSE;
        handle->dissector.old = dissector;
        handle->protocol      = find_protocol_by_id(proto);

        g_hash_table_insert(registered_dissectors, (gpointer)name,
                            (gpointer) handle);

        return handle;
}

dissector_handle_t
new_register_dissector(const char *name, new_dissector_t dissector, const int proto)
{
        struct dissector_handle *handle;

        /* Make sure the registration is unique */
        g_assert(g_hash_table_lookup(registered_dissectors, name) == NULL);

        handle                = wmem_new(wmem_epan_scope(), struct dissector_handle);
        handle->name          = name;
        handle->is_new        = TRUE;
        handle->dissector.new_d = dissector;
        handle->protocol      = find_protocol_by_id(proto);

        g_hash_table_insert(registered_dissectors, (gpointer)name,
                            (gpointer) handle);

        return handle;
}

/* Call a dissector through a handle but if the dissector rejected it
 * return 0.
 */
int
call_dissector_only(dissector_handle_t handle, tvbuff_t *tvb,
                    packet_info *pinfo, proto_tree *tree, void *data)
{
        int ret;

        g_assert(handle != NULL);
        ret = call_dissector_work(handle, tvb, pinfo, tree, TRUE, data);
        return ret;
}

/* Call a dissector through a handle and if this fails call the "data"
 * dissector.
 */
int
call_dissector_with_data(dissector_handle_t handle, tvbuff_t *tvb,
                         packet_info *pinfo, proto_tree *tree, void *data)
{
        int ret;

        ret = call_dissector_only(handle, tvb, pinfo, tree, data);
        if (ret == 0) {
                /*
                 * The protocol was disabled, or the dissector rejected
                 * it.  Just dissect this packet as data.
                 */
                g_assert(data_handle->protocol != NULL);
                call_dissector_work(data_handle, tvb, pinfo, tree, TRUE, NULL);
                return tvb_length(tvb);
        }
        return ret;
}

int
call_dissector(dissector_handle_t handle, tvbuff_t *tvb,
               packet_info *pinfo, proto_tree *tree)
{
        return call_dissector_with_data(handle, tvb, pinfo, tree, NULL);
}

/*
 * Dumps the "layer type"/"decode as" associations to stdout, similar
 * to the proto_registrar_dump_*() routines.
 *
 * There is one record per line. The fields are tab-delimited.
 *
 * Field 1 = layer type, e.g. "tcp.port"
 * Field 2 = selector in decimal
 * Field 3 = "decode as" name, e.g. "http"
 */


static void
dissector_dump_decodes_display(const gchar *table_name,
                               ftenum_t selector_type _U_, const gpointer key, const gpointer value,
                               gpointer user_data _U_)
{
        guint32             selector       = (guint32)(unsigned long) key;
        dissector_table_t   sub_dissectors = find_dissector_table(table_name);
        dtbl_entry_t       *dtbl_entry;
        dissector_handle_t  handle;
        gint                proto_id;
        const gchar        *decode_as;

        g_assert(sub_dissectors);
        switch (sub_dissectors->type) {

                case FT_UINT8:
                case FT_UINT16:
                case FT_UINT24:
                case FT_UINT32:
                        dtbl_entry = (dtbl_entry_t *)value;
                        g_assert(dtbl_entry);

                        handle   = dtbl_entry->current;
                        g_assert(handle);

                        proto_id = dissector_handle_get_protocol_index(handle);

                        if (proto_id != -1) {
                                decode_as = proto_get_protocol_filter_name(proto_id);
                                g_assert(decode_as != NULL);
                                printf("%s\t%u\t%s\n", table_name, selector, decode_as);
                        }
                        break;

        default:
                break;
        }
}

void
dissector_dump_decodes(void)
{
        dissector_all_tables_foreach(dissector_dump_decodes_display, NULL);
}

static GPtrArray* post_dissectors = NULL;
static guint num_of_postdissectors = 0;

void
register_postdissector(dissector_handle_t handle)
{
        if (!post_dissectors)
                post_dissectors = g_ptr_array_new();

        g_ptr_array_add(post_dissectors, handle);
        num_of_postdissectors++;
}

gboolean
have_postdissector(void)
{
        guint i;
        dissector_handle_t handle;

        for(i = 0; i < num_of_postdissectors; i++) {
                handle = (dissector_handle_t) g_ptr_array_index(post_dissectors,i);

                if (handle->protocol != NULL
                    && proto_is_protocol_enabled(handle->protocol)) {
                        /* We have at least one enabled postdissector */
                        return TRUE;
                }
        }
        return FALSE;
}

void
call_all_postdissectors(tvbuff_t *tvb, packet_info *pinfo, proto_tree *tree)
{
        guint i;

        for(i = 0; i < num_of_postdissectors; i++) {
                call_dissector_only((dissector_handle_t) g_ptr_array_index(post_dissectors,i),
                                    tvb,pinfo,tree, NULL);
        }
}

/*
 * Editor modelines  -  http://www.wireshark.org/tools/modelines.html
 *
 * Local variables:
 * c-basic-offset: 8
 * tab-width: 8
 * indent-tabs-mode: t
 * End:
 *
 * vi: set shiftwidth=8 tabstop=8 noexpandtab:
 * :indentSize=8:tabSize=8:noTabs=false:
 */