dlls/msvcrtd/tests/debug.c

   1 /*
   2  * Unit test suite for debug functions.
   3  *
   4  * Copyright 2004 Patrik Stridvall
   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 #include <stdarg.h>
  22
  23 #include "windef.h"
  24 #include "winbase.h"
  25 #include "winnt.h"
  26
  27 #include "crtdbg.h"
  28
  29 #include "wine/test.h"
  30
  31 /**********************************************************************/
  32
  33 static void * (__cdecl *pMSVCRTD_operator_new_dbg)(size_t, int, const char *, int) = NULL;
  34 static void * (__cdecl *pMSVCRTD_operator_delete)(void *) = NULL;
  35
  36 /* Some exports are only available in later versions */
  37 #define SETNOFAIL(x,y) x = (void*)GetProcAddress(hModule,y)
  38 #define SET(x,y) do { SETNOFAIL(x,y); ok(x != NULL, "Export '%s' not found\n", y); } while(0)
  39
  40 static BOOL init_functions(void)
  41 {
  42   HMODULE hModule = LoadLibraryA("msvcrtd.dll");
  43
  44   if (!hModule) {
  45     win_skip("LoadLibraryA failed to load msvcrtd.dll with GLE=%ld\n", GetLastError());
  46     return FALSE;
  47   }
  48
  49   if (sizeof(void *) > sizeof(int))  /* 64-bit has a different mangled name */
  50   {
  51       SET(pMSVCRTD_operator_new_dbg, "??2@YAPEAX_KHPEBDH@Z");
  52       SET(pMSVCRTD_operator_delete, "??3@YAXPEAX@Z");
  53   }
  54   else
  55   {
  56       SET(pMSVCRTD_operator_new_dbg, "??2@YAPAXIHPBDH@Z");
  57       SET(pMSVCRTD_operator_delete, "??3@YAXPAX@Z");
  58   }
  59
  60   if (pMSVCRTD_operator_new_dbg == NULL)
  61     return FALSE;
  62
  63   return TRUE;
  64 }
  65
  66 /**********************************************************************/
  67
  68 static void test_new(void)
  69 {
  70   void *mem;
  71
  72   mem = pMSVCRTD_operator_new_dbg(42, _NORMAL_BLOCK, __FILE__, __LINE__);
  73   ok(mem != NULL, "memory not allocated\n");
  74   pMSVCRTD_operator_delete(mem);
  75
  76   mem = pMSVCRTD_operator_new_dbg(42, _CRT_BLOCK, __FILE__, __LINE__);
  77   ok(mem != NULL, "memory not allocated\n");
  78   pMSVCRTD_operator_delete(mem);
  79 }
  80
  81 /**********************************************************************/
  82
  83 START_TEST(debug)
  84 {
  85   if (!init_functions())
  86     return;
  87
  88   test_new();
  89 }