vici: Add a libvici low-level client library
This commit is contained in:
Martin Willi
@@ -0,0 +1,144 @@
|
||||
/*
|
||||
* Copyright (C) 2014 Martin Willi
|
||||
* Copyright (C) 2014 revosec AG
|
||||
*
|
||||
* This program is free software; you can redistribute it and/or modify it
|
||||
* under the terms of the GNU General Public License as published by the
|
||||
* Free Software Foundation; either version 2 of the License, or (at your
|
||||
* option) any later version. See <http://www.fsf.org/copyleft/gpl.txt>.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful, but
|
||||
* WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
|
||||
* or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
||||
* for more details.
|
||||
*/
|
||||
|
||||
#include <test_suite.h>
|
||||
|
||||
#include "../vici_dispatcher.h"
|
||||
#include "../libvici.h"
|
||||
|
||||
#include <unistd.h>
|
||||
|
||||
#define URI "unix:///tmp/strongswan-vici-event-test"
|
||||
|
||||
static void event_cb(void *user, char *name, vici_res_t *ev)
|
||||
{
|
||||
int *count = (int*)user;
|
||||
|
||||
ck_assert_str_eq(name, "test");
|
||||
ck_assert(vici_parse(ev) == VICI_PARSE_KEY_VALUE);
|
||||
ck_assert_str_eq(vici_parse_name(ev), "key1");
|
||||
ck_assert_str_eq(vici_parse_value_str(ev), "value1");
|
||||
ck_assert(vici_parse(ev) == VICI_PARSE_END);
|
||||
|
||||
(*count)++;
|
||||
}
|
||||
|
||||
START_TEST(test_event)
|
||||
{
|
||||
vici_dispatcher_t *dispatcher;
|
||||
vici_conn_t *conn;
|
||||
int count = 0;
|
||||
|
||||
lib->processor->set_threads(lib->processor, 8);
|
||||
|
||||
dispatcher = vici_dispatcher_create(URI);
|
||||
ck_assert(dispatcher);
|
||||
|
||||
dispatcher->manage_event(dispatcher, "test", TRUE);
|
||||
|
||||
vici_init();
|
||||
conn = vici_connect(URI);
|
||||
ck_assert(conn);
|
||||
|
||||
ck_assert(vici_register(conn, "test", event_cb, &count) == 0);
|
||||
ck_assert(vici_register(conn, "nonexistent", event_cb, &count) != 0);
|
||||
|
||||
dispatcher->raise_event(dispatcher, "test", vici_message_create_from_args(
|
||||
VICI_KEY_VALUE, "key1", chunk_from_str("value1"),
|
||||
VICI_END));
|
||||
|
||||
while (count == 0)
|
||||
{
|
||||
usleep(1000);
|
||||
}
|
||||
|
||||
vici_disconnect(conn);
|
||||
|
||||
dispatcher->manage_event(dispatcher, "test", FALSE);
|
||||
|
||||
lib->processor->cancel(lib->processor);
|
||||
dispatcher->destroy(dispatcher);
|
||||
|
||||
vici_deinit();
|
||||
}
|
||||
END_TEST
|
||||
|
||||
START_TEST(test_stress)
|
||||
{
|
||||
vici_dispatcher_t *dispatcher;
|
||||
vici_conn_t *conn;
|
||||
int count = 0, i, total = 50;
|
||||
|
||||
lib->processor->set_threads(lib->processor, 8);
|
||||
|
||||
dispatcher = vici_dispatcher_create(URI);
|
||||
ck_assert(dispatcher);
|
||||
|
||||
dispatcher->manage_event(dispatcher, "test", TRUE);
|
||||
dispatcher->manage_event(dispatcher, "dummy", TRUE);
|
||||
|
||||
vici_init();
|
||||
conn = vici_connect(URI);
|
||||
ck_assert(conn);
|
||||
|
||||
vici_register(conn, "test", event_cb, &count);
|
||||
|
||||
for (i = 0; i < total; i++)
|
||||
{
|
||||
/* do some event re/deregistration in between */
|
||||
ck_assert(vici_register(conn, "dummy", event_cb, NULL) == 0);
|
||||
|
||||
dispatcher->raise_event(dispatcher, "test",
|
||||
vici_message_create_from_args(
|
||||
VICI_KEY_VALUE, "key1", chunk_from_str("value1"),
|
||||
VICI_END));
|
||||
|
||||
ck_assert(vici_register(conn, "dummy", NULL, NULL) == 0);
|
||||
}
|
||||
|
||||
while (count < total)
|
||||
{
|
||||
usleep(1000);
|
||||
}
|
||||
|
||||
vici_disconnect(conn);
|
||||
|
||||
dispatcher->manage_event(dispatcher, "test", FALSE);
|
||||
dispatcher->manage_event(dispatcher, "dummy", FALSE);
|
||||
|
||||
lib->processor->cancel(lib->processor);
|
||||
dispatcher->destroy(dispatcher);
|
||||
|
||||
vici_deinit();
|
||||
}
|
||||
END_TEST
|
||||
|
||||
Suite *event_suite_create()
|
||||
{
|
||||
Suite *s;
|
||||
TCase *tc;
|
||||
|
||||
s = suite_create("vici events");
|
||||
|
||||
tc = tcase_create("single");
|
||||
tcase_add_test(tc, test_event);
|
||||
suite_add_tcase(s, tc);
|
||||
|
||||
tc = tcase_create("stress");
|
||||
tcase_add_test(tc, test_stress);
|
||||
suite_add_tcase(s, tc);
|
||||
|
||||
return s;
|
||||
}
|
||||
@@ -0,0 +1,243 @@
|
||||
/*
|
||||
* Copyright (C) 2014 Martin Willi
|
||||
* Copyright (C) 2014 revosec AG
|
||||
*
|
||||
* This program is free software; you can redistribute it and/or modify it
|
||||
* under the terms of the GNU General Public License as published by the
|
||||
* Free Software Foundation; either version 2 of the License, or (at your
|
||||
* option) any later version. See <http://www.fsf.org/copyleft/gpl.txt>.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful, but
|
||||
* WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
|
||||
* or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
||||
* for more details.
|
||||
*/
|
||||
|
||||
#include <test_suite.h>
|
||||
|
||||
#include "../vici_dispatcher.h"
|
||||
#include "../libvici.h"
|
||||
|
||||
#include <unistd.h>
|
||||
|
||||
#define URI "unix:///tmp/strongswan-vici-request-test"
|
||||
|
||||
static void encode_section(vici_req_t *req)
|
||||
{
|
||||
vici_begin_section(req, "section1");
|
||||
vici_add_key_valuef(req, "key1", "value%u", 1);
|
||||
vici_add_key_value(req, "key2", "value2", strlen("value2"));
|
||||
vici_end_section(req);
|
||||
}
|
||||
|
||||
static void decode_section(vici_res_t *res)
|
||||
{
|
||||
char *str;
|
||||
int len;
|
||||
|
||||
ck_assert(vici_parse(res) == VICI_PARSE_BEGIN_SECTION);
|
||||
ck_assert_str_eq(vici_parse_name(res), "section1");
|
||||
ck_assert(vici_parse(res) == VICI_PARSE_KEY_VALUE);
|
||||
ck_assert_str_eq(vici_parse_name(res), "key1");
|
||||
ck_assert_str_eq(vici_parse_value_str(res), "value1");
|
||||
ck_assert(vici_parse(res) == VICI_PARSE_KEY_VALUE);
|
||||
ck_assert_str_eq(vici_parse_name(res), "key2");
|
||||
str = vici_parse_value(res, &len);
|
||||
ck_assert(chunk_equals(chunk_from_str("value2"), chunk_create(str, len)));
|
||||
ck_assert(vici_parse(res) == VICI_PARSE_END_SECTION);
|
||||
ck_assert(vici_parse(res) == VICI_PARSE_END);
|
||||
}
|
||||
|
||||
static void encode_list(vici_req_t *req)
|
||||
{
|
||||
vici_begin_list(req, "list1");
|
||||
vici_add_list_item(req, "item1", strlen("item1"));
|
||||
vici_add_list_itemf(req, "item%u", 2);
|
||||
vici_end_list(req);
|
||||
}
|
||||
|
||||
static void decode_list(vici_res_t *res)
|
||||
{
|
||||
char *str;
|
||||
int len;
|
||||
|
||||
ck_assert(vici_parse(res) == VICI_PARSE_BEGIN_LIST);
|
||||
ck_assert_str_eq(vici_parse_name(res), "list1");
|
||||
ck_assert(vici_parse(res) == VICI_PARSE_LIST_ITEM);
|
||||
ck_assert_str_eq(vici_parse_value_str(res), "item1");
|
||||
ck_assert(vici_parse(res) == VICI_PARSE_LIST_ITEM);
|
||||
str = vici_parse_value(res, &len);
|
||||
ck_assert(chunk_equals(chunk_from_str("item2"), chunk_create(str, len)));
|
||||
ck_assert(vici_parse(res) == VICI_PARSE_END_LIST);
|
||||
ck_assert(vici_parse(res) == VICI_PARSE_END);
|
||||
}
|
||||
|
||||
static struct {
|
||||
void (*encode)(vici_req_t* req);
|
||||
void (*decode)(vici_res_t* res);
|
||||
} echo_tests[] = {
|
||||
{ encode_section, decode_section },
|
||||
{ encode_list, decode_list },
|
||||
};
|
||||
|
||||
static vici_message_t* echo_cb(void *user, char *name,
|
||||
u_int id, vici_message_t *request)
|
||||
{
|
||||
ck_assert_str_eq(name, "echo");
|
||||
ck_assert_int_eq((uintptr_t)user, 1);
|
||||
|
||||
return vici_message_create_from_enumerator(request->create_enumerator(request));
|
||||
}
|
||||
|
||||
START_TEST(test_echo)
|
||||
{
|
||||
vici_dispatcher_t *dispatcher;
|
||||
vici_conn_t *conn;
|
||||
vici_req_t *req;
|
||||
vici_res_t *res;
|
||||
|
||||
lib->processor->set_threads(lib->processor, 8);
|
||||
|
||||
dispatcher = vici_dispatcher_create(URI);
|
||||
ck_assert(dispatcher);
|
||||
|
||||
dispatcher->manage_command(dispatcher, "echo", echo_cb, (void*)(uintptr_t)1);
|
||||
|
||||
vici_init();
|
||||
conn = vici_connect(URI);
|
||||
ck_assert(conn);
|
||||
|
||||
req = vici_begin("echo");
|
||||
echo_tests[_i].encode(req);
|
||||
res = vici_submit(req, conn);
|
||||
ck_assert(res);
|
||||
echo_tests[_i].decode(res);
|
||||
vici_free_res(res);
|
||||
|
||||
vici_disconnect(conn);
|
||||
|
||||
dispatcher->manage_command(dispatcher, "echo", NULL, NULL);
|
||||
|
||||
lib->processor->cancel(lib->processor);
|
||||
dispatcher->destroy(dispatcher);
|
||||
|
||||
vici_deinit();
|
||||
}
|
||||
END_TEST
|
||||
|
||||
START_TEST(test_missing)
|
||||
{
|
||||
vici_dispatcher_t *dispatcher;
|
||||
vici_conn_t *conn;
|
||||
vici_req_t *req;
|
||||
vici_res_t *res;
|
||||
|
||||
lib->processor->set_threads(lib->processor, 8);
|
||||
|
||||
dispatcher = vici_dispatcher_create(URI);
|
||||
ck_assert(dispatcher);
|
||||
|
||||
vici_init();
|
||||
conn = vici_connect(URI);
|
||||
ck_assert(conn);
|
||||
|
||||
req = vici_begin("nonexistent");
|
||||
encode_section(req);
|
||||
res = vici_submit(req, conn);
|
||||
ck_assert(res == NULL);
|
||||
|
||||
vici_disconnect(conn);
|
||||
|
||||
dispatcher->manage_command(dispatcher, "echo", NULL, NULL);
|
||||
|
||||
lib->processor->cancel(lib->processor);
|
||||
dispatcher->destroy(dispatcher);
|
||||
|
||||
vici_deinit();
|
||||
}
|
||||
END_TEST
|
||||
|
||||
static void event_cb(void *user, char *name, vici_res_t *ev)
|
||||
{
|
||||
int *events = (int*)user;
|
||||
|
||||
(*events)++;
|
||||
}
|
||||
|
||||
START_TEST(test_stress)
|
||||
{
|
||||
vici_dispatcher_t *dispatcher;
|
||||
vici_conn_t *conn;
|
||||
vici_req_t *req;
|
||||
vici_res_t *res;
|
||||
int i, total = 50, events = 0;
|
||||
|
||||
lib->processor->set_threads(lib->processor, 8);
|
||||
|
||||
dispatcher = vici_dispatcher_create(URI);
|
||||
ck_assert(dispatcher);
|
||||
|
||||
dispatcher->manage_command(dispatcher, "echo", echo_cb, (void*)(uintptr_t)1);
|
||||
dispatcher->manage_event(dispatcher, "dummy", TRUE);
|
||||
|
||||
vici_init();
|
||||
conn = vici_connect(URI);
|
||||
ck_assert(conn);
|
||||
|
||||
for (i = 0; i < total; i++)
|
||||
{
|
||||
/* do some event management in between */
|
||||
ck_assert(vici_register(conn, "dummy", event_cb, &events) == 0);
|
||||
dispatcher->raise_event(dispatcher, "dummy",
|
||||
vici_message_create_from_args(
|
||||
VICI_KEY_VALUE, "key1", chunk_from_str("value1"),
|
||||
VICI_END));
|
||||
|
||||
req = vici_begin("echo");
|
||||
encode_section(req);
|
||||
res = vici_submit(req, conn);
|
||||
ck_assert(res);
|
||||
decode_section(res);
|
||||
vici_free_res(res);
|
||||
|
||||
ck_assert(vici_register(conn, "dummy", NULL, NULL) == 0);
|
||||
}
|
||||
|
||||
while (events < total)
|
||||
{
|
||||
usleep(1000);
|
||||
}
|
||||
|
||||
vici_disconnect(conn);
|
||||
|
||||
dispatcher->manage_command(dispatcher, "echo", NULL, NULL);
|
||||
dispatcher->manage_event(dispatcher, "dummy", FALSE);
|
||||
|
||||
lib->processor->cancel(lib->processor);
|
||||
dispatcher->destroy(dispatcher);
|
||||
|
||||
vici_deinit();
|
||||
}
|
||||
END_TEST
|
||||
|
||||
Suite *request_suite_create()
|
||||
{
|
||||
Suite *s;
|
||||
TCase *tc;
|
||||
|
||||
s = suite_create("vici request");
|
||||
|
||||
tc = tcase_create("echo");
|
||||
tcase_add_loop_test(tc, test_echo, 0, countof(echo_tests));
|
||||
suite_add_tcase(s, tc);
|
||||
|
||||
tc = tcase_create("missing");
|
||||
tcase_add_test(tc, test_missing);
|
||||
suite_add_tcase(s, tc);
|
||||
|
||||
tc = tcase_create("stress");
|
||||
tcase_add_test(tc, test_stress);
|
||||
suite_add_tcase(s, tc);
|
||||
|
||||
return s;
|
||||
}
|
||||
Reference in New Issue
Block a user