chroot_uid.c

[詳解]

/*++
/* NAME
/*  chroot_uid 3
/* SUMMARY
/*  limit possible damage a process can do
/* SYNOPSIS
/*  #include <chroot_uid.h>
/*
/*  void    chroot_uid(root_dir, user_name)
/*  const char *root_dir;
/*  const char *user_name;
/* DESCRIPTION
/*  \fBchroot_uid\fR changes the process root to \fIroot_dir\fR and
/*  changes process privileges to those of \fIuser_name\fR.
/* DIAGNOSTICS
/*  System call errors are reported via the msg(3) interface.
/*  All errors are fatal.
/* LICENSE
/* .ad
/* .fi
/*  The Secure Mailer license must be distributed with this software.
/* AUTHOR(S)
/*  Wietse Venema
/*  IBM T.J. Watson Research
/*  P.O. Box 704
/*  Yorktown Heights, NY 10598, USA
/*--*/

/* System library. */

#include <sys_defs.h>
#include <pwd.h>
#include <unistd.h>
#include <grp.h>

/* Utility library. */

#include "msg.h"
#include "chroot_uid.h"

/* chroot_uid - restrict the damage that this program can do */

void    chroot_uid(const char *root_dir, const char *user_name)
{
    struct passwd *pwd;
    uid_t   uid;
    gid_t   gid;

    /*
     * Look up the uid/gid before entering the jail, and save them so they
     * can't be clobbered. Set up the primary and secondary groups.
     */
    if (user_name != 0) {
    if ((pwd = getpwnam(user_name)) == 0)
        msg_fatal("unknown user: %s", user_name);
    uid = pwd->pw_uid;
    gid = pwd->pw_gid;
    if (setgid(gid) < 0)
        msg_fatal("setgid(%ld): %m", (long) gid);
    if (initgroups(user_name, gid) < 0)
        msg_fatal("initgroups: %m");
    }

    /*
     * Enter the jail.
     */
    if (root_dir) {
    if (chroot(root_dir))
        msg_fatal("chroot(%s): %m", root_dir);
    if (chdir("/"))
        msg_fatal("chdir(/): %m");
    }

    /*
     * Drop the user privileges.
     */
    if (user_name != 0)
    if (setuid(uid) < 0)
        msg_fatal("setuid(%ld): %m", (long) uid);

    /*
     * Give the desperate developer a clue of what is happening.
     */
    if (msg_verbose > 1)
    msg_info("chroot %s user %s",
         root_dir ? root_dir : "(none)",
         user_name ? user_name : "(none)");
}