| Lei YU | 0783076 | 2019-03-01 17:17:28 +0800 | [diff] [blame] | 1 | #include <string> |
| 2 | #include <utility> |
| 3 | |
| 4 | #include <gtest/gtest.h> |
| 5 | |
| 6 | using PartClear = std::pair<std::string, bool>; |
| 7 | namespace utils |
| 8 | { |
| 9 | extern std::vector<PartClear> getPartsToClear(const std::string& info); |
| 10 | } |
| 11 | |
| 12 | TEST(TestItemUpdaterStatic, getPartsToClearOK) |
| 13 | { |
| 14 | constexpr auto info = |
| 15 | "Flash info:\n" |
| 16 | "-----------\n" |
| 17 | "Name = /dev/mtd6\n" |
| 18 | "Total size = 64MB Flags E:ECC, P:PRESERVED, R:READONLY, " |
| 19 | "B:BACKUP\n" |
| 20 | "Erase granule = 64KB F:REPROVISION, V:VOLATILE, C:CLEARECC\n" |
| 21 | "\n" |
| 22 | "TOC@0x00000000 Partitions:\n" |
| 23 | "-----------\n" |
| 24 | "ID=00 part 0x00000000..0x00002000 (actual=0x00002000) " |
| 25 | "[----------]\n" |
| 26 | "ID=01 HBEL 0x00008000..0x0002c000 (actual=0x00024000) " |
| 27 | "[E-----F-C-]\n" |
| 28 | "ID=02 GUARD 0x0002c000..0x00031000 (actual=0x00005000) " |
| 29 | "[E--P--F-C-]\n" |
| 30 | "ID=03 NVRAM 0x00031000..0x000c1000 (actual=0x00090000) " |
| 31 | "[---P--F---]\n" |
| 32 | "ID=04 SECBOOT 0x000c1000..0x000e5000 (actual=0x00024000) " |
| 33 | "[E--P------]\n" |
| 34 | "ID=05 DJVPD 0x000e5000..0x0012d000 (actual=0x00048000) " |
| 35 | "[E--P--F-C-]\n" |
| 36 | "ID=06 MVPD 0x0012d000..0x001bd000 (actual=0x00090000) " |
| 37 | "[E--P--F-C-]\n" |
| 38 | "ID=07 CVPD 0x001bd000..0x00205000 (actual=0x00048000) " |
| 39 | "[E--P--F-C-]\n" |
| 40 | "ID=08 HBB 0x00205000..0x00305000 (actual=0x00100000) " |
| 41 | "[EL--R-----]\n" |
| 42 | "ID=09 HBD 0x00305000..0x00425000 (actual=0x00120000) " |
| 43 | "[EL--------]\n" |
| 44 | "ID=10 HBI 0x00425000..0x013e5000 (actual=0x00fc0000) " |
| 45 | "[EL--R-----]\n" |
| 46 | "ID=11 SBE 0x013e5000..0x014a1000 (actual=0x000bc000) " |
| 47 | "[ELI-R-----]\n" |
| 48 | "ID=12 HCODE 0x014a1000..0x015c1000 (actual=0x00120000) " |
| 49 | "[EL--R-----]\n" |
| 50 | "ID=13 HBRT 0x015c1000..0x01bc1000 (actual=0x00600000) " |
| 51 | "[EL--R-----]\n" |
| 52 | "ID=14 PAYLOAD 0x01bc1000..0x01cc1000 (actual=0x00100000) " |
| 53 | "[-L--R-----]\n" |
| 54 | "ID=15 BOOTKERNEL 0x01cc1000..0x02bc1000 (actual=0x00f00000) " |
| 55 | "[-L--R-----]\n" |
| 56 | "ID=16 OCC 0x02bc1000..0x02ce1000 (actual=0x00120000) " |
| 57 | "[EL--R-----]\n" |
| 58 | "ID=17 FIRDATA 0x02ce1000..0x02ce4000 (actual=0x00003000) " |
| 59 | "[E-----F-C-]\n" |
| 60 | "ID=18 CAPP 0x02ce4000..0x02d08000 (actual=0x00024000) " |
| 61 | "[EL--R-----]\n" |
| 62 | "ID=19 BMC_INV 0x02d08000..0x02d11000 (actual=0x00009000) " |
| 63 | "[------F---]\n" |
| 64 | "ID=20 HBBL 0x02d11000..0x02d18000 (actual=0x00007000) " |
| 65 | "[EL--R-----]\n" |
| 66 | "ID=21 ATTR_TMP 0x02d18000..0x02d20000 (actual=0x00008000) " |
| 67 | "[------F---]\n" |
| 68 | "ID=22 ATTR_PERM 0x02d20000..0x02d28000 (actual=0x00008000) " |
| 69 | "[E-----F-C-]\n" |
| 70 | "ID=23 VERSION 0x02d28000..0x02d2a000 (actual=0x00002000) " |
| 71 | "[-L--R-----]\n" |
| 72 | "ID=24 IMA_CATALOG 0x02d2a000..0x02d6a000 (actual=0x00040000) " |
| 73 | "[EL--R-----]\n" |
| 74 | "ID=25 RINGOVD 0x02d6a000..0x02d8a000 (actual=0x00020000) " |
| 75 | "[----------]\n" |
| 76 | "ID=26 WOFDATA 0x02d8a000..0x0308a000 (actual=0x00300000) " |
| 77 | "[EL--R-----]\n" |
| 78 | "ID=27 HB_VOLATILE 0x0308a000..0x0308f000 (actual=0x00005000) " |
| 79 | "[E-----F-CV]\n" |
| 80 | "ID=28 MEMD 0x0308f000..0x0309d000 (actual=0x0000e000) " |
| 81 | "[EL--R-----]\n" |
| 82 | "ID=29 SBKT 0x0309d000..0x030a1000 (actual=0x00004000) " |
| 83 | "[EL--R-----]\n" |
| 84 | "ID=30 HDAT 0x030a1000..0x030a9000 (actual=0x00008000) " |
| 85 | "[EL--R-----]\n" |
| 86 | "ID=31 UVISOR 0x030a9000..0x031a9000 (actual=0x00100000) " |
| 87 | "[-L--R-----]\n" |
| 88 | "ID=32 OCMBFW 0x031a9000..0x031f4000 (actual=0x0004b000) " |
| 89 | "[EL--R-----]\n" |
| 90 | "ID=33 UVBWLIST 0x031f4000..0x03204000 (actual=0x00010000) " |
| 91 | "[-L--R-----]\n" |
| 92 | "ID=34 BACKUP_PART 0x03ff7000..0x03fff000 (actual=0x00000000) " |
| 93 | "[-----B----]"; |
| 94 | auto parts = utils::getPartsToClear(info); |
| 95 | EXPECT_EQ(11, parts.size()); |
| 96 | |
| 97 | EXPECT_EQ("HBEL", parts[0].first); |
| 98 | EXPECT_TRUE(parts[0].second); |
| 99 | |
| 100 | EXPECT_EQ("GUARD", parts[1].first); |
| 101 | EXPECT_TRUE(parts[1].second); |
| 102 | |
| 103 | EXPECT_EQ("NVRAM", parts[2].first); |
| 104 | EXPECT_FALSE(parts[2].second); |
| 105 | |
| 106 | EXPECT_EQ("HB_VOLATILE", parts[10].first); |
| 107 | EXPECT_TRUE(parts[10].second); |
| 108 | } |
| 109 | |
| 110 | TEST(TestItemUpdaterStatic, getPartsToClearNotOK) |
| 111 | { |
| 112 | // Verify the it does not crash on malformed texts |
| 113 | constexpr auto info = |
| 114 | "0x0308a000..0x0308f000(actual=0x00005000)" |
| 115 | "[E-----F-CV]\n" // missing ID and name with F |
| 116 | "ID=27 HB_VOLATILE 0x0308a000..0x0308f000 (actual=0x00005000) " |
| 117 | "E-----F-CV]\n" // missing [ |
| 118 | "ID=22 ATTR_PERM 0x02d20000..0x02d28000 (actual=0x00008000) " |
| 119 | "[E-----F-C-]\n" // The only valid one |
| 120 | "ID=28 MEMD 0x0308f000..0x0309d000 (actual=0x0000e000) " |
| 121 | "[----]\n" // missing flags |
| 122 | "SBKT 0x0309d000..0x030a1000 (actual=0x00004000) " |
| 123 | "[EL--R-----]\n"; // missing ID |
| 124 | |
| 125 | auto parts = utils::getPartsToClear(info); |
| 126 | EXPECT_EQ(1, parts.size()); |
| 127 | EXPECT_EQ("ATTR_PERM", parts[0].first); |
| 128 | EXPECT_TRUE(parts[0].second); |
| 129 | } |