/*
 * Amanda, The Advanced Maryland Automatic Network Disk Archiver
 * Copyright (c) 1991-1998 University of Maryland at College Park
 * Copyright (c) 2007-2012 Zmanda, Inc.  All Rights Reserved.
 * Copyright (c) 2013-2016 Carbonite, Inc.  All Rights Reserved.
 * 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 (!g_str_equal((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 (!g_str_equal((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 (!g_str_equal((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 gboolean
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 FALSE;
    }

    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) ? TRUE : FALSE;
}

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);
}