From 067d851d30386c553e3a84f59d81d003ff638b91 Mon Sep 17 00:00:00 2001 From: Daniel Wallace Date: Fri, 7 Dec 2012 00:01:15 -0500 Subject: unit: add %U for uids in units --- src/core/unit-printf.c | 17 +++++++++++++++-- 1 file changed, 15 insertions(+), 2 deletions(-) (limited to 'src/core') diff --git a/src/core/unit-printf.c b/src/core/unit-printf.c index a8eb60a3d4..a58c96c238 100644 --- a/src/core/unit-printf.c +++ b/src/core/unit-printf.c @@ -123,6 +123,8 @@ static char *specifier_user_name(char specifier, void *data, void *userdata) { ExecContext *c; int r; const char *username; + uid_t uid; + char *printed = NULL; assert(u); @@ -134,11 +136,21 @@ static char *specifier_user_name(char specifier, void *data, void *userdata) { /* fish username from passwd */ username = c->user; - r = get_user_creds(&username, NULL, NULL, NULL, NULL); + r = get_user_creds(&username, &uid, NULL, NULL, NULL); if (r < 0) return NULL; - return strdup(username); + switch (specifier) { + case 'U': + if (asprintf(&printed, "%d", uid) < 0) + return NULL; + break; + case 'u': + printed = strdup(username); + break; + } + + return printed; } static char *specifier_user_home(char specifier, void *data, void *userdata) { @@ -292,6 +304,7 @@ char *unit_full_printf(Unit *u, const char *format) { { 'r', specifier_cgroup_root, NULL }, { 'R', specifier_cgroup_root, NULL }, { 't', specifier_runtime, NULL }, + { 'U', specifier_user_name, NULL }, { 'u', specifier_user_name, NULL }, { 'h', specifier_user_home, NULL }, { 's', specifier_user_shell, NULL }, -- cgit v1.2.3-54-g00ecf