file.c 13.2 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/*
 * Copyright 2005 Jacek Caban
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * 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 21 22
#include "winreg.h"
#include "shlwapi.h"

23 24 25 26 27
#include "wine/debug.h"

WINE_DEFAULT_DEBUG_CHANNEL(urlmon);

typedef struct {
28
    IUnknown            IUnknown_inner;
29 30
    IInternetProtocolEx IInternetProtocolEx_iface;
    IInternetPriority   IInternetPriority_iface;
31

32 33
    IUnknown *outer;

34
    HANDLE file;
35
    ULONG size;
36
    LONG priority;
37 38 39 40

    LONG ref;
} FileProtocol;

41 42
static inline FileProtocol *impl_from_IUnknown(IUnknown *iface)
{
43
    return CONTAINING_RECORD(iface, FileProtocol, IUnknown_inner);
44 45
}

46 47 48 49
static inline FileProtocol *impl_from_IInternetProtocolEx(IInternetProtocolEx *iface)
{
    return CONTAINING_RECORD(iface, FileProtocol, IInternetProtocolEx_iface);
}
50

51 52 53 54
static inline FileProtocol *impl_from_IInternetPriority(IInternetPriority *iface)
{
    return CONTAINING_RECORD(iface, FileProtocol, IInternetPriority_iface);
}
55

56
static HRESULT WINAPI FileProtocolUnk_QueryInterface(IUnknown *iface, REFIID riid, void **ppv)
57
{
58
    FileProtocol *This = impl_from_IUnknown(iface);
59 60 61 62

    *ppv = NULL;
    if(IsEqualGUID(&IID_IUnknown, riid)) {
        TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
63
        *ppv = &This->IUnknown_inner;
64 65
    }else if(IsEqualGUID(&IID_IInternetProtocolRoot, riid)) {
        TRACE("(%p)->(IID_IInternetProtocolRoot %p)\n", This, ppv);
66
        *ppv = &This->IInternetProtocolEx_iface;
67 68
    }else if(IsEqualGUID(&IID_IInternetProtocol, riid)) {
        TRACE("(%p)->(IID_IInternetProtocol %p)\n", This, ppv);
69
        *ppv = &This->IInternetProtocolEx_iface;
70 71
    }else if(IsEqualGUID(&IID_IInternetProtocolEx, riid)) {
        TRACE("(%p)->(IID_IInternetProtocolEx %p)\n", This, ppv);
72
        *ppv = &This->IInternetProtocolEx_iface;
73 74
    }else if(IsEqualGUID(&IID_IInternetPriority, riid)) {
        TRACE("(%p)->(IID_IInternetPriority %p)\n", This, ppv);
75
        *ppv = &This->IInternetPriority_iface;
76 77 78
    }

    if(*ppv) {
79
        IUnknown_AddRef((IUnknown*)*ppv);
80 81 82 83 84 85 86
        return S_OK;
    }

    WARN("not supported interface %s\n", debugstr_guid(riid));
    return E_NOINTERFACE;
}

87
static ULONG WINAPI FileProtocolUnk_AddRef(IUnknown *iface)
88
{
89
    FileProtocol *This = impl_from_IUnknown(iface);
90
    LONG ref = InterlockedIncrement(&This->ref);
91
    TRACE("(%p) ref=%d\n", This, ref);
92 93 94
    return ref;
}

95
static ULONG WINAPI FileProtocolUnk_Release(IUnknown *iface)
96
{
97
    FileProtocol *This = impl_from_IUnknown(iface);
98 99
    LONG ref = InterlockedDecrement(&This->ref);

100
    TRACE("(%p) ref=%d\n", This, ref);
101 102

    if(!ref) {
103
        if(This->file != INVALID_HANDLE_VALUE)
104
            CloseHandle(This->file);
105
        heap_free(This);
106 107 108 109 110 111 112

        URLMON_UnlockModule();
    }

    return ref;
}

113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139
static const IUnknownVtbl FileProtocolUnkVtbl = {
    FileProtocolUnk_QueryInterface,
    FileProtocolUnk_AddRef,
    FileProtocolUnk_Release
};

static HRESULT WINAPI FileProtocol_QueryInterface(IInternetProtocolEx *iface, REFIID riid, void **ppv)
{
    FileProtocol *This = impl_from_IInternetProtocolEx(iface);
    TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
    return IUnknown_QueryInterface(This->outer, riid, ppv);
}

static ULONG WINAPI FileProtocol_AddRef(IInternetProtocolEx *iface)
{
    FileProtocol *This = impl_from_IInternetProtocolEx(iface);
    TRACE("(%p)\n", This);
    return IUnknown_AddRef(This->outer);
}

static ULONG WINAPI FileProtocol_Release(IInternetProtocolEx *iface)
{
    FileProtocol *This = impl_from_IInternetProtocolEx(iface);
    TRACE("(%p)\n", This);
    return IUnknown_Release(This->outer);
}

140
static HRESULT WINAPI FileProtocol_Start(IInternetProtocolEx *iface, LPCWSTR szUrl,
141
        IInternetProtocolSink *pOIProtSink, IInternetBindInfo *pOIBindInfo,
142
        DWORD grfPI, HANDLE_PTR dwReserved)
143
{
144
    FileProtocol *This = impl_from_IInternetProtocolEx(iface);
145
    IUri *uri;
146
    HRESULT hres;
147

148
    TRACE("(%p)->(%s %p %p %08x %lx)\n", This, debugstr_w(szUrl), pOIProtSink,
149 150
            pOIBindInfo, grfPI, dwReserved);

151 152
    hres = CreateUri(szUrl, Uri_CREATE_FILE_USE_DOS_PATH, 0, &uri);
    if(FAILED(hres))
153
        return hres;
154

155 156
    hres = IInternetProtocolEx_StartEx(&This->IInternetProtocolEx_iface, uri, pOIProtSink,
            pOIBindInfo, grfPI, (HANDLE*)dwReserved);
157

158 159
    IUri_Release(uri);
    return hres;
160 161
}

162
static HRESULT WINAPI FileProtocol_Continue(IInternetProtocolEx *iface, PROTOCOLDATA *pProtocolData)
163
{
164
    FileProtocol *This = impl_from_IInternetProtocolEx(iface);
165 166 167 168
    FIXME("(%p)->(%p)\n", This, pProtocolData);
    return E_NOTIMPL;
}

169
static HRESULT WINAPI FileProtocol_Abort(IInternetProtocolEx *iface, HRESULT hrReason,
170 171
        DWORD dwOptions)
{
172
    FileProtocol *This = impl_from_IInternetProtocolEx(iface);
173
    FIXME("(%p)->(%08x %08x)\n", This, hrReason, dwOptions);
174 175 176
    return E_NOTIMPL;
}

177
static HRESULT WINAPI FileProtocol_Terminate(IInternetProtocolEx *iface, DWORD dwOptions)
178
{
179
    FileProtocol *This = impl_from_IInternetProtocolEx(iface);
180

181
    TRACE("(%p)->(%08x)\n", This, dwOptions);
182 183 184 185

    return S_OK;
}

186
static HRESULT WINAPI FileProtocol_Suspend(IInternetProtocolEx *iface)
187
{
188
    FileProtocol *This = impl_from_IInternetProtocolEx(iface);
189 190 191 192
    FIXME("(%p)\n", This);
    return E_NOTIMPL;
}

193
static HRESULT WINAPI FileProtocol_Resume(IInternetProtocolEx *iface)
194
{
195
    FileProtocol *This = impl_from_IInternetProtocolEx(iface);
196 197 198 199
    FIXME("(%p)\n", This);
    return E_NOTIMPL;
}

200
static HRESULT WINAPI FileProtocol_Read(IInternetProtocolEx *iface, void *pv,
201 202
        ULONG cb, ULONG *pcbRead)
{
203
    FileProtocol *This = impl_from_IInternetProtocolEx(iface);
204 205
    DWORD read = 0;

206
    TRACE("(%p)->(%p %u %p)\n", This, pv, cb, pcbRead);
207

208 209 210
    if (pcbRead)
        *pcbRead = 0;

211
    if(This->file == INVALID_HANDLE_VALUE)
212 213
        return INET_E_DATA_NOT_AVAILABLE;

214 215
    if (!ReadFile(This->file, pv, cb, &read, NULL))
        return INET_E_DOWNLOAD_FAILURE;
216 217 218 219 220 221 222

    if(pcbRead)
        *pcbRead = read;
    
    return cb == read ? S_OK : S_FALSE;
}

223
static HRESULT WINAPI FileProtocol_Seek(IInternetProtocolEx *iface, LARGE_INTEGER dlibMove,
224
        DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition)
225
{
226
    FileProtocol *This = impl_from_IInternetProtocolEx(iface);
227
    FIXME("(%p)->(%d %d %p)\n", This, dlibMove.u.LowPart, dwOrigin, plibNewPosition);
228 229 230
    return E_NOTIMPL;
}

231
static HRESULT WINAPI FileProtocol_LockRequest(IInternetProtocolEx *iface, DWORD dwOptions)
232
{
233
    FileProtocol *This = impl_from_IInternetProtocolEx(iface);
234

235
    TRACE("(%p)->(%08x)\n", This, dwOptions);
236 237 238 239

    return S_OK;
}

240
static HRESULT WINAPI FileProtocol_UnlockRequest(IInternetProtocolEx *iface)
241
{
242
    FileProtocol *This = impl_from_IInternetProtocolEx(iface);
243 244 245 246 247 248

    TRACE("(%p)\n", This);

    return S_OK;
}

249 250 251 252 253 254 255 256 257 258
static inline HRESULT report_result(IInternetProtocolSink *protocol_sink, HRESULT hres, DWORD res)
{
    IInternetProtocolSink_ReportResult(protocol_sink, hres, res, NULL);
    return hres;
}

static HRESULT WINAPI FileProtocol_StartEx(IInternetProtocolEx *iface, IUri *pUri,
        IInternetProtocolSink *pOIProtSink, IInternetBindInfo *pOIBindInfo,
        DWORD grfPI, HANDLE *dwReserved)
{
259
    FileProtocol *This = impl_from_IInternetProtocolEx(iface);
260 261 262
    WCHAR path[MAX_PATH], *ptr;
    LARGE_INTEGER file_size;
    HANDLE file_handle;
263 264
    BINDINFO bindinfo;
    DWORD grfBINDF = 0;
265
    DWORD scheme, size;
266
    LPWSTR mime = NULL;
267
    BSTR ext;
268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302
    HRESULT hres;

    TRACE("(%p)->(%p %p %p %08x %p)\n", This, pUri, pOIProtSink,
            pOIBindInfo, grfPI, dwReserved);

    if(!pUri)
        return E_INVALIDARG;

    scheme = 0;
    hres = IUri_GetScheme(pUri, &scheme);
    if(FAILED(hres))
        return hres;
    if(scheme != URL_SCHEME_FILE)
        return E_INVALIDARG;

    memset(&bindinfo, 0, sizeof(bindinfo));
    bindinfo.cbSize = sizeof(BINDINFO);
    hres = IInternetBindInfo_GetBindInfo(pOIBindInfo, &grfBINDF, &bindinfo);
    if(FAILED(hres)) {
        WARN("GetBindInfo failed: %08x\n", hres);
        return hres;
    }

    ReleaseBindInfo(&bindinfo);

    if(!(grfBINDF & BINDF_FROMURLMON))
        IInternetProtocolSink_ReportProgress(pOIProtSink, BINDSTATUS_DIRECTBIND, NULL);

    if(This->file != INVALID_HANDLE_VALUE) {
        IInternetProtocolSink_ReportData(pOIProtSink,
                BSCF_FIRSTDATANOTIFICATION|BSCF_LASTDATANOTIFICATION,
                This->size, This->size);
        return S_OK;
    }

303
    IInternetProtocolSink_ReportProgress(pOIProtSink, BINDSTATUS_SENDINGREQUEST, L"");
304

305
    size = 0;
306
    hres = CoInternetParseIUri(pUri, PARSE_PATH_FROM_URL, 0, path, ARRAY_SIZE(path), &size, 0);
307
    if(FAILED(hres)) {
308
        WARN("CoInternetParseIUri failed: %08x\n", hres);
309 310 311
        return report_result(pOIProtSink, hres, 0);
    }

312 313
    file_handle = CreateFileW(path, GENERIC_READ, FILE_SHARE_READ, NULL,
            OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
314
    if(file_handle == INVALID_HANDLE_VALUE && (ptr = wcsrchr(path, '#'))) {
315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330
        /* If path contains fragment part, try without it. */
        *ptr = 0;
        file_handle = CreateFileW(path, GENERIC_READ, FILE_SHARE_READ, NULL,
                OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
    }
    if(file_handle == INVALID_HANDLE_VALUE)
        return report_result(pOIProtSink, INET_E_RESOURCE_NOT_FOUND, GetLastError());

    if(!GetFileSizeEx(file_handle, &file_size)) {
        CloseHandle(file_handle);
        return report_result(pOIProtSink, INET_E_RESOURCE_NOT_FOUND, GetLastError());
    }

    This->file = file_handle;
    This->size = file_size.u.LowPart;
    IInternetProtocolSink_ReportProgress(pOIProtSink,  BINDSTATUS_CACHEFILENAMEAVAILABLE, path);
