1 files changed, 100 insertions, 0 deletions
diff --git a/tools/profiler/gecko/ChildProfilerController.cpp b/tools/profiler/gecko/ChildProfilerController.cpp
new file mode 100644
index 0000000000..d3915b3e0b
--- /dev/null
+++ b/tools/profiler/gecko/ChildProfilerController.cpp
@@ -0,0 +1,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