]> CyberLeo.Net >> Repos - FreeBSD/releng/9.1.git/blob - contrib/libpcap/pcap-usb-linux.c
MFC r238935,238960:
[FreeBSD/releng/9.1.git] / contrib / libpcap / pcap-usb-linux.c
1 /*
2  * Copyright (c) 2006 Paolo Abeni (Italy)
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  *
9  * 1. Redistributions of source code must retain the above copyright
10  * notice, this list of conditions and the following disclaimer.
11  * 2. Redistributions in binary form must reproduce the above copyright
12  * notice, this list of conditions and the following disclaimer in the
13  * documentation and/or other materials provided with the distribution.
14  * 3. The name of the author may not be used to endorse or promote 
15  * products derived from this software without specific prior written 
16  * permission.
17  *
18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
28  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29  *
30  * USB sniffing API implementation for Linux platform
31  * By Paolo Abeni <paolo.abeni@email.it>
32  * Modifications: Kris Katterjohn <katterjohn@gmail.com>
33  *
34  */
35 #ifndef lint
36 static const char rcsid[] _U_ =
37     "@(#) $Header: /tcpdump/master/libpcap/pcap-usb-linux.c,v 1.33 2008-12-23 21:38:50 guy Exp $ (LBL)";
38 #endif
39  
40 #ifdef HAVE_CONFIG_H
41 #include "config.h"
42 #endif
43
44 #include "pcap-int.h"
45 #include "pcap-usb-linux.h"
46 #include "pcap/usb.h"
47
48 #ifdef NEED_STRERROR_H
49 #include "strerror.h"
50 #endif
51
52 #include <ctype.h>
53 #include <errno.h>
54 #include <stdlib.h>
55 #include <unistd.h>
56 #include <fcntl.h>
57 #include <string.h>
58 #include <dirent.h>
59 #include <byteswap.h>
60 #include <netinet/in.h>
61 #include <sys/ioctl.h>
62 #include <sys/mman.h>
63 #ifdef HAVE_LINUX_USBDEVICE_FS_H
64 /*
65  * We might need <linux/compiler.h> to define __user for
66  * <linux/usbdevice_fs.h>.
67  */
68 #ifdef HAVE_LINUX_COMPILER_H
69 #include <linux/compiler.h>
70 #endif /* HAVE_LINUX_COMPILER_H */
71 #include <linux/usbdevice_fs.h>
72 #endif /* HAVE_LINUX_USBDEVICE_FS_H */
73
74 #define USB_IFACE "usbmon"
75 #define USB_TEXT_DIR_OLD "/sys/kernel/debug/usbmon"
76 #define USB_TEXT_DIR "/sys/kernel/debug/usb/usbmon"
77 #define SYS_USB_BUS_DIR "/sys/bus/usb/devices"
78 #define PROC_USB_BUS_DIR "/proc/bus/usb"
79 #define USB_LINE_LEN 4096
80
81 #if __BYTE_ORDER == __LITTLE_ENDIAN
82 #define htols(s) s
83 #define htoll(l) l
84 #define htol64(ll) ll
85 #else
86 #define htols(s) bswap_16(s)
87 #define htoll(l) bswap_32(l)
88 #define htol64(ll) bswap_64(ll)
89 #endif
90
91 struct mon_bin_stats {
92         u_int32_t queued;
93         u_int32_t dropped;
94 };
95
96 struct mon_bin_get {
97         pcap_usb_header *hdr;
98         void *data;
99         size_t data_len;   /* Length of data (can be zero) */
100 };
101
102 struct mon_bin_mfetch {
103         int32_t *offvec;   /* Vector of events fetched */
104         int32_t nfetch;    /* Number of events to fetch (out: fetched) */
105         int32_t nflush;    /* Number of events to flush */
106 };
107
108 #define MON_IOC_MAGIC 0x92
109
110 #define MON_IOCQ_URB_LEN _IO(MON_IOC_MAGIC, 1)
111 #define MON_IOCX_URB  _IOWR(MON_IOC_MAGIC, 2, struct mon_bin_hdr)
112 #define MON_IOCG_STATS _IOR(MON_IOC_MAGIC, 3, struct mon_bin_stats)
113 #define MON_IOCT_RING_SIZE _IO(MON_IOC_MAGIC, 4)
114 #define MON_IOCQ_RING_SIZE _IO(MON_IOC_MAGIC, 5)
115 #define MON_IOCX_GET   _IOW(MON_IOC_MAGIC, 6, struct mon_bin_get)
116 #define MON_IOCX_MFETCH _IOWR(MON_IOC_MAGIC, 7, struct mon_bin_mfetch)
117 #define MON_IOCH_MFLUSH _IO(MON_IOC_MAGIC, 8)
118
119 #define MON_BIN_SETUP   0x1 /* setup hdr is present*/
120 #define MON_BIN_SETUP_ZERO      0x2 /* setup buffer is not available */
121 #define MON_BIN_DATA_ZERO       0x4 /* data buffer is not available */
122 #define MON_BIN_ERROR   0x8
123
124 /* forward declaration */
125 static int usb_activate(pcap_t *);
126 static int usb_stats_linux(pcap_t *, struct pcap_stat *);
127 static int usb_stats_linux_bin(pcap_t *, struct pcap_stat *);
128 static int usb_read_linux(pcap_t *, int , pcap_handler , u_char *);
129 static int usb_read_linux_bin(pcap_t *, int , pcap_handler , u_char *);
130 static int usb_read_linux_mmap(pcap_t *, int , pcap_handler , u_char *);
131 static int usb_inject_linux(pcap_t *, const void *, size_t);
132 static int usb_setdirection_linux(pcap_t *, pcap_direction_t);
133 static void usb_cleanup_linux_mmap(pcap_t *);
134
135 /* facility to add an USB device to the device list*/
136 static int 
137 usb_dev_add(pcap_if_t** alldevsp, int n, char *err_str)
138 {
139         char dev_name[10];
140         char dev_descr[30]; 
141         snprintf(dev_name, 10, USB_IFACE"%d", n);
142         snprintf(dev_descr, 30, "USB bus number %d", n);
143
144         if (pcap_add_if(alldevsp, dev_name, 0, 
145             dev_descr, err_str) < 0)
146                 return -1;
147         return 0; 
148 }
149
150 int 
151 usb_platform_finddevs(pcap_if_t **alldevsp, char *err_str)
152 {
153         struct dirent* data;
154         int ret = 0;
155         DIR* dir;
156         int n;
157         char* name;
158         size_t len;
159
160         /* try scanning sysfs usb bus directory */
161         dir = opendir(SYS_USB_BUS_DIR);
162         if (dir != NULL) {
163                 while ((ret == 0) && ((data = readdir(dir)) != 0)) {
164                         name = data->d_name;
165
166                         if (strncmp(name, "usb", 3) != 0)
167                                 continue;
168
169                         if (sscanf(&name[3], "%d", &n) == 0) 
170                                 continue;
171
172                         ret = usb_dev_add(alldevsp, n, err_str);
173                 }
174
175                 closedir(dir);
176                 return ret;
177         }
178
179         /* that didn't work; try scanning procfs usb bus directory */
180         dir = opendir(PROC_USB_BUS_DIR);
181         if (dir != NULL) {
182                 while ((ret == 0) && ((data = readdir(dir)) != 0)) {
183                         name = data->d_name;
184                         len = strlen(name);
185
186                         /* if this file name does not end with a number it's not of our interest */
187                         if ((len < 1) || !isdigit(name[--len]))
188                                 continue;
189                         while (isdigit(name[--len]));
190                         if (sscanf(&name[len+1], "%d", &n) != 1) 
191                                 continue;
192
193                         ret = usb_dev_add(alldevsp, n, err_str);
194                 }
195
196                 closedir(dir);
197                 return ret;
198         }
199
200         /* neither of them worked */
201         return 0;
202 }
203
204 static 
205 int usb_mmap(pcap_t* handle)
206 {
207         int len = ioctl(handle->fd, MON_IOCQ_RING_SIZE);
208         if (len < 0) 
209                 return 0;
210
211         handle->md.mmapbuflen = len;
212         handle->md.mmapbuf = mmap(0, handle->md.mmapbuflen, PROT_READ,
213             MAP_SHARED, handle->fd, 0);
214         return handle->md.mmapbuf != MAP_FAILED;
215 }
216
217 #define CTRL_TIMEOUT    (5*1000)        /* milliseconds */
218
219 #define USB_DIR_IN              0x80
220 #define USB_TYPE_STANDARD       0x00
221 #define USB_RECIP_DEVICE        0x00
222
223 #define USB_REQ_GET_DESCRIPTOR  6
224
225 #define USB_DT_DEVICE           1
226
227 /* probe the descriptors of the devices attached to the bus */
228 /* the descriptors will end up in the captured packet stream */
229 /* and be decoded by external apps like wireshark */
230 /* without these identifying probes packet data can't be fully decoded */
231 static void
232 probe_devices(int bus)
233 {
234         struct usbdevfs_ctrltransfer ctrl;
235         struct dirent* data;
236         int ret = 0;
237         char buf[40];
238         DIR* dir;
239
240         /* scan usb bus directories for device nodes */
241         snprintf(buf, sizeof(buf), "/dev/bus/usb/%03d", bus);
242         dir = opendir(buf);
243         if (!dir)
244                 return;
245
246         while ((ret >= 0) && ((data = readdir(dir)) != 0)) {
247                 int fd;
248                 char* name = data->d_name;
249
250                 if (name[0] == '.')
251                         continue;
252
253                 snprintf(buf, sizeof(buf), "/dev/bus/usb/%03d/%s", bus, data->d_name);
254                 
255                 fd = open(buf, O_RDWR);
256                 if (fd == -1)
257                         continue;
258
259                 /*
260                  * Sigh.  Different kernels have different member names
261                  * for this structure.
262                  */
263 #ifdef HAVE_USBDEVFS_CTRLTRANSFER_BREQUESTTYPE
264                 ctrl.bRequestType = USB_DIR_IN | USB_TYPE_STANDARD | USB_RECIP_DEVICE;
265                 ctrl.bRequest = USB_REQ_GET_DESCRIPTOR;
266                 ctrl.wValue = USB_DT_DEVICE << 8;
267                 ctrl.wIndex = 0;
268                 ctrl.wLength = sizeof(buf);
269 #else
270                 ctrl.requesttype = USB_DIR_IN | USB_TYPE_STANDARD | USB_RECIP_DEVICE;
271                 ctrl.request = USB_REQ_GET_DESCRIPTOR;
272                 ctrl.value = USB_DT_DEVICE << 8;
273                 ctrl.index = 0;
274                 ctrl.length = sizeof(buf);
275 #endif
276                 ctrl.data = buf;
277                 ctrl.timeout = CTRL_TIMEOUT;
278
279                 ret = ioctl(fd, USBDEVFS_CONTROL, &ctrl);
280
281                 close(fd);
282         }
283         closedir(dir);
284 }
285
286 pcap_t *
287 usb_create(const char *device, char *ebuf)
288 {
289         pcap_t *p;
290
291         p = pcap_create_common(device, ebuf);
292         if (p == NULL)
293                 return (NULL);
294
295         p->activate_op = usb_activate;
296         return (p);
297 }
298
299 static int
300 usb_activate(pcap_t* handle)
301 {
302         char            full_path[USB_LINE_LEN];
303
304         /* Initialize some components of the pcap structure. */
305         handle->bufsize = handle->snapshot;
306         handle->offset = 0;
307         handle->linktype = DLT_USB_LINUX;
308
309         handle->inject_op = usb_inject_linux;
310         handle->setfilter_op = install_bpf_program; /* no kernel filtering */
311         handle->setdirection_op = usb_setdirection_linux;
312         handle->set_datalink_op = NULL; /* can't change data link type */
313         handle->getnonblock_op = pcap_getnonblock_fd;
314         handle->setnonblock_op = pcap_setnonblock_fd;
315
316         /*get usb bus index from device name */
317         if (sscanf(handle->opt.source, USB_IFACE"%d", &handle->md.ifindex) != 1)
318         {
319                 snprintf(handle->errbuf, PCAP_ERRBUF_SIZE,
320                         "Can't get USB bus index from %s", handle->opt.source);
321                 return PCAP_ERROR;
322         }
323
324         /*now select the read method: try to open binary interface */
325         snprintf(full_path, USB_LINE_LEN, LINUX_USB_MON_DEV"%d", handle->md.ifindex);  
326         handle->fd = open(full_path, O_RDONLY, 0);
327         if (handle->fd >= 0)
328         {
329                 if (handle->opt.rfmon) {
330                         /*
331                          * Monitor mode doesn't apply to USB devices.
332                          */
333                         close(handle->fd);
334                         return PCAP_ERROR_RFMON_NOTSUP;
335                 }
336
337                 /* binary api is available, try to use fast mmap access */
338                 if (usb_mmap(handle)) {
339                         handle->linktype = DLT_USB_LINUX_MMAPPED;
340                         handle->stats_op = usb_stats_linux_bin;
341                         handle->read_op = usb_read_linux_mmap;
342                         handle->cleanup_op = usb_cleanup_linux_mmap;
343                         probe_devices(handle->md.ifindex);
344
345                         /*
346                          * "handle->fd" is a real file, so "select()" and
347                          * "poll()" work on it.
348                          */
349                         handle->selectable_fd = handle->fd;
350                         return 0;
351                 }
352
353                 /* can't mmap, use plain binary interface access */
354                 handle->stats_op = usb_stats_linux_bin;
355                 handle->read_op = usb_read_linux_bin;
356                 probe_devices(handle->md.ifindex);
357         }
358         else {
359                 /*Binary interface not available, try open text interface */
360                 snprintf(full_path, USB_LINE_LEN, USB_TEXT_DIR"/%dt", handle->md.ifindex);  
361                 handle->fd = open(full_path, O_RDONLY, 0);
362                 if (handle->fd < 0)
363                 {
364                         if (errno == ENOENT)
365                         {
366                                 /*
367                                  * Not found at the new location; try
368                                  * the old location.
369                                  */
370                                 snprintf(full_path, USB_LINE_LEN, USB_TEXT_DIR_OLD"/%dt", handle->md.ifindex);  
371                                 handle->fd = open(full_path, O_RDONLY, 0);
372                         }
373                         if (handle->fd < 0) {
374                                 /* no more fallback, give it up*/
375                                 snprintf(handle->errbuf, PCAP_ERRBUF_SIZE,
376                                         "Can't open USB bus file %s: %s", full_path, strerror(errno));
377                                 return PCAP_ERROR;
378                         }
379                 }
380
381                 if (handle->opt.rfmon) {
382                         /*
383                          * Monitor mode doesn't apply to USB devices.
384                          */
385                         close(handle->fd);
386                         return PCAP_ERROR_RFMON_NOTSUP;
387                 }
388
389                 handle->stats_op = usb_stats_linux;
390                 handle->read_op = usb_read_linux;
391         }
392
393         /*
394          * "handle->fd" is a real file, so "select()" and "poll()"
395          * work on it.
396          */
397         handle->selectable_fd = handle->fd;
398
399         /* for plain binary access and text access we need to allocate the read
400          * buffer */
401         handle->buffer = malloc(handle->bufsize);
402         if (!handle->buffer) {
403                 snprintf(handle->errbuf, PCAP_ERRBUF_SIZE,
404                          "malloc: %s", pcap_strerror(errno));
405                 close(handle->fd);
406                 return PCAP_ERROR;
407         }
408         return 0;
409 }
410
411 static inline int 
412 ascii_to_int(char c)
413 {
414         return c < 'A' ? c- '0': ((c<'a') ? c - 'A' + 10: c-'a'+10);
415 }
416
417 /*
418  * see <linux-kernel-source>/Documentation/usb/usbmon.txt and 
419  * <linux-kernel-source>/drivers/usb/mon/mon_text.c for urb string 
420  * format description
421  */
422 static int
423 usb_read_linux(pcap_t *handle, int max_packets, pcap_handler callback, u_char *user)
424 {
425         /* see:
426         * /usr/src/linux/Documentation/usb/usbmon.txt 
427         * for message format
428         */
429         unsigned timestamp;
430         int tag, cnt, ep_num, dev_addr, dummy, ret, urb_len, data_len;
431         char etype, pipeid1, pipeid2, status[16], urb_tag, line[USB_LINE_LEN];
432         char *string = line;
433         u_char * rawdata = handle->buffer;
434         struct pcap_pkthdr pkth;
435         pcap_usb_header* uhdr = (pcap_usb_header*)handle->buffer;
436         u_char urb_transfer=0;
437         int incoming=0;
438
439         /* ignore interrupt system call errors */
440         do {
441                 ret = read(handle->fd, line, USB_LINE_LEN - 1);
442                 if (handle->break_loop)
443                 {
444                         handle->break_loop = 0;
445                         return -2;
446                 }
447         } while ((ret == -1) && (errno == EINTR));
448         if (ret < 0)
449         {
450                 if (errno == EAGAIN)
451                         return 0;       /* no data there */
452
453                 snprintf(handle->errbuf, PCAP_ERRBUF_SIZE,
454                     "Can't read from fd %d: %s", handle->fd, strerror(errno));
455                 return -1;
456         }
457
458         /* read urb header; %n argument may increment return value, but it's 
459         * not mandatory, so does not count on it*/
460         string[ret] = 0;
461         ret = sscanf(string, "%x %d %c %c%c:%d:%d %s%n", &tag, &timestamp, &etype, 
462                 &pipeid1, &pipeid2, &dev_addr, &ep_num, status, 
463                 &cnt);
464         if (ret < 8)
465         {
466                 snprintf(handle->errbuf, PCAP_ERRBUF_SIZE,
467                     "Can't parse USB bus message '%s', too few tokens (expected 8 got %d)",
468                     string, ret);
469                 return -1;
470         }
471         uhdr->id = tag;
472         uhdr->device_address = dev_addr;
473         uhdr->bus_id = handle->md.ifindex;
474         uhdr->status = 0;
475         string += cnt;
476
477         /* don't use usbmon provided timestamp, since it have low precision*/
478         if (gettimeofday(&pkth.ts, NULL) < 0) 
479         {
480                 snprintf(handle->errbuf, PCAP_ERRBUF_SIZE,
481                         "Can't get timestamp for message '%s' %d:%s", 
482                         string, errno, strerror(errno));
483                 return -1;
484         }
485         uhdr->ts_sec = pkth.ts.tv_sec;
486         uhdr->ts_usec = pkth.ts.tv_usec;
487
488         /* parse endpoint information */
489         if (pipeid1 == 'C')
490                 urb_transfer = URB_CONTROL;
491         else if (pipeid1 == 'Z')
492                 urb_transfer = URB_ISOCHRONOUS;
493         else if (pipeid1 == 'I')
494                 urb_transfer = URB_INTERRUPT;
495         else if (pipeid1 == 'B')
496                 urb_transfer = URB_BULK;
497         if (pipeid2 == 'i') {
498                 ep_num |= URB_TRANSFER_IN;
499                 incoming = 1;
500         }
501         if (etype == 'C')
502                 incoming = !incoming;
503
504         /* direction check*/
505         if (incoming)
506         {
507                 if (handle->direction == PCAP_D_OUT)
508                         return 0;
509         }
510         else
511                 if (handle->direction == PCAP_D_IN)
512                         return 0;
513         uhdr->event_type = etype;
514         uhdr->transfer_type = urb_transfer;
515         uhdr->endpoint_number = ep_num;
516         pkth.caplen = sizeof(pcap_usb_header);
517         rawdata += sizeof(pcap_usb_header);
518
519         /* check if this is a setup packet */
520         ret = sscanf(status, "%d", &dummy);
521         if (ret != 1)
522         {
523                 /* this a setup packet, setup data can be filled with underscore if
524                 * usbmon has not been able to read them, so we must parse this fields as 
525                 * strings */
526                 pcap_usb_setup* shdr;
527                 char str1[3], str2[3], str3[5], str4[5], str5[5];
528                 ret = sscanf(string, "%s %s %s %s %s%n", str1, str2, str3, str4, 
529                 str5, &cnt);
530                 if (ret < 5)
531                 {
532                         snprintf(handle->errbuf, PCAP_ERRBUF_SIZE,
533                                 "Can't parse USB bus message '%s', too few tokens (expected 5 got %d)",
534                                 string, ret);
535                         return -1;
536                 }
537                 string += cnt;
538
539                 /* try to convert to corresponding integer */
540                 shdr = &uhdr->setup;
541                 shdr->bmRequestType = strtoul(str1, 0, 16);
542                 shdr->bRequest = strtoul(str2, 0, 16);
543                 shdr->wValue = htols(strtoul(str3, 0, 16));
544                 shdr->wIndex = htols(strtoul(str4, 0, 16));
545                 shdr->wLength = htols(strtoul(str5, 0, 16));
546
547                 uhdr->setup_flag = 0;
548         }
549         else 
550                 uhdr->setup_flag = 1;
551
552         /* read urb data */
553         ret = sscanf(string, " %d%n", &urb_len, &cnt);
554         if (ret < 1)
555         {
556                 snprintf(handle->errbuf, PCAP_ERRBUF_SIZE,
557                   "Can't parse urb length from '%s'", string);
558                 return -1;
559         }
560         string += cnt;
561
562         /* urb tag is not present if urb length is 0, so we can stop here 
563          * text parsing */
564         pkth.len = urb_len+pkth.caplen;
565         uhdr->urb_len = urb_len;
566         uhdr->data_flag = 1;
567         data_len = 0;
568         if (uhdr->urb_len == 0)
569                 goto got;
570
571         /* check for data presence; data is present if and only if urb tag is '=' */
572         if (sscanf(string, " %c", &urb_tag) != 1)
573         {
574                 snprintf(handle->errbuf, PCAP_ERRBUF_SIZE,
575                         "Can't parse urb tag from '%s'", string);
576                 return -1;
577         }
578
579         if (urb_tag != '=') 
580                 goto got;
581
582         /* skip urb tag and following space */
583         string += 3;
584
585         /* if we reach this point we got some urb data*/
586         uhdr->data_flag = 0;
587
588         /* read all urb data; if urb length is greater then the usbmon internal 
589          * buffer length used by the kernel to spool the URB, we get only
590          * a partial information.
591          * At least until linux 2.6.17 there is no way to set usbmon intenal buffer
592          * length and default value is 130. */
593         while ((string[0] != 0) && (string[1] != 0) && (pkth.caplen < handle->snapshot))
594         {
595                 rawdata[0] = ascii_to_int(string[0]) * 16 + ascii_to_int(string[1]);
596                 rawdata++;
597                 string+=2;
598                 if (string[0] == ' ')
599                         string++;
600                 pkth.caplen++;
601                 data_len++;
602         }
603
604 got:
605         uhdr->data_len = data_len;
606         if (pkth.caplen > handle->snapshot)
607                 pkth.caplen = handle->snapshot;
608
609         if (handle->fcode.bf_insns == NULL ||
610             bpf_filter(handle->fcode.bf_insns, handle->buffer,
611               pkth.len, pkth.caplen)) {
612                 handle->md.packets_read++;
613                 callback(user, &pkth, handle->buffer);
614                 return 1;
615         }
616         return 0;       /* didn't pass filter */
617 }
618
619 static int
620 usb_inject_linux(pcap_t *handle, const void *buf, size_t size)
621 {
622         snprintf(handle->errbuf, PCAP_ERRBUF_SIZE, "inject not supported on "
623                 "USB devices");
624         return (-1);
625 }
626
627 static int 
628 usb_stats_linux(pcap_t *handle, struct pcap_stat *stats)
629 {
630         int dummy, ret, consumed, cnt;
631         char string[USB_LINE_LEN];
632         char token[USB_LINE_LEN];
633         char * ptr = string;
634         int fd;
635
636         snprintf(string, USB_LINE_LEN, USB_TEXT_DIR"/%ds", handle->md.ifindex);
637         fd = open(string, O_RDONLY, 0);
638         if (fd < 0)
639         {
640                 if (errno == ENOENT)
641                 {
642                         /*
643                          * Not found at the new location; try the old
644                          * location.
645                          */
646                         snprintf(string, USB_LINE_LEN, USB_TEXT_DIR_OLD"/%ds", handle->md.ifindex);
647                         fd = open(string, O_RDONLY, 0);
648                 }
649                 if (fd < 0) {
650                         snprintf(handle->errbuf, PCAP_ERRBUF_SIZE,
651                                 "Can't open USB stats file %s: %s", 
652                                 string, strerror(errno));
653                         return -1;
654                 }
655         }
656
657         /* read stats line */
658         do {
659                 ret = read(fd, string, USB_LINE_LEN-1);
660         } while ((ret == -1) && (errno == EINTR));
661         close(fd);
662
663         if (ret < 0)
664         {
665                 snprintf(handle->errbuf, PCAP_ERRBUF_SIZE,
666                         "Can't read stats from fd %d ", fd);
667                 return -1;
668         }
669         string[ret] = 0;
670
671         /* extract info on dropped urbs */
672         for (consumed=0; consumed < ret; ) {
673                 /* from the sscanf man page: 
674                  * The C standard says: "Execution of a %n directive does 
675                  * not increment the assignment count returned at the completion
676                  * of  execution" but the Corrigendum seems to contradict this.
677                  * Do not make any assumptions on the effect of %n conversions 
678                  * on the return value and explicitly check for cnt assignmet*/
679                 int ntok;
680
681                 cnt = -1;
682                 ntok = sscanf(ptr, "%s%n", token, &cnt);
683                 if ((ntok < 1) || (cnt < 0))
684                         break;
685                 consumed += cnt;
686                 ptr += cnt;
687                 if (strcmp(token, "nreaders") == 0)
688                         ret = sscanf(ptr, "%d", &stats->ps_drop);
689                 else 
690                         ret = sscanf(ptr, "%d", &dummy);
691                 if (ntok != 1)
692                         break;
693                 consumed += cnt;
694                 ptr += cnt;
695         }
696
697         stats->ps_recv = handle->md.packets_read;
698         stats->ps_ifdrop = 0;
699         return 0;
700 }
701
702 static int 
703 usb_setdirection_linux(pcap_t *p, pcap_direction_t d)
704 {
705         p->direction = d;
706         return 0;
707 }
708
709
710 static int 
711 usb_stats_linux_bin(pcap_t *handle, struct pcap_stat *stats)
712 {
713         int ret;
714         struct mon_bin_stats st;
715         ret = ioctl(handle->fd, MON_IOCG_STATS, &st);
716         if (ret < 0)
717         {
718                 snprintf(handle->errbuf, PCAP_ERRBUF_SIZE,
719                         "Can't read stats from fd %d:%s ", handle->fd, strerror(errno));
720                 return -1;
721         }
722
723         stats->ps_recv = handle->md.packets_read + st.queued;
724         stats->ps_drop = st.dropped;
725         stats->ps_ifdrop = 0;
726         return 0;
727 }
728
729 /*
730  * see <linux-kernel-source>/Documentation/usb/usbmon.txt and 
731  * <linux-kernel-source>/drivers/usb/mon/mon_bin.c binary ABI
732  */
733 static int
734 usb_read_linux_bin(pcap_t *handle, int max_packets, pcap_handler callback, u_char *user)
735 {
736         struct mon_bin_get info;
737         int ret;
738         struct pcap_pkthdr pkth;
739         int clen = handle->snapshot - sizeof(pcap_usb_header);
740
741         /* the usb header is going to be part of 'packet' data*/
742         info.hdr = (pcap_usb_header*) handle->buffer;
743         info.data = handle->buffer + sizeof(pcap_usb_header);
744         info.data_len = clen;
745
746         /* ignore interrupt system call errors */
747         do {
748                 ret = ioctl(handle->fd, MON_IOCX_GET, &info);
749                 if (handle->break_loop)
750                 {
751                         handle->break_loop = 0;
752                         return -2;
753                 }
754         } while ((ret == -1) && (errno == EINTR));
755         if (ret < 0)
756         {
757                 if (errno == EAGAIN)
758                         return 0;       /* no data there */
759
760                 snprintf(handle->errbuf, PCAP_ERRBUF_SIZE,
761                     "Can't read from fd %d: %s", handle->fd, strerror(errno));
762                 return -1;
763         }
764
765         /* we can get less that than really captured from kernel, depending on
766          * snaplen, so adjust header accordingly */
767         if (info.hdr->data_len < clen)
768                 clen = info.hdr->data_len;
769         info.hdr->data_len = clen;
770         pkth.caplen = clen + sizeof(pcap_usb_header);
771         pkth.len = info.hdr->data_len + sizeof(pcap_usb_header);
772         pkth.ts.tv_sec = info.hdr->ts_sec;
773         pkth.ts.tv_usec = info.hdr->ts_usec;
774
775         if (handle->fcode.bf_insns == NULL ||
776             bpf_filter(handle->fcode.bf_insns, handle->buffer,
777               pkth.len, pkth.caplen)) {
778                 handle->md.packets_read++;
779                 callback(user, &pkth, handle->buffer);
780                 return 1;
781         }
782
783         return 0;       /* didn't pass filter */
784 }
785
786 /*
787  * see <linux-kernel-source>/Documentation/usb/usbmon.txt and 
788  * <linux-kernel-source>/drivers/usb/mon/mon_bin.c binary ABI
789  */
790 #define VEC_SIZE 32
791 static int
792 usb_read_linux_mmap(pcap_t *handle, int max_packets, pcap_handler callback, u_char *user)
793 {
794         struct mon_bin_mfetch fetch;
795         int32_t vec[VEC_SIZE];
796         struct pcap_pkthdr pkth;
797         pcap_usb_header* hdr;
798         int nflush = 0;
799         int packets = 0;
800         int clen, max_clen;
801
802         max_clen = handle->snapshot - sizeof(pcap_usb_header);
803
804         for (;;) {
805                 int i, ret;
806                 int limit = max_packets - packets;
807                 if (limit <= 0)
808                         limit = VEC_SIZE;
809                 if (limit > VEC_SIZE)
810                         limit = VEC_SIZE;
811
812                 /* try to fetch as many events as possible*/
813                 fetch.offvec = vec;
814                 fetch.nfetch = limit;
815                 fetch.nflush = nflush;
816                 /* ignore interrupt system call errors */
817                 do {
818                         ret = ioctl(handle->fd, MON_IOCX_MFETCH, &fetch);
819                         if (handle->break_loop)
820                         {
821                                 handle->break_loop = 0;
822                                 return -2;
823                         }
824                 } while ((ret == -1) && (errno == EINTR));
825                 if (ret < 0)
826                 {
827                         if (errno == EAGAIN)
828                                 return 0;       /* no data there */
829
830                         snprintf(handle->errbuf, PCAP_ERRBUF_SIZE,
831                             "Can't mfetch fd %d: %s", handle->fd, strerror(errno));
832                         return -1;
833                 }
834
835                 /* keep track of processed events, we will flush them later */
836                 nflush = fetch.nfetch;
837                 for (i=0; i<fetch.nfetch; ++i) {
838                         /* discard filler */
839                         hdr = (pcap_usb_header*) &handle->md.mmapbuf[vec[i]];
840                         if (hdr->event_type == '@') 
841                                 continue;
842
843                         /* we can get less that than really captured from kernel, depending on
844                         * snaplen, so adjust header accordingly */
845                         clen = max_clen;
846                         if (hdr->data_len < clen)
847                                 clen = hdr->data_len;
848
849                         /* get packet info from header*/
850                         pkth.caplen = clen + sizeof(pcap_usb_header_mmapped);
851                         pkth.len = hdr->data_len + sizeof(pcap_usb_header_mmapped);
852                         pkth.ts.tv_sec = hdr->ts_sec;
853                         pkth.ts.tv_usec = hdr->ts_usec;
854
855                         if (handle->fcode.bf_insns == NULL ||
856                             bpf_filter(handle->fcode.bf_insns, (u_char*) hdr,
857                               pkth.len, pkth.caplen)) {
858                                 handle->md.packets_read++;
859                                 callback(user, &pkth, (u_char*) hdr);
860                                 packets++;
861                         }
862                 }
863
864                 /* with max_packets <= 0 we stop afer the first chunk*/
865                 if ((max_packets <= 0) || (packets == max_packets))
866                         break;
867         }
868
869         /* flush pending events*/
870         ioctl(handle->fd, MON_IOCH_MFLUSH, nflush);
871         return packets;
872 }
873
874 static void
875 usb_cleanup_linux_mmap(pcap_t* handle)
876 {
877         /* if we have a memory-mapped buffer, unmap it */
878         if (handle->md.mmapbuf != NULL) {
879                 munmap(handle->md.mmapbuf, handle->md.mmapbuflen);
880                 handle->md.mmapbuf = NULL;
881         }
882         pcap_cleanup_live_common(handle);
883 }