/* stat_tap_ui.c

 * Routines to register UI information for stats

 *

 * 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"

#include <stdio.h>

#include <string.h>

#include <glib.h>

#include <epan/stat_tap_ui.h>

#include <wsutil/ws_assert.h>

/* structure to keep track of what stats have registered command-line

   arguments.

 */

typedef struct _stat_cmd_arg {

    stat_tap_ui *ui;

    const char *cmd;

    bool (*init)(const char *arg, void* userdata);

    void* userdata;

} stat_cmd_arg;

static wmem_list_t *stat_cmd_arg_list;

static wmem_tree_t* registered_stat_tables;

/* structure to keep track of what stats have been specified on the

   command line.

 */

typedef struct {

    stat_cmd_arg *sca;

    char *arg;

} stat_requested;

static GSList *stats_requested;

/* **********************************************************************

 * Function called from stat to register the stat's command-line argument

 * and initialization routine

 * ********************************************************************** */

static int

search_duplicate(const void *a, const void *b)

{

    return strcmp(((const stat_cmd_arg *)a)->cmd, (const char *)b);

}

static int

sort_by_name(const void *a, const void *b)

{

    return strcmp(((const stat_cmd_arg *)a)->cmd, ((const stat_cmd_arg *)b)->cmd);

}

void stat_tap_init(void)

{

    stat_cmd_arg_list = wmem_list_new(wmem_epan_scope());

    registered_stat_tables = wmem_tree_new(wmem_epan_scope());

}

void

register_stat_tap_ui(stat_tap_ui *ui, void *userdata)

{

    stat_cmd_arg *newsca;

    /* Key is already present */

    if (wmem_list_find_custom(stat_cmd_arg_list, ui->cli_string, search_duplicate))

        return;

    newsca = wmem_new(wmem_epan_scope(), stat_cmd_arg);

    newsca->cmd= wmem_strdup(wmem_epan_scope(), ui->cli_string);

    newsca->init=ui->tap_init_cb;

    newsca->userdata=userdata;

    wmem_list_insert_sorted(stat_cmd_arg_list, newsca, sort_by_name);

}

/* **********************************************************************

 * Function called for a stat command-line argument

 * ********************************************************************** */

bool

process_stat_cmd_arg(const char *optstr)

{

    wmem_list_frame_t *entry;

    stat_cmd_arg *sca;

    stat_requested *tr;

    char *stat_command = g_strdup(optstr);

    /* Renamed in Wireshark 3.0, backwards compatibility. */

    if (!strncmp(stat_command, "follow,ssl", strlen("follow,ssl"))) {

        memcpy(stat_command + 7, "tls", 3);

    }

    /* The strings "ipx" or "ipv6" must be tested before "ip" to select the

      right tap so the sorting does matter.  And it's also why the list is

      walked backwards */

    for (entry = wmem_list_tail(stat_cmd_arg_list); entry; entry = wmem_list_frame_prev(entry)) {

        sca = (stat_cmd_arg*)wmem_list_frame_data(entry);

        if (!strncmp(sca->cmd, stat_command, strlen(sca->cmd))) {

            tr=g_new(stat_requested, 1);

            tr->sca = sca;

            tr->arg = stat_command;

            stats_requested = g_slist_append(stats_requested, tr);

            return true;

        }

    }

    g_free(stat_command);

    return false;

}

/* **********************************************************************

 * Function to list all possible tap command-line arguments

 * ********************************************************************** */

static void

list_stat_cmd_args_func(void *data, void *userdata _U_)

{

    fprintf(stderr,"     %s\n", ((stat_cmd_arg*)data)->cmd);

}

void

list_stat_cmd_args(void)

{

    wmem_list_foreach(stat_cmd_arg_list, list_stat_cmd_args_func, NULL);

}

/* **********************************************************************

 * Function to process stats requested with command-line arguments

 * ********************************************************************** */

bool

start_requested_stats(void)

{

    stat_requested *sr;

    bool success = true;

    while(stats_requested){

        sr=(stat_requested *)stats_requested->data;

        success &= (*sr->sca->init)(sr->arg,sr->sca->userdata);

        stats_requested=g_slist_remove(stats_requested, sr);

        g_free(sr->arg);

        g_free(sr);

    }

    return success;

}

void register_stat_tap_table_ui(stat_tap_table_ui *ui)

{

    wmem_tree_insert_string(registered_stat_tables, ui->cli_string, ui, 0);

}

stat_tap_table_ui *stat_tap_by_name(const char *name)

{

    return (stat_tap_table_ui *) wmem_tree_lookup_string(registered_stat_tables, name, 0);

}

void stat_tap_iterate_tables(wmem_foreach_func func, void *user_data)

{

    wmem_tree_foreach(registered_stat_tables, func, user_data);

}

void stat_tap_get_filter(stat_tap_table_ui* new_stat, const char *opt_arg, const char **filter, char** err)

