urlmon_main.c 25.6 KB
Newer Older
1 2 3 4 5
/*
 * UrlMon
 *
 * Copyright (c) 2000 Patrik Stridvall
 *
6 7 8 9 10 11 12 13 14 15 16 17
 * 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
18
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19 20
 */

21 22
#include <stdarg.h>

Jacek Caban's avatar
Jacek Caban committed
23
#include "urlmon_main.h"
24

25 26
#include "winreg.h"

27 28
#define NO_SHLWAPI_REG
#include "shlwapi.h"
29
#include "wine/debug.h"
30

31
#include "urlmon.h"
32 33 34

WINE_DEFAULT_DEBUG_CHANNEL(urlmon);

35 36
LONG URLMON_refCount = 0;

37
HINSTANCE URLMON_hInstance = 0;
38
static HMODULE hCabinet = NULL;
39
static DWORD urlmon_tls = TLS_OUT_OF_INDEXES;
40

41 42
static void init_session(BOOL);

43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58
static struct list tls_list = LIST_INIT(tls_list);

static CRITICAL_SECTION tls_cs;
static CRITICAL_SECTION_DEBUG tls_cs_dbg =
{
    0, 0, &tls_cs,
    { &tls_cs_dbg.ProcessLocksList, &tls_cs_dbg.ProcessLocksList },
      0, 0, { (DWORD_PTR)(__FILE__ ": tls") }
};

static CRITICAL_SECTION tls_cs = { &tls_cs_dbg, -1, 0, 0, 0, 0 };

tls_data_t *get_tls_data(void)
{
    tls_data_t *data;

59
    if(urlmon_tls == TLS_OUT_OF_INDEXES) {
60
        DWORD tls = TlsAlloc();
61 62 63 64
        if(tls == TLS_OUT_OF_INDEXES)
            return NULL;

        tls = InterlockedCompareExchange((LONG*)&urlmon_tls, tls, TLS_OUT_OF_INDEXES);
65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88
        if(tls != urlmon_tls)
            TlsFree(tls);
    }

    data = TlsGetValue(urlmon_tls);
    if(!data) {
        data = heap_alloc_zero(sizeof(tls_data_t));
        if(!data)
            return NULL;

        EnterCriticalSection(&tls_cs);
        list_add_tail(&tls_list, &data->entry);
        LeaveCriticalSection(&tls_cs);

        TlsSetValue(urlmon_tls, data);
    }

    return data;
}

static void free_tls_list(void)
{
    tls_data_t *data;

89
    if(urlmon_tls == TLS_OUT_OF_INDEXES)
90 91 92 93 94 95 96 97 98 99 100 101 102 103 104
        return;

    while(!list_empty(&tls_list)) {
        data = LIST_ENTRY(list_head(&tls_list), tls_data_t, entry);
        list_remove(&data->entry);
        heap_free(data);
    }

    TlsFree(urlmon_tls);
}

static void detach_thread(void)
{
    tls_data_t *data;

105
    if(urlmon_tls == TLS_OUT_OF_INDEXES)
106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123
        return;

    data = TlsGetValue(urlmon_tls);
    if(!data)
        return;

    EnterCriticalSection(&tls_cs);
    list_remove(&data->entry);
    LeaveCriticalSection(&tls_cs);

    if(data->notif_hwnd) {
        WARN("notif_hwnd not destroyed\n");
        DestroyWindow(data->notif_hwnd);
    }

    heap_free(data);
}

124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139
static void process_detach(void)
{
    HINTERNET internet_session;

    internet_session = get_internet_session(NULL);
    if(internet_session)
        InternetCloseHandle(internet_session);

    if (hCabinet)
        FreeLibrary(hCabinet);

    init_session(FALSE);
    free_session();
    free_tls_list();
}

140
/***********************************************************************
141
 *		DllMain (URLMON.init)
142
 */
143
BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID fImpLoad)
144
{
145
    TRACE("%p 0x%x %p\n", hinstDLL, fdwReason, fImpLoad);
146 147 148 149

    switch(fdwReason) {
    case DLL_PROCESS_ATTACH:
        URLMON_hInstance = hinstDLL;
150
        init_session(TRUE);
151
        break;
152 153

    case DLL_PROCESS_DETACH:
154 155
        process_detach();
        break;
156 157 158 159

    case DLL_THREAD_DETACH:
        detach_thread();
        break;
160 161 162 163
    }
    return TRUE;
}

164 165

