#pragma once

#include <cia-def.h>

// For now these definitions will live here
#if !defined(STDIN_FILENO)
    #define STDIN_FILENO  0
    #define STDOUT_FILENO 1
    #define STDERR_FILENO 2
#endif

#define _SYSCALL0(n)                   _syscall0(n)
#define _SYSCALL1(n,p1)                _syscall1(n, (i64)p1)
#define _SYSCALL2(n,p1,p2)             _syscall2(n, (i64)p1, (i64)p2)
#define _SYSCALL3(n,p1,p2,p3)          _syscall3(n, (i64)p1, (i64)p2, (i64)p3)
#define _SYSCALL4(n,p1,p2,p3,p4)       _syscall4(n, (i64)p1, (i64)p2, (i64)p3, (i64)p4)
#define _SYSCALL5(n,p1,p2,p3,p4,p5)    _syscall5(n, (i64)p1, (i64)p2, (i64)p3, (i64)p4, (i64)p5)
#define _SYSCALL6(n,p1,p2,p3,p4,p5,p6) _syscall6(n, (i64)p1, (i64)p2, (i64)p3, (i64)p4, (i64)p5, (i64)p6)

#define _SYSCALL_NARGS(...) _SYSCALL_NARGS_(0, ## __VA_ARGS__, 6, 5, 4, 3, 2, 1, 0)
#define _SYSCALL_NARGS_(_0, _1_, _2_, _3_, _4_, _5_, _6_, count, ...) count
#define _SYSCALL_CONCAT(a, b) _SYSCALL_CONCAT_(a,b)
#define _SYSCALL_CONCAT_(a,b) a ## b

// Creates an expression of the sort `_SYSCALL<number of params>(<params>)`
#define syscall(n, ...) _SYSCALL_CONCAT(_SYSCALL, _SYSCALL_NARGS(__VA_ARGS__))(n, ## __VA_ARGS__)

static inline i64 _syscall0(i64 n) {
    i64 ret = 0;
    asm volatile("syscall" : "=a"(ret) : "a"(n) : "rcx", "r11", "memory");
    return ret;
}

static inline i64 _syscall1(i64 n, i64 a1) {
    i64 ret = 0;
    asm volatile("syscall" : "=a"(ret) : "a"(n), "D"(a1) : "rcx", "r11", "memory");
    return ret;
}

static inline i64 _syscall2(i64 n, i64 a1, i64 a2) {
    i64 ret = 0;
    asm volatile("syscall" : "=a"(ret) : "a"(n), "D"(a1), "S"(a2)
                          : "rcx", "r11", "memory");
    return ret;
}

static inline i64 _syscall3(i64 n, i64 a1, i64 a2, i64 a3) {
    i64 ret = 0;
    asm volatile("syscall" : "=a"(ret) : "a"(n), "D"(a1), "S"(a2),
                          "d"(a3) : "rcx", "r11", "memory");
    return ret;
}

static inline i64 _syscall4(i64 n, i64 a1, i64 a2, i64 a3, i64 a4) {
    i64 ret = 0;
    register i64 r10 asm("r10") = a4;
    asm volatile("syscall" : "=a"(ret) : "a"(n), "D"(a1), "S"(a2),
                          "d"(a3), "r"(r10): "rcx", "r11", "memory");
    return ret;
}

static inline i64 _syscall5(i64 n, i64 a1, i64 a2, i64 a3, i64 a4, i64 a5) {
    i64 ret = 0;
    register i64 r10 asm("r10") = a4;
    register i64 r8 asm("r8") = a5;
    asm volatile("syscall" : "=a"(ret) : "a"(n), "D"(a1), "S"(a2),
                          "d"(a3), "r"(r10), "r"(r8) : "rcx", "r11", "memory");
    return ret;
}

