0d37aa34f8
The new --uid command line option will show only the tasks for a given user, using the proc interface to figure out the existing tasks. Kernel work is needed to close races at startup, but this should already be useful in many use cases. Cc: David Ahern <dsahern@gmail.com> Cc: Frederic Weisbecker <fweisbec@gmail.com> Cc: Mike Galbraith <efault@gmx.de> Cc: Paul Mackerras <paulus@samba.org> Cc: Peter Zijlstra <peterz@infradead.org> Cc: Stephane Eranian <eranian@google.com> Link: http://lkml.kernel.org/n/tip-bdnspm000gw2l984a2t53o8z@git.kernel.org Signed-off-by: Arnaldo Carvalho de Melo <acme@redhat.com>
122 lines
2.4 KiB
C
122 lines
2.4 KiB
C
/*
|
|
* usage.c
|
|
*
|
|
* Various reporting routines.
|
|
* Originally copied from GIT source.
|
|
*
|
|
* Copyright (C) Linus Torvalds, 2005
|
|
*/
|
|
#include "util.h"
|
|
#include "debug.h"
|
|
|
|
static void report(const char *prefix, const char *err, va_list params)
|
|
{
|
|
char msg[1024];
|
|
vsnprintf(msg, sizeof(msg), err, params);
|
|
fprintf(stderr, " %s%s\n", prefix, msg);
|
|
}
|
|
|
|
static NORETURN void usage_builtin(const char *err)
|
|
{
|
|
fprintf(stderr, "\n Usage: %s\n", err);
|
|
exit(129);
|
|
}
|
|
|
|
static NORETURN void die_builtin(const char *err, va_list params)
|
|
{
|
|
report(" Fatal: ", err, params);
|
|
exit(128);
|
|
}
|
|
|
|
static void error_builtin(const char *err, va_list params)
|
|
{
|
|
report(" Error: ", err, params);
|
|
}
|
|
|
|
static void warn_builtin(const char *warn, va_list params)
|
|
{
|
|
report(" Warning: ", warn, params);
|
|
}
|
|
|
|
/* If we are in a dlopen()ed .so write to a global variable would segfault
|
|
* (ugh), so keep things static. */
|
|
static void (*usage_routine)(const char *err) NORETURN = usage_builtin;
|
|
static void (*die_routine)(const char *err, va_list params) NORETURN = die_builtin;
|
|
static void (*error_routine)(const char *err, va_list params) = error_builtin;
|
|
static void (*warn_routine)(const char *err, va_list params) = warn_builtin;
|
|
|
|
void set_die_routine(void (*routine)(const char *err, va_list params) NORETURN)
|
|
{
|
|
die_routine = routine;
|
|
}
|
|
|
|
void usage(const char *err)
|
|
{
|
|
usage_routine(err);
|
|
}
|
|
|
|
void die(const char *err, ...)
|
|
{
|
|
va_list params;
|
|
|
|
va_start(params, err);
|
|
die_routine(err, params);
|
|
va_end(params);
|
|
}
|
|
|
|
int error(const char *err, ...)
|
|
{
|
|
va_list params;
|
|
|
|
va_start(params, err);
|
|
error_routine(err, params);
|
|
va_end(params);
|
|
return -1;
|
|
}
|
|
|
|
void warning(const char *warn, ...)
|
|
{
|
|
va_list params;
|
|
|
|
va_start(params, warn);
|
|
warn_routine(warn, params);
|
|
va_end(params);
|
|
}
|
|
|
|
uid_t parse_target_uid(const char *str, pid_t tid, pid_t pid)
|
|
{
|
|
struct passwd pwd, *result;
|
|
char buf[1024];
|
|
|
|
if (str == NULL)
|
|
return UINT_MAX;
|
|
|
|
/* CPU and PID are mutually exclusive */
|
|
if (tid > 0 || pid > 0) {
|
|
ui__warning("PID/TID switch overriding UID\n");
|
|
sleep(1);
|
|
return UINT_MAX;
|
|
}
|
|
|
|
getpwnam_r(str, &pwd, buf, sizeof(buf), &result);
|
|
|
|
if (result == NULL) {
|
|
char *endptr;
|
|
int uid = strtol(str, &endptr, 10);
|
|
|
|
if (*endptr != '\0') {
|
|
ui__error("Invalid user %s\n", str);
|
|
return UINT_MAX - 1;
|
|
}
|
|
|
|
getpwuid_r(uid, &pwd, buf, sizeof(buf), &result);
|
|
|
|
if (result == NULL) {
|
|
ui__error("Problems obtaining information for user %s\n",
|
|
str);
|
|
return UINT_MAX - 1;
|
|
}
|
|
}
|
|
|
|
return result->pw_uid;
|
|
}
|