/*-
 * SPDX-License-Identifier: BSD-2-Clause
 *
 * Copyright (c) 2022 Dmitry Chagin <dchagin@FreeBSD.org>
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include <sys/param.h>
#include <sys/uio.h>
#include <sys/ktrace.h>
#include <err.h>
#include <errno.h>
#include <stddef.h>
#include <stdlib.h>
#include <string.h>
#include <sysdecode.h>

#include "kdump.h"

#ifdef __amd64__
#include <amd64/linux/linux.h>
#include <amd64/linux32/linux32_syscall.h>
#elif __aarch64__
#include <arm64/linux/linux.h>
#elif __i386__
#include <i386/linux/linux.h>
#endif

#include <compat/linux/linux.h>
#include <compat/linux/linux_file.h>

static void
print_linux_signal(int signo)
{
        const char *signame;

        signame = sysdecode_linux_signal(signo);
        if (signame != NULL)
                printf("%s", signame);
        else
                printf("SIG %d", signo);
}

void
ktrsyscall_linux(struct ktr_syscall *ktr, register_t **resip,
    int *resnarg, char *resc)
{
        int narg = ktr->ktr_narg;
        register_t *ip, *first;
        int quad_align, quad_slots;
        char c;

        ip = first = &ktr->ktr_args[0];
        c = *resc;
        quad_align = 0;
        quad_slots = 1;
        switch (ktr->ktr_code) {
        case LINUX_SYS_linux_faccessat:
        case LINUX_SYS_linux_fchmodat:
        case LINUX_SYS_linux_fchownat:
#ifdef LINUX_SYS_linux_newfstatat
        case LINUX_SYS_linux_newfstatat:
#endif
#ifdef LINUX_SYS_linux_fstatat64
        case LINUX_SYS_linux_fstatat64:
#endif
#ifdef LINUX_SYS_linux_futimesat
        case LINUX_SYS_linux_futimesat:
#endif
        case LINUX_SYS_linux_linkat:
        case LINUX_SYS_linux_mkdirat:
        case LINUX_SYS_linux_mknodat:
        case LINUX_SYS_linux_openat:
        case LINUX_SYS_linux_readlinkat:
        case LINUX_SYS_linux_renameat:
        case LINUX_SYS_linux_unlinkat:
        case LINUX_SYS_linux_utimensat:
                putchar('(');
                print_integer_arg_valid(sysdecode_atfd, *ip);
                c = ',';
                ip++;
                narg--;
                break;
        }
        switch (ktr->ktr_code) {
#ifdef LINUX_SYS_linux_access
        case LINUX_SYS_linux_access:
#endif
        case LINUX_SYS_linux_faccessat:
                print_number(ip, narg, c);
                putchar(',');
                print_mask_arg(sysdecode_access_mode, *ip);
                ip++;
                narg--;
                break;
#ifdef LINUX_SYS_linux_chmod
        case LINUX_SYS_linux_chmod:
#endif
        case LINUX_SYS_linux_fchmodat:
                print_number(ip, narg, c);
                putchar(',');
                decode_filemode(*ip);
                ip++;
                narg--;
                break;
        case LINUX_SYS_linux_mknodat:
                print_number(ip, narg, c);
                putchar(',');
                decode_filemode(*ip);
                ip++;
                narg--;
                break;
#ifdef LINUX_SYS_linux_mkdir
        case LINUX_SYS_linux_mkdir:
#endif
        case LINUX_SYS_linux_mkdirat:
                print_number(ip, narg, c);
                putchar(',');
                decode_filemode(*ip);
                ip++;
                narg--;
                break;
        case LINUX_SYS_linux_linkat:
        case LINUX_SYS_linux_renameat:
        case LINUX_SYS_linux_symlinkat:
                print_number(ip, narg, c);
                putchar(',');
                print_integer_arg_valid(sysdecode_atfd, *ip);
                ip++;
                narg--;
                print_number(ip, narg, c);
                break;
        case LINUX_SYS_linux_fchownat:
                print_number(ip, narg, c);
                print_number(ip, narg, c);
                print_number(ip, narg, c);
                break;
#ifdef LINUX_SYS_linux_newfstatat
        case LINUX_SYS_linux_newfstatat:
#endif
#ifdef LINUX_SYS_linux_fstatat64
        case LINUX_SYS_linux_fstatat64:
#endif
        case LINUX_SYS_linux_utimensat:
                print_number(ip, narg, c);
                print_number(ip, narg, c);
                break;
        case LINUX_SYS_linux_unlinkat:
                print_number(ip, narg, c);
                break;
        case LINUX_SYS_linux_clock_gettime:
        case LINUX_SYS_linux_clock_settime:
        case LINUX_SYS_linux_clock_getres:
        case LINUX_SYS_linux_timer_create:
                putchar('(');
                sysdecode_linux_clockid(stdout, *ip);
                c = ',';
                ip++;
                narg--;
                break;
        case LINUX_SYS_linux_clock_nanosleep:
                putchar('(');
                sysdecode_linux_clockid(stdout, *ip);
                putchar(',');
                ip++;
                narg--;
                print_mask_arg0(sysdecode_linux_clock_flags, *ip);
                c = ',';
                ip++;
                narg--;
                break;
        case LINUX_SYS_linux_clone:
                putchar('(');
                print_mask_arg(sysdecode_linux_clone_flags, *ip);
                ip++;
                narg--;
                c = ',';
                break;
        case LINUX_SYS_linux_kill:
        case LINUX_SYS_linux_tkill:
        case LINUX_SYS_linux_rt_sigqueueinfo:
                print_decimal_number(ip, narg, c);
                putchar(',');
                print_linux_signal(*ip);
                ip++;
                narg--;
                break;
        case LINUX_SYS_linux_tgkill:
        case LINUX_SYS_linux_rt_tgsigqueueinfo:
                print_decimal_number(ip, narg, c);
                print_decimal_number(ip, narg, c);
                putchar(',');
                print_linux_signal(*ip);
                ip++;
                narg--;
                break;
#ifdef LINUX_SYS_linux_open
        case LINUX_SYS_linux_open:
#endif
        case LINUX_SYS_linux_openat:
                print_number(ip, narg, c);
                putchar(',');
                print_mask_arg(sysdecode_linux_open_flags, ip[0]);
                if ((ip[0] & LINUX_O_CREAT) == LINUX_O_CREAT) {
                        putchar(',');
                        decode_filemode(ip[1]);
                }
                ip += 2;
                narg -= 2;
                break;
        case LINUX_SYS_linux_rt_sigaction:
                putchar('(');
                print_linux_signal(*ip);
                ip++;
                narg--;
                c = ',';
                break;
        case LINUX_SYS_linux_ftruncate:
        case LINUX_SYS_linux_truncate:
                print_number(ip, narg, c);
                print_number64(first, ip, narg, c);
                break;
        case LINUX_SYS_linux_getitimer:
        case LINUX_SYS_linux_setitimer:
                putchar('(');
                print_integer_arg(sysdecode_itimer, *ip);
                ip++;
                narg--;
                c = ',';
                break;
        case LINUX_SYS_linux_rt_sigprocmask:
#ifdef LINUX_SYS_linux_sigprocmask
        case LINUX_SYS_linux_sigprocmask:
#endif
                putchar('(');
                print_integer_arg(sysdecode_linux_sigprocmask_how, *ip);
                ip++;
                narg--;
                c = ',';
                break;
        }
        switch (ktr->ktr_code) {
        case LINUX_SYS_linux_fchownat:
        case LINUX_SYS_linux_faccessat:
        case LINUX_SYS_linux_fchmodat:
#ifdef LINUX_SYS_linux_newfstatat
        case LINUX_SYS_linux_newfstatat:
#endif
#ifdef LINUX_SYS_linux_fstatat64
        case LINUX_SYS_linux_fstatat64:
#endif
        case LINUX_SYS_linux_linkat:
        case LINUX_SYS_linux_unlinkat:
        case LINUX_SYS_linux_utimensat:
                putchar(',');
                print_mask_arg0(sysdecode_linux_atflags, *ip);
                ip++;
                narg--;
                break;
        }
        *resc = c;
        *resip = ip;
        *resnarg = narg;
}

#if defined(__amd64__)
void
ktrsyscall_linux32(struct ktr_syscall *ktr, register_t **resip,
    int *resnarg, char *resc)
{
        int narg = ktr->ktr_narg;
        register_t *ip, *first;
        int quad_align, quad_slots;
        char c;

        ip = first = &ktr->ktr_args[0];
        c = *resc;
        quad_align = 0;
        quad_slots = 2;
        switch (ktr->ktr_code) {
        case LINUX32_SYS_linux_faccessat:
        case LINUX32_SYS_linux_fchmodat:
        case LINUX32_SYS_linux_fchownat:
        case LINUX32_SYS_linux_fstatat64:
        case LINUX32_SYS_linux_futimesat:
        case LINUX32_SYS_linux_linkat:
        case LINUX32_SYS_linux_mkdirat:
        case LINUX32_SYS_linux_mknodat:
        case LINUX32_SYS_linux_openat:
        case LINUX32_SYS_linux_readlinkat:
        case LINUX32_SYS_linux_renameat:
        case LINUX32_SYS_linux_unlinkat:
        case LINUX32_SYS_linux_utimensat:
                putchar('(');
                print_integer_arg_valid(sysdecode_atfd, *ip);
                c = ',';
                ip++;
                narg--;
                break;
        }
        switch (ktr->ktr_code) {
        case LINUX32_SYS_linux_access:
        case LINUX32_SYS_linux_faccessat:
                print_number(ip, narg, c);
                putchar(',');
                print_mask_arg(sysdecode_access_mode, *ip);
                ip++;
                narg--;
                break;
        case LINUX32_SYS_linux_chmod:
        case LINUX32_SYS_fchmod:
        case LINUX32_SYS_linux_fchmodat:
                print_number(ip, narg, c);
                putchar(',');
                decode_filemode(*ip);
                ip++;
                narg--;
                break;
        case LINUX32_SYS_linux_mknodat:
                print_number(ip, narg, c);
                putchar(',');
                decode_filemode(*ip);
                ip++;
                narg--;
                break;
        case LINUX32_SYS_linux_mkdir:
        case LINUX32_SYS_linux_mkdirat:
                print_number(ip, narg, c);
                putchar(',');
                decode_filemode(*ip);
                ip++;
                narg--;
                break;
        case LINUX32_SYS_linux_linkat:
        case LINUX32_SYS_linux_renameat:
        case LINUX32_SYS_linux_symlinkat:
                print_number(ip, narg, c);
                putchar(',');
                print_integer_arg_valid(sysdecode_atfd, *ip);
                ip++;
                narg--;
                print_number(ip, narg, c);
                break;
        case LINUX32_SYS_linux_fchownat:
                print_number(ip, narg, c);
                print_number(ip, narg, c);
                print_number(ip, narg, c);
                break;
        case LINUX32_SYS_linux_fstatat64:
        case LINUX32_SYS_linux_utimensat:
                print_number(ip, narg, c);
                print_number(ip, narg, c);
                break;
        case LINUX32_SYS_linux_unlinkat:
                print_number(ip, narg, c);
                break;
        case LINUX32_SYS_linux_clock_gettime:
        case LINUX32_SYS_linux_clock_settime:
        case LINUX32_SYS_linux_clock_getres:
        case LINUX32_SYS_linux_timer_create:
        case LINUX32_SYS_linux_clock_gettime64:
        case LINUX32_SYS_linux_clock_settime64:
        case LINUX32_SYS_linux_clock_getres_time64:
                putchar('(');
                sysdecode_linux_clockid(stdout, *ip);
                c = ',';
                ip++;
                narg--;
                break;
        case LINUX32_SYS_linux_clock_nanosleep:
                putchar('(');
                sysdecode_linux_clockid(stdout, *ip);
                putchar(',');
                ip++;
                narg--;
                print_mask_arg0(sysdecode_linux_clock_flags, *ip);
                c = ',';
                ip++;
                narg--;
                break;
        case LINUX32_SYS_linux_clone:
                putchar('(');
                print_mask_arg(sysdecode_linux_clone_flags, *ip);
                ip++;
                narg--;
                c = ',';
                break;
        case LINUX32_SYS_linux_kill:
        case LINUX32_SYS_linux_tkill:
        case LINUX32_SYS_linux_rt_sigqueueinfo:
                print_decimal_number(ip, narg, c);
                putchar(',');
                print_linux_signal(*ip);
                ip++;
                narg--;
                break;
        case LINUX32_SYS_linux_tgkill:
        case LINUX32_SYS_linux_rt_tgsigqueueinfo:
                print_decimal_number(ip, narg, c);
                print_decimal_number(ip, narg, c);
                putchar(',');
                print_linux_signal(*ip);
                ip++;
                narg--;
                break;
        case LINUX32_SYS_linux_open:
        case LINUX32_SYS_linux_openat:
                print_number(ip, narg, c);
                putchar(',');
                print_mask_arg(sysdecode_linux_open_flags, ip[0]);
                if ((ip[0] & LINUX_O_CREAT) == LINUX_O_CREAT) {
                        putchar(',');
                        decode_filemode(ip[1]);
                }
                ip += 2;
                narg -= 2;
                break;
        case LINUX32_SYS_linux_signal:
        case LINUX32_SYS_linux_sigaction:
        case LINUX32_SYS_linux_rt_sigaction:
                putchar('(');
                print_linux_signal(*ip);
                ip++;
                narg--;
                c = ',';
                break;
        case LINUX32_SYS_linux_ftruncate:
        case LINUX32_SYS_linux_truncate:
                print_number(ip, narg, c);
                print_number64(first, ip, narg, c);
                break;
        case LINUX32_SYS_linux_getitimer:
        case LINUX32_SYS_linux_setitimer:
                putchar('(');
                print_integer_arg(sysdecode_itimer, *ip);
                ip++;
                narg--;
                c = ',';
                break;
        case LINUX32_SYS_linux_rt_sigprocmask:
        case LINUX32_SYS_linux_sigprocmask:
                putchar('(');
                print_integer_arg(sysdecode_linux_sigprocmask_how, *ip);
                ip++;
                narg--;
                c = ',';
                break;
        }
        switch (ktr->ktr_code) {
        case LINUX32_SYS_linux_fchownat:
        case LINUX32_SYS_linux_faccessat:
        case LINUX32_SYS_linux_fchmodat:
        case LINUX32_SYS_linux_fstatat64:
        case LINUX32_SYS_linux_linkat:
        case LINUX32_SYS_linux_unlinkat:
        case LINUX32_SYS_linux_utimensat:
                putchar(',');
                print_mask_arg0(sysdecode_linux_atflags, *ip);
                ip++;
                narg--;
                break;
        }
        *resc = c;
        *resip = ip;
        *resnarg = narg;
}
#endif /* __amd64__ */

static void
ktrsigset(const char *name, const l_sigset_t *mask, size_t sz)
{
        unsigned long i, c;

        printf("%s [ ", name);
        c = 0;
        for (i = 1; i <= sz * CHAR_BIT; i++) {
                if (!LINUX_SIGISMEMBER(*mask, i))
                        continue;
                if (c != 0)
                        printf(", ");
                printf("%s", sysdecode_linux_signal(i));
                c++;
        }
        if (c == 0)
                printf("empty ]\n");
        else
                printf(" ]\n");
}

bool
ktrstruct_linux(const char *name, const char *data, size_t datalen)
{
        l_sigset_t mask;

        if (strcmp(name, "l_sigset_t") == 0) {
                /* Old Linux sigset_t is one word size. */
                if (datalen < sizeof(int) || datalen > sizeof(l_sigset_t))
                        return (false);
                memcpy(&mask, data, datalen);
                ktrsigset(name, &mask, datalen);
        } else
                return (false);

        return (true);
}