src/net/url_request/url_request_simple_job.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 "net/url_request/url_request_simple_job.h"
   6
   7 #include "base/bind.h"
   8 #include "base/compiler_specific.h"
   9 #include "base/message_loop/message_loop.h"
  10 #include "net/base/io_buffer.h"
  11 #include "net/base/net_errors.h"
  12 #include "net/url_request/url_request_status.h"
  13
  14 namespace net {
  15
  16 URLRequestSimpleJob::URLRequestSimpleJob(
  17     URLRequest* request, NetworkDelegate* network_delegate)
  18     : URLRequestJob(request, network_delegate),
  19       data_offset_(0),
  20       weak_factory_(this) {}
  21
  22 void URLRequestSimpleJob::Start() {
  23   // Start reading asynchronously so that all error reporting and data
  24   // callbacks happen as they would for network requests.
  25   base::MessageLoop::current()->PostTask(
  26       FROM_HERE,
  27       base::Bind(&URLRequestSimpleJob::StartAsync, weak_factory_.GetWeakPtr()));
  28 }
  29
  30 bool URLRequestSimpleJob::GetMimeType(std::string* mime_type) const {
  31   *mime_type = mime_type_;
  32   return true;
  33 }
  34
  35 bool URLRequestSimpleJob::GetCharset(std::string* charset) {
  36   *charset = charset_;
  37   return true;
  38 }
  39
  40 URLRequestSimpleJob::~URLRequestSimpleJob() {}
  41
  42 bool URLRequestSimpleJob::ReadRawData(IOBuffer* buf, int buf_size,
  43                                       int* bytes_read) {
  44   DCHECK(bytes_read);
  45   int remaining = static_cast<int>(data_.size()) - data_offset_;
  46   if (buf_size > remaining)
  47     buf_size = remaining;
  48   memcpy(buf->data(), data_.data() + data_offset_, buf_size);
  49   data_offset_ += buf_size;
  50   *bytes_read = buf_size;
  51   return true;
  52 }
  53
  54 void URLRequestSimpleJob::StartAsync() {
  55   if (!request_)
  56     return;
  57
  58   int result = GetData(&mime_type_, &charset_, &data_,
  59                        base::Bind(&URLRequestSimpleJob::OnGetDataCompleted,
  60                                   weak_factory_.GetWeakPtr()));
  61   if (result != ERR_IO_PENDING)
  62     OnGetDataCompleted(result);
  63 }
  64
  65 void URLRequestSimpleJob::OnGetDataCompleted(int result) {
  66   if (result == OK) {
  67     // Notify that the headers are complete
  68     NotifyHeadersComplete();
  69   } else {
  70     NotifyStartError(URLRequestStatus(URLRequestStatus::FAILED, result));
  71   }
  72 }
  73
  74 }  // namespace net