libbb/bb_pwd.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * password utility routines.
   4  *
   5  * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
   6  * Copyright (C) 2008 by Tito Ragusa <farmatito@tiscali.it>
   7  *
   8  * Licensed under GPLv2 or later, see file LICENSE in this source tree.
   9  */
  10
  11 #include "libbb.h"
  12
  13 /* TODO: maybe change API to return malloced data?
  14  * This will allow to stop using libc functions returning
  15  * pointers to static data (getpwuid)
  16  */
  17
  18 struct passwd* FAST_FUNC xgetpwnam(const char *name)
  19 {
  20         struct passwd *pw = getpwnam(name);
  21         if (!pw)
  22                 bb_error_msg_and_die("unknown user %s", name);
  23         return pw;
  24 }
  25
  26 struct group* FAST_FUNC xgetgrnam(const char *name)
  27 {
  28         struct group *gr = getgrnam(name);
  29         if (!gr)
  30                 bb_error_msg_and_die("unknown group %s", name);
  31         return gr;
  32 }
  33
  34
  35 struct passwd* FAST_FUNC xgetpwuid(uid_t uid)
  36 {
  37         struct passwd *pw = getpwuid(uid);
  38         if (!pw)
  39                 bb_error_msg_and_die("unknown uid %u", (unsigned)uid);
  40         return pw;
  41 }
  42
  43 struct group* FAST_FUNC xgetgrgid(gid_t gid)
  44 {
  45         struct group *gr = getgrgid(gid);
  46         if (!gr)
  47                 bb_error_msg_and_die("unknown gid %u", (unsigned)gid);
  48         return gr;
  49 }
  50
  51 char* FAST_FUNC xuid2uname(uid_t uid)
  52 {
  53         struct passwd *pw = xgetpwuid(uid);
  54         return pw->pw_name;
  55 }
  56
  57 char* FAST_FUNC xgid2group(gid_t gid)
  58 {
  59         struct group *gr = xgetgrgid(gid);
  60         return gr->gr_name;
  61 }
  62
  63 char* FAST_FUNC uid2uname(uid_t uid)
  64 {
  65         struct passwd *pw = getpwuid(uid);
  66         return (pw) ? pw->pw_name : NULL;
  67 }
  68
  69 char* FAST_FUNC gid2group(gid_t gid)
  70 {
  71         struct group *gr = getgrgid(gid);
  72         return (gr) ? gr->gr_name : NULL;
  73 }
  74
  75 char* FAST_FUNC uid2uname_utoa(uid_t uid)
  76 {
  77         char *name = uid2uname(uid);
  78         return (name) ? name : utoa(uid);
  79 }
  80
  81 char* FAST_FUNC gid2group_utoa(gid_t gid)
  82 {
  83         char *name = gid2group(gid);
  84         return (name) ? name : utoa(gid);
  85 }
  86
  87 long FAST_FUNC xuname2uid(const char *name)
  88 {
  89         struct passwd *myuser;
  90
  91         myuser = xgetpwnam(name);
  92         return myuser->pw_uid;
  93 }
  94
  95 long FAST_FUNC xgroup2gid(const char *name)
  96 {
  97         struct group *mygroup;
  98
  99         mygroup = xgetgrnam(name);
 100         return mygroup->gr_gid;
 101 }
 102
 103 unsigned long FAST_FUNC get_ug_id(const char *s,
 104                 long FAST_FUNC (*xname2id)(const char *))
 105 {
 106         unsigned long r;
 107
 108         r = bb_strtoul(s, NULL, 10);
 109         if (errno)
 110                 return xname2id(s);
 111         return r;
 112 }
 113
 114 /* Experimental "mallocing" API.
 115  * The goal is nice: "we want to support a case when "guests" group is very large"
 116  * but the code is butt-ugly.
 117  */
 118 #if 0
 119 static char *find_latest(char last, char *cp)
 120 {
 121         if (!cp)
 122                 return last;
 123         cp += strlen(cp) + 1;
 124         if (last < cp)
 125                 last = cp;
 126         return last;
 127 }
 128
 129 struct group* FAST_FUNC xmalloc_getgrnam(const char *name)
 130 {
 131         struct {
 132                 struct group gr;
 133                 // May still be not enough!
 134                 char buf[64*1024 - sizeof(struct group) - 16];
 135         } *s;
 136         struct group *grp;
 137         int r;
 138         char *last;
 139         char **gr_mem;
 140
 141         s = xmalloc(sizeof(*s));
 142         r = getgrnam_r(name, &s->gr, s->buf, sizeof(s->buf), &grp);
 143         if (!grp) {
 144                 free(s);
 145                 return grp;
 146         }
 147         last = find_latest(s->buf, grp->gr_name);
 148         last = find_latest(last, grp->gr_passwd);
 149         gr_mem = grp->gr_mem;
 150         while (*gr_mem)
 151                 last = find_latest(last, *gr_mem++);
 152         gr_mem++; /* points past NULL */
 153         if (last < (char*)gr_mem)
 154                 last = (char*)gr_mem;
 155 //FIXME: what if we get not only truncated, but also moved here?
 156 // grp->gr_name pointer and friends are invalid now!!!
 157         s = xrealloc(s, last - (char*)s);
 158         return grp;
 159 }
 160 #endif