1/*
2 * Copyright (C) 2006, 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#ifndef WebURLResponse_H
27#define WebURLResponse_H
28
29#include "WebKit.h"
30#include <WebCore/ResourceResponse.h>
31
32class WebURLResponse : public IWebHTTPURLResponse, IWebURLResponsePrivate
33{
34public:
35    static WebURLResponse* createInstance();
36    static WebURLResponse* createInstance(const WebCore::ResourceResponse& response);
37protected:
38    WebURLResponse();
39    ~WebURLResponse();
40
41public:
42    // IUnknown
43    virtual HRESULT STDMETHODCALLTYPE QueryInterface(REFIID riid, void** ppvObject);
44    virtual ULONG STDMETHODCALLTYPE AddRef(void);
45    virtual ULONG STDMETHODCALLTYPE Release(void);
46
47    // IWebURLResponse
48    virtual HRESULT STDMETHODCALLTYPE expectedContentLength(
49        /* [retval][out] */ long long *result);
50
51    virtual HRESULT STDMETHODCALLTYPE initWithURL(
52        /* [in] */ BSTR url,
53        /* [in] */ BSTR mimeType,
54        /* [in] */ int expectedContentLength,
55        /* [in] */ BSTR textEncodingName);
56
57    virtual HRESULT STDMETHODCALLTYPE MIMEType(
58        /* [retval][out] */ BSTR *result);
59
60    virtual HRESULT STDMETHODCALLTYPE suggestedFilename(
61        /* [retval][out] */ BSTR *result);
62
63    virtual HRESULT STDMETHODCALLTYPE textEncodingName(
64        /* [retval][out] */ BSTR *result);
65
66    virtual HRESULT STDMETHODCALLTYPE URL(
67        /* [retval][out] */ BSTR *result);
68
69    // IWebHTTPURLResponse
70    virtual HRESULT STDMETHODCALLTYPE allHeaderFields(
71        /* [retval][out] */ IPropertyBag **headerFields);
72
73    virtual HRESULT STDMETHODCALLTYPE localizedStringForStatusCode(
74        /* [in] */ int statusCode,
75        /* [retval][out] */ BSTR *statusString);
76
77    virtual HRESULT STDMETHODCALLTYPE statusCode(
78        /* [retval][out] */ int *statusCode);
79
80    virtual HRESULT STDMETHODCALLTYPE isAttachment(
81        /* [retval][out] */ BOOL *attachment);
82
83    // IWebURLResponsePrivate
84    virtual HRESULT STDMETHODCALLTYPE sslPeerCertificate(
85        /* [retval][out] */ OLE_HANDLE *result);
86
87    const WebCore::ResourceResponse& resourceResponse() const;
88
89protected:
90    HRESULT suggestedFileExtension(BSTR* result);
91
92#if USE(CFNETWORK)
93    CFDictionaryRef certificateDictionary() const;
94#endif
95
96protected:
97    ULONG m_refCount;
98    WebCore::ResourceResponse m_response;
99
100#if USE(CFNETWORK)
101    mutable RetainPtr<CFDictionaryRef> m_SSLCertificateInfo;    // this ensures certificate contexts are valid for the lifetime of this WebURLResponse.
102#endif
103};
104
105#endif
106