331

332 333 334
    hres = IUri_GetExtension(pUri, &ext);
    if(SUCCEEDED(hres)) {
        if(hres == S_OK && *ext) {
335
            if((ptr = wcschr(ext, '#')))
336
                *ptr = 0;
337 338 339 340 341 342 343 344
            hres = find_mime_from_ext(ext, &mime);
            if(SUCCEEDED(hres)) {
                IInternetProtocolSink_ReportProgress(pOIProtSink,
                        (grfBINDF & BINDF_FROMURLMON) ?
                        BINDSTATUS_VERIFIEDMIMETYPEAVAILABLE : BINDSTATUS_MIMETYPEAVAILABLE,
                        mime);
                CoTaskMemFree(mime);
            }
345
        }
346
        SysFreeString(ext);
347 348 349 350 351 352 353 354 355 356
    }

    IInternetProtocolSink_ReportData(pOIProtSink,
            BSCF_FIRSTDATANOTIFICATION|BSCF_LASTDATANOTIFICATION,
            This->size, This->size);

    return report_result(pOIProtSink, S_OK, 0);
}

static const IInternetProtocolExVtbl FileProtocolExVtbl = {
357 358 359 360 361 362 363 364 365 366 367 368
    FileProtocol_QueryInterface,
    FileProtocol_AddRef,
    FileProtocol_Release,
    FileProtocol_Start,
    FileProtocol_Continue,
    FileProtocol_Abort,
    FileProtocol_Terminate,
    FileProtocol_Suspend,
    FileProtocol_Resume,
    FileProtocol_Read,
    FileProtocol_Seek,
    FileProtocol_LockRequest,
369 370
    FileProtocol_UnlockRequest,
    FileProtocol_StartEx
371 372
};

