blob: ccbe8da1830982f003039a298994d59382f0b66e (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
#pragma once
#include <QDialog>
#include <QStackedWidget>
#include <QComboBox>
#include <QProgressBar>
#include <QThread>
#include <QVBoxLayout>
#include <QLineEdit>
#include <QPushButton>
#include <QSortFilterProxyModel>
#include "binaryninjaapi.h"
#include "source.h"
#include "warp.h"
#include "source.h"
// Worker to commit a file to a container
class WarpCommitWorker : public QThread
{
Q_OBJECT
Warp::Ref<Warp::Container> m_container;
Warp::Source m_source;
Warp::Ref<Warp::File> m_file;
public:
WarpCommitWorker(Warp::Ref<Warp::Container> container, Warp::Source source, Warp::Ref<Warp::File> file,
QObject* parent = nullptr) : QThread(parent), m_container(container), m_source(source), m_file(file)
{}
void run() override
{
for (const auto& chunk : m_file->GetChunks())
{
if (auto target = chunk->GetTarget())
m_container->AddFunctions(*target, m_source, chunk->GetFunctions());
m_container->AddTypes(m_source, chunk->GetTypes());
}
const bool result = m_container->CommitSource(m_source);
emit finishedCommitting(result);
}
signals:
void finishedCommitting(bool success);
};
class CommitDialog : public QDialog
{
Q_OBJECT
public:
explicit CommitDialog(Warp::Ref<Warp::File> file, QWidget* parent = nullptr);
private slots:
void onContainerChanged(int index);
void onCreateNewSource();
void onCommit();
void onCommitFinished(bool success);
private:
void populateContainers();
Warp::Ref<Warp::File> m_file;
std::vector<Warp::Ref<Warp::Container>> m_containers;
QComboBox* m_containerCombo;
QLineEdit* m_sourceFilter;
QPushButton* m_addSourceButton;
QSortFilterProxyModel* m_proxyModel;
WarpSourcesView* m_sourcesView;
QPushButton* m_commitButton;
};
|
