Peter Torvals' Internet Cache Protocol dissector.
[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.24 1999/09/14 08:06:23 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 "packet.h"
45 #include "resolv.h"
46
47 int proto_udp = -1;             
48 int hf_udp_srcport = -1;
49 int hf_udp_dstport = -1;
50 int hf_udp_port = -1;
51 int hf_udp_length = -1;
52 int hf_udp_checksum = -1;
53
54 /* UDP structs and definitions */
55
56 typedef struct _e_udphdr {
57   guint16 uh_sport;
58   guint16 uh_dport;
59   guint16 uh_ulen;
60   guint16 uh_sum;
61 } e_udphdr;
62
63 /* UDP Ports -> should go in packet-udp.h */
64
65 #define UDP_PORT_DNS     53
66 #define UDP_PORT_BOOTPS  67
67 #define UDP_PORT_TFTP    69
68 #define UDP_PORT_IPX    213
69 #define UDP_PORT_NBNS   137
70 #define UDP_PORT_NBDGM  138
71 #define UDP_PORT_SNMP   161
72 #define UDP_PORT_ISAKMP 500
73 #define UDP_PORT_RIP    520
74 #define UDP_PORT_VINES  573
75 #define UDP_PORT_RADIUS 1645
76 #define UDP_PORT_RADIUS_NEW 1812
77 #define UDP_PORT_RADACCT 1646
78 #define UDP_PORT_RADACCT_NEW 1813
79 #define UDP_PORT_ICP    3130
80
81 struct hash_struct {
82   guint16 proto;
83   void (*dissect)(const u_char *, int, frame_data *, proto_tree *);
84   struct hash_struct *next;
85 };
86
87 static struct hash_struct *hash_table[256];
88
89 /*
90  * These routines are for UDP, will be generalized soon: RJS
91  *
92  * XXX - note that they should probably check the IP address as well as
93  * the port number, so that we don't mistakenly identify packets as, say,
94  * TFTP, merely because they have a source or destination port number
95  * equal to the port being used by a TFTP daemon on some machine other
96  * than the one they're going to or from.
97  */
98
99 struct hash_struct *udp_find_hash_ent(guint16 proto) {
100
101   int idx = proto % 256;
102   struct hash_struct *hash_ent = hash_table[idx];
103
104   while (hash_ent != NULL) {
105
106     if (hash_ent -> proto == proto)
107       return hash_ent;
108   
109     hash_ent = hash_ent -> next;
110
111   }
112
113   return NULL;
114
115 }
116
117 void udp_hash_add(guint16 proto,
118         void (*dissect)(const u_char *, int, frame_data *, proto_tree *)) {
119
120   int idx = proto % 256;   /* Simply take the remainder, hope for no collisions */
121   struct hash_struct *hash_ent = (struct hash_struct *)malloc(sizeof(struct hash_struct));
122   struct hash_struct *hash_ent2;
123   
124   hash_ent -> proto = proto;
125   hash_ent -> dissect = dissect;
126   hash_ent -> next = NULL;
127
128   if (hash_ent == NULL) {
129
130     fprintf(stderr, "Could not allocate space for hash structure in dissect_udp\n");
131     exit(1);
132   }
133
134   if (hash_table[idx]) {  /* Something, add on end */
135
136     hash_ent2 = hash_table[idx];
137
138     while (hash_ent2 -> next != NULL)
139       hash_ent2 = hash_ent2 -> next;
140
141     hash_ent2 -> next = hash_ent;     /* Bad in pathalogical cases */
142
143   }
144   else {
145
146     hash_table[idx] = hash_ent;
147
148   }
149
150 }
151
152 void init_dissect_udp(void) {
153
154   int i;
155
156   for (i = 0; i < 256; i++) {
157
158     hash_table[i] = NULL;
159
160   }
161
162   /* Now add the protocols we know about */
163
164   udp_hash_add(UDP_PORT_BOOTPS, dissect_bootp);
165   udp_hash_add(UDP_PORT_TFTP, dissect_tftp);
166
167 }
168
169 void
170 dissect_udp(const u_char *pd, int offset, frame_data *fd, proto_tree *tree) {
171   e_udphdr  uh;
172   guint16    uh_sport, uh_dport, uh_ulen, uh_sum;
173   struct hash_struct *dissect_routine = NULL;
174   proto_tree *udp_tree;
175   proto_item *ti;
176
177   /* To do: Check for {cap len,pkt len} < struct len */
178   /* Avoids alignment problems on many architectures. */
179   memcpy(&uh, &pd[offset], sizeof(e_udphdr));
180   uh_sport = ntohs(uh.uh_sport);
181   uh_dport = ntohs(uh.uh_dport);
182   uh_ulen  = ntohs(uh.uh_ulen);
183   uh_sum   = ntohs(uh.uh_sum);
184   
185   if (check_col(fd, COL_PROTOCOL))
186     col_add_str(fd, COL_PROTOCOL, "UDP");
187   if (check_col(fd, COL_INFO))
188     col_add_fstr(fd, COL_INFO, "Source port: %s  Destination port: %s",
189             get_udp_port(uh_sport), get_udp_port(uh_dport));
190   if (check_col(fd, COL_RES_SRC_PORT))
191     col_add_str(fd, COL_RES_SRC_PORT, get_udp_port(uh_sport));
192   if (check_col(fd, COL_UNRES_SRC_PORT))
193     col_add_fstr(fd, COL_UNRES_SRC_PORT, "%u", uh_sport);
194   if (check_col(fd, COL_RES_DST_PORT))
195     col_add_str(fd, COL_RES_DST_PORT, get_udp_port(uh_dport));
196   if (check_col(fd, COL_UNRES_DST_PORT))
197     col_add_fstr(fd, COL_UNRES_DST_PORT, "%u", uh_dport);
198     
199   if (tree) {
200     ti = proto_tree_add_item(tree, proto_udp, offset, 8);
201     udp_tree = proto_item_add_subtree(ti, ETT_UDP);
202
203     proto_tree_add_item_format(udp_tree, hf_udp_srcport, offset, 2, uh_sport,
204         "Source port: %s (%u)", get_udp_port(uh_sport), uh_sport);
205     proto_tree_add_item_format(udp_tree, hf_udp_dstport, offset + 2, 2, uh_dport,
206         "Destination port: %s (%u)", get_udp_port(uh_dport), uh_dport);
207
208     proto_tree_add_item_hidden(udp_tree, hf_udp_port, offset, 2, uh_sport);
209     proto_tree_add_item_hidden(udp_tree, hf_udp_port, offset+2, 2, uh_dport);
210
211     proto_tree_add_item(udp_tree, hf_udp_length, offset + 4, 2,  uh_ulen);
212     proto_tree_add_item_format(udp_tree, hf_udp_checksum, offset + 6, 2, uh_sum,
213         "Checksum: 0x%04x", uh_sum);
214   }
215
216   /* Skip over header */
217   offset += 8;
218
219   /* XXX - we should do all of this through the table of ports. */
220 #define PORT_IS(port)   (uh_sport == port || uh_dport == port)
221  if (PORT_IS(UDP_PORT_BOOTPS))
222       dissect_bootp(pd, offset, fd, tree);
223  else if (PORT_IS(UDP_PORT_DNS))
224       dissect_dns(pd, offset, fd, tree);
225  else if (PORT_IS(UDP_PORT_ISAKMP))
226       dissect_isakmp(pd, offset, fd, tree);
227  else if (PORT_IS(UDP_PORT_RIP)) {
228       /* we should check the source port too (RIP: UDP src and dst port 520) */
229       dissect_rip(pd, offset, fd, tree);
230  } else if (PORT_IS(UDP_PORT_NBNS))
231       dissect_nbns(pd, offset, fd, tree);
232  else if (PORT_IS(UDP_PORT_NBDGM))
233       dissect_nbdgm(pd, offset, fd, tree);
234  else if (PORT_IS(UDP_PORT_IPX)) /* RFC 1234 */
235       dissect_ipx(pd, offset, fd, tree);
236 #if defined(HAVE_UCD_SNMP_SNMP_H) || defined(HAVE_SNMP_SNMP_H)
237  else if (PORT_IS(UDP_PORT_SNMP))
238       dissect_snmp(pd, offset, fd, tree);
239 #endif
240  else if (PORT_IS(UDP_PORT_VINES)) {
241       /* FIXME: AFAIK, src and dst port must be the same */
242       dissect_vines_frp(pd, offset, fd, tree);
243  } else if (PORT_IS(UDP_PORT_TFTP)) {
244       /* This is the first point of call, but it adds a dynamic call */
245       udp_hash_add(MAX(uh_sport, uh_dport), dissect_tftp);  /* Add to table */
246       dissect_tftp(pd, offset, fd, tree);
247  } else if (PORT_IS(UDP_PORT_RADIUS) ||
248                 PORT_IS(UDP_PORT_RADACCT) ||
249                 PORT_IS(UDP_PORT_RADIUS_NEW) ||
250                 PORT_IS(UDP_PORT_RADACCT_NEW) ) {
251       dissect_radius(pd, offset, fd, tree);
252  } else if ( PORT_IS(UDP_PORT_ICP)) {
253         dissect_icp(pd,offset,fd,tree);
254  } else {
255       /* OK, find a routine in the table, else use the default */
256
257       if ((dissect_routine = udp_find_hash_ent(uh_sport))) {
258
259         struct hash_struct *dr2 = udp_find_hash_ent(uh_dport);
260
261         if (dr2 == NULL) {  /* Not in the table, add */
262
263           udp_hash_add(uh_dport, dissect_tftp);
264
265         }
266
267         dissect_routine -> dissect(pd, offset, fd, tree);
268       }
269       else if ((dissect_routine = udp_find_hash_ent(uh_dport))) {
270
271         dissect_routine -> dissect(pd, offset, fd, tree);
272
273       }
274       else {
275
276         dissect_data(pd, offset, fd, tree);
277       }
278   }
279 }
280
281 void
282 proto_register_udp(void)
283 {
284         static hf_register_info hf[] = {
285                 { &hf_udp_srcport,
286                 { "Source Port",        "udp.srcport", FT_UINT16, NULL }},
287
288                 { &hf_udp_dstport,
289                 { "Destination Port",   "udp.dstport", FT_UINT16, NULL }},
290
291                 { &hf_udp_port,
292                 { "Source or Destination Port", "udp.port", FT_UINT16, NULL }},
293
294                 { &hf_udp_length,
295                 { "Length",             "udp.length", FT_UINT16, NULL }},
296
297                 { &hf_udp_checksum,
298                 { "Checksum",           "udp.checksum", FT_UINT16, NULL }}
299         };
300
301         proto_udp = proto_register_protocol("User Datagram Protocol", "udp");
302         proto_register_field_array(proto_udp, hf, array_length(hf));
303 }
304