url_request_simple_job.cc revision 5d1f7b1de12d16ceb2c938c56701a3e8bfa558f7 
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 <vector>
8
9#include "base/bind.h"
10#include "base/compiler_specific.h"
11#include "base/message_loop/message_loop.h"
12#include "net/base/io_buffer.h"
13#include "net/base/net_errors.h"
14#include "net/http/http_request_headers.h"
15#include "net/http/http_util.h"
16#include "net/url_request/url_request_status.h"
17
18namespace net {
19
20URLRequestSimpleJob::URLRequestSimpleJob(
21    URLRequest* request, NetworkDelegate* network_delegate)
22    : URLRangeRequestJob(request, network_delegate),
23      data_offset_(0),
24      weak_factory_(this) {}
25
26void URLRequestSimpleJob::Start() {
27  // Start reading asynchronously so that all error reporting and data
28  // callbacks happen as they would for network requests.
29  base::MessageLoop::current()->PostTask(
30      FROM_HERE,
31      base::Bind(&URLRequestSimpleJob::StartAsync, weak_factory_.GetWeakPtr()));
32}
33
34bool URLRequestSimpleJob::GetMimeType(std::string* mime_type) const {
35  *mime_type = mime_type_;
36  return true;
37}
38
39bool URLRequestSimpleJob::GetCharset(std::string* charset) {
40  *charset = charset_;
41  return true;
42}
43
44URLRequestSimpleJob::~URLRequestSimpleJob() {}
45
46bool URLRequestSimpleJob::ReadRawData(IOBuffer* buf, int buf_size,
47                                      int* bytes_read) {
48  DCHECK(bytes_read);
49  int remaining = byte_range_.last_byte_position() - data_offset_ + 1;
50  if (buf_size > remaining)
51    buf_size = remaining;
52  memcpy(buf->data(), data_.data() + data_offset_, buf_size);
53  data_offset_ += buf_size;
54  *bytes_read = buf_size;
55  return true;
56}
57
58void URLRequestSimpleJob::StartAsync() {
59  if (!request_)
60    return;
61
62  if (ranges().size() > 1) {
63    NotifyDone(URLRequestStatus(URLRequestStatus::FAILED,
64                                ERR_REQUEST_RANGE_NOT_SATISFIABLE));
65    return;
66  }
67
68  if (!ranges().empty() && range_parse_result() == OK)
69    byte_range_ = ranges().front();
70
71  int result = GetData(&mime_type_, &charset_, &data_,
72                       base::Bind(&URLRequestSimpleJob::OnGetDataCompleted,
73                                  weak_factory_.GetWeakPtr()));
74  if (result != ERR_IO_PENDING)
75    OnGetDataCompleted(result);
76}
77
78void URLRequestSimpleJob::OnGetDataCompleted(int result) {
79  if (result == OK) {
80    // Notify that the headers are complete
81    if (!byte_range_.ComputeBounds(data_.size())) {
82      NotifyDone(URLRequestStatus(URLRequestStatus::FAILED,
83                 ERR_REQUEST_RANGE_NOT_SATISFIABLE));
84      return;
85    }
86
87    data_offset_ = byte_range_.first_byte_position();
88    int remaining_bytes = byte_range_.last_byte_position() -
89        byte_range_.first_byte_position() + 1;
90    set_expected_content_size(remaining_bytes);
91    NotifyHeadersComplete();
92  } else {
93    NotifyStartError(URLRequestStatus(URLRequestStatus::FAILED, result));
94  }
95}
96
97}  // namespace net
98