Configuration.h

#ifndef ROSE_BinaryAnalysis_Partitioner2_Configuration_H
#define ROSE_BinaryAnalysis_Partitioner2_Configuration_H
#include <featureTests.h>
#ifdef ROSE_ENABLE_BINARY_ANALYSIS
#include <Rose/BinaryAnalysis/Partitioner2/BasicTypes.h>
 
#include <Rose/FileSystem.h>
#include <Sawyer/Map.h>
#include <Sawyer/Optional.h>
#include <Rose/SourceLocation.h>
#include <set>
#include <string>
 
namespace Rose {
namespace BinaryAnalysis {
namespace Partitioner2 {
 
class BasicBlockConfiguration {
    Address address_ = 0;
    std::string comment_;
    Sawyer::Optional<Address> finalInsnVa_;
    std::set<Address> successorVas_;
    SourceLocation sourceLocation_;
 
public:
    // default constructor needed by some STL containers
    BasicBlockConfiguration() {}
 
    explicit BasicBlockConfiguration(Address va): address_(va) {}
 
    Address address() const { return address_; }
 
    const std::string& comment() const { return comment_; }
    BasicBlockConfiguration& comment(const std::string &s) { comment_ = s; return *this; }
    const SourceLocation& sourceLocation() const { return sourceLocation_; }
    BasicBlockConfiguration& sourceLocation(const SourceLocation &loc) { sourceLocation_ = loc; return *this; }
    Sawyer::Optional<Address> finalInstructionVa() const { return finalInsnVa_; }
    BasicBlockConfiguration& finalInstructionVa(const Sawyer::Optional<Address> &va) { finalInsnVa_ = va; return *this; }
    const std::set<Address>& successorVas() const { return successorVas_; }
    std::set<Address>& successorVas() { return successorVas_; }
    BasicBlockConfiguration& successorVas(const std::set<Address> &vas) { successorVas_ = vas; return *this; }
    BasicBlockConfiguration& insertSuccessorVa(Address va) { successorVas_.insert(va); return *this; }
    BasicBlockConfiguration& clearSuccessorVas() { successorVas_.clear(); return *this; }
};
 
 
class DataBlockConfiguration {
    Address address_;
    std::string name_;
    std::string comment_;
    SourceLocation sourceLocation_;
 
public:
    // Default constructor needed by some STL containers
    DataBlockConfiguration(): address_(0) {}
 
    explicit DataBlockConfiguration(Address va): address_(va) {}
 
    Address address() const { return address_; }
 
    const std::string &name() const { return name_; }
    DataBlockConfiguration& name(const std::string &s) { name_ = s; return *this; }
    const std::string& comment() const { return comment_; }
    DataBlockConfiguration& comment(const std::string &s) { comment_ = s; return *this; }
    const SourceLocation& sourceLocation() const { return sourceLocation_; }
    DataBlockConfiguration& sourceLocation(const SourceLocation &loc) { sourceLocation_ = loc; return *this; }
};
 
 
class FunctionConfiguration {
    Sawyer::Optional<Address> address_;
    std::string name_, defaultName_, comment_;
    Sawyer::Optional<int64_t> stackDelta_;
    Sawyer::Optional<bool> mayReturn_;
    SourceLocation sourceLocation_;
 
public:
    // default constructor needed by some STL containers
    FunctionConfiguration() {}
 
    explicit FunctionConfiguration(Address va, const std::string &name=""): address_(va), name_(name) {}
    explicit FunctionConfiguration(const std::string &name): name_(name) {}
    FunctionConfiguration(const Sawyer::Optional<Address> &va, const std::string &name): address_(va), name_(name) {}
    Sawyer::Optional<Address> address() const { return address_; }
 
    const std::string& name() const { return name_; }
    FunctionConfiguration& name(const std::string &s);
    const std::string& defaultName() const { return defaultName_; }
    FunctionConfiguration& defaultName(const std::string &s) { defaultName_ = s; return *this; }
    const std::string& comment() const { return comment_; }
    FunctionConfiguration& comment(const std::string &s) { comment_ = s; return *this; }
    const SourceLocation& sourceLocation() const { return sourceLocation_; }
    FunctionConfiguration& sourceLocation(const SourceLocation &loc) { sourceLocation_ = loc; return *this; }
    Sawyer::Optional<int64_t> stackDelta() const { return stackDelta_; }
    FunctionConfiguration& stackDelta(const Sawyer::Optional<int64_t> &n) { stackDelta_ = n; return *this; }
    Sawyer::Optional<bool> mayReturn() const { return mayReturn_; }
    FunctionConfiguration& mayReturn(const Sawyer::Optional<bool> &b) { mayReturn_ = b; return *this; }
};
 
class AddressConfiguration {
    Address address_ = 0;
    std::string name_;
    std::string comment_;
    SourceLocation sourceLocation_;
 
public:
    // Default constructor needed by STL containers
    AddressConfiguration() {}
 
