// SPDX-License-Identifier: GPL-2.0+
/*
 * Copyright (c) 2020 Bernd Edlinger <bernd.edlinger@hotmail.de>
 * All rights reserved.
 *
 * Check whether /proc/$pid/mem can be accessed without causing deadlocks
 * when de_thread is blocked with ->cred_guard_mutex held.
 */

#include "kselftest_harness.h"
#include <stdio.h>
#include <fcntl.h>
#include <pthread.h>
#include <signal.h>
#include <unistd.h>
#include <sys/ptrace.h>

static void *thread(void *arg)
{
        ptrace(PTRACE_TRACEME, 0, 0L, 0L);
        return NULL;
}

TEST(vmaccess)
{
        int f, pid = fork();
        char mm[64];

        if (!pid) {
                pthread_t pt;

                pthread_create(&pt, NULL, thread, NULL);
                pthread_join(pt, NULL);
                execlp("true", "true", NULL);
        }

        sleep(1);
        sprintf(mm, "/proc/%d/mem", pid);
        f = open(mm, O_RDONLY);
        ASSERT_GE(f, 0);
        close(f);
        f = kill(pid, SIGCONT);
        ASSERT_EQ(f, 0);
}

TEST(attach)
{
        int s, k, pid = fork();

        if (!pid) {
                pthread_t pt;

                pthread_create(&pt, NULL, thread, NULL);
                pthread_join(pt, NULL);
                execlp("sleep", "sleep", "2", NULL);
        }

        sleep(1);
        k = ptrace(PTRACE_ATTACH, pid, 0L, 0L);
        ASSERT_EQ(errno, EAGAIN);
        ASSERT_EQ(k, -1);
        k = waitpid(-1, &s, WNOHANG);
        ASSERT_NE(k, -1);
        ASSERT_NE(k, 0);
        ASSERT_NE(k, pid);
        ASSERT_EQ(WIFEXITED(s), 1);
        ASSERT_EQ(WEXITSTATUS(s), 0);
        sleep(1);
        k = ptrace(PTRACE_ATTACH, pid, 0L, 0L);
        ASSERT_EQ(k, 0);
        k = waitpid(-1, &s, 0);
        ASSERT_EQ(k, pid);
        ASSERT_EQ(WIFSTOPPED(s), 1);
        ASSERT_EQ(WSTOPSIG(s), SIGSTOP);
        k = ptrace(PTRACE_DETACH, pid, 0L, 0L);
        ASSERT_EQ(k, 0);
        k = waitpid(-1, &s, 0);
        ASSERT_EQ(k, pid);
        ASSERT_EQ(WIFEXITED(s), 1);
        ASSERT_EQ(WEXITSTATUS(s), 0);
        k = waitpid(-1, NULL, 0);
        ASSERT_EQ(k, -1);
        ASSERT_EQ(errno, ECHILD);
}

TEST_HARNESS_MAIN