[J-u-boot.git] / test / dm / spi.c

// SPDX-License-Identifier: GPL-2.0+
/*
 * Copyright (C) 2013 Google, Inc
 */

#include <common.h>
#include <dm.h>
#include <fdtdec.h>
#include <spi.h>
#include <spi_flash.h>
#include <asm/state.h>
#include <asm/test.h>
#include <dm/device-internal.h>
#include <dm/test.h>
#include <dm/uclass-internal.h>
#include <dm/util.h>
#include <test/test.h>
#include <test/ut.h>

/* Test that we can find buses and chip-selects */
static int dm_test_spi_find(struct unit_test_state *uts)
{
	struct sandbox_state *state = state_get_current();
	struct spi_slave *slave;
	struct udevice *bus, *dev;
	const int busnum = 0, cs = 0, mode = 0, speed = 1000000, cs_b = 2;
	struct spi_cs_info info;
	ofnode node;

	/*
	 * The post_bind() method will bind devices to chip selects. Check
	 * this then remove the emulation and the slave device.
	 */
	ut_asserteq(0, uclass_get_device_by_seq(UCLASS_SPI, busnum, &bus));
	ut_assertok(spi_cs_info(bus, cs, &info));
	node = dev_ofnode(info.dev);
	device_remove(info.dev, DM_REMOVE_NORMAL);
	device_unbind(info.dev);

	/*
	 * Even though the device is gone, the sandbox SPI drivers always
	 * reports that CS 0 is present
	 */
	ut_assertok(spi_cs_info(bus, cs, &info));
	ut_asserteq_ptr(NULL, info.dev);

	/* This finds nothing because we removed the device */
	ut_asserteq(-ENODEV, spi_find_bus_and_cs(busnum, cs, &bus, &dev));
	ut_asserteq(-ENODEV, spi_get_bus_and_cs(busnum, cs, &bus, &slave));

	/*
	 * This forces the device to be re-added, but there is no emulation
	 * connected so the probe will fail. We require that bus is left
	 * alone on failure, and that the _spi_get_bus_and_cs() does not add
	 * a 'partially-inited' device.
	 */
	ut_asserteq(-ENODEV, spi_find_bus_and_cs(busnum, cs, &bus, &dev));
	ut_asserteq(-ENOENT, _spi_get_bus_and_cs(busnum, cs, speed, mode,
						 "jedec_spi_nor", "name", &bus,
						 &slave));
	sandbox_sf_unbind_emul(state_get_current(), busnum, cs);
	ut_assertok(spi_cs_info(bus, cs, &info));
	ut_asserteq_ptr(NULL, info.dev);

	/* Add the emulation and try again */
	ut_assertok(sandbox_sf_bind_emul(state, busnum, cs, bus, node,
					 "name"));
	ut_assertok(spi_find_bus_and_cs(busnum, cs, &bus, &dev));
	ut_assertok(_spi_get_bus_and_cs(busnum, cs, speed, mode,
					"jedec_spi_nor", "name", &bus, &slave));

	ut_assertok(spi_cs_info(bus, cs, &info));
	ut_asserteq_ptr(info.dev, slave->dev);

	/* We should be able to add something to another chip select */
	ut_assertok(sandbox_sf_bind_emul(state, busnum, cs_b, bus, node,
					 "name"));
	ut_asserteq(-EINVAL, _spi_get_bus_and_cs(busnum, cs_b, speed, mode,
						 "jedec_spi_nor", "name", &bus,
						 &slave));
	ut_asserteq(-EINVAL, spi_cs_info(bus, cs_b, &info));
	ut_asserteq_ptr(NULL, info.dev);

	/*
	 * Since we are about to destroy all devices, we must tell sandbox
	 * to forget the emulation device
	 */
	sandbox_sf_unbind_emul(state_get_current(), busnum, cs);
	sandbox_sf_unbind_emul(state_get_current(), busnum, cs_b);

	return 0;
}
DM_TEST(dm_test_spi_find, UT_TESTF_SCAN_PDATA | UT_TESTF_SCAN_FDT);

