c8dca46ee7d81d1078b7e82d3450727518df5742
[openafs.git] / src / lwp / preempt.c
1 /*
2  * Copyright 2000, International Business Machines Corporation and others.
3  * All Rights Reserved.
4  *
5  * This software has been released under the terms of the IBM Public
6  * License.  For details, see the LICENSE file in the top-level source
7  * directory or online at http://www.openafs.org/dl/license10.html
8  */
9
10 /*******************************************************************\
11 *                                                                   *
12 *       Information Technology Center                               *
13 *       Carnegie-Mellon University                                  *
14 *                                                                   *
15 \*******************************************************************/
16 #include <afsconfig.h>
17 #include <afs/param.h>
18
19
20
21 #include "lwp.h"
22 #include "preempt.h"
23
24 #if defined(AFS_LINUX20_ENV) || defined(AFS_DARWIN_ENV) || defined(AFS_XBSD_ENV)
25 int PRE_Block = 0;
26
27
28 int
29 PRE_InitPreempt(struct timeval *slice)
30 {
31     return LWP_SUCCESS;
32 }
33
34 int
35 PRE_EndPreempt(void)
36 {
37     return LWP_SUCCESS;
38 }
39
40 #else
41 #include <sys/time.h>
42 #include <signal.h>
43
44 #if defined(AFS_OSF_ENV) || defined(AFS_S390_LINUX20_ENV)
45 int PRE_Block = 0;              /* used in lwp.c and process.s */
46 #else
47 char PRE_Block = 0;             /* used in lwp.c and process.s */
48 #endif
49
50 #if HAVE_SIGACTION && defined(SA_SIGINFO)
51 static void
52 AlarmHandler(int sig, siginfo_t *st, ucontext_t *scp)
53 #else
54 static void
55 AlarmHandler(int sig, int code, struct sigcontext *scp)
56 #endif
57 {
58     if (PRE_Block == 0 && lwp_cpptr->level == 0) {
59         PRE_BeginCritical();
60 #if HAVE_SIGACTION && defined(SA_SIGINFO)
61         sigprocmask(SIG_SETMASK, &scp->uc_sigmask, NULL);
62 #else
63         sigsetmask(scp->sc_mask);
64 #endif
65         LWP_DispatchProcess();
66         PRE_EndCritical();
67     }
68
69 }
70
71 int
72 PRE_InitPreempt(struct timeval *slice)
73 {
74     struct itimerval itv;
75 #if HAVE_SIGACTION && defined(SA_SIGINFO)
76     struct sigaction action;
77 #else
78     struct sigvec vec;
79 #endif
80
81     if (lwp_cpptr == 0)
82         return (LWP_EINIT);
83
84     if (slice == 0) {
85         itv.it_interval.tv_sec = itv.it_value.tv_sec = DEFAULTSLICE;
86         itv.it_interval.tv_usec = itv.it_value.tv_usec = 0;
87     } else {
88         itv.it_interval = itv.it_value = *slice;
89     }
90
91 #if HAVE_SIGACTION && defined(SA_SIGINFO)
92     memset(&action, 0, sizeof(action));
93     action.sa_sigaction = AlarmHandler;
94     action.sa_flags = SA_SIGINFO;
95
96     if ((sigaction(SIGALRM, &action, NULL) == -1)
97         || (setitimer(ITIMER_REAL, &itv, NULL) == -1))
98         return (LWP_ESYSTEM);
99 #else
100     memset(&vec, 0, sizeof(vec));
101     vec.sv_handler = AlarmHandler;
102     vec.sv_mask = vec.sv_onstack = 0;
103
104     if ((sigvec(SIGALRM, &vec, (struct sigvec *)0) == -1)
105         || (setitimer(ITIMER_REAL, &itv, (struct itimerval *)0) == -1))
106         return (LWP_ESYSTEM);
107 #endif
108
109     return (LWP_SUCCESS);
110 }
111
112 int
113 PRE_EndPreempt()
114 {
115     struct itimerval itv;
116 #if HAVE_SIGACTION && defined(SA_SIGINFO)
117     struct sigaction action;
118 #else
119     struct sigvec vec;
120 #endif
121
122     if (lwp_cpptr == 0)
123         return (LWP_EINIT);
124
125     itv.it_value.tv_sec = itv.it_value.tv_usec = 0;
126
127 #if HAVE_SIGACTION && defined(SA_SIGINFO)
128     memset(&action, 0, sizeof(action));
129     action.sa_handler = SIG_DFL;
130
131     if ((setitimer(ITIMER_REAL, &itv, NULL) == -1)
132         || (sigaction(SIGALRM, &action, NULL) == -1))
133         return (LWP_ESYSTEM);
134
135 #else
136     memset(&vec, 0, sizeof(vec));
137     vec.sv_handler = SIG_DFL;
138     vec.sv_mask = vec.sv_onstack = 0;
139
140     if ((setitimer(ITIMER_REAL, &itv, (struct itimerval *)0) == -1)
141         || (sigvec(SIGALRM, &vec, (struct sigvec *)0) == -1))
142         return (LWP_ESYSTEM);
143 #endif
144     return (LWP_SUCCESS);
145 }
146
147 #endif /* AFS_LINUX20_ENV */