OLD | NEW |
1 // Copyright (c) 2009 The Chromium OS Authors. All rights reserved. | 1 // Copyright (c) 2009 The Chromium OS 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 "update_engine/libcurl_http_fetcher.h" | 5 #include "update_engine/libcurl_http_fetcher.h" |
6 #include <algorithm> | 6 #include <algorithm> |
7 #include "base/logging.h" | 7 #include "base/logging.h" |
8 | 8 |
9 using std::max; | 9 using std::max; |
10 using std::make_pair; | 10 using std::make_pair; |
(...skipping 94 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
105 if ((transfer_size_ >= 0) && (bytes_downloaded_ < transfer_size_)) { | 105 if ((transfer_size_ >= 0) && (bytes_downloaded_ < transfer_size_)) { |
106 // Need to restart transfer | 106 // Need to restart transfer |
107 retry_count_++; | 107 retry_count_++; |
108 LOG(INFO) << "Restarting transfer b/c we finished, had downloaded " | 108 LOG(INFO) << "Restarting transfer b/c we finished, had downloaded " |
109 << bytes_downloaded_ << " bytes, but transfer_size_ is " | 109 << bytes_downloaded_ << " bytes, but transfer_size_ is " |
110 << transfer_size_ << ". retry_count: " << retry_count_; | 110 << transfer_size_ << ". retry_count: " << retry_count_; |
111 if (retry_count_ > kMaxRetriesCount) { | 111 if (retry_count_ > kMaxRetriesCount) { |
112 if (delegate_) | 112 if (delegate_) |
113 delegate_->TransferComplete(this, false); // success | 113 delegate_->TransferComplete(this, false); // success |
114 } else { | 114 } else { |
115 g_timeout_add_seconds(5, | 115 g_timeout_add_seconds(retry_seconds_, |
116 &LibcurlHttpFetcher::StaticRetryTimeoutCallback, | 116 &LibcurlHttpFetcher::StaticRetryTimeoutCallback, |
117 this); | 117 this); |
118 } | 118 } |
119 return; | 119 return; |
120 } else { | 120 } else { |
121 if (delegate_) { | 121 if (delegate_) { |
122 // success is when http_response_code is 2xx | 122 // success is when http_response_code is 2xx |
123 bool success = (http_response_code >= 200) && | 123 bool success = (http_response_code >= 200) && |
124 (http_response_code < 300); | 124 (http_response_code < 300); |
125 delegate_->TransferComplete(this, success); | 125 delegate_->TransferComplete(this, success); |
(...skipping 83 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
209 &StaticFDCallback, | 209 &StaticFDCallback, |
210 this); | 210 this); |
211 io_channels_[i] = make_pair(io_channel, tag); | 211 io_channels_[i] = make_pair(io_channel, tag); |
212 static int io_counter = 0; | 212 static int io_counter = 0; |
213 io_counter++; | 213 io_counter++; |
214 if (io_counter % 50 == 0) { | 214 if (io_counter % 50 == 0) { |
215 LOG(INFO) << "io_counter = " << io_counter; | 215 LOG(INFO) << "io_counter = " << io_counter; |
216 } | 216 } |
217 } | 217 } |
218 | 218 |
219 // Wet up a timeout callback for libcurl | 219 // Set up a timeout callback for libcurl. |
220 long ms = 0; | |
221 CHECK_EQ(curl_multi_timeout(curl_multi_handle_, &ms), CURLM_OK); | |
222 if (ms < 0) { | |
223 // From http://curl.haxx.se/libcurl/c/curl_multi_timeout.html: | |
224 // if libcurl returns a -1 timeout here, it just means that libcurl | |
225 // currently has no stored timeout value. You must not wait too long | |
226 // (more than a few seconds perhaps) before you call | |
227 // curl_multi_perform() again. | |
228 ms = idle_ms_; | |
229 } | |
230 if (!timeout_source_) { | 220 if (!timeout_source_) { |
231 LOG(INFO) << "setting up timeout source:" << ms; | 221 LOG(INFO) << "Setting up timeout source: " << idle_seconds_ << " seconds."; |
232 timeout_source_ = g_timeout_source_new_seconds(1); | 222 timeout_source_ = g_timeout_source_new_seconds(idle_seconds_); |
233 CHECK(timeout_source_); | 223 g_source_set_callback(timeout_source_, StaticTimeoutCallback, this, NULL); |
234 g_source_set_callback(timeout_source_, StaticTimeoutCallback, this, | |
235 NULL); | |
236 g_source_attach(timeout_source_, NULL); | 224 g_source_attach(timeout_source_, NULL); |
237 static int counter = 0; | |
238 counter++; | |
239 if (counter % 50 == 0) { | |
240 LOG(INFO) << "counter = " << counter; | |
241 } | |
242 } | 225 } |
243 } | 226 } |
244 | 227 |
245 bool LibcurlHttpFetcher::FDCallback(GIOChannel *source, | 228 bool LibcurlHttpFetcher::FDCallback(GIOChannel *source, |
246 GIOCondition condition) { | 229 GIOCondition condition) { |
247 CurlPerformOnce(); | 230 CurlPerformOnce(); |
248 // We handle removing of this source elsewhere, so we always return true. | 231 // We handle removing of this source elsewhere, so we always return true. |
249 // The docs say, "the function should return FALSE if the event source | 232 // The docs say, "the function should return FALSE if the event source |
250 // should be removed." | 233 // should be removed." |
251 // http://www.gtk.org/api/2.6/glib/glib-IO-Channels.html#GIOFunc | 234 // http://www.gtk.org/api/2.6/glib/glib-IO-Channels.html#GIOFunc |
(...skipping 38 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
290 curl_handle_ = NULL; | 273 curl_handle_ = NULL; |
291 } | 274 } |
292 if (curl_multi_handle_) { | 275 if (curl_multi_handle_) { |
293 CHECK_EQ(curl_multi_cleanup(curl_multi_handle_), CURLM_OK); | 276 CHECK_EQ(curl_multi_cleanup(curl_multi_handle_), CURLM_OK); |
294 curl_multi_handle_ = NULL; | 277 curl_multi_handle_ = NULL; |
295 } | 278 } |
296 transfer_in_progress_ = false; | 279 transfer_in_progress_ = false; |
297 } | 280 } |
298 | 281 |
299 } // namespace chromeos_update_engine | 282 } // namespace chromeos_update_engine |
OLD | NEW |