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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
content / browser / web_database / web_database_host_impl.h [blame]
// 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.
#ifndef CONTENT_BROWSER_WEB_DATABASE_WEB_DATABASE_HOST_IMPL_H_
#define CONTENT_BROWSER_WEB_DATABASE_WEB_DATABASE_HOST_IMPL_H_
#include <stdint.h>
#include <string>
#include "base/functional/callback_forward.h"
#include "base/memory/scoped_refptr.h"
#include "build/build_config.h"
#include "components/services/storage/public/cpp/quota_error_or.h"
#include "content/common/content_export.h"
#include "mojo/public/cpp/bindings/pending_receiver.h"
#include "mojo/public/cpp/bindings/remote.h"
#include "storage/browser/database/database_tracker.h"
#include "third_party/blink/public/mojom/webdatabase/web_database.mojom.h"
namespace url {
class Origin;
} // namespace url
namespace storage {
struct BucketInfo;
} // namespace storage
namespace content {
class CONTENT_EXPORT WebDatabaseHostImpl
: public blink::mojom::WebDatabaseHost,
public storage::DatabaseTracker::Observer {
public:
WebDatabaseHostImpl(int process_id,
scoped_refptr<storage::DatabaseTracker> db_tracker);
~WebDatabaseHostImpl() override;
static void Create(
int process_id,
scoped_refptr<storage::DatabaseTracker> db_tracker,
mojo::PendingReceiver<blink::mojom::WebDatabaseHost> receiver);
// blink::mojom::WebDatabaseHost:
void OpenFile(const std::u16string& vfs_file_name,
int32_t desired_flags,
OpenFileCallback callback) override;
void DeleteFile(const std::u16string& vfs_file_name,
bool sync_dir,
DeleteFileCallback callback) override;
void GetFileAttributes(const std::u16string& vfs_file_name,
GetFileAttributesCallback callback) override;
void GetSpaceAvailable(const url::Origin& origin,
GetSpaceAvailableCallback callback) override;
void Opened(const url::Origin& origin,
const std::u16string& database_name,
const std::u16string& database_description) override;
void Modified(const url::Origin& origin,
const std::u16string& database_name) override;
void Closed(const url::Origin& origin,
const std::u16string& database_name) override;
void HandleSqliteError(const url::Origin& origin,
const std::u16string& database_name,
int32_t error) override;
// DatabaseTracker::Observer:
void OnDatabaseSizeChanged(const std::string& origin_identifier,
const std::u16string& database_name,
int64_t database_size) override;
void OnDatabaseScheduledForDeletion(
const std::string& origin_identifier,
const std::u16string& database_name) override;
private:
void DatabaseDeleteFile(const std::u16string& vfs_file_name,
bool sync_dir,
DeleteFileCallback callback,
int reschedule_count);
// Helper function to get the mojo interface for the WebDatabase on the
// render process. Creates the WebDatabase connection if it does not already
// exist.
blink::mojom::WebDatabase& GetWebDatabase();
// blink::mojom::WebDatabaseHost methods called after ValidateOrigin()
// successfully validates the origin.
void OpenFileValidated(const std::u16string& vfs_file_name,
int32_t desired_flags,
OpenFileCallback callback);
void OpenFileWithBucketCreated(
const std::u16string& vfs_file_name,
int32_t desired_flags,
OpenFileCallback callback,
storage::QuotaErrorOr<storage::BucketInfo> bucket);
void GetFileAttributesValidated(const std::u16string& vfs_file_name,
GetFileAttributesCallback callback);
void GetSpaceAvailableValidated(const url::Origin& origin,
GetSpaceAvailableCallback callback);
void OpenedValidated(const url::Origin& origin,
const std::u16string& database_name,
const std::u16string& database_description);
void ModifiedValidated(const url::Origin& origin,
const std::u16string& database_name);
void ClosedValidated(const url::Origin& origin,
const std::u16string& database_name);
void HandleSqliteErrorValidated(const url::Origin& origin,
const std::u16string& database_name,
int32_t error);
// Asynchronously calls |callback| but only if |process_id_| has permission to
// access the passed |origin|. Must be called from within the context of a
// mojo call. Invalid calls will report a bad message, which will terminate
// the calling process.
void ValidateOrigin(const url::Origin& origin, base::OnceClosure callback);
// As above, but for calls where the origin is embedded in a VFS filename.
// Empty filenames signalling a temp file are permitted.
void ValidateOrigin(const std::u16string& vfs_file_name,
base::OnceClosure callback);
// Our render process host ID, used to bind to the correct render process.
const int process_id_;
// True if and only if this instance was added as an observer
// to DatabaseTracker.
bool observer_added_;
// Keeps track of all DB connections opened by this renderer
storage::DatabaseConnections database_connections_;
// Interface to the render process WebDatabase.
mojo::Remote<blink::mojom::WebDatabase> database_provider_;
// The database tracker for the current browser context.
const scoped_refptr<storage::DatabaseTracker> db_tracker_;
// Note: This should remain the last member so it'll be destroyed and
// invalidate its weak pointers before any other members are destroyed.
base::WeakPtrFactory<WebDatabaseHostImpl> weak_ptr_factory_{this};
};
} // namespace content
#endif // CONTENT_BROWSER_WEB_DATABASE_WEB_DATABASE_HOST_IMPL_H_