content/browser/download/save_file_resource_handler.h

   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 #ifndef CONTENT_BROWSER_DOWNLOAD_SAVE_FILE_RESOURCE_HANDLER_H_
   6 #define CONTENT_BROWSER_DOWNLOAD_SAVE_FILE_RESOURCE_HANDLER_H_
   7
   8 #include <string>
   9
  10 #include "base/memory/ref_counted.h"
  11 #include "content/browser/loader/resource_handler.h"
  12 #include "url/gurl.h"
  13
  14 namespace net {
  15 class URLRequest;
  16 }
  17
  18 namespace content {
  19 class SaveFileManager;
  20
  21 // Forwards data to the save thread.
  22 class SaveFileResourceHandler : public ResourceHandler {
  23  public:
  24   SaveFileResourceHandler(net::URLRequest* request,
  25                           int render_process_host_id,
  26                           int render_view_id,
  27                           const GURL& url,
  28                           SaveFileManager* manager);
  29   ~SaveFileResourceHandler() override;
  30
  31   // ResourceHandler Implementation:
  32   bool OnUploadProgress(uint64 position, uint64 size) override;
  33
  34   // Saves the redirected URL to final_url_, we need to use the original
  35   // URL to match original request.
  36   bool OnRequestRedirected(const net::RedirectInfo& redirect_info,
  37                            ResourceResponse* response,
  38                            bool* defer) override;
  39
  40   // Sends the download creation information to the download thread.
  41   bool OnResponseStarted(ResourceResponse* response, bool* defer) override;
  42
  43   // Pass-through implementation.
  44   bool OnWillStart(const GURL& url, bool* defer) override;
  45
  46   // Pass-through implementation.
  47   bool OnBeforeNetworkStart(const GURL& url, bool* defer) override;
  48
  49   // Creates a new buffer, which will be handed to the download thread for file
  50   // writing and deletion.
  51   bool OnWillRead(scoped_refptr<net::IOBuffer>* buf,
  52                   int* buf_size,
  53                   int min_size) override;
  54
  55   // Passes the buffer to the download file writer.
  56   bool OnReadCompleted(int bytes_read, bool* defer) override;
  57
  58   void OnResponseCompleted(const net::URLRequestStatus& status,
  59                            const std::string& security_info,
  60                            bool* defer) override;
  61
  62   // N/A to this flavor of SaveFileResourceHandler.
  63   void OnDataDownloaded(int bytes_downloaded) override;
  64
  65   // If the content-length header is not present (or contains something other
  66   // than numbers), StringToInt64 returns 0, which indicates 'unknown size' and
  67   // is handled correctly by the SaveManager.
  68   void set_content_length(const std::string& content_length);
  69
  70   void set_content_disposition(const std::string& content_disposition) {
  71     content_disposition_ = content_disposition;
  72   }
  73
  74  private:
  75   int save_id_;
  76   int render_process_id_;
  77   int render_view_id_;
  78   scoped_refptr<net::IOBuffer> read_buffer_;
  79   std::string content_disposition_;
  80   GURL url_;
  81   GURL final_url_;
  82   int64 content_length_;
  83   SaveFileManager* save_manager_;
  84
  85   static const int kReadBufSize = 32768;  // bytes
  86
  87   DISALLOW_COPY_AND_ASSIGN(SaveFileResourceHandler);
  88 };
  89
  90 }  // namespace content
  91
  92 #endif  // CONTENT_BROWSER_DOWNLOAD_SAVE_FILE_RESOURCE_HANDLER_H_