/* * MimeOle tests * * Copyright 2007 Huw Davies * * 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 */ #define COBJMACROS #define NONAMELESSUNION #include "initguid.h" #include "windows.h" #include "ole2.h" #include "ocidl.h" #include "mimeole.h" #include "wininet.h" #include #include "wine/test.h" #define DEFINE_EXPECT(func) \ static BOOL expect_ ## func = FALSE, called_ ## func = FALSE #define SET_EXPECT(func) \ expect_ ## func = TRUE #define CHECK_EXPECT(func) \ do { \ ok(expect_ ##func, "unexpected call " #func "\n"); \ expect_ ## func = FALSE; \ called_ ## func = TRUE; \ }while(0) #define CHECK_EXPECT2(func) \ do { \ ok(expect_ ##func, "unexpected call " #func "\n"); \ called_ ## func = TRUE; \ }while(0) #define CHECK_CALLED(func) \ do { \ ok(called_ ## func, "expected " #func "\n"); \ expect_ ## func = called_ ## func = FALSE; \ }while(0) DEFINE_EXPECT(Stream_Read); DEFINE_EXPECT(Stream_Stat); DEFINE_EXPECT(Stream_Seek); DEFINE_EXPECT(Stream_Seek_END); DEFINE_EXPECT(GetBindInfo); DEFINE_EXPECT(ReportProgress_MIMETYPEAVAILABLE); DEFINE_EXPECT(ReportProgress_CACHEFILENAMEAVAILABLE); DEFINE_EXPECT(ReportData); DEFINE_EXPECT(ReportResult); static const char msg1[] = "MIME-Version: 1.0\r\n" "Content-Type: multipart/mixed;\r\n" " boundary=\"------------1.5.0.6\";\r\n" " stuff=\"du;nno\";\r\n" " morestuff=\"so\\\\me\\\"thing\\\"\"\r\n" "foo: bar\r\n" "From: Huw Davies \r\n" "From: Me \r\n" "To: wine-patches \r\n" "Cc: Huw Davies ,\r\n" " \"Fred Bloggs\" \r\n" "foo: baz\r\n" "bar: fum\r\n" "\r\n" "This is a multi-part message in MIME format.\r\n" "--------------1.5.0.6\r\n" "Content-Type: text/plain; format=fixed; charset=UTF-8\r\n" "Content-Transfer-Encoding: 8bit\r\n" "\r\n" "Stuff\r\n" "--------------1.5.0.6\r\n" "Content-Type: text/plain; charset=\"us-ascii\"\r\n" "Content-Transfer-Encoding: 7bit\r\n" "\r\n" "More stuff\r\n" "--------------1.5.0.6--\r\n"; static const char mhtml_page1[] = "MIME-Version: 1.0\r\n" "Content-Type: multipart/related; type:=\"text/html\"; boundary=\"----=_NextPart_000_00\"\r\n" "\r\n" "------=_NextPart_000_00\r\n" "Content-Type: text/html; charset=\"Windows-1252\"\r\n" "Content-Transfer-Encoding: quoted-printable\r\n" "\r\n" "\r\n" "------=_NextPart_000_00\r\n" "Content-Type: Image/Jpeg\r\n" "Content-Transfer-Encoding: base64\r\n" "Content-Location: http://winehq.org/mhtmltest.html\r\n" "\r\n\t\t\t\tVGVzdA==\r\n\r\n" "------=_NextPart_000_00--"; static WCHAR *a2w(const char *str) { WCHAR *ret; int len; if(!str) return NULL; len = MultiByteToWideChar(CP_ACP, 0, str, -1, NULL, 0); ret = HeapAlloc(GetProcessHeap(), 0, len*sizeof(WCHAR)); MultiByteToWideChar(CP_ACP, 0, str, -1, ret, len); return ret; } static int strcmp_wa(const WCHAR *strw, const char *stra) { WCHAR buf[512]; MultiByteToWideChar(CP_ACP, 0, stra, -1, buf, ARRAY_SIZE(buf)); return lstrcmpW(strw, buf); } static void test_CreateVirtualStream(void) { HRESULT hr; IStream *pstm; hr = MimeOleCreateVirtualStream(&pstm); ok(hr == S_OK, "ret %08x\n", hr); IStream_Release(pstm); } static void test_CreateSecurity(void) { HRESULT hr; IMimeSecurity *sec; hr = MimeOleCreateSecurity(&sec); ok(hr == S_OK, "ret %08x\n", hr); IMimeSecurity_Release(sec); } static IStream *create_stream_from_string(const char *data) { LARGE_INTEGER off; IStream *stream; HRESULT hr; hr = CreateStreamOnHGlobal(NULL, TRUE, &stream); ok(hr == S_OK, "ret %08x\n", hr); hr = IStream_Write(stream, data, strlen(data), NULL); ok(hr == S_OK, "Write failed: %08x\n", hr); off.QuadPart = 0; hr = IStream_Seek(stream, off, STREAM_SEEK_SET, NULL); ok(hr == S_OK, "Seek failed: %08x\n", hr); return stream; } #define test_current_encoding(a,b) _test_current_encoding(__LINE__,a,b) static void _test_current_encoding(unsigned line, IMimeBody *mime_body, ENCODINGTYPE encoding) { ENCODINGTYPE current_encoding; HRESULT hres; hres = IMimeBody_GetCurrentEncoding(mime_body, ¤t_encoding); ok_(__FILE__,line)(hres == S_OK, "GetCurrentEncoding failed: %08x\n", hres); ok_(__FILE__,line)(current_encoding == encoding, "encoding = %d, expected %d\n", current_encoding, encoding); } static void test_CreateBody(void) { HRESULT hr; IMimeBody *body; HBODY handle = (void *)0xdeadbeef; IStream *in; LARGE_INTEGER off; ULARGE_INTEGER pos; ULONG count, found_param, i; MIMEPARAMINFO *param_info; IMimeAllocator *alloc; BODYOFFSETS offsets; CLSID clsid; hr = CoCreateInstance(&CLSID_IMimeBody, NULL, CLSCTX_INPROC_SERVER, &IID_IMimeBody, (void**)&body); ok(hr == S_OK, "ret %08x\n", hr); hr = IMimeBody_GetClassID(body, NULL); ok(hr == E_INVALIDARG, "ret %08x\n", hr); hr = IMimeBody_GetClassID(body, &clsid); ok(hr == S_OK, "ret %08x\n", hr); ok(IsEqualGUID(&clsid, &IID_IMimeBody), "got %s\n", wine_dbgstr_guid(&clsid)); hr = IMimeBody_GetHandle(body, &handle); ok(hr == MIME_E_NO_DATA, "ret %08x\n", hr); ok(handle == NULL, "handle %p\n", handle); in = create_stream_from_string(msg1); /* Need to call InitNew before Load otherwise Load crashes with native inetcomm */ hr = IMimeBody_InitNew(body); ok(hr == S_OK, "ret %08x\n", hr); test_current_encoding(body, IET_7BIT); hr = IMimeBody_Load(body, in); ok(hr == S_OK, "ret %08x\n", hr); off.QuadPart = 0; IStream_Seek(in, off, STREAM_SEEK_CUR, &pos); ok(pos.u.LowPart == 359, "pos %u\n", pos.u.LowPart); hr = IMimeBody_IsContentType(body, "multipart", "mixed"); ok(hr == S_OK, "ret %08x\n", hr); hr = IMimeBody_IsContentType(body, "text", "plain"); ok(hr == S_FALSE, "ret %08x\n", hr); hr = IMimeBody_IsContentType(body, NULL, "mixed"); ok(hr == S_OK, "ret %08x\n", hr); hr = IMimeBody_IsType(body, IBT_EMPTY); ok(hr == S_OK, "got %08x\n", hr); hr = IMimeBody_SetData(body, IET_8BIT, "text", "plain", &IID_IStream, in); ok(hr == S_OK, "ret %08x\n", hr); hr = IMimeBody_IsContentType(body, "text", "plain"); todo_wine ok(hr == S_OK, "ret %08x\n", hr); test_current_encoding(body, IET_8BIT); memset(&offsets, 0xcc, sizeof(offsets)); hr = IMimeBody_GetOffsets(body, &offsets); ok(hr == MIME_E_NO_DATA, "ret %08x\n", hr); ok(offsets.cbBoundaryStart == 0, "got %d\n", offsets.cbBoundaryStart); ok(offsets.cbHeaderStart == 0, "got %d\n", offsets.cbHeaderStart); ok(offsets.cbBodyStart == 0, "got %d\n", offsets.cbBodyStart); ok(offsets.cbBodyEnd == 0, "got %d\n", offsets.cbBodyEnd); hr = IMimeBody_IsType(body, IBT_EMPTY); ok(hr == S_FALSE, "got %08x\n", hr); hr = MimeOleGetAllocator(&alloc); ok(hr == S_OK, "ret %08x\n", hr); hr = IMimeBody_GetParameters(body, "nothere", &count, ¶m_info); ok(hr == MIME_E_NOT_FOUND, "ret %08x\n", hr); ok(count == 0, "got %d\n", count); ok(!param_info, "got %p\n", param_info); hr = IMimeBody_GetParameters(body, "bar", &count, ¶m_info); ok(hr == S_OK, "ret %08x\n", hr); ok(count == 0, "got %d\n", count); ok(!param_info, "got %p\n", param_info); hr = IMimeBody_GetParameters(body, "Content-Type", &count, ¶m_info); ok(hr == S_OK, "ret %08x\n", hr); todo_wine /* native adds a charset parameter */ ok(count == 4, "got %d\n", count); ok(param_info != NULL, "got %p\n", param_info); found_param = 0; for(i = 0; i < count; i++) { if(!strcmp(param_info[i].pszName, "morestuff")) { found_param++; ok(!strcmp(param_info[i].pszData, "so\\me\"thing\""), "got %s\n", param_info[i].pszData); } else if(!strcmp(param_info[i].pszName, "stuff")) { found_param++; ok(!strcmp(param_info[i].pszData, "du;nno"), "got %s\n", param_info[i].pszData); } } ok(found_param == 2, "matched %d params\n", found_param); hr = IMimeAllocator_FreeParamInfoArray(alloc, count, param_info, TRUE); ok(hr == S_OK, "ret %08x\n", hr); IMimeAllocator_Release(alloc); IStream_Release(in); IMimeBody_Release(body); } typedef struct { IStream IStream_iface; LONG ref; unsigned pos; } TestStream; static inline TestStream *impl_from_IStream(IStream *iface) { return CONTAINING_RECORD(iface, TestStream, IStream_iface); } static HRESULT WINAPI Stream_QueryInterface(IStream *iface, REFIID riid, void **ppv) { if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_ISequentialStream, riid) || IsEqualGUID(&IID_IStream, riid)) { *ppv = iface; return S_OK; } ok(0, "unexpected call %s\n", wine_dbgstr_guid(riid)); *ppv = NULL; return E_NOINTERFACE; } static ULONG WINAPI Stream_AddRef(IStream *iface) { TestStream *This = impl_from_IStream(iface); return InterlockedIncrement(&This->ref); } static ULONG WINAPI Stream_Release(IStream *iface) { TestStream *This = impl_from_IStream(iface); ULONG ref = InterlockedDecrement(&This->ref); if (!ref) HeapFree(GetProcessHeap(), 0, This); return ref; } static HRESULT WINAPI Stream_Read(IStream *iface, void *pv, ULONG cb, ULONG *pcbRead) { TestStream *This = impl_from_IStream(iface); BYTE *output = pv; unsigned i; CHECK_EXPECT(Stream_Read); for(i = 0; i < cb; i++) output[i] = '0' + This->pos++; *pcbRead = i; return S_OK; } static HRESULT WINAPI Stream_Write(IStream *iface, const void *pv, ULONG cb, ULONG *pcbWritten) { ok(0, "unexpected call\n"); return E_NOTIMPL; } static DWORD expect_seek_pos; static HRESULT WINAPI Stream_Seek(IStream *iface, LARGE_INTEGER dlibMove, DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition) { TestStream *This = impl_from_IStream(iface); if(dwOrigin == STREAM_SEEK_END) { CHECK_EXPECT(Stream_Seek_END); ok(dlibMove.QuadPart == expect_seek_pos, "unexpected seek pos %u\n", dlibMove.u.LowPart); if(plibNewPosition) plibNewPosition->QuadPart = 10; return S_OK; } CHECK_EXPECT(Stream_Seek); ok(dlibMove.QuadPart == expect_seek_pos, "unexpected seek pos %u\n", dlibMove.u.LowPart); ok(dwOrigin == STREAM_SEEK_SET, "dwOrigin = %d\n", dwOrigin); This->pos = dlibMove.QuadPart; if(plibNewPosition) plibNewPosition->QuadPart = This->pos; return S_OK; } static HRESULT WINAPI Stream_SetSize(IStream *iface, ULARGE_INTEGER libNewSize) { ok(0, "unexpected call\n"); return E_NOTIMPL; } static HRESULT WINAPI Stream_CopyTo(IStream *iface, IStream *pstm, ULARGE_INTEGER cb, ULARGE_INTEGER *pcbRead, ULARGE_INTEGER *pcbWritten) { ok(0, "unexpected call\n"); return E_NOTIMPL; } static HRESULT WINAPI Stream_Commit(IStream *iface, DWORD grfCommitFlags) { ok(0, "unexpected call\n"); return E_NOTIMPL; } static HRESULT WINAPI Stream_Revert(IStream *iface) { ok(0, "unexpected call\n"); return E_NOTIMPL; } static HRESULT WINAPI Stream_LockRegion(IStream *iface, ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType) { ok(0, "unexpected call\n"); return E_NOTIMPL; } static HRESULT WINAPI Stream_UnlockRegion(IStream *iface, ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType) { ok(0, "unexpected call\n"); return E_NOTIMPL; } static HRESULT WINAPI Stream_Stat(IStream *iface, STATSTG *pstatstg, DWORD dwStatFlag) { CHECK_EXPECT(Stream_Stat); ok(dwStatFlag == STATFLAG_NONAME, "dwStatFlag = %x\n", dwStatFlag); return E_NOTIMPL; } static HRESULT WINAPI Stream_Clone(IStream *iface, IStream **ppstm) { ok(0, "unexpected call\n"); return E_NOTIMPL; } static const IStreamVtbl StreamVtbl = { Stream_QueryInterface, Stream_AddRef, Stream_Release, Stream_Read, Stream_Write, Stream_Seek, Stream_SetSize, Stream_CopyTo, Stream_Commit, Stream_Revert, Stream_LockRegion, Stream_UnlockRegion, Stream_Stat, Stream_Clone }; static IStream *create_test_stream(void) { TestStream *stream; stream = HeapAlloc(GetProcessHeap(), 0, sizeof(*stream)); stream->IStream_iface.lpVtbl = &StreamVtbl; stream->ref = 1; stream->pos = 0; return &stream->IStream_iface; } #define test_stream_read(a,b,c,d) _test_stream_read(__LINE__,a,b,c,d) static void _test_stream_read(unsigned line, IStream *stream, HRESULT exhres, const char *exdata, unsigned read_size) { ULONG read = 0xdeadbeed, exread = strlen(exdata); char buf[1024]; HRESULT hres; if(read_size == -1) read_size = sizeof(buf)-1; hres = IStream_Read(stream, buf, read_size, &read); ok_(__FILE__,line)(hres == exhres, "Read returned %08x, expected %08x\n", hres, exhres); ok_(__FILE__,line)(read == exread, "unexpected read size %u, expected %u\n", read, exread); buf[read] = 0; ok_(__FILE__,line)(read == exread && !memcmp(buf, exdata, read), "unexpected data %s\n", buf); } static void test_SetData(void) { IStream *stream, *stream2, *test_stream; IMimeBody *body; HRESULT hr; hr = CoCreateInstance(&CLSID_IMimeBody, NULL, CLSCTX_INPROC_SERVER, &IID_IMimeBody, (void**)&body); ok(hr == S_OK, "ret %08x\n", hr); /* Need to call InitNew before Load otherwise Load crashes with native inetcomm */ hr = IMimeBody_InitNew(body); ok(hr == S_OK, "ret %08x\n", hr); stream = create_stream_from_string(msg1); hr = IMimeBody_Load(body, stream); ok(hr == S_OK, "ret %08x\n", hr); IStream_Release(stream); test_stream = create_test_stream(); hr = IMimeBody_SetData(body, IET_BINARY, "text", "plain", &IID_IStream, test_stream); ok(hr == S_OK, "ret %08x\n", hr); hr = IMimeBody_IsContentType(body, "text", "plain"); todo_wine ok(hr == S_OK, "ret %08x\n", hr); test_current_encoding(body, IET_BINARY); SET_EXPECT(Stream_Stat); SET_EXPECT(Stream_Seek_END); hr = IMimeBody_GetData(body, IET_BINARY, &stream); CHECK_CALLED(Stream_Stat); CHECK_CALLED(Stream_Seek_END); ok(hr == S_OK, "GetData failed %08x\n", hr); ok(stream != test_stream, "unexpected stream\n"); SET_EXPECT(Stream_Seek); SET_EXPECT(Stream_Read); test_stream_read(stream, S_OK, "012", 3); CHECK_CALLED(Stream_Seek); CHECK_CALLED(Stream_Read); SET_EXPECT(Stream_Stat); SET_EXPECT(Stream_Seek_END); hr = IMimeBody_GetData(body, IET_BINARY, &stream2); CHECK_CALLED(Stream_Stat); CHECK_CALLED(Stream_Seek_END); ok(hr == S_OK, "GetData failed %08x\n", hr); ok(stream2 != stream, "unexpected stream\n"); SET_EXPECT(Stream_Seek); SET_EXPECT(Stream_Read); test_stream_read(stream2, S_OK, "01", 2); CHECK_CALLED(Stream_Seek); CHECK_CALLED(Stream_Read); expect_seek_pos = 3; SET_EXPECT(Stream_Seek); SET_EXPECT(Stream_Read); test_stream_read(stream, S_OK, "345", 3); CHECK_CALLED(Stream_Seek); CHECK_CALLED(Stream_Read); IStream_Release(stream); IStream_Release(stream2); IStream_Release(test_stream); stream = create_stream_from_string(" \t\r\n|}~YWJj ZGV|}~mZw== \t"); /* "abcdefg" in base64 obscured by invalid chars */ hr = IMimeBody_SetData(body, IET_BASE64, "text", "plain", &IID_IStream, stream); IStream_Release(stream); ok(hr == S_OK, "SetData failed: %08x\n", hr); test_current_encoding(body, IET_BASE64); hr = IMimeBody_GetData(body, IET_BINARY, &stream); ok(hr == S_OK, "GetData failed %08x\n", hr); test_stream_read(stream, S_OK, "abc", 3); test_stream_read(stream, S_OK, "defg", -1); IStream_Release(stream); hr = IMimeBody_GetData(body, IET_BASE64, &stream); ok(hr == S_OK, "GetData failed %08x\n", hr); test_stream_read(stream, S_OK, " \t\r", 3); IStream_Release(stream); stream = create_stream_from_string(" =3d=3D\"one\" \t=\r\ntw= o=\nx3\n=34\r\n5"); hr = IMimeBody_SetData(body, IET_QP, "text", "plain", &IID_IStream, stream); IStream_Release(stream); ok(hr == S_OK, "SetData failed: %08x\n", hr); test_current_encoding(body, IET_QP); hr = IMimeBody_GetData(body, IET_BINARY, &stream); ok(hr == S_OK, "GetData failed %08x\n", hr); test_stream_read(stream, S_OK, " ==\"one\" \ttw=o=3\n4\r\n5", -1); IStream_Release(stream); IMimeBody_Release(body); } static void test_Allocator(void) { HRESULT hr; IMimeAllocator *alloc; hr = MimeOleGetAllocator(&alloc); ok(hr == S_OK, "ret %08x\n", hr); IMimeAllocator_Release(alloc); } static void test_CreateMessage(void) { HRESULT hr; IMimeMessage *msg; IStream *stream; LONG ref; HBODY hbody, hbody2; IMimeBody *body; BODYOFFSETS offsets; ULONG count; FINDBODY find_struct; HCHARSET hcs; HBODY handle = NULL; char text[] = "text"; HBODY *body_list; PROPVARIANT prop; static const char att_pritype[] = "att:pri-content-type"; hr = MimeOleCreateMessage(NULL, &msg); ok(hr == S_OK, "ret %08x\n", hr); stream = create_stream_from_string(msg1); hr = IMimeMessage_Load(msg, stream); ok(hr == S_OK, "ret %08x\n", hr); hr = IMimeMessage_CountBodies(msg, HBODY_ROOT, TRUE, &count); ok(hr == S_OK, "ret %08x\n", hr); ok(count == 3, "got %d\n", count); hr = IMimeMessage_CountBodies(msg, HBODY_ROOT, FALSE, &count); ok(hr == S_OK, "ret %08x\n", hr); ok(count == 3, "got %d\n", count); hr = IMimeMessage_BindToObject(msg, HBODY_ROOT, &IID_IMimeBody, (void**)&body); ok(hr == S_OK, "ret %08x\n", hr); hr = IMimeBody_GetOffsets(body, &offsets); ok(hr == S_OK, "ret %08x\n", hr); ok(offsets.cbBoundaryStart == 0, "got %d\n", offsets.cbBoundaryStart); ok(offsets.cbHeaderStart == 0, "got %d\n", offsets.cbHeaderStart); ok(offsets.cbBodyStart == 359, "got %d\n", offsets.cbBodyStart); ok(offsets.cbBodyEnd == 666, "got %d\n", offsets.cbBodyEnd); IMimeBody_Release(body); hr = IMimeMessage_GetBody(msg, IBL_ROOT, NULL, &hbody); ok(hr == S_OK, "ret %08x\n", hr); hr = IMimeBody_GetHandle(body, NULL); ok(hr == E_INVALIDARG, "ret %08x\n", hr); hr = IMimeBody_GetHandle(body, &handle); ok(hr == S_OK, "ret %08x\n", hr); ok(handle != NULL, "handle %p\n", handle); hr = IMimeMessage_GetBody(msg, IBL_PARENT, hbody, NULL); ok(hr == E_INVALIDARG, "ret %08x\n", hr); hbody2 = (HBODY)0xdeadbeef; hr = IMimeMessage_GetBody(msg, IBL_PARENT, hbody, &hbody2); ok(hr == MIME_E_NOT_FOUND, "ret %08x\n", hr); ok(hbody2 == NULL, "hbody2 %p\n", hbody2); PropVariantInit(&prop); hr = IMimeMessage_GetBodyProp(msg, hbody, att_pritype, 0, &prop); ok(hr == S_OK, "ret %08x\n", hr); ok(prop.vt == VT_LPSTR, "vt %08x\n", prop.vt); ok(!strcasecmp(prop.u.pszVal, "multipart"), "got %s\n", prop.u.pszVal); PropVariantClear(&prop); hr = IMimeMessage_GetBody(msg, IBL_FIRST, hbody, &hbody); ok(hr == S_OK, "ret %08x\n", hr); hr = IMimeMessage_BindToObject(msg, hbody, &IID_IMimeBody, (void**)&body); ok(hr == S_OK, "ret %08x\n", hr); hr = IMimeBody_GetHandle(body, &handle); ok(hr == S_OK, "ret %08x\n", hr); ok(handle == hbody, "handle %p\n", handle); hr = IMimeBody_GetOffsets(body, &offsets); ok(hr == S_OK, "ret %08x\n", hr); ok(offsets.cbBoundaryStart == 405, "got %d\n", offsets.cbBoundaryStart); ok(offsets.cbHeaderStart == 428, "got %d\n", offsets.cbHeaderStart); ok(offsets.cbBodyStart == 518, "got %d\n", offsets.cbBodyStart); ok(offsets.cbBodyEnd == 523, "got %d\n", offsets.cbBodyEnd); hr = IMimeBody_GetCharset(body, &hcs); ok(hr == S_OK, "ret %08x\n", hr); todo_wine { ok(hcs != NULL, "Expected non-NULL charset\n"); } IMimeBody_Release(body); hr = IMimeMessage_GetBody(msg, IBL_NEXT, hbody, &hbody); ok(hr == S_OK, "ret %08x\n", hr); hr = IMimeMessage_BindToObject(msg, hbody, &IID_IMimeBody, (void**)&body); ok(hr == S_OK, "ret %08x\n", hr); hr = IMimeBody_GetHandle(body, &handle); ok(hr == S_OK, "ret %08x\n", hr); ok(handle == hbody, "handle %p\n", handle); hr = IMimeBody_GetOffsets(body, &offsets); ok(hr == S_OK, "ret %08x\n", hr); ok(offsets.cbBoundaryStart == 525, "got %d\n", offsets.cbBoundaryStart); ok(offsets.cbHeaderStart == 548, "got %d\n", offsets.cbHeaderStart); ok(offsets.cbBodyStart == 629, "got %d\n", offsets.cbBodyStart); ok(offsets.cbBodyEnd == 639, "got %d\n", offsets.cbBodyEnd); IMimeBody_Release(body); find_struct.pszPriType = text; find_struct.pszSubType = NULL; hr = IMimeMessage_FindFirst(msg, &find_struct, &hbody); ok(hr == S_OK, "ret %08x\n", hr); hr = IMimeMessage_FindNext(msg, &find_struct, &hbody); ok(hr == S_OK, "ret %08x\n", hr); hr = IMimeMessage_FindNext(msg, &find_struct, &hbody); ok(hr == MIME_E_NOT_FOUND, "ret %08x\n", hr); hr = IMimeMessage_GetAttachments(msg, &count, &body_list); ok(hr == S_OK, "ret %08x\n", hr); ok(count == 2, "got %d\n", count); if(count == 2) { IMimeBody *attachment; PROPVARIANT prop; PropVariantInit(&prop); hr = IMimeMessage_BindToObject(msg, body_list[0], &IID_IMimeBody, (void**)&attachment); ok(hr == S_OK, "ret %08x\n", hr); hr = IMimeBody_IsContentType(attachment, "multipart", NULL); ok(hr == S_FALSE, "ret %08x\n", hr); test_current_encoding(attachment, IET_8BIT); prop.vt = VT_LPSTR; hr = IMimeBody_GetProp(attachment, "Content-Transfer-Encoding", 0, &prop); ok(hr == S_OK, "ret %08x\n", hr); ok(prop.vt == VT_LPSTR, "type %d\n", prop.vt); ok(!strcmp(prop.u.pszVal, "8bit"), "got %s\n", prop.u.pszVal); PropVariantClear(&prop); hr = IMimeBody_IsType(attachment, IBT_ATTACHMENT); todo_wine ok(hr == S_FALSE, "ret %08x\n", hr); IMimeBody_Release(attachment); hr = IMimeMessage_BindToObject(msg, body_list[1], &IID_IMimeBody, (void**)&attachment); ok(hr == S_OK, "ret %08x\n", hr); hr = IMimeBody_IsContentType(attachment, "multipart", NULL); ok(hr == S_FALSE, "ret %08x\n", hr); test_current_encoding(attachment, IET_7BIT); prop.vt = VT_LPSTR; hr = IMimeBody_GetProp(attachment, "Content-Transfer-Encoding", 0, &prop); ok(hr == S_OK, "ret %08x\n", hr); ok(prop.vt == VT_LPSTR, "type %d\n", prop.vt); ok(!strcmp(prop.u.pszVal, "7bit"), "got %s\n", prop.u.pszVal); PropVariantClear(&prop); hr = IMimeBody_IsType(attachment, IBT_ATTACHMENT); ok(hr == S_OK, "ret %08x\n", hr); IMimeBody_Release(attachment); } CoTaskMemFree(body_list); hr = IMimeBody_GetCharset(body, &hcs); ok(hr == S_OK, "ret %08x\n", hr); todo_wine { ok(hcs != NULL, "Expected non-NULL charset\n"); } IMimeMessage_Release(msg); ref = IStream_AddRef(stream); ok(ref == 2 || broken(ref == 1), /* win95 */ "ref %d\n", ref); IStream_Release(stream); IStream_Release(stream); } static void test_mhtml_message(void) { IMimeMessage *mime_message; IMimeBody *mime_body; HBODY *body_list; IStream *stream; ULONG count; HRESULT hres; hres = MimeOleCreateMessage(NULL, &mime_message); ok(hres == S_OK, "MimeOleCreateMessage failed: %08x\n", hres); stream = create_stream_from_string(mhtml_page1); hres = IMimeMessage_Load(mime_message, stream); IStream_Release(stream); ok(hres == S_OK, "Load failed: %08x\n", hres); hres = IMimeMessage_CountBodies(mime_message, HBODY_ROOT, TRUE, &count); ok(hres == S_OK, "CountBodies failed: %08x\n", hres); ok(count == 3, "got %d\n", count); hres = IMimeMessage_GetAttachments(mime_message, &count, &body_list); ok(hres == S_OK, "GetAttachments failed: %08x\n", hres); ok(count == 2, "count = %u\n", count); hres = IMimeMessage_BindToObject(mime_message, body_list[0], &IID_IMimeBody, (void**)&mime_body); ok(hres == S_OK, "BindToObject failed: %08x\n", hres); hres = IMimeBody_GetData(mime_body, IET_BINARY, &stream); ok(hres == S_OK, "GetData failed: %08x\n", hres); test_stream_read(stream, S_OK, "", -1); IStream_Release(stream); test_current_encoding(mime_body, IET_QP); IMimeBody_Release(mime_body); hres = IMimeMessage_BindToObject(mime_message, body_list[1], &IID_IMimeBody, (void**)&mime_body); ok(hres == S_OK, "BindToObject failed: %08x\n", hres); test_current_encoding(mime_body, IET_BASE64); hres = IMimeBody_GetData(mime_body, IET_BINARY, &stream); ok(hres == S_OK, "GetData failed: %08x\n", hres); test_stream_read(stream, S_OK, "Test", -1); IStream_Release(stream); IMimeBody_Release(mime_body); CoTaskMemFree(body_list); IMimeMessage_Release(mime_message); } static void test_MessageSetProp(void) { static const char topic[] = "wine topic"; static const WCHAR topicW[] = {'w','i','n','e',' ','t','o','p','i','c',0}; HRESULT hr; IMimeMessage *msg; IMimeBody *body; PROPVARIANT prop; hr = MimeOleCreateMessage(NULL, &msg); ok(hr == S_OK, "ret %08x\n", hr); PropVariantInit(&prop); hr = IMimeMessage_BindToObject(msg, HBODY_ROOT, &IID_IMimeBody, (void**)&body); ok(hr == S_OK, "ret %08x\n", hr); hr = IMimeBody_SetProp(body, NULL, 0, &prop); ok(hr == E_INVALIDARG, "ret %08x\n", hr); hr = IMimeBody_SetProp(body, "Thread-Topic", 0, NULL); ok(hr == E_INVALIDARG, "ret %08x\n", hr); prop.vt = VT_LPSTR; prop.u.pszVal = CoTaskMemAlloc(strlen(topic)+1); strcpy(prop.u.pszVal, topic); hr = IMimeBody_SetProp(body, "Thread-Topic", 0, &prop); ok(hr == S_OK, "ret %08x\n", hr); PropVariantClear(&prop); hr = IMimeBody_GetProp(body, NULL, 0, &prop); ok(hr == E_INVALIDARG, "ret %08x\n", hr); hr = IMimeBody_GetProp(body, "Thread-Topic", 0, NULL); ok(hr == E_INVALIDARG, "ret %08x\n", hr); hr = IMimeBody_GetProp(body, "Wine-Topic", 0, &prop); ok(hr == MIME_E_NOT_FOUND, "ret %08x\n", hr); prop.vt = VT_LPSTR; hr = IMimeBody_GetProp(body, "Thread-Topic", 0, &prop); ok(hr == S_OK, "ret %08x\n", hr); if(hr == S_OK) { ok(prop.vt == VT_LPSTR, "type %d\n", prop.vt); ok(!strcmp(prop.u.pszVal, topic), "got %s\n", prop.u.pszVal); PropVariantClear(&prop); } prop.vt = VT_LPSTR; prop.u.pszVal = CoTaskMemAlloc(strlen(topic)+1); strcpy(prop.u.pszVal, topic); hr = IMimeBody_SetProp(body, PIDTOSTR(PID_HDR_SUBJECT), 0, &prop); ok(hr == S_OK, "ret %08x\n", hr); PropVariantClear(&prop); prop.vt = VT_LPSTR; hr = IMimeBody_GetProp(body, PIDTOSTR(PID_HDR_SUBJECT), 0, &prop); ok(hr == S_OK, "ret %08x\n", hr); if(hr == S_OK) { ok(prop.vt == VT_LPSTR, "type %d\n", prop.vt); ok(!strcmp(prop.u.pszVal, topic), "got %s\n", prop.u.pszVal); PropVariantClear(&prop); } /* Using the name or PID returns the same result. */ prop.vt = VT_LPSTR; hr = IMimeBody_GetProp(body, "Subject", 0, &prop); ok(hr == S_OK, "ret %08x\n", hr); if(hr == S_OK) { ok(prop.vt == VT_LPSTR, "type %d\n", prop.vt); ok(!strcmp(prop.u.pszVal, topic), "got %s\n", prop.u.pszVal); PropVariantClear(&prop); } prop.vt = VT_LPWSTR; hr = IMimeBody_GetProp(body, "Subject", 0, &prop); ok(hr == S_OK, "ret %08x\n", hr); if(hr == S_OK) { ok(prop.vt == VT_LPWSTR, "type %d\n", prop.vt); ok(!lstrcmpW(prop.u.pwszVal, topicW), "got %s\n", wine_dbgstr_w(prop.u.pwszVal)); PropVariantClear(&prop); } prop.vt = VT_LPSTR; prop.u.pszVal = CoTaskMemAlloc(strlen(topic)+1); strcpy(prop.u.pszVal, topic); hr = IMimeBody_SetProp(body, PIDTOSTR(PID_HDR_TO), 0, &prop); ok(hr == S_OK, "ret %08x\n", hr); PropVariantClear(&prop); /* Out of Range PID */ prop.vt = VT_LPSTR; prop.u.pszVal = CoTaskMemAlloc(strlen(topic)+1); strcpy(prop.u.pszVal, topic); hr = IMimeBody_SetProp(body, PIDTOSTR(124), 0, &prop); ok(hr == MIME_E_NOT_FOUND, "ret %08x\n", hr); PropVariantClear(&prop); IMimeBody_Release(body); IMimeMessage_Release(msg); } static void test_MessageGetPropInfo(void) { static const char topic[] = "wine topic"; static const char subject[] = "wine testing"; HRESULT hr; IMimeMessage *msg; IMimeBody *body; PROPVARIANT prop; MIMEPROPINFO info; hr = MimeOleCreateMessage(NULL, &msg); ok(hr == S_OK, "ret %08x\n", hr); PropVariantInit(&prop); hr = IMimeMessage_BindToObject(msg, HBODY_ROOT, &IID_IMimeBody, (void**)&body); ok(hr == S_OK, "ret %08x\n", hr); prop.vt = VT_LPSTR; prop.u.pszVal = CoTaskMemAlloc(strlen(topic)+1); strcpy(prop.u.pszVal, topic); hr = IMimeBody_SetProp(body, "Thread-Topic", 0, &prop); ok(hr == S_OK, "ret %08x\n", hr); PropVariantClear(&prop); prop.vt = VT_LPSTR; prop.u.pszVal = CoTaskMemAlloc(strlen(subject)+1); strcpy(prop.u.pszVal, subject); hr = IMimeBody_SetProp(body, PIDTOSTR(PID_HDR_SUBJECT), 0, &prop); ok(hr == S_OK, "ret %08x\n", hr); PropVariantClear(&prop); memset(&info, 0, sizeof(info)); info.dwMask = PIM_ENCODINGTYPE | PIM_FLAGS | PIM_PROPID; hr = IMimeBody_GetPropInfo(body, NULL, &info); ok(hr == E_INVALIDARG, "ret %08x\n", hr); memset(&info, 0, sizeof(info)); info.dwMask = PIM_ENCODINGTYPE | PIM_FLAGS | PIM_PROPID; hr = IMimeBody_GetPropInfo(body, "Subject", NULL); ok(hr == E_INVALIDARG, "ret %08x\n", hr); memset(&info, 0xfe, sizeof(info)); info.dwMask = PIM_ENCODINGTYPE | PIM_FLAGS | PIM_PROPID; hr = IMimeBody_GetPropInfo(body, "Subject", &info); ok(hr == S_OK, "ret %08x\n", hr); if(hr == S_OK) { ok(info.dwMask & (PIM_ENCODINGTYPE | PIM_FLAGS| PIM_PROPID), "Invalid mask 0x%08x\n", info.dwFlags); todo_wine ok(info.dwFlags & 0x10000000, "Invalid flags 0x%08x\n", info.dwFlags); ok(info.ietEncoding == 0, "Invalid encoding %d\n", info.ietEncoding); ok(info.dwPropId == PID_HDR_SUBJECT, "Invalid propid %d\n", info.dwPropId); ok(info.cValues == 0xfefefefe, "Invalid cValues %d\n", info.cValues); } memset(&info, 0xfe, sizeof(info)); info.dwMask = 0; hr = IMimeBody_GetPropInfo(body, "Subject", &info); ok(hr == S_OK, "ret %08x\n", hr); if(hr == S_OK) { ok(info.dwMask == 0, "Invalid mask 0x%08x\n", info.dwFlags); ok(info.dwFlags == 0xfefefefe, "Invalid flags 0x%08x\n", info.dwFlags); ok(info.ietEncoding == -16843010, "Invalid encoding %d\n", info.ietEncoding); ok(info.dwPropId == -16843010, "Invalid propid %d\n", info.dwPropId); } memset(&info, 0xfe, sizeof(info)); info.dwMask = 0; info.dwPropId = 1024; info.ietEncoding = 99; hr = IMimeBody_GetPropInfo(body, "Subject", &info); ok(hr == S_OK, "ret %08x\n", hr); if(hr == S_OK) { ok(info.dwMask == 0, "Invalid mask 0x%08x\n", info.dwFlags); ok(info.dwFlags == 0xfefefefe, "Invalid flags 0x%08x\n", info.dwFlags); ok(info.ietEncoding == 99, "Invalid encoding %d\n", info.ietEncoding); ok(info.dwPropId == 1024, "Invalid propid %d\n", info.dwPropId); } memset(&info, 0, sizeof(info)); info.dwMask = PIM_ENCODINGTYPE | PIM_FLAGS | PIM_PROPID; hr = IMimeBody_GetPropInfo(body, "Invalid Property", &info); ok(hr == MIME_E_NOT_FOUND, "ret %08x\n", hr); IMimeBody_Release(body); IMimeMessage_Release(msg); } static void test_MessageOptions(void) { static const char string[] = "XXXXX"; static const char zero[] = "0"; HRESULT hr; IMimeMessage *msg; PROPVARIANT prop; hr = MimeOleCreateMessage(NULL, &msg); ok(hr == S_OK, "ret %08x\n", hr); PropVariantInit(&prop); prop.vt = VT_BOOL; prop.u.boolVal = TRUE; hr = IMimeMessage_SetOption(msg, OID_HIDE_TNEF_ATTACHMENTS, &prop); ok(hr == S_OK, "ret %08x\n", hr); PropVariantClear(&prop); hr = IMimeMessage_GetOption(msg, OID_HIDE_TNEF_ATTACHMENTS, &prop); todo_wine ok(hr == S_OK, "ret %08x\n", hr); todo_wine ok(prop.vt == VT_BOOL, "vt %08x\n", prop.vt); todo_wine ok(prop.u.boolVal == TRUE, "Hide Attachments got %d\n", prop.u.boolVal); PropVariantClear(&prop); prop.vt = VT_LPSTR; prop.u.pszVal = CoTaskMemAlloc(strlen(string)+1); strcpy(prop.u.pszVal, string); hr = IMimeMessage_SetOption(msg, OID_HIDE_TNEF_ATTACHMENTS, &prop); ok(hr == S_OK, "ret %08x\n", hr); PropVariantClear(&prop); hr = IMimeMessage_GetOption(msg, OID_HIDE_TNEF_ATTACHMENTS, &prop); todo_wine ok(hr == S_OK, "ret %08x\n", hr); todo_wine ok(prop.vt == VT_BOOL, "vt %08x\n", prop.vt); todo_wine ok(prop.u.boolVal == TRUE, "Hide Attachments got %d\n", prop.u.boolVal); PropVariantClear(&prop); /* Invalid property type doesn't change the value */ prop.vt = VT_LPSTR; prop.u.pszVal = CoTaskMemAlloc(strlen(zero)+1); strcpy(prop.u.pszVal, zero); hr = IMimeMessage_SetOption(msg, OID_HIDE_TNEF_ATTACHMENTS, &prop); ok(hr == S_OK, "ret %08x\n", hr); PropVariantClear(&prop); hr = IMimeMessage_GetOption(msg, OID_HIDE_TNEF_ATTACHMENTS, &prop); todo_wine ok(hr == S_OK, "ret %08x\n", hr); todo_wine ok(prop.vt == VT_BOOL, "vt %08x\n", prop.vt); todo_wine ok(prop.u.boolVal == TRUE, "Hide Attachments got %d\n", prop.u.boolVal); PropVariantClear(&prop); /* Invalid OID */ prop.vt = VT_BOOL; prop.u.boolVal = TRUE; hr = IMimeMessage_SetOption(msg, 0xff00000a, &prop); ok(hr == MIME_E_INVALID_OPTION_ID, "ret %08x\n", hr); PropVariantClear(&prop); /* Out of range before type. */ prop.vt = VT_I4; prop.u.lVal = 1; hr = IMimeMessage_SetOption(msg, 0xff00000a, &prop); ok(hr == MIME_E_INVALID_OPTION_ID, "ret %08x\n", hr); PropVariantClear(&prop); IMimeMessage_Release(msg); } static void test_BindToObject(void) { HRESULT hr; IMimeMessage *msg; IMimeBody *body; ULONG count; hr = MimeOleCreateMessage(NULL, &msg); ok(hr == S_OK, "ret %08x\n", hr); hr = IMimeMessage_CountBodies(msg, HBODY_ROOT, TRUE, &count); ok(hr == S_OK, "ret %08x\n", hr); ok(count == 1, "got %d\n", count); hr = IMimeMessage_BindToObject(msg, HBODY_ROOT, &IID_IMimeBody, (void**)&body); ok(hr == S_OK, "ret %08x\n", hr); IMimeBody_Release(body); IMimeMessage_Release(msg); } static void test_BodyDeleteProp(void) { static const char topic[] = "wine topic"; HRESULT hr; IMimeMessage *msg; IMimeBody *body; PROPVARIANT prop; hr = MimeOleCreateMessage(NULL, &msg); ok(hr == S_OK, "ret %08x\n", hr); PropVariantInit(&prop); hr = IMimeMessage_BindToObject(msg, HBODY_ROOT, &IID_IMimeBody, (void**)&body); ok(hr == S_OK, "ret %08x\n", hr); hr = IMimeBody_DeleteProp(body, "Subject"); ok(hr == MIME_E_NOT_FOUND, "ret %08x\n", hr); hr = IMimeBody_DeleteProp(body, PIDTOSTR(PID_HDR_SUBJECT)); ok(hr == MIME_E_NOT_FOUND, "ret %08x\n", hr); prop.vt = VT_LPSTR; prop.u.pszVal = CoTaskMemAlloc(strlen(topic)+1); strcpy(prop.u.pszVal, topic); hr = IMimeBody_SetProp(body, "Subject", 0, &prop); ok(hr == S_OK, "ret %08x\n", hr); PropVariantClear(&prop); hr = IMimeBody_DeleteProp(body, "Subject"); ok(hr == S_OK, "ret %08x\n", hr); hr = IMimeBody_GetProp(body, "Subject", 0, &prop); ok(hr == MIME_E_NOT_FOUND, "ret %08x\n", hr); prop.vt = VT_LPSTR; prop.u.pszVal = CoTaskMemAlloc(strlen(topic)+1); strcpy(prop.u.pszVal, topic); hr = IMimeBody_SetProp(body, PIDTOSTR(PID_HDR_SUBJECT), 0, &prop); ok(hr == S_OK, "ret %08x\n", hr); PropVariantClear(&prop); hr = IMimeBody_DeleteProp(body, PIDTOSTR(PID_HDR_SUBJECT)); ok(hr == S_OK, "ret %08x\n", hr); hr = IMimeBody_GetProp(body, PIDTOSTR(PID_HDR_SUBJECT), 0, &prop); ok(hr == MIME_E_NOT_FOUND, "ret %08x\n", hr); IMimeBody_Release(body); IMimeMessage_Release(msg); } static void test_MimeOleGetPropertySchema(void) { HRESULT hr; IMimePropertySchema *schema = NULL; hr = MimeOleGetPropertySchema(&schema); ok(hr == S_OK, "ret %08x\n", hr); IMimePropertySchema_Release(schema); } typedef struct { const char *url; const char *content; const char *mime; const char *data; } mhtml_binding_test_t; static const mhtml_binding_test_t binding_tests[] = { { "mhtml:file://%s", mhtml_page1, "text/html", "" }, { "mhtml:file://%s!http://winehq.org/mhtmltest.html", mhtml_page1, "Image/Jpeg", "Test" } }; static const mhtml_binding_test_t *current_binding_test; static IInternetProtocol *current_binding_protocol; static HRESULT WINAPI BindInfo_QueryInterface(IInternetBindInfo *iface, REFIID riid, void **ppv) { if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IInternetBindInfo, riid)) { *ppv = iface; return S_OK; } *ppv = NULL; ok(0, "unexpected riid %s\n", wine_dbgstr_guid(riid)); return E_NOINTERFACE; } static ULONG WINAPI BindInfo_AddRef(IInternetBindInfo *iface) { return 2; } static ULONG WINAPI BindInfo_Release(IInternetBindInfo *iface) { return 1; } static HRESULT WINAPI BindInfo_GetBindInfo(IInternetBindInfo *iface, DWORD *grfBINDF, BINDINFO *pbindinfo) { CHECK_EXPECT(GetBindInfo); ok(grfBINDF != NULL, "grfBINDF == NULL\n"); ok(pbindinfo != NULL, "pbindinfo == NULL\n"); ok(pbindinfo->cbSize == sizeof(BINDINFO), "wrong size of pbindinfo: %d\n", pbindinfo->cbSize); *grfBINDF = BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA | BINDF_FROMURLMON | BINDF_NEEDFILE; return S_OK; } static HRESULT WINAPI BindInfo_GetBindString(IInternetBindInfo *iface, ULONG ulStringType, LPOLESTR *ppwzStr, ULONG cEl, ULONG *pcElFetched) { ok(0, "unexpected call\n"); return E_NOTIMPL; } static IInternetBindInfoVtbl InternetBindInfoVtbl = { BindInfo_QueryInterface, BindInfo_AddRef, BindInfo_Release, BindInfo_GetBindInfo, BindInfo_GetBindString }; static IInternetBindInfo bind_info = { &InternetBindInfoVtbl }; static HRESULT WINAPI ServiceProvider_QueryInterface(IServiceProvider *iface, REFIID riid, void **ppv) { ok(0, "unexpected call %s\n", wine_dbgstr_guid(riid)); *ppv = NULL; return E_NOINTERFACE; } static ULONG WINAPI ServiceProvider_AddRef(IServiceProvider *iface) { return 2; } static ULONG WINAPI ServiceProvider_Release(IServiceProvider *iface) { return 1; } static HRESULT WINAPI ServiceProvider_QueryService(IServiceProvider *iface, REFGUID guidService, REFIID riid, void **ppv) { if(IsEqualGUID(&CLSID_MimeEdit, guidService)) { *ppv = NULL; return E_NOINTERFACE; } ok(0, "unexpected service %s\n", wine_dbgstr_guid(guidService)); return E_FAIL; } static const IServiceProviderVtbl ServiceProviderVtbl = { ServiceProvider_QueryInterface, ServiceProvider_AddRef, ServiceProvider_Release, ServiceProvider_QueryService }; static IServiceProvider service_provider = { &ServiceProviderVtbl }; static HRESULT WINAPI ProtocolSink_QueryInterface(IInternetProtocolSink *iface, REFIID riid, void **ppv) { if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IInternetProtocolSink, riid)) { *ppv = iface; return S_OK; } if(IsEqualGUID(&IID_IServiceProvider, riid)) { *ppv = &service_provider; return S_OK; } *ppv = NULL; ok(0, "unexpected riid %s\n", wine_dbgstr_guid(riid)); return E_NOINTERFACE; } static ULONG WINAPI ProtocolSink_AddRef(IInternetProtocolSink *iface) { return 2; } static ULONG WINAPI ProtocolSink_Release(IInternetProtocolSink *iface) { return 1; } static HRESULT WINAPI ProtocolSink_Switch(IInternetProtocolSink *iface, PROTOCOLDATA *pProtocolData) { ok(0, "unexpected call\n"); return E_NOTIMPL; } static HRESULT WINAPI ProtocolSink_ReportProgress(IInternetProtocolSink *iface, ULONG ulStatusCode, const WCHAR *szStatusText) { switch(ulStatusCode) { case BINDSTATUS_MIMETYPEAVAILABLE: CHECK_EXPECT(ReportProgress_MIMETYPEAVAILABLE); ok(!strcmp_wa(szStatusText, current_binding_test->mime), "status text %s\n", wine_dbgstr_w(szStatusText)); return S_OK; case BINDSTATUS_CACHEFILENAMEAVAILABLE: CHECK_EXPECT(ReportProgress_CACHEFILENAMEAVAILABLE); return S_OK; default: ok(0, "unexpected call %u %s\n", ulStatusCode, wine_dbgstr_w(szStatusText)); } return E_NOTIMPL; } static HRESULT WINAPI ProtocolSink_ReportData(IInternetProtocolSink *iface, DWORD grfBSCF, ULONG ulProgress, ULONG ulProgressMax) { char buf[1024]; DWORD read; HRESULT hres; CHECK_EXPECT(ReportData); ok(!ulProgress, "ulProgress = %u\n", ulProgress); ok(ulProgress == ulProgressMax, "ulProgress != ulProgressMax\n"); ok(grfBSCF == (BSCF_FIRSTDATANOTIFICATION | BSCF_INTERMEDIATEDATANOTIFICATION | BSCF_LASTDATANOTIFICATION | BSCF_DATAFULLYAVAILABLE | BSCF_AVAILABLEDATASIZEUNKNOWN), "grcf = %08x\n", grfBSCF); hres = IInternetProtocol_Read(current_binding_protocol, buf, sizeof(buf), &read); ok(hres == S_OK, "Read failed: %08x\n", hres); buf[read] = 0; ok(!strcmp(buf, current_binding_test->data), "unexpected data: %s\n", buf); hres = IInternetProtocol_Read(current_binding_protocol, buf, sizeof(buf), &read); ok(hres == S_FALSE, "Read failed: %08x\n", hres); return S_OK; } static HRESULT WINAPI ProtocolSink_ReportResult(IInternetProtocolSink *iface, HRESULT hrResult, DWORD dwError, LPCWSTR szResult) { CHECK_EXPECT(ReportResult); ok(hrResult == S_OK, "hrResult = %08x\n", hrResult); ok(!dwError, "dwError = %u\n", dwError); ok(!szResult, "szResult = %s\n", wine_dbgstr_w(szResult)); return S_OK; } static IInternetProtocolSinkVtbl InternetProtocolSinkVtbl = { ProtocolSink_QueryInterface, ProtocolSink_AddRef, ProtocolSink_Release, ProtocolSink_Switch, ProtocolSink_ReportProgress, ProtocolSink_ReportData, ProtocolSink_ReportResult }; static IInternetProtocolSink protocol_sink = { &InternetProtocolSinkVtbl }; static void test_mhtml_protocol_binding(const mhtml_binding_test_t *test) { char file_name[MAX_PATH+32], *p, urla[INTERNET_MAX_URL_LENGTH]; WCHAR test_url[INTERNET_MAX_URL_LENGTH]; IInternetProtocol *protocol; IUnknown *unk; HRESULT hres; HANDLE file; DWORD size; BOOL ret; p = file_name + GetCurrentDirectoryA(sizeof(file_name), file_name); *p++ = '\\'; strcpy(p, "winetest.mht"); file = CreateFileA(file_name, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL); ok(file != INVALID_HANDLE_VALUE, "CreateFile failed\n"); WriteFile(file, test->content, strlen(test->content), &size, NULL); CloseHandle(file); sprintf(urla, test->url, file_name); MultiByteToWideChar(CP_ACP, 0, urla, -1, test_url, ARRAY_SIZE(test_url)); hres = CoCreateInstance(&CLSID_IMimeHtmlProtocol, NULL, CLSCTX_INPROC_SERVER, &IID_IInternetProtocol, (void**)&protocol); ok(hres == S_OK, "Could not create protocol handler: %08x\n", hres); hres = IInternetProtocol_QueryInterface(protocol, &IID_IInternetProtocolEx, (void**)&unk); ok(hres == E_NOINTERFACE, "Could get IInternetProtocolEx\n"); current_binding_test = test; current_binding_protocol = protocol; SET_EXPECT(GetBindInfo); SET_EXPECT(ReportProgress_MIMETYPEAVAILABLE); SET_EXPECT(ReportProgress_CACHEFILENAMEAVAILABLE); SET_EXPECT(ReportData); SET_EXPECT(ReportResult); hres = IInternetProtocol_Start(protocol, test_url, &protocol_sink, &bind_info, 0, 0); ok(hres == S_OK, "Start failed: %08x\n", hres); CHECK_CALLED(GetBindInfo); CHECK_CALLED(ReportProgress_MIMETYPEAVAILABLE); todo_wine CHECK_CALLED(ReportProgress_CACHEFILENAMEAVAILABLE); CHECK_CALLED(ReportData); CHECK_CALLED(ReportResult); IInternetProtocol_Release(protocol); ret = DeleteFileA("winetest.mht"); ok(ret, "DeleteFile failed: %u\n", GetLastError()); } static const struct { const char *base_url; const char *relative_url; const char *expected_result; BOOL todo; } combine_tests[] = { { "mhtml:file:///c:/dir/test.mht", "http://test.org", "mhtml:file:///c:/dir/test.mht!x-usc:http://test.org" }, { "mhtml:file:///c:/dir/test.mht", "3D\"http://test.org\"", "mhtml:file:///c:/dir/test.mht!x-usc:3D\"http://test.org\"" }, { "mhtml:file:///c:/dir/test.mht", "123abc", "mhtml:file:///c:/dir/test.mht!x-usc:123abc" }, { "mhtml:file:///c:/dir/test.mht!x-usc:http://test.org", "123abc", "mhtml:file:///c:/dir/test.mht!x-usc:123abc" }, { "MhtMl:file:///c:/dir/test.mht!x-usc:http://test.org/dir/dir2/file.html", "../..", "mhtml:file:///c:/dir/test.mht!x-usc:../.." }, {"mhtml:file:///c:/dir/test.mht!x-usc:file:///c:/dir/dir2/file.html", "../..", "mhtml:file:///c:/dir/test.mht!x-usc:../.." }, { "mhtml:file:///c:/dir/test.mht!x-usc:http://test.org", "", "mhtml:file:///c:/dir/test.mht" }, { "mhtml:file:///c:/dir/test.mht!x-usc:http://test.org", "mhtml:file:///d:/file.html", "file:///d:/file.html", TRUE }, { "mhtml:file:///c:/dir/test.mht!x-usc:http://test.org", "mhtml:file:///c:/dir2/test.mht!x-usc:http://test.org", "mhtml:file:///c:/dir2/test.mht!x-usc:http://test.org", TRUE }, { "mhtml:file:///c:/dir/test.mht!http://test.org", "123abc", "mhtml:file:///c:/dir/test.mht!x-usc:123abc" }, { "mhtml:file:///c:/dir/test.mht!http://test.org", "", "mhtml:file:///c:/dir/test.mht" } }; static void test_mhtml_protocol_info(void) { WCHAR *base_url, *relative_url, combined_url[INTERNET_MAX_URL_LENGTH]; IInternetProtocolInfo *protocol_info; DWORD combined_len; unsigned i, exlen; HRESULT hres; static const WCHAR http_url[] = {'h','t','t','p',':','/','/','t','e','s','t','.','o','r','g',0}; hres = CoCreateInstance(&CLSID_IMimeHtmlProtocol, NULL, CLSCTX_INPROC_SERVER, &IID_IInternetProtocolInfo, (void**)&protocol_info); ok(hres == S_OK, "Could not create protocol info: %08x\n", hres); for(i = 0; i < ARRAY_SIZE(combine_tests); i++) { base_url = a2w(combine_tests[i].base_url); relative_url = a2w(combine_tests[i].relative_url); combined_len = 0xdeadbeef; hres = IInternetProtocolInfo_CombineUrl(protocol_info, base_url, relative_url, ICU_BROWSER_MODE, combined_url, ARRAY_SIZE(combined_url), &combined_len, 0); todo_wine_if(combine_tests[i].todo) ok(hres == S_OK, "[%u] CombineUrl failed: %08x\n", i, hres); if(SUCCEEDED(hres)) { exlen = strlen(combine_tests[i].expected_result); ok(combined_len == exlen, "[%u] combined len is %u, expected %u\n", i, combined_len, exlen); ok(!strcmp_wa(combined_url, combine_tests[i].expected_result), "[%u] combined URL is %s, expected %s\n", i, wine_dbgstr_w(combined_url), combine_tests[i].expected_result); combined_len = 0xdeadbeef; hres = IInternetProtocolInfo_CombineUrl(protocol_info, base_url, relative_url, ICU_BROWSER_MODE, combined_url, exlen, &combined_len, 0); ok(hres == E_FAIL, "[%u] CombineUrl returned: %08x\n", i, hres); ok(!combined_len, "[%u] combined_len = %u\n", i, combined_len); } HeapFree(GetProcessHeap(), 0, base_url); HeapFree(GetProcessHeap(), 0, relative_url); } hres = IInternetProtocolInfo_CombineUrl(protocol_info, http_url, http_url, ICU_BROWSER_MODE, combined_url, ARRAY_SIZE(combined_url), &combined_len, 0); ok(hres == E_FAIL, "CombineUrl failed: %08x\n", hres); IInternetProtocolInfo_Release(protocol_info); } static HRESULT WINAPI outer_QueryInterface(IUnknown *iface, REFIID riid, void **ppv) { ok(0, "unexpected call\n"); return E_NOINTERFACE; } static ULONG WINAPI outer_AddRef(IUnknown *iface) { return 2; } static ULONG WINAPI outer_Release(IUnknown *iface) { return 1; } static const IUnknownVtbl outer_vtbl = { outer_QueryInterface, outer_AddRef, outer_Release }; static BOOL broken_mhtml_resolver; static void test_mhtml_protocol(void) { IUnknown outer = { &outer_vtbl }; IClassFactory *class_factory; IUnknown *unk, *unk2; unsigned i; HRESULT hres; /* test class factory */ hres = CoGetClassObject(&CLSID_IMimeHtmlProtocol, CLSCTX_INPROC_SERVER, NULL, &IID_IUnknown, (void**)&unk); ok(hres == S_OK, "CoGetClassObject failed: %08x\n", hres); hres = IUnknown_QueryInterface(unk, &IID_IInternetProtocolInfo, (void**)&unk2); ok(hres == E_NOINTERFACE, "IInternetProtocolInfo supported\n"); hres = IUnknown_QueryInterface(unk, &IID_IClassFactory, (void**)&class_factory); ok(hres == S_OK, "Could not get IClassFactory iface: %08x\n", hres); IUnknown_Release(unk); hres = IClassFactory_CreateInstance(class_factory, &outer, &IID_IUnknown, (void**)&unk); ok(hres == S_OK, "CreateInstance returned: %08x\n", hres); hres = IUnknown_QueryInterface(unk, &IID_IInternetProtocol, (void**)&unk2); ok(hres == S_OK, "Could not get IInternetProtocol iface: %08x\n", hres); IUnknown_Release(unk2); IUnknown_Release(unk); hres = IClassFactory_CreateInstance(class_factory, (IUnknown*)0xdeadbeef, &IID_IInternetProtocol, (void**)&unk2); ok(hres == CLASS_E_NOAGGREGATION, "CreateInstance returned: %08x\n", hres); IClassFactory_Release(class_factory); if(!broken_mhtml_resolver) test_mhtml_protocol_info(); for(i = 0; i < ARRAY_SIZE(binding_tests); i++) test_mhtml_protocol_binding(binding_tests + i); } static void test_MimeOleObjectFromMoniker(void) { IMoniker *mon, *new_mon; WCHAR *mhtml_url, *url; IBindCtx *bind_ctx; IUnknown *unk; unsigned i; HRESULT hres; static const struct { const char *url; const char *mhtml_url; } tests[] = { {"file:///x:\\dir\\file.mht", "mhtml:file://x:\\dir\\file.mht"}, {"file:///x:/dir/file.mht", "mhtml:file://x:\\dir\\file.mht"}, {"http://www.winehq.org/index.html?query#hash", "mhtml:http://www.winehq.org/index.html?query#hash"}, {"../test.mht", "mhtml:../test.mht"} }; for(i = 0; i < ARRAY_SIZE(tests); i++) { url = a2w(tests[i].url); hres = CreateURLMoniker(NULL, url, &mon); ok(hres == S_OK, "CreateURLMoniker failed: %08x\n", hres); HeapFree(GetProcessHeap(), 0, url); hres = CreateBindCtx(0, &bind_ctx); ok(hres == S_OK, "CreateBindCtx failed: %08x\n", hres); hres = MimeOleObjectFromMoniker(0, mon, bind_ctx, &IID_IUnknown, (void**)&unk, &new_mon); ok(hres == S_OK || broken(!i && hres == INET_E_RESOURCE_NOT_FOUND), "MimeOleObjectFromMoniker failed: %08x\n", hres); IBindCtx_Release(bind_ctx); if(hres == INET_E_RESOURCE_NOT_FOUND) { /* winxp */ win_skip("Broken MHTML behaviour found. Skipping some tests.\n"); broken_mhtml_resolver = TRUE; return; } hres = IMoniker_GetDisplayName(new_mon, NULL, NULL, &mhtml_url); ok(hres == S_OK, "GetDisplayName failed: %08x\n", hres); ok(!strcmp_wa(mhtml_url, tests[i].mhtml_url), "[%d] unexpected mhtml URL: %s\n", i, wine_dbgstr_w(mhtml_url)); CoTaskMemFree(mhtml_url); IUnknown_Release(unk); IMoniker_Release(new_mon); IMoniker_Release(mon); } } START_TEST(mimeole) { OleInitialize(NULL); test_CreateVirtualStream(); test_CreateSecurity(); test_CreateBody(); test_SetData(); test_Allocator(); test_CreateMessage(); test_MessageSetProp(); test_MessageGetPropInfo(); test_MessageOptions(); test_BindToObject(); test_BodyDeleteProp(); test_MimeOleGetPropertySchema(); test_mhtml_message(); test_MimeOleObjectFromMoniker(); test_mhtml_protocol(); OleUninitialize(); }