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

Side by Side 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 unified diff | Download patch | Annotate | Revision Log
OLDNEW
(Empty)
1 // Copyright (c) 2012 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/base/socket_reader.h"
6
7 #include "base/compiler_specific.h"
8 #include "base/location.h"
9 #include "base/single_thread_task_runner.h"
10 #include "base/thread_task_runner_handle.h"
11 #include "net/base/io_buffer.h"
12 #include "net/base/net_errors.h"
13 #include "net/socket/socket.h"
14
15 namespace remoting {
16
17 namespace {
18 int kReadBufferSize = 4096;
19 } // namespace
20
21 SocketReader::SocketReader()
22 : socket_(NULL),
23 ALLOW_THIS_IN_INITIALIZER_LIST(weak_factory_(this)) {
24 }
25
26 SocketReader::~SocketReader() {
27 }
28
29 void SocketReader::Init(net::Socket* socket,
30 ReadResultCallback read_result_callback) {
31 DCHECK(socket);
32 DCHECK(!socket_);
33
34 socket_ = socket;
35 read_result_callback_ = read_result_callback;
36 DoRead();
37 }
38
39 void SocketReader::DoRead() {
40 while (true) {
41 read_buffer_ = new net::IOBuffer(kReadBufferSize);
42 int result = socket_->Read(
43 read_buffer_, kReadBufferSize, base::Bind(&SocketReader::OnRead,
44 weak_factory_.GetWeakPtr()));
45 HandleReadResult(result);
46 if (result <= 0)
47 break;
48 }
49 }
50
51 void SocketReader::OnRead(int result) {
52 HandleReadResult(result);
53 if (result > 0)
54 DoRead();
55 }
56
57 void SocketReader::HandleReadResult(int result) {
58 if (result != net::ERR_IO_PENDING) {
59 if (result < 0)
60 read_buffer_ = NULL;
61 base::ThreadTaskRunnerHandle::Get()->PostTask(
62 FROM_HERE,
63 base::Bind(&SocketReader::CallCallback, weak_factory_.GetWeakPtr(),
64 read_buffer_, result));
65 }
66 }
67
68 void SocketReader::CallCallback(scoped_refptr<net::IOBuffer> data, int result) {
69 read_result_callback_.Run(data, result);
70 }
71
72 } // namespace remoting
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698