123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345 |
- #include <sched.h>
- #include "libbb.h"
- typedef unsigned long ul;
- #define SZOF_UL (unsigned)(sizeof(ul))
- #define BITS_UL (unsigned)(sizeof(ul)*8)
- #define MASK_UL (unsigned)(sizeof(ul)*8 - 1)
- #if ENABLE_FEATURE_TASKSET_FANCY
- #define TASKSET_PRINTF_MASK "%s"
- static char *from_mask(const ul *mask, unsigned sz_in_bytes)
- {
- char *str = xzalloc((sz_in_bytes+1) * 2);
- char *p = str;
- for (;;) {
- ul v = *mask++;
- if (SZOF_UL == 4)
- p += sprintf(p, "%08lx", v);
- if (SZOF_UL == 8)
- p += sprintf(p, "%016lx", v);
- if (SZOF_UL == 16)
- p += sprintf(p, "%032lx", v);
- sz_in_bytes -= SZOF_UL;
- if ((int)sz_in_bytes <= 0)
- break;
- }
- while (str[0] == '0' && str[1])
- str++;
- return str;
- }
- #else
- #define TASKSET_PRINTF_MASK "%lx"
- static unsigned long from_mask(ul *mask, unsigned sz_in_bytes UNUSED_PARAM)
- {
- return *mask;
- }
- #endif
- #if ENABLE_FEATURE_TASKSET_CPULIST
- static void parse_cpulist(ul *mask, unsigned max, char *s)
- {
- char *aff = s;
- for (;;) {
- unsigned bit, end;
- unsigned stride = 1;
- bit = end = bb_strtou(s, &s, 10);
- if (*s == '-') {
- s++;
- end = bb_strtou(s, &s, 10);
- if (*s == ':') {
- s++;
- stride = bb_strtou(s, &s, 10);
- }
- }
- if ((*s != ',' && *s != '\0')
- || bit > end
- || end == UINT_MAX
- || (stride - 1) > (UINT_MAX / 4)
-
- ) {
- bb_error_msg_and_die("bad affinity '%s'", aff);
- }
- while (bit <= end && bit < max) {
- mask[bit / BITS_UL] |= (1UL << (bit & MASK_UL));
- bit += stride;
- }
- if (*s == '\0')
- break;
- s++;
- }
- }
- static void print_cpulist(const ul *mask, unsigned mask_size_in_bytes)
- {
- const ul *mask_end;
- const char *delim;
- unsigned pos;
- ul bit;
- mask_end = mask + mask_size_in_bytes / sizeof(mask[0]);
- delim = "";
- pos = 0;
- bit = 1;
- for (;;) {
- if (*mask & bit) {
- unsigned onebit = pos + 1;
- printf("%s%u", delim, pos);
- do {
- pos++;
- bit <<= 1;
- if (bit == 0) {
- mask++;
- if (mask >= mask_end)
- break;
- bit = 1;
- }
- } while (*mask & bit);
- if (onebit != pos)
- printf("-%u", pos - 1);
- delim = ",";
- }
- pos++;
- bit <<= 1;
- if (bit == 0) {
- mask++;
- if (mask >= mask_end)
- break;
- bit = 1;
- }
- }
- bb_putchar('\n');
- }
- #endif
- enum {
- OPT_p = 1 << 0,
- OPT_a = 1 << 1,
- OPT_c = (1 << 2) * ENABLE_FEATURE_TASKSET_CPULIST,
- };
- static int process_pid_str(const char *pid_str, unsigned opts, char *aff)
- {
- ul *mask;
- unsigned mask_size_in_bytes;
- const char *current_new;
- pid_t pid = !pid_str ? 0 : xatou_range(pid_str, 1, INT_MAX);
- mask_size_in_bytes = SZOF_UL;
- current_new = "current";
- print_aff:
- mask = get_malloc_cpu_affinity(pid, &mask_size_in_bytes);
- if (opts & OPT_p) {
- #if ENABLE_FEATURE_TASKSET_CPULIST
- if (opts & OPT_c) {
- printf("pid %d's %s affinity list: ", pid, current_new);
- print_cpulist(mask, mask_size_in_bytes);
- } else
- #endif
- printf("pid %d's %s affinity mask: "TASKSET_PRINTF_MASK"\n",
- pid, current_new, from_mask(mask, mask_size_in_bytes));
- if (!aff) {
-
- return 0;
- }
- current_new = "new";
- }
- memset(mask, 0, mask_size_in_bytes);
- if (!ENABLE_FEATURE_TASKSET_FANCY) {
-
-
- if (aff[0] == '0' && (aff[1]|0x20) == 'x')
- aff += 2;
- mask[0] = xstrtoul(aff, 16);
- }
- #if ENABLE_FEATURE_TASKSET_CPULIST
- else if (opts & OPT_c) {
- parse_cpulist(mask, mask_size_in_bytes * 8, aff);
- }
- #endif
- else {
- unsigned i;
- char *last_char;
-
-
- if (aff[0] == '0' && (aff[1]|0x20) == 'x')
- aff += 2;
- i = 0;
-
- last_char = strchrnul(aff, '\0');
- while (last_char > aff) {
- char c;
- ul val;
- last_char--;
- c = *last_char;
- if (isdigit(c))
- val = c - '0';
- else if ((c|0x20) >= 'a' && (c|0x20) <= 'f')
- val = (c|0x20) - ('a' - 10);
- else
- bb_error_msg_and_die("bad affinity '%s'", aff);
- if (i < mask_size_in_bytes * 8) {
- mask[i / BITS_UL] |= val << (i & MASK_UL);
-
- }
-
- i += 4;
- }
- }
-
- if (sched_setaffinity(pid, mask_size_in_bytes, (void*)mask))
- bb_perror_msg_and_die("can't %cet pid %d's affinity", 's', pid);
-
- if ((opts & OPT_p) && aff) {
- aff = NULL;
- goto print_aff;
- }
- return 0;
- }
- static int FAST_FUNC iter(const char *dn UNUSED_PARAM, struct dirent *ent, void *aff)
- {
- if (isdigit(ent->d_name[0]))
- return process_pid_str(ent->d_name, option_mask32, aff);
- return 0;
- }
- int taskset_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
- int taskset_main(int argc UNUSED_PARAM, char **argv)
- {
- const char *pid_str;
- char *aff;
- unsigned opts;
-
- opts = getopt32(argv, "^+" "pa"IF_FEATURE_TASKSET_CPULIST("c")
- "\0" "-1" );
- argv += optind;
- aff = *argv++;
- if (!(opts & OPT_p)) {
-
- if (!*argv)
- bb_show_usage();
- process_pid_str(NULL, opts, aff);
- BB_EXECVP_or_die(argv);
- }
- pid_str = aff;
- if (*argv)
- pid_str = *argv;
- else
- aff = NULL;
- if (opts & OPT_a) {
- char *dn;
- int r;
- dn = xasprintf("/proc/%s/task", pid_str);
- r = iterate_on_dir(dn, iter, aff);
- IF_FEATURE_CLEAN_UP(free(dn);)
- if (r == 0)
- return r;
-
- }
- return process_pid_str(pid_str, opts, aff);
- }
|