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     trace("LoadLibraryA failed to load msvcrtd.dll with GLE=%d\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 }
90