drm/linux: Improve put_user()
[dragonfly.git] / lib / libthread_xu / thread / thr_setschedparam.c
1 /*
2  * Copyright (c) 1998 Daniel Eischen <eischen@vigrid.com>.
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  * 3. All advertising materials mentioning features or use of this software
14  *    must display the following acknowledgement:
15  *      This product includes software developed by Daniel Eischen.
16  * 4. Neither the name of the author nor the names of any co-contributors
17  *    may be used to endorse or promote products derived from this software
18  *    without specific prior written permission.
19  *
20  * THIS SOFTWARE IS PROVIDED BY DANIEL EISCHEN AND CONTRIBUTORS ``AS IS'' AND
21  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
24  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
25  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
26  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
29  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
30  * SUCH DAMAGE.
31  *
32  */
33
34 #include "namespace.h"
35 #include <machine/tls.h>
36 #include <sys/param.h>
37 #include <errno.h>
38 #include <pthread.h>
39 #include "un-namespace.h"
40
41 #include "thr_private.h"
42
43 int
44 _pthread_setschedparam(pthread_t pthread, int policy,
45         const struct sched_param *param)
46 {
47         struct pthread *curthread = tls_get_curthread();
48         int     ret = 0;
49
50         if (pthread == curthread) {
51                 THR_LOCK(curthread);
52                 if (curthread->attr.sched_policy == policy &&
53                      curthread->attr.prio == param->sched_priority) {
54                         THR_UNLOCK(curthread);
55                         return (0);
56                 }
57                 if (policy == SCHED_OTHER) {
58                         ret = _thr_set_sched_other_prio(curthread,
59                                                 param->sched_priority);
60                 } else {
61                         ret = _thr_setscheduler(curthread->tid, policy, param);
62                 }
63                 if (ret == -1)
64                         ret = errno;
65                 else {
66                         curthread->attr.sched_policy = policy;
67                         curthread->attr.prio = param->sched_priority;
68                 }
69                 THR_UNLOCK(curthread);
70         } else if ((ret = _thr_ref_add(curthread, pthread, /*include dead*/0))
71                 == 0) {
72                 THR_THREAD_LOCK(curthread, pthread);
73                 if (pthread->attr.sched_policy == policy &&
74                      pthread->attr.prio == param->sched_priority) {
75                         THR_THREAD_UNLOCK(curthread, pthread);
76                         return (0);
77                 }
78                 if (policy == SCHED_OTHER) {
79                         ret = _thr_set_sched_other_prio(pthread,
80                                                 param->sched_priority);
81                 } else {
82                         ret = _thr_setscheduler(pthread->tid, policy, param);
83                 }
84                 if (ret == -1)
85                         ret = errno;
86                 else {
87                         pthread->attr.sched_policy = policy;
88                         pthread->attr.prio = param->sched_priority;
89                 }
90                 THR_THREAD_UNLOCK(curthread, pthread);
91                 _thr_ref_delete(curthread, pthread);
92         }
93         return (ret);
94 }
95
96 __strong_reference(_pthread_setschedparam, pthread_setschedparam);