xref: /freebsd/contrib/ntp/sntp/libevent/test/test-time.c (revision 1f4bcc459a76b7aa664f3fd557684cd0ba6da352)
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 
29 #include <sys/types.h>
30 #include <sys/stat.h>
31 #include <fcntl.h>
32 #include <stdlib.h>
33 #include <stdio.h>
34 #include <string.h>
35 #ifndef _WIN32
36 #include <unistd.h>
37 #include <sys/time.h>
38 #endif
39 #include <errno.h>
40 
41 #include "event2/event.h"
42 #include "event2/event_compat.h"
43 #include "event2/event_struct.h"
44 #include "util-internal.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 timeval tv;
85 	int i;
86 #ifdef _WIN32
87 	WORD wVersionRequested;
88 	WSADATA wsaData;
89 
90 	wVersionRequested = MAKEWORD(2, 2);
91 
92 	(void) WSAStartup(wVersionRequested, &wsaData);
93 #endif
94 
95 	evutil_weakrand_seed_(&weakrand_state, 0);
96 
97 	/* Initalize the event library */
98 	event_init();
99 
100 	for (i = 0; i < NEVENT; i++) {
101 		ev[i] = malloc(sizeof(struct event));
102 		assert(ev[i] != NULL);
103 
104 		/* Initalize one event */
105 		evtimer_set(ev[i], time_cb, ev[i]);
106 		tv.tv_sec = 0;
107 		tv.tv_usec = rand_int(50000);
108 		evtimer_add(ev[i], &tv);
109 	}
110 
111 	event_dispatch();
112 
113 
114 	printf("%d, %d\n", called, NEVENT);
115 	return (called < NEVENT);
116 }
117 
118