Move URL from WebCore to WTF
[WebKit-https.git] / Source / WebKitLegacy / win / WebResource.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  * 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 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 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 "WebKitDLL.h"
27 #include "WebResource.h"
28
29 #include "MarshallingHelpers.h"
30 #include "MemoryStream.h"
31 #include <WebCore/BString.h>
32
33 using namespace WebCore;
34
35 // WebResource ---------------------------------------------------------------------
36
37 WebResource::WebResource(IStream* data, const WTF::URL& url, const WTF::String& mimeType, const WTF::String& textEncodingName, const WTF::String& frameName)
38     : m_data(data)
39     , m_url(url)
40     , m_mimeType(mimeType)
41     , m_textEncodingName(textEncodingName)
42     , m_frameName(frameName)
43 {
44     gClassCount++;
45     gClassNameCount().add("WebResource");
46 }
47
48 WebResource::~WebResource()
49 {
50     gClassCount--;
51     gClassNameCount().remove("WebResource");
52 }
53
54 WebResource* WebResource::createInstance(RefPtr<WebCore::SharedBuffer>&& data, const WebCore::ResourceResponse& response)
55 {
56     COMPtr<MemoryStream> memoryStream = MemoryStream::createInstance(WTFMove(data));
57
58     WebResource* instance = new WebResource(memoryStream.get(), response.url(), response.mimeType(), response.textEncodingName(), String());
59     instance->AddRef();
60     return instance;
61 }
62
63 // IUnknown -------------------------------------------------------------------
64
65 HRESULT WebResource::QueryInterface(_In_ REFIID riid, _COM_Outptr_ void** ppvObject)
66 {
67     if (!ppvObject)
68         return E_POINTER;
69     *ppvObject = nullptr;
70     if (IsEqualGUID(riid, IID_IUnknown))
71         *ppvObject = static_cast<IUnknown*>(this);
72     else if (IsEqualGUID(riid, IID_IWebResource))
73         *ppvObject = static_cast<IWebResource*>(this);
74     else
75         return E_NOINTERFACE;
76
77     AddRef();
78     return S_OK;
79 }
80
81 ULONG WebResource::AddRef()
82 {
83     return ++m_refCount;
84 }
85
86 ULONG WebResource::Release()
87 {
88     ULONG newRef = --m_refCount;
89     if (!newRef)
90         delete(this);
91
92     return newRef;
93 }
94
95 // WebResource ------------------------------------------------------------------
96
97 HRESULT WebResource::initWithData(_In_opt_ IStream* data, _In_ BSTR url, _In_ BSTR mimeType, _In_ BSTR textEncodingName, _In_ BSTR frameName)
98 {
99     m_data = data;
100     m_url = MarshallingHelpers::BSTRToKURL(url);
101     m_mimeType = String(mimeType);
102     m_textEncodingName = String(textEncodingName);
103     m_frameName = String(frameName);
104
105     return S_OK;
106 }
107
108 HRESULT WebResource::data(_COM_Outptr_opt_ IStream** data)
109 {
110     if (!data)
111         return E_POINTER;
112     *data = nullptr;
113     return m_data.copyRefTo(data);
114 }
115    
116 HRESULT WebResource::URL(__deref_opt_out BSTR* url)
117 {
118     if (!url) {
119         ASSERT_NOT_REACHED();
120         return E_POINTER;
121     }
122
123     *url = BString(String(m_url.string())).release();
124     return S_OK;
125 }
126     
127 HRESULT WebResource::MIMEType(__deref_opt_out BSTR* mime)
128 {
129     if (!mime) {
130         ASSERT_NOT_REACHED();
131         return E_POINTER;
132     }
133
134     *mime = BString(m_mimeType).release();
135     return S_OK;
136 }
137    
138 HRESULT WebResource::textEncodingName(__deref_opt_out BSTR* encodingName)
139 {
140     if (!encodingName) {
141         ASSERT_NOT_REACHED();
142         return E_POINTER;
143     }
144
145     *encodingName = BString(m_textEncodingName).release();
146     return S_OK;
147 }
148     
149 HRESULT WebResource::frameName(__deref_opt_out BSTR* name)
150 {
151     if (!name) {
152         ASSERT_NOT_REACHED();
153         return E_POINTER;
154     }
155
156     *name = BString(m_frameName).release();
157     return S_OK;
158 }