Merge from vendor branch LIBARCHIVE:
[dragonfly.git] / 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/param.h>
38 #include <sys/sysctl.h>
39
40 #include <assert.h>
41 #include <stdlib.h>
42 #include <errno.h>
43 #include <kinfo.h>
44
45 int
46 kinfo_get_cpus(int *ncpus)
47 {
48         size_t len = sizeof(*ncpus);
49
50         return(sysctlbyname("hw.ncpu", ncpus, &len, NULL, 0));
51 }
52
53 int
54 kinfo_get_sched_ccpu(int *ccpu)
55 {
56         size_t len = sizeof(*ccpu);
57
58         return(sysctlbyname("kern.ccpu", ccpu, &len, NULL, 0));
59 }
60
61 int
62 kinfo_get_sched_cputime(struct kinfo_cputime *cputime)
63 {
64         struct kinfo_cputime *percpu = NULL;
65         size_t len = sizeof(struct kinfo_cputime) * SMP_MAXCPU;
66         int cpucount, error = 0;
67
68         _DIAGASSERT(cputime != NULL);
69
70         if ((percpu = malloc(len)) == NULL) {
71                 error = ENOMEM;
72                 goto done;
73         }
74
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;
84                 }
85         }
86
87         /* aggregate per-cpu statistics retrieved from kernel */
88         cpucount = len / sizeof(struct kinfo_cputime);
89         cputime_pcpu_statistics(percpu, cputime, cpucount);
90
91 done:
92         if (percpu != NULL)
93                 free(percpu);
94         return (error);
95 }
96
97 int
98 kinfo_get_sched_hz(int *hz)
99 {
100         struct kinfo_clockinfo clockinfo;
101         size_t len = sizeof(clockinfo);
102         int retval;
103
104         retval = sysctlbyname("kern.clockrate", &clockinfo, &len, NULL, 0);
105         if (retval)
106                 return(retval);
107
108         *hz = clockinfo.ci_hz;
109         return(0);
110 }
111
112 int
113 kinfo_get_sched_stathz(int *stathz)
114 {
115         struct kinfo_clockinfo clockinfo;
116         size_t len = sizeof(clockinfo);
117         int retval;
118
119         retval = sysctlbyname("kern.clockrate", &clockinfo, &len, NULL, 0);
120         if (retval)
121                 return(retval);
122
123         *stathz = clockinfo.ci_stathz;
124         return(0);
125 }
126
127 int
128 kinfo_get_sched_profhz(int *profhz)
129 {
130         struct kinfo_clockinfo clockinfo;
131         size_t len = sizeof(clockinfo);
132         int retval;
133
134         retval = sysctlbyname("kern.clockrate", &clockinfo, &len, NULL, 0);
135         if (retval)
136                 return(retval);
137
138         *profhz = clockinfo.ci_profhz;
139         return(0);
140 }