binaryninja-api/view/sharedcache/core/ObjC.h at f2c275d91afcb9cc24baaaf2e3b14b4f96b2f2e9 · Vector35/binaryninja-api · GitHub

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
#pragma once

#include <binaryninjaapi.h>
#include <objectivec/objc.h>
#include "SharedCache.h"

struct ObjCOptimizationHeader
{
	uint32_t version;
	uint32_t flags;
	uint64_t headerInfoROCacheOffset;
	uint64_t headerInfoRWCacheOffset;
	uint64_t selectorHashTableCacheOffset;
	uint64_t classHashTableCacheOffset;
	uint64_t protocolHashTableCacheOffset;
	uint64_t relativeMethodSelectorBaseAddressOffset;
};

// `objc_opt_t` from dyld/include/objc-shared-cache.h
struct LegacyObjCOptimizationHeader
{
    uint32_t version;
    uint32_t flags;
    int32_t selopt_offset;
    int32_t headeropt_ro_offset;
    int32_t unused_clsopt_offset;
    int32_t unused_protocolopt_offset;
    int32_t headeropt_rw_offset;
    int32_t unused_protocolopt2_offset;
    int32_t largeSharedCachesClassOffset;
    int32_t largeSharedCachesProtocolOffset;
    int64_t relativeMethodSelectorBaseAddressOffset;
};

namespace DSCObjC {
	class SharedCacheObjCReader : public BinaryNinja::ObjCReader
	{
		VirtualMemoryReader m_reader;

	public:
		void Read(void* dest, size_t len) override;

		std::string ReadCString(size_t maxLength = -1) override;

		uint8_t Read8() override;

		uint16_t Read16() override;

		uint32_t Read32() override;

		uint64_t Read64() override;

		int8_t ReadS8() override;

		int16_t ReadS16() override;

		int32_t ReadS32() override;

		int64_t ReadS64() override;

		uint64_t ReadPointer() override;

		uint64_t GetOffset() const override;

		void Seek(uint64_t offset) override;

		void SeekRelative(int64_t offset) override;

		VirtualMemoryReader& GetVMReader();

		SharedCacheObjCReader(VirtualMemoryReader reader);
	};

	class SharedCacheObjCProcessor : public BinaryNinja::ObjCProcessor
	{
		std::optional<uint64_t> m_customRelativeMethodSelectorBase = std::nullopt;
		uint64_t m_imageAddress;

		std::shared_ptr<BinaryNinja::ObjCReader> GetReader() override;

		void GetRelativeMethod(BinaryNinja::ObjCReader* reader, BinaryNinja::method_t& meth,
			bool typesAreOffsetsFromSelectorBase) override;

		BinaryNinja::Ref<BinaryNinja::Symbol> GetSymbol(uint64_t address) override;

		BinaryNinja::Ref<BinaryNinja::Section> GetSectionWithName(const char *sectionName) override;

	public:
		SharedCacheObjCProcessor(BinaryNinja::BinaryView* data, uint64_t imageAddress);

		uint64_t GetObjCRelativeMethodBaseAddress(BinaryNinja::ObjCReader* reader) override;
	};
}  // namespace DSCObjC