blob: 74ca012e4f3ee08ec668bc74cd500785aaf8469a (
plain)
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
|
#include "DataAccessPerfReader.h"
#include "ErrorHandling.h"
#include "PerfReader.h"
#include "llvm/Support/LineIterator.h"
#include "llvm/Support/MemoryBuffer.h"
#include <regex>
static llvm::Regex IPSampleRegex(": 0x[a-fA-F0-9]+ period:");
static llvm::Regex DataAddressRegex("addr: 0x[a-fA-F0-9]+");
namespace llvm {
void DataAccessPerfReader::parsePerfTraces() {
parsePerfTrace(PerfTraceFilename);
}
// Ignore mmap events.
void DataAccessPerfReader::parsePerfTrace(StringRef PerfTrace) {
std::regex logRegex(
R"(^.*?PERF_RECORD_SAMPLE\(.*?\):\s*(\d+)\/(\d+):\s*(0x[0-9a-fA-F]+)\s+period:\s*\d+\s+addr:\s*(0x[0-9a-fA-F]+)$)");
uint64_t UnmatchedLine = 0, MatchedLine = 0;
auto BufferOrErr = MemoryBuffer::getFile(PerfTrace);
std::error_code EC = BufferOrErr.getError();
if (EC)
exitWithError("Failed to open perf trace file: " + PerfTrace);
line_iterator LineIt(*BufferOrErr.get(), true);
for (; !LineIt.is_at_eof(); ++LineIt) {
StringRef Line = *LineIt;
// Parse MMAP event from perf trace.
// Construct a binary from the binary file path.
PerfScriptReader::MMapEvent MMap;
if (Line.contains("PERF_RECORD_MMAP2")) {
if (PerfScriptReader::extractMMapEventForBinary(Binary, Line, MMap)) {
// TODO: This is a hack to avoid mapping binary address for data section
// mappings.
if (MMap.Offset == 0) {
updateBinaryAddress(MMap);
}
}
continue;
}
if (!Line.contains("PERF_RECORD_SAMPLE")) {
// Skip lines that do not contain "PERF_RECORD_SAMPLE".
continue;
}
std::smatch matches;
const std::string LineStr = Line.str();
if (std::regex_search(LineStr.begin(), LineStr.end(), matches, logRegex)) {
if (matches.size() != 5)
continue;
uint64_t DataAddress = std::stoull(matches[4].str(), nullptr, 16);
uint64_t IP = std::stoull(matches[3].str(), nullptr, 16);
int32_t PID = std::stoi(matches[1].str());
// Check if the PID matches the filter.
if (PIDFilter && *PIDFilter != PID) {
continue;
}
// Extract the address and count.
uint64_t CanonicalDataAddress =
Binary->canonicalizeVirtualAddress(DataAddress);
AddressToCount[CanonicalDataAddress] += 1;
MatchedLine++;
}
}
}
} // namespace llvm
|
