~ubuntu-branches/ubuntu/maverick/unrar-nonfree/maverick

#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