1 /*        $NetBSD: net_stats.h,v 1.6 2024/06/29 13:01:30 riastradh Exp $        */
2 
3 /*-
4  * Copyright (c) 2008, 2020 The NetBSD Foundation, Inc.
5  * All rights reserved.
6  *
7  * This code is derived from software contributed to The NetBSD Foundation
8  * by Jason R. Thorpe.
9  *
10  * Redistribution and use in source and binary forms, with or without
11  * modification, are permitted provided that the following conditions
12  * are met:
13  * 1. Redistributions of source code must retain the above copyright
14  *    notice, this list of conditions and the following disclaimer.
15  * 2. Redistributions in binary form must reproduce the above copyright
16  *    notice, this list of conditions and the following disclaimer in the
17  *    documentation and/or other materials provided with the distribution.
18  *
19  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29  * POSSIBILITY OF SUCH DAMAGE.
30  */
31 
32 #ifndef _NET_NET_STATS_H_
33 #define _NET_NET_STATS_H_
34 
35 #ifdef _KERNEL
36 #include <sys/percpu.h>
37 
38 typedef struct net_stat_ref {
39           /*
40            * Flexible array member annoyingly requires some other
41            * non-flexible array member.
42            */
43           uint64_t  nsr_stat[1];
44           uint64_t  nsr_stats[];
45 } *net_stat_ref_t;
46 
47 #define   _NET_STAT_GETREF(stat)        ((net_stat_ref_t)percpu_getref((stat)))
48 #define   _NET_STAT_PUTREF(stat)        percpu_putref((stat))
49 
50 #define   _NET_STATINC_REF(r, x)                                                          \
51 do {                                                                                      \
52           uint64_t *_stat_ = (r)->nsr_stat;                                     \
53           _stat_[x]++;                                                                    \
54 } while (/*CONSTCOND*/0)
55 
56 #define   _NET_STATINC(stat, x)                                                           \
57 do {                                                                                      \
58           net_stat_ref_t _nsr_ = _NET_STAT_GETREF(stat);                        \
59           _NET_STATINC_REF(_nsr_, x);                                           \
60           _NET_STAT_PUTREF(stat);                                                         \
61 } while (/*CONSTCOND*/0)
62 
63 #define   _NET_STATDEC_REF(r, x)                                                          \
64 do {                                                                                      \
65           uint64_t *_stat_ = (r)->nsr_stat;                                     \
66           _stat_[x]--;                                                                    \
67 } while (/*CONSTCOND*/0)
68 
69 #define   _NET_STATDEC(stat, x)                                                           \
70 do {                                                                                      \
71           net_stat_ref_t _nsr_ = _NET_STAT_GETREF(stat);                        \
72           _NET_STATDEC_REF(_nsr_, x);                                           \
73           _NET_STAT_PUTREF(stat);                                                         \
74 } while (/*CONSTCOND*/0)
75 
76 #define   _NET_STATADD_REF(r, x, v)                                             \
77 do {                                                                                      \
78           uint64_t *_stat_ = (r)->nsr_stat;                                     \
79           _stat_[x] += (v);                                                     \
80 } while (/*CONSTCOND*/0)
81 
82 #define   _NET_STATADD(stat, x, v)                                              \
83 do {                                                                                      \
84           net_stat_ref_t _nsr_ = _NET_STAT_GETREF(stat);                        \
85           _NET_STATADD_REF(_nsr_, x, v);                                                  \
86           _NET_STAT_PUTREF(stat);                                                         \
87 } while (/*CONSTCOND*/0)
88 
89 #define   _NET_STATSUB_REF(r, x, v)                                             \
90 do {                                                                                      \
91           uint64_t *_stat_ = (r)->nsr_stat;                                     \
92           _stat_[x] -= (v);                                                     \
93 } while (/*CONSTCOND*/0)
94 
95 #define   _NET_STATSUB(stat, x, v)                                              \
96 do {                                                                                      \
97           net_stat_ref_t _nsr_ = _NET_STAT_GETREF(stat);                        \
98           _NET_STATSUB_REF(_nsr_, x, v);                                                  \
99           _NET_STAT_PUTREF(stat);                                                         \
100 } while (/*CONSTCOND*/0)
101 
102 __BEGIN_DECLS
103 struct lwp;
104 struct sysctlnode;
105 
106 int       netstat_sysctl(percpu_t *, u_int,
107                            const int *, u_int, void *,
108                            size_t *, const void *, size_t,
109                            const int *, struct lwp *, const struct sysctlnode *);
110 
111 #define   NETSTAT_SYSCTL(stat, nctrs)                                           \
112           netstat_sysctl((stat), (nctrs), name, namelen, oldp, oldlenp,         \
113                            newp, newlen, oname, l, rnode)
114 __END_DECLS
115 #endif /* _KERNEL */
116 
117 #endif /* !_NET_NET_STATS_H_ */
118