aboutsummaryrefslogtreecommitdiff
path: root/toys
diff options
context:
space:
mode:
Diffstat (limited to 'toys')
-rw-r--r--toys/chgrp.c78
1 files changed, 27 insertions, 51 deletions
diff --git a/toys/chgrp.c b/toys/chgrp.c
index 2ad8a025..5cf43896 100644
--- a/toys/chgrp.c
+++ b/toys/chgrp.c
@@ -15,7 +15,7 @@ USE_CHGRP(NEWTOY(chgrp, "<2Rfv", TOYFLAG_BIN))
config CHGRP
bool "chgrp"
- default n
+ default y
help
usage: chgrp [-R] [-f] [-v] group file...
Change group ownership of one or more files.
@@ -38,31 +38,32 @@ DEFINE_GLOBALS(
#define TT this.chgrp
-static int do_chgrp(const char *path) {
- int ret = chown(path, -1, TT.group);
- if (toys.optflags & 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);
- toys.exitval |= ret;
- return ret;
-}
-
-// Copied from toys/cp.c:cp_node()
-int chgrp_node(char *path, struct dirtree *node)
+static int do_chgrp(struct dirtree *node)
{
- char *s = path + strlen(path);
- struct dirtree *n = node;
-
- for ( ; ; n = n->parent) {
- while (s!=path) {
- if (*(--s) == '/') break;
- }
- if (!n) break;
- }
- if (s != path) s++;
+ 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);
- do_chgrp(s);
+ 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;
}
@@ -74,33 +75,8 @@ void chgrp_main(void)
TT.group_name = *toys.optargs;
group = getgrnam(TT.group_name);
- if (!group) {
- error_msg("invalid group '%s'", TT.group_name);
- toys.exitval = 1;
- return;
- }
+ if (!group) error_exit("no group '%s'", TT.group_name);
TT.group = group->gr_gid;
- if (toys.optflags & FLAG_R) {
- // Recurse into subdirectories
- for (s=toys.optargs + 1; *s; s++) {
- struct stat sb;
- if (stat(*s, &sb) == -1) {
- if (!(toys.optflags & FLAG_f))
- perror_msg("stat '%s'", *s);
- continue;
- }
- do_chgrp(*s);
- if (S_ISDIR(sb.st_mode)) {
- strncpy(toybuf, *s, sizeof(toybuf) - 1);
- toybuf[sizeof(toybuf) - 1] = 0;
- dirtree_read(toybuf, NULL, chgrp_node);
- }
- }
- } else {
- // Do not recurse
- for (s=toys.optargs + 1; *s; s++) {
- do_chgrp(*s);
- }
- }
+ for (s=toys.optargs+1; *s; s++) dirtree_read(*s, do_chgrp);
}