1 /* $OpenBSD: bpf.c,v 1.20 2007/01/08 02:51:13 krw Exp $ */ 2 /* $DragonFly: src/sbin/dhclient/bpf.c,v 1.2 2008/11/05 14:08:41 sephe Exp $ */ 3 4 /* BPF socket interface code, originally contributed by Archie Cobbs. */ 5 6 /* 7 * Copyright (c) 1995, 1996, 1998, 1999 8 * The Internet Software Consortium. All rights reserved. 9 * 10 * Redistribution and use in source and binary forms, with or without 11 * modification, are permitted provided that the following conditions 12 * are met: 13 * 14 * 1. Redistributions of source code must retain the above copyright 15 * notice, this list of conditions and the following disclaimer. 16 * 2. Redistributions in binary form must reproduce the above copyright 17 * notice, this list of conditions and the following disclaimer in the 18 * documentation and/or other materials provided with the distribution. 19 * 3. Neither the name of The Internet Software Consortium nor the names 20 * of its contributors may be used to endorse or promote products derived 21 * from this software without specific prior written permission. 22 * 23 * THIS SOFTWARE IS PROVIDED BY THE INTERNET SOFTWARE CONSORTIUM AND 24 * CONTRIBUTORS ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, 25 * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF 26 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 27 * DISCLAIMED. IN NO EVENT SHALL THE INTERNET SOFTWARE CONSORTIUM OR 28 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 29 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 30 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF 31 * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND 32 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, 33 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT 34 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 35 * SUCH DAMAGE. 36 * 37 * This software has been written for the Internet Software Consortium 38 * by Ted Lemon <mellon@fugue.com> in cooperation with Vixie 39 * Enterprises. To learn more about the Internet Software Consortium, 40 * see ``http://www.vix.com/isc''. To learn more about Vixie 41 * Enterprises, see ``http://www.vix.com''. 42 */ 43 44 #include <sys/ioctl.h> 45 #include <sys/uio.h> 46 47 #include <net/bpf.h> 48 #include <netinet/if_ether.h> 49 #include <netinet/in_systm.h> 50 #include <netinet/ip.h> 51 #include <netinet/udp.h> 52 53 #include "dhcpd.h" 54 55 #define BPF_FORMAT "/dev/bpf%d" 56 57 /* 58 * Called by get_interface_list for each interface that's discovered. 59 * Opens a packet filter for each interface and adds it to the select 60 * mask. 61 */ 62 int 63 if_register_bpf(void) 64 { 65 char filename[50]; 66 int sock, b; 67 68 /* Open a BPF device */ 69 for (b = 0; 1; b++) { 70 snprintf(filename, sizeof(filename), BPF_FORMAT, b); 71 sock = open(filename, O_RDWR, 0); 72 if (sock < 0) { 73 if (errno == EBUSY) 74 continue; 75 else 76 error("Can't find free bpf: %m"); 77 } else 78 break; 79 } 80 81 /* Set the BPF device to point at this interface. */ 82 if (ioctl(sock, BIOCSETIF, ifi->ifp) < 0) 83 error("Can't attach interface %s to bpf device %s: %m", 84 ifi->name, filename); 85 86 return (sock); 87 } 88 89 void 90 if_register_send(void) 91 { 92 int sock, on = 1; 93 94 /* 95 * If we're using the bpf API for sending and receiving, we 96 * don't need to register this interface twice. 97 */ 98 ifi->wfdesc = ifi->rfdesc; 99 100 /* 101 * Use raw socket for unicast send. 102 */ 103 if ((sock = socket(AF_INET, SOCK_RAW, IPPROTO_UDP)) == -1) 104 error("socket(SOCK_RAW): %m"); 105 if (setsockopt(sock, IPPROTO_IP, IP_HDRINCL, &on, 106 sizeof(on)) == -1) 107 error("setsockopt(IP_HDRINCL): %m"); 108 ifi->ufdesc = sock; 109 } 110 111 /* 112 * Packet filter program... 113 * 114 * XXX: Changes to the filter program may require changes to the 115 * constant offsets used in if_register_send to patch the BPF program! 116 */ 117 struct bpf_insn dhcp_bpf_filter[] = { 118 /* Make sure this is an IP packet... */ 119 BPF_STMT(BPF_LD + BPF_H + BPF_ABS, 12), 120 BPF_JUMP(BPF_JMP + BPF_JEQ + BPF_K, ETHERTYPE_IP, 0, 8), 121 122 /* Make sure it's a UDP packet... */ 123 BPF_STMT(BPF_LD + BPF_B + BPF_ABS, 23), 124 BPF_JUMP(BPF_JMP + BPF_JEQ + BPF_K, IPPROTO_UDP, 0, 6), 125 126 /* Make sure this isn't a fragment... */ 127 BPF_STMT(BPF_LD + BPF_H + BPF_ABS, 20), 128 BPF_JUMP(BPF_JMP + BPF_JSET + BPF_K, 0x1fff, 4, 0), 129 130 /* Get the IP header length... */ 131 BPF_STMT(BPF_LDX + BPF_B + BPF_MSH, 14), 132 133 /* Make sure it's to the right port... */ 134 BPF_STMT(BPF_LD + BPF_H + BPF_IND, 16), 135 BPF_JUMP(BPF_JMP + BPF_JEQ + BPF_K, 67, 0, 1), /* patch */ 136 137 /* If we passed all the tests, ask for the whole packet. */ 138 BPF_STMT(BPF_RET+BPF_K, (u_int)-1), 139 140 /* Otherwise, drop it. */ 141 BPF_STMT(BPF_RET+BPF_K, 0), 142 }; 143 144 int dhcp_bpf_filter_len = sizeof(dhcp_bpf_filter) / sizeof(struct bpf_insn); 145 146 /* 147 * Packet write filter program: 148 * 'ip and udp and src port bootps and dst port (bootps or bootpc)' 149 */ 150 struct bpf_insn dhcp_bpf_wfilter[] = { 151 BPF_STMT(BPF_LD + BPF_B + BPF_IND, 14), 152 BPF_JUMP(BPF_JMP + BPF_JEQ + BPF_K, (IPVERSION << 4) + 5, 0, 12), 153 154 /* Make sure this is an IP packet... */ 155 BPF_STMT(BPF_LD + BPF_H + BPF_ABS, 12), 156 BPF_JUMP(BPF_JMP + BPF_JEQ + BPF_K, ETHERTYPE_IP, 0, 10), 157 158 /* Make sure it's a UDP packet... */ 159 BPF_STMT(BPF_LD + BPF_B + BPF_ABS, 23), 160 BPF_JUMP(BPF_JMP + BPF_JEQ + BPF_K, IPPROTO_UDP, 0, 8), 161 162 /* Make sure this isn't a fragment... */ 163 BPF_STMT(BPF_LD + BPF_H + BPF_ABS, 20), 164 BPF_JUMP(BPF_JMP + BPF_JSET + BPF_K, 0x1fff, 6, 0), /* patched */ 165 166 /* Get the IP header length... */ 167 BPF_STMT(BPF_LDX + BPF_B + BPF_MSH, 14), 168 169 /* Make sure it's from the right port... */ 170 BPF_STMT(BPF_LD + BPF_H + BPF_IND, 14), 171 BPF_JUMP(BPF_JMP + BPF_JEQ + BPF_K, 68, 0, 3), 172 173 /* Make sure it is to the right ports ... */ 174 BPF_STMT(BPF_LD + BPF_H + BPF_IND, 16), 175 BPF_JUMP(BPF_JMP + BPF_JEQ + BPF_K, 67, 0, 1), 176 177 /* If we passed all the tests, ask for the whole packet. */ 178 BPF_STMT(BPF_RET+BPF_K, (u_int)-1), 179 180 /* Otherwise, drop it. */ 181 BPF_STMT(BPF_RET+BPF_K, 0), 182 }; 183 184 int dhcp_bpf_wfilter_len = sizeof(dhcp_bpf_wfilter) / sizeof(struct bpf_insn); 185 186 void 187 if_register_receive(void) 188 { 189 struct bpf_version v; 190 struct bpf_program p; 191 int flag = 1, sz; 192 193 /* Open a BPF device and hang it on this interface... */ 194 ifi->rfdesc = if_register_bpf(); 195 196 /* Make sure the BPF version is in range... */ 197 if (ioctl(ifi->rfdesc, BIOCVERSION, &v) < 0) 198 error("Can't get BPF version: %m"); 199 200 if (v.bv_major != BPF_MAJOR_VERSION || 201 v.bv_minor < BPF_MINOR_VERSION) 202 error("Kernel BPF version out of range - recompile dhcpd!"); 203 204 /* 205 * Set immediate mode so that reads return as soon as a packet 206 * comes in, rather than waiting for the input buffer to fill 207 * with packets. 208 */ 209 if (ioctl(ifi->rfdesc, BIOCIMMEDIATE, &flag) < 0) 210 error("Can't set immediate mode on bpf device: %m"); 211 212 /*if (ioctl(ifi->rfdesc, BIOCSFILDROP, &flag) < 0) 213 error("Can't set filter-drop mode on bpf device: %m");*/ 214 215 /* Get the required BPF buffer length from the kernel. */ 216 if (ioctl(ifi->rfdesc, BIOCGBLEN, &sz) < 0) 217 error("Can't get bpf buffer length: %m"); 218 ifi->rbuf_max = sz; 219 ifi->rbuf = malloc(ifi->rbuf_max); 220 if (!ifi->rbuf) 221 error("Can't allocate %lu bytes for bpf input buffer.", 222 (unsigned long)ifi->rbuf_max); 223 ifi->rbuf_offset = 0; 224 ifi->rbuf_len = 0; 225 226 /* Set up the bpf filter program structure. */ 227 p.bf_len = dhcp_bpf_filter_len; 228 p.bf_insns = dhcp_bpf_filter; 229 230 /* Patch the server port into the BPF program... 231 * 232 * XXX: changes to filter program may require changes to the 233 * insn number(s) used below! 234 */ 235 dhcp_bpf_filter[8].k = LOCAL_PORT; 236 237 if (ioctl(ifi->rfdesc, BIOCSETF, &p) < 0) 238 error("Can't install packet filter program: %m"); 239 240 /* Set up the bpf write filter program structure. */ 241 p.bf_len = dhcp_bpf_wfilter_len; 242 p.bf_insns = dhcp_bpf_wfilter; 243 244 if (dhcp_bpf_wfilter[7].k == 0x1fff) 245 dhcp_bpf_wfilter[7].k = htons(IP_MF|IP_OFFMASK); 246 247 if (ioctl(ifi->rfdesc, BIOCSETWF, &p) < 0) 248 error("Can't install write filter program: %m"); 249 250 if (ioctl(ifi->rfdesc, BIOCLOCK, NULL) < 0) 251 error("Cannot lock bpf"); 252 } 253 254 ssize_t 255 send_packet(struct in_addr from, struct sockaddr_in *to, 256 struct hardware *hto) 257 { 258 #define IOVCNT 2 259 unsigned char buf[256]; 260 struct iovec iov[IOVCNT]; 261 struct msghdr msg; 262 int result, bufp = 0; 263 264 if (to->sin_addr.s_addr == INADDR_BROADCAST) { 265 assemble_hw_header(buf, &bufp, hto); 266 } 267 268 assemble_udp_ip_header(buf, &bufp, from.s_addr, 269 to->sin_addr.s_addr, to->sin_port, 270 (unsigned char *)&client->packet, 271 client->packet_length); 272 273 iov[0].iov_base = (char *)buf; 274 iov[0].iov_len = bufp; 275 iov[1].iov_base = (char *)&client->packet; 276 iov[1].iov_len = client->packet_length; 277 278 if (to->sin_addr.s_addr == INADDR_BROADCAST) { 279 result = writev(ifi->wfdesc, iov, IOVCNT); 280 } else { 281 struct ip *ip = (struct ip *)buf; 282 283 /* 284 * DragonFly's raw socket expects ip_len/ip_off 285 * in host byte order. 286 */ 287 ip->ip_len = ntohs(ip->ip_len); 288 ip->ip_off = ntohs(ip->ip_off); 289 290 memset(&msg, 0, sizeof(msg)); 291 msg.msg_name = (struct sockaddr *)to; 292 msg.msg_namelen = sizeof(*to); 293 msg.msg_iov = iov; 294 msg.msg_iovlen = IOVCNT; 295 result = sendmsg(ifi->ufdesc, &msg, 0); 296 } 297 298 if (result == -1) 299 warning("send_packet: %m"); 300 return (result); 301 } 302 303 ssize_t 304 receive_packet(struct sockaddr_in *from, struct hardware *hfrom) 305 { 306 int length = 0, offset = 0; 307 struct bpf_hdr hdr; 308 309 /* 310 * All this complexity is because BPF doesn't guarantee that 311 * only one packet will be returned at a time. We're getting 312 * what we deserve, though - this is a terrible abuse of the BPF 313 * interface. Sigh. 314 */ 315 316 /* Process packets until we get one we can return or until we've 317 * done a read and gotten nothing we can return... 318 */ 319 do { 320 /* If the buffer is empty, fill it. */ 321 if (ifi->rbuf_offset == ifi->rbuf_len) { 322 length = read(ifi->rfdesc, ifi->rbuf, ifi->rbuf_max); 323 if (length <= 0) 324 return (length); 325 ifi->rbuf_offset = 0; 326 ifi->rbuf_len = BPF_WORDALIGN(length); 327 } 328 329 /* 330 * If there isn't room for a whole bpf header, something 331 * went wrong, but we'll ignore it and hope it goes 332 * away... XXX 333 */ 334 if (ifi->rbuf_len - ifi->rbuf_offset < sizeof(hdr)) { 335 ifi->rbuf_offset = ifi->rbuf_len; 336 continue; 337 } 338 339 /* Copy out a bpf header... */ 340 memcpy(&hdr, &ifi->rbuf[ifi->rbuf_offset], sizeof(hdr)); 341 342 /* 343 * If the bpf header plus data doesn't fit in what's 344 * left of the buffer, stick head in sand yet again... 345 */ 346 if (ifi->rbuf_offset + hdr.bh_hdrlen + hdr.bh_caplen > 347 ifi->rbuf_len) { 348 ifi->rbuf_offset = ifi->rbuf_len; 349 continue; 350 } 351 352 /* 353 * If the captured data wasn't the whole packet, or if 354 * the packet won't fit in the input buffer, all we can 355 * do is drop it. 356 */ 357 if (hdr.bh_caplen != hdr.bh_datalen) { 358 ifi->rbuf_offset = BPF_WORDALIGN( 359 ifi->rbuf_offset + hdr.bh_hdrlen + 360 hdr.bh_caplen); 361 continue; 362 } 363 364 /* Skip over the BPF header... */ 365 ifi->rbuf_offset += hdr.bh_hdrlen; 366 367 /* Decode the physical header... */ 368 offset = decode_hw_header(ifi->rbuf, ifi->rbuf_offset, hfrom); 369 370 /* 371 * If a physical layer checksum failed (dunno of any 372 * physical layer that supports this, but WTH), skip 373 * this packet. 374 */ 375 if (offset < 0) { 376 ifi->rbuf_offset = BPF_WORDALIGN( 377 ifi->rbuf_offset + hdr.bh_caplen); 378 continue; 379 } 380 ifi->rbuf_offset += offset; 381 hdr.bh_caplen -= offset; 382 383 /* Decode the IP and UDP headers... */ 384 offset = decode_udp_ip_header(ifi->rbuf, 385 ifi->rbuf_offset, from, NULL, hdr.bh_caplen); 386 387 /* If the IP or UDP checksum was bad, skip the packet... */ 388 if (offset < 0) { 389 ifi->rbuf_offset = BPF_WORDALIGN( 390 ifi->rbuf_offset + hdr.bh_caplen); 391 continue; 392 } 393 ifi->rbuf_offset += offset; 394 hdr.bh_caplen -= offset; 395 396 /* 397 * If there's not enough room to stash the packet data, 398 * we have to skip it (this shouldn't happen in real 399 * life, though). 400 */ 401 if (hdr.bh_caplen > sizeof(client->packet)) { 402 ifi->rbuf_offset = BPF_WORDALIGN( 403 ifi->rbuf_offset + hdr.bh_caplen); 404 continue; 405 } 406 407 /* Copy out the data in the packet... */ 408 memset(&client->packet, DHO_END, sizeof(client->packet)); 409 memcpy(&client->packet, ifi->rbuf + ifi->rbuf_offset, 410 hdr.bh_caplen); 411 ifi->rbuf_offset = BPF_WORDALIGN(ifi->rbuf_offset + 412 hdr.bh_caplen); 413 return (hdr.bh_caplen); 414 } while (!length); 415 return (0); 416 } 417