static inline i64 _syscall6(i64 n, i64 a1, i64 a2, i64 a3, i64 a4, i64 a5, i64 a6) {
    i64 ret = 0;
    register i64 r10 asm("r10") = a4;
    register i64 r8 asm("r8") = a5;
    register i64 r9 asm("r9") = a6;
    asm volatile("syscall" : "=a"(ret) : "a"(n), "D"(a1), "S"(a2),
                          "d"(a3), "r"(r10), "r"(r8), "r"(r9) : "rcx", "r11", "memory");
    return ret;
}

// Syscall numbers
#define SYS_read                   0
#define SYS_write                  1
#define SYS_open                   2
#define SYS_close                  3
#define SYS_stat                   4
#define SYS_fstat                  5
#define SYS_lstat                  6
#define SYS_poll                   7
#define SYS_lseek                  8
#define SYS_mmap                   9
#define SYS_mprotect               10
#define SYS_munmap                 11
#define SYS_brk                    12
#define SYS_rt_sigaction           13
#define SYS_rt_sigprocmask         14
#define SYS_rt_sigreturn           15
#define SYS_ioctl                  16
#define SYS_pread64                17
#define SYS_pwrite64               18
#define SYS_readv                  19
#define SYS_writev                 20
#define SYS_access                 21
#define SYS_pipe                   22
#define SYS_select                 23
#define SYS_sched_yield            24
#define SYS_mremap                 25
#define SYS_msync                  26
#define SYS_mincore                27
#define SYS_madvise                28
#define SYS_shmget                 29
#define SYS_shmat                  30
#define SYS_shmctl                 31
#define SYS_dup                    32
#define SYS_dup2                   33
#define SYS_pause                  34
#define SYS_nanosleep              35
#define SYS_getitimer              36
#define SYS_alarm                  37
#define SYS_setitimer              38
#define SYS_getpid                 39
#define SYS_sendfile               40
#define SYS_socket                 41
#define SYS_connect                42
#define SYS_accept                 43
#define SYS_sendto                 44
#define SYS_recvfrom               45
#define SYS_sendmsg                46
#define SYS_recvmsg                47
#define SYS_shutdown               48
#define SYS_bind                   49
#define SYS_listen                 50
#define SYS_getsockname            51
#define SYS_getpeername            52
#define SYS_socketpair             53
#define SYS_setsockopt             54
#define SYS_getsockopt             55
#define SYS_clone                  56
#define SYS_fork                   57
#define SYS_vfork                  58
#define SYS_execve                 59
#define SYS_exit                   60
#define SYS_wait4                  61
#define SYS_kill                   62
#define SYS_uname                  63
#define SYS_semget                 64
#define SYS_semop                  65
#define SYS_semctl                 66
#define SYS_shmdt                  67
#define SYS_msgget                 68
#define SYS_msgsnd                 69
#define SYS_msgrcv                 70
#define SYS_msgctl                 71
#define SYS_fcntl                  72
#define SYS_flock                  73
#define SYS_fsync                  74
#define SYS_fdatasync              75
#define SYS_truncate               76
#define SYS_ftruncate              77
#define SYS_getdents               78
#define SYS_getcwd                 79
#define SYS_chdir                  80
#define SYS_fchdir                 81
#define SYS_rename                 82
#define SYS_mkdir                  83
#define SYS_rmdir                  84
#define SYS_creat                  85
#define SYS_link                   86
#define SYS_unlink                 87
#define SYS_symlink                88
#define SYS_readlink               89
#define SYS_chmod                  90
#define SYS_fchmod                 91
#define SYS_chown                  92
#define SYS_fchown                 93
#define SYS_lchown                 94
#define SYS_umask                  95
#define SYS_gettimeofday           96
#define SYS_getrlimit              97
#define SYS_getrusage              98
#define SYS_sysinfo                99
#define SYS_times                  100
#define SYS_ptrace                 101
#define SYS_getuid                 102
#define SYS_syslog                 103
#define SYS_getgid                 104
#define SYS_setuid                 105
#define SYS_setgid                 106
#define SYS_geteuid                107
#define SYS_getegid                108
#define SYS_setpgid                109
#define SYS_getppid                110
#define SYS_getpgrp                111
#define SYS_setsid                 112
#define SYS_setreuid               113
#define SYS_setregid               114
#define SYS_getgroups              115
#define SYS_setgroups              116
#define SYS_setresuid              117
#define SYS_getresuid              118
#define SYS_setresgid              119
#define SYS_getresgid              120
#define SYS_getpgid                121
#define SYS_setfsuid               122
#define SYS_setfsgid               123
#define SYS_getsid                 124
#define SYS_capget                 125
#define SYS_capset                 126
#define SYS_rt_sigpending          127
#define SYS_rt_sigtimedwait        128
#define SYS_rt_sigqueueinfo        129
#define SYS_rt_sigsuspend          130
#define SYS_sigaltstack            131
#define SYS_utime                  132
#define SYS_mknod                  133
#define SYS_uselib                 134
#define SYS_personality            135
#define SYS_ustat                  136
#define SYS_statfs                 137
#define SYS_fstatfs                138
#define SYS_sysfs                  139
#define SYS_getpriority            140
#define SYS_setpriority            141
#define SYS_sched_setparam         142
#define SYS_sched_getparam         143
#define SYS_sched_setscheduler     144
#define SYS_sched_getscheduler     145
#define SYS_sched_get_priority_max 146
#define SYS_sched_get_priority_min 147
#define SYS_sched_rr_get_interval  148
#define SYS_mlock                  149
#define SYS_munlock                150
#define SYS_mlockall               151
#define SYS_munlockall             152
#define SYS_vhangup                153
#define SYS_modify_ldt             154
#define SYS_pivot_root             155
#define SYS__sysctl                156
#define SYS_prctl                  157
#define SYS_arch_prctl             158
#define SYS_adjtimex               159
#define SYS_setrlimit              160
#define SYS_chroot                 161
#define SYS_sync                   162
#define SYS_acct                   163
#define SYS_settimeofday           164
#define SYS_mount                  165
#define SYS_umount2                166
#define SYS_swapon                 167
#define SYS_swapoff                168
#define SYS_reboot                 169
#define SYS_sethostname            170
#define SYS_setdomainname          171
#define SYS_iopl                   172
#define SYS_ioperm                 173
#define SYS_create_module          174
#define SYS_init_module            175
#define SYS_delete_module          176
#define SYS_get_kernel_syms        177
#define SYS_query_module           178
#define SYS_quotactl               179
#define SYS_nfsservctl             180
#define SYS_getpmsg                181
#define SYS_putpmsg                182
#define SYS_afs_syscall            183
#define SYS_tuxcall                184
#define SYS_security               185
#define SYS_gettid                 186
#define SYS_readahead              187
#define SYS_setxattr               188
#define SYS_lsetxattr              189
#define SYS_fsetxattr              190
#define SYS_getxattr               191
#define SYS_lgetxattr              192
#define SYS_fgetxattr              193
#define SYS_listxattr              194
#define SYS_llistxattr             195
#define SYS_flistxattr             196
#define SYS_removexattr            197
#define SYS_lremovexattr           198
#define SYS_fremovexattr           199
#define SYS_tkill                  200
#define SYS_time                   201
#define SYS_futex                  202
#define SYS_sched_setaffinity      203
#define SYS_sched_getaffinity      204
#define SYS_set_thread_area        205
#define SYS_io_setup               206
#define SYS_io_destroy             207
#define SYS_io_getevents           208
#define SYS_io_submit              209
#define SYS_io_cancel              210
#define SYS_get_thread_area        211
#define SYS_lookup_dcookie         212
#define SYS_epoll_create           213
#define SYS_epoll_ctl_old          214
#define SYS_epoll_wait_old         215
#define SYS_remap_file_pages       216
#define SYS_getdents64             217
#define SYS_set_tid_address        218
#define SYS_restart_syscall        219
#define SYS_semtimedop             220
#define SYS_fadvise64              221
#define SYS_timer_create           222
#define SYS_timer_settime          223
#define SYS_timer_gettime          224
#define SYS_timer_getoverrun       225
#define SYS_timer_delete           226
#define SYS_clock_settime          227
#define SYS_clock_gettime          228
#define SYS_clock_getres           229
#define SYS_clock_nanosleep        230
#define SYS_exit_group             231
#define SYS_epoll_wait             232
#define SYS_epoll_ctl              233
#define SYS_tgkill                 234
#define SYS_utimes                 235
#define SYS_vserver                236
#define SYS_mbind                  237
#define SYS_set_mempolicy          238
#define SYS_get_mempolicy          239
#define SYS_mq_open                240
#define SYS_mq_unlink              241
#define SYS_mq_timedsend           242
#define SYS_mq_timedreceive        243
#define SYS_mq_notify              244
#define SYS_mq_getsetattr          245
#define SYS_kexec_load             246
#define SYS_waitid                 247
#define SYS_add_key                248
#define SYS_request_key            249
#define SYS_keyctl                 250
#define SYS_ioprio_set             251
#define SYS_ioprio_get             252
#define SYS_inotify_init           253
#define SYS_inotify_add_watch      254
#define SYS_inotify_rm_watch       255
#define SYS_migrate_pages          256
#define SYS_openat                 257
#define SYS_mkdirat                258
#define SYS_mknodat                259
#define SYS_fchownat               260
#define SYS_futimesat              261
#define SYS_newfstatat             262
#define SYS_unlinkat               263
#define SYS_renameat               264
#define SYS_linkat                 265
#define SYS_symlinkat              266
#define SYS_readlinkat             267
#define SYS_fchmodat               268
#define SYS_faccessat              269
#define SYS_pselect6               270
#define SYS_ppoll                  271
#define SYS_unshare                272
#define SYS_set_robust_list        273
#define SYS_get_robust_list        274
#define SYS_splice                 275
#define SYS_tee                    276
#define SYS_sync_file_range        277
#define SYS_vmsplice               278
#define SYS_move_pages             279
#define SYS_utimensat              280
#define SYS_epoll_pwait            281
#define SYS_signalfd               282
#define SYS_timerfd_create         283
#define SYS_eventfd                284
#define SYS_fallocate              285
#define SYS_timerfd_settime        286
#define SYS_timerfd_gettime        287
#define SYS_accept4                288
#define SYS_signalfd4              289
#define SYS_eventfd2               290
#define SYS_epoll_create1          291
#define SYS_dup3                   292
#define SYS_pipe2                  293
#define SYS_inotify_init1          294
#define SYS_preadv                 295
#define SYS_pwritev                296
#define SYS_rt_tgsigqueueinfo      297
#define SYS_perf_event_open        298

