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