GCC Code Coverage Report
Directory: ./ Exec Total Coverage
File: usr.sbin/snmpd/timer.c Lines: 0 43 0.0 %
Date: 2017-11-13 Branches: 0 24 0.0 %

Line Branch Exec Source
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
}