[AUTO][FILECONTROL] - version 140.0.7339.128

140.0.7339.81

140.0.7339.128

// Copyright 2017 The Chromium Authors

// Use of this source code is governed by a BSD-style license that can be

// found in the LICENSE file.

#include "chrome/browser/download/background_download_service_factory.h"

#include <memory>

#include <utility>

#include "base/feature_list.h"

#include "base/files/file_path.h"

#include "base/memory/raw_ptr.h"

#include "base/no_destructor.h"

#include "base/task/sequenced_task_runner.h"

#include "base/task/single_thread_task_runner.h"

#include "base/task/task_traits.h"

#include "base/task/thread_pool.h"

#include "build/build_config.h"

#include "chrome/browser/download/deferred_client_wrapper.h"

#include "chrome/browser/download/download_manager_utils.h"

#include "chrome/browser/download/simple_download_manager_coordinator_factory.h"

#include "chrome/browser/optimization_guide/prediction/prediction_model_download_client.h"

#include "chrome/browser/profiles/incognito_helpers.h"

#include "chrome/browser/profiles/profile.h"

#include "chrome/browser/profiles/profile_key.h"

#include "chrome/browser/transition_manager/full_browser_transition_manager.h"

#include "chrome/common/chrome_constants.h"

#include "components/background_fetch/download_client.h"

#include "components/download/content/factory/download_service_factory_helper.h"

#include "components/download/content/factory/navigation_monitor_factory.h"

#include "components/download/public/background_service/background_download_service.h"

#include "components/download/public/background_service/basic_task_scheduler.h"

#include "components/download/public/background_service/blob_context_getter_factory.h"

#include "components/download/public/background_service/clients.h"

#include "components/download/public/background_service/features.h"

#include "components/download/public/background_service/url_loader_factory_getter.h"

#include "components/download/public/common/simple_download_manager_coordinator.h"

#include "components/keyed_service/core/simple_dependency_manager.h"

#include "components/leveldb_proto/public/proto_database_provider.h"

#include "components/optimization_guide/core/optimization_guide_features.h"

#include "content/public/browser/browser_context.h"

#include "content/public/browser/browser_task_traits.h"

#include "content/public/browser/browser_thread.h"

#include "content/public/browser/download_manager.h"

#include "content/public/browser/network_service_instance.h"

#include "content/public/browser/storage_partition.h"

#if BUILDFLAG(IS_ANDROID)

#include "chrome/browser/download/android/service/download_task_scheduler.h"

#endif

#if BUILDFLAG(IS_CHROMEOS)

#include "chrome/browser/ash/plugin_vm/plugin_vm_image_download_client.h"

#endif

namespace {

std::unique_ptr<download::Client> CreateBackgroundFetchDownloadClient(

    Profile* profile) {

  return std::make_unique<background_fetch::DownloadClient>(profile);

}

#if BUILDFLAG(IS_CHROMEOS)

std::unique_ptr<download::Client> CreatePluginVmImageDownloadClient(

    Profile* profile) {

  return std::make_unique<plugin_vm::PluginVmImageDownloadClient>(profile);

}

#endif  // BUILDFLAG(IS_CHROMEOS)

std::unique_ptr<download::Client>

CreateOptimizationGuidePredictionModelDownloadClient(Profile* profile) {

  return std::make_unique<optimization_guide::PredictionModelDownloadClient>(

      profile);

}

// Called on profile created to retrieve the BlobStorageContextGetter.

void DownloadOnProfileCreated(download::BlobContextGetterCallback callback,

                              Profile* profile) {

  auto blob_context_getter = profile->GetBlobStorageContext();

  DCHECK(callback);

  std::move(callback).Run(blob_context_getter);

}

// Provides BlobContextGetter from Chrome asynchronously.

class DownloadBlobContextGetterFactory

    : public download::BlobContextGetterFactory {

 public:

  explicit DownloadBlobContextGetterFactory(SimpleFactoryKey* key) : key_(key) {

    DCHECK(key_);

  }

  DownloadBlobContextGetterFactory(const DownloadBlobContextGetterFactory&) =

      delete;

  DownloadBlobContextGetterFactory& operator=(

      const DownloadBlobContextGetterFactory&) = delete;

  ~DownloadBlobContextGetterFactory() override = default;

 private:

  // download::BlobContextGetterFactory implementation.

  void RetrieveBlobContextGetter(

      download::BlobContextGetterCallback callback) override {

    FullBrowserTransitionManager::Get()->RegisterCallbackOnProfileCreation(

        key_, base::BindOnce(&DownloadOnProfileCreated, std::move(callback)));

  }

  raw_ptr<SimpleFactoryKey> key_;

};

void OnProfileCreated(download::URLLoaderFactoryGetterCallback callback,

                      Profile* profile) {

  DCHECK(callback);

  std::move(callback).Run(profile->GetURLLoaderFactory());

}

class URLLoaderFactoryGetter : public download::URLLoaderFactoryGetter {

 public:

  explicit URLLoaderFactoryGetter(SimpleFactoryKey* key) : key_(key) {

    DCHECK(key_);

  }

  URLLoaderFactoryGetter(const URLLoaderFactoryGetter&) = delete;

  URLLoaderFactoryGetter& operator=(const URLLoaderFactoryGetter&) = delete;

  ~URLLoaderFactoryGetter() override = default;

