Source/WebKit/win/WebSecurityOrigin.cpp

   1 /*
   2  * Copyright (C) 2007 Apple 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
   6  * are met:
   7  *
   8  * 1.  Redistributions of source code must retain the above copyright
   9  *     notice, this list of conditions and the following disclaimer.
  10  * 2.  Redistributions in binary form must reproduce the above copyright
  11  *     notice, this list of conditions and the following disclaimer in the
  12  *     documentation and/or other materials provided with the distribution.
  13  * 3.  Neither the name of Apple Computer, Inc. ("Apple") nor the names of
  14  *     its contributors may be used to endorse or promote products derived
  15  *     from this software without specific prior written permission.
  16  *
  17  * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
  18  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  19  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  20  * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
  21  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  22  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  23  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  24  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  25  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  26  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  27  */
  28
  29 #include "config.h"
  30 #include "WebSecurityOrigin.h"
  31 #include "WebKitDLL.h"
  32
  33 #include <WebCore/BString.h>
  34 #include <WebCore/DatabaseTracker.h>
  35
  36 #if !ENABLE(SQL_DATABASE)
  37 #include <wtf/UnusedParam.h>
  38 #endif
  39
  40 using namespace WebCore;
  41
  42 // WebSecurityOrigin ---------------------------------------------------------------
  43 WebSecurityOrigin* WebSecurityOrigin::createInstance(SecurityOrigin* securityOrigin)
  44 {
  45     WebSecurityOrigin* origin = new WebSecurityOrigin(securityOrigin);
  46     origin->AddRef();
  47     return origin;
  48 }
  49
  50 WebSecurityOrigin::WebSecurityOrigin(SecurityOrigin* securityOrigin)
  51     : m_refCount(0)
  52     , m_securityOrigin(securityOrigin)
  53 {
  54     gClassCount++;
  55     gClassNameCount.add("WebSecurityOrigin");
  56 }
  57
  58 WebSecurityOrigin::~WebSecurityOrigin()
  59 {
  60     gClassCount--;
  61     gClassNameCount.remove("WebSecurityOrigin");
  62 }
  63
  64 // IUnknown ------------------------------------------------------------------------
  65 HRESULT STDMETHODCALLTYPE WebSecurityOrigin::QueryInterface(REFIID riid, void** ppvObject)
  66 {
  67     *ppvObject = 0;
  68     if (IsEqualGUID(riid, IID_IUnknown))
  69         *ppvObject = static_cast<IWebSecurityOrigin*>(this);
  70     else if (IsEqualGUID(riid, IID_IWebSecurityOrigin))
  71         *ppvObject = static_cast<IWebSecurityOrigin*>(this);
  72     else if (IsEqualGUID(riid, __uuidof(this)))
  73         *ppvObject = this;
  74     else
  75         return E_NOINTERFACE;
  76
  77     AddRef();
  78     return S_OK;
  79 }
  80
  81 ULONG STDMETHODCALLTYPE WebSecurityOrigin::AddRef()
  82 {
  83     return ++m_refCount;
  84 }
  85
  86 ULONG STDMETHODCALLTYPE WebSecurityOrigin::Release()
  87 {
  88     ULONG newRef = --m_refCount;
  89     if (!newRef)
  90         delete this;
  91
  92     return newRef;
  93 }
  94
  95 // IWebSecurityOrigin --------------------------------------------------------------
  96
  97 HRESULT STDMETHODCALLTYPE WebSecurityOrigin::protocol(
  98     /* [retval][out] */ BSTR* result)
  99 {
 100     if (!result)
 101         return E_POINTER;
 102
 103     *result = BString(m_securityOrigin->protocol()).release();
 104
 105     return S_OK;
 106 }
 107
 108 HRESULT STDMETHODCALLTYPE WebSecurityOrigin::host(
 109     /* [retval][out] */ BSTR* result)
 110 {
 111     if (!result)
 112         return E_POINTER;
 113
 114     *result = BString(m_securityOrigin->host()).release();
 115
 116     return S_OK;
 117 }
 118
 119 HRESULT STDMETHODCALLTYPE WebSecurityOrigin::port(
 120     /* [retval][out] */ unsigned short* result)
 121 {
 122     if (!result)
 123         return E_POINTER;
 124
 125     *result = m_securityOrigin->port();
 126
 127     return S_OK;
 128 }
 129
 130 HRESULT STDMETHODCALLTYPE WebSecurityOrigin::usage(
 131     /* [retval][out] */ unsigned long long* result)
 132 {
 133 #if ENABLE(SQL_DATABASE)
 134     if (!result)
 135         return E_POINTER;
 136
 137     *result = DatabaseTracker::tracker().usageForOrigin(m_securityOrigin.get());
 138
 139     return S_OK;
 140 #else
 141     UNUSED_PARAM(result);
 142     return E_NOTIMPL;
 143 #endif
 144 }
 145
 146 HRESULT STDMETHODCALLTYPE WebSecurityOrigin::quota(
 147     /* [retval][out] */ unsigned long long* result)
 148 {
 149 #if ENABLE(SQL_DATABASE)
 150     if (!result)
 151         return E_POINTER;
 152
 153     *result = DatabaseTracker::tracker().quotaForOrigin(m_securityOrigin.get());
 154     return S_OK;
 155 #else
 156     UNUSED_PARAM(result);
 157     return E_NOTIMPL;
 158 #endif
 159 }
 160
 161 HRESULT STDMETHODCALLTYPE WebSecurityOrigin::setQuota(
 162     /* [in] */ unsigned long long quota)
 163 {
 164 #if ENABLE(SQL_DATABASE)
 165     DatabaseTracker::tracker().setQuota(m_securityOrigin.get(), quota);
 166
 167     return S_OK;
 168 #else
 169     UNUSED_PARAM(quota);
 170     return E_NOTIMPL;
 171 #endif
 172 }