static inline i64 sys_read(u32 fd, char *buf, u64 count) {
    return syscall(SYS_read, fd, buf, count);
}

static inline i64 sys_write(u32 fd, char const *buf, u64 count) {
    return syscall(SYS_write, fd, buf, count);
}

static inline i64 sys_open(char const *filename, int flags, int mode) {
    return syscall(SYS_open, filename, flags, mode);
}

static inline i64 sys_close(u32 fd) {
    return syscall(SYS_close, fd);
}

static inline void *sys_mmap(u64 addr, u64 len, u64 prot, u64 flags, u64 fd, u64 offset) {
    return (void *)syscall(SYS_mmap, addr, len, prot, flags, fd, offset);
}

static inline i64 sys_munmap(void *addr, u64 len) {
    return syscall(SYS_munmap, addr, len);
}

static inline i64 sys_clone(u64 flags, void *stack, int *parent_tid, int *child_tid, u64 tls) {
    return syscall(SYS_clone, flags, stack, parent_tid, child_tid, tls);
}

_Noreturn static inline void sys_exit(int code) {
    syscall(SYS_exit, code);
    __builtin_unreachable();
}

static inline i64 sys_arch_prctl_set(int code, u64 value) {
    return syscall(SYS_arch_prctl, code, value);
}

static inline i64 sys_arch_prctl_get(int code, u64 *value) {
    return syscall(SYS_arch_prctl, code, value);
}

static inline i64 sys_gettid() {
    return syscall(SYS_gettid);
}