exported_func.cpp

Go to the documentation of this file.

#include "peconv/exported_func.h"
 
#include <string>
#include <algorithm>
#include <sstream>
#include <iomanip>
#include <iostream>
#include "peconv/file_util.h"
 
using namespace peconv;
 
std::string peconv::get_dll_shortname(const std::string& str)
{
    std::size_t len = str.length();
    size_t ext_pos = len;
    size_t separator_pos = 0;
    for (size_t k = len; k != 0; k--) {
        size_t i = k - 1;
        char c = str[i];
        // search first '.' from the end:
        if (c == '.' && ext_pos == len) {
            ext_pos = i;
        }
        // search first path separator from the end:
        if (c == '\\' || c == '/') {
            separator_pos = k;
            break;
        }
    }
    const size_t new_len = ext_pos - separator_pos;
    std::string name = str.substr(separator_pos, new_len);
    std::transform(name.begin(), name.end(), name.begin(), tolower);
    return name;
}
 
size_t peconv::forwarder_name_len(BYTE* fPtr)
{
    // names can be also mangled, i.e. MSVCRT.??0__non_rtti_object@std@@QAE@ABV01@@Z
    bool has_dot = false;
    size_t len = 0;
    while ((*fPtr >= 'a' && *fPtr <= 'z')
            || (*fPtr >= 'A' && *fPtr <= 'Z')
            || (*fPtr >= '0' && *fPtr <= '9')
            || (*fPtr == '.')
            || (*fPtr == '_')
            || (*fPtr == '#') 
            || (*fPtr == '@')
            || (*fPtr == '?')
            || (*fPtr == '-'))
    {
        if (*fPtr == '.') has_dot = true;
        len++;
        fPtr++;
    }
    if (*fPtr == '\0') {
        if (!has_dot) {
            return 0; //this is not a valid forwarder
        }
        return len;
    }
    return 0;
}
 
std::string peconv::get_func_name(const std::string& str)
{
    std::size_t len = str.length();
    std::size_t ext = str.find_last_of(".");
    if (ext >= len) return "";
 
    std::string name = str.substr(ext+1, len - (ext+1));
    return name;
}
 
std::string peconv::ordinal_to_string(DWORD func_ordinal)
{
    std::stringstream stream;
    stream << "#";
    stream << std::dec << func_ordinal;
    return stream.str();
}
 
bool peconv::is_ordinal_string(const std::string& func_name_str)
{
    if (func_name_str.length() < 2) return false;
    return (func_name_str[0] == '#');
}
 
DWORD peconv::ordinal_string_to_val(const std::string& func_name_str)
{
    if (!is_ordinal_string(func_name_str)) return 0;
    const char* func_name = func_name_str.c_str();
    return atoi(func_name + 1);
}
 
std::string peconv::format_dll_func(const std::string& str)
{
    std::string dllName = get_dll_shortname(str);
    std::string funcName = get_func_name(str);
    if (dllName.length() == 0 || funcName.length() == 0) {
        return "";
    }
    std::transform(dllName.begin(), dllName.end(), dllName.begin(), tolower);
    return dllName + "." + funcName;
}
 
ExportedFunc::ExportedFunc(std::string libName, std::string funcName, DWORD funcOrdinal)
{
    this->libName = ExportedFunc::formatName(libName);
    this->funcName = funcName;
    this->funcOrdinal = funcOrdinal;
    this->isByOrdinal = false;
}
 
ExportedFunc::ExportedFunc(std::string libName, DWORD funcOrdinal)
{
    this->libName = ExportedFunc::formatName(libName);
    this->funcOrdinal = funcOrdinal;
    this->isByOrdinal = true;
}
 
ExportedFunc::ExportedFunc(const ExportedFunc& other)
{
    this->libName = other.libName;
    this->funcName = other.funcName;
    this->funcOrdinal = other.funcOrdinal;
    this->isByOrdinal = other.isByOrdinal;
}
 
