1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61
   62
   63
   64
   65
   66
   67
   68
   69
   70
   71
   72
   73
   74
   75
   76
   77
   78
   79
   80
   81
   82
   83
   84
   85
   86
   87
   88
   89

content / browser / indexed_db / indexed_db_control_wrapper.cc [blame]

// 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 "content/browser/indexed_db/indexed_db_control_wrapper.h"

#include <ostream>
#include <utility>

#include "base/check.h"
#include "base/functional/bind.h"
#include "components/services/storage/public/cpp/buckets/bucket_locator.h"
#include "components/services/storage/public/mojom/storage_policy_update.mojom.h"
#include "content/browser/indexed_db/indexed_db_context_impl.h"
#include "mojo/public/cpp/bindings/pending_receiver.h"
#include "mojo/public/cpp/bindings/pending_remote.h"
#include "third_party/blink/public/common/storage_key/storage_key.h"

namespace content::indexed_db {

IndexedDBControlWrapper::IndexedDBControlWrapper(
    const base::FilePath& data_path,
    scoped_refptr<storage::SpecialStoragePolicy> special_storage_policy,
    scoped_refptr<storage::QuotaManagerProxy> quota_manager_proxy,
    mojo::PendingRemote<storage::mojom::BlobStorageContext>
        blob_storage_context,
    mojo::PendingRemote<storage::mojom::FileSystemAccessContext>
        file_system_access_context,
    scoped_refptr<base::SequencedTaskRunner> io_task_runner) {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  context_ = std::make_unique<IndexedDBContextImpl>(
      data_path, std::move(quota_manager_proxy),
      std::move(blob_storage_context), std::move(file_system_access_context),
      /*custom_task_runner=*/nullptr);

  if (special_storage_policy) {
    storage_policy_observer_.emplace(
        base::BindRepeating(
            &IndexedDBControlWrapper::OnSpecialStoragePolicyUpdated,
            base::Unretained(this)),
        io_task_runner, std::move(special_storage_policy));
  }
}

IndexedDBControlWrapper::~IndexedDBControlWrapper() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);

  IndexedDBContextImpl::Shutdown(std::move(context_));
}

void IndexedDBControlWrapper::BindIndexedDB(
    const storage::BucketLocator& bucket_locator,
    const storage::BucketClientInfo& client_info,
    mojo::PendingRemote<storage::mojom::IndexedDBClientStateChecker>
        client_state_checker_remote,
    mojo::PendingReceiver<blink::mojom::IDBFactory> receiver) {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  if (storage_policy_observer_) {
    // TODO(crbug.com/40177656): Pass the real StorageKey once
    // StoragePolicyObserver is migrated.
    storage_policy_observer_->StartTrackingOrigin(
        bucket_locator.storage_key.origin());
  }
  GetIndexedDBControl().BindIndexedDB(bucket_locator, client_info,
                                      std::move(client_state_checker_remote),
                                      std::move(receiver));
}

storage::mojom::IndexedDBControl&
IndexedDBControlWrapper::GetIndexedDBControl() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  DCHECK(
      !(indexed_db_control_.is_bound() && !indexed_db_control_.is_connected()))
      << "Rebinding is not supported yet.";

  if (!indexed_db_control_.is_bound()) {
    context_->BindControl(indexed_db_control_.BindNewPipeAndPassReceiver());
  }

  return *indexed_db_control_;
}

void IndexedDBControlWrapper::OnSpecialStoragePolicyUpdated(
    std::vector<storage::mojom::StoragePolicyUpdatePtr> policy_updates) {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  GetIndexedDBControl().ApplyPolicyUpdates(std::move(policy_updates));
}

}  // namespace content::indexed_db