svn commit: r331347 - in head: etc/mtree include sys/conf sys/dev/tcp_log sys/kern sys/netinet usr.bin/netstat

Ruslan Bukin ruslan.bukin at cl.cam.ac.uk
Thu Mar 22 14:25:07 UTC 2018


We don't have atomic_fetchadd_64 for mips32 I think

Ruslan

On Thu, Mar 22, 2018 at 09:40:08AM +0000, Jonathan T. Looney wrote:
> Author: jtl
> Date: Thu Mar 22 09:40:08 2018
> New Revision: 331347
> URL: https://svnweb.freebsd.org/changeset/base/331347
> 
> Log:
>   Add the "TCP Blackbox Recorder" which we discussed at the developer
>   summits at BSDCan and BSDCam in 2017.
>   
>   The TCP Blackbox Recorder allows you to capture events on a TCP connection
>   in a ring buffer. It stores metadata with the event. It optionally stores
>   the TCP header associated with an event (if the event is associated with a
>   packet) and also optionally stores information on the sockets.
>   
>   It supports setting a log ID on a TCP connection and using this to correlate
>   multiple connections that share a common log ID.
>   
>   You can log connections in different modes. If you are doing a coordinated
>   test with a particular connection, you may tell the system to put it in
>   mode 4 (continuous dump). Or, if you just want to monitor for errors, you
>   can put it in mode 1 (ring buffer) and dump all the ring buffers associated
>   with the connection ID when we receive an error signal for that connection
>   ID. You can set a default mode that will be applied to a particular ratio
>   of incoming connections. You can also manually set a mode using a socket
>   option.
>   
>   This commit includes only basic probes. rrs@ has added quite an abundance
>   of probes in his TCP development work. He plans to commit those soon.
>   
>   There are user-space programs which we plan to commit as ports. These read
>   the data from the log device and output pcapng files, and then let you
>   analyze the data (and metadata) in the pcapng files.
>   
>   Reviewed by:	gnn (previous version)
>   Obtained from:	Netflix, Inc.
>   Relnotes:	yes
>   Differential Revision:	https://reviews.freebsd.org/D11085
> 
> Added:
>   head/sys/dev/tcp_log/
>   head/sys/dev/tcp_log/tcp_log_dev.c   (contents, props changed)
>   head/sys/dev/tcp_log/tcp_log_dev.h   (contents, props changed)
>   head/sys/netinet/tcp_log_buf.c   (contents, props changed)
>   head/sys/netinet/tcp_log_buf.h   (contents, props changed)
> Modified:
>   head/etc/mtree/BSD.include.dist
>   head/include/Makefile
>   head/sys/conf/files
>   head/sys/kern/subr_witness.c
>   head/sys/netinet/tcp.h
>   head/sys/netinet/tcp_input.c
>   head/sys/netinet/tcp_output.c
>   head/sys/netinet/tcp_subr.c
>   head/sys/netinet/tcp_timer.c
>   head/sys/netinet/tcp_usrreq.c
>   head/sys/netinet/tcp_var.h
>   head/usr.bin/netstat/inet.c
>   head/usr.bin/netstat/main.c
>   head/usr.bin/netstat/netstat.1
>   head/usr.bin/netstat/netstat.h
> 
> Modified: head/etc/mtree/BSD.include.dist
> ==============================================================================
> --- head/etc/mtree/BSD.include.dist	Thu Mar 22 08:32:39 2018	(r331346)
> +++ head/etc/mtree/BSD.include.dist	Thu Mar 22 09:40:08 2018	(r331347)
> @@ -158,6 +158,8 @@
>          ..
>          speaker
>          ..
> +        tcp_log
> +        ..
>          usb
>          ..
>          vkbd
> 
> Modified: head/include/Makefile
> ==============================================================================
> --- head/include/Makefile	Thu Mar 22 08:32:39 2018	(r331346)
> +++ head/include/Makefile	Thu Mar 22 09:40:08 2018	(r331347)
> @@ -47,7 +47,7 @@ LSUBDIRS=	cam/ata cam/mmc cam/nvme cam/scsi \
>  	dev/hwpmc dev/hyperv \
>  	dev/ic dev/iicbus dev/io dev/lmc dev/mfi dev/mmc dev/nvme \
>  	dev/ofw dev/pbio dev/pci ${_dev_powermac_nvram} dev/ppbus dev/smbus \
> -	dev/speaker dev/vkbd dev/wi \
> +	dev/speaker dev/tcp_log dev/vkbd dev/wi \
>  	fs/devfs fs/fdescfs fs/msdosfs fs/nandfs fs/nfs fs/nullfs \
>  	fs/procfs fs/smbfs fs/udf fs/unionfs \
>  	geom/cache geom/concat geom/eli geom/gate geom/journal geom/label \
> 
> Modified: head/sys/conf/files
> ==============================================================================
> --- head/sys/conf/files	Thu Mar 22 08:32:39 2018	(r331346)
> +++ head/sys/conf/files	Thu Mar 22 09:40:08 2018	(r331347)
> @@ -3161,6 +3161,7 @@ dev/syscons/star/star_saver.c	optional star_saver
>  dev/syscons/syscons.c		optional sc
>  dev/syscons/sysmouse.c		optional sc
>  dev/syscons/warp/warp_saver.c	optional warp_saver
> +dev/tcp_log/tcp_log_dev.c	optional inet | inet6
>  dev/tdfx/tdfx_linux.c		optional tdfx_linux tdfx compat_linux
>  dev/tdfx/tdfx_pci.c		optional tdfx pci
>  dev/ti/if_ti.c			optional ti pci
> @@ -4309,6 +4310,7 @@ netinet/tcp_debug.c		optional tcpdebug
>  netinet/tcp_fastopen.c		optional inet tcp_rfc7413 | inet6 tcp_rfc7413
>  netinet/tcp_hostcache.c		optional inet | inet6
>  netinet/tcp_input.c		optional inet | inet6
> +netinet/tcp_log_buf.c		optional inet | inet6
>  netinet/tcp_lro.c		optional inet | inet6
>  netinet/tcp_output.c		optional inet | inet6
>  netinet/tcp_offload.c		optional tcp_offload inet | tcp_offload inet6
> 
> Added: head/sys/dev/tcp_log/tcp_log_dev.c
> ==============================================================================
> --- /dev/null	00:00:00 1970	(empty, because file is newly added)
> +++ head/sys/dev/tcp_log/tcp_log_dev.c	Thu Mar 22 09:40:08 2018	(r331347)
> @@ -0,0 +1,521 @@
> +/*-
> + * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
> + *
> + * Copyright (c) 2016-2017
> + *	Netflix Inc.  All rights reserved.
> + *
> + * Redistribution and use in source and binary forms, with or without
> + * modification, are permitted provided that the following conditions
> + * are met:
> + * 1. Redistributions of source code must retain the above copyright
> + *    notice, this list of conditions and the following disclaimer.
> + * 2. Redistributions in binary form must reproduce the above copyright
> + *    notice, this list of conditions and the following disclaimer in the
> + *    documentation and/or other materials provided with the distribution.
> + *
> + * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
> + * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
> + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
> + * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
> + * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
> + * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
> + * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
> + * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
> + * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
> + * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
> + * SUCH DAMAGE.
> + *
> + */
> +
> +#include <sys/cdefs.h>
> +__FBSDID("$FreeBSD$");
> +
> +#include <sys/param.h>
> +#include <sys/conf.h>
> +#include <sys/fcntl.h>
> +#include <sys/filio.h>
> +#include <sys/kernel.h>
> +#include <sys/lock.h>
> +#include <sys/malloc.h>
> +#include <sys/module.h>
> +#include <sys/poll.h>
> +#include <sys/queue.h>
> +#include <sys/refcount.h>
> +#include <sys/mutex.h>
> +#include <sys/selinfo.h>
> +#include <sys/socket.h>
> +#include <sys/socketvar.h>
> +#include <sys/sysctl.h>
> +#include <sys/tree.h>
> +#include <sys/uio.h>
> +#include <machine/atomic.h>
> +#include <sys/counter.h>
> +
> +#include <dev/tcp_log/tcp_log_dev.h>
> +
> +#ifdef TCPLOG_DEBUG_COUNTERS
> +extern counter_u64_t tcp_log_que_read;
> +extern counter_u64_t tcp_log_que_freed;
> +#endif
> +
> +static struct cdev *tcp_log_dev;
> +static struct selinfo tcp_log_sel;
> +
> +static struct log_queueh tcp_log_dev_queue_head = STAILQ_HEAD_INITIALIZER(tcp_log_dev_queue_head);
> +static struct log_infoh tcp_log_dev_reader_head = STAILQ_HEAD_INITIALIZER(tcp_log_dev_reader_head);
> +
> +MALLOC_DEFINE(M_TCPLOGDEV, "tcp_log_dev", "TCP log device data structures");
> +
> +static int	tcp_log_dev_listeners = 0;
> +
> +static struct mtx tcp_log_dev_queue_lock;
> +
> +#define	TCP_LOG_DEV_QUEUE_LOCK()	mtx_lock(&tcp_log_dev_queue_lock)
> +#define	TCP_LOG_DEV_QUEUE_UNLOCK()	mtx_unlock(&tcp_log_dev_queue_lock)
> +#define	TCP_LOG_DEV_QUEUE_LOCK_ASSERT()	mtx_assert(&tcp_log_dev_queue_lock, MA_OWNED)
> +#define	TCP_LOG_DEV_QUEUE_UNLOCK_ASSERT() mtx_assert(&tcp_log_dev_queue_lock, MA_NOTOWNED)
> +#define	TCP_LOG_DEV_QUEUE_REF(tldq)	refcount_acquire(&((tldq)->tldq_refcnt))
> +#define	TCP_LOG_DEV_QUEUE_UNREF(tldq)	refcount_release(&((tldq)->tldq_refcnt))
> +
> +static void	tcp_log_dev_clear_refcount(struct tcp_log_dev_queue *entry);
> +static void	tcp_log_dev_clear_cdevpriv(void *data);
> +static int	tcp_log_dev_open(struct cdev *dev __unused, int flags,
> +    int devtype __unused, struct thread *td __unused);
> +static int	tcp_log_dev_write(struct cdev *dev __unused,
> +    struct uio *uio __unused, int flags __unused);
> +static int	tcp_log_dev_read(struct cdev *dev __unused, struct uio *uio,
> +    int flags __unused);
> +static int	tcp_log_dev_ioctl(struct cdev *dev __unused, u_long cmd,
> +    caddr_t data, int fflag __unused, struct thread *td __unused);
> +static int	tcp_log_dev_poll(struct cdev *dev __unused, int events,
> +    struct thread *td);
> +
> +
> +enum tcp_log_dev_queue_lock_state {
> +	QUEUE_UNLOCKED = 0,
> +	QUEUE_LOCKED,
> +};
> +
> +static struct cdevsw tcp_log_cdevsw = {
> +	.d_version =	D_VERSION,
> +	.d_read =	tcp_log_dev_read,
> +	.d_open =	tcp_log_dev_open,
> +	.d_write =	tcp_log_dev_write,
> +	.d_poll =	tcp_log_dev_poll,
> +	.d_ioctl =	tcp_log_dev_ioctl,
> +#ifdef NOTYET
> +	.d_mmap =	tcp_log_dev_mmap,
> +#endif
> +	.d_name =	"tcp_log",
> +};
> +
> +static __inline void
> +tcp_log_dev_queue_validate_lock(int lockstate)
> +{
> +
> +#ifdef INVARIANTS
> +	switch (lockstate) {
> +	case QUEUE_LOCKED:
> +		TCP_LOG_DEV_QUEUE_LOCK_ASSERT();
> +		break;
> +	case QUEUE_UNLOCKED:
> +		TCP_LOG_DEV_QUEUE_UNLOCK_ASSERT();
> +		break;
> +	default:
> +		kassert_panic("%s:%d: unknown queue lock state", __func__,
> +		    __LINE__);
> +	}
> +#endif
> +}
> +
> +/*
> + * Clear the refcount. If appropriate, it will remove the entry from the
> + * queue and call the destructor.
> + *
> + * This must be called with the queue lock held.
> + */
> +static void
> +tcp_log_dev_clear_refcount(struct tcp_log_dev_queue *entry)
> +{
> +
> +	KASSERT(entry != NULL, ("%s: called with NULL entry", __func__));
> +
> +	TCP_LOG_DEV_QUEUE_LOCK_ASSERT();
> +
> +	if (TCP_LOG_DEV_QUEUE_UNREF(entry)) {
> +#ifdef TCPLOG_DEBUG_COUNTERS
> +		counter_u64_add(tcp_log_que_freed, 1);
> +#endif
> +		/* Remove the entry from the queue and call the destructor. */
> +		STAILQ_REMOVE(&tcp_log_dev_queue_head, entry, tcp_log_dev_queue,
> +		    tldq_queue);
> +		(*entry->tldq_dtor)(entry);
> +	}
> +}
> +
> +static void
> +tcp_log_dev_clear_cdevpriv(void *data)
> +{
> +	struct tcp_log_dev_info *priv;
> +	struct tcp_log_dev_queue *entry, *entry_tmp;
> +
> +	priv = (struct tcp_log_dev_info *)data;
> +	if (priv == NULL)
> +		return;
> +
> +	/*
> +	 * Lock the queue and drop our references. We hold references to all
> +	 * the entries starting with tldi_head (or, if tldi_head == NULL, all
> +	 * entries in the queue).
> +	 * 
> +	 * Because we don't want anyone adding addition things to the queue
> +	 * while we are doing this, we lock the queue.
> +	 */
> +	TCP_LOG_DEV_QUEUE_LOCK();
> +	if (priv->tldi_head != NULL) {
> +		entry = priv->tldi_head;
> +		STAILQ_FOREACH_FROM_SAFE(entry, &tcp_log_dev_queue_head,
> +		    tldq_queue, entry_tmp) {
> +			tcp_log_dev_clear_refcount(entry);
> +		}
> +	}
> +	tcp_log_dev_listeners--;
> +	KASSERT(tcp_log_dev_listeners >= 0,
> +	    ("%s: tcp_log_dev_listeners is unexpectedly negative", __func__));
> +	STAILQ_REMOVE(&tcp_log_dev_reader_head, priv, tcp_log_dev_info,
> +	    tldi_list);
> +	TCP_LOG_DEV_QUEUE_LOCK_ASSERT();
> +	TCP_LOG_DEV_QUEUE_UNLOCK();
> +	free(priv, M_TCPLOGDEV);
> +}
> +
> +static int
> +tcp_log_dev_open(struct cdev *dev __unused, int flags, int devtype __unused,
> +    struct thread *td __unused)
> +{
> +	struct tcp_log_dev_info *priv;
> +	struct tcp_log_dev_queue *entry;
> +	int rv;
> +
> +	/*
> +	 * Ideally, we shouldn't see these because of file system
> +	 * permissions.
> +	 */
> +	if (flags & (FWRITE | FEXEC | FAPPEND | O_TRUNC))
> +		return (ENODEV);
> +
> +	/* Allocate space to hold information about where we are. */
> +	priv = malloc(sizeof(struct tcp_log_dev_info), M_TCPLOGDEV,
> +	    M_ZERO | M_WAITOK);
> +
> +	/* Stash the private data away. */
> +	rv = devfs_set_cdevpriv((void *)priv, tcp_log_dev_clear_cdevpriv);
> +	if (!rv) {
> +		/*
> +		 * Increase the listener count, add this reader to the list, and
> +		 * take references on all current queues.
> +		 */
> +		TCP_LOG_DEV_QUEUE_LOCK();
> +		tcp_log_dev_listeners++;
> +		STAILQ_INSERT_HEAD(&tcp_log_dev_reader_head, priv, tldi_list);
> +		priv->tldi_head = STAILQ_FIRST(&tcp_log_dev_queue_head);
> +		if (priv->tldi_head != NULL)
> +			priv->tldi_cur = priv->tldi_head->tldq_buf;
> +		STAILQ_FOREACH(entry, &tcp_log_dev_queue_head, tldq_queue)
> +			TCP_LOG_DEV_QUEUE_REF(entry);
> +		TCP_LOG_DEV_QUEUE_UNLOCK();
> +	} else {
> +		/* Free the entry. */
> +		free(priv, M_TCPLOGDEV);
> +	}
> +	return (rv);
> +}
> +
> +static int
> +tcp_log_dev_write(struct cdev *dev __unused, struct uio *uio __unused,
> +    int flags __unused)
> +{
> +
> +	return (ENODEV);
> +}
> +
> +static __inline void
> +tcp_log_dev_rotate_bufs(struct tcp_log_dev_info *priv, int *lockstate)
> +{
> +	struct tcp_log_dev_queue *entry;
> +
> +	KASSERT(priv->tldi_head != NULL,
> +	    ("%s:%d: priv->tldi_head unexpectedly NULL",
> +	    __func__, __LINE__));
> +	KASSERT(priv->tldi_head->tldq_buf == priv->tldi_cur,
> +	    ("%s:%d: buffer mismatch (%p vs %p)",
> +	    __func__, __LINE__, priv->tldi_head->tldq_buf,
> +	    priv->tldi_cur));
> +	tcp_log_dev_queue_validate_lock(*lockstate);
> +
> +	if (*lockstate == QUEUE_UNLOCKED) {
> +		TCP_LOG_DEV_QUEUE_LOCK();
> +		*lockstate = QUEUE_LOCKED;
> +	}
> +	entry = priv->tldi_head;
> +	priv->tldi_head = STAILQ_NEXT(entry, tldq_queue);
> +	tcp_log_dev_clear_refcount(entry);
> +	priv->tldi_cur = NULL;
> +}
> +
> +static int
> +tcp_log_dev_read(struct cdev *dev __unused, struct uio *uio, int flags)
> +{
> +	struct tcp_log_common_header *buf;
> +	struct tcp_log_dev_info *priv;
> +	struct tcp_log_dev_queue *entry;
> +	ssize_t len;
> +	int lockstate, rv;
> +
> +	/* Get our private info. */
> +	rv = devfs_get_cdevpriv((void **)&priv);
> +	if (rv)
> +		return (rv);
> +
> +	lockstate = QUEUE_UNLOCKED;
> +
> +	/* Do we need to get a new buffer? */
> +	while (priv->tldi_cur == NULL ||
> +	    priv->tldi_cur->tlch_length <= priv->tldi_off) {
> +		/* Did we somehow forget to rotate? */
> +		KASSERT(priv->tldi_cur == NULL,
> +		    ("%s:%d: tldi_cur is unexpectedly non-NULL", __func__,
> +		    __LINE__));
> +		if (priv->tldi_cur != NULL)
> +			tcp_log_dev_rotate_bufs(priv, &lockstate);
> +
> +		/*
> +		 * Before we start looking at tldi_head, we need a lock on the
> +		 * queue to make sure tldi_head stays stable.
> +		 */
> +		if (lockstate == QUEUE_UNLOCKED) {
> +			TCP_LOG_DEV_QUEUE_LOCK();
> +			lockstate = QUEUE_LOCKED;
> +		}
> +
> +		/* We need the next buffer. Do we have one? */
> +		if (priv->tldi_head == NULL && (flags & FNONBLOCK)) {
> +			rv = EAGAIN;
> +			goto done;
> +		}
> +		if (priv->tldi_head == NULL) {
> +			/* Sleep and wait for more things we can read. */
> +			rv = mtx_sleep(&tcp_log_dev_listeners,
> +			    &tcp_log_dev_queue_lock, PCATCH, "tcplogdev", 0);
> +			if (rv)
> +				goto done;
> +			if (priv->tldi_head == NULL)
> +				continue;
> +		}
> +
> +		/*
> +		 * We have an entry to read. We want to try to create a
> +		 * buffer, if one doesn't already exist.
> +		 */
> +		entry = priv->tldi_head;
> +		if (entry->tldq_buf == NULL) {
> +			TCP_LOG_DEV_QUEUE_LOCK_ASSERT();
> +			buf = (*entry->tldq_xform)(entry);
> +			if (buf == NULL) {
> +				rv = EBUSY;
> +				goto done;
> +			}
> +			entry->tldq_buf = buf;
> +		}
> +
> +		priv->tldi_cur = entry->tldq_buf;
> +		priv->tldi_off = 0;
> +	}
> +
> +	/* Copy what we can from this buffer to the output buffer. */
> +	if (uio->uio_resid > 0) {
> +		/* Drop locks so we can take page faults. */
> +		if (lockstate == QUEUE_LOCKED)
> +			TCP_LOG_DEV_QUEUE_UNLOCK();
> +		lockstate = QUEUE_UNLOCKED;
> +
> +		KASSERT(priv->tldi_cur != NULL,
> +		    ("%s: priv->tldi_cur is unexpectedly NULL", __func__));
> +
> +		/* Copy as much as we can to this uio. */
> +		len = priv->tldi_cur->tlch_length - priv->tldi_off;
> +		if (len > uio->uio_resid)
> +			len = uio->uio_resid;
> +		rv = uiomove(((uint8_t *)priv->tldi_cur) + priv->tldi_off,
> +		    len, uio);
> +		if (rv != 0)
> +			goto done;
> +		priv->tldi_off += len;
> +#ifdef TCPLOG_DEBUG_COUNTERS
> +		counter_u64_add(tcp_log_que_read, len);
> +#endif
> +	}
> +	/* Are we done with this buffer? If so, find the next one. */
> +	if (priv->tldi_off >= priv->tldi_cur->tlch_length) {
> +		KASSERT(priv->tldi_off == priv->tldi_cur->tlch_length,
> +		    ("%s: offset (%ju) exceeds length (%ju)", __func__,
> +		    (uintmax_t)priv->tldi_off,
> +		    (uintmax_t)priv->tldi_cur->tlch_length));
> +		tcp_log_dev_rotate_bufs(priv, &lockstate);
> +	}
> +done:
> +	tcp_log_dev_queue_validate_lock(lockstate);
> +	if (lockstate == QUEUE_LOCKED)
> +		TCP_LOG_DEV_QUEUE_UNLOCK();
> +	return (rv);
> +}
> +
> +static int
> +tcp_log_dev_ioctl(struct cdev *dev __unused, u_long cmd, caddr_t data,
> +    int fflag __unused, struct thread *td __unused)
> +{
> +	struct tcp_log_dev_info *priv;
> +	int rv;
> +
> +	/* Get our private info. */
> +	rv = devfs_get_cdevpriv((void **)&priv);
> +	if (rv)
> +		return (rv);
> +
> +	/*
> +	 * Set things. Here, we are most concerned about the non-blocking I/O
> +	 * flag.
> +	 */
> +	rv = 0;
> +	switch (cmd) {
> +	case FIONBIO:
> +		break;
> +	case FIOASYNC:
> +		if (*(int *)data != 0)
> +			rv = EINVAL;
> +		break;
> +	default:
> +		rv = ENOIOCTL;
> +	}
> +	return (rv);
> +}
> +
> +static int
> +tcp_log_dev_poll(struct cdev *dev __unused, int events, struct thread *td)
> +{
> +	struct tcp_log_dev_info *priv;
> +	int revents;
> +
> +	/*
> +	 * Get our private info. If this fails, claim that all events are
> +	 * ready. That should prod the user to do something that will
> +	 * make the error evident to them.
> +	 */
> +	if (devfs_get_cdevpriv((void **)&priv))
> +		return (events);
> +
> +	revents = 0;
> +	if (events & (POLLIN | POLLRDNORM)) {
> +		/*
> +		 * We can (probably) read right now if we are partway through
> +		 * a buffer or if we are just about to start a buffer.
> +		 * Because we are going to read tldi_head, we should acquire
> +		 * a read lock on the queue.
> +		 */
> +		TCP_LOG_DEV_QUEUE_LOCK();
> +		if ((priv->tldi_head != NULL && priv->tldi_cur == NULL) ||
> +		    (priv->tldi_cur != NULL &&
> +		    priv->tldi_off < priv->tldi_cur->tlch_length))
> +			revents = events & (POLLIN | POLLRDNORM);
> +		else
> +			selrecord(td, &tcp_log_sel);
> +		TCP_LOG_DEV_QUEUE_UNLOCK();
> +	} else {
> +		/*
> +		 * It only makes sense to poll for reading. So, again, prod the
> +		 * user to do something that will make the error of their ways
> +		 * apparent.
> +		 */
> +		revents = events;
> +	}
> +	return (revents);
> +}
> +
> +int
> +tcp_log_dev_add_log(struct tcp_log_dev_queue *entry)
> +{
> +	struct tcp_log_dev_info *priv;
> +	int rv;
> +	bool wakeup_needed;
> +
> +	KASSERT(entry->tldq_buf != NULL || entry->tldq_xform != NULL,
> +	    ("%s: Called with both tldq_buf and tldq_xform set to NULL",
> +	    __func__));
> +	KASSERT(entry->tldq_dtor != NULL,
> +	    ("%s: Called with tldq_dtor set to NULL", __func__));
> +
> +	/* Get a lock on the queue. */
> +	TCP_LOG_DEV_QUEUE_LOCK();
> +
> +	/* If no one is listening, tell the caller to free the resources. */
> +	if (tcp_log_dev_listeners == 0) {
> +		rv = ENXIO;
> +		goto done;
> +	}
> +
> +	/* Add this to the end of the tailq. */
> +	STAILQ_INSERT_TAIL(&tcp_log_dev_queue_head, entry, tldq_queue);
> +
> +	/* Add references for all current listeners. */
> +	refcount_init(&entry->tldq_refcnt, tcp_log_dev_listeners);
> +
> +	/*
> +	 * If any listener is currently stuck on NULL, that means they are
> +	 * waiting. Point their head to this new entry.
> +	 */
> +	wakeup_needed = false;
> +	STAILQ_FOREACH(priv, &tcp_log_dev_reader_head, tldi_list)
> +		if (priv->tldi_head == NULL) {
> +			priv->tldi_head = entry;
> +			wakeup_needed = true;
> +		}
> +
> +	if (wakeup_needed) {
> +		selwakeup(&tcp_log_sel);
> +		wakeup(&tcp_log_dev_listeners);
> +	}
> +
> +	rv = 0;
> +
> +done:
> +	TCP_LOG_DEV_QUEUE_LOCK_ASSERT();
> +	TCP_LOG_DEV_QUEUE_UNLOCK();
> +	return (rv);
> +}
> +
> +static int
> +tcp_log_dev_modevent(module_t mod __unused, int type, void *data __unused)
> +{
> +
> +	/* TODO: Support intelligent unloading. */
> +	switch (type) {
> +	case MOD_LOAD:
> +		if (bootverbose)
> +			printf("tcp_log: tcp_log device\n");
> +		memset(&tcp_log_sel, 0, sizeof(tcp_log_sel));
> +		memset(&tcp_log_dev_queue_lock, 0, sizeof(struct mtx));
> +		mtx_init(&tcp_log_dev_queue_lock, "tcp_log dev",
> +			 "tcp_log device queues", MTX_DEF);
> +		tcp_log_dev = make_dev_credf(MAKEDEV_ETERNAL_KLD,
> +		    &tcp_log_cdevsw, 0, NULL, UID_ROOT, GID_WHEEL, 0400,
> +		    "tcp_log");
> +		break;
> +	default:
> +		return (EOPNOTSUPP);
> +	}
> +
> +	return (0);
> +}
> +
> +DEV_MODULE(tcp_log_dev, tcp_log_dev_modevent, NULL);
> +MODULE_VERSION(tcp_log_dev, 1);
> 
> Added: head/sys/dev/tcp_log/tcp_log_dev.h
> ==============================================================================
> --- /dev/null	00:00:00 1970	(empty, because file is newly added)
> +++ head/sys/dev/tcp_log/tcp_log_dev.h	Thu Mar 22 09:40:08 2018	(r331347)
> @@ -0,0 +1,88 @@
> +/*-
> + * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
> + *
> + * Copyright (c) 2016
> + *	Netflix Inc.  All rights reserved.
> + *
> + * Redistribution and use in source and binary forms, with or without
> + * modification, are permitted provided that the following conditions
> + * are met:
> + * 1. Redistributions of source code must retain the above copyright
> + *    notice, this list of conditions and the following disclaimer.
> + * 2. Redistributions in binary form must reproduce the above copyright
> + *    notice, this list of conditions and the following disclaimer in the
> + *    documentation and/or other materials provided with the distribution.
> + *
> + * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
> + * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
> + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
> + * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
> + * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
> + * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
> + * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
> + * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
> + * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
> + * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
> + * SUCH DAMAGE.
> + *
> + * $FreeBSD$
> + */
> +
> +#ifndef __tcp_log_dev_h__
> +#define	__tcp_log_dev_h__
> +
> +/*
> + * This is the common header for data streamed from the log device. All
> + * blocks of data need to start with this header.
> + */
> +struct tcp_log_common_header {
> +	uint32_t	tlch_version;	/* Version is specific to type. */
> +	uint32_t	tlch_type;	/* Type of entry(ies) that follow. */
> +	uint64_t	tlch_length;	/* Total length, including header. */
> +} __packed;
> +
> +#define	TCP_LOG_DEV_TYPE_BBR	1	/* black box recorder */
> +
> +#ifdef _KERNEL
> +/*
> + * This is a queue entry. All queue entries need to start with this structure
> + * so the common code can cast them to this structure; however, other modules
> + * are free to include additional data after this structure.
> + *
> + * The elements are explained here:
> + * tldq_queue: used by the common code to maintain this entry's position in the
> + *     queue.
> + * tldq_buf: should be NULL, or a pointer to a chunk of data. The data must be
> + *     as long as the common header indicates.
> + * tldq_xform: If tldq_buf is NULL, the code will call this to create the
> + *     the tldq_buf object. The function should *not* directly modify tldq_buf,
> + *     but should return the buffer (which must meet the restrictions
> + *     indicated for tldq_buf).
> + * tldq_dtor: This function is called to free the queue entry. If tldq_buf is
> + *     not NULL, the dtor function must free that, too.
> + * tldq_refcnt: used by the common code to indicate how many readers still need
> + *     this data.
> + */
> +struct tcp_log_dev_queue {
> +	STAILQ_ENTRY(tcp_log_dev_queue) tldq_queue;
> +	struct tcp_log_common_header *tldq_buf;
> +	struct tcp_log_common_header *(*tldq_xform)(struct tcp_log_dev_queue *entry);
> +	void	(*tldq_dtor)(struct tcp_log_dev_queue *entry);
> +	volatile u_int tldq_refcnt;
> +};
> +
> +STAILQ_HEAD(log_queueh, tcp_log_dev_queue);
> +
> +struct tcp_log_dev_info {
> +	STAILQ_ENTRY(tcp_log_dev_info) tldi_list;
> +	struct tcp_log_dev_queue *tldi_head;
> +	struct tcp_log_common_header *tldi_cur;
> +	off_t			tldi_off;
> +};
> +STAILQ_HEAD(log_infoh, tcp_log_dev_info);
> +
> +
> +MALLOC_DECLARE(M_TCPLOGDEV);
> +int tcp_log_dev_add_log(struct tcp_log_dev_queue *entry);
> +#endif /* _KERNEL */
> +#endif /* !__tcp_log_dev_h__ */
> 
> Modified: head/sys/kern/subr_witness.c
> ==============================================================================
> --- head/sys/kern/subr_witness.c	Thu Mar 22 08:32:39 2018	(r331346)
> +++ head/sys/kern/subr_witness.c	Thu Mar 22 09:40:08 2018	(r331347)
> @@ -640,6 +640,14 @@ static struct witness_order_list_entry order_lists[] =
>  	{ "db->db_mtx", &lock_class_sx },
>  	{ NULL, NULL },
>  	/*
> +	 * TCP log locks
> +	 */
> +	{ "TCP ID tree", &lock_class_rw },
> +	{ "tcp log id bucket", &lock_class_mtx_sleep },
> +	{ "tcpinp", &lock_class_rw },
> +	{ "TCP log expireq", &lock_class_mtx_sleep },
> +	{ NULL, NULL },
> +	/*
>  	 * spin locks
>  	 */
>  #ifdef SMP
> 
> Modified: head/sys/netinet/tcp.h
> ==============================================================================
> --- head/sys/netinet/tcp.h	Thu Mar 22 08:32:39 2018	(r331346)
> +++ head/sys/netinet/tcp.h	Thu Mar 22 09:40:08 2018	(r331347)
> @@ -168,6 +168,12 @@ struct tcphdr {
>  #define TCP_NOOPT	8	/* don't use TCP options */
>  #define TCP_MD5SIG	16	/* use MD5 digests (RFC2385) */
>  #define	TCP_INFO	32	/* retrieve tcp_info structure */
> +#define	TCP_LOG		34	/* configure event logging for connection */
> +#define	TCP_LOGBUF	35	/* retrieve event log for connection */
> +#define	TCP_LOGID	36	/* configure log ID to correlate connections */
> +#define	TCP_LOGDUMP	37	/* dump connection log events to device */
> +#define	TCP_LOGDUMPID	38	/* dump events from connections with same ID to
> +				   device */
>  #define	TCP_CONGESTION	64	/* get/set congestion control algorithm */
>  #define	TCP_CCALGOOPT	65	/* get/set cc algorithm specific options */
>  #define	TCP_KEEPINIT	128	/* N, time to establish connection */
> @@ -188,6 +194,9 @@ struct tcphdr {
>  #define	TCPI_OPT_WSCALE		0x04
>  #define	TCPI_OPT_ECN		0x08
>  #define	TCPI_OPT_TOE		0x10
> +
> +/* Maximum length of log ID. */
> +#define TCP_LOG_ID_LEN	64
>  
>  /*
>   * The TCP_INFO socket option comes from the Linux 2.6 TCP API, and permits
> 
> Modified: head/sys/netinet/tcp_input.c
> ==============================================================================
> --- head/sys/netinet/tcp_input.c	Thu Mar 22 08:32:39 2018	(r331346)
> +++ head/sys/netinet/tcp_input.c	Thu Mar 22 09:40:08 2018	(r331347)
> @@ -102,6 +102,7 @@ __FBSDID("$FreeBSD$");
>  #include <netinet6/nd6.h>
>  #include <netinet/tcp.h>
>  #include <netinet/tcp_fsm.h>
> +#include <netinet/tcp_log_buf.h>
>  #include <netinet/tcp_seq.h>
>  #include <netinet/tcp_timer.h>
>  #include <netinet/tcp_var.h>
> @@ -1592,6 +1593,8 @@ tcp_do_segment(struct mbuf *m, struct tcphdr *th, stru
>  	/* Save segment, if requested. */
>  	tcp_pcap_add(th, m, &(tp->t_inpkts));
>  #endif
> +	TCP_LOG_EVENT(tp, th, &so->so_rcv, &so->so_snd, TCP_LOG_IN, 0,
> +	    tlen, NULL, true);
>  
>  	if ((thflags & TH_SYN) && (thflags & TH_FIN) && V_drop_synfin) {
>  		if ((s = tcp_log_addrs(inc, th, NULL, NULL))) {
> 
> Added: head/sys/netinet/tcp_log_buf.c
> ==============================================================================
> --- /dev/null	00:00:00 1970	(empty, because file is newly added)
> +++ head/sys/netinet/tcp_log_buf.c	Thu Mar 22 09:40:08 2018	(r331347)
> @@ -0,0 +1,2480 @@
> +/*-
> + * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
> + *
> + * Copyright (c) 2016-2018
> + *	Netflix Inc.  All rights reserved.
> + *
> + * Redistribution and use in source and binary forms, with or without
> + * modification, are permitted provided that the following conditions
> + * are met:
> + * 1. Redistributions of source code must retain the above copyright
> + *    notice, this list of conditions and the following disclaimer.
> + * 2. Redistributions in binary form must reproduce the above copyright
> + *    notice, this list of conditions and the following disclaimer in the
> + *    documentation and/or other materials provided with the distribution.
> + *
> + * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
> + * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
> + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
> + * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
> + * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
> + * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
> + * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
> + * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
> + * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
> + * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
> + * SUCH DAMAGE.
> + *
> + */
> +
> +#include <sys/cdefs.h>
> +__FBSDID("$FreeBSD$");
> +
> +#include <sys/param.h>
> +#include <sys/kernel.h>
> +#include <sys/lock.h>
> +#include <sys/malloc.h>
> +#include <sys/mutex.h>
> +#include <sys/queue.h>
> +#include <sys/refcount.h>
> +#include <sys/rwlock.h>
> +#include <sys/socket.h>
> +#include <sys/socketvar.h>
> +#include <sys/sysctl.h>
> +#include <sys/tree.h>
> +#include <sys/counter.h>
> +
> +#include <dev/tcp_log/tcp_log_dev.h>
> +
> +#include <net/if.h>
> +#include <net/if_var.h>
> +#include <net/vnet.h>
> +
> +#include <netinet/in.h>
> +#include <netinet/in_pcb.h>
> +#include <netinet/in_var.h>
> +#include <netinet/tcp_var.h>
> +#include <netinet/tcp_log_buf.h>
> +
> +/* Default expiry time */
> +#define	TCP_LOG_EXPIRE_TIME	((sbintime_t)60 * SBT_1S)
> +
> +/* Max interval at which to run the expiry timer */
> +#define	TCP_LOG_EXPIRE_INTVL	((sbintime_t)5 * SBT_1S)
> +
> +bool	tcp_log_verbose;
> +static uma_zone_t tcp_log_bucket_zone, tcp_log_node_zone, tcp_log_zone;
> +static int	tcp_log_session_limit = TCP_LOG_BUF_DEFAULT_SESSION_LIMIT;
> +static uint32_t	tcp_log_version = TCP_LOG_BUF_VER;
> +RB_HEAD(tcp_log_id_tree, tcp_log_id_bucket);
> +static struct tcp_log_id_tree tcp_log_id_head;
> +static STAILQ_HEAD(, tcp_log_id_node) tcp_log_expireq_head =
> +    STAILQ_HEAD_INITIALIZER(tcp_log_expireq_head);
> +static struct mtx tcp_log_expireq_mtx;
> +static struct callout tcp_log_expireq_callout;
> +static uint64_t tcp_log_auto_ratio = 0;
> +static uint64_t tcp_log_auto_ratio_cur = 0;
> +static uint32_t tcp_log_auto_mode = TCP_LOG_STATE_TAIL;
> +static bool tcp_log_auto_all = false;
> +
> +RB_PROTOTYPE_STATIC(tcp_log_id_tree, tcp_log_id_bucket, tlb_rb, tcp_log_id_cmp)
> +
> +SYSCTL_NODE(_net_inet_tcp, OID_AUTO, bb, CTLFLAG_RW, 0, "TCP Black Box controls");
> +
> +SYSCTL_BOOL(_net_inet_tcp_bb, OID_AUTO, log_verbose, CTLFLAG_RW, &tcp_log_verbose,
> +    0, "Force verbose logging for TCP traces");
> +
> +SYSCTL_INT(_net_inet_tcp_bb, OID_AUTO, log_session_limit,
> +    CTLFLAG_RW, &tcp_log_session_limit, 0,
> +    "Maximum number of events maintained for each TCP session");
> +
> +SYSCTL_UMA_MAX(_net_inet_tcp_bb, OID_AUTO, log_global_limit, CTLFLAG_RW,
> +    &tcp_log_zone, "Maximum number of events maintained for all TCP sessions");
> +
> +SYSCTL_UMA_CUR(_net_inet_tcp_bb, OID_AUTO, log_global_entries, CTLFLAG_RD,
> +    &tcp_log_zone, "Current number of events maintained for all TCP sessions");
> +
> +SYSCTL_UMA_MAX(_net_inet_tcp_bb, OID_AUTO, log_id_limit, CTLFLAG_RW,
> +    &tcp_log_bucket_zone, "Maximum number of log IDs");
> +
> +SYSCTL_UMA_CUR(_net_inet_tcp_bb, OID_AUTO, log_id_entries, CTLFLAG_RD,
> +    &tcp_log_bucket_zone, "Current number of log IDs");
> +
> +SYSCTL_UMA_MAX(_net_inet_tcp_bb, OID_AUTO, log_id_tcpcb_limit, CTLFLAG_RW,
> +    &tcp_log_node_zone, "Maximum number of tcpcbs with log IDs");
> +
> +SYSCTL_UMA_CUR(_net_inet_tcp_bb, OID_AUTO, log_id_tcpcb_entries, CTLFLAG_RD,
> +    &tcp_log_node_zone, "Current number of tcpcbs with log IDs");
> +
> +SYSCTL_U32(_net_inet_tcp_bb, OID_AUTO, log_version, CTLFLAG_RD, &tcp_log_version,
> +    0, "Version of log formats exported");
> +
> +SYSCTL_U64(_net_inet_tcp_bb, OID_AUTO, log_auto_ratio, CTLFLAG_RW,
> +    &tcp_log_auto_ratio, 0, "Do auto capturing for 1 out of N sessions");
> +
> +SYSCTL_U32(_net_inet_tcp_bb, OID_AUTO, log_auto_mode, CTLFLAG_RW,
> +    &tcp_log_auto_mode, TCP_LOG_STATE_HEAD_AUTO,
> +    "Logging mode for auto-selected sessions (default is TCP_LOG_STATE_HEAD_AUTO)");
> +
> +SYSCTL_BOOL(_net_inet_tcp_bb, OID_AUTO, log_auto_all, CTLFLAG_RW,
> +    &tcp_log_auto_all, false,
> +    "Auto-select from all sessions (rather than just those with IDs)");
> +
> +#ifdef TCPLOG_DEBUG_COUNTERS
> +counter_u64_t tcp_log_queued;
> +counter_u64_t tcp_log_que_fail1;
> +counter_u64_t tcp_log_que_fail2;
> +counter_u64_t tcp_log_que_fail3;
> +counter_u64_t tcp_log_que_fail4;
> +counter_u64_t tcp_log_que_fail5;
> +counter_u64_t tcp_log_que_copyout;
> +counter_u64_t tcp_log_que_read;
> +counter_u64_t tcp_log_que_freed;
> +
> +SYSCTL_COUNTER_U64(_net_inet_tcp_bb, OID_AUTO, queued, CTLFLAG_RD,
> +    &tcp_log_queued, "Number of entries queued");
> +SYSCTL_COUNTER_U64(_net_inet_tcp_bb, OID_AUTO, fail1, CTLFLAG_RD,
> +    &tcp_log_que_fail1, "Number of entries queued but fail 1");
> +SYSCTL_COUNTER_U64(_net_inet_tcp_bb, OID_AUTO, fail2, CTLFLAG_RD,
> +    &tcp_log_que_fail2, "Number of entries queued but fail 2");
> +SYSCTL_COUNTER_U64(_net_inet_tcp_bb, OID_AUTO, fail3, CTLFLAG_RD,
> +    &tcp_log_que_fail3, "Number of entries queued but fail 3");
> +SYSCTL_COUNTER_U64(_net_inet_tcp_bb, OID_AUTO, fail4, CTLFLAG_RD,
> +    &tcp_log_que_fail4, "Number of entries queued but fail 4");
> +SYSCTL_COUNTER_U64(_net_inet_tcp_bb, OID_AUTO, fail5, CTLFLAG_RD,
> +    &tcp_log_que_fail5, "Number of entries queued but fail 4");
> +SYSCTL_COUNTER_U64(_net_inet_tcp_bb, OID_AUTO, copyout, CTLFLAG_RD,
> +    &tcp_log_que_copyout, "Number of entries copied out");
> +SYSCTL_COUNTER_U64(_net_inet_tcp_bb, OID_AUTO, read, CTLFLAG_RD,
> +    &tcp_log_que_read, "Number of entries read from the queue");
> +SYSCTL_COUNTER_U64(_net_inet_tcp_bb, OID_AUTO, freed, CTLFLAG_RD,
> +    &tcp_log_que_freed, "Number of entries freed after reading");
> +#endif
> +
> +#ifdef INVARIANTS
> +#define	TCPLOG_DEBUG_RINGBUF
> +#endif
> +
> +struct tcp_log_mem
> +{
> +	STAILQ_ENTRY(tcp_log_mem) tlm_queue;
> +	struct tcp_log_buffer	tlm_buf;
> +	struct tcp_log_verbose	tlm_v;
> +#ifdef TCPLOG_DEBUG_RINGBUF
> +	volatile int		tlm_refcnt;
> +#endif
> +};
> +
> +/* 60 bytes for the header, + 16 bytes for padding */
> +static uint8_t	zerobuf[76];
> +
> +/*
> + * Lock order:
> + * 1. TCPID_TREE
> + * 2. TCPID_BUCKET
> + * 3. INP
> + *
> + * Rules:
> + * A. You need a lock on the Tree to add/remove buckets.
> + * B. You need a lock on the bucket to add/remove nodes from the bucket.
> + * C. To change information in a node, you need the INP lock if the tln_closed
> + *    field is false. Otherwise, you need the bucket lock. (Note that the
> + *    tln_closed field can change at any point, so you need to recheck the
> + *    entry after acquiring the INP lock.)
> + * D. To remove a node from the bucket, you must have that entry locked,
> + *    according to the criteria of Rule C. Also, the node must not be on
> + *    the expiry queue.
> + * E. The exception to C is the expiry queue fields, which are locked by
> + *    the TCPLOG_EXPIREQ lock.
> + *
> + * Buckets have a reference count. Each node is a reference. Further,
> + * other callers may add reference counts to keep a bucket from disappearing.
> + * You can add a reference as long as you own a lock sufficient to keep the
> + * bucket from disappearing. For example, a common use is:
> + *   a. Have a locked INP, but need to lock the TCPID_BUCKET.
> + *   b. Add a refcount on the bucket. (Safe because the INP lock prevents
> + *      the TCPID_BUCKET from going away.)
> + *   c. Drop the INP lock.
> + *   d. Acquire a lock on the TCPID_BUCKET.
> + *   e. Acquire a lock on the INP.
> + *   f. Drop the refcount on the bucket.
> + *      (At this point, the bucket may disappear.)
> + *
> + * Expire queue lock:
> + * You can acquire this with either the bucket or INP lock. Don't reverse it.
> + * When the expire code has committed to freeing a node, it resets the expiry
> + * time to SBT_MAX. That is the signal to everyone else that they should
> + * leave that node alone.
> + */
> +static struct rwlock tcp_id_tree_lock;
> +#define	TCPID_TREE_WLOCK()		rw_wlock(&tcp_id_tree_lock)
> +#define	TCPID_TREE_RLOCK()		rw_rlock(&tcp_id_tree_lock)
> +#define	TCPID_TREE_UPGRADE()		rw_try_upgrade(&tcp_id_tree_lock)
> +#define	TCPID_TREE_WUNLOCK()		rw_wunlock(&tcp_id_tree_lock)
> +#define	TCPID_TREE_RUNLOCK()		rw_runlock(&tcp_id_tree_lock)
> +#define	TCPID_TREE_WLOCK_ASSERT()	rw_assert(&tcp_id_tree_lock, RA_WLOCKED)
> +#define	TCPID_TREE_RLOCK_ASSERT()	rw_assert(&tcp_id_tree_lock, RA_RLOCKED)
> +#define	TCPID_TREE_UNLOCK_ASSERT()	rw_assert(&tcp_id_tree_lock, RA_UNLOCKED)
> +
> +#define	TCPID_BUCKET_LOCK_INIT(tlb)	mtx_init(&((tlb)->tlb_mtx), "tcp log id bucket", NULL, MTX_DEF)
> +#define	TCPID_BUCKET_LOCK_DESTROY(tlb)	mtx_destroy(&((tlb)->tlb_mtx))
> +#define	TCPID_BUCKET_LOCK(tlb)		mtx_lock(&((tlb)->tlb_mtx))
> +#define	TCPID_BUCKET_UNLOCK(tlb)	mtx_unlock(&((tlb)->tlb_mtx))
> +#define	TCPID_BUCKET_LOCK_ASSERT(tlb)	mtx_assert(&((tlb)->tlb_mtx), MA_OWNED)
> +#define	TCPID_BUCKET_UNLOCK_ASSERT(tlb) mtx_assert(&((tlb)->tlb_mtx), MA_NOTOWNED)
> +
> +#define	TCPID_BUCKET_REF(tlb)		refcount_acquire(&((tlb)->tlb_refcnt))
> +#define	TCPID_BUCKET_UNREF(tlb)		refcount_release(&((tlb)->tlb_refcnt))
> +
> +#define	TCPLOG_EXPIREQ_LOCK()		mtx_lock(&tcp_log_expireq_mtx)
> +#define	TCPLOG_EXPIREQ_UNLOCK()		mtx_unlock(&tcp_log_expireq_mtx)
> +
> +SLIST_HEAD(tcp_log_id_head, tcp_log_id_node);
> +
> +struct tcp_log_id_bucket
> +{
> +	/*
> +	 * tlb_id must be first. This lets us use strcmp on
> +	 * (struct tcp_log_id_bucket *) and (char *) interchangeably.
> +	 */
> +	char				tlb_id[TCP_LOG_ID_LEN];
> +	RB_ENTRY(tcp_log_id_bucket)	tlb_rb;
> +	struct tcp_log_id_head		tlb_head;
> +	struct mtx			tlb_mtx;
> +	volatile u_int			tlb_refcnt;
> +};
> +
> +struct tcp_log_id_node
> +{
> +	SLIST_ENTRY(tcp_log_id_node) tln_list;
> +	STAILQ_ENTRY(tcp_log_id_node) tln_expireq; /* Locked by the expireq lock */
> +	sbintime_t		tln_expiretime;	/* Locked by the expireq lock */
> +
> +	/*
> +	 * If INP is NULL, that means the connection has closed. We've
> +	 * saved the connection endpoint information and the log entries
> +	 * in the tln_ie and tln_entries members. We've also saved a pointer
> +	 * to the enclosing bucket here. If INP is not NULL, the information is
> +	 * in the PCB and not here.
> +	 */
> +	struct inpcb		*tln_inp;
> +	struct tcpcb		*tln_tp;
> +	struct tcp_log_id_bucket *tln_bucket;
> +	struct in_endpoints	tln_ie;
> +	struct tcp_log_stailq	tln_entries;
> +	int			tln_count;
> +	volatile int		tln_closed;
> +	uint8_t			tln_af;
> +};
> +
> +enum tree_lock_state {
> +	TREE_UNLOCKED = 0,
> +	TREE_RLOCKED,
> +	TREE_WLOCKED,
> +};
> +
> +/* Do we want to select this session for auto-logging? */
> +static __inline bool
> +tcp_log_selectauto(void)
> +{
> +
> +	/*
> 
> *** DIFF OUTPUT TRUNCATED AT 1000 LINES ***
> 


More information about the svn-src-all mailing list