OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 NET_HTTP_HTTP_RESPONSE_BODY_DRAINER_H_ | 5 #ifndef NET_HTTP_HTTP_RESPONSE_BODY_DRAINER_H_ |
6 #define NET_HTTP_HTTP_RESPONSE_BODY_DRAINER_H_ | 6 #define NET_HTTP_HTTP_RESPONSE_BODY_DRAINER_H_ |
7 | 7 |
8 #include "base/macros.h" | 8 #include "base/macros.h" |
9 #include "base/memory/ref_counted.h" | 9 #include "base/memory/ref_counted.h" |
10 #include "base/memory/scoped_ptr.h" | 10 #include "base/memory/scoped_ptr.h" |
11 #include "base/timer/timer.h" | 11 #include "base/timer/timer.h" |
12 #include "net/base/completion_callback.h" | 12 #include "net/base/completion_callback.h" |
13 #include "net/base/net_export.h" | 13 #include "net/base/net_export.h" |
14 #include "net/http/http_network_session.h" | 14 #include "net/http/http_network_session.h" |
15 | 15 |
16 namespace net { | 16 namespace net { |
17 | 17 |
| 18 class HttpResponseInfo; |
18 class HttpStream; | 19 class HttpStream; |
19 class IOBuffer; | 20 class IOBuffer; |
20 | 21 |
21 class NET_EXPORT_PRIVATE HttpResponseBodyDrainer { | 22 class NET_EXPORT_PRIVATE HttpResponseBodyDrainer { |
22 public: | 23 public: |
23 // The size in bytes of the buffer we use to drain the response body that | 24 // The size in bytes of the buffer we use to drain the response body that |
24 // we want to throw away. The response body is typically a small page just a | 25 // we want to throw away. The response body is typically a small page just a |
25 // few hundred bytes long. We set a limit to prevent it from taking too long, | 26 // few hundred bytes long. We set a limit to prevent it from taking too long, |
26 // since we may as well just create a new socket then. | 27 // since we may as well just create a new socket then. |
27 static const int kDrainBodyBufferSize = 16384; | 28 static const int kDrainBodyBufferSize = 16384; |
28 static const int kTimeoutInSeconds = 5; | 29 static const int kTimeoutInSeconds = 5; |
29 | 30 |
30 explicit HttpResponseBodyDrainer(HttpStream* stream); | 31 // |response_info| is the HttpResponseInfo passed to the |stream|, which may |
| 32 // still be holding on to pointers to it. |
| 33 HttpResponseBodyDrainer(HttpStream* stream, |
| 34 scoped_ptr<HttpResponseInfo> response_info); |
31 ~HttpResponseBodyDrainer(); | 35 ~HttpResponseBodyDrainer(); |
32 | 36 |
33 // Starts reading the body until completion, or we hit the buffer limit, or we | 37 // Starts reading the body until completion, or we hit the buffer limit, or we |
34 // timeout. After Start(), |this| will eventually delete itself. If it | 38 // timeout. After Start(), |this| will eventually delete itself. If it |
35 // doesn't complete immediately, it will add itself to |session|. | 39 // doesn't complete immediately, it will add itself to |session|. |
36 void Start(HttpNetworkSession* session); | 40 void Start(HttpNetworkSession* session); |
37 | 41 |
38 private: | 42 private: |
39 enum State { | 43 enum State { |
40 STATE_DRAIN_RESPONSE_BODY, | 44 STATE_DRAIN_RESPONSE_BODY, |
41 STATE_DRAIN_RESPONSE_BODY_COMPLETE, | 45 STATE_DRAIN_RESPONSE_BODY_COMPLETE, |
42 STATE_NONE, | 46 STATE_NONE, |
43 }; | 47 }; |
44 | 48 |
45 int DoLoop(int result); | 49 int DoLoop(int result); |
46 | 50 |
47 int DoDrainResponseBody(); | 51 int DoDrainResponseBody(); |
48 int DoDrainResponseBodyComplete(int result); | 52 int DoDrainResponseBodyComplete(int result); |
49 | 53 |
50 void OnIOComplete(int result); | 54 void OnIOComplete(int result); |
51 void OnTimerFired(); | 55 void OnTimerFired(); |
52 void Finish(int result); | 56 void Finish(int result); |
53 | 57 |
54 scoped_refptr<IOBuffer> read_buf_; | 58 scoped_refptr<IOBuffer> read_buf_; |
| 59 scoped_ptr<HttpResponseInfo> response_info_; |
55 const scoped_ptr<HttpStream> stream_; | 60 const scoped_ptr<HttpStream> stream_; |
56 State next_state_; | 61 State next_state_; |
57 int total_read_; | 62 int total_read_; |
58 CompletionCallback user_callback_; | 63 CompletionCallback user_callback_; |
59 base::OneShotTimer timer_; | 64 base::OneShotTimer timer_; |
60 HttpNetworkSession* session_; | 65 HttpNetworkSession* session_; |
61 | 66 |
62 DISALLOW_COPY_AND_ASSIGN(HttpResponseBodyDrainer); | 67 DISALLOW_COPY_AND_ASSIGN(HttpResponseBodyDrainer); |
63 }; | 68 }; |
64 | 69 |
65 } // namespace net | 70 } // namespace net |
66 | 71 |
67 #endif // NET_HTTP_HTTP_RESPONSE_BODY_DRAINER_H_ | 72 #endif // NET_HTTP_HTTP_RESPONSE_BODY_DRAINER_H_ |
OLD | NEW |