sbase/chown.c
Quentin Rameau a02d66b8fc chown: fix user:group option parsing
Check that either user or group is correctly passed, and adapt
documention according to that.
2015-12-21 19:07:51 +00:00

111 lines
2.1 KiB
C

/* See LICENSE file for copyright and license details. */
#include <errno.h>
#include <grp.h>
#include <limits.h>
#include <pwd.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include "fs.h"
#include "util.h"
static int hflag = 0;
static uid_t uid = -1;
static gid_t gid = -1;
static int ret = 0;
static void
chownpwgr(const char *path, struct stat *st, void *data, struct recursor *r)
{
char *chownf_name;
int (*chownf)(const char *, uid_t, gid_t);
if (r->follow == 'P' || (r->follow == 'H' && r->depth) || (hflag && !(r->depth))) {
chownf_name = "lchown";
chownf = lchown;
} else {
chownf_name = "chown";
chownf = chown;
}
if (chownf(path, uid, gid) < 0) {
weprintf("%s %s:", chownf_name, path);
ret = 1;
} else if (st && S_ISDIR(st->st_mode)) {
recurse(path, NULL, r);
}
}
static void
usage(void)
{
eprintf("usage: %s [-h] [-R [-H | -L | -P]] owner[:[group]] file ...\n"
" %s [-h] [-R [-H | -L | -P]] :group file ...\n",
argv0, argv0);
}
int
main(int argc, char *argv[])
{
struct group *gr;
struct passwd *pw;
struct recursor r = { .fn = chownpwgr, .hist = NULL, .depth = 0, .maxdepth = 1,
.follow = 'P', .flags = 0 };
char *owner, *group;
ARGBEGIN {
case 'h':
hflag = 1;
break;
case 'r':
case 'R':
r.maxdepth = 0;
break;
case 'H':
case 'L':
case 'P':
r.follow = ARGC();
break;
default:
usage();
} ARGEND
if (argc < 2)
usage();
owner = argv[0];
if ((group = strchr(owner, ':')))
*group++ = '\0';
if (owner && *owner) {
errno = 0;
pw = getpwnam(owner);
if (pw) {
uid = pw->pw_uid;
} else {
if (errno)
eprintf("getpwnam %s:", owner);
uid = estrtonum(owner, 0, UINT_MAX);
}
}
if (group && *group) {
errno = 0;
gr = getgrnam(group);
if (gr) {
gid = gr->gr_gid;
} else {
if (errno)
eprintf("getgrnam %s:", group);
gid = estrtonum(group, 0, UINT_MAX);
}
}
if (uid == (uid_t)-1 && gid == (gid_t)-1)
usage();
for (argc--, argv++; *argv; argc--, argv++)
recurse(*argv, NULL, &r);
return ret || recurse_status;
}