hackedteam/vector-dropper

#ifndef _RESOURCE_DIRECTORY_H
#define _RESOURCE_DIRECTORY_H

#include "common.h"

#include <vector>
using namespace std;

#define ALIGN(dwToAlign, dwAlignOn) dwToAlign = (dwToAlign%dwAlignOn == 0) ? dwToAlign : dwToAlign - (dwToAlign%dwAlignOn) + dwAlignOn

class ResourceDirectoryEntry;

class ResourceDirectory
{
public:
    ResourceDirectory(PIMAGE_RESOURCE_DIRECTORY prd);
    
    virtual ~ResourceDirectory(void) {}
    
    IMAGE_RESOURCE_DIRECTORY GetInfo() { return _rdDir; }

    ResourceDirectoryEntry* GetEntry(UINT i)
    {
        if (_entries.size() < i)
            return NULL;

        return _entries[i];
    }

    void AddEntry(ResourceDirectoryEntry* entry);
    
    int CountEntries() { return _entries.size(); }
    int Find(WCHAR* name);

    int Find(WORD Id);
    
    DWORD GetSize();
    
    void Destroy();
    
    DWORD writtenAt;
    
private:
    IMAGE_RESOURCE_DIRECTORY _rdDir;
    vector<ResourceDirectoryEntry*> _entries;
};

#endif /* _RESOURCE_DIRECTORY_H */