Index: chrome/browser/component_updater/component_updater_service.cc |
diff --git a/chrome/browser/component_updater/component_updater_service.cc b/chrome/browser/component_updater/component_updater_service.cc |
index 472525b1cb84b143a07b61374814beaabd13dbac..1f4112c82cff27bf7b13a06219635e7d726c9e05 100644 |
--- a/chrome/browser/component_updater/component_updater_service.cc |
+++ b/chrome/browser/component_updater/component_updater_service.cc |
@@ -1,1166 +1,1168 @@ |
-// Copyright (c) 2012 The Chromium Authors. All rights reserved. |
-// Use of this source code is governed by a BSD-style license that can be |
-// found in the LICENSE file. |
- |
-#include "chrome/browser/component_updater/component_updater_service.h" |
- |
-#include <algorithm> |
-#include <set> |
-#include <vector> |
- |
-#include "base/at_exit.h" |
-#include "base/bind.h" |
-#include "base/compiler_specific.h" |
-#include "base/file_util.h" |
-#include "base/files/file_path.h" |
-#include "base/logging.h" |
-#include "base/memory/scoped_ptr.h" |
-#include "base/memory/weak_ptr.h" |
-#include "base/stl_util.h" |
-#include "base/strings/string_number_conversions.h" |
-#include "base/strings/string_piece.h" |
-#include "base/strings/string_util.h" |
-#include "base/strings/stringprintf.h" |
-#include "base/threading/sequenced_worker_pool.h" |
-#include "base/timer/timer.h" |
-#include "chrome/browser/browser_process.h" |
-#include "chrome/browser/component_updater/component_patcher.h" |
-#include "chrome/browser/component_updater/component_unpacker.h" |
-#include "chrome/browser/component_updater/component_updater_ping_manager.h" |
-#include "chrome/browser/component_updater/crx_update_item.h" |
-#include "chrome/common/chrome_version_info.h" |
-#include "chrome/common/extensions/update_manifest.h" |
-#include "content/public/browser/browser_thread.h" |
-#include "content/public/browser/resource_controller.h" |
-#include "content/public/browser/resource_throttle.h" |
-#include "extensions/common/extension.h" |
-#include "net/base/escape.h" |
-#include "net/base/load_flags.h" |
-#include "net/base/net_errors.h" |
-#include "net/url_request/url_fetcher.h" |
-#include "net/url_request/url_fetcher_delegate.h" |
-#include "net/url_request/url_request.h" |
-#include "net/url_request/url_request_status.h" |
-#include "url/gurl.h" |
- |
-using content::BrowserThread; |
-using extensions::Extension; |
- |
-// The component updater is designed to live until process shutdown, so |
-// base::Bind() calls are not refcounted. |
- |
-namespace { |
- |
-// Extends an omaha compatible update check url |query| string. Does |
-// not mutate the string if it would be longer than |limit| chars. |
-bool AddQueryString(const std::string& id, |
- const std::string& version, |
- const std::string& fingerprint, |
- bool ondemand, |
- size_t limit, |
- std::string* query) { |
- std::string additional = |
- base::StringPrintf("id=%s&v=%s&fp=%s&uc%s", |
- id.c_str(), |
- version.c_str(), |
- fingerprint.c_str(), |
- ondemand ? "&installsource=ondemand" : ""); |
- additional = "x=" + net::EscapeQueryParamValue(additional, true); |
- if ((additional.size() + query->size() + 1) > limit) |
- return false; |
- if (!query->empty()) |
- query->append(1, '&'); |
- query->append(additional); |
- return true; |
-} |
- |
-// Create the final omaha compatible query. The |extra| is optional and can |
-// be null. It should contain top level (non-escaped) parameters. |
-std::string MakeFinalQuery(const std::string& host, |
- const std::string& query, |
- const char* extra) { |
- std::string request(host); |
- request.append(1, '?'); |
- if (extra) { |
- request.append(extra); |
- request.append(1, '&'); |
- } |
- request.append(query); |
- return request; |
-} |
- |
-// Produces an extension-like friendly |id|. This might be removed in the |
-// future if we roll our on packing tools. |
-static std::string HexStringToID(const std::string& hexstr) { |
- std::string id; |
- for (size_t i = 0; i < hexstr.size(); ++i) { |
- int val; |
- if (base::HexStringToInt(base::StringPiece(hexstr.begin() + i, |
- hexstr.begin() + i + 1), |
- &val)) { |
- id.append(1, val + 'a'); |
- } else { |
- id.append(1, 'a'); |
- } |
- } |
- DCHECK(Extension::IdIsValid(id)); |
- return id; |
-} |
- |
-// Helper to do version check for components. |
-bool IsVersionNewer(const Version& current, const std::string& proposed) { |
- Version proposed_ver(proposed); |
- if (!proposed_ver.IsValid()) |
- return false; |
- return (current.CompareTo(proposed_ver) < 0); |
-} |
- |
-// Helper template class that allows our main class to have separate |
-// OnURLFetchComplete() callbacks for different types of url requests |
-// they are differentiated by the |Ctx| type. |
-template <typename Del, typename Ctx> |
-class DelegateWithContext : public net::URLFetcherDelegate { |
- public: |
- DelegateWithContext(Del* delegate, Ctx* context) |
- : delegate_(delegate), context_(context) {} |
- |
- virtual void OnURLFetchComplete(const net::URLFetcher* source) OVERRIDE { |
- delegate_->OnURLFetchComplete(source, context_); |
- delete this; |
- } |
- |
- private: |
- ~DelegateWithContext() {} |
- |
- Del* delegate_; |
- Ctx* context_; |
-}; |
-// This function creates the right DelegateWithContext using template inference. |
-template <typename Del, typename Ctx> |
-net::URLFetcherDelegate* MakeContextDelegate(Del* delegate, Ctx* context) { |
- return new DelegateWithContext<Del, Ctx>(delegate, context); |
-} |
- |
-// Helper to start a url request using |fetcher| with the common flags. |
-void StartFetch(net::URLFetcher* fetcher, |
- net::URLRequestContextGetter* context_getter, |
- bool save_to_file, |
- scoped_refptr<base::SequencedTaskRunner> task_runner) { |
- fetcher->SetRequestContext(context_getter); |
- fetcher->SetLoadFlags(net::LOAD_DO_NOT_SEND_COOKIES | |
- net::LOAD_DO_NOT_SAVE_COOKIES | |
- net::LOAD_DISABLE_CACHE); |
- // TODO(cpu): Define our retry and backoff policy. |
- fetcher->SetAutomaticallyRetryOn5xx(false); |
- if (save_to_file) { |
- fetcher->SaveResponseToTemporaryFile(task_runner); |
- } |
- fetcher->Start(); |
-} |
- |
-// Returns true if the url request of |fetcher| was succesful. |
-bool FetchSuccess(const net::URLFetcher& fetcher) { |
- return (fetcher.GetStatus().status() == net::URLRequestStatus::SUCCESS) && |
- (fetcher.GetResponseCode() == 200); |
-} |
- |
-// Returns the error code which occured during the fetch.The function returns 0 |
-// if the fetch was successful. If errors happen, the function could return a |
-// network error, an http response code, or the status of the fetch, if the |
-// fetch is pending or canceled. |
-int GetFetchError(const net::URLFetcher& fetcher) { |
- if (FetchSuccess(fetcher)) |
- return 0; |
- |
- const net::URLRequestStatus::Status status(fetcher.GetStatus().status()); |
- if (status == net::URLRequestStatus::FAILED) |
- return fetcher.GetStatus().error(); |
- |
- if (status == net::URLRequestStatus::IO_PENDING || |
- status == net::URLRequestStatus::CANCELED) |
- return status; |
- |
- const int response_code(fetcher.GetResponseCode()); |
- if (status == net::URLRequestStatus::SUCCESS && response_code != 200) |
- return response_code; |
- |
- return -1; |
- } |
- |
-// Returns true if a differential update is available for the update item. |
-bool IsDiffUpdateAvailable(const CrxUpdateItem* update_item) { |
- return update_item->diff_crx_url.is_valid(); |
-} |
- |
-// Returns true if a differential update is available, it has not failed yet, |
-// and the configuration allows it. |
-bool CanTryDiffUpdate(const CrxUpdateItem* update_item, |
- const ComponentUpdateService::Configurator& config) { |
- return IsDiffUpdateAvailable(update_item) && |
- !update_item->diff_update_failed && |
- config.DeltasEnabled(); |
-} |
- |
-} // namespace |
- |
-CrxUpdateItem::CrxUpdateItem() |
- : status(kNew), |
- on_demand(false), |
- diff_update_failed(false), |
- error_category(0), |
- error_code(0), |
- extra_code1(0), |
- diff_error_category(0), |
- diff_error_code(0), |
- diff_extra_code1(0) { |
-} |
- |
-CrxUpdateItem::~CrxUpdateItem() { |
-} |
- |
-CrxComponent::CrxComponent() |
- : installer(NULL), |
- observer(NULL) { |
-} |
- |
-CrxComponent::~CrxComponent() { |
-} |
- |
-std::string GetCrxComponentID(const CrxComponent& component) { |
- return HexStringToID(StringToLowerASCII(base::HexEncode(&component.pk_hash[0], |
- component.pk_hash.size()/2))); |
-} |
- |
-CrxComponentInfo::CrxComponentInfo() { |
-} |
- |
-CrxComponentInfo::~CrxComponentInfo() { |
-} |
- |
-/////////////////////////////////////////////////////////////////////////////// |
-// In charge of blocking url requests until the |crx_id| component has been |
-// updated. This class is touched solely from the IO thread. The UI thread |
-// can post tasks to it via weak pointers. By default the request is blocked |
-// unless the CrxUpdateService calls Unblock(). |
-// The lifetime is controlled by Chrome's resource loader so the component |
-// updater cannot touch objects from this class except via weak pointers. |
-class CUResourceThrottle |
- : public content::ResourceThrottle, |
- public base::SupportsWeakPtr<CUResourceThrottle> { |
- public: |
- explicit CUResourceThrottle(const net::URLRequest* request); |
- virtual ~CUResourceThrottle(); |
- // Overriden from ResourceThrottle. |
- virtual void WillStartRequest(bool* defer) OVERRIDE; |
- virtual void WillRedirectRequest(const GURL& new_url, bool* defer) OVERRIDE; |
- |
- // Component updater calls this function via PostTask to unblock the request. |
- void Unblock(); |
- |
- typedef std::vector<base::WeakPtr<CUResourceThrottle> > WeakPtrVector; |
- |
- private: |
- enum State { |
- NEW, |
- BLOCKED, |
- UNBLOCKED |
- }; |
- |
- State state_; |
-}; |
- |
-void UnblockResourceThrottle(base::WeakPtr<CUResourceThrottle> rt) { |
- BrowserThread::PostTask( |
- BrowserThread::IO, |
- FROM_HERE, |
- base::Bind(&CUResourceThrottle::Unblock, rt)); |
-} |
- |
-void UnblockandReapAllThrottles(CUResourceThrottle::WeakPtrVector* throttles) { |
- CUResourceThrottle::WeakPtrVector::iterator it; |
- for (it = throttles->begin(); it != throttles->end(); ++it) |
- UnblockResourceThrottle(*it); |
- throttles->clear(); |
-} |
- |
-////////////////////////////////////////////////////////////////////////////// |
-// The one and only implementation of the ComponentUpdateService interface. In |
-// charge of running the show. The main method is ProcessPendingItems() which |
-// is called periodically to do the upgrades/installs or the update checks. |
-// An important consideration here is to be as "low impact" as we can to the |
-// rest of the browser, so even if we have many components registered and |
-// eligible for update, we only do one thing at a time with pauses in between |
-// the tasks. Also when we do network requests there is only one |url_fetcher_| |
-// in flight at a time. |
-// There are no locks in this code, the main structure |work_items_| is mutated |
-// only from the UI thread. The unpack and installation is done in the file |
-// thread and the network requests are done in the IO thread and in the file |
-// thread. |
-class CrxUpdateService : public ComponentUpdateService { |
- public: |
- explicit CrxUpdateService(ComponentUpdateService::Configurator* config); |
- |
- virtual ~CrxUpdateService(); |
- |
- // Overrides for ComponentUpdateService. |
- virtual Status Start() OVERRIDE; |
- virtual Status Stop() OVERRIDE; |
- virtual Status RegisterComponent(const CrxComponent& component) OVERRIDE; |
- virtual Status OnDemandUpdate(const std::string& component_id) OVERRIDE; |
- virtual void GetComponents( |
- std::vector<CrxComponentInfo>* components) OVERRIDE; |
- virtual content::ResourceThrottle* GetOnDemandResourceThrottle( |
- net::URLRequest* request, const std::string& crx_id) OVERRIDE; |
- |
- // Context for a update check url request. See DelegateWithContext above. |
- struct UpdateContext { |
- base::Time start; |
- UpdateContext() : start(base::Time::Now()) {} |
- }; |
- |
- // Context for a crx download url request. See DelegateWithContext above. |
- struct CRXContext { |
- ComponentInstaller* installer; |
- std::vector<uint8> pk_hash; |
- std::string id; |
- std::string fingerprint; |
- CRXContext() : installer(NULL) {} |
- }; |
- |
- void OnURLFetchComplete(const net::URLFetcher* source, |
- UpdateContext* context); |
- |
- void OnURLFetchComplete(const net::URLFetcher* source, |
- CRXContext* context); |
- |
- private: |
- enum ErrorCategory { |
- kErrorNone = 0, |
- kNetworkError, |
- kUnpackError, |
- kInstallError, |
- }; |
- |
- enum StepDelayInterval { |
- kStepDelayShort = 0, |
- kStepDelayMedium, |
- kStepDelayLong, |
- }; |
- |
- |
- void OnParseUpdateManifestSucceeded(const UpdateManifest::Results& results); |
- |
- void OnParseUpdateManifestFailed(const std::string& error_message); |
- |
- bool AddItemToUpdateCheck(CrxUpdateItem* item, std::string* query); |
- |
- Status OnDemandUpdateInternal(CrxUpdateItem* item); |
- |
- void ProcessPendingItems(); |
- |
- CrxUpdateItem* FindReadyComponent(); |
- |
- void UpdateComponent(CrxUpdateItem* workitem); |
- |
- void AddUpdateCheckItems(std::string* query); |
- |
- void DoUpdateCheck(const std::string& query); |
- |
- void ScheduleNextRun(StepDelayInterval step_delay); |
- |
- void ParseManifest(const std::string& xml); |
- |
- void Install(const CRXContext* context, const base::FilePath& crx_path); |
- |
- void DoneInstalling(const std::string& component_id, |
- ComponentUnpacker::Error error, |
- int extended_error); |
- |
- void ChangeItemState(CrxUpdateItem* item, CrxUpdateItem::Status to); |
- |
- size_t ChangeItemStatus(CrxUpdateItem::Status from, |
- CrxUpdateItem::Status to); |
- |
- CrxUpdateItem* FindUpdateItemById(const std::string& id); |
- |
- void NotifyComponentObservers(ComponentObserver::Events event, |
- int extra) const; |
- |
- bool HasOnDemandItems() const; |
- |
- void OnNewResourceThrottle(base::WeakPtr<CUResourceThrottle> rt, |
- const std::string& crx_id); |
- |
- scoped_ptr<ComponentUpdateService::Configurator> config_; |
- |
- scoped_ptr<ComponentPatcher> component_patcher_; |
- |
- scoped_ptr<net::URLFetcher> url_fetcher_; |
- |
- scoped_ptr<component_updater::PingManager> ping_manager_; |
- |
- // A collection of every work item. |
- typedef std::vector<CrxUpdateItem*> UpdateItems; |
- UpdateItems work_items_; |
- |
- base::OneShotTimer<CrxUpdateService> timer_; |
- |
- scoped_refptr<base::SequencedTaskRunner> blocking_task_runner_; |
- |
- const Version chrome_version_; |
- |
- bool running_; |
- |
- DISALLOW_COPY_AND_ASSIGN(CrxUpdateService); |
-}; |
- |
-////////////////////////////////////////////////////////////////////////////// |
- |
-CrxUpdateService::CrxUpdateService(ComponentUpdateService::Configurator* config) |
- : config_(config), |
- component_patcher_(config->CreateComponentPatcher()), |
- ping_manager_(new component_updater::PingManager( |
- config->PingUrl(), |
- config->RequestContext())), |
- blocking_task_runner_(BrowserThread::GetBlockingPool()-> |
- GetSequencedTaskRunnerWithShutdownBehavior( |
- BrowserThread::GetBlockingPool()->GetSequenceToken(), |
- base::SequencedWorkerPool::SKIP_ON_SHUTDOWN)), |
- chrome_version_(chrome::VersionInfo().Version()), |
- running_(false) { |
-} |
- |
-CrxUpdateService::~CrxUpdateService() { |
- // Because we are a singleton, at this point only the UI thread should be |
- // alive, this simplifies the management of the work that could be in |
- // flight in other threads. |
- Stop(); |
- STLDeleteElements(&work_items_); |
-} |
- |
-ComponentUpdateService::Status CrxUpdateService::Start() { |
- // Note that RegisterComponent will call Start() when the first |
- // component is registered, so it can be called twice. This way |
- // we avoid scheduling the timer if there is no work to do. |
- running_ = true; |
- if (work_items_.empty()) |
- return kOk; |
- |
- NotifyComponentObservers(ComponentObserver::COMPONENT_UPDATER_STARTED, 0); |
- |
- timer_.Start(FROM_HERE, base::TimeDelta::FromSeconds(config_->InitialDelay()), |
- this, &CrxUpdateService::ProcessPendingItems); |
- return kOk; |
-} |
- |
-// Stop the main check + update loop. In flight operations will be |
-// completed. |
-ComponentUpdateService::Status CrxUpdateService::Stop() { |
- running_ = false; |
- timer_.Stop(); |
- return kOk; |
-} |
- |
-bool CrxUpdateService::HasOnDemandItems() const { |
- class Helper { |
- public: |
- static bool IsOnDemand(CrxUpdateItem* item) { |
- return item->on_demand; |
- } |
- }; |
- return std::find_if(work_items_.begin(), |
- work_items_.end(), |
- Helper::IsOnDemand) != work_items_.end(); |
-} |
- |
-// This function sets the timer which will call ProcessPendingItems() or |
-// ProcessRequestedItem() if there is an on_demand item. There |
-// are three kinds of waits: |
-// - a short delay, when there is immediate work to be done. |
-// - a medium delay, when there are updates to be applied within the current |
-// update cycle, or there are components that are still unchecked. |
-// - a long delay when a full check/update cycle has completed for all |
-// components. |
-void CrxUpdateService::ScheduleNextRun(StepDelayInterval step_delay) { |
- DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
- DCHECK(url_fetcher_.get() == NULL); |
- CHECK(!timer_.IsRunning()); |
- // It could be the case that Stop() had been called while a url request |
- // or unpacking was in flight, if so we arrive here but |running_| is |
- // false. In that case do not loop again. |
- if (!running_) |
- return; |
- |
- // Keep the delay short if in the middle of an update (step_delay), |
- // or there are new requested_work_items_ that have not been processed yet. |
- int64 delay_seconds = 0; |
- if (!HasOnDemandItems()) { |
- switch (step_delay) { |
- case kStepDelayShort: |
- delay_seconds = config_->StepDelay(); |
- break; |
- case kStepDelayMedium: |
- delay_seconds = config_->StepDelayMedium(); |
- break; |
- case kStepDelayLong: |
- delay_seconds = config_->NextCheckDelay(); |
- break; |
- } |
- } else { |
- delay_seconds = config_->StepDelay(); |
- } |
- |
- if (step_delay != kStepDelayShort) { |
- NotifyComponentObservers(ComponentObserver::COMPONENT_UPDATER_SLEEPING, 0); |
- |
- // Zero is only used for unit tests. |
- if (0 == delay_seconds) |
- return; |
- } |
- |
- timer_.Start(FROM_HERE, base::TimeDelta::FromSeconds(delay_seconds), |
- this, &CrxUpdateService::ProcessPendingItems); |
-} |
- |
-// Given a extension-like component id, find the associated component. |
-CrxUpdateItem* CrxUpdateService::FindUpdateItemById(const std::string& id) { |
- DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
- CrxUpdateItem::FindById finder(id); |
- UpdateItems::iterator it = std::find_if(work_items_.begin(), |
- work_items_.end(), |
- finder); |
- if (it == work_items_.end()) |
- return NULL; |
- return (*it); |
-} |
- |
-// Changes a component's status, clearing on_demand and firing notifications as |
-// necessary. By convention, this is the only function that can change a |
-// CrxUpdateItem's |status|. |
-// TODO(waffles): Do we want to add DCHECKS for valid state transitions here? |
-void CrxUpdateService::ChangeItemState(CrxUpdateItem* item, |
- CrxUpdateItem::Status to) { |
- DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
- if (to == CrxUpdateItem::kNoUpdate || |
- to == CrxUpdateItem::kUpdated || |
- to == CrxUpdateItem::kUpToDate) { |
- item->on_demand = false; |
- } |
- |
- item->status = to; |
- |
- ComponentObserver* observer = item->component.observer; |
- if (observer) { |
- switch (to) { |
- case CrxUpdateItem::kCanUpdate: |
- observer->OnEvent(ComponentObserver::COMPONENT_UPDATE_FOUND, 0); |
- break; |
- case CrxUpdateItem::kUpdatingDiff: |
- case CrxUpdateItem::kUpdating: |
- observer->OnEvent(ComponentObserver::COMPONENT_UPDATE_READY, 0); |
- break; |
- case CrxUpdateItem::kUpdated: |
- observer->OnEvent(ComponentObserver::COMPONENT_UPDATED, 0); |
- break; |
- case CrxUpdateItem::kUpToDate: |
- case CrxUpdateItem::kNoUpdate: |
- observer->OnEvent(ComponentObserver::COMPONENT_NOT_UPDATED, 0); |
- break; |
- case CrxUpdateItem::kNew: |
- case CrxUpdateItem::kChecking: |
- case CrxUpdateItem::kDownloading: |
- case CrxUpdateItem::kDownloadingDiff: |
- case CrxUpdateItem::kLastStatus: |
- // No notification for these states. |
- break; |
- } |
- } |
- |
- // Free possible pending network requests. |
- if ((to == CrxUpdateItem::kUpdated) || |
- (to == CrxUpdateItem::kUpToDate) || |
- (to == CrxUpdateItem::kNoUpdate)) { |
- UnblockandReapAllThrottles(&item->throttles); |
- } |
-} |
- |
-// Changes all the components in |work_items_| that have |from| status to |
-// |to| status and returns how many have been changed. |
-size_t CrxUpdateService::ChangeItemStatus(CrxUpdateItem::Status from, |
- CrxUpdateItem::Status to) { |
- DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
- size_t count = 0; |
- for (UpdateItems::iterator it = work_items_.begin(); |
- it != work_items_.end(); ++it) { |
- CrxUpdateItem* item = *it; |
- if (item->status != from) |
- continue; |
- ChangeItemState(item, to); |
- ++count; |
- } |
- return count; |
-} |
- |
-// Adds a component to be checked for upgrades. If the component exists it |
-// it will be replaced and the return code is kReplaced. |
-// |
-// TODO(cpu): Evaluate if we want to support un-registration. |
-ComponentUpdateService::Status CrxUpdateService::RegisterComponent( |
- const CrxComponent& component) { |
- DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
- if (component.pk_hash.empty() || |
- !component.version.IsValid() || |
- !component.installer) |
- return kError; |
- |
- std::string id = |
- HexStringToID(StringToLowerASCII(base::HexEncode(&component.pk_hash[0], |
- component.pk_hash.size()/2))); |
- CrxUpdateItem* uit; |
- uit = FindUpdateItemById(id); |
- if (uit) { |
- uit->component = component; |
- return kReplaced; |
- } |
- |
- uit = new CrxUpdateItem; |
- uit->id.swap(id); |
- uit->component = component; |
- |
- work_items_.push_back(uit); |
- // If this is the first component registered we call Start to |
- // schedule the first timer. |
- if (running_ && (work_items_.size() == 1)) |
- Start(); |
- |
- return kOk; |
-} |
- |
-// Sets a component to be checked for updates. |
-// The component to add is |item| and the |query| string is modified with the |
-// required omaha compatible query. Returns false when the query string is |
-// longer than specified by UrlSizeLimit(). |
-bool CrxUpdateService::AddItemToUpdateCheck(CrxUpdateItem* item, |
- std::string* query) { |
- if (!AddQueryString(item->id, |
- item->component.version.GetString(), |
- item->component.fingerprint, |
- item->on_demand, |
- config_->UrlSizeLimit(), |
- query)) |
- return false; |
- |
- ChangeItemState(item, CrxUpdateItem::kChecking); |
- item->last_check = base::Time::Now(); |
- item->previous_version = item->component.version; |
- item->next_version = Version(); |
- item->previous_fp = item->component.fingerprint; |
- item->next_fp.clear(); |
- item->diff_update_failed = false; |
- item->error_category = 0; |
- item->error_code = 0; |
- item->extra_code1 = 0; |
- item->diff_error_category = 0; |
- item->diff_error_code = 0; |
- item->diff_extra_code1 = 0; |
- return true; |
-} |
- |
-// Start the process of checking for an update, for a particular component |
-// that was previously registered. |
-// |component_id| is a value returned from GetCrxComponentID(). |
-ComponentUpdateService::Status CrxUpdateService::OnDemandUpdate( |
- const std::string& component_id) { |
- return OnDemandUpdateInternal(FindUpdateItemById(component_id)); |
-} |
- |
-ComponentUpdateService::Status CrxUpdateService::OnDemandUpdateInternal( |
- CrxUpdateItem* uit) { |
- if (!uit) |
- return kError; |
- // Check if the request is too soon. |
- base::TimeDelta delta = base::Time::Now() - uit->last_check; |
- if (delta < base::TimeDelta::FromSeconds(config_->OnDemandDelay())) |
- return kError; |
- |
- switch (uit->status) { |
- // If the item is already in the process of being updated, there is |
- // no point in this call, so return kInProgress. |
- case CrxUpdateItem::kChecking: |
- case CrxUpdateItem::kCanUpdate: |
- case CrxUpdateItem::kDownloadingDiff: |
- case CrxUpdateItem::kDownloading: |
- case CrxUpdateItem::kUpdatingDiff: |
- case CrxUpdateItem::kUpdating: |
- return kInProgress; |
- // Otherwise the item was already checked a while back (or it is new), |
- // set its status to kNew to give it a slightly higher priority. |
- case CrxUpdateItem::kNew: |
- case CrxUpdateItem::kUpdated: |
- case CrxUpdateItem::kUpToDate: |
- case CrxUpdateItem::kNoUpdate: |
- ChangeItemState(uit, CrxUpdateItem::kNew); |
- uit->on_demand = true; |
- break; |
- case CrxUpdateItem::kLastStatus: |
- NOTREACHED() << uit->status; |
- } |
- |
- // In case the current delay is long, set the timer to a shorter value |
- // to get the ball rolling. |
- if (timer_.IsRunning()) { |
- timer_.Stop(); |
- timer_.Start(FROM_HERE, base::TimeDelta::FromSeconds(config_->StepDelay()), |
- this, &CrxUpdateService::ProcessPendingItems); |
- } |
- |
- return kOk; |
-} |
- |
-void CrxUpdateService::GetComponents( |
- std::vector<CrxComponentInfo>* components) { |
- DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
- for (UpdateItems::const_iterator it = work_items_.begin(); |
- it != work_items_.end(); ++it) { |
- const CrxUpdateItem* item = *it; |
- CrxComponentInfo info; |
- info.id = GetCrxComponentID(item->component); |
- info.version = item->component.version.GetString(); |
- info.name = item->component.name; |
- components->push_back(info); |
- } |
-} |
- |
-// This is the main loop of the component updater. |
-void CrxUpdateService::ProcessPendingItems() { |
- DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
- CrxUpdateItem* ready_upgrade = FindReadyComponent(); |
- if (ready_upgrade) { |
- UpdateComponent(ready_upgrade); |
- return; |
- } |
- std::string query; |
- AddUpdateCheckItems(&query); |
- if (!query.empty()) { |
- DoUpdateCheck(query); |
- return; |
- } |
- // No components to update. The next check will be after a long sleep. |
- ScheduleNextRun(kStepDelayLong); |
-} |
- |
-CrxUpdateItem* CrxUpdateService::FindReadyComponent() { |
- class Helper { |
- public: |
- static bool IsReadyOnDemand(CrxUpdateItem* item) { |
- return item->on_demand && IsReady(item); |
- } |
- static bool IsReady(CrxUpdateItem* item) { |
- return item->status == CrxUpdateItem::kCanUpdate; |
- } |
- }; |
- |
- std::vector<CrxUpdateItem*>::iterator it = std::find_if( |
- work_items_.begin(), work_items_.end(), Helper::IsReadyOnDemand); |
- if (it != work_items_.end()) |
- return *it; |
- it = std::find_if(work_items_.begin(), work_items_.end(), Helper::IsReady); |
- if (it != work_items_.end()) |
- return *it; |
- return NULL; |
-} |
- |
-void CrxUpdateService::UpdateComponent(CrxUpdateItem* workitem) { |
- CRXContext* context = new CRXContext; |
- context->pk_hash = workitem->component.pk_hash; |
- context->id = workitem->id; |
- context->installer = workitem->component.installer; |
- context->fingerprint = workitem->next_fp; |
- GURL package_url; |
- if (CanTryDiffUpdate(workitem, *config_)) { |
- package_url = workitem->diff_crx_url; |
- ChangeItemState(workitem, CrxUpdateItem::kDownloadingDiff); |
- } else { |
- package_url = workitem->crx_url; |
- ChangeItemState(workitem, CrxUpdateItem::kDownloading); |
- } |
- url_fetcher_.reset(net::URLFetcher::Create( |
- 0, package_url, net::URLFetcher::GET, |
- MakeContextDelegate(this, context))); |
- StartFetch(url_fetcher_.get(), |
- config_->RequestContext(), |
- true, |
- blocking_task_runner_); |
-} |
- |
-// Given that our |work_items_| list is expected to contain relatively few |
-// items, we simply loop several times. |
-void CrxUpdateService::AddUpdateCheckItems(std::string* query){ |
- for (UpdateItems::const_iterator it = work_items_.begin(); |
- it != work_items_.end(); ++it) { |
- CrxUpdateItem* item = *it; |
- if (item->status != CrxUpdateItem::kNew) |
- continue; |
- if (!AddItemToUpdateCheck(item, query)) |
- break; |
- } |
- |
- // Next we can go back to components we already checked, here |
- // we can also batch them in a single url request, as long as |
- // we have not checked them recently. |
- const base::TimeDelta min_delta_time = |
- base::TimeDelta::FromSeconds(config_->MinimumReCheckWait()); |
- |
- for (UpdateItems::const_iterator it = work_items_.begin(); |
- it != work_items_.end(); ++it) { |
- CrxUpdateItem* item = *it; |
- if ((item->status != CrxUpdateItem::kNoUpdate) && |
- (item->status != CrxUpdateItem::kUpToDate)) |
- continue; |
- base::TimeDelta delta = base::Time::Now() - item->last_check; |
- if (delta < min_delta_time) |
- continue; |
- if (!AddItemToUpdateCheck(item, query)) |
- break; |
- } |
- |
- // Finally, we check components that we already updated as long as |
- // we have not checked them recently. |
- for (UpdateItems::const_iterator it = work_items_.begin(); |
- it != work_items_.end(); ++it) { |
- CrxUpdateItem* item = *it; |
- if (item->status != CrxUpdateItem::kUpdated) |
- continue; |
- base::TimeDelta delta = base::Time::Now() - item->last_check; |
- if (delta < min_delta_time) |
- continue; |
- if (!AddItemToUpdateCheck(item, query)) |
- break; |
- } |
-} |
- |
-void CrxUpdateService::DoUpdateCheck(const std::string& query) { |
- const std::string full_query = |
- MakeFinalQuery(config_->UpdateUrl().spec(), |
- query, |
- config_->ExtraRequestParams()); |
- |
- url_fetcher_.reset(net::URLFetcher::Create( |
- 0, GURL(full_query), net::URLFetcher::GET, |
- MakeContextDelegate(this, new UpdateContext()))); |
- StartFetch(url_fetcher_.get(), |
- config_->RequestContext(), |
- false, |
- blocking_task_runner_); |
-} |
- |
-// Called when we got a response from the update server. It consists of an xml |
-// document following the omaha update scheme. |
-void CrxUpdateService::OnURLFetchComplete(const net::URLFetcher* source, |
- UpdateContext* context) { |
- DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
- if (FetchSuccess(*source)) { |
- std::string xml; |
- source->GetResponseAsString(&xml); |
- url_fetcher_.reset(); |
- ParseManifest(xml); |
- } else { |
- url_fetcher_.reset(); |
- CrxUpdateService::OnParseUpdateManifestFailed("network error"); |
- } |
- delete context; |
-} |
- |
-// Parsing the manifest is either done right now for tests or in a sandboxed |
-// process for the production environment. This mitigates the case where an |
-// attacker was able to feed us a malicious xml string. |
-void CrxUpdateService::ParseManifest(const std::string& xml) { |
- DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
- UpdateManifest manifest; |
- if (!manifest.Parse(xml)) |
- CrxUpdateService::OnParseUpdateManifestFailed(manifest.errors()); |
- else |
- CrxUpdateService::OnParseUpdateManifestSucceeded(manifest.results()); |
-} |
- |
-// A valid Omaha update check has arrived, from only the list of components that |
-// we are currently upgrading we check for a match in which the server side |
-// version is newer, if so we queue them for an upgrade. The next time we call |
-// ProcessPendingItems() one of them will be drafted for the upgrade process. |
-void CrxUpdateService::OnParseUpdateManifestSucceeded( |
- const UpdateManifest::Results& results) { |
- DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
- size_t update_pending = 0; |
- std::vector<UpdateManifest::Result>::const_iterator it; |
- for (it = results.list.begin(); it != results.list.end(); ++it) { |
- CrxUpdateItem* crx = FindUpdateItemById(it->extension_id); |
- if (!crx) |
- continue; |
- |
- if (crx->status != CrxUpdateItem::kChecking) |
- continue; // Not updating this component now. |
- |
- if (it->version.empty()) { |
- // No version means no update available. |
- ChangeItemState(crx, CrxUpdateItem::kNoUpdate); |
- continue; |
- } |
- if (!IsVersionNewer(crx->component.version, it->version)) { |
- // Our component is up to date. |
- ChangeItemState(crx, CrxUpdateItem::kUpToDate); |
- continue; |
- } |
- if (!it->browser_min_version.empty()) { |
- if (IsVersionNewer(chrome_version_, it->browser_min_version)) { |
- // Does not apply for this chrome version. |
- ChangeItemState(crx, CrxUpdateItem::kNoUpdate); |
- continue; |
- } |
- } |
- // All test passed. Queue an upgrade for this component and fire the |
- // notifications. |
- crx->crx_url = it->crx_url; |
- crx->diff_crx_url = it->diff_crx_url; |
- ChangeItemState(crx, CrxUpdateItem::kCanUpdate); |
- crx->next_version = Version(it->version); |
- crx->next_fp = it->package_fingerprint; |
- ++update_pending; |
- } |
- |
- // All the components that are not mentioned in the manifest we |
- // consider them up to date. |
- ChangeItemStatus(CrxUpdateItem::kChecking, CrxUpdateItem::kUpToDate); |
- |
- // If there are updates pending we do a short wait, otherwise we take |
- // a longer delay until we check the components again. |
- ScheduleNextRun(update_pending > 0 ? kStepDelayShort : kStepDelayMedium); |
-} |
- |
-void CrxUpdateService::OnParseUpdateManifestFailed( |
- const std::string& error_message) { |
- DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
- size_t count = ChangeItemStatus(CrxUpdateItem::kChecking, |
- CrxUpdateItem::kNoUpdate); |
- DCHECK_GT(count, 0ul); |
- ScheduleNextRun(kStepDelayLong); |
-} |
- |
-// Called when the CRX package has been downloaded to a temporary location. |
-// Here we fire the notifications and schedule the component-specific installer |
-// to be called in the file thread. |
-void CrxUpdateService::OnURLFetchComplete(const net::URLFetcher* source, |
- CRXContext* context) { |
- DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
- |
- scoped_ptr<CRXContext> crx_context(context); |
- |
- CrxUpdateItem* crx = FindUpdateItemById(crx_context->id); |
- DCHECK(crx->status == CrxUpdateItem::kDownloadingDiff || |
- crx->status == CrxUpdateItem::kDownloading); |
- |
- if (!FetchSuccess(*source)) { |
- if (crx->status == CrxUpdateItem::kDownloadingDiff) { |
- crx->diff_error_category = kNetworkError; |
- crx->diff_error_code = GetFetchError(*source); |
- crx->diff_update_failed = true; |
- size_t count = ChangeItemStatus(CrxUpdateItem::kDownloadingDiff, |
- CrxUpdateItem::kCanUpdate); |
- DCHECK_EQ(count, 1ul); |
- url_fetcher_.reset(); |
- |
- ScheduleNextRun(kStepDelayShort); |
- return; |
- } |
- crx->error_category = kNetworkError; |
- crx->error_code = GetFetchError(*source); |
- size_t count = ChangeItemStatus(CrxUpdateItem::kDownloading, |
- CrxUpdateItem::kNoUpdate); |
- DCHECK_EQ(count, 1ul); |
- url_fetcher_.reset(); |
- |
- // At this point, since both the differential and the full downloads failed, |
- // the update for this component has finished with an error. |
- ping_manager_->OnUpdateComplete(crx); |
- |
- // Move on to the next update, if there is one available. |
- ScheduleNextRun(kStepDelayMedium); |
- } else { |
- base::FilePath temp_crx_path; |
- CHECK(source->GetResponseAsFilePath(true, &temp_crx_path)); |
- |
- size_t count = 0; |
- if (crx->status == CrxUpdateItem::kDownloadingDiff) { |
- count = ChangeItemStatus(CrxUpdateItem::kDownloadingDiff, |
- CrxUpdateItem::kUpdatingDiff); |
- } else { |
- count = ChangeItemStatus(CrxUpdateItem::kDownloading, |
- CrxUpdateItem::kUpdating); |
- } |
- DCHECK_EQ(count, 1ul); |
- |
- url_fetcher_.reset(); |
- |
- // Why unretained? See comment at top of file. |
- blocking_task_runner_->PostDelayedTask( |
- FROM_HERE, |
- base::Bind(&CrxUpdateService::Install, |
- base::Unretained(this), |
- crx_context.release(), |
- temp_crx_path), |
- base::TimeDelta::FromMilliseconds(config_->StepDelay())); |
- } |
-} |
- |
-// Install consists of digital signature verification, unpacking and then |
-// calling the component specific installer. All that is handled by the |
-// |unpacker|. If there is an error this function is in charge of deleting |
-// the files created. |
-void CrxUpdateService::Install(const CRXContext* context, |
- const base::FilePath& crx_path) { |
- // This function owns the file at |crx_path| and the |context| object. |
- ComponentUnpacker unpacker(context->pk_hash, |
- crx_path, |
- context->fingerprint, |
- component_patcher_.get(), |
- context->installer); |
- if (!base::DeleteFile(crx_path, false)) |
- NOTREACHED() << crx_path.value(); |
- // Why unretained? See comment at top of file. |
- BrowserThread::PostDelayedTask( |
- BrowserThread::UI, |
- FROM_HERE, |
- base::Bind(&CrxUpdateService::DoneInstalling, base::Unretained(this), |
- context->id, unpacker.error(), unpacker.extended_error()), |
- base::TimeDelta::FromMilliseconds(config_->StepDelay())); |
- delete context; |
-} |
- |
-// Installation has been completed. Adjust the component status and |
-// schedule the next check. Schedule a short delay before trying the full |
-// update when the differential update failed. |
-void CrxUpdateService::DoneInstalling(const std::string& component_id, |
- ComponentUnpacker::Error error, |
- int extra_code) { |
- DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
- |
- ErrorCategory error_category = kErrorNone; |
- switch (error) { |
- case ComponentUnpacker::kNone: |
- break; |
- case ComponentUnpacker::kInstallerError: |
- error_category = kInstallError; |
- break; |
- default: |
- error_category = kUnpackError; |
- break; |
- } |
- |
- const bool is_success = error == ComponentUnpacker::kNone; |
- |
- CrxUpdateItem* item = FindUpdateItemById(component_id); |
- if (item->status == CrxUpdateItem::kUpdatingDiff && !is_success) { |
- item->diff_error_category = error_category; |
- item->diff_error_code = error; |
- item->diff_extra_code1 = extra_code; |
- item->diff_update_failed = true; |
- size_t count = ChangeItemStatus(CrxUpdateItem::kUpdatingDiff, |
- CrxUpdateItem::kCanUpdate); |
- DCHECK_EQ(count, 1ul); |
- ScheduleNextRun(kStepDelayShort); |
- return; |
- } |
- |
- if (is_success) { |
- ChangeItemState(item, CrxUpdateItem::kUpdated); |
- item->component.version = item->next_version; |
- item->component.fingerprint = item->next_fp; |
- } else { |
- ChangeItemState(item, CrxUpdateItem::kNoUpdate); |
- item->error_category = error_category; |
- item->error_code = error; |
- item->extra_code1 = extra_code; |
- } |
- |
- ping_manager_->OnUpdateComplete(item); |
- |
- // Move on to the next update, if there is one available. |
- ScheduleNextRun(kStepDelayMedium); |
-} |
- |
-void CrxUpdateService::NotifyComponentObservers( |
- ComponentObserver::Events event, int extra) const { |
- for (UpdateItems::const_iterator it = work_items_.begin(); |
- it != work_items_.end(); ++it) { |
- ComponentObserver* observer = (*it)->component.observer; |
- if (observer) |
- observer->OnEvent(event, 0); |
- } |
-} |
- |
-content::ResourceThrottle* CrxUpdateService::GetOnDemandResourceThrottle( |
- net::URLRequest* request, const std::string& crx_id) { |
- // We give the raw pointer to the caller, who will delete it at will |
- // and we keep for ourselves a weak pointer to it so we can post tasks |
- // from the UI thread without having to track lifetime directly. |
- CUResourceThrottle* rt = new CUResourceThrottle(request); |
- BrowserThread::PostTask( |
- BrowserThread::UI, |
- FROM_HERE, |
- base::Bind(&CrxUpdateService::OnNewResourceThrottle, |
- base::Unretained(this), |
- rt->AsWeakPtr(), |
- crx_id)); |
- return rt; |
-} |
- |
-void CrxUpdateService::OnNewResourceThrottle( |
- base::WeakPtr<CUResourceThrottle> rt, const std::string& crx_id) { |
- DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
- // Check if we can on-demand update, else unblock the request anyway. |
- CrxUpdateItem* item = FindUpdateItemById(crx_id); |
- Status status = OnDemandUpdateInternal(item); |
- if (status == kOk || status == kInProgress) { |
- item->throttles.push_back(rt); |
- return; |
- } |
- UnblockResourceThrottle(rt); |
-} |
- |
-/////////////////////////////////////////////////////////////////////////////// |
- |
-CUResourceThrottle::CUResourceThrottle(const net::URLRequest* request) |
- : state_(NEW) { |
- DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
-} |
- |
-CUResourceThrottle::~CUResourceThrottle() { |
- DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
-} |
- |
-void CUResourceThrottle::WillStartRequest(bool* defer) { |
- if (state_ != UNBLOCKED) { |
- state_ = BLOCKED; |
- *defer = true; |
- } else { |
- *defer = false; |
- } |
-} |
- |
-void CUResourceThrottle::WillRedirectRequest(const GURL& new_url, bool* defer) { |
- WillStartRequest(defer); |
-} |
- |
-void CUResourceThrottle::Unblock() { |
- DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
- if (state_ == BLOCKED) |
- controller()->Resume(); |
- state_ = UNBLOCKED; |
-} |
- |
-// The component update factory. Using the component updater as a singleton |
-// is the job of the browser process. |
-ComponentUpdateService* ComponentUpdateServiceFactory( |
- ComponentUpdateService::Configurator* config) { |
- DCHECK(config); |
- return new CrxUpdateService(config); |
-} |
+// Copyright (c) 2012 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include "chrome/browser/component_updater/component_updater_service.h" |
+ |
+#include <algorithm> |
+#include <set> |
+#include <vector> |
+ |
+#include "base/at_exit.h" |
+#include "base/bind.h" |
+#include "base/compiler_specific.h" |
+#include "base/file_util.h" |
+#include "base/files/file_path.h" |
+#include "base/logging.h" |
+#include "base/memory/scoped_ptr.h" |
+#include "base/memory/weak_ptr.h" |
+#include "base/stl_util.h" |
+#include "base/strings/string_number_conversions.h" |
+#include "base/strings/string_piece.h" |
+#include "base/strings/string_util.h" |
+#include "base/strings/stringprintf.h" |
+#include "base/threading/sequenced_worker_pool.h" |
+#include "base/timer/timer.h" |
+#include "chrome/browser/browser_process.h" |
+#include "chrome/browser/component_updater/component_patcher.h" |
+#include "chrome/browser/component_updater/component_unpacker.h" |
+#include "chrome/browser/component_updater/component_updater_ping_manager.h" |
+#include "chrome/browser/component_updater/component_updater_utils.h" |
+#include "chrome/browser/component_updater/crx_update_item.h" |
+#include "chrome/browser/component_updater/update_manifest.h" |
+#include "chrome/common/chrome_version_info.h" |
+#include "content/public/browser/browser_thread.h" |
+#include "content/public/browser/resource_controller.h" |
+#include "content/public/browser/resource_throttle.h" |
+#include "extensions/common/extension.h" |
+#include "net/base/escape.h" |
+#include "net/base/load_flags.h" |
+#include "net/base/net_errors.h" |
+#include "net/url_request/url_fetcher.h" |
+#include "net/url_request/url_fetcher_delegate.h" |
+#include "net/url_request/url_request.h" |
+#include "net/url_request/url_request_status.h" |
+#include "url/gurl.h" |
+ |
+using content::BrowserThread; |
+ |
+// The component updater is designed to live until process shutdown, so |
+// base::Bind() calls are not refcounted. |
+ |
+namespace { |
+ |
+// Produces an extension-like friendly |id|. This might be removed in the |
+// future if we roll our on packing tools. |
+static std::string HexStringToID(const std::string& hexstr) { |
+ std::string id; |
+ for (size_t i = 0; i < hexstr.size(); ++i) { |
+ int val(0); |
+ if (base::HexStringToInt(base::StringPiece(hexstr.begin() + i, |
+ hexstr.begin() + i + 1), |
+ &val)) { |
+ id.append(1, val + 'a'); |
+ } else { |
+ id.append(1, 'a'); |
+ } |
+ } |
+ DCHECK(extensions::Extension::IdIsValid(id)); |
+ return id; |
+} |
+ |
+// Returns true if the |proposed| version is newer than |current| version. |
+bool IsVersionNewer(const Version& current, const std::string& proposed) { |
+ Version proposed_ver(proposed); |
+ return proposed_ver.IsValid() && current.CompareTo(proposed_ver) < 0; |
+} |
+ |
+// Helper template class that allows our main class to have separate |
+// OnURLFetchComplete() callbacks for different types of url requests |
+// they are differentiated by the |Ctx| type. |
+template <typename Del, typename Ctx> |
+class DelegateWithContext : public net::URLFetcherDelegate { |
+ public: |
+ DelegateWithContext(Del* delegate, Ctx* context) |
+ : delegate_(delegate), context_(context) {} |
+ |
+ virtual void OnURLFetchComplete(const net::URLFetcher* source) OVERRIDE { |
+ delegate_->OnURLFetchComplete(source, context_); |
+ delete this; |
+ } |
+ |
+ private: |
+ ~DelegateWithContext() {} |
+ |
+ Del* delegate_; |
+ Ctx* context_; |
+}; |
+// This function creates the right DelegateWithContext using template inference. |
+template <typename Del, typename Ctx> |
+net::URLFetcherDelegate* MakeContextDelegate(Del* delegate, Ctx* context) { |
+ return new DelegateWithContext<Del, Ctx>(delegate, context); |
+} |
+ |
+// Helper to start a url request using |fetcher| with the common flags. |
+void StartFetch(net::URLFetcher* fetcher, |
+ net::URLRequestContextGetter* context_getter, |
+ bool save_to_file, |
+ scoped_refptr<base::SequencedTaskRunner> task_runner) { |
+ fetcher->SetRequestContext(context_getter); |
+ fetcher->SetLoadFlags(net::LOAD_DO_NOT_SEND_COOKIES | |
+ net::LOAD_DO_NOT_SAVE_COOKIES | |
+ net::LOAD_DISABLE_CACHE); |
+ // TODO(cpu): Define our retry and backoff policy. |
+ fetcher->SetAutomaticallyRetryOn5xx(false); |
+ if (save_to_file) { |
+ fetcher->SaveResponseToTemporaryFile(task_runner); |
+ } |
+ fetcher->Start(); |
+} |
+ |
+// Returns true if the url request of |fetcher| was succesful. |
+bool FetchSuccess(const net::URLFetcher& fetcher) { |
+ return (fetcher.GetStatus().status() == net::URLRequestStatus::SUCCESS) && |
+ (fetcher.GetResponseCode() == 200); |
+} |
+ |
+// Returns the error code which occured during the fetch.The function returns 0 |
+// if the fetch was successful. If errors happen, the function could return a |
+// network error, an http response code, or the status of the fetch, if the |
+// fetch is pending or canceled. |
+int GetFetchError(const net::URLFetcher& fetcher) { |
+ if (FetchSuccess(fetcher)) |
+ return 0; |
+ |
+ const net::URLRequestStatus::Status status(fetcher.GetStatus().status()); |
+ if (status == net::URLRequestStatus::FAILED) |
+ return fetcher.GetStatus().error(); |
+ |
+ if (status == net::URLRequestStatus::IO_PENDING || |
+ status == net::URLRequestStatus::CANCELED) |
+ return status; |
+ |
+ const int response_code(fetcher.GetResponseCode()); |
+ if (status == net::URLRequestStatus::SUCCESS && response_code != 200) |
+ return response_code; |
+ |
+ return -1; |
+ } |
+ |
+// Returns true if a differential update is available for the update item. |
+bool IsDiffUpdateAvailable(const CrxUpdateItem* update_item) { |
+ return update_item->diff_crx_url.is_valid(); |
+} |
+ |
+// Returns true if a differential update is available, it has not failed yet, |
+// and the configuration allows it. |
+bool CanTryDiffUpdate(const CrxUpdateItem* update_item, |
+ const ComponentUpdateService::Configurator& config) { |
+ return IsDiffUpdateAvailable(update_item) && |
+ !update_item->diff_update_failed && |
+ config.DeltasEnabled(); |
+} |
+ |
+} // namespace |
+ |
+CrxUpdateItem::CrxUpdateItem() |
+ : status(kNew), |
+ on_demand(false), |
+ diff_update_failed(false), |
+ error_category(0), |
+ error_code(0), |
+ extra_code1(0), |
+ diff_error_category(0), |
+ diff_error_code(0), |
+ diff_extra_code1(0) { |
+} |
+ |
+CrxUpdateItem::~CrxUpdateItem() { |
+} |
+ |
+CrxComponent::CrxComponent() |
+ : installer(NULL), |
+ observer(NULL) { |
+} |
+ |
+CrxComponent::~CrxComponent() { |
+} |
+ |
+std::string GetCrxComponentID(const CrxComponent& component) { |
+ return HexStringToID(StringToLowerASCII(base::HexEncode(&component.pk_hash[0], |
+ component.pk_hash.size()/2))); |
+} |
+ |
+CrxComponentInfo::CrxComponentInfo() { |
+} |
+ |
+CrxComponentInfo::~CrxComponentInfo() { |
+} |
+ |
+/////////////////////////////////////////////////////////////////////////////// |
+// In charge of blocking url requests until the |crx_id| component has been |
+// updated. This class is touched solely from the IO thread. The UI thread |
+// can post tasks to it via weak pointers. By default the request is blocked |
+// unless the CrxUpdateService calls Unblock(). |
+// The lifetime is controlled by Chrome's resource loader so the component |
+// updater cannot touch objects from this class except via weak pointers. |
+class CUResourceThrottle |
+ : public content::ResourceThrottle, |
+ public base::SupportsWeakPtr<CUResourceThrottle> { |
+ public: |
+ explicit CUResourceThrottle(const net::URLRequest* request); |
+ virtual ~CUResourceThrottle(); |
+ |
+ // Overriden from ResourceThrottle. |
+ virtual void WillStartRequest(bool* defer) OVERRIDE; |
+ virtual void WillRedirectRequest(const GURL& new_url, bool* defer) OVERRIDE; |
+ |
+ // Component updater calls this function via PostTask to unblock the request. |
+ void Unblock(); |
+ |
+ typedef std::vector<base::WeakPtr<CUResourceThrottle> > WeakPtrVector; |
+ |
+ private: |
+ enum State { |
+ NEW, |
+ BLOCKED, |
+ UNBLOCKED |
+ }; |
+ |
+ State state_; |
+}; |
+ |
+void UnblockResourceThrottle(base::WeakPtr<CUResourceThrottle> rt) { |
+ BrowserThread::PostTask( |
+ BrowserThread::IO, |
+ FROM_HERE, |
+ base::Bind(&CUResourceThrottle::Unblock, rt)); |
+} |
+ |
+void UnblockandReapAllThrottles(CUResourceThrottle::WeakPtrVector* throttles) { |
+ CUResourceThrottle::WeakPtrVector::iterator it; |
+ for (it = throttles->begin(); it != throttles->end(); ++it) |
+ UnblockResourceThrottle(*it); |
+ throttles->clear(); |
+} |
+ |
+////////////////////////////////////////////////////////////////////////////// |
+// The one and only implementation of the ComponentUpdateService interface. In |
+// charge of running the show. The main method is ProcessPendingItems() which |
+// is called periodically to do the upgrades/installs or the update checks. |
+// An important consideration here is to be as "low impact" as we can to the |
+// rest of the browser, so even if we have many components registered and |
+// eligible for update, we only do one thing at a time with pauses in between |
+// the tasks. Also when we do network requests there is only one |url_fetcher_| |
+// in flight at a time. |
+// There are no locks in this code, the main structure |work_items_| is mutated |
+// only from the UI thread. The unpack and installation is done in the file |
+// thread and the network requests are done in the IO thread and in the file |
+// thread. |
+class CrxUpdateService : public ComponentUpdateService { |
+ public: |
+ explicit CrxUpdateService(ComponentUpdateService::Configurator* config); |
+ virtual ~CrxUpdateService(); |
+ |
+ // Overrides for ComponentUpdateService. |
+ virtual Status Start() OVERRIDE; |
+ virtual Status Stop() OVERRIDE; |
+ virtual Status RegisterComponent(const CrxComponent& component) OVERRIDE; |
+ virtual Status OnDemandUpdate(const std::string& component_id) OVERRIDE; |
+ virtual void GetComponents( |
+ std::vector<CrxComponentInfo>* components) OVERRIDE; |
+ virtual content::ResourceThrottle* GetOnDemandResourceThrottle( |
+ net::URLRequest* request, const std::string& crx_id) OVERRIDE; |
+ |
+ // Context for a update check url request. See DelegateWithContext above. |
+ struct UpdateContext { |
+ base::Time start; |
+ UpdateContext() : start(base::Time::Now()) {} |
+ }; |
+ |
+ // Context for a crx download url request. See DelegateWithContext above. |
+ struct CRXContext { |
+ ComponentInstaller* installer; |
+ std::vector<uint8> pk_hash; |
+ std::string id; |
+ std::string fingerprint; |
+ CRXContext() : installer(NULL) {} |
+ }; |
+ |
+ void OnURLFetchComplete(const net::URLFetcher* source, |
+ UpdateContext* context); |
+ |
+ void OnURLFetchComplete(const net::URLFetcher* source, |
+ CRXContext* context); |
+ |
+ private: |
+ enum ErrorCategory { |
+ kErrorNone = 0, |
+ kNetworkError, |
+ kUnpackError, |
+ kInstallError, |
+ }; |
+ |
+ enum StepDelayInterval { |
+ kStepDelayShort = 0, |
+ kStepDelayMedium, |
+ kStepDelayLong, |
+ }; |
+ |
+ void OnParseUpdateManifestSucceeded( |
+ const component_updater::UpdateManifest::Results& results); |
+ void OnParseUpdateManifestFailed(const std::string& error_message); |
+ |
+ Status OnDemandUpdateInternal(CrxUpdateItem* item); |
+ |
+ void ProcessPendingItems(); |
+ |
+ CrxUpdateItem* FindReadyComponent(); |
+ |
+ void UpdateComponent(CrxUpdateItem* workitem); |
+ |
+ void AddItemToUpdateCheck(CrxUpdateItem* item, |
+ std::string* update_check_items); |
+ |
+ void AddUpdateCheckItems(std::string* update_check_items); |
+ |
+ void DoUpdateCheck(const std::string& update_check_items); |
+ |
+ void ScheduleNextRun(StepDelayInterval step_delay); |
+ |
+ void ParseManifest(const std::string& xml); |
+ |
+ void Install(const CRXContext* context, const base::FilePath& crx_path); |
+ |
+ void DoneInstalling(const std::string& component_id, |
+ ComponentUnpacker::Error error, |
+ int extended_error); |
+ |
+ void ChangeItemState(CrxUpdateItem* item, CrxUpdateItem::Status to); |
+ |
+ size_t ChangeItemStatus(CrxUpdateItem::Status from, |
+ CrxUpdateItem::Status to); |
+ |
+ CrxUpdateItem* FindUpdateItemById(const std::string& id); |
+ |
+ void NotifyComponentObservers(ComponentObserver::Events event, |
+ int extra) const; |
+ |
+ bool HasOnDemandItems() const; |
+ |
+ void OnNewResourceThrottle(base::WeakPtr<CUResourceThrottle> rt, |
+ const std::string& crx_id); |
+ |
+ scoped_ptr<ComponentUpdateService::Configurator> config_; |
+ |
+ scoped_ptr<ComponentPatcher> component_patcher_; |
+ |
+ scoped_ptr<net::URLFetcher> url_fetcher_; |
+ |
+ scoped_ptr<component_updater::PingManager> ping_manager_; |
+ |
+ // A collection of every work item. |
+ typedef std::vector<CrxUpdateItem*> UpdateItems; |
+ UpdateItems work_items_; |
+ |
+ base::OneShotTimer<CrxUpdateService> timer_; |
+ |
+ scoped_refptr<base::SequencedTaskRunner> blocking_task_runner_; |
+ |
+ const Version chrome_version_; |
+ |
+ bool running_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(CrxUpdateService); |
+}; |
+ |
+////////////////////////////////////////////////////////////////////////////// |
+ |
+CrxUpdateService::CrxUpdateService(ComponentUpdateService::Configurator* config) |
+ : config_(config), |
+ component_patcher_(config->CreateComponentPatcher()), |
+ ping_manager_(new component_updater::PingManager( |
+ config->PingUrl(), |
+ config->RequestContext())), |
+ blocking_task_runner_(BrowserThread::GetBlockingPool()-> |
+ GetSequencedTaskRunnerWithShutdownBehavior( |
+ BrowserThread::GetBlockingPool()->GetSequenceToken(), |
+ base::SequencedWorkerPool::SKIP_ON_SHUTDOWN)), |
+ chrome_version_(chrome::VersionInfo().Version()), |
+ running_(false) { |
+} |
+ |
+CrxUpdateService::~CrxUpdateService() { |
+ // Because we are a singleton, at this point only the UI thread should be |
+ // alive, this simplifies the management of the work that could be in |
+ // flight in other threads. |
+ Stop(); |
+ STLDeleteElements(&work_items_); |
+} |
+ |
+ComponentUpdateService::Status CrxUpdateService::Start() { |
+ // Note that RegisterComponent will call Start() when the first |
+ // component is registered, so it can be called twice. This way |
+ // we avoid scheduling the timer if there is no work to do. |
+ running_ = true; |
+ if (work_items_.empty()) |
+ return kOk; |
+ |
+ NotifyComponentObservers(ComponentObserver::COMPONENT_UPDATER_STARTED, 0); |
+ |
+ timer_.Start(FROM_HERE, base::TimeDelta::FromSeconds(config_->InitialDelay()), |
+ this, &CrxUpdateService::ProcessPendingItems); |
+ return kOk; |
+} |
+ |
+// Stop the main check + update loop. In flight operations will be |
+// completed. |
+ComponentUpdateService::Status CrxUpdateService::Stop() { |
+ running_ = false; |
+ timer_.Stop(); |
+ return kOk; |
+} |
+ |
+bool CrxUpdateService::HasOnDemandItems() const { |
+ class Helper { |
+ public: |
+ static bool IsOnDemand(CrxUpdateItem* item) { |
+ return item->on_demand; |
+ } |
+ }; |
+ return std::find_if(work_items_.begin(), |
+ work_items_.end(), |
+ Helper::IsOnDemand) != work_items_.end(); |
+} |
+ |
+// This function sets the timer which will call ProcessPendingItems() or |
+// ProcessRequestedItem() if there is an on_demand item. There |
+// are three kinds of waits: |
+// - a short delay, when there is immediate work to be done. |
+// - a medium delay, when there are updates to be applied within the current |
+// update cycle, or there are components that are still unchecked. |
+// - a long delay when a full check/update cycle has completed for all |
+// components. |
+void CrxUpdateService::ScheduleNextRun(StepDelayInterval step_delay) { |
+ DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
+ DCHECK(url_fetcher_.get() == NULL); |
+ CHECK(!timer_.IsRunning()); |
+ // It could be the case that Stop() had been called while a url request |
+ // or unpacking was in flight, if so we arrive here but |running_| is |
+ // false. In that case do not loop again. |
+ if (!running_) |
+ return; |
+ |
+ // Keep the delay short if in the middle of an update (step_delay), |
+ // or there are new requested_work_items_ that have not been processed yet. |
+ int64 delay_seconds = 0; |
+ if (!HasOnDemandItems()) { |
+ switch (step_delay) { |
+ case kStepDelayShort: |
+ delay_seconds = config_->StepDelay(); |
+ break; |
+ case kStepDelayMedium: |
+ delay_seconds = config_->StepDelayMedium(); |
+ break; |
+ case kStepDelayLong: |
+ delay_seconds = config_->NextCheckDelay(); |
+ break; |
+ } |
+ } else { |
+ delay_seconds = config_->StepDelay(); |
+ } |
+ |
+ if (step_delay != kStepDelayShort) { |
+ NotifyComponentObservers(ComponentObserver::COMPONENT_UPDATER_SLEEPING, 0); |
+ |
+ // Zero is only used for unit tests. |
+ if (0 == delay_seconds) |
+ return; |
+ } |
+ |
+ timer_.Start(FROM_HERE, base::TimeDelta::FromSeconds(delay_seconds), |
+ this, &CrxUpdateService::ProcessPendingItems); |
+} |
+ |
+// Given a extension-like component id, find the associated component. |
+CrxUpdateItem* CrxUpdateService::FindUpdateItemById(const std::string& id) { |
+ DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
+ CrxUpdateItem::FindById finder(id); |
+ UpdateItems::iterator it = std::find_if(work_items_.begin(), |
+ work_items_.end(), |
+ finder); |
+ if (it == work_items_.end()) |
+ return NULL; |
+ return (*it); |
+} |
+ |
+// Changes a component's status, clearing on_demand and firing notifications as |
+// necessary. By convention, this is the only function that can change a |
+// CrxUpdateItem's |status|. |
+// TODO(waffles): Do we want to add DCHECKS for valid state transitions here? |
+void CrxUpdateService::ChangeItemState(CrxUpdateItem* item, |
+ CrxUpdateItem::Status to) { |
+ DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
+ if (to == CrxUpdateItem::kNoUpdate || |
+ to == CrxUpdateItem::kUpdated || |
+ to == CrxUpdateItem::kUpToDate) { |
+ item->on_demand = false; |
+ } |
+ |
+ item->status = to; |
+ |
+ ComponentObserver* observer = item->component.observer; |
+ if (observer) { |
+ switch (to) { |
+ case CrxUpdateItem::kCanUpdate: |
+ observer->OnEvent(ComponentObserver::COMPONENT_UPDATE_FOUND, 0); |
+ break; |
+ case CrxUpdateItem::kUpdatingDiff: |
+ case CrxUpdateItem::kUpdating: |
+ observer->OnEvent(ComponentObserver::COMPONENT_UPDATE_READY, 0); |
+ break; |
+ case CrxUpdateItem::kUpdated: |
+ observer->OnEvent(ComponentObserver::COMPONENT_UPDATED, 0); |
+ break; |
+ case CrxUpdateItem::kUpToDate: |
+ case CrxUpdateItem::kNoUpdate: |
+ observer->OnEvent(ComponentObserver::COMPONENT_NOT_UPDATED, 0); |
+ break; |
+ case CrxUpdateItem::kNew: |
+ case CrxUpdateItem::kChecking: |
+ case CrxUpdateItem::kDownloading: |
+ case CrxUpdateItem::kDownloadingDiff: |
+ case CrxUpdateItem::kLastStatus: |
+ // No notification for these states. |
+ break; |
+ } |
+ } |
+ |
+ // Free possible pending network requests. |
+ if ((to == CrxUpdateItem::kUpdated) || |
+ (to == CrxUpdateItem::kUpToDate) || |
+ (to == CrxUpdateItem::kNoUpdate)) { |
+ UnblockandReapAllThrottles(&item->throttles); |
+ } |
+} |
+ |
+// Changes all the components in |work_items_| that have |from| status to |
+// |to| status and returns how many have been changed. |
+size_t CrxUpdateService::ChangeItemStatus(CrxUpdateItem::Status from, |
+ CrxUpdateItem::Status to) { |
+ DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
+ size_t count = 0; |
+ for (UpdateItems::iterator it = work_items_.begin(); |
+ it != work_items_.end(); ++it) { |
+ CrxUpdateItem* item = *it; |
+ if (item->status != from) |
+ continue; |
+ ChangeItemState(item, to); |
+ ++count; |
+ } |
+ return count; |
+} |
+ |
+// Adds a component to be checked for upgrades. If the component exists it |
+// it will be replaced and the return code is kReplaced. |
+// |
+// TODO(cpu): Evaluate if we want to support un-registration. |
+ComponentUpdateService::Status CrxUpdateService::RegisterComponent( |
+ const CrxComponent& component) { |
+ DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
+ if (component.pk_hash.empty() || |
+ !component.version.IsValid() || |
+ !component.installer) |
+ return kError; |
+ |
+ std::string id = |
+ HexStringToID(StringToLowerASCII(base::HexEncode(&component.pk_hash[0], |
+ component.pk_hash.size()/2))); |
+ CrxUpdateItem* uit; |
+ uit = FindUpdateItemById(id); |
+ if (uit) { |
+ uit->component = component; |
+ return kReplaced; |
+ } |
+ |
+ uit = new CrxUpdateItem; |
+ uit->id.swap(id); |
+ uit->component = component; |
+ |
+ work_items_.push_back(uit); |
+ // If this is the first component registered we call Start to |
+ // schedule the first timer. |
+ if (running_ && (work_items_.size() == 1)) |
+ Start(); |
+ |
+ return kOk; |
+} |
+ |
+// Start the process of checking for an update, for a particular component |
+// that was previously registered. |
+// |component_id| is a value returned from GetCrxComponentID(). |
+ComponentUpdateService::Status CrxUpdateService::OnDemandUpdate( |
+ const std::string& component_id) { |
+ return OnDemandUpdateInternal(FindUpdateItemById(component_id)); |
+} |
+ |
+ComponentUpdateService::Status CrxUpdateService::OnDemandUpdateInternal( |
+ CrxUpdateItem* uit) { |
+ if (!uit) |
+ return kError; |
+ |
+ // Check if the request is too soon. |
+ base::TimeDelta delta = base::Time::Now() - uit->last_check; |
+ if (delta < base::TimeDelta::FromSeconds(config_->OnDemandDelay())) |
+ return kError; |
+ |
+ switch (uit->status) { |
+ // If the item is already in the process of being updated, there is |
+ // no point in this call, so return kInProgress. |
+ case CrxUpdateItem::kChecking: |
+ case CrxUpdateItem::kCanUpdate: |
+ case CrxUpdateItem::kDownloadingDiff: |
+ case CrxUpdateItem::kDownloading: |
+ case CrxUpdateItem::kUpdatingDiff: |
+ case CrxUpdateItem::kUpdating: |
+ return kInProgress; |
+ // Otherwise the item was already checked a while back (or it is new), |
+ // set its status to kNew to give it a slightly higher priority. |
+ case CrxUpdateItem::kNew: |
+ case CrxUpdateItem::kUpdated: |
+ case CrxUpdateItem::kUpToDate: |
+ case CrxUpdateItem::kNoUpdate: |
+ ChangeItemState(uit, CrxUpdateItem::kNew); |
+ uit->on_demand = true; |
+ break; |
+ case CrxUpdateItem::kLastStatus: |
+ NOTREACHED() << uit->status; |
+ } |
+ |
+ // In case the current delay is long, set the timer to a shorter value |
+ // to get the ball rolling. |
+ if (timer_.IsRunning()) { |
+ timer_.Stop(); |
+ timer_.Start(FROM_HERE, base::TimeDelta::FromSeconds(config_->StepDelay()), |
+ this, &CrxUpdateService::ProcessPendingItems); |
+ } |
+ |
+ return kOk; |
+} |
+ |
+void CrxUpdateService::GetComponents( |
+ std::vector<CrxComponentInfo>* components) { |
+ DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
+ for (UpdateItems::const_iterator it = work_items_.begin(); |
+ it != work_items_.end(); ++it) { |
+ const CrxUpdateItem* item = *it; |
+ CrxComponentInfo info; |
+ info.id = GetCrxComponentID(item->component); |
+ info.version = item->component.version.GetString(); |
+ info.name = item->component.name; |
+ components->push_back(info); |
+ } |
+} |
+ |
+// This is the main loop of the component updater. |
+void CrxUpdateService::ProcessPendingItems() { |
+ DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
+ CrxUpdateItem* ready_upgrade = FindReadyComponent(); |
+ if (ready_upgrade) { |
+ UpdateComponent(ready_upgrade); |
+ return; |
+ } |
+ std::string update_check_items; |
+ AddUpdateCheckItems(&update_check_items); |
+ if (!update_check_items.empty()) { |
+ DoUpdateCheck(update_check_items); |
+ return; |
+ } |
+ // No components to update. The next check will be after a long sleep. |
+ ScheduleNextRun(kStepDelayLong); |
+} |
+ |
+CrxUpdateItem* CrxUpdateService::FindReadyComponent() { |
+ class Helper { |
+ public: |
+ static bool IsReadyOnDemand(CrxUpdateItem* item) { |
+ return item->on_demand && IsReady(item); |
+ } |
+ static bool IsReady(CrxUpdateItem* item) { |
+ return item->status == CrxUpdateItem::kCanUpdate; |
+ } |
+ }; |
+ |
+ std::vector<CrxUpdateItem*>::iterator it = std::find_if( |
+ work_items_.begin(), work_items_.end(), Helper::IsReadyOnDemand); |
+ if (it != work_items_.end()) |
+ return *it; |
+ it = std::find_if(work_items_.begin(), work_items_.end(), Helper::IsReady); |
+ if (it != work_items_.end()) |
+ return *it; |
+ return NULL; |
+} |
+ |
+void CrxUpdateService::UpdateComponent(CrxUpdateItem* workitem) { |
+ CRXContext* context = new CRXContext; |
+ context->pk_hash = workitem->component.pk_hash; |
+ context->id = workitem->id; |
+ context->installer = workitem->component.installer; |
+ context->fingerprint = workitem->next_fp; |
+ GURL package_url; |
+ if (CanTryDiffUpdate(workitem, *config_)) { |
+ package_url = workitem->diff_crx_url; |
+ ChangeItemState(workitem, CrxUpdateItem::kDownloadingDiff); |
+ } else { |
+ package_url = workitem->crx_url; |
+ ChangeItemState(workitem, CrxUpdateItem::kDownloading); |
+ } |
+ url_fetcher_.reset(net::URLFetcher::Create( |
+ 0, package_url, net::URLFetcher::GET, |
+ MakeContextDelegate(this, context))); |
+ StartFetch(url_fetcher_.get(), |
+ config_->RequestContext(), |
+ true, |
+ blocking_task_runner_); |
+} |
+ |
+// Sets the state of the component to be checked for updates. After the |
+// function is called, the <app> element corresponding to the |item| parameter |
+// is appended to the |update_check_items|. |
+void CrxUpdateService::AddItemToUpdateCheck(CrxUpdateItem* item, |
+ std::string* update_check_items) { |
+ // The app element corresponding to an update items looks like this: |
+ // <app appid="hnimpnehoodheedghdeeijklkeaacbdc" |
+ // version="0.1.2.3" installsource="ondemand"> |
+ // <updatecheck /> |
+ // <packages> |
+ // <package fp="abcd" /> |
+ // </packages> |
+ // </app> |
+ std::string app_attributes; |
+ base::StringAppendF(&app_attributes, |
+ "appid=\"%s\" version=\"%s\"", |
+ item->id.c_str(), |
+ item->component.version.GetString().c_str()); |
+ if (item->on_demand) |
+ base::StringAppendF(&app_attributes, " installsource=\"ondemand\""); |
+ |
+ std::string app; |
+ if (item->component.fingerprint.empty()) |
+ base::StringAppendF(&app, |
+ "<app %s>" |
+ "<updatecheck />" |
+ "</app>", |
+ app_attributes.c_str()); |
+ else |
+ base::StringAppendF(&app, |
+ "<app %s>" |
+ "<updatecheck />" |
+ "<packages>" |
+ "<package fp=\"%s\"/>" |
+ "</packages>" |
+ "</app>", |
+ app_attributes.c_str(), |
+ item->component.fingerprint.c_str()); |
+ |
+ update_check_items->append(app); |
+ |
+ ChangeItemState(item, CrxUpdateItem::kChecking); |
+ item->last_check = base::Time::Now(); |
+ item->crx_url = GURL(); |
+ item->diff_crx_url = GURL(); |
+ item->previous_version = item->component.version; |
+ item->next_version = Version(); |
+ item->previous_fp = item->component.fingerprint; |
+ item->next_fp.clear(); |
+ item->diff_update_failed = false; |
+ item->error_category = 0; |
+ item->error_code = 0; |
+ item->extra_code1 = 0; |
+ item->diff_error_category = 0; |
+ item->diff_error_code = 0; |
+ item->diff_extra_code1 = 0; |
+} |
+ |
+// Builds the sequence of <app> elements in the update check and returns it |
+// in the |update_check_items| parameter. |
+void CrxUpdateService::AddUpdateCheckItems(std::string* update_check_items) { |
+ // Given that our |work_items_| list is expected to contain relatively few |
+ // items, we simply loop several times. |
+ for (UpdateItems::const_iterator it = work_items_.begin(); |
+ it != work_items_.end(); ++it) { |
+ CrxUpdateItem* item = *it; |
+ if (item->status != CrxUpdateItem::kNew) |
+ continue; |
+ AddItemToUpdateCheck(item, update_check_items); |
+ } |
+ |
+ // Next we can go back to components we already checked, here |
+ // we can also batch them in a single url request, as long as |
+ // we have not checked them recently. |
+ const base::TimeDelta min_delta_time = |
+ base::TimeDelta::FromSeconds(config_->MinimumReCheckWait()); |
+ |
+ for (UpdateItems::const_iterator it = work_items_.begin(); |
+ it != work_items_.end(); ++it) { |
+ CrxUpdateItem* item = *it; |
+ if ((item->status != CrxUpdateItem::kNoUpdate) && |
+ (item->status != CrxUpdateItem::kUpToDate)) |
+ continue; |
+ base::TimeDelta delta = base::Time::Now() - item->last_check; |
+ if (delta < min_delta_time) |
+ continue; |
+ AddItemToUpdateCheck(item, update_check_items); |
+ } |
+ |
+ // Finally, we check components that we already updated as long as |
+ // we have not checked them recently. |
+ for (UpdateItems::const_iterator it = work_items_.begin(); |
+ it != work_items_.end(); ++it) { |
+ CrxUpdateItem* item = *it; |
+ if (item->status != CrxUpdateItem::kUpdated) |
+ continue; |
+ base::TimeDelta delta = base::Time::Now() - item->last_check; |
+ if (delta < min_delta_time) |
+ continue; |
+ AddItemToUpdateCheck(item, update_check_items); |
+ } |
+} |
+ |
+// Sends an update request. The |update_check_items| parameter |
+// contains the sequence of <app> xml elements of the update check. |
+void CrxUpdateService::DoUpdateCheck(const std::string& update_check_items) { |
+ using component_updater::BuildProtocolRequest; |
+ url_fetcher_.reset(component_updater::SendProtocolRequest( |
+ config_->UpdateUrl(), |
+ BuildProtocolRequest(update_check_items), |
+ MakeContextDelegate(this, new UpdateContext()), |
+ config_->RequestContext())); |
+} |
+ |
+// Called when we got a response from the update server. It consists of an xml |
+// document following the omaha update scheme. |
+void CrxUpdateService::OnURLFetchComplete(const net::URLFetcher* source, |
+ UpdateContext* context) { |
+ DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
+ if (FetchSuccess(*source)) { |
+ std::string xml; |
+ source->GetResponseAsString(&xml); |
+ url_fetcher_.reset(); |
+ ParseManifest(xml); |
+ } else { |
+ url_fetcher_.reset(); |
+ CrxUpdateService::OnParseUpdateManifestFailed("network error"); |
+ } |
+ delete context; |
+} |
+ |
+void CrxUpdateService::ParseManifest(const std::string& xml) { |
+ DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
+ component_updater::UpdateManifest manifest; |
+ if (manifest.Parse(xml)) |
+ CrxUpdateService::OnParseUpdateManifestSucceeded(manifest.results()); |
+ else |
+ CrxUpdateService::OnParseUpdateManifestFailed(manifest.errors()); |
+} |
+ |
+// A valid Omaha update check has arrived, from only the list of components that |
+// we are currently upgrading we check for a match in which the server side |
+// version is newer, if so we queue them for an upgrade. The next time we call |
+// ProcessPendingItems() one of them will be drafted for the upgrade process. |
+void CrxUpdateService::OnParseUpdateManifestSucceeded( |
+ const component_updater::UpdateManifest::Results& results) { |
+ size_t num_updates_pending = 0; |
+ DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
+ std::vector<component_updater::UpdateManifest::Result>::const_iterator it; |
+ for (it = results.list.begin(); it != results.list.end(); ++it) { |
+ CrxUpdateItem* crx = FindUpdateItemById(it->extension_id); |
+ if (!crx) |
+ continue; |
+ |
+ if (crx->status != CrxUpdateItem::kChecking) { |
+ NOTREACHED(); |
+ continue; // Not updating this component now. |
+ } |
+ |
+ if (it->manifest.version.empty()) { |
+ // No version means no update available. |
+ ChangeItemState(crx, CrxUpdateItem::kNoUpdate); |
+ continue; |
+ } |
+ |
+ if (!IsVersionNewer(crx->component.version, it->manifest.version)) { |
+ // The component is up to date. |
+ ChangeItemState(crx, CrxUpdateItem::kUpToDate); |
+ continue; |
+ } |
+ |
+ if (!it->manifest.browser_min_version.empty()) { |
+ if (IsVersionNewer(chrome_version_, it->manifest.browser_min_version)) { |
+ // The component is not compatible with this Chrome version. |
+ ChangeItemState(crx, CrxUpdateItem::kNoUpdate); |
+ continue; |
+ } |
+ } |
+ |
+ if (it->manifest.packages.size() != 1) { |
+ // Assume one and only one package per component. |
+ ChangeItemState(crx, CrxUpdateItem::kNoUpdate); |
+ continue; |
+ } |
+ |
+ // Parse the members of the result and queue an upgrade for this component. |
+ crx->next_version = Version(it->manifest.version); |
+ |
+ typedef component_updater:: |
+ UpdateManifest::Result::Manifest::Package Package; |
+ const Package& package(it->manifest.packages[0]); |
+ crx->next_fp = package.fingerprint; |
+ |
+ // Select the first url from the list of urls until support for |
+ // fall back urls is implemented. |
+ if (!it->crx_urls.empty()) |
+ crx->crx_url = it->crx_urls[0].Resolve(package.name); |
+ if (!it->crx_diffurls.empty()) |
+ crx->diff_crx_url = it->crx_diffurls[0].Resolve(package.namediff); |
+ |
+ ChangeItemState(crx, CrxUpdateItem::kCanUpdate); |
+ ++num_updates_pending; |
+ } |
+ |
+ // All components that are not included in the update response are |
+ // considered up to date. |
+ ChangeItemStatus(CrxUpdateItem::kChecking, CrxUpdateItem::kUpToDate); |
+ |
+ // If there are updates pending we do a short wait, otherwise we take |
+ // a longer delay until we check the components again. |
+ ScheduleNextRun(num_updates_pending > 0 ? kStepDelayShort : kStepDelayMedium); |
+} |
+ |
+void CrxUpdateService::OnParseUpdateManifestFailed( |
+ const std::string& error_message) { |
+ DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
+ size_t count = ChangeItemStatus(CrxUpdateItem::kChecking, |
+ CrxUpdateItem::kNoUpdate); |
+ DCHECK_GT(count, 0ul); |
+ ScheduleNextRun(kStepDelayLong); |
+} |
+ |
+// Called when the CRX package has been downloaded to a temporary location. |
+// Here we fire the notifications and schedule the component-specific installer |
+// to be called in the file thread. |
+void CrxUpdateService::OnURLFetchComplete(const net::URLFetcher* source, |
+ CRXContext* context) { |
+ DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
+ |
+ scoped_ptr<CRXContext> crx_context(context); |
+ |
+ CrxUpdateItem* crx = FindUpdateItemById(crx_context->id); |
+ DCHECK(crx->status == CrxUpdateItem::kDownloadingDiff || |
+ crx->status == CrxUpdateItem::kDownloading); |
+ |
+ if (!FetchSuccess(*source)) { |
+ if (crx->status == CrxUpdateItem::kDownloadingDiff) { |
+ crx->diff_error_category = kNetworkError; |
+ crx->diff_error_code = GetFetchError(*source); |
+ crx->diff_update_failed = true; |
+ size_t count = ChangeItemStatus(CrxUpdateItem::kDownloadingDiff, |
+ CrxUpdateItem::kCanUpdate); |
+ DCHECK_EQ(count, 1ul); |
+ url_fetcher_.reset(); |
+ |
+ ScheduleNextRun(kStepDelayShort); |
+ return; |
+ } |
+ crx->error_category = kNetworkError; |
+ crx->error_code = GetFetchError(*source); |
+ size_t count = ChangeItemStatus(CrxUpdateItem::kDownloading, |
+ CrxUpdateItem::kNoUpdate); |
+ DCHECK_EQ(count, 1ul); |
+ url_fetcher_.reset(); |
+ |
+ // At this point, since both the differential and the full downloads failed, |
+ // the update for this component has finished with an error. |
+ ping_manager_->OnUpdateComplete(crx); |
+ |
+ // Move on to the next update, if there is one available. |
+ ScheduleNextRun(kStepDelayMedium); |
+ } else { |
+ base::FilePath temp_crx_path; |
+ CHECK(source->GetResponseAsFilePath(true, &temp_crx_path)); |
+ |
+ size_t count = 0; |
+ if (crx->status == CrxUpdateItem::kDownloadingDiff) { |
+ count = ChangeItemStatus(CrxUpdateItem::kDownloadingDiff, |
+ CrxUpdateItem::kUpdatingDiff); |
+ } else { |
+ count = ChangeItemStatus(CrxUpdateItem::kDownloading, |
+ CrxUpdateItem::kUpdating); |
+ } |
+ DCHECK_EQ(count, 1ul); |
+ |
+ url_fetcher_.reset(); |
+ |
+ // Why unretained? See comment at top of file. |
+ blocking_task_runner_->PostDelayedTask( |
+ FROM_HERE, |
+ base::Bind(&CrxUpdateService::Install, |
+ base::Unretained(this), |
+ crx_context.release(), |
+ temp_crx_path), |
+ base::TimeDelta::FromMilliseconds(config_->StepDelay())); |
+ } |
+} |
+ |
+// Install consists of digital signature verification, unpacking and then |
+// calling the component specific installer. All that is handled by the |
+// |unpacker|. If there is an error this function is in charge of deleting |
+// the files created. |
+void CrxUpdateService::Install(const CRXContext* context, |
+ const base::FilePath& crx_path) { |
+ // This function owns the file at |crx_path| and the |context| object. |
+ ComponentUnpacker unpacker(context->pk_hash, |
+ crx_path, |
+ context->fingerprint, |
+ component_patcher_.get(), |
+ context->installer); |
+ if (!base::DeleteFile(crx_path, false)) |
+ NOTREACHED() << crx_path.value(); |
+ // Why unretained? See comment at top of file. |
+ BrowserThread::PostDelayedTask( |
+ BrowserThread::UI, |
+ FROM_HERE, |
+ base::Bind(&CrxUpdateService::DoneInstalling, base::Unretained(this), |
+ context->id, unpacker.error(), unpacker.extended_error()), |
+ base::TimeDelta::FromMilliseconds(config_->StepDelay())); |
+ delete context; |
+} |
+ |
+// Installation has been completed. Adjust the component status and |
+// schedule the next check. Schedule a short delay before trying the full |
+// update when the differential update failed. |
+void CrxUpdateService::DoneInstalling(const std::string& component_id, |
+ ComponentUnpacker::Error error, |
+ int extra_code) { |
+ DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
+ |
+ ErrorCategory error_category = kErrorNone; |
+ switch (error) { |
+ case ComponentUnpacker::kNone: |
+ break; |
+ case ComponentUnpacker::kInstallerError: |
+ error_category = kInstallError; |
+ break; |
+ default: |
+ error_category = kUnpackError; |
+ break; |
+ } |
+ |
+ const bool is_success = error == ComponentUnpacker::kNone; |
+ |
+ CrxUpdateItem* item = FindUpdateItemById(component_id); |
+ if (item->status == CrxUpdateItem::kUpdatingDiff && !is_success) { |
+ item->diff_error_category = error_category; |
+ item->diff_error_code = error; |
+ item->diff_extra_code1 = extra_code; |
+ item->diff_update_failed = true; |
+ size_t count = ChangeItemStatus(CrxUpdateItem::kUpdatingDiff, |
+ CrxUpdateItem::kCanUpdate); |
+ DCHECK_EQ(count, 1ul); |
+ ScheduleNextRun(kStepDelayShort); |
+ return; |
+ } |
+ |
+ if (is_success) { |
+ ChangeItemState(item, CrxUpdateItem::kUpdated); |
+ item->component.version = item->next_version; |
+ item->component.fingerprint = item->next_fp; |
+ } else { |
+ ChangeItemState(item, CrxUpdateItem::kNoUpdate); |
+ item->error_category = error_category; |
+ item->error_code = error; |
+ item->extra_code1 = extra_code; |
+ } |
+ |
+ ping_manager_->OnUpdateComplete(item); |
+ |
+ // Move on to the next update, if there is one available. |
+ ScheduleNextRun(kStepDelayMedium); |
+} |
+ |
+void CrxUpdateService::NotifyComponentObservers( |
+ ComponentObserver::Events event, int extra) const { |
+ for (UpdateItems::const_iterator it = work_items_.begin(); |
+ it != work_items_.end(); ++it) { |
+ ComponentObserver* observer = (*it)->component.observer; |
+ if (observer) |
+ observer->OnEvent(event, 0); |
+ } |
+} |
+ |
+content::ResourceThrottle* CrxUpdateService::GetOnDemandResourceThrottle( |
+ net::URLRequest* request, const std::string& crx_id) { |
+ // We give the raw pointer to the caller, who will delete it at will |
+ // and we keep for ourselves a weak pointer to it so we can post tasks |
+ // from the UI thread without having to track lifetime directly. |
+ CUResourceThrottle* rt = new CUResourceThrottle(request); |
+ BrowserThread::PostTask( |
+ BrowserThread::UI, |
+ FROM_HERE, |
+ base::Bind(&CrxUpdateService::OnNewResourceThrottle, |
+ base::Unretained(this), |
+ rt->AsWeakPtr(), |
+ crx_id)); |
+ return rt; |
+} |
+ |
+void CrxUpdateService::OnNewResourceThrottle( |
+ base::WeakPtr<CUResourceThrottle> rt, const std::string& crx_id) { |
+ DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
+ // Check if we can on-demand update, else unblock the request anyway. |
+ CrxUpdateItem* item = FindUpdateItemById(crx_id); |
+ Status status = OnDemandUpdateInternal(item); |
+ if (status == kOk || status == kInProgress) { |
+ item->throttles.push_back(rt); |
+ return; |
+ } |
+ UnblockResourceThrottle(rt); |
+} |
+ |
+/////////////////////////////////////////////////////////////////////////////// |
+ |
+CUResourceThrottle::CUResourceThrottle(const net::URLRequest* request) |
+ : state_(NEW) { |
+ DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
+} |
+ |
+CUResourceThrottle::~CUResourceThrottle() { |
+ DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
+} |
+ |
+void CUResourceThrottle::WillStartRequest(bool* defer) { |
+ if (state_ != UNBLOCKED) { |
+ state_ = BLOCKED; |
+ *defer = true; |
+ } else { |
+ *defer = false; |
+ } |
+} |
+ |
+void CUResourceThrottle::WillRedirectRequest(const GURL& new_url, bool* defer) { |
+ WillStartRequest(defer); |
+} |
+ |
+void CUResourceThrottle::Unblock() { |
+ DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
+ if (state_ == BLOCKED) |
+ controller()->Resume(); |
+ state_ = UNBLOCKED; |
+} |
+ |
+// The component update factory. Using the component updater as a singleton |
+// is the job of the browser process. |
+ComponentUpdateService* ComponentUpdateServiceFactory( |
+ ComponentUpdateService::Configurator* config) { |
+ DCHECK(config); |
+ return new CrxUpdateService(config); |
+} |
+ |