1import os
2
3import infra.basetest
4
5
6class TestMtd(infra.basetest.BRTest):
7    config = infra.basetest.BASIC_TOOLCHAIN_CONFIG + \
8        """
9        BR2_PACKAGE_MTD=y
10        # BR2_TARGET_ROOTFS_TAR is not set
11        BR2_TARGET_ROOTFS_CPIO=y
12        """
13
14    def test_run(self):
15        cpio_file = os.path.join(self.builddir, "images", "rootfs.cpio")
16        self.emulator.boot(arch="armv7",
17                           kernel="builtin",
18                           options=["-initrd", cpio_file])
19
20        self.emulator.login()
21
22        mtd = "/dev/mtd0"
23        erasesize = 256 * 1024
24        test_sectors = 4
25        test_size = erasesize * test_sectors
26
27        output, exit_code = self.emulator.run(f"mtd_debug info {mtd}")
28        output = [x.strip() for x in output if x.strip()]
29        self.assertEqual(output, [
30            "mtd.type = MTD_NORFLASH",
31            "mtd.flags = MTD_CAP_NORFLASH",
32            "mtd.size = 134217728 (128M)",
33            "mtd.erasesize = 262144 (256K)",
34            "mtd.writesize = 1",
35            "mtd.oobsize = 0",
36            "regions = 0",
37        ])
38
39        # Test flashcp
40        self.assertRunOk(f"dd if=/dev/urandom of=random.bin bs={test_size} count=1")
41        self.assertRunOk(f"flashcp random.bin {mtd}")
42        self.assertRunOk(f"cmp -s -n {test_size} random.bin {mtd}")
43
44        # Test flash_erase
45        self.assertRunOk(f"dd if=/dev/zero bs={test_size} count=1 | tr '\\000' '\\377' >nor-erase.bin")
46        self.assertRunOk(f"flash_erase {mtd} 0 {test_sectors}")
47        self.assertRunOk(f"cmp -s -n {test_size} nor-erase.bin {mtd}")
48