exported_func.h

Go to the documentation of this file.

 
#pragma once
 
#include <windows.h>
#include <string>
#include <algorithm>
#include <set>
 
namespace peconv {
 
    const DWORD INVALID_ORD_VALUE = DWORD(-1);
 
    size_t forwarder_name_len(BYTE* fPtr); 
 
    std::string get_dll_shortname(const std::string& str);
 
    std::string get_func_name(const std::string& str);
 
    std::string ordinal_to_string(DWORD func_ordinal);
 
    bool is_ordinal_string(const std::string& str);
 
    DWORD ordinal_string_to_val(const std::string& str);
 
    std::string format_dll_func(const std::string& str);
 
    class ExportedFunc
    {
    public:
        static std::string formatName(std::string name);
 
        static bool isTheSameFuncName(const peconv::ExportedFunc& func1, const peconv::ExportedFunc& func2);
 
        static bool isTheSameDllName(const peconv::ExportedFunc& func1, const peconv::ExportedFunc& func2);
 
        static bool isTheSameFunc(const peconv::ExportedFunc& func1, const peconv::ExportedFunc& func2);
 
        std::string libName;
        std::string funcName;
        DWORD funcOrdinal;
        bool isByOrdinal;
 
        //default constructor:
        ExportedFunc() : funcOrdinal(INVALID_ORD_VALUE), isByOrdinal(false) {}
 
        ExportedFunc(const ExportedFunc& other);
        ExportedFunc(std::string libName, std::string funcName, DWORD funcOrdinal);
        ExportedFunc(std::string libName, DWORD funcOrdinal);
        ExportedFunc(const std::string &forwarderName);
 
        bool operator < (const ExportedFunc& other) const
        {
            //if only one function is named, give the preference to the named one:
            const size_t thisNameLen = this->funcName.length();
            const size_t otherNameLen = other.funcName.length();
            if (thisNameLen == 0 && otherNameLen > 0) {
                return false;
            }
            if (thisNameLen > 0 && otherNameLen == 0) {
                return true;
            }
            //select by shorter lib name:
            int cmp = libName.compare(other.libName);
            if (cmp != 0) {
                return cmp < 0;
            }
            if (thisNameLen == 0 || otherNameLen == 0) {
                return this->funcOrdinal < other.funcOrdinal;
            }
            if (thisNameLen != otherNameLen) {
                return thisNameLen < otherNameLen;
            }
            cmp = funcName.compare(other.funcName);
            return cmp < 0;
        }
 
        std::string toString() const;
 
        std::string nameToString() const;
 
        bool isValid() const
        {
            return (funcName != "" || funcOrdinal != INVALID_ORD_VALUE);
        }
    };
 
}; //namespace peconv