* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
*/
-//#include <stdarg.h>
-
#include "urlmon_main.h"
-//#include "winreg.h"
+#include <ole2.h>
-#define NO_SHLWAPI_REG
-//#include "shlwapi.h"
-#include <advpub.h>
#include <initguid.h>
-
-#include <wine/debug.h>
-
-#include "urlmon.h"
-
-WINE_DEFAULT_DEBUG_CHANNEL(urlmon);
-
DEFINE_GUID(CLSID_CUri, 0xDF2FCE13, 0x25EC, 0x45BB, 0x9D,0x4C, 0xCE,0xCD,0x47,0xC2,0x43,0x0C);
LONG URLMON_refCount = 0;
break;
case DLL_PROCESS_DETACH:
+ if (fImpLoad) break;
process_detach();
DeleteCriticalSection(&tls_cs);
break;
return S_OK;
}
-static BOOL text_richtext_filter(const BYTE *b, DWORD size)
-{
- return size > 5 && !memcmp(b, "{\\rtf", 5);
-}
-
-static BOOL text_html_filter(const BYTE *b, DWORD size)
-{
- if(size < 6)
- return FALSE;
-
- if((b[0] == '<'
- && (b[1] == 'h' || b[1] == 'H')
- && (b[2] == 't' || b[2] == 'T')
- && (b[3] == 'm' || b[3] == 'M')
- && (b[4] == 'l' || b[4] == 'L'))
- || (b[0] == '<'
- && (b[1] == 'h' || b[1] == 'H')
- && (b[2] == 'e' || b[2] == 'E')
- && (b[3] == 'a' || b[3] == 'A')
- && (b[4] == 'd' || b[4] == 'D'))) return TRUE;
-
- return FALSE;
-}
-
-static BOOL text_xml_filter(const BYTE *b, DWORD size)
-{
- if(size < 7)
- return FALSE;
-
- if(b[0] == '<' && b[1] == '?'
- && (b[2] == 'x' || b[2] == 'X')
- && (b[3] == 'm' || b[3] == 'M')
- && (b[4] == 'l' || b[4] == 'L')
- && b[5] == ' ') return TRUE;
-
- return FALSE;
-}
-
-static BOOL audio_basic_filter(const BYTE *b, DWORD size)
-{
- return size > 4
- && b[0] == '.' && b[1] == 's' && b[2] == 'n' && b[3] == 'd';
-}
-
-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';
-}
-
-static BOOL image_gif_filter(const BYTE *b, DWORD size)
-{
- 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');
-}
-
-static BOOL image_pjpeg_filter(const BYTE *b, DWORD size)
-{
- return size > 2 && b[0] == 0xff && b[1] == 0xd8;
-}
-
-static BOOL image_tiff_filter(const BYTE *b, DWORD size)
-{
- static const BYTE magic1[] = {0x4d,0x4d,0x00,0x2a};
- static const BYTE magic2[] = {0x49,0x49,0x2a,0xff};
-
- return size >= 4 && (!memcmp(b, magic1, 4) || !memcmp(b, magic2, 4));
-}
-
-static BOOL image_xpng_filter(const BYTE *b, DWORD size)
-{
- static const BYTE xpng_header[] = {0x89,'P','N','G',0x0d,0x0a,0x1a,0x0a};
- return size > sizeof(xpng_header) && !memcmp(b, xpng_header, sizeof(xpng_header));
-}
-
-static BOOL image_bmp_filter(const BYTE *b, DWORD size)
-{
- return size >= 14
- && b[0] == 0x42 && b[1] == 0x4d
- && *(const DWORD *)(b+6) == 0;
-}
-
-static BOOL video_avi_filter(const BYTE *b, DWORD size)
-{
- 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;
-}
-
-static BOOL video_mpeg_filter(const BYTE *b, DWORD size)
-{
- return size > 4
- && !b[0] && !b[1] && b[2] == 0x01
- && (b[3] == 0xb3 || b[3] == 0xba);
-}
-
-static BOOL application_postscript_filter(const BYTE *b, DWORD size)
-{
- return size > 2 && b[0] == '%' && b[1] == '!';
-}
-
-static BOOL application_pdf_filter(const BYTE *b, DWORD size)
-{
- return size > 4 && b[0] == 0x25 && b[1] == 0x50 && b[2] == 0x44 && b[3] == 0x46;
-}
-
-static BOOL application_xzip_filter(const BYTE *b, DWORD size)
-{
- return size > 2 && b[0] == 0x50 && b[1] == 0x4b;
-}
-
-static BOOL application_xgzip_filter(const BYTE *b, DWORD size)
-{
- return size > 2 && b[0] == 0x1f && b[1] == 0x8b;
-}
-
-static BOOL application_java_filter(const BYTE *b, DWORD size)
-{
- return size > 4 && b[0] == 0xca && b[1] == 0xfe && b[2] == 0xba && b[3] == 0xbe;
-}
-
-static BOOL application_xmsdownload(const BYTE *b, DWORD size)
-{
- return size > 2 && b[0] == 'M' && b[1] == 'Z';
-}
-
-static inline BOOL is_text_plain_char(BYTE b)
-{
- if(b < 0x20 && b != '\n' && b != '\r' && b != '\t')
- return FALSE;
- return TRUE;
-}
-
-static BOOL text_plain_filter(const BYTE *b, DWORD size)
-{
- const BYTE *ptr;
-
- for(ptr = b; ptr < b+size-1; ptr++) {
- if(!is_text_plain_char(*ptr))
- return FALSE;
- }
-
- return TRUE;
-}
-
-static BOOL application_octet_stream_filter(const BYTE *b, DWORD size)
-{
- return TRUE;
-}
-
-static HRESULT find_mime_from_buffer(const BYTE *buf, DWORD size, const WCHAR *proposed_mime, WCHAR **ret_mime)
-{
- LPCWSTR ret = NULL;
- int len, i, any_pos_mime = -1;
-
- static const WCHAR text_htmlW[] = {'t','e','x','t','/','h','t','m','l',0};
- static const WCHAR text_richtextW[] = {'t','e','x','t','/','r','i','c','h','t','e','x','t',0};
- static const WCHAR text_xmlW[] = {'t','e','x','t','/','x','m','l',0};
- static const WCHAR audio_basicW[] = {'a','u','d','i','o','/','b','a','s','i','c',0};
- static const WCHAR audio_wavW[] = {'a','u','d','i','o','/','w','a','v',0};
- static const WCHAR image_gifW[] = {'i','m','a','g','e','/','g','i','f',0};
- static const WCHAR image_pjpegW[] = {'i','m','a','g','e','/','p','j','p','e','g',0};
- static const WCHAR image_tiffW[] = {'i','m','a','g','e','/','t','i','f','f',0};
- static const WCHAR image_xpngW[] = {'i','m','a','g','e','/','x','-','p','n','g',0};
- static const WCHAR image_bmpW[] = {'i','m','a','g','e','/','b','m','p',0};
- static const WCHAR video_aviW[] = {'v','i','d','e','o','/','a','v','i',0};
- static const WCHAR video_mpegW[] = {'v','i','d','e','o','/','m','p','e','g',0};
- static const WCHAR app_postscriptW[] =
- {'a','p','p','l','i','c','a','t','i','o','n','/','p','o','s','t','s','c','r','i','p','t',0};
- static const WCHAR app_pdfW[] = {'a','p','p','l','i','c','a','t','i','o','n','/','p','d','f',0};
- static const WCHAR app_xzipW[] = {'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};
- static const WCHAR app_xgzipW[] = {'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};
- static const WCHAR app_javaW[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
- 'j','a','v','a',0};
- static const WCHAR app_xmsdownloadW[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
- 'x','-','m','s','d','o','w','n','l','o','a','d',0};
- static const WCHAR text_plainW[] = {'t','e','x','t','/','p','l','a','i','n','\0'};
- static const WCHAR app_octetstreamW[] = {'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;
- BOOL (*filter)(const BYTE *,DWORD);
- } mime_filters_any_pos[] = {
- {text_htmlW, text_html_filter},
- {text_xmlW, text_xml_filter}
- }, mime_filters[] = {
- {text_richtextW, text_richtext_filter},
- /* {audio_xaiffW, audio_xaiff_filter}, */
- {audio_basicW, audio_basic_filter},
- {audio_wavW, audio_wav_filter},
- {image_gifW, image_gif_filter},
- {image_pjpegW, image_pjpeg_filter},
- {image_tiffW, image_tiff_filter},
- {image_xpngW, image_xpng_filter},
- /* {image_xbitmapW, image_xbitmap_filter}, */
- {image_bmpW, image_bmp_filter},
- /* {image_xjgW, image_xjg_filter}, */
- /* {image_xemfW, image_xemf_filter}, */
- /* {image_xwmfW, image_xwmf_filter}, */
- {video_aviW, video_avi_filter},
- {video_mpegW, video_mpeg_filter},
- {app_postscriptW, application_postscript_filter},
- /* {app_base64W, application_base64_filter}, */
- /* {app_macbinhex40W, application_macbinhex40_filter}, */
- {app_pdfW, application_pdf_filter},
- /* {app_zcompressedW, application_xcompressed_filter}, */
- {app_xzipW, application_xzip_filter},
- {app_xgzipW, application_xgzip_filter},
- {app_javaW, application_java_filter},
- {app_xmsdownloadW, application_xmsdownload},
- {text_plainW, text_plain_filter},
- {app_octetstreamW, application_octet_stream_filter}
- };
-
- if(!buf || !size) {
- if(!proposed_mime)
- return E_FAIL;
-
- len = strlenW(proposed_mime)+1;
- *ret_mime = CoTaskMemAlloc(len*sizeof(WCHAR));
- if(!*ret_mime)
- return E_OUTOFMEMORY;
-
- memcpy(*ret_mime, proposed_mime, len*sizeof(WCHAR));
- return S_OK;
- }
-
- if(proposed_mime && (!strcmpW(proposed_mime, app_octetstreamW)
- || !strcmpW(proposed_mime, text_plainW)))
- proposed_mime = NULL;
-
- if(proposed_mime) {
- ret = proposed_mime;
-
- for(i=0; i < sizeof(mime_filters_any_pos)/sizeof(*mime_filters_any_pos); i++) {
- if(!strcmpW(proposed_mime, mime_filters_any_pos[i].mime)) {
- any_pos_mime = i;
- for(len=size; len>0; len--) {
- if(mime_filters_any_pos[i].filter(buf+size-len, len))
- break;
- }
- if(!len)
- ret = NULL;
- break;
- }
- }
-
- if(i == sizeof(mime_filters_any_pos)/sizeof(*mime_filters_any_pos)) {
- for(i=0; i < sizeof(mime_filters)/sizeof(*mime_filters); i++) {
- if(!strcmpW(proposed_mime, mime_filters[i].mime)) {
- if(!mime_filters[i].filter(buf, size))
- ret = NULL;
- break;
- }
- }
- }
- }
-
- /* Looks like a bug in native implementation, html and xml mimes
- * are not looked for if none of them was proposed */
- if(!proposed_mime || any_pos_mime!=-1) {
- for(len=size; !ret && len>0; len--) {
- for(i=0; i<sizeof(mime_filters_any_pos)/sizeof(*mime_filters_any_pos); i++) {
- if(mime_filters_any_pos[i].filter(buf+size-len, len)) {
- ret = mime_filters_any_pos[i].mime;
- break;
- }
- }
- }
- }
-
- i=0;
- while(!ret) {
- if(mime_filters[i].filter(buf, size))
- ret = mime_filters[i].mime;
- i++;
- }
-
- if(any_pos_mime!=-1 && ret==text_plainW)
- ret = mime_filters_any_pos[any_pos_mime].mime;
- else if(proposed_mime && ret==app_octetstreamW) {
- for(len=size; ret==app_octetstreamW && len>0; len--) {
- if(!is_text_plain_char(buf[size-len]))
- break;
- for(i=0; i<sizeof(mime_filters_any_pos)/sizeof(*mime_filters_any_pos); i++) {
- if(mime_filters_any_pos[i].filter(buf+size-len, len)) {
- ret = text_plainW;
- break;
- }
- }
- }
-
- if(ret == app_octetstreamW)
- ret = proposed_mime;
- }
- TRACE("found %s for %s\n", debugstr_w(ret), debugstr_an((const char*)buf, min(32, size)));
-
- len = strlenW(ret)+1;
- *ret_mime = CoTaskMemAlloc(len*sizeof(WCHAR));
- if(!*ret_mime)
- return E_OUTOFMEMORY;
-
- memcpy(*ret_mime, ret, len*sizeof(WCHAR));
- return S_OK;
-}
-
-/***********************************************************************
- * FindMimeFromData (URLMON.@)
- *
- * Determines the Multipurpose Internet Mail Extensions (MIME) type from the data provided.
- */
-HRESULT WINAPI FindMimeFromData(LPBC pBC, LPCWSTR pwzUrl, LPVOID pBuffer,
- DWORD cbSize, LPCWSTR pwzMimeProposed, DWORD dwMimeFlags,
- LPWSTR* ppwzMimeOut, DWORD dwReserved)
-{
- TRACE("(%p,%s,%p,%d,%s,0x%x,%p,0x%x)\n", pBC, debugstr_w(pwzUrl), pBuffer, cbSize,
- debugstr_w(pwzMimeProposed), dwMimeFlags, ppwzMimeOut, dwReserved);
-
- if(dwMimeFlags)
- WARN("dwMimeFlags=%08x\n", dwMimeFlags);
- if(dwReserved)
- WARN("dwReserved=%d\n", dwReserved);
-
- /* pBC seams to not be used */
-
- if(!ppwzMimeOut || (!pwzUrl && !pBuffer))
- return E_INVALIDARG;
-
- if(pwzMimeProposed || pBuffer)
- return find_mime_from_buffer(pBuffer, cbSize, pwzMimeProposed, ppwzMimeOut);
-
- 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)
- return HRESULT_FROM_WIN32(res);
-
- size = sizeof(mime);
- res = RegQueryValueExW(hkey, wszContentType, NULL, NULL, (LPBYTE)mime, &size);
- RegCloseKey(hkey);
- if(res != ERROR_SUCCESS)
- return HRESULT_FROM_WIN32(res);
-
- *ppwzMimeOut = CoTaskMemAlloc(size);
- memcpy(*ppwzMimeOut, mime, size);
- return S_OK;
- }
-
- return E_FAIL;
-}
-
/***********************************************************************
* GetClassFileOrMime (URLMON.@)
*