97 lines
1.6 KiB
C
97 lines
1.6 KiB
C
|
//TODO
|
||
|
|
|
||
|
|
#include <pwd.h>
|
||
|
|
#include <grp.h>
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <errno.h>
|
||
|
|
#include <dirent.h>
|
||
|
|
#include <string.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <unistd.h>
|
||
|
|
#include <sys/stat.h>
|
||
|
|
|
||
|
|
int (*chown_func)(const char *pathname, uid_t owner, gid_t group);
|
||
|
|
long gid;
|
||
|
|
long uid;
|
||
|
|
|
||
|
|
int cntree(const char *dst) {
|
||
|
|
if (chown_func(dst, uid, gid) == 0)
|
||
|
|
puts("changed");
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
void get_owner(const char *arg) {
|
||
|
|
char *group = strchr(arg, ':');
|
||
|
|
|
||
|
|
unsigned int g_flag = 1;
|
||
|
|
unsigned int u_flag = 1;
|
||
|
|
if (group == arg)
|
||
|
|
u_flag = 0;
|
||
|
|
|
||
|
|
else if (!group)
|
||
|
|
g_flag = 0;
|
||
|
|
|
||
|
|
if (g_flag) {
|
||
|
|
group[0] = '\0';
|
||
|
|
group++;
|
||
|
|
|
||
|
|
struct group *grp = getgrnam(group);
|
||
|
|
if (!grp) {
|
||
|
|
fprintf(stderr, "chown: invalid group: %s\n", group);
|
||
|
|
exit(1);
|
||
|
|
}
|
||
|
|
|
||
|
|
gid = grp->gr_gid;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (u_flag) {
|
||
|
|
struct passwd *pwd = getpwnam(arg);
|
||
|
|
if (!pwd) {
|
||
|
|
fprintf(stderr, "chown: invalid user: %s\n", arg);
|
||
|
|
exit(1);
|
||
|
|
}
|
||
|
|
|
||
|
|
uid = pwd->pw_gid;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
int main(const int argc, char **argv) {
|
||
|
|
chown_func = lchown;
|
||
|
|
|
||
|
|
int i;
|
||
|
|
for (i = 1; i < argc; i++) {
|
||
|
|
if (argv[i][0] != '-')
|
||
|
|
break;
|
||
|
|
|
||
|
|
else if (!strcmp(argv[i], "-H"))
|
||
|
|
chown_func = chown;
|
||
|
|
|
||
|
|
else if (!strcmp(argv[i], "--help")) {
|
||
|
|
printf("chown [-H (if a command line argument is a symbolic link)] [owner]:[group] [file file2...]\n");
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (argc - i == 0) {
|
||
|
|
fprintf(stderr, "chown: missing operand\n");
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
gid = -1;
|
||
|
|
uid = -1;
|
||
|
|
get_owner(argv[i++]);
|
||
|
|
|
||
|
|
if (argc - i == 0) {
|
||
|
|
fprintf(stderr, "chown: missing operand\n");
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
int ret = 0;
|
||
|
|
for (; i < argc; i++)
|
||
|
|
if (cntree(argv[i]))
|
||
|
|
ret = 1;
|
||
|
|
|
||
|
|
return ret;
|
||
|
|
}
|