Avi Drissman | 8ba1bad | 2022-09-13 19:22:36 | [diff] [blame] | 1 | // Copyright 2019 The Chromium Authors |
Ken Rockot | 1962d93 | 2019-08-21 01:47:28 | [diff] [blame] | 2 | // Use of this source code is governed by a BSD-style license that can be |
| 3 | // found in the LICENSE file. |
| 4 | |
| 5 | #include "components/services/storage/partition_impl.h" |
| 6 | |
Ken rockot | 6e4416c3 | 2020-11-23 20:10:08 | [diff] [blame] | 7 | #include <memory> |
Ken Rockot | 1962d93 | 2019-08-21 01:47:28 | [diff] [blame] | 8 | #include <utility> |
| 9 | |
Avi Drissman | 12be031 | 2023-01-11 09:16:09 | [diff] [blame] | 10 | #include "base/functional/bind.h" |
Ken rockot | 6e4416c3 | 2020-11-23 20:10:08 | [diff] [blame] | 11 | #include "base/synchronization/waitable_event.h" |
Patrick Monette | 643cdf6 | 2021-10-15 19:13:42 | [diff] [blame] | 12 | #include "base/task/sequenced_task_runner.h" |
Gabriel Charette | dd8d5985e | 2020-02-26 18:38:35 | [diff] [blame] | 13 | #include "base/task/thread_pool.h" |
Ken Rockot | af15f1d | 2020-02-14 23:08:11 | [diff] [blame] | 14 | #include "build/build_config.h" |
| 15 | #include "components/services/storage/dom_storage/local_storage_impl.h" |
| 16 | #include "components/services/storage/dom_storage/session_storage_impl.h" |
Kenichi Ishibashi | df2e01d0f | 2021-02-08 07:07:03 | [diff] [blame] | 17 | #include "components/services/storage/service_worker/service_worker_storage_control_impl.h" |
Ken Rockot | 1962d93 | 2019-08-21 01:47:28 | [diff] [blame] | 18 | #include "components/services/storage/storage_service_impl.h" |
| 19 | |
| 20 | namespace storage { |
| 21 | |
Ken Rockot | af15f1d | 2020-02-14 23:08:11 | [diff] [blame] | 22 | namespace { |
| 23 | |
| 24 | const char kSessionStorageDirectory[] = "Session Storage"; |
| 25 | |
Ken rockot | 6e4416c3 | 2020-11-23 20:10:08 | [diff] [blame] | 26 | template <typename T> |
| 27 | base::OnceClosure MakeDeferredDeleter(std::unique_ptr<T> object) { |
| 28 | return base::BindOnce( |
| 29 | [](scoped_refptr<base::SequencedTaskRunner> task_runner, T* object) { |
| 30 | task_runner->DeleteSoon(FROM_HERE, object); |
| 31 | }, |
Sean Maher | 52fa5a7 | 2022-11-14 15:53:25 | [diff] [blame] | 32 | base::SequencedTaskRunner::GetCurrentDefault(), |
Ken rockot | 6e4416c3 | 2020-11-23 20:10:08 | [diff] [blame] | 33 | // NOTE: We release `object` immediately. In the case |
| 34 | // where this task never runs, we prefer to leak the |
| 35 | // object rather than potentilaly destroying it on the |
| 36 | // wrong sequence. |
| 37 | object.release()); |
| 38 | } |
| 39 | |
| 40 | template <typename T> |
| 41 | void ShutDown(std::unique_ptr<T> object) { |
| 42 | if (T* ptr = object.get()) |
| 43 | ptr->ShutDown(MakeDeferredDeleter(std::move(object))); |
| 44 | } |
| 45 | |
Ken Rockot | af15f1d | 2020-02-14 23:08:11 | [diff] [blame] | 46 | } // namespace |
| 47 | |
Ken Rockot | 1962d93 | 2019-08-21 01:47:28 | [diff] [blame] | 48 | PartitionImpl::PartitionImpl(StorageServiceImpl* service, |
Arthur Sonzogni | c571efb | 2024-01-26 20:26:18 | [diff] [blame] | 49 | const std::optional<base::FilePath>& path) |
Ken Rockot | 1962d93 | 2019-08-21 01:47:28 | [diff] [blame] | 50 | : service_(service), path_(path) { |
| 51 | receivers_.set_disconnect_handler(base::BindRepeating( |
| 52 | &PartitionImpl::OnDisconnect, base::Unretained(this))); |
| 53 | } |
| 54 | |
Ken rockot | 6e4416c3 | 2020-11-23 20:10:08 | [diff] [blame] | 55 | PartitionImpl::~PartitionImpl() { |
| 56 | ShutDown(std::move(local_storage_)); |
| 57 | ShutDown(std::move(session_storage_)); |
| 58 | } |
Ken Rockot | 1962d93 | 2019-08-21 01:47:28 | [diff] [blame] | 59 | |
| 60 | void PartitionImpl::BindReceiver( |
| 61 | mojo::PendingReceiver<mojom::Partition> receiver) { |
| 62 | DCHECK(receivers_.empty() || path_.has_value()) |
| 63 | << "In-memory partitions must have at most one client."; |
| 64 | |
| 65 | receivers_.Add(this, std::move(receiver)); |
| 66 | } |
| 67 | |
| 68 | void PartitionImpl::BindOriginContext( |
| 69 | const url::Origin& origin, |
| 70 | mojo::PendingReceiver<mojom::OriginContext> receiver) { |
| 71 | auto iter = origin_contexts_.find(origin); |
| 72 | if (iter == origin_contexts_.end()) { |
| 73 | auto result = origin_contexts_.emplace( |
| 74 | origin, std::make_unique<OriginContextImpl>(this, origin)); |
| 75 | iter = result.first; |
| 76 | } |
| 77 | |
| 78 | iter->second->BindReceiver(std::move(receiver)); |
| 79 | } |
| 80 | |
Ken Rockot | af15f1d | 2020-02-14 23:08:11 | [diff] [blame] | 81 | void PartitionImpl::BindSessionStorageControl( |
| 82 | mojo::PendingReceiver<mojom::SessionStorageControl> receiver) { |
Ken rockot | 6e4416c3 | 2020-11-23 20:10:08 | [diff] [blame] | 83 | session_storage_ = std::make_unique<SessionStorageImpl>( |
Ken Rockot | af15f1d | 2020-02-14 23:08:11 | [diff] [blame] | 84 | path_.value_or(base::FilePath()), |
Gabriel Charette | dd8d5985e | 2020-02-26 18:38:35 | [diff] [blame] | 85 | base::ThreadPool::CreateSequencedTaskRunner( |
Ken Rockot | 4e284d3 | 2020-03-06 06:26:49 | [diff] [blame] | 86 | {base::MayBlock(), base::WithBaseSyncPrimitives(), |
| 87 | base::TaskShutdownBehavior::BLOCK_SHUTDOWN}), |
Sean Maher | 52fa5a7 | 2022-11-14 15:53:25 | [diff] [blame] | 88 | base::SequencedTaskRunner::GetCurrentDefault(), |
Xiaohan Wang | 8bef6b5 | 2022-01-14 14:59:16 | [diff] [blame] | 89 | #if BUILDFLAG(IS_ANDROID) |
Ken Rockot | af15f1d | 2020-02-14 23:08:11 | [diff] [blame] | 90 | // On Android there is no support for session storage restoring, and since |
| 91 | // the restoring code is responsible for database cleanup, we must |
| 92 | // manually delete the old database here before we open a new one. |
| 93 | SessionStorageImpl::BackingMode::kClearDiskStateOnOpen, |
| 94 | #else |
| 95 | path_.has_value() ? SessionStorageImpl::BackingMode::kRestoreDiskState |
| 96 | : SessionStorageImpl::BackingMode::kNoDisk, |
| 97 | #endif |
| 98 | std::string(kSessionStorageDirectory), std::move(receiver)); |
| 99 | } |
| 100 | |
| 101 | void PartitionImpl::BindLocalStorageControl( |
| 102 | mojo::PendingReceiver<mojom::LocalStorageControl> receiver) { |
Ken rockot | 6e4416c3 | 2020-11-23 20:10:08 | [diff] [blame] | 103 | local_storage_ = std::make_unique<LocalStorageImpl>( |
Sean Maher | 52fa5a7 | 2022-11-14 15:53:25 | [diff] [blame] | 104 | path_.value_or(base::FilePath()), |
| 105 | base::SequencedTaskRunner::GetCurrentDefault(), std::move(receiver)); |
Ken Rockot | af15f1d | 2020-02-14 23:08:11 | [diff] [blame] | 106 | } |
| 107 | |
Kenichi Ishibashi | df2e01d0f | 2021-02-08 07:07:03 | [diff] [blame] | 108 | void PartitionImpl::BindServiceWorkerStorageControl( |
| 109 | mojo::PendingReceiver<mojom::ServiceWorkerStorageControl> receiver) { |
| 110 | service_worker_storage_ = std::make_unique<ServiceWorkerStorageControlImpl>( |
| 111 | path_.value_or(base::FilePath()), |
| 112 | base::ThreadPool::CreateSequencedTaskRunner( |
| 113 | {base::MayBlock(), base::WithBaseSyncPrimitives(), |
| 114 | base::TaskShutdownBehavior::BLOCK_SHUTDOWN}), |
| 115 | std::move(receiver)); |
| 116 | } |
| 117 | |
Ken Rockot | 1962d93 | 2019-08-21 01:47:28 | [diff] [blame] | 118 | void PartitionImpl::OnDisconnect() { |
| 119 | if (receivers_.empty()) { |
| 120 | // Deletes |this|. |
| 121 | service_->RemovePartition(this); |
| 122 | } |
| 123 | } |
| 124 | |
| 125 | void PartitionImpl::RemoveOriginContext(const url::Origin& origin) { |
| 126 | origin_contexts_.erase(origin); |
| 127 | } |
| 128 | |
| 129 | } // namespace storage |