1 /*
2 * Copyright (C) 2015 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 <gtest/gtest.h>
18
19 #include <functional>
20 #include <android-base/file.h>
21 #include <android-base/test_utils.h>
22
23 #include "environment.h"
24
TEST(environment,GetCpusFromString)25 TEST(environment, GetCpusFromString) {
26 ASSERT_EQ(GetCpusFromString(""), std::vector<int>());
27 ASSERT_EQ(GetCpusFromString("0-2"), std::vector<int>({0, 1, 2}));
28 ASSERT_EQ(GetCpusFromString("0,2-3"), std::vector<int>({0, 2, 3}));
29 ASSERT_EQ(GetCpusFromString("1,0-3,3,4"), std::vector<int>({0, 1, 2, 3, 4}));
30 }
31
ModulesMatch(const char * p,const char * q)32 static bool ModulesMatch(const char* p, const char* q) {
33 if (p == nullptr && q == nullptr) {
34 return true;
35 }
36 if (p != nullptr && q != nullptr) {
37 return strcmp(p, q) == 0;
38 }
39 return false;
40 }
41
KernelSymbolsMatch(const KernelSymbol & sym1,const KernelSymbol & sym2)42 static bool KernelSymbolsMatch(const KernelSymbol& sym1, const KernelSymbol& sym2) {
43 return sym1.addr == sym2.addr &&
44 sym1.type == sym2.type &&
45 strcmp(sym1.name, sym2.name) == 0 &&
46 ModulesMatch(sym1.module, sym2.module);
47 }
48
TEST(environment,ProcessKernelSymbols)49 TEST(environment, ProcessKernelSymbols) {
50 std::string data =
51 "ffffffffa005c4e4 d __warned.41698 [libsas]\n"
52 "aaaaaaaaaaaaaaaa T _text\n"
53 "cccccccccccccccc c ccccc\n";
54 TemporaryFile tempfile;
55 ASSERT_TRUE(android::base::WriteStringToFile(data, tempfile.path));
56 KernelSymbol expected_symbol;
57 expected_symbol.addr = 0xffffffffa005c4e4ULL;
58 expected_symbol.type = 'd';
59 expected_symbol.name = "__warned.41698";
60 expected_symbol.module = "libsas";
61 ASSERT_TRUE(ProcessKernelSymbols(
62 tempfile.path, std::bind(&KernelSymbolsMatch, std::placeholders::_1, expected_symbol)));
63
64 expected_symbol.addr = 0xaaaaaaaaaaaaaaaaULL;
65 expected_symbol.type = 'T';
66 expected_symbol.name = "_text";
67 expected_symbol.module = nullptr;
68 ASSERT_TRUE(ProcessKernelSymbols(
69 tempfile.path, std::bind(&KernelSymbolsMatch, std::placeholders::_1, expected_symbol)));
70
71 expected_symbol.name = "non_existent_symbol";
72 ASSERT_FALSE(ProcessKernelSymbols(
73 tempfile.path, std::bind(&KernelSymbolsMatch, std::placeholders::_1, expected_symbol)));
74 }
75