0ad/source/ps/FileUnpacker.cpp

///////////////////////////////////////////////////////////////////////////////
//
// Name:		FileUnpacker.cpp
// Author:		Rich Cross
// Contact:		rich@wildfiregames.com
//
///////////////////////////////////////////////////////////////////////////////

#include "precompiled.h"

#include "FileUnpacker.h"
#include "res/res.h"
 
////////////////////////////////////////////////////////////////////////////////////////
// CFileUnpacker constructor
CFileUnpacker::CFileUnpacker()
{
	m_Buf = 0;
	m_Size = 0;
	m_UnpackPos = 0;
	m_Version = 0;
}

CFileUnpacker::~CFileUnpacker()
{
	if (m_Buf)
		mem_free(m_Buf);
}

////////////////////////////////////////////////////////////////////////////////////////
// Read: open and read in given file, check magic bits against those given; throw 
// variety of exceptions for missing files etc
void CFileUnpacker::Read(const char* filename,const char magicstr[4])
{
#ifdef SCED
	// HACK: to make it work with absolute pathnames, e.g. from the map
	// loading dialog box.
	// Check for "x:\<something>" style paths:
	if (! (filename[1] == ':' && filename[2] == '\\'))
	{
#endif // SCED

	// avoid vfs_load complaining about missing data files (which happens
	// too often). better to check here than squelch internal VFS error
	// reporting. we disable this in release mode to avoid a speed hit.
#ifndef NDEBUG
	if(!vfs_exists(filename))
		throw CFileOpenError();
#endif

	// load the whole thing into memory
	Handle hm = vfs_load(filename, m_Buf, m_Size);
	if(hm <= 0)
	{
		m_Buf = 0;
		m_Size = 0;
		throw CFileOpenError();
	}

	// make sure we read enough for the header
	if(m_Size < 12)
	{
		mem_free_h(hm);
		m_Buf = 0;
		m_Size = 0;
		throw CFileReadError();
	}

	// extract data from header
	u8* header = (u8*)m_Buf;
	char* magic = (char*)(header+0);
	// FIXME m_Version and datasize: Byte order? -- Simon
	m_Version = *(u32*)(header+4);
	u32 datasize = *(u32*)(header+8);

	// check we've got the right kind of file
	// .. and that we read exactly headersize+datasize
	if(strncmp(magic, magicstr, 4) != 0 ||
	   m_Size != 12+datasize)
	{
		mem_free_h(hm);
		m_Buf = 0;
		m_Size = 0;
		throw CFileTypeError();
	}

	m_UnpackPos = 12;


#ifdef SCED
	// HACK: continued from earlier
	}
	else
	{
		FILE* fp=fopen(filename,"rb");
		if (!fp)
			throw CFileOpenError();

		m_Buf = 0;
		m_Size = 0;

		char magic[4];
		u32 datasize;
		if (
			// read magic bits
			fread(magic,sizeof(char)*4,1,fp)!=1
			// check we've got the right kind of file
			|| strncmp(magic,magicstr,4)!=0
			// get version
			|| fread(&m_Version,sizeof(m_Version),1,fp)!=1
			// get size of anim data
			|| fread(&datasize,sizeof(datasize),1,fp)!=1
		) {
			fclose(fp);
			throw CFileReadError();
		}

		// allocate memory and read in a big chunk of data
		m_Buf = mem_alloc(datasize);
		if (fread(m_Buf,datasize,1,fp)!=1) {
			mem_free(m_Buf);
			m_Buf = 0;
			fclose(fp);
			throw CFileReadError();
		}

		// all done
		m_Size = datasize;
		fclose(fp);
	}
#endif // SCED
}

////////////////////////////////////////////////////////////////////////////////////////
// UnpackRaw: unpack given number of bytes from the input stream into the given array
//	- throws CFileEOFError if the end of the data stream is reached before the given 
// number of bytes have been read
void CFileUnpacker::UnpackRaw(void* rawdata,u32 rawdatalen)
{
	// got enough data to unpack?
	if (m_UnpackPos+rawdatalen<=m_Size)
	{
		// yes .. copy over
		void* src = (char*)m_Buf + m_UnpackPos;
		memcpy(rawdata, src, rawdatalen);
		m_UnpackPos += rawdatalen;
	}
	else
		// nope - throw exception
		throw CFileEOFError();
}

////////////////////////////////////////////////////////////////////////////////////////
// UnpackString: unpack a string from the raw data stream
//	- throws CFileEOFError if eof is reached before the string length has been
// satisfied
void CFileUnpacker::UnpackString(CStr& result)
{
	// get string length
	u32 length;
	UnpackRaw(&length,sizeof(length)); // FIXME Byte order? -- Simon
	
	if (m_UnpackPos + length <= m_Size)
	{
		result=std::string((char *)m_Buf+m_UnpackPos, length);
		m_UnpackPos += length;
	}
	else
		throw CFileEOFError();
}