src/third_party/WebKit/Source/modules/quota/StorageQuota.cpp

   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(ExecutionContext* executionContext, String type)
  88 {
  89     ASSERT(executionContext);
  90
  91     RefPtr<ScriptPromiseResolverWithContext> resolver = ScriptPromiseResolverWithContext::create(ScriptState::current(toIsolate(executionContext)));
  92     ScriptPromise promise = resolver->promise();
  93
  94     SecurityOrigin* securityOrigin = executionContext->securityOrigin();
  95     if (securityOrigin->isUnique()) {
  96         resolver->reject(DOMError::create(NotSupportedError));
  97         return promise;
  98     }
  99
 100     KURL storagePartition = KURL(KURL(), securityOrigin->toString());
 101     OwnPtr<StorageQuotaCallbacks> callbacks = StorageQuotaCallbacksImpl::create(resolver);
 102     blink::Platform::current()->queryStorageUsageAndQuota(storagePartition, stringToStorageQuotaType(type), callbacks.release());
 103     return promise;
 104 }
 105
 106 ScriptPromise StorageQuota::requestPersistentQuota(ScriptState* scriptState, unsigned long long newQuota)
 107 {
 108     StorageQuotaClient* client = StorageQuotaClient::from(scriptState->executionContext());
 109     if (!client) {
 110         RefPtr<ScriptPromiseResolver> resolver = ScriptPromiseResolver::create(scriptState);
 111         ScriptPromise promise = resolver->promise();
 112         resolver->reject(DOMError::create(NotSupportedError));
 113         return promise;
 114     }
 115
 116     return client->requestPersistentQuota(scriptState->executionContext(), newQuota);
 117 }
 118
 119 StorageQuota::~StorageQuota()
 120 {
 121 }
 122
 123 } // namespace WebCore