1 files changed, 169 insertions, 0 deletions
diff --git a/ipc/ipdl/test/cxx/TestInterruptRaces.cpp b/ipc/ipdl/test/cxx/TestInterruptRaces.cpp
new file mode 100644
index 0000000000..02b42cfbc3
--- /dev/null
+++ b/ipc/ipdl/test/cxx/TestInterruptRaces.cpp
@@ -0,0 +1,169 @@
+#include "TestInterruptRaces.h"
+
+#include "IPDLUnitTests.h"  // fail etc.
+
+using mozilla::ipc::MessageChannel;
+
+namespace mozilla {
+namespace _ipdltest {
+
+ipc::RacyInterruptPolicy MediateRace(const MessageChannel::MessageInfo& parent,
+                                     const MessageChannel::MessageInfo& child) {
+  return (PTestInterruptRaces::Msg_Child__ID == parent.type())
+             ? ipc::RIPParentWins
+             : ipc::RIPChildWins;
+}
+
+//-----------------------------------------------------------------------------
+// parent
+void TestInterruptRacesParent::Main() {
+  if (!SendStart()) fail("sending Start()");
+}
+
+mozilla::ipc::IPCResult TestInterruptRacesParent::RecvStartRace() {
+  MessageLoop::current()->PostTask(NewNonOwningRunnableMethod(
+      "_ipdltest::TestInterruptRacesParent::OnRaceTime", this,
+      &TestInterruptRacesParent::OnRaceTime));
+  return IPC_OK();
+}
+
+void TestInterruptRacesParent::OnRaceTime() {
+  if (!CallRace(&mChildHasReply)) fail("problem calling Race()");
+
+  if (!mChildHasReply) fail("child should have got a reply already");
+
+  mHasReply = true;
+
+  MessageLoop::current()->PostTask(
+      NewNonOwningRunnableMethod("_ipdltest::TestInterruptRacesParent::Test2",
+                                 this, &TestInterruptRacesParent::Test2));
+}
+
+mozilla::ipc::IPCResult TestInterruptRacesParent::AnswerRace(bool* hasReply) {
+  if (mHasReply) fail("apparently the parent won the Interrupt race!");
+  *hasReply = hasReply;
+  return IPC_OK();
+}
+
+void TestInterruptRacesParent::Test2() {
+  puts("  passed");
+  puts("Test 2");
+
+  mHasReply = false;
+  mChildHasReply = false;
+
+  if (!CallStackFrame()) fail("can't set up a stack frame");
+
+  puts("  passed");
+
+  MessageLoop::current()->PostTask(
+      NewNonOwningRunnableMethod("_ipdltest::TestInterruptRacesParent::Test3",
+                                 this, &TestInterruptRacesParent::Test3));
+}
+
+mozilla::ipc::IPCResult TestInterruptRacesParent::AnswerStackFrame() {
+  if (!SendWakeup()) fail("can't wake up the child");
+
+  if (!CallRace(&mChildHasReply)) fail("can't set up race condition");
+  mHasReply = true;
+
+  if (!mChildHasReply) fail("child should have got a reply already");
+
+  return IPC_OK();
+}
+
+void TestInterruptRacesParent::Test3() {
+  puts("Test 3");
+
+  if (!CallStackFrame3()) fail("can't set up a stack frame");
+
+  puts("  passed");
+
+  Close();
+}
+
+mozilla::ipc::IPCResult TestInterruptRacesParent::AnswerStackFrame3() {
+  if (!SendWakeup3()) fail("can't wake up the child");
+
+  if (!CallChild()) fail("can't set up race condition");
+
+  return IPC_OK();
+}
+
+mozilla::ipc::IPCResult TestInterruptRacesParent::AnswerParent() {
+  mAnsweredParent = true;
+  return IPC_OK();
+}
+
+mozilla::ipc::IPCResult TestInterruptRacesParent::RecvGetAnsweredParent(
+    bool* answeredParent) {
+  *answeredParent = mAnsweredParent;
+  return IPC_OK();
+}
+
+//-----------------------------------------------------------------------------
+// child
+mozilla::ipc::IPCResult TestInterruptRacesChild::RecvStart() {
+  puts("Test 1");
+
+  if (!SendStartRace()) fail("problem sending StartRace()");
+
+  bool dontcare;
+  if (!CallRace(&dontcare)) fail("problem calling Race()");
+
+  mHasReply = true;
+  return IPC_OK();
+}
+
+mozilla::ipc::IPCResult TestInterruptRacesChild::AnswerRace(bool* hasReply) {
+  if (!mHasReply) fail("apparently the child lost the Interrupt race!");
+
+  *hasReply = mHasReply;
+
+  return IPC_OK();
+}
+
+mozilla::ipc::IPCResult TestInterruptRacesChild::AnswerStackFrame() {
+  // reset for the second test
+  mHasReply = false;
+
+  if (!CallStackFrame()) fail("can't set up stack frame");
+
+  if (!mHasReply) fail("should have had reply by now");
+
+  return IPC_OK();
+}
+
+mozilla::ipc::IPCResult TestInterruptRacesChild::RecvWakeup() {
+  bool dontcare;
+  if (!CallRace(&dontcare)) fail("can't set up race condition");
+
+  mHasReply = true;
+  return IPC_OK();
+}
+
+mozilla::ipc::IPCResult TestInterruptRacesChild::AnswerStackFrame3() {
+  if (!CallStackFrame3()) fail("can't set up stack frame");
+  return IPC_OK();
+}
+
+mozilla::ipc::IPCResult TestInterruptRacesChild::RecvWakeup3() {
+  if (!CallParent()) fail("can't set up race condition");
+  return IPC_OK();
+}
+
+mozilla::ipc::IPCResult TestInterruptRacesChild::AnswerChild() {
+  bool parentAnsweredParent;
+  // the parent is supposed to win the race, which means its
+  // message, Child(), is supposed to be processed before the
+  // child's message, Parent()
+  if (!SendGetAnsweredParent(&parentAnsweredParent))
+    fail("sending GetAnsweredParent");
+
+  if (parentAnsweredParent) fail("parent was supposed to win the race!");
+
+  return IPC_OK();
+}
+
+}  // namespace _ipdltest
+}  // namespace mozilla