SKRoot-linuxKernelRoot/Lite_version/src/patch_kernel_root/patch_kernel_root.h at master · abcz316/SKRoot-linuxKernelRoot · 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
#pragma once
#include <iostream>
#include <filesystem>
#include "analyze/base_func.h"
#include "analyze/aarch64_insn.h"

#define ROOT_KEY_LEN 48
#define FOLDER_HEAD_ROOT_KEY_LEN 16

#define IF_EXIT(cond) do { if (cond) { printf("[ERROR] Patch empty addr!\n"); system("pause"); exit(0); } } while (0)

#define PATCH_AND_CONSUME(region, call_expr) \
    do { \
        size_t __sz = (call_expr); \
        IF_EXIT(!__sz); \
        IF_EXIT(!region.offset); \
        (region).consume(__sz); \
    } while (0)

struct patch_bytes_data {
	std::string str_bytes;
	size_t write_addr = 0;
};

static size_t patch_ret_cmd(const std::vector<char>& file_buf, size_t start, std::vector<patch_bytes_data>& vec_out_patch_bytes_data) {
	if (start == 0) return 0;
	uint32_t insn = *(uint32_t*)&file_buf[start];
	patch_bytes_data blob;
	if (aarch64_insn_is_paciaz(insn) || aarch64_insn_is_paciasp(insn)
		|| aarch64_insn_is_pacibz(insn) || aarch64_insn_is_pacibsp(insn) || aarch64_insn_is_bti(insn)) blob = { "DF2403D5C0035FD6", start };
	else blob = { "C0035FD6", start };
	vec_out_patch_bytes_data.push_back(blob);
	return blob.str_bytes.length() / 2;
}

static size_t patch_ret_1_cmd(const std::vector<char>& file_buf, size_t start, std::vector<patch_bytes_data>& vec_out_patch_bytes_data) {
	if (start == 0) return 0;
	uint32_t insn = *(uint32_t*)&file_buf[start];
	patch_bytes_data blob;
	if (aarch64_insn_is_paciaz(insn) || aarch64_insn_is_paciasp(insn)
		|| aarch64_insn_is_pacibz(insn) || aarch64_insn_is_pacibsp(insn) || aarch64_insn_is_bti(insn)) blob = { "DF2403D5200080D2C0035FD6", start };
	else blob = { "200080D2C0035FD6", start };
	vec_out_patch_bytes_data.push_back(blob);
	return blob.str_bytes.length() / 2;
}

static size_t patch_ret_0_cmd(const std::vector<char>& file_buf, size_t start, std::vector<patch_bytes_data>& vec_out_patch_bytes_data) {
	if (start == 0) return 0;
	uint32_t insn = *(uint32_t*)&file_buf[start];
	patch_bytes_data blob;
	if (aarch64_insn_is_paciaz(insn) || aarch64_insn_is_paciasp(insn)
		|| aarch64_insn_is_pacibz(insn) || aarch64_insn_is_pacibsp(insn) || aarch64_insn_is_bti(insn)) blob = { "DF2403D5E0031F2AC0035FD6", start };
	else blob = { "E0031F2AC0035FD6", start };
	vec_out_patch_bytes_data.push_back(blob);
	return blob.str_bytes.length() / 2;
}

static size_t patch_data(const std::vector<char>& file_buf, size_t start, void* buf, size_t buf_size, std::vector<patch_bytes_data>& vec_out_patch_bytes_data) {
	if (start == 0) return 0;
	std::string str_bytes = bytes2hex((const unsigned char*)buf, buf_size);
	vec_out_patch_bytes_data.push_back({ str_bytes, start });
	return buf_size;
}