Introduce RegistrationServer.

handler/win/registration_pipe_state_test.cc

+// Copyright 2015 The Crashpad Authors. All rights reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#include "handler/win/registration_pipe_state.h"
+
+#include <windows.h>
+
+#include "base/basictypes.h"
+#include "base/logging.h"
+#include "base/memory/scoped_ptr.h"
+#include "client/crashpad_info.h"
+#include "client/registration_protocol_win.h"
+#include "gtest/gtest.h"
+#include "handler/win/registration_test_base.h"
+#include "util/win/scoped_handle.h"
+
+namespace crashpad {
+namespace test {
+namespace {
+
+class RegistrationRegistrationPipeStateTest : public RegistrationTestBase {
+ public:
+  RegistrationRegistrationPipeStateTest() : pipe_state_() {}
+
+  void SetUp() override {
+    ScopedFileHANDLE pipe(
+        CreateNamedPipe(pipe_name().c_str(),
+                        PIPE_ACCESS_DUPLEX | FILE_FLAG_OVERLAPPED |
+                            FILE_FLAG_FIRST_PIPE_INSTANCE,
+                        PIPE_TYPE_MESSAGE | PIPE_READMODE_MESSAGE | PIPE_WAIT,
+                        1,
+                        512,  // nOutBufferSize
+                        512,  // nInBufferSize
+                        20,  // nDefaultTimeOut
+                        nullptr));  // lpSecurityAttributes
+    ASSERT_TRUE(pipe.is_valid());
+    pipe_state_.reset(new RegistrationPipeState(pipe.Pass(), &delegate()));
+  }
+
+  ~RegistrationRegistrationPipeStateTest() override {}
+
+  RegistrationPipeState& pipe_state() {
+    DCHECK(pipe_state_.get());
+    return *pipe_state_;
+  }
+
+ private:
+  scoped_ptr<RegistrationPipeState> pipe_state_;
+  DISALLOW_COPY_AND_ASSIGN(RegistrationRegistrationPipeStateTest);
+};
+
+TEST_F(RegistrationRegistrationPipeStateTest, CancelIoWhenConnectIsComplete) {
+  //  -> Connecting
+  ASSERT_TRUE(pipe_state().Initialize());
+
+  ScopedFileHANDLE client(Connect());
+
+  ASSERT_TRUE(client.is_valid());
+
+  // Connect completion.
+  ASSERT_EQ(WAIT_OBJECT_0,
+            WaitForSingleObject(pipe_state().completion_event(), INFINITE));
+
+  // Connecting -> Stopping
+  pipe_state().Stop();
+
+  // Stop completion.
+  ASSERT_EQ(WAIT_OBJECT_0,
+            WaitForSingleObject(pipe_state().completion_event(), INFINITE));
+}
+
+TEST_F(RegistrationRegistrationPipeStateTest, CancelIoWhenReadIsComplete) {
+  // -> Connecting
+  ASSERT_TRUE(pipe_state().Initialize());
+
+  ScopedFileHANDLE client(Connect());
+
+  ASSERT_TRUE(client.is_valid());
+
+  // Connect completion.
+  ASSERT_EQ(WAIT_OBJECT_0,
+            WaitForSingleObject(pipe_state().completion_event(), INFINITE));
+
+  // Connecting -> Reading
+  ASSERT_TRUE(pipe_state().OnCompletion());
+
+  RegistrationRequest request = {0};
+  CrashpadInfo crashpad_info;
+  request.client_process_id = GetCurrentProcessId();
+  request.crashpad_info_address =
+      reinterpret_cast<WinVMAddress>(&crashpad_info);
+
+  ASSERT_TRUE(WriteRequest(client.get(), &request, sizeof(request)));
+
+  // Read completion.
+  ASSERT_EQ(WAIT_OBJECT_0,
+            WaitForSingleObject(pipe_state().completion_event(), INFINITE));
+
+  // Reading -> Stopping
+  pipe_state().Stop();
+
+  // Stop completion.
+  ASSERT_EQ(WAIT_OBJECT_0,
+            WaitForSingleObject(pipe_state().completion_event(), INFINITE));
+}
+
+TEST_F(RegistrationRegistrationPipeStateTest, CancelIoWhenWriteIsComplete) {
+  // -> Connecting
+  ASSERT_TRUE(pipe_state().Initialize());
+
+  ScopedFileHANDLE client(Connect());
+
+  ASSERT_TRUE(client.is_valid());
+
+  // Connect completion.
+  ASSERT_EQ(WAIT_OBJECT_0,
+            WaitForSingleObject(pipe_state().completion_event(), INFINITE));
+
+  // Connecting -> Reading
+  ASSERT_TRUE(pipe_state().OnCompletion());
+
+  RegistrationRequest request = {0};
+  CrashpadInfo crashpad_info;
+  request.client_process_id = GetCurrentProcessId();
+  request.crashpad_info_address =
+      reinterpret_cast<WinVMAddress>(&crashpad_info);
+
+  ASSERT_TRUE(WriteRequest(client.get(), &request, sizeof(request)));
+
+  // Read completion.
+  ASSERT_EQ(WAIT_OBJECT_0,
+            WaitForSingleObject(pipe_state().completion_event(), INFINITE));
+
+  // Reading -> Writing -> Waiting for Close
+  ASSERT_TRUE(pipe_state().OnCompletion());
+
+  RegistrationResponse response = {0};
+  ASSERT_TRUE(ReadResponse(client.get(), &response));
+
+  // Waiting for Close -> Stopping
+  pipe_state().Stop();
+
+  // Stop completion.
+  ASSERT_EQ(WAIT_OBJECT_0,
+            WaitForSingleObject(pipe_state().completion_event(), INFINITE));
+}
+
+TEST_F(RegistrationRegistrationPipeStateTest, CancelIoWhenCloseIsComplete) {
+  // -> Connecting
+  ASSERT_TRUE(pipe_state().Initialize());
+
+  ScopedFileHANDLE client(Connect());
+
+  ASSERT_TRUE(client.is_valid());
+
+  // Connect completion.
+  ASSERT_EQ(WAIT_OBJECT_0,
+            WaitForSingleObject(pipe_state().completion_event(), INFINITE));
+
+  // Connecting -> Reading
+  ASSERT_TRUE(pipe_state().OnCompletion());
+
+  RegistrationRequest request = {0};
+  CrashpadInfo crashpad_info;
+  request.client_process_id = GetCurrentProcessId();
+  request.crashpad_info_address =
+      reinterpret_cast<WinVMAddress>(&crashpad_info);
+
+  ASSERT_TRUE(WriteRequest(client.get(), &request, sizeof(request)));
+
+  // Read completion.
+  ASSERT_EQ(WAIT_OBJECT_0,
+            WaitForSingleObject(pipe_state().completion_event(), INFINITE));
+
+  // Reading -> Writing -> Waiting for Close
+  ASSERT_TRUE(pipe_state().OnCompletion());
+
+  RegistrationResponse response = {0};
+  ASSERT_TRUE(ReadResponse(client.get(), &response));
+
+  client.reset();
+
+  // Wait for close completion.
+  ASSERT_EQ(WAIT_OBJECT_0,
+            WaitForSingleObject(pipe_state().completion_event(), INFINITE));
+
+  // Waiting for Close -> Stopping
+  pipe_state().Stop();
+
+  // Stop completion.
+  ASSERT_EQ(WAIT_OBJECT_0,
+            WaitForSingleObject(pipe_state().completion_event(), INFINITE));
+}
+
+TEST_F(RegistrationRegistrationPipeStateTest, FullCycle) {
+  // -> Connecting
+  ASSERT_TRUE(pipe_state().Initialize());
+
+  ScopedFileHANDLE client(Connect());
+
+  ASSERT_TRUE(client.is_valid());
+
+  // Connect completion.
+  ASSERT_EQ(WAIT_OBJECT_0,
+            WaitForSingleObject(pipe_state().completion_event(), INFINITE));
+
+  // Connecting -> Reading
+  ASSERT_TRUE(pipe_state().OnCompletion());
+
+  RegistrationRequest request = {0};
+  CrashpadInfo crashpad_info;
+  request.client_process_id = GetCurrentProcessId();
+  request.crashpad_info_address =
+      reinterpret_cast<WinVMAddress>(&crashpad_info);
+
+  ASSERT_TRUE(WriteRequest(client.get(), &request, sizeof(request)));
+
+  // Read completion.
+  ASSERT_EQ(WAIT_OBJECT_0,
+            WaitForSingleObject(pipe_state().completion_event(), INFINITE));
+
+  // Reading -> Writing -> Waiting for Close
+  ASSERT_TRUE(pipe_state().OnCompletion());
+
+  RegistrationResponse response = {0};
+  ASSERT_TRUE(ReadResponse(client.get(), &response));
+
+  client.reset();
+
+  // Wait for close completion.
+  ASSERT_EQ(WAIT_OBJECT_0,
+            WaitForSingleObject(pipe_state().completion_event(), INFINITE));
+
+  // Waiting for Close -> Reset -> Connecting
+  ASSERT_TRUE(pipe_state().OnCompletion());
+
+  client = Connect();
+  ASSERT_TRUE(client.is_valid());
+
+  pipe_state().Stop();
+
+  ASSERT_EQ(WAIT_OBJECT_0,
+            WaitForSingleObject(pipe_state().completion_event(), INFINITE));
+}
+
+}  // namespace
+}  // namespace test
+}  // namespace crashpad