/***********************************************************************
Patrik Stridvall's avatar
Patrik Stridvall committed
166
 *		DllInstall (URLMON.@)
167
 */
168
HRESULT WINAPI DllInstall(BOOL bInstall, LPCWSTR cmdline)
169
{
170
  FIXME("(%s, %s): stub\n", bInstall?"TRUE":"FALSE",
171 172 173 174 175
	debugstr_w(cmdline));

  return S_OK;
}

176 177 178
/***********************************************************************
 *		DllCanUnloadNow (URLMON.@)
 */
179
HRESULT WINAPI DllCanUnloadNow(void)
180
{
181
    return URLMON_refCount != 0 ? S_FALSE : S_OK;
182 183
}

184 185 186 187


/******************************************************************************
 * Urlmon ClassFactory
188
 */
189
typedef struct {
190
    const IClassFactoryVtbl *lpClassFactoryVtbl;
191 192

    HRESULT (*pfnCreateInstance)(IUnknown *pUnkOuter, LPVOID *ppObj);
193
} ClassFactory;
194

195
#define CLASSFACTORY(x)  ((IClassFactory*)  &(x)->lpClassFactoryVtbl)
196

197
static HRESULT WINAPI CF_QueryInterface(IClassFactory *iface, REFIID riid, LPVOID *ppv)
198
{
199 200 201 202 203 204 205 206 207
    *ppv = NULL;

    if(IsEqualGUID(riid, &IID_IUnknown)) {
        TRACE("(%p)->(IID_IUnknown %p)\n", iface, ppv);
        *ppv = iface;
    }else if(IsEqualGUID(riid, &IID_IClassFactory)) {
        TRACE("(%p)->(IID_IClassFactory %p)\n", iface, ppv);
        *ppv = iface;
    }
208

209 210
    if(*ppv) {
	IUnknown_AddRef((IUnknown*)*ppv);
211 212 213
	return S_OK;
    }

214
    WARN("(%p)->(%s,%p),not found\n", iface, debugstr_guid(riid), ppv);
215 216 217
    return E_NOINTERFACE;
}

218
static ULONG WINAPI CF_AddRef(IClassFactory *iface)
219
{
220 221
    URLMON_LockModule();
    return 2;
222 223
}

224
static ULONG WINAPI CF_Release(IClassFactory *iface)
225
{
226 227
    URLMON_UnlockModule();
    return 1;
228 229
}

230

231
static HRESULT WINAPI CF_CreateInstance(IClassFactory *iface, IUnknown *pOuter,
232 233
                                        REFIID riid, LPVOID *ppobj)
{
234
    ClassFactory *This = (ClassFactory*)iface;
235 236 237 238 239 240 241 242 243 244 245 246 247 248 249
    HRESULT hres;
    LPUNKNOWN punk;
    
    TRACE("(%p)->(%p,%s,%p)\n",This,pOuter,debugstr_guid(riid),ppobj);

    *ppobj = NULL;
    if(SUCCEEDED(hres = This->pfnCreateInstance(pOuter, (LPVOID *) &punk))) {
        hres = IUnknown_QueryInterface(punk, riid, ppobj);
        IUnknown_Release(punk);
    }
    return hres;
}

static HRESULT WINAPI CF_LockServer(LPCLASSFACTORY iface,BOOL dolock)
{
250 251 252 253 254 255 256
    TRACE("(%d)\n", dolock);

    if (dolock)
	   URLMON_LockModule();
    else
	   URLMON_UnlockModule();

257 258 259
    return S_OK;
}

260
static const IClassFactoryVtbl ClassFactoryVtbl =
261 262 263 264 265 266 267 268
{
    CF_QueryInterface,
    CF_AddRef,
    CF_Release,
    CF_CreateInstance,
    CF_LockServer
};

269 270 271 272
static const ClassFactory FileProtocolCF =
    { &ClassFactoryVtbl, FileProtocol_Construct};
static const ClassFactory FtpProtocolCF =
    { &ClassFactoryVtbl, FtpProtocol_Construct};
273 274
static const ClassFactory GopherProtocolCF =
    { &ClassFactoryVtbl, GopherProtocol_Construct};
275 276
static const ClassFactory HttpProtocolCF =
    { &ClassFactoryVtbl, HttpProtocol_Construct};
277 278
static const ClassFactory HttpSProtocolCF =
    { &ClassFactoryVtbl, HttpSProtocol_Construct};
