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