From Didier Gautheron:
[metze/wireshark/wip.git] / epan / dissectors / packet-wol.c
1 /* packet-wol.c
2  * Routines for WOL dissection
3  * Copyright 2007, Christopher Maynard <Chris.Maynard[AT]gtech.com>
4  *
5  * $Id$
6  *
7  * Wireshark - Network traffic analyzer
8  * By Gerald Combs <gerald@wireshark.org>
9  * Copyright 1998 Gerald Combs
10  *
11  * This dissector for "Wake On LAN" was not copied from any other existing
12  * dissector.  It uses the template from SVN23520 docs/README.devloper, which
13  * was the latest one available at the time of this writing.  This dissector is
14  * a heuristic one though, so appropriate changes have made to the template
15  * as needed.
16  *
17  * The "Wake On LAN" dissector was written based primarily on the AMD white
18  * paper, available from: http://www.amd.com/us-en/assets/content_type/white_papers_and_tech_docs/20213.pdf.
19  *
20  * In addition, testing of the dissector was conducted using 2 utilities
21  * downloaded from http://www.moldaner.de/wakeonlan/wakeonlan.html and
22  * http://www.depicus.com/wake-on-lan/, as well as with the ether-wake utility
23  * on a Linux Fedora Core 4 system.
24  *
25  * From what I can tell from the tools available, even though the white paper
26  * indicates that the so-called, "MagicPacket" can be located anywhere within
27  * the Ethernet frame, in practice, there seem to be only 2 variations of the
28  * implementation of the MagicPacket.  Ether-wake implements it as an Ethernet
29  * frame with ether type 0x0842 (ETHERTYPE_WOL), and the other tools all seem
30  * to implement it as a UDP packet, both with the payload as nothing but the
31  * MagicPacket.
32  *
33  * To keep things simple, this dissector will only indicate a frame as
34  * Wake-On-Lan if the MagicPacket is found for a frame marked as etherytpe
35  * 0x0842 or if it's a UDP packet.  To fully support Wake-On-Lan dissection
36  * though, we would need a way to have this dissector called only if the frame
37  * hasn't already been classified as some other type of dissector ... but I
38  * don't know how to do that?  The only alternative I am aware of would be to
39  * register as a heuristic dissector for pretty much every possible protocol
40  * there is, which seems unreasonable to do to me.
41  *
42  * This program is free software; you can redistribute it and/or
43  * modify it under the terms of the GNU General Public License
44  * as published by the Free Software Foundation; either version 2
45  * of the License, or (at your option) any later version.
46  *
47  * This program is distributed in the hope that it will be useful,
48  * but WITHOUT ANY WARRANTY; without even the implied warranty of
49  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
50  * GNU General Public License for more details.
51  *
52  * You should have received a copy of the GNU General Public License
53  * along with this program; if not, write to the Free Software
54  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
55  */
56
57 #ifdef HAVE_CONFIG_H
58 # include "config.h"
59 #endif
60
61 #include <stdio.h>
62 #include <stdlib.h>
63 #include <string.h>
64
65 #include <glib.h>
66
67 #include <epan/addr_resolv.h>
68 #include <epan/packet.h>
69 #include <epan/etypes.h>
70
71 /* IF PROTO exposes code to other dissectors, then it must be exported
72    in a header file. If not, a header file is not needed at all. */
73 /* #include "packet-wol.h" */
74
75 /* Initialize the protocol and registered fields */
76 static int proto_wol = -1;
77 static int hf_wol_sync = -1;
78 static int hf_wol_mac = -1;
79 static int hf_wol_passwd = -1;
80
81 /* Global sample preference ("controls" display of numbers) */
82 /* static gboolean gPREF_HEX = FALSE; */
83
84 /* Initialize the subtree pointers */
85 static gint ett_wol = -1;
86 static gint ett_wol_macblock = -1;
87
88 /* Code to actually dissect the packets */
89 static int
90 dissect_wol(tvbuff_t *tvb, packet_info *pinfo, proto_tree *tree)
91 {
92     guint len;
93     gint offset;
94     guint8 sync[6] = {0xff, 0xff, 0xff, 0xff, 0xff, 0xff};
95     guint8 *mac;
96     const guint8 *passwd;
97
98 /* Set up structures needed to add the protocol subtree and manage it */
99     proto_item *ti;
100     proto_item *mti;
101     proto_tree *wol_tree;
102     proto_tree *mac_tree;
103
104 /*  First, if at all possible, do some heuristics to check if the packet cannot
105  *  possibly belong to your protocol.  This is especially important for
106  *  protocols directly on top of TCP or UDP where port collisions are
107  *  common place (e.g., even though your protocol uses a well known port,
108  *  someone else may set up, for example, a web server on that port which,
109  *  if someone analyzed that web server's traffic in Wireshark, would result
110  *  in Wireshark handing an HTTP packet to your dissector).  For example:
111  */
112     /* Check that there's enough data */
113     len = tvb_length(tvb);
114     if ( len < 102 )    /* wol's smallest packet size is 102 */
115         return (0);
116
117     /* Get some values from the packet header, probably using tvb_get_*() */
118
119     /* Regardless of what the AMD white paper states, don't search the entire
120      * tvb for the synchronization stream.  My feeling is that this could be
121      * quite expensive and seriously hinder Wireshark performance.  For now,
122      * unless we need to change it later, just compare the 1st 6 bytes. */
123     if ( tvb_memeql(tvb, 0, sync, 6) != 0 )
124         return (0);
125
126     /* So far so good.  Now get the next 6 bytes, which we'll assume is the
127      * target's MAC address, and do 15 memory chunk comparisons, since if this
128      * is a real MagicPacket, the target's MAC will be duplicated 16 times. */
129     mac = ep_tvb_memdup(tvb, 6, 6);
130     for ( offset = 12; offset < 102; offset += 6 )
131         if ( tvb_memeql(tvb, offset, mac, 6) != 0 )
132             return (0);
133
134     /* OK, we're going to assume it's a MagicPacket.  If there's a password,
135      * grab it now, and in case there's any extra bytes after the only 3 valid
136      * and expected lengths, truncate the length so the extra byte(s) aren't
137      * included as being part of the WOL payload. */
138     if ( len >= 106 && len < 108 )
139     {
140         len = 106;
141         passwd = ip_to_str(ep_tvb_memdup(tvb, 102, 4));
142     }
143     else if ( len >= 108 )
144     {
145         len = 108;
146         passwd = ether_to_str(ep_tvb_memdup(tvb, 102, 6));
147     }
148     else
149     {
150         len = 102;
151         passwd = NULL;
152     }
153
154 /* Make entries in Protocol column and Info column on summary display */
155     col_set_str(pinfo->cinfo, COL_PROTOCOL, "WOL");
156
157 /* This field shows up as the "Info" column in the display; you should use
158    it, if possible, to summarize what's in the packet, so that a user looking
159    at the list of packets can tell what type of packet it is. See section 1.5
160    for more information.
161
162    Before changing the contents of a column you should make sure the column is
163    active by calling "check_col(pinfo->cinfo, COL_*)". If it is not active
164    don't bother setting it.
165
166    If you are setting the column to a constant string, use "col_set_str()",
167    as it's more efficient than the other "col_set_XXX()" calls.
168
169    If you're setting it to a string you've constructed, or will be
170    appending to the column later, use "col_add_str()".
171
172    "col_add_fstr()" can be used instead of "col_add_str()"; it takes
173    "printf()"-like arguments.  Don't use "col_add_fstr()" with a format
174    string of "%s" - just use "col_add_str()" or "col_set_str()", as it's
175    more efficient than "col_add_fstr()".
176
177    If you will be fetching any data from the packet before filling in
178    the Info column, clear that column first, in case the calls to fetch
179    data from the packet throw an exception because they're fetching data
180    past the end of the packet, so that the Info column doesn't have data
181    left over from the previous dissector; do
182
183     col_clear(pinfo->cinfo, COL_INFO);
184
185    */
186
187     if ( check_col(pinfo->cinfo, COL_INFO) )
188     {
189         col_add_fstr(pinfo->cinfo, COL_INFO, "MagicPacket for %s (%s)",
190             get_ether_name(mac), ether_to_str(mac));
191
192         /* NOTE: ether-wake uses a dotted-decimal format for specifying a
193          * 4-byte password or an Ethernet mac address format for specifying
194          * a 6-byte password, so display them in that format, even if the
195          * password isn't really an IP or MAC address. */
196         if ( passwd )
197             col_append_fstr(pinfo->cinfo, COL_INFO, ", password %s", passwd);
198     }
199
200 /* A protocol dissector can be called in 2 different ways:
201
202     (a) Operational dissection
203
204         In this mode, Wireshark is only interested in the way protocols
205         interact, protocol conversations are created, packets are
206         reassembled and handed over to higher-level protocol dissectors.
207         In this mode Wireshark does not build a so-called "protocol
208         tree".
209
210     (b) Detailed dissection
211
212         In this mode, Wireshark is also interested in all details of
213         a given protocol, so a "protocol tree" is created.
214
215    Wireshark distinguishes between the 2 modes with the proto_tree pointer:
216     (a) <=> tree == NULL
217     (b) <=> tree != NULL
218
219    In the interest of speed, if "tree" is NULL, avoid building a
220    protocol tree and adding stuff to it, or even looking at any packet
221    data needed only if you're building the protocol tree, if possible.
222
223    Note, however, that you must fill in column information, create
224    conversations, reassemble packets, build any other persistent state
225    needed for dissection, and call subdissectors regardless of whether
226    "tree" is NULL or not.  This might be inconvenient to do without
227    doing most of the dissection work; the routines for adding items to
228    the protocol tree can be passed a null protocol tree pointer, in
229    which case they'll return a null item pointer, and
230    "proto_item_add_subtree()" returns a null tree pointer if passed a
231    null item pointer, so, if you're careful not to dereference any null
232    tree or item pointers, you can accomplish this by doing all the
233    dissection work.  This might not be as efficient as skipping that
234    work if you're not building a protocol tree, but if the code would
235    have a lot of tests whether "tree" is null if you skipped that work,
236    you might still be better off just doing all that work regardless of
237    whether "tree" is null or not. */
238     if (tree) {
239
240 /* NOTE: The offset and length values in the call to
241    "proto_tree_add_item()" define what data bytes to highlight in the hex
242    display window when the line in the protocol tree display
243    corresponding to that item is selected.
244
245    Supplying a length of -1 is the way to highlight all data from the
246    offset to the end of the packet. */
247
248 /* create display subtree for the protocol */
249         ti = proto_tree_add_item(tree, proto_wol, tvb, 0, len, FALSE);
250         proto_item_append_text(ti, ", MAC: %s (%s)", get_ether_name(mac),
251             ether_to_str(mac));
252         if ( passwd )
253             proto_item_append_text(ti, ", password: %s", passwd);
254         wol_tree = proto_item_add_subtree(ti, ett_wol);
255
256 /* add an item to the subtree, see section 1.6 for more information */
257         proto_tree_add_item(wol_tree, hf_wol_sync, tvb, 0, 6, FALSE);
258
259 /* Continue adding tree items to process the packet here */
260         mti = proto_tree_add_text(wol_tree, tvb, 6, 96, "MAC: %s (%s)",
261             get_ether_name(mac), ether_to_str(mac));
262         mac_tree = proto_item_add_subtree(mti, ett_wol_macblock);
263         for ( offset = 6; offset < 102; offset += 6 )
264             proto_tree_add_ether(mac_tree, hf_wol_mac, tvb, offset, 6, mac);
265
266         if ( len == 106 )
267             proto_tree_add_bytes_format(wol_tree, hf_wol_passwd, tvb, offset,
268                 4, passwd, "Password: %s", passwd);
269         else if ( len == 108 )
270             proto_tree_add_bytes_format(wol_tree, hf_wol_passwd, tvb, offset,
271                 6, passwd, "Password: %s", passwd);
272     }
273
274 /* If this protocol has a sub-dissector call it here, see section 1.8 */
275
276 /* Return the amount of data this dissector was able to dissect */
277     if ( pinfo->ethertype == ETHERTYPE_WOL )
278         return (len);
279
280     /* Heuristic dissectors return TRUE/FALSE. */
281     return (TRUE);
282 }
283
284
285 /* Register the protocol with Wireshark */
286
287 /* this format is require because a script is used to build the C function
288    that calls all the protocol registration.
289 */
290
291 void
292 proto_register_wol(void)
293 {
294 /* Setup list of header fields  See Section 1.6.1 for details*/
295     static hf_register_info hf[] = {
296         { &hf_wol_sync,
297             { "Sync stream", "wol.sync",
298             FT_BYTES, BASE_NONE, NULL, 0, NULL, HFILL }},
299         { &hf_wol_mac,
300             { "MAC", "wol.mac",
301             FT_ETHER, BASE_NONE, NULL, 0, NULL, HFILL }},
302         { &hf_wol_passwd,
303             { "Password", "wol.passwd",
304             FT_BYTES, BASE_NONE, NULL, 0, NULL, HFILL }}
305     };
306
307 /* Setup protocol subtree array */
308     static gint *ett[] = {
309         &ett_wol,
310         &ett_wol_macblock
311     };
312
313 /* Register the protocol name and description */
314     proto_wol = proto_register_protocol("Wake On LAN", "WOL", "wol");
315
316 /* Required function calls to register the header fields and subtrees used */
317     proto_register_field_array(proto_wol, hf, array_length(hf));
318     proto_register_subtree_array(ett, array_length(ett));
319 }
320
321
322 /* If this dissector uses sub-dissector registration add a registration routine.
323    This exact format is required because a script is used to find these
324    routines and create the code that calls these routines.
325
326 */
327 void
328 proto_reg_handoff_wol(void)
329 {
330     dissector_handle_t wol_handle;
331
332 /*  Use new_create_dissector_handle() to indicate that dissect_wol()
333  *  returns the number of bytes it dissected (or 0 if it thinks the packet
334  *  does not belong to PROTONAME).
335  */
336     wol_handle = new_create_dissector_handle(dissect_wol, proto_wol);
337
338     /* We don't really want to register with EVERY possible dissector,
339      * do we?  I know that the AMD white paper specifies that the
340      * MagicPacket could be present in any frame, but are we seriously
341      * going to register WOL with every other dissector!?  I think not.
342      *
343      * Unless anyone has a better idea, just register with only those that
344      * are in "common usage" and grow this list as needed.  Yeah, I'm sure
345      * we'll miss some, but how else to do this ... add a thousand of
346      * these dissector_add()'s and heur_dissector_add()'s??? */
347     dissector_add("ethertype", ETHERTYPE_WOL, wol_handle);
348     heur_dissector_add("udp", dissect_wol, proto_wol);
349 }
350