libc/sys/t_sigqueue.c

/* $NetBSD: t_sigqueue.c,v 1.7 2017/01/13 20:44:10 christos Exp $ */

/*-
 * Copyright (c) 2011 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * This code is derived from software contributed to The NetBSD Foundation
 * by Christos Zoulas.
 *
 * 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 NETBSD FOUNDATION, INC. 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 FOUNDATION 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/cdefs.h>
__RCSID("$NetBSD: t_sigqueue.c,v 1.7 2017/01/13 20:44:10 christos Exp $");

#include <atf-c.h>
#include <err.h>
#include <errno.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <sched.h>
#include <unistd.h>

static void         handler(int, siginfo_t *, void *);

#define VALUE (int)0xc001dad1
static int value;

static void
handler(int signo __unused, siginfo_t *info, void *data __unused)
{
          value = info->si_value.sival_int;
          kill(0, SIGINFO);
}

ATF_TC(sigqueue_basic);
ATF_TC_HEAD(sigqueue_basic, tc)
{
          atf_tc_set_md_var(tc, "descr", "Checks sigqueue(3) sigval delivery");
}

ATF_TC_BODY(sigqueue_basic, tc)
{
          struct sigaction sa;
          union sigval sv;

          sa.sa_sigaction = handler;
          sigemptyset(&sa.sa_mask);
          sa.sa_flags = SA_SIGINFO;

          if (sigaction(SIGUSR1, &sa, NULL) != 0)
                    atf_tc_fail("sigaction failed");

          sv.sival_int = VALUE;

          if (sigqueue(0, SIGUSR1, sv) != 0)
                    atf_tc_fail("sigqueue failed");

          sched_yield();
          ATF_REQUIRE_EQ(sv.sival_int, value);
}

ATF_TC(sigqueue_err);
ATF_TC_HEAD(sigqueue_err, tc)
{
          atf_tc_set_md_var(tc, "descr", "Test errors from sigqueue(3)");
}

ATF_TC_BODY(sigqueue_err, tc)
{
          static union sigval sv;

          errno = 0;
          ATF_REQUIRE_ERRNO(EINVAL, sigqueue(getpid(), -1, sv) == -1);
}

static int signals[] = {
          SIGINT, SIGRTMIN + 1, SIGINT, SIGRTMIN + 0, SIGRTMIN + 2,
          SIGQUIT, SIGRTMIN + 1
};
#ifdef __arraycount
#define CNT         __arraycount(signals)
#else
#define CNT         (sizeof(signals) / sizeof(signals[0]))
#endif

static sig_atomic_t count = 0;
static int delivered[CNT];

static void
myhandler(int signo, siginfo_t *info, void *context __unused)
{
          delivered[count++] = signo;
          printf("Signal #%zu: signo: %d\n", (size_t)count, signo);
}

static int
asc(const void *a, const void *b)
{
          const int *ia = a, *ib = b;
          return *ib - *ia;
}

/*
 * given a array of signals to be delivered in tosend of size len
 * place in ordered the signals to be delivered in delivery order
 * and return the number of signals that should be delivered
 */
static size_t
sigorder(int *ordered, const int *tosend, size_t len)
{
          memcpy(ordered, tosend, len * sizeof(*tosend));
          qsort(ordered, len, sizeof(*ordered), asc);
          if (len == 1)
                    return len;

          size_t i, j;
          for (i = 0, j = 0; i < len - 1; i++) {
                    if (ordered[i] >= SIGRTMIN)
                              continue;
                    if (j == 0)
                              j = i + 1;
                    while (ordered[i] == ordered[j] && j < len)
                              j++;
                    if (j == len)
                              break;
                    ordered[i + 1] = ordered[j];
          }
          return i + 1;
}

ATF_TC(sigqueue_rt);
ATF_TC_HEAD(sigqueue_rt, tc)
{
          atf_tc_set_md_var(tc, "descr", "Test queuing of real-time signals");
}

ATF_TC_BODY(sigqueue_rt, tc)
{
          pid_t pid;
          union sigval val;
          struct sigaction act;
          int ordered[CNT];
          struct sigaction oact[CNT];
          size_t ndelivered;

          ndelivered = sigorder(ordered, signals, CNT);

          act.sa_flags = SA_SIGINFO;
          act.sa_sigaction = myhandler;
          sigemptyset(&act.sa_mask);
          for (size_t i = 0; i < ndelivered; i++)
                    ATF_REQUIRE(sigaction(ordered[i], &act, &oact[i]) != -1);

          val.sival_int = 0;
          pid = getpid();

          sigset_t mask, orig;
          sigemptyset(&mask);
          for (size_t i = 0; i < CNT; i++)
                    if (sigaddset(&mask, signals[i]) == -1)
                              warn("sigaddset");

          ATF_REQUIRE(sigprocmask(SIG_BLOCK, &mask, &orig) != -1);

          for (size_t i = 0; i < CNT; i++)
                    ATF_REQUIRE(sigqueue(pid, signals[i], val) != -1);

          ATF_REQUIRE(sigprocmask(SIG_UNBLOCK, &mask, &orig) != -1);
          sleep(1);
          ATF_CHECK_MSG((size_t)count == ndelivered,
              "count %zu != ndelivered %zu", (size_t)count, ndelivered);
          for (size_t i = 0; i < ndelivered; i++)
                    ATF_REQUIRE_MSG(ordered[i] == delivered[i],
                        "%zu: ordered %d != delivered %d",
                        i, ordered[i], delivered[i]);

          if ((size_t)count > ndelivered)
                    for (size_t i = ndelivered; i < (size_t)count; i++)
                              printf("Undelivered signal #%zu: %d\n", i, ordered[i]);

          for (size_t i = 0; i < ndelivered; i++)
                    ATF_REQUIRE(sigaction(signals[i], &oact[i], NULL) != -1);
}

ATF_TP_ADD_TCS(tp)
{

          ATF_TP_ADD_TC(tp, sigqueue_basic);
          ATF_TP_ADD_TC(tp, sigqueue_err);
          ATF_TP_ADD_TC(tp, sigqueue_rt);

          return atf_no_error();
}