mirror of
https://git.proxmox.com/git/mirror_corosync
synced 2025-07-27 01:22:10 +00:00
CTS: Add a condb test agent and python tests.
git-svn-id: http://svn.fedorahosted.org/svn/corosync/trunk@2736 fd59a12c-fef9-0310-b244-a6a79926bd2f
This commit is contained in:
parent
8df2e184f9
commit
988be543cf
247
cts/agents/common_test_agent.c
Normal file
247
cts/agents/common_test_agent.c
Normal file
@ -0,0 +1,247 @@
|
||||
/*
|
||||
* Copyright (c) 2010 Red Hat, Inc.
|
||||
*
|
||||
* All rights reserved.
|
||||
*
|
||||
* Author: Angus Salkeld (asalkeld@redhat.com)
|
||||
*
|
||||
* This software licensed under BSD license, the text of which follows:
|
||||
*
|
||||
* Redistribution and use in source and binary forms, with or without
|
||||
* modification, are permitted provided that the following conditions are met:
|
||||
*
|
||||
* - Redistributions of source code must retain the above copyright notice,
|
||||
* this list of conditions and the following disclaimer.
|
||||
* - Redistributions in binary form must reproduce the above copyright notice,
|
||||
* this list of conditions and the following disclaimer in the documentation
|
||||
* and/or other materials provided with the distribution.
|
||||
* - Neither the name of the MontaVista Software, Inc. nor the names of its
|
||||
* contributors may be used to endorse or promote products derived from this
|
||||
* software without specific prior written permission.
|
||||
*
|
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
||||
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
|
||||
* THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#include <errno.h>
|
||||
#include <unistd.h>
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <assert.h>
|
||||
#include <string.h>
|
||||
#include <sys/types.h>
|
||||
#include <sys/socket.h>
|
||||
#include <netinet/in.h>
|
||||
#include <arpa/inet.h>
|
||||
#include <netdb.h>
|
||||
#include <syslog.h>
|
||||
#include <poll.h>
|
||||
#include <unistd.h>
|
||||
#include <fcntl.h>
|
||||
#include <corosync/hdb.h>
|
||||
#include <corosync/totem/coropoll.h>
|
||||
#include "common_test_agent.h"
|
||||
|
||||
|
||||
int32_t parse_debug = 0;
|
||||
static char big_and_buf_rx[HOW_BIG_AND_BUF];
|
||||
ta_do_command_fn do_command;
|
||||
static hdb_handle_t poll_handle;
|
||||
|
||||
|
||||
hdb_handle_t ta_poll_handle_get(void)
|
||||
{
|
||||
return poll_handle;
|
||||
}
|
||||
|
||||
static void ta_handle_command (int sock, char* msg)
|
||||
{
|
||||
int num_args;
|
||||
char *saveptr = NULL;
|
||||
char *str = strdup (msg);
|
||||
char *str_len;
|
||||
char *str_arg;
|
||||
char *args[5];
|
||||
int i = 0;
|
||||
int a = 0;
|
||||
char* func = NULL;
|
||||
|
||||
if (parse_debug)
|
||||
syslog (LOG_DEBUG,"%s (MSG:%s)\n", __func__, msg);
|
||||
|
||||
str_len = strtok_r (str, ":", &saveptr);
|
||||
assert (str_len);
|
||||
|
||||
num_args = atoi (str_len) * 2;
|
||||
for (i = 0; i < num_args / 2; i++) {
|
||||
str_len = strtok_r (NULL, ":", &saveptr);
|
||||
str_arg = strtok_r (NULL, ":", &saveptr);
|
||||
if (func == NULL) {
|
||||
/* first "arg" is the function */
|
||||
if (parse_debug)
|
||||
syslog (LOG_DEBUG, "(LEN:%s, FUNC:%s)", str_len, str_arg);
|
||||
func = str_arg;
|
||||
a = 0;
|
||||
} else {
|
||||
args[a] = str_arg;
|
||||
a++;
|
||||
if (parse_debug)
|
||||
syslog (LOG_DEBUG, "(LEN:%s, ARG:%s)", str_len, str_arg);
|
||||
}
|
||||
}
|
||||
do_command (sock, func, args, a+1);
|
||||
|
||||
free (str);
|
||||
}
|
||||
|
||||
static int server_process_data_fn (hdb_handle_t handle,
|
||||
int fd,
|
||||
int revents,
|
||||
void *data)
|
||||
{
|
||||
char *saveptr;
|
||||
char *msg;
|
||||
char *cmd;
|
||||
int32_t nbytes;
|
||||
|
||||
if ((nbytes = recv (fd, big_and_buf_rx, sizeof (big_and_buf_rx), 0)) <= 0) {
|
||||
/* got error or connection closed by client */
|
||||
if (nbytes == 0) {
|
||||
/* connection closed */
|
||||
syslog (LOG_WARNING, "socket %d hung up: exiting...\n", fd);
|
||||
} else {
|
||||
syslog (LOG_ERR,"recv() failed: %s", strerror(errno));
|
||||
}
|
||||
close (fd);
|
||||
poll_stop (handle);
|
||||
} else {
|
||||
big_and_buf_rx[nbytes] = '\0';
|
||||
|
||||
msg = strtok_r (big_and_buf_rx, ";", &saveptr);
|
||||
assert (msg);
|
||||
while (msg) {
|
||||
cmd = strdup (msg);
|
||||
ta_handle_command (fd, cmd);
|
||||
free (cmd);
|
||||
msg = strtok_r (NULL, ";", &saveptr);
|
||||
}
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int server_accept_fn (hdb_handle_t handle,
|
||||
int fd, int revents, void *data)
|
||||
{
|
||||
socklen_t addrlen;
|
||||
struct sockaddr_in in_addr;
|
||||
int new_fd;
|
||||
int res;
|
||||
|
||||
addrlen = sizeof (struct sockaddr_in);
|
||||
|
||||
retry_accept:
|
||||
new_fd = accept (fd, (struct sockaddr *)&in_addr, &addrlen);
|
||||
if (new_fd == -1 && errno == EINTR) {
|
||||
goto retry_accept;
|
||||
}
|
||||
|
||||
if (new_fd == -1) {
|
||||
syslog (LOG_ERR,
|
||||
"Could not accept connection: %s\n", strerror (errno));
|
||||
return (0); /* This is an error, but -1 would indicate disconnect from poll loop */
|
||||
}
|
||||
|
||||
res = fcntl (new_fd, F_SETFL, O_NONBLOCK);
|
||||
if (res == -1) {
|
||||
syslog (LOG_ERR,
|
||||
"Could not set non-blocking operation on connection: %s\n",
|
||||
strerror (errno));
|
||||
close (new_fd);
|
||||
return (0); /* This is an error, but -1 would indicate disconnect from poll loop */
|
||||
}
|
||||
|
||||
poll_dispatch_add (poll_handle, new_fd, POLLIN|POLLNVAL, NULL, server_process_data_fn);
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
static int create_server_sockect (int server_port)
|
||||
{
|
||||
int listener;
|
||||
int yes = 1;
|
||||
int rv;
|
||||
struct addrinfo hints, *ai, *p;
|
||||
char server_port_str[16];
|
||||
|
||||
/* get a socket and bind it
|
||||
*/
|
||||
sprintf(server_port_str, "%d", server_port);
|
||||
memset (&hints, 0, sizeof hints);
|
||||
hints.ai_family = AF_UNSPEC;
|
||||
hints.ai_socktype = SOCK_STREAM;
|
||||
hints.ai_flags = AI_PASSIVE;
|
||||
if ((rv = getaddrinfo (NULL, server_port_str, &hints, &ai)) != 0) {
|
||||
syslog (LOG_ERR, "%s\n", gai_strerror (rv));
|
||||
exit (1);
|
||||
}
|
||||
|
||||
for (p = ai; p != NULL; p = p->ai_next) {
|
||||
listener = socket (p->ai_family, p->ai_socktype, p->ai_protocol);
|
||||
if (listener < 0) {
|
||||
continue;
|
||||
}
|
||||
|
||||
/* lose the pesky "address already in use" error message
|
||||
*/
|
||||
if (setsockopt (listener, SOL_SOCKET, SO_REUSEADDR,
|
||||
&yes, sizeof(int)) < 0) {
|
||||
syslog (LOG_ERR, "setsockopt() failed: %s\n", strerror (errno));
|
||||
}
|
||||
|
||||
if (bind (listener, p->ai_addr, p->ai_addrlen) < 0) {
|
||||
syslog (LOG_ERR, "bind() failed: %s\n", strerror (errno));
|
||||
close (listener);
|
||||
continue;
|
||||
}
|
||||
|
||||
break;
|
||||
}
|
||||
|
||||
if (p == NULL) {
|
||||
syslog (LOG_ERR, "failed to bind\n");
|
||||
exit (2);
|
||||
}
|
||||
|
||||
freeaddrinfo (ai);
|
||||
|
||||
if (listen (listener, 10) == -1) {
|
||||
syslog (LOG_ERR, "listen() failed: %s", strerror(errno));
|
||||
exit (3);
|
||||
}
|
||||
|
||||
return listener;
|
||||
}
|
||||
|
||||
int test_agent_run(int server_port, ta_do_command_fn func)
|
||||
{
|
||||
int listener;
|
||||
|
||||
do_command = func;
|
||||
poll_handle = poll_create ();
|
||||
|
||||
listener = create_server_sockect (server_port);
|
||||
poll_dispatch_add (poll_handle, listener, POLLIN|POLLNVAL, NULL, server_accept_fn);
|
||||
|
||||
return poll_run (poll_handle);
|
||||
}
|
||||
|
43
cts/agents/common_test_agent.h
Normal file
43
cts/agents/common_test_agent.h
Normal file
@ -0,0 +1,43 @@
|
||||
/*
|
||||
* Copyright (c) 2010 Red Hat, Inc.
|
||||
*
|
||||
* All rights reserved.
|
||||
*
|
||||
* Author: Angus Salkeld (asalkeld@redhat.com)
|
||||
*
|
||||
* This software licensed under BSD license, the text of which follows:
|
||||
*
|
||||
* Redistribution and use in source and binary forms, with or without
|
||||
* modification, are permitted provided that the following conditions are met:
|
||||
*
|
||||
* - Redistributions of source code must retain the above copyright notice,
|
||||
* this list of conditions and the following disclaimer.
|
||||
* - Redistributions in binary form must reproduce the above copyright notice,
|
||||
* this list of conditions and the following disclaimer in the documentation
|
||||
* and/or other materials provided with the distribution.
|
||||
* - Neither the name of the MontaVista Software, Inc. nor the names of its
|
||||
* contributors may be used to endorse or promote products derived from this
|
||||
* software without specific prior written permission.
|
||||
*
|
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
||||
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
|
||||
* THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
extern int32_t parse_debug;
|
||||
#define HOW_BIG_AND_BUF 4096
|
||||
|
||||
typedef void (*ta_do_command_fn) (int sock, char* func, char*args[], int num_args);
|
||||
|
||||
int test_agent_run(int server_port, ta_do_command_fn func);
|
||||
|
||||
hdb_handle_t ta_poll_handle_get(void);
|
||||
|
||||
|
565
cts/agents/confdb_test_agent.c
Normal file
565
cts/agents/confdb_test_agent.c
Normal file
@ -0,0 +1,565 @@
|
||||
/*
|
||||
* Copyright (c) 2008, 2009 Red Hat Inc
|
||||
*
|
||||
* All rights reserved.
|
||||
*
|
||||
* Author: Christine Caulfield <ccaulfie@redhat.com>
|
||||
*
|
||||
* This software licensed under BSD license, the text of which follows:
|
||||
*
|
||||
* Redistribution and use in source and binary forms, with or without
|
||||
* modification, are permitted provided that the following conditions are met:
|
||||
*
|
||||
* - Redistributions of source code must retain the above copyright notice,
|
||||
* this list of conditions and the following disclaimer.
|
||||
* - Redistributions in binary form must reproduce the above copyright notice,
|
||||
* this list of conditions and the following disclaimer in the documentation
|
||||
* and/or other materials provided with the distribution.
|
||||
* - Neither the name of the MontaVista Software, Inc. nor the names of its
|
||||
* contributors may be used to endorse or promote products derived from this
|
||||
* software without specific prior written permission.
|
||||
*
|
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
||||
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
|
||||
* THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#include <config.h>
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <errno.h>
|
||||
#include <unistd.h>
|
||||
#include <string.h>
|
||||
#include <sys/types.h>
|
||||
#include <sys/socket.h>
|
||||
#include <netinet/in.h>
|
||||
#include <arpa/inet.h>
|
||||
#include <netdb.h>
|
||||
#include <sys/un.h>
|
||||
#include <syslog.h>
|
||||
|
||||
#include <corosync/corotypes.h>
|
||||
#include <corosync/confdb.h>
|
||||
#include "common_test_agent.h"
|
||||
|
||||
|
||||
#define OK_STR "OK"
|
||||
#define FAIL_STR "FAIL"
|
||||
#define NOT_SUPPORTED_STR "NOT_SUPPORTED"
|
||||
|
||||
#define INCDEC_VALUE 45
|
||||
|
||||
confdb_callbacks_t callbacks = {
|
||||
.confdb_key_change_notify_fn = NULL,
|
||||
.confdb_object_create_change_notify_fn = NULL,
|
||||
.confdb_object_delete_change_notify_fn = NULL
|
||||
};
|
||||
|
||||
typedef enum {
|
||||
NTF_OBJECT_CREATED,
|
||||
NTF_OBJECT_DELETED,
|
||||
NTF_KEY_CREATED,
|
||||
NTF_KEY_REPLACED,
|
||||
NTF_KEY_DELETED,
|
||||
NTF_NONE,
|
||||
} ntf_callback_type_t;
|
||||
|
||||
static ntf_callback_type_t callback_type;
|
||||
static char ntf_object_name[256];
|
||||
static size_t ntf_object_name_len;
|
||||
static char ntf_key_name[256];
|
||||
static size_t ntf_key_name_len;
|
||||
static char ntf_key_value[256];
|
||||
static size_t ntf_key_value_len;
|
||||
|
||||
static void ta_key_change_notify (
|
||||
confdb_handle_t handle,
|
||||
confdb_change_type_t change_type,
|
||||
hdb_handle_t parent_object_handle,
|
||||
hdb_handle_t object_handle,
|
||||
const void *object_name,
|
||||
size_t object_name_len,
|
||||
const void *key_name,
|
||||
size_t key_name_len,
|
||||
const void *key_value,
|
||||
size_t key_value_len)
|
||||
{
|
||||
switch (change_type) {
|
||||
case OBJECT_KEY_CREATED:
|
||||
callback_type = NTF_KEY_CREATED;
|
||||
break;
|
||||
case OBJECT_KEY_DELETED:
|
||||
callback_type = NTF_KEY_DELETED;
|
||||
break;
|
||||
case OBJECT_KEY_REPLACED:
|
||||
callback_type = NTF_KEY_REPLACED;
|
||||
break;
|
||||
default:
|
||||
assert (0);
|
||||
break;
|
||||
}
|
||||
ntf_object_name_len = object_name_len;
|
||||
memcpy (ntf_object_name, object_name, object_name_len);
|
||||
|
||||
ntf_key_name_len = key_name_len;
|
||||
memcpy (ntf_key_name, key_name, key_name_len);
|
||||
|
||||
ntf_key_value_len = key_value_len;
|
||||
memcpy (ntf_key_value, key_value, key_value_len);
|
||||
}
|
||||
|
||||
static void ta_object_create_notify (
|
||||
confdb_handle_t handle,
|
||||
hdb_handle_t parent_object_handle,
|
||||
hdb_handle_t object_handle,
|
||||
const void *name_pt,
|
||||
size_t name_len)
|
||||
{
|
||||
callback_type = NTF_OBJECT_CREATED;
|
||||
ntf_object_name_len = name_len;
|
||||
memcpy (ntf_object_name, name_pt, name_len);
|
||||
}
|
||||
|
||||
static void ta_object_delete_notify (
|
||||
confdb_handle_t handle,
|
||||
hdb_handle_t parent_object_handle,
|
||||
const void *name_pt,
|
||||
size_t name_len)
|
||||
{
|
||||
callback_type = NTF_OBJECT_DELETED;
|
||||
ntf_object_name_len = name_len;
|
||||
memcpy (ntf_object_name, name_pt, name_len);
|
||||
}
|
||||
|
||||
confdb_callbacks_t valid_callbacks = {
|
||||
.confdb_key_change_notify_fn = ta_key_change_notify,
|
||||
.confdb_object_create_change_notify_fn = ta_object_create_notify,
|
||||
.confdb_object_delete_change_notify_fn = ta_object_delete_notify
|
||||
};
|
||||
|
||||
static void set_get_test (int sock)
|
||||
{
|
||||
confdb_handle_t handle;
|
||||
char response[100];
|
||||
int res;
|
||||
hdb_handle_t object_handle;
|
||||
confdb_value_types_t type;
|
||||
char key_value[256];
|
||||
char key2_value[256];
|
||||
size_t value_len;
|
||||
size_t value2_len;
|
||||
|
||||
syslog (LOG_ERR, "%s START", __func__);
|
||||
|
||||
snprintf (response, 100, "%s", OK_STR);
|
||||
|
||||
res = confdb_initialize (&handle, &callbacks);
|
||||
if (res != CS_OK) {
|
||||
syslog (LOG_ERR, "Could not initialize confdb error %d", res);
|
||||
goto send_response;
|
||||
}
|
||||
/* Add a scratch object and put 2 keys into it */
|
||||
res = confdb_object_create (handle, OBJECT_PARENT_HANDLE,
|
||||
"testconfdb", strlen("testconfdb"), &object_handle);
|
||||
if (res != CS_OK) {
|
||||
syslog (LOG_ERR, "error creating 'testconfdb' object: %d", res);
|
||||
goto send_response;
|
||||
}
|
||||
|
||||
res = confdb_key_create (handle, object_handle,
|
||||
"testkey", strlen ("testkey"),
|
||||
"one", strlen ("one"));
|
||||
if (res != CS_OK) {
|
||||
syslog (LOG_ERR, "error creating 'testconfdb' key 1: %d", res);
|
||||
goto send_response;
|
||||
}
|
||||
|
||||
res = confdb_key_replace (handle, object_handle,
|
||||
"testkey", strlen ("testkey"),
|
||||
"one", strlen ("one"),
|
||||
"newone", strlen ("newone"));
|
||||
if (res != CS_OK) {
|
||||
syslog (LOG_ERR, "error replace 'testconfdb' key 2: %d", res);
|
||||
goto send_response;
|
||||
}
|
||||
|
||||
res = confdb_key_get_typed (handle, object_handle,
|
||||
"testkey", key_value, &value_len, &type);
|
||||
if (res != CS_OK) {
|
||||
syslog (LOG_ERR, "Could not get \"testkey\" key: %d", res);
|
||||
goto send_response;
|
||||
}
|
||||
if (strcmp (key_value, "newone") != 0) {
|
||||
syslog (LOG_ERR, "Key not set correctly");
|
||||
goto send_response;
|
||||
}
|
||||
if (type != CONFDB_VALUETYPE_ANY) {
|
||||
syslog (LOG_ERR, "Key type not set correctly");
|
||||
goto send_response;
|
||||
}
|
||||
res = confdb_key_get (handle, object_handle,
|
||||
"testkey", strlen ("testkey"), key2_value, &value2_len);
|
||||
if (res != CS_OK) {
|
||||
syslog (LOG_ERR, "Could not get \"testkey\" key: %d", res);
|
||||
goto send_response;
|
||||
}
|
||||
if (value2_len != value_len) {
|
||||
syslog (LOG_ERR, "value length from confdb_key_get:%u and confdb_key_get_typed:%u differ.",
|
||||
(uint32_t)value_len, (uint32_t)value2_len);
|
||||
goto send_response;
|
||||
}
|
||||
|
||||
res = confdb_key_delete (handle, object_handle,
|
||||
"testkey", strlen ("testkey"), key2_value, value2_len);
|
||||
if (res != CS_OK) {
|
||||
syslog (LOG_ERR, "Could not get \"testkey\" key: %d", res);
|
||||
goto send_response;
|
||||
}
|
||||
|
||||
/* Remove it.
|
||||
Check that it doesn't exist when the full tree dump runs next */
|
||||
res = confdb_object_destroy(handle, object_handle);
|
||||
if (res != CS_OK) {
|
||||
syslog (LOG_ERR, "error destroying 'testconfdb' object: %d", res);
|
||||
goto send_response;
|
||||
}
|
||||
|
||||
snprintf (response, 100, "%s", OK_STR);
|
||||
|
||||
send_response:
|
||||
syslog (LOG_ERR, "%s %s", __func__, response);
|
||||
send (sock, response, strlen (response) + 1, 0);
|
||||
confdb_finalize (handle);
|
||||
}
|
||||
|
||||
static void increment_decrement_test (int sock)
|
||||
{
|
||||
char response[100];
|
||||
int res;
|
||||
uint32_t incdec_value;
|
||||
hdb_handle_t object_handle;
|
||||
confdb_handle_t handle;
|
||||
confdb_handle_t par_handle;
|
||||
|
||||
snprintf (response, 100, "%s", FAIL_STR);
|
||||
|
||||
res = confdb_initialize (&handle, &callbacks);
|
||||
if (res != CS_OK) {
|
||||
syslog (LOG_ERR, "Could not initialize confdb error %d", res);
|
||||
goto send_response;
|
||||
}
|
||||
/* Add a scratch object and put 1 keys into it */
|
||||
res = confdb_object_create(handle, OBJECT_PARENT_HANDLE,
|
||||
"testconfdb", strlen("testconfdb"), &object_handle);
|
||||
if (res != CS_OK) {
|
||||
syslog (LOG_ERR, "error creating 'testconfdb' object: %d", res);
|
||||
goto send_response;
|
||||
}
|
||||
|
||||
res = confdb_object_parent_get (handle, object_handle, &par_handle);
|
||||
if (res != CS_OK) {
|
||||
syslog (LOG_ERR, "error getting parent of 'testconfdb' object: %d", res);
|
||||
goto send_response;
|
||||
}
|
||||
if (par_handle != OBJECT_PARENT_HANDLE) {
|
||||
syslog (LOG_ERR, "wrong parent handle");
|
||||
goto send_response;
|
||||
}
|
||||
|
||||
|
||||
incdec_value = INCDEC_VALUE;
|
||||
res = confdb_key_create_typed (handle, object_handle, "incdec",
|
||||
&incdec_value, sizeof(incdec_value), CONFDB_VALUETYPE_UINT32);
|
||||
if (res != CS_OK) {
|
||||
syslog (LOG_ERR, "error creating 'testconfdb' key 4: %d\n", res);
|
||||
goto send_response;
|
||||
}
|
||||
res = confdb_key_increment(handle, object_handle, "incdec", strlen("incdec"), &incdec_value);
|
||||
if (res != CS_OK) {
|
||||
syslog (LOG_ERR, "error incrementing 'testconfdb' key 4: %d\n", res);
|
||||
goto send_response;
|
||||
}
|
||||
if (incdec_value == INCDEC_VALUE + 1) {
|
||||
syslog (LOG_INFO, "incremented value = %d\n", incdec_value);
|
||||
}
|
||||
else {
|
||||
syslog (LOG_ERR, "ERROR: incremented value = %d (should be %d)\n", incdec_value, INCDEC_VALUE+1);
|
||||
goto send_response;
|
||||
}
|
||||
res = confdb_key_decrement(handle, object_handle, "incdec", strlen("incdec"), &incdec_value);
|
||||
if (res != CS_OK) {
|
||||
syslog (LOG_ERR, "error decrementing 'testconfdb' key 4: %d\n", res);
|
||||
goto send_response;
|
||||
}
|
||||
if (incdec_value == INCDEC_VALUE) {
|
||||
syslog (LOG_ERR, "decremented value = %d\n", incdec_value);
|
||||
}
|
||||
else {
|
||||
syslog (LOG_ERR, "ERROR: decremented value = %d (should be %d)\n", incdec_value, INCDEC_VALUE);
|
||||
goto send_response;
|
||||
}
|
||||
/* Remove it.
|
||||
Check that it doesn't exist when the full tree dump runs next */
|
||||
res = confdb_object_destroy(handle, object_handle);
|
||||
if (res != CS_OK) {
|
||||
syslog (LOG_ERR, "error destroying 'testconfdb' object: %d\n", res);
|
||||
goto send_response;
|
||||
}
|
||||
|
||||
snprintf (response, 100, "%s", OK_STR);
|
||||
|
||||
send_response:
|
||||
confdb_finalize (handle);
|
||||
send (sock, response, strlen (response) + 1, 0);
|
||||
}
|
||||
|
||||
|
||||
static void object_find_test (int sock)
|
||||
{
|
||||
char response[100];
|
||||
confdb_handle_t handle;
|
||||
int result;
|
||||
hdb_handle_t totem_handle;
|
||||
char key_value[256];
|
||||
size_t value_len;
|
||||
|
||||
snprintf (response, 100, "%s", FAIL_STR);
|
||||
|
||||
result = confdb_initialize (&handle, &callbacks);
|
||||
if (result != CS_OK) {
|
||||
syslog (LOG_ERR, "Could not initialize confdb error %d\n", result);
|
||||
goto send_response;
|
||||
}
|
||||
|
||||
/* Find "totem" and dump bits of it again, to test the direct APIs */
|
||||
result = confdb_object_find_start(handle, OBJECT_PARENT_HANDLE);
|
||||
if (result != CS_OK) {
|
||||
syslog (LOG_ERR, "Could not start object_find %d\n", result);
|
||||
goto send_response;
|
||||
}
|
||||
|
||||
result = confdb_object_find(handle, OBJECT_PARENT_HANDLE, "totem", strlen("totem"), &totem_handle);
|
||||
if (result != CS_OK) {
|
||||
syslog (LOG_ERR, "Could not object_find \"totem\": %d\n", result);
|
||||
goto send_response;
|
||||
}
|
||||
|
||||
result = confdb_key_get(handle, totem_handle, "version", strlen("version"), key_value, &value_len);
|
||||
if (result != CS_OK) {
|
||||
syslog (LOG_ERR, "Could not get \"version\" key: %d\n", result);
|
||||
goto send_response;
|
||||
}
|
||||
|
||||
result = confdb_object_find_destroy (handle, OBJECT_PARENT_HANDLE);
|
||||
if (result != CS_OK) {
|
||||
syslog (LOG_ERR, "Could not destroy find object %d\n", result);
|
||||
goto send_response;
|
||||
}
|
||||
|
||||
|
||||
snprintf (response, 100, "%s", OK_STR);
|
||||
|
||||
send_response:
|
||||
confdb_finalize (handle);
|
||||
send (sock, response, strlen (response) + 1, 0);
|
||||
}
|
||||
|
||||
static void notification_test (int sock)
|
||||
{
|
||||
char response[100];
|
||||
confdb_handle_t handle;
|
||||
int res;
|
||||
hdb_handle_t object_handle;
|
||||
hdb_handle_t new_object_handle;
|
||||
uint16_t incdec_value;
|
||||
uint32_t incdec_value_out;
|
||||
|
||||
snprintf (response, 100, "%s", FAIL_STR);
|
||||
|
||||
res = confdb_initialize (&handle, &valid_callbacks);
|
||||
if (res != CS_OK) {
|
||||
syslog (LOG_ERR, "Could not initialize confdb error %d\n", res);
|
||||
goto send_response;
|
||||
}
|
||||
|
||||
/* Add a base scratch object (we don't want to track the parent object) */
|
||||
res = confdb_object_create(handle, OBJECT_PARENT_HANDLE,
|
||||
"testconfdb", strlen("testconfdb"), &object_handle);
|
||||
if (res != CS_OK) {
|
||||
syslog (LOG_ERR, "error creating 'testconfdb' object: %d", res);
|
||||
goto send_response;
|
||||
}
|
||||
|
||||
res = confdb_track_changes (handle, object_handle, 1 /*OBJECT_TRACK_DEPTH_RECURSIVE*/);
|
||||
if (res != CS_OK) {
|
||||
syslog (LOG_ERR, "can't track changes on object: %d", res);
|
||||
goto send_response;
|
||||
}
|
||||
|
||||
/* Test 'object created' notification
|
||||
*/
|
||||
callback_type = NTF_NONE;
|
||||
|
||||
res = confdb_object_create(handle, object_handle,
|
||||
"duck", strlen("duck"), &new_object_handle);
|
||||
if (res != CS_OK) {
|
||||
syslog (LOG_ERR, "error creating 'duck' object: %d", res);
|
||||
goto send_response;
|
||||
}
|
||||
|
||||
confdb_dispatch (handle, CS_DISPATCH_ALL);
|
||||
|
||||
if (callback_type != NTF_OBJECT_CREATED) {
|
||||
syslog (LOG_ERR, "no notification received for the creation of 'duck'");
|
||||
goto send_response;
|
||||
}
|
||||
if (strcmp ("duck", ntf_object_name) != 0) {
|
||||
syslog (LOG_ERR, "expected notification for 'duck' but got %s", ntf_object_name);
|
||||
goto send_response;
|
||||
}
|
||||
|
||||
/* Test 'key created' notification
|
||||
*/
|
||||
callback_type = NTF_NONE;
|
||||
|
||||
incdec_value = INCDEC_VALUE;
|
||||
res = confdb_key_create_typed (handle, new_object_handle, "incdec",
|
||||
&incdec_value, sizeof(incdec_value), CONFDB_VALUETYPE_UINT16);
|
||||
if (res != CS_OK) {
|
||||
syslog (LOG_ERR, "error creating 'testconfdb' key 4: %d\n", res);
|
||||
goto send_response;
|
||||
}
|
||||
|
||||
confdb_dispatch (handle, CS_DISPATCH_ALL);
|
||||
|
||||
if (callback_type != NTF_KEY_CREATED) {
|
||||
syslog (LOG_ERR, "no notification received for the creation of key 'incdec'");
|
||||
goto send_response;
|
||||
}
|
||||
if (strcmp ("incdec", ntf_key_name) != 0) {
|
||||
syslog (LOG_ERR, "expected notification for 'incdec' but got %s", ntf_key_name);
|
||||
goto send_response;
|
||||
}
|
||||
|
||||
/* Test 'key replaced' notification
|
||||
*/
|
||||
callback_type = NTF_NONE;
|
||||
|
||||
res = confdb_key_increment(handle, new_object_handle, "incdec", strlen("incdec"), &incdec_value_out);
|
||||
if (res != CS_OK) {
|
||||
syslog (LOG_ERR, "error incrementing 'testconfdb' key 4: %d\n", res);
|
||||
goto send_response;
|
||||
}
|
||||
|
||||
confdb_dispatch (handle, CS_DISPATCH_ALL);
|
||||
|
||||
if (callback_type != NTF_KEY_REPLACED) {
|
||||
syslog (LOG_ERR, "no notification received for the incrementing of key 'incdec'");
|
||||
goto send_response;
|
||||
}
|
||||
if (strcmp ("incdec", ntf_key_name) != 0) {
|
||||
syslog (LOG_ERR, "expected notification for 'incdec' but got %s", ntf_key_name);
|
||||
goto send_response;
|
||||
}
|
||||
|
||||
/* Test 'key destroyed' notification
|
||||
*/
|
||||
callback_type = NTF_NONE;
|
||||
|
||||
res = confdb_key_delete (handle, new_object_handle,
|
||||
"incdec", strlen ("incdec"), ntf_key_value, ntf_key_value_len);
|
||||
if (res != CS_OK) {
|
||||
syslog (LOG_ERR, "Could not delete \"incdec\" key: %d", res);
|
||||
goto send_response;
|
||||
}
|
||||
|
||||
confdb_dispatch (handle, CS_DISPATCH_ALL);
|
||||
|
||||
if (callback_type != NTF_KEY_DELETED) {
|
||||
syslog (LOG_ERR, "no notification received for the deletion of key 'incdec'");
|
||||
goto send_response;
|
||||
}
|
||||
if (strcmp ("incdec", ntf_key_name) != 0) {
|
||||
syslog (LOG_ERR, "expected notification for 'incdec' but got %s", ntf_key_name);
|
||||
goto send_response;
|
||||
}
|
||||
|
||||
/* Test 'object destroyed' notification
|
||||
*/
|
||||
callback_type = NTF_NONE;
|
||||
|
||||
res = confdb_object_destroy(handle, new_object_handle);
|
||||
if (res != CS_OK) {
|
||||
syslog (LOG_ERR, "error destroying 'testconfdb' object: %d", res);
|
||||
goto send_response;
|
||||
}
|
||||
|
||||
confdb_dispatch (handle, CS_DISPATCH_ALL);
|
||||
|
||||
if (callback_type != NTF_OBJECT_DELETED) {
|
||||
syslog (LOG_ERR, "no notification received for the deletion of 'duck'");
|
||||
goto send_response;
|
||||
}
|
||||
if (strcmp ("duck", ntf_object_name) != 0) {
|
||||
syslog (LOG_ERR, "expected notification for 'duck' but got %s", ntf_object_name);
|
||||
goto send_response;
|
||||
}
|
||||
confdb_stop_track_changes (handle);
|
||||
confdb_object_destroy(handle, object_handle);
|
||||
|
||||
snprintf (response, 100, "%s", OK_STR);
|
||||
|
||||
send_response:
|
||||
send (sock, response, strlen (response) + 1, 0);
|
||||
confdb_finalize (handle);
|
||||
}
|
||||
|
||||
|
||||
|
||||
static void do_command (int sock, char* func, char*args[], int num_args)
|
||||
{
|
||||
char response[100];
|
||||
|
||||
if (parse_debug)
|
||||
syslog (LOG_DEBUG,"RPC:%s() called.", func);
|
||||
|
||||
if (strcmp ("set_get_test", func) == 0) {
|
||||
set_get_test (sock);
|
||||
} else if (strcmp ("increment_decrement_test", func) == 0) {
|
||||
increment_decrement_test (sock);
|
||||
} else if (strcmp ("object_find_test", func) == 0) {
|
||||
object_find_test (sock);
|
||||
} else if (strcmp ("notification_test", func) == 0) {
|
||||
notification_test (sock);
|
||||
} else {
|
||||
syslog (LOG_ERR,"%s RPC:%s not supported!", __func__, func);
|
||||
snprintf (response, 100, "%s", NOT_SUPPORTED_STR);
|
||||
send (sock, response, strlen (response) + 1, 0);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
int main (int argc, char *argv[])
|
||||
{
|
||||
int ret;
|
||||
|
||||
openlog (NULL, LOG_CONS|LOG_PID, LOG_DAEMON);
|
||||
syslog (LOG_ERR, "confdb_test_agent STARTING");
|
||||
|
||||
parse_debug = 1;
|
||||
ret = test_agent_run (9035, do_command);
|
||||
syslog (LOG_ERR, "confdb_test_agent EXITING");
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user