Create a broker interface for the new Mojo EDK so that the browser can create and duplicate messa...

mojo/edk/system/parent_token_serializer_win.cc

+// Copyright 2015 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "mojo/edk/system/parent_token_serializer_win.h"
+
+#include "base/bind.h"
+#include "base/lazy_instance.h"
+#include "mojo/edk/embedder/platform_channel_pair.h"
+#include "mojo/edk/system/configuration.h"
+#include "mojo/edk/system/parent_token_serializer_state_win.h"
+#include "mojo/edk/system/token_serializer_messages_win.h"
+
+namespace mojo {
+namespace edk {
+
+namespace {
+static const int kDefaultReadBufferSize = 256;
+}
+
+ParentTokenSerializer::ParentTokenSerializer(HANDLE child_process,
+                                             ScopedPlatformHandle pipe)
+    : child_process_(child_process),
+      pipe_(pipe.Pass()),
+      num_bytes_read_(0) {
+  memset(&read_context_.overlapped, 0, sizeof(read_context_.overlapped));
+  read_context_.handler = this;
+  memset(&write_context_.overlapped, 0, sizeof(write_context_.overlapped));
+  write_context_.handler = this;
+
+  read_data_.resize(kDefaultReadBufferSize);
+  ParentTokenSerializerState::GetInstance()->token_serialize_thread()->PostTask(
+      FROM_HERE,
+      base::Bind(&ParentTokenSerializer::RegisterIOHandler,
+                 base::Unretained(this)));
+}
+
+ParentTokenSerializer::~ParentTokenSerializer() {
+}
+
+void ParentTokenSerializer::RegisterIOHandler() {
+  base::MessageLoopForIO::current()->RegisterIOHandler(
+      pipe_.get().handle, this);
+  BeginRead();
+}
+
+void ParentTokenSerializer::BeginRead() {
+  BOOL rv = ReadFile(pipe_.get().handle, &read_data_[num_bytes_read_],
+                     static_cast<int>(read_data_.size() - num_bytes_read_),
+                     nullptr, &read_context_.overlapped);
+  if (rv || GetLastError() == ERROR_IO_PENDING)
+    return;
+
+  if (rv == ERROR_BROKEN_PIPE) {
+    delete this;
+    return;
+  }
+
+  NOTREACHED() << "Unknown error in ParentTokenSerializer " << rv;
+}
+
+void ParentTokenSerializer::OnIOCompleted(
+    base::MessageLoopForIO::IOContext* context,
+    DWORD bytes_transferred,
+    DWORD error) {
+  if (context != &read_context_)
+    return;
+
+  if (error == ERROR_BROKEN_PIPE) {
+    delete this;
+    return;  // Child process exited or crashed.
+  }
+
+  if (error != ERROR_SUCCESS) {
+    NOTREACHED() << "Error " << error << " in ParentTokenSerializer.";
+    delete this;
+    return;
+  }
+
+  num_bytes_read_ += bytes_transferred;
+  CHECK_GE(num_bytes_read_, sizeof(uint32_t));
+  TokenSerializerMessage* message =
+      reinterpret_cast<TokenSerializerMessage*>(&read_data_[0]);
+  if (num_bytes_read_ < message->size) {
+    read_data_.resize(message->size);
+    BeginRead();
+    return;
+  }
+
+  if (message->id == CREATE_PLATFORM_CHANNEL_PAIR) {
+    PlatformChannelPair channel_pair;
+    uint32_t response_size = 2 * sizeof(HANDLE);
+    write_data_.resize(response_size);
+    HANDLE* handles = reinterpret_cast<HANDLE*>(&write_data_[0]);
+    handles[0] = DuplicateToChild(
+        channel_pair.PassServerHandle().release().handle);
+    handles[1] = DuplicateToChild(
+        channel_pair.PassClientHandle().release().handle);
+  } else if (message->id == HANDLE_TO_TOKEN) {
+    uint32_t count =
+        (message->size - kTokenSerializerMessageHeaderSize) / sizeof(HANDLE);
+    if (count > GetConfiguration().max_message_num_handles) {
+      NOTREACHED() << "Too many handles from child process. Closing channel.";
+      delete this;
+      return;
+    }
+    uint32_t response_size = count * sizeof(uint64_t);
+    write_data_.resize(response_size);
+    uint64_t* tokens = reinterpret_cast<uint64_t*>(&write_data_[0]);
+    std::vector<PlatformHandle> duplicated_handles(count);
+    for (uint32_t i = 0; i < count; ++i) {
+      duplicated_handles[i] =
+          PlatformHandle(DuplicateFromChild(message->handles[i]));
+    }
+    ParentTokenSerializerState::GetInstance()->HandleToToken(
+        &duplicated_handles[0], count, tokens);
+  } else if (message->id == TOKEN_TO_HANDLE) {
+    uint32_t count =
+        (message->size - kTokenSerializerMessageHeaderSize) /
+        sizeof(uint64_t);
+    if (count > GetConfiguration().max_message_num_handles) {
+      NOTREACHED() << "Too many tokens from child process. Closing channel.";
+      delete this;
+      return;
+    }
+    uint32_t response_size = count * sizeof(HANDLE);
+    write_data_.resize(response_size);
+    HANDLE* handles = reinterpret_cast<HANDLE*>(&write_data_[0]);
+    std::vector<PlatformHandle> temp_handles(count);
+    ParentTokenSerializerState::GetInstance()->TokenToHandle(
+        &message->tokens[0], count, &temp_handles[0]);
+    for (uint32_t i = 0; i < count; ++i) {
+      if (temp_handles[i].is_valid()) {
+        handles[i] = DuplicateToChild(temp_handles[i].handle);
+      } else {
+        NOTREACHED() << "Unknown token";
+        handles[i] = INVALID_HANDLE_VALUE;
+      }
+    }
+  } else {
+    NOTREACHED() << "Unknown command. Stopping reading.";
+    delete this;
+    return;
+  }
+
+  BOOL rv = WriteFile(pipe_.get().handle, &write_data_[0],
+                      static_cast<int>(write_data_.size()), NULL,
+                      &write_context_.overlapped);
+  DCHECK(rv || GetLastError() == ERROR_IO_PENDING);
+
+  // Start reading again.
+  num_bytes_read_ = 0;
+  BeginRead();
+}
+
+
+HANDLE ParentTokenSerializer::DuplicateToChild(HANDLE handle) {
+  HANDLE rv = INVALID_HANDLE_VALUE;
+  BOOL result = DuplicateHandle(base::GetCurrentProcessHandle(), handle,
+                                child_process_, &rv, 0, FALSE,
+                                DUPLICATE_SAME_ACCESS | DUPLICATE_CLOSE_SOURCE);
+  DCHECK(result);
+  return rv;
+}
+
+HANDLE ParentTokenSerializer::DuplicateFromChild(HANDLE handle) {
+  HANDLE rv = INVALID_HANDLE_VALUE;
+  BOOL result = DuplicateHandle(child_process_, handle,
+                                base::GetCurrentProcessHandle(), &rv, 0, FALSE,
+                                DUPLICATE_SAME_ACCESS | DUPLICATE_CLOSE_SOURCE);
+  DCHECK(result);
+  return rv;
+}
+
+}  // namespace edk
+}  // namespace mojo