dll_export_iterator.hpp

Go to the documentation of this file.

// ===-- include/psystem/dll_export_iterator.hpp ---------- -*- C++ -*- --=== //
// Copyright (c) 2015 Matt Bisson.  All rights reserved.

#pragma once
#ifndef PSYSTEM_DLL_EXPORT_ITERATOR_HPP
#define PSYSTEM_DLL_EXPORT_ITERATOR_HPP

#include "psystem_ns.hpp"

namespace psystem {

class dll_export_iterator
    // : public std::iterator<
    //     std::forward_iterator_tag,
    //     std::pair<psystem::address_t, char const *>,
    //     size_t>
{
public:
    dll_export_iterator(
        uint8_t const *base_address,
        integral_address_t const va_offset,
        uint16_t const index,
        uint32_t const *export_rva,
        uint16_t const *idx_with_name,
        uint32_t const *export_name_rva) noexcept;

public:
    dll_export_iterator& operator++();

    bool operator==(dll_export_iterator const& other) const;

    bool operator!=(dll_export_iterator const& other) const;

public:
    char const *get_export_name() const
    {
        return reinterpret_cast<char const*>(
            m_base_address + *m_export_name_rva + m_va_offset);
    }

private:
    uint8_t const *m_base_address;

    integral_address_t m_va_offset;

public:
    uint16_t const *m_idx_with_name;

    uint16_t m_current_idx;

    uint32_t const *m_export_rva;

    uint32_t const *m_export_name_rva;
};

} // namespace psystem

#endif // PSYSTEM_DLL_EXPORT_ITERATOR_HPP