335 lines
11 KiB
C
335 lines
11 KiB
C
/*
|
|
* 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 "windows.h"
|
|
#include "ole2.h"
|
|
#include "ocidl.h"
|
|
|
|
#include "initguid.h"
|
|
#include "mimeole.h"
|
|
|
|
#include <stdio.h>
|
|
#include <assert.h>
|
|
|
|
#include "wine/test.h"
|
|
|
|
static 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 <huw@codeweavers.com>\r\n"
|
|
"From: Me <xxx@codeweavers.com>\r\n"
|
|
"To: wine-patches <wine-patches@winehq.org>\r\n"
|
|
"Cc: Huw Davies <huw@codeweavers.com>,\r\n"
|
|
" \"Fred Bloggs\" <fred@bloggs.com>\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 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 void test_CreateBody(void)
|
|
{
|
|
HRESULT hr;
|
|
IMimeBody *body;
|
|
HBODY handle = (void *)0xdeadbeef;
|
|
IStream *in;
|
|
LARGE_INTEGER off;
|
|
ULARGE_INTEGER pos;
|
|
ENCODINGTYPE enc;
|
|
ULONG count, found_param, i;
|
|
MIMEPARAMINFO *param_info;
|
|
IMimeAllocator *alloc;
|
|
BODYOFFSETS offsets;
|
|
|
|
hr = CoCreateInstance(&CLSID_IMimeBody, NULL, CLSCTX_INPROC_SERVER, &IID_IMimeBody, (void**)&body);
|
|
ok(hr == S_OK, "ret %08x\n", hr);
|
|
|
|
hr = IMimeBody_GetHandle(body, &handle);
|
|
ok(hr == MIME_E_NO_DATA, "ret %08x\n", hr);
|
|
ok(handle == NULL, "handle %p\n", handle);
|
|
|
|
hr = CreateStreamOnHGlobal(NULL, TRUE, &in);
|
|
ok(hr == S_OK, "ret %08x\n", hr);
|
|
IStream_Write(in, msg1, sizeof(msg1) - 1, NULL);
|
|
off.QuadPart = 0;
|
|
IStream_Seek(in, off, STREAM_SEEK_SET, NULL);
|
|
|
|
/* Need to call InitNew before Load otherwise Load crashes with native inetcomm */
|
|
hr = IMimeBody_InitNew(body);
|
|
ok(hr == S_OK, "ret %08x\n", hr);
|
|
|
|
hr = IMimeBody_GetCurrentEncoding(body, &enc);
|
|
ok(hr == S_OK, "ret %08x\n", hr);
|
|
ok(enc == IET_7BIT, "encoding %d\n", enc);
|
|
|
|
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_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);
|
|
hr = IMimeBody_GetCurrentEncoding(body, &enc);
|
|
ok(hr == S_OK, "ret %08x\n", hr);
|
|
ok(enc == IET_8BIT, "encoding %d\n", enc);
|
|
|
|
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 = 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);
|
|
}
|
|
|
|
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;
|
|
LARGE_INTEGER pos;
|
|
LONG ref;
|
|
HBODY hbody;
|
|
IMimeBody *body;
|
|
BODYOFFSETS offsets;
|
|
ULONG count;
|
|
FINDBODY find_struct;
|
|
HCHARSET hcs;
|
|
INETCSETINFO csi;
|
|
|
|
char text[] = "text";
|
|
HBODY *body_list;
|
|
PROPVARIANT prop;
|
|
static char att_pritype[] = "att:pri-content-type";
|
|
|
|
hr = MimeOleCreateMessage(NULL, &msg);
|
|
ok(hr == S_OK, "ret %08x\n", hr);
|
|
|
|
CreateStreamOnHGlobal(NULL, TRUE, &stream);
|
|
IStream_Write(stream, msg1, sizeof(msg1) - 1, NULL);
|
|
pos.QuadPart = 0;
|
|
IStream_Seek(stream, pos, STREAM_SEEK_SET, NULL);
|
|
|
|
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);
|
|
|
|
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_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);
|
|
ok(hcs == NULL, "ret %p\n", hcs);
|
|
|
|
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_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);
|
|
CoTaskMemFree(body_list);
|
|
|
|
hr = IMimeMessage_GetCharset(body, &hcs);
|
|
ok(hr == S_OK, "ret %08x\n", hr);
|
|
ok(hcs == NULL, "ret %p\n", hcs);
|
|
hr = MimeOleGetCharsetInfo(hcs, &csi);
|
|
ok(hr == E_INVALIDARG, "ret %08x\n", hr);
|
|
|
|
IMimeMessage_Release(msg);
|
|
|
|
ref = IStream_AddRef(stream);
|
|
ok(ref == 2, "ref %d\n", ref);
|
|
IStream_Release(stream);
|
|
|
|
IStream_Release(stream);
|
|
}
|
|
|
|
START_TEST(mimeole)
|
|
{
|
|
OleInitialize(NULL);
|
|
test_CreateVirtualStream();
|
|
test_CreateSecurity();
|
|
test_CreateBody();
|
|
test_Allocator();
|
|
test_CreateMessage();
|
|
OleUninitialize();
|
|
}
|