logo

utils-std

Collection of commonly available Unix tools git clone https://anongit.hacktivis.me/git/utils-std.git/

renice.c (3300B)

    

  1. // utils-std: Collection of commonly available Unix tools
  2. // SPDX-FileCopyrightText: 2017 Haelwenn (lanodan) Monnier <contact+utils@hacktivis.me>
  3. // SPDX-License-Identifier: MPL-2.0
  5. #define _POSIX_C_SOURCE 200809L
  6. #define _XOPEN_SOURCE 700 // getpriority, setpriority
  8. #include <errno.h>
  9. #include <stdio.h>        // fprintf
  10. #include <stdlib.h>       // abort
  11. #include <string.h>       // strerror
  12. #include <sys/resource.h> // getpriority, setpriority
  13. #include <unistd.h>       // getopt, nice
  15. static int
  16. renice(int which, id_t who, int adj)
  17. {
  18. 	const char *which_s = NULL;
  19. 	switch(which)
  20. 	{
  21. 	case PRIO_PROCESS:
  22. 		which_s = "PRIO_PROCESS";
  23. 		break;
  24. 	case PRIO_PGRP:
  25. 		which_s = "PRIO_PGRP";
  26. 		break;
  27. 	case PRIO_USER:
  28. 		which_s = "PRIO_USER";
  29. 		break;
  30. 	default:
  31. 		which_s = NULL;
  32. 	}
  34. 	errno = 0;
  35. 	int prio = getpriority(which, who);
  36. 	if(errno != 0)
  37. 	{
  38. 		fprintf(stderr, "renice: error: getpriority(%s, %d): %s\n", which_s, who, strerror(errno));
  39. 		return -1;
  40. 	}
  42. 	prio += adj;
  44. 	if(setpriority(which, who, prio) != 0)
  45. 	{
  46. 		fprintf(stderr,
  47. 		        "renice: error: setpriority(%s, %d, %d): %s\n",
  48. 		        which_s,
  49. 		        who,
  50. 		        prio,
  51. 		        strerror(errno));
  52. 		return -1;
  53. 	}
  55. 	return 0;
  56. }
  58. static void
  59. usage(void)
  60. {
  61. 	fprintf(stderr, "\
  62. Usage: renice [-p] -n adj PID...\n\
  63.        renice  -g  -n adj PGID...\n\
  64.        renice  -u  -n adj UID...\n\
  65. ");
  66. }
  68. int
  69. main(int argc, char *argv[])
  70. {
  71. 	long adj = 0;
  72. 	int which = PRIO_PROCESS;
  74. 	int c = -1;
  75. 	while((c = getopt(argc, argv, ":gpun:")) != -1)
  76. 	{
  77. 		char *endptr = NULL;
  79. 		switch(c)
  80. 		{
  81. 		case 'g':
  82. 			which = PRIO_PGRP;
  83. 			break;
  84. 		case 'p':
  85. 			which = PRIO_PROCESS;
  86. 			break;
  87. 		case 'u':
  88. 			which = PRIO_USER;
  89. 			break;
  90. 		case 'n':
  91. 			adj = strtol(optarg, &endptr, 10);
  93. 			if(endptr && *endptr != 0) errno = EINVAL;
  94. 			if(errno != 0)
  95. 			{
  96. 				fprintf(stderr,
  97. 				        "renice: error: Failed parsing '%s' as a number: %s\n",
  98. 				        optarg,
  99. 				        strerror(errno));
  100. 				usage();
  101. 				return 1;
  102. 			}
  103. 			if(adj < PRIO_MIN)
  104. 			{
  105. 				fprintf(stderr, "renice: error: '-n %ld' is lower than PRIO_MIN(%d)\n", adj, PRIO_MIN);
  106. 				return 1;
  107. 			}
  108. 			if(adj > PRIO_MAX)
  109. 			{
  110. 				fprintf(stderr, "renice: error: '-n %ld' is greater than PRIO_MAX(%d)\n", adj, PRIO_MAX);
  111. 				return 1;
  112. 			}
  113. 			if(adj == 0)
  114. 			{
  115. 				fprintf(stderr, "renice: error: '-n 0' makes no sense\n");
  116. 				return 1;
  117. 			}
  118. 			break;
  119. 		case ':':
  120. 			fprintf(stderr, "renice: error: Missing operand for option: '-%c'\n", optopt);
  121. 			usage();
  122. 			return 1;
  123. 		case '?':
  124. 			fprintf(stderr, "renice: error: Unrecognised option: '-%c'\n", optopt);
  125. 			usage();
  126. 			return 1;
  127. 		default:
  128. 			abort();
  129. 		}
  130. 	}
  132. 	argc -= optind;
  133. 	argv += optind;
  135. 	if(argc == 0)
  136. 	{
  137. 		fprintf(stderr, "renice: error: No IDs passed\n");
  138. 		usage();
  139. 		return 1;
  140. 	}
  142. 	if(adj == 0)
  143. 	{
  144. 		fprintf(stderr, "renice: error: Mandatory option '-n adj' not passed\n");
  145. 		usage();
  146. 		return 1;
  147. 	}
  149. 	for(int i = 0; i < argc; i++)
  150. 	{
  151. 		char *endptr = NULL;
  152. 		char *arg = argv[i];
  154. 		long who = strtol(arg, &endptr, 0);
  155. 		if(endptr && *endptr != 0) errno = EINVAL;
  157. 		if(errno != 0)
  158. 		{
  159. 			fprintf(stderr,
  160. 			        "renice: error: Failed parsing argument '%s' as a number: %s\n",
  161. 			        arg,
  162. 			        strerror(errno));
  163. 			usage();
  164. 			return 1;
  165. 		}
  167. 		if(renice(which, who, adj) != 0) return 1;
  168. 	}
  170. 	return 0;
  171. }