1 |
|
|
/* $OpenBSD: timer.c,v 1.7 2016/10/28 08:01:53 rzalamena Exp $ */ |
2 |
|
|
|
3 |
|
|
/* |
4 |
|
|
* Copyright (c) 2008 Reyk Floeter <reyk@openbsd.org> |
5 |
|
|
* |
6 |
|
|
* Permission to use, copy, modify, and distribute this software for any |
7 |
|
|
* purpose with or without fee is hereby granted, provided that the above |
8 |
|
|
* copyright notice and this permission notice appear in all copies. |
9 |
|
|
* |
10 |
|
|
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES |
11 |
|
|
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF |
12 |
|
|
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR |
13 |
|
|
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES |
14 |
|
|
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN |
15 |
|
|
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF |
16 |
|
|
* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. |
17 |
|
|
*/ |
18 |
|
|
|
19 |
|
|
#include <sys/queue.h> |
20 |
|
|
#include <sys/types.h> |
21 |
|
|
#include <sys/time.h> |
22 |
|
|
#include <sys/sched.h> |
23 |
|
|
#include <sys/socket.h> |
24 |
|
|
#include <sys/sysctl.h> |
25 |
|
|
|
26 |
|
|
#include <net/if.h> |
27 |
|
|
#include <net/if_types.h> |
28 |
|
|
#include <netinet/in.h> |
29 |
|
|
#include <netinet/ip.h> |
30 |
|
|
#include <netinet/ip_var.h> |
31 |
|
|
#include <arpa/inet.h> |
32 |
|
|
|
33 |
|
|
#include <stdlib.h> |
34 |
|
|
#include <stdio.h> |
35 |
|
|
#include <errno.h> |
36 |
|
|
#include <event.h> |
37 |
|
|
#include <fcntl.h> |
38 |
|
|
#include <string.h> |
39 |
|
|
#include <unistd.h> |
40 |
|
|
#include <pwd.h> |
41 |
|
|
|
42 |
|
|
#include "snmpd.h" |
43 |
|
|
#include "mib.h" |
44 |
|
|
|
45 |
|
|
void timer_cpu(int, short, void *); |
46 |
|
|
int percentages(int, int64_t *, int64_t *, int64_t *, int64_t *); |
47 |
|
|
|
48 |
|
|
static int64_t **cp_time; |
49 |
|
|
static int64_t **cp_old; |
50 |
|
|
static int64_t **cp_diff; |
51 |
|
|
struct event cpu_ev; |
52 |
|
|
|
53 |
|
|
void |
54 |
|
|
timer_cpu(int fd, short event, void *arg) |
55 |
|
|
{ |
56 |
|
|
struct event *ev = (struct event *)arg; |
57 |
|
|
struct timeval tv = { 60, 0 }; /* every 60 seconds */ |
58 |
|
|
int mib[3] = { CTL_KERN, KERN_CPTIME2, 0 }, n; |
59 |
|
|
size_t len; |
60 |
|
|
int64_t *cptime2; |
61 |
|
|
|
62 |
|
|
len = CPUSTATES * sizeof(int64_t); |
63 |
|
|
for (n = 0; n < snmpd_env->sc_ncpu; n++) { |
64 |
|
|
mib[2] = n; |
65 |
|
|
cptime2 = snmpd_env->sc_cpustates + (CPUSTATES * n); |
66 |
|
|
if (sysctl(mib, 3, cp_time[n], &len, NULL, 0) == -1) |
67 |
|
|
continue; |
68 |
|
|
(void)percentages(CPUSTATES, cptime2, cp_time[n], |
69 |
|
|
cp_old[n], cp_diff[n]); |
70 |
|
|
#ifdef DEBUG |
71 |
|
|
log_debug("timer_cpu: cpu%d %lld%% idle in %llds", n, |
72 |
|
|
(cptime2[CP_IDLE] > 1000 ? |
73 |
|
|
1000 : (cptime2[CP_IDLE] / 10)), (long long) tv.tv_sec); |
74 |
|
|
#endif |
75 |
|
|
} |
76 |
|
|
|
77 |
|
|
evtimer_add(ev, &tv); |
78 |
|
|
} |
79 |
|
|
|
80 |
|
|
void |
81 |
|
|
timer_init(void) |
82 |
|
|
{ |
83 |
|
|
int mib[] = { CTL_HW, HW_NCPU }, i; |
84 |
|
|
size_t len; |
85 |
|
|
|
86 |
|
|
len = sizeof(snmpd_env->sc_ncpu); |
87 |
|
|
if (sysctl(mib, 2, &snmpd_env->sc_ncpu, &len, NULL, 0) == -1) |
88 |
|
|
fatal("sysctl"); |
89 |
|
|
|
90 |
|
|
snmpd_env->sc_cpustates = calloc(snmpd_env->sc_ncpu, |
91 |
|
|
CPUSTATES * sizeof(int64_t)); |
92 |
|
|
cp_time = calloc(snmpd_env->sc_ncpu, sizeof(int64_t *)); |
93 |
|
|
cp_old = calloc(snmpd_env->sc_ncpu, sizeof(int64_t *)); |
94 |
|
|
cp_diff = calloc(snmpd_env->sc_ncpu, sizeof(int64_t *)); |
95 |
|
|
if (snmpd_env->sc_cpustates == NULL || |
96 |
|
|
cp_time == NULL || cp_old == NULL || cp_diff == NULL) |
97 |
|
|
fatal("calloc"); |
98 |
|
|
for (i = 0; i < snmpd_env->sc_ncpu; i++) { |
99 |
|
|
cp_time[i] = calloc(CPUSTATES, sizeof(int64_t)); |
100 |
|
|
cp_old[i] = calloc(CPUSTATES, sizeof(int64_t)); |
101 |
|
|
cp_diff[i] = calloc(CPUSTATES, sizeof(int64_t)); |
102 |
|
|
if (cp_time[i] == NULL || cp_old[i] == NULL || |
103 |
|
|
cp_diff[i] == NULL) |
104 |
|
|
fatal("calloc"); |
105 |
|
|
} |
106 |
|
|
|
107 |
|
|
evtimer_set(&cpu_ev, timer_cpu, &cpu_ev); |
108 |
|
|
timer_cpu(0, EV_TIMEOUT, &cpu_ev); |
109 |
|
|
} |
110 |
|
|
|
111 |
|
|
/* |
112 |
|
|
* percentages() function to calculate CPU utilization. |
113 |
|
|
* Source code derived from the top(1) utility: |
114 |
|
|
* |
115 |
|
|
* Copyright (c) 1984, 1989, William LeFebvre, Rice University |
116 |
|
|
* Copyright (c) 1989, 1990, 1992, William LeFebvre, Northwestern University |
117 |
|
|
* |
118 |
|
|
* Redistribution and use in source and binary forms, with or without |
119 |
|
|
* modification, are permitted provided that the following conditions |
120 |
|
|
* are met: |
121 |
|
|
* 1. Redistributions of source code must retain the above copyright |
122 |
|
|
* notice, this list of conditions and the following disclaimer. |
123 |
|
|
* 2. Redistributions in binary form must reproduce the above copyright |
124 |
|
|
* notice, this list of conditions and the following disclaimer in the |
125 |
|
|
* documentation and/or other materials provided with the distribution. |
126 |
|
|
* |
127 |
|
|
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR |
128 |
|
|
* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES |
129 |
|
|
* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. |
130 |
|
|
* IN NO EVENT SHALL THE AUTHOR OR HIS EMPLOYER BE LIABLE FOR ANY DIRECT, INDIRECT, |
131 |
|
|
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT |
132 |
|
|
* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
133 |
|
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
134 |
|
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
135 |
|
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF |
136 |
|
|
* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
137 |
|
|
*/ |
138 |
|
|
int |
139 |
|
|
percentages(int cnt, int64_t *out, int64_t *new, int64_t *old, int64_t *diffs) |
140 |
|
|
{ |
141 |
|
|
int64_t change, total_change, *dp, half_total; |
142 |
|
|
int i; |
143 |
|
|
|
144 |
|
|
/* initialization */ |
145 |
|
|
total_change = 0; |
146 |
|
|
dp = diffs; |
147 |
|
|
|
148 |
|
|
/* calculate changes for each state and the overall change */ |
149 |
|
|
for (i = 0; i < cnt; i++) { |
150 |
|
|
if ((change = *new - *old) < 0) { |
151 |
|
|
/* this only happens when the counter wraps */ |
152 |
|
|
change = (*new - *old); |
153 |
|
|
} |
154 |
|
|
total_change += (*dp++ = change); |
155 |
|
|
*old++ = *new++; |
156 |
|
|
} |
157 |
|
|
|
158 |
|
|
/* avoid divide by zero potential */ |
159 |
|
|
if (total_change == 0) |
160 |
|
|
total_change = 1; |
161 |
|
|
|
162 |
|
|
/* calculate percentages based on overall change, rounding up */ |
163 |
|
|
half_total = total_change / 2l; |
164 |
|
|
for (i = 0; i < cnt; i++) |
165 |
|
|
*out++ = ((*diffs++ * 1000 + half_total) / total_change); |
166 |
|
|
|
167 |
|
|
/* return the total in case the caller wants to use it */ |
168 |
|
|
return (total_change); |
169 |
|
|
} |