1e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci/* 2e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci * Copyright (C) 2018 The Android Open Source Project 3e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci * 4e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci * Licensed under the Apache License, Version 2.0 (the "License"); 5e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci * you may not use this file except in compliance with the License. 6e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci * You may obtain a copy of the License at 7e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci * 8e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci * http://www.apache.org/licenses/LICENSE-2.0 9e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci * 10e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci * Unless required by applicable law or agreed to in writing, software 11e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci * distributed under the License is distributed on an "AS IS" BASIS, 12e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci * See the License for the specific language governing permissions and 14e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci * limitations under the License. 15e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci */ 16e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci 17484524a246ffe453f8cd89b698a279c23b0bde1fTej Singh#define DEBUG false // STOPSHIP if true 18d0dd6c7cefb23e0f521aac533087281be2c30bddSami Kyostila#include "config/ConfigKey.h" 19e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci#include "Log.h" 20e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci 21e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci#include "frameworks/base/cmds/statsd/src/statsd_config.pb.h" // Alert 22e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci 23e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci#include <android-base/unique_fd.h> 24e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci#include <errno.h> 25a74556061ab5257c39a0ba752410da8fbdadd175Primiano Tucci#include <fcntl.h> 26d0dd6c7cefb23e0f521aac533087281be2c30bddSami Kyostila#include <inttypes.h> 27e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci#include <sys/types.h> 28e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci#include <sys/wait.h> 29e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci#include <unistd.h> 30e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci 31e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci#include <string> 32e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci 33e4d44919aa90f16285a9966dcc632c128b932374Primiano Tuccinamespace { 34e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucciconst char kDropboxTag[] = "perfetto"; 35e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci} 36e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci 37e4d44919aa90f16285a9966dcc632c128b932374Primiano Tuccinamespace android { 38e4d44919aa90f16285a9966dcc632c128b932374Primiano Tuccinamespace os { 39e4d44919aa90f16285a9966dcc632c128b932374Primiano Tuccinamespace statsd { 40e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci 41d0dd6c7cefb23e0f521aac533087281be2c30bddSami Kyostilabool CollectPerfettoTraceAndUploadToDropbox(const PerfettoDetails& config, 42d0dd6c7cefb23e0f521aac533087281be2c30bddSami Kyostila int64_t alert_id, 43d0dd6c7cefb23e0f521aac533087281be2c30bddSami Kyostila const ConfigKey& configKey) { 44484524a246ffe453f8cd89b698a279c23b0bde1fTej Singh VLOG("Starting trace collection through perfetto"); 45e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci 46e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci if (!config.has_trace_config()) { 47e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci ALOGE("The perfetto trace config is empty, aborting"); 48e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci return false; 49e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci } 50e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci 51d0dd6c7cefb23e0f521aac533087281be2c30bddSami Kyostila char alertId[20]; 52d0dd6c7cefb23e0f521aac533087281be2c30bddSami Kyostila char configId[20]; 53d0dd6c7cefb23e0f521aac533087281be2c30bddSami Kyostila char configUid[20]; 54d0dd6c7cefb23e0f521aac533087281be2c30bddSami Kyostila snprintf(alertId, sizeof(alertId), "%" PRId64, alert_id); 55d0dd6c7cefb23e0f521aac533087281be2c30bddSami Kyostila snprintf(configId, sizeof(configId), "%" PRId64, configKey.GetId()); 56d0dd6c7cefb23e0f521aac533087281be2c30bddSami Kyostila snprintf(configUid, sizeof(configUid), "%d", configKey.GetUid()); 57d0dd6c7cefb23e0f521aac533087281be2c30bddSami Kyostila 58e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci android::base::unique_fd readPipe; 59e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci android::base::unique_fd writePipe; 60e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci if (!android::base::Pipe(&readPipe, &writePipe)) { 61e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci ALOGE("pipe() failed while calling the Perfetto client: %s", strerror(errno)); 62e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci return false; 63e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci } 64e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci 65e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci pid_t pid = fork(); 66e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci if (pid < 0) { 67e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci ALOGE("fork() failed while calling the Perfetto client: %s", strerror(errno)); 68e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci return false; 69e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci } 70e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci 71e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci if (pid == 0) { 72e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci // Child process. 73e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci 74e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci // No malloc calls or library calls after this point. Remember that even 75e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci // ALOGx (aka android_printLog()) can use dynamic memory for vsprintf(). 76e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci 77e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci writePipe.reset(); // Close the write end (owned by the main process). 78e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci 79e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci // Replace stdin with |readPipe| so the main process can write into it. 80e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci if (dup2(readPipe.get(), STDIN_FILENO) < 0) _exit(1); 81a74556061ab5257c39a0ba752410da8fbdadd175Primiano Tucci readPipe.reset(); 82a74556061ab5257c39a0ba752410da8fbdadd175Primiano Tucci 83a74556061ab5257c39a0ba752410da8fbdadd175Primiano Tucci // Replace stdout/stderr with /dev/null and close any other file 84a74556061ab5257c39a0ba752410da8fbdadd175Primiano Tucci // descriptor. This is to avoid SELinux complaining about perfetto 85a74556061ab5257c39a0ba752410da8fbdadd175Primiano Tucci // trying to access files accidentally left open by statsd (i.e. files 86a74556061ab5257c39a0ba752410da8fbdadd175Primiano Tucci // that have been opened without the O_CLOEXEC flag). 87a74556061ab5257c39a0ba752410da8fbdadd175Primiano Tucci int devNullFd = open("/dev/null", O_RDWR | O_CLOEXEC); 88a74556061ab5257c39a0ba752410da8fbdadd175Primiano Tucci if (dup2(devNullFd, STDOUT_FILENO) < 0) _exit(2); 89a74556061ab5257c39a0ba752410da8fbdadd175Primiano Tucci if (dup2(devNullFd, STDERR_FILENO) < 0) _exit(3); 90a74556061ab5257c39a0ba752410da8fbdadd175Primiano Tucci close(devNullFd); 91a74556061ab5257c39a0ba752410da8fbdadd175Primiano Tucci for (int i = 0; i < 1024; i++) { 92a74556061ab5257c39a0ba752410da8fbdadd175Primiano Tucci if (i != STDIN_FILENO && i != STDOUT_FILENO && i != STDERR_FILENO) close(i); 93a74556061ab5257c39a0ba752410da8fbdadd175Primiano Tucci } 94a74556061ab5257c39a0ba752410da8fbdadd175Primiano Tucci 95e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci execl("/system/bin/perfetto", "perfetto", "--background", "--config", "-", "--dropbox", 96d0dd6c7cefb23e0f521aac533087281be2c30bddSami Kyostila kDropboxTag, "--alert-id", alertId, "--config-id", configId, "--config-uid", 97d0dd6c7cefb23e0f521aac533087281be2c30bddSami Kyostila configUid, nullptr); 98e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci 99a74556061ab5257c39a0ba752410da8fbdadd175Primiano Tucci // execl() doesn't return in case of success, if we get here something 100a74556061ab5257c39a0ba752410da8fbdadd175Primiano Tucci // failed. 101a74556061ab5257c39a0ba752410da8fbdadd175Primiano Tucci _exit(4); 102e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci } 103e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci 104e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci // Main process. 105e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci 106e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci readPipe.reset(); // Close the read end (owned by the child process). 107e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci 108e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci // Using fopen() because fwrite() has the right logic to chunking write() 109e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci // over a pipe (see __sfvwrite()). 110e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci FILE* writePipeStream = fdopen(writePipe.get(), "wb"); 111e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci if (!writePipeStream) { 112e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci ALOGE("fdopen() failed while calling the Perfetto client: %s", strerror(errno)); 113e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci return false; 114e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci } 115e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci 116e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci std::string cfgProto = config.trace_config().SerializeAsString(); 117e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci size_t bytesWritten = fwrite(cfgProto.data(), 1, cfgProto.size(), writePipeStream); 118e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci fclose(writePipeStream); 119e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci if (bytesWritten != cfgProto.size() || cfgProto.size() == 0) { 120e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci ALOGE("fwrite() failed (ret: %zd) while calling the Perfetto client: %s", bytesWritten, 121e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci strerror(errno)); 122e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci return false; 123e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci } 124e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci 125a74556061ab5257c39a0ba752410da8fbdadd175Primiano Tucci // This does NOT wait for the full duration of the trace. It just waits until 126a74556061ab5257c39a0ba752410da8fbdadd175Primiano Tucci // the process has read the config from stdin and detached. 127e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci int childStatus = 0; 128e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci waitpid(pid, &childStatus, 0); 129e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci if (!WIFEXITED(childStatus) || WEXITSTATUS(childStatus) != 0) { 130e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci ALOGE("Child process failed (0x%x) while calling the Perfetto client", childStatus); 131e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci return false; 132e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci } 133e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci 134484524a246ffe453f8cd89b698a279c23b0bde1fTej Singh VLOG("CollectPerfettoTraceAndUploadToDropbox() succeeded"); 135e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci return true; 136e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci} 137e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci 138e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci} // namespace statsd 139e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci} // namespace os 140e4d44919aa90f16285a9966dcc632c128b932374Primiano Tucci} // namespace android 141