进程间通信_共享内存类_C++

Bertha 。 2022-08-12 12:00 153阅读 0赞

转自http://blog.csdn.net/fengrx/article/details/4069088\#

/\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*

\++ File Name : FFMClass.h  
\++ Description: 共享內存類  
\---------------------------------------------------------------  
\++ Author: Fei ZhaoDong  
\++ Create time: 2004/3/25 上午 10:00:00  
\++ Version: 1.1  
\++ Modifier:  
\++ Activities:  
\++ Update List: 2004/3/30 下午 02:59:45  
\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*/  
// FFMClass.h: interface for the CFFMClass.  
//  
//  
\#ifndef FZD\_FFMCLASS\_H  
\#define FZD\_FFMCLASS\_H  
//\#include <Winbase.h>  
\#include <aclapi.h>  
  
  
//  
// Macro definition  
// 以下為共享內存定義  
\#define DEFAULT\_FILENAME NULL // 默認的文件名  
\#define DEFAULT\_MAPNAME "Local//\_FZD\_MAP\_" // 默認的共享內存名  
\#define DEFAULT\_MAPSIZE (0xFFFF + 1) // 默認的共享內存大小  
const DWORD NETRGUSER\_CFM\_CODE = 0x1211DBFF; // 校驗碼, 用於命令數據  
const DWORD NETRGUSER\_CMD\_NONE = 0; // 初始化指令碼, 無指令  
// 以下為錯誤碼定義  
\#define ERROR\_LEN 256 // 錯誤描述長度  
\#define ERROR\_INVALID\_CMDCODE 0xE00001FF // 已經存在完全一樣的共享內存  
\#define ERROR\_NO\_MAPFILE 0xE00002FF // 未分配共享內存文件  
\#define ERROR\_INVALID\_CFMCODE 0xE00003FF // 校驗碼不匹配  
//  
// 內存文件格式定義  
\#pragma pack(1)  
// 用於存儲命令數據的內存文件格式  
typedef struct \_tagDATA\_HEADER  
\{  
DWORD dwConfirmCode; // 校驗碼  
DWORD nCommandCode; // 指令識別碼  
DWORD dwDataSize; // 數據的大小  
BYTE dwReserved\[19\]; // 保留  
BYTE bInfo\[1\]; // 數據起始地址  
\_tagDATA\_HEADER()  
\{  
dwConfirmCode = NETRGUSER\_CFM\_CODE;  
nCommandCode = NETRGUSER\_CMD\_NONE;  
dwDataSize = 0;  
ZeroMemory(dwReserved, 19);  
ZeroMemory(bInfo, 1);  
\}  
\}DATA\_HEADER, \*LPDATA\_HEADER;  
typedef DWORD (WINAPI \*PSetEntriesInAcl)(ULONG, PEXPLICIT\_ACCESS, PACL, PACL\*);  
// 用於存儲應答數據的共享內存文件格式 (暫不用)  
typedef struct \_tagANS\_HEADER  
\{  
\}ANS\_HEADER, \*LPANS\_HEADER;  
  
  
\#pragma pack()  
  
  
//  
// 類定義，共享內存服務端  
class CFFMServer  
\{  
public:  
CFFMServer();  
virtual ~CFFMServer();  
CFFMServer(char \*szFileName, char \*szMapName, DWORD dwSize);  
protected:  
PSetEntriesInAcl m\_fnpSetEntriesInAcl;  
HANDLE m\_hFile; // 映射文件句柄  
HANDLE m\_hFileMap; // 內存文件句柄  
LPVOID m\_lpFileMapBuffer; // 緩衝區指針  
char \*m\_pFileName; // 映射文件名  
char \*m\_pMapName; // 內存文件名  
DWORD m\_dwSize; // 緩衝區大小  
BOOL m\_bCreateFlag; // 是否已創建共享內存  
DWORD m\_dwLastError; // 錯誤代碼  
private:  
void \_Init(); // 初始化參數  
BOOL \_IsWinNTLater(); // 判斷當前操作系統  
public:  
BOOL Create(char \*szFileName = DEFAULT\_FILENAME,  
char \*szMapName = DEFAULT\_MAPNAME,  
DWORD dwSize = DEFAULT\_MAPSIZE); // 新建共享內存  
LPVOID GetBuffer(); // 獲取內存文件指針  
DWORD GetSize(); // 獲取內存文件大小  
void Destory(); // 銷毀已有的共享內存  
BOOL WriteCmdData( // 寫入命令數據  
DWORD nCommandCode,  
DWORD dwDataSize,  
const LPVOID pBuf);  
\};  
//  
// 類定義，共享內存客戶端  
class CFFMClient  
\{  
public:  
CFFMClient();  
virtual ~CFFMClient();  
CFFMClient(DWORD dwAccess, char \*szMapName, DWORD dwSize);  
protected:  
HANDLE m\_hFileMap; // 內存文件句柄  
LPVOID m\_lpFileMapBuffer; // 內存文件指針  
char \*m\_pMapName; // 內存文件名  
DWORD m\_dwSize; // 緩衝區大小  
BOOL m\_bOpenFlag; // 是否已經打開了一個內存文件  
DWORD m\_dwLastError; // 錯誤代碼  
private:  
void \_Init(); // 初始化參數  
public:  
BOOL Open(DWORD dwAccess = FILE\_MAP\_READ | FILE\_MAP\_WRITE,  
char \*szMapName = DEFAULT\_MAPNAME,  
DWORD dwSize = 0); // 打開一個內存文件  
LPVOID GetBuffer(); // 獲取當前內存文件的指針  
void Destory(); // 關閉當前對內存文件的訪問  
DWORD GetSize(); // 獲取內存文件大小  
BOOL GetCmdDataSize(DWORD \*pDataSize); // 讀取命令數據大小  
BOOL ReadCmdData( // 讀取命令數據  
DWORD dwCommandCode,  
DWORD dwBufSize,  
LPVOID pOutBuf);  
BOOL WriteCmdData( // 寫入命令數據  
DWORD memSize,  
DWORD nCommandCode,  
DWORD dwDataSize,  
const LPVOID pBuf);  
\};

