blob: b0592d3657a82815212951841fd53ac5b3ee0867 [file] [log] [blame]
// SPDX-License-Identifier: GPL-2.0-or-later
/*
*
* BlueZ - Bluetooth protocol stack for Linux
*
* Copyright (C) 2014 Intel Corporation. All rights reserved.
*
*
*/
#ifdef HAVE_CONFIG_H
#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"
#include "src/shared/tester.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 {
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 = util_memdup(pdus, sizeof(pdus)); \
tester_add(name, &data, NULL, function, NULL); \
} while (0)
static void test_debug(const char *str, void *user_data)
{
const char *prefix = user_data;
tester_debug("%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 void destroy_context(struct context *context)
{
if (context->source > 0)
g_source_remove(context->source);
bt_uhid_unref(context->uhid);
test_free(context->data);
g_free(context);
}
static gboolean context_quit(gpointer user_data)
{
struct context *context = user_data;
if (context == NULL)
return FALSE;
if (context->process > 0)
g_source_remove(context->process);
destroy_context(context);
tester_test_passed();
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 (tester_use_debug())
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 (tester_use_debug())
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];
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 const struct uhid_event ev_create = {
.type = UHID_CREATE,
};
static const struct uhid_event ev_destroy = {
.type = UHID_DESTROY,
};
static const struct uhid_event ev_feature_answer = {
.type = UHID_FEATURE_ANSWER,
};
static const struct uhid_event ev_input = {
.type = UHID_INPUT,
};
static const struct uhid_event ev_output = {
.type = UHID_OUTPUT,
};
static const struct uhid_event ev_feature = {
.type = UHID_FEATURE,
};
static void test_client(gconstpointer data)
{
struct context *context = create_context(data);
int err;
err = bt_uhid_create(context->uhid, "", NULL, NULL, 0, 0, 0, 0,
BT_UHID_NONE, NULL, 0);
if (err < 0)
tester_test_failed();
if (g_str_equal(context->data->test_name, "/uhid/command/destroy")) {
err = bt_uhid_destroy(context->uhid, true);
if (err < 0)
tester_test_failed();
}
if (g_str_equal(context->data->test_name,
"/uhid/command/feature_answer")) {
err = bt_uhid_send(context->uhid, &ev_feature_answer);
if (err < 0)
tester_test_failed();
}
if (g_str_equal(context->data->test_name, "/uhid/command/input")) {
err = bt_uhid_input(context->uhid, 0, NULL, 0);
if (err < 0)
tester_test_failed();
}
context_quit(context);
}
static void handle_output(struct uhid_event *ev, void *user_data)
{
g_assert_cmpint(ev->type, ==, UHID_OUTPUT);
context_quit(user_data);
}
static void handle_feature(struct uhid_event *ev, void *user_data)
{
g_assert_cmpint(ev->type, ==, UHID_FEATURE);
context_quit(user_data);
}
static void test_server(gconstpointer data)
{
struct context *context = create_context(data);
bt_uhid_register(context->uhid, UHID_OUTPUT, handle_output, context);
bt_uhid_register(context->uhid, UHID_FEATURE, handle_feature, context);
g_idle_add(send_pdu, context);
}
int main(int argc, char *argv[])
{
tester_init(&argc, &argv);
define_test("/uhid/command/create", test_client, event(&ev_create));
define_test("/uhid/command/destroy", test_client, event(&ev_destroy));
define_test("/uhid/command/feature_answer", test_client,
event(&ev_feature_answer));
define_test("/uhid/command/input", test_client, event(&ev_input));
define_test("/uhid/event/output", test_server, event(&ev_output));
define_test("/uhid/event/feature", test_server, event(&ev_feature));
return tester_run();
}