utils.h revision 5633586f0efc54dbaecad3deeb8568786317832d
167d3abd7b26a741347b33402ad32f5c6735ca0bdYabin Cui/* 267d3abd7b26a741347b33402ad32f5c6735ca0bdYabin Cui * Copyright (C) 2015 The Android Open Source Project 367d3abd7b26a741347b33402ad32f5c6735ca0bdYabin Cui * 467d3abd7b26a741347b33402ad32f5c6735ca0bdYabin Cui * Licensed under the Apache License, Version 2.0 (the "License"); 567d3abd7b26a741347b33402ad32f5c6735ca0bdYabin Cui * you may not use this file except in compliance with the License. 667d3abd7b26a741347b33402ad32f5c6735ca0bdYabin Cui * You may obtain a copy of the License at 767d3abd7b26a741347b33402ad32f5c6735ca0bdYabin Cui * 867d3abd7b26a741347b33402ad32f5c6735ca0bdYabin Cui * http://www.apache.org/licenses/LICENSE-2.0 967d3abd7b26a741347b33402ad32f5c6735ca0bdYabin Cui * 1067d3abd7b26a741347b33402ad32f5c6735ca0bdYabin Cui * Unless required by applicable law or agreed to in writing, software 1167d3abd7b26a741347b33402ad32f5c6735ca0bdYabin Cui * distributed under the License is distributed on an "AS IS" BASIS, 1267d3abd7b26a741347b33402ad32f5c6735ca0bdYabin Cui * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 1367d3abd7b26a741347b33402ad32f5c6735ca0bdYabin Cui * See the License for the specific language governing permissions and 1467d3abd7b26a741347b33402ad32f5c6735ca0bdYabin Cui * limitations under the License. 1567d3abd7b26a741347b33402ad32f5c6735ca0bdYabin Cui */ 1667d3abd7b26a741347b33402ad32f5c6735ca0bdYabin Cui 1767d3abd7b26a741347b33402ad32f5c6735ca0bdYabin Cui#ifndef SIMPLE_PERF_UTILS_H_ 1867d3abd7b26a741347b33402ad32f5c6735ca0bdYabin Cui#define SIMPLE_PERF_UTILS_H_ 1967d3abd7b26a741347b33402ad32f5c6735ca0bdYabin Cui 2067d3abd7b26a741347b33402ad32f5c6735ca0bdYabin Cui#include <stddef.h> 21cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui 22323e945313b190373b3fcfe578e25ee8390a76d3Yabin Cui#include <string> 239759e1b1ce76185aa539aeea2fb1cbd8382156e7Yabin Cui#include <vector> 2467d3abd7b26a741347b33402ad32f5c6735ca0bdYabin Cui 258f680f60dc800bec880c5c35bfbc1ac36165e1f3Yabin Cui#include <android-base/logging.h> 26b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui#include <android-base/macros.h> 27be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui#include <ziparchive/zip_archive.h> 28b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui 297d59bb49fb47fbc82ef5c77d7aebf7174fd996e1Yabin Cui#define ALIGN(value, alignment) (((value) + (alignment)-1) & ~((alignment)-1)) 3067d3abd7b26a741347b33402ad32f5c6735ca0bdYabin Cui 31ffaa912ca48e87d25d516780c15cdd7c50de43dbYabin Cui#ifdef _WIN32 32ffaa912ca48e87d25d516780c15cdd7c50de43dbYabin Cui#define CLOSE_ON_EXEC_MODE "" 33ffaa912ca48e87d25d516780c15cdd7c50de43dbYabin Cui#else 34ffaa912ca48e87d25d516780c15cdd7c50de43dbYabin Cui#define CLOSE_ON_EXEC_MODE "e" 35ffaa912ca48e87d25d516780c15cdd7c50de43dbYabin Cui#endif 36621a533ec28dbbf8634211c684456b4bbc1bc0fbYabin Cui 37cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui// OneTimeAllocator is used to allocate memory many times and free only once at the end. 38cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui// It reduces the cost to free each allocated memory. 39cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cuiclass OneTimeFreeAllocator { 40cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui public: 41cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui OneTimeFreeAllocator(size_t unit_size = 8192u) 42cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui : unit_size_(unit_size), cur_(nullptr), end_(nullptr) { 43cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui } 44cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui 45cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui ~OneTimeFreeAllocator() { 46cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui Clear(); 47cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui } 48cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui 49cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui void Clear(); 50cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui const char* AllocateString(const std::string& s); 51cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui 52cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui private: 53cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui const size_t unit_size_; 54cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui std::vector<char*> v_; 55cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui char* cur_; 56cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui char* end_; 57cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui}; 58cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui 59b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cuiclass FileHelper { 60b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui public: 61be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui static FileHelper OpenReadOnly(const std::string& filename); 62be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui static FileHelper OpenWriteOnly(const std::string& filename); 63be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui 64be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui FileHelper(FileHelper&& other) { 65be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui fd_ = other.fd_; 66be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui other.fd_ = -1; 67be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui } 68be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui 69b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui ~FileHelper(); 70b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui 71b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui explicit operator bool() const { 72b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui return fd_ != -1; 73b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui } 74b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui 75b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui int fd() const { 76b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui return fd_; 77b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui } 78b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui 79b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui private: 80be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui FileHelper(int fd) : fd_(fd) {} 81b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui int fd_; 82b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui 83b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui DISALLOW_COPY_AND_ASSIGN(FileHelper); 84b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui}; 85b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui 86be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cuiclass ArchiveHelper { 87be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui public: 88be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui ArchiveHelper(int fd, const std::string& debug_filename); 89be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui ~ArchiveHelper(); 90be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui 91be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui explicit operator bool() const { 92be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui return valid_; 93be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui } 94be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui ZipArchiveHandle &archive_handle() { 95be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui return handle_; 96be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui } 97be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui 98be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui private: 99be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui ZipArchiveHandle handle_; 100be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui bool valid_; 101be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui 102be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui DISALLOW_COPY_AND_ASSIGN(ArchiveHelper); 103be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui}; 104be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui 105d713f959ec1fe07ed993e2c9f4166b52aa98a58cYabin Cuitemplate <class T> 106d713f959ec1fe07ed993e2c9f4166b52aa98a58cYabin Cuivoid MoveFromBinaryFormat(T& data, const char*& p) { 107d713f959ec1fe07ed993e2c9f4166b52aa98a58cYabin Cui data = *reinterpret_cast<const T*>(p); 108d713f959ec1fe07ed993e2c9f4166b52aa98a58cYabin Cui p += sizeof(T); 109d713f959ec1fe07ed993e2c9f4166b52aa98a58cYabin Cui} 110d713f959ec1fe07ed993e2c9f4166b52aa98a58cYabin Cui 1117d59bb49fb47fbc82ef5c77d7aebf7174fd996e1Yabin Cuivoid PrintIndented(size_t indent, const char* fmt, ...); 1127d59bb49fb47fbc82ef5c77d7aebf7174fd996e1Yabin Cui 1139759e1b1ce76185aa539aeea2fb1cbd8382156e7Yabin Cuibool IsPowerOfTwo(uint64_t value); 1149759e1b1ce76185aa539aeea2fb1cbd8382156e7Yabin Cui 1157d59bb49fb47fbc82ef5c77d7aebf7174fd996e1Yabin Cuivoid GetEntriesInDir(const std::string& dirpath, std::vector<std::string>* files, 1167d59bb49fb47fbc82ef5c77d7aebf7174fd996e1Yabin Cui std::vector<std::string>* subdirs); 117b032de7a9e1975a838cd6df2e7c8df3c7f70a3ceYabin Cuibool IsDir(const std::string& dirpath); 118797116bfb9ad10d858086680c7465d65c399ef6eYabin Cuibool IsRegularFile(const std::string& filename); 119b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cuiuint64_t GetFileSize(const std::string& filename); 120be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cuibool MkdirWithParents(const std::string& path); 121323e945313b190373b3fcfe578e25ee8390a76d3Yabin Cui 12205400539d15c50a52a388daba2e831b86824cc8eYabin Cuibool XzDecompress(const std::string& compressed_data, std::string* decompressed_data); 12305400539d15c50a52a388daba2e831b86824cc8eYabin Cui 1248f680f60dc800bec880c5c35bfbc1ac36165e1f3Yabin Cuibool GetLogSeverity(const std::string& name, android::base::LogSeverity* severity); 1258f680f60dc800bec880c5c35bfbc1ac36165e1f3Yabin Cui 1265633586f0efc54dbaecad3deeb8568786317832dYabin Cuibool IsRoot(); 1275633586f0efc54dbaecad3deeb8568786317832dYabin Cui 12867d3abd7b26a741347b33402ad32f5c6735ca0bdYabin Cui#endif // SIMPLE_PERF_UTILS_H_ 129