| 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 #include "net/url_request/url_request_job_manager.h" | 5 #include "net/url_request/url_request_job_manager.h" |
| 6 | 6 |
| 7 #include <algorithm> | 7 #include <algorithm> |
| 8 | 8 |
| 9 #include "build/build_config.h" | 9 #include "build/build_config.h" |
| 10 #include "base/string_util.h" | 10 #include "base/string_util.h" |
| 11 #include "net/base/load_flags.h" | 11 #include "net/base/load_flags.h" |
| 12 #include "net/base/net_errors.h" | 12 #include "net/base/net_errors.h" |
| 13 #include "net/url_request/url_request_about_job.h" | 13 #include "net/url_request/url_request_about_job.h" |
| 14 #include "net/url_request/url_request_data_job.h" | 14 #include "net/url_request/url_request_data_job.h" |
| 15 #include "net/url_request/url_request_error_job.h" | 15 #include "net/url_request/url_request_error_job.h" |
| 16 #include "net/url_request/url_request_file_job.h" | 16 #include "net/url_request/url_request_file_job.h" |
| 17 #include "net/url_request/url_request_ftp_job.h" | 17 #include "net/url_request/url_request_ftp_job.h" |
| 18 #include "net/url_request/url_request_http_job.h" | 18 #include "net/url_request/url_request_http_job.h" |
| 19 | 19 |
| 20 // The built-in set of protocol factories | 20 // The built-in set of protocol factories |
| 21 namespace { | 21 namespace { |
| 22 | 22 |
| 23 struct SchemeToFactory { | 23 struct SchemeToFactory { |
| 24 const char* scheme; | 24 const char* scheme; |
| 25 URLRequest::ProtocolFactory* factory; | 25 net::URLRequest::ProtocolFactory* factory; |
| 26 }; | 26 }; |
| 27 | 27 |
| 28 } // namespace | 28 } // namespace |
| 29 | 29 |
| 30 static const SchemeToFactory kBuiltinFactories[] = { | 30 static const SchemeToFactory kBuiltinFactories[] = { |
| 31 { "http", URLRequestHttpJob::Factory }, | 31 { "http", URLRequestHttpJob::Factory }, |
| 32 { "https", URLRequestHttpJob::Factory }, | 32 { "https", URLRequestHttpJob::Factory }, |
| 33 { "file", URLRequestFileJob::Factory }, | 33 { "file", URLRequestFileJob::Factory }, |
| 34 { "ftp", URLRequestFtpJob::Factory }, | 34 { "ftp", URLRequestFtpJob::Factory }, |
| 35 { "about", URLRequestAboutJob::Factory }, | 35 { "about", URLRequestAboutJob::Factory }, |
| 36 { "data", URLRequestDataJob::Factory }, | 36 { "data", URLRequestDataJob::Factory }, |
| 37 }; | 37 }; |
| 38 | 38 |
| 39 URLRequestJobManager::URLRequestJobManager() : enable_file_access_(false) { | 39 URLRequestJobManager::URLRequestJobManager() : enable_file_access_(false) { |
| 40 #ifndef NDEBUG | 40 #ifndef NDEBUG |
| 41 allowed_thread_ = 0; | 41 allowed_thread_ = 0; |
| 42 allowed_thread_initialized_ = false; | 42 allowed_thread_initialized_ = false; |
| 43 #endif | 43 #endif |
| 44 } | 44 } |
| 45 | 45 |
| 46 URLRequestJobManager::~URLRequestJobManager() {} | 46 URLRequestJobManager::~URLRequestJobManager() {} |
| 47 | 47 |
| 48 URLRequestJob* URLRequestJobManager::CreateJob(URLRequest* request) const { | 48 URLRequestJob* URLRequestJobManager::CreateJob(net::URLRequest* request) const { |
| 49 #ifndef NDEBUG | 49 #ifndef NDEBUG |
| 50 DCHECK(IsAllowedThread()); | 50 DCHECK(IsAllowedThread()); |
| 51 #endif | 51 #endif |
| 52 | 52 |
| 53 // If we are given an invalid URL, then don't even try to inspect the scheme. | 53 // If we are given an invalid URL, then don't even try to inspect the scheme. |
| 54 if (!request->url().is_valid()) | 54 if (!request->url().is_valid()) |
| 55 return new URLRequestErrorJob(request, net::ERR_INVALID_URL); | 55 return new URLRequestErrorJob(request, net::ERR_INVALID_URL); |
| 56 | 56 |
| 57 // We do this here to avoid asking interceptors about unsupported schemes. | 57 // We do this here to avoid asking interceptors about unsupported schemes. |
| 58 const std::string& scheme = request->url().scheme(); // already lowercase | 58 const std::string& scheme = request->url().scheme(); // already lowercase |
| (...skipping 34 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 93 } | 93 } |
| 94 | 94 |
| 95 // If we reached here, then it means that a registered protocol factory | 95 // If we reached here, then it means that a registered protocol factory |
| 96 // wasn't interested in handling the URL. That is fairly unexpected, and we | 96 // wasn't interested in handling the URL. That is fairly unexpected, and we |
| 97 // don't know have a specific error to report here :-( | 97 // don't know have a specific error to report here :-( |
| 98 LOG(WARNING) << "Failed to map: " << request->url().spec(); | 98 LOG(WARNING) << "Failed to map: " << request->url().spec(); |
| 99 return new URLRequestErrorJob(request, net::ERR_FAILED); | 99 return new URLRequestErrorJob(request, net::ERR_FAILED); |
| 100 } | 100 } |
| 101 | 101 |
| 102 URLRequestJob* URLRequestJobManager::MaybeInterceptRedirect( | 102 URLRequestJob* URLRequestJobManager::MaybeInterceptRedirect( |
| 103 URLRequest* request, | 103 net::URLRequest* request, |
| 104 const GURL& location) const { | 104 const GURL& location) const { |
| 105 #ifndef NDEBUG | 105 #ifndef NDEBUG |
| 106 DCHECK(IsAllowedThread()); | 106 DCHECK(IsAllowedThread()); |
| 107 #endif | 107 #endif |
| 108 if ((request->load_flags() & net::LOAD_DISABLE_INTERCEPT) || | 108 if ((request->load_flags() & net::LOAD_DISABLE_INTERCEPT) || |
| 109 (request->status().status() == URLRequestStatus::CANCELED) || | 109 (request->status().status() == URLRequestStatus::CANCELED) || |
| 110 !request->url().is_valid() || | 110 !request->url().is_valid() || |
| 111 !SupportsScheme(request->url().scheme())) | 111 !SupportsScheme(request->url().scheme())) |
| 112 return NULL; | 112 return NULL; |
| 113 | 113 |
| 114 InterceptorList::const_iterator i; | 114 InterceptorList::const_iterator i; |
| 115 for (i = interceptors_.begin(); i != interceptors_.end(); ++i) { | 115 for (i = interceptors_.begin(); i != interceptors_.end(); ++i) { |
| 116 URLRequestJob* job = (*i)->MaybeInterceptRedirect(request, location); | 116 URLRequestJob* job = (*i)->MaybeInterceptRedirect(request, location); |
| 117 if (job) | 117 if (job) |
| 118 return job; | 118 return job; |
| 119 } | 119 } |
| 120 return NULL; | 120 return NULL; |
| 121 } | 121 } |
| 122 | 122 |
| 123 URLRequestJob* URLRequestJobManager::MaybeInterceptResponse( | 123 URLRequestJob* URLRequestJobManager::MaybeInterceptResponse( |
| 124 URLRequest* request) const { | 124 net::URLRequest* request) const { |
| 125 #ifndef NDEBUG | 125 #ifndef NDEBUG |
| 126 DCHECK(IsAllowedThread()); | 126 DCHECK(IsAllowedThread()); |
| 127 #endif | 127 #endif |
| 128 if ((request->load_flags() & net::LOAD_DISABLE_INTERCEPT) || | 128 if ((request->load_flags() & net::LOAD_DISABLE_INTERCEPT) || |
| 129 (request->status().status() == URLRequestStatus::CANCELED) || | 129 (request->status().status() == URLRequestStatus::CANCELED) || |
| 130 !request->url().is_valid() || | 130 !request->url().is_valid() || |
| 131 !SupportsScheme(request->url().scheme())) | 131 !SupportsScheme(request->url().scheme())) |
| 132 return NULL; | 132 return NULL; |
| 133 | 133 |
| 134 InterceptorList::const_iterator i; | 134 InterceptorList::const_iterator i; |
| (...skipping 13 matching lines...) Expand all Loading... |
| 148 return true; | 148 return true; |
| 149 } | 149 } |
| 150 | 150 |
| 151 for (size_t i = 0; i < arraysize(kBuiltinFactories); ++i) | 151 for (size_t i = 0; i < arraysize(kBuiltinFactories); ++i) |
| 152 if (LowerCaseEqualsASCII(scheme, kBuiltinFactories[i].scheme)) | 152 if (LowerCaseEqualsASCII(scheme, kBuiltinFactories[i].scheme)) |
| 153 return true; | 153 return true; |
| 154 | 154 |
| 155 return false; | 155 return false; |
| 156 } | 156 } |
| 157 | 157 |
| 158 URLRequest::ProtocolFactory* URLRequestJobManager::RegisterProtocolFactory( | 158 net::URLRequest::ProtocolFactory* URLRequestJobManager::RegisterProtocolFactory( |
| 159 const std::string& scheme, | 159 const std::string& scheme, |
| 160 URLRequest::ProtocolFactory* factory) { | 160 net::URLRequest::ProtocolFactory* factory) { |
| 161 #ifndef NDEBUG | 161 #ifndef NDEBUG |
| 162 DCHECK(IsAllowedThread()); | 162 DCHECK(IsAllowedThread()); |
| 163 #endif | 163 #endif |
| 164 | 164 |
| 165 AutoLock locked(lock_); | 165 AutoLock locked(lock_); |
| 166 | 166 |
| 167 URLRequest::ProtocolFactory* old_factory; | 167 net::URLRequest::ProtocolFactory* old_factory; |
| 168 FactoryMap::iterator i = factories_.find(scheme); | 168 FactoryMap::iterator i = factories_.find(scheme); |
| 169 if (i != factories_.end()) { | 169 if (i != factories_.end()) { |
| 170 old_factory = i->second; | 170 old_factory = i->second; |
| 171 } else { | 171 } else { |
| 172 old_factory = NULL; | 172 old_factory = NULL; |
| 173 } | 173 } |
| 174 if (factory) { | 174 if (factory) { |
| 175 factories_[scheme] = factory; | 175 factories_[scheme] = factory; |
| 176 } else if (i != factories_.end()) { // uninstall any old one | 176 } else if (i != factories_.end()) { // uninstall any old one |
| 177 factories_.erase(i); | 177 factories_.erase(i); |
| 178 } | 178 } |
| 179 return old_factory; | 179 return old_factory; |
| 180 } | 180 } |
| 181 | 181 |
| 182 void URLRequestJobManager::RegisterRequestInterceptor( | 182 void URLRequestJobManager::RegisterRequestInterceptor( |
| 183 URLRequest::Interceptor* interceptor) { | 183 net::URLRequest::Interceptor* interceptor) { |
| 184 #ifndef NDEBUG | 184 #ifndef NDEBUG |
| 185 DCHECK(IsAllowedThread()); | 185 DCHECK(IsAllowedThread()); |
| 186 #endif | 186 #endif |
| 187 | 187 |
| 188 AutoLock locked(lock_); | 188 AutoLock locked(lock_); |
| 189 | 189 |
| 190 DCHECK(std::find(interceptors_.begin(), interceptors_.end(), interceptor) == | 190 DCHECK(std::find(interceptors_.begin(), interceptors_.end(), interceptor) == |
| 191 interceptors_.end()); | 191 interceptors_.end()); |
| 192 interceptors_.push_back(interceptor); | 192 interceptors_.push_back(interceptor); |
| 193 } | 193 } |
| 194 | 194 |
| 195 void URLRequestJobManager::UnregisterRequestInterceptor( | 195 void URLRequestJobManager::UnregisterRequestInterceptor( |
| 196 URLRequest::Interceptor* interceptor) { | 196 net::URLRequest::Interceptor* interceptor) { |
| 197 #ifndef NDEBUG | 197 #ifndef NDEBUG |
| 198 DCHECK(IsAllowedThread()); | 198 DCHECK(IsAllowedThread()); |
| 199 #endif | 199 #endif |
| 200 | 200 |
| 201 AutoLock locked(lock_); | 201 AutoLock locked(lock_); |
| 202 | 202 |
| 203 InterceptorList::iterator i = | 203 InterceptorList::iterator i = |
| 204 std::find(interceptors_.begin(), interceptors_.end(), interceptor); | 204 std::find(interceptors_.begin(), interceptors_.end(), interceptor); |
| 205 DCHECK(i != interceptors_.end()); | 205 DCHECK(i != interceptors_.end()); |
| 206 interceptors_.erase(i); | 206 interceptors_.erase(i); |
| 207 } | 207 } |
| OLD | NEW |