diff options
author | Denys Vlasenko <vda.linux@googlemail.com> | 2021-06-24 12:02:17 +0200 |
---|---|---|
committer | Denys Vlasenko <vda.linux@googlemail.com> | 2021-06-24 12:02:17 +0200 |
commit | 6d61eb13635ac1121762f55376989af3d6c6ebad (patch) | |
tree | 613a092df0c92f00c04a90e2b88d50b95e66367d /util-linux | |
parent | 1746218beebc7e180f3eaed905277f9f46983ac4 (diff) |
taskset: use iterate_on_dir()
function old new delta
iter - 27 +27
process_pid_str 854 856 +2
taskset_main 181 135 -46
------------------------------------------------------------------------------
(add/remove: 1/0 grow/shrink: 1/1 up/down: 29/-46) Total: -17 bytes
Signed-off-by: Denys Vlasenko <vda.linux@googlemail.com>
Diffstat (limited to 'util-linux')
-rw-r--r-- | util-linux/taskset.c | 35 |
1 files changed, 17 insertions, 18 deletions
diff --git a/util-linux/taskset.c b/util-linux/taskset.c index 47e989792..afe2f04d2 100644 --- a/util-linux/taskset.c +++ b/util-linux/taskset.c @@ -213,7 +213,7 @@ enum { OPT_c = (1 << 2) * ENABLE_FEATURE_TASKSET_CPULIST, }; -static void process_pid_str(const char *pid_str, unsigned opts, char *aff) +static int process_pid_str(const char *pid_str, unsigned opts, char *aff) { ul *mask; unsigned mask_size_in_bytes; @@ -237,7 +237,7 @@ static void process_pid_str(const char *pid_str, unsigned opts, char *aff) /* Either it was just "-p <pid>", * or it was "-p <aff> <pid>" and we came here * for the second time (see goto below) */ - return; + return 0; } current_new = "new"; } @@ -312,6 +312,14 @@ static void process_pid_str(const char *pid_str, unsigned opts, char *aff) aff = NULL; goto print_aff; /* print new affinity and exit */ } + 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; @@ -341,29 +349,20 @@ int taskset_main(int argc UNUSED_PARAM, char **argv) pid_str = aff; if (*argv) /* "-p <aff> <pid> ...rest.is.ignored..." */ - pid_str = *argv; /* NB: *argv != NULL in this case */ + pid_str = *argv; else aff = NULL; if (opts & OPT_a) { char *dn; - DIR *dir; - struct dirent *ent; + int r; dn = xasprintf("/proc/%s/task", pid_str); - dir = opendir(dn); + r = iterate_on_dir(dn, iter, aff); IF_FEATURE_CLEAN_UP(free(dn);) - if (!dir) { - goto no_threads; - } - while ((ent = readdir(dir)) != NULL) { - if (isdigit(ent->d_name[0])) - process_pid_str(ent->d_name, opts, aff); - } - IF_FEATURE_CLEAN_UP(closedir(dir);) - } else { - no_threads: - process_pid_str(pid_str, opts, aff); + if (r == 0) + return r; /* EXIT_SUCCESS */ + /* else: no /proc/PID/task, act as if no -a was given */ } - return EXIT_SUCCESS; + return process_pid_str(pid_str, opts, aff); } |