开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 查看源码
MEMORY.CPP
资源名称：MSDN_VC98.zip　[点击查看]
上传用户：bangxh
上传日期：2007-01-31
资源大小：42235k
文件大小：3k
源码类别：

Windows编程

开发平台：

Visual C++

MEMORY.CPP：源码内容
  1. /*++
  3. Copyright (c) 1996 Microsoft Corporation
  5. Module Name:
  7.     Memory.cpp
  9. Abstract:
  11.     This module provides all the memory management functions for all Sample
  12.     provider components
  15. Author:
  17. Environment:
  19.     User mode
  21. Revision History :
  23. --*/
  24. #include "adssmp.h"
  26. #define WORD_ALIGN_DOWN(addr) ((LPBYTE)((DWORD)addr &= ~1))
  27. #define DWORD_ALIGN_UP(size) ((size+3)&~3)
  30. LPVOID
  31. AllocProvMem(
  32.     DWORD cb
  33. )
  34. {
  35.     LPDWORD  pMem;
  36.     DWORD    cbNew ;
  38. #if DBG
  39.     cb = DWORD_ALIGN_UP(cb);
  41.     cbNew = cb + (sizeof(DWORD) + sizeof(LONGLONG));
  42. #else
  43.     cbNew = cb ;
  44. #endif
  46.     pMem=(LPDWORD)LocalAlloc(LPTR, cbNew);
  49. #if DBG
  51.     if (!pMem) {
  52.         SetLastError(ERROR_NOT_ENOUGH_MEMORY);
  53.         return 0;
  54.     }
  56.     *pMem=cb;
  57.     *(LPDWORD)((LPBYTE)pMem+cbNew-sizeof(DWORD))=0xdeadfeed;
  58.     pMem = (LPDWORD) (((LPBYTE)pMem)+sizeof(LONGLONG));
  59. #endif
  61.     return (LPVOID)(pMem);
  62. }
  64. BOOL
  65. FreeProvMem(
  66.    LPVOID pMem
  67. )
  68. {
  69.     LPBYTE   pNewMem;
  71.     pNewMem = (LPBYTE)pMem;
  73. #if DBG
  74.     DWORD    cb;
  75.     DWORD    cbNew = 0;
  77.     pNewMem -= sizeof(LONGLONG) ;
  79.     cb = *((LPDWORD)pNewMem);
  80.     cbNew = cb + sizeof(DWORD) + sizeof(LONGLONG);
  82.     if (*(LPDWORD)(pNewMem + cbNew - sizeof(DWORD)) != 0xdeadbeef) {
  83.         ADsAssert(!"Freeing memory not allocated by AllocProvMem") ;
  84.         return FALSE;
  85.     }
  87.     // fill freed memory with known pattern
  88.     memset(pMem, 0x5C, cb);
  90. #endif
  93.     return(LocalFree((LPVOID)pNewMem) == NULL);
  94. }
  96. LPVOID
  97. ReallocProvMem(
  98.    LPVOID pOldMem,
  99.    DWORD cbOld,
  100.    DWORD cbNew
  101. )
  102. {
  103.     LPVOID pNewMem;
  105.     pNewMem=AllocProvMem(cbNew);
  107.     if (pOldMem && pNewMem) {
  108.         memcpy(pNewMem, pOldMem, min(cbNew, cbOld));
  109.         FreeProvMem(pOldMem);
  110.     }
  112.     return pNewMem;
  113. }
  116. LPWSTR
  117. AllocProvStr(
  118.     LPWSTR pStr
  119. )
  120. {
  121.    LPWSTR pMem;
  123.    if (!pStr)
  124.       return 0;
  126.    if (pMem = (LPWSTR)AllocProvMem( wcslen(pStr)*sizeof(WCHAR) + sizeof(WCHAR) ))
  127.       wcscpy(pMem, pStr);
  129.    return pMem;
  130. }
  132. BOOL
  133. FreeProvStr(
  134.    LPWSTR pStr
  135. )
  136. {
  137.    return pStr ? FreeProvMem(pStr)
  138.                : FALSE;
  139. }
  141. BOOL
  142. ReallocProvStr(
  143.    LPWSTR *ppStr,
  144.    LPWSTR pStr
  145. )
  146. {
  147.    FreeProvStr(*ppStr);
  148.    *ppStr=AllocProvStr(pStr);
  150.    return TRUE;
  151. }
  153. STDAPI
  154. ProvAllocString(const OLECHAR * pch, BSTR * pBSTR)
  155. {
  156.     HRESULT hr = S_OK;
  158.     ADsAssert(pBSTR);
  159.     if (!pch)
  160.     {
  161.         *pBSTR = NULL;
  162.         return S_OK;
  163.     }
  165.     *pBSTR = SysAllocString(pch);
  166.     hr = (*pBSTR) ? S_OK : E_OUTOFMEMORY;
  167.     RRETURN(hr);
  168. }