1 files changed, 107 insertions, 0 deletions
diff --git a/ipc/ipdl/test/cxx/TestStackHooks.h b/ipc/ipdl/test/cxx/TestStackHooks.h
new file mode 100644
index 0000000000..315a3c1ae5
--- /dev/null
+++ b/ipc/ipdl/test/cxx/TestStackHooks.h
@@ -0,0 +1,107 @@
+#ifndef mozilla__ipdltest_TestStackHooks_h
+#define mozilla__ipdltest_TestStackHooks_h 1
+
+#include "mozilla/_ipdltest/IPDLUnitTests.h"
+
+#include "mozilla/_ipdltest/PTestStackHooksParent.h"
+#include "mozilla/_ipdltest/PTestStackHooksChild.h"
+
+namespace mozilla {
+namespace _ipdltest {
+
+class TestStackHooksParent : public PTestStackHooksParent {
+  friend class PTestStackHooksParent;
+
+ public:
+  TestStackHooksParent();
+  virtual ~TestStackHooksParent();
+
+  static bool RunTestInProcesses() { return true; }
+  static bool RunTestInThreads() { return true; }
+
+  void Main();
+
+ protected:
+  mozilla::ipc::IPCResult RecvAsync() {
+    if (!mOnStack) fail("not on C++ stack?!");
+    return IPC_OK();
+  }
+
+  mozilla::ipc::IPCResult RecvSync() {
+    if (!mOnStack) fail("not on C++ stack?!");
+    return IPC_OK();
+  }
+
+  mozilla::ipc::IPCResult AnswerRpc() {
+    if (!mOnStack) fail("not on C++ stack?!");
+    return IPC_OK();
+  }
+
+  mozilla::ipc::IPCResult AnswerStackFrame();
+
+  virtual void ActorDestroy(ActorDestroyReason why) override {
+    if (NormalShutdown != why) fail("unexpected destruction!");
+    passed("ok");
+    QuitParent();
+  }
+
+  virtual void EnteredCxxStack() override { mOnStack = true; }
+  virtual void ExitedCxxStack() override { mOnStack = false; }
+
+  virtual void EnteredCall() override { ++mIncallDepth; }
+  virtual void ExitedCall() override { --mIncallDepth; }
+
+ private:
+  bool mOnStack;
+  int mIncallDepth;
+};
+
+class TestStackHooksChild : public PTestStackHooksChild {
+  friend class PTestStackHooksChild;
+
+ public:
+  TestStackHooksChild();
+  virtual ~TestStackHooksChild();
+
+  void RunTests();
+
+ protected:
+  mozilla::ipc::IPCResult RecvStart();
+
+  mozilla::ipc::IPCResult AnswerStackFrame();
+
+  virtual void ActorDestroy(ActorDestroyReason why) override {
+    if (NormalShutdown != why) fail("unexpected destruction!");
+
+    if (mEntered != mExited) fail("unbalanced enter/exit notifications");
+
+    if (mOnStack)
+      fail("computing mOnStack went awry; should have failed above assertion");
+
+    QuitChild();
+  }
+
+  virtual void EnteredCxxStack() override {
+    ++mEntered;
+    mOnStack = true;
+  }
+  virtual void ExitedCxxStack() override {
+    ++mExited;
+    mOnStack = false;
+  }
+
+  virtual void EnteredCall() override { ++mIncallDepth; }
+  virtual void ExitedCall() override { --mIncallDepth; }
+
+ private:
+  bool mOnStack;
+  int mEntered;
+  int mExited;
+  int mIncallDepth;
+  int32_t mNumAnswerStackFrame;
+};
+
+}  // namespace _ipdltest
+}  // namespace mozilla
+
+#endif  // ifndef mozilla__ipdltest_TestStackHooks_h