lib/libkinfo/kinfo_sched.c

   1 /*
   2  * Copyright (c) 2004 The DragonFly Project.  All rights reserved.
   3  *
   4  * This code is derived from software contributed to The DragonFly Project
   5  * by Joerg Sonnenberger <joerg@bec.de>.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  *
  11  * 1. Redistributions of source code must retain the above copyright
  12  *    notice, this list of conditions and the following disclaimer.
  13  * 2. Redistributions in binary form must reproduce the above copyright
  14  *    notice, this list of conditions and the following disclaimer in
  15  *    the documentation and/or other materials provided with the
  16  *    distribution.
  17  * 3. Neither the name of The DragonFly Project nor the names of its
  18  *    contributors may be used to endorse or promote products derived
  19  *    from this software without specific, prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  22  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  23  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
  24  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE
  25  * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
  26  * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
  27  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  28  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
  29  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  30  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
  31  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  32  * SUCH DAMAGE.
  33  *
  34  * $DragonFly: src/lib/libkinfo/kinfo_sched.c,v 1.3 2005/04/27 16:16:30 hmp Exp $
  35  */
  36
  37 #include <sys/kinfo.h>
  38 #include <sys/param.h>
  39 #include <sys/sysctl.h>
  40
  41 #include <assert.h>
  42 #include <stdlib.h>
  43 #include <errno.h>
  44 #include <kinfo.h>
  45
  46 int
  47 kinfo_get_cpus(int *ncpus)
  48 {
  49         size_t len = sizeof(*ncpus);
  50
  51         return(sysctlbyname("hw.ncpu", ncpus, &len, NULL, 0));
  52 }
  53
  54 int
  55 kinfo_get_sched_ccpu(int *ccpu)
  56 {
  57         size_t len = sizeof(*ccpu);
  58
  59         return(sysctlbyname("kern.ccpu", ccpu, &len, NULL, 0));
  60 }
  61
  62 int
  63 kinfo_get_sched_cputime(struct kinfo_cputime *cputime)
  64 {
  65         struct kinfo_cputime *percpu = NULL;
  66         size_t len = sizeof(struct kinfo_cputime) * SMP_MAXCPU;
  67         int cpucount, error = 0;
  68
  69         _DIAGASSERT(cputime != NULL);
  70
  71         if ((percpu = malloc(len)) == NULL) {
  72                 error = ENOMEM;
  73                 goto done;
  74         }
  75
  76         /* retrieve verbatim per-cpu statistics from kernel */
  77         if (sysctlbyname("kern.cputime", percpu, &len, NULL, 0) < 0) {
  78                 error = errno;
  79                 goto done;
  80         } else {
  81                 percpu = reallocf(percpu, len);
  82                 if (percpu == NULL) {
  83                         error = ENOMEM;
  84                         goto done;
  85                 }
  86         }
  87
  88         /* aggregate per-cpu statistics retrieved from kernel */
  89         cpucount = len / sizeof(struct kinfo_cputime);
  90         cputime_pcpu_statistics(percpu, cputime, cpucount);
  91
  92 done:
  93         if (percpu != NULL)
  94                 free(percpu);
  95         return (error);
  96 }
  97
  98 int
  99 kinfo_get_sched_hz(int *hz)
 100 {
 101         struct kinfo_clockinfo clockinfo;
 102         size_t len = sizeof(clockinfo);
 103         int retval;
 104
 105         retval = sysctlbyname("kern.clockrate", &clockinfo, &len, NULL, 0);
 106         if (retval)
 107                 return(retval);
 108
 109         *hz = clockinfo.ci_hz;
 110         return(0);
 111 }
 112
 113 int
 114 kinfo_get_sched_stathz(int *stathz)
 115 {
 116         struct kinfo_clockinfo clockinfo;
 117         size_t len = sizeof(clockinfo);
 118         int retval;
 119
 120         retval = sysctlbyname("kern.clockrate", &clockinfo, &len, NULL, 0);
 121         if (retval)
 122                 return(retval);
 123
 124         *stathz = clockinfo.ci_stathz;
 125         return(0);
 126 }
 127
 128 int
 129 kinfo_get_sched_profhz(int *profhz)
 130 {
 131         struct kinfo_clockinfo clockinfo;
 132         size_t len = sizeof(clockinfo);
 133         int retval;
 134
 135         retval = sysctlbyname("kern.clockrate", &clockinfo, &len, NULL, 0);
 136         if (retval)
 137                 return(retval);
 138
 139         *profhz = clockinfo.ci_profhz;
 140         return(0);
 141 }