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 = 0;
 249         xt->cc_recv = 0;
 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         /*
 323          * Special case TCP to include the connections in time wait
 324          */
 325         if (proto == IPPROTO_TCP) {
 326                 n = get_tcp_inp_list(inp_list, n, gencnt);
 327         } else {
 328                 for (inp = pcbinfo->ipi_listhead->lh_first, i = 0; inp && i < n;
 329                      inp = inp->inp_list.le_next) {
 330                         if (inp->inp_gencnt <= gencnt &&
 331                             inp->inp_state != INPCB_STATE_DEAD)
 332                                 inp_list[i++] = inp;
 333                 }
 334                 n = i;
 335         }
 336
 337
 338         error = 0;
 339         for (i = 0; i < n; i++) {
 340                 inp = inp_list[i];
 341                 if (inp->inp_gencnt <= gencnt &&
 342                     inp->inp_state != INPCB_STATE_DEAD) {
 343                         struct xinpcb_n *xi = (struct xinpcb_n *)buf;
 344                         struct xsocket_n *xso = (struct xsocket_n *)
 345                             ADVANCE64(xi, sizeof (*xi));
 346                         struct xsockbuf_n *xsbrcv = (struct xsockbuf_n *)
 347                             ADVANCE64(xso, sizeof (*xso));
 348                         struct xsockbuf_n *xsbsnd = (struct xsockbuf_n *)
 349                             ADVANCE64(xsbrcv, sizeof (*xsbrcv));
 350                         struct xsockstat_n *xsostats = (struct xsockstat_n *)
 351                             ADVANCE64(xsbsnd, sizeof (*xsbsnd));
 352
 353                         bzero(buf, item_size);
 354
 355                         inpcb_to_xinpcb_n(inp, xi);
 356                         sotoxsocket_n(inp->inp_socket, xso);
 357                         sbtoxsockbuf_n(inp->inp_socket ?
 358                             &inp->inp_socket->so_rcv : NULL, xsbrcv);
 359                         sbtoxsockbuf_n(inp->inp_socket ?
 360                             &inp->inp_socket->so_snd : NULL, xsbsnd);
 361                         sbtoxsockstat_n(inp->inp_socket, xsostats);
 362                         if (proto == IPPROTO_TCP) {
 363                                 struct  xtcpcb_n *xt = (struct xtcpcb_n *)
 364                                     ADVANCE64(xsostats, sizeof (*xsostats));
 365
 366                                 /*
 367                                  * inp->inp_ppcb, can only be NULL on
 368                                  * an initialization race window.
 369                                  * No need to lock.
 370                                  */
 371                                 if (inp->inp_ppcb == NULL)
 372                                         continue;
 373
 374                                 tcpcb_to_xtcpcb_n((struct tcpcb *)
 375                                     inp->inp_ppcb, xt);
 376                         }
 377                         error = SYSCTL_OUT(req, buf, item_size);
 378                 }
 379         }
 380         if (!error) {
 381                 /*
 382                  * Give the user an updated idea of our state.
 383                  * If the generation differs from what we told
 384                  * her before, she knows that something happened
 385                  * while we were processing this request, and it
 386                  * might be necessary to retry.
 387                  */
 388                 bzero(&xig, sizeof (xig));
 389                 xig.xig_len = sizeof (xig);
 390                 xig.xig_gen = pcbinfo->ipi_gencnt;
 391                 xig.xig_sogen = so_gencnt;
 392                 xig.xig_count = pcbinfo->ipi_count;
 393                 error = SYSCTL_OUT(req, &xig, sizeof (xig));
 394         }
 395 done:
 396         lck_rw_done(pcbinfo->ipi_lock);
 397         if (inp_list != NULL)
 398                 FREE(inp_list, M_TEMP);
 399         if (buf != NULL)
 400                 FREE(buf, M_TEMP);
 401         return (error);
 402 }
 403
 404 __private_extern__ void
 405 inpcb_get_ports_used(uint32_t ifindex, int protocol, uint32_t flags,
 406     bitstr_t *bitfield, struct inpcbinfo *pcbinfo)
 407 {
 408         struct inpcb *inp;
 409         struct socket *so;
 410         inp_gen_t gencnt;
 411         bool iswildcard, wildcardok, nowakeok;
 412         bool recvanyifonly, extbgidleok;
 413         bool activeonly;
 414
 415         wildcardok = ((flags & INPCB_GET_PORTS_USED_WILDCARDOK) != 0);
 416         nowakeok = ((flags & INPCB_GET_PORTS_USED_NOWAKEUPOK) != 0);
 417         recvanyifonly = ((flags & INPCB_GET_PORTS_USED_RECVANYIFONLY) != 0);
 418         extbgidleok = ((flags & INPCB_GET_PORTS_USED_EXTBGIDLEONLY) != 0);
 419         activeonly = ((flags & INPCB_GET_PORTS_USED_ACTIVEONLY) != 0);
 420
 421         lck_rw_lock_shared(pcbinfo->ipi_lock);
 422         gencnt = pcbinfo->ipi_gencnt;
 423
 424         for (inp = LIST_FIRST(pcbinfo->ipi_listhead); inp;
 425             inp = LIST_NEXT(inp, inp_list)) {
 426                 uint16_t port;
 427
 428                 if (inp->inp_gencnt > gencnt ||
 429                     inp->inp_state == INPCB_STATE_DEAD ||
 430                     inp->inp_wantcnt == WNT_STOPUSING)
 431                         continue;
 432
 433                 if ((so = inp->inp_socket) == NULL ||
 434                     (so->so_state & SS_DEFUNCT) ||
 435                     (so->so_state & SS_ISDISCONNECTED))
 436                         continue;
 437
 438                 if (!(protocol == PF_UNSPEC ||
 439                     (protocol == PF_INET && (inp->inp_vflag & INP_IPV4)) ||
 440                     (protocol == PF_INET6 && (inp->inp_vflag & INP_IPV6))))
 441                         continue;
 442
 443                 iswildcard = (((inp->inp_vflag & INP_IPV4) &&
 444                     inp->inp_laddr.s_addr == INADDR_ANY) ||
 445                     ((inp->inp_vflag & INP_IPV6) &&
 446                     IN6_IS_ADDR_UNSPECIFIED(&inp->in6p_laddr)));
 447
 448                 if (!wildcardok && iswildcard)
 449                         continue;
 450
 451                 if ((so->so_options & SO_NOWAKEFROMSLEEP) &&
 452                         !nowakeok)
 453                         continue;
 454
 455                 if (!(inp->inp_flags & INP_RECV_ANYIF) &&
 456                         recvanyifonly)
 457                         continue;
 458
 459                 if (!(so->so_flags1 & SOF1_EXTEND_BK_IDLE_WANTED) &&
 460                         extbgidleok)
 461                         continue;
 462
 463                 if (!iswildcard &&
 464                     !(ifindex == 0 || inp->inp_last_outifp == NULL ||
 465                     ifindex == inp->inp_last_outifp->if_index))
 466                         continue;
 467
 468                 if (SOCK_PROTO(inp->inp_socket) == IPPROTO_UDP &&
 469                     so->so_state & SS_CANTRCVMORE)
 470                         continue;
 471
 472                 if (SOCK_PROTO(inp->inp_socket) == IPPROTO_TCP) {
 473                         struct  tcpcb *tp = sototcpcb(inp->inp_socket);
 474
 475                         /*
 476                          * Workaround race where inp_ppcb is NULL during
 477                          * socket initialization
 478                          */
 479                         if (tp == NULL)
 480                                 continue;
 481
 482                         switch (tp->t_state) {
 483                                 case TCPS_CLOSED:
 484                                         continue;
 485                                         /* NOT REACHED */
 486                                 case TCPS_LISTEN:
 487                                 case TCPS_SYN_SENT:
 488                                 case TCPS_SYN_RECEIVED:
 489                                 case TCPS_ESTABLISHED:
 490                                 case TCPS_FIN_WAIT_1:
 491                                         /*
 492                                          * Note: FIN_WAIT_1 is an active state
 493                                          * because we need our FIN to be
 494                                          * acknowledged
 495                                          */
 496                                         break;
 497                                 case TCPS_CLOSE_WAIT:
 498                                 case TCPS_CLOSING:
 499                                 case TCPS_LAST_ACK:
 500                                 case TCPS_FIN_WAIT_2:
 501                                         /*
 502                                          * In the closing states, the connection
 503                                          * is not idle when there is outgoing
 504                                          * data having to be acknowledged
 505                                          */
 506                                         if (activeonly && so->so_snd.sb_cc == 0)
 507                                                 continue;
 508                                         break;
 509                                 case TCPS_TIME_WAIT:
 510                                         continue;
 511                                         /* NOT REACHED */
 512                         }
 513                 }
 514                 /*
 515                  * Final safeguard to exclude unspecified local port
 516                  */
 517                 port = ntohs(inp->inp_lport);
 518                 if (port == 0)
 519                         continue;
 520                 bit_set(bitfield, port);
 521         }
 522         lck_rw_done(pcbinfo->ipi_lock);
 523 }
 524
 525 __private_extern__ uint32_t
 526 inpcb_count_opportunistic(unsigned int ifindex, struct inpcbinfo *pcbinfo,
 527     u_int32_t flags)
 528 {
 529         uint32_t opportunistic = 0;
 530         struct inpcb *inp;
 531         inp_gen_t gencnt;
 532
 533         lck_rw_lock_shared(pcbinfo->ipi_lock);
 534         gencnt = pcbinfo->ipi_gencnt;
 535         for (inp = LIST_FIRST(pcbinfo->ipi_listhead);
 536             inp != NULL; inp = LIST_NEXT(inp, inp_list)) {
 537                 if (inp->inp_gencnt <= gencnt &&
 538                     inp->inp_state != INPCB_STATE_DEAD &&
 539                     inp->inp_socket != NULL &&
 540                     so_get_opportunistic(inp->inp_socket) &&
 541                     inp->inp_last_outifp != NULL &&
 542                     ifindex == inp->inp_last_outifp->if_index) {
 543                         opportunistic++;
 544                         struct socket *so = inp->inp_socket;
 545                         if ((flags & INPCB_OPPORTUNISTIC_SETCMD) &&
 546                             (so->so_state & SS_ISCONNECTED)) {
 547                                 socket_lock(so, 1);
 548                                 if (flags & INPCB_OPPORTUNISTIC_THROTTLEON) {
 549                                         so->so_flags |= SOF_SUSPENDED;
 550                                         soevent(so,
 551                                             (SO_FILT_HINT_LOCKED |
 552                                             SO_FILT_HINT_SUSPEND));
 553                                 } else {
 554                                         so->so_flags &= ~(SOF_SUSPENDED);
 555                                         soevent(so,
 556                                             (SO_FILT_HINT_LOCKED |
 557                                             SO_FILT_HINT_RESUME));
 558                                 }
 559                                 SOTHROTTLELOG("throttle[%d]: so 0x%llx "
 560                                     "[%d,%d] %s\n", so->last_pid,
 561                                     (uint64_t)VM_KERNEL_ADDRPERM(so),
 562                                     SOCK_DOM(so), SOCK_TYPE(so),
 563                                     (so->so_flags & SOF_SUSPENDED) ?
 564                                     "SUSPENDED" : "RESUMED");
 565                                 socket_unlock(so, 1);
 566                         }
 567                 }
 568         }
 569
 570         lck_rw_done(pcbinfo->ipi_lock);
 571
 572         return (opportunistic);
 573 }
 574
 575 __private_extern__ uint32_t
 576 inpcb_find_anypcb_byaddr(struct ifaddr *ifa, struct inpcbinfo *pcbinfo)
 577 {
 578         struct inpcb *inp;
 579         inp_gen_t gencnt = pcbinfo->ipi_gencnt;
 580         struct socket *so = NULL;
 581         int af;
 582
 583         if ((ifa->ifa_addr->sa_family != AF_INET) &&
 584             (ifa->ifa_addr->sa_family != AF_INET6)) {
 585                 return (0);
 586         }
 587
 588         lck_rw_lock_shared(pcbinfo->ipi_lock);
 589         for (inp = LIST_FIRST(pcbinfo->ipi_listhead);
 590             inp != NULL; inp = LIST_NEXT(inp, inp_list)) {
 591
 592                 if (inp->inp_gencnt <= gencnt &&
 593                     inp->inp_state != INPCB_STATE_DEAD &&
 594                     inp->inp_socket != NULL) {
 595                         so = inp->inp_socket;
 596                         af = SOCK_DOM(so);
 597                         if (af != ifa->ifa_addr->sa_family)
 598                                 continue;
 599                         if (inp->inp_last_outifp != ifa->ifa_ifp)
 600                                 continue;
 601
 602                         if (af == AF_INET) {
 603                                 if (inp->inp_laddr.s_addr ==
 604                                     (satosin(ifa->ifa_addr))->sin_addr.s_addr) {
 605                                         lck_rw_done(pcbinfo->ipi_lock);
 606                                         return (1);
 607                                 }
 608                         }
 609                         if (af == AF_INET6) {
 610                                 if (IN6_ARE_ADDR_EQUAL(IFA_IN6(ifa),
 611                                     &inp->in6p_laddr)) {
 612                                         lck_rw_done(pcbinfo->ipi_lock);
 613                                         return (1);
 614                                 }
 615                         }
 616                 }
 617         }
 618         lck_rw_done(pcbinfo->ipi_lock);
 619         return (0);
 620 }