PS4 Kernel Defines Maker

PS4_Kernel_Defines_Maker.py

#!/usr/bin/env python

'''

SocraticBliss (R)
    
PS4 Kernel Defines Maker - Creates a non-rebased kernel offsets header file

Inspired by CelesteBlue

* Instructions
IDA -> Load Kernel Database -> File -> Script File -> Pick dis plz

'''

import ida_funcs
import ida_nalt
import idautils

base = ida_nalt.get_imagebase()
print('Kernel Base: %#x' % base)

'''
HEN_INSTALLER_FUNCTIONS = ['xfast_syscall', 'printf', 'memcpy', 'kmem_alloc']
HEN_INSTALLER_VARIABLES = ['PRISON_0', 'ROOTVNODE', 'KERN_MAP']
HEN_INSTALLER_HOOKS = ['sceSblAuthMgrIsLoadable2', 'sceSblAuthMgrVerifySelfHeader']
'''

with open ('%s_defines.h' % ida_nalt.get_root_filename(), 'w') as defines:
    
    for function in idautils.Functions():
        name = ida_funcs.get_func_name(function)
        
        if 'sub_' not in name:
            #print('Function EA: %#x' % function)
            #print('Offset: %#x' % (function - base))
            defines.write('#define %s   %#x\n' % (name.replace('.', '_'), function - base))
    
print('Completed!')
print('Saved to %s_defines.h' % (ida_nalt.get_input_file_path()))