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
|
#!/usr/bin/env python3
from math import ceil as c
def get_offsets(addr):
if addr % 4096:
print("You must align to 4096!")
return 0
if addr > 2 ** 47:
print("Address too big!")
return 0
pte = (addr >> 12) & 0x1ff
pde = (addr >> 21) & 0x1ff
pdpe =(addr >> 30) & 0x1ff
pml4 =(addr >> 39) & 0x1ff
print("pte:\t{}\npde:\t{}\npdpe:\t{}\npml4:\t{}".format(pte, pde, pdpe, pml4))
def get_table_size(addr):
pte_cnt = c(c(addr / (1 << 12)) / 512) * 512
pde_cnt = c(c(addr / (1 << 21)) / 512) * 512
pdpe_cnt = c(c(addr / (1 << 30)) / 512) * 512
pml4_cnt = c(c(addr / (1 << 39)) / 512) * 512
return((pte_cnt + pde_cnt + pdpe_cnt + pml4_cnt) * 8)
def get_palloc_table_size(tb, base, size):
tsize = 24
for x in range(8):
bs = tsize
tsize += c(((size / (0x1000 * (1 << x))) / 64)) * 8
print(tsize - bs)
print("buddy: {} - {} > {}".format(
hex(bs + tb),
hex((tsize + tb) - 1),
(size / (0x1000 * (1 << x))) / 64
))
return(tsize)
# tablebase, base, size
free_chunks = [[0x200000, 0x100000, 3220041728], [0x22fce0, 0x100000000, 1073741824]]
ts = get_palloc_table_size(free_chunks[0][0], free_chunks[0][1], free_chunks[0][2])
print(hex(ts))
|