/* dm_test_spi_switch_slaves - Helper function to check whether spi_claim_bus
 *                             operates correctly with two spi slaves.
 *
 * Check that switching back and forth between two slaves claiming the bus
 * will update dm_spi_bus->speed and sandbox_spi bus speed/mode correctly.
 *
 * @uts - unit test state
 * @slave_a - first spi slave used for testing
 * @slave_b - second spi slave used for testing
 */
static int dm_test_spi_switch_slaves(struct unit_test_state *uts,
				     struct spi_slave *slave_a,
				     struct spi_slave *slave_b)
{
	struct udevice *bus;
	struct dm_spi_bus *bus_data;

	/* Check that slaves are on the same bus */
	ut_asserteq_ptr(dev_get_parent(slave_a->dev),
			dev_get_parent(slave_b->dev));

	bus = dev_get_parent(slave_a->dev);
	bus_data = dev_get_uclass_priv(bus);

	ut_assertok(spi_claim_bus(slave_a));
	ut_asserteq(slave_a->max_hz, bus_data->speed);
	ut_asserteq(slave_a->max_hz, sandbox_spi_get_speed(bus));
	ut_asserteq(slave_a->mode, sandbox_spi_get_mode(bus));
	spi_release_bus(slave_a);

	ut_assertok(spi_claim_bus(slave_b));
	ut_asserteq(slave_b->max_hz, bus_data->speed);
	ut_asserteq(slave_b->max_hz, sandbox_spi_get_speed(bus));
	ut_asserteq(slave_b->mode, sandbox_spi_get_mode(bus));
	spi_release_bus(slave_b);

	ut_assertok(spi_claim_bus(slave_a));
	ut_asserteq(slave_a->max_hz, bus_data->speed);
	ut_asserteq(slave_a->max_hz, sandbox_spi_get_speed(bus));
	ut_asserteq(slave_a->mode, sandbox_spi_get_mode(bus));
	spi_release_bus(slave_a);

	return 0;
}

static int dm_test_spi_claim_bus(struct unit_test_state *uts)
{
	struct udevice *bus;
	struct spi_slave *slave_a, *slave_b;
	struct dm_spi_slave_plat *slave_plat;
	const int busnum = 0, cs_a = 0, cs_b = 1;

	/* Get spi slave on CS0 */
	ut_assertok(spi_get_bus_and_cs(busnum, cs_a, &bus, &slave_a));
	/* Get spi slave on CS1 */
	ut_assertok(spi_get_bus_and_cs(busnum, cs_b, &bus, &slave_b));

	/* Different max_hz, different mode. */
	ut_assert(slave_a->max_hz != slave_b->max_hz);
	ut_assert(slave_a->mode != slave_b->mode);
	dm_test_spi_switch_slaves(uts, slave_a, slave_b);

	/* Different max_hz, same mode. */
	slave_a->mode = slave_b->mode;
	dm_test_spi_switch_slaves(uts, slave_a, slave_b);

	/*
	 * Same max_hz, different mode.
	 * Restore original mode for slave_a, from platdata.
	 */
	slave_plat = dev_get_parent_plat(slave_a->dev);
	slave_a->mode = slave_plat->mode;
	slave_a->max_hz = slave_b->max_hz;
	dm_test_spi_switch_slaves(uts, slave_a, slave_b);

	return 0;
}
DM_TEST(dm_test_spi_claim_bus, UT_TESTF_SCAN_PDATA | UT_TESTF_SCAN_FDT);

