Annotation of embedaddon/strongswan/src/libcharon/plugins/vici/suites/test_event.c, revision 1.1.1.1
1.1 misho 1: /*
2: * Copyright (C) 2014 Martin Willi
3: * Copyright (C) 2014 revosec AG
4: *
5: * This program is free software; you can redistribute it and/or modify it
6: * under the terms of the GNU General Public License as published by the
7: * Free Software Foundation; either version 2 of the License, or (at your
8: * option) any later version. See <http://www.fsf.org/copyleft/gpl.txt>.
9: *
10: * This program is distributed in the hope that it will be useful, but
11: * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
12: * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
13: * for more details.
14: */
15:
16: #include <test_suite.h>
17:
18: #include "../vici_dispatcher.h"
19: #include "../libvici.h"
20:
21: #include <unistd.h>
22:
23: #ifdef WIN32
24: # define URI "tcp://127.0.0.1:6543"
25: #else /* !WIN32 */
26: # define URI "unix:///tmp/strongswan-vici-event-test"
27: #endif /* !WIN32 */
28:
29: static void event_cb(void *user, char *name, vici_res_t *ev)
30: {
31: int *count = (int*)user;
32:
33: ck_assert_str_eq(name, "test");
34: ck_assert(vici_parse(ev) == VICI_PARSE_KEY_VALUE);
35: ck_assert_str_eq(vici_parse_name(ev), "key1");
36: ck_assert_str_eq(vici_parse_value_str(ev), "value1");
37: ck_assert(vici_parse(ev) == VICI_PARSE_END);
38:
39: (*count)++;
40: }
41:
42: START_TEST(test_event)
43: {
44: vici_dispatcher_t *dispatcher;
45: vici_conn_t *conn;
46: int count = 0;
47:
48: lib->processor->set_threads(lib->processor, 8);
49:
50: dispatcher = vici_dispatcher_create(URI);
51: ck_assert(dispatcher);
52:
53: dispatcher->manage_event(dispatcher, "test", TRUE);
54:
55: vici_init();
56: conn = vici_connect(URI);
57: ck_assert(conn);
58:
59: ck_assert(vici_register(conn, "test", event_cb, &count) == 0);
60: ck_assert(vici_register(conn, "nonexistent", event_cb, &count) != 0);
61:
62: dispatcher->raise_event(dispatcher, "test", 0, vici_message_create_from_args(
63: VICI_KEY_VALUE, "key1", chunk_from_str("value1"),
64: VICI_END));
65:
66: while (count == 0)
67: {
68: usleep(1000);
69: }
70:
71: vici_disconnect(conn);
72:
73: dispatcher->manage_event(dispatcher, "test", FALSE);
74:
75: lib->processor->cancel(lib->processor);
76: dispatcher->destroy(dispatcher);
77:
78: vici_deinit();
79: }
80: END_TEST
81:
82: #define EVENT_COUNT 500
83:
84: CALLBACK(raise_cb, vici_message_t*,
85: vici_dispatcher_t *dispatcher, char *name, u_int id, vici_message_t *req)
86: {
87: u_int i;
88:
89: for (i = 0; i < EVENT_COUNT; i++)
90: {
91: dispatcher->raise_event(dispatcher, "event", id,
92: vici_message_create_from_args(
93: VICI_KEY_VALUE, "counter", chunk_from_thing(i),
94: VICI_END));
95: }
96: return vici_message_create_from_args(VICI_END);
97: }
98:
99: CALLBACK(raise_event_cb, void,
100: int *count, char *name, vici_res_t *ev)
101: {
102: u_int *value, len;
103:
104: ck_assert_str_eq(name, "event");
105: ck_assert(vici_parse(ev) == VICI_PARSE_KEY_VALUE);
106: ck_assert_str_eq(vici_parse_name(ev), "counter");
107: value = vici_parse_value(ev, &len);
108: ck_assert_int_eq(len, sizeof(*value));
109: ck_assert(vici_parse(ev) == VICI_PARSE_END);
110:
111: ck_assert_int_eq(*count, *value);
112: (*count)++;
113: }
114:
115: START_TEST(test_raise_events)
116: {
117: vici_dispatcher_t *dispatcher;
118: vici_res_t *res;
119: vici_conn_t *conn;
120: int count = 0;
121:
122: lib->processor->set_threads(lib->processor, 8);
123:
124: dispatcher = vici_dispatcher_create(URI);
125: ck_assert(dispatcher);
126:
127: dispatcher->manage_event(dispatcher, "event", TRUE);
128: dispatcher->manage_command(dispatcher, "raise", raise_cb, dispatcher);
129:
130: vici_init();
131: conn = vici_connect(URI);
132: ck_assert(conn);
133:
134: ck_assert(vici_register(conn, "event", raise_event_cb, &count) == 0);
135:
136: res = vici_submit(vici_begin("raise"), conn);
137:
138: ck_assert_int_eq(count, EVENT_COUNT);
139: ck_assert(res);
140: vici_free_res(res);
141:
142: vici_disconnect(conn);
143:
144: dispatcher->manage_event(dispatcher, "event", FALSE);
145: dispatcher->manage_command(dispatcher, "raise", NULL, NULL);
146:
147: lib->processor->cancel(lib->processor);
148: dispatcher->destroy(dispatcher);
149:
150: vici_deinit();
151: }
152: END_TEST
153:
154: START_TEST(test_stress)
155: {
156: vici_dispatcher_t *dispatcher;
157: vici_conn_t *conn;
158: int count = 0, i, total = 50;
159:
160: lib->processor->set_threads(lib->processor, 8);
161:
162: dispatcher = vici_dispatcher_create(URI);
163: ck_assert(dispatcher);
164:
165: dispatcher->manage_event(dispatcher, "test", TRUE);
166: dispatcher->manage_event(dispatcher, "dummy", TRUE);
167:
168: vici_init();
169: conn = vici_connect(URI);
170: ck_assert(conn);
171:
172: vici_register(conn, "test", event_cb, &count);
173:
174: for (i = 0; i < total; i++)
175: {
176: /* do some event re/deregistration in between */
177: ck_assert(vici_register(conn, "dummy", event_cb, NULL) == 0);
178:
179: dispatcher->raise_event(dispatcher, "test", 0,
180: vici_message_create_from_args(
181: VICI_KEY_VALUE, "key1", chunk_from_str("value1"),
182: VICI_END));
183:
184: ck_assert(vici_register(conn, "dummy", NULL, NULL) == 0);
185: }
186:
187: while (count < total)
188: {
189: usleep(1000);
190: }
191:
192: vici_disconnect(conn);
193:
194: dispatcher->manage_event(dispatcher, "test", FALSE);
195: dispatcher->manage_event(dispatcher, "dummy", FALSE);
196:
197: lib->processor->cancel(lib->processor);
198: dispatcher->destroy(dispatcher);
199:
200: vici_deinit();
201: }
202: END_TEST
203:
204: Suite *event_suite_create()
205: {
206: Suite *s;
207: TCase *tc;
208:
209: s = suite_create("vici events");
210:
211: tc = tcase_create("single");
212: tcase_add_test(tc, test_event);
213: suite_add_tcase(s, tc);
214:
215: tc = tcase_create("raise events");
216: tcase_add_test(tc, test_raise_events);
217: suite_add_tcase(s, tc);
218:
219: tc = tcase_create("stress");
220: tcase_add_test(tc, test_stress);
221: suite_add_tcase(s, tc);
222:
223: return s;
224: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>
![[BACK]](/icons/cvsweb/back.gif){kind=icon}
Return to test_event.c CVS log ![[TXT]](/icons/cvsweb/text.gif){kind=icon}
![[DIR]](/icons/cvsweb/dir.gif){kind=icon}
Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / strongswan / src / libcharon / plugins / vici / suites