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 	pthread_t 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);
97