www.pudn.com > WindowsDisk.zip > DiskInfo.cpp

#include "stdafx.h" 
#include "DiskInfo.h" 
 
CDiskInfo::CDiskInfo() 
{ 
	GetPlatform(); 
	switch(dwCurrentPlatform) 
	{ 
	case Win95: 
		break; 
	case Win98: 
		hDev=CreateFile("\\\\.\\vwin32",0,0,NULL,0,FILE_FLAG_DELETE_ON_CLOSE,NULL); 
		break; 
	case Win2000: 
		break; 
	} 
} 
CDiskInfo::~CDiskInfo() 
{ 
	if(hDev!=INVALID_HANDLE_VALUE)CloseHandle(hDev); 
} 
 
void CDiskInfo::GetPlatform() 
{ 
	OSVERSIONINFO osv; 
    osv.dwOSVersionInfoSize=sizeof(OSVERSIONINFO); 
	BOOL bRet=GetVersionEx(&osv); 
	if(bRet) 
	{ 
		switch(osv.dwPlatformId) 
		{ 
		case VER_PLATFORM_WIN32s: 
			dwCurrentPlatform=Win95; 
			break; 
		case VER_PLATFORM_WIN32_WINDOWS: 
			if(osv.dwMinorVersion==0)dwCurrentPlatform=Win95; 
			if(osv.dwMinorVersion>0)dwCurrentPlatform=Win98; 
			break; 
		case VER_PLATFORM_WIN32_NT: 
			dwCurrentPlatform=Win2000; 
			break; 
		} 
	} 
} 
 
