| OLD | NEW |
| 1 // Copyright (c) 2006-2008 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2006-2008 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef CHROME_BROWSER_RENDERER_HOST_BUFFERED_RESOURCE_HANDLER_H_ | 5 #ifndef CHROME_BROWSER_RENDERER_HOST_BUFFERED_RESOURCE_HANDLER_H_ |
| 6 #define CHROME_BROWSER_RENDERER_HOST_BUFFERED_RESOURCE_HANDLER_H_ | 6 #define CHROME_BROWSER_RENDERER_HOST_BUFFERED_RESOURCE_HANDLER_H_ |
| 7 | 7 |
| 8 #include <string> | 8 #include <string> |
| 9 | 9 |
| 10 #include "chrome/browser/renderer_host/resource_handler.h" | 10 #include "chrome/browser/renderer_host/resource_handler.h" |
| 11 | 11 |
| 12 class ResourceDispatcherHost; | 12 class ResourceDispatcherHost; |
| 13 | 13 |
| 14 // Used to buffer a request until enough data has been received. | 14 // Used to buffer a request until enough data has been received. |
| 15 class BufferedResourceHandler : public ResourceHandler { | 15 class BufferedResourceHandler : public ResourceHandler { |
| 16 public: | 16 public: |
| 17 BufferedResourceHandler(ResourceHandler* handler, | 17 BufferedResourceHandler(ResourceHandler* handler, |
| 18 ResourceDispatcherHost* host, | 18 ResourceDispatcherHost* host, |
| 19 URLRequest* request); | 19 URLRequest* request); |
| 20 | 20 |
| 21 // ResourceHandler implementation: | 21 // ResourceHandler implementation: |
| 22 bool OnUploadProgress(int request_id, uint64 position, uint64 size); | 22 bool OnUploadProgress(int request_id, uint64 position, uint64 size); |
| 23 bool OnRequestRedirected(int request_id, const GURL& new_url); | 23 bool OnRequestRedirected(int request_id, const GURL& new_url); |
| 24 bool OnResponseStarted(int request_id, ResourceResponse* response); | 24 bool OnResponseStarted(int request_id, ResourceResponse* response); |
| 25 bool OnWillRead(int request_id, char** buf, int* buf_size, int min_size); | 25 bool OnWillRead(int request_id, net::IOBuffer** buf, int* buf_size, |
| 26 int min_size); |
| 26 bool OnReadCompleted(int request_id, int* bytes_read); | 27 bool OnReadCompleted(int request_id, int* bytes_read); |
| 27 bool OnResponseCompleted(int request_id, const URLRequestStatus& status); | 28 bool OnResponseCompleted(int request_id, const URLRequestStatus& status); |
| 28 | 29 |
| 29 private: | 30 private: |
| 30 // Returns true if we should delay OnResponseStarted forwarding. | 31 // Returns true if we should delay OnResponseStarted forwarding. |
| 31 bool DelayResponse(); | 32 bool DelayResponse(); |
| 32 | 33 |
| 33 // Returns true if there will be a need to parse the DocType of the document | 34 // Returns true if there will be a need to parse the DocType of the document |
| 34 // to determine the right way to handle it. | 35 // to determine the right way to handle it. |
| 35 bool ShouldBuffer(const GURL& url, const std::string& mime_type); | 36 bool ShouldBuffer(const GURL& url, const std::string& mime_type); |
| 36 | 37 |
| 37 // Returns true if there is enough information to process the DocType. | 38 // Returns true if there is enough information to process the DocType. |
| 38 bool DidBufferEnough(int bytes_read); | 39 bool DidBufferEnough(int bytes_read); |
| 39 | 40 |
| 40 // Returns true if we have to keep buffering data. | 41 // Returns true if we have to keep buffering data. |
| 41 bool KeepBuffering(int bytes_read); | 42 bool KeepBuffering(int bytes_read); |
| 42 | 43 |
| 43 // Sends a pending OnResponseStarted notification. |in_complete| is true if | 44 // Sends a pending OnResponseStarted notification. |in_complete| is true if |
| 44 // this is invoked from |OnResponseCompleted|. | 45 // this is invoked from |OnResponseCompleted|. |
| 45 bool CompleteResponseStarted(int request_id, bool in_complete); | 46 bool CompleteResponseStarted(int request_id, bool in_complete); |
| 46 | 47 |
| 47 scoped_refptr<ResourceHandler> real_handler_; | 48 scoped_refptr<ResourceHandler> real_handler_; |
| 48 scoped_refptr<ResourceResponse> response_; | 49 scoped_refptr<ResourceResponse> response_; |
| 49 ResourceDispatcherHost* host_; | 50 ResourceDispatcherHost* host_; |
| 50 URLRequest* request_; | 51 URLRequest* request_; |
| 51 char* read_buffer_; | 52 scoped_refptr<net::IOBuffer> read_buffer_; |
| 53 scoped_refptr<net::IOBuffer> my_buffer_; |
| 52 int read_buffer_size_; | 54 int read_buffer_size_; |
| 53 int bytes_read_; | 55 int bytes_read_; |
| 54 bool sniff_content_; | 56 bool sniff_content_; |
| 55 bool should_buffer_; | 57 bool should_buffer_; |
| 56 bool buffering_; | 58 bool buffering_; |
| 57 bool finished_; | 59 bool finished_; |
| 58 | 60 |
| 59 DISALLOW_COPY_AND_ASSIGN(BufferedResourceHandler); | 61 DISALLOW_COPY_AND_ASSIGN(BufferedResourceHandler); |
| 60 }; | 62 }; |
| 61 | 63 |
| 62 #endif // CHROME_BROWSER_RENDERER_HOST_BUFFERED_RESOURCE_HANDLER_H_ | 64 #endif // CHROME_BROWSER_RENDERER_HOST_BUFFERED_RESOURCE_HANDLER_H_ |
| OLD | NEW |