{

    unsigned len = (unsigned) strlen(new_stat->cli_string);

    *filter=NULL;

    *err=NULL;

    if (!strncmp(opt_arg, new_stat->cli_string, len))

    {

        if (opt_arg[len] == ',')

        {

           *filter = opt_arg + len+1;

        }

    }

    if (new_stat->stat_filter_check_cb)

        new_stat->stat_filter_check_cb(opt_arg, filter, err);

}

stat_tap_table* stat_tap_init_table(const char *name, int num_fields, int num_elements,

                const char *filter_string)

{

    stat_tap_table* new_table = g_new0(stat_tap_table, 1);

    new_table->title = name;

    new_table->num_elements = num_elements;

    new_table->num_fields = num_fields;

    new_table->filter_string = filter_string;

    new_table->elements = g_new0(stat_tap_table_item_type*, num_elements);

    return new_table;

}

stat_tap_table *stat_tap_find_table(stat_tap_table_ui *ui, const char *name)

{

    unsigned i = 0;

    stat_tap_table *stat_table;

    if (ui->tables == NULL)

        return NULL;

    for (i = 0; i < ui->tables->len; i++) {

        stat_table = g_array_index(ui->tables, stat_tap_table *, i);

        if (!g_strcmp0(stat_table->title, name)) {

            return stat_table;

        }

    }

    return NULL;

}

void stat_tap_add_table(stat_tap_table_ui* new_stat, stat_tap_table* table)

{

    if (new_stat->tables == NULL)

        new_stat->tables = g_array_new(false, true, sizeof(stat_tap_table*));

    g_array_insert_val(new_stat->tables, new_stat->tables->len, table);

}

void stat_tap_init_table_row(stat_tap_table *stat_table, unsigned table_index, unsigned num_fields, const stat_tap_table_item_type* fields)

{

    /* we have discovered a new procedure. Extend the table accordingly */

    if(table_index>=stat_table->num_elements){

        unsigned old_num_elements=stat_table->num_elements;

        unsigned i;

        stat_table->num_elements=table_index+1;

        stat_table->elements = (stat_tap_table_item_type**)g_realloc(stat_table->elements, sizeof(stat_tap_table_item_type*)*(stat_table->num_elements));

        for(i=old_num_elements;i<stat_table->num_elements;i++){

            stat_table->elements[i] = g_new0(stat_tap_table_item_type, stat_table->num_fields);

        }

    }

    memcpy(stat_table->elements[table_index], fields, num_fields*sizeof(stat_tap_table_item_type));

}

stat_tap_table_item_type* stat_tap_get_field_data(const stat_tap_table *stat_table, unsigned table_index, unsigned field_index)

{

    stat_tap_table_item_type* field_value;

    ws_assert(table_index < stat_table->num_elements);

    field_value = stat_table->elements[table_index];

    ws_assert(field_index < stat_table->num_fields);

    return &field_value[field_index];

}

void stat_tap_set_field_data(stat_tap_table *stat_table, unsigned table_index, unsigned field_index, stat_tap_table_item_type* field_data)

{

    stat_tap_table_item_type* field_value;

    ws_assert(table_index < stat_table->num_elements);

    field_value = stat_table->elements[table_index];

    ws_assert(field_index < stat_table->num_fields);

    field_value[field_index] = *field_data;

}

void reset_stat_table(stat_tap_table_ui* new_stat)

{

    unsigned i = 0;

    stat_tap_table *stat_table;

    for (i = 0; i < new_stat->tables->len; i++)

    {

        stat_table = g_array_index(new_stat->tables, stat_tap_table*, i);

        if (new_stat->stat_tap_reset_table_cb)

            new_stat->stat_tap_reset_table_cb(stat_table);

    }

}

void free_stat_tables(stat_tap_table_ui* new_stat)

{

    unsigned i = 0, element, field_index;

    stat_tap_table *stat_table;

    stat_tap_table_item_type* field_data;

    for (i = 0; i < new_stat->tables->len; i++)

    {

        stat_table = g_array_index(new_stat->tables, stat_tap_table*, i);

        for (element = 0; element < stat_table->num_elements; element++)

        {

            for (field_index = 0; field_index < stat_table->num_fields; field_index++)

            {

                field_data = stat_tap_get_field_data(stat_table, element, field_index);

                /* Give dissector a crack at it */

                /* XXX Should this be per-row instead? */

                if (new_stat->stat_tap_free_table_item_cb)

                    new_stat->stat_tap_free_table_item_cb(stat_table, element, field_index, field_data);

            }

            g_free(stat_table->elements[element]);

        }

        g_free(stat_table->elements);

        g_free(stat_table);

    }

    g_array_set_size(new_stat->tables, 0);

}

/*

 * Editor modelines

 *

 * Local Variables:

 * c-basic-offset: 4

 * tab-width: 8

 * indent-tabs-mode: nil

 * End:

 *

 * ex: set shiftwidth=4 tabstop=8 expandtab:

 * :indentSize=4:tabSize=8:noTabs=true:

 */