diff options
| author | Vadim Dashevskiy <watcherhd@gmail.com> | 2012-10-12 09:56:01 +0000 |
|---|---|---|
| committer | Vadim Dashevskiy <watcherhd@gmail.com> | 2012-10-12 09:56:01 +0000 |
| commit | 69fe21d6e2262d6fb2b14278b7a20364468cbdcf (patch) | |
| tree | d7a5c3805ba838e98e1c5288fe06962a3201c694 /plugins/Dbx_tree/src/MappedMemory.cpp | |
| parent | bcc2cdbf65af07f2593577a4c59f93d25c2e3de9 (diff) | |
Dbx_tree: folders restructurization
git-svn-id: http://svn.miranda-ng.org/main/trunk@1883 1316c22d-e87f-b044-9b9b-93d7a3e3ba9c
Diffstat (limited to 'plugins/Dbx_tree/src/MappedMemory.cpp')
| -rw-r--r-- | plugins/Dbx_tree/src/MappedMemory.cpp | 167 |
1 files changed, 167 insertions, 0 deletions
diff --git a/plugins/Dbx_tree/src/MappedMemory.cpp b/plugins/Dbx_tree/src/MappedMemory.cpp new file mode 100644 index 0000000000..09a1471a90 --- /dev/null +++ b/plugins/Dbx_tree/src/MappedMemory.cpp @@ -0,0 +1,167 @@ +/*
+
+dbx_tree: tree database driver for Miranda IM
+
+Copyright 2007-2010 Michael "Protogenes" Kunz,
+
+This program is free software; you can redistribute it and/or
+modify it under the terms of the GNU General Public License
+as published by the Free Software Foundation; either version 2
+of the License, or (at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program; if not, write to the Free Software
+Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
+
+*/
+
+#include "MappedMemory.h"
+#include "Logger.h"
+
+typedef BOOL (WINAPI *TUnmapViewOfFile)(LPCVOID);
+typedef BOOL (WINAPI *TFlushViewOfFile)(LPCVOID, SIZE_T);
+typedef HANDLE (WINAPI *TCreateFileMappingA)(HANDLE, LPSECURITY_ATTRIBUTES, DWORD, DWORD, DWORD, LPCSTR);
+typedef LPVOID (WINAPI *TMapViewOfFile)(HANDLE, DWORD, DWORD, DWORD, SIZE_T);
+
+HMODULE myKernelLib = NULL;
+TUnmapViewOfFile myUnmapViewOfFile = NULL;
+TFlushViewOfFile myFlushViewOfFile = NULL;
+TCreateFileMappingA myCreateFileMappingA = NULL;
+TMapViewOfFile myMapViewOfFile = NULL;
+
+bool CMappedMemory::InitMMAP()
+{
+ if (!myKernelLib)
+ myKernelLib = GetModuleHandleA("kernel32.dll"); // is always loaded
+
+ if (myKernelLib)
+ {
+ myUnmapViewOfFile = (TUnmapViewOfFile) GetProcAddress(myKernelLib, "UnmapViewOfFile");
+ myFlushViewOfFile = (TFlushViewOfFile) GetProcAddress(myKernelLib, "FlushViewOfFile");
+ myCreateFileMappingA = (TCreateFileMappingA) GetProcAddress(myKernelLib, "CreateFileMappingA");
+ myMapViewOfFile = (TMapViewOfFile) GetProcAddress(myKernelLib, "MapViewOfFile");
+ }
+
+ return myUnmapViewOfFile && myFlushViewOfFile && myCreateFileMappingA && myMapViewOfFile;
+}
+CMappedMemory::CMappedMemory(const TCHAR* FileName)
+: CFileAccess(FileName)
+{
+ SYSTEM_INFO sysinfo;
+
+ m_AllocSize = 0;
+ m_DirectFile = 0;
+ m_FileMapping = 0;
+ m_Base = NULL;
+
+ GetSystemInfo(&sysinfo);
+ m_AllocGranularity = sysinfo.dwAllocationGranularity; // usually 64kb
+ m_MinAllocGranularity = m_AllocGranularity; // must be at least one segment
+ m_MaxAllocGranularity = m_AllocGranularity << 4; // usually 1mb for fast increasing
+
+ m_DirectFile = CreateFile(FileName, GENERIC_READ | GENERIC_WRITE, 0, NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL | FILE_FLAG_RANDOM_ACCESS, 0);
+ if (m_DirectFile == INVALID_HANDLE_VALUE)
+ LOGSYS(logCRITICAL, _T("CreateFile failed"));
+
+ uint32_t size = GetFileSize(m_DirectFile, NULL);
+ size = (size + m_AllocGranularity - 1) & ~(m_AllocGranularity - 1);
+
+ if (size == 0)
+ size = m_AllocGranularity;
+
+ _SetSize(size);
+ m_AllocSize = size;
+
+ InitJournal();
+}
+
+CMappedMemory::~CMappedMemory()
+{
+ if (m_Base)
+ {
+ myFlushViewOfFile(m_Base, NULL);
+ myUnmapViewOfFile(m_Base);
+ }
+ if (m_FileMapping)
+ CloseHandle(m_FileMapping);
+
+ if (m_DirectFile)
+ {
+ if (INVALID_SET_FILE_POINTER != SetFilePointer(m_DirectFile, m_Size, NULL, FILE_BEGIN))
+ SetEndOfFile(m_DirectFile);
+
+ FlushFileBuffers(m_DirectFile);
+ CloseHandle(m_DirectFile);
+ }
+}
+
+
+uint32_t CMappedMemory::_Read(void* Buf, uint32_t Source, uint32_t Size)
+{
+ memcpy(Buf, m_Base + Source, Size);
+ return Size;
+}
+uint32_t CMappedMemory::_Write(void* Buf, uint32_t Dest, uint32_t Size)
+{
+ memcpy(m_Base + Dest, Buf, Size);
+ return Size;
+}
+
+uint32_t CMappedMemory::_SetSize(uint32_t Size)
+{
+ if (m_Base)
+ {
+ myFlushViewOfFile(m_Base, 0);
+ myUnmapViewOfFile(m_Base);
+ }
+ if (m_FileMapping)
+ CloseHandle(m_FileMapping);
+
+ m_Base = NULL;
+ m_FileMapping = NULL;
+
+ if (INVALID_SET_FILE_POINTER == SetFilePointer(m_DirectFile, Size, NULL, FILE_BEGIN))
+ {
+ LOGSYS(logERROR, _T("SetFilePointer failed"));
+ return 0;
+ }
+
+ if (!SetEndOfFile(m_DirectFile))
+ {
+ LOGSYS(logERROR, _T("Cannot set end of file"));
+ return 0;
+ }
+
+ m_FileMapping = myCreateFileMappingA(m_DirectFile, NULL, PAGE_READWRITE, 0, Size, NULL);
+
+ if (!m_FileMapping)
+ {
+ LOGSYS(logERROR, _T("CreateFileMapping failed"));
+ return 0;
+ }
+
+ m_Base = (uint8_t*) myMapViewOfFile(m_FileMapping, FILE_MAP_ALL_ACCESS, 0, 0, 0);
+ if (!m_Base)
+ {
+ LOGSYS(logERROR, _T("MapViewOfFile failed"));
+ return 0;
+ }
+
+ return Size;
+}
+
+void CMappedMemory::_Invalidate(uint32_t Dest, uint32_t Size)
+{
+ memset(m_Base + Dest, 0, Size);
+}
+
+void CMappedMemory::_Flush()
+{
+ myFlushViewOfFile(m_Base, NULL);
+ FlushFileBuffers(m_DirectFile);
+}
|