signals_unittest.cc 8.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222
  1. // Copyright 2017 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "build/build_config.h"
  5. #include "mojo/core/test/mojo_test_base.h"
  6. #include "mojo/public/c/system/buffer.h"
  7. #include "mojo/public/c/system/data_pipe.h"
  8. #include "mojo/public/c/system/functions.h"
  9. #include "mojo/public/c/system/message_pipe.h"
  10. #include "mojo/public/c/system/trap.h"
  11. #include "mojo/public/c/system/types.h"
  12. namespace mojo {
  13. namespace core {
  14. namespace {
  15. using SignalsTest = test::MojoTestBase;
  16. TEST_F(SignalsTest, QueryInvalidArguments) {
  17. MojoHandleSignalsState state = {0, 0};
  18. EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT,
  19. MojoQueryHandleSignalsState(MOJO_HANDLE_INVALID, &state));
  20. MojoHandle a, b;
  21. CreateMessagePipe(&a, &b);
  22. EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT,
  23. MojoQueryHandleSignalsState(a, nullptr));
  24. }
  25. TEST_F(SignalsTest, QueryMessagePipeSignals) {
  26. MojoHandleSignalsState state = {0, 0};
  27. MojoHandle a, b;
  28. CreateMessagePipe(&a, &b);
  29. EXPECT_EQ(MOJO_RESULT_OK, MojoQueryHandleSignalsState(a, &state));
  30. EXPECT_EQ(MOJO_HANDLE_SIGNAL_WRITABLE, state.satisfied_signals);
  31. EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE |
  32. MOJO_HANDLE_SIGNAL_PEER_CLOSED |
  33. MOJO_HANDLE_SIGNAL_PEER_REMOTE |
  34. MOJO_HANDLE_SIGNAL_QUOTA_EXCEEDED,
  35. state.satisfiable_signals);
  36. EXPECT_EQ(MOJO_RESULT_OK, MojoQueryHandleSignalsState(b, &state));
  37. EXPECT_EQ(MOJO_HANDLE_SIGNAL_WRITABLE, state.satisfied_signals);
  38. EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE |
  39. MOJO_HANDLE_SIGNAL_PEER_CLOSED |
  40. MOJO_HANDLE_SIGNAL_PEER_REMOTE |
  41. MOJO_HANDLE_SIGNAL_QUOTA_EXCEEDED,
  42. state.satisfiable_signals);
  43. WriteMessage(a, "ok");
  44. EXPECT_EQ(MOJO_RESULT_OK, WaitForSignals(b, MOJO_HANDLE_SIGNAL_READABLE));
  45. EXPECT_EQ(MOJO_RESULT_OK, MojoQueryHandleSignalsState(b, &state));
  46. EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE,
  47. state.satisfied_signals);
  48. EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE |
  49. MOJO_HANDLE_SIGNAL_PEER_CLOSED |
  50. MOJO_HANDLE_SIGNAL_PEER_REMOTE |
  51. MOJO_HANDLE_SIGNAL_QUOTA_EXCEEDED,
  52. state.satisfiable_signals);
  53. EXPECT_EQ("ok", ReadMessage(b));
  54. EXPECT_EQ(MOJO_RESULT_OK, MojoQueryHandleSignalsState(b, &state));
  55. EXPECT_EQ(MOJO_HANDLE_SIGNAL_WRITABLE, state.satisfied_signals);
  56. EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE |
  57. MOJO_HANDLE_SIGNAL_PEER_CLOSED |
  58. MOJO_HANDLE_SIGNAL_PEER_REMOTE |
  59. MOJO_HANDLE_SIGNAL_QUOTA_EXCEEDED,
  60. state.satisfiable_signals);
  61. EXPECT_EQ(MOJO_RESULT_OK, MojoClose(a));
  62. EXPECT_EQ(MOJO_RESULT_OK, WaitForSignals(b, MOJO_HANDLE_SIGNAL_PEER_CLOSED));
  63. EXPECT_EQ(MOJO_RESULT_OK, MojoQueryHandleSignalsState(b, &state));
  64. EXPECT_EQ(MOJO_HANDLE_SIGNAL_PEER_CLOSED, state.satisfied_signals);
  65. EXPECT_EQ(MOJO_HANDLE_SIGNAL_PEER_CLOSED | MOJO_HANDLE_SIGNAL_QUOTA_EXCEEDED,
  66. state.satisfiable_signals);
  67. }
  68. TEST_F(SignalsTest, LocalPeers) {
  69. MojoHandleSignalsState state = {0, 0};
  70. MojoHandle a, b, c, d;
  71. CreateMessagePipe(&a, &b);
  72. CreateMessagePipe(&c, &d);
  73. EXPECT_EQ(MOJO_RESULT_OK, MojoQueryHandleSignalsState(a, &state));
  74. EXPECT_TRUE(state.satisfiable_signals & MOJO_HANDLE_SIGNAL_PEER_REMOTE);
  75. EXPECT_FALSE(state.satisfied_signals & MOJO_HANDLE_SIGNAL_PEER_REMOTE);
  76. EXPECT_EQ(MOJO_RESULT_OK, MojoQueryHandleSignalsState(b, &state));
  77. EXPECT_TRUE(state.satisfiable_signals & MOJO_HANDLE_SIGNAL_PEER_REMOTE);
  78. EXPECT_FALSE(state.satisfied_signals & MOJO_HANDLE_SIGNAL_PEER_REMOTE);
  79. EXPECT_EQ(MOJO_RESULT_OK, MojoQueryHandleSignalsState(c, &state));
  80. EXPECT_TRUE(state.satisfiable_signals & MOJO_HANDLE_SIGNAL_PEER_REMOTE);
  81. EXPECT_FALSE(state.satisfied_signals & MOJO_HANDLE_SIGNAL_PEER_REMOTE);
  82. EXPECT_EQ(MOJO_RESULT_OK, MojoQueryHandleSignalsState(d, &state));
  83. EXPECT_TRUE(state.satisfiable_signals & MOJO_HANDLE_SIGNAL_PEER_REMOTE);
  84. EXPECT_FALSE(state.satisfied_signals & MOJO_HANDLE_SIGNAL_PEER_REMOTE);
  85. // Verify that sending a local pipe over a local pipe doesn't change the
  86. // perceived locality of the peer.
  87. const char kMessage[] = "ayyy";
  88. WriteMessageWithHandles(a, kMessage, &c, 1);
  89. EXPECT_EQ(kMessage, ReadMessageWithHandles(b, &c, 1));
  90. WriteMessage(c, kMessage);
  91. EXPECT_EQ(kMessage, ReadMessage(d));
  92. EXPECT_EQ(MOJO_RESULT_OK, MojoQueryHandleSignalsState(c, &state));
  93. EXPECT_TRUE(state.satisfiable_signals & MOJO_HANDLE_SIGNAL_PEER_REMOTE);
  94. EXPECT_FALSE(state.satisfied_signals & MOJO_HANDLE_SIGNAL_PEER_REMOTE);
  95. EXPECT_EQ(MOJO_RESULT_OK, MojoQueryHandleSignalsState(d, &state));
  96. EXPECT_TRUE(state.satisfiable_signals & MOJO_HANDLE_SIGNAL_PEER_REMOTE);
  97. EXPECT_FALSE(state.satisfied_signals & MOJO_HANDLE_SIGNAL_PEER_REMOTE);
  98. // Sanity check: a closed peer can never signal remoteness.
  99. EXPECT_EQ(MOJO_RESULT_OK, MojoClose(c));
  100. EXPECT_EQ(MOJO_RESULT_OK, MojoQueryHandleSignalsState(d, &state));
  101. EXPECT_FALSE(state.satisfiable_signals & MOJO_HANDLE_SIGNAL_PEER_REMOTE);
  102. EXPECT_EQ(MOJO_RESULT_OK, MojoClose(d));
  103. EXPECT_EQ(MOJO_RESULT_OK, MojoClose(a));
  104. EXPECT_EQ(MOJO_RESULT_OK, MojoClose(b));
  105. }
  106. #if !BUILDFLAG(IS_IOS)
  107. TEST_F(SignalsTest, RemotePeers) {
  108. MojoHandleSignalsState state = {0, 0};
  109. MojoHandle a, b;
  110. CreateMessagePipe(&a, &b);
  111. EXPECT_EQ(MOJO_RESULT_OK, MojoQueryHandleSignalsState(a, &state));
  112. EXPECT_TRUE(state.satisfiable_signals & MOJO_HANDLE_SIGNAL_PEER_REMOTE);
  113. EXPECT_FALSE(state.satisfied_signals & MOJO_HANDLE_SIGNAL_PEER_REMOTE);
  114. EXPECT_EQ(MOJO_RESULT_OK,
  115. WaitForSignals(a, MOJO_HANDLE_SIGNAL_PEER_REMOTE,
  116. MOJO_TRIGGER_CONDITION_SIGNALS_UNSATISFIED));
  117. EXPECT_EQ(MOJO_RESULT_OK, MojoQueryHandleSignalsState(b, &state));
  118. EXPECT_TRUE(state.satisfiable_signals & MOJO_HANDLE_SIGNAL_PEER_REMOTE);
  119. EXPECT_FALSE(state.satisfied_signals & MOJO_HANDLE_SIGNAL_PEER_REMOTE);
  120. EXPECT_EQ(MOJO_RESULT_OK,
  121. WaitForSignals(b, MOJO_HANDLE_SIGNAL_PEER_REMOTE,
  122. MOJO_TRIGGER_CONDITION_SIGNALS_UNSATISFIED));
  123. RunTestClient("RemotePeersClient", [&](MojoHandle h) {
  124. // The bootstrap pipe should eventually signal remoteness.
  125. EXPECT_EQ(MOJO_RESULT_OK,
  126. WaitForSignals(h, MOJO_HANDLE_SIGNAL_PEER_REMOTE,
  127. MOJO_TRIGGER_CONDITION_SIGNALS_SATISFIED));
  128. // And so should |a| after we send its peer.
  129. WriteMessageWithHandles(h, ":)", &b, 1);
  130. EXPECT_EQ(MOJO_RESULT_OK,
  131. WaitForSignals(a, MOJO_HANDLE_SIGNAL_PEER_REMOTE,
  132. MOJO_TRIGGER_CONDITION_SIGNALS_SATISFIED));
  133. EXPECT_EQ(MOJO_RESULT_OK, MojoQueryHandleSignalsState(a, &state));
  134. EXPECT_TRUE(state.satisfied_signals & MOJO_HANDLE_SIGNAL_PEER_REMOTE);
  135. // And so should |c| after we fuse |d| to |a|.
  136. MojoHandle c, d;
  137. CreateMessagePipe(&c, &d);
  138. EXPECT_EQ(MOJO_RESULT_OK, MojoFuseMessagePipes(d, a, nullptr));
  139. EXPECT_EQ(MOJO_RESULT_OK,
  140. WaitForSignals(c, MOJO_HANDLE_SIGNAL_PEER_REMOTE,
  141. MOJO_TRIGGER_CONDITION_SIGNALS_SATISFIED));
  142. EXPECT_EQ(MOJO_RESULT_OK, MojoQueryHandleSignalsState(c, &state));
  143. EXPECT_TRUE(state.satisfied_signals & MOJO_HANDLE_SIGNAL_PEER_REMOTE);
  144. // We fused c-d to a-b, so we'll just sort of "rename" |c| back to |a| so
  145. // the system resembles the state it was in before we did that.
  146. a = c;
  147. WriteMessage(h, "OK!");
  148. // Read |b| back before joining the client.
  149. EXPECT_EQ("O_O", ReadMessageWithHandles(h, &b, 1));
  150. // Wait for |a| to see its peer as local again.
  151. EXPECT_EQ(MOJO_RESULT_OK,
  152. WaitForSignals(a, MOJO_HANDLE_SIGNAL_PEER_REMOTE,
  153. MOJO_TRIGGER_CONDITION_SIGNALS_UNSATISFIED));
  154. EXPECT_EQ(MOJO_RESULT_OK, MojoQueryHandleSignalsState(a, &state));
  155. EXPECT_FALSE(state.satisfied_signals & MOJO_HANDLE_SIGNAL_PEER_REMOTE);
  156. });
  157. }
  158. DEFINE_TEST_CLIENT_TEST_WITH_PIPE(RemotePeersClient, SignalsTest, h) {
  159. // The bootstrap pipe should eventually signal remoteness.
  160. EXPECT_EQ(MOJO_RESULT_OK,
  161. WaitForSignals(h, MOJO_HANDLE_SIGNAL_PEER_REMOTE,
  162. MOJO_TRIGGER_CONDITION_SIGNALS_SATISFIED));
  163. MojoHandle b;
  164. EXPECT_EQ(":)", ReadMessageWithHandles(h, &b, 1));
  165. // And so should |b|.
  166. EXPECT_EQ(MOJO_RESULT_OK,
  167. WaitForSignals(b, MOJO_HANDLE_SIGNAL_PEER_REMOTE,
  168. MOJO_TRIGGER_CONDITION_SIGNALS_SATISFIED));
  169. // Wait for the test to signal that it's ready to read |b| back.
  170. EXPECT_EQ("OK!", ReadMessage(h));
  171. // Now send |b| back home.
  172. WriteMessageWithHandles(h, "O_O", &b, 1);
  173. }
  174. #endif // !BUILDFLAG(IS_IOS)
  175. } // namespace
  176. } // namespace core
  177. } // namespace mojo