279 280
static const ClassFactory MkProtocolCF =
    { &ClassFactoryVtbl, MkProtocol_Construct};
281 282 283 284
static const ClassFactory SecurityManagerCF =
    { &ClassFactoryVtbl, SecManagerImpl_Construct};
static const ClassFactory ZoneManagerCF =
    { &ClassFactoryVtbl, ZoneMgrImpl_Construct};
285 286
static const ClassFactory StdURLMonikerCF =
    { &ClassFactoryVtbl, StdURLMoniker_Construct};
287 288
static const ClassFactory MimeFilterCF =
    { &ClassFactoryVtbl, MimeFilter_Construct};
289 290 291 292 293
 
struct object_creation_info
{
    const CLSID *clsid;
    IClassFactory *cf;
294
    LPCWSTR protocol;
295 296
};

297 298
static const WCHAR wszFile[] = {'f','i','l','e',0};
static const WCHAR wszFtp[]  = {'f','t','p',0};
299
static const WCHAR wszGopher[]  = {'g','o','p','h','e','r',0};
300
static const WCHAR wszHttp[] = {'h','t','t','p',0};
301
static const WCHAR wszHttps[] = {'h','t','t','p','s',0};
302
static const WCHAR wszMk[]   = {'m','k',0};
303

304 305
static const struct object_creation_info object_creation[] =
{
306 307
    { &CLSID_FileProtocol,            CLASSFACTORY(&FileProtocolCF),    wszFile },
    { &CLSID_FtpProtocol,             CLASSFACTORY(&FtpProtocolCF),     wszFtp  },
308
    { &CLSID_GopherProtocol,          CLASSFACTORY(&GopherProtocolCF),  wszGopher },
309
    { &CLSID_HttpProtocol,            CLASSFACTORY(&HttpProtocolCF),    wszHttp },
310
    { &CLSID_HttpSProtocol,           CLASSFACTORY(&HttpSProtocolCF),   wszHttps },
311
    { &CLSID_MkProtocol,              CLASSFACTORY(&MkProtocolCF),      wszMk },
312
    { &CLSID_InternetSecurityManager, CLASSFACTORY(&SecurityManagerCF), NULL    },
313
    { &CLSID_InternetZoneManager,     CLASSFACTORY(&ZoneManagerCF),     NULL    },
314 315
    { &CLSID_StdURLMoniker,           CLASSFACTORY(&StdURLMonikerCF),   NULL    },
    { &CLSID_DeCompMimeFilter,        CLASSFACTORY(&MimeFilterCF),      NULL    }
316 317
};

318 319
static void init_session(BOOL init)
{
320
    unsigned int i;
321 322 323

    for(i=0; i < sizeof(object_creation)/sizeof(object_creation[0]); i++) {

324 325 326 327
        if(object_creation[i].protocol)
            register_urlmon_namespace(object_creation[i].cf, object_creation[i].clsid,
                                      object_creation[i].protocol, init);
    }
328 329
}

330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347
/*******************************************************************************
 * DllGetClassObject [URLMON.@]
 * Retrieves class object from a DLL object
 *
 * NOTES
 *    Docs say returns STDAPI
 *
 * PARAMS
 *    rclsid [I] CLSID for the class object
 *    riid   [I] Reference to identifier of interface for class object
 *    ppv    [O] Address of variable to receive interface pointer for riid
 *
 * RETURNS
 *    Success: S_OK
 *    Failure: CLASS_E_CLASSNOTAVAILABLE, E_OUTOFMEMORY, E_INVALIDARG,
 *             E_UNEXPECTED
 */

348
HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppv)
349
{
350
    unsigned int i;
351
    HRESULT hr;
352 353 354 355 356 357
    
    TRACE("(%s,%s,%p)\n", debugstr_guid(rclsid), debugstr_guid(riid), ppv);
    
    for (i=0; i < sizeof(object_creation)/sizeof(object_creation[0]); i++)
    {
	if (IsEqualGUID(object_creation[i].clsid, rclsid))
358
	    return IClassFactory_QueryInterface(object_creation[i].cf, riid, ppv);
359 360
    }

361 362 363 364
    hr = URLMON_DllGetClassObject(rclsid, riid, ppv);
    if(SUCCEEDED(hr))
        return hr;

365 366
    FIXME("%s: no class found.\n", debugstr_guid(rclsid));
    return CLASS_E_CLASSNOTAVAILABLE;
367 368 369
}


