Использование общей памяти в библиотеке Dynamic-Link
В следующем примере показано, как функция точки входа DLL может использовать объект сопоставления файлов для настройки памяти, которую могут совместно использовать процессы, загружающие библиотеку DLL. Общая память DLL сохраняется только при загрузке библиотеки DLL. Приложения могут использовать функции SetSharedMem и GetSharedMem для доступа к общей памяти.
Библиотека DLL, реализующая общую память
В примере сопоставление файлов используется для сопоставления блока именованной общей памяти с виртуальным адресным пространством каждого процесса, загружающего библиотеку DLL. Для этого функция точки входа должна:
- Вызовите функцию CreateFileMapping , чтобы получить дескриптор объекта сопоставления файлов. Первый процесс, загружающий библиотеку DLL, создает объект сопоставления файлов. Последующие процессы открывают дескриптор для существующего объекта . Дополнительные сведения см. в разделе Создание объекта File-Mapping.
- Вызовите функцию MapViewOfFile , чтобы сопоставить представление с виртуальным адресным пространством. Это позволяет процессу получить доступ к общей памяти. Дополнительные сведения см. в разделе Создание представления файлов.
Обратите внимание, что хотя можно указать атрибуты безопасности по умолчанию, передав значение NULL для параметра lpAttributescreateFileMapping, вы можете использовать структуру SECURITY_ATTRIBUTES для обеспечения дополнительной безопасности.
// The DLL code
#include <windows.h>
#include <memory.h>
#define SHMEMSIZE 4096
static LPVOID lpvMem = NULL; // pointer to shared memory
static HANDLE hMapObject = NULL; // handle to file mapping
// The DLL entry-point function sets up shared memory using a
// named file-mapping object.
BOOL WINAPI DllMain(HINSTANCE hinstDLL, // DLL module handle
DWORD fdwReason, // reason called
LPVOID lpvReserved) // reserved
{
BOOL fInit, fIgnore;
switch (fdwReason)
{
// DLL load due to process initialization or LoadLibrary
case DLL_PROCESS_ATTACH:
// Create a named file mapping object
hMapObject = CreateFileMapping(
INVALID_HANDLE_VALUE, // use paging file
NULL, // default security attributes
PAGE_READWRITE, // read/write access
0, // size: high 32-bits
SHMEMSIZE, // size: low 32-bits
TEXT("dllmemfilemap")); // name of map object
if (hMapObject == NULL)
return FALSE;
// The first process to attach initializes memory
fInit = (GetLastError() != ERROR_ALREADY_EXISTS);
// Get a pointer to the file-mapped shared memory
lpvMem = MapViewOfFile(
hMapObject, // object to map view of
FILE_MAP_WRITE, // read/write access
0, // high offset: map from
0, // low offset: beginning
0); // default: map entire file
if (lpvMem == NULL)
return FALSE;
// Initialize memory if this is the first process
if (fInit)
memset(lpvMem, '\0', SHMEMSIZE);
break;
// The attached process creates a new thread
case DLL_THREAD_ATTACH:
break;
// The thread of the attached process terminates
case DLL_THREAD_DETACH:
break;
// DLL unload due to process termination or FreeLibrary
case DLL_PROCESS_DETACH:
// Unmap shared memory from the process's address space
fIgnore = UnmapViewOfFile(lpvMem);
// Close the process's handle to the file-mapping object
fIgnore = CloseHandle(hMapObject);
break;
default:
break;
}
return TRUE;
UNREFERENCED_PARAMETER(hinstDLL);
UNREFERENCED_PARAMETER(lpvReserved);
}
// The export mechanism used here is the __declspec(export)
// method supported by Microsoft Visual Studio, but any
// other export method supported by your development
// environment may be substituted.
#ifdef __cplusplus // If used by C++ code,
extern "C" { // we need to export the C interface
#endif
// SetSharedMem sets the contents of the shared memory
__declspec(dllexport) VOID __cdecl SetSharedMem(LPWSTR lpszBuf)
{
LPWSTR lpszTmp;
DWORD dwCount=1;
// Get the address of the shared memory block
lpszTmp = (LPWSTR) lpvMem;
// Copy the null-terminated string into shared memory
while (*lpszBuf && dwCount<SHMEMSIZE)
{
*lpszTmp++ = *lpszBuf++;
dwCount++;
}
*lpszTmp = '\0';
}
// GetSharedMem gets the contents of the shared memory
__declspec(dllexport) VOID __cdecl GetSharedMem(LPWSTR lpszBuf, DWORD cchSize)
{
LPWSTR lpszTmp;
// Get the address of the shared memory block
lpszTmp = (LPWSTR) lpvMem;
// Copy from shared memory into the caller's buffer
while (*lpszTmp && --cchSize)
*lpszBuf++ = *lpszTmp++;
*lpszBuf = '\0';
}
#ifdef __cplusplus
}
#endif
Общая память может быть сопоставлена с разным адресом в каждом процессе. По этой причине каждый процесс имеет собственный экземпляр lpvMem, который объявляется как глобальная переменная, чтобы он был доступен для всех функций DLL. В примере предполагается, что глобальные данные DLL не являются общими, поэтому каждый процесс, загружающий библиотеку DLL, имеет собственный экземпляр lpvMem.
Обратите внимание, что общая память освобождается при закрытии последнего дескриптора объекта сопоставления файлов. Чтобы создать постоянную общую память, необходимо убедиться, что какой-то процесс всегда имеет открытый дескриптор для объекта сопоставления файлов.
Процессы, использующие общую память
В следующих процессах используется общая память, предоставляемая библиотекой DLL, определенной выше. Первый процесс вызывает Метод SetSharedMem для записи строки, а второй — GetSharedMem для получения этой строки.
В этом процессе используется функция SetSharedMem, реализованная библиотекой DLL, для записи строки "This is a test string" в общую память. Он также запускает дочерний процесс, который считывает строку из общей памяти.
// Parent process
#include <windows.h>
#include <tchar.h>
#include <stdio.h>
extern "C" VOID __cdecl SetSharedMem(LPWSTR lpszBuf);
HANDLE CreateChildProcess(LPTSTR szCmdline)
{
PROCESS_INFORMATION piProcInfo;
STARTUPINFO siStartInfo;
BOOL bFuncRetn = FALSE;
// Set up members of the PROCESS_INFORMATION structure.
ZeroMemory( &piProcInfo, sizeof(PROCESS_INFORMATION) );
// Set up members of the STARTUPINFO structure.
ZeroMemory( &siStartInfo, sizeof(STARTUPINFO) );
siStartInfo.cb = sizeof(STARTUPINFO);
// Create the child process.
bFuncRetn = CreateProcess(NULL,
szCmdline, // command line
NULL, // process security attributes
NULL, // primary thread security attributes
TRUE, // handles are inherited
0, // creation flags
NULL, // use parent's environment
NULL, // use parent's current directory
&siStartInfo, // STARTUPINFO pointer
&piProcInfo); // receives PROCESS_INFORMATION
if (bFuncRetn == 0)
{
printf("CreateProcess failed (%)\n", GetLastError());
return INVALID_HANDLE_VALUE;
}
else
{
CloseHandle(piProcInfo.hThread);
return piProcInfo.hProcess;
}
}
int _tmain(int argc, TCHAR *argv[])
{
HANDLE hProcess;
if (argc == 1)
{
printf("Please specify an input file");
ExitProcess(0);
}
// Call the DLL function
printf("\nProcess is writing to shared memory...\n\n");
SetSharedMem(L"This is a test string");
// Start the child process that will read the memory
hProcess = CreateChildProcess(argv[1]);
// Ensure this process is around until the child process terminates
if (INVALID_HANDLE_VALUE != hProcess)
{
WaitForSingleObject(hProcess, INFINITE);
CloseHandle(hProcess);
}
return 0;
}
В этом процессе используется функция GetSharedMem, реализованная библиотекой DLL для чтения строки из общей памяти. Он запускается родительским процессом выше.
// Child process
#include <windows.h>
#include <tchar.h>
#include <stdio.h>
extern "C" VOID __cdecl GetSharedMem(LPWSTR lpszBuf, DWORD cchSize);
int _tmain( void )
{
WCHAR cBuf[MAX_PATH];
GetSharedMem(cBuf, MAX_PATH);
printf("Child process read from shared memory: %S\n", cBuf);
return 0;
}
Связанные темы