    explicit AddressConfiguration(Address va)
        : address_(va) {}
 
    Address address() const { return address_; }
 
    const std::string &name() const { return name_; }
    AddressConfiguration& name(const std::string &s) { name_ = s; return *this; }
    const std::string &comment() const { return comment_; }
    AddressConfiguration& comment(const std::string &s) { comment_ = s; return *this; }
    const SourceLocation& sourceLocation() const { return sourceLocation_; }
    AddressConfiguration& sourceLocation(const SourceLocation &loc) { sourceLocation_ = loc; return *this; }
};
 
class Configuration {
public:
    typedef Sawyer::Container::Map<Address, BasicBlockConfiguration> BasicBlockConfigurations;
    typedef Sawyer::Container::Map<Address, DataBlockConfiguration> DataBlockConfigurations;
    typedef Sawyer::Container::Map<Address, FunctionConfiguration> FunctionConfigurationsByAddress;
    typedef Sawyer::Container::Map<std::string, FunctionConfiguration> FunctionConfigurationsByName;
    typedef Sawyer::Container::Map<Address, AddressConfiguration> AddressConfigurations;
 
private:
    BasicBlockConfigurations bblockConfigurations_;
    DataBlockConfigurations dblockConfigurations_;
    FunctionConfigurationsByAddress functionConfigurationsByAddress_;
    FunctionConfigurationsByName functionConfigurationsByName_;
    AddressConfigurations addressConfigurations_;
 
public:
    static std::string fileFormatDoc();
 
    void loadFromFile(const FileSystem::Path &fileName);
 
    const BasicBlockConfigurations& basicBlocks() const { return bblockConfigurations_; }
    BasicBlockConfigurations& basicBlocks() { return bblockConfigurations_; }
    const BasicBlockConfiguration& basicBlock(Address) const;
 
    const DataBlockConfigurations& dataBlocks() const { return dblockConfigurations_; }
    DataBlockConfigurations& dataBlocks() { return dblockConfigurations_; }
    const DataBlockConfiguration& dataBlock(Address) const;
 
    const FunctionConfigurationsByAddress& functionConfigurationsByAddress() const { return functionConfigurationsByAddress_; }
    FunctionConfigurationsByAddress& functionConfigurationsByAddress() { return functionConfigurationsByAddress_; }
    const FunctionConfigurationsByName& functionConfigurationsByName() const { return functionConfigurationsByName_; }
    FunctionConfigurationsByName& functionConfigurationsByName() { return functionConfigurationsByName_; }
    const FunctionConfiguration& function(Address) const;
    const FunctionConfiguration& function(const std::string &name) const;
    const AddressConfigurations& addresses() const { return addressConfigurations_; }
    AddressConfigurations& addresses() { return addressConfigurations_; }
    const AddressConfiguration& address(Address) const;
 
    BasicBlockConfiguration& insertMaybeBasicBlock(Address va);
 
    DataBlockConfiguration& insertMaybeDataBlock(Address va);
 
    FunctionConfiguration& insertMaybeFunction(Address va, const std::string &name="");
    FunctionConfiguration& insertMaybeFunction(const std::string &name);
    AddressConfiguration& insertMaybeAddress(Address va);
 
    bool insertConfiguration(const BasicBlockConfiguration&);
 
    bool insertConfiguration(const DataBlockConfiguration&);
 
    bool insertConfiguration(const FunctionConfiguration&);
 
    bool insertConfiguration(const AddressConfiguration&);
 
    std::string basicBlockComment(Address bblockVa) const;
 
    Sawyer::Optional<Address> basicBlockFinalInstructionVa(Address bblockVa) const;
 
    std::set<Address> basicBlockSuccessorVas(Address bblockVa) const;
 
    std::string dataBlockName(Address dblockVa) const;
 
    std::string dataBlockComment(Address dblockVa) const;
 
    std::string functionName(Address functionVa) const;
 
    std::string functionDefaultName(Address functionVa) const;
 
    std::string functionComment(Address functionVa) const;
    std::string functionComment(const std::string &functionName) const;
    std::string functionComment(const FunctionPtr&) const;
    Sawyer::Optional<int64_t> functionStackDelta(Address functionVa) const;
    Sawyer::Optional<int64_t> functionStackDelta(const std::string &functionName) const;
    Sawyer::Optional<int64_t> functionStackDelta(const FunctionPtr&) const;
    Sawyer::Optional<bool> functionMayReturn(Address functionVa) const;
    Sawyer::Optional<bool> functionMayReturn(const std::string &functionName) const;
    Sawyer::Optional<bool> functionMayReturn(const FunctionPtr&) const;
    std::string addressComment(Address va) const;
 
    std::string comment(Address va) const;
 
    void print(std::ostream&) const;
};
 
} // namespace
} // namespace
} // namespace
 
std::ostream& operator<<(std::ostream&, const Rose::BinaryAnalysis::Partitioner2::Configuration&);
 
#endif
#endif