370 371 372
/***********************************************************************
 *		DllRegisterServerEx (URLMON.@)
 */
373
HRESULT WINAPI DllRegisterServerEx(void)
374 375 376 377 378 379
{
    FIXME("(void): stub\n");

    return E_FAIL;
}

380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399
/**************************************************************************
 *                 IsValidURL (URLMON.@)
 * 
 * Determines if a specified string is a valid URL.
 *
 * PARAMS
 *  pBC        [I] ignored, must be NULL.
 *  szURL      [I] string that represents the URL in question.
 *  dwReserved [I] reserved and must be zero.
 *
 * RETURNS
 *  Success: S_OK.
 *  Failure: S_FALSE.
 *  returns E_INVALIDARG if one or more of the args is invalid.
 *
 * TODO:
 *  test functionality against windows to see what a valid URL is.
 */
HRESULT WINAPI IsValidURL(LPBC pBC, LPCWSTR szURL, DWORD dwReserved)
{
400
    FIXME("(%p, %s, %d): stub\n", pBC, debugstr_w(szURL), dwReserved);
401 402

    if (pBC || dwReserved || !szURL)
403
        return E_INVALIDARG;
404

405 406
    return S_OK;
}
407 408 409 410 411 412 413 414 415

/**************************************************************************
 *                 FaultInIEFeature (URLMON.@)
 *
 *  Undocumented.  Appears to be used by native shdocvw.dll.
 */
HRESULT WINAPI FaultInIEFeature( HWND hwnd, uCLSSPEC * pClassSpec,
                                 QUERYCONTEXT *pQuery, DWORD flags )
{
416
    FIXME("%p %p %p %08x\n", hwnd, pClassSpec, pQuery, flags);
417 418
    return E_NOTIMPL;
}
419 420 421 422 423 424 425

/**************************************************************************
 *                 CoGetClassObjectFromURL (URLMON.@)
 */
HRESULT WINAPI CoGetClassObjectFromURL( REFCLSID rclsid, LPCWSTR szCodeURL, DWORD dwFileVersionMS,
                                        DWORD dwFileVersionLS, LPCWSTR szContentType,
                                        LPBINDCTX pBindCtx, DWORD dwClsContext, LPVOID pvReserved,
426
                                        REFIID riid, LPVOID *ppv )
427
{
428
    FIXME("(%s %s %d %d %s %p %d %p %s %p) Stub!\n", debugstr_guid(rclsid), debugstr_w(szCodeURL),
429 430 431 432
	dwFileVersionMS, dwFileVersionLS, debugstr_w(szContentType), pBindCtx, dwClsContext, pvReserved,
	debugstr_guid(riid), ppv);
    return E_NOINTERFACE;
}
433 434 435 436 437 438 439 440 441 442 443 444 445 446

/***********************************************************************
 *           ReleaseBindInfo (URLMON.@)
 *
 * Release the resources used by the specified BINDINFO structure.
 *
 * PARAMS
 *  pbindinfo [I] BINDINFO to release.
 *
 * RETURNS
 *  Nothing.
 */
void WINAPI ReleaseBindInfo(BINDINFO* pbindinfo)
{
447 448
    DWORD size;

449 450
    TRACE("(%p)\n", pbindinfo);

451
    if(!pbindinfo || !(size = pbindinfo->cbSize))
452 453 454
        return;

    CoTaskMemFree(pbindinfo->szExtraInfo);
455 456 457 458
    ReleaseStgMedium(&pbindinfo->stgmedData);

    if(offsetof(BINDINFO, szExtraInfo) < size)
        CoTaskMemFree(pbindinfo->szCustomVerb);
459

460
    if(pbindinfo->pUnk && offsetof(BINDINFO, pUnk) < size)
461
        IUnknown_Release(pbindinfo->pUnk);
462 463 464

    memset(pbindinfo, 0, size);
    pbindinfo->cbSize = size;
465
}
466

467 468 469 470 471 472 473 474 475 476
/***********************************************************************
 *           CopyStgMedium (URLMON.@)
 */
