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

Unified Diff: remoting/base/socket_reader.cc

Issue 11361197: Rename SocketReaderBase to SocketReader and move it to remoting/base (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Created 8 years, 1 month 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/base/socket_reader.cc
diff --git a/remoting/base/socket_reader.cc b/remoting/base/socket_reader.cc
new file mode 100644
index 0000000000000000000000000000000000000000..098be337b1eab49c4927da42d0b8bd46366dcca5
--- /dev/null
+++ b/remoting/base/socket_reader.cc
@@ -0,0 +1,72 @@
+// Copyright (c) 2012 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/base/socket_reader.h"
+
+#include "base/compiler_specific.h"
+#include "base/location.h"
+#include "base/single_thread_task_runner.h"
+#include "base/thread_task_runner_handle.h"
+#include "net/base/io_buffer.h"
+#include "net/base/net_errors.h"
+#include "net/socket/socket.h"
+
+namespace remoting {
+
+namespace {
+int kReadBufferSize = 4096;
+} // namespace
+
+SocketReader::SocketReader()
+ : socket_(NULL),
+ ALLOW_THIS_IN_INITIALIZER_LIST(weak_factory_(this)) {
+}
+
+SocketReader::~SocketReader() {
+}
+
+void SocketReader::Init(net::Socket* socket,
+ ReadResultCallback read_result_callback) {
+ DCHECK(socket);
+ DCHECK(!socket_);
+
+ socket_ = socket;
+ read_result_callback_ = read_result_callback;
+ DoRead();
+}
+
+void SocketReader::DoRead() {
+ while (true) {
+ read_buffer_ = new net::IOBuffer(kReadBufferSize);
+ int result = socket_->Read(
+ read_buffer_, kReadBufferSize, base::Bind(&SocketReader::OnRead,
+ weak_factory_.GetWeakPtr()));
+ HandleReadResult(result);
+ if (result <= 0)
+ break;
+ }
+}
+
+void SocketReader::OnRead(int result) {
+ HandleReadResult(result);
+ if (result > 0)
+ DoRead();
+}
+
+void SocketReader::HandleReadResult(int result) {
+ if (result != net::ERR_IO_PENDING) {
+ if (result < 0)
+ read_buffer_ = NULL;
+ base::ThreadTaskRunnerHandle::Get()->PostTask(
+ FROM_HERE,
+ base::Bind(&SocketReader::CallCallback, weak_factory_.GetWeakPtr(),
+ read_buffer_, result));
+ }
+}
+
+void SocketReader::CallCallback(scoped_refptr<net::IOBuffer> data, int result) {
+ read_result_callback_.Run(data, result);
+}
+
+} // namespace remoting

Powered by Google App Engine
This is Rietveld 408576698