/*      $NetBSD: net_stats.h,v 1.6 2024/06/29 13:01:30 riastradh Exp $  */

/*-
* Copyright (c) 2008, 2020 The NetBSD Foundation, Inc.
* All rights reserved.
*
* This code is derived from software contributed to The NetBSD Foundation
* by Jason R. Thorpe.
*
* 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 NETBSD FOUNDATION, INC. 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 FOUNDATION 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.
*/

#ifndef _NET_NET_STATS_H_
#define _NET_NET_STATS_H_

#ifdef _KERNEL
#include <sys/percpu.h>

typedef struct net_stat_ref {
       /*
        * Flexible array member annoyingly requires some other
        * non-flexible array member.
        */
       uint64_t        nsr_stat[1];
       uint64_t        nsr_stats[];
} *net_stat_ref_t;

#define _NET_STAT_GETREF(stat)  ((net_stat_ref_t)percpu_getref((stat)))
#define _NET_STAT_PUTREF(stat)  percpu_putref((stat))

#define _NET_STATINC_REF(r, x)                                          \
do {                                                                    \
       uint64_t *_stat_ = (r)->nsr_stat;                               \
       _stat_[x]++;                                                    \
} while (/*CONSTCOND*/0)

#define _NET_STATINC(stat, x)                                           \
do {                                                                    \
       net_stat_ref_t _nsr_ = _NET_STAT_GETREF(stat);                  \
       _NET_STATINC_REF(_nsr_, x);                                     \
       _NET_STAT_PUTREF(stat);                                         \
} while (/*CONSTCOND*/0)

#define _NET_STATDEC_REF(r, x)                                          \
do {                                                                    \
       uint64_t *_stat_ = (r)->nsr_stat;                               \
       _stat_[x]--;                                                    \
} while (/*CONSTCOND*/0)

#define _NET_STATDEC(stat, x)                                           \
do {                                                                    \
       net_stat_ref_t _nsr_ = _NET_STAT_GETREF(stat);                  \
       _NET_STATDEC_REF(_nsr_, x);                                     \
       _NET_STAT_PUTREF(stat);                                         \
} while (/*CONSTCOND*/0)

#define _NET_STATADD_REF(r, x, v)                                       \
do {                                                                    \
       uint64_t *_stat_ = (r)->nsr_stat;                               \
       _stat_[x] += (v);                                               \
} while (/*CONSTCOND*/0)

#define _NET_STATADD(stat, x, v)                                        \
do {                                                                    \
       net_stat_ref_t _nsr_ = _NET_STAT_GETREF(stat);                  \
       _NET_STATADD_REF(_nsr_, x, v);                                  \
       _NET_STAT_PUTREF(stat);                                         \
} while (/*CONSTCOND*/0)

#define _NET_STATSUB_REF(r, x, v)                                       \
do {                                                                    \
       uint64_t *_stat_ = (r)->nsr_stat;                               \
       _stat_[x] -= (v);                                               \
} while (/*CONSTCOND*/0)

#define _NET_STATSUB(stat, x, v)                                        \
do {                                                                    \
       net_stat_ref_t _nsr_ = _NET_STAT_GETREF(stat);                  \
       _NET_STATSUB_REF(_nsr_, x, v);                                  \
       _NET_STAT_PUTREF(stat);                                         \
} while (/*CONSTCOND*/0)

__BEGIN_DECLS
struct lwp;
struct sysctlnode;

int     netstat_sysctl(percpu_t *, u_int,
                      const int *, u_int, void *,
                      size_t *, const void *, size_t,
                      const int *, struct lwp *, const struct sysctlnode *);

#define NETSTAT_SYSCTL(stat, nctrs)                                     \
       netstat_sysctl((stat), (nctrs), name, namelen, oldp, oldlenp,   \
                      newp, newlen, oname, l, rnode)
__END_DECLS
#endif /* _KERNEL */

#endif /* !_NET_NET_STATS_H_ */