From: Luiz Augusto von Dentz <luiz.von.dentz@xxxxxxxxx> --- unit/test-uhid.c | 210 +++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 210 insertions(+) diff --git a/unit/test-uhid.c b/unit/test-uhid.c index e0fb411..e4b53e6 100644 --- a/unit/test-uhid.c +++ b/unit/test-uhid.c @@ -25,13 +25,223 @@ #include <config.h> #endif +#include <unistd.h> +#include <stdlib.h> +#include <stdbool.h> +#include <inttypes.h> +#include <string.h> +#include <fcntl.h> +#include <sys/socket.h> + #include <glib.h> #include "src/shared/uhid.h" +#include "src/shared/util.h" + +struct test_pdu { + bool valid; + const uint8_t *data; + size_t size; +}; + +struct test_data { + char *test_name; + struct test_pdu *pdu_list; +}; + +struct context { + GMainLoop *main_loop; + struct bt_uhid *uhid; + guint source; + guint process; + int fd; + unsigned int pdu_offset; + const struct test_data *data; +}; + +#define event(args...) \ + { \ + .valid = true, \ + .data = (void *) args, \ + .size = sizeof(*args), \ + } + +#define define_test(name, function, args...) \ + do { \ + const struct test_pdu pdus[] = { \ + args, { } \ + }; \ + static struct test_data data; \ + data.test_name = g_strdup(name); \ + data.pdu_list = g_malloc(sizeof(pdus)); \ + memcpy(data.pdu_list, pdus, sizeof(pdus)); \ + g_test_add_data_func(name, &data, function); \ + } while (0) + +static void test_debug(const char *str, void *user_data) +{ + const char *prefix = user_data; + + g_print("%s%s\n", prefix, str); +} + +static void test_free(gconstpointer user_data) +{ + const struct test_data *data = user_data; + + g_free(data->test_name); + g_free(data->pdu_list); +} + +static gboolean context_quit(gpointer user_data) +{ + struct context *context = user_data; + + if (context->process > 0) + g_source_remove(context->process); + + g_main_loop_quit(context->main_loop); + + return FALSE; +} + +static gboolean send_pdu(gpointer user_data) +{ + struct context *context = user_data; + const struct test_pdu *pdu; + ssize_t len; + + pdu = &context->data->pdu_list[context->pdu_offset++]; + + len = write(context->fd, pdu->data, pdu->size); + + if (g_test_verbose()) + util_hexdump('<', pdu->data, len, test_debug, "uHID: "); + + g_assert_cmpint(len, ==, pdu->size); + + context->process = 0; + return FALSE; +} + +static void context_process(struct context *context) +{ + if (!context->data->pdu_list[context->pdu_offset].valid) { + context_quit(context); + return; + } + + context->process = g_idle_add(send_pdu, context); +} + +static gboolean test_handler(GIOChannel *channel, GIOCondition cond, + gpointer user_data) +{ + struct context *context = user_data; + const struct test_pdu *pdu; + unsigned char buf[sizeof(struct uhid_event)]; + ssize_t len; + int fd; + + pdu = &context->data->pdu_list[context->pdu_offset++]; + + if (cond & (G_IO_NVAL | G_IO_ERR | G_IO_HUP)) { + context->source = 0; + g_print("%s: cond %x\n", __func__, cond); + return FALSE; + } + + fd = g_io_channel_unix_get_fd(channel); + + len = read(fd, buf, sizeof(buf)); + + g_assert(len > 0); + + if (g_test_verbose()) + util_hexdump('>', buf, len, test_debug, "uHID: "); + + g_assert_cmpint(len, ==, pdu->size); + + g_assert(memcmp(buf, pdu->data, pdu->size) == 0); + + context_process(context); + + return TRUE; +} + +static struct context *create_context(gconstpointer data) +{ + struct context *context = g_new0(struct context, 1); + GIOChannel *channel; + int err, sv[2]; + + context->main_loop = g_main_loop_new(NULL, FALSE); + g_assert(context->main_loop); + + err = socketpair(AF_UNIX, SOCK_SEQPACKET | SOCK_CLOEXEC, 0, sv); + g_assert(err == 0); + + context->uhid = bt_uhid_new(sv[0]); + g_assert(context->uhid != NULL); + + channel = g_io_channel_unix_new(sv[1]); + + g_io_channel_set_close_on_unref(channel, TRUE); + g_io_channel_set_encoding(channel, NULL, NULL); + g_io_channel_set_buffered(channel, FALSE); + + context->source = g_io_add_watch(channel, + G_IO_IN | G_IO_HUP | G_IO_ERR | G_IO_NVAL, + test_handler, context); + g_assert(context->source > 0); + + g_io_channel_unref(channel); + + context->fd = sv[1]; + context->data = data; + + return context; +} + +static void destroy_context(struct context *context) +{ + if (context->source > 0) + g_source_remove(context->source); + + bt_uhid_unref(context->uhid); + + g_main_loop_unref(context->main_loop); + + test_free(context->data); + g_free(context); +} + +static void execute_context(struct context *context) +{ + g_main_loop_run(context->main_loop); + + destroy_context(context); +} + +static const struct uhid_event ev_create = { + .type = UHID_CREATE, +}; + +static void test_client(gconstpointer data) +{ + struct context *context = create_context(data); + + if (g_str_equal(context->data->test_name, "/uhid/command/create")) + bt_uhid_send(context->uhid, &ev_create); + + execute_context(context); +} int main(int argc, char *argv[]) { g_test_init(&argc, &argv, NULL); + define_test("/uhid/command/create", test_client, event(&ev_create)); + return g_test_run(); } -- 1.9.0 -- To unsubscribe from this list: send the line "unsubscribe linux-bluetooth" in the body of a message to majordomo@xxxxxxxxxxxxxxx More majordomo info at http://vger.kernel.org/majordomo-info.html