summaryrefslogtreecommitdiffstats
path: root/ipc/ipdl/test/cxx/TestStackHooks.h
blob: 315a3c1ae505d9e224a9bff44b24e98d0a92e0e3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
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