view toys/chgrp.c @ 582:b88bc7dcdb48

Update chgrp so -R works, tweaking DIRTREE_COMEAGAIN design along the way.
author Rob Landley <rob@landley.net>
date Sun, 27 May 2012 00:56:17 -0500
parents 4a91ede70548
children 9c2277b92b86
line wrap: on
line source

/* vi: set sw=4 ts=4:
 *
 * chgrp.c - Change group ownership
 *
 * Copyright 2012 Georgi Chorbadzhiyski <georgi@unixsol.org>
 *
 * See http://pubs.opengroup.org/onlinepubs/009695399/utilities/chgrp.html
 *
 * TODO: Add support for -h
 * TODO: Add support for -H
 * TODO: Add support for -L
 * TODO: Add support for -P

USE_CHGRP(NEWTOY(chgrp, "<2Rfv", TOYFLAG_BIN))

config CHGRP
	bool "chgrp"
	default y
	help
	  usage: chgrp [-R] [-f] [-v] group file...
	  Change group ownership of one or more files.

	  -R	recurse into subdirectories.
	  -f	suppress most error messages.
	  -v	verbose output.
*/

#include "toys.h"

#define FLAG_R 4
#define FLAG_f 2
#define FLAG_v 1

DEFINE_GLOBALS(
	gid_t group;
	char *group_name;
)

#define TT this.chgrp

static int do_chgrp(struct dirtree *node)
{
	int fd, ret = 1, flags = toys.optflags;

	if (!dirtree_notdotdot(node)) return 0;

	// Handle recursion, and make it depth first
	if (S_ISDIR(node->st.st_mode)) {
		if (!node->extra) node->extra = dup(node->data);
		if ((flags & FLAG_R) && node->data != -1) return DIRTREE_COMEAGAIN;
		fd = node->extra;
	} else fd = openat(node->parent ? node->parent->data : AT_FDCWD,
		node->name, 0);

	if (fd != -1) ret = fchown(fd, -1, TT.group);

	if (ret || (flags & FLAG_v)) {
		char *path = dirtree_path(node, 0);
		if (flags & FLAG_v)
			xprintf("chgrp(%s, %s)\n", TT.group_name, path);
		if (ret == -1 && !(toys.optflags & FLAG_f))
			perror_msg("changing group of '%s' to '%s'", path, TT.group_name);
		free(path);
	}
	close(fd);
	toys.exitval |= ret;

	return 0;
}

void chgrp_main(void)
{
	char **s;
	struct group *group;

	TT.group_name = *toys.optargs;
	group = getgrnam(TT.group_name);
	if (!group) error_exit("no group '%s'", TT.group_name);
	TT.group = group->gr_gid;

	for (s=toys.optargs+1; *s; s++) dirtree_read(*s, do_chgrp);
}