summaryrefslogtreecommitdiff
path: root/view/sharedcache/core/ObjC.cpp
blob: c763b321e9241564904aa9bbba4de5382863114d (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
113
114
115
116
#include "ObjC.h"

using namespace BinaryNinja;
using namespace DSCObjC;
using namespace SharedCacheCore;

DSCObjCReader::DSCObjCReader(SharedCache* cache, size_t addressSize) :
	m_reader(VMReader(cache->GetVMMap())), m_addressSize(addressSize)
{
}

void DSCObjCReader::Read(void* dest, size_t len)
{
	m_reader.Read(dest, len);
}

std::string DSCObjCReader::ReadCString()
{
	return m_reader.ReadCString(m_reader.GetOffset());
}

uint8_t DSCObjCReader::Read8()
{
	return m_reader.Read8();
}

uint16_t DSCObjCReader::Read16()
{
	return m_reader.Read16();
}

uint32_t DSCObjCReader::Read32()
{
	return m_reader.Read32();
}

uint64_t DSCObjCReader::Read64()
{
	return m_reader.Read64();
}

int8_t DSCObjCReader::ReadS8()
{
	return m_reader.ReadS8();
}

int16_t DSCObjCReader::ReadS16()
{
	return m_reader.ReadS16();
}

int32_t DSCObjCReader::ReadS32()
{
	return m_reader.ReadS32();
}

int64_t DSCObjCReader::ReadS64()
{
	return m_reader.ReadS64();
}

uint64_t DSCObjCReader::ReadPointer()
{
	return m_reader.ReadPointer();
}

uint64_t DSCObjCReader::GetOffset() const
{
	return m_reader.GetOffset();
}

void DSCObjCReader::Seek(uint64_t offset)
{
	m_reader.Seek(offset);
}

void DSCObjCReader::SeekRelative(int64_t offset)
{
	m_reader.SeekRelative(offset);
}

VMReader& DSCObjCReader::GetVMReader()
{
	return m_reader;
}

std::shared_ptr<ObjCReader> DSCObjCProcessor::GetReader()
{
	return std::make_shared<DSCObjCReader>(m_cache, m_data->GetAddressSize());
}

void DSCObjCProcessor::GetRelativeMethod(ObjCReader* reader, method_t& meth)
{
	if (m_customRelativeMethodSelectorBase.has_value())
	{
		meth.name = m_customRelativeMethodSelectorBase.value() + reader->ReadS32();
		meth.types = reader->GetOffset() + reader->ReadS32();
		meth.imp = reader->GetOffset() + reader->ReadS32();
	}
	else
	{
		ObjCProcessor::GetRelativeMethod(reader, meth);
	}
}

uint64_t DSCObjCProcessor::GetObjCRelativeMethodBaseAddress(ObjCReader* reader)
{
	auto objCRelativeMethodsBaseAddr = m_cache->GetObjCRelativeMethodBaseAddress(static_cast<DSCObjCReader*>(reader)->GetVMReader());
	m_customRelativeMethodSelectorBase = objCRelativeMethodsBaseAddr;
	return objCRelativeMethodsBaseAddr;
}

DSCObjCProcessor::DSCObjCProcessor(BinaryView* data, SharedCache* cache, bool isBackedByDatabase) :
	ObjCProcessor(data, "SharedCache.ObjC", isBackedByDatabase, true), m_cache(cache)
{
}