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

#include <windows.h>

class CTLSRipper
{
public:

    CTLSRipper();

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

private:

    // private functions
    BOOL LoadFile(PBYTE pbFile);

    // member variables
    BOOL                 m_bStrip;
    BOOL                 m_bOutput;
    DWORD                m_dwNullDwordVa;
    IMAGE_TLS_DIRECTORY  m_TLS;
    PBYTE                m_pbFile;
    PIMAGE_TLS_DIRECTORY m_pTLS;

};

#endif