| Christopher Ferris | 723cf9b | 2017-01-19 20:08:48 -0800 | [diff] [blame] | 1 | /* | 
|  | 2 | * Copyright (C) 2016 The Android Open Source Project | 
|  | 3 | * | 
|  | 4 | * Licensed under the Apache License, Version 2.0 (the "License"); | 
|  | 5 | * you may not use this file except in compliance with the License. | 
|  | 6 | * You may obtain a copy of the License at | 
|  | 7 | * | 
|  | 8 | *      http://www.apache.org/licenses/LICENSE-2.0 | 
|  | 9 | * | 
|  | 10 | * Unless required by applicable law or agreed to in writing, software | 
|  | 11 | * distributed under the License is distributed on an "AS IS" BASIS, | 
|  | 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | 
|  | 13 | * See the License for the specific language governing permissions and | 
|  | 14 | * limitations under the License. | 
|  | 15 | */ | 
|  | 16 |  | 
|  | 17 | #include <sys/mman.h> | 
|  | 18 |  | 
|  | 19 | #include <android-base/file.h> | 
|  | 20 | #include <android-base/test_utils.h> | 
|  | 21 | #include <gtest/gtest.h> | 
|  | 22 |  | 
|  | 23 | #include "Maps.h" | 
|  | 24 |  | 
|  | 25 | #include "LogFake.h" | 
|  | 26 |  | 
|  | 27 | class MapsTest : public ::testing::Test { | 
|  | 28 | protected: | 
|  | 29 | void SetUp() override { | 
|  | 30 | ResetLogs(); | 
|  | 31 | } | 
|  | 32 | }; | 
|  | 33 |  | 
|  | 34 | TEST_F(MapsTest, parse_permissions) { | 
|  | 35 | MapsBuffer maps("1000-2000 ---- 00000000 00:00 0\n" | 
|  | 36 | "2000-3000 r--- 00000000 00:00 0\n" | 
|  | 37 | "3000-4000 -w-- 00000000 00:00 0\n" | 
|  | 38 | "4000-5000 --x- 00000000 00:00 0\n" | 
|  | 39 | "5000-6000 rwx- 00000000 00:00 0\n"); | 
|  | 40 |  | 
|  | 41 | ASSERT_TRUE(maps.Parse()); | 
|  | 42 | ASSERT_EQ(5U, maps.Total()); | 
|  | 43 | auto it = maps.begin(); | 
|  | 44 | ASSERT_EQ(PROT_NONE, it->flags); | 
|  | 45 | ASSERT_EQ(0x1000U, it->start); | 
|  | 46 | ASSERT_EQ(0x2000U, it->end); | 
|  | 47 | ASSERT_EQ(0U, it->offset); | 
|  | 48 | ASSERT_EQ("", it->name); | 
|  | 49 | ++it; | 
|  | 50 | ASSERT_EQ(PROT_READ, it->flags); | 
|  | 51 | ASSERT_EQ(0x2000U, it->start); | 
|  | 52 | ASSERT_EQ(0x3000U, it->end); | 
|  | 53 | ASSERT_EQ(0U, it->offset); | 
|  | 54 | ASSERT_EQ("", it->name); | 
|  | 55 | ++it; | 
|  | 56 | ASSERT_EQ(PROT_WRITE, it->flags); | 
|  | 57 | ASSERT_EQ(0x3000U, it->start); | 
|  | 58 | ASSERT_EQ(0x4000U, it->end); | 
|  | 59 | ASSERT_EQ(0U, it->offset); | 
|  | 60 | ASSERT_EQ("", it->name); | 
|  | 61 | ++it; | 
|  | 62 | ASSERT_EQ(PROT_EXEC, it->flags); | 
|  | 63 | ASSERT_EQ(0x4000U, it->start); | 
|  | 64 | ASSERT_EQ(0x5000U, it->end); | 
|  | 65 | ASSERT_EQ(0U, it->offset); | 
|  | 66 | ASSERT_EQ("", it->name); | 
|  | 67 | ++it; | 
|  | 68 | ASSERT_EQ(PROT_READ | PROT_WRITE | PROT_EXEC, it->flags); | 
|  | 69 | ASSERT_EQ(0x5000U, it->start); | 
|  | 70 | ASSERT_EQ(0x6000U, it->end); | 
|  | 71 | ASSERT_EQ(0U, it->offset); | 
|  | 72 | ASSERT_EQ("", it->name); | 
|  | 73 | ++it; | 
|  | 74 | ASSERT_EQ(it, maps.end()); | 
|  | 75 | } | 
|  | 76 |  | 
|  | 77 | TEST_F(MapsTest, parse_name) { | 
|  | 78 | MapsBuffer maps("720b29b000-720b29e000 rw-p 00000000 00:00 0\n" | 
|  | 79 | "720b29e000-720b29f000 rw-p 00000000 00:00 0 /system/lib/fake.so\n" | 
|  | 80 | "720b29f000-720b2a0000 rw-p 00000000 00:00 0"); | 
|  | 81 |  | 
|  | 82 | ASSERT_TRUE(maps.Parse()); | 
|  | 83 | ASSERT_EQ(3U, maps.Total()); | 
|  | 84 | auto it = maps.begin(); | 
|  | 85 | ASSERT_EQ("", it->name); | 
|  | 86 | ASSERT_EQ(0x720b29b000U, it->start); | 
|  | 87 | ASSERT_EQ(0x720b29e000U, it->end); | 
|  | 88 | ASSERT_EQ(0U, it->offset); | 
|  | 89 | ASSERT_EQ(PROT_READ | PROT_WRITE, it->flags); | 
|  | 90 | ++it; | 
|  | 91 | ASSERT_EQ("/system/lib/fake.so", it->name); | 
|  | 92 | ASSERT_EQ(0x720b29e000U, it->start); | 
|  | 93 | ASSERT_EQ(0x720b29f000U, it->end); | 
|  | 94 | ASSERT_EQ(0U, it->offset); | 
|  | 95 | ASSERT_EQ(PROT_READ | PROT_WRITE, it->flags); | 
|  | 96 | ++it; | 
|  | 97 | ASSERT_EQ("", it->name); | 
|  | 98 | ASSERT_EQ(0x720b29f000U, it->start); | 
|  | 99 | ASSERT_EQ(0x720b2a0000U, it->end); | 
|  | 100 | ASSERT_EQ(0U, it->offset); | 
|  | 101 | ASSERT_EQ(PROT_READ | PROT_WRITE, it->flags); | 
|  | 102 | ++it; | 
|  | 103 | ASSERT_EQ(it, maps.end()); | 
|  | 104 | } | 
|  | 105 |  | 
|  | 106 | TEST_F(MapsTest, parse_offset) { | 
|  | 107 | MapsBuffer maps("a000-e000 rw-p 00000000 00:00 0 /system/lib/fake.so\n" | 
|  | 108 | "e000-f000 rw-p 00a12345 00:00 0 /system/lib/fake.so\n"); | 
|  | 109 |  | 
|  | 110 | ASSERT_TRUE(maps.Parse()); | 
|  | 111 | ASSERT_EQ(2U, maps.Total()); | 
|  | 112 | auto it = maps.begin(); | 
|  | 113 | ASSERT_EQ(0U, it->offset); | 
|  | 114 | ASSERT_EQ(0xa000U, it->start); | 
|  | 115 | ASSERT_EQ(0xe000U, it->end); | 
|  | 116 | ASSERT_EQ(PROT_READ | PROT_WRITE, it->flags); | 
|  | 117 | ASSERT_EQ("/system/lib/fake.so", it->name); | 
|  | 118 | ++it; | 
|  | 119 | ASSERT_EQ(0xa12345U, it->offset); | 
|  | 120 | ASSERT_EQ(0xe000U, it->start); | 
|  | 121 | ASSERT_EQ(0xf000U, it->end); | 
|  | 122 | ASSERT_EQ(PROT_READ | PROT_WRITE, it->flags); | 
|  | 123 | ASSERT_EQ("/system/lib/fake.so", it->name); | 
|  | 124 | ++it; | 
|  | 125 | ASSERT_EQ(maps.end(), it); | 
|  | 126 | } | 
|  | 127 |  | 
|  | 128 | TEST_F(MapsTest, file_smoke) { | 
|  | 129 | TemporaryFile tf; | 
|  | 130 | ASSERT_TRUE(tf.fd != -1); | 
|  | 131 |  | 
|  | 132 | ASSERT_TRUE(android::base::WriteStringToFile( | 
|  | 133 | "720b29b000-720b29e000 r-xp a0000000 00:00 0   /fake.so\n" | 
|  | 134 | "720b2b0000-720b2e0000 r-xp b0000000 00:00 0   /fake2.so\n" | 
|  | 135 | "720b2e0000-720b2f0000 r-xp c0000000 00:00 0   /fake3.so\n", | 
|  | 136 | tf.path, 0660, getuid(), getgid())); | 
|  | 137 |  | 
|  | 138 | MapsFile maps(tf.path); | 
|  | 139 |  | 
|  | 140 | ASSERT_TRUE(maps.Parse()); | 
|  | 141 | ASSERT_EQ(3U, maps.Total()); | 
|  | 142 | auto it = maps.begin(); | 
|  | 143 | ASSERT_EQ(0x720b29b000U, it->start); | 
|  | 144 | ASSERT_EQ(0x720b29e000U, it->end); | 
|  | 145 | ASSERT_EQ(0xa0000000U, it->offset); | 
|  | 146 | ASSERT_EQ(PROT_READ | PROT_EXEC, it->flags); | 
|  | 147 | ASSERT_EQ("/fake.so", it->name); | 
|  | 148 | ++it; | 
|  | 149 | ASSERT_EQ(0x720b2b0000U, it->start); | 
|  | 150 | ASSERT_EQ(0x720b2e0000U, it->end); | 
|  | 151 | ASSERT_EQ(0xb0000000U, it->offset); | 
|  | 152 | ASSERT_EQ(PROT_READ | PROT_EXEC, it->flags); | 
|  | 153 | ASSERT_EQ("/fake2.so", it->name); | 
|  | 154 | ++it; | 
|  | 155 | ASSERT_EQ(0x720b2e0000U, it->start); | 
|  | 156 | ASSERT_EQ(0x720b2f0000U, it->end); | 
|  | 157 | ASSERT_EQ(0xc0000000U, it->offset); | 
|  | 158 | ASSERT_EQ(PROT_READ | PROT_EXEC, it->flags); | 
|  | 159 | ASSERT_EQ("/fake3.so", it->name); | 
|  | 160 | ++it; | 
|  | 161 | ASSERT_EQ(it, maps.end()); | 
|  | 162 | } | 
|  | 163 |  | 
|  | 164 | TEST_F(MapsTest, find) { | 
|  | 165 | MapsBuffer maps("1000-2000 r--p 00000010 00:00 0 /system/lib/fake1.so\n" | 
|  | 166 | "3000-4000 -w-p 00000020 00:00 0 /system/lib/fake2.so\n" | 
|  | 167 | "6000-8000 --xp 00000030 00:00 0 /system/lib/fake3.so\n" | 
|  | 168 | "a000-b000 rw-p 00000040 00:00 0 /system/lib/fake4.so\n" | 
|  | 169 | "e000-f000 rwxp 00000050 00:00 0 /system/lib/fake5.so\n"); | 
|  | 170 | ASSERT_TRUE(maps.Parse()); | 
|  | 171 | ASSERT_EQ(5U, maps.Total()); | 
|  | 172 |  | 
|  | 173 | ASSERT_TRUE(maps.Find(0x500) == nullptr); | 
|  | 174 | ASSERT_TRUE(maps.Find(0x2000) == nullptr); | 
|  | 175 | ASSERT_TRUE(maps.Find(0x5010) == nullptr); | 
|  | 176 | ASSERT_TRUE(maps.Find(0x9a00) == nullptr); | 
|  | 177 | ASSERT_TRUE(maps.Find(0xf000) == nullptr); | 
|  | 178 | ASSERT_TRUE(maps.Find(0xf010) == nullptr); | 
|  | 179 |  | 
|  | 180 | MapInfo* info = maps.Find(0x1000); | 
|  | 181 | ASSERT_TRUE(info != nullptr); | 
|  | 182 | ASSERT_EQ(0x1000U, info->start); | 
|  | 183 | ASSERT_EQ(0x2000U, info->end); | 
|  | 184 | ASSERT_EQ(0x10U, info->offset); | 
|  | 185 | ASSERT_EQ(PROT_READ, info->flags); | 
|  | 186 | ASSERT_EQ("/system/lib/fake1.so", info->name); | 
|  | 187 |  | 
|  | 188 | info = maps.Find(0x3020); | 
|  | 189 | ASSERT_TRUE(info != nullptr); | 
|  | 190 | ASSERT_EQ(0x3000U, info->start); | 
|  | 191 | ASSERT_EQ(0x4000U, info->end); | 
|  | 192 | ASSERT_EQ(0x20U, info->offset); | 
|  | 193 | ASSERT_EQ(PROT_WRITE, info->flags); | 
|  | 194 | ASSERT_EQ("/system/lib/fake2.so", info->name); | 
|  | 195 |  | 
|  | 196 | info = maps.Find(0x6020); | 
|  | 197 | ASSERT_TRUE(info != nullptr); | 
|  | 198 | ASSERT_EQ(0x6000U, info->start); | 
|  | 199 | ASSERT_EQ(0x8000U, info->end); | 
|  | 200 | ASSERT_EQ(0x30U, info->offset); | 
|  | 201 | ASSERT_EQ(PROT_EXEC, info->flags); | 
|  | 202 | ASSERT_EQ("/system/lib/fake3.so", info->name); | 
|  | 203 |  | 
|  | 204 | info = maps.Find(0xafff); | 
|  | 205 | ASSERT_TRUE(info != nullptr); | 
|  | 206 | ASSERT_EQ(0xa000U, info->start); | 
|  | 207 | ASSERT_EQ(0xb000U, info->end); | 
|  | 208 | ASSERT_EQ(0x40U, info->offset); | 
|  | 209 | ASSERT_EQ(PROT_READ | PROT_WRITE, info->flags); | 
|  | 210 | ASSERT_EQ("/system/lib/fake4.so", info->name); | 
|  | 211 |  | 
|  | 212 | info = maps.Find(0xe500); | 
|  | 213 | ASSERT_TRUE(info != nullptr); | 
|  | 214 | ASSERT_EQ(0xe000U, info->start); | 
|  | 215 | ASSERT_EQ(0xf000U, info->end); | 
|  | 216 | ASSERT_EQ(0x50U, info->offset); | 
|  | 217 | ASSERT_EQ(PROT_READ | PROT_WRITE | PROT_EXEC, info->flags); | 
|  | 218 | ASSERT_EQ("/system/lib/fake5.so", info->name); | 
|  | 219 | } |