make unistd file location search dependent on platform

Author: tanelpoder

lib/0xtools/psnproc.py

@@ -322,7 +322,13 @@ def extract_system_call_ids(unistd_64_fh):
 def get_system_call_names():
     psn_dir=os.path.dirname(os.path.realpath(__file__))
     kernel_ver=platform.release().split('-')[0]
-    unistd_64_paths = ['/usr/include/asm-generic/unistd.h', '/usr/include/asm/unistd_64.h', '/usr/include/x86_64-linux-gnu/asm/unistd_64.h', '/usr/include/asm-x86_64/unistd.h', '/usr/include/asm/unistd.h', psn_dir+'/syscall_64_'+kernel_ver+'.h', psn_dir+'/syscall_64.h']
+
+    # this probably needds to be improved for better platform support
+    if platform.machine() == 'aarch64':
+        unistd_64_paths = ['/usr/include/asm-generic/unistd.h']
+    else:
+        unistd_64_paths = ['/usr/include/asm/unistd_64.h', '/usr/include/x86_64-linux-gnu/asm/unistd_64.h', '/usr/include/asm-x86_64/unistd.h', '/usr/include/asm/unistd.h', psn_dir+'/syscall_64_'+kernel_ver+'.h', psn_dir+'/syscall_64.h']
+
     for path in unistd_64_paths:
         try:
             with open(path) as f: