1 // Copyright 2018 The Fuchsia Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "logical-to-physical-map.h"
6
7 #include <fbl/array.h>
8 #include <fbl/vector.h>
9 #include <unittest/unittest.h>
10 #include <zircon/types.h>
11
12 #include <utility>
13
14 namespace nand {
15
CheckMultiple(LogicalToPhysicalMap ltop_map,fbl::Vector<fbl::Vector<uint32_t>> expected)16 bool CheckMultiple(LogicalToPhysicalMap ltop_map, fbl::Vector<fbl::Vector<uint32_t>> expected) {
17 BEGIN_HELPER;
18 for (uint32_t copy = 0; copy < expected.size(); copy++) {
19 EXPECT_EQ(ltop_map.LogicalBlockCount(copy), expected[copy].size());
20 uint32_t i;
21 for (i = 0; i < expected[copy].size(); i++) {
22 uint32_t physical;
23 zx_status_t status = ltop_map.GetPhysical(copy, i, &physical);
24 ASSERT_EQ(status, ZX_OK);
25 EXPECT_EQ(physical, expected[copy][i]);
26 }
27 uint32_t physical;
28 zx_status_t status = ltop_map.GetPhysical(copy, i, &physical);
29 EXPECT_EQ(status, ZX_ERR_OUT_OF_RANGE);
30 }
31 END_HELPER;
32 }
33
Check(LogicalToPhysicalMap ltop_map,fbl::Vector<uint32_t> expected)34 bool Check(LogicalToPhysicalMap ltop_map, fbl::Vector<uint32_t> expected) {
35 fbl::Vector<fbl::Vector<uint32_t>> expected_;
36 expected_.push_back(std::move(expected));
37 return CheckMultiple(std::move(ltop_map), std::move(expected_));
38 }
39
SimpleTest()40 bool SimpleTest() {
41 BEGIN_TEST;
42 LogicalToPhysicalMap ltop_map(1, 5, fbl::Array<uint32_t>());
43 EXPECT_TRUE(Check(std::move(ltop_map), {0, 1, 2, 3, 4}));
44 END_TEST;
45 }
46
SingleBadTest()47 bool SingleBadTest() {
48 BEGIN_TEST;
49 fbl::Array<uint32_t> bad_blocks(new uint32_t[1], 1);
50 bad_blocks[0] = 2;
51 LogicalToPhysicalMap ltop_map(1, 5, std::move(bad_blocks));
52 EXPECT_TRUE(Check(std::move(ltop_map), {0, 1, 3, 4}));
53 END_TEST;
54 }
55
FirstBadTest()56 bool FirstBadTest() {
57 BEGIN_TEST;
58 fbl::Array<uint32_t> bad_blocks(new uint32_t[1], 1);
59 bad_blocks[0] = 0;
60 LogicalToPhysicalMap ltop_map(1, 5, std::move(bad_blocks));
61 EXPECT_TRUE(Check(std::move(ltop_map), {1, 2, 3, 4}));
62 END_TEST;
63 }
64
LastBadTest()65 bool LastBadTest() {
66 BEGIN_TEST;
67 fbl::Array<uint32_t> bad_blocks(new uint32_t[1], 1);
68 bad_blocks[0] = 4;
69 LogicalToPhysicalMap ltop_map(1, 5, std::move(bad_blocks));
70 EXPECT_TRUE(Check(std::move(ltop_map), {0, 1, 2, 3}));
71 END_TEST;
72 }
73
MultipleBadTest()74 bool MultipleBadTest() {
75 BEGIN_TEST;
76 fbl::Array<uint32_t> bad_blocks(new uint32_t[3], 3);
77 bad_blocks[0] = 0;
78 bad_blocks[1] = 2;
79 bad_blocks[2] = 4;
80 LogicalToPhysicalMap ltop_map(1, 5, std::move(bad_blocks));
81 EXPECT_TRUE(Check(std::move(ltop_map), {1, 3}));
82 END_TEST;
83 }
84
AllBadTest()85 bool AllBadTest() {
86 BEGIN_TEST;
87 fbl::Array<uint32_t> bad_blocks(new uint32_t[3], 3);
88 bad_blocks[0] = 0;
89 bad_blocks[1] = 1;
90 bad_blocks[2] = 2;
91 LogicalToPhysicalMap ltop_map(1, 3, std::move(bad_blocks));
92
93 EXPECT_EQ(ltop_map.LogicalBlockCount(0), 0);
94 uint32_t physical;
95 zx_status_t status = ltop_map.GetPhysical(0, 0, &physical);
96 EXPECT_EQ(status, ZX_ERR_OUT_OF_RANGE);
97 END_TEST;
98 }
99
MultipleCopiesTest()100 bool MultipleCopiesTest() {
101 BEGIN_TEST;
102 LogicalToPhysicalMap ltop_map(4, 8, fbl::Array<uint32_t>());
103 fbl::Vector<fbl::Vector<uint32_t>> expected;
104 expected.push_back({0, 1});
105 expected.push_back({2, 3});
106 expected.push_back({4, 5});
107 expected.push_back({6, 7});
108 EXPECT_TRUE(CheckMultiple(std::move(ltop_map), std::move(expected)));
109 END_TEST;
110 }
111
MultipleCopiesSomeBadTest()112 bool MultipleCopiesSomeBadTest() {
113 BEGIN_TEST;
114 fbl::Array<uint32_t> bad_blocks(new uint32_t[5], 5);
115 bad_blocks[0] = 0;
116 bad_blocks[1] = 1;
117 bad_blocks[2] = 3;
118 bad_blocks[3] = 5;
119 bad_blocks[4] = 6;
120 LogicalToPhysicalMap ltop_map(2, 8, std::move(bad_blocks));
121 fbl::Vector<fbl::Vector<uint32_t>> expected;
122 expected.push_back({2});
123 expected.push_back({4, 7});
124 EXPECT_TRUE(CheckMultiple(std::move(ltop_map), std::move(expected)));
125 END_TEST;
126 }
127
128 } // namespace nand
129
130 BEGIN_TEST_CASE(LtopTests)
131 RUN_TEST(nand::SimpleTest)
132 RUN_TEST(nand::SingleBadTest)
133 RUN_TEST(nand::FirstBadTest)
134 RUN_TEST(nand::LastBadTest)
135 RUN_TEST(nand::AllBadTest)
136 RUN_TEST(nand::MultipleCopiesTest)
137 RUN_TEST(nand::MultipleCopiesSomeBadTest)
138 END_TEST_CASE(LtopTests);
139