xref: /developtools/hiperf/src/main.cpp (revision 48f512ce) 
1/*
2 * Copyright (c) 2021 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 *     http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16#include <cstdio>
17#include <iostream>
18
19#include "command.h"
20#include "debug_logger.h"
21#if defined(is_ohos) && is_ohos
22#include "hiperf_hilog.h"
23#endif
24#include "option_debug.h"
25#include "subcommand.h"
26#include "subcommand_help.h"
27#include "utilities.h"
28#if SUPPORT_PERF_EVENT
29#include "subcommand_list.h"
30#include "subcommand_record.h"
31#include "subcommand_stat.h"
32#endif
33#include "subcommand_dump.h"
34#include "subcommand_report.h"
35
36using namespace std;
37using namespace OHOS::Developtools::HiPerf;
38
39#ifdef FUZZER_TEST
40#define main HiperfFuzzerMain
41#endif
42
43int main(const int argc, const char *argv[])
44{
45    if (!GetDeveloperMode() && !IsAllowProfilingUid()) {
46        printf("error: not in developermode, exit.\n");
47        return -1;
48    }
49
50    if (argc < 0 || argc > 128) { // 128 : max input argument counts
51        printf("The number of input arguments exceeds the upper limit.\n");
52        return -1;
53    }
54    std::ios::sync_with_stdio(false);
55    cin.tie(nullptr);
56
57#if defined(is_ohos) && is_ohos
58    if (IsRoot() && setgid(2000) != 0) { // 2000 is shell group
59        printf("setgid failed errno: %d.\n", errno);
60    }
61    WriteStringToFile("/proc/self/oom_score_adj", "0");
62    if (signal(SIGPIPE, SIG_IGN) == SIG_ERR) {
63        HIPERF_HILOGI(MODULE_DEFAULT, "ignore SIGPIPE failed.");
64    }
65    HIPERF_HILOGI(MODULE_DEFAULT, "hiperf start.");
66#endif
67
68    // pass the argv to next
69    vector<string> args;
70    for (int i = 1; i < argc; i++) {
71        args.push_back(argv[i]);
72    }
73
74    if (args.empty()) {
75        printf("no command input.\n");
76        args.push_back("help"); // no cmd like help cmd
77    }
78
79// register all the main command
80#ifdef HIPERF_DEBUG
81    RegisterMainCommandDebug();
82#endif
83
84    // register all the sub command
85    SubCommandHelp::RegisterSubCommandHelp();
86
87#if SUPPORT_PERF_EVENT
88    RegisterSubCommandStat();
89    SubCommandList::RegisterSubCommandList();
90    SubCommandRecord::RegisterSubCommandRecord();
91#endif
92
93    SubCommandDump::RegisterSubCommandDump();
94    SubCommandReport::RegisterSubCommandReport();
95#ifdef FUZZER_TEST
96    bool isRecordCmd = false;
97    if (args[0] == "record") {
98        isRecordCmd = true;
99    }
100#endif
101    // tell sub cmd to do the process
102    Command::DispatchCommands(args);
103
104#ifdef FUZZER_TEST
105    SubCommand::ClearSubCommands();
106    Option::ClearMainOptions();
107
108    if (isRecordCmd) {
109        const uint64_t msWaitSysReleaseThread = 100;
110        std::this_thread::sleep_for(std::chrono::milliseconds(msWaitSysReleaseThread));
111    }
112#endif
113
114    HLOGD("normal exit.");
115    return 0;
116}
117