*/
#include "urlmon_main.h"
-#include <wine/debug.h>
-
-WINE_DEFAULT_DEBUG_CHANNEL(urlmon);
typedef struct {
IInternetProtocol IInternetProtocol_iface;
*ppobj = &ret->IInternetProtocol_iface;
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_url(const WCHAR *url, WCHAR **ret)
+{
+ const WCHAR *ptr;
+ DWORD res, size;
+ WCHAR mime[64];
+ HKEY hkey;
+
+ static const WCHAR content_typeW[] = {'C','o','n','t','e','n','t',' ','T','y','p','e','\0'};
+
+ ptr = strrchrW(url, '.');
+ 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, content_typeW, NULL, NULL, (LPBYTE)mime, &size);
+ RegCloseKey(hkey);
+ if(res != ERROR_SUCCESS)
+ return HRESULT_FROM_WIN32(res);
+
+ TRACE("found MIME %s\n", debugstr_w(mime));
+
+ *ret = CoTaskMemAlloc(size);
+ memcpy(*ret, mime, size);
+ return S_OK;
+}
+
+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 {
+ const WCHAR *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}
+};
+
+static BOOL is_known_mime_type(const WCHAR *mime)
+{
+ unsigned i;
+
+ for(i=0; i < sizeof(mime_filters_any_pos)/sizeof(*mime_filters_any_pos); i++) {
+ if(!strcmpW(mime, mime_filters_any_pos[i].mime))
+ return TRUE;
+ }
+
+ for(i=0; i < sizeof(mime_filters)/sizeof(*mime_filters); i++) {
+ if(!strcmpW(mime, mime_filters[i].mime))
+ return TRUE;
+ }
+
+ return FALSE;
+}
+
+static HRESULT find_mime_from_buffer(const BYTE *buf, DWORD size, const WCHAR *proposed_mime, const WCHAR *url, WCHAR **ret_mime)
+{
+ int len, i, any_pos_mime = -1;
+ const WCHAR *ret = NULL;
+
+ 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;
+ }
+
+ if(url && (ret == app_octetstreamW || ret == text_plainW)) {
+ WCHAR *url_mime;
+ HRESULT hres;
+
+ hres = find_mime_from_url(url, &url_mime);
+ if(SUCCEEDED(hres)) {
+ if(!is_known_mime_type(url_mime)) {
+ *ret_mime = url_mime;
+ return hres;
+ }
+ CoTaskMemFree(url_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, pwzUrl, ppwzMimeOut);
+
+ if(pwzUrl)
+ return find_mime_from_url(pwzUrl, ppwzMimeOut);
+
+ return E_FAIL;
+}