xref: /freebsd/contrib/ntp/sntp/libevent/test/test-time.c (revision ba3c1f5972d7b90feb6e6da47905ff2757e0fe57)
1 /*
2  * Copyright (c) 2002-2007 Niels Provos <provos@citi.umich.edu>
3  * Copyright (c) 2007-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 #include "event2/event-config.h"
28 #include "util-internal.h"
29 
30 #include <sys/types.h>
31 #include <sys/stat.h>
32 #include <fcntl.h>
33 #include <stdlib.h>
34 #include <stdio.h>
35 #include <string.h>
36 #ifndef _WIN32
37 #include <unistd.h>
38 #include <sys/time.h>
39 #endif
40 #include <errno.h>
41 
42 #include "event2/event.h"
43 #include "event2/event_compat.h"
44 #include "event2/event_struct.h"
45 
46 int called = 0;
47 
48 #define NEVENT	20000
49 
50 struct event *ev[NEVENT];
51 
52 struct evutil_weakrand_state weakrand_state;
53 
54 static int
55 rand_int(int n)
56 {
57 	return evutil_weakrand_(&weakrand_state) % n;
58 }
59 
60 static void
61 time_cb(evutil_socket_t fd, short event, void *arg)
62 {
63 	struct timeval tv;
64 	int i, j;
65 
66 	called++;
67 
68 	if (called < 10*NEVENT) {
69 		for (i = 0; i < 10; i++) {
70 			j = rand_int(NEVENT);
71 			tv.tv_sec = 0;
72 			tv.tv_usec = rand_int(50000);
73 			if (tv.tv_usec % 2 || called < NEVENT)
74 				evtimer_add(ev[j], &tv);
75 			else
76 				evtimer_del(ev[j]);
77 		}
78 	}
79 }
80 
81 int
82 main(int argc, char **argv)
83 {
84 	struct event_base *base;
85 	struct timeval tv;
86 	int i;
87 
88 #ifdef _WIN32
89 	WORD wVersionRequested;
90 	WSADATA wsaData;
91 
92 	wVersionRequested = MAKEWORD(2, 2);
93 
94 	(void) WSAStartup(wVersionRequested, &wsaData);
95 #endif
96 
97 	evutil_weakrand_seed_(&weakrand_state, 0);
98 
99 	if (getenv("EVENT_DEBUG_LOGGING_ALL")) {
100 		event_enable_debug_logging(EVENT_DBG_ALL);
101 	}
102 
103 	base = event_base_new();
104 
105 	for (i = 0; i < NEVENT; i++) {
106 		ev[i] = evtimer_new(base, time_cb, event_self_cbarg());
107 		tv.tv_sec = 0;
108 		tv.tv_usec = rand_int(50000);
109 		evtimer_add(ev[i], &tv);
110 	}
111 
112 	i = event_base_dispatch(base);
113 
114 	printf("event_base_dispatch=%d, called=%d, EVENT=%d\n",
115 		i, called, NEVENT);
116 
117 	if (i == 1 && called >= NEVENT) {
118 		return EXIT_SUCCESS;
119 	} else {
120 		return EXIT_FAILURE;
121 	}
122 }
123 
124