int main(int argc, char * argv[])
{
int ch, ret;
bool flagx = false, flagX = false, flagb = false;
int flagd = -1, flagg = -1;
struct task_qos_policy qosinfo = { LATENCY_QOS_TIER_UNSPECIFIED, THROUGHPUT_QOS_TIER_UNSPECIFIED };
while ((ch = getopt(argc, argv, "xXbd:g:t:l:")) != -1) {
switch (ch) {
case 'x':
flagx = true;
break;
case 'X':
flagX = true;
break;
case 'b':
flagb = true;
break;
case 'd':
flagd = parse_disk_policy(optarg);
if (flagd == -1) {
warnx("Could not parse '%s' as a disk policy", optarg);
usage();
}
break;
case 'g':
flagg = parse_disk_policy(optarg);
if (flagg == -1) {
warnx("Could not parse '%s' as a disk policy", optarg);
usage();
}
break;
case 't':
qosinfo.task_throughput_qos_tier = parse_qos_tier(optarg, QOS_PARAMETER_THROUGHPUT);
if (qosinfo.task_throughput_qos_tier == -1) {
warnx("Could not parse '%s' as a qos tier", optarg);
usage();
}
break;
case 'l':
qosinfo.task_latency_qos_tier = parse_qos_tier(optarg, QOS_PARAMETER_LATENCY);
if (qosinfo.task_latency_qos_tier == -1) {
warnx("Could not parse '%s' as a qos tier", optarg);
usage();
}
break;
case '?':
default:
usage();
}
}
argc -= optind;
argv += optind;
if (argc == 0) {
usage();
}
if (flagx && flagX){
warnx("Incompatible options -x, -X");
usage();
}
if (flagx) {
ret = setiopolicy_np(IOPOL_TYPE_VFS_HFS_CASE_SENSITIVITY, IOPOL_SCOPE_PROCESS, IOPOL_VFS_HFS_CASE_SENSITIVITY_FORCE_CASE_SENSITIVE);
if (ret == -1) {
err(EX_SOFTWARE, "setiopolicy_np(IOPOL_TYPE_VFS_HFS_CASE_SENSITIVITY...)");
}
}
if (flagX) {
ret = setiopolicy_np(IOPOL_TYPE_VFS_HFS_CASE_SENSITIVITY, IOPOL_SCOPE_PROCESS, IOPOL_VFS_HFS_CASE_SENSITIVITY_DEFAULT);
if (ret == -1) {
err(EX_SOFTWARE, "setiopolicy_np(IOPOL_TYPE_VFS_HFS_CASE_SENSITIVITY...)");
}
}
if (flagb) {
ret = setpriority(PRIO_DARWIN_PROCESS, 0, PRIO_DARWIN_BG);
if (ret == -1) {
err(EX_SOFTWARE, "setpriority()");
}
}
if (flagd >= 0) {
ret = setiopolicy_np(IOPOL_TYPE_DISK, IOPOL_SCOPE_PROCESS, flagd);
if (ret == -1) {
err(EX_SOFTWARE, "setiopolicy_np(...IOPOL_SCOPE_PROCESS...)");
}
}
if (flagg >= 0){
ret = setiopolicy_np(IOPOL_TYPE_DISK, IOPOL_SCOPE_DARWIN_BG, flagg);
if (ret == -1) {
err(EX_SOFTWARE, "setiopolicy_np(...IOPOL_SCOPE_DARWIN_BG...)");
}
}
if (qosinfo.task_latency_qos_tier != LATENCY_QOS_TIER_UNSPECIFIED ||
qosinfo.task_throughput_qos_tier != THROUGHPUT_QOS_TIER_UNSPECIFIED){
ret = task_policy_set(mach_task_self(), TASK_OVERRIDE_QOS_POLICY, (task_policy_t)&qosinfo, TASK_QOS_POLICY_COUNT);
if (ret != KERN_SUCCESS){
err(EX_SOFTWARE, "task_policy_set(...TASK_OVERRIDE_QOS_POLICY...)");
}
}
ret = execvp(argv[0], argv);
if (ret == -1) {
err(EX_NOINPUT, "Could not execute %s", argv[0]);
}
return EX_OSERR;
}
int main(int argc, char * argv[])
{
int ch, ret;
pid_t pid = 0;
posix_spawnattr_t attr;
extern char **environ;
bool flagx = false, flagX = false, flagb = false, flagB = false;
int flagd = -1, flagg = -1;
struct task_qos_policy qosinfo = { LATENCY_QOS_TIER_UNSPECIFIED, THROUGHPUT_QOS_TIER_UNSPECIFIED };
uint64_t qos_clamp = POSIX_SPAWN_PROC_CLAMP_NONE;
while ((ch = getopt(argc, argv, "xXbBd:g:c:t:l:p:")) != -1) {
switch (ch) {
case 'x':
flagx = true;
break;
case 'X':
flagX = true;
break;
case 'b':
flagb = true;
break;
case 'B':
flagB = true;
break;
case 'd':
flagd = parse_disk_policy(optarg);
if (flagd == -1) {
warnx("Could not parse '%s' as a disk policy", optarg);
usage();
}
break;
case 'g':
flagg = parse_disk_policy(optarg);
if (flagg == -1) {
warnx("Could not parse '%s' as a disk policy", optarg);
usage();
}
break;
case 'c':
qos_clamp = parse_qos_clamp(optarg);
if (qos_clamp == POSIX_SPAWN_PROC_CLAMP_NONE) {
warnx("Could not parse '%s' as a QoS clamp", optarg);
usage();
}
break;
case 't':
qosinfo.task_throughput_qos_tier = parse_qos_tier(optarg, QOS_PARAMETER_THROUGHPUT);
if (qosinfo.task_throughput_qos_tier == -1) {
warnx("Could not parse '%s' as a qos tier", optarg);
usage();
}
break;
case 'l':
qosinfo.task_latency_qos_tier = parse_qos_tier(optarg, QOS_PARAMETER_LATENCY);
if (qosinfo.task_latency_qos_tier == -1) {
warnx("Could not parse '%s' as a qos tier", optarg);
usage();
}
break;
case 'p':
pid = atoi(optarg);
if (pid == 0) {
warnx("Invalid pid '%s' specified", optarg);
usage();
}
break;
case '?':
default:
usage();
}
}
argc -= optind;
argv += optind;
if (pid == 0 && argc == 0) {
usage();
}
if (pid != 0 && (flagx || flagX || flagg != -1 || flagd != -1)) {
warnx("Incompatible option(s) used with -p");
usage();
}
if (flagx && flagX) {
warnx("Incompatible options -x, -X");
usage();
}
if (flagb && flagB) {
warnx("Incompatible options -b, -B");
usage();
}
if (flagB && pid == 0) {
warnx("The -B option can only be used with the -p option");
usage();
}
if (flagx) {
ret = setiopolicy_np(IOPOL_TYPE_VFS_HFS_CASE_SENSITIVITY, IOPOL_SCOPE_PROCESS, IOPOL_VFS_HFS_CASE_SENSITIVITY_FORCE_CASE_SENSITIVE);
if (ret == -1) {
err(EX_SOFTWARE, "setiopolicy_np(IOPOL_TYPE_VFS_HFS_CASE_SENSITIVITY...)");
}
}
if (flagX) {
ret = setiopolicy_np(IOPOL_TYPE_VFS_HFS_CASE_SENSITIVITY, IOPOL_SCOPE_PROCESS, IOPOL_VFS_HFS_CASE_SENSITIVITY_DEFAULT);
if (ret == -1) {
err(EX_SOFTWARE, "setiopolicy_np(IOPOL_TYPE_VFS_HFS_CASE_SENSITIVITY...)");
}
}
if (flagb) {
ret = setpriority(PRIO_DARWIN_PROCESS, pid, PRIO_DARWIN_BG);
if (ret == -1) {
err(EX_SOFTWARE, "setpriority()");
}
}
if (flagB) {
ret = setpriority(PRIO_DARWIN_PROCESS, pid, 0);
if (ret == -1) {
err(EX_SOFTWARE, "setpriority()");
}
}
if (flagd >= 0) {
ret = setiopolicy_np(IOPOL_TYPE_DISK, IOPOL_SCOPE_PROCESS, flagd);
if (ret == -1) {
err(EX_SOFTWARE, "setiopolicy_np(...IOPOL_SCOPE_PROCESS...)");
}
}
if (flagg >= 0) {
ret = setiopolicy_np(IOPOL_TYPE_DISK, IOPOL_SCOPE_DARWIN_BG, flagg);
if (ret == -1) {
err(EX_SOFTWARE, "setiopolicy_np(...IOPOL_SCOPE_DARWIN_BG...)");
}
}
if (qosinfo.task_latency_qos_tier != LATENCY_QOS_TIER_UNSPECIFIED ||
qosinfo.task_throughput_qos_tier != THROUGHPUT_QOS_TIER_UNSPECIFIED) {
mach_port_t task;
if (pid) {
ret = task_for_pid(mach_task_self(), pid, &task);
if (ret != KERN_SUCCESS) {
err(EX_SOFTWARE, "task_for_pid(%d) failed", pid);
return EX_OSERR;
}
} else {
task = mach_task_self();
}
ret = task_policy_set((task_t)task, TASK_OVERRIDE_QOS_POLICY, (task_policy_t)&qosinfo, TASK_QOS_POLICY_COUNT);
if (ret != KERN_SUCCESS) {
err(EX_SOFTWARE, "task_policy_set(...TASK_OVERRIDE_QOS_POLICY...)");
}
}
if (pid != 0)
return 0;
ret = posix_spawnattr_init(&attr);
if (ret != 0) errc(EX_NOINPUT, ret, "posix_spawnattr_init");
ret = posix_spawnattr_setflags(&attr, POSIX_SPAWN_SETEXEC);
if (ret != 0) errc(EX_NOINPUT, ret, "posix_spawnattr_setflags");
if (qos_clamp != POSIX_SPAWN_PROC_CLAMP_NONE) {
ret = posix_spawnattr_set_qos_clamp_np(&attr, qos_clamp);
if (ret != 0) errc(EX_NOINPUT, ret, "posix_spawnattr_set_qos_clamp_np");
}
ret = posix_spawnp(&pid, argv[0], NULL, &attr, argv, environ);
if (ret != 0) errc(EX_NOINPUT, ret, "posix_spawn");
return EX_OSERR;
}
