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
|
// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "content/browser/devtools/tracing_process_set_monitor.h"
#include "base/auto_reset.h"
#include "content/public/browser/render_process_host.h"
namespace content {
// static
std::unique_ptr<TracingProcessSetMonitor> TracingProcessSetMonitor::Start(
DevToolsSession& root_session,
ProcessAddedCallback callback) {
return base::WrapUnique(
new TracingProcessSetMonitor(root_session, std::move(callback)));
}
void TracingProcessSetMonitor::SessionAttached(DevToolsSession& session) {
CHECK(session.GetRootSession() == &*root_session_);
auto* const host =
static_cast<DevToolsAgentHostImpl*>(session.GetAgentHost());
CHECK(host);
if (bool inserted = hosts_.insert(host).second; inserted) {
MaybeAddProcess(host);
}
}
void TracingProcessSetMonitor::DevToolsAgentHostDetached(
DevToolsAgentHost* host) {
hosts_.erase(host);
}
void TracingProcessSetMonitor::DevToolsAgentHostDestroyed(
DevToolsAgentHost* host) {
hosts_.erase(host);
}
void TracingProcessSetMonitor::DevToolsAgentHostProcessChanged(
DevToolsAgentHost* host) {
if (!hosts_.contains(host)) {
return;
}
MaybeAddProcess(host);
}
void TracingProcessSetMonitor::MaybeAddProcess(DevToolsAgentHost* host) {
base::ProcessId pid =
static_cast<DevToolsAgentHostImpl*>(host)->GetProcessId();
if (pid == base::kNullProcessId) {
return;
}
AddProcess(pid);
}
void TracingProcessSetMonitor::AddProcess(base::ProcessId pid) {
const bool inserted = known_pids_.insert(pid).second;
if (inserted && !in_init_) {
process_added_callback_.Run(pid);
}
}
TracingProcessSetMonitor::TracingProcessSetMonitor(
DevToolsSession& root_session,
ProcessAddedCallback callback)
: root_session_(root_session),
process_added_callback_(std::move(callback)) {
base::AutoReset<bool> auto_reset(&in_init_, true);
session_observation_.Observe(&root_session);
DevToolsAgentHost::AddObserver(this);
SessionAttached(root_session);
}
TracingProcessSetMonitor::~TracingProcessSetMonitor() {
DevToolsAgentHost::RemoveObserver(this);
}
} // namespace content
|
