Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(926)

Unified Diff: remoting/host/security_key/remote_security_key_message_reader.cc

Issue 2167473003: Revert of Renaming Gnubby and RemoteSecurityKey files/classes/members (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 4 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: remoting/host/security_key/remote_security_key_message_reader.cc
diff --git a/remoting/host/security_key/remote_security_key_message_reader.cc b/remoting/host/security_key/remote_security_key_message_reader.cc
new file mode 100644
index 0000000000000000000000000000000000000000..2cc3e762955c382e611504e4cf579796c191636d
--- /dev/null
+++ b/remoting/host/security_key/remote_security_key_message_reader.cc
@@ -0,0 +1,134 @@
+// Copyright 2016 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/security_key/remote_security_key_message_reader.h"
+
+#include <cstdint>
+#include <string>
+#include <utility>
+
+#include "base/bind.h"
+#include "base/files/file.h"
+#include "base/macros.h"
+#include "base/single_thread_task_runner.h"
+#include "base/stl_util.h"
+#include "base/threading/thread_task_runner_handle.h"
+#include "remoting/host/security_key/security_key_message.h"
+
+namespace remoting {
+
+RemoteSecurityKeyMessageReader::RemoteSecurityKeyMessageReader(
+ base::File input_file)
+ : read_stream_(std::move(input_file)),
+ reader_thread_("RemoteSecurityKeyMessageReader"),
+ weak_factory_(this) {
+ base::Thread::Options options;
+ options.message_loop_type = base::MessageLoop::TYPE_IO;
+ reader_thread_.StartWithOptions(options);
+
+ read_task_runner_ = reader_thread_.task_runner();
+ main_task_runner_ = base::ThreadTaskRunnerHandle::Get();
+}
+
+RemoteSecurityKeyMessageReader::~RemoteSecurityKeyMessageReader() {
+ DCHECK(main_task_runner_->RunsTasksOnCurrentThread());
+
+ // In order to ensure the reader thread is stopped cleanly, we close the
+ // stream it is blocking on and then wait for the thread to exit.
+ read_stream_.Close();
+ reader_thread_.Stop();
+}
+
+void RemoteSecurityKeyMessageReader::Start(
+ SecurityKeyMessageCallback message_callback,
+ base::Closure error_callback) {
+ DCHECK(main_task_runner_->RunsTasksOnCurrentThread());
+
+ message_callback_ = message_callback;
+ error_callback_ = error_callback;
+
+ // base::Unretained is safe since this class owns the thread running this task
+ // which will be destroyed before this instance is.
+ read_task_runner_->PostTask(
+ FROM_HERE, base::Bind(&RemoteSecurityKeyMessageReader::ReadMessage,
+ base::Unretained(this)));
+}
+
+void RemoteSecurityKeyMessageReader::ReadMessage() {
+ DCHECK(read_task_runner_->RunsTasksOnCurrentThread());
+
+ while (true) {
+ if (!read_stream_.IsValid()) {
+ LOG(ERROR) << "Cannot read from invalid stream.";
+ NotifyError();
+ return;
+ }
+
+ // Read the message header to retrieve the remaining message length.
+ uint32_t total_message_size_bytes;
+ int read_result = read_stream_.ReadAtCurrentPos(
+ reinterpret_cast<char*>(&total_message_size_bytes),
+ SecurityKeyMessage::kHeaderSizeBytes);
+ if (read_result != SecurityKeyMessage::kHeaderSizeBytes) {
+ // 0 means EOF which is normal and should not be logged as an error.
+ if (read_result != 0) {
+ LOG(ERROR) << "Failed to read message header, read returned "
+ << read_result;
+ }
+ NotifyError();
+ return;
+ }
+
+ if (!SecurityKeyMessage::IsValidMessageSize(total_message_size_bytes)) {
+ LOG(ERROR) << "Message size too large: " << total_message_size_bytes;
+ NotifyError();
+ return;
+ }
+
+ std::string message_data(total_message_size_bytes, '\0');
+ read_result = read_stream_.ReadAtCurrentPos(string_as_array(&message_data),
+ total_message_size_bytes);
+ // The static cast is safe as we know the value is smaller than max int.
+ if (read_result != static_cast<int>(total_message_size_bytes)) {
+ LOG(ERROR) << "Failed to read message: " << read_result;
+ NotifyError();
+ return;
+ }
+
+ std::unique_ptr<SecurityKeyMessage> message(new SecurityKeyMessage());
+ if (!message->ParseMessage(message_data)) {
+ LOG(ERROR) << "Invalid message data received.";
+ NotifyError();
+ return;
+ }
+
+ // Notify callback of the new message received.
+ main_task_runner_->PostTask(
+ FROM_HERE,
+ base::Bind(&RemoteSecurityKeyMessageReader::InvokeMessageCallback,
+ weak_factory_.GetWeakPtr(), base::Passed(&message)));
+ }
+}
+
+void RemoteSecurityKeyMessageReader::NotifyError() {
+ DCHECK(read_task_runner_->RunsTasksOnCurrentThread());
+
+ main_task_runner_->PostTask(
+ FROM_HERE,
+ base::Bind(&RemoteSecurityKeyMessageReader::InvokeErrorCallback,
+ weak_factory_.GetWeakPtr()));
+}
+
+void RemoteSecurityKeyMessageReader::InvokeMessageCallback(
+ std::unique_ptr<SecurityKeyMessage> message) {
+ DCHECK(main_task_runner_->RunsTasksOnCurrentThread());
+ message_callback_.Run(std::move(message));
+}
+
+void RemoteSecurityKeyMessageReader::InvokeErrorCallback() {
+ DCHECK(main_task_runner_->RunsTasksOnCurrentThread());
+ error_callback_.Run();
+}
+
+} // namespace remoting

Powered by Google App Engine
This is Rietveld 408576698