forked from haiwan/sledge
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
74 lines
1.3 KiB
74 lines
1.3 KiB
#ifndef SFRT_SOFTINT_H
|
|
#define SFRT_SOFTINT_H
|
|
|
|
#include <stdbool.h>
|
|
#include <assert.h>
|
|
#include <signal.h>
|
|
|
|
static inline int
|
|
softint_mask(int sig)
|
|
{
|
|
sigset_t set;
|
|
int ret;
|
|
|
|
assert(sig == SIGALRM || sig == SIGUSR1);
|
|
/* all threads created by the calling thread will have sig blocked */
|
|
sigemptyset(&set);
|
|
sigaddset(&set, sig);
|
|
ret = pthread_sigmask(SIG_BLOCK, &set, NULL);
|
|
if (ret != 0) {
|
|
errno = ret;
|
|
perror("pthread_sigmask");
|
|
exit(-1);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static inline int
|
|
softint_unmask(int sig)
|
|
{
|
|
sigset_t set;
|
|
int ret;
|
|
|
|
assert(sig == SIGALRM || sig == SIGUSR1);
|
|
/* all threads created by the calling thread will have sig unblocked */
|
|
sigemptyset(&set);
|
|
sigaddset(&set, sig);
|
|
ret = pthread_sigmask(SIG_UNBLOCK, &set, NULL);
|
|
if (ret != 0) {
|
|
errno = ret;
|
|
perror("pthread_sigmask");
|
|
exit(-1);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
extern __thread volatile sig_atomic_t softint_off;
|
|
|
|
static inline void
|
|
softint_disable(void)
|
|
{
|
|
while (__sync_bool_compare_and_swap(&softint_off, 0, 1) == false) ;
|
|
}
|
|
|
|
static inline void
|
|
softint_enable(void)
|
|
{
|
|
if (__sync_bool_compare_and_swap(&softint_off, 1, 0) == false) assert(0);
|
|
}
|
|
|
|
static inline int
|
|
softint_enabled(void)
|
|
{
|
|
return (softint_off == 0);
|
|
}
|
|
|
|
void softint_init(void);
|
|
|
|
void softint_timer_arm(void);
|
|
void softint_timer_disarm(void);
|
|
|
|
#endif /* SFRT_SOFTINT_H */
|