PENodeWrapper.h

Go to the documentation of this file.

#pragma once
 
#include "../ExeNodeWrapper.h"
#include "pe_formats.h"
 
class PEFile;
 
class PEElementWrapper : public ExeElementWrapper
{
public:
    PEElementWrapper(PEFile* pe);
    virtual ~PEElementWrapper() {}
 
    PEFile* getPE() { return m_PE; }
 
protected:
    PEFile *m_PE;
 
friend class PEFile;
};
 
//----
 
class PENodeWrapper : public ExeNodeWrapper
{
public:
    PENodeWrapper(PEFile* pe, PENodeWrapper* parent = NULL);
    PENodeWrapper(PEFile* pe, PENodeWrapper* parent, size_t entryNumber);
 
    virtual ~PENodeWrapper() {}
 
    PEFile* getPE() { return m_PE; }
    virtual PENodeWrapper* getParentNode() { return peParentNode; }
 
protected:
    PEFile *m_PE;
    PENodeWrapper *peParentNode;
 
friend class PEFile;
};