58 lines
1.7 KiB
C
58 lines
1.7 KiB
C
#include <stdio.h>
|
|
#include <sys/types.h>
|
|
#include <unistd.h>
|
|
#include <assert.h>
|
|
#include "util.h"
|
|
|
|
int main(int argc, char *argv[]) {
|
|
if(argc < 7) {
|
|
fprintf(stderr, "Usage: zfsaddhomedir homedir refquota skeldir uid gid mode acl\n");
|
|
return 1;
|
|
}
|
|
|
|
{
|
|
char *homedir = argv[1];
|
|
char *skeldir = argv[3];
|
|
char refquota[32];
|
|
char *mode = argv[6];
|
|
char *acl = (argc >= 8) ? argv[7] : NULL;
|
|
uid_t uid, gid;
|
|
char *zfs_bin = "/usr/sbin/zfs";
|
|
char *chmod_bin = "/usr/bin/chmod";
|
|
char *rsync_bin = "/usr/bin/rsync";
|
|
char *dataset = homedir + 1;
|
|
char *create_argv[] = { "zfs", "create", dataset, NULL };
|
|
char *quota_argv[] = { "zfs", "set", refquota, dataset, NULL };
|
|
char *mode_argv[] = { "chmod", mode, homedir, NULL };
|
|
char *acl_argv[] = { "chmod", acl, homedir, NULL };
|
|
char *rsync_argv[] = { "rsync", "-avH", skeldir, homedir, NULL };
|
|
|
|
assert(homedir[0]);
|
|
uid = atol(argv[4]);
|
|
gid = atol(argv[5]);
|
|
snprintf(refquota, sizeof(refquota), "refquota=%s", argv[2]);
|
|
|
|
if(spawnv(zfs_bin, create_argv))
|
|
return 1;
|
|
if(spawnv(zfs_bin, quota_argv))
|
|
return 1;
|
|
if(spawnv(chmod_bin, mode_argv))
|
|
return 1;
|
|
if(acl && spawnv(chmod_bin, acl_argv))
|
|
return 1;
|
|
|
|
if (chown(homedir, uid, gid)) {
|
|
errorpe("failed to chown %s", homedir);
|
|
return 1;
|
|
}
|
|
|
|
if(seteuid(uid) != 0 || setegid(gid) != 0)
|
|
errorpe("failed to seteuid(%d) or setegid(%d)", uid, gid);
|
|
return 1;
|
|
if(spawnv(rsync_bin, rsync_argv))
|
|
return 1;
|
|
}
|
|
|
|
return 0;
|
|
}
|