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/basictypes.h" | 8 #include "base/basictypes.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 HttpStreamBase; | 18 class HttpStream; |
19 class IOBuffer; | 19 class IOBuffer; |
20 | 20 |
21 class NET_EXPORT_PRIVATE HttpResponseBodyDrainer { | 21 class NET_EXPORT_PRIVATE HttpResponseBodyDrainer { |
22 public: | 22 public: |
23 // The size in bytes of the buffer we use to drain the response body that | 23 // 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 | 24 // 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, | 25 // 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. | 26 // since we may as well just create a new socket then. |
27 static const int kDrainBodyBufferSize = 16384; | 27 static const int kDrainBodyBufferSize = 16384; |
28 static const int kTimeoutInSeconds = 5; | 28 static const int kTimeoutInSeconds = 5; |
29 | 29 |
30 explicit HttpResponseBodyDrainer(HttpStreamBase* stream); | 30 explicit HttpResponseBodyDrainer(HttpStream* stream); |
31 ~HttpResponseBodyDrainer(); | 31 ~HttpResponseBodyDrainer(); |
32 | 32 |
33 // Starts reading the body until completion, or we hit the buffer limit, or we | 33 // 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 | 34 // timeout. After Start(), |this| will eventually delete itself. If it |
35 // doesn't complete immediately, it will add itself to |session|. | 35 // doesn't complete immediately, it will add itself to |session|. |
36 void Start(HttpNetworkSession* session); | 36 void Start(HttpNetworkSession* session); |
37 | 37 |
38 private: | 38 private: |
39 enum State { | 39 enum State { |
40 STATE_DRAIN_RESPONSE_BODY, | 40 STATE_DRAIN_RESPONSE_BODY, |
41 STATE_DRAIN_RESPONSE_BODY_COMPLETE, | 41 STATE_DRAIN_RESPONSE_BODY_COMPLETE, |
42 STATE_NONE, | 42 STATE_NONE, |
43 }; | 43 }; |
44 | 44 |
45 int DoLoop(int result); | 45 int DoLoop(int result); |
46 | 46 |
47 int DoDrainResponseBody(); | 47 int DoDrainResponseBody(); |
48 int DoDrainResponseBodyComplete(int result); | 48 int DoDrainResponseBodyComplete(int result); |
49 | 49 |
50 void OnIOComplete(int result); | 50 void OnIOComplete(int result); |
51 void OnTimerFired(); | 51 void OnTimerFired(); |
52 void Finish(int result); | 52 void Finish(int result); |
53 | 53 |
54 scoped_refptr<IOBuffer> read_buf_; | 54 scoped_refptr<IOBuffer> read_buf_; |
55 const scoped_ptr<HttpStreamBase> stream_; | 55 const scoped_ptr<HttpStream> stream_; |
56 State next_state_; | 56 State next_state_; |
57 int total_read_; | 57 int total_read_; |
58 CompletionCallback user_callback_; | 58 CompletionCallback user_callback_; |
59 base::OneShotTimer<HttpResponseBodyDrainer> timer_; | 59 base::OneShotTimer<HttpResponseBodyDrainer> timer_; |
60 HttpNetworkSession* session_; | 60 HttpNetworkSession* session_; |
61 | 61 |
62 DISALLOW_COPY_AND_ASSIGN(HttpResponseBodyDrainer); | 62 DISALLOW_COPY_AND_ASSIGN(HttpResponseBodyDrainer); |
63 }; | 63 }; |
64 | 64 |
65 } // namespace net | 65 } // namespace net |
66 | 66 |
67 #endif // NET_HTTP_HTTP_RESPONSE_BODY_DRAINER_H_ | 67 #endif // NET_HTTP_HTTP_RESPONSE_BODY_DRAINER_H_ |
OLD | NEW |