summaryrefslogtreecommitdiff
path: root/demangle.cpp
blob: 3217905cb39bdc4672da7ae8dc9b804914b18a34 (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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
#include "binaryninjaapi.h"
#include <string>
using namespace std;

namespace BinaryNinja {
	bool DemangleLLVM(const std::string& mangledName, QualifiedName& outVarName,
		BinaryView* view)
	{
		const bool simplify = Settings::Instance()->Get<bool>("analysis.types.templateSimplifier", view);
		return DemangleLLVM(mangledName, outVarName, simplify);
	}

	bool DemangleLLVM(const std::string& mangledName, QualifiedName& outVarName,
		const bool simplify)
	{
		char** localVarName = nullptr;
		size_t localSize = 0;
		if (!BNDemangleLLVM(mangledName.c_str(), &localVarName, &localSize, simplify))
			return false;
		for (size_t i = 0; i < localSize; i++)
		{
			outVarName.push_back(localVarName[i]);
		}
		BNFreeDemangledName(&localVarName, localSize);
		return true;
	}

	bool DemangleMS(Architecture* arch, const std::string& mangledName, Ref<Type>& outType, QualifiedName& outVarName,
	    BinaryView* view)
	{
		const bool simplify = Settings::Instance()->Get<bool>("analysis.types.templateSimplifier", view);
		return DemangleMS(arch, mangledName, outType, outVarName, simplify);
	}

	bool DemangleMS(Architecture* arch, const std::string& mangledName, Ref<Type>& outType, QualifiedName& outVarName,
	    const bool simplify)
	{
		BNType* localType = nullptr;
		char** localVarName = nullptr;
		size_t localSize = 0;
		if (!BNDemangleMS(arch->GetObject(), mangledName.c_str(), &localType, &localVarName, &localSize, simplify))
			return false;
		outType = localType ? new Type(localType) : nullptr;
		for (size_t i = 0; i < localSize; i++)
		{
			outVarName.push_back(localVarName[i]);
		}
		BNFreeDemangledName(&localVarName, localSize);
		return true;
	}

	bool DemangleGNU3(Ref<Architecture> arch, const std::string& mangledName, Ref<Type>& outType, QualifiedName& outVarName,
	    BinaryView* view)
	{
		const bool simplify = Settings::Instance()->Get<bool>("analysis.types.templateSimplifier", view);
		return DemangleGNU3(arch, mangledName, outType, outVarName, simplify);
	}

	bool DemangleGNU3(Ref<Architecture> arch, const std::string& mangledName, Ref<Type>& outType, QualifiedName& outVarName,
	    const bool simplify)
	{
		BNType* localType;
		char** localVarName = nullptr;
		size_t localSize = 0;
		if (!BNDemangleGNU3(arch->GetObject(), mangledName.c_str(), &localType, &localVarName, &localSize, simplify))
			return false;
		outType = localType ? new Type(localType) : nullptr;
		outVarName.clear();
		for (size_t i = 0; i < localSize; i++)
		{
			outVarName.push_back(localVarName[i]);
		}
		BNFreeDemangledName(&localVarName, localSize);
		return true;
	}


	bool IsGNU3MangledString(const std::string& mangledName)
	{
		return BNIsGNU3MangledString(mangledName.c_str());
	}


	string SimplifyToString(const string& input)
	{
		return BNRustSimplifyStrToStr(input.c_str());
	}


	string SimplifyToString(const QualifiedName& input)
	{
		return BNRustSimplifyStrToStr(input.GetString().c_str());
	}


	QualifiedName SimplifyToQualifiedName(const string& input, bool simplify)
	{
		BNQualifiedName name = BNRustSimplifyStrToFQN(input.c_str(), simplify);
		QualifiedName result = QualifiedName::FromAPIObject(&name);
		BNFreeQualifiedName(&name);
		return result;
	}


	QualifiedName SimplifyToQualifiedName(const QualifiedName& input)
	{
		BNQualifiedName name = BNRustSimplifyStrToFQN(input.GetString().c_str(), true);
		QualifiedName result = QualifiedName::FromAPIObject(&name);
		BNFreeQualifiedName(&name);
		return result;
	}
}  // namespace BinaryNinja