1 /*
2 * CDDL HEADER START
3 *
4 * The contents of this file are subject to the terms of the
5 * Common Development and Distribution License (the "License").
6 * You may not use this file except in compliance with the License.
7 *
8 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9 * or http://www.opensolaris.org/os/licensing.
10 * See the License for the specific language governing permissions
11 * and limitations under the License.
12 *
13 * When distributing Covered Code, include this CDDL HEADER in each
14 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15 * If applicable, add the following below this CDDL HEADER, with the
16 * fields enclosed by brackets "[]" replaced with your own identifying
17 * information: Portions Copyright [yyyy] [name of copyright owner]
18 *
19 * CDDL HEADER END
20 */
21
22 /*
23 * Copyright 2008 Sun Microsystems, Inc. All rights reserved.
24 * Use is subject to license terms.
25 */
26
27 /* Copyright (c) 1984, 1986, 1987, 1988, 1989 AT&T */
28 /* All Rights Reserved */
29
30 /*
31 * University Copyright- Copyright (c) 1982, 1986, 1988
32 * The Regents of the University of California
33 * All Rights Reserved
34 *
35 * University Acknowledgment- Portions of this document are derived from
36 * software developed by the University of California, Berkeley, and its
37 * contributors.
38 */
39
40 #pragma ident "%Z%%M% %I% %E% SMI"
41
42 #include "lint.h"
43 #include <string.h>
44 #include <limits.h>
45 #include <sys/types.h>
46 #include <sys/time.h>
47 #include <sys/resource.h>
48 #include <sys/procset.h>
49 #include <sys/priocntl.h>
50 #include <limits.h>
51 #include <errno.h>
52 #include <priv.h>
53
54 static idtype_t
prio_to_idtype(int which)55 prio_to_idtype(int which)
56 {
57 switch (which) {
58
59 case PRIO_PROCESS:
60 return (P_PID);
61
62 case PRIO_PGRP:
63 return (P_PGID);
64
65 case PRIO_USER:
66 return (P_UID);
67
68 case PRIO_GROUP:
69 return (P_GID);
70
71 case PRIO_SESSION:
72 return (P_SID);
73
74 case PRIO_LWP:
75 return (P_LWPID);
76
77 case PRIO_TASK:
78 return (P_TASKID);
79
80 case PRIO_PROJECT:
81 return (P_PROJID);
82
83 case PRIO_ZONE:
84 return (P_ZONEID);
85
86 case PRIO_CONTRACT:
87 return (P_CTID);
88
89 default:
90 return (-1);
91 }
92 }
93
94 static int
old_idtype(int which)95 old_idtype(int which)
96 {
97 switch (which) {
98 case PRIO_PROCESS:
99 case PRIO_PGRP:
100 case PRIO_USER:
101 return (1);
102 default:
103 return (0);
104 }
105 }
106
107 int
getpriority(int which,id_t who)108 getpriority(int which, id_t who)
109 {
110 id_t id;
111 idtype_t idtype;
112 pcnice_t pcnice;
113
114 if ((idtype = prio_to_idtype(which)) == -1) {
115 errno = EINVAL;
116 return (-1);
117 }
118
119 if (who < 0) {
120 if (old_idtype(which)) {
121 errno = EINVAL;
122 return (-1);
123 } else if (who != P_MYID) {
124 errno = EINVAL;
125 return (-1);
126 }
127 }
128
129 /*
130 * The POSIX standard requires that a 0 value for the who argument
131 * should specify the current process, process group, or user.
132 * For all other id types we can treat zero as normal id value.
133 */
134 if (who == 0 && old_idtype(which))
135 id = P_MYID;
136 else
137 id = who;
138
139 pcnice.pc_val = 0;
140 pcnice.pc_op = PC_GETNICE;
141
142 if (priocntl(idtype, id, PC_DONICE, &pcnice) == -1)
143 return (-1);
144 else
145 return (pcnice.pc_val);
146 }
147
148 int
setpriority(int which,id_t who,int prio)149 setpriority(int which, id_t who, int prio)
150 {
151 id_t id;
152 idtype_t idtype;
153 pcnice_t pcnice;
154 int ret;
155
156 if ((idtype = prio_to_idtype(which)) == -1) {
157 errno = EINVAL;
158 return (-1);
159 }
160
161 if (who < 0) {
162 if (old_idtype(which)) {
163 errno = EINVAL;
164 return (-1);
165 } else if (who != P_MYID) {
166 errno = EINVAL;
167 return (-1);
168 }
169 }
170
171 if (who == 0 && old_idtype(which))
172 id = P_MYID;
173 else
174 id = who;
175
176 if (prio > NZERO - 1)
177 prio = NZERO - 1;
178 else if (prio < -NZERO)
179 prio = -NZERO;
180
181 pcnice.pc_val = prio;
182 pcnice.pc_op = PC_SETNICE;
183
184 ret = priocntl(idtype, id, PC_DONICE, &pcnice);
185
186 if (ret != 0 && errno == EPERM) {
187 pcnice_t gpcnice = { 0, PC_GETNICE };
188 priv_set_t *pset = NULL;
189
190 /*
191 * The priocntl PC_DONICE subcommand returns EPERM if we lack
192 * sufficient privileges to carry out the operation, but
193 * setpriority(3C) may need to return EACCES. We can't just
194 * change EPERM to EACCES, because there are other conditions
195 * which legitimately cause EPERM (such as an euid/ruid mismatch
196 * between the current process and the target.).
197 *
198 * setpriority(3C) must return EACCES if we lack the privilege
199 * checked for below and we are trying to increase the process
200 * priority (by lowering the numeric value of its priority).
201 */
202 if (priocntl(idtype, id, PC_DONICE, &gpcnice) == 0 &&
203 prio < gpcnice.pc_val) {
204 if ((pset = priv_allocset()) != NULL &&
205 getppriv(PRIV_EFFECTIVE, pset) == 0 &&
206 !priv_ismember(pset, "proc_priocntl"))
207 errno = EACCES;
208 if (pset != NULL)
209 priv_freeset(pset);
210 }
211 }
212
213 return (ret);
214 }
215