OLD | NEW |
| (Empty) |
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 | |
3 // found in the LICENSE file. | |
4 | |
5 #ifndef NET_HTTP_HTTP_RESPONSE_BODY_DRAINER_H_ | |
6 #define NET_HTTP_HTTP_RESPONSE_BODY_DRAINER_H_ | |
7 | |
8 #include "base/basictypes.h" | |
9 #include "base/memory/ref_counted.h" | |
10 #include "base/memory/scoped_ptr.h" | |
11 #include "base/timer/timer.h" | |
12 #include "net/base/completion_callback.h" | |
13 #include "net/base/net_export.h" | |
14 #include "net/http/http_network_session.h" | |
15 | |
16 namespace net { | |
17 | |
18 class HttpStream; | |
19 class IOBuffer; | |
20 | |
21 class NET_EXPORT_PRIVATE HttpResponseBodyDrainer { | |
22 public: | |
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 | |
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. | |
27 static const int kDrainBodyBufferSize = 16384; | |
28 static const int kTimeoutInSeconds = 5; | |
29 | |
30 explicit HttpResponseBodyDrainer(HttpStream* stream); | |
31 ~HttpResponseBodyDrainer(); | |
32 | |
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 | |
35 // doesn't complete immediately, it will add itself to |session|. | |
36 void Start(HttpNetworkSession* session); | |
37 | |
38 private: | |
39 enum State { | |
40 STATE_DRAIN_RESPONSE_BODY, | |
41 STATE_DRAIN_RESPONSE_BODY_COMPLETE, | |
42 STATE_NONE, | |
43 }; | |
44 | |
45 int DoLoop(int result); | |
46 | |
47 int DoDrainResponseBody(); | |
48 int DoDrainResponseBodyComplete(int result); | |
49 | |
50 void OnIOComplete(int result); | |
51 void OnTimerFired(); | |
52 void Finish(int result); | |
53 | |
54 scoped_refptr<IOBuffer> read_buf_; | |
55 const scoped_ptr<HttpStream> stream_; | |
56 State next_state_; | |
57 int total_read_; | |
58 CompletionCallback user_callback_; | |
59 base::OneShotTimer<HttpResponseBodyDrainer> timer_; | |
60 HttpNetworkSession* session_; | |
61 | |
62 DISALLOW_COPY_AND_ASSIGN(HttpResponseBodyDrainer); | |
63 }; | |
64 | |
65 } // namespace net | |
66 | |
67 #endif // NET_HTTP_HTTP_RESPONSE_BODY_DRAINER_H_ | |
OLD | NEW |