Home | History | Annotate | Download | only in simpleperf 
      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 <gtest/gtest.h>
     18 
     19 #include <android-base/file.h>
     20 #include <android-base/strings.h>
     21 #include <android-base/test_utils.h>
     22 
     23 #include <memory>
     24 
     25 #include "command.h"
     26 #include "get_test_data.h"
     27 #include "record.h"
     28 #include "record_file.h"
     29 #include "test_util.h"
     30 
     31 static std::unique_ptr<Command> KmemCmd() {
     32   return CreateCommandInstance("kmem");
     33 }
     34 
     35 struct ReportResult {
     36   bool success;
     37   std::string content;
     38   std::vector<std::string> lines;
     39 };
     40 
     41 static void KmemReportRawFile(const std::string& perf_data,
     42                               const std::vector<std::string>& additional_args,
     43                               ReportResult* result) {
     44   result->success = false;
     45   TemporaryFile tmp_file;
     46   std::vector<std::string> args = {"report", "-i", perf_data, "-o",
     47                                    tmp_file.path};
     48   args.insert(args.end(), additional_args.begin(), additional_args.end());
     49   ASSERT_TRUE(KmemCmd()->Run(args));
     50   ASSERT_TRUE(android::base::ReadFileToString(tmp_file.path, &result->content));
     51   ASSERT_TRUE(!result->content.empty());
     52   std::vector<std::string> raw_lines =
     53       android::base::Split(result->content, "\n");
     54   result->lines.clear();
     55   for (const auto& line : raw_lines) {
     56     std::string s = android::base::Trim(line);
     57     if (!s.empty()) {
     58       result->lines.push_back(s);
     59     }
     60   }
     61   ASSERT_GE(result->lines.size(), 2u);
     62   result->success = true;
     63 }
     64 
     65 static void KmemReportFile(const std::string& perf_data,
     66                            const std::vector<std::string>& additional_args,
     67                            ReportResult* result) {
     68   KmemReportRawFile(GetTestData(perf_data), additional_args, result);
     69 }
     70 
     71 #if defined(__linux__)
     72 #include "environment.h"
     73 
     74 static bool RunKmemRecordCmd(std::vector<std::string> v,
     75                              const char* output_file = nullptr) {
     76   std::unique_ptr<TemporaryFile> tmpfile;
     77   std::string out_file;
     78   if (output_file != nullptr) {
     79     out_file = output_file;
     80   } else {
     81     tmpfile.reset(new TemporaryFile);
     82     out_file = tmpfile->path;
     83   }
     84   v.insert(v.begin(), "record");
     85   v.insert(v.end(), {"-o", out_file, "sleep", SLEEP_SEC});
     86   return KmemCmd()->Run(v);
     87 }
     88 
     89 TEST(kmem_cmd, record_slab) {
     90   TEST_IN_ROOT(ASSERT_TRUE(RunKmemRecordCmd({"--slab"})));
     91 }
     92 
     93 TEST(kmem_cmd, record_fp_callchain_sampling) {
     94   TEST_IN_ROOT(ASSERT_TRUE(RunKmemRecordCmd({"--slab", "-g"})));
     95   TEST_IN_ROOT(ASSERT_TRUE(RunKmemRecordCmd({"--slab", "--call-graph", "fp"})));
     96 }
     97 
     98 TEST(kmem_cmd, record_and_report) {
     99   TemporaryFile tmp_file;
    100   TEST_IN_ROOT({
    101     ASSERT_TRUE(RunKmemRecordCmd({"--slab"}, tmp_file.path));
    102     ReportResult result;
    103     KmemReportRawFile(tmp_file.path, {}, &result);
    104     ASSERT_TRUE(result.success);
    105   });
    106 }
    107 
    108 TEST(kmem_cmd, record_and_report_callgraph) {
    109   TemporaryFile tmp_file;
    110   TEST_IN_ROOT({
    111     ASSERT_TRUE(RunKmemRecordCmd({"--slab", "-g"}, tmp_file.path));
    112     ReportResult result;
    113     KmemReportRawFile(tmp_file.path, {"-g"}, &result);
    114     ASSERT_TRUE(result.success);
    115   });
    116 }
    117 
    118 #endif
    119 
    120 TEST(kmem_cmd, report) {
    121   ReportResult result;
    122   KmemReportFile(PERF_DATA_WITH_KMEM_SLAB_CALLGRAPH_RECORD, {}, &result);
    123   ASSERT_TRUE(result.success);
    124   ASSERT_NE(result.content.find("kmem:kmalloc"), std::string::npos);
    125   ASSERT_NE(result.content.find("__alloc_skb"), std::string::npos);
    126 }
    127 
    128 TEST(kmem_cmd, report_all_sort_options) {
    129   ReportResult result;
    130   KmemReportFile(
    131       PERF_DATA_WITH_KMEM_SLAB_CALLGRAPH_RECORD,
    132       {"--slab-sort",
    133        "hit,caller,ptr,bytes_req,bytes_alloc,fragment,gfp_flags,pingpong"},
    134       &result);
    135   ASSERT_TRUE(result.success);
    136   ASSERT_NE(result.content.find("Ptr"), std::string::npos);
    137   ASSERT_NE(result.content.find("GfpFlags"), std::string::npos);
    138 }
    139 
    140 TEST(kmem_cmd, report_callgraph) {
    141   ReportResult result;
    142   KmemReportFile(PERF_DATA_WITH_KMEM_SLAB_CALLGRAPH_RECORD, {"-g"}, &result);
    143   ASSERT_TRUE(result.success);
    144   ASSERT_NE(result.content.find("kmem:kmalloc"), std::string::npos);
    145   ASSERT_NE(result.content.find("__alloc_skb"), std::string::npos);
    146   ASSERT_NE(result.content.find("system_call_fastpath"), std::string::npos);
    147 }
    148