remoting/base/socket_reader.cc

   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