/* Test that sandbox SPI works correctly */
static int dm_test_spi_xfer(struct unit_test_state *uts)
{
	struct spi_slave *slave;
	struct udevice *bus;
	const int busnum = 0, cs = 0;
	const char dout[5] = {0x9f};
	unsigned char din[5];

	ut_assertok(spi_get_bus_and_cs(busnum, cs, &bus, &slave));
	ut_assertok(spi_claim_bus(slave));
	ut_assertok(spi_xfer(slave, 40, dout, din,
			     SPI_XFER_BEGIN | SPI_XFER_END));
	ut_asserteq(0xff, din[0]);
	ut_asserteq(0x20, din[1]);
	ut_asserteq(0x20, din[2]);
	ut_asserteq(0x15, din[3]);
	spi_release_bus(slave);

	/*
	 * Since we are about to destroy all devices, we must tell sandbox
	 * to forget the emulation device
	 */
#if CONFIG_IS_ENABLED(DM_SPI_FLASH)
	sandbox_sf_unbind_emul(state_get_current(), busnum, cs);
#endif

	return 0;
}
DM_TEST(dm_test_spi_xfer, UT_TESTF_SCAN_PDATA | UT_TESTF_SCAN_FDT);
Commit	Line	Data
83d290c5	1	// SPDX-License-Identifier: GPL-2.0+
ebcab48a SG	2	/*
ebcab48a SG	3	* Copyright (C) 2013 Google, Inc
ebcab48a SG	4	*/
	5
	6	#include <common.h>
	7	#include <dm.h>
	8	#include <fdtdec.h>
	9	#include <spi.h>
	10	#include <spi_flash.h>
e721b882	11	#include <asm/state.h>
a5624c6b	12	#include <asm/test.h>
ebcab48a SG	13	#include <dm/device-internal.h>
	14	#include <dm/test.h>
	15	#include <dm/uclass-internal.h>
ebcab48a	16	#include <dm/util.h>
0e1fad43	17	#include <test/test.h>
e721b882	18	#include <test/ut.h>
ebcab48a SG	19
ebcab48a SG	20	/* Test that we can find buses and chip-selects */
e721b882	21	static int dm_test_spi_find(struct unit_test_state *uts)
ebcab48a SG	22	{
	23	struct sandbox_state *state = state_get_current();
	24	struct spi_slave *slave;
	25	struct udevice bus, dev;
1dc53ce7	26	const int busnum = 0, cs = 0, mode = 0, speed = 1000000, cs_b = 2;
ebcab48a	27	struct spi_cs_info info;
008dcddf	28	ofnode node;
ebcab48a	29
ebcab48a	30	/*
91195485 SG	31	* The post_bind() method will bind devices to chip selects. Check
91195485 SG	32	* this then remove the emulation and the slave device.
ebcab48a SG	33	*/
	34	ut_asserteq(0, uclass_get_device_by_seq(UCLASS_SPI, busnum, &bus));
	35	ut_assertok(spi_cs_info(bus, cs, &info));
008dcddf	36	node = dev_ofnode(info.dev);
706865af	37	device_remove(info.dev, DM_REMOVE_NORMAL);
ebcab48a SG	38	device_unbind(info.dev);
	39
	40	/*
	41	* Even though the device is gone, the sandbox SPI drivers always
	42	* reports that CS 0 is present
	43	*/
	44	ut_assertok(spi_cs_info(bus, cs, &info));
d0cff03e	45	ut_asserteq_ptr(NULL, info.dev);
ebcab48a SG	46
	47	/* This finds nothing because we removed the device */
	48	ut_asserteq(-ENODEV, spi_find_bus_and_cs(busnum, cs, &bus, &dev));
012afa83	49	ut_asserteq(-ENODEV, spi_get_bus_and_cs(busnum, cs, &bus, &slave));
ebcab48a SG	50
	51	/*
	52	* This forces the device to be re-added, but there is no emulation
	53	* connected so the probe will fail. We require that bus is left
61708bb0	54	* alone on failure, and that the _spi_get_bus_and_cs() does not add
ebcab48a SG	55	* a 'partially-inited' device.
	56	*/
	57	ut_asserteq(-ENODEV, spi_find_bus_and_cs(busnum, cs, &bus, &dev));
61708bb0 PC	58	ut_asserteq(-ENOENT, _spi_get_bus_and_cs(busnum, cs, speed, mode,
	59	"jedec_spi_nor", "name", &bus,
	60	&slave));
d0cff03e	61	sandbox_sf_unbind_emul(state_get_current(), busnum, cs);
ebcab48a	62	ut_assertok(spi_cs_info(bus, cs, &info));
d0cff03e	63	ut_asserteq_ptr(NULL, info.dev);
ebcab48a SG	64
ebcab48a SG	65	/* Add the emulation and try again */
008dcddf	66	ut_assertok(sandbox_sf_bind_emul(state, busnum, cs, bus, node,
ebcab48a SG	67	"name"));
ebcab48a SG	68	ut_assertok(spi_find_bus_and_cs(busnum, cs, &bus, &dev));
61708bb0 PC	69	ut_assertok(_spi_get_bus_and_cs(busnum, cs, speed, mode,
61708bb0 PC	70	"jedec_spi_nor", "name", &bus, &slave));
ebcab48a SG	71
	72	ut_assertok(spi_cs_info(bus, cs, &info));
	73	ut_asserteq_ptr(info.dev, slave->dev);
	74
	75	/* We should be able to add something to another chip select */
008dcddf	76	ut_assertok(sandbox_sf_bind_emul(state, busnum, cs_b, bus, node,
ebcab48a	77	"name"));
61708bb0 PC	78	ut_asserteq(-EINVAL, _spi_get_bus_and_cs(busnum, cs_b, speed, mode,
	79	"jedec_spi_nor", "name", &bus,
	80	&slave));
bfcd9298 BM	81	ut_asserteq(-EINVAL, spi_cs_info(bus, cs_b, &info));
bfcd9298 BM	82	ut_asserteq_ptr(NULL, info.dev);
ebcab48a SG	83
	84	/*
	85	* Since we are about to destroy all devices, we must tell sandbox
	86	* to forget the emulation device
	87	*/
	88	sandbox_sf_unbind_emul(state_get_current(), busnum, cs);
	89	sandbox_sf_unbind_emul(state_get_current(), busnum, cs_b);
	90
	91	return 0;
	92	}
