2 * Copyright 2005-2009 Jacek Caban for CodeWeavers
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.
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.
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
19 #include "urlmon_main.h"
21 #define NO_SHLWAPI_REG
24 #include "wine/debug.h"
26 WINE_DEFAULT_DEBUG_CHANNEL(urlmon
);
31 IUnknown IUnknown_outer
;
32 IInternetProtocolEx IInternetProtocolEx_iface
;
33 IInternetPriority IInternetPriority_iface
;
34 IWinInetHttpInfo IWinInetHttpInfo_iface
;
40 static inline FtpProtocol
*impl_from_IUnknown(IUnknown
*iface
)
42 return CONTAINING_RECORD(iface
, FtpProtocol
, IUnknown_outer
);
45 static inline FtpProtocol
*impl_from_IInternetProtocolEx(IInternetProtocolEx
*iface
)
47 return CONTAINING_RECORD(iface
, FtpProtocol
, IInternetProtocolEx_iface
);
50 static inline FtpProtocol
*impl_from_IInternetPriority(IInternetPriority
*iface
)
52 return CONTAINING_RECORD(iface
, FtpProtocol
, IInternetPriority_iface
);
54 static inline FtpProtocol
*impl_from_IWinInetHttpInfo(IWinInetHttpInfo
*iface
)
57 return CONTAINING_RECORD(iface
, FtpProtocol
, IWinInetHttpInfo_iface
);
60 static inline FtpProtocol
*impl_from_Protocol(Protocol
*prot
)
62 return CONTAINING_RECORD(prot
, FtpProtocol
, base
);
65 static HRESULT
FtpProtocol_open_request(Protocol
*prot
, IUri
*uri
, DWORD request_flags
,
66 HINTERNET internet_session
, IInternetBindInfo
*bind_info
)
68 FtpProtocol
*This
= impl_from_Protocol(prot
);
72 hres
= IUri_GetAbsoluteUri(uri
, &url
);
76 This
->base
.request
= InternetOpenUrlW(internet_session
, url
, NULL
, 0,
77 request_flags
|INTERNET_FLAG_EXISTING_CONNECT
|INTERNET_FLAG_PASSIVE
,
78 (DWORD_PTR
)&This
->base
);
80 if (!This
->base
.request
&& GetLastError() != ERROR_IO_PENDING
) {
81 WARN("InternetOpenUrl failed: %d\n", GetLastError());
82 return INET_E_RESOURCE_NOT_FOUND
;
88 static HRESULT
FtpProtocol_end_request(Protocol
*prot
)
93 static HRESULT
FtpProtocol_start_downloading(Protocol
*prot
)
95 FtpProtocol
*This
= impl_from_Protocol(prot
);
99 res
= FtpGetFileSize(This
->base
.request
, &size
);
101 This
->base
.content_length
= size
;
103 WARN("FtpGetFileSize failed: %d\n", GetLastError());
108 static void FtpProtocol_close_connection(Protocol
*prot
)
112 static void FtpProtocol_on_error(Protocol
*prot
, DWORD error
)
114 FIXME("(%p) %d - stub\n", prot
, error
);
117 static const ProtocolVtbl AsyncProtocolVtbl
= {
118 FtpProtocol_open_request
,
119 FtpProtocol_end_request
,
120 FtpProtocol_start_downloading
,
121 FtpProtocol_close_connection
,
125 static HRESULT WINAPI
FtpProtocolUnk_QueryInterface(IUnknown
*iface
, REFIID riid
, void **ppv
)
127 FtpProtocol
*This
= impl_from_IUnknown(iface
);
129 if(IsEqualGUID(&IID_IUnknown
, riid
)) {
130 TRACE("(%p)->(IID_IUnknown %p)\n", This
, ppv
);
131 *ppv
= &This
->IUnknown_outer
;
132 }else if(IsEqualGUID(&IID_IInternetProtocolRoot
, riid
)) {
133 TRACE("(%p)->(IID_IInternetProtocolRoot %p)\n", This
, ppv
);
134 *ppv
= &This
->IInternetProtocolEx_iface
;
135 }else if(IsEqualGUID(&IID_IInternetProtocol
, riid
)) {
136 TRACE("(%p)->(IID_IInternetProtocol %p)\n", This
, ppv
);
137 *ppv
= &This
->IInternetProtocolEx_iface
;
138 }else if(IsEqualGUID(&IID_IInternetProtocolEx
, riid
)) {
139 TRACE("(%p)->(IID_IInternetProtocolEx %p)\n", This
, ppv
);
140 *ppv
= &This
->IInternetProtocolEx_iface
;
141 }else if(IsEqualGUID(&IID_IInternetPriority
, riid
)) {
142 TRACE("(%p)->(IID_IInternetPriority %p)\n", This
, ppv
);
143 *ppv
= &This
->IInternetPriority_iface
;
144 }else if(IsEqualGUID(&IID_IWinInetInfo
, riid
)) {
145 TRACE("(%p)->(IID_IWinInetInfo %p)\n", This
, ppv
);
146 *ppv
= &This
->IWinInetHttpInfo_iface
;
147 }else if(IsEqualGUID(&IID_IWinInetHttpInfo
, riid
)) {
148 TRACE("(%p)->(IID_IWinInetHttpInfo %p)\n", This
, ppv
);
149 *ppv
= &This
->IWinInetHttpInfo_iface
;
152 WARN("not supported interface %s\n", debugstr_guid(riid
));
153 return E_NOINTERFACE
;
156 IUnknown_AddRef((IUnknown
*)*ppv
);
160 static ULONG WINAPI
FtpProtocolUnk_AddRef(IUnknown
*iface
)
162 FtpProtocol
*This
= impl_from_IUnknown(iface
);
163 LONG ref
= InterlockedIncrement(&This
->ref
);
164 TRACE("(%p) ref=%d\n", This
, ref
);
168 static ULONG WINAPI
FtpProtocolUnk_Release(IUnknown
*iface
)
170 FtpProtocol
*This
= impl_from_IUnknown(iface
);
171 LONG ref
= InterlockedDecrement(&This
->ref
);
173 TRACE("(%p) ref=%d\n", This
, ref
);
176 protocol_close_connection(&This
->base
);
179 URLMON_UnlockModule();
185 static const IUnknownVtbl FtpProtocolUnkVtbl
= {
186 FtpProtocolUnk_QueryInterface
,
187 FtpProtocolUnk_AddRef
,
188 FtpProtocolUnk_Release
191 static HRESULT WINAPI
FtpProtocol_QueryInterface(IInternetProtocolEx
*iface
, REFIID riid
, void **ppv
)
193 FtpProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
194 TRACE("(%p)->(%s %p)\n", This
, debugstr_guid(riid
), ppv
);
195 return IUnknown_QueryInterface(This
->outer
, riid
, ppv
);
198 static ULONG WINAPI
FtpProtocol_AddRef(IInternetProtocolEx
*iface
)
200 FtpProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
201 TRACE("(%p)\n", This
);
202 return IUnknown_AddRef(This
->outer
);
205 static ULONG WINAPI
FtpProtocol_Release(IInternetProtocolEx
*iface
)
207 FtpProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
208 TRACE("(%p)\n", This
);
209 return IUnknown_Release(This
->outer
);
212 static HRESULT WINAPI
FtpProtocol_Start(IInternetProtocolEx
*iface
, LPCWSTR szUrl
,
213 IInternetProtocolSink
*pOIProtSink
, IInternetBindInfo
*pOIBindInfo
,
214 DWORD grfPI
, HANDLE_PTR dwReserved
)
216 FtpProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
220 TRACE("(%p)->(%s %p %p %08x %lx)\n", This
, debugstr_w(szUrl
), pOIProtSink
,
221 pOIBindInfo
, grfPI
, dwReserved
);
223 hres
= CreateUri(szUrl
, 0, 0, &uri
);
227 hres
= IInternetProtocolEx_StartEx(&This
->IInternetProtocolEx_iface
, uri
, pOIProtSink
,
228 pOIBindInfo
, grfPI
, (HANDLE
*)dwReserved
);
234 static HRESULT WINAPI
FtpProtocol_Continue(IInternetProtocolEx
*iface
, PROTOCOLDATA
*pProtocolData
)
236 FtpProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
238 TRACE("(%p)->(%p)\n", This
, pProtocolData
);
240 return protocol_continue(&This
->base
, pProtocolData
);
243 static HRESULT WINAPI
FtpProtocol_Abort(IInternetProtocolEx
*iface
, HRESULT hrReason
,
246 FtpProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
248 TRACE("(%p)->(%08x %08x)\n", This
, hrReason
, dwOptions
);
250 return protocol_abort(&This
->base
, hrReason
);
253 static HRESULT WINAPI
FtpProtocol_Terminate(IInternetProtocolEx
*iface
, DWORD dwOptions
)
255 FtpProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
257 TRACE("(%p)->(%08x)\n", This
, dwOptions
);
259 protocol_close_connection(&This
->base
);
263 static HRESULT WINAPI
FtpProtocol_Suspend(IInternetProtocolEx
*iface
)
265 FtpProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
266 FIXME("(%p)\n", This
);
270 static HRESULT WINAPI
FtpProtocol_Resume(IInternetProtocolEx
*iface
)
272 FtpProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
273 FIXME("(%p)\n", This
);
277 static HRESULT WINAPI
FtpProtocol_Read(IInternetProtocolEx
*iface
, void *pv
,
278 ULONG cb
, ULONG
*pcbRead
)
280 FtpProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
282 TRACE("(%p)->(%p %u %p)\n", This
, pv
, cb
, pcbRead
);
284 return protocol_read(&This
->base
, pv
, cb
, pcbRead
);
287 static HRESULT WINAPI
FtpProtocol_Seek(IInternetProtocolEx
*iface
, LARGE_INTEGER dlibMove
,
288 DWORD dwOrigin
, ULARGE_INTEGER
*plibNewPosition
)
290 FtpProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
291 FIXME("(%p)->(%d %d %p)\n", This
, dlibMove
.u
.LowPart
, dwOrigin
, plibNewPosition
);
295 static HRESULT WINAPI
FtpProtocol_LockRequest(IInternetProtocolEx
*iface
, DWORD dwOptions
)
297 FtpProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
299 TRACE("(%p)->(%08x)\n", This
, dwOptions
);
301 return protocol_lock_request(&This
->base
);
304 static HRESULT WINAPI
FtpProtocol_UnlockRequest(IInternetProtocolEx
*iface
)
306 FtpProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
308 TRACE("(%p)\n", This
);
310 return protocol_unlock_request(&This
->base
);
313 static HRESULT WINAPI
FtpProtocol_StartEx(IInternetProtocolEx
*iface
, IUri
*pUri
,
314 IInternetProtocolSink
*pOIProtSink
, IInternetBindInfo
*pOIBindInfo
,
315 DWORD grfPI
, HANDLE
*dwReserved
)
317 FtpProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
321 TRACE("(%p)->(%p %p %p %08x %p)\n", This
, pUri
, pOIProtSink
,
322 pOIBindInfo
, grfPI
, dwReserved
);
324 hres
= IUri_GetScheme(pUri
, &scheme
);
327 if(scheme
!= URL_SCHEME_FTP
)
330 return protocol_start(&This
->base
, (IInternetProtocol
*)&This
->IInternetProtocolEx_iface
, pUri
,
331 pOIProtSink
, pOIBindInfo
);
334 static const IInternetProtocolExVtbl FtpProtocolVtbl
= {
335 FtpProtocol_QueryInterface
,
339 FtpProtocol_Continue
,
341 FtpProtocol_Terminate
,
346 FtpProtocol_LockRequest
,
347 FtpProtocol_UnlockRequest
,
351 static HRESULT WINAPI
FtpPriority_QueryInterface(IInternetPriority
*iface
, REFIID riid
, void **ppv
)
353 FtpProtocol
*This
= impl_from_IInternetPriority(iface
);
354 return IInternetProtocolEx_QueryInterface(&This
->IInternetProtocolEx_iface
, riid
, ppv
);
357 static ULONG WINAPI
FtpPriority_AddRef(IInternetPriority
*iface
)
359 FtpProtocol
*This
= impl_from_IInternetPriority(iface
);
360 return IInternetProtocolEx_AddRef(&This
->IInternetProtocolEx_iface
);
363 static ULONG WINAPI
FtpPriority_Release(IInternetPriority
*iface
)
365 FtpProtocol
*This
= impl_from_IInternetPriority(iface
);
366 return IInternetProtocolEx_Release(&This
->IInternetProtocolEx_iface
);
369 static HRESULT WINAPI
FtpPriority_SetPriority(IInternetPriority
*iface
, LONG nPriority
)
371 FtpProtocol
*This
= impl_from_IInternetPriority(iface
);
373 TRACE("(%p)->(%d)\n", This
, nPriority
);
375 This
->base
.priority
= nPriority
;
379 static HRESULT WINAPI
FtpPriority_GetPriority(IInternetPriority
*iface
, LONG
*pnPriority
)
381 FtpProtocol
*This
= impl_from_IInternetPriority(iface
);
383 TRACE("(%p)->(%p)\n", This
, pnPriority
);
385 *pnPriority
= This
->base
.priority
;
389 static const IInternetPriorityVtbl FtpPriorityVtbl
= {
390 FtpPriority_QueryInterface
,
393 FtpPriority_SetPriority
,
394 FtpPriority_GetPriority
397 static HRESULT WINAPI
HttpInfo_QueryInterface(IWinInetHttpInfo
*iface
, REFIID riid
, void **ppv
)
399 FtpProtocol
*This
= impl_from_IWinInetHttpInfo(iface
);
400 return IInternetProtocolEx_QueryInterface(&This
->IInternetProtocolEx_iface
, riid
, ppv
);
403 static ULONG WINAPI
HttpInfo_AddRef(IWinInetHttpInfo
*iface
)
405 FtpProtocol
*This
= impl_from_IWinInetHttpInfo(iface
);
406 return IInternetProtocolEx_AddRef(&This
->IInternetProtocolEx_iface
);
409 static ULONG WINAPI
HttpInfo_Release(IWinInetHttpInfo
*iface
)
411 FtpProtocol
*This
= impl_from_IWinInetHttpInfo(iface
);
412 return IInternetProtocolEx_Release(&This
->IInternetProtocolEx_iface
);
415 static HRESULT WINAPI
HttpInfo_QueryOption(IWinInetHttpInfo
*iface
, DWORD dwOption
,
416 void *pBuffer
, DWORD
*pcbBuffer
)
418 FtpProtocol
*This
= impl_from_IWinInetHttpInfo(iface
);
419 TRACE("(%p)->(%x %p %p)\n", This
, dwOption
, pBuffer
, pcbBuffer
);
421 if(!This
->base
.request
)
424 if(!InternetQueryOptionW(This
->base
.request
, dwOption
, pBuffer
, pcbBuffer
))
429 static HRESULT WINAPI
HttpInfo_QueryInfo(IWinInetHttpInfo
*iface
, DWORD dwOption
,
430 void *pBuffer
, DWORD
*pcbBuffer
, DWORD
*pdwFlags
, DWORD
*pdwReserved
)
432 FtpProtocol
*This
= impl_from_IWinInetHttpInfo(iface
);
433 TRACE("(%p)->(%x %p %p %p %p)\n", This
, dwOption
, pBuffer
, pcbBuffer
, pdwFlags
, pdwReserved
);
435 if(!This
->base
.request
)
438 if(!HttpQueryInfoW(This
->base
.request
, dwOption
, pBuffer
, pcbBuffer
, pdwFlags
))
443 static const IWinInetHttpInfoVtbl WinInetHttpInfoVtbl
= {
444 HttpInfo_QueryInterface
,
447 HttpInfo_QueryOption
,
451 HRESULT
FtpProtocol_Construct(IUnknown
*outer
, void **ppv
)
455 TRACE("(%p %p)\n", outer
, ppv
);
459 ret
= heap_alloc_zero(sizeof(FtpProtocol
));
461 ret
->base
.vtbl
= &AsyncProtocolVtbl
;
462 ret
->IUnknown_outer
.lpVtbl
= &FtpProtocolUnkVtbl
;
463 ret
->IInternetProtocolEx_iface
.lpVtbl
= &FtpProtocolVtbl
;
464 ret
->IInternetPriority_iface
.lpVtbl
= &FtpPriorityVtbl
;
465 ret
->IWinInetHttpInfo_iface
.lpVtbl
= &WinInetHttpInfoVtbl
;
467 ret
->outer
= outer
? outer
: &ret
->IUnknown_outer
;
469 *ppv
= &ret
->IUnknown_outer
;