ProcessPrx.h

/***************************************************************
 * PRXTool : Utility for PSP executables.
 * (c) TyRaNiD 2k5
 *
 * ProcessPrx.h - Definition of a class to process PRX files.
 ***************************************************************/

#ifndef __PROCESSPRX_H__
#define __PROCESSPRX_H__

#include "ProcessElf.h"
#include "VirtualMem.h"
#include "prxtypes.h"
#include "NidMgr.h"
#include "disasm.h"

/* Define ProcessPrx derived from ProcessElf */
class CProcessPrx : public CProcessElf
{
	PspModule m_modInfo;
	CNidMgr   m_defNidMgr;
	CNidMgr*  m_pCurrNidMgr;
	CVirtualMem m_vMem;
	bool m_blPrxLoaded;
	/* Pointer to the allocated relocation entries, if available */
	ElfReloc  *m_pElfRelocs;
	/* Number of relocations */
	int m_iRelocCount;
	ImmMap m_imms;
	SymbolMap m_syms;
	u32 m_dwBase;
	u32 m_stubBottom;
	bool m_blXmlDump;

	bool FillModule(u8 *pData, u32 iAddr);
	bool CreateFakeSections();
	void FreeMemory();
	int  LoadSingleImport(PspModuleImport *pImport, u32 addr);
	bool LoadImports();
	int  LoadSingleExport(PspModuleExport *pExport, u32 addr);
	bool LoadExports();
	int  CountRelocs();
	int  LoadRelocsTypeA(struct ElfReloc *pRelocs);
	int  LoadRelocsTypeB(struct ElfReloc *pRelocs);
	bool LoadRelocs();
	bool BuildMaps();
	void BuildSymbols(SymbolMap &syms, u32 dwBase);
	void FreeSymbols(SymbolMap &syms);
	void FreeImms(ImmMap &imms);
	void FixupRelocs(u32 dwBase, ImmMap &imms);
	bool ReadString(u32 dwAddr, std::string &str, bool unicode, u32 *dwRet);
	void DumpStrings(FILE *fp, u32 dwAddr, u32 iSize, unsigned char *pData);
	void PrintRow(FILE *fp, const u32* row, s32 row_size, u32 addr);
	void DumpData(FILE *fp, u32 dwAddr, u32 iSize, unsigned char *pData);
	void Disasm(FILE *fp, u32 dwAddr, u32 iSize, unsigned char *pData, ImmMap &imms, u32 dwBase);
	void DisasmXML(FILE *fp, u32 dwAddr, u32 iSize, unsigned char *pData, ImmMap &imms, u32 dwBase);
	void CalcElfSize(size_t &iTotal, size_t &iSectCount, size_t &iStrSize);
	bool OutputElfHeader(FILE *fp, size_t iSectCount);
	bool OutputSections(FILE *fp, size_t iElfHeadSize, size_t iSectCount, size_t iStrSize);
	int  FindFuncExtent(u32 dwStart, u8 *pTouchMap);
	void MapFuncExtents(SymbolMap &syms);
public:
	CProcessPrx(u32 dwBase);
	virtual ~CProcessPrx();
	virtual bool LoadFromFile(const char *szFilename);
	virtual bool LoadFromBinFile(const char *szFilename, unsigned int dwDataBase);

	bool PrxToElf(FILE *fp);

	void SetXmlDump();
	PspModule* GetModuleInfo();
	ElfReloc* GetRelocs(int &iCount);
	ElfSymbol* GetSymbols(int &iCount);
	PspLibImport *GetImports();
	PspLibExport *GetExports();
	void SetNidMgr(CNidMgr* nidMgr);
	void Dump(FILE *fp, const char *disopts);
	void DumpXML(FILE *fp, const char *disopts);
	SymbolEntry *GetSymbolEntryFromAddr(u32 dwAddr);
};

#endif