BOOL CDiskInfo::Int25_ReadSectors(BYTE bDrive,DWORD dwStartSector,WORD wSectors,LPBYTE lpSectBuff) 
{ 
      BOOL           fResult; 
      DWORD          cb; 
      DIOC_REGISTERS reg = {0}; 
      DISKIO         dio = {0}; 
 
      dio.dwStartSector = dwStartSector; 
      dio.wSectors      = wSectors; 
      dio.dwBuffer      = (DWORD)lpSectBuff; 
 
      reg.reg_EAX = bDrive - 1;    // Int 25h drive numbers are 0-based. 
      reg.reg_EBX = (DWORD)&dio; 
      reg.reg_ECX = 0xFFFF;        // use DISKIO struct 
 
      fResult = DeviceIoControl(hDev, VWIN32_DIOC_DOS_INT25, 
                                ®, sizeof(reg), 
                                ®, sizeof(reg), &cb, 0); 
 
      // Determine if the DeviceIoControl call and the read succeeded. 
      fResult = fResult && !(reg.reg_Flags & CARRY_FLAG); 
 
      return fResult; 
}    
 
 
BOOL CDiskInfo::Int26_WriteSectors(BYTE bDrive,DWORD dwStartSector,WORD wSectors,LPBYTE lpSectBuff) 
{ 
  
	BOOL           fResult; 
    DWORD          cb; 
    DIOC_REGISTERS reg = {0}; 
    DISKIO         dio = {0}; 
 
	 
    dio.dwStartSector = dwStartSector; 
    dio.wSectors      = wSectors; 
    dio.dwBuffer      = (DWORD)lpSectBuff; 
 
    reg.reg_EAX = bDrive - 1;    // Int 26h drive numbers are 0-based. 
    reg.reg_EBX = (DWORD)&dio; 
    reg.reg_ECX = 0xFFFF;        // use DISKIO struct 
 
    fResult = DeviceIoControl(hDev, VWIN32_DIOC_DOS_INT26, 
                              ®, sizeof(reg), 
                              ®, sizeof(reg), &cb, 0); 
 
    // Determine if the DeviceIoControl call and the write succeeded. 
    fResult = fResult && !(reg.reg_Flags & CARRY_FLAG); 
 
    return fResult; 
} 
 
 
BOOL CDiskInfo::Int21_AccessSectors (WORD CMD,BYTE bDrive,DWORD dwStartSector,WORD wSectors,LPBYTE lpSectBuff) 
{ 
	BOOL           fResult; 
    DWORD          cb; 
    DIOC_REGISTERS reg = {0}; 
    DISKIO         dio; 
 
    dio.dwStartSector = dwStartSector; 
    dio.wSectors      = wSectors; 
    dio.dwBuffer      = (DWORD)lpSectBuff; 
 
    reg.reg_EAX = 0x7305;   // Ext_ABSDiskReadWrite 
    reg.reg_EBX = (DWORD)&dio; 
    reg.reg_ECX = -1; 
    reg.reg_EDX = bDrive;   // Int 21h, fn 7305h drive numbers are 1-based 
	                        // 0 = default, 1 = A, 2 = B, 3 = C, etc. 
 
	if(CMD==WRITESECTORS) 
	{ 
		reg.reg_ESI = 0x6001;    
	} 
 
    fResult = DeviceIoControl(hDev, VWIN32_DIOC_DOS_DRIVEINFO, 
                              ®, sizeof(reg), 
                              ®, sizeof(reg), &cb, 0); 
 
    // Determine if the DeviceIoControl call and the read succeeded. 
    fResult = fResult && !(reg.reg_Flags & CARRY_FLAG); 
 
    return fResult; 
} 
BOOL CDiskInfo::Win2000_AccessSectors(WORD CMD,BYTE bDrive,DWORD dwStartSector,WORD wSectors,LPBYTE lpSectBuff) 
{ 
	if(bDrive==0)return 0; 
	DWORD dwAccess=0; 
	BOOL bRet=0; 
	DWORD dwCB; 
	char devName[]="\\\\.\\A:"; 
	devName[4]='A'+bDrive-1; 
	if(CMD==WRITESECTORS)dwAccess=GENERIC_WRITE; 
    if(CMD==READSECTORS)dwAccess=GENERIC_READ; 
 
	hDev=CreateFile(devName,dwAccess,FILE_SHARE_WRITE,NULL,OPEN_EXISTING,0,NULL); 
	if(hDev==INVALID_HANDLE_VALUE)return 0; 
 
	dwCB=SetFilePointer(hDev,SECTORSIZE*dwStartSector,0,FILE_BEGIN); 
	switch(CMD) 
	{ 
	case READSECTORS: 
		bRet=ReadFile(hDev,lpSectBuff,SECTORSIZE*wSectors,&dwCB,NULL); 
		break; 
	case WRITESECTORS: 
		bRet=WriteFile(hDev,lpSectBuff,SECTORSIZE*wSectors,&dwCB,NULL); 
		break; 
	} 
	CloseHandle(hDev); 
	return bRet; 
} 
 
 
BOOL CDiskInfo::ReadSectors(BYTE bDrive,DWORD dwStartSector,WORD wSectors,LPBYTE lpSectBuff) 
{ 
	switch(dwCurrentPlatform) 
	{ 
	case Win98: 
		return Int21_AccessSectors(READSECTORS,bDrive,dwStartSector,wSectors,lpSectBuff); 
	case Win95: 
		break; 
	case Win2000: 
		return Win2000_AccessSectors(READSECTORS,bDrive,dwStartSector,wSectors,lpSectBuff); 
	} 
	return 0; 
} 
BOOL CDiskInfo::WriteSectors(BYTE bDrive,DWORD dwStartSector,WORD wSectors,LPBYTE lpSectBuff) 
{ 
	switch(dwCurrentPlatform) 
	{ 
	case Win95: 
		break; 
	case Win98: 
		return Int21_AccessSectors(WRITESECTORS,bDrive,dwStartSector,wSectors,lpSectBuff); 
	case Win2000: 
		return Win2000_AccessSectors(WRITESECTORS,bDrive,dwStartSector,wSectors,lpSectBuff); 
	} 
	return 0; 
}