pe_exe.h

Go to the documentation of this file.

/* xoreos - A reimplementation of BioWare's Aurora engine
 *
 * xoreos is the legal property of its developers, whose names
 * can be found in the AUTHORS file distributed with this source
 * distribution.
 *
 * xoreos is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 3
 * of the License, or (at your option) any later version.
 *
 * xoreos is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with xoreos. If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef COMMON_PE_EXE_H
#define COMMON_PE_EXE_H

#include <map>
#include <vector>

#include <boost/noncopyable.hpp>

#include "src/common/types.h"
#include "src/common/ustring.h"
#include "src/common/scopedptr.h"

namespace Common {

class SeekableReadStream;
class UString;

class PEResourceID {
public:
    PEResourceID();
    PEResourceID(UString x);
    PEResourceID(uint32 x);

    PEResourceID &operator=(UString string);
    PEResourceID &operator=(uint32 x);

    bool operator==(const UString &x) const;
    bool operator==(const uint32 &x) const;
    bool operator==(const PEResourceID &x) const;

    bool operator<(const PEResourceID &x) const;

    UString getString() const;
    uint32 getID() const;
    UString toString() const;

private:
    enum IDType {
        kIDTypeNull,      
        kIDTypeNumerical, 
        kIDTypeString     
    } _idType;

    UString _name;        
    uint32 _id;           
};

enum PEResourceType {
    kPECursor = 0x1,
    kPEBitmap = 0x2,
    kPEIcon = 0x3,
    kPEMenu = 0x4,
    kPEDialog = 0x5,
    kPEString = 0x6,
    kPEFontDir = 0x7,
    kPEFont = 0x8,
    kPEAccelerator = 0x9,
    kPERCData = 0xA,
    kPEMessageTable = 0xB,
    kPEGroupCursor = 0xC,
    kPEGroupIcon = 0xE,
    kPEVersion = 0x10,
    kPEDlgInclude = 0x11,
    kPEPlugPlay = 0x13,
    kPEVXD = 0x14,
    kPEAniCursor = 0x15,
    kPEAniIcon = 0x16,
    kPEHTML = 0x17,
    kPEManifest = 0x18
};

class PEResources : boost::noncopyable {
public:
    PEResources(SeekableReadStream *exe);
    ~PEResources();

    const std::vector<PEResourceID> getTypeList() const;

    const std::vector<PEResourceID> getNameList(const PEResourceID &type) const;

    const std::vector<PEResourceID> getLangList(const PEResourceID &type, const PEResourceID &name) const;

    SeekableReadStream *getResource(const PEResourceID &type, const PEResourceID &name);

    SeekableReadStream *getResource(const PEResourceID &type, const PEResourceID &name,
                                    const PEResourceID &lang);

private:
    struct Section {
        uint32 virtualAddress;
        uint32 size;
        uint32 offset;
    };

    struct Resource {
        uint32 offset;
        uint32 size;
    };

    typedef std::map<PEResourceID, Resource> LangMap;
    typedef std::map<PEResourceID,  LangMap> NameMap;
    typedef std::map<PEResourceID,  NameMap> TypeMap;

    std::map<UString, Section> _sections;


    ScopedPtr<SeekableReadStream> _exe;

    PEResourceID _curType;
    PEResourceID _curName;
    PEResourceID _curLang;

    TypeMap _resources;


    bool loadFromEXE(SeekableReadStream &exe);

    void parseResourceLevel(SeekableReadStream &exe, Section &section, uint32 offset, int level);
};

} // End of namespace Common

#endif // COMMON_PE_EXE_H