msi: Avoid accessing a freed object.
[wine.git] / dlls / mapi32 / imalloc.c
blob1562b7b0fee0fe68e9145e8d77ae43f565cb79c6
1 /*
2 * MAPI Default IMalloc implementation
4 * Copyright 2004 Jon Griffiths
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.
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.
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
21 #include <stdarg.h>
23 #define COBJMACROS
24 #define NONAMELESSUNION
25 #define NONAMELESSSTRUCT
26 #include "windef.h"
27 #include "winbase.h"
28 #include "winreg.h"
29 #include "winuser.h"
30 #include "winerror.h"
31 #include "winternl.h"
32 #include "objbase.h"
33 #include "shlwapi.h"
34 #include "mapiutil.h"
35 #include "wine/debug.h"
37 WINE_DEFAULT_DEBUG_CHANNEL(mapi);
39 static const IMallocVtbl MAPI_IMalloc_vt;
41 typedef struct
43 const IMallocVtbl *lpVtbl;
44 LONG lRef;
45 } MAPI_IMALLOC;
47 static MAPI_IMALLOC MAPI_IMalloc = { &MAPI_IMalloc_vt, 0u };
49 extern LONG MAPI_ObjectCount; /* In mapi32_main.c */
51 /*************************************************************************
52 * MAPIGetDefaultMalloc@0 (MAPI32.59)
54 * Get the default MAPI IMalloc interface.
56 * PARAMS
57 * None.
59 * RETURNS
60 * A pointer to the MAPI default allocator.
62 LPMALLOC WINAPI MAPIGetDefaultMalloc(void)
64 TRACE("()\n");
66 IMalloc_AddRef((LPMALLOC)&MAPI_IMalloc);
67 return (LPMALLOC)&MAPI_IMalloc;
70 /**************************************************************************
71 * IMAPIMalloc_QueryInterface
73 static HRESULT WINAPI IMAPIMalloc_fnQueryInterface(LPMALLOC iface, REFIID refiid,
74 LPVOID *ppvObj)
76 TRACE("(%s,%p)\n", debugstr_guid(refiid), ppvObj);
78 if (IsEqualIID(refiid, &IID_IUnknown) ||
79 IsEqualIID(refiid, &IID_IMalloc))
81 *ppvObj = (LPMALLOC) &MAPI_IMalloc;
82 TRACE("Returning IMalloc (%p)\n", *ppvObj);
83 return S_OK;
85 TRACE("Returning E_NOINTERFACE\n");
86 return E_NOINTERFACE;
89 /**************************************************************************
90 * IMAPIMalloc_AddRef
92 static ULONG WINAPI IMAPIMalloc_fnAddRef(LPMALLOC iface)
94 TRACE("(%p)\n", iface);
95 InterlockedIncrement(&MAPI_ObjectCount);
96 return 1u;
99 /**************************************************************************
100 * IMAPIMalloc_Release
102 static ULONG WINAPI IMAPIMalloc_fnRelease(LPMALLOC iface)
104 TRACE("(%p)\n", iface);
105 InterlockedDecrement(&MAPI_ObjectCount);
106 return 1u;
109 /**************************************************************************
110 * IMAPIMalloc_Alloc
112 static LPVOID WINAPI IMAPIMalloc_fnAlloc(LPMALLOC iface, DWORD cb)
114 TRACE("(%p)->(%d)\n", iface, cb);
116 return LocalAlloc(LMEM_FIXED, cb);
119 /**************************************************************************
120 * IMAPIMalloc_Realloc
122 static LPVOID WINAPI IMAPIMalloc_fnRealloc(LPMALLOC iface, LPVOID pv, DWORD cb)
124 TRACE("(%p)->(%p, %d)\n", iface, pv, cb);
126 if (!pv)
127 return LocalAlloc(LMEM_FIXED, cb);
129 if (cb)
130 return LocalReAlloc((HANDLE) pv, cb, LMEM_MOVEABLE);
132 LocalFree((HANDLE) pv);
133 return NULL;
136 /**************************************************************************
137 * IMAPIMalloc_Free
139 static void WINAPI IMAPIMalloc_fnFree(LPMALLOC iface, LPVOID pv)
141 TRACE("(%p)->(%p)\n", iface, pv);
142 LocalFree((HANDLE) pv);
145 /**************************************************************************
146 * IMAPIMalloc_GetSize
148 static DWORD WINAPI IMAPIMalloc_fnGetSize(LPMALLOC iface, LPVOID pv)
150 TRACE("(%p)->(%p)\n", iface, pv);
151 return LocalSize((HANDLE) pv);
154 /**************************************************************************
155 * IMAPIMalloc_DidAlloc
157 static INT WINAPI IMAPIMalloc_fnDidAlloc(LPMALLOC iface, LPVOID pv)
159 TRACE("(%p)->(%p)\n", iface, pv);
160 return -1;
163 /**************************************************************************
164 * IMAPIMalloc_HeapMinimize
166 static void WINAPI IMAPIMalloc_fnHeapMinimize(LPMALLOC iface)
168 TRACE("(%p)\n", iface);
171 static const IMallocVtbl MAPI_IMalloc_vt =
173 IMAPIMalloc_fnQueryInterface,
174 IMAPIMalloc_fnAddRef,
175 IMAPIMalloc_fnRelease,
176 IMAPIMalloc_fnAlloc,
177 IMAPIMalloc_fnRealloc,
178 IMAPIMalloc_fnFree,
179 IMAPIMalloc_fnGetSize,
180 IMAPIMalloc_fnDidAlloc,
181 IMAPIMalloc_fnHeapMinimize