git.gag.com Git - debian/sudo/blob - common/aix.c

   1 /*
   2  * Copyright (c) 2008, 2010-2011 Todd C. Miller <Todd.Miller@courtesan.com>
   3  *
   4  * Permission to use, copy, modify, and distribute this software for any
   5  * purpose with or without fee is hereby granted, provided that the above
   6  * copyright notice and this permission notice appear in all copies.
   7  *
   8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
   9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  15  */
  16
  17 #include <config.h>
  18
  19 #include <sys/types.h>
  20 #include <sys/resource.h>
  21
  22 #include <stdio.h>
  23 #ifdef STDC_HEADERS
  24 # include <stdlib.h>
  25 # include <stddef.h>
  26 #else
  27 # ifdef HAVE_STDLIB_H
  28 #  include <stdlib.h>
  29 # endif
  30 #endif /* STDC_HEADERS */
  31 #include <usersec.h>
  32 #include <uinfo.h>
  33
  34 #include "missing.h"
  35 #include "alloc.h"
  36 #include "error.h"
  37 #include "sudo_debug.h"
  38
  39 #define DEFAULT_TEXT_DOMAIN     "sudo"
  40 #include "gettext.h"
  41
  42 #ifdef HAVE_GETUSERATTR
  43
  44 #ifndef HAVE_SETRLIMIT64
  45 # define setrlimit64(a, b) setrlimit(a, b)
  46 # define rlimit64 rlimit
  47 # define rlim64_t rlim_t
  48 # define RLIM64_INFINITY RLIM_INFINITY
  49 #endif /* HAVE_SETRLIMIT64 */
  50
  51 #ifndef RLIM_SAVED_MAX
  52 # define RLIM_SAVED_MAX RLIM64_INFINITY
  53 #endif
  54
  55 struct aix_limit {
  56     int resource;
  57     char *soft;
  58     char *hard;
  59     int factor;
  60 };
  61
  62 static struct aix_limit aix_limits[] = {
  63     { RLIMIT_FSIZE, S_UFSIZE, S_UFSIZE_HARD, 512 },
  64     { RLIMIT_CPU, S_UCPU, S_UCPU_HARD, 1 },
  65     { RLIMIT_DATA, S_UDATA, S_UDATA_HARD, 512 },
  66     { RLIMIT_STACK, S_USTACK, S_USTACK_HARD, 512 },
  67     { RLIMIT_RSS, S_URSS, S_URSS_HARD, 512 },
  68     { RLIMIT_CORE, S_UCORE, S_UCORE_HARD, 512 },
  69     { RLIMIT_NOFILE, S_UNOFILE, S_UNOFILE_HARD, 1 }
  70 };
  71
  72 static int
  73 aix_getlimit(char *user, char *lim, rlim64_t *valp)
  74 {
  75     int val;
  76     debug_decl(aix_getlimit, SUDO_DEBUG_UTIL)
  77
  78     if (getuserattr(user, lim, &val, SEC_INT) != 0)
  79         debug_return_int(-1);
  80     *valp = val;
  81     debug_return_int(0);
  82 }
  83
  84 static void
  85 aix_setlimits(char *user)
  86 {
  87     struct rlimit64 rlim;
  88     rlim64_t val;
  89     int n;
  90     debug_decl(aix_setlimits, SUDO_DEBUG_UTIL)
  91
  92     if (setuserdb(S_READ) != 0)
  93         error(1, "unable to open userdb");
  94
  95     /*
  96      * For each resource limit, get the soft/hard values for the user
  97      * and set those values via setrlimit64().  Must be run as euid 0.
  98      */
  99     for (n = 0; n < sizeof(aix_limits) / sizeof(aix_limits[0]); n++) {
 100         /*
 101          * We have two strategies, depending on whether or not the
 102          * hard limit has been defined.
 103          */
 104         if (aix_getlimit(user, aix_limits[n].hard, &val) == 0) {
 105             rlim.rlim_max = val == -1 ? RLIM64_INFINITY : val * aix_limits[n].factor;
 106             if (aix_getlimit(user, aix_limits[n].soft, &val) == 0)
 107                 rlim.rlim_cur = val == -1 ? RLIM64_INFINITY : val * aix_limits[n].factor;
 108             else
 109                 rlim.rlim_cur = rlim.rlim_max;  /* soft not specd, use hard */
 110         } else {
 111             /* No hard limit set, try soft limit. */
 112             if (aix_getlimit(user, aix_limits[n].soft, &val) == 0)
 113                 rlim.rlim_cur = val == -1 ? RLIM64_INFINITY : val * aix_limits[n].factor;
 114
 115             /* Set hard limit per AIX /etc/security/limits documentation. */
 116             switch (aix_limits[n].resource) {
 117                 case RLIMIT_CPU:
 118                 case RLIMIT_FSIZE:
 119                     rlim.rlim_max = rlim.rlim_cur;
 120                     break;
 121                 case RLIMIT_STACK:
 122                     rlim.rlim_max = RLIM_SAVED_MAX;
 123                     break;
 124                 default:
 125                     rlim.rlim_max = RLIM64_INFINITY;
 126                     break;
 127             }
 128         }
 129         (void)setrlimit64(aix_limits[n].resource, &rlim);
 130     }
 131     enduserdb();
 132     debug_return;
 133 }
 134
 135 #ifdef HAVE_SETAUTHDB
 136 /*
 137  * Look up administrative domain for user (SYSTEM in /etc/security/user) and
 138  * set it as the default for the process.  This ensures that password and
 139  * group lookups are made against the correct source (files, NIS, LDAP, etc).
 140  */
 141 void
 142 aix_setauthdb(char *user)
 143 {
 144     char *registry;
 145     debug_decl(aix_setauthdb, SUDO_DEBUG_UTIL)
 146
 147     if (user != NULL) {
 148         if (setuserdb(S_READ) != 0)
 149             error(1, _("unable to open userdb"));
 150         if (getuserattr(user, S_REGISTRY, &registry, SEC_CHAR) == 0) {
 151             if (setauthdb(registry, NULL) != 0)
 152                 error(1, _("unable to switch to registry \"%s\" for %s"),
 153                     registry, user);
 154         }
 155         enduserdb();
 156     }
 157     debug_return;
 158 }
 159
 160 /*
 161  * Restore the saved administrative domain, if any.
 162  */
 163 void
 164 aix_restoreauthdb(void)
 165 {
 166     debug_decl(aix_setauthdb, SUDO_DEBUG_UTIL)
 167
 168     if (setauthdb(NULL, NULL) != 0)
 169         error(1, _("unable to restore registry"));
 170
 171     debug_return;
 172 }
 173 #endif
 174
 175 void
 176 aix_prep_user(char *user, const char *tty)
 177 {
 178     char *info;
 179     int len;
 180     debug_decl(aix_setauthdb, SUDO_DEBUG_UTIL)
 181
 182     /* set usrinfo, like login(1) does */
 183     len = easprintf(&info, "NAME=%s%cLOGIN=%s%cLOGNAME=%s%cTTY=%s%c",
 184         user, '\0', user, '\0', user, '\0', tty ? tty : "", '\0');
 185     (void)usrinfo(SETUINFO, info, len);
 186     efree(info);
 187
 188 #ifdef HAVE_SETAUTHDB
 189     /* set administrative domain */
 190     aix_setauthdb(user);
 191 #endif
 192
 193     /* set resource limits */
 194     aix_setlimits(user);
 195
 196     debug_return;
 197 }
 198 #endif /* HAVE_GETUSERATTR */