test/cmd/rw.c

   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);