utils.h revision be7ec66eaa4f995bd9068637f7c7d5718173922c
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 25b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui#include <android-base/macros.h> 26be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui#include <ziparchive/zip_archive.h> 27b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui 287d59bb49fb47fbc82ef5c77d7aebf7174fd996e1Yabin Cui#define ALIGN(value, alignment) (((value) + (alignment)-1) & ~((alignment)-1)) 2967d3abd7b26a741347b33402ad32f5c6735ca0bdYabin Cui 30ffaa912ca48e87d25d516780c15cdd7c50de43dbYabin Cui#ifdef _WIN32 31ffaa912ca48e87d25d516780c15cdd7c50de43dbYabin Cui#define CLOSE_ON_EXEC_MODE "" 32ffaa912ca48e87d25d516780c15cdd7c50de43dbYabin Cui#else 33ffaa912ca48e87d25d516780c15cdd7c50de43dbYabin Cui#define CLOSE_ON_EXEC_MODE "e" 34ffaa912ca48e87d25d516780c15cdd7c50de43dbYabin Cui#endif 35621a533ec28dbbf8634211c684456b4bbc1bc0fbYabin Cui 36cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui// OneTimeAllocator is used to allocate memory many times and free only once at the end. 37cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui// It reduces the cost to free each allocated memory. 38cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cuiclass OneTimeFreeAllocator { 39cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui public: 40cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui OneTimeFreeAllocator(size_t unit_size = 8192u) 41cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui : unit_size_(unit_size), cur_(nullptr), end_(nullptr) { 42cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui } 43cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui 44cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui ~OneTimeFreeAllocator() { 45cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui Clear(); 46cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui } 47cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui 48cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui void Clear(); 49cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui const char* AllocateString(const std::string& s); 50cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui 51cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui private: 52cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui const size_t unit_size_; 53cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui std::vector<char*> v_; 54cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui char* cur_; 55cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui char* end_; 56cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui}; 57cc2e59e2478d330c89eaceda0dcc1f05ee32fbf8Yabin Cui 58b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cuiclass FileHelper { 59b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui public: 60be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui static FileHelper OpenReadOnly(const std::string& filename); 61be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui static FileHelper OpenWriteOnly(const std::string& filename); 62be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui 63be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui FileHelper(FileHelper&& other) { 64be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui fd_ = other.fd_; 65be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui other.fd_ = -1; 66be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui } 67be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui 68b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui ~FileHelper(); 69b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui 70b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui explicit operator bool() const { 71b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui return fd_ != -1; 72b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui } 73b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui 74b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui int fd() const { 75b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui return fd_; 76b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui } 77b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui 78b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui private: 79be7ec66eaa4f995bd9068637f7c7d5718173922cYabin Cui FileHelper(int fd) : fd_(fd) {} 80b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui int fd_; 81b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui 82b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui DISALLOW_COPY_AND_ASSIGN(FileHelper); 83b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui}; 84b1a885b014540a2f7798b5a35ea0f0ec150d93eeYabin Cui 85be7ec66eaa4f995bd9068637f7c7d5718173922cYabin 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 12267d3abd7b26a741347b33402ad32f5c6735ca0bdYabin Cui#endif // SIMPLE_PERF_UTILS_H_ 123