xref: /freebsd/contrib/libpcap/pcap-enet.c (revision 535af610)
1 /*
2  * Stanford Enetfilter subroutines for tcpdump
3  *
4  * Based on the MERIT NNstat etherifrt.c and the Ultrix pcap-pf.c
5  * subroutines.
6  *
7  * Rayan Zachariassen, CA*Net
8  */
9 
10 #ifdef HAVE_CONFIG_H
11 #include <config.h>
12 #endif
13 
14 #include <sys/types.h>
15 #include <sys/time.h>
16 #include <sys/file.h>
17 #include <sys/ioctl.h>
18 #include <sys/socket.h>
19 
20 #include <net/if.h>
21 #include <pcap/bpf.h>
22 #include <net/enet.h>
23 
24 #include <netinet/in.h>
25 #include <netinet/if_ether.h>
26 
27 #include <stdio.h>
28 #include <errno.h>
29 
30 #include "interface.h"
31 
32 struct packet_header {
33 #ifdef	IBMRTPC
34 	struct LengthWords	length;
35 	struct tap_header	tap;
36 #endif	/* IBMRTPC */
37 	u_char			packet[8]
38 };
39 
40 extern int errno;
41 
42 #define BUFSPACE (4*1024)
43 
44 /* Forwards */
45 static void efReadError(int, char *);
46 
47 void
48 readloop(int cnt, int if_fd, struct bpf_program *fp, printfunc printit)
49 {
50 #ifdef	IBMRTPC
51 	register struct packet_header *ph;
52 	register u_char *bp;
53 	register int inc;
54 #else	/* !IBMRTPC */
55 	static struct timeval tv = { 0 };
56 #endif	/* IBMRTPC */
57 	register int cc, caplen;
58 	register struct bpf_insn *fcode = fp->bf_insns;
59 	union {
60 		struct packet_header hdr;
61 		u_char	p[BUFSPACE];
62 		u_short	s;
63 	} buf;
64 
65 	while (1) {
66 		if ((cc = read(if_fd, (char *)buf.p, sizeof(buf))) < 0)
67 			efReadError(if_fd, "reader");
68 
69 #ifdef	IBMRTPC
70 		/*
71 		 * Loop through each packet.
72 		 */
73 		bp = buf.p;
74 		while (cc > 0) {
75 			ph = (struct packet_header *)bp;
76 			caplen = ph->tap.th_wirelen > snaplen ? snaplen : ph->tap
77 .th_wirelen ;
78 			if (pcap_filter(fcode, (char *)ph->packet,
79 						ph->tap.th_wirelen, caplen)) {
80 				if (cnt >= 0 && --cnt < 0)
81 					goto out;
82 				(*printit)((char *)ph->packet,
83 					(struct timeval *)ph->tap.th_timestamp,
84 					ph->tap.th_wirelen, caplen);
85 			}
86 			inc = ph->length.PacketOffset;
87 			cc -= inc;
88 			bp += inc;
89 		}
90 #else	/* !IBMRTPC */
91 		caplen = cc > snaplen ? snaplen : cc ;
92 		if (pcap_filter(fcode, buf.hdr.packet, cc, caplen)) {
93 			if (cnt >= 0 && --cnt < 0)
94 				goto out;
95 			(*printit)(buf.hdr.packet, &tv, cc, caplen);
96 		}
97 #endif	/* IBMRTPC */
98 	}
99  out:
100 	wrapup(if_fd);
101 }
102 
103 /* Call ONLY if read() has returned an error on packet filter */
104 static void
105 efReadError(int fid, char *msg)
106 {
107 	if (errno == EINVAL) {	/* read MAXINT bytes already! */
108 		if (lseek(fid, 0, 0) < 0) {
109 			perror("tcpdump: efReadError/lseek");
110 			exit(-1);
111 		}
112 		else
113 			return;
114 	}
115 	else {
116 		(void) fprintf(stderr, "tcpdump: ");
117 		perror(msg);
118 		exit(-1);
119 	}
120 }
121 
122 void
123 wrapup(int fd)
124 {
125 #ifdef	IBMRTPC
126 	struct enstats es;
127 
128 	if (ioctl(fd, EIOSTATS, &es) == -1) {
129 		perror("tcpdump: enet ioctl EIOSTATS error");
130 		exit(-1);
131 	}
132 
133 	fprintf(stderr, "%d packets queued", es.enStat_Rcnt);
134 	if (es.enStat_Rdrops > 0)
135 		fprintf(stderr, ", %d dropped", es.enStat_Rdrops);
136 	if (es.enStat_Reads > 0)
137 		fprintf(stderr, ", %d tcpdump %s", es.enStat_Reads,
138 				es.enStat_Reads > 1 ? "reads" : "read");
139 	if (es.enStat_MaxRead > 1)
140 		fprintf(stderr, ", %d packets in largest read",
141 			es.enStat_MaxRead);
142 	putc('\n', stderr);
143 #endif	/* IBMRTPC */
144 	close(fd);
145 }
146 
147 int
148 initdevice(char *device, int pflag, int *linktype)
149 {
150 	struct eniocb ctl;
151 	struct enfilter filter;
152 	u_int maxwaiting;
153 	int if_fd;
154 
155 #ifdef	IBMRTPC
156 	GETENETDEVICE(0, O_RDONLY, &if_fd);
157 #else	/* !IBMRTPC */
158 	if_fd = open("/dev/enet", O_RDONLY, 0);
159 #endif	/* IBMRTPC */
160 
161 	if (if_fd == -1) {
162 		perror("tcpdump: enet open error");
163 		error(
164 "your system may not be properly configured; see \"man enet(4)\"");
165 		exit(-1);
166 	}
167 
168 	/*  Get operating parameters. */
169 
170 	if (ioctl(if_fd, EIOCGETP, (char *)&ctl) == -1) {
171 		perror("tcpdump: enet ioctl EIOCGETP error");
172 		exit(-1);
173 	}
174 
175 	/*  Set operating parameters. */
176 
177 #ifdef	IBMRTPC
178 	ctl.en_rtout = 1 * ctl.en_hz;
179 	ctl.en_tr_etherhead = 1;
180 	ctl.en_tap_network = 1;
181 	ctl.en_multi_packet = 1;
182 	ctl.en_maxlen = BUFSPACE;
183 #else	/* !IBMRTPC */
184 	ctl.en_rtout = 64;	/* randomly picked value for HZ */
185 #endif	/* IBMRTPC */
186 	if (ioctl(if_fd, EIOCSETP, &ctl) == -1) {
187 		perror("tcpdump: enet ioctl EIOCSETP error");
188 		exit(-1);
189 	}
190 
191 	/*  Flush the receive queue, since we've changed
192 	    the operating parameters and we otherwise might
193 	    receive data without headers. */
194 
195 	if (ioctl(if_fd, EIOCFLUSH) == -1) {
196 		perror("tcpdump: enet ioctl EIOCFLUSH error");
197 		exit(-1);
198 	}
199 
200 	/*  Set the receive queue depth to its maximum. */
201 
202 	maxwaiting = ctl.en_maxwaiting;
203 	if (ioctl(if_fd, EIOCSETW, &maxwaiting) == -1) {
204 		perror("tcpdump: enet ioctl EIOCSETW error");
205 		exit(-1);
206 	}
207 
208 #ifdef	IBMRTPC
209 	/*  Clear statistics. */
210 
211 	if (ioctl(if_fd, EIOCLRSTAT, 0) == -1) {
212 		perror("tcpdump: enet ioctl EIOCLRSTAT error");
213 		exit(-1);
214 	}
215 #endif	/* IBMRTPC */
216 
217 	/*  Set the filter (accept all packets). */
218 
219 	filter.enf_Priority = 3;
220 	filter.enf_FilterLen = 0;
221 	if (ioctl(if_fd, EIOCSETF, &filter) == -1) {
222 		perror("tcpdump: enet ioctl EIOCSETF error");
223 		exit(-1);
224 	}
225 	/*
226 	 * "enetfilter" supports only ethernets.
227 	 */
228 	*linktype = DLT_EN10MB;
229 
230 	return(if_fd);
231 }
232