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 * Copyright 2007 Sun Microsystems, Inc. All rights reserved.
23 * Use is subject to license terms.
24 */
25
26 #pragma ident "%Z%%M% %I% %E% SMI"
27
28 #define _LARGEFILE64_SOURCE
29
30 #include <stdlib.h>
31 #include <unistd.h>
32 #include <errno.h>
33 #include <strings.h>
34 #include "libproc.h"
35 #include <sys/rctl_impl.h>
36
37 /*
38 * getrctl() system call -- executed by subject process
39 */
40 int
pr_getrctl(struct ps_prochandle * Pr,const char * rname,rctlblk_t * old_blk,rctlblk_t * new_blk,int rflag)41 pr_getrctl(struct ps_prochandle *Pr, const char *rname,
42 rctlblk_t *old_blk, rctlblk_t *new_blk, int rflag)
43 {
44 sysret_t rval;
45 argdes_t argd[6];
46 argdes_t *adp;
47 int error;
48
49 if (Pr == NULL) /* no subject process */
50 return (getrctl(rname, old_blk, new_blk, rflag));
51
52 adp = &argd[0];
53 adp->arg_value = 0; /* switch for getrctl in rctlsys */
54 adp->arg_object = NULL;
55 adp->arg_type = AT_BYVAL;
56 adp->arg_inout = AI_INPUT;
57 adp->arg_size = 0;
58
59 adp++;
60 adp->arg_value = 0;
61 adp->arg_object = (void *)rname;
62 adp->arg_type = AT_BYREF;
63 adp->arg_inout = AI_INPUT;
64 adp->arg_size = strlen(rname) + 1;
65
66 adp++;
67 if (old_blk == NULL) {
68 adp->arg_value = 0;
69 adp->arg_object = NULL;
70 adp->arg_type = AT_BYVAL;
71 adp->arg_inout = AI_INPUT;
72 adp->arg_size = 0;
73 } else {
74 adp->arg_value = 0;
75 adp->arg_object = old_blk;
76 adp->arg_type = AT_BYREF;
77 adp->arg_inout = AI_INPUT;
78 adp->arg_size = rctlblk_size();
79 }
80
81 adp++;
82 if (new_blk == NULL) {
83 adp->arg_value = 0;
84 adp->arg_object = NULL;
85 adp->arg_type = AT_BYVAL;
86 adp->arg_inout = AI_OUTPUT;
87 adp->arg_size = 0;
88 } else {
89 adp->arg_value = 0;
90 adp->arg_object = new_blk;
91 adp->arg_type = AT_BYREF;
92 adp->arg_inout = AI_INOUT;
93 adp->arg_size = rctlblk_size();
94 }
95
96 adp++;
97 adp->arg_value = 0; /* obufsz isn't used by getrctl() */
98 adp->arg_object = NULL;
99 adp->arg_type = AT_BYVAL;
100 adp->arg_inout = AI_INPUT;
101 adp->arg_size = 0;
102
103 adp++;
104 adp->arg_value = rflag;
105 adp->arg_object = NULL;
106 adp->arg_type = AT_BYVAL;
107 adp->arg_inout = AI_INPUT;
108 adp->arg_size = 0;
109
110 error = Psyscall(Pr, &rval, SYS_rctlsys, 6, &argd[0]);
111
112 if (error) {
113 errno = (error > 0) ? error : ENOSYS;
114 return (-1);
115 }
116 return (rval.sys_rval1);
117 }
118
119 /*
120 * setrctl() system call -- executed by subject process
121 */
122 int
pr_setrctl(struct ps_prochandle * Pr,const char * rname,rctlblk_t * old_blk,rctlblk_t * new_blk,int rflag)123 pr_setrctl(struct ps_prochandle *Pr, const char *rname,
124 rctlblk_t *old_blk, rctlblk_t *new_blk, int rflag)
125 {
126 sysret_t rval;
127 argdes_t argd[6];
128 argdes_t *adp;
129 int error;
130
131 if (Pr == NULL) /* no subject process */
132 return (setrctl(rname, old_blk, new_blk, rflag));
133
134 adp = &argd[0];
135 adp->arg_value = 1; /* switch for setrctl in rctlsys */
136 adp->arg_object = NULL;
137 adp->arg_type = AT_BYVAL;
138 adp->arg_inout = AI_INPUT;
139 adp->arg_size = 0;
140
141 adp++;
142 adp->arg_value = 0;
143 adp->arg_object = (void *)rname;
144 adp->arg_type = AT_BYREF;
145 adp->arg_inout = AI_INPUT;
146 adp->arg_size = strlen(rname) + 1;
147
148 adp++;
149 if (old_blk == NULL) {
150 adp->arg_value = 0;
151 adp->arg_object = NULL;
152 adp->arg_type = AT_BYVAL;
153 adp->arg_inout = AI_INPUT;
154 adp->arg_size = 0;
155 } else {
156 adp->arg_value = 0;
157 adp->arg_object = old_blk;
158 adp->arg_type = AT_BYREF;
159 adp->arg_inout = AI_INPUT;
160 adp->arg_size = rctlblk_size();
161 }
162
163 adp++;
164 if (new_blk == NULL) {
165 adp->arg_value = 0;
166 adp->arg_object = NULL;
167 adp->arg_type = AT_BYVAL;
168 adp->arg_inout = AI_INPUT;
169 adp->arg_size = 0;
170 } else {
171 adp->arg_value = 0;
172 adp->arg_object = new_blk;
173 adp->arg_type = AT_BYREF;
174 adp->arg_inout = AI_INPUT;
175 adp->arg_size = rctlblk_size();
176 }
177
178 adp++;
179 adp->arg_value = 0; /* obufsz isn't used by setrctl() */
180 adp->arg_object = NULL;
181 adp->arg_type = AT_BYVAL;
182 adp->arg_inout = AI_INPUT;
183 adp->arg_size = 0;
184
185 adp++;
186 adp->arg_value = rflag;
187 adp->arg_object = NULL;
188 adp->arg_type = AT_BYVAL;
189 adp->arg_inout = AI_INPUT;
190 adp->arg_size = 0;
191
192 error = Psyscall(Pr, &rval, SYS_rctlsys, 6, &argd[0]);
193
194 if (error) {
195 errno = (error > 0) ? error : ENOSYS;
196 return (-1);
197 }
198 return (rval.sys_rval1);
199 }
200
201 /*
202 * setprojrctl() system call -- executed by subject process
203 */
204 int
pr_setprojrctl(struct ps_prochandle * Pr,const char * rname,rctlblk_t * new_blk,size_t size,int rflag)205 pr_setprojrctl(struct ps_prochandle *Pr, const char *rname,
206 rctlblk_t *new_blk, size_t size, int rflag)
207 {
208 sysret_t rval;
209 argdes_t argd[6];
210 argdes_t *adp;
211 int error;
212
213 if (Pr == NULL) /* no subject process */
214 return (setprojrctl(rname, new_blk, size, rflag));
215
216 adp = &argd[0];
217 adp->arg_value = 4; /* switch for setprojrctls in rctlsys */
218 adp->arg_object = NULL;
219 adp->arg_type = AT_BYVAL;
220 adp->arg_inout = AI_INPUT;
221 adp->arg_size = 0;
222
223 adp++;
224 adp->arg_value = 0;
225 adp->arg_object = (void *)rname;
226 adp->arg_type = AT_BYREF;
227 adp->arg_inout = AI_INPUT;
228 adp->arg_size = strlen(rname) + 1;
229
230 adp++;
231 adp->arg_value = 0; /* old_blk is not used by setprojrctls() */
232 adp->arg_object = NULL;
233 adp->arg_type = AT_BYVAL;
234 adp->arg_inout = AI_INPUT;
235 adp->arg_size = 0;
236
237
238 adp++;
239 if (new_blk == NULL) {
240 adp->arg_value = 0;
241 adp->arg_object = NULL;
242 adp->arg_type = AT_BYVAL;
243 adp->arg_inout = AI_INPUT;
244 adp->arg_size = 0;
245 } else {
246 adp->arg_value = 0;
247 adp->arg_object = new_blk;
248 adp->arg_type = AT_BYREF;
249 adp->arg_inout = AI_INPUT;
250 adp->arg_size = rctlblk_size() * size;
251 }
252
253 adp++;
254 adp->arg_value = size; /* obufsz is used by setrctls() */
255 adp->arg_object = NULL;
256 adp->arg_type = AT_BYVAL;
257 adp->arg_inout = AI_INPUT;
258 adp->arg_size = 0;
259
260 adp++;
261 adp->arg_value = rflag;
262 adp->arg_object = NULL;
263 adp->arg_type = AT_BYVAL;
264 adp->arg_inout = AI_INPUT;
265 adp->arg_size = 0;
266
267 error = Psyscall(Pr, &rval, SYS_rctlsys, 6, &argd[0]);
268
269 if (error) {
270 errno = (error > 0) ? error : ENOSYS;
271 return (-1);
272 }
273 return (rval.sys_rval1);
274 }
275