1 // Copyright 2013 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
5 #include "remoting/host/native_messaging/native_messaging_channel.h"
6
7 #include "base/basictypes.h"
8 #include "base/bind.h"
9 #include "base/callback.h"
10 #include "base/callback_helpers.h"
11 #include "base/location.h"
12 #include "base/values.h"
13
14 #if defined(OS_POSIX)
15 #include <unistd.h>
16 #endif
17
18 namespace {
19
DuplicatePlatformFile(base::PlatformFile handle)20 base::PlatformFile DuplicatePlatformFile(base::PlatformFile handle) {
21 base::PlatformFile result;
22 #if defined(OS_WIN)
23 if (!DuplicateHandle(GetCurrentProcess(),
24 handle,
25 GetCurrentProcess(),
26 &result,
27 0,
28 FALSE,
29 DUPLICATE_CLOSE_SOURCE | DUPLICATE_SAME_ACCESS)) {
30 PLOG(ERROR) << "Failed to duplicate handle " << handle;
31 return base::kInvalidPlatformFileValue;
32 }
33 return result;
34 #elif defined(OS_POSIX)
35 result = dup(handle);
36 base::ClosePlatformFile(handle);
37 return result;
38 #else
39 #error Not implemented.
40 #endif
41 }
42
43 } // namespace
44
45 namespace remoting {
46
NativeMessagingChannel(base::PlatformFile input,base::PlatformFile output)47 NativeMessagingChannel::NativeMessagingChannel(
48 base::PlatformFile input,
49 base::PlatformFile output)
50 : native_messaging_reader_(DuplicatePlatformFile(input)),
51 native_messaging_writer_(new NativeMessagingWriter(
52 DuplicatePlatformFile(output))),
53 weak_factory_(this) {
54 weak_ptr_ = weak_factory_.GetWeakPtr();
55 }
56
~NativeMessagingChannel()57 NativeMessagingChannel::~NativeMessagingChannel() {
58 }
59
Start(const SendMessageCallback & received_message,const base::Closure & quit_closure)60 void NativeMessagingChannel::Start(const SendMessageCallback& received_message,
61 const base::Closure& quit_closure) {
62 DCHECK(CalledOnValidThread());
63 DCHECK(received_message_.is_null());
64 DCHECK(quit_closure_.is_null());
65
66 received_message_ = received_message;
67 quit_closure_ = quit_closure;
68
69 native_messaging_reader_.Start(
70 base::Bind(&NativeMessagingChannel::ProcessMessage, weak_ptr_),
71 base::Bind(&NativeMessagingChannel::Shutdown, weak_ptr_));
72 }
73
ProcessMessage(scoped_ptr<base::Value> message)74 void NativeMessagingChannel::ProcessMessage(scoped_ptr<base::Value> message) {
75 DCHECK(CalledOnValidThread());
76
77 if (message->GetType() != base::Value::TYPE_DICTIONARY) {
78 LOG(ERROR) << "Expected DictionaryValue";
79 Shutdown();
80 return;
81 }
82
83 scoped_ptr<base::DictionaryValue> message_dict(
84 static_cast<base::DictionaryValue*>(message.release()));
85 received_message_.Run(message_dict.Pass());
86 }
87
SendMessage(scoped_ptr<base::DictionaryValue> message)88 void NativeMessagingChannel::SendMessage(
89 scoped_ptr<base::DictionaryValue> message) {
90 DCHECK(CalledOnValidThread());
91
92 bool success = message && native_messaging_writer_;
93 if (success)
94 success = native_messaging_writer_->WriteMessage(*message);
95
96 if (!success) {
97 // Close the write pipe so no more responses will be sent.
98 native_messaging_writer_.reset();
99 Shutdown();
100 }
101 }
102
Shutdown()103 void NativeMessagingChannel::Shutdown() {
104 DCHECK(CalledOnValidThread());
105
106 if (!quit_closure_.is_null())
107 base::ResetAndReturn(&quit_closure_).Run();
108 }
109
110 } // namespace remoting
111