| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "modules/quota/StorageManager.h" | 5 #include "modules/quota/StorageManager.h" |
| 6 | 6 |
| 7 #include "bindings/core/v8/ScriptPromiseResolver.h" | 7 #include "bindings/core/v8/ScriptPromiseResolver.h" |
| 8 #include "bindings/modules/v8/V8StorageEstimate.h" | 8 #include "bindings/modules/v8/V8StorageEstimate.h" |
| 9 #include "core/dom/DOMException.h" | 9 #include "core/dom/DOMException.h" |
| 10 #include "core/dom/Document.h" | 10 #include "core/dom/Document.h" |
| 11 #include "core/dom/ExceptionCode.h" | 11 #include "core/dom/ExceptionCode.h" |
| 12 #include "modules/permissions/Permissions.h" | 12 #include "modules/permissions/Permissions.h" |
| 13 #include "modules/quota/StorageEstimate.h" | 13 #include "modules/quota/StorageEstimate.h" |
| 14 #include "platform/StorageQuotaCallbacks.h" | 14 #include "platform/StorageQuotaCallbacks.h" |
| 15 #include "platform/UserGestureIndicator.h" |
| 15 #include "public/platform/Platform.h" | 16 #include "public/platform/Platform.h" |
| 16 #include "public/platform/WebCallbacks.h" | 17 #include "wtf/Functional.h" |
| 17 #include "public/platform/modules/permissions/WebPermissionClient.h" | |
| 18 #include "public/platform/modules/permissions/WebPermissionStatus.h" | |
| 19 | 18 |
| 20 namespace blink { | 19 namespace blink { |
| 21 | 20 |
| 21 using mojom::blink::PermissionName; |
| 22 using mojom::blink::PermissionService; |
| 23 using mojom::blink::PermissionStatus; |
| 24 |
| 22 namespace { | 25 namespace { |
| 23 | 26 |
| 24 class DurableStorageCallbacks final : public WebPermissionCallback { | |
| 25 public: | |
| 26 DurableStorageCallbacks(ScriptPromiseResolver* resolver) | |
| 27 : m_resolver(resolver) | |
| 28 { | |
| 29 } | |
| 30 | |
| 31 void onSuccess(WebPermissionStatus status) override | |
| 32 { | |
| 33 m_resolver->resolve(status == WebPermissionStatusGranted); | |
| 34 } | |
| 35 void onError() override | |
| 36 { | |
| 37 ASSERT_NOT_REACHED(); | |
| 38 } | |
| 39 | |
| 40 private: | |
| 41 Persistent<ScriptPromiseResolver> m_resolver; | |
| 42 }; | |
| 43 | |
| 44 class EstimateCallbacks final : public StorageQuotaCallbacks { | 27 class EstimateCallbacks final : public StorageQuotaCallbacks { |
| 45 WTF_MAKE_NONCOPYABLE(EstimateCallbacks); | 28 WTF_MAKE_NONCOPYABLE(EstimateCallbacks); |
| 46 public: | 29 public: |
| 47 explicit EstimateCallbacks(ScriptPromiseResolver* resolver) | 30 explicit EstimateCallbacks(ScriptPromiseResolver* resolver) |
| 48 : m_resolver(resolver) {} | 31 : m_resolver(resolver) {} |
| 49 | 32 |
| 50 ~EstimateCallbacks() override {} | 33 ~EstimateCallbacks() override {} |
| 51 | 34 |
| 52 void didQueryStorageUsageAndQuota(unsigned long long usageInBytes, unsigned
long long quotaInBytes) override | 35 void didQueryStorageUsageAndQuota(unsigned long long usageInBytes, unsigned
long long quotaInBytes) override |
| 53 { | 36 { |
| (...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 86 if (securityOrigin->isUnique()) { | 69 if (securityOrigin->isUnique()) { |
| 87 resolver->reject(DOMException::create(NotSupportedError)); | 70 resolver->reject(DOMException::create(NotSupportedError)); |
| 88 return promise; | 71 return promise; |
| 89 } | 72 } |
| 90 String errorMessage; | 73 String errorMessage; |
| 91 if (!executionContext->isSecureContext(errorMessage)) { | 74 if (!executionContext->isSecureContext(errorMessage)) { |
| 92 resolver->reject(DOMException::create(SecurityError, errorMessage)); | 75 resolver->reject(DOMException::create(SecurityError, errorMessage)); |
| 93 return promise; | 76 return promise; |
| 94 } | 77 } |
| 95 ASSERT(executionContext->isDocument()); | 78 ASSERT(executionContext->isDocument()); |
| 96 WebPermissionClient* permissionClient = Permissions::getClient(executionCont
ext); | 79 PermissionService* permissionService = getPermissionService(scriptState->get
ExecutionContext()); |
| 97 if (!permissionClient) { | 80 if (!permissionService) { |
| 98 resolver->reject(DOMException::create(InvalidStateError, "In its current
state, the global scope can't request permissions.")); | 81 resolver->reject(DOMException::create(InvalidStateError, "In its current
state, the global scope can't request permissions.")); |
| 99 return promise; | 82 return promise; |
| 100 } | 83 } |
| 101 permissionClient->requestPermission(WebPermissionTypeDurableStorage, KURL(KU
RL(), scriptState->getExecutionContext()->getSecurityOrigin()->toString()), new
DurableStorageCallbacks(resolver)); | 84 permissionService->RequestPermission(PermissionName::DURABLE_STORAGE, script
State->getExecutionContext()->getSecurityOrigin()->toString(), UserGestureIndica
tor::processingUserGesture(), convertToBaseCallback(WTF::bind(&StorageManager::p
ermissionRequestComplete, wrapPersistent(this), wrapPersistent(resolver)))); |
| 102 | 85 |
| 103 return promise; | 86 return promise; |
| 104 } | 87 } |
| 105 | 88 |
| 106 ScriptPromise StorageManager::persisted(ScriptState* scriptState) | 89 ScriptPromise StorageManager::persisted(ScriptState* scriptState) |
| 107 { | 90 { |
| 108 ScriptPromiseResolver* resolver = ScriptPromiseResolver::create(scriptState)
; | 91 ScriptPromiseResolver* resolver = ScriptPromiseResolver::create(scriptState)
; |
| 109 ScriptPromise promise = resolver->promise(); | 92 ScriptPromise promise = resolver->promise(); |
| 110 WebPermissionClient* permissionClient = Permissions::getClient(scriptState->
getExecutionContext()); | 93 PermissionService* permissionService = getPermissionService(scriptState->get
ExecutionContext()); |
| 111 if (!permissionClient) { | 94 if (!permissionService) { |
| 112 resolver->reject(DOMException::create(InvalidStateError, "In its current
state, the global scope can't query permissions.")); | 95 resolver->reject(DOMException::create(InvalidStateError, "In its current
state, the global scope can't query permissions.")); |
| 113 return promise; | 96 return promise; |
| 114 } | 97 } |
| 115 permissionClient->queryPermission(WebPermissionTypeDurableStorage, KURL(KURL
(), scriptState->getExecutionContext()->getSecurityOrigin()->toString()), new Du
rableStorageCallbacks(resolver)); | 98 permissionService->HasPermission(PermissionName::DURABLE_STORAGE, scriptStat
e->getExecutionContext()->getSecurityOrigin()->toString(), convertToBaseCallback
(WTF::bind(&StorageManager::permissionRequestComplete, wrapPersistent(this), wra
pPersistent(resolver)))); |
| 116 return promise; | 99 return promise; |
| 117 } | 100 } |
| 118 | 101 |
| 119 ScriptPromise StorageManager::estimate(ScriptState* scriptState) | 102 ScriptPromise StorageManager::estimate(ScriptState* scriptState) |
| 120 { | 103 { |
| 121 ScriptPromiseResolver* resolver = ScriptPromiseResolver::create(scriptState)
; | 104 ScriptPromiseResolver* resolver = ScriptPromiseResolver::create(scriptState)
; |
| 122 ScriptPromise promise = resolver->promise(); | 105 ScriptPromise promise = resolver->promise(); |
| 123 ExecutionContext* executionContext = scriptState->getExecutionContext(); | 106 ExecutionContext* executionContext = scriptState->getExecutionContext(); |
| 124 SecurityOrigin* securityOrigin = executionContext->getSecurityOrigin(); | 107 SecurityOrigin* securityOrigin = executionContext->getSecurityOrigin(); |
| 125 if (securityOrigin->isUnique()) { | 108 if (securityOrigin->isUnique()) { |
| 126 resolver->reject(DOMException::create(NotSupportedError)); | 109 resolver->reject(DOMException::create(NotSupportedError)); |
| 127 return promise; | 110 return promise; |
| 128 } | 111 } |
| 129 // IDL has: [SecureContext] | 112 // IDL has: [SecureContext] |
| 130 String errorMessage; | 113 String errorMessage; |
| 131 if (!executionContext->isSecureContext(errorMessage)) { | 114 if (!executionContext->isSecureContext(errorMessage)) { |
| 132 resolver->reject(DOMException::create(SecurityError, errorMessage)); | 115 resolver->reject(DOMException::create(SecurityError, errorMessage)); |
| 133 return promise; | 116 return promise; |
| 134 } | 117 } |
| 135 | 118 |
| 136 KURL storagePartition = KURL(KURL(), securityOrigin->toString()); | 119 KURL storagePartition = KURL(KURL(), securityOrigin->toString()); |
| 137 Platform::current()->queryStorageUsageAndQuota(storagePartition, WebStorageQ
uotaTypeTemporary, new EstimateCallbacks(resolver)); | 120 Platform::current()->queryStorageUsageAndQuota(storagePartition, WebStorageQ
uotaTypeTemporary, new EstimateCallbacks(resolver)); |
| 138 return promise; | 121 return promise; |
| 139 } | 122 } |
| 140 | 123 |
| 141 DEFINE_TRACE(StorageManager) | 124 DEFINE_TRACE(StorageManager) |
| 142 { | 125 { |
| 143 } | 126 } |
| 144 | 127 |
| 128 mojom::blink::PermissionService* StorageManager::getPermissionService(ExecutionC
ontext* executionContext) |
| 129 { |
| 130 if (!m_permissionService && Permissions::connectToService(executionContext,
mojo::GetProxy(&m_permissionService))) |
| 131 m_permissionService.set_connection_error_handler(convertToBaseCallback(W
TF::bind(&StorageManager::permissionServiceConnectionError, wrapWeakPersistent(t
his)))); |
| 132 return m_permissionService.get(); |
| 133 } |
| 134 |
| 135 void StorageManager::permissionServiceConnectionError() |
| 136 { |
| 137 m_permissionService.reset(); |
| 138 } |
| 139 |
| 140 void StorageManager::permissionRequestComplete(ScriptPromiseResolver* resolver,
PermissionStatus status) |
| 141 { |
| 142 if (!resolver->getExecutionContext() || resolver->getExecutionContext()->act
iveDOMObjectsAreStopped()) |
| 143 return; |
| 144 resolver->resolve(status == PermissionStatus::GRANTED); |
| 145 } |
| 146 |
| 145 } // namespace blink | 147 } // namespace blink |
| OLD | NEW |