114 lines
2.7 KiB
C++
114 lines
2.7 KiB
C++
#ifndef _RAR_VM_
|
|
#define _RAR_VM_
|
|
|
|
#define VM_STANDARDFILTERS
|
|
|
|
#ifndef SFX_MODULE
|
|
#define VM_OPTIMIZE
|
|
#endif
|
|
|
|
|
|
#define VM_MEMSIZE 0x40000
|
|
#define VM_MEMMASK (VM_MEMSIZE-1)
|
|
#define VM_GLOBALMEMADDR 0x3C000
|
|
#define VM_GLOBALMEMSIZE 0x2000
|
|
#define VM_FIXEDGLOBALSIZE 64
|
|
|
|
enum VM_Commands
|
|
{
|
|
VM_MOV, VM_CMP, VM_ADD, VM_SUB, VM_JZ, VM_JNZ, VM_INC, VM_DEC,
|
|
VM_JMP, VM_XOR, VM_AND, VM_OR, VM_TEST, VM_JS, VM_JNS, VM_JB,
|
|
VM_JBE, VM_JA, VM_JAE, VM_PUSH, VM_POP, VM_CALL, VM_RET, VM_NOT,
|
|
VM_SHL, VM_SHR, VM_SAR, VM_NEG, VM_PUSHA,VM_POPA, VM_PUSHF,VM_POPF,
|
|
VM_MOVZX,VM_MOVSX,VM_XCHG, VM_MUL, VM_DIV, VM_ADC, VM_SBB, VM_PRINT,
|
|
|
|
#ifdef VM_OPTIMIZE
|
|
VM_MOVB, VM_MOVD, VM_CMPB, VM_CMPD,
|
|
|
|
VM_ADDB, VM_ADDD, VM_SUBB, VM_SUBD, VM_INCB, VM_INCD, VM_DECB, VM_DECD,
|
|
VM_NEGB, VM_NEGD,
|
|
#endif
|
|
|
|
VM_STANDARD
|
|
};
|
|
|
|
enum VM_StandardFilters {
|
|
VMSF_NONE, VMSF_E8, VMSF_E8E9, VMSF_ITANIUM, VMSF_RGB, VMSF_AUDIO,
|
|
VMSF_DELTA, VMSF_UPCASE
|
|
};
|
|
|
|
enum VM_Flags {VM_FC=1,VM_FZ=2,VM_FS=0x80000000};
|
|
|
|
enum VM_OpType {VM_OPREG,VM_OPINT,VM_OPREGMEM,VM_OPNONE};
|
|
|
|
struct VM_PreparedOperand
|
|
{
|
|
VM_OpType Type;
|
|
uint Data;
|
|
uint Base;
|
|
uint *Addr;
|
|
};
|
|
|
|
struct VM_PreparedCommand
|
|
{
|
|
VM_Commands OpCode;
|
|
bool ByteMode;
|
|
VM_PreparedOperand Op1,Op2;
|
|
};
|
|
|
|
|
|
struct VM_PreparedProgram
|
|
{
|
|
VM_PreparedProgram()
|
|
{
|
|
AltCmd=NULL;
|
|
FilteredDataSize=0;
|
|
CmdCount=0;
|
|
}
|
|
|
|
Array<VM_PreparedCommand> Cmd;
|
|
VM_PreparedCommand *AltCmd;
|
|
int CmdCount;
|
|
|
|
Array<byte> GlobalData;
|
|
Array<byte> StaticData; // static data contained in DB operators
|
|
uint InitR[7];
|
|
|
|
byte *FilteredData;
|
|
uint FilteredDataSize;
|
|
};
|
|
|
|
class RarVM:private BitInput
|
|
{
|
|
private:
|
|
inline uint GetValue(bool ByteMode,uint *Addr);
|
|
inline void SetValue(bool ByteMode,uint *Addr,uint Value);
|
|
inline uint* GetOperand(VM_PreparedOperand *CmdOp);
|
|
void DecodeArg(VM_PreparedOperand &Op,bool ByteMode);
|
|
#ifdef VM_OPTIMIZE
|
|
void Optimize(VM_PreparedProgram *Prg);
|
|
#endif
|
|
bool ExecuteCode(VM_PreparedCommand *PreparedCode,uint CodeSize);
|
|
#ifdef VM_STANDARDFILTERS
|
|
VM_StandardFilters IsStandardFilter(byte *Code,uint CodeSize);
|
|
void ExecuteStandardFilter(VM_StandardFilters FilterType);
|
|
uint FilterItanium_GetBits(byte *Data,int BitPos,int BitCount);
|
|
void FilterItanium_SetBits(byte *Data,uint BitField,int BitPos,int BitCount);
|
|
#endif
|
|
|
|
byte *Mem;
|
|
uint R[8];
|
|
uint Flags;
|
|
public:
|
|
RarVM();
|
|
~RarVM();
|
|
void Init();
|
|
void Prepare(byte *Code,uint CodeSize,VM_PreparedProgram *Prg);
|
|
void Execute(VM_PreparedProgram *Prg);
|
|
void SetLowEndianValue(uint *Addr,uint Value);
|
|
void SetMemory(uint Pos,byte *Data,uint DataSize);
|
|
static uint ReadData(BitInput &Inp);
|
|
};
|
|
|
|
#endif
|