blob: 6aff3670e793f39db424cff6fa501bca2f51f43f (
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
|
/*
* Copyright (c) 2022-2023 Jon Palmisciano. All rights reserved.
*
* Use of this source code is governed by the BSD 3-Clause license; the full
* terms of the license can be found in the LICENSE.txt file.
*/
#include "GlobalState.h"
#include <set>
#include <shared_mutex>
#include <unordered_map>
static std::unordered_map<BinaryViewID, MessageHandler*> g_messageHandlers;
static std::shared_mutex g_messageHandlerLock;
static std::unordered_map<BinaryViewID, SharedAnalysisInfo> g_viewInfos;
static std::shared_mutex g_viewInfoLock;
static std::set<BinaryViewID> g_ignoredViews;
MessageHandler* GlobalState::messageHandler(BinaryViewRef bv)
{
std::unique_lock<std::shared_mutex> lock(g_messageHandlerLock);
auto messageHandler = g_messageHandlers.find(id(bv));
if (messageHandler != g_messageHandlers.end())
return messageHandler->second;
auto newMessageHandler = new MessageHandler(bv);
g_messageHandlers[id(bv)] = newMessageHandler;
return newMessageHandler;
}
BinaryViewID GlobalState::id(BinaryViewRef bv)
{
return bv->GetFile()->GetSessionId();
}
void GlobalState::addIgnoredView(BinaryViewRef bv)
{
g_ignoredViews.insert(id(std::move(bv)));
}
bool GlobalState::viewIsIgnored(BinaryViewRef bv)
{
return g_ignoredViews.count(id(std::move(bv))) > 0;
}
SharedAnalysisInfo GlobalState::analysisInfo(BinaryViewRef data)
{
{
std::shared_lock<std::shared_mutex> lock(g_viewInfoLock);
if (auto it = g_viewInfos.find(id(data)); it != g_viewInfos.end())
if (data->GetStart() == it->second->imageBase)
return it->second;
}
std::unique_lock<std::shared_mutex> lock(g_viewInfoLock);
SharedAnalysisInfo info = std::make_shared<AnalysisInfo>();
info->imageBase = data->GetStart();
if (auto objcStubs = data->GetSectionByName("__objc_stubs"))
{
info->objcStubsStartEnd = {objcStubs->GetStart(), objcStubs->GetEnd()};
info->hasObjcStubs = true;
}
auto meta = data->QueryMetadata("Objective-C");
if (!meta)
{
g_viewInfos[id(data)] = info;
return info;
}
auto metaKVS = meta->GetKeyValueStore();
if (metaKVS["version"]->GetUnsignedInteger() != 1)
{
BinaryNinja::LogError("workflow_objc: Invalid metadata version received!");
g_viewInfos[id(data)] = info;
return info;
}
for (const auto& selAndImps : metaKVS["selRefImplementations"]->GetArray())
info->selRefToImp[selAndImps->GetArray()[0]->GetUnsignedInteger()] = selAndImps->GetArray()[1]->GetUnsignedIntegerList();
for (const auto& selAndImps : metaKVS["selImplementations"]->GetArray())
info->selToImp[selAndImps->GetArray()[0]->GetUnsignedInteger()] = selAndImps->GetArray()[1]->GetUnsignedIntegerList();
g_viewInfos[id(data)] = info;
return info;
}
bool GlobalState::hasAnalysisInfo(BinaryViewRef data)
{
return data->QueryMetadata("Objective-C") != nullptr;
}
bool GlobalState::hasFlag(BinaryViewRef bv, const std::string& flag)
{
return bv->QueryMetadata(flag);
}
void GlobalState::setFlag(BinaryViewRef bv, const std::string& flag)
{
bv->StoreMetadata(flag, new BinaryNinja::Metadata("YES"));
}
|
