Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(4452)

Unified Diff: chrome/browser/component_updater/component_updater_service.cc

Issue 74893002: Changed the update protocol for component updater from v2 to v3. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Addressed feedback from cdn@ Created 7 years, 1 month ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
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);
+}
+

Powered by Google App Engine
This is Rietveld 408576698