+++ /dev/null
-//
-// main.cpp
-// msa
-//
-// Created by James McIlree on 1/30/14.
-// Copyright (c) 2014 Apple. All rights reserved.
-//
-
-#include <CPPUtil/CPPUtil.h>
-
-#include "global.h"
-
-bool isVerbose = true;
-bool shouldPrintVersion = true;
-std::vector<std::string> procsOfInterest;
-bool interestedInEverything = false;
-
-__attribute__((noreturn)) void usage(const char *errorMsg) {
- if (errorMsg) {
- printf("%s\n", errorMsg);
- exit(1);
- }
-
- // const char* BOLD = "\033[1m";
- // const char* UNBOLD = "\033[0m";
-
- // printf("01234567890123456789012345678901234567890123456789012345678901234567890123456789\n");
- printf("msa [options]\n\n");
- printf(" GLOBAL OPTIONS\n\n");
- printf(" -h, --help Print this message\n");
- printf(" --verbose Print additional information\n");
- printf(" --version Print version info\n");
- printf("\n");
- printf(" TRACE COLLECTION OPTIONS\n\n");
- printf(" -i, --initialize # Set the size of the kernel trace buffer\n");
- printf(" --no-voucher-contents Disable collecting voucher contents\n");
- printf(" -L path Capture and save trace output to path\n");
- printf("\n");
- printf(" OUTPUT OPTIONS\n\n");
- printf(" --lifecycle all|user|none\n");
- printf(" Set filter level for lifecycle events\n");
- printf(" --mach-msg all|user|voucher|none\n");
- printf(" Set filter level for mach msg events\n");
- printf(" -o, --output path Print output to path\n");
- printf(" --raw-timestamps Print timestamps as raw values, not deltas\n");
- printf(" --mach-absolute-time Print timestamps in mach absolute time\n");
- printf(" --event-index Print the index of each event\n");
- printf("\n");
- exit(1);
-}
-
-template <typename SIZE>
-static bool check_interest_name(const MachineProcess<SIZE>& process) {
- if (interestedInEverything)
- return true;
-
- const char* name = process.name();
- for (auto& proc : procsOfInterest) {
- if (strcmp(name, proc.c_str()) == 0)
- return true;
- }
-
- return false;
-}
-
-static std::unique_ptr<Action> create_read_trace_file_action(const char* trace_file_path) {
- if (Path::is_file(trace_file_path, true)) {
- char resolved_path[PATH_MAX];
- if (realpath(trace_file_path, resolved_path)) {
- return std::make_unique<ReadTraceFileAction>(resolved_path);
- }
- }
- char* errmsg = NULL;
- asprintf(&errmsg, "%s does not exist or is not a file", trace_file_path);
- usage(errmsg);
-}
-
-static std::vector<std::unique_ptr<Action>> parse_arguments(int argc, const char* argv[], Globals& globals) {
- int i = 1;
-
- std::vector<std::unique_ptr<Action>> actions;
-
- while (i < argc) {
- const char* arg = argv[i];
- if ((strcmp(arg, "-h") == 0) || (strcasecmp(arg, "--help") == 0)) {
- usage(NULL);
- } else if ((strcmp(arg, "-v") == 0) || strcasecmp(arg, "--verbose") == 0) {
- globals.set_is_verbose(true);
- } else if (strcasecmp(arg, "--version") == 0) {
- shouldPrintVersion = true;
- } else if ((strcmp(arg, "-i") == 0) || strcasecmp(arg, "--initialize") == 0) {
- if (++i >= argc)
- usage("--initialize requires an argument");
-
- arg = argv[i];
- char* endptr;
- uint32_t temp = (uint32_t)strtoul(arg, &endptr, 0);
- if (*endptr == 0) {
- globals.set_trace_buffer_size(temp);
- } else {
- usage("Unable to parse --initialize argument");
- }
- } else if (strcasecmp(arg, "--no-voucher-contents") == 0) {
- globals.set_should_trace_voucher_contents(false);
- } else if (strcasecmp(arg, "-L") == 0) {
- if (++i >= argc)
- usage("-L requires an argument");
-
- arg = argv[i];
- actions.push_back(std::make_unique<WriteTraceFileAction>(arg));
- } else if (strcasecmp(arg, "--lifecycle") == 0) {
- if (++i >= argc)
- usage("--lifecycle requires an argument");
-
- arg = argv[i];
- if (strcasecmp(arg, "all") == 0) {
- globals.set_lifecycle_filter(kLifecycleFilter::All);
- } else if (strcasecmp(arg, "user") == 0) {
- globals.set_lifecycle_filter(kLifecycleFilter::User);
- } else if (strcasecmp(arg, "none") == 0) {
- globals.set_lifecycle_filter(kLifecycleFilter::None);
- } else {
- usage("Unrecognized --lifecycle value");
- }
- } else if (strcasecmp(arg, "--mach-msg") == 0) {
- if (++i >= argc)
- usage("--mach-msg requires an argument");
-
- arg = argv[i];
- if (strcasecmp(arg, "all") == 0) {
- globals.set_mach_msg_filter(kMachMsgFilter::All);
- } else if (strcasecmp(arg, "user") == 0) {
- globals.set_mach_msg_filter(kMachMsgFilter::User);
- } else if (strcasecmp(arg, "voucher") == 0) {
- globals.set_mach_msg_filter(kMachMsgFilter::Voucher);
- } else if (strcasecmp(arg, "none") == 0) {
- globals.set_mach_msg_filter(kMachMsgFilter::None);
- } else {
- usage("Unrecognized --mach-msg value");
- }
- } else if ((strcmp(arg, "-o") == 0) || strcasecmp(arg, "--output") == 0) {
- if (++i >= argc)
- usage("--output requires an argument");
-
- FileDescriptor desc(argv[i], O_WRONLY | O_CREAT | O_TRUNC, 0644);
- if (!desc.is_open()) {
- char* errmsg = NULL;
- asprintf(&errmsg, "Unable to create output file at %s", argv[i]);
- usage(errmsg);
- }
- globals.set_output_fd(std::move(desc));
- } else if (strcasecmp(arg, "--raw-timestamps") == 0) {
- globals.set_should_zero_base_timestamps(false);
- } else if (strcasecmp(arg, "--mach-absolute-time") == 0) {
- globals.set_should_print_mach_absolute_timestamps(true);
- } else if (strcasecmp(arg, "--event-index") == 0) {
- globals.set_should_print_event_index(true);
- } else {
- //
- // Last attempts to divine argument type/intent.
- //
- std::string temp(arg);
-
- if (ends_with(temp, ".trace")) {
- actions.push_back(create_read_trace_file_action(argv[i]));
- goto no_error;
- }
-
- //
- // ERROR!
- //
- char error_buffer[PATH_MAX];
- snprintf(error_buffer, sizeof(error_buffer), "Unhandled argument: %s", arg);
- usage(error_buffer);
- }
-
- no_error:
-
- i++;
- }
-
- if (actions.empty()) {
- actions.push_back(std::make_unique<LiveTraceAction>());
- }
-
- return actions;
-}
-
-int main(int argc, const char * argv[])
-{
- //
- // Use host values as defaults.
- // User overrides as needed via flags.
- //
- Globals globals;
- auto actions = parse_arguments(argc, argv, globals);
-
- interestedInEverything = procsOfInterest.empty();
-
- // globals.set_should_print_mach_absolute_timestamps(true);
-
- for (auto& action : actions) {
- action->execute(globals);
- }
-
- return 0;
-}
-
projects / apple / system_cmds.git / blobdiff