Add expert info configuration framework. Bug 2412 (https://bugs.wireshark.org/bugzilla/show_bug.cgi?id=2412).

Expert info "fields" can now be registered/addressed by name.  Right now, the basic framework allows expert info fields to become "display filters".  However more could be done, like user preferences overriding default severity level, speeding up expert info dialog load time by not needing to redissect a file, etc.

Long term goal is to have all expert_info filterable and have the functionality of expert_add_info_format() include the "registered index".  expert_add_info_format_text() is the workaround until all current calls to expert_add_info_format() have been updated with either expert_add_info() or expert_add_info_format_text().  Then the remaining expert_add_info_format_text() will be renamed to expert_add_info_format().

svn path=/trunk/; revision=49559
This commit is contained in:
Michael Mann 2013-05-24 17:59:36 +00:00
parent ecd3073813
commit 0d1a4b2920
5 changed files with 277 additions and 11 deletions

View File

@ -103,13 +103,14 @@ epan_init(void (*register_all_protocols_func)(register_cb cb, gpointer client_da
#endif
tap_init();
prefs_init();
expert_init();
proto_init(register_all_protocols_func, register_all_handoffs_func,
cb, client_data);
packet_init();
dfilter_init();
final_registration_all_protocols();
/*host_name_lookup_init();*//* We load the hostname file in cf_open, no need to do it here? */
expert_init();
expert_packet_init();
#ifdef HAVE_LUA
wslua_init(cb, client_data);
#endif
@ -123,6 +124,7 @@ epan_cleanup(void)
proto_cleanup();
prefs_cleanup();
packet_cleanup();
expert_cleanup();
oid_resolv_cleanup();
#ifdef HAVE_LIBGNUTLS
gnutls_global_deinit();

View File

@ -26,6 +26,8 @@
#include "config.h"
#include <stdio.h>
#include "packet.h"
#include "expert.h"
#include "emem.h"
@ -47,8 +49,36 @@ static int hf_expert_msg = -1;
static int hf_expert_group = -1;
static int hf_expert_severity = -1;
struct expert_module
{
const char* proto_name;
int proto_id; /* Cache this for registering hfs */
GList *experts; /* expert_infos for this protocol */
GList *last_expert; /* pointer to end of list of expert_infos */
};
/* List which stores protocols and expert_info that have been registered */
typedef struct _gpa_expertinfo_t {
guint32 len;
guint32 allocated_len;
expert_field_info **ei;
} gpa_expertinfo_t;
static gpa_expertinfo_t gpa_expertinfo;
/*
* List of all modules with expert info.
*/
static emem_tree_t *expert_modules = NULL;
#define EXPERT_REGISTRAR_GET_NTH(eiindex, expinfo) \
if((guint)eiindex >= gpa_expertinfo.len && getenv("WIRESHARK_ABORT_ON_DISSECTOR_BUG")) \
g_error("Unregistered expert info! index=%d", eiindex); \
DISSECTOR_ASSERT_HINT((guint)eiindex < gpa_expertinfo.len, "Unregistered expert info!");\
expinfo = gpa_expertinfo.ei[eiindex];
void
expert_init(void)
expert_packet_init(void)
{
static hf_register_info hf[] = {
{ &hf_expert_msg,
@ -78,13 +108,34 @@ expert_init(void)
}
highest_severity = 0;
if (expert_modules == NULL) {
expert_modules = pe_tree_create(EMEM_TREE_TYPE_RED_BLACK, "expert_modules");
}
}
void
expert_init(void)
{
gpa_expertinfo.len = 0;
gpa_expertinfo.allocated_len = 0;
gpa_expertinfo.ei = NULL;
}
void
expert_packet_cleanup(void)
{
}
void
expert_cleanup(void)
{
if (gpa_expertinfo.allocated_len) {
gpa_expertinfo.len = 0;
gpa_expertinfo.allocated_len = 0;
g_free(gpa_expertinfo.ei);
gpa_expertinfo.ei = NULL;
}
}
@ -94,6 +145,97 @@ expert_get_highest_severity(void)
return highest_severity;
}
expert_module_t *expert_register_protocol(int id)
{
expert_module_t *module;
protocol_t *protocol;
protocol = find_protocol_by_id(id);
module = g_new(expert_module_t,1);
module->proto_id = id;
module->proto_name = proto_get_protocol_short_name(protocol);
module->experts = NULL;
module->last_expert = NULL;
/*
* Insert this module into the appropriate place in the tree.
*/
pe_tree_insert_string(expert_modules, module->proto_name, module, EMEM_TREE_STRING_NOCASE);
return module;
}
static int
expert_register_field_init(expert_field_info *expinfo, expert_module_t* module)
{
expinfo->protocol = module->proto_name;
/* if we always add and never delete, then id == len - 1 is correct */
if (gpa_expertinfo.len >= gpa_expertinfo.allocated_len) {
if (!gpa_expertinfo.ei) {
gpa_expertinfo.allocated_len = PRE_ALLOC_EXPERT_FIELDS_MEM;
gpa_expertinfo.ei = (expert_field_info **)g_malloc(sizeof(expert_field_info *)*PRE_ALLOC_EXPERT_FIELDS_MEM);
} else {
gpa_expertinfo.allocated_len += 1000;
gpa_expertinfo.ei = (expert_field_info **)g_realloc(gpa_expertinfo.ei,
sizeof(expert_field_info *)*gpa_expertinfo.allocated_len);
}
}
gpa_expertinfo.ei[gpa_expertinfo.len] = expinfo;
gpa_expertinfo.len++;
expinfo->id = gpa_expertinfo.len - 1;
return expinfo->id;
}
/* for use with static arrays only, since we don't allocate our own copies
of the expert_field_info struct contained within the exp_register_info struct */
void
expert_register_field_array(expert_module_t* module, ei_register_info *exp, const int num_records)
{
int i;
ei_register_info *ptr = exp;
for (i = 0; i < num_records; i++, ptr++) {
/*
* Make sure we haven't registered this yet.
* Most fields have variables associated with them
* that are initialized to -1; some have array elements,
* or possibly uninitialized variables, so we also allow
* 0 (which is unlikely to be the field ID we get back
* from "expert_register_field_init()").
*/
if (ptr->ids->ei != -1 && ptr->ids->ei != 0) {
fprintf(stderr,
"Duplicate field detected in call to expert_register_field_array: '%s' is already registered\n",
ptr->eiinfo.summary);
return;
}
if (module != NULL) {
if (module->experts == NULL) {
module->experts = g_list_append(NULL, ptr);
module->last_expert = module->experts;
} else {
module->last_expert =
g_list_append(module->last_expert, ptr)->next;
}
}
/* Register the field with the experts */
ptr->ids->ei = expert_register_field_init(&ptr->eiinfo, module);
/* Register with the header field info, so it's display filterable */
ptr->eiinfo.hf_info.p_id = &ptr->ids->hf;
ptr->eiinfo.hf_info.hfinfo.abbrev = ptr->eiinfo.name;
ptr->eiinfo.hf_info.hfinfo.blurb = ptr->eiinfo.summary;
proto_register_field_array(module->proto_id, &ptr->eiinfo.hf_info, 1);
}
}
/* set's the PI_ flags to a protocol item
* (and its parent items till the toplevel) */
@ -131,7 +273,8 @@ expert_create_tree(proto_item *pi, int group, int severity, const char *msg)
}
static void
expert_set_info_vformat(packet_info *pinfo, proto_item *pi, int group, int severity, const char *format, va_list ap)
expert_set_info_vformat(packet_info *pinfo, proto_item *pi, int group, int severity, int hf_index, gboolean use_vaformat,
const char *format, va_list ap)
{
char formatted[ITEM_LABEL_LENGTH];
int tap;
@ -158,12 +301,27 @@ expert_set_info_vformat(packet_info *pinfo, proto_item *pi, int group, int sever
col_add_str(pinfo->cinfo, COL_EXPERT, val_to_str(severity, expert_severity_vals, "Unknown (%u)"));
g_vsnprintf(formatted, ITEM_LABEL_LENGTH, format, ap);
if (use_vaformat) {
g_vsnprintf(formatted, ITEM_LABEL_LENGTH, format, ap);
} else {
g_strlcpy(formatted, format, ITEM_LABEL_LENGTH);
}
tree = expert_create_tree(pi, group, severity, formatted);
ti = proto_tree_add_string(tree, hf_expert_msg, NULL, 0, 0, formatted);
PROTO_ITEM_SET_GENERATED(ti);
if (hf_index == -1) {
/* If no filterable expert info, just add the message */
ti = proto_tree_add_string(tree, hf_expert_msg, NULL, 0, 0, formatted);
PROTO_ITEM_SET_GENERATED(ti);
} else {
/* If filterable expert info, hide the "generic" form of the message,
and generate the formatted filterable expert info */
ti = proto_tree_add_none_format(tree, hf_index, NULL, 0, 0, "%s", formatted);
PROTO_ITEM_SET_GENERATED(ti);
ti = proto_tree_add_string(tree, hf_expert_msg, NULL, 0, 0, formatted);
PROTO_ITEM_SET_HIDDEN(ti);
}
ti = proto_tree_add_uint_format_value(tree, hf_expert_severity, NULL, 0, 0, severity,
"%s", val_to_str_const(severity, expert_severity_vals, "Unknown"));
PROTO_ITEM_SET_GENERATED(ti);
@ -201,7 +359,36 @@ expert_add_info_format(packet_info *pinfo, proto_item *pi, int group, int severi
va_list ap;
va_start(ap, format);
expert_set_info_vformat(pinfo, pi, group, severity, format, ap);
expert_set_info_vformat(pinfo, pi, group, severity, -1, TRUE, format, ap);
va_end(ap);
}
void
expert_add_info(packet_info *pinfo, proto_item *pi, expert_field* expindex)
{
va_list ap;
expert_field_info* eiinfo;
/* Look up the item */
EXPERT_REGISTRAR_GET_NTH(expindex->ei, eiinfo);
/* Not used by expert_set_info_vformat, but need the variable initialized */
va_start(ap, eiinfo);
expert_set_info_vformat(pinfo, pi, eiinfo->group, eiinfo->severity, *eiinfo->hf_info.p_id, FALSE, eiinfo->summary, ap);
va_end(ap);
}
void
expert_add_info_format_text(packet_info *pinfo, proto_item *pi, expert_field* expindex, const char *format, ...)
{
va_list ap;
expert_field_info* eiinfo;
/* Look up the item */
EXPERT_REGISTRAR_GET_NTH(expindex->ei, eiinfo);
va_start(ap, format);
expert_set_info_vformat(pinfo, pi, eiinfo->group, eiinfo->severity, *eiinfo->hf_info.p_id, TRUE, format, ap);
va_end(ap);
}

View File

@ -46,6 +46,39 @@ typedef struct expert_info_s {
proto_item *pitem;
} expert_info_t;
/* Expert Info and Display hf data */
typedef struct expert_field
{
int ei;
int hf;
} expert_field;
#define EI_INIT {-1, -1}
typedef struct expert_field_info {
/* ---------- set by dissector --------- */
const char *name;
int group;
int severity;
const gchar *summary;
/* ------- set by register routines (prefilled by EXPFILL macro, see below) ------ */
int id;
const gchar *protocol;
hf_register_info hf_info;
} expert_field_info;
#define EXPFILL 0, NULL, \
{0, {"Expert Info", NULL, FT_NONE, BASE_NONE, NULL, 0, NULL, HFILL}}
typedef struct ei_register_info {
expert_field *ids; /**< written to by register() function */
expert_field_info eiinfo; /**< the field info to be registered */
} ei_register_info;
typedef struct expert_module expert_module_t;
static const value_string expert_group_vals[] = {
{ PI_CHECKSUM, "Checksum" },
{ PI_SEQUENCE, "Sequence" },
@ -71,6 +104,8 @@ static const value_string expert_severity_vals[] = {
{ 0, NULL }
};
#define PRE_ALLOC_EXPERT_FIELDS_MEM 5000
/* "proto_expert" is exported from libwireshark.dll.
* Thus we need a special declaration.
*/
@ -79,12 +114,28 @@ WS_DLL_PUBLIC int proto_expert;
extern void
expert_init(void);
extern void
expert_packet_init(void);
extern void
expert_cleanup(void);
extern void
expert_packet_cleanup(void);
WS_DLL_PUBLIC int
expert_get_highest_severity(void);
/** Add an expert info.
Add an expert info tree to a protocol item using registered expert info item
@param pinfo Packet info of the currently processed packet. May be NULL if
pi is supplied
@param pi Current protocol item (or NULL)
@param eiindex The registered expert info item
*/
WS_DLL_PUBLIC void
expert_add_info(packet_info *pinfo, proto_item *pi, expert_field* eiindex);
/** Add an expert info.
Add an expert info tree to a protocol item, with classification and message.
@param pinfo Packet info of the currently processed packet. May be NULL if
@ -99,6 +150,32 @@ expert_add_info_format(packet_info *pinfo, proto_item *pi, int group,
int severity, const char *format, ...)
G_GNUC_PRINTF(5, 6);
/** Add an expert info.
Add an expert info tree to a protocol item, using registered expert info item,
but with a formatted message.
@param pinfo Packet info of the currently processed packet. May be NULL if
pi is supplied
@param pi Current protocol item (or NULL)
@param eiindex The registered expert info item
@param format Printf-style format string for additional arguments
*/
WS_DLL_PUBLIC void
expert_add_info_format_text(packet_info *pinfo, proto_item *pi, expert_field* eiindex,
const char *format, ...) G_GNUC_PRINTF(5, 6);
/*
* Register that a protocol has expert info.
*/
WS_DLL_PUBLIC expert_module_t *expert_register_protocol(int id);
/** Register a expert field array.
@param module the protocol handle from expert_register_protocol()
@param ei the ei_register_info array
@param num_records the number of records in exp */
WS_DLL_PUBLIC void
expert_register_field_array(expert_module_t* module, ei_register_info *ei, const int num_records);
/** Add an expert info about not dissected "item"
Add an expert info tree to a not dissected protocol item.
@param tvb The tvb associated with the item.

View File

@ -160,7 +160,7 @@ init_dissection(void)
stream_init();
/* Initialize the expert infos */
expert_init();
expert_packet_init();
}
void
@ -185,7 +185,7 @@ cleanup_dissection(void)
stream_cleanup();
/* Initialize the expert infos */
expert_cleanup();
expert_packet_cleanup();
wmem_leave_file_scope();

View File

@ -5105,7 +5105,7 @@ tmp_fld_check_assert(header_field_info *hfinfo)
}
}
#define PROTO_PRE_ALLOC_HF_FIELDS_MEM 120000
#define PROTO_PRE_ALLOC_HF_FIELDS_MEM (120000+PRE_ALLOC_EXPERT_FIELDS_MEM)
static int
proto_register_field_init(header_field_info *hfinfo, const int parent)
{