1 // Copyright 2019 The Chromium Authors
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 "base/android/orderfile/orderfile_instrumentation.h"
6
7 #include <thread>
8
9 #include "base/android/library_loader/anchor_functions.h"
10 #include "base/strings/stringprintf.h"
11 #include "base/time/time.h"
12 #include "testing/gtest/include/gtest/gtest.h"
13 #include "testing/perf/perf_test.h"
14
15 namespace base {
16 namespace android {
17 namespace orderfile {
18
19 namespace {
20 const size_t kStep = sizeof(int);
21
CallRecordAddress(int iterations,size_t addr_count)22 void CallRecordAddress(int iterations, size_t addr_count) {
23 for (int i = 0; i < iterations; i++) {
24 for (size_t caller_addr = kStartOfTextForTesting + kStep;
25 caller_addr < addr_count; caller_addr += kStep) {
26 for (size_t callee_addr = caller_addr + kStep; callee_addr < addr_count;
27 callee_addr += kStep) {
28 RecordAddressForTesting(callee_addr, caller_addr);
29 }
30 }
31 }
32 }
33
RunBenchmark(size_t iterations,size_t addresses_count,int threads)34 void RunBenchmark(size_t iterations, size_t addresses_count, int threads) {
35 ResetForTesting();
36 auto iterate = [iterations, addresses_count]() {
37 CallRecordAddress(iterations, addresses_count);
38 };
39 if (threads != 1) {
40 for (int i = 0; i < threads - 1; ++i)
41 std::thread(iterate).detach();
42 }
43 auto tick = base::TimeTicks::Now();
44 iterate();
45 auto tock = base::TimeTicks::Now();
46 double nanos = static_cast<double>((tock - tick).InNanoseconds());
47 size_t addresses = (addresses_count - kStartOfTextForTesting - 1) / kStep;
48 double calls_count = (addresses * (addresses - 1)) / 2;
49 auto ns_per_call = nanos / (iterations * calls_count);
50 auto modifier =
51 base::StringPrintf("_%zu_%zu_%d", iterations, addresses_count, threads);
52 perf_test::PrintResult("RecordAddressCostPerCall", modifier, "", ns_per_call,
53 "ns", true);
54 }
55
CheckValid(size_t iterations,size_t addr_count)56 void CheckValid(size_t iterations, size_t addr_count) {
57 // |reached| is expected to be ordered by callee offset
58 auto reached = GetOrderedOffsetsForTesting();
59 size_t buckets_per_callee = 9; // kTotalBuckets * 2 + 1.
60 size_t callers_per_callee = 3;
61 size_t addresses = (addr_count - kStartOfTextForTesting - 1) / kStep;
62 EXPECT_EQ((addresses - 1) * buckets_per_callee, reached.size());
63 size_t expected_callee = kStartOfTextForTesting + 2 * kStep;
64
65 for (size_t i = 0; i < reached.size(); i += buckets_per_callee) {
66 EXPECT_EQ(reached[i] / 4, (expected_callee - kStartOfTextForTesting) / 4);
67 size_t callee_index = i / buckets_per_callee;
68 for (size_t j = 0; j < callers_per_callee; j++) {
69 EXPECT_EQ(reached[i + j * 2 + 1],
70 j > callee_index ? 0UL : (j + 1) * kStep);
71 EXPECT_EQ(reached[i + j * 2 + 2], j > callee_index ? 0UL : iterations);
72 }
73 size_t misses = callee_index > 2 ? (callee_index - 2) * iterations : 0UL;
74 EXPECT_EQ(reached[i + 7], 0UL);
75 EXPECT_EQ(reached[i + 8], misses);
76 expected_callee += kStep;
77 }
78 }
79
80 } // namespace
81
82 class OrderfileInstrumentationTest : public ::testing::Test {
83 // Any tests need to run ResetForTesting() when they start. Because this
84 // perftest is built with instrumentation enabled, all code including
85 // ::testing::Test is instrumented. If ResetForTesting() is called earlier,
86 // for example in setUp(), any test harness code between setUp() and the
87 // actual test will change the instrumentation offset record in unpredictable
88 // ways and make these tests unreliable.
89 };
90
TEST_F(OrderfileInstrumentationTest,SequentialTest_10_5000)91 TEST_F(OrderfileInstrumentationTest, SequentialTest_10_5000) {
92 size_t iterations = 10;
93 size_t addr_count = 5000;
94 ResetForTesting();
95 CallRecordAddress(iterations, addr_count);
96 Disable();
97 CheckValid(iterations, addr_count);
98 }
99
TEST_F(OrderfileInstrumentationTest,SequentialTest_10_10000)100 TEST_F(OrderfileInstrumentationTest, SequentialTest_10_10000) {
101 size_t iterations = 10;
102 size_t addr_count = 10000;
103 ResetForTesting();
104 CallRecordAddress(iterations, addr_count);
105 Disable();
106 CheckValid(iterations, addr_count);
107 }
108
TEST_F(OrderfileInstrumentationTest,OutOfBoundsCaller)109 TEST_F(OrderfileInstrumentationTest, OutOfBoundsCaller) {
110 ResetForTesting();
111 RecordAddressForTesting(1234, kStartOfTextForTesting);
112 RecordAddressForTesting(1234, kEndOfTextForTesting + 1);
113 Disable();
114 auto reached = GetOrderedOffsetsForTesting();
115 EXPECT_EQ(reached.size(), 9UL);
116 EXPECT_EQ(reached[0] / 4, (1234 - kStartOfTextForTesting) / 4);
117 for (size_t i = 1; i < 8; i++) {
118 EXPECT_EQ(reached[i], 0UL);
119 }
120 EXPECT_EQ(reached[8], 2UL);
121 }
122
TEST(OrderfileInstrumentationPerfTest,RecordAddress_10_2000)123 TEST(OrderfileInstrumentationPerfTest, RecordAddress_10_2000) {
124 RunBenchmark(10, 2000, 1);
125 }
126
TEST(OrderfileInstrumentationPerfTest,RecordAddress_100_2000)127 TEST(OrderfileInstrumentationPerfTest, RecordAddress_100_2000) {
128 RunBenchmark(100, 2000, 1);
129 }
130
TEST(OrderfileInstrumentationPerfTest,RecordAddress_1000_2000_2)131 TEST(OrderfileInstrumentationPerfTest, RecordAddress_1000_2000_2) {
132 RunBenchmark(100, 2000, 2);
133 }
134
TEST(OrderfileInstrumentationPerfTest,RecordAddress_1000_2000_3)135 TEST(OrderfileInstrumentationPerfTest, RecordAddress_1000_2000_3) {
136 RunBenchmark(100, 2000, 3);
137 }
138
TEST(OrderfileInstrumentationPerfTest,RecordAddress_1000_2000_4)139 TEST(OrderfileInstrumentationPerfTest, RecordAddress_1000_2000_4) {
140 RunBenchmark(100, 2000, 4);
141 }
142
TEST(OrderfileInstrumentationPerfTest,RecordAddress_1000_2000_6)143 TEST(OrderfileInstrumentationPerfTest, RecordAddress_1000_2000_6) {
144 RunBenchmark(100, 2000, 6);
145 }
146
147 } // namespace orderfile
148 } // namespace android
149 } // namespace base
150
151 // Custom runner implementation since base's one requires JNI on Android.
main(int argc,char ** argv)152 int main(int argc, char** argv) {
153 testing::InitGoogleTest(&argc, argv);
154 return RUN_ALL_TESTS();
155 }
156