1 /*
2 * Copyright 2018 The Hafnium Authors.
3 *
4 * Use of this source code is governed by a BSD-style
5 * license that can be found in the LICENSE file or at
6 * https://opensource.org/licenses/BSD-3-Clause.
7 */
8
9 #include <gmock/gmock.h>
10
11 extern "C" {
12 #include "hf/boot_params.h"
13 #include "hf/fdt_handler.h"
14 #include "hf/mpool.h"
15 }
16
17 #include <memory>
18
19 namespace
20 {
21 using ::testing::Eq;
22 using ::testing::NotNull;
23
24 constexpr size_t TEST_HEAP_SIZE = PAGE_SIZE * 10;
25
26 /*
27 * /dts-v1/;
28 *
29 * / {
30 * #address-cells = <2>;
31 * #size-cells = <2>;
32 *
33 * memory@0 {
34 * device_type = "memory";
35 * reg = <0x00000000 0x00000000 0x00000000 0x20000000
36 * 0x00000000 0x30000000 0x00000000 0x00010000>;
37 * };
38 * memory@1 {
39 * device_type = "memory";
40 * reg = <0x00000000 0x30020000 0x00000000 0x00010000>;
41 * };
42 *
43 * chosen {
44 * linux,initrd-start = <0x00000000>;
45 * linux,initrd-end = <0x00000000>;
46 * };
47 * };
48 *
49 * $ dtc --boot-cpu 0 --in-format dts --out-format dtb --out-version 17 test.dts
50 * | xxd -i
51 */
52
53 constexpr uint8_t test_dtb[] = {
54 0xd0, 0x0d, 0xfe, 0xed, 0x00, 0x00, 0x01, 0x7f, 0x00, 0x00, 0x00, 0x38,
55 0x00, 0x00, 0x01, 0x30, 0x00, 0x00, 0x00, 0x28, 0x00, 0x00, 0x00, 0x11,
56 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x4f,
57 0x00, 0x00, 0x00, 0xf8, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
58 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01,
59 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, 0x04,
60 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x03,
61 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x0f, 0x00, 0x00, 0x00, 0x02,
62 0x00, 0x00, 0x00, 0x01, 0x6d, 0x65, 0x6d, 0x6f, 0x72, 0x79, 0x40, 0x30,
63 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, 0x07,
64 0x00, 0x00, 0x00, 0x1b, 0x6d, 0x65, 0x6d, 0x6f, 0x72, 0x79, 0x00, 0x00,
65 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, 0x20, 0x00, 0x00, 0x00, 0x27,
66 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
67 0x20, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x30, 0x00, 0x00, 0x00,
68 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02,
69 0x00, 0x00, 0x00, 0x01, 0x6d, 0x65, 0x6d, 0x6f, 0x72, 0x79, 0x40, 0x31,
70 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, 0x07,
71 0x00, 0x00, 0x00, 0x1b, 0x6d, 0x65, 0x6d, 0x6f, 0x72, 0x79, 0x00, 0x00,
72 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, 0x27,
73 0x00, 0x00, 0x00, 0x00, 0x30, 0x02, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
74 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x01,
75 0x63, 0x68, 0x6f, 0x73, 0x65, 0x6e, 0x00, 0x00, 0x00, 0x00, 0x00, 0x03,
76 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x2b, 0x00, 0x00, 0x00, 0x00,
77 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x3e,
78 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x02,
79 0x00, 0x00, 0x00, 0x09, 0x23, 0x61, 0x64, 0x64, 0x72, 0x65, 0x73, 0x73,
80 0x2d, 0x63, 0x65, 0x6c, 0x6c, 0x73, 0x00, 0x23, 0x73, 0x69, 0x7a, 0x65,
81 0x2d, 0x63, 0x65, 0x6c, 0x6c, 0x73, 0x00, 0x64, 0x65, 0x76, 0x69, 0x63,
82 0x65, 0x5f, 0x74, 0x79, 0x70, 0x65, 0x00, 0x72, 0x65, 0x67, 0x00, 0x6c,
83 0x69, 0x6e, 0x75, 0x78, 0x2c, 0x69, 0x6e, 0x69, 0x74, 0x72, 0x64, 0x2d,
84 0x73, 0x74, 0x61, 0x72, 0x74, 0x00, 0x6c, 0x69, 0x6e, 0x75, 0x78, 0x2c,
85 0x69, 0x6e, 0x69, 0x74, 0x72, 0x64, 0x2d, 0x65, 0x6e, 0x64, 0x00};
86
TEST(fdt,find_memory_ranges)87 TEST(fdt, find_memory_ranges)
88 {
89 struct mpool ppool;
90 std::unique_ptr<uint8_t[]> test_heap(new uint8_t[TEST_HEAP_SIZE]);
91
92 mpool_init(&ppool, sizeof(struct mm_page_table));
93 mpool_add_chunk(&ppool, test_heap.get(), TEST_HEAP_SIZE);
94 mm_init(&ppool);
95
96 struct fdt fdt;
97 struct boot_params params = {};
98
99 struct mm_stage1_locked mm_stage1_locked = mm_lock_stage1();
100 struct string memory = STRING_INIT("memory");
101 ASSERT_TRUE(fdt_map(&fdt, mm_stage1_locked,
102 pa_init((uintpaddr_t)&test_dtb), &ppool));
103 fdt_find_memory_ranges(&fdt, &memory, params.mem_ranges,
104 ¶ms.mem_ranges_count, MAX_MEM_RANGES);
105 ASSERT_TRUE(fdt_unmap(&fdt, mm_stage1_locked, &ppool));
106 mm_unlock_stage1(&mm_stage1_locked);
107
108 EXPECT_THAT(params.mem_ranges_count, Eq(3));
109 EXPECT_THAT(pa_addr(params.mem_ranges[0].begin), Eq(0x00000000));
110 EXPECT_THAT(pa_addr(params.mem_ranges[0].end), Eq(0x20000000));
111 EXPECT_THAT(pa_addr(params.mem_ranges[1].begin), Eq(0x30000000));
112 EXPECT_THAT(pa_addr(params.mem_ranges[1].end), Eq(0x30010000));
113 EXPECT_THAT(pa_addr(params.mem_ranges[2].begin), Eq(0x30020000));
114 EXPECT_THAT(pa_addr(params.mem_ranges[2].end), Eq(0x30030000));
115 }
116
117 } /* namespace */
118