plugins/mate/packet-mate.c

   1 /* packet-mate.c
   2  * Routines for the mate Facility's Pseudo-Protocol dissection
   3  *
   4  * Copyright 2004, Luis E. Garcia Ontanon <gopo@webflies.org>
   5  *
   6  * $Id$
   7  *
   8  * Wireshark - Network traffic analyzer
   9  * By Gerald Combs <gerald@wireshark.org>
  10  * Copyright 1998 Gerald Combs
  11  *
  12  * This program is free software; you can redistribute it and/or
  13  * modify it under the terms of the GNU General Public License
  14  * as published by the Free Software Foundation; either version 2
  15  * of the License, or (at your option) any later version.
  16  *
  17  * This program is distributed in the hope that it will be useful,
  18  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  19  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  20  * GNU General Public License for more details.
  21  *
  22  * You should have received a copy of the GNU General Public License
  23  * along with this program; if not, write to the Free Software
  24  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  25  */
  26
  27
  28 /**************************************************************************
  29  * This is the pseudo protocol dissector for the mate module.          ***
  30  * It is intended for this to be just the user interface to the module. ***
  31  **************************************************************************/
  32
  33 #include "mate.h"
  34
  35 static int mate_tap_data = 0;
  36 static mate_config* mc = NULL;
  37
  38 static int proto_mate = -1;
  39
  40 static const gchar* pref_mate_config_filename = "";
  41 static const gchar* current_mate_config_filename = NULL;
  42
  43 static proto_item *mate_i = NULL;
  44
  45 static void pdu_attrs_tree(proto_tree* tree, tvbuff_t *tvb, mate_pdu* pdu) {
  46         AVPN* c;
  47         proto_item *avpl_i;
  48         proto_tree *avpl_t;
  49         int* hfi_p;
  50
  51         avpl_i = proto_tree_add_text(tree,tvb,0,0,"%s Attributes",pdu->cfg->name);
  52         avpl_t = proto_item_add_subtree(avpl_i, pdu->cfg->ett_attr);
  53
  54         for ( c = pdu->avpl->null.next; c->avp; c = c->next) {
  55                 hfi_p = (int *)g_hash_table_lookup(pdu->cfg->my_hfids,(char*)c->avp->n);
  56
  57                 if (hfi_p) {
  58                         proto_tree_add_string(avpl_t,*hfi_p,tvb,0,0,c->avp->v);
  59                 } else {
  60                         g_warning("MATE: error: undefined attribute: mate.%s.%s",pdu->cfg->name,c->avp->n);
  61                         proto_tree_add_text(avpl_t,tvb,0,0,"Undefined attribute: %s=%s",c->avp->n, c->avp->v);
  62                 }
  63         }
  64 }
  65
  66 static void gop_attrs_tree(proto_tree* tree, tvbuff_t *tvb, mate_gop* gop) {
  67         AVPN* c;
  68         proto_item *avpl_i;
  69         proto_tree *avpl_t;
  70         int* hfi_p;
  71
  72         avpl_i = proto_tree_add_text(tree,tvb,0,0,"%s Attributes",gop->cfg->name);
  73         avpl_t = proto_item_add_subtree(avpl_i, gop->cfg->ett_attr);
  74
  75         for ( c = gop->avpl->null.next; c->avp; c = c->next) {
  76                 hfi_p = (int *)g_hash_table_lookup(gop->cfg->my_hfids,(char*)c->avp->n);
  77
  78                 if (hfi_p) {
  79                         proto_tree_add_string(avpl_t,*hfi_p,tvb,0,0,c->avp->v);
  80                 } else {
  81                         g_warning("MATE: error: undefined attribute: mate.%s.%s",gop->cfg->name,c->avp->n);
  82                         proto_tree_add_text(avpl_t,tvb,0,0,"Undefined attribute: %s=%s",c->avp->n, c->avp->v);
  83                 }
  84         }
  85 }
  86
  87 static void gog_attrs_tree(proto_tree* tree, tvbuff_t *tvb, mate_gog* gog) {
  88         AVPN* c;
  89         proto_item *avpl_i;
  90         proto_tree *avpl_t;
  91         int* hfi_p;
  92
  93         avpl_i = proto_tree_add_text(tree,tvb,0,0,"%s Attributes",gog->cfg->name);
  94         avpl_t = proto_item_add_subtree(avpl_i, gog->cfg->ett_attr);
  95
  96         for ( c = gog->avpl->null.next; c->avp; c = c->next) {
  97                 hfi_p = (int *)g_hash_table_lookup(gog->cfg->my_hfids,(char*)c->avp->n);
  98
  99                 if (hfi_p) {
 100                         proto_tree_add_string(avpl_t,*hfi_p,tvb,0,0,c->avp->v);
 101                 } else {
 102                         g_warning("MATE: error: undefined attribute: mate.%s.%s",gog->cfg->name,c->avp->n);
 103                         proto_tree_add_text(avpl_t,tvb,0,0,"Undefined attribute: %s=%s",c->avp->n, c->avp->v);
 104                 }
 105         }
 106 }
 107
 108 static void mate_gop_tree(proto_tree* pdu_tree, tvbuff_t *tvb, mate_gop* gop);
 109
 110 static void mate_gog_tree(proto_tree* tree, tvbuff_t *tvb, mate_gog* gog, mate_gop* gop) {
 111         proto_item *gog_item;
 112         proto_tree *gog_tree;
 113         proto_item *gog_time_item;
 114         proto_tree *gog_time_tree;
 115         proto_item *gog_gops_item;
 116         proto_tree *gog_gops_tree;
 117         mate_gop* gog_gops;
 118         proto_item *gog_gop_item;
 119         proto_tree *gog_gop_tree;
 120         mate_pdu* pdu;
 121
 122         gog_item = proto_tree_add_uint(tree,gog->cfg->hfid,tvb,0,0,gog->id);
 123         gog_tree = proto_item_add_subtree(gog_item,gog->cfg->ett);
 124
 125         gog_attrs_tree(gog_tree,tvb,gog);
 126
 127         if (gog->cfg->show_times) {
 128                 gog_time_item = proto_tree_add_text(gog_tree,tvb,0,0,"%s Times",gog->cfg->name);
 129                 gog_time_tree = proto_item_add_subtree(gog_time_item, gog->cfg->ett_times);
 130
 131                 proto_tree_add_float(gog_time_tree, gog->cfg->hfid_start_time, tvb, 0, 0, gog->start_time);
 132                 proto_tree_add_float(gog_time_tree, gog->cfg->hfid_last_time, tvb, 0, 0, gog->last_time - gog->start_time);
 133         }
 134
 135         gog_gops_item = proto_tree_add_uint(gog_tree, gog->cfg->hfid_gog_num_of_gops,
 136                                                                            tvb, 0, 0, gog->num_of_gops);
 137
 138         gog_gops_tree = proto_item_add_subtree(gog_gops_item, gog->cfg->ett_children);
 139
 140         for (gog_gops = gog->gops; gog_gops; gog_gops = gog_gops->next) {
 141
 142                 if (gop != gog_gops) {
 143                         if (gog->cfg->gop_tree_mode == GOP_FULL_TREE) {
 144                                 mate_gop_tree(gog_gops_tree, tvb, gog_gops);
 145                         } else {
 146                                 gog_gop_item = proto_tree_add_uint(gog_gops_tree,gog_gops->cfg->hfid,tvb,0,0,gog_gops->id);
 147
 148                                 if (gog->cfg->gop_tree_mode == GOP_BASIC_TREE) {
 149                                         gog_gop_tree = proto_item_add_subtree(gog_gop_item, gog->cfg->ett_gog_gop);
 150
 151                                         proto_tree_add_text(gog_gop_tree, tvb,0,0, "Started at: %f", gog_gops->start_time);
 152
 153
 154                                         proto_tree_add_text(gog_gop_tree, tvb,0,0, "%s Duration: %f",
 155                                                                                 gog_gops->cfg->name, gog_gops->last_time - gog_gops->start_time);
 156
 157                                         if (gog_gops->released)
 158                                                 proto_tree_add_text(gog_gop_tree, tvb,0,0, "%s has been released, Time: %f",
 159                                                                                         gog_gops->cfg->name, gog_gops->release_time - gog_gops->start_time);
 160
 161                                         proto_tree_add_text(gog_gop_tree, tvb,0,0, "Number of Pdus: %u",gog_gops->num_of_pdus);
 162
 163                                         if (gop->pdus && gop->cfg->pdu_tree_mode != GOP_NO_TREE) {
 164                                                 proto_tree_add_uint(gog_gop_tree,gog->cfg->hfid_gog_gopstart,tvb,0,0,gog_gops->pdus->frame);
 165
 166                                                 for (pdu = gog_gops->pdus->next ; pdu; pdu = pdu->next) {
 167                                                         if (pdu->is_stop) {
 168                                                                 proto_tree_add_uint(gog_gop_tree,gog->cfg->hfid_gog_gopstop,tvb,0,0,pdu->frame);
 169                                                                 break;
 170                                                         }
 171                                                 }
 172                                         }
 173                                 }
 174
 175                         }
 176                 } else {
 177                          proto_tree_add_uint_format(gog_gops_tree,gop->cfg->hfid,tvb,0,0,gop->id,"current %s Gop: %d",gop->cfg->name,gop->id);
 178                 }
 179         }
 180 }
 181
 182 static void mate_gop_tree(proto_tree* tree, tvbuff_t *tvb, mate_gop* gop) {
 183         proto_item *gop_item;
 184         proto_tree *gop_time_tree;
 185         proto_item *gop_time_item;
 186         proto_tree *gop_tree;
 187         proto_item *gop_pdu_item;
 188         proto_tree *gop_pdu_tree;
 189         mate_pdu* gop_pdus;
 190         float rel_time;
 191         float pdu_rel_time;
 192         const gchar* pdu_str;
 193         const gchar* type_str;
 194         guint32 pdu_item;
 195
 196         gop_item = proto_tree_add_uint(tree,gop->cfg->hfid,tvb,0,0,gop->id);
 197         gop_tree = proto_item_add_subtree(gop_item, gop->cfg->ett);
 198
 199         if (gop->gop_key) proto_tree_add_text(gop_tree,tvb,0,0,"GOP Key: %s",gop->gop_key);
 200
 201         gop_attrs_tree(gop_tree,tvb,gop);
 202
 203         if (gop->cfg->show_times) {
 204                 gop_time_item = proto_tree_add_text(gop_tree,tvb,0,0,"%s Times",gop->cfg->name);
 205                 gop_time_tree = proto_item_add_subtree(gop_time_item, gop->cfg->ett_times);
 206
 207                 proto_tree_add_float(gop_time_tree, gop->cfg->hfid_start_time, tvb, 0, 0, gop->start_time);
 208
 209                 if (gop->released) {
 210                         proto_tree_add_float(gop_time_tree, gop->cfg->hfid_stop_time, tvb, 0, 0, gop->release_time - gop->start_time);
 211                         proto_tree_add_float(gop_time_tree, gop->cfg->hfid_last_time, tvb, 0, 0, gop->last_time - gop->start_time);
 212                 } else {
 213                         proto_tree_add_float(gop_time_tree, gop->cfg->hfid_last_time, tvb, 0, 0, gop->last_time - gop->start_time);
 214                 }
 215         }
 216
 217         gop_pdu_item = proto_tree_add_uint(gop_tree, gop->cfg->hfid_gop_num_pdus, tvb, 0, 0,gop->num_of_pdus);
 218
 219         if (gop->cfg->pdu_tree_mode != GOP_NO_TREE) {
 220
 221                 gop_pdu_tree = proto_item_add_subtree(gop_pdu_item, gop->cfg->ett_children);
 222
 223                 rel_time = gop->start_time;
 224
 225                 type_str = (gop->cfg->pdu_tree_mode == GOP_FRAME_TREE ) ? "in frame:" : "id:";
 226
 227                 for (gop_pdus = gop->pdus; gop_pdus; gop_pdus = gop_pdus->next) {
 228
 229                         pdu_item = (gop->cfg->pdu_tree_mode == GOP_FRAME_TREE ) ? gop_pdus->frame : gop_pdus->id;
 230
 231                         if (gop_pdus->is_start) {
 232                                 pdu_str = "Start ";
 233                         } else if (gop_pdus->is_stop) {
 234                                 pdu_str = "Stop ";
 235                         } else if (gop_pdus->after_release) {
 236                                 pdu_str = "After stop ";
 237                         } else {
 238                                 pdu_str = "";
 239                         }
 240
 241                         pdu_rel_time = gop_pdus->time_in_gop != 0.0 ? gop_pdus->time_in_gop - rel_time : (float) 0.0;
 242
 243                         proto_tree_add_uint_format(gop_pdu_tree,gop->cfg->hfid_gop_pdu,tvb,0,0,pdu_item,
 244                                                                            "%sPDU: %s %i (%f : %f)",pdu_str, type_str,
 245                                                                            pdu_item, gop_pdus->time_in_gop,
 246                                                                            pdu_rel_time);
 247
 248                         rel_time = gop_pdus->time_in_gop;
 249
 250                 }
 251         }
 252 }
 253
 254
 255 static void mate_pdu_tree(mate_pdu *pdu, tvbuff_t *tvb, proto_tree* tree) {
 256         proto_item *pdu_item;
 257         proto_tree *pdu_tree;
 258
 259         if ( ! pdu ) return;
 260
 261         if (pdu->gop && pdu->gop->gog) {
 262                 proto_item_append_text(mate_i," %s:%d->%s:%d->%s:%d",
 263                                                            pdu->cfg->name,pdu->id,
 264                                                            pdu->gop->cfg->name,pdu->gop->id,
 265                                                            pdu->gop->gog->cfg->name,pdu->gop->gog->id);
 266         } else if (pdu->gop) {
 267                 proto_item_append_text(mate_i," %s:%d->%s:%d",
 268                                                            pdu->cfg->name,pdu->id,
 269                                                            pdu->gop->cfg->name,pdu->gop->id);
 270         } else {
 271                 proto_item_append_text(mate_i," %s:%d",pdu->cfg->name,pdu->id);
 272         }
 273
 274         pdu_item = proto_tree_add_uint(tree,pdu->cfg->hfid,tvb,0,0,pdu->id);
 275         pdu_tree = proto_item_add_subtree(pdu_item, pdu->cfg->ett);
 276         proto_tree_add_float(pdu_tree,pdu->cfg->hfid_pdu_rel_time, tvb, 0, 0, pdu->rel_time);
 277
 278         if (pdu->gop) {
 279                 proto_tree_add_float(pdu_tree,pdu->cfg->hfid_pdu_time_in_gop, tvb, 0, 0, pdu->time_in_gop);
 280                 mate_gop_tree(tree,tvb,pdu->gop);
 281
 282                 if (pdu->gop->gog)
 283                         mate_gog_tree(tree,tvb,pdu->gop->gog,pdu->gop);
 284         }
 285
 286         if (pdu->avpl) {
 287                 pdu_attrs_tree(pdu_tree,tvb,pdu);
 288         }
 289 }
 290
 291 static void mate_tree(tvbuff_t *tvb, packet_info *pinfo, proto_tree *tree) {
 292         mate_pdu* pdus;
 293         proto_tree *mate_t;
 294
 295         if ( ! mc || ! tree ) return;
 296
 297         mate_analyze_frame(pinfo,tree);
 298
 299         if (( pdus = mate_get_pdus(pinfo->fd->num) )) {
 300                 for ( ; pdus; pdus = pdus->next_in_frame) {
 301                         mate_i = proto_tree_add_protocol_format(tree,mc->hfid_mate,tvb,0,0,"MATE");
 302                         mate_t = proto_item_add_subtree(mate_i, mc->ett_root);
 303                         mate_pdu_tree(pdus,tvb,mate_t);
 304                 }
 305         }
 306 }
 307
 308 static int mate_packet(void *prs _U_,  packet_info* tree _U_, epan_dissect_t *edt _U_, const void *dummy _U_) {
 309         /* nothing to do yet */
 310         return 0;
 311 }
 312
 313 extern
 314 void
 315 proto_reg_handoff_mate(void)
 316 {
 317         GString* tap_error = NULL;
 318
 319         if ( *pref_mate_config_filename != '\0' ) {
 320
 321                 if (current_mate_config_filename) {
 322                         report_failure("Mate cannot reconfigure itself.\n"
 323                                                    "for changes to be applied you have to restart wireshark\n");
 324                         return;
 325                 }
 326
 327                 if (!mc) {
 328                         mc = mate_make_config(pref_mate_config_filename,proto_mate);
 329
 330                         if (mc) {
 331                                 /* XXX: alignment warnings, what do they mean? */
 332                                 proto_register_field_array(proto_mate, (hf_register_info*)(void *)mc->hfrs->data, mc->hfrs->len );
 333                                 proto_register_subtree_array((gint**)(void*)mc->ett->data, mc->ett->len);
 334                                 register_init_routine(initialize_mate_runtime);
 335
 336                                 tap_error = register_tap_listener("frame", &mate_tap_data,
 337                                     (char*) mc->tap_filter,
 338                                     0,
 339                                     (tap_reset_cb) NULL,
 340                                     mate_packet,
 341                                     (tap_draw_cb) NULL);
 342
 343                                 if ( tap_error ) {
 344                                         g_warning("mate: couldn't (re)register tap: %s",tap_error->str);
 345                                         g_string_free(tap_error, TRUE);
 346                                         mate_tap_data = 0;
 347                                         return;
 348                                 }
 349
 350                                 initialize_mate_runtime();
 351                         }
 352
 353                         current_mate_config_filename = pref_mate_config_filename;
 354
 355                 }
 356         }
 357 }
 358
 359 extern
 360 void
 361 proto_register_mate(void)
 362 {
 363         module_t *mate_module;
 364         dissector_handle_t mate_handle;
 365
 366         proto_mate = proto_register_protocol("Meta Analysis Tracing Engine", "MATE", "mate");
 367         register_dissector("mate",mate_tree,proto_mate);
 368         mate_module = prefs_register_protocol(proto_mate, proto_reg_handoff_mate);
 369         prefs_register_filename_preference(mate_module, "config",
 370                                            "Configuration Filename",
 371                                            "The name of the file containing the mate module's configuration",
 372                                            &pref_mate_config_filename);
 373
 374         mate_handle = find_dissector("mate");
 375         register_postdissector(mate_handle);
 376 }
 377