forked from public/pyceo
parent
0d52c0475b
commit
37eb3e6465
@ -0,0 +1,121 @@ |
||||
#include <stdio.h> |
||||
#include <unistd.h> |
||||
#include <sys/types.h> |
||||
#include <sys/stat.h> |
||||
#include <sys/acl.h> |
||||
#include <dirent.h> |
||||
#include <pwd.h> |
||||
#include <fcntl.h> |
||||
|
||||
#include "homedir.h" |
||||
#include "util.h" |
||||
#include "config.h" |
||||
|
||||
int ceo_create_home(char *homedir, uid_t uid, gid_t gid) { |
||||
int mask; |
||||
DIR *skel; |
||||
struct dirent *skelent; |
||||
|
||||
mask = umask(0); |
||||
|
||||
if (mkdir(homedir, 0755)) { |
||||
errorpe("failed to create %s", homedir); |
||||
return -1; |
||||
} |
||||
|
||||
skel = opendir(skeleton_dir); |
||||
if (!skel) { |
||||
errorpe("failed to open %s", skeleton_dir); |
||||
return -1; |
||||
} |
||||
|
||||
while ((skelent = readdir(skel))) { |
||||
struct stat sb; |
||||
char src[PATH_MAX], dest[PATH_MAX]; |
||||
|
||||
if (!strcmp(skelent->d_name, ".") || !strcmp(skelent->d_name, "..")) |
||||
continue; |
||||
|
||||
snprintf(src, sizeof(src), "%s/%s", skeleton_dir, skelent->d_name); |
||||
snprintf(dest, sizeof(dest), "%s/%s", homedir, skelent->d_name); |
||||
lstat(src, &sb); |
||||
|
||||
if (sb.st_uid || sb.st_gid) { |
||||
warn("not creating %s due to ownership", dest); |
||||
continue; |
||||
} |
||||
|
||||
if (S_ISREG(sb.st_mode)) { |
||||
int bytes; |
||||
char buf[4096]; |
||||
|
||||
int srcfd = open(src, O_RDONLY); |
||||
if (srcfd == -1) { |
||||
warnpe("open: %s", src); |
||||
continue; |
||||
} |
||||
|
||||
int destfd = open(dest, O_WRONLY|O_CREAT|O_EXCL, sb.st_mode & 0777); |
||||
if (destfd == -1) { |
||||
warnpe("open: %s", dest); |
||||
close(srcfd); |
||||
continue; |
||||
} |
||||
|
||||
for (;;) { |
||||
bytes = read(srcfd, buf, sizeof(buf)); |
||||
if (!bytes) |
||||
break; |
||||
if (bytes < 0) { |
||||
warnpe("read"); |
||||
break; |
||||
} |
||||
if (write(destfd, buf, bytes) < 0) { |
||||
warnpe("write"); |
||||
break; |
||||
} |
||||
} |
||||
|
||||
if (fchown(destfd, uid, gid)) |
||||
errorpe("chown: %s", dest); |
||||
|
||||
close(srcfd); |
||||
close(destfd); |
||||
} else if (S_ISDIR(sb.st_mode)) { |
||||
if (mkdir(dest, sb.st_mode & 0777)) { |
||||
warnpe("mkdir: %s", dest); |
||||
continue; |
||||
} |
||||
if (chown(dest, uid, gid)) |
||||
errorpe("chown: %s", dest); |
||||
} else if (S_ISLNK(sb.st_mode)) { |
||||
char lnkdest[PATH_MAX]; |
||||
int bytes; |
||||
bytes = readlink(src, lnkdest, sizeof(lnkdest)); |
||||
lnkdest[bytes] = '\0'; |
||||
if (bytes == -1) { |
||||
warnpe("readlink: %s", src); |
||||
continue; |
||||
} |
||||
if (symlink(lnkdest, dest)) { |
||||
warnpe("symlink: %s", dest); |
||||
continue; |
||||
} |
||||
if (lchown(dest, uid, gid)) |
||||
errorpe("lchown: %s", dest); |
||||
} else { |
||||
warn("not creating %s", dest); |
||||
} |
||||
} |
||||
|
||||
closedir(skel); |
||||
|
||||
if (chown(homedir, uid, gid)) { |
||||
errorpe("failed to chown %s", homedir); |
||||
return -1; |
||||
} |
||||
|
||||
umask(mask); |
||||
|
||||
return 0; |
||||
} |
@ -0,0 +1,3 @@ |
||||
#include <sys/acl.h> |
||||
|
||||
int ceo_create_home(char *homedir, uid_t uid, gid_t gid); |
Loading…
Reference in new issue