Subversion Repositories spk

#ifndef __CATFILE_H__

#define __CATFILE_H__

#include "File.h"

#include "File_IO.h"

#include "lists.h"

#include "spkdll.h"

enum {CATERR_NONE, CATERR_NODATFILE, CATERR_NOCATFILE, CATERR_FILEEMPTY, CATERR_READCAT, CATERR_DECRYPT, CATERR_MISMATCH, CATERR_NOFILE, CATERR_CANTREAD, CATERR_CANTCREATEDIR, CATERR_INVALIDDEST,

		CATERR_CREATED, CATERR_MALLOC};

enum {CATFILE_NONE, CATFILE_READ, CATFILE_DECYPTED};

enum {CATREAD_JUSTCONTENTS, CATREAD_CAT, CATREAD_CATDECRYPT, CATREAD_DAT};

typedef struct SInCatFile {

	SInCatFile () { lSize = 0; sData = 0; lOffset = 0; bDelete = false; bDecrypted = false; }

	Utils::WString sFile;

	size_t lSize;

	unsigned char  *sData;

	size_t lOffset;

	bool	bDelete;

	bool	bDecrypted;

} SInCatFile;

unsigned char SPKEXPORT *PCKData ( unsigned char *data, size_t oldsize, size_t *newsize, bool bXor = true );

class SPKEXPORT CCatFile

{

public:

	enum CatFiletype {

		FILETYPE_PLAIN,

		FILETYPE_DEFLATE,

		FILETYPE_PCK,

	};

public:

	static Utils::WString PckChangeExtension(const Utils::WString &f);

	static Utils::WString RenameFileExtension(SInCatFile *f);

	static bool IsAddonDir(const Utils::WString &dir);

	static bool CheckPackedExtension(const Utils::WString &sFilename);

public:

	CCatFile ();

	~CCatFile ();

	int  open(const Utils::WString &catfile, const Utils::WString &addon, int readtype = CATREAD_CATDECRYPT, bool create = true);

	bool DecryptData ();

	bool DecryptData ( unsigned char *data, size_t size );

	void DecryptDAT(unsigned char *buffer, size_t size);

	void DecryptDAT(SInCatFile *pFile);

	bool readFiles ();

	void LoadDatFile ();

	bool checkExtensionPck(const Utils::WString &filename) const;

	unsigned char *readData(const Utils::WString &filename, size_t *size);

	unsigned char *readData(SInCatFile *c, size_t *size);

	SInCatFile *findData(const Utils::WString &filename) const;

	bool readFileToData(const Utils::WString &filename);

	bool readFileToData(SInCatFile *c);

	bool markRemoveFile(SInCatFile *f);

	bool markRemoveFile(const Utils::WString &filename);

	int GetEndOffset();

	size_t GetNumFiles() const;

	SInCatFile *GetFile(unsigned int num) const;

	unsigned char *UnpackFile ( SInCatFile *c, size_t *size);

	static bool Opened(int error, bool allowCreate = true);

	bool removeFile(SInCatFile *f);

	bool removeFile(const Utils::WString &filename);

	bool WriteCatFile ();

	bool appendFile(const Utils::WString &sFilename, const Utils::WString &to, bool pck = true, bool bXor = true, Utils::WString *sChangeTo = NULL);

	bool appendData (unsigned char *data, size_t size, const Utils::WString &to, bool pck = true, bool bXor = true );

	bool addData(const Utils::WString &catfile, unsigned char *data, size_t size, const Utils::WString &to, bool pck = true, bool create = true );

	std::vector<SInCatFile *> *GetFiles() const;

	bool extractFile(const Utils::WString &filename, const Utils::WString &to = Utils::WString::Null(), bool preserve = false);

	bool extractFile(SInCatFile* f, const Utils::WString &to = Utils::WString::Null(), bool preserve = false);

	bool extractAll(const Utils::WString &dir);

	void clearError () { m_iError = CATERR_NONE; _sError.clear(); }

	int error () const { return m_iError; }

	const Utils::WString &errorString() const { return _sError; }

	Utils::WString getErrorString () const;

	const Utils::WString &internalDatFilename() const;

	const Utils::WString &catFilename() const { return m_fCatFile.fullFilename(); }

	const Utils::WString &datFilename() const { return m_fDatFile.fullFilename(); }

	bool writeFromCat(const Utils::WString &catfile, const Utils::WString &file);

	bool writeFromCat(CCatFile* fcat, const Utils::WString &file);

	void findFiles(Utils::WStringList &list, const Utils::WString &filemask) const;

private:

	void _clearFiles();

	int _checkFiletype(const unsigned char *pBuffer, int iSize);

	void RemoveData ();

	CFileIO m_fCatFile;

	CFileIO m_fDatFile;

	unsigned char *m_sData;

	size_t m_lSize;

	char m_iDataType;

	std::vector<SInCatFile *> *_lFiles;

	Utils::WString _sError;

	Utils::WString _sAddonDir;

	Utils::WString _sReadFilename;

	int m_iError;

	bool m_bCreate;

	bool m_bCatChanged;

};

#endif //__CATFILE_H__