| // Copyright 2013 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 "remoting/host/native_messaging/pipe_messaging_channel.h" |
| |
| #include <utility> |
| |
| #include "base/bind.h" |
| #include "base/callback.h" |
| #include "base/callback_helpers.h" |
| #include "base/location.h" |
| #include "base/values.h" |
| #include "build/build_config.h" |
| |
| #if defined(OS_POSIX) |
| #include <unistd.h> |
| |
| #include "base/posix/eintr_wrapper.h" |
| #endif |
| |
| namespace { |
| |
| base::File DuplicatePlatformFile(base::File file) { |
| base::PlatformFile result; |
| #if defined(OS_WIN) |
| if (!DuplicateHandle(GetCurrentProcess(), |
| file.TakePlatformFile(), |
| GetCurrentProcess(), |
| &result, |
| 0, |
| FALSE, |
| DUPLICATE_CLOSE_SOURCE | DUPLICATE_SAME_ACCESS)) { |
| PLOG(ERROR) << "Failed to duplicate handle " << file.GetPlatformFile(); |
| return base::File(); |
| } |
| return base::File(result); |
| #elif defined(OS_POSIX) |
| result = HANDLE_EINTR(dup(file.GetPlatformFile())); |
| return base::File(result); |
| #else |
| #error Not implemented. |
| #endif |
| } |
| |
| } // namespace |
| |
| namespace remoting { |
| |
| PipeMessagingChannel::PipeMessagingChannel(base::File input, base::File output) |
| : native_messaging_reader_(DuplicatePlatformFile(std::move(input))), |
| native_messaging_writer_( |
| new NativeMessagingWriter(DuplicatePlatformFile(std::move(output)))), |
| event_handler_(nullptr), |
| weak_factory_(this) { |
| weak_ptr_ = weak_factory_.GetWeakPtr(); |
| } |
| |
| PipeMessagingChannel::~PipeMessagingChannel() { |
| DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); |
| } |
| |
| void PipeMessagingChannel::Start(EventHandler* event_handler) { |
| DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); |
| DCHECK(!event_handler_); |
| |
| event_handler_ = event_handler; |
| DCHECK(event_handler_); |
| |
| native_messaging_reader_.Start( |
| base::Bind(&PipeMessagingChannel::ProcessMessage, weak_ptr_), |
| base::Bind(&PipeMessagingChannel::Shutdown, weak_ptr_)); |
| } |
| |
| void PipeMessagingChannel::ProcessMessage( |
| std::unique_ptr<base::Value> message) { |
| DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); |
| |
| if (event_handler_) |
| event_handler_->OnMessage(std::move(message)); |
| } |
| |
| void PipeMessagingChannel::SendMessage(std::unique_ptr<base::Value> message) { |
| DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); |
| |
| bool success = message && native_messaging_writer_; |
| if (success) |
| success = native_messaging_writer_->WriteMessage(*message); |
| |
| if (!success) { |
| // Close the write pipe so no more responses will be sent. |
| native_messaging_writer_.reset(); |
| Shutdown(); |
| } |
| } |
| |
| void PipeMessagingChannel::Shutdown() { |
| DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); |
| |
| if (event_handler_) { |
| // Set |event_handler_| to nullptr to indicate the object is in a shutdown |
| // cycle. Since event_handler->OnDisconnect() will destroy the current |
| // object, |event_handler_| will become a dangling pointer after |
| // OnDisconnect() returns. Therefore, we set |event_handler_| to nullptr |
| // beforehand. |
| EventHandler* handler = event_handler_; |
| event_handler_ = nullptr; |
| handler->OnDisconnect(); |
| } |
| } |
| |
| } // namespace remoting |