HRESULT WINAPI CopyStgMedium(const STGMEDIUM *src, STGMEDIUM *dst)
{
    TRACE("(%p %p)\n", src, dst);

    if(!src || !dst)
        return E_POINTER;

477
    *dst = *src;
478 479 480 481

    switch(dst->tymed) {
    case TYMED_NULL:
        break;
482 483 484 485 486 487 488
    case TYMED_FILE:
        if(src->u.lpszFileName && !src->pUnkForRelease) {
            DWORD size = (strlenW(src->u.lpszFileName)+1)*sizeof(WCHAR);
            dst->u.lpszFileName = CoTaskMemAlloc(size);
            memcpy(dst->u.lpszFileName, src->u.lpszFileName, size);
        }
        break;
489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506
    case TYMED_ISTREAM:
        if(dst->u.pstm)
            IStream_AddRef(dst->u.pstm);
        break;
    case TYMED_ISTORAGE:
        if(dst->u.pstg)
            IStorage_AddRef(dst->u.pstg);
        break;
    default:
        FIXME("Unimplemented tymed %d\n", src->tymed);
    }

    if(dst->pUnkForRelease)
        IUnknown_AddRef(dst->pUnkForRelease);

    return S_OK;
}

507 508 509 510
static BOOL text_richtext_filter(const BYTE *b, DWORD size)
{
    return size > 5 && !memcmp(b, "{\\rtf", 5);
}
511

512
static BOOL text_html_filter(const BYTE *b, DWORD size)
513
{
514
    DWORD i;
515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530

    if(size < 5)
        return FALSE;

    for(i=0; i < size-5; i++) {
        if(b[i] == '<'
           && (b[i+1] == 'h' || b[i+1] == 'H')
           && (b[i+2] == 't' || b[i+2] == 'T')
           && (b[i+3] == 'm' || b[i+3] == 'M')
           && (b[i+4] == 'l' || b[i+4] == 'L'))
            return TRUE;
    }

    return FALSE;
}

531 532 533 534 535 536
static BOOL audio_basic_filter(const BYTE *b, DWORD size)
{
    return size > 4
        && b[0] == '.' && b[1] == 's' && b[2] == 'n' && b[3] == 'd';
}

537 538 539 540 541 542 543
static BOOL audio_wav_filter(const BYTE *b, DWORD size)
{
    return size > 12
        && b[0] == 'R' && b[1] == 'I' && b[2] == 'F' && b[3] == 'F'
        && b[8] == 'W' && b[9] == 'A' && b[10] == 'V' && b[11] == 'E';
}

544
static BOOL image_gif_filter(const BYTE *b, DWORD size)
545 546 547 548 549 550 551 552 553 554
{
    return size >= 6
        && (b[0] == 'G' || b[0] == 'g')
        && (b[1] == 'I' || b[1] == 'i')
        && (b[2] == 'F' || b[2] == 'f')
        &&  b[3] == '8'
        && (b[4] == '7' || b[4] == '9')
        && (b[5] == 'A' || b[5] == 'a');
}

555
static BOOL image_pjpeg_filter(const BYTE *b, DWORD size)
556
{
Jacek Caban's avatar
Jacek Caban committed
557
    return size > 2 && b[0] == 0xff && b[1] == 0xd8;
558 559
}

560
static BOOL image_tiff_filter(const BYTE *b, DWORD size)
561
{
Jacek Caban's avatar
Jacek Caban committed
562
    return size > 2 && b[0] == 0x4d && b[1] == 0x4d;
563 564
}

565
static BOOL image_xpng_filter(const BYTE *b, DWORD size)
566 567
{
    static const BYTE xpng_header[] = {0x89,'P','N','G',0x0d,0x0a,0x1a,0x0a};
Jacek Caban's avatar
Jacek Caban committed
568
    return size > sizeof(xpng_header) && !memcmp(b, xpng_header, sizeof(xpng_header));
569 570
}

571
static BOOL image_bmp_filter(const BYTE *b, DWORD size)
572 573
{
    return size >= 14
Jacek Caban's avatar
Jacek Caban committed
574
        && b[0] == 0x42 && b[1] == 0x4d
575
        && *(const DWORD *)(b+6) == 0;
576 577
}

578
static BOOL video_avi_filter(const BYTE *b, DWORD size)
579 580 581 582 583 584
{
    return size > 12
        && b[0] == 'R' && b[1] == 'I' && b[2] == 'F' && b[3] == 'F'
        && b[8] == 'A' && b[9] == 'V' && b[10] == 'I' && b[11] == 0x20;
}

585
static BOOL video_mpeg_filter(const BYTE *b, DWORD size)
586 587 588 589 590 591
{
    return size > 4
        && !b[0] && !b[1] && b[2] == 0x01
        && (b[3] == 0xb3 || b[3] == 0xba);
}