373 374 375
static HRESULT WINAPI FilePriority_QueryInterface(IInternetPriority *iface,
                                                  REFIID riid, void **ppv)
{
376 377
    FileProtocol *This = impl_from_IInternetPriority(iface);
    return IInternetProtocolEx_QueryInterface(&This->IInternetProtocolEx_iface, riid, ppv);
378 379 380 381
}

static ULONG WINAPI FilePriority_AddRef(IInternetPriority *iface)
{
382 383
    FileProtocol *This = impl_from_IInternetPriority(iface);
    return IInternetProtocolEx_AddRef(&This->IInternetProtocolEx_iface);
384 385 386 387
}

static ULONG WINAPI FilePriority_Release(IInternetPriority *iface)
{
388 389
    FileProtocol *This = impl_from_IInternetPriority(iface);
    return IInternetProtocolEx_Release(&This->IInternetProtocolEx_iface);
390 391 392 393
}

static HRESULT WINAPI FilePriority_SetPriority(IInternetPriority *iface, LONG nPriority)
{
394
    FileProtocol *This = impl_from_IInternetPriority(iface);
395

396
    TRACE("(%p)->(%d)\n", This, nPriority);
397 398 399 400 401 402 403

    This->priority = nPriority;
    return S_OK;
}

static HRESULT WINAPI FilePriority_GetPriority(IInternetPriority *iface, LONG *pnPriority)
{
404
    FileProtocol *This = impl_from_IInternetPriority(iface);
405 406 407 408 409 410 411 412 413 414 415 416 417 418 419

    TRACE("(%p)->(%p)\n", This, pnPriority);

    *pnPriority = This->priority;
    return S_OK;
}

static const IInternetPriorityVtbl FilePriorityVtbl = {
    FilePriority_QueryInterface,
    FilePriority_AddRef,
    FilePriority_Release,
    FilePriority_SetPriority,
    FilePriority_GetPriority
};

420
HRESULT FileProtocol_Construct(IUnknown *outer, LPVOID *ppobj)
421 422 423
{
    FileProtocol *ret;

424
    TRACE("(%p %p)\n", outer, ppobj);
425 426 427

    URLMON_LockModule();

428
    ret = heap_alloc(sizeof(FileProtocol));
429

430
    ret->IUnknown_inner.lpVtbl = &FileProtocolUnkVtbl;
431 432
    ret->IInternetProtocolEx_iface.lpVtbl = &FileProtocolExVtbl;
    ret->IInternetPriority_iface.lpVtbl = &FilePriorityVtbl;
433
    ret->file = INVALID_HANDLE_VALUE;
434
    ret->priority = 0;
435
    ret->ref = 1;
436
    ret->outer = outer ? outer : &ret->IUnknown_inner;
437

438
    *ppobj = &ret->IUnknown_inner;
439 440
    return S_OK;
}