LXR linux/tools/testing/selftests/powerpc/harness.c

   1/*
   2 * Copyright 2013, Michael Ellerman, IBM Corp.
   3 * Licensed under GPLv2.
   4 */
   5
   6#include <errno.h>
   7#include <signal.h>
   8#include <stdbool.h>
   9#include <stdio.h>
  10#include <stdlib.h>
  11#include <sys/types.h>
  12#include <sys/wait.h>
  13#include <unistd.h>
  14#include <elf.h>
  15#include <fcntl.h>
  16#include <link.h>
  17#include <sys/stat.h>
  18
  19#include "subunit.h"
  20#include "utils.h"
  21
  22#define KILL_TIMEOUT    5
  23
  24static uint64_t timeout = 120;
  25
  26int run_test(int (test_function)(void), char *name)
  27{
  28        bool terminated;
  29        int rc, status;
  30        pid_t pid;
  31
  32        /* Make sure output is flushed before forking */
  33        fflush(stdout);
  34
  35        pid = fork();
  36        if (pid == 0) {
  37                setpgid(0, 0);
  38                exit(test_function());
  39        } else if (pid == -1) {
  40                perror("fork");
  41                return 1;
  42        }
  43
  44        setpgid(pid, pid);
  45
  46        /* Wake us up in timeout seconds */
  47        alarm(timeout);
  48        terminated = false;
  49
  50wait:
  51        rc = waitpid(pid, &status, 0);
  52        if (rc == -1) {
  53                if (errno != EINTR) {
  54                        printf("unknown error from waitpid\n");
  55                        return 1;
  56                }
  57
  58                if (terminated) {
  59                        printf("!! force killing %s\n", name);
  60                        kill(-pid, SIGKILL);
  61                        return 1;
  62                } else {
  63                        printf("!! killing %s\n", name);
  64                        kill(-pid, SIGTERM);
  65                        terminated = true;
  66                        alarm(KILL_TIMEOUT);
  67                        goto wait;
  68                }
  69        }
  70
  71        /* Kill anything else in the process group that is still running */
  72        kill(-pid, SIGTERM);
  73
  74        if (WIFEXITED(status))
  75                status = WEXITSTATUS(status);
  76        else {
  77                if (WIFSIGNALED(status))
  78                        printf("!! child died by signal %d\n", WTERMSIG(status));
  79                else
  80                        printf("!! child died by unknown cause\n");
  81
  82                status = 1; /* Signal or other */
  83        }
  84
  85        return status;
  86}
  87
  88static void sig_handler(int signum)
  89{
  90        /* Just wake us up from waitpid */
  91}
  92
  93static struct sigaction sig_action = {
  94        .sa_handler = sig_handler,
  95};
  96
  97void test_harness_set_timeout(uint64_t time)
  98{
  99        timeout = time;
 100}
 101
 102int test_harness(int (test_function)(void), char *name)
 103{
 104        int rc;
 105
 106        test_start(name);
 107        test_set_git_version(GIT_VERSION);
 108
 109        if (sigaction(SIGINT, &sig_action, NULL)) {
 110                perror("sigaction (sigint)");
 111                test_error(name);
 112                return 1;
 113        }
 114
 115        if (sigaction(SIGALRM, &sig_action, NULL)) {
 116                perror("sigaction (sigalrm)");
 117                test_error(name);
 118                return 1;
 119        }
 120
 121        rc = run_test(test_function, name);
 122
 123        if (rc == MAGIC_SKIP_RETURN_VALUE) {
 124                test_skip(name);
 125                /* so that skipped test is not marked as failed */
 126                rc = 0;
 127        } else
 128                test_finish(name, rc);
 129
 130        return rc;
 131}
 132