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  * 1. Redistributions of source code must retain the above copyright

     8  *    notice, this list of conditions and the following disclaimer.

     9  * 2. Redistributions in binary form must reproduce the above copyright

    10  *    notice, this list of conditions and the following disclaimer in the

    11  *    documentation and/or other materials provided with the distribution.

    12  *

    13  * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY

    14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE

    15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

    16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR

    17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,

    18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,

    19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR

    20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY

    21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

    22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE

    23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 

    24  */

    25 

    26 #include "config.h"

    27 #include "WebKitDLL.h"

    28 #include <initguid.h>

    29 #include "WebURLAuthenticationChallengeSender.h"

    30 

    31 #include "COMPtr.h"

    32 #include "WebKit.h"

    33 #include "WebURLAuthenticationChallenge.h"

    34 #include "WebURLCredential.h"

    35 

    36 #pragma warning(push, 0)

    37 #include <WebCore/ResourceHandle.h>

    38 #pragma warning(pop)

    39 

    40 using namespace WebCore;

    41 

    42 // WebURLAuthenticationChallengeSender ----------------------------------------------------------------

    43 

    44 WebURLAuthenticationChallengeSender::WebURLAuthenticationChallengeSender(PassRefPtr<ResourceHandle> handle)

    45     : m_refCount(0)

    46     , m_handle(handle)

    47 {

    48     ASSERT(m_handle);

    49     gClassCount++;

    50 }

    51 

    52 WebURLAuthenticationChallengeSender::~WebURLAuthenticationChallengeSender()

    53 {

    54     gClassCount--;

    55 }

    56 

    57 WebURLAuthenticationChallengeSender* WebURLAuthenticationChallengeSender::createInstance(PassRefPtr<WebCore::ResourceHandle> handle)

    58 {

    59     WebURLAuthenticationChallengeSender* instance = new WebURLAuthenticationChallengeSender(handle);

    60     instance->AddRef();

    61     return instance;

    62 }

    63 

    64 // IUnknown -------------------------------------------------------------------

    65 

    66 HRESULT STDMETHODCALLTYPE WebURLAuthenticationChallengeSender::QueryInterface(REFIID riid, void** ppvObject)

    67 {

    68     *ppvObject = 0;

    69     if (IsEqualGUID(riid, IID_IUnknown))

    70         *ppvObject = static_cast<IUnknown*>(this);

    71     else if (IsEqualGUID(riid, __uuidof(WebURLAuthenticationChallengeSender)))

    72         *ppvObject = static_cast<WebURLAuthenticationChallengeSender*>(this);

    73     else if (IsEqualGUID(riid, IID_IWebURLAuthenticationChallengeSender))

    74         *ppvObject = static_cast<IWebURLAuthenticationChallengeSender*>(this);

    75     else

    76         return E_NOINTERFACE;

    77 

    78     AddRef();

    79     return S_OK;

    80 }

    81 

    82 ULONG STDMETHODCALLTYPE WebURLAuthenticationChallengeSender::AddRef(void)

    83 {

    84     return ++m_refCount;

    85 }

    86 

    87 ULONG STDMETHODCALLTYPE WebURLAuthenticationChallengeSender::Release(void)

    88 {

    89     ULONG newRef = --m_refCount;

    90     if (!newRef)

    91         delete(this);

    92 

    93     return newRef;

    94 }

    95 

    96 // IWebURLAuthenticationChallengeSender -------------------------------------------------------------------

    97 

    98 HRESULT STDMETHODCALLTYPE WebURLAuthenticationChallengeSender::cancelAuthenticationChallenge(

    99         /* [in] */ IWebURLAuthenticationChallenge* challenge)

   100 {

   101     COMPtr<WebURLAuthenticationChallenge> webChallenge(Query, challenge);

   102     if (!webChallenge)

   103         return E_FAIL;

   104 

   105     m_handle->receivedCancellation(webChallenge->authenticationChallenge());

   106     return S_OK;

   107 }

   108 

   109 HRESULT STDMETHODCALLTYPE WebURLAuthenticationChallengeSender::continueWithoutCredentialForAuthenticationChallenge(

   110         /* [in] */ IWebURLAuthenticationChallenge* challenge)

   111 {

   112     COMPtr<WebURLAuthenticationChallenge> webChallenge(Query, challenge);

   113     if (!webChallenge)

   114         return E_FAIL;

   115 

   116     m_handle->receivedRequestToContinueWithoutCredential(webChallenge->authenticationChallenge());

   117     return S_OK;

   118 }

   119 

   120 HRESULT STDMETHODCALLTYPE WebURLAuthenticationChallengeSender::useCredential(

   121         /* [in] */ IWebURLCredential* credential, 

   122         /* [in] */ IWebURLAuthenticationChallenge* challenge)

   123 {

   124     COMPtr<WebURLAuthenticationChallenge> webChallenge(Query, challenge);

   125     if (!webChallenge)

   126         return E_FAIL;

   127     

   128     COMPtr<WebURLCredential> webCredential;

   129     if (!credential || FAILED(credential->QueryInterface(__uuidof(WebURLCredential), (void**)&webCredential)))

   130         return E_FAIL;

   131 

   132     m_handle->receivedCredential(webChallenge->authenticationChallenge(), webCredential->credential());

   133     return S_OK;

   134 }

   135 

   136 // WebURLAuthenticationChallengeSender ----------------------------------------------------------------

   137 

   138 ResourceHandle* WebURLAuthenticationChallengeSender::resourceHandle() const

   139 {

   140     return m_handle.get();

   141 }

   142