1 /*-
2 * Copyright (C) 1995, 1996, 1997, and 1998 WIDE Project.
3 * Copyright (c) 2010-2011 Juniper Networks, Inc.
4 * All rights reserved.
5 *
6 * Portions of this software were developed by Robert N. M. Watson under
7 * contract to Juniper Networks, Inc.
8 *
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted provided that the following conditions
11 * are met:
12 * 1. Redistributions of source code must retain the above copyright
13 * notice, this list of conditions and the following disclaimer.
14 * 2. Redistributions in binary form must reproduce the above copyright
15 * notice, this list of conditions and the following disclaimer in the
16 * documentation and/or other materials provided with the distribution.
17 * 3. Neither the name of the project nor the names of its contributors
18 * may be used to endorse or promote products derived from this software
19 * without specific prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY THE PROJECT AND CONTRIBUTORS ``AS IS'' AND
22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED. IN NO EVENT SHALL THE PROJECT OR CONTRIBUTORS BE LIABLE
25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31 * SUCH DAMAGE.
32 *
33 * $KAME: in6_pcb.c,v 1.31 2001/05/21 05:45:10 jinmei Exp $
34 */
35
36 /*-
37 * Copyright (c) 1982, 1986, 1991, 1993
38 * The Regents of the University of California. All rights reserved.
39 *
40 * Redistribution and use in source and binary forms, with or without
41 * modification, are permitted provided that the following conditions
42 * are met:
43 * 1. Redistributions of source code must retain the above copyright
44 * notice, this list of conditions and the following disclaimer.
45 * 2. Redistributions in binary form must reproduce the above copyright
46 * notice, this list of conditions and the following disclaimer in the
47 * documentation and/or other materials provided with the distribution.
48 * 4. Neither the name of the University nor the names of its contributors
49 * may be used to endorse or promote products derived from this software
50 * without specific prior written permission.
51 *
52 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
53 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
54 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
55 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
56 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
57 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
58 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
59 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
60 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
61 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
62 * SUCH DAMAGE.
63 *
64 * @(#)in_pcb.c 8.2 (Berkeley) 1/4/94
65 */
66
67 #include <sys/cdefs.h>
68 __FBSDID("$FreeBSD$");
69
70 #include "opt_inet.h"
71 #include "opt_inet6.h"
72 #include "opt_ipsec.h"
73 #include "opt_pcbgroup.h"
74 #include "opt_rss.h"
75
76 #include <sys/param.h>
77 #include <sys/systm.h>
78 #include <sys/malloc.h>
79 #include <sys/mbuf.h>
80 #include <sys/domain.h>
81 #include <sys/protosw.h>
82 #include <sys/socket.h>
83 #include <sys/socketvar.h>
84 #include <sys/sockio.h>
85 #include <sys/errno.h>
86 #include <sys/time.h>
87 #include <sys/priv.h>
88 #include <sys/proc.h>
89 #include <sys/jail.h>
90
91 #include <vm/uma.h>
92
93 #include <net/if.h>
94 #include <net/if_var.h>
95 #include <net/if_llatbl.h>
96 #include <net/if_types.h>
97 #include <net/route.h>
98
99 #include <netinet/in.h>
100 #include <netinet/in_var.h>
101 #include <netinet/in_systm.h>
102 #include <netinet/tcp_var.h>
103 #include <netinet/ip6.h>
104 #include <netinet/ip_var.h>
105
106 #include <netinet6/ip6_var.h>
107 #include <netinet6/nd6.h>
108 #include <netinet/in_pcb.h>
109 #include <netinet6/in6_pcb.h>
110 #include <netinet6/scope6_var.h>
111
112 static struct inpcb *in6_pcblookup_hash_locked(struct inpcbinfo *,
113 struct in6_addr *, u_int, struct in6_addr *, u_int, int, struct ifnet *);
114
115 int
in6_pcbbind(struct inpcb * inp,struct sockaddr * nam,struct ucred * cred)116 in6_pcbbind(struct inpcb *inp, struct sockaddr *nam,
117 struct ucred *cred)
118 {
119 struct socket *so = inp->inp_socket;
120 struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *)NULL;
121 struct inpcbinfo *pcbinfo = inp->inp_pcbinfo;
122 u_short lport = 0;
123 int error, lookupflags = 0;
124 int reuseport = (so->so_options & SO_REUSEPORT);
125
126 INP_WLOCK_ASSERT(inp);
127 INP_HASH_WLOCK_ASSERT(pcbinfo);
128
129 if (TAILQ_EMPTY(&V_in6_ifaddrhead)) /* XXX broken! */
130 return (EADDRNOTAVAIL);
131 if (inp->inp_lport || !IN6_IS_ADDR_UNSPECIFIED(&inp->in6p_laddr))
132 return (EINVAL);
133 if ((so->so_options & (SO_REUSEADDR|SO_REUSEPORT)) == 0)
134 lookupflags = INPLOOKUP_WILDCARD;
135 if (nam == NULL) {
136 if ((error = prison_local_ip6(cred, &inp->in6p_laddr,
137 ((inp->inp_flags & IN6P_IPV6_V6ONLY) != 0))) != 0)
138 return (error);
139 } else {
140 sin6 = (struct sockaddr_in6 *)nam;
141 if (nam->sa_len != sizeof(*sin6))
142 return (EINVAL);
143 /*
144 * family check.
145 */
146 if (nam->sa_family != AF_INET6)
147 return (EAFNOSUPPORT);
148
149 if ((error = sa6_embedscope(sin6, V_ip6_use_defzone)) != 0)
150 return(error);
151
152 if ((error = prison_local_ip6(cred, &sin6->sin6_addr,
153 ((inp->inp_flags & IN6P_IPV6_V6ONLY) != 0))) != 0)
154 return (error);
155
156 lport = sin6->sin6_port;
157 if (IN6_IS_ADDR_MULTICAST(&sin6->sin6_addr)) {
158 /*
159 * Treat SO_REUSEADDR as SO_REUSEPORT for multicast;
160 * allow compepte duplication of binding if
161 * SO_REUSEPORT is set, or if SO_REUSEADDR is set
162 * and a multicast address is bound on both
163 * new and duplicated sockets.
164 */
165 if ((so->so_options & (SO_REUSEADDR|SO_REUSEPORT)) != 0)
166 reuseport = SO_REUSEADDR|SO_REUSEPORT;
167 } else if (!IN6_IS_ADDR_UNSPECIFIED(&sin6->sin6_addr)) {
168 struct ifaddr *ifa;
169
170 sin6->sin6_port = 0; /* yech... */
171 if ((ifa = ifa_ifwithaddr((struct sockaddr *)sin6)) ==
172 NULL &&
173 (inp->inp_flags & INP_BINDANY) == 0) {
174 return (EADDRNOTAVAIL);
175 }
176
177 /*
178 * XXX: bind to an anycast address might accidentally
179 * cause sending a packet with anycast source address.
180 * We should allow to bind to a deprecated address, since
181 * the application dares to use it.
182 */
183 if (ifa != NULL &&
184 ((struct in6_ifaddr *)ifa)->ia6_flags &
185 (IN6_IFF_ANYCAST|IN6_IFF_NOTREADY|IN6_IFF_DETACHED)) {
186 ifa_free(ifa);
187 return (EADDRNOTAVAIL);
188 }
189 if (ifa != NULL)
190 ifa_free(ifa);
191 }
192 if (lport) {
193 struct inpcb *t;
194 struct tcptw *tw;
195
196 /* GROSS */
197 if (ntohs(lport) <= V_ipport_reservedhigh &&
198 ntohs(lport) >= V_ipport_reservedlow &&
199 priv_check_cred(cred, PRIV_NETINET_RESERVEDPORT,
200 0))
201 return (EACCES);
202 if (!IN6_IS_ADDR_MULTICAST(&sin6->sin6_addr) &&
203 priv_check_cred(inp->inp_cred,
204 PRIV_NETINET_REUSEPORT, 0) != 0) {
205 t = in6_pcblookup_local(pcbinfo,
206 &sin6->sin6_addr, lport,
207 INPLOOKUP_WILDCARD, cred);
208 if (t &&
209 ((inp->inp_flags2 & INP_BINDMULTI) == 0) &&
210 ((t->inp_flags & INP_TIMEWAIT) == 0) &&
211 (so->so_type != SOCK_STREAM ||
212 IN6_IS_ADDR_UNSPECIFIED(&t->in6p_faddr)) &&
213 (!IN6_IS_ADDR_UNSPECIFIED(&sin6->sin6_addr) ||
214 !IN6_IS_ADDR_UNSPECIFIED(&t->in6p_laddr) ||
215 (t->inp_flags2 & INP_REUSEPORT) == 0) &&
216 (inp->inp_cred->cr_uid !=
217 t->inp_cred->cr_uid))
218 return (EADDRINUSE);
219
220 /*
221 * If the socket is a BINDMULTI socket, then
222 * the credentials need to match and the
223 * original socket also has to have been bound
224 * with BINDMULTI.
225 */
226 if (t && (! in_pcbbind_check_bindmulti(inp, t)))
227 return (EADDRINUSE);
228
229 #ifdef INET
230 if ((inp->inp_flags & IN6P_IPV6_V6ONLY) == 0 &&
231 IN6_IS_ADDR_UNSPECIFIED(&sin6->sin6_addr)) {
232 struct sockaddr_in sin;
233
234 in6_sin6_2_sin(&sin, sin6);
235 t = in_pcblookup_local(pcbinfo,
236 sin.sin_addr, lport,
237 INPLOOKUP_WILDCARD, cred);
238 if (t &&
239 ((inp->inp_flags2 & INP_BINDMULTI) == 0) &&
240 ((t->inp_flags &
241 INP_TIMEWAIT) == 0) &&
242 (so->so_type != SOCK_STREAM ||
243 ntohl(t->inp_faddr.s_addr) ==
244 INADDR_ANY) &&
245 (inp->inp_cred->cr_uid !=
246 t->inp_cred->cr_uid))
247 return (EADDRINUSE);
248
249 if (t && (! in_pcbbind_check_bindmulti(inp, t)))
250 return (EADDRINUSE);
251 }
252 #endif
253 }
254 t = in6_pcblookup_local(pcbinfo, &sin6->sin6_addr,
255 lport, lookupflags, cred);
256 if (t && (t->inp_flags & INP_TIMEWAIT)) {
257 /*
258 * XXXRW: If an incpb has had its timewait
259 * state recycled, we treat the address as
260 * being in use (for now). This is better
261 * than a panic, but not desirable.
262 */
263 tw = intotw(t);
264 if (tw == NULL ||
265 (reuseport & tw->tw_so_options) == 0)
266 return (EADDRINUSE);
267 } else if (t && (reuseport & inp_so_options(t)) == 0) {
268 return (EADDRINUSE);
269 }
270 #ifdef INET
271 if ((inp->inp_flags & IN6P_IPV6_V6ONLY) == 0 &&
272 IN6_IS_ADDR_UNSPECIFIED(&sin6->sin6_addr)) {
273 struct sockaddr_in sin;
274
275 in6_sin6_2_sin(&sin, sin6);
276 t = in_pcblookup_local(pcbinfo, sin.sin_addr,
277 lport, lookupflags, cred);
278 if (t && t->inp_flags & INP_TIMEWAIT) {
279 tw = intotw(t);
280 if (tw == NULL)
281 return (EADDRINUSE);
282 if ((reuseport & tw->tw_so_options) == 0
283 && (ntohl(t->inp_laddr.s_addr) !=
284 INADDR_ANY || ((inp->inp_vflag &
285 INP_IPV6PROTO) ==
286 (t->inp_vflag & INP_IPV6PROTO))))
287 return (EADDRINUSE);
288 } else if (t &&
289 (reuseport & inp_so_options(t)) == 0 &&
290 (ntohl(t->inp_laddr.s_addr) != INADDR_ANY ||
291 (t->inp_vflag & INP_IPV6PROTO) != 0))
292 return (EADDRINUSE);
293 }
294 #endif
295 }
296 inp->in6p_laddr = sin6->sin6_addr;
297 }
298 if (lport == 0) {
299 if ((error = in6_pcbsetport(&inp->in6p_laddr, inp, cred)) != 0) {
300 /* Undo an address bind that may have occurred. */
301 inp->in6p_laddr = in6addr_any;
302 return (error);
303 }
304 } else {
305 inp->inp_lport = lport;
306 if (in_pcbinshash(inp) != 0) {
307 inp->in6p_laddr = in6addr_any;
308 inp->inp_lport = 0;
309 return (EAGAIN);
310 }
311 }
312 return (0);
313 }
314
315 /*
316 * Transform old in6_pcbconnect() into an inner subroutine for new
317 * in6_pcbconnect(): Do some validity-checking on the remote
318 * address (in mbuf 'nam') and then determine local host address
319 * (i.e., which interface) to use to access that remote host.
320 *
321 * This preserves definition of in6_pcbconnect(), while supporting a
322 * slightly different version for T/TCP. (This is more than
323 * a bit of a kludge, but cleaning up the internal interfaces would
324 * have forced minor changes in every protocol).
325 */
326 static int
in6_pcbladdr(struct inpcb * inp,struct sockaddr * nam,struct in6_addr * plocal_addr6)327 in6_pcbladdr(struct inpcb *inp, struct sockaddr *nam,
328 struct in6_addr *plocal_addr6)
329 {
330 struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *)nam;
331 int error = 0;
332 int scope_ambiguous = 0;
333 struct in6_addr in6a;
334
335 INP_WLOCK_ASSERT(inp);
336 INP_HASH_WLOCK_ASSERT(inp->inp_pcbinfo); /* XXXRW: why? */
337
338 if (nam->sa_len != sizeof (*sin6))
339 return (EINVAL);
340 if (sin6->sin6_family != AF_INET6)
341 return (EAFNOSUPPORT);
342 if (sin6->sin6_port == 0)
343 return (EADDRNOTAVAIL);
344
345 if (sin6->sin6_scope_id == 0 && !V_ip6_use_defzone)
346 scope_ambiguous = 1;
347 if ((error = sa6_embedscope(sin6, V_ip6_use_defzone)) != 0)
348 return(error);
349
350 if (!TAILQ_EMPTY(&V_in6_ifaddrhead)) {
351 /*
352 * If the destination address is UNSPECIFIED addr,
353 * use the loopback addr, e.g ::1.
354 */
355 if (IN6_IS_ADDR_UNSPECIFIED(&sin6->sin6_addr))
356 sin6->sin6_addr = in6addr_loopback;
357 }
358 if ((error = prison_remote_ip6(inp->inp_cred, &sin6->sin6_addr)) != 0)
359 return (error);
360
361 error = in6_selectsrc_socket(sin6, inp->in6p_outputopts,
362 inp, inp->inp_cred, scope_ambiguous, &in6a, NULL);
363 if (error)
364 return (error);
365
366 /*
367 * Do not update this earlier, in case we return with an error.
368 *
369 * XXX: this in6_selectsrc_socket result might replace the bound local
370 * address with the address specified by setsockopt(IPV6_PKTINFO).
371 * Is it the intended behavior?
372 */
373 *plocal_addr6 = in6a;
374
375 /*
376 * Don't do pcblookup call here; return interface in
377 * plocal_addr6
378 * and exit to caller, that will do the lookup.
379 */
380
381 return (0);
382 }
383
384 /*
385 * Outer subroutine:
386 * Connect from a socket to a specified address.
387 * Both address and port must be specified in argument sin.
388 * If don't have a local address for this socket yet,
389 * then pick one.
390 */
391 int
in6_pcbconnect_mbuf(struct inpcb * inp,struct sockaddr * nam,struct ucred * cred,struct mbuf * m)392 in6_pcbconnect_mbuf(struct inpcb *inp, struct sockaddr *nam,
393 struct ucred *cred, struct mbuf *m)
394 {
395 struct inpcbinfo *pcbinfo = inp->inp_pcbinfo;
396 struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *)nam;
397 struct in6_addr addr6;
398 int error;
399
400 INP_WLOCK_ASSERT(inp);
401 INP_HASH_WLOCK_ASSERT(pcbinfo);
402
403 /*
404 * Call inner routine, to assign local interface address.
405 * in6_pcbladdr() may automatically fill in sin6_scope_id.
406 */
407 if ((error = in6_pcbladdr(inp, nam, &addr6)) != 0)
408 return (error);
409
410 if (in6_pcblookup_hash_locked(pcbinfo, &sin6->sin6_addr,
411 sin6->sin6_port,
412 IN6_IS_ADDR_UNSPECIFIED(&inp->in6p_laddr)
413 ? &addr6 : &inp->in6p_laddr,
414 inp->inp_lport, 0, NULL) != NULL) {
415 return (EADDRINUSE);
416 }
417 if (IN6_IS_ADDR_UNSPECIFIED(&inp->in6p_laddr)) {
418 if (inp->inp_lport == 0) {
419 error = in6_pcbbind(inp, (struct sockaddr *)0, cred);
420 if (error)
421 return (error);
422 }
423 inp->in6p_laddr = addr6;
424 }
425 inp->in6p_faddr = sin6->sin6_addr;
426 inp->inp_fport = sin6->sin6_port;
427 /* update flowinfo - draft-itojun-ipv6-flowlabel-api-00 */
428 inp->inp_flow &= ~IPV6_FLOWLABEL_MASK;
429 if (inp->inp_flags & IN6P_AUTOFLOWLABEL)
430 inp->inp_flow |=
431 (htonl(ip6_randomflowlabel()) & IPV6_FLOWLABEL_MASK);
432
433 in_pcbrehash_mbuf(inp, m);
434
435 return (0);
436 }
437
438 int
in6_pcbconnect(struct inpcb * inp,struct sockaddr * nam,struct ucred * cred)439 in6_pcbconnect(struct inpcb *inp, struct sockaddr *nam, struct ucred *cred)
440 {
441
442 return (in6_pcbconnect_mbuf(inp, nam, cred, NULL));
443 }
444
445 void
in6_pcbdisconnect(struct inpcb * inp)446 in6_pcbdisconnect(struct inpcb *inp)
447 {
448
449 INP_WLOCK_ASSERT(inp);
450 INP_HASH_WLOCK_ASSERT(inp->inp_pcbinfo);
451
452 bzero((caddr_t)&inp->in6p_faddr, sizeof(inp->in6p_faddr));
453 inp->inp_fport = 0;
454 /* clear flowinfo - draft-itojun-ipv6-flowlabel-api-00 */
455 inp->inp_flow &= ~IPV6_FLOWLABEL_MASK;
456 in_pcbrehash(inp);
457 }
458
459 struct sockaddr *
in6_sockaddr(in_port_t port,struct in6_addr * addr_p)460 in6_sockaddr(in_port_t port, struct in6_addr *addr_p)
461 {
462 struct sockaddr_in6 *sin6;
463
464 sin6 = malloc(sizeof *sin6, M_SONAME, M_WAITOK);
465 bzero(sin6, sizeof *sin6);
466 sin6->sin6_family = AF_INET6;
467 sin6->sin6_len = sizeof(*sin6);
468 sin6->sin6_port = port;
469 sin6->sin6_addr = *addr_p;
470 (void)sa6_recoverscope(sin6); /* XXX: should catch errors */
471
472 return (struct sockaddr *)sin6;
473 }
474
475 struct sockaddr *
in6_v4mapsin6_sockaddr(in_port_t port,struct in_addr * addr_p)476 in6_v4mapsin6_sockaddr(in_port_t port, struct in_addr *addr_p)
477 {
478 struct sockaddr_in sin;
479 struct sockaddr_in6 *sin6_p;
480
481 bzero(&sin, sizeof sin);
482 sin.sin_family = AF_INET;
483 sin.sin_len = sizeof(sin);
484 sin.sin_port = port;
485 sin.sin_addr = *addr_p;
486
487 sin6_p = malloc(sizeof *sin6_p, M_SONAME,
488 M_WAITOK);
489 in6_sin_2_v4mapsin6(&sin, sin6_p);
490
491 return (struct sockaddr *)sin6_p;
492 }
493
494 int
in6_getsockaddr(struct socket * so,struct sockaddr ** nam)495 in6_getsockaddr(struct socket *so, struct sockaddr **nam)
496 {
497 struct inpcb *inp;
498 struct in6_addr addr;
499 in_port_t port;
500
501 inp = sotoinpcb(so);
502 KASSERT(inp != NULL, ("in6_getsockaddr: inp == NULL"));
503
504 INP_RLOCK(inp);
505 port = inp->inp_lport;
506 addr = inp->in6p_laddr;
507 INP_RUNLOCK(inp);
508
509 *nam = in6_sockaddr(port, &addr);
510 return 0;
511 }
512
513 int
in6_getpeeraddr(struct socket * so,struct sockaddr ** nam)514 in6_getpeeraddr(struct socket *so, struct sockaddr **nam)
515 {
516 struct inpcb *inp;
517 struct in6_addr addr;
518 in_port_t port;
519
520 inp = sotoinpcb(so);
521 KASSERT(inp != NULL, ("in6_getpeeraddr: inp == NULL"));
522
523 INP_RLOCK(inp);
524 port = inp->inp_fport;
525 addr = inp->in6p_faddr;
526 INP_RUNLOCK(inp);
527
528 *nam = in6_sockaddr(port, &addr);
529 return 0;
530 }
531
532 int
in6_mapped_sockaddr(struct socket * so,struct sockaddr ** nam)533 in6_mapped_sockaddr(struct socket *so, struct sockaddr **nam)
534 {
535 struct inpcb *inp;
536 int error;
537
538 inp = sotoinpcb(so);
539 KASSERT(inp != NULL, ("in6_mapped_sockaddr: inp == NULL"));
540
541 #ifdef INET
542 if ((inp->inp_vflag & (INP_IPV4 | INP_IPV6)) == INP_IPV4) {
543 error = in_getsockaddr(so, nam);
544 if (error == 0)
545 in6_sin_2_v4mapsin6_in_sock(nam);
546 } else
547 #endif
548 {
549 /* scope issues will be handled in in6_getsockaddr(). */
550 error = in6_getsockaddr(so, nam);
551 }
552
553 return error;
554 }
555
556 int
in6_mapped_peeraddr(struct socket * so,struct sockaddr ** nam)557 in6_mapped_peeraddr(struct socket *so, struct sockaddr **nam)
558 {
559 struct inpcb *inp;
560 int error;
561
562 inp = sotoinpcb(so);
563 KASSERT(inp != NULL, ("in6_mapped_peeraddr: inp == NULL"));
564
565 #ifdef INET
566 if ((inp->inp_vflag & (INP_IPV4 | INP_IPV6)) == INP_IPV4) {
567 error = in_getpeeraddr(so, nam);
568 if (error == 0)
569 in6_sin_2_v4mapsin6_in_sock(nam);
570 } else
571 #endif
572 /* scope issues will be handled in in6_getpeeraddr(). */
573 error = in6_getpeeraddr(so, nam);
574
575 return error;
576 }
577
578 /*
579 * Pass some notification to all connections of a protocol
580 * associated with address dst. The local address and/or port numbers
581 * may be specified to limit the search. The "usual action" will be
582 * taken, depending on the ctlinput cmd. The caller must filter any
583 * cmds that are uninteresting (e.g., no error in the map).
584 * Call the protocol specific routine (if any) to report
585 * any errors for each matching socket.
586 */
587 void
in6_pcbnotify(struct inpcbinfo * pcbinfo,struct sockaddr * dst,u_int fport_arg,const struct sockaddr * src,u_int lport_arg,int cmd,void * cmdarg,struct inpcb * (* notify)(struct inpcb *,int))588 in6_pcbnotify(struct inpcbinfo *pcbinfo, struct sockaddr *dst,
589 u_int fport_arg, const struct sockaddr *src, u_int lport_arg,
590 int cmd, void *cmdarg,
591 struct inpcb *(*notify)(struct inpcb *, int))
592 {
593 struct inpcb *inp, *inp_temp;
594 struct sockaddr_in6 sa6_src, *sa6_dst;
595 u_short fport = fport_arg, lport = lport_arg;
596 u_int32_t flowinfo;
597 int errno;
598
599 if ((unsigned)cmd >= PRC_NCMDS || dst->sa_family != AF_INET6)
600 return;
601
602 sa6_dst = (struct sockaddr_in6 *)dst;
603 if (IN6_IS_ADDR_UNSPECIFIED(&sa6_dst->sin6_addr))
604 return;
605
606 /*
607 * note that src can be NULL when we get notify by local fragmentation.
608 */
609 sa6_src = (src == NULL) ? sa6_any : *(const struct sockaddr_in6 *)src;
610 flowinfo = sa6_src.sin6_flowinfo;
611
612 /*
613 * Redirects go to all references to the destination,
614 * and use in6_rtchange to invalidate the route cache.
615 * Dead host indications: also use in6_rtchange to invalidate
616 * the cache, and deliver the error to all the sockets.
617 * Otherwise, if we have knowledge of the local port and address,
618 * deliver only to that socket.
619 */
620 if (PRC_IS_REDIRECT(cmd) || cmd == PRC_HOSTDEAD) {
621 fport = 0;
622 lport = 0;
623 bzero((caddr_t)&sa6_src.sin6_addr, sizeof(sa6_src.sin6_addr));
624
625 if (cmd != PRC_HOSTDEAD)
626 notify = in6_rtchange;
627 }
628 errno = inet6ctlerrmap[cmd];
629 INP_INFO_WLOCK(pcbinfo);
630 LIST_FOREACH_SAFE(inp, pcbinfo->ipi_listhead, inp_list, inp_temp) {
631 INP_WLOCK(inp);
632 if ((inp->inp_vflag & INP_IPV6) == 0) {
633 INP_WUNLOCK(inp);
634 continue;
635 }
636
637 /*
638 * If the error designates a new path MTU for a destination
639 * and the application (associated with this socket) wanted to
640 * know the value, notify.
641 * XXX: should we avoid to notify the value to TCP sockets?
642 */
643 if (cmd == PRC_MSGSIZE && cmdarg != NULL)
644 ip6_notify_pmtu(inp, (struct sockaddr_in6 *)dst,
645 *(u_int32_t *)cmdarg);
646
647 /*
648 * Detect if we should notify the error. If no source and
649 * destination ports are specifed, but non-zero flowinfo and
650 * local address match, notify the error. This is the case
651 * when the error is delivered with an encrypted buffer
652 * by ESP. Otherwise, just compare addresses and ports
653 * as usual.
654 */
655 if (lport == 0 && fport == 0 && flowinfo &&
656 inp->inp_socket != NULL &&
657 flowinfo == (inp->inp_flow & IPV6_FLOWLABEL_MASK) &&
658 IN6_ARE_ADDR_EQUAL(&inp->in6p_laddr, &sa6_src.sin6_addr))
659 goto do_notify;
660 else if (!IN6_ARE_ADDR_EQUAL(&inp->in6p_faddr,
661 &sa6_dst->sin6_addr) ||
662 inp->inp_socket == 0 ||
663 (lport && inp->inp_lport != lport) ||
664 (!IN6_IS_ADDR_UNSPECIFIED(&sa6_src.sin6_addr) &&
665 !IN6_ARE_ADDR_EQUAL(&inp->in6p_laddr,
666 &sa6_src.sin6_addr)) ||
667 (fport && inp->inp_fport != fport)) {
668 INP_WUNLOCK(inp);
669 continue;
670 }
671
672 do_notify:
673 if (notify) {
674 if ((*notify)(inp, errno))
675 INP_WUNLOCK(inp);
676 } else
677 INP_WUNLOCK(inp);
678 }
679 INP_INFO_WUNLOCK(pcbinfo);
680 }
681
682 /*
683 * Lookup a PCB based on the local address and port. Caller must hold the
684 * hash lock. No inpcb locks or references are acquired.
685 */
686 struct inpcb *
in6_pcblookup_local(struct inpcbinfo * pcbinfo,struct in6_addr * laddr,u_short lport,int lookupflags,struct ucred * cred)687 in6_pcblookup_local(struct inpcbinfo *pcbinfo, struct in6_addr *laddr,
688 u_short lport, int lookupflags, struct ucred *cred)
689 {
690 struct inpcb *inp;
691 int matchwild = 3, wildcard;
692
693 KASSERT((lookupflags & ~(INPLOOKUP_WILDCARD)) == 0,
694 ("%s: invalid lookup flags %d", __func__, lookupflags));
695
696 INP_HASH_WLOCK_ASSERT(pcbinfo);
697
698 if ((lookupflags & INPLOOKUP_WILDCARD) == 0) {
699 struct inpcbhead *head;
700 /*
701 * Look for an unconnected (wildcard foreign addr) PCB that
702 * matches the local address and port we're looking for.
703 */
704 head = &pcbinfo->ipi_hashbase[INP_PCBHASH(
705 INP6_PCBHASHKEY(&in6addr_any), lport, 0,
706 pcbinfo->ipi_hashmask)];
707 LIST_FOREACH(inp, head, inp_hash) {
708 /* XXX inp locking */
709 if ((inp->inp_vflag & INP_IPV6) == 0)
710 continue;
711 if (IN6_IS_ADDR_UNSPECIFIED(&inp->in6p_faddr) &&
712 IN6_ARE_ADDR_EQUAL(&inp->in6p_laddr, laddr) &&
713 inp->inp_lport == lport) {
714 /* Found. */
715 if (cred == NULL ||
716 prison_equal_ip6(cred->cr_prison,
717 inp->inp_cred->cr_prison))
718 return (inp);
719 }
720 }
721 /*
722 * Not found.
723 */
724 return (NULL);
725 } else {
726 struct inpcbporthead *porthash;
727 struct inpcbport *phd;
728 struct inpcb *match = NULL;
729 /*
730 * Best fit PCB lookup.
731 *
732 * First see if this local port is in use by looking on the
733 * port hash list.
734 */
735 porthash = &pcbinfo->ipi_porthashbase[INP_PCBPORTHASH(lport,
736 pcbinfo->ipi_porthashmask)];
737 LIST_FOREACH(phd, porthash, phd_hash) {
738 if (phd->phd_port == lport)
739 break;
740 }
741 if (phd != NULL) {
742 /*
743 * Port is in use by one or more PCBs. Look for best
744 * fit.
745 */
746 LIST_FOREACH(inp, &phd->phd_pcblist, inp_portlist) {
747 wildcard = 0;
748 if (cred != NULL &&
749 !prison_equal_ip6(cred->cr_prison,
750 inp->inp_cred->cr_prison))
751 continue;
752 /* XXX inp locking */
753 if ((inp->inp_vflag & INP_IPV6) == 0)
754 continue;
755 if (!IN6_IS_ADDR_UNSPECIFIED(&inp->in6p_faddr))
756 wildcard++;
757 if (!IN6_IS_ADDR_UNSPECIFIED(
758 &inp->in6p_laddr)) {
759 if (IN6_IS_ADDR_UNSPECIFIED(laddr))
760 wildcard++;
761 else if (!IN6_ARE_ADDR_EQUAL(
762 &inp->in6p_laddr, laddr))
763 continue;
764 } else {
765 if (!IN6_IS_ADDR_UNSPECIFIED(laddr))
766 wildcard++;
767 }
768 if (wildcard < matchwild) {
769 match = inp;
770 matchwild = wildcard;
771 if (matchwild == 0)
772 break;
773 }
774 }
775 }
776 return (match);
777 }
778 }
779
780 void
in6_pcbpurgeif0(struct inpcbinfo * pcbinfo,struct ifnet * ifp)781 in6_pcbpurgeif0(struct inpcbinfo *pcbinfo, struct ifnet *ifp)
782 {
783 struct inpcb *in6p;
784 struct ip6_moptions *im6o;
785 int i, gap;
786
787 INP_INFO_WLOCK(pcbinfo);
788 LIST_FOREACH(in6p, pcbinfo->ipi_listhead, inp_list) {
789 INP_WLOCK(in6p);
790 im6o = in6p->in6p_moptions;
791 if ((in6p->inp_vflag & INP_IPV6) && im6o != NULL) {
792 /*
793 * Unselect the outgoing ifp for multicast if it
794 * is being detached.
795 */
796 if (im6o->im6o_multicast_ifp == ifp)
797 im6o->im6o_multicast_ifp = NULL;
798 /*
799 * Drop multicast group membership if we joined
800 * through the interface being detached.
801 */
802 gap = 0;
803 for (i = 0; i < im6o->im6o_num_memberships; i++) {
804 if (im6o->im6o_membership[i]->in6m_ifp ==
805 ifp) {
806 in6_mc_leave(im6o->im6o_membership[i],
807 NULL);
808 gap++;
809 } else if (gap != 0) {
810 im6o->im6o_membership[i - gap] =
811 im6o->im6o_membership[i];
812 }
813 }
814 im6o->im6o_num_memberships -= gap;
815 }
816 INP_WUNLOCK(in6p);
817 }
818 INP_INFO_WUNLOCK(pcbinfo);
819 }
820
821 /*
822 * Check for alternatives when higher level complains
823 * about service problems. For now, invalidate cached
824 * routing information. If the route was created dynamically
825 * (by a redirect), time to try a default gateway again.
826 */
827 void
in6_losing(struct inpcb * in6p)828 in6_losing(struct inpcb *in6p)
829 {
830
831 if (in6p->inp_route6.ro_rt) {
832 RTFREE(in6p->inp_route6.ro_rt);
833 in6p->inp_route6.ro_rt = (struct rtentry *)NULL;
834 }
835 if (in6p->inp_route.ro_lle)
836 LLE_FREE(in6p->inp_route.ro_lle); /* zeros ro_lle */
837 return;
838 }
839
840 /*
841 * After a routing change, flush old routing
842 * and allocate a (hopefully) better one.
843 */
844 struct inpcb *
in6_rtchange(struct inpcb * inp,int errno)845 in6_rtchange(struct inpcb *inp, int errno)
846 {
847
848 if (inp->inp_route6.ro_rt) {
849 RTFREE(inp->inp_route6.ro_rt);
850 inp->inp_route6.ro_rt = (struct rtentry *)NULL;
851 }
852 if (inp->inp_route.ro_lle)
853 LLE_FREE(inp->inp_route.ro_lle); /* zeros ro_lle */
854 return inp;
855 }
856
857 #ifdef PCBGROUP
858 /*
859 * Lookup PCB in hash list, using pcbgroup tables.
860 */
861 static struct inpcb *
in6_pcblookup_group(struct inpcbinfo * pcbinfo,struct inpcbgroup * pcbgroup,struct in6_addr * faddr,u_int fport_arg,struct in6_addr * laddr,u_int lport_arg,int lookupflags,struct ifnet * ifp)862 in6_pcblookup_group(struct inpcbinfo *pcbinfo, struct inpcbgroup *pcbgroup,
863 struct in6_addr *faddr, u_int fport_arg, struct in6_addr *laddr,
864 u_int lport_arg, int lookupflags, struct ifnet *ifp)
865 {
866 struct inpcbhead *head;
867 struct inpcb *inp, *tmpinp;
868 u_short fport = fport_arg, lport = lport_arg;
869 bool locked;
870
871 /*
872 * First look for an exact match.
873 */
874 tmpinp = NULL;
875 INP_GROUP_LOCK(pcbgroup);
876 head = &pcbgroup->ipg_hashbase[INP_PCBHASH(
877 INP6_PCBHASHKEY(faddr), lport, fport, pcbgroup->ipg_hashmask)];
878 LIST_FOREACH(inp, head, inp_pcbgrouphash) {
879 /* XXX inp locking */
880 if ((inp->inp_vflag & INP_IPV6) == 0)
881 continue;
882 if (IN6_ARE_ADDR_EQUAL(&inp->in6p_faddr, faddr) &&
883 IN6_ARE_ADDR_EQUAL(&inp->in6p_laddr, laddr) &&
884 inp->inp_fport == fport &&
885 inp->inp_lport == lport) {
886 /*
887 * XXX We should be able to directly return
888 * the inp here, without any checks.
889 * Well unless both bound with SO_REUSEPORT?
890 */
891 if (prison_flag(inp->inp_cred, PR_IP6))
892 goto found;
893 if (tmpinp == NULL)
894 tmpinp = inp;
895 }
896 }
897 if (tmpinp != NULL) {
898 inp = tmpinp;
899 goto found;
900 }
901
902 /*
903 * Then look for a wildcard match in the pcbgroup.
904 */
905 if ((lookupflags & INPLOOKUP_WILDCARD) != 0) {
906 struct inpcb *local_wild = NULL, *local_exact = NULL;
907 struct inpcb *jail_wild = NULL;
908 int injail;
909
910 /*
911 * Order of socket selection - we always prefer jails.
912 * 1. jailed, non-wild.
913 * 2. jailed, wild.
914 * 3. non-jailed, non-wild.
915 * 4. non-jailed, wild.
916 */
917 head = &pcbgroup->ipg_hashbase[
918 INP_PCBHASH(INADDR_ANY, lport, 0, pcbgroup->ipg_hashmask)];
919 LIST_FOREACH(inp, head, inp_pcbgrouphash) {
920 /* XXX inp locking */
921 if ((inp->inp_vflag & INP_IPV6) == 0)
922 continue;
923
924 if (!IN6_IS_ADDR_UNSPECIFIED(&inp->in6p_faddr) ||
925 inp->inp_lport != lport) {
926 continue;
927 }
928
929 injail = prison_flag(inp->inp_cred, PR_IP6);
930 if (injail) {
931 if (prison_check_ip6(inp->inp_cred,
932 laddr) != 0)
933 continue;
934 } else {
935 if (local_exact != NULL)
936 continue;
937 }
938
939 if (IN6_ARE_ADDR_EQUAL(&inp->in6p_laddr, laddr)) {
940 if (injail)
941 goto found;
942 else
943 local_exact = inp;
944 } else if (IN6_IS_ADDR_UNSPECIFIED(&inp->in6p_laddr)) {
945 if (injail)
946 jail_wild = inp;
947 else
948 local_wild = inp;
949 }
950 } /* LIST_FOREACH */
951
952 inp = jail_wild;
953 if (inp == NULL)
954 inp = jail_wild;
955 if (inp == NULL)
956 inp = local_exact;
957 if (inp == NULL)
958 inp = local_wild;
959 if (inp != NULL)
960 goto found;
961 }
962
963 /*
964 * Then look for a wildcard match, if requested.
965 */
966 if ((lookupflags & INPLOOKUP_WILDCARD) != 0) {
967 struct inpcb *local_wild = NULL, *local_exact = NULL;
968 struct inpcb *jail_wild = NULL;
969 int injail;
970
971 /*
972 * Order of socket selection - we always prefer jails.
973 * 1. jailed, non-wild.
974 * 2. jailed, wild.
975 * 3. non-jailed, non-wild.
976 * 4. non-jailed, wild.
977 */
978 head = &pcbinfo->ipi_wildbase[INP_PCBHASH(
979 INP6_PCBHASHKEY(&in6addr_any), lport, 0,
980 pcbinfo->ipi_wildmask)];
981 LIST_FOREACH(inp, head, inp_pcbgroup_wild) {
982 /* XXX inp locking */
983 if ((inp->inp_vflag & INP_IPV6) == 0)
984 continue;
985
986 if (!IN6_IS_ADDR_UNSPECIFIED(&inp->in6p_faddr) ||
987 inp->inp_lport != lport) {
988 continue;
989 }
990
991 injail = prison_flag(inp->inp_cred, PR_IP6);
992 if (injail) {
993 if (prison_check_ip6(inp->inp_cred,
994 laddr) != 0)
995 continue;
996 } else {
997 if (local_exact != NULL)
998 continue;
999 }
1000
1001 if (IN6_ARE_ADDR_EQUAL(&inp->in6p_laddr, laddr)) {
1002 if (injail)
1003 goto found;
1004 else
1005 local_exact = inp;
1006 } else if (IN6_IS_ADDR_UNSPECIFIED(&inp->in6p_laddr)) {
1007 if (injail)
1008 jail_wild = inp;
1009 else
1010 local_wild = inp;
1011 }
1012 } /* LIST_FOREACH */
1013
1014 inp = jail_wild;
1015 if (inp == NULL)
1016 inp = jail_wild;
1017 if (inp == NULL)
1018 inp = local_exact;
1019 if (inp == NULL)
1020 inp = local_wild;
1021 if (inp != NULL)
1022 goto found;
1023 } /* if ((lookupflags & INPLOOKUP_WILDCARD) != 0) */
1024 INP_GROUP_UNLOCK(pcbgroup);
1025 return (NULL);
1026
1027 found:
1028 if (lookupflags & INPLOOKUP_WLOCKPCB)
1029 locked = INP_TRY_WLOCK(inp);
1030 else if (lookupflags & INPLOOKUP_RLOCKPCB)
1031 locked = INP_TRY_RLOCK(inp);
1032 else
1033 panic("%s: locking buf", __func__);
1034 if (!locked)
1035 in_pcbref(inp);
1036 INP_GROUP_UNLOCK(pcbgroup);
1037 if (!locked) {
1038 if (lookupflags & INPLOOKUP_WLOCKPCB) {
1039 INP_WLOCK(inp);
1040 if (in_pcbrele_wlocked(inp))
1041 return (NULL);
1042 } else {
1043 INP_RLOCK(inp);
1044 if (in_pcbrele_rlocked(inp))
1045 return (NULL);
1046 }
1047 }
1048 #ifdef INVARIANTS
1049 if (lookupflags & INPLOOKUP_WLOCKPCB)
1050 INP_WLOCK_ASSERT(inp);
1051 else
1052 INP_RLOCK_ASSERT(inp);
1053 #endif
1054 return (inp);
1055 }
1056 #endif /* PCBGROUP */
1057
1058 /*
1059 * Lookup PCB in hash list.
1060 */
1061 static struct inpcb *
in6_pcblookup_hash_locked(struct inpcbinfo * pcbinfo,struct in6_addr * faddr,u_int fport_arg,struct in6_addr * laddr,u_int lport_arg,int lookupflags,struct ifnet * ifp)1062 in6_pcblookup_hash_locked(struct inpcbinfo *pcbinfo, struct in6_addr *faddr,
1063 u_int fport_arg, struct in6_addr *laddr, u_int lport_arg,
1064 int lookupflags, struct ifnet *ifp)
1065 {
1066 struct inpcbhead *head;
1067 struct inpcb *inp, *tmpinp;
1068 u_short fport = fport_arg, lport = lport_arg;
1069
1070 KASSERT((lookupflags & ~(INPLOOKUP_WILDCARD)) == 0,
1071 ("%s: invalid lookup flags %d", __func__, lookupflags));
1072
1073 INP_HASH_LOCK_ASSERT(pcbinfo);
1074
1075 /*
1076 * First look for an exact match.
1077 */
1078 tmpinp = NULL;
1079 head = &pcbinfo->ipi_hashbase[INP_PCBHASH(
1080 INP6_PCBHASHKEY(faddr), lport, fport, pcbinfo->ipi_hashmask)];
1081 LIST_FOREACH(inp, head, inp_hash) {
1082 /* XXX inp locking */
1083 if ((inp->inp_vflag & INP_IPV6) == 0)
1084 continue;
1085 if (IN6_ARE_ADDR_EQUAL(&inp->in6p_faddr, faddr) &&
1086 IN6_ARE_ADDR_EQUAL(&inp->in6p_laddr, laddr) &&
1087 inp->inp_fport == fport &&
1088 inp->inp_lport == lport) {
1089 /*
1090 * XXX We should be able to directly return
1091 * the inp here, without any checks.
1092 * Well unless both bound with SO_REUSEPORT?
1093 */
1094 if (prison_flag(inp->inp_cred, PR_IP6))
1095 return (inp);
1096 if (tmpinp == NULL)
1097 tmpinp = inp;
1098 }
1099 }
1100 if (tmpinp != NULL)
1101 return (tmpinp);
1102
1103 /*
1104 * Then look for a wildcard match, if requested.
1105 */
1106 if ((lookupflags & INPLOOKUP_WILDCARD) != 0) {
1107 struct inpcb *local_wild = NULL, *local_exact = NULL;
1108 struct inpcb *jail_wild = NULL;
1109 int injail;
1110
1111 /*
1112 * Order of socket selection - we always prefer jails.
1113 * 1. jailed, non-wild.
1114 * 2. jailed, wild.
1115 * 3. non-jailed, non-wild.
1116 * 4. non-jailed, wild.
1117 */
1118 head = &pcbinfo->ipi_hashbase[INP_PCBHASH(
1119 INP6_PCBHASHKEY(&in6addr_any), lport, 0,
1120 pcbinfo->ipi_hashmask)];
1121 LIST_FOREACH(inp, head, inp_hash) {
1122 /* XXX inp locking */
1123 if ((inp->inp_vflag & INP_IPV6) == 0)
1124 continue;
1125
1126 if (!IN6_IS_ADDR_UNSPECIFIED(&inp->in6p_faddr) ||
1127 inp->inp_lport != lport) {
1128 continue;
1129 }
1130
1131 injail = prison_flag(inp->inp_cred, PR_IP6);
1132 if (injail) {
1133 if (prison_check_ip6(inp->inp_cred,
1134 laddr) != 0)
1135 continue;
1136 } else {
1137 if (local_exact != NULL)
1138 continue;
1139 }
1140
1141 if (IN6_ARE_ADDR_EQUAL(&inp->in6p_laddr, laddr)) {
1142 if (injail)
1143 return (inp);
1144 else
1145 local_exact = inp;
1146 } else if (IN6_IS_ADDR_UNSPECIFIED(&inp->in6p_laddr)) {
1147 if (injail)
1148 jail_wild = inp;
1149 else
1150 local_wild = inp;
1151 }
1152 } /* LIST_FOREACH */
1153
1154 if (jail_wild != NULL)
1155 return (jail_wild);
1156 if (local_exact != NULL)
1157 return (local_exact);
1158 if (local_wild != NULL)
1159 return (local_wild);
1160 } /* if ((lookupflags & INPLOOKUP_WILDCARD) != 0) */
1161
1162 /*
1163 * Not found.
1164 */
1165 return (NULL);
1166 }
1167
1168 /*
1169 * Lookup PCB in hash list, using pcbinfo tables. This variation locks the
1170 * hash list lock, and will return the inpcb locked (i.e., requires
1171 * INPLOOKUP_LOCKPCB).
1172 */
1173 static struct inpcb *
in6_pcblookup_hash(struct inpcbinfo * pcbinfo,struct in6_addr * faddr,u_int fport,struct in6_addr * laddr,u_int lport,int lookupflags,struct ifnet * ifp)1174 in6_pcblookup_hash(struct inpcbinfo *pcbinfo, struct in6_addr *faddr,
1175 u_int fport, struct in6_addr *laddr, u_int lport, int lookupflags,
1176 struct ifnet *ifp)
1177 {
1178 struct inpcb *inp;
1179 bool locked;
1180
1181 INP_HASH_RLOCK(pcbinfo);
1182 inp = in6_pcblookup_hash_locked(pcbinfo, faddr, fport, laddr, lport,
1183 (lookupflags & ~(INPLOOKUP_RLOCKPCB | INPLOOKUP_WLOCKPCB)), ifp);
1184 if (inp != NULL) {
1185 if (lookupflags & INPLOOKUP_WLOCKPCB)
1186 locked = INP_TRY_WLOCK(inp);
1187 else if (lookupflags & INPLOOKUP_RLOCKPCB)
1188 locked = INP_TRY_RLOCK(inp);
1189 else
1190 panic("%s: locking bug", __func__);
1191 if (!locked)
1192 in_pcbref(inp);
1193 INP_HASH_RUNLOCK(pcbinfo);
1194 if (!locked) {
1195 if (lookupflags & INPLOOKUP_WLOCKPCB) {
1196 INP_WLOCK(inp);
1197 if (in_pcbrele_wlocked(inp))
1198 return (NULL);
1199 } else {
1200 INP_RLOCK(inp);
1201 if (in_pcbrele_rlocked(inp))
1202 return (NULL);
1203 }
1204 }
1205 #ifdef INVARIANTS
1206 if (lookupflags & INPLOOKUP_WLOCKPCB)
1207 INP_WLOCK_ASSERT(inp);
1208 else
1209 INP_RLOCK_ASSERT(inp);
1210 #endif
1211 } else
1212 INP_HASH_RUNLOCK(pcbinfo);
1213 return (inp);
1214 }
1215
1216 /*
1217 * Public inpcb lookup routines, accepting a 4-tuple, and optionally, an mbuf
1218 * from which a pre-calculated hash value may be extracted.
1219 *
1220 * Possibly more of this logic should be in in6_pcbgroup.c.
1221 */
1222 struct inpcb *
in6_pcblookup(struct inpcbinfo * pcbinfo,struct in6_addr * faddr,u_int fport,struct in6_addr * laddr,u_int lport,int lookupflags,struct ifnet * ifp)1223 in6_pcblookup(struct inpcbinfo *pcbinfo, struct in6_addr *faddr, u_int fport,
1224 struct in6_addr *laddr, u_int lport, int lookupflags, struct ifnet *ifp)
1225 {
1226 #if defined(PCBGROUP) && !defined(RSS)
1227 struct inpcbgroup *pcbgroup;
1228 #endif
1229
1230 KASSERT((lookupflags & ~INPLOOKUP_MASK) == 0,
1231 ("%s: invalid lookup flags %d", __func__, lookupflags));
1232 KASSERT((lookupflags & (INPLOOKUP_RLOCKPCB | INPLOOKUP_WLOCKPCB)) != 0,
1233 ("%s: LOCKPCB not set", __func__));
1234
1235 /*
1236 * When not using RSS, use connection groups in preference to the
1237 * reservation table when looking up 4-tuples. When using RSS, just
1238 * use the reservation table, due to the cost of the Toeplitz hash
1239 * in software.
1240 *
1241 * XXXRW: This policy belongs in the pcbgroup code, as in principle
1242 * we could be doing RSS with a non-Toeplitz hash that is affordable
1243 * in software.
1244 */
1245 #if defined(PCBGROUP) && !defined(RSS)
1246 if (in_pcbgroup_enabled(pcbinfo)) {
1247 pcbgroup = in6_pcbgroup_bytuple(pcbinfo, laddr, lport, faddr,
1248 fport);
1249 return (in6_pcblookup_group(pcbinfo, pcbgroup, faddr, fport,
1250 laddr, lport, lookupflags, ifp));
1251 }
1252 #endif
1253 return (in6_pcblookup_hash(pcbinfo, faddr, fport, laddr, lport,
1254 lookupflags, ifp));
1255 }
1256
1257 struct inpcb *
in6_pcblookup_mbuf(struct inpcbinfo * pcbinfo,struct in6_addr * faddr,u_int fport,struct in6_addr * laddr,u_int lport,int lookupflags,struct ifnet * ifp,struct mbuf * m)1258 in6_pcblookup_mbuf(struct inpcbinfo *pcbinfo, struct in6_addr *faddr,
1259 u_int fport, struct in6_addr *laddr, u_int lport, int lookupflags,
1260 struct ifnet *ifp, struct mbuf *m)
1261 {
1262 #ifdef PCBGROUP
1263 struct inpcbgroup *pcbgroup;
1264 #endif
1265
1266 KASSERT((lookupflags & ~INPLOOKUP_MASK) == 0,
1267 ("%s: invalid lookup flags %d", __func__, lookupflags));
1268 KASSERT((lookupflags & (INPLOOKUP_RLOCKPCB | INPLOOKUP_WLOCKPCB)) != 0,
1269 ("%s: LOCKPCB not set", __func__));
1270
1271 #ifdef PCBGROUP
1272 /*
1273 * If we can use a hardware-generated hash to look up the connection
1274 * group, use that connection group to find the inpcb. Otherwise
1275 * fall back on a software hash -- or the reservation table if we're
1276 * using RSS.
1277 *
1278 * XXXRW: As above, that policy belongs in the pcbgroup code.
1279 */
1280 if (in_pcbgroup_enabled(pcbinfo) &&
1281 M_HASHTYPE_TEST(m, M_HASHTYPE_NONE) == 0) {
1282 pcbgroup = in6_pcbgroup_byhash(pcbinfo, M_HASHTYPE_GET(m),
1283 m->m_pkthdr.flowid);
1284 if (pcbgroup != NULL)
1285 return (in6_pcblookup_group(pcbinfo, pcbgroup, faddr,
1286 fport, laddr, lport, lookupflags, ifp));
1287 #ifndef RSS
1288 pcbgroup = in6_pcbgroup_bytuple(pcbinfo, laddr, lport, faddr,
1289 fport);
1290 return (in6_pcblookup_group(pcbinfo, pcbgroup, faddr, fport,
1291 laddr, lport, lookupflags, ifp));
1292 #endif
1293 }
1294 #endif
1295 return (in6_pcblookup_hash(pcbinfo, faddr, fport, laddr, lport,
1296 lookupflags, ifp));
1297 }
1298
1299 void
init_sin6(struct sockaddr_in6 * sin6,struct mbuf * m)1300 init_sin6(struct sockaddr_in6 *sin6, struct mbuf *m)
1301 {
1302 struct ip6_hdr *ip;
1303
1304 ip = mtod(m, struct ip6_hdr *);
1305 bzero(sin6, sizeof(*sin6));
1306 sin6->sin6_len = sizeof(*sin6);
1307 sin6->sin6_family = AF_INET6;
1308 sin6->sin6_addr = ip->ip6_src;
1309
1310 (void)sa6_recoverscope(sin6); /* XXX: should catch errors... */
1311
1312 return;
1313 }
1314