mirror of https://gitee.com/openkylin/linux.git
123 lines
2.4 KiB
C
123 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, const char *tid, const char *pid)
|
|
{
|
|
struct passwd pwd, *result;
|
|
char buf[1024];
|
|
|
|
if (str == NULL)
|
|
return UINT_MAX;
|
|
|
|
/* UID and PID are mutually exclusive */
|
|
if (tid || pid) {
|
|
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;
|
|
}
|