1 files changed, 122 insertions, 0 deletions
diff --git a/ipc/ipdl/test/cxx/TestStackHooks.cpp b/ipc/ipdl/test/cxx/TestStackHooks.cpp
new file mode 100644
index 0000000000..9800ac54e7
--- /dev/null
+++ b/ipc/ipdl/test/cxx/TestStackHooks.cpp
@@ -0,0 +1,122 @@
+#include "TestStackHooks.h"
+
+#include "base/task.h"
+#include "IPDLUnitTests.h"  // fail etc.
+
+namespace mozilla {
+namespace _ipdltest {
+
+//-----------------------------------------------------------------------------
+// parent
+
+TestStackHooksParent::TestStackHooksParent()
+    : mOnStack(false), mIncallDepth(0) {
+  MOZ_COUNT_CTOR(TestStackHooksParent);
+}
+
+TestStackHooksParent::~TestStackHooksParent() {
+  MOZ_COUNT_DTOR(TestStackHooksParent);
+}
+
+void TestStackHooksParent::Main() {
+  if (!SendStart()) fail("sending Start()");
+}
+
+mozilla::ipc::IPCResult TestStackHooksParent::AnswerStackFrame() {
+  if (!mOnStack) fail("not on C++ stack?!");
+
+  if (!CallStackFrame()) fail("calling StackFrame()");
+
+  if (!mOnStack) fail("not on C++ stack?!");
+
+  if (1 != mIncallDepth) fail("missed EnteredCall or ExitedCall hook");
+
+  return IPC_OK();
+}
+
+//-----------------------------------------------------------------------------
+// child
+
+TestStackHooksChild::TestStackHooksChild()
+    : mOnStack(false),
+      mEntered(0),
+      mExited(0),
+      mIncallDepth(0),
+      mNumAnswerStackFrame(0) {
+  MOZ_COUNT_CTOR(TestStackHooksChild);
+}
+
+TestStackHooksChild::~TestStackHooksChild() {
+  MOZ_COUNT_DTOR(TestStackHooksChild);
+}
+
+namespace {
+void RunTestsFn() {
+  static_cast<TestStackHooksChild*>(gChildActor)->RunTests();
+}
+}  // namespace
+
+mozilla::ipc::IPCResult TestStackHooksChild::RecvStart() {
+  if (!mOnStack) fail("missed stack notification");
+
+  if (0 != mIncallDepth) fail("EnteredCall/ExitedCall malfunction");
+
+  // kick off tests from a runnable so that we can start with
+  // MessageChannel code on the C++ stack
+  MessageLoop::current()->PostTask(
+      NewRunnableFunction("RunTestsFn", RunTestsFn));
+
+  return IPC_OK();
+}
+
+mozilla::ipc::IPCResult TestStackHooksChild::AnswerStackFrame() {
+  ++mNumAnswerStackFrame;
+
+  if (!mOnStack) fail("missed stack notification");
+
+  if (1 != mIncallDepth) fail("missed EnteredCall or ExitedCall hook");
+
+  if (mNumAnswerStackFrame == 1) {
+    if (!SendAsync()) fail("sending Async()");
+  } else if (mNumAnswerStackFrame == 2) {
+    if (!SendSync()) fail("sending Sync()");
+  } else {
+    fail("unexpected state");
+  }
+
+  if (!mOnStack) fail("bad stack exit notification");
+
+  return IPC_OK();
+}
+
+void TestStackHooksChild::RunTests() {
+  // 1 because of RecvStart()
+  if (1 != mEntered) fail("missed stack notification");
+  if (mOnStack) fail("spurious stack notification");
+  if (0 != mIncallDepth) fail("EnteredCall/ExitedCall malfunction");
+
+  if (!SendAsync()) fail("sending Async()");
+  if (mOnStack) fail("spurious stack notification");
+  if (0 != mIncallDepth) fail("EnteredCall/ExitedCall malfunction");
+  if (2 != mEntered) fail("missed stack notification");
+
+  if (!SendSync()) fail("sending Sync()");
+  if (mOnStack) fail("spurious stack notification");
+  if (0 != mIncallDepth) fail("EnteredCall/ExitedCall malfunction");
+  if (3 != mEntered) fail("missed stack notification");
+
+  if (!CallRpc()) fail("calling RPC()");
+  if (mOnStack) fail("spurious stack notification");
+  if (0 != mIncallDepth) fail("EnteredCall/ExitedCall malfunction");
+  if (4 != mEntered) fail("missed stack notification");
+
+  if (!CallStackFrame()) fail("calling StackFrame()");
+  if (mOnStack) fail("spurious stack notification");
+  if (0 != mIncallDepth) fail("EnteredCall/ExitedCall malfunction");
+  if (5 != mEntered) fail("missed stack notification");
+
+  Close();
+}
+
+}  // namespace _ipdltest
+}  // namespace mozilla