sys/net/net_stats.c

   1 /*      $NetBSD: net_stats.c,v 1.3 2008/04/28 20:24:09 martin 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.3 2008/04/28 20:24:09 martin 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 }