view toys/id.c @ 542:06763d1a2f9d

Tighten up the code a bit, and use actual process group id instead of what /etc/passwd says.
author Rob Landley <rob@landley.net>
date Mon, 12 Mar 2012 20:56:56 -0500
parents 31215cc6c9f2
children 60b97ba66a70
line wrap: on
line source

/* 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]);

}