about summary refs log tree commit diff
path: root/support/delayed_exit.c
blob: bc64ff0d3e7966b2b57491ec7f3568de05ba75f9 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
/* Time-triggered process termination.
   Copyright (C) 2016-2023 Free Software Foundation, Inc.
   This file is part of the GNU C Library.

   The GNU C Library is free software; you can redistribute it and/or
   modify it under the terms of the GNU Lesser General Public
   License as published by the Free Software Foundation; either
   version 2.1 of the License, or (at your option) any later version.

   The GNU C Library is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
   Lesser General Public License for more details.

   You should have received a copy of the GNU Lesser General Public
   License along with the GNU C Library; if not, see
   <https://www.gnu.org/licenses/>.  */

#include <support/xthread.h>
#include <support/xsignal.h>

#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <support/check.h>
#include <support/support.h>
#include <time.h>
#include <unistd.h>

struct delayed_exit_request
{
  void (*exitfunc) (int);
  int seconds;
};

static void *
delayed_exit_thread (void *closure)
{
  struct delayed_exit_request *request = closure;
  void (*exitfunc) (int) = request->exitfunc;
  struct timespec delay = { request->seconds, 0 };
  struct timespec remaining = { 0 };
  free (request);

  if (nanosleep (&delay, &remaining) != 0)
    FAIL_EXIT1 ("nanosleep: %m");
  /* Exit the process successfully.  */
  exitfunc (0);
  return NULL;
}

static void
delayed_exit_1 (int seconds, void (*exitfunc) (int))
{
  /* Create the new thread with all signals blocked.  */
  sigset_t all_blocked;
  sigfillset (&all_blocked);
  sigset_t old_set;
  xpthread_sigmask (SIG_SETMASK, &all_blocked, &old_set);
  struct delayed_exit_request *request = xmalloc (sizeof (*request));
  request->seconds = seconds;
  request->exitfunc = exitfunc;
  /* Create a detached thread. */
  pthread_t thr = xpthread_create (NULL, delayed_exit_thread, request);
  xpthread_detach (thr);
  /* Restore the original signal mask.  */
  xpthread_sigmask (SIG_SETMASK, &old_set, NULL);
}

void
delayed_exit (int seconds)
{
  delayed_exit_1 (seconds, exit);
}

void
delayed__exit (int seconds)
{
  delayed_exit_1 (seconds, _exit);
}