]> The Tcpdump Group git mirrors - libpcap/blob - pcap-win32.c
Added support for Endace Dag cards under Windows
[libpcap] / pcap-win32.c
1 /*
2 * Copyright (c) 1999 - 2003
3 * NetGroup, Politecnico di Torino (Italy)
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 *
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
15 * 3. Neither the name of the Politecnico di Torino nor the names of its
16 * contributors may be used to endorse or promote products derived from
17 * this software without specific prior written permission.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
22 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
23 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
24 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
25 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
26 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
27 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
28 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
29 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30 *
31 */
32
33 #ifndef lint
34 static const char rcsid[] _U_ =
35 "@(#) $Header: /tcpdump/master/libpcap/pcap-win32.c,v 1.20 2004-01-28 14:06:20 risso Exp $ (LBL)";
36 #endif
37
38 #include <pcap-int.h>
39 #include <packet32.h>
40 #include <Ntddndis.h>
41 #ifdef HAVE_DAG_API
42 #include <dagnew.h>
43 #include <dagapi.h>
44 #endif /* HAVE_DAG_API */
45 #ifdef __MINGW32__
46 int* _errno();
47 #define errno (*_errno())
48 #endif /* __MINGW32__ */
49
50 static int pcap_setfilter_win32_npf(pcap_t *, struct bpf_program *);
51 static int pcap_setfilter_win32_dag(pcap_t *, struct bpf_program *);
52 static int pcap_getnonblock_win32(pcap_t *, char *);
53 static int pcap_setnonblock_win32(pcap_t *, int, char *);
54
55 #define PcapBufSize 256000 /*dimension of the buffer in the pcap_t structure*/
56 #define SIZE_BUF 1000000
57
58 /* Equivalent to ntohs(), but a lot faster under Windows */
59 #define SWAPS(_X) ((_X & 0xff) << 8) | (_X >> 8)
60
61 /*
62 * Header that the WinPcap driver associates to the packets.
63 * Once was in bpf.h
64 */
65 struct bpf_hdr {
66 struct timeval bh_tstamp; /* time stamp */
67 bpf_u_int32 bh_caplen; /* length of captured portion */
68 bpf_u_int32 bh_datalen; /* original length of packet */
69 u_short bh_hdrlen; /* length of bpf header (this struct
70 plus alignment padding) */
71 };
72
73 /* Start winsock */
74 int
75 wsockinit()
76 {
77 WORD wVersionRequested;
78 WSADATA wsaData;
79 int err;
80 wVersionRequested = MAKEWORD( 1, 1);
81 err = WSAStartup( wVersionRequested, &wsaData );
82 if ( err != 0 )
83 {
84 return -1;
85 }
86 return 0;
87 }
88
89
90 static int
91 pcap_stats_win32(pcap_t *p, struct pcap_stat *ps)
92 {
93
94 if(PacketGetStats(p->adapter, (struct bpf_stat*)ps) != TRUE){
95 snprintf(p->errbuf, PCAP_ERRBUF_SIZE, "PacketGetStats error: %s", pcap_win32strerror());
96 return -1;
97 }
98
99 return 0;
100 }
101
102 static int
103 pcap_read_win32_npf(pcap_t *p, int cnt, pcap_handler callback, u_char *user)
104 {
105 int cc;
106 int n = 0;
107 register u_char *bp, *ep;
108
109 cc = p->cc;
110 if (p->cc == 0) {
111 /*
112 * Has "pcap_breakloop()" been called?
113 */
114 if (p->break_loop) {
115 /*
116 * Yes - clear the flag that indicates that it
117 * has, and return -2 to indicate that we were
118 * told to break out of the loop.
119 */
120 p->break_loop = 0;
121 return (-2);
122 }
123
124 /* capture the packets */
125 if(PacketReceivePacket(p->adapter,p->Packet,TRUE)==FALSE){
126 snprintf(p->errbuf, PCAP_ERRBUF_SIZE, "read error: PacketReceivePacket failed");
127 return (-1);
128 }
129
130 cc = p->Packet->ulBytesReceived;
131
132 bp = p->Packet->Buffer;
133 }
134 else
135 bp = p->bp;
136
137 /*
138 * Loop through each packet.
139 */
140 #define bhp ((struct bpf_hdr *)bp)
141 ep = bp + cc;
142 while (1) {
143 register int caplen, hdrlen;
144
145 /*
146 * Has "pcap_breakloop()" been called?
147 * If so, return immediately - if we haven't read any
148 * packets, clear the flag and return -2 to indicate
149 * that we were told to break out of the loop, otherwise
150 * leave the flag set, so that the *next* call will break
151 * out of the loop without having read any packets, and
152 * return the number of packets we've processed so far.
153 */
154 if (p->break_loop) {
155 if (n == 0) {
156 p->break_loop = 0;
157 return (-2);
158 } else {
159 p->bp = bp;
160 p->cc = ep - bp;
161 return (n);
162 }
163 }
164 if (bp >= ep)
165 break;
166
167 caplen = bhp->bh_caplen;
168 hdrlen = bhp->bh_hdrlen;
169
170 /*
171 * XXX A bpf_hdr matches a pcap_pkthdr.
172 */
173 (*callback)(user, (struct pcap_pkthdr*)bp, bp + hdrlen);
174 bp += BPF_WORDALIGN(caplen + hdrlen);
175 if (++n >= cnt && cnt > 0) {
176 p->bp = bp;
177 p->cc = ep - bp;
178 return (n);
179 }
180 }
181 #undef bhp
182 p->cc = 0;
183 return (n);
184 }
185
186 #ifdef HAVE_DAG_API
187 static int
188 pcap_read_win32_dag(pcap_t *p, int cnt, pcap_handler callback, u_char *user)
189 {
190 u_char *dp = NULL;
191 int packet_len = 0, caplen = 0;
192 struct pcap_pkthdr pcap_header;
193 u_char *endofbuf;
194 int n = 0;
195 dag_record_t *header;
196 unsigned erf_record_len;
197 ULONGLONG ts;
198 int cc;
199 unsigned swt;
200 unsigned dfp = p->adapter->DagFastProcess;
201
202 cc = p->cc;
203 if (cc == 0) /* Get new packets only if we have processed all the ones of the previous read */
204 {
205 /* Get new packets from the network */
206 if(PacketReceivePacket(p->adapter, p->Packet, TRUE)==FALSE){
207 snprintf(p->errbuf, PCAP_ERRBUF_SIZE, "read error: PacketReceivePacket failed");
208 return (-1);
209 }
210
211 cc = p->Packet->ulBytesReceived;
212 if(cc == 0)
213 /* The timeout has expired but we no packets arrived */
214 return 0;
215 header = (dag_record_t*)p->adapter->DagBuffer;
216 }
217 else
218 header = (dag_record_t*)p->bp;
219
220 endofbuf = (char*)header + cc;
221
222 /*
223 * Cycle through the packets
224 */
225 do
226 {
227 erf_record_len = SWAPS(header->rlen);
228 if((char*)header + erf_record_len > endofbuf)
229 break;
230
231 /* Increase the number of captured packets */
232 p->md.stat.ps_recv++;
233
234 /* Find the beginning of the packet */
235 dp = ((u_char *)header) + dag_record_size;
236
237 /* Determine actual packet len */
238 switch(header->type)
239 {
240 case TYPE_ATM:
241 packet_len = ATM_SNAPLEN;
242 caplen = ATM_SNAPLEN;
243 dp += 4;
244
245 break;
246
247 case TYPE_ETH:
248 swt = SWAPS(header->wlen);
249 packet_len = swt - (p->md.dag_fcs_bits);
250 caplen = erf_record_len - dag_record_size - 2;
251 if (caplen > packet_len)
252 {
253 caplen = packet_len;
254 }
255 dp += 2;
256
257 break;
258
259 case TYPE_HDLC_POS:
260 swt = SWAPS(header->wlen);
261 packet_len = swt - (p->md.dag_fcs_bits);
262 caplen = erf_record_len - dag_record_size;
263 if (caplen > packet_len)
264 {
265 caplen = packet_len;
266 }
267
268 break;
269 }
270
271 if(caplen > p->snapshot)
272 caplen = p->snapshot;
273
274 /*
275 * Has "pcap_breakloop()" been called?
276 * If so, return immediately - if we haven't read any
277 * packets, clear the flag and return -2 to indicate
278 * that we were told to break out of the loop, otherwise
279 * leave the flag set, so that the *next* call will break
280 * out of the loop without having read any packets, and
281 * return the number of packets we've processed so far.
282 */
283 if (p->break_loop)
284 {
285 if (n == 0)
286 {
287 p->break_loop = 0;
288 return (-2);
289 }
290 else
291 {
292 p->bp = (char*)header;
293 p->cc = endofbuf - (char*)header;
294 return (n);
295 }
296 }
297
298 if(!dfp)
299 {
300 /* convert between timestamp formats */
301 ts = header->ts;
302 pcap_header.ts.tv_sec = (int)(ts >> 32);
303 ts = (ts & 0xffffffffi64) * 1000000;
304 ts += 0x80000000; /* rounding */
305 pcap_header.ts.tv_usec = (int)(ts >> 32);
306 if (pcap_header.ts.tv_usec >= 1000000) {
307 pcap_header.ts.tv_usec -= 1000000;
308 pcap_header.ts.tv_sec++;
309 }
310 }
311
312 /* No underlaying filtering system. We need to filter on our own */
313 if (p->fcode.bf_insns)
314 {
315 if (bpf_filter(p->fcode.bf_insns, dp, packet_len, caplen) == 0)
316 {
317 /* Move to next packet */
318 header = (dag_record_t*)((char*)header + erf_record_len);
319 continue;
320 }
321 }
322
323 /* Fill the header for the user suppplied callback function */
324 pcap_header.caplen = caplen;
325 pcap_header.len = packet_len;
326
327 /* Call the callback function */
328 (*callback)(user, &pcap_header, dp);
329
330 /* Move to next packet */
331 header = (dag_record_t*)((char*)header + erf_record_len);
332
333 /* Stop if the number of packets requested by user has been reached*/
334 if (++n >= cnt && cnt > 0)
335 {
336 p->bp = (char*)header;
337 p->cc = endofbuf - (char*)header;
338 return (n);
339 }
340 }
341 while((u_char*)header < endofbuf);
342
343 return 1;
344 }
345 #endif /* HAVE_DAG_API */
346
347 static void
348 pcap_close_win32(pcap_t *p)
349 {
350 if (p->buffer != NULL)
351 free(p->buffer);
352 if (p->adapter != NULL) {
353 PacketCloseAdapter(p->adapter);
354 p->adapter = NULL;
355 }
356 }
357
358 pcap_t *
359 pcap_open_live(const char *device, int snaplen, int promisc, int to_ms,
360 char *ebuf)
361 {
362 register pcap_t *p;
363 NetType type;
364
365 /* Init WinSock */
366 wsockinit();
367
368 p = (pcap_t *)malloc(sizeof(*p));
369 if (p == NULL)
370 {
371 snprintf(ebuf, PCAP_ERRBUF_SIZE, "malloc: %s", pcap_strerror(errno));
372 return (NULL);
373 }
374 memset(p, 0, sizeof(*p));
375 p->adapter=NULL;
376
377 p->adapter = PacketOpenAdapter((char*)device);
378
379 if (p->adapter == NULL)
380 {
381 /* Adapter detected but we are not able to open it. Return failure. */
382 snprintf(ebuf, PCAP_ERRBUF_SIZE, "Error opening adapter: %s", pcap_win32strerror());
383 return NULL;
384 }
385
386 /*get network type*/
387 if(PacketGetNetType (p->adapter,&type) == FALSE)
388 {
389 snprintf(ebuf, PCAP_ERRBUF_SIZE, "Cannot determine the network type: %s", pcap_win32strerror());
390 goto bad;
391 }
392
393 /*Set the linktype*/
394 switch (type.LinkType)
395 {
396 case NdisMediumWan:
397 p->linktype = DLT_EN10MB;
398 break;
399
400 case NdisMedium802_3:
401 p->linktype = DLT_EN10MB;
402 /*
403 * This is (presumably) a real Ethernet capture; give it a
404 * link-layer-type list with DLT_EN10MB and DLT_DOCSIS, so
405 * that an application can let you choose it, in case you're
406 * capturing DOCSIS traffic that a Cisco Cable Modem
407 * Termination System is putting out onto an Ethernet (it
408 * doesn't put an Ethernet header onto the wire, it puts raw
409 * DOCSIS frames out on the wire inside the low-level
410 * Ethernet framing).
411 */
412 p->dlt_list = (u_int *) malloc(sizeof(u_int) * 2);
413 /*
414 * If that fails, just leave the list empty.
415 */
416 if (p->dlt_list != NULL) {
417 p->dlt_list[0] = DLT_EN10MB;
418 p->dlt_list[1] = DLT_DOCSIS;
419 p->dlt_count = 2;
420 }
421 break;
422
423 case NdisMediumFddi:
424 p->linktype = DLT_FDDI;
425 break;
426
427 case NdisMedium802_5:
428 p->linktype = DLT_IEEE802;
429 break;
430
431 case NdisMediumArcnetRaw:
432 p->linktype = DLT_ARCNET;
433 break;
434
435 case NdisMediumArcnet878_2:
436 p->linktype = DLT_ARCNET;
437 break;
438
439 case NdisMediumAtm:
440 p->linktype = DLT_ATM_RFC1483;
441 break;
442
443 case NdisMediumCHDLC:
444 p->linktype = DLT_CHDLC;
445 break;
446
447 case NdisMediumPPPSerial:
448 p->linktype = DLT_PPP_SERIAL;
449 break;
450
451 case NdisMediumNull:
452 p->linktype = DLT_NULL;
453 break;
454
455 default:
456 p->linktype = DLT_EN10MB; /*an unknown adapter is assumed to be ethernet*/
457 break;
458 }
459
460 /* Set promisquous mode */
461 if (promisc) PacketSetHwFilter(p->adapter,NDIS_PACKET_TYPE_PROMISCUOUS);
462 else PacketSetHwFilter(p->adapter,NDIS_PACKET_TYPE_ALL_LOCAL);
463
464 /* Set the buffer size */
465 p->bufsize = PcapBufSize;
466
467 /* allocate Packet structure used during the capture */
468 if((p->Packet = PacketAllocatePacket())==NULL)
469 {
470 snprintf(ebuf, PCAP_ERRBUF_SIZE, "failed to allocate the PACKET structure");
471 goto bad;
472 }
473
474 if(!(p->adapter->Flags & INFO_FLAG_DAG_CARD))
475 {
476 /*
477 * Traditional Adapter
478 */
479
480 p->buffer = (u_char *)malloc(PcapBufSize);
481 if (p->buffer == NULL)
482 {
483 snprintf(ebuf, PCAP_ERRBUF_SIZE, "malloc: %s", pcap_strerror(errno));
484 goto bad;
485 }
486
487 PacketInitPacket(p->Packet,(BYTE*)p->buffer,p->bufsize);
488
489 p->snapshot = snaplen;
490
491 /* allocate the standard buffer in the driver */
492 if(PacketSetBuff( p->adapter, SIZE_BUF)==FALSE)
493 {
494 snprintf(ebuf, PCAP_ERRBUF_SIZE,"driver error: not enough memory to allocate the kernel buffer\n");
495 goto bad;
496 }
497
498 /* tell the driver to copy the buffer only if it contains at least 16K */
499 if(PacketSetMinToCopy(p->adapter,16000)==FALSE)
500 {
501 snprintf(ebuf, PCAP_ERRBUF_SIZE,"Error calling PacketSetMinToCopy: %s\n", pcap_win32strerror());
502 goto bad;
503 }
504 }
505 else
506 #ifdef HAVE_DAG_API
507 {
508 /*
509 * Dag Card
510 */
511 LONG status;
512 HKEY dagkey;
513 DWORD lptype;
514 DWORD lpcbdata;
515 int postype = 0;
516 char keyname[512];
517
518 snprintf(keyname, sizeof(keyname), "%s\\CardParams\\%s",
519 "SYSTEM\\CurrentControlSet\\Services\\DAG",
520 strstr(_strlwr((char*)device), "dag"));
521 do
522 {
523 status = RegOpenKeyEx(HKEY_LOCAL_MACHINE, keyname, 0, KEY_READ, &dagkey);
524 if(status != ERROR_SUCCESS)
525 break;
526
527 status = RegQueryValueEx(dagkey,
528 "PosType",
529 NULL,
530 &lptype,
531 (char*)&postype,
532 &lpcbdata);
533
534 if(status != ERROR_SUCCESS)
535 {
536 postype = 0;
537 }
538
539 RegCloseKey(dagkey);
540 }
541 while(FALSE);
542
543
544 p->snapshot = PacketSetSnapLen(p->adapter, snaplen);
545
546 /* Set the length of the FCS associated to any packet. This value
547 * will be subtracted to the packet length */
548 p->md.dag_fcs_bits = p->adapter->DagFcsLen;
549 }
550 #else
551 goto bad;
552 #endif /* HAVE_DAG_API */
553
554 PacketSetReadTimeout(p->adapter, to_ms);
555
556 #ifdef HAVE_DAG_API
557 if(p->adapter->Flags & INFO_FLAG_DAG_CARD)
558 {
559 /* install dag specific handlers for read and setfilter */
560 p->read_op = pcap_read_win32_dag;
561 p->setfilter_op = pcap_setfilter_win32_dag;
562 }
563 else
564 {
565 #endif /* HAVE_DAG_API */
566 /* install traditional npf handlers for read and setfilter */
567 p->read_op = pcap_read_win32_npf;
568 p->setfilter_op = pcap_setfilter_win32_npf;
569 #ifdef HAVE_DAG_API
570 }
571 #endif /* HAVE_DAG_API */
572 p->set_datalink_op = NULL; /* can't change data link type */
573 p->getnonblock_op = pcap_getnonblock_win32;
574 p->setnonblock_op = pcap_setnonblock_win32;
575 p->stats_op = pcap_stats_win32;
576 p->close_op = pcap_close_win32;
577
578 return (p);
579 bad:
580 if (p->adapter)
581 PacketCloseAdapter(p->adapter);
582 if (p->buffer != NULL)
583 free(p->buffer);
584 if(p->Packet)
585 PacketFreePacket(p->Packet);
586 /*
587 * Get rid of any link-layer type list we allocated.
588 */
589 if (p->dlt_list != NULL)
590 free(p->dlt_list);
591 free(p);
592 return (NULL);
593 }
594
595
596 static int
597 pcap_setfilter_win32_npf(pcap_t *p, struct bpf_program *fp)
598 {
599 if(PacketSetBpf(p->adapter,fp)==FALSE){
600 /* kernel filter not installed. */
601 snprintf(p->errbuf, PCAP_ERRBUF_SIZE, "Driver error: cannot set bpf filter: %s", pcap_win32strerror());
602 return (-1);
603 }
604 return (0);
605 }
606
607 /*
608 * We filter at user level, since the kernel driver does't process the packets
609 */
610 static int
611 pcap_setfilter_win32_dag(pcap_t *p, struct bpf_program *fp) {
612
613 if(!fp)
614 {
615 strncpy(p->errbuf, "setfilter: No filter specified", sizeof(p->errbuf));
616 return -1;
617 }
618
619 /* Install a user level filter */
620 if (install_bpf_program(p, fp) < 0)
621 {
622 snprintf(p->errbuf, sizeof(p->errbuf),
623 "setfilter, unable to install the filter: %s", pcap_strerror(errno));
624 return -1;
625 }
626
627 p->md.use_bpf = 0;
628
629 return (0);
630 }
631
632 static int
633 pcap_getnonblock_win32(pcap_t *p, char *errbuf)
634 {
635 /*
636 * XXX - if there were a PacketGetReadTimeout() call, we
637 * would use it, and return 1 if the timeout is -1
638 * and 0 otherwise.
639 */
640 return (p->nonblock);
641 }
642
643 static int
644 pcap_setnonblock_win32(pcap_t *p, int nonblock, char *errbuf)
645 {
646 int newtimeout;
647
648 if (nonblock) {
649 /*
650 * Set the read timeout to -1 for non-blocking mode.
651 */
652 newtimeout = -1;
653 } else {
654 /*
655 * Restore the timeout set when the device was opened.
656 * (Note that this may be -1, in which case we're not
657 * really leaving non-blocking mode.)
658 */
659 newtimeout = p->timeout;
660 }
661 if (!PacketSetReadTimeout(p->adapter, newtimeout)) {
662 snprintf(p->errbuf, PCAP_ERRBUF_SIZE,
663 "PacketSetReadTimeout: %s", pcap_win32strerror());
664 return (-1);
665 }
666 p->nonblock = (newtimeout == -1);
667 return (0);
668 }
669
670 /* Set the driver working mode */
671 int
672 pcap_setmode(pcap_t *p, int mode){
673
674 if (p->adapter==NULL)
675 {
676 snprintf(p->errbuf, PCAP_ERRBUF_SIZE, "impossible to set mode while reading from a file");
677 return -1;
678 }
679
680 if(PacketSetMode(p->adapter,mode)==FALSE)
681 {
682 snprintf(p->errbuf, PCAP_ERRBUF_SIZE, "driver error: working mode not recognized");
683 return -1;
684 }
685
686 return 0;
687 }
688
689 /* Send a packet to the network */
690 int
691 pcap_sendpacket(pcap_t *p, u_char *buf, int size){
692 LPPACKET PacketToSend;
693
694 if (p->adapter==NULL)
695 {
696 snprintf(p->errbuf, PCAP_ERRBUF_SIZE, "Writing a packet is allowed only on a physical adapter");
697 return -1;
698 }
699
700 PacketToSend=PacketAllocatePacket();
701 PacketInitPacket(PacketToSend,buf,size);
702 if(PacketSendPacket(p->adapter,PacketToSend,TRUE) == FALSE){
703 PacketFreePacket(PacketToSend);
704 return -1;
705 }
706
707 PacketFreePacket(PacketToSend);
708 return 0;
709 }
710
711 /* Set the dimension of the kernel-level capture buffer */
712 int
713 pcap_setbuff(pcap_t *p, int dim)
714 {
715 if (p->adapter==NULL)
716 {
717 snprintf(p->errbuf, PCAP_ERRBUF_SIZE, "The kernel buffer size cannot be set while reading from a file");
718 return -1;
719 }
720
721 if(PacketSetBuff(p->adapter,dim)==FALSE)
722 {
723 snprintf(p->errbuf, PCAP_ERRBUF_SIZE, "driver error: not enough memory to allocate the kernel buffer");
724 return -1;
725 }
726 return 0;
727 }
728
729 /*set the minimum amount of data that will release a read call*/
730 int
731 pcap_setmintocopy(pcap_t *p, int size)
732 {
733 if (p->adapter==NULL)
734 {
735 snprintf(p->errbuf, PCAP_ERRBUF_SIZE, "Impossible to set the mintocopy parameter on an offline capture");
736 return -1;
737 }
738
739 if(PacketSetMinToCopy(p->adapter, size)==FALSE)
740 {
741 snprintf(p->errbuf, PCAP_ERRBUF_SIZE, "driver error: unable to set the requested mintocopy size");
742 return -1;
743 }
744 return 0;
745 }