| // SPDX-License-Identifier: GPL-2.0+ | |
| /* | |
| * Tests for mbr command | |
| * | |
| * Copyright 2023 Matrox Video | |
| * Written by Alex Gendin <agendin@matrox.com> | |
| */ | |
| #include <dm.h> | |
| #include <console.h> | |
| #include <dm/test.h> | |
| #include <mapmem.h> | |
| #include <part.h> | |
| #include <asm/global_data.h> | |
| #include <dm/device-internal.h> | |
| #include <dm/lists.h> | |
| #include <linux/sizes.h> | |
| #include <test/ut.h> | |
| DECLARE_GLOBAL_DATA_PTR; | |
| #define BLKSZ SZ_512 /* block size */ | |
| /* | |
| * Requirements for running test manually: | |
| * mmc6.img - File size needs to be at least 12 MiB | |
| * | |
| * Command to create mmc6.img: | |
| * $ dd if=/dev/zero of=mmc6.img bs=12M count=1 | |
| * | |
| * To run this test manually, place mmc6.img into the same directory as u-boot, | |
| * then run: | |
| * $ ./u-boot -Tc 'ut mbr' | |
| * | |
| * To run this test as part of U-Boot test: | |
| * $ ./test/py/test.py --bd sandbox --build -k ut_dm -v | |
| * Note: mmc6.img will be created by the test suit. | |
| */ | |
| static char * mbr_parts_header = "setenv mbr_parts '"; | |
| static char * mbr_parts_p1 = "uuid_disk=0x12345678;name=p1,start=8M,bootable,size=1M,id=0x0e"; | |
| static char * mbr_parts_p2 = ";name=p2,size=1M,id=0x0e"; | |
| static char * mbr_parts_p3 = ";name=p3,size=1M,id=0x0e"; | |
| static char * mbr_parts_p4 = ";name=p4,size=1M,id=0x0e"; | |
| static char * mbr_parts_p5 = ";name=[ext],size=2M,id=0x05;name=p5,size=1M,id=0x0e"; | |
| static char * mbr_parts_tail = "'"; | |
| /* | |
| * One MBR partition | |
| 000001b0 00 00 00 00 00 00 00 00 78 56 34 12 00 00 80 05 |........xV4.....| | |
| 000001c0 05 01 0e 25 24 01 00 40 00 00 00 08 00 00 00 00 |...%$..@........| | |
| 000001d0 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 |................| | |
| 000001e0 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 |................| | |
| 000001f0 00 00 00 00 00 00 00 00 00 00 00 00 00 00 55 aa |..............U.| | |
| */ | |
| static unsigned int mbr_cmp_start = 0x1b8; | |
| static unsigned mbr_cmp_size = 0x48; | |
| static unsigned char mbr_parts_ref_p1[] = { | |
| 0x78, 0x56, 0x34, 0x12, 0x00, 0x00, 0x80, 0x05, | |
| 0x05, 0x01, 0x0e, 0x25, 0x24, 0x01, 0x00, 0x40, 0x00, 0x00, 0x00, 0x08, 0x00, 0x00, 0x00, 0x00, | |
| 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, | |
| 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, | |
| 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x55, 0xaa | |
| }; | |
| /* | |
| * Two MBR partitions | |
| 000001b0 00 00 00 00 00 00 00 00 78 56 34 12 00 00 80 05 |........xV4.....| | |
| 000001c0 05 01 0e 25 24 01 00 40 00 00 00 08 00 00 00 25 |...%$..@.......%| | |
| 000001d0 25 01 0e 46 05 01 00 48 00 00 00 08 00 00 00 00 |%..F...H........| | |
| 000001e0 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 |................| | |
| 000001f0 00 00 00 00 00 00 00 00 00 00 00 00 00 00 55 aa |..............U.| | |
| */ | |
| static unsigned char mbr_parts_ref_p2[] = { | |
| 0x78, 0x56, 0x34, 0x12, 0x00, 0x00, 0x80, 0x05, | |
| 0x05, 0x01, 0x0e, 0x25, 0x24, 0x01, 0x00, 0x40, 0x00, 0x00, 0x00, 0x08, 0x00, 0x00, 0x00, 0x25, | |
| 0x25, 0x01, 0x0e, 0x46, 0x05, 0x01, 0x00, 0x48, 0x00, 0x00, 0x00, 0x08, 0x00, 0x00, 0x00, 0x00, | |
| 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, | |
| 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x55, 0xaa | |
| }; | |
| /* | |
| * Three MBR partitions | |
| 000001b0 00 00 00 00 00 00 00 00 78 56 34 12 00 00 80 05 |........xV4.....| | |
| 000001c0 05 01 0e 25 24 01 00 40 00 00 00 08 00 00 00 25 |...%$..@.......%| | |
| 000001d0 25 01 0e 46 05 01 00 48 00 00 00 08 00 00 00 46 |%..F...H.......F| | |
| 000001e0 06 01 0e 66 25 01 00 50 00 00 00 08 00 00 00 00 |...f%..P........| | |
| 000001f0 00 00 00 00 00 00 00 00 00 00 00 00 00 00 55 aa |..............U.| | |
| */ | |
| static unsigned char mbr_parts_ref_p3[] = { | |
| 0x78, 0x56, 0x34, 0x12, 0x00, 0x00, 0x80, 0x05, | |
| 0x05, 0x01, 0x0e, 0x25, 0x24, 0x01, 0x00, 0x40, 0x00, 0x00, 0x00, 0x08, 0x00, 0x00, 0x00, 0x25, | |
| 0x25, 0x01, 0x0e, 0x46, 0x05, 0x01, 0x00, 0x48, 0x00, 0x00, 0x00, 0x08, 0x00, 0x00, 0x00, 0x46, | |
| 0x06, 0x01, 0x0e, 0x66, 0x25, 0x01, 0x00, 0x50, 0x00, 0x00, 0x00, 0x08, 0x00, 0x00, 0x00, 0x00, | |
| 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x55, 0xaa | |
| }; | |
| /* | |
| * Four MBR partitions | |
| 000001b0 00 00 00 00 00 00 00 00 78 56 34 12 00 00 80 05 |........xV4.....| | |
| 000001c0 05 01 0e 25 24 01 00 40 00 00 00 08 00 00 00 25 |...%$..@.......%| | |
| 000001d0 25 01 0e 46 05 01 00 48 00 00 00 08 00 00 00 46 |%..F...H.......F| | |
| 000001e0 06 01 0e 66 25 01 00 50 00 00 00 08 00 00 00 66 |...f%..P.......f| | |
| 000001f0 26 01 0e 87 06 01 00 58 00 00 00 08 00 00 55 aa |&......X......U.| | |
| */ | |
| static unsigned char mbr_parts_ref_p4[] = { | |
| 0x78, 0x56, 0x34, 0x12, 0x00, 0x00, 0x80, 0x05, | |
| 0x05, 0x01, 0x0e, 0x25, 0x24, 0x01, 0x00, 0x40, 0x00, 0x00, 0x00, 0x08, 0x00, 0x00, 0x00, 0x25, | |
| 0x25, 0x01, 0x0e, 0x46, 0x05, 0x01, 0x00, 0x48, 0x00, 0x00, 0x00, 0x08, 0x00, 0x00, 0x00, 0x46, | |
| 0x06, 0x01, 0x0e, 0x66, 0x25, 0x01, 0x00, 0x50, 0x00, 0x00, 0x00, 0x08, 0x00, 0x00, 0x00, 0x66, | |
| 0x26, 0x01, 0x0e, 0x87, 0x06, 0x01, 0x00, 0x58, 0x00, 0x00, 0x00, 0x08, 0x00, 0x00, 0x55, 0xaa | |
| }; | |
| /* | |
| * Five MBR partitions | |
| 000001b0 00 00 00 00 00 00 00 00 78 56 34 12 00 00 80 05 |........xV4.....| | |
| 000001c0 05 01 0e 25 24 01 00 40 00 00 00 08 00 00 00 25 |...%$..@.......%| | |
| 000001d0 25 01 0e 46 05 01 00 48 00 00 00 08 00 00 00 46 |%..F...H.......F| | |
| 000001e0 06 01 0e 66 25 01 00 50 00 00 00 08 00 00 00 66 |...f%..P.......f| | |
| 000001f0 26 01 05 a7 26 01 00 58 00 00 00 10 00 00 55 aa |&...&..X......U.| | |
| */ | |
| static unsigned char mbr_parts_ref_p5[] = { | |
| 0x78, 0x56, 0x34, 0x12, 0x00, 0x00, 0x80, 0x05, | |
| 0x05, 0x01, 0x0e, 0x25, 0x24, 0x01, 0x00, 0x40, 0x00, 0x00, 0x00, 0x08, 0x00, 0x00, 0x00, 0x25, | |
| 0x25, 0x01, 0x0e, 0x46, 0x05, 0x01, 0x00, 0x48, 0x00, 0x00, 0x00, 0x08, 0x00, 0x00, 0x00, 0x46, | |
| 0x06, 0x01, 0x0e, 0x66, 0x25, 0x01, 0x00, 0x50, 0x00, 0x00, 0x00, 0x08, 0x00, 0x00, 0x00, 0x66, | |
| 0x26, 0x01, 0x05, 0xa7, 0x26, 0x01, 0x00, 0x58, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x55, 0xaa | |
| }; | |
| static unsigned ebr_cmp_start = 0x1B8; | |
| static unsigned ebr_cmp_size = 0x48; | |
| /* | |
| * EBR | |
| 00b001b0 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 87 |................| | |
| 00b001c0 07 01 0e a7 26 01 00 08 00 00 00 08 00 00 00 00 |....&...........| | |
| 00b001d0 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 |................| | |
| 00b001e0 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 |................| | |
| 00b001f0 00 00 00 00 00 00 00 00 00 00 00 00 00 00 55 aa |..............U.| | |
| */ | |
| static unsigned char ebr_parts_ref_p5[] = { | |
| 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x87, | |
| 0x07, 0x01, 0x0e, 0xa7, 0x26, 0x01, 0x00, 0x08, 0x00, 0x00, 0x00, 0x08, 0x00, 0x00, 0x00, 0x00, | |
| 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, | |
| 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, | |
| 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x55, 0xaa | |
| }; | |
| /* Fill write buffers with pseudo-random data */ | |
| static void init_write_buffers(char *mbr_wb, size_t mbr_wb_size, | |
| char *ebr_wb, size_t ebr_wb_size, unsigned seed) | |
| { | |
| while (mbr_wb_size--) { | |
| *mbr_wb++ = seed; | |
| seed *= 43; | |
| seed += 17 + mbr_wb_size/4; | |
| } | |
| while (ebr_wb_size--) { | |
| *ebr_wb++ = seed; | |
| seed *= 43; | |
| seed += 17 + ebr_wb_size/4; | |
| } | |
| } | |
| /* Build string with MBR partition(s) layout */ | |
| static unsigned build_mbr_parts(char *buf, size_t buf_size, unsigned num_parts) | |
| { | |
| size_t bytes_remaining, cur_str_size; | |
| char * cur_buf; | |
| if (!num_parts || num_parts > 5 || !buf) | |
| return 1; | |
| cur_buf = buf; | |
| *cur_buf = '\0'; | |
| bytes_remaining = buf_size; | |
| cur_str_size = sizeof(mbr_parts_header); | |
| if (cur_str_size + 1 > bytes_remaining) | |
| return 1; | |
| strcat(cur_buf, mbr_parts_header); | |
| bytes_remaining -= cur_str_size; | |
| if (num_parts >= 1) { | |
| cur_str_size = sizeof(mbr_parts_p1); | |
| if (cur_str_size + 1 > bytes_remaining) | |
| return 1; | |
| strcat(cur_buf, mbr_parts_p1); | |
| bytes_remaining -= cur_str_size; | |
| if (num_parts >= 2) { | |
| cur_str_size = sizeof(mbr_parts_p2); | |
| if (cur_str_size + 1 > bytes_remaining) | |
| return 1; | |
| strcat(cur_buf, mbr_parts_p2); | |
| bytes_remaining -= cur_str_size; | |
| if (num_parts >= 3) { | |
| cur_str_size = sizeof(mbr_parts_p3); | |
| if (cur_str_size + 1 > bytes_remaining) | |
| return 1; | |
| strcat(cur_buf, mbr_parts_p3); | |
| bytes_remaining -= cur_str_size; | |
| if (num_parts == 4) { | |
| cur_str_size = sizeof(mbr_parts_p4); | |
| if (cur_str_size + 1 > bytes_remaining) | |
| return 1; | |
| strcat(cur_buf, mbr_parts_p4); | |
| bytes_remaining -= cur_str_size; | |
| } | |
| else if (num_parts == 5) { | |
| cur_str_size = sizeof(mbr_parts_p5); | |
| if (cur_str_size + 1 > bytes_remaining) | |
| return 1; | |
| strcat(cur_buf, mbr_parts_p5); | |
| bytes_remaining -= cur_str_size; | |
| } | |
| } | |
| } | |
| } | |
| cur_str_size = sizeof(mbr_parts_tail); | |
| if (cur_str_size + 1 > bytes_remaining) | |
| return 1; | |
| strcat(cur_buf, mbr_parts_tail); | |
| return 0; | |
| } | |
| static int mbr_test_run(struct unit_test_state *uts) | |
| { | |
| struct blk_desc *mmc_dev_desc; | |
| unsigned char *mbr_wbuf, *ebr_wbuf, *rbuf; | |
| char mbr_parts_buf[256]; | |
| ulong addr = 0x1000; /* start address for buffers */ | |
| ulong mbr_wa = addr, ebr_wa = addr + BLKSZ, ra = addr + BLKSZ * 2; | |
| ulong ebr_blk, mbr_parts_max; | |
| struct udevice *dev; | |
| ofnode root, node; | |
| /* Enable the mmc6 node for this test */ | |
| root = oftree_root(oftree_default()); | |
| node = ofnode_find_subnode(root, "mmc6"); | |
| ut_assert(ofnode_valid(node)); | |
| ut_assertok(lists_bind_fdt(gd->dm_root, node, &dev, NULL, false)); | |
| /* | |
| * 1 byte for null character | |
| * 2 reserved bytes | |
| */ | |
| mbr_parts_max = 1 + 2 + | |
| strlen(mbr_parts_header) + | |
| strlen(mbr_parts_p1) + | |
| strlen(mbr_parts_p2) + | |
| strlen(mbr_parts_p3) + | |
| max(strlen(mbr_parts_p4), strlen(mbr_parts_p5)) + | |
| strlen(mbr_parts_tail); | |
| ut_assertf(sizeof(mbr_parts_buf) >= mbr_parts_max, "Buffer avail: %zd; buffer req: %ld\n", | |
| sizeof(mbr_parts_buf), mbr_parts_max); | |
| mbr_wbuf = map_sysmem(mbr_wa, BLKSZ); | |
| ebr_wbuf = map_sysmem(ebr_wa, BLKSZ); | |
| rbuf = map_sysmem(ra, BLKSZ); | |
| ebr_blk = (ulong)0xb00000 / BLKSZ; | |
| /* Make sure mmc6 exists */ | |
| ut_asserteq(6, blk_get_device_by_str("mmc", "6", &mmc_dev_desc)); | |
| ut_assertok(run_commandf("mmc dev 6")); | |
| ut_assert_nextline("switch to partitions #0, OK"); | |
| ut_assert_nextline("mmc6 is current device"); | |
| ut_assert_console_end(); | |
| /* Make sure mmc6 is 12+ MiB in size */ | |
| ut_assertok(run_commandf("mmc read %lx %lx 1", ra, | |
| (ulong)0xbffe00 / BLKSZ)); | |
| /* Test one MBR partition */ | |
| init_write_buffers(mbr_wbuf, BLKSZ, ebr_wbuf, BLKSZ, __LINE__); | |
| ut_assertok(build_mbr_parts(mbr_parts_buf, sizeof(mbr_parts_buf), 1)); | |
| ut_assertok(run_commandf("write mmc 6:0 %lx 0 1", mbr_wa)); | |
| memset(rbuf, '\0', BLKSZ); | |
| ut_assertok(run_commandf("read mmc 6:0 %lx 0 1", ra)); | |
| ut_assertok(memcmp(mbr_wbuf, rbuf, BLKSZ)); | |
| ut_assertok(run_commandf("write mmc 6:0 %lx %lx 1", ebr_wa, ebr_blk)); | |
| memset(rbuf, '\0', BLKSZ); | |
| ut_assertok(run_commandf("read mmc 6:0 %lx %lx 1", ra, ebr_blk)); | |
| ut_assertok(memcmp(ebr_wbuf, rbuf, BLKSZ)); | |
| ut_assertf(0 == run_commandf(mbr_parts_buf), "Invalid partitions string: %s\n", mbr_parts_buf); | |
| ut_assertok(run_commandf("mbr write mmc 6")); | |
| ut_assert_nextlinen("MMC read: dev # 6"); | |
| ut_assert_nextline("MBR: write success!"); | |
| ut_assertok(run_commandf("mbr verify mmc 6")); | |
| ut_assert_nextline("MBR: verify success!"); | |
| memset(rbuf, '\0', BLKSZ); | |
| ut_assertok(run_commandf("read mmc 6:0 %lx %lx 1", ra, ebr_blk)); | |
| ut_assertok(memcmp(ebr_wbuf, rbuf, BLKSZ)); | |
| ut_assert_console_end(); | |
| /* | |
| 000001b0 00 00 00 00 00 00 00 00 78 56 34 12 00 00 80 05 |........xV4.....| | |
| 000001c0 05 01 0e 25 24 01 00 40 00 00 00 08 00 00 00 00 |...%$..@........| | |
| 000001d0 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 |................| | |
| 000001e0 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 |................| | |
| 000001f0 00 00 00 00 00 00 00 00 00 00 00 00 00 00 55 aa |..............U.| | |
| */ | |
| memset(rbuf, '\0', BLKSZ); | |
| ut_assertok(run_commandf("read mmc 6:0 %lx 0 1", ra)); | |
| for (unsigned i = 0; i < mbr_cmp_size; i++) { | |
| ut_assertf(rbuf[mbr_cmp_start + i] == mbr_parts_ref_p1[i], | |
| "1P MBR+0x%04X: expected %#02X, actual: %#02X\n", | |
| mbr_cmp_start + i, mbr_parts_ref_p1[i], rbuf[mbr_cmp_start + i]); | |
| } | |
| /* Test two MBR partitions */ | |
| init_write_buffers(mbr_wbuf, sizeof(mbr_wbuf), ebr_wbuf, sizeof(ebr_wbuf), __LINE__); | |
| ut_assertok(build_mbr_parts(mbr_parts_buf, sizeof(mbr_parts_buf), 2)); | |
| ut_assertok(run_commandf("write mmc 6:0 %lx 0 1", mbr_wa)); | |
| memset(rbuf, '\0', BLKSZ); | |
| ut_assertok(run_commandf("read mmc 6:0 %lx 0 1", ra)); | |
| ut_assertok(memcmp(mbr_wbuf, rbuf, BLKSZ)); | |
| ut_assertok(run_commandf("write mmc 6:0 %lx %lx 1", ebr_wa, ebr_blk)); | |
| memset(rbuf, '\0', BLKSZ); | |
| ut_assertok(run_commandf("read mmc 6:0 %lx %lx 1", ra, ebr_blk)); | |
| ut_assertok(memcmp(ebr_wbuf, rbuf, BLKSZ)); | |
| ut_assertf(0 == run_commandf(mbr_parts_buf), "Invalid partitions string: %s\n", mbr_parts_buf); | |
| ut_assertok(run_commandf("mbr write mmc 6")); | |
| ut_assert_nextline("MBR: write success!"); | |
| ut_assertok(run_commandf("mbr verify mmc 6")); | |
| ut_assert_nextline("MBR: verify success!"); | |
| memset(rbuf, '\0', BLKSZ); | |
| ut_assertok(run_commandf("read mmc 6:0 %lx %lx 1", ra, ebr_blk)); | |
| ut_assertok(memcmp(ebr_wbuf, rbuf, BLKSZ)); | |
| ut_assert_console_end(); | |
| /* | |
| 000001b0 00 00 00 00 00 00 00 00 78 56 34 12 00 00 80 05 |........xV4.....| | |
| 000001c0 05 01 0e 25 24 01 00 40 00 00 00 08 00 00 00 25 |...%$..@.......%| | |
| 000001d0 25 01 0e 46 05 01 00 48 00 00 00 08 00 00 00 00 |%..F...H........| | |
| 000001e0 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 |................| | |
| 000001f0 00 00 00 00 00 00 00 00 00 00 00 00 00 00 55 aa |..............U.| | |
| */ | |
| memset(rbuf, '\0', BLKSZ); | |
| ut_assertok(run_commandf("read mmc 6:0 %lx 0 1", ra)); | |
| for (unsigned i = 0; i < mbr_cmp_size; i++) { | |
| ut_assertf(rbuf[mbr_cmp_start + i] == mbr_parts_ref_p2[i], | |
| "2P MBR+0x%04X: expected %#02X, actual: %#02X\n", | |
| mbr_cmp_start + i, mbr_parts_ref_p2[i], rbuf[mbr_cmp_start + i]); | |
| } | |
| /* Test three MBR partitions */ | |
| init_write_buffers(mbr_wbuf, sizeof(mbr_wbuf), ebr_wbuf, sizeof(ebr_wbuf), __LINE__); | |
| ut_assertok(build_mbr_parts(mbr_parts_buf, sizeof(mbr_parts_buf), 3)); | |
| ut_assertok(run_commandf("write mmc 6:0 %lx 0 1", mbr_wa)); | |
| memset(rbuf, '\0', BLKSZ); | |
| ut_assertok(run_commandf("read mmc 6:0 %lx 0 1", ra)); | |
| ut_assertok(memcmp(mbr_wbuf, rbuf, BLKSZ)); | |
| ut_assertok(run_commandf("write mmc 6:0 %lx %lx 1", ebr_wa, ebr_blk)); | |
| memset(rbuf, '\0', BLKSZ); | |
| ut_assertok(run_commandf("read mmc 6:0 %lx %lx 1", ra, ebr_blk)); | |
| ut_assertok(memcmp(ebr_wbuf, rbuf, BLKSZ)); | |
| ut_assertf(0 == run_commandf(mbr_parts_buf), "Invalid partitions string: %s\n", mbr_parts_buf); | |
| ut_assertok(run_commandf("mbr write mmc 6")); | |
| ut_assert_nextline("MBR: write success!"); | |
| ut_assertok(run_commandf("mbr verify mmc 6")); | |
| ut_assert_nextline("MBR: verify success!"); | |
| memset(rbuf, '\0', BLKSZ); | |
| ut_assertok(run_commandf("read mmc 6:0 %lx %lx 1", ra, ebr_blk)); | |
| ut_assertok(memcmp(ebr_wbuf, rbuf, BLKSZ)); | |
| ut_assert_console_end(); | |
| /* | |
| 000001b0 00 00 00 00 00 00 00 00 78 56 34 12 00 00 80 05 |........xV4.....| | |
| 000001c0 05 01 0e 25 24 01 00 40 00 00 00 08 00 00 00 25 |...%$..@.......%| | |
| 000001d0 25 01 0e 46 05 01 00 48 00 00 00 08 00 00 00 46 |%..F...H.......F| | |
| 000001e0 06 01 0e 66 25 01 00 50 00 00 00 08 00 00 00 00 |...f%..P........| | |
| 000001f0 00 00 00 00 00 00 00 00 00 00 00 00 00 00 55 aa |..............U.| | |
| */ | |
| memset(rbuf, '\0', BLKSZ); | |
| ut_assertok(run_commandf("read mmc 6:0 %lx 0 1", ra)); | |
| for (unsigned i = 0; i < mbr_cmp_size; i++) { | |
| ut_assertf(rbuf[mbr_cmp_start + i] == mbr_parts_ref_p3[i], | |
| "3P MBR+0x%04X: expected %#02X, actual: %#02X\n", | |
| mbr_cmp_start + i, mbr_parts_ref_p3[i], rbuf[mbr_cmp_start + i]); | |
| } | |
| /* Test four MBR partitions */ | |
| init_write_buffers(mbr_wbuf, sizeof(mbr_wbuf), ebr_wbuf, sizeof(ebr_wbuf), __LINE__); | |
| ut_assertok(build_mbr_parts(mbr_parts_buf, sizeof(mbr_parts_buf), 4)); | |
| ut_assertok(run_commandf("write mmc 6:0 %lx 0 1", mbr_wa)); | |
| memset(rbuf, '\0', BLKSZ); | |
| ut_assertok(run_commandf("read mmc 6:0 %lx 0 1", ra)); | |
| ut_assertok(memcmp(mbr_wbuf, rbuf, BLKSZ)); | |
| ut_assertok(run_commandf("write mmc 6:0 %lx %lx 1", ebr_wa, ebr_blk)); | |
| memset(rbuf, '\0', BLKSZ); | |
| ut_assertok(run_commandf("read mmc 6:0 %lx %lx 1", ra, ebr_blk)); | |
| ut_assertok(memcmp(ebr_wbuf, rbuf, BLKSZ)); | |
| ut_assertf(0 == run_commandf(mbr_parts_buf), "Invalid partitions string: %s\n", mbr_parts_buf); | |
| ut_assertok(run_commandf("mbr write mmc 6")); | |
| ut_assert_nextline("MBR: write success!"); | |
| ut_assertok(run_commandf("mbr verify mmc 6")); | |
| ut_assert_nextline("MBR: verify success!"); | |
| memset(rbuf, '\0', BLKSZ); | |
| ut_assertok(run_commandf("read mmc 6:0 %lx %lx 1", ra, ebr_blk)); | |
| ut_assertok(memcmp(ebr_wbuf, rbuf, BLKSZ)); | |
| ut_assert_console_end(); | |
| /* | |
| 000001b0 00 00 00 00 00 00 00 00 78 56 34 12 00 00 80 05 |........xV4.....| | |
| 000001c0 05 01 0e 25 24 01 00 40 00 00 00 08 00 00 00 25 |...%$..@.......%| | |
| 000001d0 25 01 0e 46 05 01 00 48 00 00 00 08 00 00 00 46 |%..F...H.......F| | |
| 000001e0 06 01 0e 66 25 01 00 50 00 00 00 08 00 00 00 66 |...f%..P.......f| | |
| 000001f0 26 01 0e 87 06 01 00 58 00 00 00 08 00 00 55 aa |&......X......U.| | |
| */ | |
| memset(rbuf, '\0', BLKSZ); | |
| ut_assertok(run_commandf("read mmc 6:0 %lx 0 1", ra)); | |
| for (unsigned i = 0; i < mbr_cmp_size; i++) { | |
| ut_assertf(rbuf[mbr_cmp_start + i] == mbr_parts_ref_p4[i], | |
| "4P MBR+0x%04X: expected %#02X, actual: %#02X\n", | |
| mbr_cmp_start + i, mbr_parts_ref_p4[i], rbuf[mbr_cmp_start + i]); | |
| } | |
| /* Test five MBR partitions */ | |
| init_write_buffers(mbr_wbuf, sizeof(mbr_wbuf), ebr_wbuf, sizeof(ebr_wbuf), __LINE__); | |
| ut_assertok(build_mbr_parts(mbr_parts_buf, sizeof(mbr_parts_buf), 5)); | |
| ut_assertok(run_commandf("write mmc 6:0 %lx 0 1", mbr_wa)); | |
| memset(rbuf, '\0', BLKSZ); | |
| ut_assertok(run_commandf("read mmc 6:0 %lx 0 1", ra)); | |
| ut_assertok(memcmp(mbr_wbuf, rbuf, BLKSZ)); | |
| ut_assertok(run_commandf("write mmc 6:0 %lx %lx 1", ebr_wa, ebr_blk)); | |
| memset(rbuf, '\0', BLKSZ); | |
| ut_assertok(run_commandf("read mmc 6:0 %lx %lx 1", ra, ebr_blk)); | |
| ut_assertok(memcmp(ebr_wbuf, rbuf, BLKSZ)); | |
| ut_assertf(0 == run_commandf(mbr_parts_buf), "Invalid partitions string: %s\n", mbr_parts_buf); | |
| ut_assertf(0 == run_commandf("mbr write mmc 6"), "Invalid partitions string: %s\n", mbr_parts_buf); | |
| ut_assert_nextline("MBR: write success!"); | |
| ut_assertok(run_commandf("mbr verify mmc 6")); | |
| ut_assert_nextline("MBR: verify success!"); | |
| ut_assert_console_end(); | |
| /* | |
| 000001b0 00 00 00 00 00 00 00 00 78 56 34 12 00 00 80 05 |........xV4.....| | |
| 000001c0 05 01 0e 25 24 01 00 40 00 00 00 08 00 00 00 25 |...%$..@.......%| | |
| 000001d0 25 01 0e 46 05 01 00 48 00 00 00 08 00 00 00 46 |%..F...H.......F| | |
| 000001e0 06 01 0e 66 25 01 00 50 00 00 00 08 00 00 00 66 |...f%..P.......f| | |
| 000001f0 26 01 05 a7 26 01 00 58 00 00 00 10 00 00 55 aa |&...&..X......U.| | |
| */ | |
| memset(rbuf, '\0', BLKSZ); | |
| ut_assertok(run_commandf("read mmc 6:0 %lx 0 1", ra)); | |
| for (unsigned i = 0; i < mbr_cmp_size; i++) { | |
| ut_assertf(rbuf[mbr_cmp_start + i] == mbr_parts_ref_p5[i], | |
| "5P MBR+0x%04X: expected %#02X, actual: %#02X\n", | |
| mbr_cmp_start + i, mbr_parts_ref_p5[i], rbuf[mbr_cmp_start + i]); | |
| } | |
| /* | |
| 00b001b0 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 87 |................| | |
| 00b001c0 07 01 0e a7 26 01 00 08 00 00 00 08 00 00 00 00 |....&...........| | |
| 00b001d0 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 |................| | |
| 00b001e0 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 |................| | |
| 00b001f0 00 00 00 00 00 00 00 00 00 00 00 00 00 00 55 aa |..............U.| | |
| */ | |
| memset(rbuf, '\0', BLKSZ); | |
| ut_assertok(run_commandf("read mmc 6:0 %lx %lx 1", ra, ebr_blk)); | |
| for (unsigned i = 0; i < ebr_cmp_size; i++) { | |
| ut_assertf(rbuf[ebr_cmp_start + i] == ebr_parts_ref_p5[i], | |
| "5P EBR+0x%04X: expected %#02X, actual: %#02X\n", | |
| ebr_cmp_start + i, ebr_parts_ref_p5[i], rbuf[ebr_cmp_start + i]); | |
| } | |
| unmap_sysmem(mbr_wbuf); | |
| unmap_sysmem(ebr_wbuf); | |
| unmap_sysmem(rbuf); | |
| return 0; | |
| } | |
| /* Declare mbr test */ | |
| UNIT_TEST(mbr_test_run, UTF_CONSOLE, mbr); |