save_file_resource_handler.cc revision 3345a6884c488ff3a535c2c9acdd33d74b37e311 
1// Copyright (c) 2010 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 "chrome/browser/renderer_host/save_file_resource_handler.h"
6
7#include "base/logging.h"
8#include "base/message_loop.h"
9#include "base/string_number_conversions.h"
10#include "chrome/browser/chrome_thread.h"
11#include "chrome/browser/download/save_file_manager.h"
12#include "net/base/io_buffer.h"
13#include "net/url_request/url_request_status.h"
14
15SaveFileResourceHandler::SaveFileResourceHandler(int render_process_host_id,
16                                                 int render_view_id,
17                                                 const GURL& url,
18                                                 SaveFileManager* manager)
19    : save_id_(-1),
20      render_process_id_(render_process_host_id),
21      render_view_id_(render_view_id),
22      url_(url),
23      content_length_(0),
24      save_manager_(manager) {
25}
26
27bool SaveFileResourceHandler::OnUploadProgress(int request_id,
28                                               uint64 position,
29                                               uint64 size) {
30  return true;
31}
32
33bool SaveFileResourceHandler::OnRequestRedirected(int request_id,
34                                                  const GURL& url,
35                                                  ResourceResponse* response,
36                                                  bool* defer) {
37  final_url_ = url;
38  return true;
39}
40
41bool SaveFileResourceHandler::OnResponseStarted(int request_id,
42                                                ResourceResponse* response) {
43  save_id_ = save_manager_->GetNextId();
44  // |save_manager_| consumes (deletes):
45  SaveFileCreateInfo* info = new SaveFileCreateInfo;
46  info->url = url_;
47  info->final_url = final_url_;
48  info->total_bytes = content_length_;
49  info->save_id = save_id_;
50  info->render_process_id = render_process_id_;
51  info->render_view_id = render_view_id_;
52  info->request_id = request_id;
53  info->content_disposition = content_disposition_;
54  info->save_source = SaveFileCreateInfo::SAVE_FILE_FROM_NET;
55  ChromeThread::PostTask(
56      ChromeThread::FILE, FROM_HERE,
57      NewRunnableMethod(save_manager_,
58                        &SaveFileManager::StartSave,
59                        info));
60  return true;
61}
62
63bool SaveFileResourceHandler::OnWillStart(int request_id,
64                                          const GURL& url,
65                                          bool* defer) {
66  return true;
67}
68
69bool SaveFileResourceHandler::OnWillRead(int request_id, net::IOBuffer** buf,
70                                         int* buf_size, int min_size) {
71  DCHECK(buf && buf_size);
72  if (!read_buffer_) {
73    *buf_size = min_size < 0 ? kReadBufSize : min_size;
74    read_buffer_ = new net::IOBuffer(*buf_size);
75  }
76  *buf = read_buffer_.get();
77  return true;
78}
79
80bool SaveFileResourceHandler::OnReadCompleted(int request_id, int* bytes_read) {
81  DCHECK(read_buffer_);
82  // We are passing ownership of this buffer to the save file manager.
83  net::IOBuffer* buffer = NULL;
84  read_buffer_.swap(&buffer);
85  ChromeThread::PostTask(
86      ChromeThread::FILE, FROM_HERE,
87      NewRunnableMethod(save_manager_,
88                        &SaveFileManager::UpdateSaveProgress,
89                        save_id_,
90                        buffer,
91                        *bytes_read));
92  return true;
93}
94
95bool SaveFileResourceHandler::OnResponseCompleted(
96    int request_id,
97    const URLRequestStatus& status,
98    const std::string& security_info) {
99  ChromeThread::PostTask(
100      ChromeThread::FILE, FROM_HERE,
101      NewRunnableMethod(save_manager_,
102                        &SaveFileManager::SaveFinished,
103                        save_id_,
104                        url_,
105                        render_process_id_,
106                        status.is_success() && !status.is_io_pending()));
107  read_buffer_ = NULL;
108  return true;
109}
110
111void SaveFileResourceHandler::OnRequestClosed() {
112}
113
114void SaveFileResourceHandler::set_content_length(
115    const std::string& content_length) {
116  base::StringToInt64(content_length, &content_length_);
117}
118