1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
/* vi: set sw=4 ts=4:
*
* id.c - print real and effective user and group IDs
*
* Copyright 2012 Sony Network Entertainment, Inc.
*
* by Tim Bird <tim.bird@am.sony.com>
*
* See http://www.opengroup.org/onlinepubs/009695399/utilities/id.html
USE_ID(NEWTOY(id, "nGgru", TOYFLAG_BIN))
config ID
bool "id"
default y
help
usage: id [-nGgru]
Print user and group ID.
-n print names instead of numeric IDs (to be used with -Ggu)
-G Show only the group IDs
-g Show only the effective group ID
-r Show real ID instead of effective ID
-u Show only the effective user ID
*/
#include "toys.h"
#define FLAG_n (1<<4)
#define FLAG_G (1<<3)
#define FLAG_g (1<<2)
#define FLAG_r (1<<1)
#define FLAG_u 1
void pretty_print(struct passwd *pw, struct group *grp, struct group **grps,
int n)
{
int i = 0;
printf("uid=%u(%s) gid=%u(%s)", pw->pw_uid, pw->pw_name,
grp->gr_gid, grp->gr_name);
if (n) printf(" groups=");
while (i < n) {
printf("%d(%s)", grps[i]->gr_gid, grps[i]->gr_name);
if (++i<n) xputc(',');
}
xputc('\n');
}
void id_main(void)
{
int flags = toys.optflags, i, ngroups;
struct passwd *pw;
struct group *grp, **grps;
uid_t uid;
gid_t gid, *groups;
/* check if a username is given */
if (*toys.optargs) {
if (!(pw = getpwnam(*toys.optargs)))
error_exit("no such user '%s'", *toys.optargs);
uid = pw->pw_uid;
gid = pw->pw_gid;
} else {
/* show effective, unless user specifies real */
if (flags & FLAG_r) {
uid = getuid();
gid = getgid();
} else {
uid = geteuid();
gid = getegid();
}
}
if (!(pw = getpwuid(uid)) || !(grp = getgrgid(gid)))
perror_exit(0);
if (flags & FLAG_u) {
if (flags & FLAG_n) xputs(pw->pw_name);
else printf("%d\n", pw->pw_uid);
return;
}
if (flags & FLAG_g) {
if (flags & FLAG_n) xputs(grp->gr_name);
else printf("%d\n", grp->gr_gid);
return;
}
ngroups = sysconf(_SC_NGROUPS_MAX);
if (ngroups<1) ngroups = 32;
groups = xmalloc(ngroups * sizeof(gid_t));
if (getgrouplist(pw->pw_name, pw->pw_gid, groups, &ngroups) < 0)
perror_exit(0);
grps = xmalloc(ngroups * sizeof(struct group *));
for (i = 0; i < ngroups; i++) {
struct group *tmp;
grps[i] = xmalloc(sizeof(struct group));
size_t f = sysconf(_SC_GETGR_R_SIZE_MAX);
char *buf = xmalloc(f);
if (getgrgid_r(groups[i], grps[i], buf, f, &tmp) < 0 || !tmp) {
perror_msg(0);
continue;
}
}
if (flags & FLAG_G) {
for (i = 0; i < ngroups; i++) {
if (i) xputc(' ');
if (flags & FLAG_n) printf("%s", grps[i]->gr_name);
else printf("%d", grps[i]->gr_gid);
}
printf("\n");
return;
}
pretty_print(pw, grp, grps, ngroups);
for (i=0; i < ngroups; i++)
free(grps[i]);
}
|
