src/dbg/_scriptapi_label.cpp - development - githubmirror/x64dbg

src/dbg/_scriptapi_label.cpp
86 строк2 KB
ZehMatt
Add temporary label support for scripting
19 июл 2021, 18:36
19 июл 2021, 18:36c85d52b
#include "_scriptapi_label.h"
#include "_scriptapi_module.h"
#include "label.h"

SCRIPT_EXPORT bool Script::Label::Set(duint addr, const char* text, bool manual)
{
    return LabelSet(addr, text, manual);
}

SCRIPT_EXPORT bool Script::Label::Set(duint addr, const char* text, bool manual, bool temporary)
{
    return LabelSet(addr, text, manual, temporary);
}

SCRIPT_EXPORT bool Script::Label::Set(const LabelInfo* info)
{
    if(!info)
        return false;
    auto base = Module::BaseFromName(info->mod);
    if(!base)
        return false;
    return Set(base + info->rva, info->text, info->manual, false);
}

SCRIPT_EXPORT bool Script::Label::FromString(const char* label, duint* addr)
{
    return LabelFromString(label, addr);
}

SCRIPT_EXPORT bool Script::Label::Get(duint addr, char* text)
{
    return LabelGet(addr, text);
}

SCRIPT_EXPORT bool Script::Label::IsTemporary(duint addr)
{
    return LabelIsTemporary(addr);
}

SCRIPT_EXPORT bool Script::Label::GetInfo(duint addr, LabelInfo* info)
{
    LABELSINFO label;
    if(!LabelGetInfo(addr, &label))
        return false;
    if(info)
    {
        strcpy_s(info->mod, label.mod().c_str());
        info->rva = label.addr;
        strcpy_s(info->text, label.text.c_str());
        info->manual = label.manual;
    }
    return true;
}

SCRIPT_EXPORT bool Script::Label::Delete(duint addr)
{
    return LabelDelete(addr);
}

SCRIPT_EXPORT void Script::Label::DeleteRange(duint start, duint end)
{
    LabelDelRange(start, end, false);
}

SCRIPT_EXPORT void Script::Label::Clear()
{
    LabelClear();
}

SCRIPT_EXPORT bool Script::Label::GetList(ListOf(LabelInfo) list)
{
    std::vector<LABELSINFO> labelList;
    LabelGetList(labelList);
    std::vector<LabelInfo> labelScriptList;
    labelScriptList.reserve(labelList.size());
    for(const auto & label : labelList)
    {
        LabelInfo scriptLabel;
        strcpy_s(scriptLabel.mod, label.mod().c_str());
        scriptLabel.rva = label.addr;
        strcpy_s(scriptLabel.text, label.text.c_str());
        scriptLabel.manual = label.manual;
        labelScriptList.push_back(scriptLabel);
    }
    return BridgeList<LabelInfo>::CopyData(list, labelScriptList);
}