1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
/*
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 "Interface.h"
#include "DirectAccess.h"
#include "Logger.h"
CDirectAccess::CDirectAccess(const TCHAR* FileName)
: CFileAccess(FileName)
{
m_File = CreateFile(FileName, GENERIC_READ | GENERIC_WRITE, 0, NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL | FILE_FLAG_RANDOM_ACCESS, 0);
CHECKSYS(m_File != INVALID_HANDLE_VALUE,
logCRITICAL, _T("CreateFile failed"));
m_MinAllocGranularity = 0x00001000; // 4kb to avoid heavy fragmentation
m_AllocGranularity = 0x00008000; // 32kb
m_MaxAllocGranularity = 0x00100000; // 1mb for fast increasing
uint32_t size = GetFileSize(m_File, NULL);
size = (size + m_AllocGranularity - 1) & ~(m_AllocGranularity - 1);
if (size == 0)
size = m_AllocGranularity;
m_AllocSize = size;
InitJournal();
}
CDirectAccess::~CDirectAccess()
{
if (m_File)
{
if (INVALID_SET_FILE_POINTER != SetFilePointer(m_File, m_Size, NULL, FILE_BEGIN))
SetEndOfFile(m_File);
CloseHandle(m_File);
}
}
uint32_t CDirectAccess::_Read(void* Buf, uint32_t Source, uint32_t Size)
{
DWORD read = 0;
CHECKSYS(INVALID_SET_FILE_POINTER != SetFilePointer(m_File, Source, NULL, FILE_BEGIN),
logERROR, _T("SetFilePointer failed"));
CHECKSYS(ReadFile(m_File, Buf, Size, &read, NULL),
logERROR, _T("ReadFile failed"));
return read;
}
uint32_t CDirectAccess::_Write(void* Buf, uint32_t Dest, uint32_t Size)
{
DWORD written = 0;
CHECKSYS(INVALID_SET_FILE_POINTER != SetFilePointer(m_File, Dest, NULL, FILE_BEGIN),
logERROR, _T("SetFilePointer failed"));
CHECKSYS(WriteFile(m_File, Buf, Size, &written, NULL),
logERROR, _T("WriteFile failed"));
return written;
}
uint32_t CDirectAccess::_SetSize(uint32_t Size)
{
CHECKSYS(INVALID_SET_FILE_POINTER != SetFilePointer(m_File, Size, NULL, FILE_BEGIN),
logERROR, _T("SetFilePointer failed"));
CHECKSYS(SetEndOfFile(m_File),
logERROR, _T("SetEndOfFile failed"));
return Size;
}
void CDirectAccess::_Invalidate(uint32_t Dest, uint32_t Size)
{
DWORD written;
uint8_t buf[4096];
memset(buf, 0, sizeof(buf));
CHECKSYS(INVALID_SET_FILE_POINTER != SetFilePointer(m_File, Dest, NULL, FILE_BEGIN),
logERROR, _T("SetFilePointer failed"));
while (Size > sizeof(buf))
{
Size -= sizeof(buf);
CHECKSYS(WriteFile(m_File, buf, sizeof(buf), &written, NULL),
logERROR, _T("WriteFile failed"));
}
CHECKSYS(WriteFile(m_File, buf, Size, &written, NULL),
logERROR, _T("WriteFile failed"));
}
void CDirectAccess::_Flush()
{
CHECKSYS(FlushFileBuffers(m_File),
logERROR, _T("FlushFileBuffers failed"));
}
|
