chrome/browser/policy/messaging_layer/upload/upload_client.cc

// Copyright 2020 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/policy/messaging_layer/upload/upload_client.h"

#include "base/memory/ptr_util.h"
#include "base/task/sequenced_task_runner.h"
#include "build/build_config.h"
#include "chrome/browser/policy/messaging_layer/upload/file_upload_impl.h"
#include "chrome/browser/policy/messaging_layer/upload/record_handler_impl.h"
#include "chrome/browser/policy/messaging_layer/upload/server_uploader.h"
#include "chrome/browser/policy/messaging_layer/util/upload_declarations.h"
#include "components/reporting/client/report_queue_provider.h"
#include "components/reporting/resources/resource_manager.h"
#include "components/reporting/util/status.h"
#include "components/reporting/util/status_macros.h"
#include "components/reporting/util/statusor.h"
#include "components/user_manager/user_manager.h"
#include "content/public/browser/browser_context.h"
#include "content/public/browser/browser_thread.h"
#include "content/public/browser/storage_partition.h"

namespace reporting {

namespace {

FileUploadJob::Delegate::SmartPtr CreateFileUploadDelegate() {
#if BUILDFLAG(IS_CHROMEOS)
  return FileUploadJob::Delegate::SmartPtr(
      new FileUploadDelegate(),
      base::OnTaskRunnerDeleter(::content::GetUIThreadTaskRunner({})));
#else   // !BUILDFLAG(IS_CHROMEOS)
  // No file uploads for all other configurations.
  return FileUploadJob::Delegate::SmartPtr(
      nullptr, base::OnTaskRunnerDeleter(
                   base::SequencedTaskRunner::GetCurrentDefault()));
#endif  // BUILDFLAG(IS_CHROMEOS)
}
}  // namespace

// static
void UploadClient::Create(CreatedCallback created_cb) {
  std::move(created_cb).Run(base::WrapUnique(new UploadClient()));
}

void UploadClient::EnqueueUpload(
    bool need_encryption_key,
    int config_file_version,
    std::vector<EncryptedRecord> records,
    ScopedReservation scoped_reservation,
    UploadEnqueuedCallback enqueued_cb,
    ReportSuccessfulUploadCallback report_upload_success_cb,
    EncryptionKeyAttachedCallback encryption_key_attached_cb,
    ConfigFileAttachedCallback config_file_attached_cb) {
  if (records.empty() && !need_encryption_key) {
    // Do nothing, just return status.
    std::move(enqueued_cb)
        .Run(base::unexpected(
            Status(error::INVALID_ARGUMENT, "No action required")));
    return;
  }

  Start<ServerUploader>(
      need_encryption_key, config_file_version, std::move(records),
      std::move(scoped_reservation),
      std::make_unique<RecordHandlerImpl>(
          sequenced_task_runner_,
          base::BindRepeating(&CreateFileUploadDelegate)),
      std::move(enqueued_cb), std::move(report_upload_success_cb),
      std::move(encryption_key_attached_cb), std::move(config_file_attached_cb),
      base::DoNothing(), sequenced_task_runner_);
}

UploadClient::UploadClient()
    : sequenced_task_runner_(content::GetUIThreadTaskRunner()) {}

UploadClient::~UploadClient() = default;

}  // namespace reporting