From Olivier Jacques: Add #defines to disable XML validation.
[obnox/wireshark/wip.git] / tap-protocolinfo.c
1 /* tap-protocolinfo.c
2  * protohierstat   2002 Ronnie Sahlberg
3  *
4  * $Id$
5  *
6  * Ethereal - Network traffic analyzer
7  * By Gerald Combs <gerald@ethereal.com>
8  * Copyright 1998 Gerald Combs
9  * 
10  * This program is free software; you can redistribute it and/or
11  * modify it under the terms of the GNU General Public License
12  * as published by the Free Software Foundation; either version 2
13  * of the License, or (at your option) any later version.
14  * 
15  * This program is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18  * GNU General Public License for more details.
19  * 
20  * You should have received a copy of the GNU General Public License
21  * along with this program; if not, write to the Free Software
22  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
23  */
24
25 /* This module provides Protocol Column Info tap for tethereal */
26
27 #ifdef HAVE_CONFIG_H
28 # include "config.h"
29 #endif
30
31 #include <stdio.h>
32
33 #ifdef HAVE_SYS_TYPES_H
34 # include <sys/types.h>
35 #endif
36
37 #include <string.h>
38 #include "epan/epan_dissect.h"
39 #include "epan/column-utils.h"
40 #include "epan/proto.h"
41 #include <epan/tap.h>
42 #include "register.h"
43
44 typedef struct _pci_t {
45         char *filter;
46         int hf_index;
47 } pci_t;
48
49
50 static int
51 protocolinfo_packet(void *prs, packet_info *pinfo, epan_dissect_t *edt, void *dummy _U_)
52 {
53         pci_t *rs=prs;
54         GPtrArray *gp;
55         guint i;
56         char *str;
57
58         /*
59          * XXX - there needs to be a way for "protocolinfo_init()" to
60          * find out whether the columns are being generated and, if not,
61          * to report an error and exit, as the whole point of this tap
62          * is to modify the columns, and if the columns aren't being
63          * displayed, that makes this tap somewhat pointless.
64          *
65          * To prevent a crash, we check whether pinfo->cinfo is null
66          * and, if so, we report that error and exit.
67          */
68         if (pinfo->cinfo == NULL) {
69                 fprintf(stderr, "tethereal: the proto,colinfo tap doesn't work if the columns aren't being printed.\n");
70                 exit(1);
71         }
72         gp=proto_get_finfo_ptr_array(edt->tree, rs->hf_index);
73         if(!gp){
74                 return 0;
75         }
76
77         for(i=0;i<gp->len;i++){
78                 str=proto_construct_dfilter_string(gp->pdata[i], NULL);
79                 if(str){
80                         col_append_fstr(pinfo->cinfo, COL_INFO, "  %s",str);
81                         g_free(str);
82                 }
83         }
84         return 0;
85 }
86
87
88
89 static void
90 protocolinfo_init(char *optarg)
91 {
92         pci_t *rs;
93         char *field=NULL;
94         char *filter=NULL;
95         header_field_info *hfi;
96         GString *error_string;
97
98         if(!strncmp("proto,colinfo,",optarg,14)){
99                 filter=optarg+14;
100                 field=strchr(filter,',');
101                 if(field){
102                         field+=1;  /* skip the ',' */
103                 }
104         }
105         if(!field){
106                 fprintf(stderr, "tethereal: invalid \"-z proto,colinfo,<filter>,<field>\" argument\n");
107                 exit(1);
108         }
109
110         hfi=proto_registrar_get_byname(field);
111         if(!hfi){
112                 fprintf(stderr, "tethereal: Field \"%s\" does not exist.\n", field);
113                 exit(1);
114         }
115
116         rs=g_malloc(sizeof(pci_t));
117         rs->hf_index=hfi->id;
118         if((field-filter)>1){
119                 rs->filter=g_malloc(field-filter);
120                 strncpy(rs->filter,filter,(field-filter)-1);
121                 rs->filter[(field-filter)-1]=0;
122         } else {
123                 rs->filter=NULL;
124         }
125
126         error_string=register_tap_listener("frame", rs, rs->filter, NULL, protocolinfo_packet, NULL);
127         if(error_string){
128                 /* error, we failed to attach to the tap. complain and clean up */
129                 fprintf(stderr, "tethereal: Couldn't register proto,colinfo tap: %s\n",
130                     error_string->str);
131                 g_string_free(error_string, TRUE);
132                 if(rs->filter){
133                         g_free(rs->filter);
134                 }
135                 g_free(rs);
136
137                 exit(1);
138         }
139 }
140
141
142 void
143 register_tap_listener_protocolinfo(void)
144 {
145         register_ethereal_tap("proto,colinfo,", protocolinfo_init);
146 }
147