1 /* 2 * cOPyright (c) 2004 Jeffrey M. Hsu. All rights reserved. 3 * Copyright (c) 2004 The DragonFly Project. All rights reserved. 4 * 5 * This code is derived from software contributed to The DragonFly Project 6 * by Jeffrey M. Hsu. 7 * 8 * Redistribution and use in source and binary forms, with or without 9 * modification, are permitted provided that the following conditions 10 * are met: 11 * 1. Redistributions of source code must retain the above copyright 12 * notice, this list of conditions and the following disclaimer. 13 * 2. Redistributions in binary form must reproduce the above copyright 14 * notice, this list of conditions and the following disclaimer in the 15 * documentation and/or other materials provided with the distribution. 16 * 3. Neither the name of The DragonFly Project nor the names of its 17 * contributors may be used to endorse or promote products derived 18 * from this software without specific, prior written permission. 19 * 20 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 21 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 22 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS 23 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE 24 * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, 25 * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING, 26 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 27 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED 28 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, 29 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT 30 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 31 * SUCH DAMAGE. 32 */ 33 34 /* 35 * Copyright (c) 2004 Jeffrey M. Hsu. All rights reserved. 36 * 37 * License terms: all terms for the DragonFly license above plus the following: 38 * 39 * 4. All advertising materials mentioning features or use of this software 40 * must display the following acknowledgement: 41 * 42 * This product includes software developed by Jeffrey M. Hsu 43 * for the DragonFly Project. 44 * 45 * This requirement may be waived with permission from Jeffrey Hsu. 46 * This requirement will sunset and may be removed on July 8 2005, 47 * after which the standard DragonFly license (as shown above) will 48 * apply. 49 */ 50 51 /* 52 * Copyright (c) 1982, 1986, 1991, 1993, 1995 53 * The Regents of the University of California. All rights reserved. 54 * 55 * Redistribution and use in source and binary forms, with or without 56 * modification, are permitted provided that the following conditions 57 * are met: 58 * 1. Redistributions of source code must retain the above copyright 59 * notice, this list of conditions and the following disclaimer. 60 * 2. Redistributions in binary form must reproduce the above copyright 61 * notice, this list of conditions and the following disclaimer in the 62 * documentation and/or other materials provided with the distribution. 63 * 3. All advertising materials mentioning features or use of this software 64 * must display the following acknowledgement: 65 * This product includes software developed by the University of 66 * California, Berkeley and its contributors. 67 * 4. Neither the name of the University nor the names of its contributors 68 * may be used to endorse or promote products derived from this software 69 * without specific prior written permission. 70 * 71 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 72 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 73 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 74 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 75 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 76 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 77 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 78 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 79 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 80 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 81 * SUCH DAMAGE. 82 * 83 * @(#)in_pcb.c 8.4 (Berkeley) 5/24/95 84 * $FreeBSD: src/sys/netinet/in_pcb.c,v 1.59.2.27 2004/01/02 04:06:42 ambrisko Exp $ 85 * $DragonFly: src/sys/netinet/in_pcb.c,v 1.34 2005/03/04 03:48:25 hsu Exp $ 86 */ 87 88 #include "opt_ipsec.h" 89 #include "opt_inet6.h" 90 91 #include <sys/param.h> 92 #include <sys/systm.h> 93 #include <sys/malloc.h> 94 #include <sys/mbuf.h> 95 #include <sys/domain.h> 96 #include <sys/protosw.h> 97 #include <sys/socket.h> 98 #include <sys/socketvar.h> 99 #include <sys/proc.h> 100 #include <sys/jail.h> 101 #include <sys/kernel.h> 102 #include <sys/sysctl.h> 103 104 #include <machine/limits.h> 105 106 #include <vm/vm_zone.h> 107 108 #include <net/if.h> 109 #include <net/if_types.h> 110 #include <net/route.h> 111 112 #include <netinet/in.h> 113 #include <netinet/in_pcb.h> 114 #include <netinet/in_var.h> 115 #include <netinet/ip_var.h> 116 #ifdef INET6 117 #include <netinet/ip6.h> 118 #include <netinet6/ip6_var.h> 119 #endif /* INET6 */ 120 121 #ifdef IPSEC 122 #include <netinet6/ipsec.h> 123 #include <netproto/key/key.h> 124 #endif 125 126 #ifdef FAST_IPSEC 127 #if defined(IPSEC) || defined(IPSEC_ESP) 128 #error "Bad idea: don't compile with both IPSEC and FAST_IPSEC!" 129 #endif 130 131 #include <netproto/ipsec/ipsec.h> 132 #include <netproto/ipsec/key.h> 133 #define IPSEC 134 #endif /* FAST_IPSEC */ 135 136 struct in_addr zeroin_addr; 137 138 /* 139 * These configure the range of local port addresses assigned to 140 * "unspecified" outgoing connections/packets/whatever. 141 */ 142 int ipport_lowfirstauto = IPPORT_RESERVED - 1; /* 1023 */ 143 int ipport_lowlastauto = IPPORT_RESERVEDSTART; /* 600 */ 144 145 int ipport_firstauto = IPPORT_RESERVED; /* 1024 */ 146 int ipport_lastauto = IPPORT_USERRESERVED; /* 5000 */ 147 148 int ipport_hifirstauto = IPPORT_HIFIRSTAUTO; /* 49152 */ 149 int ipport_hilastauto = IPPORT_HILASTAUTO; /* 65535 */ 150 151 static __inline void 152 RANGECHK(int var, int min, int max) 153 { 154 if (var < min) 155 var = min; 156 else if (var > max) 157 var = max; 158 } 159 160 static int 161 sysctl_net_ipport_check(SYSCTL_HANDLER_ARGS) 162 { 163 int error; 164 165 error = sysctl_handle_int(oidp, oidp->oid_arg1, oidp->oid_arg2, req); 166 if (!error) { 167 RANGECHK(ipport_lowfirstauto, 1, IPPORT_RESERVED - 1); 168 RANGECHK(ipport_lowlastauto, 1, IPPORT_RESERVED - 1); 169 170 RANGECHK(ipport_firstauto, IPPORT_RESERVED, USHRT_MAX); 171 RANGECHK(ipport_lastauto, IPPORT_RESERVED, USHRT_MAX); 172 173 RANGECHK(ipport_hifirstauto, IPPORT_RESERVED, USHRT_MAX); 174 RANGECHK(ipport_hilastauto, IPPORT_RESERVED, USHRT_MAX); 175 } 176 return (error); 177 } 178 179 SYSCTL_NODE(_net_inet_ip, IPPROTO_IP, portrange, CTLFLAG_RW, 0, "IP Ports"); 180 181 SYSCTL_PROC(_net_inet_ip_portrange, OID_AUTO, lowfirst, CTLTYPE_INT|CTLFLAG_RW, 182 &ipport_lowfirstauto, 0, &sysctl_net_ipport_check, "I", ""); 183 SYSCTL_PROC(_net_inet_ip_portrange, OID_AUTO, lowlast, CTLTYPE_INT|CTLFLAG_RW, 184 &ipport_lowlastauto, 0, &sysctl_net_ipport_check, "I", ""); 185 SYSCTL_PROC(_net_inet_ip_portrange, OID_AUTO, first, CTLTYPE_INT|CTLFLAG_RW, 186 &ipport_firstauto, 0, &sysctl_net_ipport_check, "I", ""); 187 SYSCTL_PROC(_net_inet_ip_portrange, OID_AUTO, last, CTLTYPE_INT|CTLFLAG_RW, 188 &ipport_lastauto, 0, &sysctl_net_ipport_check, "I", ""); 189 SYSCTL_PROC(_net_inet_ip_portrange, OID_AUTO, hifirst, CTLTYPE_INT|CTLFLAG_RW, 190 &ipport_hifirstauto, 0, &sysctl_net_ipport_check, "I", ""); 191 SYSCTL_PROC(_net_inet_ip_portrange, OID_AUTO, hilast, CTLTYPE_INT|CTLFLAG_RW, 192 &ipport_hilastauto, 0, &sysctl_net_ipport_check, "I", ""); 193 194 /* 195 * in_pcb.c: manage the Protocol Control Blocks. 196 * 197 * NOTE: It is assumed that most of these functions will be called at 198 * splnet(). XXX - There are, unfortunately, a few exceptions to this 199 * rule that should be fixed. 200 * 201 * NOTE: The caller should initialize the cpu field to the cpu running the 202 * protocol stack associated with this inpcbinfo. 203 */ 204 205 void 206 in_pcbinfo_init(struct inpcbinfo *pcbinfo) 207 { 208 LIST_INIT(&pcbinfo->pcblisthead); 209 pcbinfo->cpu = -1; 210 } 211 212 /* 213 * Allocate a PCB and associate it with the socket. 214 */ 215 int 216 in_pcballoc(struct socket *so, struct inpcbinfo *pcbinfo) 217 { 218 struct inpcb *inp; 219 #ifdef IPSEC 220 int error; 221 #endif 222 223 inp = zalloc(pcbinfo->ipi_zone); 224 if (inp == NULL) 225 return (ENOBUFS); 226 bzero(inp, sizeof *inp); 227 inp->inp_gencnt = ++pcbinfo->ipi_gencnt; 228 inp->inp_pcbinfo = inp->inp_cpcbinfo = pcbinfo; 229 inp->inp_socket = so; 230 #ifdef IPSEC 231 error = ipsec_init_policy(so, &inp->inp_sp); 232 if (error != 0) { 233 zfree(pcbinfo->ipi_zone, inp); 234 return (error); 235 } 236 #endif 237 #ifdef INET6 238 if (INP_SOCKAF(so) == AF_INET6 && ip6_v6only) 239 inp->inp_flags |= IN6P_IPV6_V6ONLY; 240 if (ip6_auto_flowlabel) 241 inp->inp_flags |= IN6P_AUTOFLOWLABEL; 242 #endif 243 so->so_pcb = inp; 244 LIST_INSERT_HEAD(&pcbinfo->pcblisthead, inp, inp_list); 245 pcbinfo->ipi_count++; 246 return (0); 247 } 248 249 int 250 in_pcbbind(struct inpcb *inp, struct sockaddr *nam, struct thread *td) 251 { 252 struct socket *so = inp->inp_socket; 253 struct proc *p = td->td_proc; 254 unsigned short *lastport; 255 struct sockaddr_in *sin; 256 struct inpcbinfo *pcbinfo = inp->inp_pcbinfo; 257 u_short lport = 0; 258 int wild = 0, reuseport = (so->so_options & SO_REUSEPORT); 259 int error, prison = 0; 260 261 KKASSERT(p); 262 263 if (TAILQ_EMPTY(&in_ifaddrhead)) /* XXX broken! */ 264 return (EADDRNOTAVAIL); 265 if (inp->inp_lport != 0 || inp->inp_laddr.s_addr != INADDR_ANY) 266 return (EINVAL); /* already bound */ 267 if (!(so->so_options & (SO_REUSEADDR|SO_REUSEPORT))) 268 wild = 1; /* neither SO_REUSEADDR nor SO_REUSEPORT is set */ 269 if (nam != NULL) { 270 sin = (struct sockaddr_in *)nam; 271 if (nam->sa_len != sizeof *sin) 272 return (EINVAL); 273 #ifdef notdef 274 /* 275 * We should check the family, but old programs 276 * incorrectly fail to initialize it. 277 */ 278 if (sin->sin_family != AF_INET) 279 return (EAFNOSUPPORT); 280 #endif 281 if (sin->sin_addr.s_addr != INADDR_ANY && 282 prison_ip(td, 0, &sin->sin_addr.s_addr)) 283 return (EINVAL); 284 lport = sin->sin_port; 285 if (IN_MULTICAST(ntohl(sin->sin_addr.s_addr))) { 286 /* 287 * Treat SO_REUSEADDR as SO_REUSEPORT for multicast; 288 * allow complete duplication of binding if 289 * SO_REUSEPORT is set, or if SO_REUSEADDR is set 290 * and a multicast address is bound on both 291 * new and duplicated sockets. 292 */ 293 if (so->so_options & SO_REUSEADDR) 294 reuseport = SO_REUSEADDR | SO_REUSEPORT; 295 } else if (sin->sin_addr.s_addr != INADDR_ANY) { 296 sin->sin_port = 0; /* yech... */ 297 bzero(&sin->sin_zero, sizeof sin->sin_zero); 298 if (ifa_ifwithaddr((struct sockaddr *)sin) == NULL) 299 return (EADDRNOTAVAIL); 300 } 301 if (lport != 0) { 302 struct inpcb *t; 303 304 /* GROSS */ 305 if (ntohs(lport) < IPPORT_RESERVED && 306 p && suser_cred(p->p_ucred, PRISON_ROOT)) 307 return (EACCES); 308 if (p && p->p_ucred->cr_prison) 309 prison = 1; 310 if (so->so_cred->cr_uid != 0 && 311 !IN_MULTICAST(ntohl(sin->sin_addr.s_addr))) { 312 t = in_pcblookup_local(inp->inp_pcbinfo, 313 sin->sin_addr, lport, 314 prison ? 0 : INPLOOKUP_WILDCARD); 315 if (t && 316 (!in_nullhost(sin->sin_addr) || 317 !in_nullhost(t->inp_laddr) || 318 (t->inp_socket->so_options & 319 SO_REUSEPORT) == 0) && 320 (so->so_cred->cr_uid != 321 t->inp_socket->so_cred->cr_uid)) { 322 #ifdef INET6 323 if (!in_nullhost(sin->sin_addr) || 324 !in_nullhost(t->inp_laddr) || 325 INP_SOCKAF(so) == 326 INP_SOCKAF(t->inp_socket)) 327 #endif 328 return (EADDRINUSE); 329 } 330 } 331 if (prison && prison_ip(td, 0, &sin->sin_addr.s_addr)) 332 return (EADDRNOTAVAIL); 333 t = in_pcblookup_local(pcbinfo, sin->sin_addr, 334 lport, prison ? 0 : wild); 335 if (t && !(reuseport & t->inp_socket->so_options)) { 336 #ifdef INET6 337 if (!in_nullhost(sin->sin_addr) || 338 !in_nullhost(t->inp_laddr) || 339 INP_SOCKAF(so) == INP_SOCKAF(t->inp_socket)) 340 #endif 341 return (EADDRINUSE); 342 } 343 } 344 inp->inp_laddr = sin->sin_addr; 345 } 346 if (lport == 0) { 347 ushort first, last; 348 int count; 349 350 if (inp->inp_laddr.s_addr != INADDR_ANY && 351 prison_ip(td, 0, &inp->inp_laddr.s_addr )) { 352 inp->inp_laddr.s_addr = INADDR_ANY; 353 return (EINVAL); 354 } 355 inp->inp_flags |= INP_ANONPORT; 356 357 if (inp->inp_flags & INP_HIGHPORT) { 358 first = ipport_hifirstauto; /* sysctl */ 359 last = ipport_hilastauto; 360 lastport = &pcbinfo->lasthi; 361 } else if (inp->inp_flags & INP_LOWPORT) { 362 if (p && 363 (error = suser_cred(p->p_ucred, PRISON_ROOT))) { 364 inp->inp_laddr.s_addr = INADDR_ANY; 365 return (error); 366 } 367 first = ipport_lowfirstauto; /* 1023 */ 368 last = ipport_lowlastauto; /* 600 */ 369 lastport = &pcbinfo->lastlow; 370 } else { 371 first = ipport_firstauto; /* sysctl */ 372 last = ipport_lastauto; 373 lastport = &pcbinfo->lastport; 374 } 375 /* 376 * Simple check to ensure all ports are not used up causing 377 * a deadlock here. 378 * 379 * We split the two cases (up and down) so that the direction 380 * is not being tested on each round of the loop. 381 */ 382 if (first > last) { 383 /* 384 * counting down 385 */ 386 count = first - last; 387 388 do { 389 if (count-- < 0) { /* completely used? */ 390 inp->inp_laddr.s_addr = INADDR_ANY; 391 return (EADDRNOTAVAIL); 392 } 393 --*lastport; 394 if (*lastport > first || *lastport < last) 395 *lastport = first; 396 lport = htons(*lastport); 397 } while (in_pcblookup_local(pcbinfo, 398 inp->inp_laddr, lport, wild)); 399 } else { 400 /* 401 * counting up 402 */ 403 count = last - first; 404 405 do { 406 if (count-- < 0) { /* completely used? */ 407 inp->inp_laddr.s_addr = INADDR_ANY; 408 return (EADDRNOTAVAIL); 409 } 410 ++*lastport; 411 if (*lastport < first || *lastport > last) 412 *lastport = first; 413 lport = htons(*lastport); 414 } while (in_pcblookup_local(pcbinfo, 415 inp->inp_laddr, lport, wild)); 416 } 417 } 418 inp->inp_lport = lport; 419 if (prison_ip(td, 0, &inp->inp_laddr.s_addr)) { 420 inp->inp_laddr.s_addr = INADDR_ANY; 421 inp->inp_lport = 0; 422 return (EINVAL); 423 } 424 if (in_pcbinsporthash(inp) != 0) { 425 inp->inp_laddr.s_addr = INADDR_ANY; 426 inp->inp_lport = 0; 427 return (EAGAIN); 428 } 429 return (0); 430 } 431 432 /* 433 * Transform old in_pcbconnect() into an inner subroutine for new 434 * in_pcbconnect(): Do some validity-checking on the remote 435 * address (in mbuf 'nam') and then determine local host address 436 * (i.e., which interface) to use to access that remote host. 437 * 438 * This preserves definition of in_pcbconnect(), while supporting a 439 * slightly different version for T/TCP. (This is more than 440 * a bit of a kludge, but cleaning up the internal interfaces would 441 * have forced minor changes in every protocol). 442 */ 443 int 444 in_pcbladdr(inp, nam, plocal_sin) 445 struct inpcb *inp; 446 struct sockaddr *nam; 447 struct sockaddr_in **plocal_sin; 448 { 449 struct in_ifaddr *ia; 450 struct sockaddr_in *sin = (struct sockaddr_in *)nam; 451 452 if (nam->sa_len != sizeof *sin) 453 return (EINVAL); 454 if (sin->sin_family != AF_INET) 455 return (EAFNOSUPPORT); 456 if (sin->sin_port == 0) 457 return (EADDRNOTAVAIL); 458 if (!TAILQ_EMPTY(&in_ifaddrhead)) { 459 ia = TAILQ_FIRST(&in_ifaddrhead); 460 /* 461 * If the destination address is INADDR_ANY, 462 * use the primary local address. 463 * If the supplied address is INADDR_BROADCAST, 464 * and the primary interface supports broadcast, 465 * choose the broadcast address for that interface. 466 */ 467 if (sin->sin_addr.s_addr == INADDR_ANY) 468 sin->sin_addr = IA_SIN(ia)->sin_addr; 469 else if (sin->sin_addr.s_addr == (u_long)INADDR_BROADCAST && 470 (ia->ia_ifp->if_flags & IFF_BROADCAST)) 471 sin->sin_addr = satosin(&ia->ia_broadaddr)->sin_addr; 472 } 473 if (inp->inp_laddr.s_addr == INADDR_ANY) { 474 struct route *ro; 475 476 ia = (struct in_ifaddr *)NULL; 477 /* 478 * If route is known or can be allocated now, 479 * our src addr is taken from the i/f, else punt. 480 * Note that we should check the address family of the cached 481 * destination, in case of sharing the cache with IPv6. 482 */ 483 ro = &inp->inp_route; 484 if (ro->ro_rt && 485 (!(ro->ro_rt->rt_flags & RTF_UP) || 486 ro->ro_dst.sa_family != AF_INET || 487 satosin(&ro->ro_dst)->sin_addr.s_addr != 488 sin->sin_addr.s_addr || 489 inp->inp_socket->so_options & SO_DONTROUTE)) { 490 RTFREE(ro->ro_rt); 491 ro->ro_rt = (struct rtentry *)NULL; 492 } 493 if (!(inp->inp_socket->so_options & SO_DONTROUTE) && /*XXX*/ 494 (ro->ro_rt == (struct rtentry *)NULL || 495 ro->ro_rt->rt_ifp == (struct ifnet *)NULL)) { 496 /* No route yet, so try to acquire one */ 497 bzero(&ro->ro_dst, sizeof(struct sockaddr_in)); 498 ro->ro_dst.sa_family = AF_INET; 499 ro->ro_dst.sa_len = sizeof(struct sockaddr_in); 500 ((struct sockaddr_in *) &ro->ro_dst)->sin_addr = 501 sin->sin_addr; 502 rtalloc(ro); 503 } 504 /* 505 * If we found a route, use the address 506 * corresponding to the outgoing interface 507 * unless it is the loopback (in case a route 508 * to our address on another net goes to loopback). 509 */ 510 if (ro->ro_rt && !(ro->ro_rt->rt_ifp->if_flags & IFF_LOOPBACK)) 511 ia = ifatoia(ro->ro_rt->rt_ifa); 512 if (ia == NULL) { 513 u_short fport = sin->sin_port; 514 515 sin->sin_port = 0; 516 ia = ifatoia(ifa_ifwithdstaddr(sintosa(sin))); 517 if (ia == NULL) 518 ia = ifatoia(ifa_ifwithnet(sintosa(sin))); 519 sin->sin_port = fport; 520 if (ia == NULL) 521 ia = TAILQ_FIRST(&in_ifaddrhead); 522 if (ia == NULL) 523 return (EADDRNOTAVAIL); 524 } 525 /* 526 * If the destination address is multicast and an outgoing 527 * interface has been set as a multicast option, use the 528 * address of that interface as our source address. 529 */ 530 if (IN_MULTICAST(ntohl(sin->sin_addr.s_addr)) && 531 inp->inp_moptions != NULL) { 532 struct ip_moptions *imo; 533 struct ifnet *ifp; 534 535 imo = inp->inp_moptions; 536 if (imo->imo_multicast_ifp != NULL) { 537 ifp = imo->imo_multicast_ifp; 538 TAILQ_FOREACH(ia, &in_ifaddrhead, ia_link) 539 if (ia->ia_ifp == ifp) 540 break; 541 if (ia == NULL) 542 return (EADDRNOTAVAIL); 543 } 544 } 545 /* 546 * Don't do pcblookup call here; return interface in plocal_sin 547 * and exit to caller, that will do the lookup. 548 */ 549 *plocal_sin = &ia->ia_addr; 550 551 } 552 return (0); 553 } 554 555 /* 556 * Outer subroutine: 557 * Connect from a socket to a specified address. 558 * Both address and port must be specified in argument sin. 559 * If don't have a local address for this socket yet, 560 * then pick one. 561 */ 562 int 563 in_pcbconnect(struct inpcb *inp, struct sockaddr *nam, struct thread *td) 564 { 565 struct sockaddr_in *if_sin; 566 struct sockaddr_in *sin = (struct sockaddr_in *)nam; 567 struct sockaddr_in sa; 568 struct ucred *cr = td->td_proc ? td->td_proc->p_ucred : NULL; 569 int error; 570 571 if (cr && cr->cr_prison != NULL && in_nullhost(inp->inp_laddr)) { 572 bzero(&sa, sizeof sa); 573 sa.sin_addr.s_addr = htonl(cr->cr_prison->pr_ip); 574 sa.sin_len = sizeof sa; 575 sa.sin_family = AF_INET; 576 error = in_pcbbind(inp, (struct sockaddr *)&sa, td); 577 if (error) 578 return (error); 579 } 580 581 /* Call inner routine to assign local interface address. */ 582 if ((error = in_pcbladdr(inp, nam, &if_sin)) != 0) 583 return (error); 584 585 if (in_pcblookup_hash(inp->inp_cpcbinfo, sin->sin_addr, sin->sin_port, 586 inp->inp_laddr.s_addr ? inp->inp_laddr : if_sin->sin_addr, 587 inp->inp_lport, FALSE, NULL) != NULL) { 588 return (EADDRINUSE); 589 } 590 if (inp->inp_laddr.s_addr == INADDR_ANY) { 591 if (inp->inp_lport == 0) { 592 error = in_pcbbind(inp, (struct sockaddr *)NULL, td); 593 if (error) 594 return (error); 595 } 596 inp->inp_laddr = if_sin->sin_addr; 597 } 598 inp->inp_faddr = sin->sin_addr; 599 inp->inp_fport = sin->sin_port; 600 in_pcbinsconnhash(inp); 601 return (0); 602 } 603 604 void 605 in_pcbdisconnect(inp) 606 struct inpcb *inp; 607 { 608 609 inp->inp_faddr.s_addr = INADDR_ANY; 610 inp->inp_fport = 0; 611 in_pcbremconnhash(inp); 612 if (inp->inp_socket->so_state & SS_NOFDREF) 613 in_pcbdetach(inp); 614 } 615 616 void 617 in_pcbdetach(inp) 618 struct inpcb *inp; 619 { 620 struct socket *so = inp->inp_socket; 621 struct inpcbinfo *ipi = inp->inp_pcbinfo; 622 623 #ifdef IPSEC 624 ipsec4_delete_pcbpolicy(inp); 625 #endif /*IPSEC*/ 626 inp->inp_gencnt = ++ipi->ipi_gencnt; 627 in_pcbremlists(inp); 628 so->so_pcb = 0; 629 sofree(so); 630 if (inp->inp_options) 631 m_free(inp->inp_options); 632 if (inp->inp_route.ro_rt) 633 rtfree(inp->inp_route.ro_rt); 634 ip_freemoptions(inp->inp_moptions); 635 inp->inp_vflag = 0; 636 zfree(ipi->ipi_zone, inp); 637 } 638 639 /* 640 * The calling convention of in_setsockaddr() and in_setpeeraddr() was 641 * modified to match the pru_sockaddr() and pru_peeraddr() entry points 642 * in struct pr_usrreqs, so that protocols can just reference then directly 643 * without the need for a wrapper function. The socket must have a valid 644 * (i.e., non-nil) PCB, but it should be impossible to get an invalid one 645 * except through a kernel programming error, so it is acceptable to panic 646 * (or in this case trap) if the PCB is invalid. (Actually, we don't trap 647 * because there actually /is/ a programming error somewhere... XXX) 648 */ 649 int 650 in_setsockaddr(so, nam) 651 struct socket *so; 652 struct sockaddr **nam; 653 { 654 int s; 655 struct inpcb *inp; 656 struct sockaddr_in *sin; 657 658 /* 659 * Do the malloc first in case it blocks. 660 */ 661 MALLOC(sin, struct sockaddr_in *, sizeof *sin, M_SONAME, 662 M_WAITOK | M_ZERO); 663 sin->sin_family = AF_INET; 664 sin->sin_len = sizeof *sin; 665 666 s = splnet(); 667 inp = so->so_pcb; 668 if (!inp) { 669 splx(s); 670 free(sin, M_SONAME); 671 return (ECONNRESET); 672 } 673 sin->sin_port = inp->inp_lport; 674 sin->sin_addr = inp->inp_laddr; 675 splx(s); 676 677 *nam = (struct sockaddr *)sin; 678 return (0); 679 } 680 681 int 682 in_setpeeraddr(so, nam) 683 struct socket *so; 684 struct sockaddr **nam; 685 { 686 int s; 687 struct inpcb *inp; 688 struct sockaddr_in *sin; 689 690 /* 691 * Do the malloc first in case it blocks. 692 */ 693 MALLOC(sin, struct sockaddr_in *, sizeof *sin, M_SONAME, 694 M_WAITOK | M_ZERO); 695 sin->sin_family = AF_INET; 696 sin->sin_len = sizeof *sin; 697 698 s = splnet(); 699 inp = so->so_pcb; 700 if (!inp) { 701 splx(s); 702 free(sin, M_SONAME); 703 return (ECONNRESET); 704 } 705 sin->sin_port = inp->inp_fport; 706 sin->sin_addr = inp->inp_faddr; 707 splx(s); 708 709 *nam = (struct sockaddr *)sin; 710 return (0); 711 } 712 713 void 714 in_pcbnotifyall(head, faddr, errno, notify) 715 struct inpcbhead *head; 716 struct in_addr faddr; 717 void (*notify) (struct inpcb *, int); 718 { 719 struct inpcb *inp, *ninp; 720 int s; 721 722 /* 723 * note: if INP_PLACEMARKER is set we must ignore the rest of 724 * the structure and skip it. 725 */ 726 s = splnet(); 727 LIST_FOREACH_MUTABLE(inp, head, inp_list, ninp) { 728 if (inp->inp_flags & INP_PLACEMARKER) 729 continue; 730 #ifdef INET6 731 if (!(inp->inp_vflag & INP_IPV4)) 732 continue; 733 #endif 734 if (inp->inp_faddr.s_addr != faddr.s_addr || 735 inp->inp_socket == NULL) 736 continue; 737 (*notify)(inp, errno); /* can remove inp from list! */ 738 } 739 splx(s); 740 } 741 742 void 743 in_pcbpurgeif0(head, ifp) 744 struct inpcb *head; 745 struct ifnet *ifp; 746 { 747 struct inpcb *inp; 748 struct ip_moptions *imo; 749 int i, gap; 750 751 for (inp = head; inp != NULL; inp = LIST_NEXT(inp, inp_list)) { 752 if (inp->inp_flags & INP_PLACEMARKER) 753 continue; 754 imo = inp->inp_moptions; 755 if ((inp->inp_vflag & INP_IPV4) && imo != NULL) { 756 /* 757 * Unselect the outgoing interface if it is being 758 * detached. 759 */ 760 if (imo->imo_multicast_ifp == ifp) 761 imo->imo_multicast_ifp = NULL; 762 763 /* 764 * Drop multicast group membership if we joined 765 * through the interface being detached. 766 */ 767 for (i = 0, gap = 0; i < imo->imo_num_memberships; 768 i++) { 769 if (imo->imo_membership[i]->inm_ifp == ifp) { 770 in_delmulti(imo->imo_membership[i]); 771 gap++; 772 } else if (gap != 0) 773 imo->imo_membership[i - gap] = 774 imo->imo_membership[i]; 775 } 776 imo->imo_num_memberships -= gap; 777 } 778 } 779 } 780 781 /* 782 * Check for alternatives when higher level complains 783 * about service problems. For now, invalidate cached 784 * routing information. If the route was created dynamically 785 * (by a redirect), time to try a default gateway again. 786 */ 787 void 788 in_losing(struct inpcb *inp) 789 { 790 struct rtentry *rt; 791 struct rt_addrinfo rtinfo; 792 793 if ((rt = inp->inp_route.ro_rt)) { 794 bzero(&rtinfo, sizeof(struct rt_addrinfo)); 795 rtinfo.rti_info[RTAX_DST] = rt_key(rt); 796 rtinfo.rti_info[RTAX_GATEWAY] = rt->rt_gateway; 797 rtinfo.rti_info[RTAX_NETMASK] = rt_mask(rt); 798 rtinfo.rti_flags = rt->rt_flags; 799 rt_missmsg(RTM_LOSING, &rtinfo, rt->rt_flags, 0); 800 if (rt->rt_flags & RTF_DYNAMIC) 801 rtrequest1(RTM_DELETE, &rtinfo, NULL); 802 inp->inp_route.ro_rt = NULL; 803 rtfree(rt); 804 /* 805 * A new route can be allocated 806 * the next time output is attempted. 807 */ 808 } 809 } 810 811 /* 812 * After a routing change, flush old routing 813 * and allocate a (hopefully) better one. 814 */ 815 void 816 in_rtchange(inp, errno) 817 struct inpcb *inp; 818 int errno; 819 { 820 if (inp->inp_route.ro_rt) { 821 rtfree(inp->inp_route.ro_rt); 822 inp->inp_route.ro_rt = NULL; 823 /* 824 * A new route can be allocated the next time 825 * output is attempted. 826 */ 827 } 828 } 829 830 /* 831 * Lookup a PCB based on the local address and port. 832 */ 833 struct inpcb * 834 in_pcblookup_local(pcbinfo, laddr, lport_arg, wild_okay) 835 struct inpcbinfo *pcbinfo; 836 struct in_addr laddr; 837 u_int lport_arg; 838 int wild_okay; 839 { 840 struct inpcb *inp; 841 int matchwild = 3, wildcard; 842 u_short lport = lport_arg; 843 844 struct inpcbporthead *porthash; 845 struct inpcbport *phd; 846 struct inpcb *match = NULL; 847 848 /* 849 * Best fit PCB lookup. 850 * 851 * First see if this local port is in use by looking on the 852 * port hash list. 853 */ 854 porthash = &pcbinfo->porthashbase[INP_PCBPORTHASH(lport, 855 pcbinfo->porthashmask)]; 856 LIST_FOREACH(phd, porthash, phd_hash) { 857 if (phd->phd_port == lport) 858 break; 859 } 860 if (phd != NULL) { 861 /* 862 * Port is in use by one or more PCBs. Look for best 863 * fit. 864 */ 865 LIST_FOREACH(inp, &phd->phd_pcblist, inp_portlist) { 866 wildcard = 0; 867 #ifdef INET6 868 if ((inp->inp_vflag & INP_IPV4) == 0) 869 continue; 870 #endif 871 if (inp->inp_faddr.s_addr != INADDR_ANY) 872 wildcard++; 873 if (inp->inp_laddr.s_addr != INADDR_ANY) { 874 if (laddr.s_addr == INADDR_ANY) 875 wildcard++; 876 else if (inp->inp_laddr.s_addr != laddr.s_addr) 877 continue; 878 } else { 879 if (laddr.s_addr != INADDR_ANY) 880 wildcard++; 881 } 882 if (wildcard && !wild_okay) 883 continue; 884 if (wildcard < matchwild) { 885 match = inp; 886 matchwild = wildcard; 887 if (matchwild == 0) { 888 break; 889 } 890 } 891 } 892 } 893 return (match); 894 } 895 896 /* 897 * Lookup PCB in hash list. 898 */ 899 struct inpcb * 900 in_pcblookup_hash(pcbinfo, faddr, fport_arg, laddr, lport_arg, wildcard, ifp) 901 struct inpcbinfo *pcbinfo; 902 struct in_addr faddr, laddr; 903 u_int fport_arg, lport_arg; 904 boolean_t wildcard; 905 struct ifnet *ifp; 906 { 907 struct inpcbhead *head; 908 struct inpcb *inp; 909 u_short fport = fport_arg, lport = lport_arg; 910 911 /* 912 * First look for an exact match. 913 */ 914 head = &pcbinfo->hashbase[INP_PCBCONNHASH(faddr.s_addr, fport, 915 laddr.s_addr, lport, pcbinfo->hashmask)]; 916 LIST_FOREACH(inp, head, inp_hash) { 917 #ifdef INET6 918 if (!(inp->inp_vflag & INP_IPV4)) 919 continue; 920 #endif 921 if (in_hosteq(inp->inp_faddr, faddr) && 922 in_hosteq(inp->inp_laddr, laddr) && 923 inp->inp_fport == fport && inp->inp_lport == lport) { 924 /* found */ 925 return (inp); 926 } 927 } 928 929 if (wildcard) { 930 struct inpcb *local_wild = NULL; 931 #ifdef INET6 932 struct inpcb *local_wild_mapped = NULL; 933 #endif 934 struct inpcontainer *ic; 935 struct inpcontainerhead *chead; 936 937 chead = &pcbinfo->wildcardhashbase[ 938 INP_PCBWILDCARDHASH(lport, pcbinfo->wildcardhashmask)]; 939 LIST_FOREACH(ic, chead, ic_list) { 940 inp = ic->ic_inp; 941 #ifdef INET6 942 if (!(inp->inp_vflag & INP_IPV4)) 943 continue; 944 #endif 945 if (inp->inp_lport == lport) { 946 if (ifp && ifp->if_type == IFT_FAITH && 947 !(inp->inp_flags & INP_FAITH)) 948 continue; 949 if (inp->inp_laddr.s_addr == laddr.s_addr) 950 return (inp); 951 if (inp->inp_laddr.s_addr == INADDR_ANY) { 952 #ifdef INET6 953 if (INP_CHECK_SOCKAF(inp->inp_socket, 954 AF_INET6)) 955 local_wild_mapped = inp; 956 else 957 #endif 958 local_wild = inp; 959 } 960 } 961 } 962 #ifdef INET6 963 if (local_wild == NULL) 964 return (local_wild_mapped); 965 #endif 966 return (local_wild); 967 } 968 969 /* 970 * Not found. 971 */ 972 return (NULL); 973 } 974 975 /* 976 * Insert PCB into connection hash table. 977 */ 978 void 979 in_pcbinsconnhash(struct inpcb *inp) 980 { 981 struct inpcbinfo *pcbinfo = inp->inp_cpcbinfo; 982 struct inpcbhead *bucket; 983 u_int32_t hashkey_faddr, hashkey_laddr; 984 985 #ifdef INET6 986 if (inp->inp_vflag & INP_IPV6) { 987 hashkey_faddr = inp->in6p_faddr.s6_addr32[3] /* XXX JH */; 988 hashkey_laddr = inp->in6p_laddr.s6_addr32[3] /* XXX JH */; 989 } else { 990 #endif 991 hashkey_faddr = inp->inp_faddr.s_addr; 992 hashkey_laddr = inp->inp_laddr.s_addr; 993 #ifdef INET6 994 } 995 #endif 996 997 KASSERT(!(inp->inp_flags & INP_CONNECTED), ("already on hash list")); 998 inp->inp_flags |= INP_CONNECTED; 999 1000 /* 1001 * Insert into the connection hash table. 1002 */ 1003 bucket = &pcbinfo->hashbase[INP_PCBCONNHASH(hashkey_faddr, 1004 inp->inp_fport, hashkey_laddr, inp->inp_lport, pcbinfo->hashmask)]; 1005 LIST_INSERT_HEAD(bucket, inp, inp_hash); 1006 } 1007 1008 /* 1009 * Remove PCB from connection hash table. 1010 */ 1011 void 1012 in_pcbremconnhash(struct inpcb *inp) 1013 { 1014 KASSERT(inp->inp_flags & INP_CONNECTED, ("inp not connected")); 1015 LIST_REMOVE(inp, inp_hash); 1016 inp->inp_flags &= ~INP_CONNECTED; 1017 } 1018 1019 /* 1020 * Insert PCB into port hash table. 1021 */ 1022 int 1023 in_pcbinsporthash(struct inpcb *inp) 1024 { 1025 struct inpcbinfo *pcbinfo = inp->inp_pcbinfo; 1026 struct inpcbporthead *pcbporthash; 1027 struct inpcbport *phd; 1028 1029 /* 1030 * Insert into the port hash table. 1031 */ 1032 pcbporthash = &pcbinfo->porthashbase[ 1033 INP_PCBPORTHASH(inp->inp_lport, pcbinfo->porthashmask)]; 1034 1035 /* Go through port list and look for a head for this lport. */ 1036 LIST_FOREACH(phd, pcbporthash, phd_hash) 1037 if (phd->phd_port == inp->inp_lport) 1038 break; 1039 1040 /* If none exists, malloc one and tack it on. */ 1041 if (phd == NULL) { 1042 MALLOC(phd, struct inpcbport *, sizeof(struct inpcbport), 1043 M_PCB, M_INTWAIT | M_NULLOK); 1044 if (phd == NULL) 1045 return (ENOBUFS); /* XXX */ 1046 phd->phd_port = inp->inp_lport; 1047 LIST_INIT(&phd->phd_pcblist); 1048 LIST_INSERT_HEAD(pcbporthash, phd, phd_hash); 1049 } 1050 1051 inp->inp_phd = phd; 1052 LIST_INSERT_HEAD(&phd->phd_pcblist, inp, inp_portlist); 1053 1054 return (0); 1055 } 1056 1057 void 1058 in_pcbinswildcardhash_oncpu(struct inpcb *inp, struct inpcbinfo *pcbinfo) 1059 { 1060 struct inpcontainer *ic; 1061 struct inpcontainerhead *bucket; 1062 1063 bucket = &pcbinfo->wildcardhashbase[ 1064 INP_PCBWILDCARDHASH(inp->inp_lport, pcbinfo->wildcardhashmask)]; 1065 1066 ic = malloc(sizeof(struct inpcontainer), M_TEMP, M_INTWAIT); 1067 ic->ic_inp = inp; 1068 LIST_INSERT_HEAD(bucket, ic, ic_list); 1069 } 1070 1071 /* 1072 * Insert PCB into wildcard hash table. 1073 */ 1074 void 1075 in_pcbinswildcardhash(struct inpcb *inp) 1076 { 1077 struct inpcbinfo *pcbinfo = inp->inp_pcbinfo; 1078 1079 KKASSERT(pcbinfo != NULL); 1080 1081 in_pcbinswildcardhash_oncpu(inp, pcbinfo); 1082 inp->inp_flags |= INP_WILDCARD; 1083 } 1084 1085 void 1086 in_pcbremwildcardhash_oncpu(struct inpcb *inp, struct inpcbinfo *pcbinfo) 1087 { 1088 struct inpcontainer *ic; 1089 struct inpcontainerhead *head; 1090 1091 /* find bucket */ 1092 head = &pcbinfo->wildcardhashbase[ 1093 INP_PCBWILDCARDHASH(inp->inp_lport, pcbinfo->wildcardhashmask)]; 1094 1095 LIST_FOREACH(ic, head, ic_list) { 1096 if (ic->ic_inp == inp) 1097 goto found; 1098 } 1099 return; /* not found! */ 1100 1101 found: 1102 LIST_REMOVE(ic, ic_list); /* remove container from bucket chain */ 1103 free(ic, M_TEMP); /* deallocate container */ 1104 } 1105 1106 /* 1107 * Remove PCB from wildcard hash table. 1108 */ 1109 void 1110 in_pcbremwildcardhash(struct inpcb *inp) 1111 { 1112 struct inpcbinfo *pcbinfo = inp->inp_pcbinfo; 1113 1114 KASSERT(inp->inp_flags & INP_WILDCARD, ("inp not wildcard")); 1115 in_pcbremwildcardhash_oncpu(inp, pcbinfo); 1116 inp->inp_flags &= ~INP_WILDCARD; 1117 } 1118 1119 /* 1120 * Remove PCB from various lists. 1121 */ 1122 void 1123 in_pcbremlists(inp) 1124 struct inpcb *inp; 1125 { 1126 if (inp->inp_lport) { 1127 struct inpcbport *phd = inp->inp_phd; 1128 1129 LIST_REMOVE(inp, inp_portlist); 1130 if (LIST_FIRST(&phd->phd_pcblist) == NULL) { 1131 LIST_REMOVE(phd, phd_hash); 1132 free(phd, M_PCB); 1133 } 1134 } 1135 if (inp->inp_flags & INP_WILDCARD) { 1136 in_pcbremwildcardhash(inp); 1137 } else if (inp->inp_flags & INP_CONNECTED) { 1138 in_pcbremconnhash(inp); 1139 } 1140 LIST_REMOVE(inp, inp_list); 1141 inp->inp_pcbinfo->ipi_count--; 1142 } 1143 1144 int 1145 prison_xinpcb(struct thread *td, struct inpcb *inp) 1146 { 1147 struct ucred *cr; 1148 1149 if (td->td_proc == NULL) 1150 return (0); 1151 cr = td->td_proc->p_ucred; 1152 if (cr->cr_prison == NULL) 1153 return (0); 1154 if (ntohl(inp->inp_laddr.s_addr) == cr->cr_prison->pr_ip) 1155 return (0); 1156 return (1); 1157 } 1158 1159 int 1160 in_pcblist_global(SYSCTL_HANDLER_ARGS) 1161 { 1162 struct inpcbinfo *pcbinfo = arg1; 1163 struct inpcb *inp, *marker; 1164 struct xinpcb xi; 1165 int error, i, n; 1166 inp_gen_t gencnt; 1167 1168 /* 1169 * The process of preparing the TCB list is too time-consuming and 1170 * resource-intensive to repeat twice on every request. 1171 */ 1172 if (req->oldptr == NULL) { 1173 n = pcbinfo->ipi_count; 1174 req->oldidx = (n + n/8 + 10) * sizeof(struct xinpcb); 1175 return 0; 1176 } 1177 1178 if (req->newptr != NULL) 1179 return EPERM; 1180 1181 /* 1182 * OK, now we're committed to doing something. Re-fetch ipi_count 1183 * after obtaining the generation count. 1184 */ 1185 gencnt = pcbinfo->ipi_gencnt; 1186 n = pcbinfo->ipi_count; 1187 1188 marker = malloc(sizeof(struct inpcb), M_TEMP, M_WAITOK|M_ZERO); 1189 marker->inp_flags |= INP_PLACEMARKER; 1190 LIST_INSERT_HEAD(&pcbinfo->pcblisthead, marker, inp_list); 1191 1192 i = 0; 1193 error = 0; 1194 1195 while ((inp = LIST_NEXT(marker, inp_list)) != NULL && i < n) { 1196 LIST_REMOVE(marker, inp_list); 1197 LIST_INSERT_AFTER(inp, marker, inp_list); 1198 1199 if (inp->inp_flags & INP_PLACEMARKER) 1200 continue; 1201 if (inp->inp_gencnt > gencnt) 1202 continue; 1203 if (prison_xinpcb(req->td, inp)) 1204 continue; 1205 xi.xi_len = sizeof xi; 1206 bcopy(inp, &xi.xi_inp, sizeof *inp); 1207 if (inp->inp_socket) 1208 sotoxsocket(inp->inp_socket, &xi.xi_socket); 1209 if ((error = SYSCTL_OUT(req, &xi, sizeof xi)) != 0) 1210 break; 1211 ++i; 1212 } 1213 LIST_REMOVE(marker, inp_list); 1214 if (error == 0 && i < n) { 1215 bzero(&xi, sizeof xi); 1216 xi.xi_len = sizeof xi; 1217 while (i < n) { 1218 error = SYSCTL_OUT(req, &xi, sizeof xi); 1219 ++i; 1220 } 1221 } 1222 free(marker, M_TEMP); 1223 return(error); 1224 } 1225