cutter/src/widgets/RizinGraphWidget.cpp

#include "RizinGraphWidget.h"
#include "ui_RizinGraphWidget.h"

#include <QJsonValue>
#include <QJsonArray>
#include <QJsonObject>

RizinGraphWidget::RizinGraphWidget(MainWindow *main)
    : CutterDockWidget(main),
      ui(new Ui::RizinGraphWidget),
      graphView(new GenericRizinGraphView(this, main))
{
    ui->setupUi(this);
    ui->verticalLayout->addWidget(graphView);
    connect(ui->refreshButton, &QPushButton::pressed, this, [this]() { graphView->refreshView(); });
    struct GraphType
    {
        QChar commandChar;
        QString label;
    } types[] = {
        { 'a', tr("Data reference graph (aga)") },
        { 'A', tr("Global data references graph (agA)") },
        // {'c', tr("c - Function callgraph")},
        // {'C', tr("C - Global callgraph")},
        // {'f', tr("f - Basic blocks function graph")},
        { 'i', tr("Imports graph (agi)") },
        { 'r', tr("References graph (agr)") },
        { 'R', tr("Global references graph (agR)") },
        { 'x', tr("Cross references graph (agx)") },
        { 'g', tr("Custom graph (agg)") },
        { ' ', tr("User command") },
    };
    for (auto &graphType : types) {
        if (graphType.commandChar != ' ') {
            ui->graphType->addItem(graphType.label, graphType.commandChar);
        } else {
            ui->graphType->addItem(graphType.label, QVariant());
        }
    }
    connect<void (QComboBox::*)(int)>(ui->graphType, &QComboBox::currentIndexChanged, this,
                                      &RizinGraphWidget::typeChanged);
    connect(ui->customCommand, &QLineEdit::textEdited, this,
            [this]() { graphView->setGraphCommand(ui->customCommand->text()); });
    connect(ui->customCommand, &QLineEdit::returnPressed, this, [this]() {
        graphView->setGraphCommand(ui->customCommand->text());
        graphView->refreshView();
    });
    ui->customCommand->hide();
    typeChanged();
}

RizinGraphWidget::~RizinGraphWidget() {}

void RizinGraphWidget::typeChanged()
{
    auto currentData = ui->graphType->currentData();
    if (currentData.isNull()) {
        ui->customCommand->setVisible(true);
        graphView->setGraphCommand(ui->customCommand->text());
        ui->customCommand->setFocus();
    } else {
        ui->customCommand->setVisible(false);
        auto command = QString("ag%1").arg(currentData.toChar());
        graphView->setGraphCommand(command);
        graphView->refreshView();
    }
}

GenericRizinGraphView::GenericRizinGraphView(RizinGraphWidget *parent, MainWindow *main)
    : SimpleTextGraphView(parent, main), refreshDeferrer(nullptr, this)
{
    refreshDeferrer.registerFor(parent);
    connect(&refreshDeferrer, &RefreshDeferrer::refreshNow, this,
            &GenericRizinGraphView::refreshView);
}

void GenericRizinGraphView::setGraphCommand(QString cmd)
{
    graphCommand = cmd;
}

void GenericRizinGraphView::refreshView()
{
    if (!refreshDeferrer.attemptRefresh(nullptr)) {
        return;
    }
    SimpleTextGraphView::refreshView();
}

void GenericRizinGraphView::loadCurrentGraph()
{
    blockContent.clear();
    blocks.clear();

    if (graphCommand.isEmpty()) {
        return;
    }

    QJsonDocument functionsDoc = Core()->cmdj(QString("%1j").arg(graphCommand));
    auto nodes = functionsDoc.object()["nodes"].toArray();

    for (const QJsonValueRef &value : nodes) {
        QJsonObject block = value.toObject();
        uint64_t id = block["id"].toVariant().toULongLong();

        QString content;
        QString title = block["title"].toString();
        QString body = block["body"].toString();
        if (!title.isEmpty() && !body.isEmpty()) {
            content = title + "/n" + body;
        } else {
            content = title + body;
        }

        auto edges = block["out_nodes"].toArray();
        GraphLayout::GraphBlock layoutBlock;
        layoutBlock.entry = id;
        for (auto edge : edges) {
            auto targetId = edge.toVariant().toULongLong();
            layoutBlock.edges.emplace_back(targetId);
        }

        addBlock(std::move(layoutBlock), content);
    }

    cleanupEdges(blocks);

    computeGraphPlacement();

    if (graphCommand != lastShownCommand) {
        selectedBlock = NO_BLOCK_SELECTED;
        lastShownCommand = graphCommand;
        center();
    }
}