2 * Copyright 2005 Jacek Caban
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"
22 IInternetProtocolEx IInternetProtocolEx_iface
;
23 IInternetPriority IInternetPriority_iface
;
32 static inline FileProtocol
*impl_from_IInternetProtocolEx(IInternetProtocolEx
*iface
)
34 return CONTAINING_RECORD(iface
, FileProtocol
, IInternetProtocolEx_iface
);
37 static inline FileProtocol
*impl_from_IInternetPriority(IInternetPriority
*iface
)
39 return CONTAINING_RECORD(iface
, FileProtocol
, IInternetPriority_iface
);
42 static HRESULT WINAPI
FileProtocol_QueryInterface(IInternetProtocolEx
*iface
, REFIID riid
, void **ppv
)
44 FileProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
47 if(IsEqualGUID(&IID_IUnknown
, riid
)) {
48 TRACE("(%p)->(IID_IUnknown %p)\n", This
, ppv
);
49 *ppv
= &This
->IInternetProtocolEx_iface
;
50 }else if(IsEqualGUID(&IID_IInternetProtocolRoot
, riid
)) {
51 TRACE("(%p)->(IID_IInternetProtocolRoot %p)\n", This
, ppv
);
52 *ppv
= &This
->IInternetProtocolEx_iface
;
53 }else if(IsEqualGUID(&IID_IInternetProtocol
, riid
)) {
54 TRACE("(%p)->(IID_IInternetProtocol %p)\n", This
, ppv
);
55 *ppv
= &This
->IInternetProtocolEx_iface
;
56 }else if(IsEqualGUID(&IID_IInternetProtocolEx
, riid
)) {
57 TRACE("(%p)->(IID_IInternetProtocolEx %p)\n", This
, ppv
);
58 *ppv
= &This
->IInternetProtocolEx_iface
;
59 }else if(IsEqualGUID(&IID_IInternetPriority
, riid
)) {
60 TRACE("(%p)->(IID_IInternetPriority %p)\n", This
, ppv
);
61 *ppv
= &This
->IInternetPriority_iface
;
65 IInternetProtocolEx_AddRef(iface
);
69 WARN("not supported interface %s\n", debugstr_guid(riid
));
73 static ULONG WINAPI
FileProtocol_AddRef(IInternetProtocolEx
*iface
)
75 FileProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
76 LONG ref
= InterlockedIncrement(&This
->ref
);
77 TRACE("(%p) ref=%d\n", This
, ref
);
81 static ULONG WINAPI
FileProtocol_Release(IInternetProtocolEx
*iface
)
83 FileProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
84 LONG ref
= InterlockedDecrement(&This
->ref
);
86 TRACE("(%p) ref=%d\n", This
, ref
);
89 if(This
->file
!= INVALID_HANDLE_VALUE
)
90 CloseHandle(This
->file
);
93 URLMON_UnlockModule();
99 static HRESULT WINAPI
FileProtocol_Start(IInternetProtocolEx
*iface
, LPCWSTR szUrl
,
100 IInternetProtocolSink
*pOIProtSink
, IInternetBindInfo
*pOIBindInfo
,
101 DWORD grfPI
, HANDLE_PTR dwReserved
)
103 FileProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
107 TRACE("(%p)->(%s %p %p %08x %lx)\n", This
, debugstr_w(szUrl
), pOIProtSink
,
108 pOIBindInfo
, grfPI
, dwReserved
);
110 hres
= CreateUri(szUrl
, Uri_CREATE_FILE_USE_DOS_PATH
, 0, &uri
);
114 hres
= IInternetProtocolEx_StartEx(&This
->IInternetProtocolEx_iface
, uri
, pOIProtSink
,
115 pOIBindInfo
, grfPI
, (HANDLE
*)dwReserved
);
121 static HRESULT WINAPI
FileProtocol_Continue(IInternetProtocolEx
*iface
, PROTOCOLDATA
*pProtocolData
)
123 FileProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
124 FIXME("(%p)->(%p)\n", This
, pProtocolData
);
128 static HRESULT WINAPI
FileProtocol_Abort(IInternetProtocolEx
*iface
, HRESULT hrReason
,
131 FileProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
132 FIXME("(%p)->(%08x %08x)\n", This
, hrReason
, dwOptions
);
136 static HRESULT WINAPI
FileProtocol_Terminate(IInternetProtocolEx
*iface
, DWORD dwOptions
)
138 FileProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
140 TRACE("(%p)->(%08x)\n", This
, dwOptions
);
145 static HRESULT WINAPI
FileProtocol_Suspend(IInternetProtocolEx
*iface
)
147 FileProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
148 FIXME("(%p)\n", This
);
152 static HRESULT WINAPI
FileProtocol_Resume(IInternetProtocolEx
*iface
)
154 FileProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
155 FIXME("(%p)\n", This
);
159 static HRESULT WINAPI
FileProtocol_Read(IInternetProtocolEx
*iface
, void *pv
,
160 ULONG cb
, ULONG
*pcbRead
)
162 FileProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
165 TRACE("(%p)->(%p %u %p)\n", This
, pv
, cb
, pcbRead
);
170 if(This
->file
== INVALID_HANDLE_VALUE
)
171 return INET_E_DATA_NOT_AVAILABLE
;
173 if (!ReadFile(This
->file
, pv
, cb
, &read
, NULL
))
174 return INET_E_DOWNLOAD_FAILURE
;
179 return cb
== read
? S_OK
: S_FALSE
;
182 static HRESULT WINAPI
FileProtocol_Seek(IInternetProtocolEx
*iface
, LARGE_INTEGER dlibMove
,
183 DWORD dwOrigin
, ULARGE_INTEGER
*plibNewPosition
)
185 FileProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
186 FIXME("(%p)->(%d %d %p)\n", This
, dlibMove
.u
.LowPart
, dwOrigin
, plibNewPosition
);
190 static HRESULT WINAPI
FileProtocol_LockRequest(IInternetProtocolEx
*iface
, DWORD dwOptions
)
192 FileProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
194 TRACE("(%p)->(%08x)\n", This
, dwOptions
);
199 static HRESULT WINAPI
FileProtocol_UnlockRequest(IInternetProtocolEx
*iface
)
201 FileProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
203 TRACE("(%p)\n", This
);
208 static inline HRESULT
report_result(IInternetProtocolSink
*protocol_sink
, HRESULT hres
, DWORD res
)
210 IInternetProtocolSink_ReportResult(protocol_sink
, hres
, res
, NULL
);
214 static HRESULT
open_file(FileProtocol
*This
, const WCHAR
*path
, IInternetProtocolSink
*protocol_sink
)
219 file
= CreateFileW(path
, GENERIC_READ
, FILE_SHARE_READ
, NULL
,
220 OPEN_EXISTING
, FILE_ATTRIBUTE_NORMAL
, NULL
);
221 if(file
== INVALID_HANDLE_VALUE
)
222 return report_result(protocol_sink
, INET_E_RESOURCE_NOT_FOUND
, GetLastError());
224 if(!GetFileSizeEx(file
, &size
)) {
226 return report_result(protocol_sink
, INET_E_RESOURCE_NOT_FOUND
, GetLastError());
230 This
->size
= size
.u
.LowPart
;
232 IInternetProtocolSink_ReportProgress(protocol_sink
,
233 BINDSTATUS_CACHEFILENAMEAVAILABLE
, path
);
237 static HRESULT WINAPI
FileProtocol_StartEx(IInternetProtocolEx
*iface
, IUri
*pUri
,
238 IInternetProtocolSink
*pOIProtSink
, IInternetBindInfo
*pOIBindInfo
,
239 DWORD grfPI
, HANDLE
*dwReserved
)
241 FileProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
242 WCHAR path
[MAX_PATH
];
251 TRACE("(%p)->(%p %p %p %08x %p)\n", This
, pUri
, pOIProtSink
,
252 pOIBindInfo
, grfPI
, dwReserved
);
258 hres
= IUri_GetScheme(pUri
, &scheme
);
261 if(scheme
!= URL_SCHEME_FILE
)
264 memset(&bindinfo
, 0, sizeof(bindinfo
));
265 bindinfo
.cbSize
= sizeof(BINDINFO
);
266 hres
= IInternetBindInfo_GetBindInfo(pOIBindInfo
, &grfBINDF
, &bindinfo
);
268 WARN("GetBindInfo failed: %08x\n", hres
);
272 ReleaseBindInfo(&bindinfo
);
274 if(!(grfBINDF
& BINDF_FROMURLMON
))
275 IInternetProtocolSink_ReportProgress(pOIProtSink
, BINDSTATUS_DIRECTBIND
, NULL
);
277 if(This
->file
!= INVALID_HANDLE_VALUE
) {
278 IInternetProtocolSink_ReportData(pOIProtSink
,
279 BSCF_FIRSTDATANOTIFICATION
|BSCF_LASTDATANOTIFICATION
,
280 This
->size
, This
->size
);
284 IInternetProtocolSink_ReportProgress(pOIProtSink
, BINDSTATUS_SENDINGREQUEST
, &null_char
);
287 hres
= CoInternetParseIUri(pUri
, PARSE_PATH_FROM_URL
, 0, path
, sizeof(path
)/sizeof(WCHAR
), &size
, 0);
289 WARN("CoInternetParseIUri failed: %08x\n", hres
);
290 return report_result(pOIProtSink
, hres
, 0);
293 hres
= open_file(This
, path
, pOIProtSink
);
297 hres
= IUri_GetDisplayUri(pUri
, &url
);
299 hres
= FindMimeFromData(NULL
, url
, NULL
, 0, NULL
, 0, &mime
, 0);
301 if(SUCCEEDED(hres
)) {
302 IInternetProtocolSink_ReportProgress(pOIProtSink
,
303 (grfBINDF
& BINDF_FROMURLMON
) ?
304 BINDSTATUS_VERIFIEDMIMETYPEAVAILABLE
: BINDSTATUS_MIMETYPEAVAILABLE
,
310 IInternetProtocolSink_ReportData(pOIProtSink
,
311 BSCF_FIRSTDATANOTIFICATION
|BSCF_LASTDATANOTIFICATION
,
312 This
->size
, This
->size
);
314 return report_result(pOIProtSink
, S_OK
, 0);
317 static const IInternetProtocolExVtbl FileProtocolExVtbl
= {
318 FileProtocol_QueryInterface
,
320 FileProtocol_Release
,
322 FileProtocol_Continue
,
324 FileProtocol_Terminate
,
325 FileProtocol_Suspend
,
329 FileProtocol_LockRequest
,
330 FileProtocol_UnlockRequest
,
334 static HRESULT WINAPI
FilePriority_QueryInterface(IInternetPriority
*iface
,
335 REFIID riid
, void **ppv
)
337 FileProtocol
*This
= impl_from_IInternetPriority(iface
);
338 return IInternetProtocolEx_QueryInterface(&This
->IInternetProtocolEx_iface
, riid
, ppv
);
341 static ULONG WINAPI
FilePriority_AddRef(IInternetPriority
*iface
)
343 FileProtocol
*This
= impl_from_IInternetPriority(iface
);
344 return IInternetProtocolEx_AddRef(&This
->IInternetProtocolEx_iface
);
347 static ULONG WINAPI
FilePriority_Release(IInternetPriority
*iface
)
349 FileProtocol
*This
= impl_from_IInternetPriority(iface
);
350 return IInternetProtocolEx_Release(&This
->IInternetProtocolEx_iface
);
353 static HRESULT WINAPI
FilePriority_SetPriority(IInternetPriority
*iface
, LONG nPriority
)
355 FileProtocol
*This
= impl_from_IInternetPriority(iface
);
357 TRACE("(%p)->(%d)\n", This
, nPriority
);
359 This
->priority
= nPriority
;
363 static HRESULT WINAPI
FilePriority_GetPriority(IInternetPriority
*iface
, LONG
*pnPriority
)
365 FileProtocol
*This
= impl_from_IInternetPriority(iface
);
367 TRACE("(%p)->(%p)\n", This
, pnPriority
);
369 *pnPriority
= This
->priority
;
373 static const IInternetPriorityVtbl FilePriorityVtbl
= {
374 FilePriority_QueryInterface
,
376 FilePriority_Release
,
377 FilePriority_SetPriority
,
378 FilePriority_GetPriority
381 HRESULT
FileProtocol_Construct(IUnknown
*pUnkOuter
, LPVOID
*ppobj
)
385 TRACE("(%p %p)\n", pUnkOuter
, ppobj
);
389 ret
= heap_alloc(sizeof(FileProtocol
));
391 ret
->IInternetProtocolEx_iface
.lpVtbl
= &FileProtocolExVtbl
;
392 ret
->IInternetPriority_iface
.lpVtbl
= &FilePriorityVtbl
;
393 ret
->file
= INVALID_HANDLE_VALUE
;
397 *ppobj
= &ret
->IInternetProtocolEx_iface
;