1// SPDX-License-Identifier: GPL-2.0
  2/*
  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}