| Rasmus Villemoes | ceefc04 | 2023-03-02 09:12:25 +0100 | [diff] [blame] | 1 | // SPDX-License-Identifier: GPL-2.0-or-later |
| 2 | /* |
| 3 | * Tests for read and write commands |
| 4 | */ |
| 5 | |
| 6 | #include <common.h> |
| 7 | #include <dm/test.h> |
| 8 | #include <mapmem.h> |
| 9 | #include <part.h> |
| 10 | #include <test/test.h> |
| 11 | #include <test/ut.h> |
| 12 | |
| 13 | static int setup_partitions(struct unit_test_state *uts, struct blk_desc **mmc_dev_desc) |
| 14 | { |
| 15 | char str_disk_guid[UUID_STR_LEN + 1]; |
| 16 | struct disk_partition parts[2] = { |
| 17 | { |
| 18 | .start = 48, /* GPT data takes up the first 34 blocks or so */ |
| 19 | .size = 4, |
| 20 | .name = "data", |
| 21 | }, |
| 22 | { |
| 23 | .start = 52, |
| 24 | .size = 10, |
| 25 | .name = "log", |
| 26 | }, |
| 27 | }; |
| 28 | |
| 29 | ut_asserteq(2, blk_get_device_by_str("mmc", "2", mmc_dev_desc)); |
| 30 | if (CONFIG_IS_ENABLED(RANDOM_UUID)) { |
| 31 | gen_rand_uuid_str(parts[0].uuid, UUID_STR_FORMAT_STD); |
| 32 | gen_rand_uuid_str(parts[1].uuid, UUID_STR_FORMAT_STD); |
| 33 | gen_rand_uuid_str(str_disk_guid, UUID_STR_FORMAT_STD); |
| 34 | } |
| 35 | ut_assertok(gpt_restore(*mmc_dev_desc, str_disk_guid, parts, |
| 36 | ARRAY_SIZE(parts))); |
| 37 | return 0; |
| 38 | } |
| 39 | |
| 40 | /* Fill the write buffer with pseudo-random data, clear the read buffer. */ |
| 41 | static void init_buffers(char *rb, char *wb, size_t size, unsigned seed) |
| 42 | { |
| 43 | memset(rb, 0, size); |
| 44 | while (size--) { |
| 45 | *wb++ = seed; |
| 46 | seed *= 43; |
| 47 | seed += 17 + size/4; |
| 48 | } |
| 49 | } |
| 50 | |
| 51 | static int dm_test_read_write(struct unit_test_state *uts) |
| 52 | { |
| 53 | struct blk_desc *dev_desc; |
| 54 | char wbuf[1024], rbuf[1024]; |
| 55 | ulong wa, ra; |
| 56 | |
| 57 | #define INIT_BUFFERS() init_buffers(rbuf, wbuf, sizeof(rbuf), __LINE__) |
| 58 | |
| 59 | ut_assertok(setup_partitions(uts, &dev_desc)); |
| 60 | |
| 61 | wa = map_to_sysmem(wbuf); |
| 62 | ra = map_to_sysmem(rbuf); |
| 63 | |
| 64 | /* Simple test, write to/read from same partition. */ |
| 65 | INIT_BUFFERS(); |
| 66 | ut_assertok(run_commandf("write mmc 2:1 0x%lx 0 2", wa)); |
| 67 | ut_assertok(run_commandf("read mmc 2:1 0x%lx 0 2", ra)); |
| 68 | ut_assertok(memcmp(wbuf, rbuf, sizeof(wbuf))); |
| 69 | ut_assertok(run_commandf("read mmc 2:1 0x%lx 1 1", ra)); |
| 70 | ut_assertok(memcmp(&wbuf[512], rbuf, 512)); |
| 71 | |
| 72 | /* Use name for write, number for read. */ |
| 73 | INIT_BUFFERS(); |
| 74 | ut_assertok(run_commandf("write mmc 2#log 0x%lx 0 2", wa)); |
| 75 | ut_assertok(run_commandf("read mmc 2:2 0x%lx 0 2", ra)); |
| 76 | ut_assertok(memcmp(wbuf, rbuf, sizeof(wbuf))); |
| 77 | |
| 78 | /* Use full device for write, name for read. */ |
| 79 | INIT_BUFFERS(); |
| 80 | ut_assertok(run_commandf("write mmc 2:0 0x%lx 0x30 2", wa)); |
| 81 | ut_assertok(run_commandf("read mmc 2#data 0x%lx 0 2", ra)); |
| 82 | ut_assertok(memcmp(wbuf, rbuf, sizeof(wbuf))); |
| 83 | |
| 84 | /* Use name for write, full device for read */ |
| 85 | INIT_BUFFERS(); |
| 86 | ut_assertok(run_commandf("write mmc 2#log 0x%lx 1 2", wa)); |
| 87 | ut_assertok(run_commandf("read mmc 2:0 0x%lx 0x35 2", ra)); |
| 88 | ut_assertok(memcmp(wbuf, rbuf, sizeof(wbuf))); |
| 89 | |
| 90 | /* Read/write outside partition bounds should be rejected upfront. */ |
| 91 | console_record_reset_enable(); |
| 92 | ut_asserteq(1, run_commandf("read mmc 2#data 0x%lx 3 2", ra)); |
| 93 | ut_assert_nextlinen("read out of range"); |
| 94 | ut_assert_console_end(); |
| 95 | |
| 96 | console_record_reset_enable(); |
| 97 | ut_asserteq(1, run_commandf("write mmc 2#log 0x%lx 9 2", wa)); |
| 98 | ut_assert_nextlinen("write out of range"); |
| 99 | ut_assert_console_end(); |
| 100 | |
| 101 | return 0; |
| 102 | } |
| 103 | |
| 104 | DM_TEST(dm_test_read_write, UT_TESTF_SCAN_FDT | UT_TESTF_CONSOLE_REC); |