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"
20 #include "wine/debug.h"
22 WINE_DEFAULT_DEBUG_CHANNEL(urlmon
);
25 const IInternetProtocolVtbl
*lpInternetProtocolVtbl
;
26 const IInternetPriorityVtbl
*lpInternetPriorityVtbl
;
34 #define PROTOCOL(x) ((IInternetProtocol*) &(x)->lpInternetProtocolVtbl)
35 #define PRIORITY(x) ((IInternetPriority*) &(x)->lpInternetPriorityVtbl)
37 #define PROTOCOL_THIS(iface) DEFINE_THIS(FileProtocol, InternetProtocol, iface)
39 static HRESULT WINAPI
FileProtocol_QueryInterface(IInternetProtocol
*iface
, REFIID riid
, void **ppv
)
41 FileProtocol
*This
= PROTOCOL_THIS(iface
);
44 if(IsEqualGUID(&IID_IUnknown
, riid
)) {
45 TRACE("(%p)->(IID_IUnknown %p)\n", This
, ppv
);
46 *ppv
= PROTOCOL(This
);
47 }else if(IsEqualGUID(&IID_IInternetProtocolRoot
, riid
)) {
48 TRACE("(%p)->(IID_IInternetProtocolRoot %p)\n", This
, ppv
);
49 *ppv
= PROTOCOL(This
);
50 }else if(IsEqualGUID(&IID_IInternetProtocol
, riid
)) {
51 TRACE("(%p)->(IID_IInternetProtocol %p)\n", This
, ppv
);
52 *ppv
= PROTOCOL(This
);
53 }else if(IsEqualGUID(&IID_IInternetPriority
, riid
)) {
54 TRACE("(%p)->(IID_IInternetPriority %p)\n", This
, ppv
);
55 *ppv
= PRIORITY(This
);
59 IInternetProtocol_AddRef(iface
);
63 WARN("not supported interface %s\n", debugstr_guid(riid
));
67 static ULONG WINAPI
FileProtocol_AddRef(IInternetProtocol
*iface
)
69 FileProtocol
*This
= PROTOCOL_THIS(iface
);
70 LONG ref
= InterlockedIncrement(&This
->ref
);
71 TRACE("(%p) ref=%d\n", This
, ref
);
75 static ULONG WINAPI
FileProtocol_Release(IInternetProtocol
*iface
)
77 FileProtocol
*This
= PROTOCOL_THIS(iface
);
78 LONG ref
= InterlockedDecrement(&This
->ref
);
80 TRACE("(%p) ref=%d\n", This
, ref
);
84 CloseHandle(This
->file
);
87 URLMON_UnlockModule();
93 static HRESULT WINAPI
FileProtocol_Start(IInternetProtocol
*iface
, LPCWSTR szUrl
,
94 IInternetProtocolSink
*pOIProtSink
, IInternetBindInfo
*pOIBindInfo
,
95 DWORD grfPI
, DWORD dwReserved
)
97 FileProtocol
*This
= PROTOCOL_THIS(iface
);
102 LPWSTR url
, mime
= NULL
, file_name
;
104 BOOL first_call
= FALSE
;
107 static const WCHAR wszFile
[] = {'f','i','l','e',':'};
109 TRACE("(%p)->(%s %p %p %08x %d)\n", This
, debugstr_w(szUrl
), pOIProtSink
,
110 pOIBindInfo
, grfPI
, dwReserved
);
112 memset(&bindinfo
, 0, sizeof(bindinfo
));
113 bindinfo
.cbSize
= sizeof(BINDINFO
);
114 hres
= IInternetBindInfo_GetBindInfo(pOIBindInfo
, &grfBINDF
, &bindinfo
);
116 WARN("GetBindInfo failed: %08x\n", hres
);
120 ReleaseBindInfo(&bindinfo
);
122 if(!szUrl
|| !*szUrl
)
125 if(lstrlenW(szUrl
) < sizeof(wszFile
)/sizeof(WCHAR
)
126 || memcmp(szUrl
, wszFile
, sizeof(wszFile
)))
129 len
= lstrlenW(szUrl
)+16;
130 url
= heap_alloc(len
*sizeof(WCHAR
));
131 hres
= CoInternetParseUrl(szUrl
, PARSE_ENCODE
, 0, url
, len
, &len
, 0);
137 if(!(grfBINDF
& BINDF_FROMURLMON
))
138 IInternetProtocolSink_ReportProgress(pOIProtSink
, BINDSTATUS_DIRECTBIND
, NULL
);
145 IInternetProtocolSink_ReportProgress(pOIProtSink
, BINDSTATUS_SENDINGREQUEST
, &null_char
);
147 file_name
= url
+sizeof(wszFile
)/sizeof(WCHAR
);
148 if(file_name
[0] == '/' && file_name
[1] == '/')
150 if(*file_name
== '/')
153 for(ptr
= file_name
; *ptr
; ptr
++) {
154 if(*ptr
== '?' || *ptr
== '#') {
160 if(file_name
[1] == '|')
163 This
->file
= CreateFileW(file_name
, GENERIC_READ
, FILE_SHARE_READ
, NULL
,
164 OPEN_EXISTING
, FILE_ATTRIBUTE_NORMAL
, NULL
);
166 if(This
->file
== INVALID_HANDLE_VALUE
) {
168 IInternetProtocolSink_ReportResult(pOIProtSink
, INET_E_RESOURCE_NOT_FOUND
,
169 GetLastError(), NULL
);
171 return INET_E_RESOURCE_NOT_FOUND
;
174 IInternetProtocolSink_ReportProgress(pOIProtSink
,
175 BINDSTATUS_CACHEFILENAMEAVAILABLE
, file_name
);
177 hres
= FindMimeFromData(NULL
, url
, NULL
, 0, NULL
, 0, &mime
, 0);
178 if(SUCCEEDED(hres
)) {
179 IInternetProtocolSink_ReportProgress(pOIProtSink
,
180 (grfBINDF
& BINDF_FROMURLMON
) ?
181 BINDSTATUS_VERIFIEDMIMETYPEAVAILABLE
: BINDSTATUS_MIMETYPEAVAILABLE
,
189 if(GetFileSizeEx(This
->file
, &size
))
190 IInternetProtocolSink_ReportData(pOIProtSink
,
191 BSCF_FIRSTDATANOTIFICATION
|BSCF_LASTDATANOTIFICATION
,
192 size
.u
.LowPart
, size
.u
.LowPart
);
195 IInternetProtocolSink_ReportResult(pOIProtSink
, S_OK
, 0, NULL
);
200 static HRESULT WINAPI
FileProtocol_Continue(IInternetProtocol
*iface
, PROTOCOLDATA
*pProtocolData
)
202 FileProtocol
*This
= PROTOCOL_THIS(iface
);
203 FIXME("(%p)->(%p)\n", This
, pProtocolData
);
207 static HRESULT WINAPI
FileProtocol_Abort(IInternetProtocol
*iface
, HRESULT hrReason
,
210 FileProtocol
*This
= PROTOCOL_THIS(iface
);
211 FIXME("(%p)->(%08x %08x)\n", This
, hrReason
, dwOptions
);
215 static HRESULT WINAPI
FileProtocol_Terminate(IInternetProtocol
*iface
, DWORD dwOptions
)
217 FileProtocol
*This
= PROTOCOL_THIS(iface
);
219 TRACE("(%p)->(%08x)\n", This
, dwOptions
);
224 static HRESULT WINAPI
FileProtocol_Suspend(IInternetProtocol
*iface
)
226 FileProtocol
*This
= PROTOCOL_THIS(iface
);
227 FIXME("(%p)\n", This
);
231 static HRESULT WINAPI
FileProtocol_Resume(IInternetProtocol
*iface
)
233 FileProtocol
*This
= PROTOCOL_THIS(iface
);
234 FIXME("(%p)\n", This
);
238 static HRESULT WINAPI
FileProtocol_Read(IInternetProtocol
*iface
, void *pv
,
239 ULONG cb
, ULONG
*pcbRead
)
241 FileProtocol
*This
= PROTOCOL_THIS(iface
);
244 TRACE("(%p)->(%p %u %p)\n", This
, pv
, cb
, pcbRead
);
250 return INET_E_DATA_NOT_AVAILABLE
;
252 if (!ReadFile(This
->file
, pv
, cb
, &read
, NULL
))
253 return INET_E_DOWNLOAD_FAILURE
;
258 return cb
== read
? S_OK
: S_FALSE
;
261 static HRESULT WINAPI
FileProtocol_Seek(IInternetProtocol
*iface
, LARGE_INTEGER dlibMove
,
262 DWORD dwOrigin
, ULARGE_INTEGER
*plibNewPosition
)
264 FileProtocol
*This
= PROTOCOL_THIS(iface
);
265 FIXME("(%p)->(%d %d %p)\n", This
, dlibMove
.u
.LowPart
, dwOrigin
, plibNewPosition
);
269 static HRESULT WINAPI
FileProtocol_LockRequest(IInternetProtocol
*iface
, DWORD dwOptions
)
271 FileProtocol
*This
= PROTOCOL_THIS(iface
);
273 TRACE("(%p)->(%08x)\n", This
, dwOptions
);
278 static HRESULT WINAPI
FileProtocol_UnlockRequest(IInternetProtocol
*iface
)
280 FileProtocol
*This
= PROTOCOL_THIS(iface
);
282 TRACE("(%p)\n", This
);
289 static const IInternetProtocolVtbl FileProtocolVtbl
= {
290 FileProtocol_QueryInterface
,
292 FileProtocol_Release
,
294 FileProtocol_Continue
,
296 FileProtocol_Terminate
,
297 FileProtocol_Suspend
,
301 FileProtocol_LockRequest
,
302 FileProtocol_UnlockRequest
305 #define PRIORITY_THIS(iface) DEFINE_THIS(FileProtocol, InternetPriority, iface)
307 static HRESULT WINAPI
FilePriority_QueryInterface(IInternetPriority
*iface
,
308 REFIID riid
, void **ppv
)
310 FileProtocol
*This
= PRIORITY_THIS(iface
);
311 return IInternetProtocol_QueryInterface(PROTOCOL(This
), riid
, ppv
);
314 static ULONG WINAPI
FilePriority_AddRef(IInternetPriority
*iface
)
316 FileProtocol
*This
= PRIORITY_THIS(iface
);
317 return IInternetProtocol_AddRef(PROTOCOL(This
));
320 static ULONG WINAPI
FilePriority_Release(IInternetPriority
*iface
)
322 FileProtocol
*This
= PRIORITY_THIS(iface
);
323 return IInternetProtocol_Release(PROTOCOL(This
));
326 static HRESULT WINAPI
FilePriority_SetPriority(IInternetPriority
*iface
, LONG nPriority
)
328 FileProtocol
*This
= PRIORITY_THIS(iface
);
330 TRACE("(%p)->(%d)\n", This
, nPriority
);
332 This
->priority
= nPriority
;
336 static HRESULT WINAPI
FilePriority_GetPriority(IInternetPriority
*iface
, LONG
*pnPriority
)
338 FileProtocol
*This
= PRIORITY_THIS(iface
);
340 TRACE("(%p)->(%p)\n", This
, pnPriority
);
342 *pnPriority
= This
->priority
;
348 static const IInternetPriorityVtbl FilePriorityVtbl
= {
349 FilePriority_QueryInterface
,
351 FilePriority_Release
,
352 FilePriority_SetPriority
,
353 FilePriority_GetPriority
356 HRESULT
FileProtocol_Construct(IUnknown
*pUnkOuter
, LPVOID
*ppobj
)
360 TRACE("(%p %p)\n", pUnkOuter
, ppobj
);
364 ret
= heap_alloc(sizeof(FileProtocol
));
366 ret
->lpInternetProtocolVtbl
= &FileProtocolVtbl
;
367 ret
->lpInternetPriorityVtbl
= &FilePriorityVtbl
;
372 *ppobj
= PROTOCOL(ret
);