]> git.saurik.com Git - apple/system_cmds.git/blame - taskpolicy.tproj/taskpolicy.c
system_cmds-643.1.1.tar.gz
[apple/system_cmds.git] / taskpolicy.tproj / taskpolicy.c
CommitLineData
fc6d9e4b
A
1/*
2 * Copyright (c) 2013 Apple Inc. All rights reserved.
3 *
4 * @APPLE_LICENSE_HEADER_START@
5 *
6 * "Portions Copyright (c) 1999 Apple Computer, Inc. All Rights
7 * Reserved. This file contains Original Code and/or Modifications of
8 * Original Code as defined in and that are subject to the Apple Public
9 * Source License Version 1.0 (the 'License'). You may not use this file
10 * except in compliance with the License. Please obtain a copy of the
11 * License at http://www.apple.com/publicsource and read it before using
12 * this file.
13 *
14 * The Original Code and all software distributed under the License are
15 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
16 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
17 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE OR NON-INFRINGEMENT. Please see the
19 * License for the specific language governing rights and limitations
20 * under the License."
21 *
22 * @APPLE_LICENSE_HEADER_END@
23 */
24
25#include <stdio.h>
26#include <stdlib.h>
27#include <unistd.h>
28#include <string.h>
29#include <sys/resource.h>
30#include <err.h>
31#include <sys/errno.h>
32#include <stdbool.h>
33#include <sysexits.h>
34#include <mach/mach.h>
35#include <mach/task_policy.h>
36
1a7e3f61
A
37#include <spawn.h>
38#include <spawn_private.h>
39#include <sys/spawn_internal.h>
40
fc6d9e4b
A
41#define QOS_PARAMETER_LATENCY 0
42#define QOS_PARAMETER_THROUGHPUT 1
43
1a7e3f61
A
44extern char **environ;
45
fc6d9e4b
A
46static void usage(void);
47static int parse_disk_policy(const char *strpolicy);
48static int parse_qos_tier(const char *strpolicy, int parameter);
1a7e3f61 49static uint64_t parse_qos_clamp(const char *qos_string);
fc6d9e4b
A
50
51int main(int argc, char * argv[])
52{
53 int ch, ret;
1a7e3f61
A
54 pid_t pid = 0;
55 posix_spawnattr_t attr;
56 extern char **environ;
57 bool flagx = false, flagX = false, flagb = false, flagB = false;
fc6d9e4b
A
58 int flagd = -1, flagg = -1;
59 struct task_qos_policy qosinfo = { LATENCY_QOS_TIER_UNSPECIFIED, THROUGHPUT_QOS_TIER_UNSPECIFIED };
1a7e3f61 60 uint64_t qos_clamp = POSIX_SPAWN_PROC_CLAMP_NONE;
fc6d9e4b 61
1a7e3f61 62 while ((ch = getopt(argc, argv, "xXbBd:g:c:t:l:p:")) != -1) {
fc6d9e4b
A
63 switch (ch) {
64 case 'x':
65 flagx = true;
66 break;
67 case 'X':
68 flagX = true;
69 break;
70 case 'b':
71 flagb = true;
72 break;
1a7e3f61
A
73 case 'B':
74 flagB = true;
75 break;
fc6d9e4b
A
76 case 'd':
77 flagd = parse_disk_policy(optarg);
78 if (flagd == -1) {
79 warnx("Could not parse '%s' as a disk policy", optarg);
80 usage();
81 }
82 break;
83 case 'g':
84 flagg = parse_disk_policy(optarg);
85 if (flagg == -1) {
86 warnx("Could not parse '%s' as a disk policy", optarg);
87 usage();
88 }
89 break;
1a7e3f61
A
90 case 'c':
91 qos_clamp = parse_qos_clamp(optarg);
92 if (qos_clamp == POSIX_SPAWN_PROC_CLAMP_NONE) {
93 warnx("Could not parse '%s' as a QoS clamp", optarg);
94 usage();
95 }
96 break;
fc6d9e4b
A
97 case 't':
98 qosinfo.task_throughput_qos_tier = parse_qos_tier(optarg, QOS_PARAMETER_THROUGHPUT);
99 if (qosinfo.task_throughput_qos_tier == -1) {
100 warnx("Could not parse '%s' as a qos tier", optarg);
101 usage();
102 }
103 break;
104 case 'l':
105 qosinfo.task_latency_qos_tier = parse_qos_tier(optarg, QOS_PARAMETER_LATENCY);
106 if (qosinfo.task_latency_qos_tier == -1) {
107 warnx("Could not parse '%s' as a qos tier", optarg);
108 usage();
109 }
110 break;
1a7e3f61
A
111 case 'p':
112 pid = atoi(optarg);
113 if (pid == 0) {
114 warnx("Invalid pid '%s' specified", optarg);
115 usage();
116 }
117 break;
fc6d9e4b
A
118 case '?':
119 default:
120 usage();
121 }
122 }
123 argc -= optind;
124 argv += optind;
125
1a7e3f61
A
126 if (pid == 0 && argc == 0) {
127 usage();
128 }
129
130 if (pid != 0 && (flagx || flagX || flagg != -1 || flagd != -1)) {
131 warnx("Incompatible option(s) used with -p");
fc6d9e4b
A
132 usage();
133 }
134
135 if (flagx && flagX){
136 warnx("Incompatible options -x, -X");
137 usage();
138 }
139
1a7e3f61
A
140 if (flagb && flagB) {
141 warnx("Incompatible options -b, -B");
142 usage();
143 }
144
145 if (flagB && pid == 0) {
146 warnx("The -B option can only be used with the -p option");
147 usage();
148 }
149
fc6d9e4b
A
150 if (flagx) {
151 ret = setiopolicy_np(IOPOL_TYPE_VFS_HFS_CASE_SENSITIVITY, IOPOL_SCOPE_PROCESS, IOPOL_VFS_HFS_CASE_SENSITIVITY_FORCE_CASE_SENSITIVE);
152 if (ret == -1) {
153 err(EX_SOFTWARE, "setiopolicy_np(IOPOL_TYPE_VFS_HFS_CASE_SENSITIVITY...)");
154 }
155 }
156
157 if (flagX) {
158 ret = setiopolicy_np(IOPOL_TYPE_VFS_HFS_CASE_SENSITIVITY, IOPOL_SCOPE_PROCESS, IOPOL_VFS_HFS_CASE_SENSITIVITY_DEFAULT);
159 if (ret == -1) {
160 err(EX_SOFTWARE, "setiopolicy_np(IOPOL_TYPE_VFS_HFS_CASE_SENSITIVITY...)");
161 }
162 }
163
164 if (flagb) {
1a7e3f61
A
165 ret = setpriority(PRIO_DARWIN_PROCESS, pid, PRIO_DARWIN_BG);
166 if (ret == -1) {
167 err(EX_SOFTWARE, "setpriority()");
168 }
169 }
170
171 if (flagB) {
172 ret = setpriority(PRIO_DARWIN_PROCESS, pid, 0);
fc6d9e4b
A
173 if (ret == -1) {
174 err(EX_SOFTWARE, "setpriority()");
175 }
176 }
177
178 if (flagd >= 0) {
179 ret = setiopolicy_np(IOPOL_TYPE_DISK, IOPOL_SCOPE_PROCESS, flagd);
180 if (ret == -1) {
181 err(EX_SOFTWARE, "setiopolicy_np(...IOPOL_SCOPE_PROCESS...)");
182 }
183 }
184
185 if (flagg >= 0){
186 ret = setiopolicy_np(IOPOL_TYPE_DISK, IOPOL_SCOPE_DARWIN_BG, flagg);
187 if (ret == -1) {
188 err(EX_SOFTWARE, "setiopolicy_np(...IOPOL_SCOPE_DARWIN_BG...)");
189 }
190 }
191
192 if (qosinfo.task_latency_qos_tier != LATENCY_QOS_TIER_UNSPECIFIED ||
193 qosinfo.task_throughput_qos_tier != THROUGHPUT_QOS_TIER_UNSPECIFIED){
1a7e3f61
A
194 mach_port_t task;
195 if (pid) {
196 ret = task_for_pid(mach_task_self(), pid, &task);
197 if (ret != KERN_SUCCESS) {
198 err(EX_SOFTWARE, "task_for_pid(%d) failed", pid);
199 return EX_OSERR;
200 }
201 } else {
202 task = mach_task_self();
203 }
204 ret = task_policy_set((task_t)task, TASK_OVERRIDE_QOS_POLICY, (task_policy_t)&qosinfo, TASK_QOS_POLICY_COUNT);
fc6d9e4b
A
205 if (ret != KERN_SUCCESS){
206 err(EX_SOFTWARE, "task_policy_set(...TASK_OVERRIDE_QOS_POLICY...)");
207 }
208 }
209
1a7e3f61
A
210 if (pid != 0)
211 return 0;
212
213
214 ret = posix_spawnattr_init(&attr);
215 if (ret != 0) errc(EX_NOINPUT, ret, "posix_spawnattr_init");
216
217 ret = posix_spawnattr_setflags(&attr, POSIX_SPAWN_SETEXEC);
218 if (ret != 0) errc(EX_NOINPUT, ret, "posix_spawnattr_setflags");
219
220 if (qos_clamp != POSIX_SPAWN_PROC_CLAMP_NONE) {
221 ret = posix_spawnattr_set_qos_clamp_np(&attr, qos_clamp);
222 if (ret != 0) errc(EX_NOINPUT, ret, "posix_spawnattr_set_qos_clamp_np");
223 }
224
225 ret = posix_spawnp(&pid, argv[0], NULL, &attr, argv, environ);
226 if (ret != 0) errc(EX_NOINPUT, ret, "posix_spawn");
227
fc6d9e4b
A
228 return EX_OSERR;
229}
230
231static void usage(void)
232{
1a7e3f61
A
233 fprintf(stderr, "Usage: %s [-x|-X] [-d <policy>] [-g policy] [-c clamp] [-b] [-t <tier>]\n"
234 " [-l <tier>] <program> [<pargs> [...]]\n", getprogname());
235 fprintf(stderr, " %s [-b|-B] [-t <tier>] [-l <tier>] -p pid\n", getprogname());
fc6d9e4b
A
236 exit(EX_USAGE);
237}
238
239static int parse_disk_policy(const char *strpolicy)
240{
241 long policy;
242 char *endptr = NULL;
243
244 /* first try as an integer */
245 policy = strtol(strpolicy, &endptr, 0);
246 if (endptr && (endptr[0] == '\0') && (strpolicy[0] != '\0')) {
247 /* parsed complete string as a number */
248 return (int)policy;
249 }
250
251 if (0 == strcasecmp(strpolicy, "DEFAULT") ) {
252 return IOPOL_DEFAULT;
253 } else if (0 == strcasecmp(strpolicy, "IMPORTANT")) {
254 return IOPOL_IMPORTANT;
255 } else if (0 == strcasecmp(strpolicy, "PASSIVE")) {
256 return IOPOL_PASSIVE;
257 } else if (0 == strcasecmp(strpolicy, "THROTTLE")) {
258 return IOPOL_THROTTLE;
259 } else if (0 == strcasecmp(strpolicy, "UTILITY")) {
260 return IOPOL_UTILITY;
261 } else if (0 == strcasecmp(strpolicy, "STANDARD")) {
262 return IOPOL_STANDARD;
263 } else {
264 return -1;
265 }
266}
267
268static int parse_qos_tier(const char *strtier, int parameter){
269 long policy;
270 char *endptr = NULL;
271
272 /* first try as an integer */
273 policy = strtol(strtier, &endptr, 0);
274 if (endptr && (endptr[0] == '\0') && (strtier[0] != '\0')) {
275 switch (policy) {
276 case 0:
277 return parameter ? THROUGHPUT_QOS_TIER_0 : LATENCY_QOS_TIER_0;
278 break;
279 case 1:
280 return parameter ? THROUGHPUT_QOS_TIER_1 : LATENCY_QOS_TIER_1;
281 break;
282 case 2:
283 return parameter ? THROUGHPUT_QOS_TIER_2 : LATENCY_QOS_TIER_2;
284 break;
285 case 3:
286 return parameter ? THROUGHPUT_QOS_TIER_3 : LATENCY_QOS_TIER_3;
287 break;
288 case 4:
289 return parameter ? THROUGHPUT_QOS_TIER_4 : LATENCY_QOS_TIER_4;
290 break;
291 case 5:
292 return parameter ? THROUGHPUT_QOS_TIER_5 : LATENCY_QOS_TIER_5;
293 break;
294 default:
295 return -1;
296 break;
297 }
298 }
299
300 return -1;
301}
1a7e3f61
A
302
303static uint64_t parse_qos_clamp(const char *qos_string) {
304
305 if (0 == strcasecmp(qos_string, "utility") ) {
306 return POSIX_SPAWN_PROC_CLAMP_UTILITY;
307 } else if (0 == strcasecmp(qos_string, "background")) {
308 return POSIX_SPAWN_PROC_CLAMP_BACKGROUND;
309 } else if (0 == strcasecmp(qos_string, "maintenance")) {
310 return POSIX_SPAWN_PROC_CLAMP_MAINTENANCE;
311 } else {
312 return POSIX_SPAWN_PROC_CLAMP_NONE;
313 }
314}