blob: f9dcb007c2c60c8dd15fba2a99ef7e5a065bbeb5 [file] [log] [blame] [edit]
/*
*
* 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();
}