OLD | NEW |
| (Empty) |
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 NET_PROXY_PROXY_SCRIPT_FETCHER_IMPL_H_ | |
6 #define NET_PROXY_PROXY_SCRIPT_FETCHER_IMPL_H_ | |
7 | |
8 #include <string> | |
9 | |
10 #include "base/basictypes.h" | |
11 #include "base/compiler_specific.h" | |
12 #include "base/memory/ref_counted.h" | |
13 #include "base/memory/scoped_ptr.h" | |
14 #include "base/memory/weak_ptr.h" | |
15 #include "base/strings/string16.h" | |
16 #include "base/time/time.h" | |
17 #include "net/proxy/proxy_script_fetcher.h" | |
18 #include "net/url_request/url_request.h" | |
19 | |
20 class GURL; | |
21 | |
22 namespace net { | |
23 | |
24 class URLRequestContext; | |
25 | |
26 // Implementation of ProxyScriptFetcher that downloads scripts using the | |
27 // specified request context. | |
28 class NET_EXPORT ProxyScriptFetcherImpl : public ProxyScriptFetcher, | |
29 public URLRequest::Delegate { | |
30 public: | |
31 // Creates a ProxyScriptFetcher that issues requests through | |
32 // |url_request_context|. |url_request_context| must remain valid for the | |
33 // lifetime of ProxyScriptFetcherImpl. | |
34 // Note that while a request is in progress, we will be holding a reference | |
35 // to |url_request_context|. Be careful not to create cycles between the | |
36 // fetcher and the context; you can break such cycles by calling Cancel(). | |
37 explicit ProxyScriptFetcherImpl(URLRequestContext* url_request_context); | |
38 | |
39 ~ProxyScriptFetcherImpl() override; | |
40 | |
41 // Used by unit-tests to modify the default limits. | |
42 base::TimeDelta SetTimeoutConstraint(base::TimeDelta timeout); | |
43 size_t SetSizeConstraint(size_t size_bytes); | |
44 | |
45 void OnResponseCompleted(URLRequest* request); | |
46 | |
47 // ProxyScriptFetcher methods: | |
48 int Fetch(const GURL& url, | |
49 base::string16* text, | |
50 const net::CompletionCallback& callback) override; | |
51 void Cancel() override; | |
52 URLRequestContext* GetRequestContext() const override; | |
53 | |
54 // URLRequest::Delegate methods: | |
55 void OnAuthRequired(URLRequest* request, | |
56 AuthChallengeInfo* auth_info) override; | |
57 void OnSSLCertificateError(URLRequest* request, | |
58 const SSLInfo& ssl_info, | |
59 bool is_hsts_ok) override; | |
60 void OnResponseStarted(URLRequest* request) override; | |
61 void OnReadCompleted(URLRequest* request, int num_bytes) override; | |
62 | |
63 private: | |
64 enum { kBufSize = 4096 }; | |
65 | |
66 // Read more bytes from the response. | |
67 void ReadBody(URLRequest* request); | |
68 | |
69 // Handles a response from Read(). Returns true if we should continue trying | |
70 // to read. |num_bytes| is 0 for EOF, and < 0 on errors. | |
71 bool ConsumeBytesRead(URLRequest* request, int num_bytes); | |
72 | |
73 // Called once the request has completed to notify the caller of | |
74 // |response_code_| and |response_text_|. | |
75 void FetchCompleted(); | |
76 | |
77 // Clear out the state for the current request. | |
78 void ResetCurRequestState(); | |
79 | |
80 // Callback for time-out task of request with id |id|. | |
81 void OnTimeout(int id); | |
82 | |
83 // The context used for making network requests. | |
84 URLRequestContext* const url_request_context_; | |
85 | |
86 // Buffer that URLRequest writes into. | |
87 scoped_refptr<IOBuffer> buf_; | |
88 | |
89 // The next ID to use for |cur_request_| (monotonically increasing). | |
90 int next_id_; | |
91 | |
92 // The current (in progress) request, or NULL. | |
93 scoped_ptr<URLRequest> cur_request_; | |
94 | |
95 // State for current request (only valid when |cur_request_| is not NULL): | |
96 | |
97 // Unique ID for the current request. | |
98 int cur_request_id_; | |
99 | |
100 // Callback to invoke on completion of the fetch. | |
101 net::CompletionCallback callback_; | |
102 | |
103 // Holds the error condition that was hit on the current request, or OK. | |
104 int result_code_; | |
105 | |
106 // Holds the bytes read so far. Will not exceed |max_response_bytes|. | |
107 std::string bytes_read_so_far_; | |
108 | |
109 // This buffer is owned by the owner of |callback|, and will be filled with | |
110 // UTF16 response on completion. | |
111 base::string16* result_text_; | |
112 | |
113 // The maximum number of bytes to allow in responses. | |
114 size_t max_response_bytes_; | |
115 | |
116 // The maximum amount of time to wait for download to complete. | |
117 base::TimeDelta max_duration_; | |
118 | |
119 // Factory for creating the time-out task. This takes care of revoking | |
120 // outstanding tasks when |this| is deleted. | |
121 base::WeakPtrFactory<ProxyScriptFetcherImpl> weak_factory_; | |
122 | |
123 DISALLOW_COPY_AND_ASSIGN(ProxyScriptFetcherImpl); | |
124 }; | |
125 | |
126 } // namespace net | |
127 | |
128 #endif // NET_PROXY_PROXY_SCRIPT_FETCHER_IMPL_H_ | |
OLD | NEW |