diff options
Diffstat (limited to 'unit/test-mgmt.c')
-rw-r--r-- | unit/test-mgmt.c | 243 |
1 files changed, 243 insertions, 0 deletions
diff --git a/unit/test-mgmt.c b/unit/test-mgmt.c new file mode 100644 index 00000000..f9dcb007 --- /dev/null +++ b/unit/test-mgmt.c @@ -0,0 +1,243 @@ +/* + * + * BlueZ - Bluetooth protocol stack for Linux + * + * Copyright (C) 2012 Intel Corporation. All rights reserved. + * + * + * 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. + * + * 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. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA + * + */ + +#ifdef HAVE_CONFIG_H +#include <config.h> +#endif + +#include <stdbool.h> +#include <unistd.h> +#include <sys/socket.h> + +#include <glib.h> + +#include "lib/bluetooth.h" +#include "lib/mgmt.h" + +#include "src/shared/mgmt.h" + +struct context { + GMainLoop *main_loop; + struct mgmt *mgmt_client; + guint server_source; + GList *handler_list; +}; + +enum action { + ACTION_PASSED, + ACTION_IGNORE, +}; + +struct handler { + const void *cmd_data; + uint16_t cmd_size; + bool match_prefix; + enum action action; +}; + +static void mgmt_debug(const char *str, void *user_data) +{ + const char *prefix = user_data; + + g_print("%s%s\n", prefix, str); +} + +static void context_quit(struct context *context) +{ + g_main_loop_quit(context->main_loop); +} + +static void check_actions(struct context *context, + const void *data, uint16_t size) +{ + GList *list; + + for (list = g_list_first(context->handler_list); list; + list = g_list_next(list)) { + struct handler *handler = list->data; + + if (handler->match_prefix) { + if (size < handler->cmd_size) + continue; + } else { + if (size != handler->cmd_size) + continue; + } + + if (memcmp(data, handler->cmd_data, handler->cmd_size)) + continue; + + switch (handler->action) { + case ACTION_PASSED: + context_quit(context); + return; + case ACTION_IGNORE: + return; + } + } + + g_test_message("Command not handled\n"); + g_assert_not_reached(); +} + +static gboolean server_handler(GIOChannel *channel, GIOCondition cond, + gpointer user_data) +{ + struct context *context = user_data; + unsigned char buf[512]; + ssize_t result; + int fd; + + if (cond & (G_IO_NVAL | G_IO_ERR | G_IO_HUP)) + return FALSE; + + fd = g_io_channel_unix_get_fd(channel); + + result = read(fd, buf, sizeof(buf)); + if (result < 0) + return FALSE; + + check_actions(context, buf, result); + + return TRUE; +} + +static struct context *create_context(void) +{ + 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); + + channel = g_io_channel_unix_new(sv[0]); + + 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->server_source = g_io_add_watch(channel, + G_IO_IN | G_IO_HUP | G_IO_ERR | G_IO_NVAL, + server_handler, context); + g_assert(context->server_source > 0); + + g_io_channel_unref(channel); + + context->mgmt_client = mgmt_new(sv[1]); + g_assert(context->mgmt_client); + + if (g_test_verbose() == TRUE) + mgmt_set_debug(context->mgmt_client, + mgmt_debug, "mgmt: ", NULL); + + mgmt_set_close_on_unref(context->mgmt_client, true); + + return context; +} + +static void execute_context(struct context *context) +{ + g_main_loop_run(context->main_loop); + + g_list_free_full(context->handler_list, g_free); + + g_source_remove(context->server_source); + + mgmt_unref(context->mgmt_client); + + g_main_loop_unref(context->main_loop); + + g_free(context); +} + +static void add_action(struct context *context, + const void *cmd_data, uint16_t cmd_size, + bool match_prefix, enum action action) +{ + struct handler *handler = g_new0(struct handler, 1); + + handler->cmd_data = cmd_data; + handler->cmd_size = cmd_size; + handler->match_prefix = match_prefix; + handler->action = action; + + context->handler_list = g_list_append(context->handler_list, handler); +} + +struct command_test_data { + uint16_t opcode; + uint16_t index; + uint16_t length; + const void *param; + const void *cmd_data; + uint16_t cmd_size; +}; + +static const unsigned char read_version_command[] = + { 0x01, 0x00, 0xff, 0xff, 0x00, 0x00 }; + +static const struct command_test_data command_test_1 = { + .opcode = MGMT_OP_READ_VERSION, + .index = MGMT_INDEX_NONE, + .cmd_data = read_version_command, + .cmd_size = sizeof(read_version_command), +}; + +static const unsigned char read_info_command[] = + { 0x04, 0x00, 0x00, 0x02, 0x00, 0x00 }; + +static const struct command_test_data command_test_2 = { + .opcode = MGMT_OP_READ_INFO, + .index = 512, + .cmd_data = read_info_command, + .cmd_size = sizeof(read_info_command), +}; + +static void test_command(gconstpointer data) +{ + const struct command_test_data *test = data; + struct context *context = create_context(); + + add_action(context, test->cmd_data, test->cmd_size, + false, ACTION_PASSED); + + mgmt_send(context->mgmt_client, test->opcode, test->index, + test->length, test->param, + NULL ,NULL, NULL); + + execute_context(context); +} + +int main(int argc, char *argv[]) +{ + g_test_init(&argc, &argv, NULL); + + g_test_add_data_func("/mgmt/command/1", &command_test_1, test_command); + g_test_add_data_func("/mgmt/command/2", &command_test_2, test_command); + + return g_test_run(); +} |