Add routines for adding items to a protocol tree that take arguments of
[obnox/wireshark/wip.git] / packet-udp.c
1 /* packet-udp.c
2  * Routines for UDP packet disassembly
3  *
4  * $Id: packet-udp.c,v 1.72 2000/05/31 05:07:52 guy Exp $
5  *
6  * Ethereal - Network traffic analyzer
7  * By Gerald Combs <gerald@zing.org>
8  * Copyright 1998 Gerald Combs
9  *
10  * Richard Sharpe, 13-Feb-1999, added dispatch table support and 
11  *                              support for tftp.
12  * 
13  * This program is free software; you can redistribute it and/or
14  * modify it under the terms of the GNU General Public License
15  * as published by the Free Software Foundation; either version 2
16  * of the License, or (at your option) any later version.
17  * 
18  * This program is distributed in the hope that it will be useful,
19  * but WITHOUT ANY WARRANTY; without even the implied warranty of
20  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
21  * GNU General Public License for more details.
22  * 
23  * You should have received a copy of the GNU General Public License
24  * along with this program; if not, write to the Free Software
25  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
26  */
27  
28 #ifdef HAVE_CONFIG_H
29 # include "config.h"
30 #endif
31
32 #ifdef HAVE_SYS_TYPES_H
33 # include <sys/types.h>
34 #endif
35
36 #ifdef HAVE_NETINET_IN_H
37 # include <netinet/in.h>
38 #endif
39
40 #include <stdio.h>
41 #include <stdlib.h>
42
43 #include <glib.h>
44 #include "globals.h"
45 #include "resolv.h"
46
47 #include "plugins.h"
48 #include "packet-udp.h"
49
50 #include "packet-ip.h"
51 #include "packet-rpc.h"
52 #include "packet-tftp.h"
53 #include "packet-vines.h"
54
55 static int proto_udp = -1;              
56 static int hf_udp_srcport = -1;
57 static int hf_udp_dstport = -1;
58 static int hf_udp_port = -1;
59 static int hf_udp_length = -1;
60 static int hf_udp_checksum = -1;
61
62 static gint ett_udp = -1;
63
64 /* UDP structs and definitions */
65
66 typedef struct _e_udphdr {
67   guint16 uh_sport;
68   guint16 uh_dport;
69   guint16 uh_ulen;
70   guint16 uh_sum;
71 } e_udphdr;
72
73 /* UDP Ports -> should go in packet-udp.h */
74
75 #define UDP_PORT_TFTP    69
76 #define UDP_PORT_VINES  573
77
78 static dissector_table_t udp_dissector_table;
79
80 /* Determine if there is a sub-dissector and call it.  This has been */
81 /* separated into a stand alone routine to other protocol dissectors */
82 /* can call to it, ie. socks    */
83
84 void
85 decode_udp_ports( const u_char *pd, int offset, frame_data *fd,
86         proto_tree *tree, int uh_sport, int uh_dport){
87   dissector_t sub_dissector;
88
89 /* determine if this packet is part of a conversation and call dissector */
90 /* for the conversation if available */
91
92   sub_dissector = find_conversation_dissector( &pi.src, &pi.dst, PT_UDP,
93                 uh_sport, uh_dport);
94   if (sub_dissector){
95         (sub_dissector)(pd, offset, fd, tree);
96         return;
97   }
98
99   /* ONC RPC.  We can't base this on anything in the UDP header; we have
100      to look at the payload.  If "dissect_rpc()" returns TRUE, it was
101      an RPC packet, otherwise it's some other type of packet. */
102   if (dissect_rpc(pd, offset, fd, tree))
103     return;
104
105   /* try to apply the plugins */
106 #ifdef HAVE_PLUGINS
107   {
108       plugin *pt_plug = plugin_list;
109
110       if (enabled_plugins_number > 0) {
111           while (pt_plug) {
112               if (pt_plug->enabled && !strcmp(pt_plug->protocol, "udp") &&
113                   tree && dfilter_apply(pt_plug->filter, tree, pd, fd->cap_len)) {
114                   pt_plug->dissector(pd, offset, fd, tree);
115                   return;
116               }
117               pt_plug = pt_plug->next;
118           }
119       }
120   }
121 #endif
122
123   /* do lookup with the subdissector table */
124   if (dissector_try_port(udp_dissector_table, uh_sport, pd, offset, fd, tree) ||
125       dissector_try_port(udp_dissector_table, uh_dport, pd, offset, fd, tree))
126     return;
127
128   /* XXX - we should do these with the subdissector table as well. */
129 #define PORT_IS(port)   (uh_sport == port || uh_dport == port)
130   if (PORT_IS(UDP_PORT_VINES)) {
131     /* FIXME: AFAIK, src and dst port must be the same */
132     dissect_vines_frp(pd, offset, fd, tree);
133   } else if (PORT_IS(UDP_PORT_TFTP)) {
134     /* This is the first point of call, but it adds a dynamic call */
135     dissector_add("udp.port", MAX(uh_sport, uh_dport), dissect_tftp);  /* Add to table */
136     dissect_tftp(pd, offset, fd, tree);
137   } else
138     dissect_data(pd, offset, fd, tree);
139 }
140
141
142 static void
143 dissect_udp(const u_char *pd, int offset, frame_data *fd, proto_tree *tree) {
144   e_udphdr  uh;
145   guint16    uh_sport, uh_dport, uh_ulen, uh_sum;
146   proto_tree *udp_tree;
147   proto_item *ti;
148
149   if (!BYTES_ARE_IN_FRAME(offset, sizeof(e_udphdr))) {
150     dissect_data(pd, offset, fd, tree);
151     return;
152   }
153
154   /* Avoids alignment problems on many architectures. */
155   memcpy(&uh, &pd[offset], sizeof(e_udphdr));
156   uh_sport = ntohs(uh.uh_sport);
157   uh_dport = ntohs(uh.uh_dport);
158   uh_ulen  = ntohs(uh.uh_ulen);
159   uh_sum   = ntohs(uh.uh_sum);
160   
161   if (check_col(fd, COL_PROTOCOL))
162     col_add_str(fd, COL_PROTOCOL, "UDP");
163   if (check_col(fd, COL_INFO))
164     col_add_fstr(fd, COL_INFO, "Source port: %s  Destination port: %s",
165             get_udp_port(uh_sport), get_udp_port(uh_dport));
166     
167   if (tree) {
168     ti = proto_tree_add_item(tree, proto_udp, NullTVB, offset, 8, FALSE);
169     udp_tree = proto_item_add_subtree(ti, ett_udp);
170
171     proto_tree_add_uint_format(udp_tree, hf_udp_srcport, NullTVB, offset, 2, uh_sport,
172         "Source port: %s (%u)", get_udp_port(uh_sport), uh_sport);
173     proto_tree_add_uint_format(udp_tree, hf_udp_dstport, NullTVB, offset + 2, 2, uh_dport,
174         "Destination port: %s (%u)", get_udp_port(uh_dport), uh_dport);
175
176     proto_tree_add_uint_hidden(udp_tree, hf_udp_port, NullTVB, offset, 2, uh_sport);
177     proto_tree_add_uint_hidden(udp_tree, hf_udp_port, NullTVB, offset+2, 2, uh_dport);
178
179     proto_tree_add_uint(udp_tree, hf_udp_length, NullTVB, offset + 4, 2,  uh_ulen);
180     proto_tree_add_uint_format(udp_tree, hf_udp_checksum, NullTVB, offset + 6, 2, uh_sum,
181         "Checksum: 0x%04x", uh_sum);
182   }
183
184   /* Skip over header */
185   offset += 8;
186
187   pi.ptype = PT_UDP;
188   pi.srcport = uh_sport;
189   pi.destport = uh_dport;
190
191 /* call sub-dissectors */
192   decode_udp_ports( pd, offset, fd, tree, uh_sport, uh_dport);
193
194 }
195
196 void
197 proto_register_udp(void)
198 {
199         static hf_register_info hf[] = {
200                 { &hf_udp_srcport,
201                 { "Source Port",        "udp.srcport", FT_UINT16, BASE_DEC, NULL, 0x0,
202                         "" }},
203
204                 { &hf_udp_dstport,
205                 { "Destination Port",   "udp.dstport", FT_UINT16, BASE_DEC, NULL, 0x0,
206                         "" }},
207
208                 { &hf_udp_port,
209                 { "Source or Destination Port", "udp.port", FT_UINT16, BASE_DEC,  NULL, 0x0,
210                         "" }},
211
212                 { &hf_udp_length,
213                 { "Length",             "udp.length", FT_UINT16, BASE_DEC, NULL, 0x0,
214                         "" }},
215
216                 { &hf_udp_checksum,
217                 { "Checksum",           "udp.checksum", FT_UINT16, BASE_HEX, NULL, 0x0,
218                         "" }},
219         };
220         static gint *ett[] = {
221                 &ett_udp,
222         };
223
224         proto_udp = proto_register_protocol("User Datagram Protocol", "udp");
225         proto_register_field_array(proto_udp, hf, array_length(hf));
226         proto_register_subtree_array(ett, array_length(ett));
227
228 /* subdissector code */
229         udp_dissector_table = register_dissector_table("udp.port");
230 }
231
232 void
233 proto_reg_handoff_udp(void)
234 {
235         dissector_add("ip.proto", IP_PROTO_UDP, dissect_udp);
236 }