// Author:   Brandon LaCombe
// Date:     February 3, 2006
// License:  Public Domain
#ifndef _CRELOCATIONRIPPER_H
#define _CRELOCATIONRIPPER_H

#include <windows.h>

// private type definitions needed for header compilation
typedef struct _RVA_ITEM RVA_ITEM, *PRVA_ITEM;

class CRelocationRipper
{
public:

    CRelocationRipper();
    ~CRelocationRipper();

    // interface functions
    VOID  AddRelocation(DWORD dwRva);
    VOID  Strip(BOOL bStrip);
    VOID  Rip(PVOID pvFile);
    DWORD GetSize();
    VOID  Export(PVOID pvOutput);

private:

    // private functions
    PRVA_ITEM CreateRva(DWORD dwRva);
    VOID      DeleteRvaList();
    BOOL      LoadFile(PBYTE pbFile);
    VOID      StripRelocationsFromFile();

    // member variables
    BOOL                   m_bOutput;
    BOOL                   m_bStrip;
    HANDLE                 m_hHeap;
    PBYTE                  m_pbFile;
    PIMAGE_BASE_RELOCATION m_pReloc;
    PRVA_ITEM              m_pRvaList;
};

#endif