1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
/*
* Copyright (C) 2015 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef SIMPLE_PERF_UTILS_H_
#define SIMPLE_PERF_UTILS_H_
#include <signal.h>
#include <stddef.h>
#include <stdio.h>
#include <stdlib.h>
#include <string>
#include <vector>
#define ALIGN(value, alignment) (((value) + (alignment)-1) & ~((alignment)-1))
class LineReader {
public:
LineReader(FILE* fp) : fp_(fp), buf_(nullptr), bufsize_(0) {
}
~LineReader() {
free(buf_);
fclose(fp_);
}
char* ReadLine() {
if (getline(&buf_, &bufsize_, fp_) != -1) {
return buf_;
}
return nullptr;
}
size_t MaxLineSize() {
return bufsize_;
}
private:
FILE* fp_;
char* buf_;
size_t bufsize_;
};
class SignalHandlerRegister {
public:
SignalHandlerRegister(const std::vector<int>& signums, void (*handler)(int)) {
for (auto& sig : signums) {
sig_t old_handler = signal(sig, handler);
saved_signal_handlers_.push_back(std::make_pair(sig, old_handler));
}
}
~SignalHandlerRegister() {
for (auto& pair : saved_signal_handlers_) {
signal(pair.first, pair.second);
}
}
private:
std::vector<std::pair<int, sig_t>> saved_signal_handlers_;
};
// OneTimeAllocator is used to allocate memory many times and free only once at the end.
// It reduces the cost to free each allocated memory.
class OneTimeFreeAllocator {
public:
OneTimeFreeAllocator(size_t unit_size = 8192u)
: unit_size_(unit_size), cur_(nullptr), end_(nullptr) {
}
~OneTimeFreeAllocator() {
Clear();
}
void Clear();
const char* AllocateString(const std::string& s);
private:
const size_t unit_size_;
std::vector<char*> v_;
char* cur_;
char* end_;
};
template <class T>
void MoveFromBinaryFormat(T& data, const char*& p) {
data = *reinterpret_cast<const T*>(p);
p += sizeof(T);
}
void PrintIndented(size_t indent, const char* fmt, ...);
bool IsPowerOfTwo(uint64_t value);
void GetEntriesInDir(const std::string& dirpath, std::vector<std::string>* files,
std::vector<std::string>* subdirs);
bool IsDir(const std::string& dirpath);
bool RemovePossibleFile(const std::string& filename);
bool StringToPid(const std::string& s, int* pid);
#endif // SIMPLE_PERF_UTILS_H_
|