X-Git-Url: https://git.gag.com/?a=blobdiff_plain;f=common-src%2Fipc-binary-test.c;fp=common-src%2Fipc-binary-test.c;h=11945c33ed73d7c85bf212748f3c1b430cff51bc;hb=fd48f3e498442f0cbff5f3606c7c403d0566150e;hp=0000000000000000000000000000000000000000;hpb=96f35b20267e8b1a1c846d476f27fcd330e0b018;p=debian%2Famanda

diff --git a/common-src/ipc-binary-test.c b/common-src/ipc-binary-test.c
new file mode 100644
index 0000000..11945c3
--- /dev/null
+++ b/common-src/ipc-binary-test.c
@@ -0,0 +1,203 @@
+/*
+ * Amanda, The Advanced Maryland Automatic Network Disk Archiver
+ * Copyright (c) 1991-1998 University of Maryland at College Park
+ * All Rights Reserved.
+ *
+ * Permission to use, copy, modify, distribute, and sell this software and its
+ * documentation for any purpose is hereby granted without fee, provided that
+ * the above copyright notice appear in all copies and that both that
+ * copyright notice and this permission notice appear in supporting
+ * documentation, and that the name of U.M. not be used in advertising or
+ * publicity pertaining to distribution of the software without specific,
+ * written prior permission.  U.M. makes no representations about the
+ * suitability of this software for any purpose.  It is provided "as is"
+ * without express or implied warranty.
+ *
+ * U.M. DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE, INCLUDING ALL
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS, IN NO EVENT SHALL U.M.
+ * BE LIABLE FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+ * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION
+ * OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
+ * CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+ *
+ * Authors: the Amanda Development Team.  Its members are listed in a
+ * file named AUTHORS, in the root directory of this distribution.
+ */
+
+#include "amanda.h"
+#include "testutils.h"
+#include "glib-util.h"
+#include "ipc-binary.h"
+
+enum {
+    MY_PROTO_CMD1 = 1,
+    MY_PROTO_CMD2 = 2,
+};
+
+enum {
+    MY_PROTO_HOSTNAME = 1,
+    MY_PROTO_DISK = 2,
+    MY_PROTO_DATA = 3,
+};
+
+struct proto_and_fd {
+    ipc_binary_proto_t *proto;
+    int fd;
+};
+
+/*
+ * Test a basic synchronous conversation between two endpoints.  Because
+ * sync is implemented in terms of the async functions, this tests all of
+ * the module's functionality.
+ */
+static gpointer
+test_sync_child(gpointer d)
+{
+    struct proto_and_fd *data = (struct proto_and_fd *)d;
+    ipc_binary_proto_t *proto = data->proto;
+    ipc_binary_channel_t *chan;
+    ipc_binary_message_t *msg;
+    int fd = data->fd;
+
+    g_usleep(G_USEC_PER_SEC / 8);
+
+    chan = ipc_binary_new_channel(proto);
+    msg = ipc_binary_new_message(chan, MY_PROTO_CMD1);
+    ipc_binary_add_arg(msg, MY_PROTO_HOSTNAME, 0, "localhost", 0);
+    if (ipc_binary_write_message(chan, fd, msg) < 0) return NULL;
+
+    g_usleep(G_USEC_PER_SEC / 8);
+
+    msg = ipc_binary_new_message(chan, MY_PROTO_CMD1);
+    ipc_binary_add_arg(msg, MY_PROTO_HOSTNAME, 0, "otherhost", 0);
+    ipc_binary_add_arg(msg, MY_PROTO_DISK, 0, "/usr", 0);
+    if (ipc_binary_write_message(chan, fd, msg) < 0) return NULL;
+
+    msg = ipc_binary_new_message(chan, MY_PROTO_CMD2);
+    ipc_binary_add_arg(msg, MY_PROTO_DATA, 9, "some-data", 0);
+    if (ipc_binary_write_message(chan, fd, msg) < 0) return NULL;
+
+    return GINT_TO_POINTER(1);
+}
+
+static int
+test_sync_parent(ipc_binary_proto_t *proto, int fd)
+{
+    ipc_binary_channel_t *chan;
+    ipc_binary_message_t *msg;
+
+    chan = ipc_binary_new_channel(proto);
+    tu_dbg("parent: created channel\n");
+
+    msg = ipc_binary_read_message(chan, fd);
+    tu_dbg("parent: read message 1\n");
+    if (msg->cmd_id != MY_PROTO_CMD1) {
+	tu_dbg("got bad cmd_id %d\n", (int)msg->cmd_id);
+	return 0;
+    }
+    if (msg->args[MY_PROTO_HOSTNAME].data == NULL) {
+	tu_dbg("got NULL hostname\n");
+	return 0;
+    }
+    if (0 != strcmp((gchar *)msg->args[MY_PROTO_HOSTNAME].data, "localhost")) {
+	tu_dbg("got bad hostname %s\n", (gchar *)msg->args[MY_PROTO_HOSTNAME].data);
+	return 0;
+    }
+    if (msg->args[MY_PROTO_DISK].data != NULL) {
+	tu_dbg("got non-NULL disk\n");
+	return 0;
+    }
+    ipc_binary_free_message(msg);
+
+    msg = ipc_binary_read_message(chan, fd);
+    tu_dbg("parent: read message 2\n");
+    if (msg->cmd_id != MY_PROTO_CMD1) {
+	tu_dbg("got bad cmd_id %d\n", (int)msg->cmd_id);
+	return 0;
+    }
+    if (msg->args[MY_PROTO_HOSTNAME].data == NULL) {
+	tu_dbg("got NULL hostname\n");
+	return 0;
+    }
+    if (0 != strcmp((gchar *)msg->args[MY_PROTO_HOSTNAME].data, "otherhost")) {
+	tu_dbg("got bad hostname %s\n", (gchar *)msg->args[MY_PROTO_HOSTNAME].data);
+	return 0;
+    }
+    if (msg->args[MY_PROTO_DISK].data == NULL) {
+	tu_dbg("got NULL disk\n");
+	return 0;
+    }
+    if (0 != strcmp((gchar *)msg->args[MY_PROTO_DISK].data, "/usr")) {
+	tu_dbg("got bad disk %s\n", (gchar *)msg->args[MY_PROTO_DISK].data);
+	return 0;
+    }
+    ipc_binary_free_message(msg);
+
+    g_usleep(G_USEC_PER_SEC / 8);
+
+    msg = ipc_binary_read_message(chan, fd);
+    tu_dbg("parent: read message 3\n");
+    if (msg->cmd_id != MY_PROTO_CMD2) {
+	tu_dbg("got bad cmd_id %d\n", (int)msg->cmd_id);
+	return 0;
+    }
+    if (msg->args[MY_PROTO_DATA].data == NULL) {
+	tu_dbg("got NULL data\n");
+	return 0;
+    }
+    if (msg->args[MY_PROTO_DATA].len != 9) {
+	tu_dbg("got data length %d, expected 9\n", (int)msg->args[MY_PROTO_DATA].len);
+	return 0;
+    }
+    if (0 != strncmp((gchar *)msg->args[MY_PROTO_DATA].data, "some-data", 9)) {
+	tu_dbg("got bad data\n");
+	return 0;
+    }
+    ipc_binary_free_message(msg);
+
+    return 1;
+}
+
+static int
+test_sync(void)
+{
+    int rv;
+    int p[2];
+    ipc_binary_proto_t *proto;
+    ipc_binary_cmd_t *cmd;
+    struct proto_and_fd data;
+    GThread *child;
+
+    if (pipe(p) == -1) {
+	perror("pipe");
+	return 0;
+    }
+
+    proto = ipc_binary_proto_new(0xE10E);
+    cmd = ipc_binary_proto_add_cmd(proto, MY_PROTO_CMD1);
+    ipc_binary_cmd_add_arg(cmd, MY_PROTO_HOSTNAME, IPC_BINARY_STRING);
+    ipc_binary_cmd_add_arg(cmd, MY_PROTO_DISK, IPC_BINARY_STRING|IPC_BINARY_OPTIONAL);
+    cmd = ipc_binary_proto_add_cmd(proto, MY_PROTO_CMD2);
+    ipc_binary_cmd_add_arg(cmd, MY_PROTO_DATA, 0);
+
+    /* start the child thread */
+    data.proto = proto;
+    data.fd = p[1];
+    child = g_thread_create(test_sync_child, &data, TRUE, NULL);
+
+    /* run the parent and collect the results */
+    rv = test_sync_parent(proto, p[0]) && GPOINTER_TO_INT(g_thread_join(child));
+    return rv;
+}
+
+int
+main(int argc, char **argv)
+{
+    static TestUtilsTest tests[] = {
+	TU_TEST(test_sync, 60),
+	TU_END()
+    };
+
+    glib_init();
+    return testutils_run_tests(argc, argv, tests);
+}