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
|
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#include "ChildProfilerController.h"
#include "ProfilerChild.h"
#include "mozilla/ipc/Endpoint.h"
#include "nsThreadUtils.h"
using namespace mozilla::ipc;
namespace mozilla {
/* static */
already_AddRefed<ChildProfilerController> ChildProfilerController::Create(
mozilla::ipc::Endpoint<PProfilerChild>&& aEndpoint) {
MOZ_RELEASE_ASSERT(NS_IsMainThread());
RefPtr<ChildProfilerController> cpc = new ChildProfilerController();
cpc->Init(std::move(aEndpoint));
return cpc.forget();
}
ChildProfilerController::ChildProfilerController() {
MOZ_COUNT_CTOR(ChildProfilerController);
}
void ChildProfilerController::Init(Endpoint<PProfilerChild>&& aEndpoint) {
if (NS_SUCCEEDED(
NS_NewNamedThread("ProfilerChild", getter_AddRefs(mThread)))) {
// Now that mThread has been set, run SetupProfilerChild on the thread.
mThread->Dispatch(
NewRunnableMethod<Endpoint<PProfilerChild>&&>(
"ChildProfilerController::SetupProfilerChild", this,
&ChildProfilerController::SetupProfilerChild, std::move(aEndpoint)),
NS_DISPATCH_NORMAL);
}
}
nsCString ChildProfilerController::GrabShutdownProfileAndShutdown() {
nsCString shutdownProfile;
ShutdownAndMaybeGrabShutdownProfileFirst(&shutdownProfile);
return shutdownProfile;
}
void ChildProfilerController::Shutdown() {
ShutdownAndMaybeGrabShutdownProfileFirst(nullptr);
}
void ChildProfilerController::ShutdownAndMaybeGrabShutdownProfileFirst(
nsCString* aOutShutdownProfile) {
if (mThread) {
mThread->Dispatch(NewRunnableMethod<nsCString*>(
"ChildProfilerController::ShutdownProfilerChild",
this, &ChildProfilerController::ShutdownProfilerChild,
aOutShutdownProfile),
NS_DISPATCH_NORMAL);
// Shut down the thread. This call will spin until all runnables (including
// the ShutdownProfilerChild runnable) have been processed.
mThread->Shutdown();
mThread = nullptr;
}
}
ChildProfilerController::~ChildProfilerController() {
MOZ_COUNT_DTOR(ChildProfilerController);
MOZ_ASSERT(!mThread,
"Please call Shutdown before destroying ChildProfilerController");
MOZ_ASSERT(!mProfilerChild);
}
void ChildProfilerController::SetupProfilerChild(
Endpoint<PProfilerChild>&& aEndpoint) {
MOZ_RELEASE_ASSERT(mThread == NS_GetCurrentThread());
MOZ_ASSERT(aEndpoint.IsValid());
mProfilerChild = new ProfilerChild();
Endpoint<PProfilerChild> endpoint = std::move(aEndpoint);
if (!endpoint.Bind(mProfilerChild)) {
MOZ_CRASH("Failed to bind ProfilerChild!");
}
}
void ChildProfilerController::ShutdownProfilerChild(
nsCString* aOutShutdownProfile) {
MOZ_RELEASE_ASSERT(mThread == NS_GetCurrentThread());
if (aOutShutdownProfile) {
*aOutShutdownProfile = mProfilerChild->GrabShutdownProfile();
}
mProfilerChild->Destroy();
mProfilerChild = nullptr;
}
} // namespace mozilla
|
