platform/chrome/cros_kunit_util.c

0001 // SPDX-License-Identifier: GPL-2.0
0002 /*
0003  * CrOS Kunit tests utilities.
0004  */
0005
0006 #include <kunit/test.h>
0007
0008 #include <linux/list.h>
0009 #include <linux/minmax.h>
0010 #include <linux/platform_data/cros_ec_commands.h>
0011 #include <linux/platform_data/cros_ec_proto.h>
0012
0013 #include "cros_ec.h"
0014 #include "cros_kunit_util.h"
0015
0016 int cros_kunit_ec_xfer_mock_default_result;
0017 int cros_kunit_ec_xfer_mock_default_ret;
0018 int cros_kunit_ec_cmd_xfer_mock_called;
0019 int cros_kunit_ec_pkt_xfer_mock_called;
0020
0021 static struct list_head cros_kunit_ec_xfer_mock_in;
0022 static struct list_head cros_kunit_ec_xfer_mock_out;
0023
0024 int cros_kunit_ec_xfer_mock(struct cros_ec_device *ec_dev, struct cros_ec_command *msg)
0025 {
0026     struct ec_xfer_mock *mock;
0027
0028     mock = list_first_entry_or_null(&cros_kunit_ec_xfer_mock_in, struct ec_xfer_mock, list);
0029     if (!mock) {
0030         msg->result = cros_kunit_ec_xfer_mock_default_result;
0031         return cros_kunit_ec_xfer_mock_default_ret;
0032     }
0033
0034     list_del(&mock->list);
0035
0036     memcpy(&mock->msg, msg, sizeof(*msg));
0037     if (msg->outsize) {
0038         mock->i_data = kunit_kzalloc(mock->test, msg->outsize, GFP_KERNEL);
0039         if (mock->i_data)
0040             memcpy(mock->i_data, msg->data, msg->outsize);
0041     }
0042
0043     msg->result = mock->result;
0044     if (msg->insize)
0045         memcpy(msg->data, mock->o_data, min(msg->insize, mock->o_data_len));
0046
0047     list_add_tail(&mock->list, &cros_kunit_ec_xfer_mock_out);
0048
0049     return mock->ret;
0050 }
0051
0052 int cros_kunit_ec_cmd_xfer_mock(struct cros_ec_device *ec_dev, struct cros_ec_command *msg)
0053 {
0054     ++cros_kunit_ec_cmd_xfer_mock_called;
0055     return cros_kunit_ec_xfer_mock(ec_dev, msg);
0056 }
0057
0058 int cros_kunit_ec_pkt_xfer_mock(struct cros_ec_device *ec_dev, struct cros_ec_command *msg)
0059 {
0060     ++cros_kunit_ec_pkt_xfer_mock_called;
0061     return cros_kunit_ec_xfer_mock(ec_dev, msg);
0062 }
0063
0064 struct ec_xfer_mock *cros_kunit_ec_xfer_mock_add(struct kunit *test, size_t size)
0065 {
0066     return cros_kunit_ec_xfer_mock_addx(test, size, EC_RES_SUCCESS, size);
0067 }
0068
0069 struct ec_xfer_mock *cros_kunit_ec_xfer_mock_addx(struct kunit *test,
0070                           int ret, int result, size_t size)
0071 {
0072     struct ec_xfer_mock *mock;
0073
0074     mock = kunit_kzalloc(test, sizeof(*mock), GFP_KERNEL);
0075     if (!mock)
0076         return NULL;
0077
0078     list_add_tail(&mock->list, &cros_kunit_ec_xfer_mock_in);
0079     mock->test = test;
0080
0081     mock->ret = ret;
0082     mock->result = result;
0083     mock->o_data = kunit_kzalloc(test, size, GFP_KERNEL);
0084     if (!mock->o_data)
0085         return NULL;
0086     mock->o_data_len = size;
0087
0088     return mock;
0089 }
0090
0091 struct ec_xfer_mock *cros_kunit_ec_xfer_mock_next(void)
0092 {
0093     struct ec_xfer_mock *mock;
0094
0095     mock = list_first_entry_or_null(&cros_kunit_ec_xfer_mock_out, struct ec_xfer_mock, list);
0096     if (mock)
0097         list_del(&mock->list);
0098
0099     return mock;
0100 }
0101
0102 int cros_kunit_readmem_mock_offset;
0103 u8 *cros_kunit_readmem_mock_data;
0104 int cros_kunit_readmem_mock_ret;
0105
0106 int cros_kunit_readmem_mock(struct cros_ec_device *ec_dev, unsigned int offset,
0107                 unsigned int bytes, void *dest)
0108 {
0109     cros_kunit_readmem_mock_offset = offset;
0110
0111     memcpy(dest, cros_kunit_readmem_mock_data, bytes);
0112
0113     return cros_kunit_readmem_mock_ret;
0114 }
0115
0116 void cros_kunit_mock_reset(void)
0117 {
0118     cros_kunit_ec_xfer_mock_default_result = 0;
0119     cros_kunit_ec_xfer_mock_default_ret = 0;
0120     cros_kunit_ec_cmd_xfer_mock_called = 0;
0121     cros_kunit_ec_pkt_xfer_mock_called = 0;
0122     INIT_LIST_HEAD(&cros_kunit_ec_xfer_mock_in);
0123     INIT_LIST_HEAD(&cros_kunit_ec_xfer_mock_out);
0124
0125     cros_kunit_readmem_mock_offset = 0;
0126     cros_kunit_readmem_mock_data = NULL;
0127     cros_kunit_readmem_mock_ret = 0;
0128 }
0129
0130 MODULE_LICENSE("GPL");