#include "TestBadActor.h"
#include "IPDLUnitTests.h"
#include "mozilla/Unused.h"

namespace mozilla {
namespace _ipdltest {

void TestBadActorParent::Main() {
  // This test is designed to test a race condition where the child sends us
  // a message on an actor that we've already destroyed. The child process
  // should die, and the parent process should not abort.

  PTestBadActorSubParent* child = SendPTestBadActorSubConstructor();
  if (!child) fail("Sending constructor");

  Unused << child->Call__delete__(child);
}

// By default, fatal errors kill the parent process, but this makes it
// hard to test, so instead we use the previous behavior and kill the
// child process.
void TestBadActorParent::HandleFatalError(const char* aErrorMsg) {
  if (!!strcmp(aErrorMsg, "incoming message racing with actor deletion")) {
    fail("wrong fatal error");
  }

  ipc::ScopedProcessHandle otherProcessHandle;
  if (!base::OpenProcessHandle(OtherPid(), &otherProcessHandle.rwget())) {
    fail("couldn't open child process");
  } else {
    if (!base::KillProcess(otherProcessHandle, 0, false)) {
      fail("terminating child process");
    }
  }
}

PTestBadActorSubParent* TestBadActorParent::AllocPTestBadActorSubParent() {
  return new TestBadActorSubParent();
}

mozilla::ipc::IPCResult TestBadActorSubParent::RecvPing() {
  fail("Shouldn't have received ping.");
  return IPC_FAIL_NO_REASON(this);
}

PTestBadActorSubChild* TestBadActorChild::AllocPTestBadActorSubChild() {
  return new TestBadActorSubChild();
}

mozilla::ipc::IPCResult TestBadActorChild::RecvPTestBadActorSubConstructor(
    PTestBadActorSubChild* actor) {
  if (!actor->SendPing()) {
    fail("Couldn't send ping to an actor which supposedly isn't dead yet.");
  }
  return IPC_OK();
}

}  // namespace _ipdltest
}  // namespace mozilla
