cmd/acct/accton.c

/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License, Version 1.0 only
 * (the "License").  You may not use this file except in compliance
 * with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */
/*  Copyright (c) 1984, 1986, 1987, 1988, 1989 AT&T */
/*    All Rights Reserved   */

/*
 * Copyright 2005 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */
#pragma ident   "%Z%%M% %I% %E% SMI"

/*
 *  accton - calls syscall with super-user privileges
 */

#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);
        }

    /* was if(s->st_mode & 0777 != 0664) */
    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);
}