From 7aff1e4cb053394db57c2814d5fe1e6493e0cc75 Mon Sep 17 00:00:00 2001
From: watcherhd <watcherhd@e753b5eb-9565-29b2-b5c5-2cc6f99dfbcb>
Date: Sat, 26 Nov 2011 14:19:43 +0000
Subject: Project folders rename part 2

git-svn-id: http://miranda-plugins.googlecode.com/svn/trunk@214 e753b5eb-9565-29b2-b5c5-2cc6f99dfbcb
---
 Dbx_tree/MappedMemory.cpp | 167 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 167 insertions(+)
 create mode 100644 Dbx_tree/MappedMemory.cpp

(limited to 'Dbx_tree/MappedMemory.cpp')

diff --git a/Dbx_tree/MappedMemory.cpp b/Dbx_tree/MappedMemory.cpp
new file mode 100644
index 0000000..09a1471
--- /dev/null
+++ b/Dbx_tree/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);
+}
-- 
cgit v1.2.3