592 593 594 595 596
static BOOL application_postscript_filter(const BYTE *b, DWORD size)
{
    return size > 2 && b[0] == '%' && b[1] == '!';
}

597
static BOOL application_pdf_filter(const BYTE *b, DWORD size)
598 599 600 601
{
    return size > 4 && b[0] == 0x25 && b[1] == 0x50 && b[2] == 0x44 && b[3] == 0x46;
}

602
static BOOL application_xzip_filter(const BYTE *b, DWORD size)
603 604 605 606
{
    return size > 2 && b[0] == 0x50 && b[1] == 0x4b;
}

607
static BOOL application_xgzip_filter(const BYTE *b, DWORD size)
608 609 610 611
{
    return size > 2 && b[0] == 0x1f && b[1] == 0x8b;
}

612
static BOOL application_java_filter(const BYTE *b, DWORD size)
613 614 615 616
{
    return size > 4 && b[0] == 0xca && b[1] == 0xfe && b[2] == 0xba && b[3] == 0xbe;
}

617
static BOOL application_xmsdownload(const BYTE *b, DWORD size)
618 619 620 621
{
    return size > 2 && b[0] == 'M' && b[1] == 'Z';
}

622
static BOOL text_plain_filter(const BYTE *b, DWORD size)
623
{
Jacek Caban's avatar
Jacek Caban committed
624
    const BYTE *ptr;
625

Jacek Caban's avatar
Jacek Caban committed
626
    for(ptr = b; ptr < b+size-1; ptr++) {
627 628 629 630 631 632 633
        if(*ptr < 0x20 && *ptr != '\n' && *ptr != '\r' && *ptr != '\t')
            return FALSE;
    }

    return TRUE;
}

634
static BOOL application_octet_stream_filter(const BYTE *b, DWORD size)
635 636 637 638
{
    return TRUE;
}

639 640 641 642 643
/***********************************************************************
 *           FindMimeFromData (URLMON.@)
 *
 * Determines the Multipurpose Internet Mail Extensions (MIME) type from the data provided.
 */
