From d5d6965af7e69367babf40b7fb1d7fca8617cabe Mon Sep 17 00:00:00 2001
From: George Hazan <george.hazan@gmail.com>
Date: Thu, 26 Jul 2012 14:56:53 +0000
Subject: dbtool is divided into two parts: DbChecker & appendix for db3x_mmap
 plugin

git-svn-id: http://svn.miranda-ng.org/main/trunk@1195 1316c22d-e87f-b044-9b9b-93d7a3e3ba9c
---
 plugins/Db3x_mmap/src/dbtool/disk.cpp | 112 ++++++++++++++++++++++++++++++++++
 1 file changed, 112 insertions(+)
 create mode 100644 plugins/Db3x_mmap/src/dbtool/disk.cpp

(limited to 'plugins/Db3x_mmap/src/dbtool/disk.cpp')

diff --git a/plugins/Db3x_mmap/src/dbtool/disk.cpp b/plugins/Db3x_mmap/src/dbtool/disk.cpp
new file mode 100644
index 0000000000..2cf9136a71
--- /dev/null
+++ b/plugins/Db3x_mmap/src/dbtool/disk.cpp
@@ -0,0 +1,112 @@
+/*
+Miranda Database Tool
+Copyright (C) 2001-2005  Richard Hughes
+
+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 "dbtool.h"
+
+extern DWORD spaceProcessed, sourceFileSize;
+
+int SignatureValid(DWORD ofs, DWORD signature)
+{
+	DWORD sig;
+
+	if (ofs+sizeof(sig) >= sourceFileSize)	{
+		AddToStatus(STATUS_ERROR, TranslateT("Invalid offset found (database truncated?)"));
+		return 0;
+	}
+
+	sig = *(DWORD*)(opts.pFile+ofs);
+
+	return sig == signature;
+}
+
+int PeekSegment(DWORD ofs, PVOID buf, int cbBytes)
+{
+	DWORD bytesRead;
+
+	if (ofs >= sourceFileSize) {
+		AddToStatus(STATUS_ERROR, TranslateT("Invalid offset found"));
+		return ERROR_SEEK;
+	}
+
+	if (ofs+cbBytes>sourceFileSize)
+		bytesRead = sourceFileSize - ofs;
+	else
+		bytesRead = cbBytes;
+
+	if (bytesRead == 0) {
+		AddToStatus(STATUS_ERROR, TranslateT("Error reading, database truncated? (%u)"), GetLastError());
+		return ERROR_READ_FAULT;
+	}
+
+	CopyMemory(buf, opts.pFile+ofs, bytesRead);
+
+	if ((int)bytesRead<cbBytes) return ERROR_HANDLE_EOF;
+	return ERROR_SUCCESS;
+}
+
+int ReadSegment(DWORD ofs, PVOID buf, int cbBytes)
+{
+	int ret;
+
+	ret = PeekSegment(ofs, buf, cbBytes);
+	if (ret != ERROR_SUCCESS && ret != ERROR_HANDLE_EOF) return ret;
+
+	if (opts.bAggressive) {
+		if (ofs+cbBytes>sourceFileSize) {
+			AddToStatus(STATUS_WARNING, TranslateT("Can't write to working file, aggressive mode may be too aggressive now"));
+			ZeroMemory(opts.pFile+ofs, sourceFileSize-ofs);
+		}
+		else
+			ZeroMemory(opts.pFile+ofs, cbBytes);
+	}
+	spaceProcessed += cbBytes;
+	return ERROR_SUCCESS;
+}
+
+DWORD WriteSegment(DWORD ofs, PVOID buf, int cbBytes)
+{
+	DWORD bytesWritten;
+	if (opts.bCheckOnly) return 0xbfbfbfbf;
+	if (ofs == WSOFS_END) {
+		ofs = dbhdr.ofsFileEnd;
+		dbhdr.ofsFileEnd += cbBytes;
+	}
+	SetFilePointer(opts.hOutFile, ofs, NULL, FILE_BEGIN);
+	WriteFile(opts.hOutFile, buf, cbBytes, &bytesWritten, NULL);
+	if ((int)bytesWritten<cbBytes) {
+		AddToStatus(STATUS_FATAL, TranslateT("Can't write to output file - disk full? (%u)"), GetLastError());
+		return WS_ERROR;
+	}
+	return ofs;
+}
+
+
+int ReadWrittenSegment(DWORD ofs, PVOID buf, int cbBytes)
+{
+	DWORD bytesRead;
+	if (opts.bCheckOnly) return 0xbfbfbfbf;
+	if (ofs + cbBytes > dbhdr.ofsFileEnd)
+		return ERROR_SEEK;
+
+	SetFilePointer(opts.hOutFile, ofs, NULL, FILE_BEGIN);
+	ReadFile(opts.hOutFile, buf, cbBytes, &bytesRead, NULL);
+	if ((int)bytesRead<cbBytes)
+		return ERROR_READ_FAULT;
+
+	return ERROR_SUCCESS;
+}
-- 
cgit v1.2.3