HLLib/HLLib/FileStream.cpp

359 lines
7.2 KiB
C++

/*
* HLLib
* Copyright (C) 2006-2010 Ryan Gregg
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later
* version.
*/
#include "HLLib.h"
#include "FileStream.h"
using namespace HLLib;
using namespace HLLib::Streams;
#ifdef _WIN32
CFileStream::CFileStream(const hlChar *lpFileName) : hFile(0), uiMode(HL_MODE_INVALID)
#else
CFileStream::CFileStream(const hlChar *lpFileName) : iFile(-1), uiMode(HL_MODE_INVALID)
#endif
{
this->lpFileName = new hlChar[strlen(lpFileName) + 1];
strcpy(this->lpFileName, lpFileName);
}
CFileStream::~CFileStream()
{
this->Close();
delete []this->lpFileName;
}
HLStreamType CFileStream::GetType() const
{
return HL_STREAM_FILE;
}
const hlChar *CFileStream::GetFileName() const
{
return this->lpFileName;
}
hlBool CFileStream::GetOpened() const
{
#ifdef _WIN32
return this->hFile != 0;
#else
return this->iFile >= 0;
#endif
}
hlUInt CFileStream::GetMode() const
{
return this->uiMode;
}
hlBool CFileStream::Open(hlUInt uiMode)
{
this->Close();
#ifdef _WIN32
DWORD dwDesiredAccess = ((uiMode & HL_MODE_READ) ? GENERIC_READ : 0) | ((uiMode & HL_MODE_WRITE) ? GENERIC_WRITE : 0);
DWORD dwShareMode = (uiMode & HL_MODE_VOLATILE) ? FILE_SHARE_READ | FILE_SHARE_WRITE : ((uiMode & HL_MODE_READ) && !(uiMode & HL_MODE_WRITE) ? FILE_SHARE_READ : 0);
DWORD dwCreationDisposition = (uiMode & HL_MODE_WRITE) && (uiMode & HL_MODE_CREATE) ? (bOverwriteFiles ? CREATE_ALWAYS : CREATE_NEW) : ((uiMode & HL_MODE_READ) || (uiMode & HL_MODE_WRITE) ? OPEN_EXISTING : 0);
if(dwDesiredAccess == 0 || dwCreationDisposition == 0)
{
LastError.SetErrorMessageFormated("Invalid open mode (%#.8x).", uiMode);
return hlFalse;
}
this->hFile = CreateFile(this->lpFileName, dwDesiredAccess, dwShareMode, NULL, dwCreationDisposition, FILE_ATTRIBUTE_NORMAL, NULL);
if(this->hFile == INVALID_HANDLE_VALUE)
{
LastError.SetSystemErrorMessage("Error opening file.");
this->hFile = 0;
return hlFalse;
}
#else
hlInt iMode;
if((uiMode & HL_MODE_READ) && (uiMode & HL_MODE_WRITE))
{
iMode = O_RDWR;
}
else if(uiMode & HL_MODE_READ)
{
iMode = O_RDONLY;
}
else if(uiMode & HL_MODE_WRITE)
{
iMode = O_WRONLY;
}
if((uiMode & HL_MODE_WRITE) && (uiMode & HL_MODE_CREATE))
{
iMode |= bOverwriteFiles ? O_CREAT | O_TRUNC : O_CREAT | O_EXCL;
}
if((uiMode & (HL_MODE_READ | HL_MODE_WRITE)) == 0)
{
LastError.SetErrorMessageFormated("Invalid open mode (%#.8x).", uiMode);
return hlFalse;
}
this->iFile = open(this->lpFileName, iMode | O_BINARY | O_RANDOM, S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH);
if(this->iFile < 0)
{
LastError.SetSystemErrorMessage("Error opening file.");
this->iFile = -1;
return hlFalse;
}
#endif
this->uiMode = uiMode;
return hlTrue;
}
hlVoid CFileStream::Close()
{
if(this->GetOpened())
{
#ifdef _WIN32
CloseHandle(this->hFile);
this->hFile = 0;
#else
close(this->iFile);
this->iFile = -1;
#endif
this->uiMode = HL_MODE_INVALID;
}
}
hlULongLong CFileStream::GetStreamSize() const
{
if(!this->GetOpened())
{
return 0;
}
#ifdef _WIN32
LARGE_INTEGER liFileSize;
return GetFileSizeEx(this->hFile, &liFileSize) ? static_cast<hlULongLong>(liFileSize.QuadPart) : 0;
#else
struct stat Stat;
return fstat(this->iFile, &Stat) < 0 ? 0 : Stat.st_size;
#endif
}
hlULongLong CFileStream::GetStreamPointer() const
{
if(!this->GetOpened())
{
return 0;
}
#ifdef _WIN32
LARGE_INTEGER liOffset;
liOffset.QuadPart = 0;
LARGE_INTEGER liPointer;
return SetFilePointerEx(this->hFile, liOffset, &liPointer, FILE_CURRENT) ? static_cast<hlULongLong>(liPointer.QuadPart) : 0;
#else
return (hlULongLong)lseek(this->iFile, 0, SEEK_CUR);
#endif
}
hlULongLong CFileStream::Seek(hlLongLong iOffset, HLSeekMode eSeekMode)
{
if(!this->GetOpened())
{
return 0;
}
#ifdef _WIN32
DWORD dwMode = FILE_BEGIN;
switch(eSeekMode)
{
/*case HL_SEEK_BEGINNING:
dwMode = FILE_BEGIN;
break;*/
case HL_SEEK_CURRENT:
dwMode = FILE_CURRENT;
break;
case HL_SEEK_END:
dwMode = FILE_END;
break;
}
LARGE_INTEGER liOffset;
liOffset.QuadPart = iOffset;
LARGE_INTEGER liPointer;
return SetFilePointerEx(this->hFile, liOffset, &liPointer, dwMode) ? static_cast<hlULongLong>(liPointer.QuadPart) : 0;
#else
hlInt iMode = SEEK_SET;
switch(eSeekMode)
{
case HL_SEEK_BEGINNING:
break;
case HL_SEEK_CURRENT:
iMode = SEEK_CUR;
break;
case HL_SEEK_END:
iMode = SEEK_END;
break;
}
return (hlULongLong)lseek(this->iFile, iOffset, iMode);
#endif
}
hlBool CFileStream::Read(hlChar &cChar)
{
if(!this->GetOpened())
{
return hlFalse;
}
if((this->uiMode & HL_MODE_READ) == 0)
{
LastError.SetErrorMessage("Stream not in read mode.");
return hlFalse;
}
#ifdef _WIN32
hlULong ulBytesRead = 0;
if(!ReadFile(this->hFile, &cChar, 1, &ulBytesRead, NULL))
{
LastError.SetSystemErrorMessage("ReadFile() failed.");
}
return ulBytesRead == 1;
#else
hlInt iBytesRead = read(this->iFile, &cChar, 1);
if(iBytesRead < 0)
{
LastError.SetSystemErrorMessage("read() failed.");
}
return iBytesRead == 1;
#endif
}
hlUInt CFileStream::Read(hlVoid *lpData, hlUInt uiBytes)
{
if(!this->GetOpened())
{
return 0;
}
if((this->uiMode & HL_MODE_READ) == 0)
{
LastError.SetErrorMessage("Stream not in read mode.");
return 0;
}
#ifdef _WIN32
hlULong ulBytesRead = 0;
if(!ReadFile(this->hFile, lpData, uiBytes, &ulBytesRead, NULL))
{
LastError.SetSystemErrorMessage("ReadFile() failed.");
}
return (hlUInt)ulBytesRead;
#else
hlInt iBytesRead = read(this->iFile, lpData, uiBytes);
if(iBytesRead < 0)
{
LastError.SetSystemErrorMessage("read() failed.");
}
return (hlUInt)iBytesRead;
#endif
}
hlBool CFileStream::Write(hlChar cChar)
{
if(!this->GetOpened())
{
return hlFalse;
}
if((this->uiMode & HL_MODE_WRITE) == 0)
{
LastError.SetErrorMessage("Stream not in write mode.");
return hlFalse;
}
#ifdef _WIN32
hlULong ulBytesWritten = 0;
if(!WriteFile(this->hFile, &cChar, 1, &ulBytesWritten, NULL))
{
LastError.SetSystemErrorMessage("WriteFile() failed.");
}
return ulBytesWritten == 1;
#else
hlInt iBytesWritten = write(this->iFile, &cChar, 1);
if(iBytesWritten < 0)
{
LastError.SetSystemErrorMessage("write() failed.");
}
return iBytesWritten == 1;
#endif
}
hlUInt CFileStream::Write(const hlVoid *lpData, hlUInt uiBytes)
{
if(!this->GetOpened())
{
return 0;
}
if((this->uiMode & HL_MODE_WRITE) == 0)
{
LastError.SetErrorMessage("Stream not in write mode.");
return 0;
}
#ifdef _WIN32
hlULong ulBytesWritten = 0;
if(!WriteFile(this->hFile, lpData, uiBytes, &ulBytesWritten, NULL))
{
LastError.SetSystemErrorMessage("WriteFile() failed.");
}
return (hlUInt)ulBytesWritten;
#else
hlInt iBytesWritten = write(this->iFile, lpData, uiBytes);
if(iBytesWritten < 0)
{
LastError.SetSystemErrorMessage("write() failed.");
}
return (hlUInt)iBytesWritten;
#endif
}