sys/net/netisr.c

   1 /*
   2  * Copyright (c) 2003, 2004 Matthew Dillon. All rights reserved.
   3  * Copyright (c) 2003, 2004 Jeffrey M. Hsu.  All rights reserved.
   4  * Copyright (c) 2003 Jonathan Lemon.  All rights reserved.
   5  * Copyright (c) 2003, 2004 The DragonFly Project.  All rights reserved.
   6  *
   7  * This code is derived from software contributed to The DragonFly Project
   8  * by Jonathan Lemon, Jeffrey M. Hsu, and Matthew Dillon.
   9  *
  10  * Jonathan Lemon gave Jeffrey Hsu permission to combine his copyright
  11  * into this one around July 8 2004.
  12  *
  13  * Redistribution and use in source and binary forms, with or without
  14  * modification, are permitted provided that the following conditions
  15  * are met:
  16  * 1. Redistributions of source code must retain the above copyright
  17  *    notice, this list of conditions and the following disclaimer.
  18  * 2. Redistributions in binary form must reproduce the above copyright
  19  *    notice, this list of conditions and the following disclaimer in the
  20  *    documentation and/or other materials provided with the distribution.
  21  * 3. Neither the name of The DragonFly Project nor the names of its
  22  *    contributors may be used to endorse or promote products derived
  23  *    from this software without specific, prior written permission.
  24  *
  25  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  26  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  27  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
  28  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE
  29  * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
  30  * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
  31  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  32  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
  33  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  34  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
  35  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  36  * SUCH DAMAGE.
  37  */
  38
  39 #include <sys/param.h>
  40 #include <sys/systm.h>
  41 #include <sys/kernel.h>
  42 #include <sys/malloc.h>
  43 #include <sys/msgport.h>
  44 #include <sys/proc.h>
  45 #include <sys/interrupt.h>
  46 #include <sys/socket.h>
  47 #include <sys/sysctl.h>
  48 #include <sys/socketvar.h>
  49 #include <net/if.h>
  50 #include <net/if_var.h>
  51 #include <net/netisr2.h>
  52 #include <machine/cpufunc.h>
  53 #include <machine/smp.h>
  54
  55 #include <sys/thread2.h>
  56 #include <sys/msgport2.h>
  57 #include <net/netmsg2.h>
  58 #include <sys/mplock2.h>
  59
  60 static void netmsg_service_loop(void *arg);
  61 static void netisr_hashfn0(struct mbuf **mp, int hoff);
  62 static void netisr_nohashck(struct mbuf *, const struct pktinfo *);
  63
  64 struct netmsg_port_registration {
  65         TAILQ_ENTRY(netmsg_port_registration) npr_entry;
  66         lwkt_port_t     npr_port;
  67 };
  68
  69 struct netmsg_rollup {
  70         TAILQ_ENTRY(netmsg_rollup) ru_entry;
  71         netisr_ru_t     ru_func;
  72         int             ru_prio;
  73 };
  74
  75 struct netmsg_barrier {
  76         struct netmsg_base      base;
  77         volatile cpumask_t      *br_cpumask;
  78         volatile uint32_t       br_done;
  79 };
  80
  81 #define NETISR_BR_NOTDONE       0x1
  82 #define NETISR_BR_WAITDONE      0x80000000
  83
  84 struct netisr_barrier {
  85         struct netmsg_barrier   *br_msgs[MAXCPU];
  86         int                     br_isset;
  87 };
  88
  89 static struct netisr netisrs[NETISR_MAX];
  90 static TAILQ_HEAD(,netmsg_port_registration) netreglist;
  91 static TAILQ_HEAD(,netmsg_rollup) netrulist;
  92
  93 /* Per-CPU thread to handle any protocol.  */
  94 struct thread netisr_cpu[MAXCPU];
  95 lwkt_port netisr_afree_rport;
  96 lwkt_port netisr_afree_free_so_rport;
  97 lwkt_port netisr_adone_rport;
  98 lwkt_port netisr_apanic_rport;
  99 lwkt_port netisr_sync_port;
 100
 101 static int (*netmsg_fwd_port_fn)(lwkt_port_t, lwkt_msg_t);
 102
 103 SYSCTL_NODE(_net, OID_AUTO, netisr, CTLFLAG_RW, 0, "netisr");
 104
 105 /*
 106  * netisr_afree_rport replymsg function, only used to handle async
 107  * messages which the sender has abandoned to their fate.
 108  */
 109 static void
 110 netisr_autofree_reply(lwkt_port_t port, lwkt_msg_t msg)
 111 {
 112         kfree(msg, M_LWKTMSG);
 113 }
 114
 115 static void
 116 netisr_autofree_free_so_reply(lwkt_port_t port, lwkt_msg_t msg)
 117 {
 118         sofree(((netmsg_t)msg)->base.nm_so);
 119         kfree(msg, M_LWKTMSG);
 120 }
 121
 122 /*
 123  * We need a custom putport function to handle the case where the
 124  * message target is the current thread's message port.  This case
 125  * can occur when the TCP or UDP stack does a direct callback to NFS and NFS
 126  * then turns around and executes a network operation synchronously.
 127  *
 128  * To prevent deadlocking, we must execute these self-referential messages
 129  * synchronously, effectively turning the message into a glorified direct
 130  * procedure call back into the protocol stack.  The operation must be
 131  * complete on return or we will deadlock, so panic if it isn't.
 132  *
 133  * However, the target function is under no obligation to immediately
 134  * reply the message.  It may forward it elsewhere.
 135  */
 136 static int
 137 netmsg_put_port(lwkt_port_t port, lwkt_msg_t lmsg)
 138 {
 139         netmsg_base_t nmsg = (void *)lmsg;
 140
 141         if ((lmsg->ms_flags & MSGF_SYNC) && port == &curthread->td_msgport) {
 142                 nmsg->nm_dispatch((netmsg_t)nmsg);
 143                 return(EASYNC);
 144         } else {
 145                 return(netmsg_fwd_port_fn(port, lmsg));
 146         }
 147 }
 148
 149 /*
 150  * UNIX DOMAIN sockets still have to run their uipc functions synchronously,
 151  * because they depend on the user proc context for a number of things
 152  * (like creds) which we have not yet incorporated into the message structure.
 153  *
 154  * However, we maintain or message/port abstraction.  Having a special
 155  * synchronous port which runs the commands synchronously gives us the
 156  * ability to serialize operations in one place later on when we start
 157  * removing the BGL.
 158  */
 159 static int
 160 netmsg_sync_putport(lwkt_port_t port, lwkt_msg_t lmsg)
 161 {
 162         netmsg_base_t nmsg = (void *)lmsg;
 163
 164         KKASSERT((lmsg->ms_flags & MSGF_DONE) == 0);
 165
 166         lmsg->ms_target_port = port;    /* required for abort */
 167         nmsg->nm_dispatch((netmsg_t)nmsg);
 168         return(EASYNC);
 169 }
 170
 171 static void
 172 netisr_init(void)
 173 {
 174         int i;
 175
 176         TAILQ_INIT(&netreglist);
 177         TAILQ_INIT(&netrulist);
 178
 179         /*
 180          * Create default per-cpu threads for generic protocol handling.
 181          */
 182         for (i = 0; i < ncpus; ++i) {
 183                 lwkt_create(netmsg_service_loop, NULL, NULL,
 184                             &netisr_cpu[i],
 185                             TDF_NOSTART|TDF_FORCE_SPINPORT|TDF_FIXEDCPU,
 186                             i, "netisr_cpu %d", i);
 187                 netmsg_service_port_init(&netisr_cpu[i].td_msgport);
 188                 lwkt_schedule(&netisr_cpu[i]);
 189         }
 190
 191         /*
 192          * The netisr_afree_rport is a special reply port which automatically
 193          * frees the replied message.  The netisr_adone_rport simply marks
 194          * the message as being done.  The netisr_apanic_rport panics if
 195          * the message is replied to.
 196          */
 197         lwkt_initport_replyonly(&netisr_afree_rport, netisr_autofree_reply);
 198         lwkt_initport_replyonly(&netisr_afree_free_so_rport,
 199                                 netisr_autofree_free_so_reply);
 200         lwkt_initport_replyonly_null(&netisr_adone_rport);
 201         lwkt_initport_panic(&netisr_apanic_rport);
 202
 203         /*
 204          * The netisr_syncport is a special port which executes the message
 205          * synchronously and waits for it if EASYNC is returned.
 206          */
 207         lwkt_initport_putonly(&netisr_sync_port, netmsg_sync_putport);
 208 }
 209
 210 SYSINIT(netisr, SI_SUB_PRE_DRIVERS, SI_ORDER_FIRST, netisr_init, NULL);
 211
 212 /*
 213  * Finish initializing the message port for a netmsg service.  This also
 214  * registers the port for synchronous cleanup operations such as when an
 215  * ifnet is being destroyed.  There is no deregistration API yet.
 216  */
 217 void
 218 netmsg_service_port_init(lwkt_port_t port)
 219 {
 220         struct netmsg_port_registration *reg;
 221
 222         /*
 223          * Override the putport function.  Our custom function checks for
 224          * self-references and executes such commands synchronously.
 225          */
 226         if (netmsg_fwd_port_fn == NULL)
 227                 netmsg_fwd_port_fn = port->mp_putport;
 228         KKASSERT(netmsg_fwd_port_fn == port->mp_putport);
 229         port->mp_putport = netmsg_put_port;
 230
 231         /*
 232          * Keep track of ports using the netmsg API so we can synchronize
 233          * certain operations (such as freeing an ifnet structure) across all
 234          * consumers.
 235          */
 236         reg = kmalloc(sizeof(*reg), M_TEMP, M_WAITOK|M_ZERO);
 237         reg->npr_port = port;
 238         TAILQ_INSERT_TAIL(&netreglist, reg, npr_entry);
 239 }
 240
 241 /*
 242  * This function synchronizes the caller with all netmsg services.  For
 243  * example, if an interface is being removed we must make sure that all
 244  * packets related to that interface complete processing before the structure
 245  * can actually be freed.  This sort of synchronization is an alternative to
 246  * ref-counting the netif, removing the ref counting overhead in favor of
 247  * placing additional overhead in the netif freeing sequence (where it is
 248  * inconsequential).
 249  */
 250 void
 251 netmsg_service_sync(void)
 252 {
 253         struct netmsg_port_registration *reg;
 254         struct netmsg_base smsg;
 255
 256         netmsg_init(&smsg, NULL, &curthread->td_msgport, 0, netmsg_sync_handler);
 257
 258         TAILQ_FOREACH(reg, &netreglist, npr_entry) {
 259                 lwkt_domsg(reg->npr_port, &smsg.lmsg, 0);
 260         }
 261 }
 262
 263 /*
 264  * The netmsg function simply replies the message.  API semantics require
 265  * EASYNC to be returned if the netmsg function disposes of the message.
 266  */
 267 void
 268 netmsg_sync_handler(netmsg_t msg)
 269 {
 270         lwkt_replymsg(&msg->lmsg, 0);
 271 }
 272
 273 /*
 274  * Generic netmsg service loop.  Some protocols may roll their own but all
 275  * must do the basic command dispatch function call done here.
 276  */
 277 static void
 278 netmsg_service_loop(void *arg)
 279 {
 280         struct netmsg_rollup *ru;
 281         netmsg_base_t msg;
 282         thread_t td = curthread;
 283         int limit;
 284
 285         td->td_type = TD_TYPE_NETISR;
 286
 287         while ((msg = lwkt_waitport(&td->td_msgport, 0))) {
 288                 /*
 289                  * Run up to 512 pending netmsgs.
 290                  */
 291                 limit = 512;
 292                 do {
 293                         KASSERT(msg->nm_dispatch != NULL,
 294                                 ("netmsg_service isr %d badmsg",
 295                                 msg->lmsg.u.ms_result));
 296                         /*
 297                          * Don't match so_port, if the msg explicitly
 298                          * asks us to ignore its so_port.
 299                          */
 300                         if ((msg->lmsg.ms_flags & MSGF_IGNSOPORT) == 0 &&
 301                             msg->nm_so &&
 302                             msg->nm_so->so_port != &td->td_msgport) {
 303                                 /*
 304                                  * Sockets undergoing connect or disconnect
 305                                  * ops can change ports on us.  Chase the
 306                                  * port.
 307                                  */
 308 #ifdef foo
 309                                 /*
 310                                  * This could be quite common for protocols
 311                                  * which support asynchronous pru_connect,
 312                                  * e.g. TCP, so kprintf socket port chasing
 313                                  * could be too verbose for the console.
 314                                  */
 315                                 kprintf("netmsg_service_loop: Warning, "
 316                                         "port changed so=%p\n", msg->nm_so);
 317 #endif
 318                                 lwkt_forwardmsg(msg->nm_so->so_port,
 319                                                 &msg->lmsg);
 320                         } else {
 321                                 /*
 322                                  * We are on the correct port, dispatch it.
 323                                  */
 324                                 msg->nm_dispatch((netmsg_t)msg);
 325                         }
 326                         if (--limit == 0)
 327                                 break;
 328                 } while ((msg = lwkt_getport(&td->td_msgport)) != NULL);
 329
 330                 /*
 331                  * Run all registered rollup functions for this cpu
 332                  * (e.g. tcp_willblock()).
 333                  */
 334                 TAILQ_FOREACH(ru, &netrulist, ru_entry)
 335                         ru->ru_func();
 336         }
 337 }
 338
 339 /*
 340  * Forward a packet to a netisr service function.
 341  *
 342  * If the packet has not been assigned to a protocol thread we call
 343  * the port characterization function to assign it.  The caller must
 344  * clear M_HASH (or not have set it in the first place) if the caller
 345  * wishes the packet to be recharacterized.
 346  */
 347 int
 348 netisr_queue(int num, struct mbuf *m)
 349 {
 350         struct netisr *ni;
 351         struct netmsg_packet *pmsg;
 352         lwkt_port_t port;
 353
 354         KASSERT((num > 0 && num <= NELEM(netisrs)),
 355                 ("Bad isr %d", num));
 356
 357         ni = &netisrs[num];
 358         if (ni->ni_handler == NULL) {
 359                 kprintf("Unregistered isr %d\n", num);
 360                 m_freem(m);
 361                 return (EIO);
 362         }
 363
 364         /*
 365          * Figure out which protocol thread to send to.  This does not
 366          * have to be perfect but performance will be really good if it
 367          * is correct.  Major protocol inputs such as ip_input() will
 368          * re-characterize the packet as necessary.
 369          */
 370         if ((m->m_flags & M_HASH) == 0) {
 371                 ni->ni_hashfn(&m, 0);
 372                 if (m == NULL)
 373                         return (EIO);
 374                 if ((m->m_flags & M_HASH) == 0) {
 375                         kprintf("netisr_queue(%d): packet hash failed\n", num);
 376                         m_freem(m);
 377                         return (EIO);
 378                 }
 379         }
 380
 381         /*
 382          * Get the protocol port based on the packet hash, initialize
 383          * the netmsg, and send it off.
 384          */
 385         port = netisr_hashport(m->m_pkthdr.hash);
 386         pmsg = &m->m_hdr.mh_netmsg;
 387         netmsg_init(&pmsg->base, NULL, &netisr_apanic_rport,
 388                     0, ni->ni_handler);
 389         pmsg->nm_packet = m;
 390         pmsg->base.lmsg.u.ms_result = num;
 391         lwkt_sendmsg(port, &pmsg->base.lmsg);
 392
 393         return (0);
 394 }
 395
 396 /*
 397  * Run a netisr service function on the packet.
 398  *
 399  * The packet must have been correctly characterized!
 400  */
 401 int
 402 netisr_handle(int num, struct mbuf *m)
 403 {
 404         struct netisr *ni;
 405         struct netmsg_packet *pmsg;
 406         lwkt_port_t port;
 407
 408         /*
 409          * Get the protocol port based on the packet hash
 410          */
 411         KASSERT((m->m_flags & M_HASH), ("packet not characterized"));
 412         port = netisr_hashport(m->m_pkthdr.hash);
 413         KASSERT(&curthread->td_msgport == port, ("wrong msgport"));
 414
 415         KASSERT((num > 0 && num <= NELEM(netisrs)), ("bad isr %d", num));
 416         ni = &netisrs[num];
 417         if (ni->ni_handler == NULL) {
 418                 kprintf("unregistered isr %d\n", num);
 419                 m_freem(m);
 420                 return EIO;
 421         }
 422
 423         /*
 424          * Initialize the netmsg, and run the handler directly.
 425          */
 426         pmsg = &m->m_hdr.mh_netmsg;
 427         netmsg_init(&pmsg->base, NULL, &netisr_apanic_rport,
 428                     0, ni->ni_handler);
 429         pmsg->nm_packet = m;
 430         pmsg->base.lmsg.u.ms_result = num;
 431         ni->ni_handler((netmsg_t)&pmsg->base);
 432
 433         return 0;
 434 }
 435
 436 /*
 437  * Pre-characterization of a deeper portion of the packet for the
 438  * requested isr.
 439  *
 440  * The base of the ISR type (e.g. IP) that we want to characterize is
 441  * at (hoff) relative to the beginning of the mbuf.  This allows
 442  * e.g. ether_characterize() to not have to adjust the m_data/m_len.
 443  */
 444 void
 445 netisr_characterize(int num, struct mbuf **mp, int hoff)
 446 {
 447         struct netisr *ni;
 448         struct mbuf *m;
 449
 450         /*
 451          * Validation
 452          */
 453         m = *mp;
 454         KKASSERT(m != NULL);
 455
 456         if (num < 0 || num >= NETISR_MAX) {
 457                 if (num == NETISR_MAX) {
 458                         m->m_flags |= M_HASH;
 459                         m->m_pkthdr.hash = 0;
 460                         return;
 461                 }
 462                 panic("Bad isr %d", num);
 463         }
 464
 465         /*
 466          * Valid netisr?
 467          */
 468         ni = &netisrs[num];
 469         if (ni->ni_handler == NULL) {
 470                 kprintf("Unregistered isr %d\n", num);
 471                 m_freem(m);
 472                 *mp = NULL;
 473         }
 474
 475         /*
 476          * Characterize the packet
 477          */
 478         if ((m->m_flags & M_HASH) == 0) {
 479                 ni->ni_hashfn(mp, hoff);
 480                 m = *mp;
 481                 if (m && (m->m_flags & M_HASH) == 0)
 482                         kprintf("netisr_queue(%d): packet hash failed\n", num);
 483         }
 484 }
 485
 486 void
 487 netisr_register(int num, netisr_fn_t handler, netisr_hashfn_t hashfn)
 488 {
 489         struct netisr *ni;
 490
 491         KASSERT((num > 0 && num <= NELEM(netisrs)),
 492                 ("netisr_register: bad isr %d", num));
 493         KKASSERT(handler != NULL);
 494
 495         if (hashfn == NULL)
 496                 hashfn = netisr_hashfn0;
 497
 498         ni = &netisrs[num];
 499
 500         ni->ni_handler = handler;
 501         ni->ni_hashck = netisr_nohashck;
 502         ni->ni_hashfn = hashfn;
 503         netmsg_init(&ni->ni_netmsg, NULL, &netisr_adone_rport, 0, NULL);
 504 }
 505
 506 void
 507 netisr_register_hashcheck(int num, netisr_hashck_t hashck)
 508 {
 509         struct netisr *ni;
 510
 511         KASSERT((num > 0 && num <= NELEM(netisrs)),
 512                 ("netisr_register: bad isr %d", num));
 513
 514         ni = &netisrs[num];
 515         ni->ni_hashck = hashck;
 516 }
 517
 518 void
 519 netisr_register_rollup(netisr_ru_t ru_func, int prio)
 520 {
 521         struct netmsg_rollup *new_ru, *ru;
 522
 523         new_ru = kmalloc(sizeof(*new_ru), M_TEMP, M_WAITOK|M_ZERO);
 524         new_ru->ru_func = ru_func;
 525         new_ru->ru_prio = prio;
 526
 527         /*
 528          * Higher priority "rollup" appears first
 529          */
 530         TAILQ_FOREACH(ru, &netrulist, ru_entry) {
 531                 if (ru->ru_prio < new_ru->ru_prio) {
 532                         TAILQ_INSERT_BEFORE(ru, new_ru, ru_entry);
 533                         return;
 534                 }
 535         }
 536         TAILQ_INSERT_TAIL(&netrulist, new_ru, ru_entry);
 537 }
 538
 539 /*
 540  * Return a default protocol control message processing thread port
 541  */
 542 lwkt_port_t
 543 cpu0_ctlport(int cmd __unused, struct sockaddr *sa __unused,
 544              void *extra __unused)
 545 {
 546         return (&netisr_cpu[0].td_msgport);
 547 }
 548
 549 /*
 550  * This is a default netisr packet characterization function which
 551  * sets M_HASH.  If a netisr is registered with a NULL hashfn function
 552  * this one is assigned.
 553  *
 554  * This function makes no attempt to validate the packet.
 555  */
 556 static void
 557 netisr_hashfn0(struct mbuf **mp, int hoff __unused)
 558 {
 559         struct mbuf *m = *mp;
 560
 561         m->m_flags |= M_HASH;
 562         m->m_pkthdr.hash = 0;
 563 }
 564
 565 /*
 566  * schednetisr() is used to call the netisr handler from the appropriate
 567  * netisr thread for polling and other purposes.
 568  *
 569  * This function may be called from a hard interrupt or IPI and must be
 570  * MP SAFE and non-blocking.  We use a fixed per-cpu message instead of
 571  * trying to allocate one.  We must get ourselves onto the target cpu
 572  * to safely check the MSGF_DONE bit on the message but since the message
 573  * will be sent to that cpu anyway this does not add any extra work beyond
 574  * what lwkt_sendmsg() would have already had to do to schedule the target
 575  * thread.
 576  */
 577 static void
 578 schednetisr_remote(void *data)
 579 {
 580         int num = (int)(intptr_t)data;
 581         struct netisr *ni = &netisrs[num];
 582         lwkt_port_t port = &netisr_cpu[0].td_msgport;
 583         netmsg_base_t pmsg;
 584
 585         pmsg = &netisrs[num].ni_netmsg;
 586         if (pmsg->lmsg.ms_flags & MSGF_DONE) {
 587                 netmsg_init(pmsg, NULL, &netisr_adone_rport, 0, ni->ni_handler);
 588                 pmsg->lmsg.u.ms_result = num;
 589                 lwkt_sendmsg(port, &pmsg->lmsg);
 590         }
 591 }
 592
 593 void
 594 schednetisr(int num)
 595 {
 596         KASSERT((num > 0 && num <= NELEM(netisrs)),
 597                 ("schednetisr: bad isr %d", num));
 598         KKASSERT(netisrs[num].ni_handler != NULL);
 599         if (mycpu->gd_cpuid != 0) {
 600                 lwkt_send_ipiq(globaldata_find(0),
 601                                schednetisr_remote, (void *)(intptr_t)num);
 602         } else {
 603                 crit_enter();
 604                 schednetisr_remote((void *)(intptr_t)num);
 605                 crit_exit();
 606         }
 607 }
 608
 609 static void
 610 netisr_barrier_dispatch(netmsg_t nmsg)
 611 {
 612         struct netmsg_barrier *msg = (struct netmsg_barrier *)nmsg;
 613
 614         atomic_clear_cpumask(msg->br_cpumask, mycpu->gd_cpumask);
 615         if (*msg->br_cpumask == 0)
 616                 wakeup(msg->br_cpumask);
 617
 618         for (;;) {
 619                 uint32_t done = msg->br_done;
 620
 621                 cpu_ccfence();
 622                 if ((done & NETISR_BR_NOTDONE) == 0)
 623                         break;
 624
 625                 tsleep_interlock(&msg->br_done, 0);
 626                 if (atomic_cmpset_int(&msg->br_done,
 627                     done, done | NETISR_BR_WAITDONE))
 628                         tsleep(&msg->br_done, PINTERLOCKED, "nbrdsp", 0);
 629         }
 630
 631         lwkt_replymsg(&nmsg->lmsg, 0);
 632 }
 633
 634 struct netisr_barrier *
 635 netisr_barrier_create(void)
 636 {
 637         struct netisr_barrier *br;
 638
 639         br = kmalloc(sizeof(*br), M_LWKTMSG, M_WAITOK | M_ZERO);
 640         return br;
 641 }
 642
 643 void
 644 netisr_barrier_set(struct netisr_barrier *br)
 645 {
 646         volatile cpumask_t other_cpumask;
 647         int i, cur_cpuid;
 648
 649         KKASSERT(&curthread->td_msgport == netisr_cpuport(0));
 650         KKASSERT(!br->br_isset);
 651
 652         other_cpumask = mycpu->gd_other_cpus & smp_active_mask;
 653         cur_cpuid = mycpuid;
 654
 655         for (i = 0; i < ncpus; ++i) {
 656                 struct netmsg_barrier *msg;
 657
 658                 if (i == cur_cpuid)
 659                         continue;
 660
 661                 msg = kmalloc(sizeof(struct netmsg_barrier),
 662                               M_LWKTMSG, M_WAITOK);
 663
 664                 /*
 665                  * Don't use priority message here; mainly to keep
 666                  * it ordered w/ the previous data packets sent by
 667                  * the caller.
 668                  */
 669                 netmsg_init(&msg->base, NULL, &netisr_afree_rport, 0,
 670                     netisr_barrier_dispatch);
 671                 msg->br_cpumask = &other_cpumask;
 672                 msg->br_done = NETISR_BR_NOTDONE;
 673
 674                 KKASSERT(br->br_msgs[i] == NULL);
 675                 br->br_msgs[i] = msg;
 676         }
 677
 678         for (i = 0; i < ncpus; ++i) {
 679                 if (i == cur_cpuid)
 680                         continue;
 681                 lwkt_sendmsg(netisr_cpuport(i), &br->br_msgs[i]->base.lmsg);
 682         }
 683
 684         while (other_cpumask != 0) {
 685                 tsleep_interlock(&other_cpumask, 0);
 686                 if (other_cpumask != 0)
 687                         tsleep(&other_cpumask, PINTERLOCKED, "nbrset", 0);
 688         }
 689         br->br_isset = 1;
 690 }
 691
 692 void
 693 netisr_barrier_rem(struct netisr_barrier *br)
 694 {
 695         int i, cur_cpuid;
 696
 697         KKASSERT(&curthread->td_msgport == netisr_cpuport(0));
 698         KKASSERT(br->br_isset);
 699
 700         cur_cpuid = mycpuid;
 701         for (i = 0; i < ncpus; ++i) {
 702                 struct netmsg_barrier *msg = br->br_msgs[i];
 703                 uint32_t done;
 704
 705                 msg = br->br_msgs[i];
 706                 br->br_msgs[i] = NULL;
 707
 708                 if (i == cur_cpuid)
 709                         continue;
 710
 711                 done = atomic_swap_int(&msg->br_done, 0);
 712                 if (done & NETISR_BR_WAITDONE)
 713                         wakeup(&msg->br_done);
 714         }
 715         br->br_isset = 0;
 716 }
 717
 718 static void
 719 netisr_nohashck(struct mbuf *m, const struct pktinfo *pi __unused)
 720 {
 721         m->m_flags &= ~M_HASH;
 722 }
 723
 724 void
 725 netisr_hashcheck(int num, struct mbuf *m, const struct pktinfo *pi)
 726 {
 727         struct netisr *ni;
 728
 729         if (num < 0 || num >= NETISR_MAX)
 730                 panic("Bad isr %d", num);
 731
 732         /*
 733          * Valid netisr?
 734          */
 735         ni = &netisrs[num];
 736         if (ni->ni_handler == NULL)
 737                 panic("Unregistered isr %d", num);
 738
 739         ni->ni_hashck(m, pi);
 740 }