644 645 646 647
HRESULT WINAPI FindMimeFromData(LPBC pBC, LPCWSTR pwzUrl, LPVOID pBuffer,
        DWORD cbSize, LPCWSTR pwzMimeProposed, DWORD dwMimeFlags,
        LPWSTR* ppwzMimeOut, DWORD dwReserved)
{
648
    TRACE("(%p,%s,%p,%d,%s,0x%x,%p,0x%x)\n", pBC, debugstr_w(pwzUrl), pBuffer, cbSize,
649 650 651
            debugstr_w(pwzMimeProposed), dwMimeFlags, ppwzMimeOut, dwReserved);

    if(dwMimeFlags)
652
        WARN("dwMimeFlags=%08x\n", dwMimeFlags);
653
    if(dwReserved)
654
        WARN("dwReserved=%d\n", dwReserved);
655 656 657 658 659 660

    /* pBC seams to not be used */

    if(!ppwzMimeOut || (!pwzUrl && !pBuffer))
        return E_INVALIDARG;

661
    if(pwzMimeProposed && (!pBuffer || (pBuffer && !cbSize))) {
662 663 664 665 666 667 668 669 670 671 672 673
        DWORD len;

        if(!pwzMimeProposed)
            return E_FAIL;

        len = strlenW(pwzMimeProposed)+1;
        *ppwzMimeOut = CoTaskMemAlloc(len*sizeof(WCHAR));
        memcpy(*ppwzMimeOut, pwzMimeProposed, len*sizeof(WCHAR));
        return S_OK;
    }

    if(pBuffer) {
674
        const BYTE *buf = pBuffer;
675
        DWORD len;
676
        LPCWSTR ret = NULL;
677
        unsigned int i;
678

679
        static const WCHAR wszTextHtml[] = {'t','e','x','t','/','h','t','m','l',0};
680
        static const WCHAR wszTextRichtext[] = {'t','e','x','t','/','r','i','c','h','t','e','x','t',0};
681
        static const WCHAR wszAudioBasic[] = {'a','u','d','i','o','/','b','a','s','i','c',0};
682
        static const WCHAR wszAudioWav[] = {'a','u','d','i','o','/','w','a','v',0};
683
        static const WCHAR wszImageGif[] = {'i','m','a','g','e','/','g','i','f',0};
684
        static const WCHAR wszImagePjpeg[] = {'i','m','a','g','e','/','p','j','p','e','g',0};
685
        static const WCHAR wszImageTiff[] = {'i','m','a','g','e','/','t','i','f','f',0};
686
        static const WCHAR wszImageXPng[] = {'i','m','a','g','e','/','x','-','p','n','g',0};
687
        static const WCHAR wszImageBmp[] = {'i','m','a','g','e','/','b','m','p',0};
688
        static const WCHAR wszVideoAvi[] = {'v','i','d','e','o','/','a','v','i',0};
689
        static const WCHAR wszVideoMpeg[] = {'v','i','d','e','o','/','m','p','e','g',0};
690 691
        static const WCHAR wszAppPostscript[] =
            {'a','p','p','l','i','c','a','t','i','o','n','/','p','o','s','t','s','c','r','i','p','t',0};
692 693
        static const WCHAR wszAppPdf[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
            'p','d','f',0};
694 695
        static const WCHAR wszAppXZip[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
            'x','-','z','i','p','-','c','o','m','p','r','e','s','s','e','d',0};
696 697
        static const WCHAR wszAppXGzip[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
            'x','-','g','z','i','p','-','c','o','m','p','r','e','s','s','e','d',0};
698 699
        static const WCHAR wszAppJava[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
            'j','a','v','a',0};
700 701
        static const WCHAR wszAppXMSDownload[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
            'x','-','m','s','d','o','w','n','l','o','a','d',0};
702
        static const WCHAR wszTextPlain[] = {'t','e','x','t','/','p','l','a','i','n','\0'};
703 704 705 706 707
        static const WCHAR wszAppOctetStream[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
            'o','c','t','e','t','-','s','t','r','e','a','m','\0'};

        static const struct {
            LPCWSTR mime;
708
            BOOL (*filter)(const BYTE *,DWORD);
709
        } mime_filters[] = {
710
            {wszTextHtml,       text_html_filter},
711
            {wszTextRichtext,   text_richtext_filter},
712
         /* {wszAudioXAiff,     audio_xaiff_filter}, */
713
            {wszAudioBasic,     audio_basic_filter},
714
            {wszAudioWav,       audio_wav_filter},
715
            {wszImageGif,       image_gif_filter},
716
            {wszImagePjpeg,     image_pjpeg_filter},
717
            {wszImageTiff,      image_tiff_filter},
718
            {wszImageXPng,      image_xpng_filter},
719
         /* {wszImageXBitmap,   image_xbitmap_filter}, */
720
            {wszImageBmp,       image_bmp_filter},
721 722 723
         /* {wszImageXJg,       image_xjg_filter}, */
         /* {wszImageXEmf,      image_xemf_filter}, */
         /* {wszImageXWmf,      image_xwmf_filter}, */
724
            {wszVideoAvi,       video_avi_filter},
725
            {wszVideoMpeg,      video_mpeg_filter},
726
            {wszAppPostscript,  application_postscript_filter},
727 728
         /* {wszAppBase64,      application_base64_filter}, */
         /* {wszAppMacbinhex40, application_macbinhex40_filter}, */
729
            {wszAppPdf,         application_pdf_filter},
730
         /* {wszAppXCompressed, application_xcompressed_filter}, */
731
            {wszAppXZip,        application_xzip_filter},
732
            {wszAppXGzip,       application_xgzip_filter},
733
            {wszAppJava,        application_java_filter},
734
            {wszAppXMSDownload, application_xmsdownload},
735 736 737
            {wszTextPlain,      text_plain_filter},
            {wszAppOctetStream, application_octet_stream_filter}
        };
738 739 740 741

        if(!cbSize)
            return E_FAIL;

742 743 744 745 746 747 748
        if(pwzMimeProposed && strcmpW(pwzMimeProposed, wszAppOctetStream)) {
            for(i=0; i < sizeof(mime_filters)/sizeof(*mime_filters); i++) {
                if(!strcmpW(pwzMimeProposed, mime_filters[i].mime))
                    break;
            }

            if(i == sizeof(mime_filters)/sizeof(*mime_filters)
Jacek Caban's avatar
Jacek Caban committed
749
                    || mime_filters[i].filter(buf, cbSize)) {
750 751 752 753 754 755 756 757
                len = strlenW(pwzMimeProposed)+1;
                *ppwzMimeOut = CoTaskMemAlloc(len*sizeof(WCHAR));
                memcpy(*ppwzMimeOut, pwzMimeProposed, len*sizeof(WCHAR));
                return S_OK;
            }
        }

        i=0;
758
        while(!ret) {
Jacek Caban's avatar
Jacek Caban committed
759
            if(mime_filters[i].filter(buf, cbSize))
760 761
                ret = mime_filters[i].mime;
            i++;
762 763
        }

Jacek Caban's avatar
Jacek Caban committed
764 765 766 767 768
        TRACE("found %s for data\n"
              "%02x %02x %02x %02x  %02x %02x %02x %02x  %02x %02x %02x %02x  %02x %02x %02x %02x\n",
              debugstr_w(ret), buf[0],buf[1],buf[2],buf[3], buf[4],buf[5],buf[6],buf[7],
              buf[8],buf[9],buf[10],buf[11], buf[12],buf[13],buf[14],buf[15]);

769 770 771 772 773
        if(pwzMimeProposed) {
            if(i == sizeof(mime_filters)/sizeof(*mime_filters))
                ret = pwzMimeProposed;

            /* text/html is a special case */
774
            if(!strcmpW(pwzMimeProposed, wszTextHtml) && !strcmpW(ret, wszTextPlain))
775 776 777
                ret = wszTextHtml;
        }

778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798
        len = strlenW(ret)+1;
        *ppwzMimeOut = CoTaskMemAlloc(len*sizeof(WCHAR));
        memcpy(*ppwzMimeOut, ret, len*sizeof(WCHAR));
        return S_OK;
    }

    if(pwzUrl) {
        HKEY hkey;
        DWORD res, size;
        LPCWSTR ptr;
        WCHAR mime[64];

        static const WCHAR wszContentType[] =
                {'C','o','n','t','e','n','t',' ','T','y','p','e','\0'};

        ptr = strrchrW(pwzUrl, '.');
        if(!ptr)
            return E_FAIL;

        res = RegOpenKeyW(HKEY_CLASSES_ROOT, ptr, &hkey);
        if(res != ERROR_SUCCESS)
799
            return HRESULT_FROM_WIN32(res);
800 801 802 803 804

        size = sizeof(mime);
        res = RegQueryValueExW(hkey, wszContentType, NULL, NULL, (LPBYTE)mime, &size);
        RegCloseKey(hkey);
        if(res != ERROR_SUCCESS)
805
            return HRESULT_FROM_WIN32(res);
806 807 808 809 810 811 812 813

        *ppwzMimeOut = CoTaskMemAlloc(size);
        memcpy(*ppwzMimeOut, mime, size);
        return S_OK;
    }

    return E_FAIL;
}
814

