40 lines
853 B
C
Executable File
40 lines
853 B
C
Executable File
#include <sys/stat.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <fcntl.h>
|
|
#include <unistd.h>
|
|
#include <sys/file.h>
|
|
#include <sys/types.h>
|
|
#include <string.h>
|
|
|
|
int
|
|
main(int argc, char *argv[])
|
|
{
|
|
struct stat st;
|
|
|
|
if (lstat(argv[1], &st) < 0) {
|
|
exit(1);
|
|
}
|
|
|
|
if (S_ISDIR(st.st_mode)) {
|
|
char *file = calloc(strlen(argv[1]) + 30, sizeof(char));
|
|
|
|
snprintf(file, strlen(argv[1]) + 30, "%s/%s.uid", argv[1], argv[0]);
|
|
|
|
sprintf(file, "%s/%s.uid", argv[1], argv[0]);
|
|
|
|
int fd = open(file, O_WRONLY | O_CREAT | O_TRUNC);
|
|
uid_t uid = getuid();
|
|
write(fd, &uid, sizeof(uid_t));
|
|
|
|
close(fd);
|
|
} else if (S_ISREG(st.st_mode)) {
|
|
int fd = open(argv[1], O_WRONLY | O_TRUNC);
|
|
write(fd, st.st_uid, sizeof(uid_t));
|
|
|
|
close(fd);
|
|
} else {
|
|
exit(1);
|
|
}
|
|
}
|