first steps to support MSVC 2008 Express Edition
[obnox/wireshark/wip.git] / capture-pcap-util.c
1 /* capture-pcap-util.c
2  * Utility routines for packet capture
3  *
4  * $Id$
5  *
6  * Wireshark - Network traffic analyzer
7  * By Gerald Combs <gerald@wireshark.org>
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 #ifdef HAVE_CONFIG_H
26 # include "config.h"
27 #endif
28
29 #ifdef HAVE_LIBPCAP
30
31 #include <glib.h>
32
33 #include <stdlib.h>
34 #include <stdio.h>
35 #include <limits.h>
36 #include <string.h>
37
38 #include <pcap.h>
39
40 #ifdef HAVE_SYS_TYPES_H
41 # include <sys/types.h>
42 #endif
43
44 #ifdef HAVE_SYS_SOCKET_H
45 #include <sys/socket.h>
46 #endif
47
48 #include <wtap.h>
49 #include <libpcap.h>
50
51 #include "capture-pcap-util.h"
52 #include "capture-pcap-util-int.h"
53
54 #ifndef _WIN32
55 #include <netinet/in.h>
56 #endif
57
58
59 /*
60  * Get the data-link type for a libpcap device.
61  * This works around AIX 5.x's non-standard and incompatible-with-the-
62  * rest-of-the-universe libpcap.
63  */
64 int
65 get_pcap_linktype(pcap_t *pch, const char *devname
66 #ifndef _AIX
67         _U_
68 #endif
69 )
70 {
71         int linktype;
72 #ifdef _AIX
73         const char *ifacename;
74 #endif
75
76         linktype = pcap_datalink(pch);
77 #ifdef _AIX
78
79         /*
80          * The libpcap that comes with AIX 5.x uses RFC 1573 ifType values
81          * rather than DLT_ values for link-layer types; the ifType values
82          * for LAN devices are:
83          *
84          *      Ethernet        6
85          *      802.3           7
86          *      Token Ring      9
87          *      FDDI            15
88          *
89          * and the ifType value for a loopback device is 24.
90          *
91          * The AIX names for LAN devices begin with:
92          *
93          *      Ethernet                en
94          *      802.3                   et
95          *      Token Ring              tr
96          *      FDDI                    fi
97          *
98          * and the AIX names for loopback devices begin with "lo".
99          *
100          * (The difference between "Ethernet" and "802.3" is presumably
101          * whether packets have an Ethernet header, with a packet type,
102          * or an 802.3 header, with a packet length, followed by an 802.2
103          * header and possibly a SNAP header.)
104          *
105          * If the device name matches "linktype" interpreted as an ifType
106          * value, rather than as a DLT_ value, we will assume this is AIX's
107          * non-standard, incompatible libpcap, rather than a standard libpcap,
108          * and will map the link-layer type to the standard DLT_ value for
109          * that link-layer type, as that's what the rest of Wireshark expects.
110          *
111          * (This means the capture files won't be readable by a tcpdump
112          * linked with AIX's non-standard libpcap, but so it goes.  They
113          * *will* be readable by standard versions of tcpdump, Wireshark,
114          * and so on.)
115          *
116          * XXX - if we conclude we're using AIX libpcap, should we also
117          * set a flag to cause us to assume the time stamps are in
118          * seconds-and-nanoseconds form, and to convert them to
119          * seconds-and-microseconds form before processing them and
120          * writing them out?
121          */
122
123         /*
124          * Find the last component of the device name, which is the
125          * interface name.
126          */
127         ifacename = strchr(devname, '/');
128         if (ifacename == NULL)
129                 ifacename = devname;
130
131         /* See if it matches any of the LAN device names. */
132         if (strncmp(ifacename, "en", 2) == 0) {
133                 if (linktype == 6) {
134                         /*
135                          * That's the RFC 1573 value for Ethernet; map it
136                          * to DLT_EN10MB.
137                          */
138                         linktype = 1;
139                 }
140         } else if (strncmp(ifacename, "et", 2) == 0) {
141                 if (linktype == 7) {
142                         /*
143                          * That's the RFC 1573 value for 802.3; map it to
144                          * DLT_EN10MB.
145                          * (libpcap, tcpdump, Wireshark, etc. don't care if
146                          * it's Ethernet or 802.3.)
147                          */
148                         linktype = 1;
149                 }
150         } else if (strncmp(ifacename, "tr", 2) == 0) {
151                 if (linktype == 9) {
152                         /*
153                          * That's the RFC 1573 value for 802.5 (Token Ring);
154                          * map it to DLT_IEEE802, which is what's used for
155                          * Token Ring.
156                          */
157                         linktype = 6;
158                 }
159         } else if (strncmp(ifacename, "fi", 2) == 0) {
160                 if (linktype == 15) {
161                         /*
162                          * That's the RFC 1573 value for FDDI; map it to
163                          * DLT_FDDI.
164                          */
165                         linktype = 10;
166                 }
167         } else if (strncmp(ifacename, "lo", 2) == 0) {
168                 if (linktype == 24) {
169                         /*
170                          * That's the RFC 1573 value for "software loopback"
171                          * devices; map it to DLT_NULL, which is what's used
172                          * for loopback devices on BSD.
173                          */
174                         linktype = 0;
175                 }
176         }
177 #endif
178
179         return linktype;
180 }
181
182 if_info_t *
183 if_info_new(char *name, char *description)
184 {
185         if_info_t *if_info;
186
187         if_info = g_malloc(sizeof (if_info_t));
188         if_info->name = g_strdup(name);
189         if (description == NULL)
190                 if_info->description = NULL;
191         else
192                 if_info->description = g_strdup(description);
193         if_info->ip_addr = NULL;
194         if_info->loopback = FALSE;
195         return if_info;
196 }
197
198 void
199 if_info_add_address(if_info_t *if_info, struct sockaddr *addr)
200 {
201         if_addr_t *ip_addr;
202         struct sockaddr_in *ai;
203 #ifdef INET6
204         struct sockaddr_in6 *ai6;
205 #endif
206
207         switch (addr->sa_family) {
208
209         case AF_INET:
210                 ai = (struct sockaddr_in *)addr;
211                 ip_addr = g_malloc(sizeof(*ip_addr));
212                 ip_addr->type = AT_IPv4;
213                 ip_addr->ip_addr.ip4_addr =
214                     *((guint32 *)&(ai->sin_addr.s_addr));
215                 if_info->ip_addr = g_slist_append(if_info->ip_addr, ip_addr);
216                 break;
217
218 #ifdef INET6
219         case AF_INET6:
220                 ai6 = (struct sockaddr_in6 *)addr;
221                 ip_addr = g_malloc(sizeof(*ip_addr));
222                 ip_addr->type = AT_IPv6;
223                 memcpy((void *)&ip_addr->ip_addr.ip6_addr,
224                     (void *)&ai6->sin6_addr.s6_addr,
225                     sizeof ip_addr->ip_addr.ip6_addr);
226                 if_info->ip_addr = g_slist_append(if_info->ip_addr, ip_addr);
227                 break;
228 #endif
229         }
230 }
231
232 #ifdef HAVE_PCAP_FINDALLDEVS
233 /*
234  * Get all IP address information, and the loopback flag, for the given
235  * interface.
236  */
237 static void
238 if_info_ip(if_info_t *if_info, pcap_if_t *d)
239 {
240         pcap_addr_t *a;
241
242         /* Loopback flag */
243         if_info->loopback = (d->flags & PCAP_IF_LOOPBACK) ? TRUE : FALSE;
244
245         /* All addresses */
246         for (a = d->addresses; a != NULL; a = a->next) {
247                 if (a->addr != NULL)
248                         if_info_add_address(if_info, a->addr);
249         }
250 }
251
252 GList *
253 get_interface_list_findalldevs(int *err, char **err_str)
254 {
255         GList  *il = NULL;
256         pcap_if_t *alldevs, *dev;
257         if_info_t *if_info;
258         char errbuf[PCAP_ERRBUF_SIZE];
259
260         if (pcap_findalldevs(&alldevs, errbuf) == -1) {
261                 *err = CANT_GET_INTERFACE_LIST;
262                 if (err_str != NULL)
263                         *err_str = cant_get_if_list_error_message(errbuf);
264                 return NULL;
265         }
266
267         if (alldevs == NULL) {
268                 /*
269                  * No interfaces found.
270                  */
271                 *err = NO_INTERFACES_FOUND;
272                 if (err_str != NULL)
273                         *err_str = NULL;
274                 return NULL;
275         }
276
277         for (dev = alldevs; dev != NULL; dev = dev->next) {
278                 if_info = if_info_new(dev->name, dev->description);
279                 il = g_list_append(il, if_info);
280                 if_info_ip(if_info, dev);
281         }
282         pcap_freealldevs(alldevs);
283
284         return il;
285 }
286 #endif /* HAVE_PCAP_FINDALLDEVS */
287
288 static void
289 free_if_info_addr_cb(gpointer addr, gpointer user_data _U_)
290 {
291         g_free(addr);
292 }
293
294 static void
295 free_if_cb(gpointer data, gpointer user_data _U_)
296 {
297         if_info_t *if_info = data;
298
299         g_free(if_info->name);
300         if (if_info->description != NULL)
301                 g_free(if_info->description);
302
303         g_slist_foreach(if_info->ip_addr, free_if_info_addr_cb, NULL);
304         g_slist_free(if_info->ip_addr);
305         g_free(if_info);
306 }
307
308 void
309 free_interface_list(GList *if_list)
310 {
311         g_list_foreach(if_list, free_if_cb, NULL);
312         g_list_free(if_list);
313 }
314
315 /*
316  * Get the data-link types available for a libpcap device.
317  */
318 static data_link_info_t *
319 create_data_link_info(int dlt)
320 {
321         data_link_info_t *data_link_info;
322 #ifdef HAVE_PCAP_DATALINK_VAL_TO_NAME
323         const char *typename;
324 #endif
325         int wtap_encap;
326
327         data_link_info = g_malloc(sizeof (data_link_info_t));
328         data_link_info->dlt = dlt;
329 #ifdef HAVE_PCAP_DATALINK_VAL_TO_NAME
330         typename = pcap_datalink_val_to_name(dlt);
331         if (typename != NULL)
332                 data_link_info->name = g_strdup(typename);
333         else
334 #endif
335                 data_link_info->name = g_strdup_printf("DLT %d", dlt);
336         wtap_encap = wtap_pcap_encap_to_wtap_encap(dlt);
337         if (wtap_encap == WTAP_ENCAP_UNKNOWN) {
338                 /*
339                  * We don't support this in Wiretap.
340                  * However, we should, so you can capture on it.
341                  * Put in an entry for it, with no description.
342                  */
343                 data_link_info->description = NULL;
344         } else {
345                 /*
346                  * If this is null, that's a bug in
347                  * "wtap_pcap_encap_to_wtap_encap()" - it should always
348                  * return a valid encapsulation type - so we assume it's
349                  * not null.
350                  */
351                 data_link_info->description =
352                     g_strdup(wtap_encap_string(wtap_encap));
353         }
354         return data_link_info;
355 }
356
357 GList *
358 get_pcap_linktype_list(char *devname, char **err_str)
359 {
360         GList *linktype_list = NULL;
361         pcap_t *pch;
362         int deflt;
363         char errbuf[PCAP_ERRBUF_SIZE];
364 #ifdef HAVE_PCAP_SET_DATALINK
365         int *linktypes;
366         int i, nlt;
367 #endif
368         data_link_info_t *data_link_info;
369
370         pch = pcap_open_live(devname, MIN_PACKET_SIZE, 0, 0, errbuf);
371         if (pch == NULL) {
372                 if (err_str != NULL)
373                         *err_str = g_strdup(errbuf);
374                 return NULL;
375         }
376         deflt = get_pcap_linktype(pch, devname);
377 #ifdef HAVE_PCAP_LIST_DATALINKS
378         nlt = pcap_list_datalinks(pch, &linktypes);
379         if (nlt == 0 || linktypes == NULL) {
380                 pcap_close(pch);
381                 if (err_str != NULL)
382                         *err_str = NULL;        /* an empty list doesn't mean an error */
383                 return NULL;
384         }
385         for (i = 0; i < nlt; i++) {
386                 data_link_info = create_data_link_info(linktypes[i]);
387
388                 /*
389                  * XXX - for 802.11, make the most detailed 802.11
390                  * version the default, rather than the one the
391                  * device has as the default?
392                  */
393                 if (linktypes[i] == deflt)
394                         linktype_list = g_list_prepend(linktype_list,
395                             data_link_info);
396                 else
397                         linktype_list = g_list_append(linktype_list,
398                             data_link_info);
399         }
400         free(linktypes);
401 #else
402         data_link_info = create_data_link_info(deflt);
403         linktype_list = g_list_append(linktype_list, data_link_info);
404 #endif
405
406         pcap_close(pch);
407         return linktype_list;
408 }
409
410 static void
411 free_linktype_cb(gpointer data, gpointer user_data _U_)
412 {
413         data_link_info_t *linktype_info = data;
414
415         g_free(linktype_info->name);
416         if (linktype_info->description != NULL)
417                 g_free(linktype_info->description);
418 }
419
420 void
421 free_pcap_linktype_list(GList *linktype_list)
422 {
423         g_list_foreach(linktype_list, free_linktype_cb, NULL);
424         g_list_free(linktype_list);
425 }
426
427 /* Set the data link type on a pcap. */
428 const char *
429 set_pcap_linktype(pcap_t *pch, char *devname
430 #ifdef HAVE_PCAP_SET_DATALINK
431         _U_
432 #endif
433         , int dlt)
434 {
435 #ifdef HAVE_PCAP_SET_DATALINK
436         if (pcap_set_datalink(pch, dlt) == 0)
437                 return NULL;    /* no error */
438         return pcap_geterr(pch);
439 #else
440         /* Let them set it to the type it is; reject any other request. */
441         if (get_pcap_linktype(pch, devname) == dlt)
442                 return NULL;    /* no error */
443         return "That DLT isn't one of the DLTs supported by this device";
444 #endif
445 }
446
447
448 #ifdef HAVE_PCAP_DATALINK_VAL_TO_NAME
449 const char *
450 linktype_val_to_name(int dlt)
451 {
452     return pcap_datalink_val_to_name(dlt);
453 }
454 #endif
455
456 #ifdef HAVE_PCAP_DATALINK_NAME_TO_VAL
457 int linktype_name_to_val(const char *linktype)
458 {
459     return pcap_datalink_name_to_val(linktype);
460 }
461 #endif
462
463 #endif /* HAVE_LIBPCAP */