815 816 817 818 819 820 821 822 823 824 825 826 827 828 829
/***********************************************************************
 *           GetClassFileOrMime (URLMON.@)
 *
 * Determines the class ID from the bind context, file name or MIME type.
 */
HRESULT WINAPI GetClassFileOrMime(LPBC pBC, LPCWSTR pszFilename,
        LPVOID pBuffer, DWORD cbBuffer, LPCWSTR pszMimeType, DWORD dwReserved,
        CLSID *pclsid)
{
    FIXME("(%p, %s, %p, %d, %p, 0x%08x, %p): stub\n", pBC,
        debugstr_w(pszFilename), pBuffer, cbBuffer, debugstr_w(pszMimeType),
        dwReserved, pclsid);
    return E_NOTIMPL;
}

830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845
/***********************************************************************
 * Extract (URLMON.@)
 */
HRESULT WINAPI Extract(void *dest, LPCSTR szCabName)
{
    HRESULT (WINAPI *pExtract)(void *, LPCSTR);

    if (!hCabinet)
        hCabinet = LoadLibraryA("cabinet.dll");

    if (!hCabinet) return HRESULT_FROM_WIN32(GetLastError());
    pExtract = (void *)GetProcAddress(hCabinet, "Extract");
    if (!pExtract) return HRESULT_FROM_WIN32(GetLastError());

    return pExtract(dest, szCabName);
}
846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863

/***********************************************************************
 *           IsLoggingEnabledA (URLMON.@)
 */
BOOL WINAPI IsLoggingEnabledA(LPCSTR url)
{
    FIXME("(%s)\n", debugstr_a(url));
    return FALSE;
}

/***********************************************************************
 *           IsLoggingEnabledW (URLMON.@)
 */
BOOL WINAPI IsLoggingEnabledW(LPCWSTR url)
{
    FIXME("(%s)\n", debugstr_w(url));
    return FALSE;
}