萬盛學電腦網

 萬盛學電腦網 >> 服務器教程 >> windows下利用文件映射實現共享內存

windows下利用文件映射實現共享內存

windows下利用文件映射實現共享內存的辦法比較簡單,下面是實現代碼,細節用注釋說明.
調用類似linux下shm的操作.該類沒有進行太多的測試,歡迎提出問題和bug~~:)

#include <windows.h>#include <string>#include <iostream>#include <assert.h>using std::string;using std::cout;using std::endl;#pragma warning(disable: 4311)class shareMemory{private: LPWSTR shm_name_u; bool  is_create_file; void * sh_base; HANDLE semaphore; HANDLE file_mapping; int  addr_len;public:/*create_file用來說明是用磁盤文件映射還是頁文件映射,如果用磁盤文件映射,共享內存不會出現存儲器release後出現違規訪問的問題,但是會在磁盤上建立一個文件,文件的名稱由參數shm_name給定.如果用頁文件映射,則不會在磁盤上建立一個文件*/ shareMemory(const string& shm_name, bool create_file=false):is_create_file(create_file) {     const char * _c_shm_name = shm_name.c_str();  int _size =(int)shm_name.length()+1;  shm_name_u=(LPWSTR)malloc(_size*2);  MultiByteToWideChar(CP_ACP,0,_c_shm_name,_size,shm_name_u,_size);  semaphore =  CreateSemaphore(NULL,1,1,NULL);  sh_base = NULL;  file_mapping = INVALID_HANDLE_VALUE; } void * shm_open(void* addr,const int length, DWORD protect) {   addr_len = length;  HANDLE _file_handle = INVALID_HANDLE_VALUE;  if(is_create_file)  {  _file_handle=   CreateFile(shm_name_u,GENERIC_READ|GENERIC_WRITE,0,   NULL,OPEN_ALWAYS,FILE_ATTRIBUTE_NORMAL,0);#ifdef _DEBUG  if(_file_handle==INVALID_HANDLE_VALUE)   cout<<"can not create file! we use page file instead!";#endif  //end _DEBUG  }/*打開指定名稱的文件映射,如果不存在則創建一個*/  file_mapping = OpenFileMapping(PAGE_READWRITE,false,shm_name_u );  if(file_mapping!=NULL)    goto file_mapping_exist;   file_mapping =    CreateFileMapping(_file_handle,NULL,PAGE_READWRITE,   0,length,shm_name_u);#ifdef _DEBUG   assert(file_mapping);#endif    if(file_mapping==NULL)    return NULL;file_mapping_exist:  sh_base = MapViewOfFileEx(file_mapping,protect,0,0,length,addr);  CloseHandle(_file_handle);  return sh_base; }        /*往共享內存中寫數據,返回寫出數據的字節個數*/ int  shm_write(void *dest, void * src, int size) {     if(!check_adress(dest))   return -1;  int _write_count = (int)sh_base+addr_len - (int)dest;  if(_write_count>size)   _write_count = size;/*利用semaphore進行保護映射的區域(同一進程的不同線程調用時候才進行保護)*/  WaitForSingleObject(semaphore,INFINITE);  memcpy(dest,src,_write_count);   ReleaseSemaphore(semaphore,1,NULL);  FlushViewOfFile(sh_base,_write_count);  return _write_count; } /*從共享內存中讀數據,返回讀出的數據字節個數**/int  shm_read(void* src, void * dest, int size) {  if(!check_adress(src))   return -1;    int _read_count = (int)sh_base+addr_len -(int) src; if(_read_count>size)  _read_count = size; memcpy(dest,src,_read_count); return _read_count; } ~shareMemory() {    UnmapViewOfFile(sh_base);   free(shm_name_u);   CloseHandle(semaphore);   CloseHandle(file_mapping); }private :/*進行地址檢測*/ bool check_adress(void* addr) { if( ( (int)addr <(int)sh_base) || ((int)addr > (int)sh_base+addr_len) )  {   SetLastError(ERROR_ACCESS_DENIED);#ifdef _DEBUG   printf("access denied,the destination address out of the map view!");#endif //_DEBUG   return  false;  } return true; }};

測試:
寫進程的主函數:

int main(){ shareMemory sm("boost", false); void * bs = sm.shm_open(NULL,1000*4,FILE_MAP_WRITE); if(bs==NULL)  return -1; int a[10]; for(int i=0; i<10; ++i)  a[i] = i; sm.shm_write(bs,a,10*4); Sleep(100000);}
讀進程的主函數:

 

 

int main(){ shareMemory sm("boost", false); void * bs = sm.shm_open(NULL,1000,FILE_MAP_READ); if(bs==NULL) {   cout<<"null";  return -1; }    int b[10]; sm.shm_read(bs,b,10*4); for(int i=0; i<10; ++i)  cout<<b[i]<<"  ";}


 

						
copyright © 萬盛學電腦網 all rights reserved