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

#include <windows.h>

class CFileHeaderRipper
{
public:

    CFileHeaderRipper();
    ~CFileHeaderRipper();

    // interface functions
    VOID  CombineSections(BOOL bCombine);
    VOID  NewHeader(BOOL bNewHeader);
    VOID  SetSectionName(PTSTR pszSectionName);
    VOID  SetFileAlignment(DWORD dwFileAlignment);
    VOID  Rip(PVOID pvFile);
    DWORD GetSize();
    VOID  Export(PVOID pvOutput);

private:

    // private functions
    VOID CreateHeader(PBYTE pbFile);
    VOID RipHeader(PBYTE pbFile);

    // member variables
    BOOL  m_bCombine;
    BOOL  m_bNewHeader;
    DWORD m_dwFileAlignment;
    DWORD m_dwHeaderSize;
    PBYTE m_pbHeader;
    PTSTR m_pszSectionName;
};

#endif