nexmon – Rev 1

Subversion Repositories:
Rev:
/* tap-stats_tree.c
 * tshark's tap implememntation of stats_tree
 * 2005, Luis E. G. Ontanon
 *
 * 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 <string.h>
#include <stdio.h>
#include <glib.h>

#include <wsutil/report_err.h>

#include <epan/stats_tree_priv.h>
#include <epan/stat_tap_ui.h>

void register_tap_listener_stats_tree_stat(void);

/* actually unused */
struct _st_node_pres {
        void *dummy;
};

struct _tree_pres {
        void **dummy;
};

struct _tree_cfg_pres {
        gchar *init_string;
};

static void
draw_stats_tree(void *psp)
{
        stats_tree *st = (stats_tree *)psp;
        GString *s;

        s= stats_tree_format_as_str(st, ST_FORMAT_PLAIN, stats_tree_get_default_sort_col(st),
                                    stats_tree_is_default_sort_DESC(st));

        printf("%s", s->str);
        g_string_free(s, TRUE);
}

static void
init_stats_tree(const char *opt_arg, void *userdata _U_)
{
        char *abbr = stats_tree_get_abbr(opt_arg);
        GString *error_string;
        stats_tree_cfg *cfg = NULL;
        stats_tree *st = NULL;

        if (abbr) {
                cfg = stats_tree_get_cfg_by_abbr(abbr);

                if (cfg != NULL) {
                        if (strncmp (opt_arg, cfg->pr->init_string, strlen(cfg->pr->init_string)) == 0) {
                                st = stats_tree_new(cfg, NULL, opt_arg+strlen(cfg->pr->init_string));
                        } else {
                                report_failure("Wrong stats_tree (%s) found when looking at ->init_string", abbr);
                                return;
                        }
                } else {
                        report_failure("no such stats_tree (%s) found in stats_tree registry", abbr);
                        return;
                }

                g_free(abbr);

        } else {
                report_failure("could not obtain stats_tree abbr (%s) from arg '%s'", abbr, opt_arg);
                return;
        }

        error_string = register_tap_listener(st->cfg->tapname,
                                             st,
                                             st->filter,
                                             st->cfg->flags,
                                             stats_tree_reset,
                                             stats_tree_packet,
                                             draw_stats_tree);

        if (error_string) {
                report_failure("stats_tree for: %s failed to attach to the tap: %s", cfg->name, error_string->str);
                return;
        }

        if (cfg->init) cfg->init(st);

}

static void
register_stats_tree_tap (gpointer k _U_, gpointer v, gpointer p _U_)
{
        stats_tree_cfg *cfg = (stats_tree_cfg *)v;
        stat_tap_ui ui_info;

        cfg->pr = (tree_cfg_pres *)g_malloc(sizeof(tree_cfg_pres));
        cfg->pr->init_string = g_strdup_printf("%s,tree", cfg->abbr);

        ui_info.group = REGISTER_STAT_GROUP_GENERIC;
        ui_info.title = NULL;
        ui_info.cli_string = cfg->pr->init_string;
        ui_info.tap_init_cb = init_stats_tree;
        ui_info.nparams = 0;
        ui_info.params = NULL;
        register_stat_tap_ui(&ui_info, NULL);

}

static void
free_tree_presentation(stats_tree *st)
{
        g_free(st->pr);
}


void
register_tap_listener_stats_tree_stat(void)
{
        stats_tree_presentation(register_stats_tree_tap, NULL,
                free_tree_presentation, 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:
 */