1 /* 2 * Copyright (C) 2014 Google Inc. All rights reserved. 3 * 4 * Redistribution and use in source and binary forms, with or without 5 * modification, are permitted provided that the following conditions are 6 * met: 7 * 8 * * Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * * Redistributions in binary form must reproduce the above 11 * copyright notice, this list of conditions and the following disclaimer 12 * in the documentation and/or other materials provided with the 13 * distribution. 14 * * Neither the name of Google Inc. nor the names of its 15 * contributors may be used to endorse or promote products derived from 16 * this software without specific prior written permission. 17 * 18 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 19 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 20 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 21 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 22 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 23 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 24 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 25 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 26 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 27 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 28 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 29 */ 30 31 #include "config.h" 32 #include "modules/quota/StorageQuota.h" 33 34 #include "bindings/v8/ScriptPromise.h" 35 #include "bindings/v8/ScriptPromiseResolver.h" 36 #include "bindings/v8/ScriptPromiseResolverWithContext.h" 37 #include "core/dom/DOMError.h" 38 #include "core/dom/ExceptionCode.h" 39 #include "core/dom/ExecutionContext.h" 40 #include "modules/quota/StorageQuotaCallbacksImpl.h" 41 #include "modules/quota/StorageQuotaClient.h" 42 #include "platform/weborigin/KURL.h" 43 #include "platform/weborigin/SecurityOrigin.h" 44 #include "public/platform/Platform.h" 45 #include "public/platform/WebStorageQuotaCallbacks.h" 46 #include "public/platform/WebStorageQuotaType.h" 47 48 namespace WebCore { 49 50 namespace { 51 52 struct StorageTypeMapping { 53 blink::WebStorageQuotaType type; 54 const char* const name; 55 }; 56 57 const StorageTypeMapping storageTypeMappings[] = { 58 { blink::WebStorageQuotaTypeTemporary, "temporary" }, 59 { blink::WebStorageQuotaTypePersistent, "persistent" }, 60 }; 61 62 blink::WebStorageQuotaType stringToStorageQuotaType(const String& type) 63 { 64 for (size_t i = 0; i < WTF_ARRAY_LENGTH(storageTypeMappings); ++i) { 65 if (storageTypeMappings[i].name == type) 66 return storageTypeMappings[i].type; 67 } 68 ASSERT_NOT_REACHED(); 69 return blink::WebStorageQuotaTypeTemporary; 70 } 71 72 } // namespace 73 74 StorageQuota::StorageQuota() 75 { 76 ScriptWrappable::init(this); 77 } 78 79 Vector<String> StorageQuota::supportedTypes() const 80 { 81 Vector<String> types; 82 for (size_t i = 0; i < WTF_ARRAY_LENGTH(storageTypeMappings); ++i) 83 types.append(storageTypeMappings[i].name); 84 return types; 85 } 86 87 ScriptPromise StorageQuota::queryInfo(ScriptState* scriptState, String type) 88 { 89 RefPtr<ScriptPromiseResolverWithContext> resolver = ScriptPromiseResolverWithContext::create(scriptState); 90 ScriptPromise promise = resolver->promise(); 91 92 SecurityOrigin* securityOrigin = scriptState->executionContext()->securityOrigin(); 93 if (securityOrigin->isUnique()) { 94 resolver->reject(DOMError::create(NotSupportedError)); 95 return promise; 96 } 97 98 KURL storagePartition = KURL(KURL(), securityOrigin->toString()); 99 OwnPtr<StorageQuotaCallbacks> callbacks = StorageQuotaCallbacksImpl::create(resolver); 100 blink::Platform::current()->queryStorageUsageAndQuota(storagePartition, stringToStorageQuotaType(type), callbacks.release()); 101 return promise; 102 } 103 104 ScriptPromise StorageQuota::requestPersistentQuota(ScriptState* scriptState, unsigned long long newQuota) 105 { 106 StorageQuotaClient* client = StorageQuotaClient::from(scriptState->executionContext()); 107 if (!client) { 108 RefPtr<ScriptPromiseResolver> resolver = ScriptPromiseResolver::create(scriptState); 109 ScriptPromise promise = resolver->promise(); 110 resolver->reject(DOMError::create(NotSupportedError)); 111 return promise; 112 } 113 114 return client->requestPersistentQuota(scriptState, newQuota); 115 } 116 117 StorageQuota::~StorageQuota() 118 { 119 } 120 121 } // namespace WebCore 122