dlls/inetcomm/tests/mimeole.c

   1 /*
   2  * MimeOle tests
   3  *
   4  * Copyright 2007 Huw Davies
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #define COBJMACROS
  22 #define NONAMELESSUNION
  23
  24 #include "initguid.h"
  25 #include "windows.h"
  26 #include "ole2.h"
  27 #include "ocidl.h"
  28
  29 #include "mimeole.h"
  30
  31 #include <stdio.h>
  32
  33 #include "wine/test.h"
  34
  35 static char msg1[] =
  36     "MIME-Version: 1.0\r\n"
  37     "Content-Type: multipart/mixed;\r\n"
  38     " boundary=\"------------1.5.0.6\";\r\n"
  39     " stuff=\"du;nno\";\r\n"
  40     " morestuff=\"so\\\\me\\\"thing\\\"\"\r\n"
  41     "foo: bar\r\n"
  42     "From: Huw Davies <huw@codeweavers.com>\r\n"
  43     "From: Me <xxx@codeweavers.com>\r\n"
  44     "To: wine-patches <wine-patches@winehq.org>\r\n"
  45     "Cc: Huw Davies <huw@codeweavers.com>,\r\n"
  46     "    \"Fred Bloggs\"   <fred@bloggs.com>\r\n"
  47     "foo: baz\r\n"
  48     "bar: fum\r\n"
  49     "\r\n"
  50     "This is a multi-part message in MIME format.\r\n"
  51     "--------------1.5.0.6\r\n"
  52     "Content-Type: text/plain; format=fixed; charset=UTF-8\r\n"
  53     "Content-Transfer-Encoding: 8bit\r\n"
  54     "\r\n"
  55     "Stuff\r\n"
  56     "--------------1.5.0.6\r\n"
  57     "Content-Type: text/plain; charset=\"us-ascii\"\r\n"
  58     "Content-Transfer-Encoding: 7bit\r\n"
  59     "\r\n"
  60     "More stuff\r\n"
  61     "--------------1.5.0.6--\r\n";
  62
  63 static void test_CreateVirtualStream(void)
  64 {
  65     HRESULT hr;
  66     IStream *pstm;
  67
  68     hr = MimeOleCreateVirtualStream(&pstm);
  69     ok(hr == S_OK, "ret %08x\n", hr);
  70
  71     IStream_Release(pstm);
  72 }
  73
  74 static void test_CreateSecurity(void)
  75 {
  76     HRESULT hr;
  77     IMimeSecurity *sec;
  78
  79     hr = MimeOleCreateSecurity(&sec);
  80     ok(hr == S_OK, "ret %08x\n", hr);
  81
  82     IMimeSecurity_Release(sec);
  83 }
  84
  85 static void test_CreateBody(void)
  86 {
  87     HRESULT hr;
  88     IMimeBody *body;
  89     HBODY handle = (void *)0xdeadbeef;
  90     IStream *in;
  91     LARGE_INTEGER off;
  92     ULARGE_INTEGER pos;
  93     ENCODINGTYPE enc;
  94     ULONG count, found_param, i;
  95     MIMEPARAMINFO *param_info;
  96     IMimeAllocator *alloc;
  97     BODYOFFSETS offsets;
  98
  99     hr = CoCreateInstance(&CLSID_IMimeBody, NULL, CLSCTX_INPROC_SERVER, &IID_IMimeBody, (void**)&body);
 100     ok(hr == S_OK, "ret %08x\n", hr);
 101
 102     hr = IMimeBody_GetHandle(body, &handle);
 103     ok(hr == MIME_E_NO_DATA, "ret %08x\n", hr);
 104     ok(handle == NULL, "handle %p\n", handle);
 105
 106     hr = CreateStreamOnHGlobal(NULL, TRUE, &in);
 107     ok(hr == S_OK, "ret %08x\n", hr);
 108     IStream_Write(in, msg1, sizeof(msg1) - 1, NULL);
 109     off.QuadPart = 0;
 110     IStream_Seek(in, off, STREAM_SEEK_SET, NULL);
 111
 112     /* Need to call InitNew before Load otherwise Load crashes with native inetcomm */
 113     hr = IMimeBody_InitNew(body);
 114     ok(hr == S_OK, "ret %08x\n", hr);
 115
 116     hr = IMimeBody_GetCurrentEncoding(body, &enc);
 117     ok(hr == S_OK, "ret %08x\n", hr);
 118     ok(enc == IET_7BIT, "encoding %d\n", enc);
 119
 120     hr = IMimeBody_Load(body, in);
 121     ok(hr == S_OK, "ret %08x\n", hr);
 122     off.QuadPart = 0;
 123     IStream_Seek(in, off, STREAM_SEEK_CUR, &pos);
 124     ok(pos.u.LowPart == 359, "pos %u\n", pos.u.LowPart);
 125
 126     hr = IMimeBody_IsContentType(body, "multipart", "mixed");
 127     ok(hr == S_OK, "ret %08x\n", hr);
 128     hr = IMimeBody_IsContentType(body, "text", "plain");
 129     ok(hr == S_FALSE, "ret %08x\n", hr);
 130     hr = IMimeBody_IsContentType(body, NULL, "mixed");
 131     ok(hr == S_OK, "ret %08x\n", hr);
 132     hr = IMimeBody_IsType(body, IBT_EMPTY);
 133     ok(hr == S_OK, "got %08x\n", hr);
 134
 135     hr = IMimeBody_SetData(body, IET_8BIT, "text", "plain", &IID_IStream, in);
 136     ok(hr == S_OK, "ret %08x\n", hr);
 137     hr = IMimeBody_IsContentType(body, "text", "plain");
 138     todo_wine
 139         ok(hr == S_OK, "ret %08x\n", hr);
 140     hr = IMimeBody_GetCurrentEncoding(body, &enc);
 141     ok(hr == S_OK, "ret %08x\n", hr);
 142     ok(enc == IET_8BIT, "encoding %d\n", enc);
 143
 144     memset(&offsets, 0xcc, sizeof(offsets));
 145     hr = IMimeBody_GetOffsets(body, &offsets);
 146     ok(hr == MIME_E_NO_DATA, "ret %08x\n", hr);
 147     ok(offsets.cbBoundaryStart == 0, "got %d\n", offsets.cbBoundaryStart);
 148     ok(offsets.cbHeaderStart == 0, "got %d\n", offsets.cbHeaderStart);
 149     ok(offsets.cbBodyStart == 0, "got %d\n", offsets.cbBodyStart);
 150     ok(offsets.cbBodyEnd == 0, "got %d\n", offsets.cbBodyEnd);
 151
 152     hr = IMimeBody_IsType(body, IBT_EMPTY);
 153     ok(hr == S_FALSE, "got %08x\n", hr);
 154
 155     hr = MimeOleGetAllocator(&alloc);
 156     ok(hr == S_OK, "ret %08x\n", hr);
 157
 158     hr = IMimeBody_GetParameters(body, "nothere", &count, &param_info);
 159     ok(hr == MIME_E_NOT_FOUND, "ret %08x\n", hr);
 160     ok(count == 0, "got %d\n", count);
 161     ok(!param_info, "got %p\n", param_info);
 162
 163     hr = IMimeBody_GetParameters(body, "bar", &count, &param_info);
 164     ok(hr == S_OK, "ret %08x\n", hr);
 165     ok(count == 0, "got %d\n", count);
 166     ok(!param_info, "got %p\n", param_info);
 167
 168     hr = IMimeBody_GetParameters(body, "Content-Type", &count, &param_info);
 169     ok(hr == S_OK, "ret %08x\n", hr);
 170     todo_wine  /* native adds a charset parameter */
 171         ok(count == 4, "got %d\n", count);
 172     ok(param_info != NULL, "got %p\n", param_info);
 173
 174     found_param = 0;
 175     for(i = 0; i < count; i++)
 176     {
 177         if(!strcmp(param_info[i].pszName, "morestuff"))
 178         {
 179             found_param++;
 180             ok(!strcmp(param_info[i].pszData, "so\\me\"thing\""),
 181                "got %s\n", param_info[i].pszData);
 182         }
 183         else if(!strcmp(param_info[i].pszName, "stuff"))
 184         {
 185             found_param++;
 186             ok(!strcmp(param_info[i].pszData, "du;nno"),
 187                "got %s\n", param_info[i].pszData);
 188         }
 189     }
 190     ok(found_param == 2, "matched %d params\n", found_param);
 191
 192     hr = IMimeAllocator_FreeParamInfoArray(alloc, count, param_info, TRUE);
 193     ok(hr == S_OK, "ret %08x\n", hr);
 194     IMimeAllocator_Release(alloc);
 195
 196     IStream_Release(in);
 197     IMimeBody_Release(body);
 198 }
 199
 200 static void test_Allocator(void)
 201 {
 202     HRESULT hr;
 203     IMimeAllocator *alloc;
 204
 205     hr = MimeOleGetAllocator(&alloc);
 206     ok(hr == S_OK, "ret %08x\n", hr);
 207     IMimeAllocator_Release(alloc);
 208 }
 209
 210 static void test_CreateMessage(void)
 211 {
 212     HRESULT hr;
 213     IMimeMessage *msg;
 214     IStream *stream;
 215     LARGE_INTEGER pos;
 216     LONG ref;
 217     HBODY hbody;
 218     IMimeBody *body;
 219     BODYOFFSETS offsets;
 220     ULONG count;
 221     FINDBODY find_struct;
 222     HCHARSET hcs;
 223
 224     char text[] = "text";
 225     HBODY *body_list;
 226     PROPVARIANT prop;
 227     static char att_pritype[] = "att:pri-content-type";
 228
 229     hr = MimeOleCreateMessage(NULL, &msg);
 230     ok(hr == S_OK, "ret %08x\n", hr);
 231
 232     CreateStreamOnHGlobal(NULL, TRUE, &stream);
 233     IStream_Write(stream, msg1, sizeof(msg1) - 1, NULL);
 234     pos.QuadPart = 0;
 235     IStream_Seek(stream, pos, STREAM_SEEK_SET, NULL);
 236
 237     hr = IMimeMessage_Load(msg, stream);
 238     ok(hr == S_OK, "ret %08x\n", hr);
 239
 240     hr = IMimeMessage_CountBodies(msg, HBODY_ROOT, TRUE, &count);
 241     ok(hr == S_OK, "ret %08x\n", hr);
 242     ok(count == 3, "got %d\n", count);
 243
 244     hr = IMimeMessage_CountBodies(msg, HBODY_ROOT, FALSE, &count);
 245     ok(hr == S_OK, "ret %08x\n", hr);
 246     ok(count == 3, "got %d\n", count);
 247
 248     hr = IMimeMessage_BindToObject(msg, HBODY_ROOT, &IID_IMimeBody, (void**)&body);
 249     ok(hr == S_OK, "ret %08x\n", hr);
 250     hr = IMimeBody_GetOffsets(body, &offsets);
 251     ok(hr == S_OK, "ret %08x\n", hr);
 252     ok(offsets.cbBoundaryStart == 0, "got %d\n", offsets.cbBoundaryStart);
 253     ok(offsets.cbHeaderStart == 0, "got %d\n", offsets.cbHeaderStart);
 254     ok(offsets.cbBodyStart == 359, "got %d\n", offsets.cbBodyStart);
 255     ok(offsets.cbBodyEnd == 666, "got %d\n", offsets.cbBodyEnd);
 256     IMimeBody_Release(body);
 257
 258     hr = IMimeMessage_GetBody(msg, IBL_ROOT, NULL, &hbody);
 259     ok(hr == S_OK, "ret %08x\n", hr);
 260
 261     PropVariantInit(&prop);
 262     hr = IMimeMessage_GetBodyProp(msg, hbody, att_pritype, 0, &prop);
 263     ok(hr == S_OK, "ret %08x\n", hr);
 264     ok(prop.vt == VT_LPSTR, "vt %08x\n", prop.vt);
 265     ok(!strcasecmp(prop.u.pszVal, "multipart"), "got %s\n", prop.u.pszVal);
 266     PropVariantClear(&prop);
 267
 268     hr = IMimeMessage_GetBody(msg, IBL_FIRST, hbody, &hbody);
 269     ok(hr == S_OK, "ret %08x\n", hr);
 270     hr = IMimeMessage_BindToObject(msg, hbody, &IID_IMimeBody, (void**)&body);
 271     ok(hr == S_OK, "ret %08x\n", hr);
 272     hr = IMimeBody_GetOffsets(body, &offsets);
 273     ok(hr == S_OK, "ret %08x\n", hr);
 274     ok(offsets.cbBoundaryStart == 405, "got %d\n", offsets.cbBoundaryStart);
 275     ok(offsets.cbHeaderStart == 428, "got %d\n", offsets.cbHeaderStart);
 276     ok(offsets.cbBodyStart == 518, "got %d\n", offsets.cbBodyStart);
 277     ok(offsets.cbBodyEnd == 523, "got %d\n", offsets.cbBodyEnd);
 278
 279     hr = IMimeBody_GetCharset(body, &hcs);
 280     ok(hr == S_OK, "ret %08x\n", hr);
 281     todo_wine
 282     {
 283         ok(hcs != NULL, "Expected non-NULL charset\n");
 284     }
 285
 286     IMimeBody_Release(body);
 287
 288     hr = IMimeMessage_GetBody(msg, IBL_NEXT, hbody, &hbody);
 289     ok(hr == S_OK, "ret %08x\n", hr);
 290     hr = IMimeMessage_BindToObject(msg, hbody, &IID_IMimeBody, (void**)&body);
 291     ok(hr == S_OK, "ret %08x\n", hr);
 292     hr = IMimeBody_GetOffsets(body, &offsets);
 293     ok(hr == S_OK, "ret %08x\n", hr);
 294     ok(offsets.cbBoundaryStart == 525, "got %d\n", offsets.cbBoundaryStart);
 295     ok(offsets.cbHeaderStart == 548, "got %d\n", offsets.cbHeaderStart);
 296     ok(offsets.cbBodyStart == 629, "got %d\n", offsets.cbBodyStart);
 297     ok(offsets.cbBodyEnd == 639, "got %d\n", offsets.cbBodyEnd);
 298     IMimeBody_Release(body);
 299
 300     find_struct.pszPriType = text;
 301     find_struct.pszSubType = NULL;
 302
 303     hr = IMimeMessage_FindFirst(msg, &find_struct, &hbody);
 304     ok(hr == S_OK, "ret %08x\n", hr);
 305
 306     hr = IMimeMessage_FindNext(msg, &find_struct, &hbody);
 307     ok(hr == S_OK, "ret %08x\n", hr);
 308
 309     hr = IMimeMessage_FindNext(msg, &find_struct, &hbody);
 310     ok(hr == MIME_E_NOT_FOUND, "ret %08x\n", hr);
 311
 312     hr = IMimeMessage_GetAttachments(msg, &count, &body_list);
 313     ok(hr == S_OK, "ret %08x\n", hr);
 314     ok(count == 2, "got %d\n", count);
 315     CoTaskMemFree(body_list);
 316
 317     hr = IMimeMessage_GetCharset(body, &hcs);
 318     ok(hr == S_OK, "ret %08x\n", hr);
 319     todo_wine
 320     {
 321         ok(hcs != NULL, "Expected non-NULL charset\n");
 322     }
 323
 324     IMimeMessage_Release(msg);
 325
 326     ref = IStream_AddRef(stream);
 327     ok(ref == 2 ||
 328        broken(ref == 1), /* win95 */
 329        "ref %d\n", ref);
 330     IStream_Release(stream);
 331
 332     IStream_Release(stream);
 333 }
 334
 335 START_TEST(mimeole)
 336 {
 337     OleInitialize(NULL);
 338     test_CreateVirtualStream();
 339     test_CreateSecurity();
 340     test_CreateBody();
 341     test_Allocator();
 342     test_CreateMessage();
 343     OleUninitialize();
 344 }