#!/usr/bin/env python3
import sys
lines = []
try:
while True:
lines.append(input())
except EOFError:
pass
def remove_comments(text):
return ' '.join(t.split('/*')[0] for t in text.split('*/'))
functions = frozenset(sys.argv[1:]) if len(sys.argv) > 1 else None
lines = remove_comments(' '.join(lines)).replace('\t', ' ')
lines = [line.strip() for line in lines.split(';') if line]
lines = [' '.join(word for word in line.split(' ')[1:] if word) for line in lines
if line.replace('//', '').strip().startswith('asmlinkage ')]
lines = sorted(lines)
def map_type(typ):
orig_typ = typ
if typ.startswith('signed ') and typ != 'signed char':
typ = typ[len('signed '):]
typemap = {
'void' : 'LIBSYSCALLS_TYPE_VOID',
'char' : 'LIBSYSCALLS_TYPE_CHAR',
'signed char' : 'LIBSYSCALLS_TYPE_SCHAR',
'unsigned char' : 'LIBSYSCALLS_TYPE_UCHAR',
'short int' : 'LIBSYSCALLS_TYPE_SHORT',
'unsigned short int' : 'LIBSYSCALLS_TYPE_USHORT',
'int' : 'LIBSYSCALLS_TYPE_INT',
'unsigned int' : 'LIBSYSCALLS_TYPE_UINT',
'long int' : 'LIBSYSCALLS_TYPE_LONG',
'unsigned long int' : 'LIBSYSCALLS_TYPE_ULONG',
'long long int' : 'LIBSYSCALLS_TYPE_LLONG',
'unsigned long long int' : 'LIBSYSCALLS_TYPE_ULLONG',
'size_t' : 'LIBSYSCALLS_TYPE_SIZE',
'ssize_t' : 'LIBSYSCALLS_TYPE_SSIZE',
'ptrdiff_t' : 'LIBSYSCALLS_TYPE_PTRDIFF',
'float' : 'LIBSYSCALLS_TYPE_FLOAT',
'double' : 'LIBSYSCALLS_TYPE_DOUBLE',
'long double' : 'LIBSYSCALLS_TYPE_LDOUBLE',
'int8_t' : 'LIBSYSCALLS_TYPE_INT8',
'uint8_t' : 'LIBSYSCALLS_TYPE_UINT8',
'int16_t' : 'LIBSYSCALLS_TYPE_INT16',
'uint16_t' : 'LIBSYSCALLS_TYPE_UINT16',
'int32_t' : 'LIBSYSCALLS_TYPE_INT32',
'uint32_t' : 'LIBSYSCALLS_TYPE_UINT32',
'int64_t' : 'LIBSYSCALLS_TYPE_INT64',
'uint64_t' : 'LIBSYSCALLS_TYPE_UINT64',
'struct sockaddr' : 'LIBSYSCALLS_TYPE_STRUCT_SOCKADDR',
'struct timex' : 'LIBSYSCALLS_TYPE_STRUCT_TIMEX',
'struct old_timex32' : 'LIBSYSCALLS_TYPE_STRUCT_OLD_TIMEX32',
'struct iovec' : 'LIBSYSCALLS_TYPE_STRUCT_IOVEC_ARRAY',
'struct statfs' : 'LIBSYSCALLS_TYPE_STRUCT_STATFS',
'struct statfs64' : 'LIBSYSCALLS_TYPE_STRUCT_STATFS64',
'struct stat' : 'LIBSYSCALLS_TYPE_STRUCT_STAT',
'struct stat64' : 'LIBSYSCALLS_TYPE_STRUCT_STAT64',
'struct old_stat' : 'LIBSYSCALLS_TYPE_STRUCT_OLD_STAT',
'struct mount_attr' : 'LIBSYSCALLS_TYPE_STRUCT_MOUNT_ATTR',
'struct ustat' : 'LIBSYSCALLS_TYPE_STRUCT_USTAT',
'struct statx' : 'LIBSYSCALLS_TYPE_STRUCT_STATX',
'struct linux_dirent' : 'LIBSYSCALLS_TYPE_STRUCT_LINUX_DIRENT',
'struct linux_dirent64' : 'LIBSYSCALLS_TYPE_STRUCT_LINUX_DIRENT64',
'struct old_linux_dirent' : 'LIBSYSCALLS_TYPE_STRUCT_OLD_LINUX_DIRENT',
'struct open_how' : 'LIBSYSCALLS_TYPE_STRUCT_OPEN_HOW',
'struct iob' : 'LIBSYSCALLS_TYPE_STRUCT_IOB',
'struct io_event' : 'LIBSYSCALLS_TYPE_STRUCT_IO_EVENT',
'struct timespec' : 'LIBSYSCALLS_TYPE_STRUCT_TIMESPEC',
'struct old_timespec32' : 'LIBSYSCALLS_TYPE_STRUCT_OLD_TIMESPEC32',
'struct aio_sigset' : 'LIBSYSCALLS_TYPE_STRUCT_AIO_SIGSET',
'struct io_uring_params' : 'LIBSYSCALLS_TYPE_STRUCT_IO_URING_PARAMS',
'struct epoll_event' : 'LIBSYSCALLS_TYPE_STRUCT_EPOLL_EVENT',
'struct pollfd' : 'LIBSYSCALLS_TYPE_STRUCT_POLLFD',
'struct utimbuf' : 'LIBSYSCALLS_TYPE_STRUCT_UTIMBUF',
'struct old_utimbuf32' : 'LIBSYSCALLS_TYPE_STRUCT_OLD_UTIMBUF32',
'struct old_timeval' : 'LIBSYSCALLS_TYPE_STRUCT_OLD_TIMEVAL',
'struct old_timeval32' : 'LIBSYSCALLS_TYPE_STRUCT_OLD_TIMEVAL32',
'struct sel_arg_struct' : 'LIBSYSCALLS_TYPE_STRUCT_SEL_ARG_STRUCT',
'struct tms' : 'LIBSYSCALLS_TYPE_STRUCT_TMS',
'struct itimerspec' : 'LIBSYSCALLS_TYPE_STRUCT_ITIMERSPEC',
'struct old_itimerspec32' : 'LIBSYSCALLS_TYPE_STRUCT_OLD_ITIMERSPEC32',
'struct siginfo' : 'LIBSYSCALLS_TYPE_STRUCT_SIGINFO',
'struct rusage' : 'LIBSYSCALLS_TYPE_STRUCT_RUSAGE',
'struct robust_list_head' : 'LIBSYSCALLS_TYPE_STRUCT_ROBUST_LIST_HEAD',
'struct futex_waitv' : 'LIBSYSCALLS_TYPE_STRUCT_FUTEX_WAITV',
'struct old_itimerval' : 'LIBSYSCALLS_TYPE_STRUCT_OLD_ITIMERVAL',
'struct kexec_segment' : 'LIBSYSCALLS_TYPE_STRUCT_KEXEC_SEGMENT',
'struct sigevent' : 'LIBSYSCALLS_TYPE_STRUCT_SIGEVENT',
'struct sched_param' : 'LIBSYSCALLS_TYPE_STRUCT_SCHED_PARAM',
'struct sigaltstack' : 'LIBSYSCALLS_TYPE_STRUCT_SIGALTSTACK',
'struct sigaction' : 'LIBSYSCALLS_TYPE_STRUCT_SIGACTION',
'struct old_sigaction' : 'LIBSYSCALLS_TYPE_STRUCT_OLD_SIGACTION',
'struct new_utsname' : 'LIBSYSCALLS_TYPE_STRUCT_NEW_UTSNAME',
'struct old_utsname' : 'LIBSYSCALLS_TYPE_STRUCT_OLD_UTSNAME',
'struct oldold_utsname' : 'LIBSYSCALLS_TYPE_STRUCT_OLDOLD_UTSNAME',
'struct rlimit' : 'LIBSYSCALLS_TYPE_STRUCT_RLIMIT',
'struct rlimit64' : 'LIBSYSCALLS_TYPE_STRUCT_RLIMIT64',
'struct file_handle' : 'LIBSYSCALLS_TYPE_STRUCT_FILE_HANDLE',
'struct timezone' : 'LIBSYSCALLS_TYPE_STRUCT_TIMEZONE',
'struct sysinfo' : 'LIBSYSCALLS_TYPE_STRUCT_SYSINFO',
'struct mq_attr' : 'LIBSYSCALLS_TYPE_STRUCT_MQ_ATTR',
'struct msqid_ds' : 'LIBSYSCALLS_TYPE_STRUCT_MSQID_DS',
'struct msgbuf' : 'LIBSYSCALLS_TYPE_STRUCT_MSGBUF',
'struct semid_ds' : 'LIBSYSCALLS_TYPE_STRUCT_SEMID_DS',
'struct sembuf' : 'LIBSYSCALLS_TYPE_STRUCT_SEMBUF',
'struct shmid_ds' : 'LIBSYSCALLS_TYPE_STRUCT_SHMID_DS',
'struct user_msghdr' : 'LIBSYSCALLS_TYPE_STRUCT_MSGHDR',
'struct mmsghdr' : 'LIBSYSCALLS_TYPE_STRUCT_MMSGHDR',
'struct clone_args' : 'LIBSYSCALLS_TYPE_STRUCT_CLONE_ARGS',
'struct perf_event_attr' : 'LIBSYSCALLS_TYPE_STRUCT_PERF_EVENT_ATTR',
'struct sched_attr' : 'LIBSYSCALLS_TYPE_STRUCT_SCHED_ATTR',
'struct rseq' : 'LIBSYSCALLS_TYPE_STRUCT_RSEQ',
'struct landlock_ruleset_attr' : 'LIBSYSCALLS_TYPE_STRUCT_LANDLOCK_RULESET_ATTR',
'struct cachestat_range' : 'LIBSYSCALLS_TYPE_STRUCT_CACHESTAT_RANGE',
'struct cachestat' : 'LIBSYSCALLS_TYPE_STRUCT_CACHESTAT',
'struct mmap_arg_struct' : 'LIBSYSCALLS_TYPE_STRUCT_MMAP_ARG_STRUCT',
'union bpf_attr' : 'LIBSYSCALLS_TYPE_UNION_BPF_ATTR',
'u8' : 'uint8_t',
'u16' : 'uint16_t',
'u32' : 'uint32_t',
'u64' : 'uint64_t',
's8' : 'int8_t',
's16' : 'int16_t',
's32' : 'int32_t',
's64' : 'int64_t',
'gid_t' : 'gid32_t',
'compat_gid_t' : 'gid32_t',
'gid32_t' : 'u32',
'old_gid_t' : 'unsigned short int',
'uid_t' : 'uid32_t',
'compat_uid_t' : 'uid32_t',
'uid32_t' : 'u32',
'old_uid_t' : 'unsigned short int',
'gid16_t' : 'u16',
'uid16_t' : 'u16',
'pid_t' : 'int',
'umode_t' : 'unsigned short int',
'aio_context_t' : 'unsigned long int',
'sigset_t' : 'unsigned long int',
'loff_t' : 'long long int',
'off_t' : 'long int',
'qid_t' : 'uid32_t',
'clockid_t' : 'int',
'timer_t' : 'int',
'mqd_t' : 'int',
'key_t' : 'int',
'rwf_t' : 'int',
'time_t' : 'long int',
'old_time_t' : 'long int',
'old_time32_t' : 's32',
'time64_t' : 's64', # also defined as `long long int`
'key_serial_t' : 'int32_t',
'old_sigset_t' : 'unsigned long int',
'intptr_t' : 'long int',
'uintptr_t' : 'unsigned long int',
'compat_ulong_t' : 'u32',
'fd_set' : 'LIBSYSCALLS_TYPE_FD_SET'
}
while not typ.startswith('LIBSYSCALLS'):
if typ == '???':
typ = typemap.get(typ, 'LIBSYSCALLS_TYPE_UNKNOWN /* TO'+'DO */')
else:
typ = typemap.get(typ, 'LIBSYSCALLS_TYPE_UNKNOWN /* TO'+'DO (%s) */' % typ)
if typ.startswith('LIBSYSCALLS_TYPE_UNKNOWN') and (orig_typ.split(' ')[0] in ('struct', 'union')):
typ = 'LIBSYSCALLS_TYPE_STRUCT /* TO'+'DO */'
return typ
def get_type(param):
old_param = [p for p in param.replace('*', ' * ').split(' ') if p and p != '__user']
param = []
width = ''
sign = ''
constness = ''
require = None
have_type = False
for p in old_param:
if p == 'volatile':
pass
elif p == 'restrict':
pass
elif p == 'long':
width += 'long '
require = 'int'
elif p == 'short':
width += 'short '
require = 'int'
elif p == 'signed':
sign += 'signed '
require = 'int'
elif p == 'unsigned':
sign += 'unsigned '
require = 'int'
elif p == 'struct':
sign += 'struct '
elif p == 'union':
sign += 'union '
elif p == 'enum':
sign += 'enum '
elif p == 'const':
constness = 'const'
else:
if require and p != require:
param.append(sign + width + require)
have_type = True
if p == '*':
param.append(constness + p)
constness = ''
elif not have_type:
p = sign + width + p
if p.startswith('struct __kernel_'):
p = 'struct ' + p[len('struct __kernel_'):]
elif p.startswith('struct __old_kernel_'):
p = 'struct old_' + p[len('struct __old_kernel_'):]
elif p.startswith('__kernel_'):
p = p[len('__kernel_'):]
elif p.startswith('__old_kernel_'):
p = 'old_' + p[len('__old_kernel_'):]
elif p.startswith('__'):
p = p[len('__'):]
param.append(p)
have_type = True
sign = ''
width = ''
require = None
if require:
param.append(sign + width.strip() + require)
if len(param) == 1:
return (map_type(param[0]), False, False)
if param == ['char', 'const*']:
return ('LIBSYSCALLS_TYPE_STRING', False, False)
if param == ['void', '*']:
return ('LIBSYSCALLS_TYPE_MEMORY_ADDRESS', False, False)
if len(param) == 2 and param[1] == '*':
return (map_type(param[0]), False, True)
if len(param) == 2 and param[1] == 'const*':
return (map_type(param[0]), True, False)
return ('LIBSYSCALLS_TYPE_UNKNOWN /* TO'+'DO */', False, False)
def format_mask(args):
if len(args) == 0:
return '0'
if len(args) == 1:
return '_' + str(args[0] + 1)
return '(%s)' % ' | ' .join('_' + str(arg + 1) for arg in sorted(args))
queued = {}
for line in lines:
line = line.split(')')[0]
[name, params] = line.split('(')
name = name.split(' ')
return_type, name = ' '.join(name[:-1]), name[-1]
while name[0] == '*':
return_type += '*'
name = name[1:]
if functions is not None and name not in functions:
continue
if return_type == 'long':
return_type = '???'
params = [param.strip() for param in params.split(',')]
if params == ['void']:
params = []
param_types = []
in_pointer_mask = []
out_pointer_mask = []
for i, param in enumerate(params):
(typ, is_in, is_out) = get_type(param)
param_types.append(typ)
if is_in:
in_pointer_mask.append(i)
if is_out:
out_pointer_mask.append(i)
code = []
(return_type, return_type_ptr_in, return_type_ptr_out) = get_type(return_type)
if return_type_ptr_in or return_type_ptr_out:
return_type += ' *'
return_type = return_type.replace('* *', '**')
if False:
code.append('\nstatic struct libsyscalls_syscall_abi libsyscalls_linux_syscall_%s = {' % name)
code.append('\t.category = LIBSYSCALLS_CAT_SUPPORT_PENDING /* TO'+'DO */,')
code.append('\t/* TO'+'DO .subcategory = , */')
code.append('\t.min_argument_count = %i,%s' % (len(params), (' /* TO'+'DO */' if len(params) > 1 else '')))
code.append('\t.max_argument_count = %i,' % len(params))
code.append('\t.in_pointer_mask = %s,' % format_mask(in_pointer_mask))
code.append('\t.out_pointer_mask = %s,' % format_mask(out_pointer_mask))
code.append('\t.symbolic_mask = 0, /* TO'+'DO */')
code.append('\t.symbolic_return = 0,')
code.append('\t.expect_zero = 1, /* TO'+'DO */')
code.append('\t.symbol_printer = ~0, /* TO'+'DO */')
code.append('\t.return_type = %s%s' % (return_type, (',' if len(params) else '')))
if len(params):
code.append('\t.parameters_types = {')
for i, param_type in enumerate(param_types):
code.append('\t\t' + param_type + ('' if i + 1 == len(param_types) else ','))
code.append('\t}')
code.append('};')
else:
paramstr = ''
for i, param_type in enumerate(param_types):
if i:
paramstr += ', '
paramstr += {(False, False) : '_',
(True, False) : 'IN',
(False, True) : 'OUT',
(True, True) : 'BI'}[(i in in_pointer_mask), (i in out_pointer_mask)]
paramstr += ',' + param_type.replace('LIBSYSCALLS_TYPE_', '')
code.append('\nSYS(%s, /*CATEGORY*/, /*SUBCATEGORY*/, %i, ZERO(%s),' %
(name, len(params), return_type.replace('LIBSYSCALLS_TYPE_', '')))
code.append('\tPARAMS(%s),' % paramstr)
code.append('\t.symbolic_return = 0,')
code.append('\t.symbol_printer = ~0')
code.append(');')
code = '\n'.join(code)
if functions is None:
print(code)
else:
queued[name] = code
not_found = []
for name in sys.argv[1:]:
if name in queued:
print(queued[name])
else:
not_found.append(name)
if not_found:
for name in not_found:
print('Could not find ' + name, file = sys.stderr)
sys.exit(1)
