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
|
// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "base/android/thread_instruction_count.h"
#include <linux/perf_event.h>
#include <sys/syscall.h>
#include <unistd.h>
#include "base/check_op.h"
#include "base/logging.h"
#include "base/no_destructor.h"
#include "base/numerics/safe_conversions.h"
#include "base/threading/thread_local_storage.h"
namespace base {
namespace android {
namespace {
constexpr int kPerfFdOpenFailed = -1;
ThreadLocalStorage::Slot& InstructionCounterFdSlot() {
static NoDestructor<ThreadLocalStorage::Slot> fd_slot([](void* fd_ptr) {
int fd = checked_cast<int>(reinterpret_cast<intptr_t>(fd_ptr));
if (fd > 0) {
close(fd);
}
});
return *fd_slot;
}
// Opens a new file descriptor that emits the value of
// PERF_COUNT_HW_INSTRUCTIONS in userspace (excluding kernel and hypervisor
// instructions) for the given |thread_id|, or 0 for the calling thread.
//
// Returns kPerfFdOpenFailed if opening the file descriptor failed.
int OpenInstructionCounterFdForThread(int thread_id) {
struct perf_event_attr pe = {0};
pe.type = PERF_TYPE_HARDWARE;
pe.size = sizeof(struct perf_event_attr);
pe.config = PERF_COUNT_HW_INSTRUCTIONS;
pe.exclude_kernel = 1;
pe.exclude_hv = 1;
long fd = syscall(__NR_perf_event_open, &pe, thread_id, /* cpu */ -1,
/* group_fd */ -1, /* flags */ 0);
if (fd < 0) {
PLOG(ERROR) << "perf_event_open: omitting instruction counters";
return kPerfFdOpenFailed;
}
return checked_cast<int>(fd);
}
// Retrieves the active perf counter FD for the current thread, performing
// lazy-initialization if necessary.
int InstructionCounterFdForCurrentThread() {
auto& slot = InstructionCounterFdSlot();
int fd = checked_cast<int>(reinterpret_cast<intptr_t>(slot.Get()));
if (fd == 0) {
fd = OpenInstructionCounterFdForThread(0);
slot.Set(reinterpret_cast<void*>(fd));
}
return fd;
}
} // namespace
// static
bool ThreadInstructionCount::IsSupported() {
return InstructionCounterFdForCurrentThread() > 0;
}
// static
ThreadInstructionCount ThreadInstructionCount::Now() {
DCHECK(IsSupported());
int fd = InstructionCounterFdForCurrentThread();
if (fd <= 0) {
return ThreadInstructionCount();
}
uint64_t instructions = 0;
ssize_t bytes_read = read(fd, &instructions, sizeof(instructions));
CHECK_EQ(bytes_read, static_cast<ssize_t>(sizeof(instructions)))
<< "Short reads of small size from kernel memory is not expected. If "
"this fails, use HANDLE_EINTR.";
return ThreadInstructionCount(instructions);
}
} // namespace android
} // namespace base
|
