include: Add definition of IKsControl interface in dmksctrl.h.
[wine] / dlls / urlmon / mk.c
1 /*
2  * Copyright 2007 Jacek Caban for CodeWeavers
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * This library is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with this library; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
17  */
18
19 #include "urlmon_main.h"
20 #include "wine/debug.h"
21
22 #define NO_SHLWAPI_REG
23 #include "shlwapi.h"
24
25 WINE_DEFAULT_DEBUG_CHANNEL(urlmon);
26
27 typedef struct {
28     IInternetProtocolEx IInternetProtocolEx_iface;
29
30     LONG ref;
31
32     IStream *stream;
33 } MkProtocol;
34
35 static inline MkProtocol *impl_from_IInternetProtocolEx(IInternetProtocolEx *iface)
36 {
37     return CONTAINING_RECORD(iface, MkProtocol, IInternetProtocolEx_iface);
38 }
39
40 static HRESULT WINAPI MkProtocol_QueryInterface(IInternetProtocolEx *iface, REFIID riid, void **ppv)
41 {
42     MkProtocol *This = impl_from_IInternetProtocolEx(iface);
43
44     *ppv = NULL;
45     if(IsEqualGUID(&IID_IUnknown, riid)) {
46         TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
47         *ppv = &This->IInternetProtocolEx_iface;
48     }else if(IsEqualGUID(&IID_IInternetProtocolRoot, riid)) {
49         TRACE("(%p)->(IID_IInternetProtocolRoot %p)\n", This, ppv);
50         *ppv = &This->IInternetProtocolEx_iface;
51     }else if(IsEqualGUID(&IID_IInternetProtocol, riid)) {
52         TRACE("(%p)->(IID_IInternetProtocol %p)\n", This, ppv);
53         *ppv = &This->IInternetProtocolEx_iface;
54     }else if(IsEqualGUID(&IID_IInternetProtocolEx, riid)) {
55         TRACE("(%p)->(IID_IInternetProtocolEx %p)\n", This, ppv);
56         *ppv = &This->IInternetProtocolEx_iface;
57     }
58
59     if(*ppv) {
60         IInternetProtocol_AddRef(iface);
61         return S_OK;
62     }
63
64     WARN("not supported interface %s\n", debugstr_guid(riid));
65     return E_NOINTERFACE;
66 }
67
68 static ULONG WINAPI MkProtocol_AddRef(IInternetProtocolEx *iface)
69 {
70     MkProtocol *This = impl_from_IInternetProtocolEx(iface);
71     LONG ref = InterlockedIncrement(&This->ref);
72     TRACE("(%p) ref=%d\n", This, ref);
73     return ref;
74 }
75
76 static ULONG WINAPI MkProtocol_Release(IInternetProtocolEx *iface)
77 {
78     MkProtocol *This = impl_from_IInternetProtocolEx(iface);
79     LONG ref = InterlockedDecrement(&This->ref);
80
81     TRACE("(%p) ref=%d\n", This, ref);
82
83     if(!ref) {
84         if(This->stream)
85             IStream_Release(This->stream);
86
87         heap_free(This);
88
89         URLMON_UnlockModule();
90     }
91
92     return ref;
93 }
94
95 static HRESULT report_result(IInternetProtocolSink *sink, HRESULT hres, DWORD dwError)
96 {
97     IInternetProtocolSink_ReportResult(sink, hres, dwError, NULL);
98     return hres;
99 }
100
101 static HRESULT WINAPI MkProtocol_Start(IInternetProtocolEx *iface, LPCWSTR szUrl,
102         IInternetProtocolSink *pOIProtSink, IInternetBindInfo *pOIBindInfo,
103         DWORD grfPI, HANDLE_PTR dwReserved)
104 {
105     MkProtocol *This = impl_from_IInternetProtocolEx(iface);
106     HRESULT hres;
107     IUri *uri;
108
109     TRACE("(%p)->(%s %p %p %08x %lx)\n", This, debugstr_w(szUrl), pOIProtSink,
110             pOIBindInfo, grfPI, dwReserved);
111
112     hres = CreateUri(szUrl, 0, 0, &uri);
113     if(FAILED(hres))
114         return hres;
115
116     hres = IInternetProtocolEx_StartEx(&This->IInternetProtocolEx_iface, uri, pOIProtSink,
117             pOIBindInfo, grfPI, (HANDLE*)dwReserved);
118
119     IUri_Release(uri);
120     return hres;
121 }
122
123 static HRESULT WINAPI MkProtocol_Continue(IInternetProtocolEx *iface, PROTOCOLDATA *pProtocolData)
124 {
125     MkProtocol *This = impl_from_IInternetProtocolEx(iface);
126     FIXME("(%p)->(%p)\n", This, pProtocolData);
127     return E_NOTIMPL;
128 }
129
130 static HRESULT WINAPI MkProtocol_Abort(IInternetProtocolEx *iface, HRESULT hrReason,
131         DWORD dwOptions)
132 {
133     MkProtocol *This = impl_from_IInternetProtocolEx(iface);
134     FIXME("(%p)->(%08x %08x)\n", This, hrReason, dwOptions);
135     return E_NOTIMPL;
136 }
137
138 static HRESULT WINAPI MkProtocol_Terminate(IInternetProtocolEx *iface, DWORD dwOptions)
139 {
140     MkProtocol *This = impl_from_IInternetProtocolEx(iface);
141
142     TRACE("(%p)->(%08x)\n", This, dwOptions);
143
144     return S_OK;
145 }
146
147 static HRESULT WINAPI MkProtocol_Suspend(IInternetProtocolEx *iface)
148 {
149     MkProtocol *This = impl_from_IInternetProtocolEx(iface);
150     FIXME("(%p)\n", This);
151     return E_NOTIMPL;
152 }
153
154 static HRESULT WINAPI MkProtocol_Resume(IInternetProtocolEx *iface)
155 {
156     MkProtocol *This = impl_from_IInternetProtocolEx(iface);
157     FIXME("(%p)\n", This);
158     return E_NOTIMPL;
159 }
160
161 static HRESULT WINAPI MkProtocol_Read(IInternetProtocolEx *iface, void *pv,
162         ULONG cb, ULONG *pcbRead)
163 {
164     MkProtocol *This = impl_from_IInternetProtocolEx(iface);
165
166     TRACE("(%p)->(%p %u %p)\n", This, pv, cb, pcbRead);
167
168     if(!This->stream)
169         return E_FAIL;
170
171     return IStream_Read(This->stream, pv, cb, pcbRead);
172 }
173
174 static HRESULT WINAPI MkProtocol_Seek(IInternetProtocolEx *iface, LARGE_INTEGER dlibMove,
175         DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition)
176 {
177     MkProtocol *This = impl_from_IInternetProtocolEx(iface);
178     FIXME("(%p)->(%d %d %p)\n", This, dlibMove.u.LowPart, dwOrigin, plibNewPosition);
179     return E_NOTIMPL;
180 }
181
182 static HRESULT WINAPI MkProtocol_LockRequest(IInternetProtocolEx *iface, DWORD dwOptions)
183 {
184     MkProtocol *This = impl_from_IInternetProtocolEx(iface);
185
186     TRACE("(%p)->(%08x)\n", This, dwOptions);
187
188     return S_OK;
189 }
190
191 static HRESULT WINAPI MkProtocol_UnlockRequest(IInternetProtocolEx *iface)
192 {
193     MkProtocol *This = impl_from_IInternetProtocolEx(iface);
194
195     TRACE("(%p)\n", This);
196
197     return S_OK;
198 }
199
200 static HRESULT WINAPI MkProtocol_StartEx(IInternetProtocolEx *iface, IUri *pUri,
201         IInternetProtocolSink *pOIProtSink, IInternetBindInfo *pOIBindInfo,
202         DWORD grfPI, HANDLE *dwReserved)
203 {
204     MkProtocol *This = impl_from_IInternetProtocolEx(iface);
205     LPWSTR mime, progid, display_name, colon_ptr;
206     DWORD path_size = INTERNET_MAX_URL_LENGTH;
207     DWORD bindf=0, eaten=0, scheme=0, len;
208     BSTR url, path_tmp, path = NULL;
209     IParseDisplayName *pdn;
210     BINDINFO bindinfo;
211     STATSTG statstg;
212     IMoniker *mon;
213     HRESULT hres;
214     CLSID clsid;
215
216     TRACE("(%p)->(%p %p %p %08x %p)\n", This, pUri, pOIProtSink,
217             pOIBindInfo, grfPI, dwReserved);
218
219     hres = IUri_GetScheme(pUri, &scheme);
220     if(FAILED(hres))
221         return hres;
222     if(scheme != URL_SCHEME_MK)
223         return INET_E_INVALID_URL;
224
225     memset(&bindinfo, 0, sizeof(bindinfo));
226     bindinfo.cbSize = sizeof(BINDINFO);
227     hres = IInternetBindInfo_GetBindInfo(pOIBindInfo, &bindf, &bindinfo);
228     if(FAILED(hres)) {
229         WARN("GetBindInfo failed: %08x\n", hres);
230         return hres;
231     }
232
233     ReleaseBindInfo(&bindinfo);
234
235     IInternetProtocolSink_ReportProgress(pOIProtSink, BINDSTATUS_SENDINGREQUEST, NULL);
236
237     hres = IUri_GetDisplayUri(pUri, &url);
238     if(FAILED(hres))
239         return hres;
240     hres = FindMimeFromData(NULL, url, NULL, 0, NULL, 0, &mime, 0);
241     SysFreeString(url);
242     if(SUCCEEDED(hres)) {
243         IInternetProtocolSink_ReportProgress(pOIProtSink, BINDSTATUS_MIMETYPEAVAILABLE, mime);
244         CoTaskMemFree(mime);
245     }
246
247     hres = IUri_GetPath(pUri, &path_tmp);
248     if(FAILED(hres))
249         return hres;
250     path = heap_alloc(path_size);
251     hres = UrlUnescapeW((LPWSTR)path_tmp, path, &path_size, 0);
252     SysFreeString(path_tmp);
253     if (FAILED(hres))
254     {
255         heap_free(path);
256         return report_result(pOIProtSink, INET_E_RESOURCE_NOT_FOUND, ERROR_INVALID_PARAMETER);
257     }
258     progid = path+1; /* skip '@' symbol */
259     colon_ptr = strchrW(path, ':');
260     if(!colon_ptr)
261     {
262         heap_free(path);
263         return report_result(pOIProtSink, INET_E_RESOURCE_NOT_FOUND, ERROR_INVALID_PARAMETER);
264     }
265
266     len = strlenW(path);
267     display_name = heap_alloc((len+1)*sizeof(WCHAR));
268     memcpy(display_name, path, (len+1)*sizeof(WCHAR));
269
270     progid[colon_ptr-progid] = 0; /* overwrite ':' with NULL terminator */
271     hres = CLSIDFromProgID(progid, &clsid);
272     heap_free(path);
273     if(FAILED(hres))
274     {
275         heap_free(display_name);
276         return report_result(pOIProtSink, INET_E_RESOURCE_NOT_FOUND, ERROR_INVALID_PARAMETER);
277     }
278
279     hres = CoCreateInstance(&clsid, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
280             &IID_IParseDisplayName, (void**)&pdn);
281     if(FAILED(hres)) {
282         WARN("Could not create object %s\n", debugstr_guid(&clsid));
283         heap_free(display_name);
284         return report_result(pOIProtSink, hres, ERROR_INVALID_PARAMETER);
285     }
286
287     hres = IParseDisplayName_ParseDisplayName(pdn, NULL /* FIXME */, display_name, &eaten, &mon);
288     heap_free(display_name);
289     IParseDisplayName_Release(pdn);
290     if(FAILED(hres)) {
291         WARN("ParseDisplayName failed: %08x\n", hres);
292         return report_result(pOIProtSink, hres, ERROR_INVALID_PARAMETER);
293     }
294
295     if(This->stream) {
296         IStream_Release(This->stream);
297         This->stream = NULL;
298     }
299
300     hres = IMoniker_BindToStorage(mon, NULL /* FIXME */, NULL, &IID_IStream, (void**)&This->stream);
301     IMoniker_Release(mon);
302     if(FAILED(hres)) {
303         WARN("BindToStorage failed: %08x\n", hres);
304         return report_result(pOIProtSink, hres, ERROR_INVALID_PARAMETER);
305     }
306
307     hres = IStream_Stat(This->stream, &statstg, STATFLAG_NONAME);
308     if(FAILED(hres)) {
309         WARN("Stat failed: %08x\n", hres);
310         return report_result(pOIProtSink, hres, ERROR_INVALID_PARAMETER);
311     }
312
313     IInternetProtocolSink_ReportData(pOIProtSink,
314             BSCF_FIRSTDATANOTIFICATION | BSCF_LASTDATANOTIFICATION,
315             statstg.cbSize.u.LowPart, statstg.cbSize.u.LowPart);
316     return report_result(pOIProtSink, S_OK, ERROR_SUCCESS);
317 }
318
319 static const IInternetProtocolExVtbl MkProtocolVtbl = {
320     MkProtocol_QueryInterface,
321     MkProtocol_AddRef,
322     MkProtocol_Release,
323     MkProtocol_Start,
324     MkProtocol_Continue,
325     MkProtocol_Abort,
326     MkProtocol_Terminate,
327     MkProtocol_Suspend,
328     MkProtocol_Resume,
329     MkProtocol_Read,
330     MkProtocol_Seek,
331     MkProtocol_LockRequest,
332     MkProtocol_UnlockRequest,
333     MkProtocol_StartEx
334 };
335
336 HRESULT MkProtocol_Construct(IUnknown *pUnkOuter, LPVOID *ppobj)
337 {
338     MkProtocol *ret;
339
340     TRACE("(%p %p)\n", pUnkOuter, ppobj);
341
342     URLMON_LockModule();
343
344     ret = heap_alloc(sizeof(MkProtocol));
345
346     ret->IInternetProtocolEx_iface.lpVtbl = &MkProtocolVtbl;
347     ret->ref = 1;
348     ret->stream = NULL;
349
350     /* NOTE:
351      * Native returns NULL ppobj and S_OK in CreateInstance if called with IID_IUnknown riid.
352      */
353     *ppobj = &ret->IInternetProtocolEx_iface;
354
355     return S_OK;
356 }