\#endif // FZD\_FFMCLASS\_H

/\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*  
\++ File Name : FFMClass.cpp  
\++ Description: 共享內存類  
\---------------------------------------------------------------  
\++ Author: Fei ZhaoDong  
\++ Create time: 2004/3/25 上午 10:00:00  
\++ Version: 1.0  
\++ Modifier:  
\++ Activities:  
\++ Update List: 2004/3/29 下午 02:59:45  
\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*/  
// FFMClass.cpp: implementation of the CFFMClass.  
//  
//  
\#include "stdafx.h"  
\#include "FFMClass.h"  
//\#include <aclapi.h>  
//\#include <windows.h>  
//\#include <aclapi.h>  
//\#include <lmerr.h>  
//  
//\#include <stdio.h>  
//  
// CFFMServer  
//  
CFFMServer::CFFMServer()  
\{  
m\_dwLastError = 0;  
m\_fnpSetEntriesInAcl = NULL;  
\_Init();  
\}  
CFFMServer::~CFFMServer()  
\{  
Destory();  
\}  
CFFMServer::CFFMServer(char \*szFileName, char \*szMapName, DWORD dwSize)  
\{  
// 以自定義設置創建共享內存塊  
\_Init();  
Create(szFileName, szMapName, dwSize);  
\}  
// 初始化各個參數  
void CFFMServer::\_Init()  
\{  
m\_hFile = NULL;  
m\_hFileMap = NULL;  
m\_lpFileMapBuffer = NULL;  
m\_pFileName = NULL;  
m\_pMapName = NULL;  
m\_dwSize = 0;  
m\_bCreateFlag = FALSE;  
\}  
// 判斷是否NT4.0以上操作系統  
BOOL CFFMServer::\_IsWinNTLater()  
\{  
OSVERSIONINFO Ver;  
BOOL bAbleVersion = FALSE;  
Ver.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);  
if (GetVersionEx(&Ver))  
\{  
if (Ver.dwPlatformId == VER\_PLATFORM\_WIN32\_NT  
&& Ver.dwMajorVersion >= 4)  
\{  
bAbleVersion = TRUE;  
\}  
\}  
else  
\{  
m\_dwLastError = GetLastError();  
\}  
return bAbleVersion;  
\}  
// 釋放當前共享內存,並重新初始化參數  
void CFFMServer::Destory()  
\{  
if (m\_lpFileMapBuffer != NULL)  
\{  
UnmapViewOfFile(m\_lpFileMapBuffer);  
m\_lpFileMapBuffer = NULL;  
\}  
if (m\_hFileMap != NULL)  
\{  
CloseHandle(m\_hFileMap);  
m\_hFileMap = NULL;  
\}  
if (m\_hFile && m\_hFile != INVALID\_HANDLE\_VALUE)  
\{  
CloseHandle(m\_hFile);  
m\_hFile = NULL;  
\}  
if (m\_pFileName != NULL)  
\{  
free(m\_pFileName);  
m\_pFileName = NULL;  
\}  
if (m\_pMapName != NULL)  
\{  
free(m\_pMapName);  
m\_pMapName = NULL;  
\}  
\_Init();  
\}  
static void FreeSidEx(PSID oSID)  
\{  
try  
\{  
FreeSid(oSID);  
\}  
catch(...)  
\{  
\}  
\}  
// 創建共享內存塊  
BOOL CFFMServer::Create(char \*szFileName, char \*szMapName, DWORD dwSize)  
\{  
// 釋放已有的共享內存塊  
if (m\_bCreateFlag)  
\{  
Destory();  
\}  
// 拷貝各個參數  
if (szFileName)  
\{  
m\_pFileName = \_strdup(szFileName);  
\}  
if (szMapName)  
\{  
m\_pMapName = \_strdup(szMapName);  
\}  
else  
\{  
m\_pMapName = \_strdup(DEFAULT\_MAPNAME);  
\}  
if (dwSize > 0)  
\{  
m\_dwSize = dwSize;  
\}  
else  
\{  
m\_dwSize = DEFAULT\_MAPSIZE;  
\}  
// 以下創建共享內存  
if (m\_pFileName)  
\{  
m\_hFile = CreateFile(  
m\_pFileName,  
GENERIC\_READ|GENERIC\_WRITE,  
FILE\_SHARE\_READ|FILE\_SHARE\_WRITE,  
NULL,  
OPEN\_ALWAYS,  
FILE\_ATTRIBUTE\_NORMAL,  
NULL  
);  
\}  
else  
\{  
// 默認情況下,在頁面文件中創建共享內存  
m\_hFile = INVALID\_HANDLE\_VALUE;  
\}  
if (\_IsWinNTLater())  
\{  
// Set DACL  
const int NUM\_ACES = 2; // number if ACEs int DACL  
// evryone -- read  
// creator -- full access  
// 初始化參數  
PSID pEveryoneSID = NULL; // everyone群組SID  
PSID pCreatorSID = NULL; // creator群組SID  
PACL pFileMapACL = NULL; // 準備新內存文件的DACL  
PSECURITY\_DESCRIPTOR pSD = NULL; // 內存文件的SD  
SECURITY\_ATTRIBUTES saFileMap; // 內存文件的SA  
EXPLICIT\_ACCESS ea\[NUM\_ACES\]; // 外部訪問結構  
BOOL bHasErr = FALSE; // 返回值  
// 以下創建SID  
SID\_IDENTIFIER\_AUTHORITY SIDAuthWorld = SECURITY\_WORLD\_SID\_AUTHORITY;  
SID\_IDENTIFIER\_AUTHORITY SIDAuthCreator = SECURITY\_CREATOR\_SID\_AUTHORITY;  
// Evryone  
if (!AllocateAndInitializeSid(&SIDAuthWorld, 1, SECURITY\_WORLD\_RID,  
0, 0, 0, 0, 0, 0, 0, &pEveryoneSID))  
\{  
bHasErr = TRUE;  
goto Finish;  
\}  
// Creator  
if (!AllocateAndInitializeSid(&SIDAuthCreator, 1, SECURITY\_CREATOR\_OWNER\_RID,  
0, 0, 0, 0, 0, 0, 0, &pCreatorSID))  
\{  
bHasErr = TRUE;  
goto Finish;  
\}  
// 填充ACE  
ZeroMemory(&ea, NUM\_ACES \* sizeof(EXPLICIT\_ACCESS));  
// S-1-1-0 evryone, 唯讀權限  
ea\[0\].grfAccessPermissions = GENERIC\_READ | GENERIC\_WRITE;  
ea\[0\].grfAccessMode = SET\_ACCESS;  
ea\[0\].grfInheritance= NO\_INHERITANCE;  
ea\[0\].Trustee.TrusteeForm = TRUSTEE\_IS\_SID;  
ea\[0\].Trustee.TrusteeType = TRUSTEE\_IS\_WELL\_KNOWN\_GROUP;  
ea\[0\].Trustee.ptstrName = (LPTSTR) pEveryoneSID;  
// S-1-3-0 creator owner, 完全權限  
ea\[1\].grfAccessPermissions = STANDARD\_RIGHTS\_ALL;  
ea\[1\].grfAccessMode = SET\_ACCESS;  
ea\[1\].grfInheritance= NO\_INHERITANCE;  
ea\[1\].Trustee.TrusteeForm = TRUSTEE\_IS\_SID;  
ea\[1\].Trustee.TrusteeType = TRUSTEE\_IS\_WELL\_KNOWN\_GROUP;  
ea\[1\].Trustee.ptstrName = (LPTSTR) pCreatorSID;  
// 創建並填充ACL  
if(NULL == m\_fnpSetEntriesInAcl)  
\{  
HINSTANCE hLib = ::LoadLibrary("Advapi32.dll");  
if(NULL != hLib)  
\{  
m\_fnpSetEntriesInAcl = (PSetEntriesInAcl)GetProcAddress(hLib,"SetEntriesInAclA");  
::FreeLibrary(hLib);  
hLib = NULL;  
\}  
\}  
if (ERROR\_SUCCESS != m\_fnpSetEntriesInAcl(NUM\_ACES, ea, NULL, &pFileMapACL))  
\{  
bHasErr = TRUE;  
goto Finish;  
\}  
// 創建並初始化SD  
pSD = (PSECURITY\_DESCRIPTOR)LocalAlloc(LPTR, SECURITY\_DESCRIPTOR\_MIN\_LENGTH);  
if (NULL == pSD)  
\{  
bHasErr = TRUE;  
goto Finish;  
\}  
if (!InitializeSecurityDescriptor(pSD, SECURITY\_DESCRIPTOR\_REVISION))  
\{  
bHasErr = TRUE;  
goto Finish;  
\}  
// 添加ACL到SD中去  
if (!SetSecurityDescriptorDacl(pSD,  
TRUE, // fDaclPresent flag  
pFileMapACL,  
FALSE)) // not a default DACL  
\{  
bHasErr = TRUE;  
goto Finish;  
\}  
// 設置SA  
saFileMap.nLength = sizeof(SECURITY\_ATTRIBUTES);  
saFileMap.bInheritHandle = FALSE;  
saFileMap.lpSecurityDescriptor = pSD;  
// 創建共享內存文件  
if (m\_hFile != NULL)  
\{  
m\_hFileMap = CreateFileMapping(  
m\_hFile,  
&saFileMap,  
PAGE\_READWRITE,  
0,  
m\_dwSize,  
m\_pMapName);  
if (NULL == m\_hFileMap)  
\{  
m\_hFileMap = OpenFileMapping(FILE\_MAP\_READ | FILE\_MAP\_WRITE,  
TRUE, m\_pMapName);  
if (NULL == m\_hFileMap)  
\{  
m\_dwLastError = GetLastError();  
Destory();  
goto Finish;  
\}  
\}  
\}  
Finish:  
//if (pSD != NULL)  
//\{  
//LocalFree(pSD);  
//\}  
pSD = NULL;  
if (pFileMapACL != NULL)  
\{  
LocalFree(pFileMapACL);  
pFileMapACL = NULL;  
\}  
if (pEveryoneSID != NULL)  
\{  
FreeSidEx(pEveryoneSID);  
pEveryoneSID = NULL;  
\}  
if (pCreatorSID != NULL)  
\{  
FreeSidEx(pCreatorSID);  
pCreatorSID = NULL;  
\}  
if (bHasErr)  
\{  
m\_dwLastError = GetLastError();  
return FALSE;  
\}  
\}  
else  
\{  
// 創建共享內存文件  
if (m\_hFile)  
\{  
m\_hFileMap = CreateFileMapping(  
m\_hFile,  
NULL,  
PAGE\_READWRITE,  
0,  
m\_dwSize,  
m\_pMapName);  
if (NULL == m\_hFileMap)  
\{  
m\_dwLastError = GetLastError();  
Destory();  
return FALSE;  
\}  
\}  
\}  
// 映射文件指針到用戶  
if (m\_hFileMap)  
\{  
m\_lpFileMapBuffer = MapViewOfFile(  
m\_hFileMap,  
FILE\_MAP\_ALL\_ACCESS,  
0,  
0,  
m\_dwSize);  
if (NULL == m\_lpFileMapBuffer)  
\{  
m\_dwLastError = GetLastError();  
Destory();  
return FALSE;  
\}  
\}  
m\_bCreateFlag = TRUE;  
return TRUE;  
\}  
// 獲取內存文件指針  
LPVOID CFFMServer::GetBuffer()  
\{  
return (m\_lpFileMapBuffer)?(m\_lpFileMapBuffer):(NULL);  
\}  
// 獲取內存文件大小  
DWORD CFFMServer::GetSize()  
\{  
return m\_dwSize;  
\}  
BOOL CFFMServer::WriteCmdData(DWORD nCommandCode, DWORD dwDataSize, const LPVOID pBuf)  
\{  
// 檢驗數據的合理性  
if (NULL == GetBuffer())  
\{  
m\_dwLastError = ERROR\_NO\_MAPFILE;  
SetLastError(ERROR\_NO\_MAPFILE);  
return FALSE;  
\}  
if (NETRGUSER\_CMD\_NONE == nCommandCode)  
\{  
m\_dwLastError = ERROR\_INVALID\_CMDCODE;  
SetLastError(ERROR\_INVALID\_CMDCODE);  
return FALSE;  
\}  
if (dwDataSize > 0 && pBuf == NULL)  
\{  
m\_dwLastError = ERROR\_INVALID\_USER\_BUFFER;  
SetLastError(ERROR\_INVALID\_USER\_BUFFER);  
return FALSE;  
\}  
if (dwDataSize + sizeof(DATA\_HEADER) > GetSize())  
\{  
m\_dwLastError = ERROR\_BUFFER\_OVERFLOW;  
SetLastError(ERROR\_BUFFER\_OVERFLOW);  
return FALSE;  
\}  
// 填寫數據結構  
// 文件頭  
DATA\_HEADER dataHeader;  
dataHeader.nCommandCode = nCommandCode;  
dataHeader.dwDataSize = dwDataSize;  
ZeroMemory(GetBuffer(), GetSize());  
memcpy(GetBuffer(), &dataHeader, sizeof(DATA\_HEADER));  
// 數據塊  
LPDATA\_HEADER pData = (LPDATA\_HEADER)GetBuffer();  
memcpy(pData->bInfo, pBuf, dwDataSize);  
return TRUE;  
\}  
//  
// CFFMClient  
//  
CFFMClient::CFFMClient()  
\{  
m\_dwLastError = 0;  
\_Init();  
\}  
CFFMClient::~CFFMClient()  
\{  
Destory();  
\}  
CFFMClient::CFFMClient(DWORD dwAccess, char \*szMapName, DWORD dwSize)  
\{  
\_Init();  
if (!Open(dwAccess, szMapName, dwSize))  
\{  
Destory();  
\}  
\}  
// 初始化參數  
void CFFMClient::\_Init()  
\{  
m\_hFileMap = NULL;  
m\_lpFileMapBuffer = NULL;  
m\_pMapName = NULL;  
m\_bOpenFlag = FALSE;  
\}  
// 關閉當前對內存文件的訪問  
void CFFMClient::Destory()  
\{  
if (m\_lpFileMapBuffer)  
\{  
UnmapViewOfFile(m\_lpFileMapBuffer);  
m\_lpFileMapBuffer = NULL;  
\}  
if (m\_hFileMap)  
\{  
CloseHandle(m\_hFileMap);  
m\_hFileMap = NULL;  
\}  
if (m\_pMapName)  
\{  
free(m\_pMapName);  
m\_pMapName = NULL;  
\}  
\_Init();  
\}  
// 打開一個內存文件  
BOOL CFFMClient::Open(DWORD dwAccess, char \*szMapName, DWORD dwSize)  
\{  
if (m\_bOpenFlag)  
\{  
Destory();  
\}  
if (szMapName != NULL)  
\{  
m\_pMapName = \_strdup(szMapName);  
\}  
else  
\{  
m\_pMapName = \_strdup(DEFAULT\_MAPNAME);  
\}  
m\_hFileMap = OpenFileMapping(dwAccess, TRUE, m\_pMapName);  
if (NULL == m\_hFileMap)  
\{  
m\_dwLastError = GetLastError();  
Destory();  
return FALSE;  
\}  
m\_lpFileMapBuffer = MapViewOfFile(m\_hFileMap, dwAccess, 0, 0, dwSize);  
if (NULL == m\_lpFileMapBuffer)  
\{  
m\_dwLastError = GetLastError();  
Destory();  
return FALSE;  
\}  
m\_bOpenFlag = TRUE;  
return TRUE;  
\}  
// 獲取當前內存文件的指針  
LPVOID CFFMClient::GetBuffer()  
\{  
return (m\_lpFileMapBuffer)?(m\_lpFileMapBuffer):(NULL);  
\}  
// 讀取命令數據大小  
BOOL CFFMClient::GetCmdDataSize(DWORD \*pDataSize)  
\{  
\_ASSERT(pDataSize != NULL);  
\*pDataSize = 0;  
LPDATA\_HEADER pHeader = (LPDATA\_HEADER)GetBuffer();  
if (NULL == pHeader)  
\{  
m\_dwLastError = ERROR\_NO\_MAPFILE;  
SetLastError(ERROR\_NO\_MAPFILE);  
return FALSE;  
\}  
if (NETRGUSER\_CFM\_CODE != pHeader->dwConfirmCode)  
\{  
m\_dwLastError = ERROR\_INVALID\_CFMCODE;  
SetLastError(ERROR\_INVALID\_CFMCODE);  
return FALSE;  
\}  
if (NETRGUSER\_CMD\_NONE == pHeader->nCommandCode)  
\{  
m\_dwLastError = ERROR\_INVALID\_CMDCODE;  
SetLastError(ERROR\_INVALID\_CMDCODE);  
return FALSE;  
\}  
\*pDataSize = pHeader->dwDataSize;  
return TRUE;  
\}  
// 讀取命令數據  
BOOL CFFMClient::ReadCmdData(DWORD dwCommandCode, DWORD dwBufSize, LPVOID pOutBuf)  
\{  
\_ASSERT (pOutBuf != NULL);  
LPDATA\_HEADER pHeader = (LPDATA\_HEADER)GetBuffer();  
if (NULL == pHeader)  
\{  
m\_dwLastError = ERROR\_NO\_MAPFILE;  
SetLastError(ERROR\_NO\_MAPFILE);  
return FALSE;  
\}  
if (NETRGUSER\_CFM\_CODE != pHeader->dwConfirmCode)  
\{  
m\_dwLastError = ERROR\_INVALID\_CFMCODE;  
SetLastError(ERROR\_INVALID\_CFMCODE);  
return FALSE;  
\}  
if (NETRGUSER\_CMD\_NONE == pHeader->nCommandCode)  
\{  
m\_dwLastError = ERROR\_INVALID\_CMDCODE;  
SetLastError(ERROR\_INVALID\_CMDCODE);  
return FALSE;  
\}  
if (pHeader->dwDataSize > dwBufSize)  
\{  
m\_dwLastError = ERROR\_BUFFER\_OVERFLOW;  
SetLastError(ERROR\_BUFFER\_OVERFLOW);  
return FALSE;  
\}  
if (pHeader->nCommandCode != dwCommandCode)  
\{  
m\_dwLastError = ERROR\_INVALID\_CMDCODE;  
SetLastError(ERROR\_INVALID\_CMDCODE);  
return FALSE;  
\}  
ZeroMemory(pOutBuf, dwBufSize);  
// 拷貝數據到緩衝區  
memcpy(pOutBuf, pHeader->bInfo, pHeader->dwDataSize);  
return TRUE;  
\}  
BOOL CFFMClient::WriteCmdData(DWORD memSize, DWORD nCommandCode, DWORD dwDataSize, const LPVOID pBuf)  
\{  
if(!memSize)  
memSize = DEFAULT\_MAPSIZE;  
m\_dwSize = memSize;  
// 檢驗數據的合理性  
if (NULL == GetBuffer())  
\{  
m\_dwLastError = ERROR\_NO\_MAPFILE;  
SetLastError(ERROR\_NO\_MAPFILE);  
return FALSE;  
\}  
if (NETRGUSER\_CMD\_NONE == nCommandCode)  
\{  
m\_dwLastError = ERROR\_INVALID\_CMDCODE;  
SetLastError(ERROR\_INVALID\_CMDCODE);  
return FALSE;  
\}  
if (dwDataSize > 0 && pBuf == NULL)  
\{  
m\_dwLastError = ERROR\_INVALID\_USER\_BUFFER;  
SetLastError(ERROR\_INVALID\_USER\_BUFFER);  
return FALSE;  
\}  
if (dwDataSize + sizeof(DATA\_HEADER) > GetSize())  
\{  
m\_dwLastError = ERROR\_BUFFER\_OVERFLOW;  
SetLastError(ERROR\_BUFFER\_OVERFLOW);  
return FALSE;  
\}  
// 填寫數據結構  
// 文件頭  
DATA\_HEADER dataHeader;  
dataHeader.nCommandCode = nCommandCode;  
dataHeader.dwDataSize = dwDataSize;  
ZeroMemory(GetBuffer(), GetSize());  
memcpy(GetBuffer(), &dataHeader, sizeof(DATA\_HEADER));  
// 數據塊  
LPDATA\_HEADER pData = (LPDATA\_HEADER)GetBuffer();  
memcpy(pData->bInfo, pBuf, dwDataSize);  
return TRUE;  
\}  
// 獲取內存文件大小  
DWORD CFFMClient::GetSize()  
\{  
return m\_dwSize;  
\}

http://blog.csdn.net/chollima/article/details/5515141