dlls/mapi32/tests/imalloc.c

   1 /*
   2  * Unit test suite for MAPI IMalloc functions
   3  *
   4  * Copyright 2004 Jon Griffiths
   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 #include "wine/test.h"
  23 #include "windef.h"
  24 #include "winbase.h"
  25 #include "winuser.h"
  26 #include "winerror.h"
  27 #include "winnt.h"
  28 #include "mapiutil.h"
  29 #include "mapi32_test.h"
  30
  31 static HMODULE hMapi32 = 0;
  32
  33 static SCODE    (WINAPI *pScInitMapiUtil)(ULONG);
  34 static LPMALLOC (WINAPI *pMAPIGetDefaultMalloc)(void);
  35
  36 static void test_IMalloc(void)
  37 {
  38     LPVOID lpMem;
  39     ULONG ulRef;
  40     int iRet;
  41     HRESULT hRet;
  42     LPMALLOC lpMalloc;
  43     LPVOID lpVoid;
  44
  45     pMAPIGetDefaultMalloc = (void*)GetProcAddress(hMapi32,
  46                                                   "MAPIGetDefaultMalloc@0");
  47     if (!pMAPIGetDefaultMalloc)
  48     {
  49         win_skip("MAPIGetDefaultMalloc is not available\n");
  50         return;
  51     }
  52
  53     lpMalloc = pMAPIGetDefaultMalloc();
  54     ok(lpMalloc != NULL, "Expected MAPIGetDefaultMalloc to return non-NULL\n");
  55     if (!lpMalloc)
  56     {
  57         skip("MAPIGetDefaultMalloc failed\n");
  58         return;
  59     }
  60
  61     lpVoid = NULL;
  62     hRet = IMalloc_QueryInterface(lpMalloc, &IID_IUnknown, &lpVoid);
  63     ok (hRet == S_OK && lpVoid != NULL,
  64         "IID_IUnknown: expected S_OK, non-null, got 0x%08x, %p\n",
  65         hRet, lpVoid);
  66
  67     lpVoid = NULL;
  68     hRet = IMalloc_QueryInterface(lpMalloc, &IID_IMalloc, &lpVoid);
  69     ok (hRet == S_OK && lpVoid != NULL,
  70         "IID_IIMalloc: expected S_OK, non-null, got 0x%08x, %p\n",
  71         hRet, lpVoid);
  72
  73     /* Prove that native mapi uses LocalAlloc/LocalFree */
  74     lpMem = IMalloc_Alloc(lpMalloc, 61);
  75     ok (lpMem && IMalloc_GetSize(lpMalloc, lpMem) == LocalSize(lpMem),
  76         "Expected non-null, same size, got %p, %s size\n", lpMem,
  77         lpMem ? "different" : "same");
  78
  79     iRet = IMalloc_DidAlloc(lpMalloc, lpMem);
  80     ok (iRet == -1, "DidAlloc, expected -1. got %d\n", iRet);
  81
  82     IMalloc_HeapMinimize(lpMalloc);
  83
  84     LocalFree(lpMem);
  85
  86     ulRef = IMalloc_AddRef(lpMalloc);
  87     ok (ulRef == 1u, "AddRef expected 1, returned %d\n", ulRef);
  88
  89     ulRef = IMalloc_Release(lpMalloc);
  90     ok (ulRef == 1u, "AddRef expected 1, returned %d\n", ulRef);
  91
  92     IMalloc_Release(lpMalloc);
  93 }
  94
  95 START_TEST(imalloc)
  96 {
  97     SCODE ret;
  98
  99     if (!HaveDefaultMailClient())
 100     {
 101         win_skip("No default mail client installed\n");
 102         return;
 103     }
 104
 105     hMapi32 = LoadLibraryA("mapi32.dll");
 106
 107     pScInitMapiUtil = (void*)GetProcAddress(hMapi32, "ScInitMapiUtil@4");
 108     if (!pScInitMapiUtil)
 109     {
 110         win_skip("ScInitMapiUtil is not available\n");
 111         FreeLibrary(hMapi32);
 112         return;
 113     }
 114
 115     SetLastError(0xdeadbeef);
 116     ret = pScInitMapiUtil(0);
 117     if ((ret != S_OK) && (GetLastError() == ERROR_PROC_NOT_FOUND))
 118     {
 119         win_skip("ScInitMapiUtil is not implemented\n");
 120         FreeLibrary(hMapi32);
 121         return;
 122     }
 123     else if ((ret == E_FAIL) && (GetLastError() == ERROR_INVALID_HANDLE))
 124     {
 125         win_skip("ScInitMapiUtil doesn't work on some Win98 and WinME systems\n");
 126         FreeLibrary(hMapi32);
 127         return;
 128     }
 129
 130     test_IMalloc();
 131
 132     FreeLibrary(hMapi32);
 133 }