12672dea3f1112b13678103023011c797ca283bacYabin Cui/* 22672dea3f1112b13678103023011c797ca283bacYabin Cui * Copyright (C) 2015 The Android Open Source Project 32672dea3f1112b13678103023011c797ca283bacYabin Cui * 42672dea3f1112b13678103023011c797ca283bacYabin Cui * Licensed under the Apache License, Version 2.0 (the "License"); 52672dea3f1112b13678103023011c797ca283bacYabin Cui * you may not use this file except in compliance with the License. 62672dea3f1112b13678103023011c797ca283bacYabin Cui * You may obtain a copy of the License at 72672dea3f1112b13678103023011c797ca283bacYabin Cui * 82672dea3f1112b13678103023011c797ca283bacYabin Cui * http://www.apache.org/licenses/LICENSE-2.0 92672dea3f1112b13678103023011c797ca283bacYabin Cui * 102672dea3f1112b13678103023011c797ca283bacYabin Cui * Unless required by applicable law or agreed to in writing, software 112672dea3f1112b13678103023011c797ca283bacYabin Cui * distributed under the License is distributed on an "AS IS" BASIS, 122672dea3f1112b13678103023011c797ca283bacYabin Cui * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 132672dea3f1112b13678103023011c797ca283bacYabin Cui * See the License for the specific language governing permissions and 142672dea3f1112b13678103023011c797ca283bacYabin Cui * limitations under the License. 152672dea3f1112b13678103023011c797ca283bacYabin Cui */ 162672dea3f1112b13678103023011c797ca283bacYabin Cui 172672dea3f1112b13678103023011c797ca283bacYabin Cui#include <gtest/gtest.h> 182672dea3f1112b13678103023011c797ca283bacYabin Cui 19b032de7a9e1975a838cd6df2e7c8df3c7f70a3ceYabin Cui#include "sample_tree.h" 20b032de7a9e1975a838cd6df2e7c8df3c7f70a3ceYabin Cui 212672dea3f1112b13678103023011c797ca283bacYabin Cuistruct ExpectedSampleInMap { 222672dea3f1112b13678103023011c797ca283bacYabin Cui int pid; 232672dea3f1112b13678103023011c797ca283bacYabin Cui int tid; 2441d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui const char* comm; 2541d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui std::string dso_name; 262672dea3f1112b13678103023011c797ca283bacYabin Cui uint64_t map_start_addr; 272672dea3f1112b13678103023011c797ca283bacYabin Cui size_t sample_count; 282672dea3f1112b13678103023011c797ca283bacYabin Cui}; 292672dea3f1112b13678103023011c797ca283bacYabin Cui 302672dea3f1112b13678103023011c797ca283bacYabin Cuistatic void SampleMatchExpectation(const SampleEntry& sample, const ExpectedSampleInMap& expected, 312672dea3f1112b13678103023011c797ca283bacYabin Cui bool* has_error) { 322672dea3f1112b13678103023011c797ca283bacYabin Cui *has_error = true; 3341d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui ASSERT_TRUE(sample.thread != nullptr); 3441d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui ASSERT_EQ(expected.pid, sample.thread->pid); 3541d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui ASSERT_EQ(expected.tid, sample.thread->tid); 3641d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui ASSERT_STREQ(expected.comm, sample.thread_comm); 37ec12ed9010128483993a87d68edc02d3a89d56cfYabin Cui ASSERT_TRUE(sample.map != nullptr); 38c84856093e8bf4350d30fc521dc0f1c800c5270bYabin Cui ASSERT_EQ(expected.dso_name, sample.map->dso->Path()); 39ec12ed9010128483993a87d68edc02d3a89d56cfYabin Cui ASSERT_EQ(expected.map_start_addr, sample.map->start_addr); 402672dea3f1112b13678103023011c797ca283bacYabin Cui ASSERT_EQ(expected.sample_count, sample.sample_count); 412672dea3f1112b13678103023011c797ca283bacYabin Cui *has_error = false; 422672dea3f1112b13678103023011c797ca283bacYabin Cui} 432672dea3f1112b13678103023011c797ca283bacYabin Cui 442672dea3f1112b13678103023011c797ca283bacYabin Cuistatic void CheckSampleCallback(const SampleEntry& sample, 452672dea3f1112b13678103023011c797ca283bacYabin Cui std::vector<ExpectedSampleInMap>& expected_samples, size_t* pos) { 462672dea3f1112b13678103023011c797ca283bacYabin Cui ASSERT_LT(*pos, expected_samples.size()); 472672dea3f1112b13678103023011c797ca283bacYabin Cui bool has_error; 482672dea3f1112b13678103023011c797ca283bacYabin Cui SampleMatchExpectation(sample, expected_samples[*pos], &has_error); 492672dea3f1112b13678103023011c797ca283bacYabin Cui ASSERT_FALSE(has_error) << "Error matching sample at pos " << *pos; 502672dea3f1112b13678103023011c797ca283bacYabin Cui ++*pos; 512672dea3f1112b13678103023011c797ca283bacYabin Cui} 522672dea3f1112b13678103023011c797ca283bacYabin Cui 532672dea3f1112b13678103023011c797ca283bacYabin Cuistatic int CompareSampleFunction(const SampleEntry& sample1, const SampleEntry& sample2) { 5441d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui if (sample1.thread->pid != sample2.thread->pid) { 5541d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui return sample1.thread->pid - sample2.thread->pid; 562672dea3f1112b13678103023011c797ca283bacYabin Cui } 5741d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui if (sample1.thread->tid != sample2.thread->tid) { 5841d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui return sample1.thread->tid - sample2.thread->tid; 592672dea3f1112b13678103023011c797ca283bacYabin Cui } 6041d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui if (strcmp(sample1.thread_comm, sample2.thread_comm) != 0) { 6141d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui return strcmp(sample1.thread_comm, sample2.thread_comm); 6241d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui } 63c84856093e8bf4350d30fc521dc0f1c800c5270bYabin Cui if (sample1.map->dso->Path() != sample2.map->dso->Path()) { 64c84856093e8bf4350d30fc521dc0f1c800c5270bYabin Cui return sample1.map->dso->Path() > sample2.map->dso->Path() ? 1 : -1; 652672dea3f1112b13678103023011c797ca283bacYabin Cui } 66ec12ed9010128483993a87d68edc02d3a89d56cfYabin Cui if (sample1.map->start_addr != sample2.map->start_addr) { 67ec12ed9010128483993a87d68edc02d3a89d56cfYabin Cui return sample1.map->start_addr - sample2.map->start_addr; 682672dea3f1112b13678103023011c797ca283bacYabin Cui } 692672dea3f1112b13678103023011c797ca283bacYabin Cui return 0; 702672dea3f1112b13678103023011c797ca283bacYabin Cui} 712672dea3f1112b13678103023011c797ca283bacYabin Cui 72b47de4af4d9a1ceffa74a148f6e89be4dbb62bcdYabin Cuivoid VisitSampleTree(SampleTree* sample_tree, 73b47de4af4d9a1ceffa74a148f6e89be4dbb62bcdYabin Cui const std::vector<ExpectedSampleInMap>& expected_samples) { 74b47de4af4d9a1ceffa74a148f6e89be4dbb62bcdYabin Cui size_t pos = 0; 75b47de4af4d9a1ceffa74a148f6e89be4dbb62bcdYabin Cui sample_tree->VisitAllSamples( 76b47de4af4d9a1ceffa74a148f6e89be4dbb62bcdYabin Cui std::bind(&CheckSampleCallback, std::placeholders::_1, expected_samples, &pos)); 77b47de4af4d9a1ceffa74a148f6e89be4dbb62bcdYabin Cui ASSERT_EQ(expected_samples.size(), pos); 78b47de4af4d9a1ceffa74a148f6e89be4dbb62bcdYabin Cui} 79b47de4af4d9a1ceffa74a148f6e89be4dbb62bcdYabin Cui 802672dea3f1112b13678103023011c797ca283bacYabin Cuiclass SampleTreeTest : public testing::Test { 812672dea3f1112b13678103023011c797ca283bacYabin Cui protected: 822672dea3f1112b13678103023011c797ca283bacYabin Cui virtual void SetUp() { 8360a0ea96c0fb9e807c899759256df5e20bd904bdYabin Cui thread_tree.AddThread(1, 1, "p1t1"); 8460a0ea96c0fb9e807c899759256df5e20bd904bdYabin Cui thread_tree.AddThread(1, 11, "p1t11"); 8560a0ea96c0fb9e807c899759256df5e20bd904bdYabin Cui thread_tree.AddThread(2, 2, "p2t2"); 8660a0ea96c0fb9e807c899759256df5e20bd904bdYabin Cui thread_tree.AddThreadMap(1, 1, 1, 5, 0, 0, "process1_thread1"); 8760a0ea96c0fb9e807c899759256df5e20bd904bdYabin Cui thread_tree.AddThreadMap(1, 1, 6, 5, 0, 0, "process1_thread1_map2"); 8860a0ea96c0fb9e807c899759256df5e20bd904bdYabin Cui thread_tree.AddThreadMap(1, 11, 1, 10, 0, 0, "process1_thread11"); 8960a0ea96c0fb9e807c899759256df5e20bd904bdYabin Cui thread_tree.AddThreadMap(2, 2, 1, 20, 0, 0, "process2_thread2"); 9060a0ea96c0fb9e807c899759256df5e20bd904bdYabin Cui thread_tree.AddKernelMap(10, 20, 0, 0, "kernel"); 9160a0ea96c0fb9e807c899759256df5e20bd904bdYabin Cui sample_tree = std::unique_ptr<SampleTree>(new SampleTree(&thread_tree, CompareSampleFunction)); 922672dea3f1112b13678103023011c797ca283bacYabin Cui } 932672dea3f1112b13678103023011c797ca283bacYabin Cui 94b47de4af4d9a1ceffa74a148f6e89be4dbb62bcdYabin Cui void VisitSampleTree(const std::vector<ExpectedSampleInMap>& expected_samples) { 95b47de4af4d9a1ceffa74a148f6e89be4dbb62bcdYabin Cui ::VisitSampleTree(sample_tree.get(), expected_samples); 962672dea3f1112b13678103023011c797ca283bacYabin Cui } 972672dea3f1112b13678103023011c797ca283bacYabin Cui 9860a0ea96c0fb9e807c899759256df5e20bd904bdYabin Cui ThreadTree thread_tree; 992672dea3f1112b13678103023011c797ca283bacYabin Cui std::unique_ptr<SampleTree> sample_tree; 1002672dea3f1112b13678103023011c797ca283bacYabin Cui}; 1012672dea3f1112b13678103023011c797ca283bacYabin Cui 1022672dea3f1112b13678103023011c797ca283bacYabin CuiTEST_F(SampleTreeTest, ip_in_map) { 10322ec7fa2032610b6868e4fb8997bb28aee0dea84Yabin Cui sample_tree->AddSample(1, 1, 1, 0, 0, false); 10441d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui sample_tree->AddSample(1, 1, 2, 0, 0, false); 10522ec7fa2032610b6868e4fb8997bb28aee0dea84Yabin Cui sample_tree->AddSample(1, 1, 5, 0, 0, false); 1062672dea3f1112b13678103023011c797ca283bacYabin Cui std::vector<ExpectedSampleInMap> expected_samples = { 10741d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui {1, 1, "p1t1", "process1_thread1", 1, 3}, 1082672dea3f1112b13678103023011c797ca283bacYabin Cui }; 1092672dea3f1112b13678103023011c797ca283bacYabin Cui VisitSampleTree(expected_samples); 1102672dea3f1112b13678103023011c797ca283bacYabin Cui} 1112672dea3f1112b13678103023011c797ca283bacYabin Cui 1122672dea3f1112b13678103023011c797ca283bacYabin CuiTEST_F(SampleTreeTest, different_pid) { 11322ec7fa2032610b6868e4fb8997bb28aee0dea84Yabin Cui sample_tree->AddSample(1, 1, 1, 0, 0, false); 11422ec7fa2032610b6868e4fb8997bb28aee0dea84Yabin Cui sample_tree->AddSample(2, 2, 1, 0, 0, false); 1152672dea3f1112b13678103023011c797ca283bacYabin Cui std::vector<ExpectedSampleInMap> expected_samples = { 11641d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui {1, 1, "p1t1", "process1_thread1", 1, 1}, {2, 2, "p2t2", "process2_thread2", 1, 1}, 1172672dea3f1112b13678103023011c797ca283bacYabin Cui }; 1182672dea3f1112b13678103023011c797ca283bacYabin Cui VisitSampleTree(expected_samples); 1192672dea3f1112b13678103023011c797ca283bacYabin Cui} 1202672dea3f1112b13678103023011c797ca283bacYabin Cui 1212672dea3f1112b13678103023011c797ca283bacYabin CuiTEST_F(SampleTreeTest, different_tid) { 12222ec7fa2032610b6868e4fb8997bb28aee0dea84Yabin Cui sample_tree->AddSample(1, 1, 1, 0, 0, false); 12322ec7fa2032610b6868e4fb8997bb28aee0dea84Yabin Cui sample_tree->AddSample(1, 11, 1, 0, 0, false); 1242672dea3f1112b13678103023011c797ca283bacYabin Cui std::vector<ExpectedSampleInMap> expected_samples = { 12541d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui {1, 1, "p1t1", "process1_thread1", 1, 1}, {1, 11, "p1t11", "process1_thread11", 1, 1}, 12641d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui }; 12741d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui VisitSampleTree(expected_samples); 12841d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui} 12941d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui 13041d4ba9f6f2781155a0519a784606d5382cda88fYabin CuiTEST_F(SampleTreeTest, different_comm) { 13141d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui sample_tree->AddSample(1, 1, 1, 0, 0, false); 13260a0ea96c0fb9e807c899759256df5e20bd904bdYabin Cui thread_tree.AddThread(1, 1, "p1t1_comm2"); 13341d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui sample_tree->AddSample(1, 1, 1, 0, 0, false); 13441d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui std::vector<ExpectedSampleInMap> expected_samples = { 13541d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui {1, 1, "p1t1", "process1_thread1", 1, 1}, {1, 1, "p1t1_comm2", "process1_thread1", 1, 1}, 1362672dea3f1112b13678103023011c797ca283bacYabin Cui }; 1372672dea3f1112b13678103023011c797ca283bacYabin Cui VisitSampleTree(expected_samples); 1382672dea3f1112b13678103023011c797ca283bacYabin Cui} 1392672dea3f1112b13678103023011c797ca283bacYabin Cui 1402672dea3f1112b13678103023011c797ca283bacYabin CuiTEST_F(SampleTreeTest, different_map) { 14122ec7fa2032610b6868e4fb8997bb28aee0dea84Yabin Cui sample_tree->AddSample(1, 1, 1, 0, 0, false); 14241d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui sample_tree->AddSample(1, 1, 6, 0, 0, false); 1432672dea3f1112b13678103023011c797ca283bacYabin Cui std::vector<ExpectedSampleInMap> expected_samples = { 14441d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui {1, 1, "p1t1", "process1_thread1", 1, 1}, {1, 1, "p1t1", "process1_thread1_map2", 6, 1}, 1452672dea3f1112b13678103023011c797ca283bacYabin Cui }; 1462672dea3f1112b13678103023011c797ca283bacYabin Cui VisitSampleTree(expected_samples); 1472672dea3f1112b13678103023011c797ca283bacYabin Cui} 1482672dea3f1112b13678103023011c797ca283bacYabin Cui 1492672dea3f1112b13678103023011c797ca283bacYabin CuiTEST_F(SampleTreeTest, unmapped_sample) { 15022ec7fa2032610b6868e4fb8997bb28aee0dea84Yabin Cui sample_tree->AddSample(1, 1, 0, 0, 0, false); 15122ec7fa2032610b6868e4fb8997bb28aee0dea84Yabin Cui sample_tree->AddSample(1, 1, 31, 0, 0, false); 15222ec7fa2032610b6868e4fb8997bb28aee0dea84Yabin Cui sample_tree->AddSample(1, 1, 70, 0, 0, false); 1532672dea3f1112b13678103023011c797ca283bacYabin Cui // Match the unknown map. 1542672dea3f1112b13678103023011c797ca283bacYabin Cui std::vector<ExpectedSampleInMap> expected_samples = { 15541d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui {1, 1, "p1t1", "unknown", 0, 3}, 1562672dea3f1112b13678103023011c797ca283bacYabin Cui }; 1572672dea3f1112b13678103023011c797ca283bacYabin Cui VisitSampleTree(expected_samples); 1582672dea3f1112b13678103023011c797ca283bacYabin Cui} 1592672dea3f1112b13678103023011c797ca283bacYabin Cui 1602672dea3f1112b13678103023011c797ca283bacYabin CuiTEST_F(SampleTreeTest, map_kernel) { 16141d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui sample_tree->AddSample(1, 1, 10, 0, 0, true); 16241d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui sample_tree->AddSample(1, 1, 10, 0, 0, false); 1632672dea3f1112b13678103023011c797ca283bacYabin Cui std::vector<ExpectedSampleInMap> expected_samples = { 16441d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui {1, 1, "p1t1", "kernel", 10, 1}, {1, 1, "p1t1", "process1_thread1_map2", 6, 1}, 1652672dea3f1112b13678103023011c797ca283bacYabin Cui }; 1662672dea3f1112b13678103023011c797ca283bacYabin Cui VisitSampleTree(expected_samples); 1672672dea3f1112b13678103023011c797ca283bacYabin Cui} 168b47de4af4d9a1ceffa74a148f6e89be4dbb62bcdYabin Cui 169b47de4af4d9a1ceffa74a148f6e89be4dbb62bcdYabin CuiTEST(sample_tree, overlapped_map) { 17060a0ea96c0fb9e807c899759256df5e20bd904bdYabin Cui ThreadTree thread_tree; 17160a0ea96c0fb9e807c899759256df5e20bd904bdYabin Cui SampleTree sample_tree(&thread_tree, CompareSampleFunction); 17260a0ea96c0fb9e807c899759256df5e20bd904bdYabin Cui thread_tree.AddThread(1, 1, "thread1"); 17360a0ea96c0fb9e807c899759256df5e20bd904bdYabin Cui thread_tree.AddThreadMap(1, 1, 1, 10, 0, 0, "map1"); // Add map 1. 17460a0ea96c0fb9e807c899759256df5e20bd904bdYabin Cui sample_tree.AddSample(1, 1, 5, 0, 0, false); // Hit map 1. 17560a0ea96c0fb9e807c899759256df5e20bd904bdYabin Cui thread_tree.AddThreadMap(1, 1, 5, 20, 0, 0, "map2"); // Add map 2. 17660a0ea96c0fb9e807c899759256df5e20bd904bdYabin Cui sample_tree.AddSample(1, 1, 6, 0, 0, false); // Hit map 2. 177b7f481f59126456b0e708a76f40fa88224f3531bYabin Cui sample_tree.AddSample(1, 1, 4, 0, 0, false); // Hit map 1. 17860a0ea96c0fb9e807c899759256df5e20bd904bdYabin Cui thread_tree.AddThreadMap(1, 1, 2, 7, 0, 0, "map3"); // Add map 3. 17960a0ea96c0fb9e807c899759256df5e20bd904bdYabin Cui sample_tree.AddSample(1, 1, 7, 0, 0, false); // Hit map 3. 180b7f481f59126456b0e708a76f40fa88224f3531bYabin Cui sample_tree.AddSample(1, 1, 10, 0, 0, false); // Hit map 2. 181b47de4af4d9a1ceffa74a148f6e89be4dbb62bcdYabin Cui 182b47de4af4d9a1ceffa74a148f6e89be4dbb62bcdYabin Cui std::vector<ExpectedSampleInMap> expected_samples = { 183b7f481f59126456b0e708a76f40fa88224f3531bYabin Cui {1, 1, "thread1", "map1", 1, 2}, 18441d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui {1, 1, "thread1", "map2", 5, 1}, 185b7f481f59126456b0e708a76f40fa88224f3531bYabin Cui {1, 1, "thread1", "map2", 9, 1}, 18641d4ba9f6f2781155a0519a784606d5382cda88fYabin Cui {1, 1, "thread1", "map3", 2, 1}, 187b47de4af4d9a1ceffa74a148f6e89be4dbb62bcdYabin Cui }; 18860a0ea96c0fb9e807c899759256df5e20bd904bdYabin Cui VisitSampleTree(&sample_tree, expected_samples); 189b47de4af4d9a1ceffa74a148f6e89be4dbb62bcdYabin Cui} 190