/* -*- 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 "mozilla/ipc/IOThread.h"
#include "mozilla/ipc/NodeController.h"

#if defined(XP_WIN)
#  include <objbase.h>
#endif

namespace mozilla {
namespace ipc {

//
// IOThread
//

/* static */
IOThread* IOThread::sSingleton = nullptr;

IOThread::IOThread(const char* aName) : base::Thread(aName) {
  sSingleton = this;
}

IOThread::~IOThread() {
  MOZ_ASSERT(!IsRunning());
  sSingleton = nullptr;
}

void IOThread::StartThread() {
  // Failure to create the IPC I/O thread is unrecoverable.
  // NOTE: This will block if successful for the `Init()` virtual method to have
  // been run.
  if (!StartWithOptions(
          base::Thread::Options{MessageLoop::TYPE_IO, /* stack size */ 0})) {
    MOZ_CRASH("Failed to create IPC I/O Thread");
  }
}

void IOThread::StopThread() {
  // This will block until CleanUp() has been called, and the IPC I/O thread has
  // been joined.
  Stop();
}

//
// IOThreadParent
//

IOThreadParent::IOThreadParent() : IOThread("IPC I/O Parent") { StartThread(); }

IOThreadParent::~IOThreadParent() { StopThread(); }

void IOThreadParent::Init() {
#if defined(XP_WIN)
  // Initializes the COM library on the current thread.
  CoInitialize(nullptr);
#endif

  // Initialize the ports library in the current thread.
  NodeController::InitBrokerProcess();
}

void IOThreadParent::CleanUp() {
  NodeController::CleanUp();

#if defined(XP_WIN)
  // Closes the COM library on the current thread. CoInitialize must
  // be balanced by a corresponding call to CoUninitialize.
  CoUninitialize();
#endif
}

//
// IOThreadChild
//

IOThreadChild::IOThreadChild(IPC::Channel::ChannelHandle aClientHandle,
                             base::ProcessId aParentPid)
    : IOThread("IPC I/O Child"),
      mClientHandle(std::move(aClientHandle)),
      mParentPid(std::move(aParentPid)) {
  StartThread();
}

IOThreadChild::~IOThreadChild() { StopThread(); }

void IOThreadChild::Init() {
  auto channel = MakeUnique<IPC::Channel>(
      std::move(mClientHandle), IPC::Channel::MODE_CLIENT, mParentPid);
#if defined(XP_WIN)
  channel->StartAcceptingHandles(IPC::Channel::MODE_CLIENT);
#elif defined(XP_DARWIN)
  channel->StartAcceptingMachPorts(IPC::Channel::MODE_CLIENT);
#endif

  mInitialPort = mozilla::ipc::NodeController::InitChildProcess(
      std::move(channel), mParentPid);
}

void IOThreadChild::CleanUp() { NodeController::CleanUp(); }

}  // namespace ipc
}  // namespace mozilla