renice_u.c

/*
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
 * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
 * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED 
 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
#include <sys/time.h>
#include <sys/resource.h>
#include <sched.h>

static void usage(const char *s) {
	fprintf(stderr, "Usage: %s { [-r] <priority> <pids ...> | -g <pid> }\n", s);
}

void print_prio(pid_t pid) {
	int sched;
	struct sched_param sp;

	printf("pid %d's priority: %d\n", pid, getpriority(PRIO_PROCESS, pid));

	printf("scheduling class: ");
	sched = sched_getscheduler(pid);
	switch(sched) {
		case SCHED_FIFO:
			puts("FIFO");
			break;
		case SCHED_RR:
			puts("RR");
			break;
		case SCHED_OTHER:
			puts("Normal");
			break;
		case -1:
			perror("sched_getscheduler");
			break;
		default:
			puts("Unknown");
	}

	sched_getparam(pid, &sp);
	printf("RT prio: %d (of %d to %d)\n", sp.sched_priority,
		sched_get_priority_min(sched), sched_get_priority_max(sched));
}

int renice_main(int argc, char *argv[]) {
	int prio;
	int realtime = 0;

#if 0
	argc--;
	argv++;

	if(argc < 1) {
		usage(name);
		return -1;
	}

	if(strcmp("-r", argv[0]) == 0) {
		// do realtime priority adjustment
		realtime = 1;
		argc--;
		argv++;
	}

	if(strcmp("-g", argv[0]) == 0) {
		if(argc < 2) {
			usage(name);
			return -1;
		}
		print_prio(atoi(argv[1]));
		return 0;
	}
#else
	while(1) {
		int c = getopt(argc, argv, "rg:h");
		if(c == -1) break;
		switch(c) {
			case 'r':
				realtime = 1;
				break;
			case 'g':
				print_prio(atoi(optarg));
				return 0;
			case 'h':
				usage(argv[0]);
				return 0;
			case '?':
				return -1;
		}
	}
#endif
	//fprintf(stderr, "argc = %d, optind = %d\n", argc, optind);
	if(argc < optind + 2) {
		usage(argv[0]);
		return -1;
	}

	prio = atoi(argv[optind++]);
	argc -= optind;
	argv += optind;

	while(argc) {
		pid_t pid;

		pid = atoi(*argv);
		argc--;
		argv++;

		if(realtime) {
			struct sched_param sp = { .sched_priority = prio };
			int ret = sched_setscheduler(pid, SCHED_RR, &sp);
			if(ret) {
				perror("sched_set_scheduler");
				return EXIT_FAILURE;
			}
		} else {
			int ret = setpriority(PRIO_PROCESS, pid, prio);
			if(ret) {
				perror("setpriority");
				return EXIT_FAILURE;
			}
		}
	}

	return 0;
}