backgroundtask.cpp


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
75
76
77
78
79
80
81
82

#include "binaryninjaapi.h"

using namespace std;
using namespace BinaryNinja;


BackgroundTask::BackgroundTask(BNBackgroundTask* task)
{
	m_object = task;
}


BackgroundTask::BackgroundTask(const string& initialText, bool canCancel)
{
	m_object = BNBeginBackgroundTask(initialText.c_str(), canCancel);
}


bool BackgroundTask::CanCancel() const
{
	return BNCanCancelBackgroundTask(m_object);
}


bool BackgroundTask::IsCancelled() const
{
	return BNIsBackgroundTaskCancelled(m_object);
}


bool BackgroundTask::IsFinished() const
{
	return BNIsBackgroundTaskFinished(m_object);
}


string BackgroundTask::GetProgressText() const
{
	char* text = BNGetBackgroundTaskProgressText(m_object);
	string result = text;
	BNFreeString(text);
	return result;
}


uint64_t BackgroundTask::GetRuntimeSeconds() const
{
	return BNGetBackgroundTaskRuntimeSeconds(m_object);
}


void BackgroundTask::Cancel()
{
	BNCancelBackgroundTask(m_object);
}


void BackgroundTask::Finish()
{
	BNFinishBackgroundTask(m_object);
}


void BackgroundTask::SetProgressText(const string& text)
{
	BNSetBackgroundTaskProgressText(m_object, text.c_str());
}


vector<Ref<BackgroundTask>> BackgroundTask::GetRunningTasks()
{
	size_t count;
	BNBackgroundTask** tasks = BNGetRunningBackgroundTasks(&count);

	vector<Ref<BackgroundTask>> result;
	result.reserve(count);
	for (size_t i = 0; i < count; i++)
		result.push_back(new BackgroundTask(BNNewBackgroundTaskReference(tasks[i])));

	BNFreeBackgroundTaskList(tasks, count);
	return result;
}