Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 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_ftp_job.h" | 5 #include "net/url_request/url_request_ftp_job.h" |
| 6 | 6 |
| 7 #include "base/compiler_specific.h" | 7 #include "base/compiler_specific.h" |
| 8 #include "base/message_loop.h" | 8 #include "base/message_loop.h" |
| 9 #include "base/utf_string_conversions.h" | 9 #include "base/utf_string_conversions.h" |
| 10 #include "net/base/auth.h" | 10 #include "net/base/auth.h" |
| 11 #include "net/base/host_port_pair.h" | 11 #include "net/base/host_port_pair.h" |
| 12 #include "net/base/net_errors.h" | 12 #include "net/base/net_errors.h" |
| 13 #include "net/base/net_util.h" | 13 #include "net/base/net_util.h" |
| 14 #include "net/ftp/ftp_response_info.h" | 14 #include "net/ftp/ftp_response_info.h" |
| 15 #include "net/ftp/ftp_transaction_factory.h" | 15 #include "net/ftp/ftp_transaction_factory.h" |
| 16 #include "net/url_request/url_request.h" | 16 #include "net/url_request/url_request.h" |
| 17 #include "net/url_request/url_request_context.h" | 17 #include "net/url_request/url_request_context.h" |
| 18 #include "net/url_request/url_request_error_job.h" | 18 #include "net/url_request/url_request_error_job.h" |
| 19 | 19 |
| 20 namespace net { | 20 namespace net { |
| 21 | 21 |
| 22 URLRequestFtpJob::URLRequestFtpJob(URLRequest* request) | 22 URLRequestFtpJob::URLRequestFtpJob( |
| 23 URLRequest* request, | |
| 24 NetworkDelegate* network_delegate, | |
| 25 FtpTransactionFactory* ftp_transaction_factory, | |
| 26 FtpAuthCache* ftp_auth_cache) | |
| 23 : URLRequestJob(request), | 27 : URLRequestJob(request), |
| 24 read_in_progress_(false), | 28 read_in_progress_(false), |
| 25 ALLOW_THIS_IN_INITIALIZER_LIST(weak_factory_(this)) { | 29 ALLOW_THIS_IN_INITIALIZER_LIST(weak_factory_(this)), |
| 30 network_delegate_(network_delegate), | |
| 31 ftp_transaction_factory_(ftp_transaction_factory), | |
| 32 ftp_auth_cache_(ftp_auth_cache) { | |
| 33 DCHECK(ftp_transaction_factory); | |
| 34 DCHECK(ftp_auth_cache); | |
| 35 DCHECK(network_delegate); | |
| 26 } | 36 } |
| 27 | 37 |
| 28 // static | 38 // static |
| 29 URLRequestJob* URLRequestFtpJob::Factory(URLRequest* request, | 39 URLRequestJob* URLRequestFtpJob::Factory(URLRequest* request, |
| 30 const std::string& scheme) { | 40 const std::string& scheme) { |
| 31 DCHECK_EQ(scheme, "ftp"); | 41 DCHECK_EQ(scheme, "ftp"); |
| 32 | 42 |
| 33 int port = request->url().IntPort(); | 43 int port = request->url().IntPort(); |
| 34 if (request->url().has_port() && | 44 if (request->url().has_port() && |
| 35 !IsPortAllowedByFtp(port) && !IsPortAllowedByOverride(port)) | 45 !IsPortAllowedByFtp(port) && !IsPortAllowedByOverride(port)) |
| 36 return new URLRequestErrorJob(request, ERR_UNSAFE_PORT); | 46 return new URLRequestErrorJob(request, ERR_UNSAFE_PORT); |
| 37 | 47 |
| 38 DCHECK(request->context()); | 48 return new URLRequestFtpJob(request, |
|
Paweł Hajdan Jr.
2012/06/07 19:30:21
Please keep the DCHECKs. In fact, add even more: f
erikwright (departed)
2012/06/07 19:40:40
I'm indifferent as to DCHECKing network_delegate e
shalev
2012/06/21 20:04:55
Done.
| |
| 39 DCHECK(request->context()->ftp_transaction_factory()); | 49 request->context()->network_delegate(), |
| 40 return new URLRequestFtpJob(request); | 50 request->context()->ftp_transaction_factory(), |
| 51 request->context()->ftp_auth_cache()); | |
| 41 } | 52 } |
| 42 | 53 |
| 43 bool URLRequestFtpJob::GetMimeType(std::string* mime_type) const { | 54 bool URLRequestFtpJob::GetMimeType(std::string* mime_type) const { |
| 44 if (transaction_->GetResponseInfo()->is_directory_listing) { | 55 if (transaction_->GetResponseInfo()->is_directory_listing) { |
| 45 *mime_type = "text/vnd.chromium.ftp-dir"; | 56 *mime_type = "text/vnd.chromium.ftp-dir"; |
| 46 return true; | 57 return true; |
| 47 } | 58 } |
| 48 return false; | 59 return false; |
| 49 } | 60 } |
| 50 | 61 |
| 51 HostPortPair URLRequestFtpJob::GetSocketAddress() const { | 62 HostPortPair URLRequestFtpJob::GetSocketAddress() const { |
| 52 if (!transaction_.get()) { | 63 if (!transaction_.get()) { |
| 53 return HostPortPair(); | 64 return HostPortPair(); |
| 54 } | 65 } |
| 55 return transaction_->GetResponseInfo()->socket_address; | 66 return transaction_->GetResponseInfo()->socket_address; |
| 56 } | 67 } |
| 57 | 68 |
| 58 URLRequestFtpJob::~URLRequestFtpJob() { | 69 URLRequestFtpJob::~URLRequestFtpJob() { |
| 59 } | 70 } |
| 60 | 71 |
| 61 void URLRequestFtpJob::StartTransaction() { | 72 void URLRequestFtpJob::StartTransaction() { |
| 62 // Create a transaction. | 73 // Create a transaction. |
| 63 DCHECK(!transaction_.get()); | 74 DCHECK(!transaction_.get()); |
| 64 DCHECK(request_->context()); | 75 DCHECK(ftp_transaction_factory_); |
|
erikwright (departed)
2012/06/07 19:14:51
Remove.
shalev
2012/06/21 20:04:55
Done.
shalev
2012/06/21 20:04:55
Done.
| |
| 65 DCHECK(request_->context()->ftp_transaction_factory()); | |
| 66 | 76 |
| 67 transaction_.reset( | 77 transaction_.reset(ftp_transaction_factory_->CreateTransaction()); |
| 68 request_->context()->ftp_transaction_factory()->CreateTransaction()); | |
| 69 | 78 |
| 70 // No matter what, we want to report our status as IO pending since we will | 79 // No matter what, we want to report our status as IO pending since we will |
| 71 // be notifying our consumer asynchronously via OnStartCompleted. | 80 // be notifying our consumer asynchronously via OnStartCompleted. |
| 72 SetStatus(URLRequestStatus(URLRequestStatus::IO_PENDING, 0)); | 81 SetStatus(URLRequestStatus(URLRequestStatus::IO_PENDING, 0)); |
| 73 int rv; | 82 int rv; |
| 74 if (transaction_.get()) { | 83 if (transaction_.get()) { |
| 75 rv = transaction_->Start( | 84 rv = transaction_->Start( |
| 76 &request_info_, | 85 &request_info_, |
| 77 base::Bind(&URLRequestFtpJob::OnStartCompleted, | 86 base::Bind(&URLRequestFtpJob::OnStartCompleted, |
| 78 base::Unretained(this)), | 87 base::Unretained(this)), |
| (...skipping 18 matching lines...) Expand all Loading... | |
| 97 // FTP obviously doesn't have HTTP Content-Length header. We have to pass | 106 // FTP obviously doesn't have HTTP Content-Length header. We have to pass |
| 98 // the content size information manually. | 107 // the content size information manually. |
| 99 set_expected_content_size( | 108 set_expected_content_size( |
| 100 transaction_->GetResponseInfo()->expected_content_size); | 109 transaction_->GetResponseInfo()->expected_content_size); |
| 101 | 110 |
| 102 if (result == OK) { | 111 if (result == OK) { |
| 103 NotifyHeadersComplete(); | 112 NotifyHeadersComplete(); |
| 104 } else if (transaction_->GetResponseInfo()->needs_auth) { | 113 } else if (transaction_->GetResponseInfo()->needs_auth) { |
| 105 GURL origin = request_->url().GetOrigin(); | 114 GURL origin = request_->url().GetOrigin(); |
| 106 if (server_auth_ && server_auth_->state == AUTH_STATE_HAVE_AUTH) { | 115 if (server_auth_ && server_auth_->state == AUTH_STATE_HAVE_AUTH) { |
| 107 request_->context()->ftp_auth_cache()->Remove( | 116 ftp_auth_cache_->Remove(origin, server_auth_->credentials); |
| 108 origin, server_auth_->credentials); | |
| 109 } else if (!server_auth_) { | 117 } else if (!server_auth_) { |
| 110 server_auth_ = new AuthData(); | 118 server_auth_ = new AuthData(); |
| 111 } | 119 } |
| 112 server_auth_->state = AUTH_STATE_NEED_AUTH; | 120 server_auth_->state = AUTH_STATE_NEED_AUTH; |
| 113 | 121 |
| 114 FtpAuthCache::Entry* cached_auth = | 122 FtpAuthCache::Entry* cached_auth = |
| 115 request_->context()->ftp_auth_cache()->Lookup(origin); | 123 ftp_auth_cache_->Lookup(origin); |
| 116 | |
| 117 if (cached_auth) { | 124 if (cached_auth) { |
| 118 // Retry using cached auth data. | 125 // Retry using cached auth data. |
| 119 SetAuth(cached_auth->credentials); | 126 SetAuth(cached_auth->credentials); |
| 120 } else { | 127 } else { |
| 121 // Prompt for a username/password. | 128 // Prompt for a username/password. |
| 122 NotifyHeadersComplete(); | 129 NotifyHeadersComplete(); |
| 123 } | 130 } |
| 124 } else { | 131 } else { |
| 125 NotifyDone(URLRequestStatus(URLRequestStatus::FAILED, result)); | 132 NotifyDone(URLRequestStatus(URLRequestStatus::FAILED, result)); |
| 126 } | 133 } |
| (...skipping 70 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 197 DCHECK(auth_info->scheme.empty()); | 204 DCHECK(auth_info->scheme.empty()); |
| 198 DCHECK(auth_info->realm.empty()); | 205 DCHECK(auth_info->realm.empty()); |
| 199 result->swap(auth_info); | 206 result->swap(auth_info); |
| 200 } | 207 } |
| 201 | 208 |
| 202 void URLRequestFtpJob::SetAuth(const AuthCredentials& credentials) { | 209 void URLRequestFtpJob::SetAuth(const AuthCredentials& credentials) { |
| 203 DCHECK(NeedsAuth()); | 210 DCHECK(NeedsAuth()); |
| 204 server_auth_->state = AUTH_STATE_HAVE_AUTH; | 211 server_auth_->state = AUTH_STATE_HAVE_AUTH; |
| 205 server_auth_->credentials = credentials; | 212 server_auth_->credentials = credentials; |
| 206 | 213 |
| 207 request_->context()->ftp_auth_cache()->Add(request_->url().GetOrigin(), | 214 ftp_auth_cache_->Add(request_->url().GetOrigin(), |
| 208 server_auth_->credentials); | 215 server_auth_->credentials); |
| 209 | 216 |
| 210 RestartTransactionWithAuth(); | 217 RestartTransactionWithAuth(); |
| 211 } | 218 } |
| 212 | 219 |
| 213 void URLRequestFtpJob::CancelAuth() { | 220 void URLRequestFtpJob::CancelAuth() { |
| 214 DCHECK(NeedsAuth()); | 221 DCHECK(NeedsAuth()); |
| 215 server_auth_->state = AUTH_STATE_CANCELED; | 222 server_auth_->state = AUTH_STATE_CANCELED; |
| 216 | 223 |
| 217 // Once the auth is cancelled, we proceed with the request as though | 224 // Once the auth is cancelled, we proceed with the request as though |
| 218 // there were no auth. Schedule this for later so that we don't cause | 225 // there were no auth. Schedule this for later so that we don't cause |
| (...skipping 26 matching lines...) Expand all Loading... | |
| 245 if (rv == ERR_IO_PENDING) { | 252 if (rv == ERR_IO_PENDING) { |
| 246 read_in_progress_ = true; | 253 read_in_progress_ = true; |
| 247 SetStatus(URLRequestStatus(URLRequestStatus::IO_PENDING, 0)); | 254 SetStatus(URLRequestStatus(URLRequestStatus::IO_PENDING, 0)); |
| 248 } else { | 255 } else { |
| 249 NotifyDone(URLRequestStatus(URLRequestStatus::FAILED, rv)); | 256 NotifyDone(URLRequestStatus(URLRequestStatus::FAILED, rv)); |
| 250 } | 257 } |
| 251 return false; | 258 return false; |
| 252 } | 259 } |
| 253 | 260 |
| 254 } // namespace net | 261 } // namespace net |
| OLD | NEW |