ExportedFunc::ExportedFunc(const std::string &forwarderName)
{
    this->libName = get_dll_shortname(forwarderName);
    std::string func_name_str =  get_func_name(forwarderName);
    if (func_name_str.length() < 2) {
        this->funcOrdinal = -1;
        this->funcName = "";
        this->isByOrdinal = false;
#ifdef _DEBUG
        std::cerr << "Invalid function data" << std::endl;
#endif
        return;
    }
    if (is_ordinal_string(func_name_str)) {
        // it is an ordinal in a string form, i.e.: "COMBASE.#110"
        this->funcOrdinal = peconv::ordinal_string_to_val(func_name_str);
        this->isByOrdinal = true;
        this->funcName = "";
        //std::cout << "[O] Adding forwarded func: " << forwarderName << " parsed: " << this->toString() << std::endl;
    } else {
        this->funcName = func_name_str;
        this->isByOrdinal = false;
        this->funcOrdinal = 0;
        //std::cout << "[N] Adding forwarded func:" << this->toString() << std::endl;
    }
}
 
std::string ExportedFunc::formatName(std::string name)
{
    if (name.length() == 0) {
        return "";
    }
    std::transform(name.begin(), name.end(), name.begin(), tolower);
    return name;
}
 
bool ExportedFunc::isTheSameFuncName(const peconv::ExportedFunc& func1, const peconv::ExportedFunc& func2)
{
    if (!func1.isByOrdinal && !func2.isByOrdinal) {
        if (func1.funcName == func2.funcName) {
            return true;
        }
    }
    if (func1.funcOrdinal == func2.funcOrdinal) {
        return true;
    }
    return false;
}
 
namespace peconv
{
    bool is_valid_extension(const std::string &ext)
    {
        if (ext.length() > 3) {
            //probably not an extension
            return false;
        }
        for (size_t j = 0; j < ext.length(); j++) {
            if (!isalpha(ext[j])) {
                return false;
            }
        }
        return true;
    }
 
    std::string remove_module_extension(IN const std::string str)
    {
        size_t len = str.length();
        size_t ext_pos = find_extension_pos(str);
        if (ext_pos == len) {
            return str;
        }
        std::string ext = str.substr(ext_pos + 1);
        if (is_valid_extension(ext)) {
            std::string  str1 = str.substr(0, ext_pos);
            return str1;
        }
        return str;
    }
}; //namespace peconv
 
bool ExportedFunc::isTheSameDllName(const peconv::ExportedFunc& func1, const peconv::ExportedFunc& func2)
{
    const std::string file1 = peconv::get_file_name(func1.libName);
    const std::string file2 = peconv::get_file_name(func2.libName);
    if (file1 == file2) {
        return true;
    }
    const std::string short1 = peconv::remove_module_extension(file1);
    const std::string short2 = peconv::remove_module_extension(file2);
    if (short1 == short2) {
        return true;
    }
    return false;
}
 
bool ExportedFunc::isTheSameFunc(const peconv::ExportedFunc& func1, const peconv::ExportedFunc& func2)
{
    if (!peconv::ExportedFunc::isTheSameFuncName(func1, func2)) {
        return false;
    }
    if (!isTheSameDllName(func1, func2)) {
        return false;
    }
    return true;
}
 
std::string ExportedFunc::toString() const
{
    if (!isValid()) {
        return "[Invalid func]";
    }
    std::stringstream stream;
    stream << this->libName;
    stream << ".";
    if (!this->isByOrdinal) {
        stream << this->funcName;
        stream << " ";
    }
    stream << ordinal_to_string(this->funcOrdinal);
    return stream.str();
}
 
std::string ExportedFunc::nameToString() const
{
    if (!isValid()) {
        return "";
    }
    if (this->isByOrdinal) {
        return ordinal_to_string(this->funcOrdinal);
    }
    return this->funcName;
}