path: root/src/OSSupport/GZipFile.cpp

// GZipFile.cpp

// Implements the cGZipFile class representing a RAII wrapper over zlib's GZip file routines

#include "Globals.h"
#include "GZipFile.h"





cGZipFile::cGZipFile(void) :
	m_File(nullptr), m_Mode(fmRead)
{
}





cGZipFile::~cGZipFile()
{
	Close();
}





bool cGZipFile::Open(const AString & a_FileName, eMode a_Mode)
{
	if (m_File != nullptr)
	{
		ASSERT(!"A file is already open in this object");
		return false;
	}
	m_File = gzopen(a_FileName.c_str(), (a_Mode == fmRead) ? "r" : "w");
	m_Mode = a_Mode;
	return (m_File != nullptr);
}





void cGZipFile::Close(void)
{
	if (m_File != nullptr)
	{
		gzclose(m_File);
		m_File = nullptr;
	}
}





int cGZipFile::ReadRestOfFile(AString & a_Contents)
{
	if (m_File == nullptr)
	{
		ASSERT(!"No file has been opened");
		return -1;
	}

	if (m_Mode != fmRead)
	{
		ASSERT(!"Bad file mode, cannot read");
		return -1;
	}

	// Since the gzip format doesn't really support getting the uncompressed length, we need to read incrementally. Yuck!
	int NumBytesRead = 0;
	int TotalBytes = 0;
	char Buffer[64 KiB];
	while ((NumBytesRead = gzread(m_File, Buffer, sizeof(Buffer))) > 0)
	{
		TotalBytes += NumBytesRead;
		a_Contents.append(Buffer, static_cast<size_t>(NumBytesRead));
	}
	// NumBytesRead is < 0 on error
	return (NumBytesRead >= 0) ? TotalBytes : NumBytesRead;
}





bool cGZipFile::Write(const char * a_Contents, int a_Size)
{
	if (m_File == nullptr)
	{
		ASSERT(!"No file has been opened");
		return false;
	}

	if (m_Mode != fmWrite)
	{
		ASSERT(!"Bad file mode, cannot write");
		return false;
	}

	return (gzwrite(m_File, a_Contents, static_cast<unsigned int>(a_Size)) != 0);
}