PMPatch/patch.h at master · AppleLife/PMPatch · 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
#ifndef __PATCH_H__
#define __PATCH_H__

#include <stdio.h>
#include <stdlib.h>
#include "EDK2/Compress.h"
#include "EDK2/Decompress.h"

/* Module UUIDs */
extern const UINT8 PWRMGMT_UUID[];
extern const UINT8 CPUPEI_UUID[];
extern const UINT8 VOLUME_TOP_UUID[];

/* Patch strings */
extern const UINT8 PWRMGMT_PATCH_STRING[];
extern const UINT8 CPUPEI_PATCH_STRING[];
extern const UINT8 PWRMGMT_PATCHED_STRING[];
extern const UINT8 CPUPEI_PATCHED_STRING[];

/* Data offsets and sizes*/
/* Common */
#define MODULE_UUID_LENGTH                16
#define MODULE_STATE_OFFSET               23
#define MODULE_SIZE_OFFSET                20
#define MODULE_HEADER_CHECKSUM_OFFSET     16
#define MODULE_DATA_CHECKSUM_OFFSET       17
#define MODULE_DATA_CHECKSUM_START        24
/* PowerManagement */
#define PWRMGMT_COMPRESSED_SIZE_OFFSET    172
#define PWRMGMT_COMPRESSED_DATA_OFFSET    9
#define PWRMGMT_DATA_OFFSET               181

/* Error messages array for PowerManagement module patcher - index of that array is an error code */
extern const UINT8* PATCH_PWRMGMT_ERROR_MESSAGES[];
/* Error messages array for CpuPei module patcher - index of that array is an error code */
extern const UINT8* PATCH_CPUPEI_ERROR_MESSAGES[];

/* Finds pattern in string */
/* Returns pointer to the first symbol of found pattern, or NULL if not found */
UINT8* find_pattern(UINT8* string, UINT32 slen, const UINT8* pattern, UINT32 plen);

/* Calculates 2's complement 8-bit checksum of data from data[0] to data[length-1] and stores it to *checksum */
/* Returns 1 on success or 0 on error */
int calculate_checksum(UINT8* data, UINT32 length, UINT8* checksum);

/* Converts UINT32 to 3 bytes in reversed order. */
/* Returns 1 on success or 0 on error */
int int2size(UINT32 size, UINT8* module_size);

/* Converts 3 bytes in reversed order to UINT32. */
/* Returns 1 on success or 0 on error */
int size2int(UINT8* module_size, UINT32* size);

/* Patches PowerManagement module */
/* Returns 1 on success or 0 on error and error_code is set to non-zero */
int patch_powermanagement_module(UINT8* module, UINT8* error_code);

/* Patches CpuPei module */
/* Returns 1 on success or 0 on error and error_code is set to non-zero */
int patch_cpupei_module(UINT8* module, UINT8* error_code);

#endif /* __PATCH_H__ */