e180c2b1	93	DM_TEST(dm_test_spi_find, UT_TESTF_SCAN_PDATA \| UT_TESTF_SCAN_FDT);
ebcab48a	94
a5624c6b OP	95	/* dm_test_spi_switch_slaves - Helper function to check whether spi_claim_bus
	96	* operates correctly with two spi slaves.
	97	*
	98	* Check that switching back and forth between two slaves claiming the bus
	99	* will update dm_spi_bus->speed and sandbox_spi bus speed/mode correctly.
	100	*
	101	* @uts - unit test state
	102	* @slave_a - first spi slave used for testing
	103	* @slave_b - second spi slave used for testing
	104	*/
	105	static int dm_test_spi_switch_slaves(struct unit_test_state *uts,
	106	struct spi_slave *slave_a,
	107	struct spi_slave *slave_b)
	108	{
	109	struct udevice *bus;
	110	struct dm_spi_bus *bus_data;
	111
	112	/* Check that slaves are on the same bus */
	113	ut_asserteq_ptr(dev_get_parent(slave_a->dev),
	114	dev_get_parent(slave_b->dev));
	115
	116	bus = dev_get_parent(slave_a->dev);
	117	bus_data = dev_get_uclass_priv(bus);
	118
	119	ut_assertok(spi_claim_bus(slave_a));
	120	ut_asserteq(slave_a->max_hz, bus_data->speed);
	121	ut_asserteq(slave_a->max_hz, sandbox_spi_get_speed(bus));
	122	ut_asserteq(slave_a->mode, sandbox_spi_get_mode(bus));
	123	spi_release_bus(slave_a);
	124
	125	ut_assertok(spi_claim_bus(slave_b));
	126	ut_asserteq(slave_b->max_hz, bus_data->speed);
	127	ut_asserteq(slave_b->max_hz, sandbox_spi_get_speed(bus));
	128	ut_asserteq(slave_b->mode, sandbox_spi_get_mode(bus));
	129	spi_release_bus(slave_b);
	130
	131	ut_assertok(spi_claim_bus(slave_a));
	132	ut_asserteq(slave_a->max_hz, bus_data->speed);
	133	ut_asserteq(slave_a->max_hz, sandbox_spi_get_speed(bus));
	134	ut_asserteq(slave_a->mode, sandbox_spi_get_mode(bus));
	135	spi_release_bus(slave_a);
	136
	137	return 0;
	138	}
	139
	140	static int dm_test_spi_claim_bus(struct unit_test_state *uts)
	141	{
	142	struct udevice *bus;
	143	struct spi_slave slave_a, slave_b;
	144	struct dm_spi_slave_plat *slave_plat;
012afa83	145	const int busnum = 0, cs_a = 0, cs_b = 1;
a5624c6b OP	146
a5624c6b OP	147	/* Get spi slave on CS0 */
012afa83	148	ut_assertok(spi_get_bus_and_cs(busnum, cs_a, &bus, &slave_a));
a5624c6b	149	/* Get spi slave on CS1 */
012afa83	150	ut_assertok(spi_get_bus_and_cs(busnum, cs_b, &bus, &slave_b));
a5624c6b OP	151
	152	/* Different max_hz, different mode. */
	153	ut_assert(slave_a->max_hz != slave_b->max_hz);
	154	ut_assert(slave_a->mode != slave_b->mode);
	155	dm_test_spi_switch_slaves(uts, slave_a, slave_b);
	156
	157	/* Different max_hz, same mode. */
	158	slave_a->mode = slave_b->mode;
	159	dm_test_spi_switch_slaves(uts, slave_a, slave_b);
	160
	161	/*
	162	* Same max_hz, different mode.
	163	* Restore original mode for slave_a, from platdata.
	164	*/
	165	slave_plat = dev_get_parent_plat(slave_a->dev);
	166	slave_a->mode = slave_plat->mode;
	167	slave_a->max_hz = slave_b->max_hz;
	168	dm_test_spi_switch_slaves(uts, slave_a, slave_b);
	169
	170	return 0;
	171	}
	172	DM_TEST(dm_test_spi_claim_bus, UT_TESTF_SCAN_PDATA \| UT_TESTF_SCAN_FDT);
	173
