1 | /* $NetBSD: net_stats.c,v 1.4 2008/05/04 07:22:14 thorpej Exp $ */ |
2 | |
3 | /*- |
4 | * Copyright (c) 2008 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 | #include <sys/cdefs.h> |
33 | __KERNEL_RCSID(0, "$NetBSD: net_stats.c,v 1.4 2008/05/04 07:22:14 thorpej Exp $" ); |
34 | |
35 | #include <sys/types.h> |
36 | #include <sys/systm.h> |
37 | #include <sys/sysctl.h> |
38 | #include <sys/kmem.h> |
39 | |
40 | #include <net/net_stats.h> |
41 | |
42 | typedef struct { |
43 | uint64_t *ctx_counters; /* pointer to collated counter array */ |
44 | u_int ctx_ncounters; /* number of counters in array */ |
45 | } netstat_sysctl_context; |
46 | |
47 | /* |
48 | * netstat_convert_to_user_cb -- |
49 | * Internal routine used as a call-back for percpu data enumeration. |
50 | */ |
51 | static void |
52 | netstat_convert_to_user_cb(void *v1, void *v2, struct cpu_info *ci) |
53 | { |
54 | const uint64_t *local_counters = v1; |
55 | netstat_sysctl_context *ctx = v2; |
56 | u_int i; |
57 | |
58 | for (i = 0; i < ctx->ctx_ncounters; i++) |
59 | ctx->ctx_counters[i] += local_counters[i]; |
60 | } |
61 | |
62 | /* |
63 | * netstat_sysctl -- |
64 | * Common routine for collating and reporting network statistics |
65 | * that are gathered per-CPU. Statistics counters are assumed |
66 | * to be arrays of uint64_t's. |
67 | */ |
68 | int |
69 | netstat_sysctl(percpu_t *stat, u_int ncounters, SYSCTLFN_ARGS) |
70 | { |
71 | netstat_sysctl_context ctx; |
72 | struct sysctlnode node; |
73 | uint64_t *counters; |
74 | size_t countersize; |
75 | int rv; |
76 | |
77 | countersize = sizeof(uint64_t) * ncounters; |
78 | |
79 | counters = kmem_zalloc(countersize, KM_SLEEP); |
80 | if (counters == NULL) |
81 | return (ENOMEM); |
82 | |
83 | ctx.ctx_counters = counters; |
84 | ctx.ctx_ncounters = ncounters; |
85 | |
86 | percpu_foreach(stat, netstat_convert_to_user_cb, &ctx); |
87 | |
88 | node = *rnode; |
89 | node.sysctl_data = counters; |
90 | node.sysctl_size = countersize; |
91 | rv = sysctl_lookup(SYSCTLFN_CALL(&node)); |
92 | |
93 | kmem_free(counters, countersize); |
94 | |
95 | return (rv); |
96 | } |
97 | |