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
15namespace remoting {
16
17namespace {
18int kReadBufferSize = 4096;
19}  // namespace
20
21SocketReader::SocketReader()
22    : socket_(NULL),
23      weak_factory_(this) {
24}
25
26SocketReader::~SocketReader() {
27}
28
29void 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
39void SocketReader::DoRead() {
40  while (true) {
41    read_buffer_ = new net::IOBuffer(kReadBufferSize);
42    int result = socket_->Read(
43        read_buffer_.get(),
44        kReadBufferSize,
45        base::Bind(&SocketReader::OnRead, weak_factory_.GetWeakPtr()));
46    HandleReadResult(result);
47    if (result <= 0)
48      break;
49  }
50}
51
52void SocketReader::OnRead(int result) {
53  HandleReadResult(result);
54  if (result > 0)
55    DoRead();
56}
57
58void SocketReader::HandleReadResult(int result) {
59  if (result != net::ERR_IO_PENDING) {
60    if (result < 0)
61      read_buffer_ = NULL;
62    base::ThreadTaskRunnerHandle::Get()->PostTask(
63        FROM_HERE,
64        base::Bind(&SocketReader::CallCallback, weak_factory_.GetWeakPtr(),
65                   read_buffer_, result));
66  }
67}
68
69void SocketReader::CallCallback(scoped_refptr<net::IOBuffer> data, int result) {
70  read_result_callback_.Run(data, result);
71}
72
73}  // namespace remoting
74