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 #include <net/ntstat.h>
  75
  76 #include <netinet/in.h>
  77 #include <netinet/in_pcb.h>
  78 #include <netinet/in_var.h>
  79 #include <netinet/ip_var.h>
  80
  81 #include <netinet/udp.h>
  82 #include <netinet/udp_var.h>
  83
  84 #include <netinet/tcp.h>
  85 #include <netinet/tcp_fsm.h>
  86 #include <netinet/tcp_seq.h>
  87 #include <netinet/tcp_timer.h>
  88 #include <netinet/tcp_var.h>
  89 #include <netinet6/in6_var.h>
  90
  91 #ifndef ROUNDUP64
  92 #define ROUNDUP64(x) P2ROUNDUP((x), sizeof (u_int64_t))
  93 #endif
  94
  95 #ifndef ADVANCE64
  96 #define ADVANCE64(p, n) (void*)((char *)(p) + ROUNDUP64(n))
  97 #endif
  98
  99 static void inpcb_to_xinpcb_n(struct inpcb *, struct xinpcb_n *);
 100 static void tcpcb_to_xtcpcb_n(struct tcpcb *, struct xtcpcb_n *);
 101
 102 __private_extern__ void
 103 sotoxsocket_n(struct socket *so, struct xsocket_n *xso)
 104 {
 105         xso->xso_len = sizeof (struct xsocket_n);
 106         xso->xso_kind = XSO_SOCKET;
 107
 108         if (so != NULL) {
 109                 xso->xso_so = (uint64_t)VM_KERNEL_ADDRPERM(so);
 110                 xso->so_type = so->so_type;
 111                 xso->so_options = so->so_options;
 112                 xso->so_linger = so->so_linger;
 113                 xso->so_state = so->so_state;
 114                 xso->so_pcb = (uint64_t)VM_KERNEL_ADDRPERM(so->so_pcb);
 115                 if (so->so_proto) {
 116                         xso->xso_protocol = SOCK_PROTO(so);
 117                         xso->xso_family = SOCK_DOM(so);
 118                 } else {
 119                         xso->xso_protocol = xso->xso_family = 0;
 120                 }
 121                 xso->so_qlen = so->so_qlen;
 122                 xso->so_incqlen = so->so_incqlen;
 123                 xso->so_qlimit = so->so_qlimit;
 124                 xso->so_timeo = so->so_timeo;
 125                 xso->so_error = so->so_error;
 126                 xso->so_pgid = so->so_pgid;
 127                 xso->so_oobmark = so->so_oobmark;
 128                 xso->so_uid = kauth_cred_getuid(so->so_cred);
 129                 xso->so_last_pid = so->last_pid;
 130                 xso->so_e_pid = so->e_pid;
 131         }
 132 }
 133
 134 __private_extern__ void
 135 sbtoxsockbuf_n(struct sockbuf *sb, struct xsockbuf_n *xsb)
 136 {
 137         xsb->xsb_len = sizeof (struct xsockbuf_n);
 138         xsb->xsb_kind = (sb->sb_flags & SB_RECV) ? XSO_RCVBUF : XSO_SNDBUF;
 139
 140         if (sb != NULL) {
 141                 xsb->sb_cc = sb->sb_cc;
 142                 xsb->sb_hiwat = sb->sb_hiwat;
 143                 xsb->sb_mbcnt = sb->sb_mbcnt;
 144                 xsb->sb_mbmax = sb->sb_mbmax;
 145                 xsb->sb_lowat = sb->sb_lowat;
 146                 xsb->sb_flags = sb->sb_flags;
 147                 xsb->sb_timeo = (short)(sb->sb_timeo.tv_sec * hz) +
 148                     sb->sb_timeo.tv_usec / tick;
 149                 if (xsb->sb_timeo == 0 && sb->sb_timeo.tv_usec != 0)
 150                         xsb->sb_timeo = 1;
 151         }
 152 }
 153
 154 __private_extern__ void
 155 sbtoxsockstat_n(struct socket *so, struct xsockstat_n *xst)
 156 {
 157         int i;
 158
 159         xst->xst_len = sizeof (struct xsockstat_n);
 160         xst->xst_kind = XSO_STATS;
 161
 162         for (i = 0; i < SO_TC_STATS_MAX; i++) {
 163                 xst->xst_tc_stats[i].rxpackets = so->so_tc_stats[i].rxpackets;
 164                 xst->xst_tc_stats[i].rxbytes = so->so_tc_stats[i].rxbytes;
 165                 xst->xst_tc_stats[i].txpackets = so->so_tc_stats[i].txpackets;
 166                 xst->xst_tc_stats[i].txbytes = so->so_tc_stats[i].txbytes;
 167         }
 168 }
 169
 170 static void
 171 inpcb_to_xinpcb_n(struct inpcb *inp, struct xinpcb_n *xinp)
 172 {
 173         xinp->xi_len = sizeof (struct xinpcb_n);
 174         xinp->xi_kind = XSO_INPCB;
 175         xinp->xi_inpp = (uint64_t)VM_KERNEL_ADDRPERM(inp);
 176         xinp->inp_fport = inp->inp_fport;
 177         xinp->inp_lport = inp->inp_lport;
 178         xinp->inp_ppcb = (uint64_t)VM_KERNEL_ADDRPERM(inp->inp_ppcb);
 179         xinp->inp_gencnt = inp->inp_gencnt;
 180         xinp->inp_flags = inp->inp_flags;
 181         xinp->inp_flow = inp->inp_flow;
 182         xinp->inp_vflag = inp->inp_vflag;
 183         xinp->inp_ip_ttl = inp->inp_ip_ttl;
 184         xinp->inp_ip_p = inp->inp_ip_p;
 185         xinp->inp_dependfaddr.inp6_foreign = inp->inp_dependfaddr.inp6_foreign;
 186         xinp->inp_dependladdr.inp6_local = inp->inp_dependladdr.inp6_local;
 187         xinp->inp_depend4.inp4_ip_tos = inp->inp_depend4.inp4_ip_tos;
 188         xinp->inp_depend6.inp6_hlim = 0;
 189         xinp->inp_depend6.inp6_cksum = inp->inp_depend6.inp6_cksum;
 190         xinp->inp_depend6.inp6_ifindex = 0;
 191         xinp->inp_depend6.inp6_hops = inp->inp_depend6.inp6_hops;
 192         xinp->inp_flowhash = inp->inp_flowhash;
 193         xinp->inp_flags2 = inp->inp_flags2;
 194 }
 195
 196 __private_extern__ void
 197 tcpcb_to_xtcpcb_n(struct tcpcb *tp, struct xtcpcb_n *xt)
 198 {
 199         xt->xt_len = sizeof (struct xtcpcb_n);
 200         xt->xt_kind = XSO_TCPCB;
 201
 202         xt->t_segq = (uint32_t)VM_KERNEL_ADDRPERM(tp->t_segq.lh_first);
 203         xt->t_dupacks = tp->t_dupacks;
 204         xt->t_timer[TCPT_REXMT_EXT] = tp->t_timer[TCPT_REXMT];
 205         xt->t_timer[TCPT_PERSIST_EXT] = tp->t_timer[TCPT_PERSIST];
 206         xt->t_timer[TCPT_KEEP_EXT] = tp->t_timer[TCPT_KEEP];
 207         xt->t_timer[TCPT_2MSL_EXT] = tp->t_timer[TCPT_2MSL];
 208         xt->t_state = tp->t_state;
 209         xt->t_flags = tp->t_flags;
 210         xt->t_force = (tp->t_flagsext & TF_FORCE) ? 1 : 0;
 211         xt->snd_una = tp->snd_una;
 212         xt->snd_max = tp->snd_max;
 213         xt->snd_nxt = tp->snd_nxt;
 214         xt->snd_up = tp->snd_up;
 215         xt->snd_wl1 = tp->snd_wl1;
 216         xt->snd_wl2 = tp->snd_wl2;
 217         xt->iss = tp->iss;
 218         xt->irs = tp->irs;
 219         xt->rcv_nxt = tp->rcv_nxt;
 220         xt->rcv_adv = tp->rcv_adv;
 221         xt->rcv_wnd = tp->rcv_wnd;
 222         xt->rcv_up = tp->rcv_up;
 223         xt->snd_wnd = tp->snd_wnd;
 224         xt->snd_cwnd = tp->snd_cwnd;
 225         xt->snd_ssthresh = tp->snd_ssthresh;
 226         xt->t_maxopd = tp->t_maxopd;
 227         xt->t_rcvtime = tp->t_rcvtime;
 228         xt->t_starttime = tp->t_starttime;
 229         xt->t_rtttime = tp->t_rtttime;
 230         xt->t_rtseq = tp->t_rtseq;
 231         xt->t_rxtcur = tp->t_rxtcur;
 232         xt->t_maxseg = tp->t_maxseg;
 233         xt->t_srtt = tp->t_srtt;
 234         xt->t_rttvar = tp->t_rttvar;
 235         xt->t_rxtshift = tp->t_rxtshift;
 236         xt->t_rttmin = tp->t_rttmin;
 237         xt->t_rttupdated = tp->t_rttupdated;
 238         xt->max_sndwnd = tp->max_sndwnd;
 239         xt->t_softerror = tp->t_softerror;
 240         xt->t_oobflags = tp->t_oobflags;
 241         xt->t_iobc = tp->t_iobc;
 242         xt->snd_scale = tp->snd_scale;
 243         xt->rcv_scale = tp->rcv_scale;
 244         xt->request_r_scale = tp->request_r_scale;
 245         xt->requested_s_scale = tp->requested_s_scale;
 246         xt->ts_recent = tp->ts_recent;
 247         xt->ts_recent_age = tp->ts_recent_age;
 248         xt->last_ack_sent = tp->last_ack_sent;
 249         xt->cc_send = 0;
 250         xt->cc_recv = 0;
 251         xt->snd_recover = tp->snd_recover;
 252         xt->snd_cwnd_prev = tp->snd_cwnd_prev;
 253         xt->snd_ssthresh_prev = tp->snd_ssthresh_prev;
 254 }
 255
 256 __private_extern__ int
 257 get_pcblist_n(short proto, struct sysctl_req *req, struct inpcbinfo *pcbinfo)
 258 {
 259         int error = 0;
 260         int i, n;
 261         struct inpcb *inp, **inp_list = NULL;
 262         inp_gen_t gencnt;
 263         struct xinpgen xig;
 264         void *buf = NULL;
 265         size_t item_size = ROUNDUP64(sizeof (struct xinpcb_n)) +
 266             ROUNDUP64(sizeof (struct xsocket_n)) +
 267             2 * ROUNDUP64(sizeof (struct xsockbuf_n)) +
 268             ROUNDUP64(sizeof (struct xsockstat_n));
 269
 270         if (proto == IPPROTO_TCP)
 271                 item_size += ROUNDUP64(sizeof (struct xtcpcb_n));
 272
 273         if (req->oldptr == USER_ADDR_NULL) {
 274                 n = pcbinfo->ipi_count;
 275                 req->oldidx = 2 * (sizeof (xig)) + (n + n/8 + 1) * item_size;
 276                 return 0;
 277         }
 278
 279         if (req->newptr != USER_ADDR_NULL) {
 280                 return EPERM;
 281         }
 282
 283
 284         /*
 285          * The process of preparing the PCB list is too time-consuming and
 286          * resource-intensive to repeat twice on every request.
 287          */
 288         lck_rw_lock_exclusive(pcbinfo->ipi_lock);
 289         /*
 290          * OK, now we're committed to doing something.
 291          */
 292         gencnt = pcbinfo->ipi_gencnt;
 293         n = pcbinfo->ipi_count;
 294
 295         bzero(&xig, sizeof (xig));
 296         xig.xig_len = sizeof (xig);
 297         xig.xig_count = n;
 298         xig.xig_gen = gencnt;
 299         xig.xig_sogen = so_gencnt;
 300         error = SYSCTL_OUT(req, &xig, sizeof (xig));
 301         if (error) {
 302                 goto done;
 303         }
 304         /*
 305          * We are done if there is no pcb
 306          */
 307         if (xig.xig_count == 0) {
 308                 goto done;
 309         }
 310
 311         buf = _MALLOC(item_size, M_TEMP, M_WAITOK);
 312         if (buf == NULL) {
 313                 error = ENOMEM;
 314                 goto done;
 315         }
 316
 317         inp_list = _MALLOC(n * sizeof (*inp_list), M_TEMP, M_WAITOK);
 318         if (inp_list == NULL) {
 319                 error = ENOMEM;
 320                 goto done;
 321         }
 322
 323         /*
 324          * Special case TCP to include the connections in time wait
 325          */
 326         if (proto == IPPROTO_TCP) {
 327                 n = get_tcp_inp_list(inp_list, n, gencnt);
 328         } else {
 329                 for (inp = pcbinfo->ipi_listhead->lh_first, i = 0; inp && i < n;
 330                      inp = inp->inp_list.le_next) {
 331                         if (inp->inp_gencnt <= gencnt &&
 332                             inp->inp_state != INPCB_STATE_DEAD)
 333                                 inp_list[i++] = inp;
 334                 }
 335                 n = i;
 336         }
 337
 338
 339         error = 0;
 340         for (i = 0; i < n; i++) {
 341                 inp = inp_list[i];
 342                 if (inp->inp_gencnt <= gencnt &&
 343                     inp->inp_state != INPCB_STATE_DEAD) {
 344                         struct xinpcb_n *xi = (struct xinpcb_n *)buf;
 345                         struct xsocket_n *xso = (struct xsocket_n *)
 346                             ADVANCE64(xi, sizeof (*xi));
 347                         struct xsockbuf_n *xsbrcv = (struct xsockbuf_n *)
 348                             ADVANCE64(xso, sizeof (*xso));
 349                         struct xsockbuf_n *xsbsnd = (struct xsockbuf_n *)
 350                             ADVANCE64(xsbrcv, sizeof (*xsbrcv));
 351                         struct xsockstat_n *xsostats = (struct xsockstat_n *)
 352                             ADVANCE64(xsbsnd, sizeof (*xsbsnd));
 353
 354                         bzero(buf, item_size);
 355
 356                         inpcb_to_xinpcb_n(inp, xi);
 357                         sotoxsocket_n(inp->inp_socket, xso);
 358                         sbtoxsockbuf_n(inp->inp_socket ?
 359                             &inp->inp_socket->so_rcv : NULL, xsbrcv);
 360                         sbtoxsockbuf_n(inp->inp_socket ?
 361                             &inp->inp_socket->so_snd : NULL, xsbsnd);
 362                         sbtoxsockstat_n(inp->inp_socket, xsostats);
 363                         if (proto == IPPROTO_TCP) {
 364                                 struct  xtcpcb_n *xt = (struct xtcpcb_n *)
 365                                     ADVANCE64(xsostats, sizeof (*xsostats));
 366
 367                                 /*
 368                                  * inp->inp_ppcb, can only be NULL on
 369                                  * an initialization race window.
 370                                  * No need to lock.
 371                                  */
 372                                 if (inp->inp_ppcb == NULL)
 373                                         continue;
 374
 375                                 tcpcb_to_xtcpcb_n((struct tcpcb *)
 376                                     inp->inp_ppcb, xt);
 377                         }
 378                         error = SYSCTL_OUT(req, buf, item_size);
 379                         if (error) {
 380                                 break;
 381                         }
 382                 }
 383         }
 384
 385         if (!error) {
 386                 /*
 387                  * Give the user an updated idea of our state.
 388                  * If the generation differs from what we told
 389                  * her before, she knows that something happened
 390                  * while we were processing this request, and it
 391                  * might be necessary to retry.
 392                  */
 393                 bzero(&xig, sizeof (xig));
 394                 xig.xig_len = sizeof (xig);
 395                 xig.xig_gen = pcbinfo->ipi_gencnt;
 396                 xig.xig_sogen = so_gencnt;
 397                 xig.xig_count = pcbinfo->ipi_count;
 398                 error = SYSCTL_OUT(req, &xig, sizeof (xig));
 399         }
 400 done:
 401         lck_rw_done(pcbinfo->ipi_lock);
 402
 403         if (inp_list != NULL)
 404                 FREE(inp_list, M_TEMP);
 405         if (buf != NULL)
 406                 FREE(buf, M_TEMP);
 407         return (error);
 408 }
 409
 410 __private_extern__ void
 411 inpcb_get_ports_used(uint32_t ifindex, int protocol, uint32_t flags,
 412     bitstr_t *bitfield, struct inpcbinfo *pcbinfo)
 413 {
 414         struct inpcb *inp;
 415         struct socket *so;
 416         inp_gen_t gencnt;
 417         bool iswildcard, wildcardok, nowakeok;
 418         bool recvanyifonly, extbgidleok;
 419         bool activeonly;
 420
 421         wildcardok = ((flags & INPCB_GET_PORTS_USED_WILDCARDOK) != 0);
 422         nowakeok = ((flags & INPCB_GET_PORTS_USED_NOWAKEUPOK) != 0);
 423         recvanyifonly = ((flags & INPCB_GET_PORTS_USED_RECVANYIFONLY) != 0);
 424         extbgidleok = ((flags & INPCB_GET_PORTS_USED_EXTBGIDLEONLY) != 0);
 425         activeonly = ((flags & INPCB_GET_PORTS_USED_ACTIVEONLY) != 0);
 426
 427         lck_rw_lock_shared(pcbinfo->ipi_lock);
 428         gencnt = pcbinfo->ipi_gencnt;
 429
 430         for (inp = LIST_FIRST(pcbinfo->ipi_listhead); inp;
 431             inp = LIST_NEXT(inp, inp_list)) {
 432                 uint16_t port;
 433
 434                 if (inp->inp_gencnt > gencnt ||
 435                     inp->inp_state == INPCB_STATE_DEAD ||
 436                     inp->inp_wantcnt == WNT_STOPUSING)
 437                         continue;
 438
 439                 if ((so = inp->inp_socket) == NULL ||
 440                     (so->so_state & SS_DEFUNCT) ||
 441                     (so->so_state & SS_ISDISCONNECTED))
 442                         continue;
 443
 444                 if (!(protocol == PF_UNSPEC ||
 445                     (protocol == PF_INET && (inp->inp_vflag & INP_IPV4)) ||
 446                     (protocol == PF_INET6 && (inp->inp_vflag & INP_IPV6))))
 447                         continue;
 448
 449                 iswildcard = (((inp->inp_vflag & INP_IPV4) &&
 450                     inp->inp_laddr.s_addr == INADDR_ANY) ||
 451                     ((inp->inp_vflag & INP_IPV6) &&
 452                     IN6_IS_ADDR_UNSPECIFIED(&inp->in6p_laddr)));
 453
 454                 if (!wildcardok && iswildcard)
 455                         continue;
 456
 457                 if ((so->so_options & SO_NOWAKEFROMSLEEP) &&
 458                         !nowakeok)
 459                         continue;
 460
 461                 if (!(inp->inp_flags & INP_RECV_ANYIF) &&
 462                         recvanyifonly)
 463                         continue;
 464
 465                 if (!(so->so_flags1 & SOF1_EXTEND_BK_IDLE_WANTED) &&
 466                         extbgidleok)
 467                         continue;
 468
 469                 if (!iswildcard &&
 470                     !(ifindex == 0 || inp->inp_last_outifp == NULL ||
 471                     ifindex == inp->inp_last_outifp->if_index))
 472                         continue;
 473
 474                 if (SOCK_PROTO(inp->inp_socket) == IPPROTO_UDP &&
 475                     so->so_state & SS_CANTRCVMORE)
 476                         continue;
 477
 478                 if (SOCK_PROTO(inp->inp_socket) == IPPROTO_TCP) {
 479                         struct  tcpcb *tp = sototcpcb(inp->inp_socket);
 480
 481                         /*
 482                          * Workaround race where inp_ppcb is NULL during
 483                          * socket initialization
 484                          */
 485                         if (tp == NULL)
 486                                 continue;
 487
 488                         switch (tp->t_state) {
 489                                 case TCPS_CLOSED:
 490                                         continue;
 491                                         /* NOT REACHED */
 492                                 case TCPS_LISTEN:
 493                                 case TCPS_SYN_SENT:
 494                                 case TCPS_SYN_RECEIVED:
 495                                 case TCPS_ESTABLISHED:
 496                                 case TCPS_FIN_WAIT_1:
 497                                         /*
 498                                          * Note: FIN_WAIT_1 is an active state
 499                                          * because we need our FIN to be
 500                                          * acknowledged
 501                                          */
 502                                         break;
 503                                 case TCPS_CLOSE_WAIT:
 504                                 case TCPS_CLOSING:
 505                                 case TCPS_LAST_ACK:
 506                                 case TCPS_FIN_WAIT_2:
 507                                         /*
 508                                          * In the closing states, the connection
 509                                          * is not idle when there is outgoing
 510                                          * data having to be acknowledged
 511                                          */
 512                                         if (activeonly && so->so_snd.sb_cc == 0)
 513                                                 continue;
 514                                         break;
 515                                 case TCPS_TIME_WAIT:
 516                                         continue;
 517                                         /* NOT REACHED */
 518                         }
 519                 }
 520                 /*
 521                  * Final safeguard to exclude unspecified local port
 522                  */
 523                 port = ntohs(inp->inp_lport);
 524                 if (port == 0)
 525                         continue;
 526                 bitstr_set(bitfield, port);
 527         }
 528         lck_rw_done(pcbinfo->ipi_lock);
 529 }
 530
 531 __private_extern__ uint32_t
 532 inpcb_count_opportunistic(unsigned int ifindex, struct inpcbinfo *pcbinfo,
 533     u_int32_t flags)
 534 {
 535         uint32_t opportunistic = 0;
 536         struct inpcb *inp;
 537         inp_gen_t gencnt;
 538
 539         lck_rw_lock_shared(pcbinfo->ipi_lock);
 540         gencnt = pcbinfo->ipi_gencnt;
 541         for (inp = LIST_FIRST(pcbinfo->ipi_listhead);
 542             inp != NULL; inp = LIST_NEXT(inp, inp_list)) {
 543                 if (inp->inp_gencnt <= gencnt &&
 544                     inp->inp_state != INPCB_STATE_DEAD &&
 545                     inp->inp_socket != NULL &&
 546                     so_get_opportunistic(inp->inp_socket) &&
 547                     inp->inp_last_outifp != NULL &&
 548                     ifindex == inp->inp_last_outifp->if_index) {
 549                         opportunistic++;
 550                         struct socket *so = inp->inp_socket;
 551                         if ((flags & INPCB_OPPORTUNISTIC_SETCMD) &&
 552                             (so->so_state & SS_ISCONNECTED)) {
 553                                 socket_lock(so, 1);
 554                                 if (flags & INPCB_OPPORTUNISTIC_THROTTLEON) {
 555                                         so->so_flags |= SOF_SUSPENDED;
 556                                         soevent(so,
 557                                             (SO_FILT_HINT_LOCKED |
 558                                             SO_FILT_HINT_SUSPEND));
 559                                 } else {
 560                                         so->so_flags &= ~(SOF_SUSPENDED);
 561                                         soevent(so,
 562                                             (SO_FILT_HINT_LOCKED |
 563                                             SO_FILT_HINT_RESUME));
 564                                 }
 565                                 SOTHROTTLELOG("throttle[%d]: so 0x%llx "
 566                                     "[%d,%d] %s\n", so->last_pid,
 567                                     (uint64_t)VM_KERNEL_ADDRPERM(so),
 568                                     SOCK_DOM(so), SOCK_TYPE(so),
 569                                     (so->so_flags & SOF_SUSPENDED) ?
 570                                     "SUSPENDED" : "RESUMED");
 571                                 socket_unlock(so, 1);
 572                         }
 573                 }
 574         }
 575
 576         lck_rw_done(pcbinfo->ipi_lock);
 577
 578         return (opportunistic);
 579 }
 580
 581 __private_extern__ uint32_t
 582 inpcb_find_anypcb_byaddr(struct ifaddr *ifa, struct inpcbinfo *pcbinfo)
 583 {
 584         struct inpcb *inp;
 585         inp_gen_t gencnt = pcbinfo->ipi_gencnt;
 586         struct socket *so = NULL;
 587         int af;
 588
 589         if ((ifa->ifa_addr->sa_family != AF_INET) &&
 590             (ifa->ifa_addr->sa_family != AF_INET6)) {
 591                 return (0);
 592         }
 593
 594         lck_rw_lock_shared(pcbinfo->ipi_lock);
 595         for (inp = LIST_FIRST(pcbinfo->ipi_listhead);
 596             inp != NULL; inp = LIST_NEXT(inp, inp_list)) {
 597
 598                 if (inp->inp_gencnt <= gencnt &&
 599                     inp->inp_state != INPCB_STATE_DEAD &&
 600                     inp->inp_socket != NULL) {
 601                         so = inp->inp_socket;
 602                         af = SOCK_DOM(so);
 603                         if (af != ifa->ifa_addr->sa_family)
 604                                 continue;
 605                         if (inp->inp_last_outifp != ifa->ifa_ifp)
 606                                 continue;
 607
 608                         if (af == AF_INET) {
 609                                 if (inp->inp_laddr.s_addr ==
 610                                     (satosin(ifa->ifa_addr))->sin_addr.s_addr) {
 611                                         lck_rw_done(pcbinfo->ipi_lock);
 612                                         return (1);
 613                                 }
 614                         }
 615                         if (af == AF_INET6) {
 616                                 if (IN6_ARE_ADDR_EQUAL(IFA_IN6(ifa),
 617                                     &inp->in6p_laddr)) {
 618                                         lck_rw_done(pcbinfo->ipi_lock);
 619                                         return (1);
 620                                 }
 621                         }
 622                 }
 623         }
 624         lck_rw_done(pcbinfo->ipi_lock);
 625         return (0);
 626 }