D:/opendownloadmanager/ODM-1.x/vmsFile.cpp

Go to the documentation of this file.

/*
  Free Download Manager Copyright (c) 2003-2007 FreeDownloadManager.ORG
  Open Download Manager Copyright (c) 2008-2010 OpenDownloadManager.ORG
*/        

#include "stdafx.h"
#include "vmsFile.h"
using namespace vmsFDM;        

vmsFile::vmsFile(HANDLE hFile, bool bAutoClose)
{
        m_hFile = hFile;
        m_bAutoClose = bAutoClose;
        m_bThrowOnPartialRead = true;
}

vmsFile::~vmsFile()
{
        if (m_bAutoClose)
                Close ();
}

void vmsFile::Create(LPCTSTR pszFileName, DWORD dwAccess, DWORD dwShareMode, DWORD dwCreationDisposition, DWORD dwFlags)
{
        Close ();

        m_hFile = CreateFile (pszFileName, dwAccess, dwShareMode, NULL, dwCreationDisposition,
                dwFlags, NULL);

        if (m_hFile == INVALID_HANDLE_VALUE)
                throw GetLastError ();
}

void vmsFile::Close()
{
        if (m_hFile != INVALID_HANDLE_VALUE)
        {
                if (FALSE == CloseHandle (Detach ()))
                        throw GetLastError ();
        }
}

void vmsFile::Read(LPVOID pvBuffer, DWORD dwToRead, DWORD *pdwRead)
{
        if (m_hFile == INVALID_HANDLE_VALUE)
                throw ERROR_INVALID_HANDLE;

        DWORD dwRead = 0;

        if (FALSE == ReadFile (m_hFile, pvBuffer, dwToRead, &dwRead, NULL))
                throw GetLastError ();

        if (pdwRead)
                *pdwRead = dwRead;

        if (m_bThrowOnPartialRead && dwRead != dwToRead)
                throw ERROR_BAD_FORMAT;
}

void vmsFile::Write(LPCVOID pvBuffer, DWORD dwToWrite)
{
        if (m_hFile == INVALID_HANDLE_VALUE)
                throw ERROR_INVALID_HANDLE;

        DWORD dwWritten = 0;

        if (FALSE == WriteFile (m_hFile, pvBuffer, dwToWrite, &dwWritten, NULL))
                throw GetLastError ();

        if (dwWritten != dwToWrite)
                throw ERROR_BAD_LENGTH;
}

void vmsFile::Attach(HANDLE hFile, bool bAutoClose)
{
        Close ();
        m_hFile = hFile;
        m_bAutoClose = bAutoClose;
}

HANDLE vmsFile::Detach()
{
        HANDLE h = m_hFile;
        m_hFile = INVALID_HANDLE_VALUE;
        return h;
}

DWORD vmsFile::GetSize()
{
        if (m_hFile == INVALID_HANDLE_VALUE)
                throw ERROR_INVALID_HANDLE;

        return GetFileSize (m_hFile, NULL);
}

void vmsFile::Seek(DWORD dwDistance, DWORD dwMethod)
{
        SetFilePointer (m_hFile, dwDistance, NULL, dwMethod);
}

void vmsFile::WriteString(LPCSTR psz)
{
        if (FALSE == fsSaveStrToFile (psz, m_hFile))
                throw GetLastError ();
}

void vmsFile::ReadString(fsString& str)
{
        LPSTR psz;
        if (FALSE == fsReadStrFromFile (&psz, m_hFile))
                throw GetLastError ();
        str  = psz;
        delete [] psz;
}

void vmsFile::WriteDouble(double f)
{
        Write (&f, sizeof (f));
}

void vmsFile::WriteInt64(__int64 i)
{
        Write (&i, sizeof (i));
}

void vmsFile::WriteInt(int i)
{
        Write (&i, sizeof (i));
}

void vmsFile::ReadInt(int &i)
{
        Read (&i, sizeof (i));
}

void vmsFile::ReadInt64(__int64 &i)
{
        Read (&i, sizeof (i));
}

void vmsFile::ReadDouble(double &f)
{
        Read (&f, sizeof (f));
}

---