129 lines · c
1// SPDX-License-Identifier: GPL-2.02/*3 * CrOS Kunit tests utilities.4 */5 6#include <kunit/test.h>7 8#include <linux/list.h>9#include <linux/minmax.h>10#include <linux/platform_data/cros_ec_commands.h>11#include <linux/platform_data/cros_ec_proto.h>12 13#include "cros_ec.h"14#include "cros_ec_proto_test_util.h"15 16int cros_kunit_ec_xfer_mock_default_result;17int cros_kunit_ec_xfer_mock_default_ret;18int cros_kunit_ec_cmd_xfer_mock_called;19int cros_kunit_ec_pkt_xfer_mock_called;20 21static struct list_head cros_kunit_ec_xfer_mock_in;22static struct list_head cros_kunit_ec_xfer_mock_out;23 24int cros_kunit_ec_xfer_mock(struct cros_ec_device *ec_dev, struct cros_ec_command *msg)25{26 struct ec_xfer_mock *mock;27 28 mock = list_first_entry_or_null(&cros_kunit_ec_xfer_mock_in, struct ec_xfer_mock, list);29 if (!mock) {30 msg->result = cros_kunit_ec_xfer_mock_default_result;31 return cros_kunit_ec_xfer_mock_default_ret;32 }33 34 list_del(&mock->list);35 36 memcpy(&mock->msg, msg, sizeof(*msg));37 if (msg->outsize) {38 mock->i_data = kunit_kzalloc(mock->test, msg->outsize, GFP_KERNEL);39 if (mock->i_data)40 memcpy(mock->i_data, msg->data, msg->outsize);41 }42 43 msg->result = mock->result;44 if (msg->insize)45 memcpy(msg->data, mock->o_data, min(msg->insize, mock->o_data_len));46 47 list_add_tail(&mock->list, &cros_kunit_ec_xfer_mock_out);48 49 return mock->ret;50}51 52int cros_kunit_ec_cmd_xfer_mock(struct cros_ec_device *ec_dev, struct cros_ec_command *msg)53{54 ++cros_kunit_ec_cmd_xfer_mock_called;55 return cros_kunit_ec_xfer_mock(ec_dev, msg);56}57 58int cros_kunit_ec_pkt_xfer_mock(struct cros_ec_device *ec_dev, struct cros_ec_command *msg)59{60 ++cros_kunit_ec_pkt_xfer_mock_called;61 return cros_kunit_ec_xfer_mock(ec_dev, msg);62}63 64struct ec_xfer_mock *cros_kunit_ec_xfer_mock_add(struct kunit *test, size_t size)65{66 return cros_kunit_ec_xfer_mock_addx(test, size, EC_RES_SUCCESS, size);67}68 69struct ec_xfer_mock *cros_kunit_ec_xfer_mock_addx(struct kunit *test,70 int ret, int result, size_t size)71{72 struct ec_xfer_mock *mock;73 74 mock = kunit_kzalloc(test, sizeof(*mock), GFP_KERNEL);75 if (!mock)76 return NULL;77 78 list_add_tail(&mock->list, &cros_kunit_ec_xfer_mock_in);79 mock->test = test;80 81 mock->ret = ret;82 mock->result = result;83 mock->o_data = kunit_kzalloc(test, size, GFP_KERNEL);84 if (!mock->o_data)85 return NULL;86 mock->o_data_len = size;87 88 return mock;89}90 91struct ec_xfer_mock *cros_kunit_ec_xfer_mock_next(void)92{93 struct ec_xfer_mock *mock;94 95 mock = list_first_entry_or_null(&cros_kunit_ec_xfer_mock_out, struct ec_xfer_mock, list);96 if (mock)97 list_del(&mock->list);98 99 return mock;100}101 102int cros_kunit_readmem_mock_offset;103u8 *cros_kunit_readmem_mock_data;104int cros_kunit_readmem_mock_ret;105 106int cros_kunit_readmem_mock(struct cros_ec_device *ec_dev, unsigned int offset,107 unsigned int bytes, void *dest)108{109 cros_kunit_readmem_mock_offset = offset;110 111 memcpy(dest, cros_kunit_readmem_mock_data, bytes);112 113 return cros_kunit_readmem_mock_ret;114}115 116void cros_kunit_mock_reset(void)117{118 cros_kunit_ec_xfer_mock_default_result = 0;119 cros_kunit_ec_xfer_mock_default_ret = 0;120 cros_kunit_ec_cmd_xfer_mock_called = 0;121 cros_kunit_ec_pkt_xfer_mock_called = 0;122 INIT_LIST_HEAD(&cros_kunit_ec_xfer_mock_in);123 INIT_LIST_HEAD(&cros_kunit_ec_xfer_mock_out);124 125 cros_kunit_readmem_mock_offset = 0;126 cros_kunit_readmem_mock_data = NULL;127 cros_kunit_readmem_mock_ret = 0;128}129