Sweden-Number/dlls/urlmon/mimefilter.c

677 lines
20 KiB
C
Raw Normal View History

/*
* Copyright 2009 Jacek Caban for CodeWeavers
*
* 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
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
*/
#include "urlmon_main.h"
#include "wine/debug.h"
WINE_DEFAULT_DEBUG_CHANNEL(urlmon);
typedef struct {
IInternetProtocol IInternetProtocol_iface;
IInternetProtocolSink IInternetProtocolSink_iface;
LONG ref;
} MimeFilter;
static inline MimeFilter *impl_from_IInternetProtocol(IInternetProtocol *iface)
{
return CONTAINING_RECORD(iface, MimeFilter, IInternetProtocol_iface);
}
static HRESULT WINAPI MimeFilterProtocol_QueryInterface(IInternetProtocol *iface, REFIID riid, void **ppv)
{
MimeFilter *This = impl_from_IInternetProtocol(iface);
*ppv = NULL;
if(IsEqualGUID(&IID_IUnknown, riid)) {
TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
*ppv = &This->IInternetProtocol_iface;
}else if(IsEqualGUID(&IID_IInternetProtocolRoot, riid)) {
TRACE("(%p)->(IID_IInternetProtocolRoot %p)\n", This, ppv);
*ppv = &This->IInternetProtocol_iface;
}else if(IsEqualGUID(&IID_IInternetProtocol, riid)) {
TRACE("(%p)->(IID_IInternetProtocol %p)\n", This, ppv);
*ppv = &This->IInternetProtocol_iface;
}else if(IsEqualGUID(&IID_IInternetProtocolSink, riid)) {
TRACE("(%p)->(IID_IInternetProtocolSink %p)\n", This, ppv);
*ppv = &This->IInternetProtocolSink_iface;
}
if(*ppv) {
IInternetProtocol_AddRef(iface);
return S_OK;
}
WARN("not supported interface %s\n", debugstr_guid(riid));
return E_NOINTERFACE;
}
static ULONG WINAPI MimeFilterProtocol_AddRef(IInternetProtocol *iface)
{
MimeFilter *This = impl_from_IInternetProtocol(iface);
LONG ref = InterlockedIncrement(&This->ref);
TRACE("(%p) ref=%ld\n", This, ref);
return ref;
}
static ULONG WINAPI MimeFilterProtocol_Release(IInternetProtocol *iface)
{
MimeFilter *This = impl_from_IInternetProtocol(iface);
LONG ref = InterlockedDecrement(&This->ref);
TRACE("(%p) ref=%ld\n", This, ref);
if(!ref) {
heap_free(This);
URLMON_UnlockModule();
}
return ref;
}
static HRESULT WINAPI MimeFilterProtocol_Start(IInternetProtocol *iface, LPCWSTR szUrl,
IInternetProtocolSink *pOIProtSink, IInternetBindInfo *pOIBindInfo,
DWORD grfPI, HANDLE_PTR dwReserved)
{
MimeFilter *This = impl_from_IInternetProtocol(iface);
FIXME("(%p)->(%s %p %p %08lx %Ix)\n", This, debugstr_w(szUrl), pOIProtSink,
pOIBindInfo, grfPI, dwReserved);
return E_NOTIMPL;
}
static HRESULT WINAPI MimeFilterProtocol_Continue(IInternetProtocol *iface, PROTOCOLDATA *pProtocolData)
{
MimeFilter *This = impl_from_IInternetProtocol(iface);
FIXME("(%p)->(%p)\n", This, pProtocolData);
return E_NOTIMPL;
}
static HRESULT WINAPI MimeFilterProtocol_Abort(IInternetProtocol *iface, HRESULT hrReason,
DWORD dwOptions)
{
MimeFilter *This = impl_from_IInternetProtocol(iface);
FIXME("(%p)->(%08lx %08lx)\n", This, hrReason, dwOptions);
return E_NOTIMPL;
}
static HRESULT WINAPI MimeFilterProtocol_Terminate(IInternetProtocol *iface, DWORD dwOptions)
{
MimeFilter *This = impl_from_IInternetProtocol(iface);
FIXME("(%p)->(%08lx)\n", This, dwOptions);
return E_NOTIMPL;
}
static HRESULT WINAPI MimeFilterProtocol_Suspend(IInternetProtocol *iface)
{
MimeFilter *This = impl_from_IInternetProtocol(iface);
FIXME("(%p)\n", This);
return E_NOTIMPL;
}
static HRESULT WINAPI MimeFilterProtocol_Resume(IInternetProtocol *iface)
{
MimeFilter *This = impl_from_IInternetProtocol(iface);
FIXME("(%p)\n", This);
return E_NOTIMPL;
}
static HRESULT WINAPI MimeFilterProtocol_Read(IInternetProtocol *iface, void *pv,
ULONG cb, ULONG *pcbRead)
{
MimeFilter *This = impl_from_IInternetProtocol(iface);
FIXME("(%p)->(%p %lu %p)\n", This, pv, cb, pcbRead);
return E_NOTIMPL;
}
static HRESULT WINAPI MimeFilterProtocol_Seek(IInternetProtocol *iface, LARGE_INTEGER dlibMove,
DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition)
{
MimeFilter *This = impl_from_IInternetProtocol(iface);
FIXME("(%p)->(%ld %ld %p)\n", This, dlibMove.u.LowPart, dwOrigin, plibNewPosition);
return E_NOTIMPL;
}
static HRESULT WINAPI MimeFilterProtocol_LockRequest(IInternetProtocol *iface, DWORD dwOptions)
{
MimeFilter *This = impl_from_IInternetProtocol(iface);
FIXME("(%p)->(%08lx)\n", This, dwOptions);
return E_NOTIMPL;
}
static HRESULT WINAPI MimeFilterProtocol_UnlockRequest(IInternetProtocol *iface)
{
MimeFilter *This = impl_from_IInternetProtocol(iface);
FIXME("(%p)\n", This);
return E_NOTIMPL;
}
static const IInternetProtocolVtbl MimeFilterProtocolVtbl = {
MimeFilterProtocol_QueryInterface,
MimeFilterProtocol_AddRef,
MimeFilterProtocol_Release,
MimeFilterProtocol_Start,
MimeFilterProtocol_Continue,
MimeFilterProtocol_Abort,
MimeFilterProtocol_Terminate,
MimeFilterProtocol_Suspend,
MimeFilterProtocol_Resume,
MimeFilterProtocol_Read,
MimeFilterProtocol_Seek,
MimeFilterProtocol_LockRequest,
MimeFilterProtocol_UnlockRequest
};
static inline MimeFilter *impl_from_IInternetProtocolSink(IInternetProtocolSink *iface)
{
return CONTAINING_RECORD(iface, MimeFilter, IInternetProtocolSink_iface);
}
static HRESULT WINAPI MimeFilterSink_QueryInterface(IInternetProtocolSink *iface,
REFIID riid, void **ppv)
{
MimeFilter *This = impl_from_IInternetProtocolSink(iface);
return IInternetProtocol_QueryInterface(&This->IInternetProtocol_iface, riid, ppv);
}
static ULONG WINAPI MimeFilterSink_AddRef(IInternetProtocolSink *iface)
{
MimeFilter *This = impl_from_IInternetProtocolSink(iface);
return IInternetProtocol_AddRef(&This->IInternetProtocol_iface);
}
static ULONG WINAPI MimeFilterSink_Release(IInternetProtocolSink *iface)
{
MimeFilter *This = impl_from_IInternetProtocolSink(iface);
return IInternetProtocol_Release(&This->IInternetProtocol_iface);
}
static HRESULT WINAPI MimeFilterSink_Switch(IInternetProtocolSink *iface,
PROTOCOLDATA *pProtocolData)
{
MimeFilter *This = impl_from_IInternetProtocolSink(iface);
FIXME("(%p)->(%p)\n", This, pProtocolData);
return E_NOTIMPL;
}
static HRESULT WINAPI MimeFilterSink_ReportProgress(IInternetProtocolSink *iface,
ULONG ulStatusCode, LPCWSTR szStatusText)
{
MimeFilter *This = impl_from_IInternetProtocolSink(iface);
FIXME("(%p)->(%lu %s)\n", This, ulStatusCode, debugstr_w(szStatusText));
return E_NOTIMPL;
}
static HRESULT WINAPI MimeFilterSink_ReportData(IInternetProtocolSink *iface,
DWORD grfBSCF, ULONG ulProgress, ULONG ulProgressMax)
{
MimeFilter *This = impl_from_IInternetProtocolSink(iface);
FIXME("(%p)->(%ld %lu %lu)\n", This, grfBSCF, ulProgress, ulProgressMax);
return E_NOTIMPL;
}
static HRESULT WINAPI MimeFilterSink_ReportResult(IInternetProtocolSink *iface,
HRESULT hrResult, DWORD dwError, LPCWSTR szResult)
{
MimeFilter *This = impl_from_IInternetProtocolSink(iface);
FIXME("(%p)->(%08lx %ld %s)\n", This, hrResult, dwError, debugstr_w(szResult));
return E_NOTIMPL;
}
static const IInternetProtocolSinkVtbl InternetProtocolSinkVtbl = {
MimeFilterSink_QueryInterface,
MimeFilterSink_AddRef,
MimeFilterSink_Release,
MimeFilterSink_Switch,
MimeFilterSink_ReportProgress,
MimeFilterSink_ReportData,
MimeFilterSink_ReportResult
};
HRESULT MimeFilter_Construct(IUnknown *pUnkOuter, LPVOID *ppobj)
{
MimeFilter *ret;
TRACE("(%p %p)\n", pUnkOuter, ppobj);
URLMON_LockModule();
ret = heap_alloc_zero(sizeof(MimeFilter));
ret->IInternetProtocol_iface.lpVtbl = &MimeFilterProtocolVtbl;
ret->IInternetProtocolSink_iface.lpVtbl = &InternetProtocolSinkVtbl;
ret->ref = 1;
*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'))
|| (b[0] == '<'
&& (b[1] == 'b' || b[1] == 'B')
&& (b[2] == 'o' || b[2] == 'O')
&& (b[3] == 'd' || b[3] == 'D')
&& (b[4] == 'y' || b[4] == 'Y'))) 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;
}
HRESULT find_mime_from_ext(const WCHAR *ext, WCHAR **ret)
{
DWORD res, size;
WCHAR mime[64];
HKEY hkey;
res = RegOpenKeyW(HKEY_CLASSES_ROOT, ext, &hkey);
if(res != ERROR_SUCCESS)
return HRESULT_FROM_WIN32(res);
size = sizeof(mime);
res = RegQueryValueExW(hkey, L"Content Type", NULL, NULL, (BYTE*)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 HRESULT find_mime_from_url(const WCHAR *url, WCHAR **ret)
{
const WCHAR *ptr, *end_ptr;
WCHAR *ext = NULL;
HRESULT hres;
for(end_ptr = url; *end_ptr; end_ptr++) {
if(*end_ptr == '?' || *end_ptr == '#')
break;
}
for(ptr = end_ptr; ptr >= url; ptr--) {
if(*ptr == '.')
break;
}
if(ptr < url)
return E_FAIL;
if(*end_ptr) {
unsigned len = end_ptr-ptr;
ext = heap_alloc((len+1)*sizeof(WCHAR));
if(!ext)
return E_OUTOFMEMORY;
memcpy(ext, ptr, len*sizeof(WCHAR));
ext[len] = 0;
}
hres = find_mime_from_ext(ext ? ext : ptr, ret);
heap_free(ext);
return hres;
}
static const WCHAR text_plainW[] = L"text/plain";
static const WCHAR app_octetstreamW[] = L"application/octet-stream";
static const struct {
const WCHAR *mime;
BOOL (*filter)(const BYTE *,DWORD);
} mime_filters_any_pos[] = {
{L"text/html", text_html_filter},
{L"text/xml", text_xml_filter}
}, mime_filters[] = {
{L"text/richtext", text_richtext_filter},
/* {L"audio/x-aiff", audio_xaiff_filter}, */
{L"audio/basic", audio_basic_filter},
{L"audio/wav", audio_wav_filter},
{L"image/gif", image_gif_filter},
{L"image/pjpeg", image_pjpeg_filter},
{L"image/tiff", image_tiff_filter},
{L"image/x-png", image_xpng_filter},
/* {L"image/x-bitmap", image_xbitmap_filter}, */
{L"image/bmp", image_bmp_filter},
/* {L"image/x-jg", image_xjg_filter}, */
/* {L"image/x-emf", image_xemf_filter}, */
/* {L"image/x-wmf", image_xwmf_filter}, */
{L"video/avi", video_avi_filter},
{L"video/mpeg", video_mpeg_filter},
{L"application/postscript", application_postscript_filter},
/* {L"application/base64", application_base64_filter}, */
/* {L"application/mac-binhex40", application_macbinhex40_filter}, */
{L"application/pdf", application_pdf_filter},
/* {L"application/x-compressed", application_xcompressed_filter}, */
{L"application/x-zip-compressed", application_xzip_filter},
{L"application/x-gzip-compressed", application_xgzip_filter},
{L"application/java", application_java_filter},
{L"application/x-msdownload", 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 < ARRAY_SIZE(mime_filters_any_pos); i++) {
if(!wcscmp(mime, mime_filters_any_pos[i].mime))
return TRUE;
}
for(i=0; i < ARRAY_SIZE(mime_filters); i++) {
if(!wcscmp(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 = lstrlenW(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 && (!wcscmp(proposed_mime, app_octetstreamW)
|| !wcscmp(proposed_mime, text_plainW)))
proposed_mime = NULL;
if(proposed_mime) {
ret = proposed_mime;
for(i=0; i < ARRAY_SIZE(mime_filters_any_pos); i++) {
if(!wcscmp(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 == ARRAY_SIZE(mime_filters_any_pos)) {
for(i=0; i < ARRAY_SIZE(mime_filters); i++) {
if(!wcscmp(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<ARRAY_SIZE(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<ARRAY_SIZE(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 = lstrlenW(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,%ld,%s,0x%lx,%p,0x%lx)\n", pBC, debugstr_w(pwzUrl), pBuffer, cbSize,
debugstr_w(pwzMimeProposed), dwMimeFlags, ppwzMimeOut, dwReserved);
if(dwMimeFlags)
WARN("dwMimeFlags=%08lx\n", dwMimeFlags);
if(dwReserved)
WARN("dwReserved=%ld\n", dwReserved);
/* pBC seems to not be used */
if(!ppwzMimeOut || (!pwzUrl && !pBuffer))
return E_INVALIDARG;
if(pwzMimeProposed || pBuffer)
return find_mime_from_buffer(pBuffer, cbSize, pwzMimeProposed, pwzUrl, ppwzMimeOut);
return find_mime_from_url(pwzUrl, ppwzMimeOut);
}