1 /* $NetBSD: test-changelist.c,v 1.1.1.3 2021/04/07 02:43:15 christos Exp $ */
2 /*
3 * Copyright (c) 2010-2012 Niels Provos and Nick Mathewson
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 * 3. The name of the author may not be used to endorse or promote products
14 * derived from this software without specific prior written permission.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
17 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
18 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
19 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
20 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
21 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
22 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
23 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
25 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26 */
27
28 #include "event2/event-config.h"
29 #include <sys/cdefs.h>
30 __RCSID("$NetBSD: test-changelist.c,v 1.1.1.3 2021/04/07 02:43:15 christos Exp $");
31
32 #ifdef _WIN32
33 #include <winsock2.h>
34 #include <windows.h>
35 #else
36 #include <unistd.h>
37 #endif
38 #include <sys/types.h>
39 #include <sys/stat.h>
40 #ifdef EVENT__HAVE_SYS_TIME_H
41 #include <sys/time.h>
42 #endif
43
44 #ifdef EVENT__HAVE_SYS_SOCKET_H
45 #include <sys/socket.h>
46 #endif
47 #include <fcntl.h>
48 #include <stdlib.h>
49 #include <stdio.h>
50 #include <string.h>
51 #include <errno.h>
52
53 #include "event2/event.h"
54 #include "event2/util.h"
55 #include <time.h>
56
57 struct cpu_usage_timer {
58 #ifdef _WIN32
59 HANDLE thread;
60 FILETIME usertimeBegin;
61 FILETIME kerneltimeBegin;
62 #else
63 clock_t ticksBegin;
64 #endif
65 struct timeval timeBegin;
66 };
67 static void
start_cpu_usage_timer(struct cpu_usage_timer * timer)68 start_cpu_usage_timer(struct cpu_usage_timer *timer)
69 {
70 #ifdef _WIN32
71 int r;
72 FILETIME createtime, exittime;
73 timer->thread = GetCurrentThread();
74 r = GetThreadTimes(timer->thread, &createtime, &exittime,
75 &timer->usertimeBegin, &timer->kerneltimeBegin);
76 if (r==0) printf("GetThreadTimes failed.");
77 #else
78 timer->ticksBegin = clock();
79 #endif
80
81 evutil_gettimeofday(&timer->timeBegin, NULL);
82 }
83 #ifdef _WIN32
84 static ev_int64_t
filetime_to_100nsec(const FILETIME * ft)85 filetime_to_100nsec(const FILETIME *ft)
86 {
87 /* Number of 100-nanosecond units */
88 ev_int64_t n = ft->dwHighDateTime;
89 n <<= 32;
90 n += ft->dwLowDateTime;
91 return n;
92 }
93 static double
filetime_diff(const FILETIME * ftStart,const FILETIME * ftEnd)94 filetime_diff(const FILETIME *ftStart, const FILETIME *ftEnd)
95 {
96 ev_int64_t s, e, diff;
97 double r;
98 s = filetime_to_100nsec(ftStart);
99 e = filetime_to_100nsec(ftEnd);
100 diff = e - s;
101 r = (double) diff;
102 return r / 1.0e7;
103 }
104 #endif
105
106 static void
get_cpu_usage(struct cpu_usage_timer * timer,double * secElapsedOut,double * secUsedOut,double * usageOut)107 get_cpu_usage(struct cpu_usage_timer *timer, double *secElapsedOut,
108 double *secUsedOut, double *usageOut)
109 {
110 #ifdef _WIN32
111 double usertime_seconds, kerneltime_seconds;
112 FILETIME createtime, exittime, usertimeEnd, kerneltimeEnd;
113 int r;
114 #else
115 clock_t ticksEnd;
116 #endif
117 struct timeval timeEnd, timeDiff;
118 double secondsPassed, secondsUsed;
119
120 #ifdef _WIN32
121 r = GetThreadTimes(timer->thread, &createtime, &exittime,
122 &usertimeEnd, &kerneltimeEnd);
123 if (r==0) printf("GetThreadTimes failed.");
124 usertime_seconds = filetime_diff(&timer->usertimeBegin, &usertimeEnd);
125 kerneltime_seconds = filetime_diff(&timer->kerneltimeBegin, &kerneltimeEnd);
126 secondsUsed = kerneltime_seconds + usertime_seconds;
127 #else
128 ticksEnd = clock();
129 secondsUsed = (ticksEnd - timer->ticksBegin) / (double)CLOCKS_PER_SEC;
130 #endif
131 evutil_gettimeofday(&timeEnd, NULL);
132 evutil_timersub(&timeEnd, &timer->timeBegin, &timeDiff);
133 secondsPassed = timeDiff.tv_sec + (timeDiff.tv_usec / 1.0e6);
134
135 *secElapsedOut = secondsPassed;
136 *secUsedOut = secondsUsed;
137 *usageOut = secondsUsed / secondsPassed;
138 }
139
140 static void
write_cb(evutil_socket_t fd,short event,void * arg)141 write_cb(evutil_socket_t fd, short event, void *arg)
142 {
143 printf("write callback. should only see this once\n");
144
145 /* got what we want remove the event */
146 event_del(*(struct event**)arg);
147
148 /* opps changed my mind add it back again */
149 event_add(*(struct event**)arg,NULL);
150
151 /* not a good day for decisiveness, I really didn't want it after all */
152 event_del(*(struct event**)arg);
153
154 }
155
156 static void
timeout_cb(evutil_socket_t fd,short event,void * arg)157 timeout_cb(evutil_socket_t fd, short event, void *arg)
158 {
159 printf("timeout fired, time to end test\n");
160 event_del(*(struct event**)arg);
161 return;
162 }
163
164 int
main(int argc,char ** argv)165 main(int argc, char **argv)
166 {
167 struct event* ev;
168 struct event* timeout;
169 struct event_base* base;
170
171 evutil_socket_t pair[2];
172 struct timeval tv;
173 struct cpu_usage_timer timer;
174
175 double usage, secPassed, secUsed;
176
177 #ifdef _WIN32
178 WORD wVersionRequested;
179 WSADATA wsaData;
180
181 wVersionRequested = MAKEWORD(2, 2);
182
183 (void) WSAStartup(wVersionRequested, &wsaData);
184 #endif
185 if (evutil_socketpair(AF_UNIX, SOCK_STREAM, 0, pair) == -1)
186 return (1);
187
188 /* Initialize the event library */
189 if (!(base = event_base_new()))
190 return (1);
191
192 /* Initialize a timeout to terminate the test */
193 timeout = evtimer_new(base,timeout_cb,&timeout);
194 /* and watch for writability on one end of the pipe */
195 ev = event_new(base,pair[1],EV_WRITE | EV_PERSIST, write_cb, &ev);
196
197 tv.tv_sec = 1;
198 tv.tv_usec = 500*1000;
199
200 evtimer_add(timeout, &tv);
201
202 event_add(ev, NULL);
203
204 start_cpu_usage_timer(&timer);
205
206 event_base_dispatch(base);
207
208 event_free(ev);
209 event_free(timeout);
210 event_base_free(base);
211
212 get_cpu_usage(&timer, &secPassed, &secUsed, &usage);
213
214 /* attempt to calculate our cpu usage over the test should be
215 virtually nil */
216
217 printf("usec used=%d, usec passed=%d, cpu usage=%.2f%%\n",
218 (int)(secUsed*1e6),
219 (int)(secPassed*1e6),
220 usage*100);
221
222 if (usage > 50.0) /* way too high */
223 return 1;
224
225 return 0;
226 }
227
228