/*      $NetBSD: t_backtrace_sandbox.c,v 1.3 2025/01/30 16:13:51 christos Exp $ */

/*-
 * Copyright (c) 2025 Kyle Evans <kevans@FreeBSD.org>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */
#include <sys/cdefs.h>
__RCSID("$NetBSD: t_backtrace_sandbox.c,v 1.3 2025/01/30 16:13:51 christos Exp $");

#include <sys/param.h>
#include <sys/wait.h>
#ifdef __FreeBSD__
#include <sys/capsicum.h>
#define __arraycount(a) nitems(a)
#endif

#include <execinfo.h>
#include <signal.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>

#include <atf-c.h>

#define BT_FUNCTIONS            10

ATF_TC(backtrace_sandbox);
ATF_TC_HEAD(backtrace_sandbox, tc)
{
        atf_tc_set_md_var(tc, "descr",
            "Test backtrace_sandbox_init(3) in sandbox");
#ifndef __FreeBSD__
        atf_tc_set_md_var(tc, "require.user", "root");
#endif
}

ATF_TC_BODY(backtrace_sandbox, tc)
{
        void *addr[BT_FUNCTIONS];
        char **syms;
        size_t frames;
        pid_t pid;
        int status;

        frames = backtrace(addr, __arraycount(addr));
        ATF_REQUIRE(frames > 0);

        syms = backtrace_symbols_fmt(addr, frames, "%n");
        ATF_REQUIRE(strcmp(syms[0], "atfu_backtrace_sandbox_body") == 0);

        pid = fork();
        ATF_REQUIRE(pid >= 0);

        if (pid == 0) {

                backtrace_sandbox_init();
#ifdef __FreeBSD__
                cap_enter();
#else
                if (chroot("/tmp") != 0)
                        _exit(EXIT_FAILURE);
#endif

                syms = backtrace_symbols_fmt(addr, frames, "%n");
                if (strcmp(syms[0], "atfu_backtrace_sandbox_body") != 0)
                        _exit(EXIT_FAILURE);

                backtrace_sandbox_fini();

                _exit(EXIT_SUCCESS);
        }

        (void)wait(&status);

        if (!WIFEXITED(status) || WEXITSTATUS(status) != EXIT_SUCCESS)
                atf_tc_fail("resolving symbols in chroot failed");

}

ATF_TP_ADD_TCS(tp)
{

        ATF_TP_ADD_TC(tp, backtrace_sandbox);

        return atf_no_error();
}