OLD | NEW |
| (Empty) |
1 // Copyright 2016 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/security_key/security_key_message_reader.h" | |
6 | |
7 #include <cstdint> | |
8 #include <string> | |
9 #include <utility> | |
10 | |
11 #include "base/bind.h" | |
12 #include "base/files/file.h" | |
13 #include "base/macros.h" | |
14 #include "base/single_thread_task_runner.h" | |
15 #include "base/stl_util.h" | |
16 #include "base/threading/thread_task_runner_handle.h" | |
17 #include "remoting/host/security_key/security_key_message.h" | |
18 | |
19 namespace remoting { | |
20 | |
21 SecurityKeyMessageReader::SecurityKeyMessageReader(base::File input_file) | |
22 : read_stream_(std::move(input_file)), | |
23 reader_thread_("SecurityKeyMessageReader"), | |
24 weak_factory_(this) { | |
25 base::Thread::Options options; | |
26 options.message_loop_type = base::MessageLoop::TYPE_IO; | |
27 reader_thread_.StartWithOptions(options); | |
28 | |
29 read_task_runner_ = reader_thread_.task_runner(); | |
30 main_task_runner_ = base::ThreadTaskRunnerHandle::Get(); | |
31 } | |
32 | |
33 SecurityKeyMessageReader::~SecurityKeyMessageReader() { | |
34 DCHECK(main_task_runner_->RunsTasksOnCurrentThread()); | |
35 | |
36 // In order to ensure the reader thread is stopped cleanly, we close the | |
37 // stream it is blocking on and then wait for the thread to exit. | |
38 read_stream_.Close(); | |
39 reader_thread_.Stop(); | |
40 } | |
41 | |
42 void SecurityKeyMessageReader::Start( | |
43 SecurityKeyMessageCallback message_callback, | |
44 base::Closure error_callback) { | |
45 DCHECK(main_task_runner_->RunsTasksOnCurrentThread()); | |
46 | |
47 message_callback_ = message_callback; | |
48 error_callback_ = error_callback; | |
49 | |
50 // base::Unretained is safe since this class owns the thread running this task | |
51 // which will be destroyed before this instance is. | |
52 read_task_runner_->PostTask(FROM_HERE, | |
53 base::Bind(&SecurityKeyMessageReader::ReadMessage, | |
54 base::Unretained(this))); | |
55 } | |
56 | |
57 void SecurityKeyMessageReader::ReadMessage() { | |
58 DCHECK(read_task_runner_->RunsTasksOnCurrentThread()); | |
59 | |
60 while (true) { | |
61 if (!read_stream_.IsValid()) { | |
62 LOG(ERROR) << "Cannot read from invalid stream."; | |
63 NotifyError(); | |
64 return; | |
65 } | |
66 | |
67 // Read the message header to retrieve the remaining message length. | |
68 uint32_t total_message_size_bytes; | |
69 int read_result = read_stream_.ReadAtCurrentPos( | |
70 reinterpret_cast<char*>(&total_message_size_bytes), | |
71 SecurityKeyMessage::kHeaderSizeBytes); | |
72 if (read_result != SecurityKeyMessage::kHeaderSizeBytes) { | |
73 // 0 means EOF which is normal and should not be logged as an error. | |
74 if (read_result != 0) { | |
75 LOG(ERROR) << "Failed to read message header, read returned " | |
76 << read_result; | |
77 } | |
78 NotifyError(); | |
79 return; | |
80 } | |
81 | |
82 if (!SecurityKeyMessage::IsValidMessageSize(total_message_size_bytes)) { | |
83 LOG(ERROR) << "Message size too large: " << total_message_size_bytes; | |
84 NotifyError(); | |
85 return; | |
86 } | |
87 | |
88 std::string message_data(total_message_size_bytes, '\0'); | |
89 read_result = read_stream_.ReadAtCurrentPos(string_as_array(&message_data), | |
90 total_message_size_bytes); | |
91 // The static cast is safe as we know the value is smaller than max int. | |
92 if (read_result != static_cast<int>(total_message_size_bytes)) { | |
93 LOG(ERROR) << "Failed to read message: " << read_result; | |
94 NotifyError(); | |
95 return; | |
96 } | |
97 | |
98 std::unique_ptr<SecurityKeyMessage> message(new SecurityKeyMessage()); | |
99 if (!message->ParseMessage(message_data)) { | |
100 LOG(ERROR) << "Invalid message data received."; | |
101 NotifyError(); | |
102 return; | |
103 } | |
104 | |
105 // Notify callback of the new message received. | |
106 main_task_runner_->PostTask( | |
107 FROM_HERE, | |
108 base::Bind(&SecurityKeyMessageReader::InvokeMessageCallback, | |
109 weak_factory_.GetWeakPtr(), base::Passed(&message))); | |
110 } | |
111 } | |
112 | |
113 void SecurityKeyMessageReader::NotifyError() { | |
114 DCHECK(read_task_runner_->RunsTasksOnCurrentThread()); | |
115 | |
116 main_task_runner_->PostTask( | |
117 FROM_HERE, base::Bind(&SecurityKeyMessageReader::InvokeErrorCallback, | |
118 weak_factory_.GetWeakPtr())); | |
119 } | |
120 | |
121 void SecurityKeyMessageReader::InvokeMessageCallback( | |
122 std::unique_ptr<SecurityKeyMessage> message) { | |
123 DCHECK(main_task_runner_->RunsTasksOnCurrentThread()); | |
124 message_callback_.Run(std::move(message)); | |
125 } | |
126 | |
127 void SecurityKeyMessageReader::InvokeErrorCallback() { | |
128 DCHECK(main_task_runner_->RunsTasksOnCurrentThread()); | |
129 error_callback_.Run(); | |
130 } | |
131 | |
132 } // namespace remoting | |
OLD | NEW |