 private:

  void RetrieveURLLoaderFactory(

      download::URLLoaderFactoryGetterCallback callback) override {

    FullBrowserTransitionManager::Get()->RegisterCallbackOnProfileCreation(

        key_, base::BindOnce(&OnProfileCreated, std::move(callback)));

  }

  raw_ptr<SimpleFactoryKey> key_;

};

}  // namespace

// static

BackgroundDownloadServiceFactory*

BackgroundDownloadServiceFactory::GetInstance() {

  static base::NoDestructor<BackgroundDownloadServiceFactory> instance;

  return instance.get();

}

// static

download::BackgroundDownloadService*

BackgroundDownloadServiceFactory::GetForKey(SimpleFactoryKey* key) {

  return static_cast<download::BackgroundDownloadService*>(

      GetInstance()->GetServiceForKey(key, true));

}

BackgroundDownloadServiceFactory::BackgroundDownloadServiceFactory()

    : SimpleKeyedServiceFactory("download::BackgroundDownloadService",

                                SimpleDependencyManager::GetInstance()) {

  DependsOn(SimpleDownloadManagerCoordinatorFactory::GetInstance());

  DependsOn(download::NavigationMonitorFactory::GetInstance());

}

BackgroundDownloadServiceFactory::~BackgroundDownloadServiceFactory() = default;

std::unique_ptr<KeyedService>

BackgroundDownloadServiceFactory::BuildServiceInstanceFor(

    SimpleFactoryKey* key) const {

  auto clients = std::make_unique<download::DownloadClientMap>();

  ProfileKey* profile_key = ProfileKey::FromSimpleFactoryKey(key);

  clients->insert(std::make_pair(

      download::DownloadClient::BACKGROUND_FETCH,

      std::make_unique<download::DeferredClientWrapper>(

          base::BindOnce(&CreateBackgroundFetchDownloadClient), key)));

#if BUILDFLAG(IS_CHROMEOS)

  if (!key->IsOffTheRecord()) {

    clients->insert(std::make_pair(

        download::DownloadClient::PLUGIN_VM_IMAGE,

        std::make_unique<download::DeferredClientWrapper>(

            base::BindOnce(&CreatePluginVmImageDownloadClient), key)));

  }

#endif  // BUILDFLAG(IS_CHROMEOS)

  if (!key->IsOffTheRecord()) {

    clients->insert(std::make_pair(

        download::DownloadClient::OPTIMIZATION_GUIDE_PREDICTION_MODELS,

        std::make_unique<download::DeferredClientWrapper>(

            base::BindOnce(

                &CreateOptimizationGuidePredictionModelDownloadClient),

            key)));

  }

  // Build in memory download service for incognito profile.

  if (key->IsOffTheRecord()) {

    auto blob_context_getter_factory =

        std::make_unique<DownloadBlobContextGetterFactory>(key);

    scoped_refptr<base::SingleThreadTaskRunner> io_task_runner =

        content::GetIOThreadTaskRunner({});

    return download::BuildInMemoryDownloadService(

        key, std::move(clients), content::GetNetworkConnectionTracker(),

        base::FilePath(), std::move(blob_context_getter_factory),

        io_task_runner, std::make_unique<::URLLoaderFactoryGetter>(key));

  } else {

    // Build download service for normal profile.

    base::FilePath storage_dir;

    if (!key->IsOffTheRecord() && !key->GetPath().empty()) {

      storage_dir =

          key->GetPath().Append(chrome::kDownloadServiceStorageDirname);

    }

    scoped_refptr<base::SequencedTaskRunner> background_task_runner =

        base::ThreadPool::CreateSequencedTaskRunner(

            {base::MayBlock(), base::TaskPriority::BEST_EFFORT});

    std::unique_ptr<download::TaskScheduler> task_scheduler;

#if BUILDFLAG(IS_ANDROID)

    task_scheduler =

        std::make_unique<download::android::DownloadTaskScheduler>();

#else

    task_scheduler =

        std::make_unique<download::BasicTaskScheduler>(base::BindRepeating(

            [](SimpleFactoryKey* key) {

              return BackgroundDownloadServiceFactory::GetForKey(key);

            },

            key));

#endif

    // Some tests doesn't initialize DownloadManager when profile is created,

    // and cause the download service to fail. Call

    // InitializeSimpleDownloadManager() to initialize the DownloadManager

    // whenever profile becomes available.

    DownloadManagerUtils::InitializeSimpleDownloadManager(profile_key);

    leveldb_proto::ProtoDatabaseProvider* proto_db_provider =

        profile_key->GetProtoDatabaseProvider();

    return download::BuildDownloadService(

        key, std::move(clients), content::GetNetworkConnectionTracker(),

        storage_dir, SimpleDownloadManagerCoordinatorFactory::GetForKey(key),

        proto_db_provider, background_task_runner, std::move(task_scheduler));

  }

}

SimpleFactoryKey* BackgroundDownloadServiceFactory::GetKeyToUse(

    SimpleFactoryKey* key) const {

  return key;

}

      status: "stable",

    },

    {

      // TODO(crbug.com/440374239): Disabled due to https://crbug.com/440374239.

      name: "ClipElementVisibleBoundsInLocalRoot",

      status: "stable",

    },

    {

      name: "ClipPathNestedRasterOptimization",