1/*
2 * Copyright (c) 2009 Mark Heily <mark@heily.com>
3 *
4 * Permission to use, copy, modify, and distribute this software for any
5 * purpose with or without fee is hereby granted, provided that the above
6 * copyright notice and this permission notice appear in all copies.
7 *
8 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15 *
16 * $FreeBSD$
17 */
18
19#include "common.h"
20
21int kqfd;
22
23void
24test_kevent_timer_add(void)
25{
26    const char *test_id = "kevent(EVFILT_TIMER, EV_ADD)";
27    struct kevent kev;
28
29    test_begin(test_id);
30
31    EV_SET(&kev, 1, EVFILT_TIMER, EV_ADD, 0, 1000, NULL);
32    if (kevent(kqfd, &kev, 1, NULL, 0, NULL) < 0)
33        err(1, "%s", test_id);
34
35    success();
36}
37
38void
39test_kevent_timer_del(void)
40{
41    const char *test_id = "kevent(EVFILT_TIMER, EV_DELETE)";
42    struct kevent kev;
43
44    test_begin(test_id);
45
46    EV_SET(&kev, 1, EVFILT_TIMER, EV_DELETE, 0, 0, NULL);
47    if (kevent(kqfd, &kev, 1, NULL, 0, NULL) < 0)
48        err(1, "%s", test_id);
49
50    test_no_kevents();
51
52    success();
53}
54
55void
56test_kevent_timer_get(void)
57{
58    const char *test_id = "kevent(EVFILT_TIMER, wait)";
59    struct kevent kev;
60
61    test_begin(test_id);
62
63    EV_SET(&kev, 1, EVFILT_TIMER, EV_ADD, 0, 1000, NULL);
64    if (kevent(kqfd, &kev, 1, NULL, 0, NULL) < 0)
65        err(1, "%s", test_id);
66
67    kev.flags |= EV_CLEAR;
68    kev.data = 1;
69    kevent_cmp(&kev, kevent_get(kqfd));
70
71    EV_SET(&kev, 1, EVFILT_TIMER, EV_DELETE, 0, 0, NULL);
72    if (kevent(kqfd, &kev, 1, NULL, 0, NULL) < 0)
73        err(1, "%s", test_id);
74
75    success();
76}
77
78static void
79test_oneshot(void)
80{
81    const char *test_id = "kevent(EVFILT_TIMER, EV_ONESHOT)";
82    struct kevent kev;
83
84    test_begin(test_id);
85
86    test_no_kevents();
87
88    EV_SET(&kev, vnode_fd, EVFILT_TIMER, EV_ADD | EV_ONESHOT, 0, 500,NULL);
89    if (kevent(kqfd, &kev, 1, NULL, 0, NULL) < 0)
90        err(1, "%s", test_id);
91
92    /* Retrieve the event */
93    kev.flags = EV_ADD | EV_CLEAR | EV_ONESHOT;
94    kev.data = 1;
95    kevent_cmp(&kev, kevent_get(kqfd));
96
97    /* Check if the event occurs again */
98    sleep(3);
99    test_no_kevents();
100
101
102    success();
103}
104
105static void
106test_periodic(void)
107{
108    const char *test_id = "kevent(EVFILT_TIMER, periodic)";
109    struct kevent kev;
110
111    test_begin(test_id);
112
113    test_no_kevents();
114
115    EV_SET(&kev, vnode_fd, EVFILT_TIMER, EV_ADD, 0, 1000,NULL);
116    if (kevent(kqfd, &kev, 1, NULL, 0, NULL) < 0)
117        err(1, "%s", test_id);
118
119    /* Retrieve the event */
120    kev.flags = EV_ADD | EV_CLEAR;
121    kev.data = 1;
122    kevent_cmp(&kev, kevent_get(kqfd));
123
124    /* Check if the event occurs again */
125    sleep(1);
126    kevent_cmp(&kev, kevent_get(kqfd));
127
128    /* Delete the event */
129    kev.flags = EV_DELETE;
130    if (kevent(kqfd, &kev, 1, NULL, 0, NULL) < 0)
131        err(1, "%s", test_id);
132
133    success();
134}
135
136static void
137disable_and_enable(void)
138{
139    const char *test_id = "kevent(EVFILT_TIMER, EV_DISABLE and EV_ENABLE)";
140    struct kevent kev;
141
142    test_begin(test_id);
143
144    test_no_kevents();
145
146    /* Add the watch and immediately disable it */
147    EV_SET(&kev, vnode_fd, EVFILT_TIMER, EV_ADD | EV_ONESHOT, 0, 2000,NULL);
148    if (kevent(kqfd, &kev, 1, NULL, 0, NULL) < 0)
149        err(1, "%s", test_id);
150    kev.flags = EV_DISABLE;
151    if (kevent(kqfd, &kev, 1, NULL, 0, NULL) < 0)
152        err(1, "%s", test_id);
153    test_no_kevents();
154
155    /* Re-enable and check again */
156    kev.flags = EV_ENABLE;
157    if (kevent(kqfd, &kev, 1, NULL, 0, NULL) < 0)
158        err(1, "%s", test_id);
159
160    kev.flags = EV_ADD | EV_CLEAR | EV_ONESHOT;
161    kev.data = 1;
162    kevent_cmp(&kev, kevent_get(kqfd));
163
164    success();
165}
166
167void
168test_evfilt_timer()
169{
170	kqfd = kqueue();
171    test_kevent_timer_add();
172    test_kevent_timer_del();
173    test_kevent_timer_get();
174    test_oneshot();
175    test_periodic();
176    disable_and_enable();
177	close(kqfd);
178}
179