// SPDX-License-Identifier: GPL-2.0

/* kselftest for acct() system call
 *  The acct() system call enables or disables process accounting.
 */

#include <stdio.h>
#include <errno.h>
#include <string.h>
#include <sys/wait.h>

#include "kselftest.h"

int main(void)
{
        char filename[] = "process_log";
        FILE *fp;
        pid_t child_pid;
        int sz;

        // Setting up kselftest framework
        ksft_print_header();
        ksft_set_plan(1);

        // Check if test is run a root
        if (geteuid()) {
                ksft_exit_skip("This test needs root to run!\n");
                return 1;
        }

        // Create file to log closed processes
        fp = fopen(filename, "w");

        if (!fp) {
                ksft_test_result_error("%s.\n", strerror(errno));
                ksft_finished();
                return 1;
        }

        acct(filename);

        // Handle error conditions
        if (errno) {
                ksft_test_result_error("%s.\n", strerror(errno));
                fclose(fp);
                ksft_finished();
                return 1;
        }

        // Create child process and wait for it to terminate.

        child_pid = fork();

        if (child_pid < 0) {
                ksft_test_result_error("Creating a child process to log failed\n");
                acct(NULL);
                return 1;
        } else if (child_pid > 0) {
                wait(NULL);
                fseek(fp, 0L, SEEK_END);
                sz = ftell(fp);

                acct(NULL);

                if (sz <= 0) {
                        ksft_test_result_fail("Terminated child process not logged\n");
                        ksft_exit_fail();
                        return 1;
                }

                ksft_test_result_pass("Successfully logged terminated process.\n");
                fclose(fp);
                ksft_exit_pass();
                return 0;
        }

        return 1;
}