1
0
mirror of git://git.suckless.org/sbase synced 2024-12-22 23:13:44 +00:00
sbase/chown.c
FRIGN 8dc92fbd6c Refactor enmasse() and recurse() to reflect depth
The HLP-changes to sbase have been a great addition of functionality,
but they kind of "polluted" the enmasse() and recurse() prototypes.
As this will come in handy in the future, knowing at which "depth"
you are inside a recursing function is an important functionality.

Instead of having a special HLP-flag passed to enmasse, each sub-
function needs to provide it on its own and can calculate results
based on the current depth (for instance, 'H' implies 'P' at
depth > 0).
A special case is recurse(), because it actually depends on the
follow-type. A new flag "recurse_follow" brings consistency into
what used to be spread across different naming conventions (fflag,
HLP_flag, ...).

This also fixes numerous bugs with the behaviour of HLP in the
tools using it.
2015-03-02 22:50:38 +01:00

104 lines
1.8 KiB
C

/* See LICENSE file for copyright and license details. */
#include <errno.h>
#include <grp.h>
#include <pwd.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include "util.h"
static int rflag = 0;
static uid_t uid = -1;
static gid_t gid = -1;
static int ret = 0;
static char *chownf_name = "chown";
static int (*chownf)(const char *, uid_t, gid_t) = chown;
static void
chownpwgr(const char *path, int depth)
{
if (chownf(path, uid, gid) < 0) {
weprintf("%s %s:", chownf_name, path);
ret = 1;
}
if (rflag)
recurse(path, chownpwgr, depth);
}
static void
usage(void)
{
eprintf("usage: %s [-h] [-R [-H | -L | -P]] [owner][:[group]] file...\n", argv0);
}
int
main(int argc, char *argv[])
{
char *owner, *group, *end;
struct passwd *pw;
struct group *gr;
ARGBEGIN {
case 'h':
chownf_name = "lchown";
chownf = lchown;
break;
case 'r':
case 'R':
rflag = 1;
break;
case 'H':
case 'L':
case 'P':
recurse_follow = ARGC();
break;
default:
usage();
} ARGEND;
if (argc == 0)
usage();
if (recurse_follow == 'P') {
chownf_name = "lchown";
chownf = lchown;
}
owner = argv[0];
argv++;
argc--;
if ((group = strchr(owner, ':')))
*group++ = '\0';
if (owner && *owner) {
errno = 0;
pw = getpwnam(owner);
if (pw) {
uid = pw->pw_uid;
} else {
if (errno != 0)
eprintf("getpwnam %s:", owner);
uid = strtoul(owner, &end, 10);
if (*end != '\0')
eprintf("getpwnam %s: no such user\n", owner);
}
}
if (group && *group) {
errno = 0;
gr = getgrnam(group);
if (gr) {
gid = gr->gr_gid;
} else {
if (errno != 0)
eprintf("getgrnam %s:", group);
gid = strtoul(group, &end, 10);
if (*end != '\0')
eprintf("getgrnam %s: no such group\n", group);
}
}
for (; argc > 0; argc--, argv++)
chownpwgr(argv[0], 0);
return ret;
}