bsd/netinet/in_pcblist.c

   1 /*
   2  * Copyright (c) 2010-2014 Apple Inc. All rights reserved.
   3  *
   4  * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
   5  *
   6  * This file contains Original Code and/or Modifications of Original Code
   7  * as defined in and that are subject to the Apple Public Source License
   8  * Version 2.0 (the 'License'). You may not use this file except in
   9  * compliance with the License. The rights granted to you under the License
  10  * may not be used to create, or enable the creation or redistribution of,
  11  * unlawful or unlicensed copies of an Apple operating system, or to
  12  * circumvent, violate, or enable the circumvention or violation of, any
  13  * terms of an Apple operating system software license agreement.
  14  *
  15  * Please obtain a copy of the License at
  16  * http://www.opensource.apple.com/apsl/ and read it before using this file.
  17  *
  18  * The Original Code and all software distributed under the License are
  19  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  20  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  21  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
  22  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
  23  * Please see the License for the specific language governing rights and
  24  * limitations under the License.
  25  *
  26  * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
  27  */
  28 /*
  29  * Copyright (c) 1982, 1986, 1990, 1993
  30  *      The Regents of the University of California.  All rights reserved.
  31  *
  32  * Redistribution and use in source and binary forms, with or without
  33  * modification, are permitted provided that the following conditions
  34  * are met:
  35  * 1. Redistributions of source code must retain the above copyright
  36  *    notice, this list of conditions and the following disclaimer.
  37  * 2. Redistributions in binary form must reproduce the above copyright
  38  *    notice, this list of conditions and the following disclaimer in the
  39  *    documentation and/or other materials provided with the distribution.
  40  * 3. All advertising materials mentioning features or use of this software
  41  *    must display the following acknowledgement:
  42  *      This product includes software developed by the University of
  43  *      California, Berkeley and its contributors.
  44  * 4. Neither the name of the University nor the names of its contributors
  45  *    may be used to endorse or promote products derived from this software
  46  *    without specific prior written permission.
  47  *
  48  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  49  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  50  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  51  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  52  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  53  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  54  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  55  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  56  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  57  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  58  * SUCH DAMAGE.
  59  */
  60
  61 #include <sys/types.h>
  62 #include <sys/malloc.h>
  63 #include <sys/socket.h>
  64 #include <sys/socketvar.h>
  65 #include <sys/protosw.h>
  66 #include <sys/domain.h>
  67 #include <sys/kernel.h>
  68 #include <sys/sysctl.h>
  69 #include <sys/dtrace.h>
  70 #include <sys/kauth.h>
  71
  72 #include <net/route.h>
  73 #include <net/if_var.h>
  74
  75 #include <netinet/in.h>
  76 #include <netinet/in_pcb.h>
  77 #include <netinet/in_var.h>
  78 #include <netinet/ip_var.h>
  79
  80 #include <netinet/udp.h>
  81 #include <netinet/udp_var.h>
  82
  83 #include <netinet/tcp.h>
  84 #include <netinet/tcp_fsm.h>
  85 #include <netinet/tcp_seq.h>
  86 #include <netinet/tcp_timer.h>
  87 #include <netinet/tcp_var.h>
  88 #include <netinet6/in6_var.h>
  89
  90 #ifndef ROUNDUP64
  91 #define ROUNDUP64(x) P2ROUNDUP((x), sizeof (u_int64_t))
  92 #endif
  93
  94 #ifndef ADVANCE64
  95 #define ADVANCE64(p, n) (void*)((char *)(p) + ROUNDUP64(n))
  96 #endif
  97
  98 static void inpcb_to_xinpcb_n(struct inpcb *, struct xinpcb_n *);
  99 static void tcpcb_to_xtcpcb_n(struct tcpcb *, struct xtcpcb_n *);
 100
 101 __private_extern__ void
 102 sotoxsocket_n(struct socket *so, struct xsocket_n *xso)
 103 {
 104         xso->xso_len = sizeof (struct xsocket_n);
 105         xso->xso_kind = XSO_SOCKET;
 106
 107         if (so != NULL) {
 108                 xso->xso_so = (uint64_t)VM_KERNEL_ADDRPERM(so);
 109                 xso->so_type = so->so_type;
 110                 xso->so_options = so->so_options;
 111                 xso->so_linger = so->so_linger;
 112                 xso->so_state = so->so_state;
 113                 xso->so_pcb = (uint64_t)VM_KERNEL_ADDRPERM(so->so_pcb);
 114                 if (so->so_proto) {
 115                         xso->xso_protocol = SOCK_PROTO(so);
 116                         xso->xso_family = SOCK_DOM(so);
 117                 } else {
 118                         xso->xso_protocol = xso->xso_family = 0;
 119                 }
 120                 xso->so_qlen = so->so_qlen;
 121                 xso->so_incqlen = so->so_incqlen;
 122                 xso->so_qlimit = so->so_qlimit;
 123                 xso->so_timeo = so->so_timeo;
 124                 xso->so_error = so->so_error;
 125                 xso->so_pgid = so->so_pgid;
 126                 xso->so_oobmark = so->so_oobmark;
 127                 xso->so_uid = kauth_cred_getuid(so->so_cred);
 128                 xso->so_last_pid = so->last_pid;
 129                 xso->so_e_pid = so->e_pid;
 130         }
 131 }
 132
 133 __private_extern__ void
 134 sbtoxsockbuf_n(struct sockbuf *sb, struct xsockbuf_n *xsb)
 135 {
 136         xsb->xsb_len = sizeof (struct xsockbuf_n);
 137         xsb->xsb_kind = (sb->sb_flags & SB_RECV) ? XSO_RCVBUF : XSO_SNDBUF;
 138
 139         if (sb != NULL) {
 140                 xsb->sb_cc = sb->sb_cc;
 141                 xsb->sb_hiwat = sb->sb_hiwat;
 142                 xsb->sb_mbcnt = sb->sb_mbcnt;
 143                 xsb->sb_mbmax = sb->sb_mbmax;
 144                 xsb->sb_lowat = sb->sb_lowat;
 145                 xsb->sb_flags = sb->sb_flags;
 146                 xsb->sb_timeo = (short)(sb->sb_timeo.tv_sec * hz) +
 147                     sb->sb_timeo.tv_usec / tick;
 148                 if (xsb->sb_timeo == 0 && sb->sb_timeo.tv_usec != 0)
 149                         xsb->sb_timeo = 1;
 150         }
 151 }
 152
 153 __private_extern__ void
 154 sbtoxsockstat_n(struct socket *so, struct xsockstat_n *xst)
 155 {
 156         int i;
 157
 158         xst->xst_len = sizeof (struct xsockstat_n);
 159         xst->xst_kind = XSO_STATS;
 160
 161         for (i = 0; i < SO_TC_STATS_MAX; i++) {
 162                 xst->xst_tc_stats[i].rxpackets = so->so_tc_stats[i].rxpackets;
 163                 xst->xst_tc_stats[i].rxbytes = so->so_tc_stats[i].rxbytes;
 164                 xst->xst_tc_stats[i].txpackets = so->so_tc_stats[i].txpackets;
 165                 xst->xst_tc_stats[i].txbytes = so->so_tc_stats[i].txbytes;
 166         }
 167 }
 168
 169 static void
 170 inpcb_to_xinpcb_n(struct inpcb *inp, struct xinpcb_n *xinp)
 171 {
 172         xinp->xi_len = sizeof (struct xinpcb_n);
 173         xinp->xi_kind = XSO_INPCB;
 174         xinp->xi_inpp = (uint64_t)VM_KERNEL_ADDRPERM(inp);
 175         xinp->inp_fport = inp->inp_fport;
 176         xinp->inp_lport = inp->inp_lport;
 177         xinp->inp_ppcb = (uint64_t)VM_KERNEL_ADDRPERM(inp->inp_ppcb);
 178         xinp->inp_gencnt = inp->inp_gencnt;
 179         xinp->inp_flags = inp->inp_flags;
 180         xinp->inp_flow = inp->inp_flow;
 181         xinp->inp_vflag = inp->inp_vflag;
 182         xinp->inp_ip_ttl = inp->inp_ip_ttl;
 183         xinp->inp_ip_p = inp->inp_ip_p;
 184         xinp->inp_dependfaddr.inp6_foreign = inp->inp_dependfaddr.inp6_foreign;
 185         xinp->inp_dependladdr.inp6_local = inp->inp_dependladdr.inp6_local;
 186         xinp->inp_depend4.inp4_ip_tos = inp->inp_depend4.inp4_ip_tos;
 187         xinp->inp_depend6.inp6_hlim = 0;
 188         xinp->inp_depend6.inp6_cksum = inp->inp_depend6.inp6_cksum;
 189         xinp->inp_depend6.inp6_ifindex = 0;
 190         xinp->inp_depend6.inp6_hops = inp->inp_depend6.inp6_hops;
 191         xinp->inp_flowhash = inp->inp_flowhash;
 192         xinp->inp_flags2 = inp->inp_flags2;
 193 }
 194
 195 __private_extern__ void
 196 tcpcb_to_xtcpcb_n(struct tcpcb *tp, struct xtcpcb_n *xt)
 197 {
 198         xt->xt_len = sizeof (struct xtcpcb_n);
 199         xt->xt_kind = XSO_TCPCB;
 200
 201         xt->t_segq = (uint32_t)VM_KERNEL_ADDRPERM(tp->t_segq.lh_first);
 202         xt->t_dupacks = tp->t_dupacks;
 203         xt->t_timer[TCPT_REXMT_EXT] = tp->t_timer[TCPT_REXMT];
 204         xt->t_timer[TCPT_PERSIST_EXT] = tp->t_timer[TCPT_PERSIST];
 205         xt->t_timer[TCPT_KEEP_EXT] = tp->t_timer[TCPT_KEEP];
 206         xt->t_timer[TCPT_2MSL_EXT] = tp->t_timer[TCPT_2MSL];
 207         xt->t_state = tp->t_state;
 208         xt->t_flags = tp->t_flags;
 209         xt->t_force = (tp->t_flagsext & TF_FORCE) ? 1 : 0;
 210         xt->snd_una = tp->snd_una;
 211         xt->snd_max = tp->snd_max;
 212         xt->snd_nxt = tp->snd_nxt;
 213         xt->snd_up = tp->snd_up;
 214         xt->snd_wl1 = tp->snd_wl1;
 215         xt->snd_wl2 = tp->snd_wl2;
 216         xt->iss = tp->iss;
 217         xt->irs = tp->irs;
 218         xt->rcv_nxt = tp->rcv_nxt;
 219         xt->rcv_adv = tp->rcv_adv;
 220         xt->rcv_wnd = tp->rcv_wnd;
 221         xt->rcv_up = tp->rcv_up;
 222         xt->snd_wnd = tp->snd_wnd;
 223         xt->snd_cwnd = tp->snd_cwnd;
 224         xt->snd_ssthresh = tp->snd_ssthresh;
 225         xt->t_maxopd = tp->t_maxopd;
 226         xt->t_rcvtime = tp->t_rcvtime;
 227         xt->t_starttime = tp->t_starttime;
 228         xt->t_rtttime = tp->t_rtttime;
 229         xt->t_rtseq = tp->t_rtseq;
 230         xt->t_rxtcur = tp->t_rxtcur;
 231         xt->t_maxseg = tp->t_maxseg;
 232         xt->t_srtt = tp->t_srtt;
 233         xt->t_rttvar = tp->t_rttvar;
 234         xt->t_rxtshift = tp->t_rxtshift;
 235         xt->t_rttmin = tp->t_rttmin;
 236         xt->t_rttupdated = tp->t_rttupdated;
 237         xt->max_sndwnd = tp->max_sndwnd;
 238         xt->t_softerror = tp->t_softerror;
 239         xt->t_oobflags = tp->t_oobflags;
 240         xt->t_iobc = tp->t_iobc;
 241         xt->snd_scale = tp->snd_scale;
 242         xt->rcv_scale = tp->rcv_scale;
 243         xt->request_r_scale = tp->request_r_scale;
 244         xt->requested_s_scale = tp->requested_s_scale;
 245         xt->ts_recent = tp->ts_recent;
 246         xt->ts_recent_age = tp->ts_recent_age;
 247         xt->last_ack_sent = tp->last_ack_sent;
 248         xt->cc_send = tp->cc_send;
 249         xt->cc_recv = tp->cc_recv;
 250         xt->snd_recover = tp->snd_recover;
 251         xt->snd_cwnd_prev = tp->snd_cwnd_prev;
 252         xt->snd_ssthresh_prev = tp->snd_ssthresh_prev;
 253 }
 254
 255 __private_extern__ int
 256 get_pcblist_n(short proto, struct sysctl_req *req, struct inpcbinfo *pcbinfo)
 257 {
 258         int error = 0;
 259         int i, n;
 260         struct inpcb *inp, **inp_list = NULL;
 261         inp_gen_t gencnt;
 262         struct xinpgen xig;
 263         void *buf = NULL;
 264         size_t item_size = ROUNDUP64(sizeof (struct xinpcb_n)) +
 265             ROUNDUP64(sizeof (struct xsocket_n)) +
 266             2 * ROUNDUP64(sizeof (struct xsockbuf_n)) +
 267             ROUNDUP64(sizeof (struct xsockstat_n));
 268
 269         if (proto == IPPROTO_TCP)
 270                 item_size += ROUNDUP64(sizeof (struct xtcpcb_n));
 271
 272         /*
 273          * The process of preparing the PCB list is too time-consuming and
 274          * resource-intensive to repeat twice on every request.
 275          */
 276         lck_rw_lock_exclusive(pcbinfo->ipi_lock);
 277         if (req->oldptr == USER_ADDR_NULL) {
 278                 n = pcbinfo->ipi_count;
 279                 req->oldidx = 2 * (sizeof (xig)) + (n + n/8) * item_size;
 280                 goto done;
 281         }
 282
 283         if (req->newptr != USER_ADDR_NULL) {
 284                 error = EPERM;
 285                 goto done;
 286         }
 287
 288         /*
 289          * OK, now we're committed to doing something.
 290          */
 291         gencnt = pcbinfo->ipi_gencnt;
 292         n = pcbinfo->ipi_count;
 293
 294         bzero(&xig, sizeof (xig));
 295         xig.xig_len = sizeof (xig);
 296         xig.xig_count = n;
 297         xig.xig_gen = gencnt;
 298         xig.xig_sogen = so_gencnt;
 299         error = SYSCTL_OUT(req, &xig, sizeof (xig));
 300         if (error) {
 301                 goto done;
 302         }
 303         /*
 304          * We are done if there is no pcb
 305          */
 306         if (n == 0) {
 307                 goto done;
 308         }
 309
 310         buf = _MALLOC(item_size, M_TEMP, M_WAITOK);
 311         if (buf == NULL) {
 312                 error = ENOMEM;
 313                 goto done;
 314         }
 315
 316         inp_list = _MALLOC(n * sizeof (*inp_list), M_TEMP, M_WAITOK);
 317         if (inp_list == NULL) {
 318                 error = ENOMEM;
 319                 goto done;
 320         }
 321
 322         for (inp = pcbinfo->ipi_listhead->lh_first, i = 0; inp && i < n;
 323             inp = inp->inp_list.le_next) {
 324                 if (inp->inp_gencnt <= gencnt &&
 325                     inp->inp_state != INPCB_STATE_DEAD)
 326                         inp_list[i++] = inp;
 327         }
 328         n = i;
 329
 330         error = 0;
 331         for (i = 0; i < n; i++) {
 332                 inp = inp_list[i];
 333                 if (inp->inp_gencnt <= gencnt &&
 334                     inp->inp_state != INPCB_STATE_DEAD) {
 335                         struct xinpcb_n *xi = (struct xinpcb_n *)buf;
 336                         struct xsocket_n *xso = (struct xsocket_n *)
 337                             ADVANCE64(xi, sizeof (*xi));
 338                         struct xsockbuf_n *xsbrcv = (struct xsockbuf_n *)
 339                             ADVANCE64(xso, sizeof (*xso));
 340                         struct xsockbuf_n *xsbsnd = (struct xsockbuf_n *)
 341                             ADVANCE64(xsbrcv, sizeof (*xsbrcv));
 342                         struct xsockstat_n *xsostats = (struct xsockstat_n *)
 343                             ADVANCE64(xsbsnd, sizeof (*xsbsnd));
 344
 345                         bzero(buf, item_size);
 346
 347                         inpcb_to_xinpcb_n(inp, xi);
 348                         sotoxsocket_n(inp->inp_socket, xso);
 349                         sbtoxsockbuf_n(inp->inp_socket ?
 350                             &inp->inp_socket->so_rcv : NULL, xsbrcv);
 351                         sbtoxsockbuf_n(inp->inp_socket ?
 352                             &inp->inp_socket->so_snd : NULL, xsbsnd);
 353                         sbtoxsockstat_n(inp->inp_socket, xsostats);
 354                         if (proto == IPPROTO_TCP) {
 355                                 struct  xtcpcb_n *xt = (struct xtcpcb_n *)
 356                                     ADVANCE64(xsostats, sizeof (*xsostats));
 357
 358                                 /*
 359                                  * inp->inp_ppcb, can only be NULL on
 360                                  * an initialization race window.
 361                                  * No need to lock.
 362                                  */
 363                                 if (inp->inp_ppcb == NULL)
 364                                         continue;
 365
 366                                 tcpcb_to_xtcpcb_n((struct tcpcb *)
 367                                     inp->inp_ppcb, xt);
 368                         }
 369                         error = SYSCTL_OUT(req, buf, item_size);
 370                 }
 371         }
 372         if (!error) {
 373                 /*
 374                  * Give the user an updated idea of our state.
 375                  * If the generation differs from what we told
 376                  * her before, she knows that something happened
 377                  * while we were processing this request, and it
 378                  * might be necessary to retry.
 379                  */
 380                 bzero(&xig, sizeof (xig));
 381                 xig.xig_len = sizeof (xig);
 382                 xig.xig_gen = pcbinfo->ipi_gencnt;
 383                 xig.xig_sogen = so_gencnt;
 384                 xig.xig_count = pcbinfo->ipi_count;
 385                 error = SYSCTL_OUT(req, &xig, sizeof (xig));
 386         }
 387 done:
 388         lck_rw_done(pcbinfo->ipi_lock);
 389         if (inp_list != NULL)
 390                 FREE(inp_list, M_TEMP);
 391         if (buf != NULL)
 392                 FREE(buf, M_TEMP);
 393         return (error);
 394 }
 395
 396 __private_extern__ void
 397 inpcb_get_ports_used(uint32_t ifindex, int protocol, uint32_t flags,
 398     bitstr_t *bitfield, struct inpcbinfo *pcbinfo)
 399 {
 400         struct inpcb *inp;
 401         struct socket *so;
 402         inp_gen_t gencnt;
 403         bool iswildcard, wildcardok, nowakeok;
 404         bool recvanyifonly, extbgidleok;
 405         bool activeonly;
 406
 407         wildcardok = ((flags & INPCB_GET_PORTS_USED_WILDCARDOK) != 0);
 408         nowakeok = ((flags & INPCB_GET_PORTS_USED_NOWAKEUPOK) != 0);
 409         recvanyifonly = ((flags & INPCB_GET_PORTS_USED_RECVANYIFONLY) != 0);
 410         extbgidleok = ((flags & INPCB_GET_PORTS_USED_EXTBGIDLEONLY) != 0);
 411         activeonly = ((flags & INPCB_GET_PORTS_USED_ACTIVEONLY) != 0);
 412
 413         lck_rw_lock_shared(pcbinfo->ipi_lock);
 414         gencnt = pcbinfo->ipi_gencnt;
 415
 416         for (inp = LIST_FIRST(pcbinfo->ipi_listhead); inp;
 417             inp = LIST_NEXT(inp, inp_list)) {
 418                 uint16_t port;
 419
 420                 if (inp->inp_gencnt > gencnt ||
 421                     inp->inp_state == INPCB_STATE_DEAD ||
 422                     inp->inp_wantcnt == WNT_STOPUSING)
 423                         continue;
 424
 425                 if ((so = inp->inp_socket) == NULL ||
 426                     (so->so_state & SS_DEFUNCT) ||
 427                     (so->so_state & SS_ISDISCONNECTED))
 428                         continue;
 429
 430                 if (!(protocol == PF_UNSPEC ||
 431                     (protocol == PF_INET && (inp->inp_vflag & INP_IPV4)) ||
 432                     (protocol == PF_INET6 && (inp->inp_vflag & INP_IPV6))))
 433                         continue;
 434
 435                 iswildcard = (((inp->inp_vflag & INP_IPV4) &&
 436                     inp->inp_laddr.s_addr == INADDR_ANY) ||
 437                     ((inp->inp_vflag & INP_IPV6) &&
 438                     IN6_IS_ADDR_UNSPECIFIED(&inp->in6p_laddr)));
 439
 440                 if (!wildcardok && iswildcard)
 441                         continue;
 442
 443                 if ((so->so_options & SO_NOWAKEFROMSLEEP) &&
 444                         !nowakeok)
 445                         continue;
 446
 447                 if (!(inp->inp_flags & INP_RECV_ANYIF) &&
 448                         recvanyifonly)
 449                         continue;
 450
 451                 if (!(so->so_flags1 & SOF1_EXTEND_BK_IDLE_WANTED) &&
 452                         extbgidleok)
 453                         continue;
 454
 455                 if (!iswildcard &&
 456                     !(ifindex == 0 || inp->inp_last_outifp == NULL ||
 457                     ifindex == inp->inp_last_outifp->if_index))
 458                         continue;
 459
 460                 if (SOCK_PROTO(inp->inp_socket) == IPPROTO_UDP &&
 461                     so->so_state & SS_CANTRCVMORE)
 462                         continue;
 463
 464                 if (SOCK_PROTO(inp->inp_socket) == IPPROTO_TCP) {
 465                         struct  tcpcb *tp = sototcpcb(inp->inp_socket);
 466
 467                         switch (tp->t_state) {
 468                                 case TCPS_CLOSED:
 469                                         continue;
 470                                         /* NOT REACHED */
 471                                 case TCPS_LISTEN:
 472                                 case TCPS_SYN_SENT:
 473                                 case TCPS_SYN_RECEIVED:
 474                                 case TCPS_ESTABLISHED:
 475                                 case TCPS_FIN_WAIT_1:
 476                                         /*
 477                                          * Note: FIN_WAIT_1 is an active state
 478                                          * because we need our FIN to be
 479                                          * acknowledged
 480                                          */
 481                                         break;
 482                                 case TCPS_CLOSE_WAIT:
 483                                 case TCPS_CLOSING:
 484                                 case TCPS_LAST_ACK:
 485                                 case TCPS_FIN_WAIT_2:
 486                                         /*
 487                                          * In the closing states, the connection
 488                                          * is not idle when there is outgoing
 489                                          * data having to be acknowledged
 490                                          */
 491                                         if (activeonly && so->so_snd.sb_cc == 0)
 492                                                 continue;
 493                                         break;
 494                                 case TCPS_TIME_WAIT:
 495                                         continue;
 496                                         /* NOT REACHED */
 497                         }
 498                 }
 499                 /*
 500                  * Final safeguard to exclude unspecified local port
 501                  */
 502                 port = ntohs(inp->inp_lport);
 503                 if (port == 0)
 504                         continue;
 505                 bit_set(bitfield, port);
 506         }
 507         lck_rw_done(pcbinfo->ipi_lock);
 508 }
 509
 510 __private_extern__ uint32_t
 511 inpcb_count_opportunistic(unsigned int ifindex, struct inpcbinfo *pcbinfo,
 512     u_int32_t flags)
 513 {
 514         uint32_t opportunistic = 0;
 515         struct inpcb *inp;
 516         inp_gen_t gencnt;
 517
 518         lck_rw_lock_shared(pcbinfo->ipi_lock);
 519         gencnt = pcbinfo->ipi_gencnt;
 520         for (inp = LIST_FIRST(pcbinfo->ipi_listhead);
 521             inp != NULL; inp = LIST_NEXT(inp, inp_list)) {
 522                 if (inp->inp_gencnt <= gencnt &&
 523                     inp->inp_state != INPCB_STATE_DEAD &&
 524                     inp->inp_socket != NULL &&
 525                     so_get_opportunistic(inp->inp_socket) &&
 526                     inp->inp_last_outifp != NULL &&
 527                     ifindex == inp->inp_last_outifp->if_index) {
 528                         opportunistic++;
 529                         struct socket *so = inp->inp_socket;
 530                         if ((flags & INPCB_OPPORTUNISTIC_SETCMD) &&
 531                             (so->so_state & SS_ISCONNECTED)) {
 532                                 socket_lock(so, 1);
 533                                 if (flags & INPCB_OPPORTUNISTIC_THROTTLEON) {
 534                                         so->so_flags |= SOF_SUSPENDED;
 535                                         soevent(so,
 536                                             (SO_FILT_HINT_LOCKED |
 537                                             SO_FILT_HINT_SUSPEND));
 538                                 } else {
 539                                         so->so_flags &= ~(SOF_SUSPENDED);
 540                                         soevent(so,
 541                                             (SO_FILT_HINT_LOCKED |
 542                                             SO_FILT_HINT_RESUME));
 543                                 }
 544                                 SOTHROTTLELOG(("throttle[%d]: so 0x%llx "
 545                                     "[%d,%d] %s\n", so->last_pid,
 546                                     (uint64_t)VM_KERNEL_ADDRPERM(so),
 547                                     SOCK_DOM(so), SOCK_TYPE(so),
 548                                     (so->so_flags & SOF_SUSPENDED) ?
 549                                     "SUSPENDED" : "RESUMED"));
 550                                 socket_unlock(so, 1);
 551                         }
 552                 }
 553         }
 554
 555         lck_rw_done(pcbinfo->ipi_lock);
 556
 557         return (opportunistic);
 558 }
 559
 560 __private_extern__ uint32_t
 561 inpcb_find_anypcb_byaddr(struct ifaddr *ifa, struct inpcbinfo *pcbinfo)
 562 {
 563         struct inpcb *inp;
 564         inp_gen_t gencnt = pcbinfo->ipi_gencnt;
 565         struct socket *so = NULL;
 566         int af;
 567
 568         if ((ifa->ifa_addr->sa_family != AF_INET) &&
 569             (ifa->ifa_addr->sa_family != AF_INET6)) {
 570                 return (0);
 571         }
 572
 573         lck_rw_lock_shared(pcbinfo->ipi_lock);
 574         for (inp = LIST_FIRST(pcbinfo->ipi_listhead);
 575             inp != NULL; inp = LIST_NEXT(inp, inp_list)) {
 576
 577                 if (inp->inp_gencnt <= gencnt &&
 578                     inp->inp_state != INPCB_STATE_DEAD &&
 579                     inp->inp_socket != NULL) {
 580                         so = inp->inp_socket;
 581                         af = SOCK_DOM(so);
 582                         if (af != ifa->ifa_addr->sa_family)
 583                                 continue;
 584                         if (inp->inp_last_outifp != ifa->ifa_ifp)
 585                                 continue;
 586
 587                         if (af == AF_INET) {
 588                                 if (inp->inp_laddr.s_addr ==
 589                                     (satosin(ifa->ifa_addr))->sin_addr.s_addr) {
 590                                         lck_rw_done(pcbinfo->ipi_lock);
 591                                         return (1);
 592                                 }
 593                         }
 594                         if (af == AF_INET6) {
 595                                 if (IN6_ARE_ADDR_EQUAL(IFA_IN6(ifa),
 596                                     &inp->in6p_laddr)) {
 597                                         lck_rw_done(pcbinfo->ipi_lock);
 598                                         return (1);
 599                                 }
 600                         }
 601                 }
 602         }
 603         lck_rw_done(pcbinfo->ipi_lock);
 604         return (0);
 605 }