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

#include <windows.h>

// private type definitions needed for header compilation
typedef struct _EXPORT_DIRECTORY EXPORT_DIRECTORY, *PEXPORT_DIRECTORY;

class CExportRipper
{
public:

    CExportRipper();
    ~CExportRipper();

    // interface functions
    VOID  MoveData(BOOL bMoveData);
    VOID  Strip(BOOL bStrip);
    VOID  Rip(PVOID pvFile);
    DWORD GetSize();
    VOID  Export(PVOID pvOutput, DWORD dwBaseRva);

private:

    // private functions
    VOID ExtractExportDirectory();
    VOID FreeExportDirectory();
    BOOL LoadFile(PBYTE pbFile);

    // member variables
    BOOL              m_bStrip;
    BOOL              m_bMoveData;
    DWORD             m_dwHeaderSize;
    DWORD             m_dwStringSize;
    DWORD             m_dwExportsSize;
    HANDLE            m_hHeap;
    PBYTE             m_pbFile;
    PBYTE             m_pbExports;
    PEXPORT_DIRECTORY m_pExportDirectory;
};

#endif