* Copyright 2024, Haiku, Inc. All Rights Reserved.
* Distributed under the terms of the MIT License.
*
* Authors:
* Jérôme Duval, jerome.duval@gmail.com
* Inspired from Android signal_test.cpp
*/
#include <errno.h>
#include <pthread.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <sys/cdefs.h>
#include <sys/types.h>
#include <unistd.h>
#define ASSERT_EQ(x,y) { if (x != y) { fprintf(stderr, "assert failed %s %s\n", #x, #y); exit(1); }}
#define ASSERT_ERRNO(x) ASSERT_EQ(errno, x)
class ScopedSignalHandler
{
public:
ScopedSignalHandler(int signal, void (*handler)(int, siginfo_t*, void*))
: signalNumber(signal)
{
sigemptyset(&action.sa_mask);
action.sa_flags = SA_SIGINFO;
action.sa_sigaction = handler;
sigaction(signalNumber, &action, &oldAction);
}
~ScopedSignalHandler()
{
sigaction(signalNumber, &oldAction, NULL);
}
private:
struct sigaction action;
struct sigaction oldAction;
const int signalNumber;
};
static int gSigqueueSignalHandlerCallCount = 0;
static void
SigqueueSignalHandler(int signum, siginfo_t* info, void*)
{
ASSERT_EQ(SIGALRM, signum);
ASSERT_EQ(SIGALRM, info->si_signo);
ASSERT_EQ(SI_QUEUE, info->si_code);
ASSERT_EQ(1, info->si_value.sival_int);
++gSigqueueSignalHandlerCallCount;
}
void
signal_sigqueue()
{
gSigqueueSignalHandlerCallCount = 0;
ScopedSignalHandler ssh(SIGALRM, SigqueueSignalHandler);
sigval sigval = {.sival_int = 1};
errno = 0;
ASSERT_EQ(0, sigqueue(getpid(), SIGALRM, sigval));
ASSERT_ERRNO(0);
ASSERT_EQ(1, gSigqueueSignalHandlerCallCount);
}
void
signal_pthread_sigqueue_self()
{
gSigqueueSignalHandlerCallCount = 0;
ScopedSignalHandler ssh(SIGALRM, SigqueueSignalHandler);
sigval sigval = {.sival_int = 1};
errno = 0;
ASSERT_EQ(0, pthread_sigqueue(pthread_self(), SIGALRM, sigval));
ASSERT_ERRNO(0);
ASSERT_EQ(1, gSigqueueSignalHandlerCallCount);
}
void
signal_pthread_sigqueue_other()
{
gSigqueueSignalHandlerCallCount = 0;
ScopedSignalHandler ssh(SIGALRM, SigqueueSignalHandler);
sigval sigval = {.sival_int = 1};
sigset_t mask;
sigfillset(&mask);
pthread_sigmask(SIG_SETMASK, &mask, nullptr);
pthread_t thread;
int rc = pthread_create(&thread, nullptr,
[](void*) -> void* {
sigset_t mask;
sigemptyset(&mask);
sigsuspend(&mask);
return nullptr;
}, nullptr);
ASSERT_EQ(0, rc);
errno = 0;
ASSERT_EQ(0, pthread_sigqueue(thread, SIGALRM, sigval));
ASSERT_ERRNO(0);
pthread_join(thread, nullptr);
ASSERT_EQ(1, gSigqueueSignalHandlerCallCount);
}
extern "C" int
main()
{
printf("signal_sigqueue\n");
signal_sigqueue();
printf("signal_pthread_sigqueue_self\n");
signal_pthread_sigqueue_self();
printf("signal_pthread_sigqueue_other\n");
signal_pthread_sigqueue_other();
}
