/***************************************************************************
*   Copyright (C) 2001-2008 by Józef Starosczyk                           *
*   ixen@copyhandler.com                                                  *
*                                                                         *
*   This program is free software; you can redistribute it and/or modify  *
*   it under the terms of the GNU Library General Public License          *
*   (version 2) as published by the Free Software Foundation;             *
*                                                                         *
*   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 Library 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.             *
***************************************************************************/
#ifndef __DEVICEIO_H__
#define __DEVICEIO_H__

#include <boost/scoped_array.hpp>

// only NT
static bool GetSignature(LPCTSTR lpszDrive, LPTSTR lpszBuffer, int iSize)
{
	boost::scoped_array<TCHAR> szMapping(new TCHAR[1024]);
	boost::scoped_array<TCHAR> szQuery(new TCHAR[16384]);
	boost::scoped_array<TCHAR> szSymbolic(new TCHAR[1024]);

	// read mappings
	if(QueryDosDevice(lpszDrive, szMapping.get(), 1024) == 0)
		return false;

	// search for all, to find out in which string is the signature
	DWORD dwCount, dwCount2;
	if ((dwCount=QueryDosDevice(NULL, szQuery.get(), 16384)) == 0)
	{
		TRACE("Encountered error #%lu @QueryDosDevice\n", GetLastError());
		return false;
	}

	DWORD dwOffset = 0, dwOffset2 = 0;
	TCHAR* pszSignature = NULL;
	TCHAR* pszOffset = NULL;
	while(dwOffset < dwCount)
	{
		if(_tcsncmp(szQuery.get() + dwOffset, _T("STORAGE#Volume#"), _tcslen(_T("STORAGE#Volume#"))) == 0)
		{
			if((dwCount2 = QueryDosDevice(szQuery.get() + dwOffset, szSymbolic.get(), 1024)) == 0)
				return false;

			// now search for 'Signature' and extract (from szQuery+dwOffset)
			pszSignature=_tcsstr(szQuery.get() + dwOffset, _T("Signature"));
			if (pszSignature == NULL)
			{
				dwOffset += boost::numeric_cast<DWORD>(_tcslen(szQuery.get() + dwOffset) + 1);
				continue;
			}
			pszOffset=_tcsstr(pszSignature, _T("Offset"));
			if (pszOffset == NULL)
			{
				dwOffset += boost::numeric_cast<DWORD>(_tcslen(szQuery.get() + dwOffset) + 1);
				continue;
			}

			// for better string copying
			pszOffset[0]=_T('\0');

			// read values from szSymbolic and compare with szMapping
			dwOffset2=0;
			while(dwOffset2 < dwCount2)
			{
				// compare szSymbolic+dwOffset2 with szMapping
				if (_tcscmp(szMapping.get(), szSymbolic.get() + dwOffset2) == 0)
				{
					// found Signature & Offset - copy
					int iCnt=reinterpret_cast<int>(pszOffset)-reinterpret_cast<int>(pszSignature)+1;
					_tcsncpy(lpszBuffer, pszSignature, (iCnt > iSize) ? iSize : iCnt);
					return true;
				}

				dwOffset2 += boost::numeric_cast<DWORD>(_tcslen(szSymbolic.get()) + 1);
			}
		}

		dwOffset += boost::numeric_cast<DWORD>(_tcslen(szQuery.get() + dwOffset) + 1);
	}

	return false;
}

static bool IsSamePhysicalDisk(int iDrvNum1, int iDrvNum2)
{
	OSVERSIONINFO osvi;
	osvi.dwOSVersionInfoSize=sizeof(OSVERSIONINFO);

	GetVersionEx(&osvi);
	if(osvi.dwPlatformId == VER_PLATFORM_WIN32_NT)
	{
		TCHAR drv1[3], drv2[3];
		
		drv1[0]=static_cast<TCHAR>(iDrvNum1+_T('A'));
		drv1[1]=_T(':');
		drv1[2]=_T('\0');
		drv2[0]=static_cast<TCHAR>(iDrvNum2+_T('A'));
		drv2[1]=_T(':');
		drv2[2]=_T('\0');
		
		TCHAR szSign1[512], szSign2[512];
		return (GetSignature(drv1, szSign1, 512) && GetSignature(drv2, szSign2, 512) && _tcscmp(szSign1, szSign2) == 0);
	}

	return false;
}

#endif