Quelle  hid-uclogic-core-test.c   Sprache: C

// SPDX-License-Identifier: GPL-2.0+

/*
 *  HID driver for UC-Logic devices not fully compliant with HID standard
 *
 *  Copyright (c) 2022 José Expósito <jose.exposito89@gmail.com>
 */

#include <kunit/test.h>
#include "./hid-uclogic-params.h"

#define MAX_EVENT_SIZE 12

struct uclogic_raw_event_hook_test {
 u8 event[MAX_EVENT_SIZE];
 size_t size;
 bool expected;
};

static struct uclogic_raw_event_hook_test hook_events[] = {
 {
  .event = { 0xA1, 0xB2, 0xC3, 0xD4 },
  .size = 4,
 },
 {
  .event = { 0x1F, 0x2E, 0x3D, 0x4C, 0x5B, 0x6A },
  .size = 6,
 },
};

static struct uclogic_raw_event_hook_test test_events[] = {
 {
  .event = { 0xA1, 0xB2, 0xC3, 0xD4 },
  .size = 4,
  .expected = true,
 },
 {
  .event = { 0x1F, 0x2E, 0x3D, 0x4C, 0x5B, 0x6A },
  .size = 6,
  .expected = true,
 },
 {
  .event = { 0xA1, 0xB2, 0xC3 },
  .size = 3,
  .expected = false,
 },
 {
  .event = { 0xA1, 0xB2, 0xC3, 0xD4, 0x00 },
  .size = 5,
  .expected = false,
 },
 {
  .event = { 0x2E, 0x3D, 0x4C, 0x5B, 0x6A, 0x1F },
  .size = 6,
  .expected = false,
 },
};

static void fake_work(struct work_struct *work)
{

}

static void hid_test_uclogic_exec_event_hook_test(struct kunit *test)
{
 struct uclogic_params p = {0, };
 struct uclogic_raw_event_hook *filter;
 bool res;
 int n;

 /* Initialize the list of events to hook */
 p.event_hooks = kunit_kzalloc(test, sizeof(*p.event_hooks), GFP_KERNEL);
 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, p.event_hooks);
 INIT_LIST_HEAD(&p.event_hooks->list);

 for (n = 0; n < ARRAY_SIZE(hook_events); n++) {
  filter = kunit_kzalloc(test, sizeof(*filter), GFP_KERNEL);
  KUNIT_ASSERT_NOT_ERR_OR_NULL(test, filter);

  filter->size = hook_events[n].size;
  filter->event = kunit_kzalloc(test, filter->size, GFP_KERNEL);
  KUNIT_ASSERT_NOT_ERR_OR_NULL(test, filter->event);
  memcpy(filter->event, &hook_events[n].event[0], filter->size);

  INIT_WORK(&filter->work, fake_work);

  list_add_tail(&filter->list, &p.event_hooks->list);
 }

 /* Test uclogic_exec_event_hook() */
 for (n = 0; n < ARRAY_SIZE(test_events); n++) {
  res = uclogic_exec_event_hook(&p, &test_events[n].event[0],
           test_events[n].size);
  KUNIT_ASSERT_EQ(test, res, test_events[n].expected);
 }
}

static struct kunit_case hid_uclogic_core_test_cases[] = {
 KUNIT_CASE(hid_test_uclogic_exec_event_hook_test),
 {}
};

static struct kunit_suite hid_uclogic_core_test_suite = {
 .name = "hid_uclogic_core_test",
 .test_cases = hid_uclogic_core_test_cases,
};

kunit_test_suite(hid_uclogic_core_test_suite);

MODULE_DESCRIPTION("KUnit tests for the UC-Logic driver");
MODULE_LICENSE("GPL");
MODULE_AUTHOR("José Expósito ");