1*4882a593Smuzhiyun /*
2*4882a593Smuzhiyun * Copyright (C) 2013 Google, Inc
3*4882a593Smuzhiyun *
4*4882a593Smuzhiyun * SPDX-License-Identifier: GPL-2.0+
5*4882a593Smuzhiyun */
6*4882a593Smuzhiyun
7*4882a593Smuzhiyun #include <common.h>
8*4882a593Smuzhiyun #include <dm.h>
9*4882a593Smuzhiyun #include <fdtdec.h>
10*4882a593Smuzhiyun #include <spi.h>
11*4882a593Smuzhiyun #include <spi_flash.h>
12*4882a593Smuzhiyun #include <asm/state.h>
13*4882a593Smuzhiyun #include <asm/test.h>
14*4882a593Smuzhiyun #include <dm/test.h>
15*4882a593Smuzhiyun #include <dm/util.h>
16*4882a593Smuzhiyun #include <test/ut.h>
17*4882a593Smuzhiyun
18*4882a593Smuzhiyun /* Test that sandbox SPI flash works correctly */
dm_test_spi_flash(struct unit_test_state * uts)19*4882a593Smuzhiyun static int dm_test_spi_flash(struct unit_test_state *uts)
20*4882a593Smuzhiyun {
21*4882a593Smuzhiyun struct udevice *dev, *emul;
22*4882a593Smuzhiyun int full_size = 0x200000;
23*4882a593Smuzhiyun int size = 0x10000;
24*4882a593Smuzhiyun u8 *src, *dst;
25*4882a593Smuzhiyun uint map_size;
26*4882a593Smuzhiyun ulong map_base;
27*4882a593Smuzhiyun uint offset;
28*4882a593Smuzhiyun int i;
29*4882a593Smuzhiyun
30*4882a593Smuzhiyun src = map_sysmem(0x20000, full_size);
31*4882a593Smuzhiyun ut_assertok(os_write_file("spi.bin", src, full_size));
32*4882a593Smuzhiyun ut_assertok(uclass_first_device_err(UCLASS_SPI_FLASH, &dev));
33*4882a593Smuzhiyun
34*4882a593Smuzhiyun dst = map_sysmem(0x20000 + full_size, full_size);
35*4882a593Smuzhiyun ut_assertok(spi_flash_read_dm(dev, 0, size, dst));
36*4882a593Smuzhiyun ut_assertok(memcmp(src, dst, size));
37*4882a593Smuzhiyun
38*4882a593Smuzhiyun /* Erase */
39*4882a593Smuzhiyun ut_assertok(spi_flash_erase_dm(dev, 0, size));
40*4882a593Smuzhiyun ut_assertok(spi_flash_read_dm(dev, 0, size, dst));
41*4882a593Smuzhiyun for (i = 0; i < size; i++)
42*4882a593Smuzhiyun ut_asserteq(dst[i], 0xff);
43*4882a593Smuzhiyun
44*4882a593Smuzhiyun /* Write some new data */
45*4882a593Smuzhiyun for (i = 0; i < size; i++)
46*4882a593Smuzhiyun src[i] = i;
47*4882a593Smuzhiyun ut_assertok(spi_flash_write_dm(dev, 0, size, src));
48*4882a593Smuzhiyun ut_assertok(spi_flash_read_dm(dev, 0, size, dst));
49*4882a593Smuzhiyun ut_assertok(memcmp(src, dst, size));
50*4882a593Smuzhiyun
51*4882a593Smuzhiyun /* Try the write-protect stuff */
52*4882a593Smuzhiyun ut_assertok(uclass_first_device_err(UCLASS_SPI_EMUL, &emul));
53*4882a593Smuzhiyun ut_asserteq(0, spl_flash_get_sw_write_prot(dev));
54*4882a593Smuzhiyun sandbox_sf_set_block_protect(emul, 1);
55*4882a593Smuzhiyun ut_asserteq(1, spl_flash_get_sw_write_prot(dev));
56*4882a593Smuzhiyun sandbox_sf_set_block_protect(emul, 0);
57*4882a593Smuzhiyun ut_asserteq(0, spl_flash_get_sw_write_prot(dev));
58*4882a593Smuzhiyun
59*4882a593Smuzhiyun /* Check mapping */
60*4882a593Smuzhiyun ut_assertok(dm_spi_get_mmap(dev, &map_base, &map_size, &offset));
61*4882a593Smuzhiyun ut_asserteq(0x1000, map_base);
62*4882a593Smuzhiyun ut_asserteq(0x2000, map_size);
63*4882a593Smuzhiyun ut_asserteq(0x100, offset);
64*4882a593Smuzhiyun
65*4882a593Smuzhiyun /*
66*4882a593Smuzhiyun * Since we are about to destroy all devices, we must tell sandbox
67*4882a593Smuzhiyun * to forget the emulation device
68*4882a593Smuzhiyun */
69*4882a593Smuzhiyun sandbox_sf_unbind_emul(state_get_current(), 0, 0);
70*4882a593Smuzhiyun
71*4882a593Smuzhiyun return 0;
72*4882a593Smuzhiyun }
73*4882a593Smuzhiyun DM_TEST(dm_test_spi_flash, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
74*4882a593Smuzhiyun
75*4882a593Smuzhiyun /* Functional test that sandbox SPI flash works correctly */
dm_test_spi_flash_func(struct unit_test_state * uts)76*4882a593Smuzhiyun static int dm_test_spi_flash_func(struct unit_test_state *uts)
77*4882a593Smuzhiyun {
78*4882a593Smuzhiyun /*
79*4882a593Smuzhiyun * Create an empty test file and run the SPI flash tests. This is a
80*4882a593Smuzhiyun * long way from being a unit test, but it does test SPI device and
81*4882a593Smuzhiyun * emulator binding, probing, the SPI flash emulator including
82*4882a593Smuzhiyun * device tree decoding, plus the file-based backing store of SPI.
83*4882a593Smuzhiyun *
84*4882a593Smuzhiyun * More targeted tests could be created to perform the above steps
85*4882a593Smuzhiyun * one at a time. This might not increase test coverage much, but
86*4882a593Smuzhiyun * it would make bugs easier to find. It's not clear whether the
87*4882a593Smuzhiyun * benefit is worth the extra complexity.
88*4882a593Smuzhiyun */
89*4882a593Smuzhiyun ut_asserteq(0, run_command_list(
90*4882a593Smuzhiyun "sb save hostfs - 0 spi.bin 200000;"
91*4882a593Smuzhiyun "sf probe;"
92*4882a593Smuzhiyun "sf test 0 10000", -1, 0));
93*4882a593Smuzhiyun /*
94*4882a593Smuzhiyun * Since we are about to destroy all devices, we must tell sandbox
95*4882a593Smuzhiyun * to forget the emulation device
96*4882a593Smuzhiyun */
97*4882a593Smuzhiyun sandbox_sf_unbind_emul(state_get_current(), 0, 0);
98*4882a593Smuzhiyun
99*4882a593Smuzhiyun return 0;
100*4882a593Smuzhiyun }
101*4882a593Smuzhiyun DM_TEST(dm_test_spi_flash, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
102