ebcab48a	174	/* Test that sandbox SPI works correctly */
e721b882	175	static int dm_test_spi_xfer(struct unit_test_state *uts)
ebcab48a SG	176	{
	177	struct spi_slave *slave;
	178	struct udevice *bus;
012afa83	179	const int busnum = 0, cs = 0;
ebcab48a SG	180	const char dout[5] = {0x9f};
	181	unsigned char din[5];
	182
012afa83	183	ut_assertok(spi_get_bus_and_cs(busnum, cs, &bus, &slave));
ebcab48a SG	184	ut_assertok(spi_claim_bus(slave));
	185	ut_assertok(spi_xfer(slave, 40, dout, din,
	186	SPI_XFER_BEGIN \| SPI_XFER_END));
	187	ut_asserteq(0xff, din[0]);
	188	ut_asserteq(0x20, din[1]);
	189	ut_asserteq(0x20, din[2]);
	190	ut_asserteq(0x15, din[3]);
	191	spi_release_bus(slave);
	192
	193	/*
	194	* Since we are about to destroy all devices, we must tell sandbox
	195	* to forget the emulation device
	196	*/
56c40460	197	#if CONFIG_IS_ENABLED(DM_SPI_FLASH)
ebcab48a SG	198	sandbox_sf_unbind_emul(state_get_current(), busnum, cs);
	199	#endif
	200
	201	return 0;
	202	}
e180c2b1	203	DM_TEST(dm_test_spi_xfer, UT_TESTF_SCAN_PDATA \| UT_TESTF_SCAN_FDT);