+++ /dev/null
-/*
- * Copyright (C) 2011 Davidlohr Bueso <dave@gnu.org>
- * Copyright (C) 2011-2020 Karel Zak <kzak@redhat.com>
- *
- * procutils.c: General purpose procfs parsing utilities
- *
- * No copyright is claimed. This code is in the public domain; do with
- * it what you wish.
- */
-
-#include <stdio.h>
-#include <stdlib.h>
-#include <string.h>
-#include <errno.h>
-#include <sys/stat.h>
-#include <sys/vfs.h>
-#include <sys/types.h>
-#include <dirent.h>
-#include <ctype.h>
-
-#include "procutils.h"
-#include "statfs_magic.h"
-#include "fileutils.h"
-#include "all-io.h"
-#include "c.h"
-
-/*
- * @pid: process ID for which we want to obtain the threads group
- *
- * Returns: newly allocated tasks structure
- */
-struct proc_tasks *proc_open_tasks(pid_t pid)
-{
- struct proc_tasks *tasks;
- char path[PATH_MAX];
-
- snprintf(path, sizeof(path), "/proc/%d/task/", pid);
-
- tasks = malloc(sizeof(struct proc_tasks));
- if (tasks) {
- tasks->dir = opendir(path);
- if (tasks->dir)
- return tasks;
- }
-
- free(tasks);
- return NULL;
-}
-
-/*
- * @tasks: allocated tasks structure
- *
- * Returns: nothing
- */
-void proc_close_tasks(struct proc_tasks *tasks)
-{
- if (tasks && tasks->dir)
- closedir(tasks->dir);
- free(tasks);
-}
-
-/*
- * @tasks: allocated task structure
- * @tid: [output] one of the thread IDs belonging to the thread group
- * If when an error occurs, it is set to 0.
- *
- * Returns: 0 on success, 1 on end, -1 on failure or no more threads
- */
-int proc_next_tid(struct proc_tasks *tasks, pid_t *tid)
-{
- struct dirent *d;
- char *end;
-
- if (!tasks || !tid)
- return -EINVAL;
-
- *tid = 0;
- errno = 0;
-
- do {
- d = readdir(tasks->dir);
- if (!d)
- return errno ? -1 : 1; /* error or end-of-dir */
-
- if (!isdigit((unsigned char) *d->d_name))
- continue;
- errno = 0;
- *tid = (pid_t) strtol(d->d_name, &end, 10);
- if (errno || d->d_name == end || (end && *end))
- return -1;
-
- } while (!*tid);
-
- return 0;
-}
-
-/* returns process command path, use free() for result */
-static char *proc_file_strdup(pid_t pid, const char *name)
-{
- char buf[BUFSIZ], *res = NULL;
- ssize_t sz = 0;
- size_t i;
- int fd;
-
- snprintf(buf, sizeof(buf), "/proc/%d/%s", (int) pid, name);
- fd = open(buf, O_RDONLY);
- if (fd < 0)
- goto done;
-
- sz = read_all(fd, buf, sizeof(buf));
- if (sz <= 0)
- goto done;
-
- for (i = 0; i < (size_t) sz; i++) {
-
- if (buf[i] == '\0')
- buf[i] = ' ';
- }
- buf[sz - 1] = '\0';
- res = strdup(buf);
-done:
- if (fd >= 0)
- close(fd);
- return res;
-}
-
-/* returns process command path, use free() for result */
-char *proc_get_command(pid_t pid)
-{
- return proc_file_strdup(pid, "cmdline");
-}
-
-/* returns process command name, use free() for result */
-char *proc_get_command_name(pid_t pid)
-{
- return proc_file_strdup(pid, "comm");
-}
-
-struct proc_processes *proc_open_processes(void)
-{
- struct proc_processes *ps;
-
- ps = calloc(1, sizeof(struct proc_processes));
- if (ps) {
- ps->dir = opendir("/proc");
- if (ps->dir)
- return ps;
- }
-
- free(ps);
- return NULL;
-}
-
-void proc_close_processes(struct proc_processes *ps)
-{
- if (ps && ps->dir)
- closedir(ps->dir);
- free(ps);
-}
-
-void proc_processes_filter_by_name(struct proc_processes *ps, const char *name)
-{
- ps->fltr_name = name;
- ps->has_fltr_name = name ? 1 : 0;
-}
-
-void proc_processes_filter_by_uid(struct proc_processes *ps, uid_t uid)
-{
- ps->fltr_uid = uid;
- ps->has_fltr_uid = 1;
-}
-
-int proc_next_pid(struct proc_processes *ps, pid_t *pid)
-{
- struct dirent *d;
-
- if (!ps || !pid)
- return -EINVAL;
-
- *pid = 0;
- errno = 0;
-
- do {
- char buf[BUFSIZ], *p;
-
- errno = 0;
- d = readdir(ps->dir);
- if (!d)
- return errno ? -1 : 1; /* error or end-of-dir */
-
-
- if (!isdigit((unsigned char) *d->d_name))
- continue;
-
- /* filter out by UID */
- if (ps->has_fltr_uid) {
- struct stat st;
-
- if (fstatat(dirfd(ps->dir), d->d_name, &st, 0))
- continue;
- if (ps->fltr_uid != st.st_uid)
- continue;
- }
-
- /* filter out by NAME */
- if (ps->has_fltr_name) {
- char procname[256];
- FILE *f;
-
- snprintf(buf, sizeof(buf), "%s/stat", d->d_name);
- f = fopen_at(dirfd(ps->dir), buf, O_CLOEXEC|O_RDONLY, "r");
- if (!f)
- continue;
-
- p = fgets(buf, sizeof(buf), f);
- fclose(f);
- if (!p)
- continue;
-
- if (sscanf(buf, "%*d (%255[^)])", procname) != 1)
- continue;
-
- /* ok, we got the process name. */
- if (strcmp(procname, ps->fltr_name) != 0)
- continue;
- }
-
- p = NULL;
- errno = 0;
- *pid = (pid_t) strtol(d->d_name, &p, 10);
- if (errno || d->d_name == p || (p && *p))
- return errno ? -errno : -1;
-
- return 0;
- } while (1);
-
- return 0;
-}
-
-/* checks if fd is file in a procfs;
- * returns 1 if true, 0 if false or couldn't determine */
-int proc_is_procfs(int fd)
-{
- struct statfs st;
- int ret;
-
- do {
- errno = 0;
- ret = fstatfs(fd, &st);
-
- if (ret < 0) {
- if (errno != EINTR && errno != EAGAIN)
- return 0;
- xusleep(250000);
- }
- } while (ret != 0);
-
- return st.f_type == STATFS_PROC_MAGIC;
-}
-
-#ifdef TEST_PROGRAM_PROCUTILS
-
-static int test_tasks(int argc, char *argv[])
-{
- pid_t tid, pid;
- struct proc_tasks *ts;
-
- if (argc != 2)
- return EXIT_FAILURE;
-
- pid = strtol(argv[1], (char **) NULL, 10);
- printf("PID=%d, TIDs:", pid);
-
- ts = proc_open_tasks(pid);
- if (!ts)
- err(EXIT_FAILURE, "open list of tasks failed");
-
- while (proc_next_tid(ts, &tid) == 0)
- printf(" %d", tid);
-
- printf("\n");
- proc_close_tasks(ts);
- return EXIT_SUCCESS;
-}
-
-static int test_processes(int argc, char *argv[])
-{
- pid_t pid;
- struct proc_processes *ps;
-
- ps = proc_open_processes();
- if (!ps)
- err(EXIT_FAILURE, "open list of processes failed");
-
- if (argc >= 3 && strcmp(argv[1], "--name") == 0)
- proc_processes_filter_by_name(ps, argv[2]);
-
- if (argc >= 3 && strcmp(argv[1], "--uid") == 0)
- proc_processes_filter_by_uid(ps, (uid_t) atol(argv[2]));
-
- while (proc_next_pid(ps, &pid) == 0)
- printf(" %d", pid);
-
- printf("\n");
- proc_close_processes(ps);
- return EXIT_SUCCESS;
-}
-
-static int test_isprocfs(int argc, char *argv[])
-{
- const char *name = argc > 1 ? argv[1] : "/proc";
- int fd = open(name, O_RDONLY);
- int is = 0;
-
- if (fd >= 0) {
- is = proc_is_procfs(fd);
- close(fd);
- } else
- err(EXIT_FAILURE, "cannot open %s", name);
-
- printf("%s: %s procfs\n", name, is ? "is" : "is NOT");
- return is ? EXIT_SUCCESS : EXIT_FAILURE;
-}
-
-
-int main(int argc, char *argv[])
-{
- if (argc < 2) {
- fprintf(stderr, "usage: %1$s --tasks <pid>\n"
- " %1$s --is-procfs [<dir>]\n"
- " %1$s --processes [---name <name>] [--uid <uid>]\n",
- program_invocation_short_name);
- return EXIT_FAILURE;
- }
-
- if (strcmp(argv[1], "--tasks") == 0)
- return test_tasks(argc - 1, argv + 1);
- if (strcmp(argv[1], "--processes") == 0)
- return test_processes(argc - 1, argv + 1);
- if (strcmp(argv[1], "--is-procfs") == 0)
- return test_isprocfs(argc - 1, argv + 1);
-
- return EXIT_FAILURE;
-}
-#endif /* TEST_PROGRAM_PROCUTILS */
projects / thirdparty / util-linux.git / commitdiff
