librdmacm: allow for graceful cleanup of verbs

[mirror/winof/.git] / ulp / librdmacm / src / cma.h

diff --git a/ulp/librdmacm/src/cma.h b/ulp/librdmacm/src/cma.h
index fb65026..918a793 100644 (file)
--- a/ulp/librdmacm/src/cma.h
+++ b/ulp/librdmacm/src/cma.h
@@ -32,15 +32,18 @@
 #define CMA_H\r
 \r
 extern CRITICAL_SECTION lock;\r
 #define CMA_H\r
 \r
 extern CRITICAL_SECTION lock;\r

+extern HANDLE heap;\r
+\r
+void ucma_cleanup();\r

 \r
 __inline void* __cdecl operator new(size_t size)\r
 {\r
 \r
 __inline void* __cdecl operator new(size_t size)\r
 {\r

-       return HeapAlloc(GetProcessHeap(), 0, size);\r
+       return HeapAlloc(heap, 0, size);\r

 }\r
 \r
 __inline void __cdecl operator delete(void *pObj)\r
 {\r
 }\r
 \r
 __inline void __cdecl operator delete(void *pObj)\r
 {\r

-       HeapFree(GetProcessHeap(), 0, pObj);\r
+       HeapFree(heap, 0, pObj);\r

 }\r
 \r
 #endif /* CMA_H */\r
 }\r
 \r
 #endif /* CMA_H */\r