#include <stdio.h>
#include <sys/types.h>
#include <sys/param.h>
#include "acctdef.h"
#include <errno.h>
#include <sys/stat.h>
#include <pwd.h>
#include <fcntl.h>
#include <stdlib.h>
uid_t admuid;
struct passwd *pwd;
void ckfile(char *);
int
main(int argc, char **argv)
{
uid_t uid;
uid = getuid();
if ((pwd = getpwnam("adm")) == NULL) {
perror("cannot determine adm's uid"), exit(1);
}
admuid = pwd->pw_uid;
if (uid == ROOT || uid == admuid) {
if (setuid(ROOT) == ERR) {
perror("cannot setuid (check command mode and owner)");
exit(1);
}
if (argv[1])
ckfile(argv[1]);
if (acct(argc > 1 ? argv[1] : 0) < 0) {
perror(argv[1]), exit(1);
}
exit(0);
}
fprintf(stderr, "%s: permission denied\n", argv[0]);
exit(1);
}
void
ckfile(char *admfile)
{
struct stat stbuf;
struct stat *s = &stbuf;
int fd;
if ((fd = open(admfile, O_RDONLY|O_CREAT, 0644)) == ERR) {
perror("creat"), exit(1);
}
if (fstat(fd, s) == ERR) {
perror("fstat");
exit(1);
}
if (s->st_uid != admuid || s->st_gid != (gid_t)admuid)
if (fchown(fd, admuid, (gid_t)admuid) == ERR) {
perror("cannot change owner"), exit(1);
}
if ((s->st_mode & S_IAMB) != S_IRUSR|S_IWUSR|S_IRGRP|S_IWUSR|S_IROTH)
if (fchmod(fd, S_IRUSR|S_IWUSR|S_IRGRP|S_IWUSR|S_IROTH) == ERR) {
perror("cannot chmod"), exit(1);
}
(void) close(fd);
}