Prepare to support polling(4) on multiple cpus:
[dragonfly/port-amd64.git] / lib / libkinfo / kinfo_sched.c
bloba342aecb8141b249a26c596ad18093b0b758acf0
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:
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.
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.
34 * $DragonFly: src/lib/libkinfo/kinfo_sched.c,v 1.3 2005/04/27 16:16:30 hmp Exp $
37 #include <sys/param.h>
38 #include <sys/sysctl.h>
40 #include <assert.h>
41 #include <stdlib.h>
42 #include <errno.h>
43 #include <kinfo.h>
45 int
46 kinfo_get_cpus(int *ncpus)
48 size_t len = sizeof(*ncpus);
50 return(sysctlbyname("hw.ncpu", ncpus, &len, NULL, 0));
53 int
54 kinfo_get_sched_ccpu(int *ccpu)
56 size_t len = sizeof(*ccpu);
58 return(sysctlbyname("kern.ccpu", ccpu, &len, NULL, 0));
61 int
62 kinfo_get_sched_cputime(struct kinfo_cputime *cputime)
64 struct kinfo_cputime *percpu = NULL;
65 size_t len = sizeof(struct kinfo_cputime) * SMP_MAXCPU;
66 int cpucount, error = 0;
68 _DIAGASSERT(cputime != NULL);
70 if ((percpu = malloc(len)) == NULL) {
71 error = ENOMEM;
72 goto done;
75 /* retrieve verbatim per-cpu statistics from kernel */
76 if (sysctlbyname("kern.cputime", percpu, &len, NULL, 0) < 0) {
77 error = errno;
78 goto done;
79 } else {
80 percpu = reallocf(percpu, len);
81 if (percpu == NULL) {
82 error = ENOMEM;
83 goto done;
87 /* aggregate per-cpu statistics retrieved from kernel */
88 cpucount = len / sizeof(struct kinfo_cputime);
89 cputime_pcpu_statistics(percpu, cputime, cpucount);
91 done:
92 if (percpu != NULL)
93 free(percpu);
94 return (error);
97 int
98 kinfo_get_sched_hz(int *hz)
100 struct kinfo_clockinfo clockinfo;
101 size_t len = sizeof(clockinfo);
102 int retval;
104 retval = sysctlbyname("kern.clockrate", &clockinfo, &len, NULL, 0);
105 if (retval)
106 return(retval);
108 *hz = clockinfo.ci_hz;
109 return(0);
113 kinfo_get_sched_stathz(int *stathz)
115 struct kinfo_clockinfo clockinfo;
116 size_t len = sizeof(clockinfo);
117 int retval;
119 retval = sysctlbyname("kern.clockrate", &clockinfo, &len, NULL, 0);
120 if (retval)
121 return(retval);
123 *stathz = clockinfo.ci_stathz;
124 return(0);
128 kinfo_get_sched_profhz(int *profhz)
130 struct kinfo_clockinfo clockinfo;
131 size_t len = sizeof(clockinfo);
132 int retval;
134 retval = sysctlbyname("kern.clockrate", &clockinfo, &len, NULL, 0);
135 if (retval)
136 return(retval);
138 *profhz = clockinfo.ci_profhz;
139 return(0);