volume.c File Reference

#include <windows.h>
#include <stdio.h>
#include <tchar.h>
#include "volume.h"

Include dependency graph for volume.c:

Go to the source code of this file.

Functions
BOOL	OpenVolume (LPCTSTR lpszVolumeName)
void	CloseVolume (void)
BOOL	ReadVolumeSector (ULONG SectorNumber, PVOID SectorBuffer)
BOOL	WriteVolumeSector (ULONG SectorNumber, PVOID SectorBuffer)

Variables
static HANDLE	hDiskVolume = NULL

Function Documentation

CloseVolume()

void CloseVolume

(

void

)

Definition at line 68 of file volume.c.

{
    CloseHandle(hDiskVolume);
}

Referenced by main().

OpenVolume()

BOOL OpenVolume

(

LPCTSTR

lpszVolumeName

)

Definition at line 28 of file volume.c.

{
    TCHAR    RealVolumeName[MAX_PATH];

    //
    // If they passed in a drive letter (e.g. "A:")
    // then try to open the physical device volume,
    // otherwise we will assume it is a disk image
    // file they are writing to. (not fully supported yet)
    //
    if ((_tcslen(lpszVolumeName) == 2) && (lpszVolumeName[1] == _T(':')))
    {
        _tcscpy(RealVolumeName, _T("\\\\.\\"));
        _tcscat(RealVolumeName, lpszVolumeName);
    }
    else
    {
        _tcscpy(RealVolumeName, lpszVolumeName);
    }

    _tprintf(_T("Opening volume \'%s\'\n"), lpszVolumeName);

    hDiskVolume = CreateFile(RealVolumeName,
                            GENERIC_READ | GENERIC_WRITE,
                            FILE_SHARE_READ | FILE_SHARE_WRITE,
                            NULL,
                            OPEN_EXISTING,
                            0,
                            NULL);

    if (hDiskVolume == INVALID_HANDLE_VALUE)
    {
        _tprintf(_T("%s:%d: "), __FILE__, __LINE__);
        _tprintf(_T("Failed. Error code %ld.\n"), GetLastError());
        return FALSE;
    }

    return TRUE;
}

Referenced by DismountMain().

ReadVolumeSector()

BOOL ReadVolumeSector	(	ULONG	SectorNumber,
		PVOID	SectorBuffer
	)

Definition at line 73 of file volume.c.

{
    DWORD    dwNumberOfBytesRead;
    DWORD    dwFilePosition;
    BOOL    bRetVal;

    //
    // FIXME: this doesn't seem to handle the situation
    // properly when SectorNumber is bigger than the
    // amount of sectors on the disk. Seems to me that
    // the call to SetFilePointer() should just give an
    // out of bounds error or something but it doesn't.
    //
    dwFilePosition = SetFilePointer(hDiskVolume, (SectorNumber * 512), NULL, FILE_BEGIN);
    if (dwFilePosition != (SectorNumber * 512))
    {
        _tprintf(_T("%s:%d: "), __FILE__, __LINE__);
        _tprintf(_T("SetFilePointer() failed. Error code %ld.\n"), GetLastError());
        return FALSE;
    }

    bRetVal = ReadFile(hDiskVolume, SectorBuffer, 512, &dwNumberOfBytesRead, NULL);
    if (!bRetVal || (dwNumberOfBytesRead != 512))
    {
        _tprintf(_T("%s:%d: "), __FILE__, __LINE__);
        _tprintf(_T("ReadFile() failed. Error code %ld.\n"), GetLastError());
        return FALSE;
    }

    return TRUE;
}

Referenced by BackupBootSector(), and InstallBootSector().

WriteVolumeSector()

BOOL WriteVolumeSector	(	ULONG	SectorNumber,
		PVOID	SectorBuffer
	)

Definition at line 105 of file volume.c.

{
    DWORD    dwNumberOfBytesWritten;
    DWORD    dwFilePosition;
    BOOL    bRetVal;

    //
    // FIXME: this doesn't seem to handle the situation
    // properly when SectorNumber is bigger than the
    // amount of sectors on the disk. Seems to me that
    // the call to SetFilePointer() should just give an
    // out of bounds error or something but it doesn't.
    //
    dwFilePosition = SetFilePointer(hDiskVolume, (SectorNumber * 512), NULL, FILE_BEGIN);
    if (dwFilePosition != (SectorNumber * 512))
    {
        _tprintf(_T("%s:%d: "), __FILE__, __LINE__);
        _tprintf(_T("SetFilePointer() failed. Error code %ld.\n"), GetLastError());
        return FALSE;
    }

    bRetVal = WriteFile(hDiskVolume, SectorBuffer, 512, &dwNumberOfBytesWritten, NULL);
    if (!bRetVal || (dwNumberOfBytesWritten != 512))
    {
        _tprintf(_T("%s:%d: "), __FILE__, __LINE__);
        _tprintf(_T("WriteFile() failed. Error code %ld.\n"), GetLastError());
        return FALSE;
    }

    return TRUE;
}

Referenced by InstallBootSector().

Variable Documentation

hDiskVolume

HANDLE hDiskVolume = NULL

static

Definition at line 26 of file volume.c.

Referenced by CloseVolume(), OpenVolume(), ReadVolumeSector(), and WriteVolumeSector().