Annotation of embedaddon/sudo/common/aix.c, revision 1.1.1.1

1.1       misho       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: 
                     38: #define DEFAULT_TEXT_DOMAIN    "sudo"
                     39: #include "gettext.h"
                     40: 
                     41: #ifdef HAVE_GETUSERATTR
                     42: 
                     43: #ifndef HAVE_SETRLIMIT64
                     44: # define setrlimit64(a, b) setrlimit(a, b)
                     45: # define rlimit64 rlimit
                     46: # define rlim64_t rlim_t
                     47: # define RLIM64_INFINITY RLIM_INFINITY
                     48: #endif /* HAVE_SETRLIMIT64 */
                     49: 
                     50: #ifndef RLIM_SAVED_MAX
                     51: # define RLIM_SAVED_MAX        RLIM64_INFINITY
                     52: #endif
                     53: 
                     54: struct aix_limit {
                     55:     int resource;
                     56:     char *soft;
                     57:     char *hard;
                     58:     int factor;
                     59: };
                     60: 
                     61: static struct aix_limit aix_limits[] = {
                     62:     { RLIMIT_FSIZE, S_UFSIZE, S_UFSIZE_HARD, 512 },
                     63:     { RLIMIT_CPU, S_UCPU, S_UCPU_HARD, 1 },
                     64:     { RLIMIT_DATA, S_UDATA, S_UDATA_HARD, 512 },
                     65:     { RLIMIT_STACK, S_USTACK, S_USTACK_HARD, 512 },
                     66:     { RLIMIT_RSS, S_URSS, S_URSS_HARD, 512 },
                     67:     { RLIMIT_CORE, S_UCORE, S_UCORE_HARD, 512 },
                     68:     { RLIMIT_NOFILE, S_UNOFILE, S_UNOFILE_HARD, 1 }
                     69: };
                     70: 
                     71: static int
                     72: aix_getlimit(char *user, char *lim, rlim64_t *valp)
                     73: {
                     74:     int val;
                     75: 
                     76:     if (getuserattr(user, lim, &val, SEC_INT) != 0)
                     77:        return -1;
                     78:     *valp = val;
                     79:     return 0;
                     80: }
                     81: 
                     82: static void
                     83: aix_setlimits(char *user)
                     84: {
                     85:     struct rlimit64 rlim;
                     86:     rlim64_t val;
                     87:     int n;
                     88: 
                     89:     if (setuserdb(S_READ) != 0)
                     90:        error(1, "unable to open userdb");
                     91: 
                     92:     /*
                     93:      * For each resource limit, get the soft/hard values for the user
                     94:      * and set those values via setrlimit64().  Must be run as euid 0.
                     95:      */
                     96:     for (n = 0; n < sizeof(aix_limits) / sizeof(aix_limits[0]); n++) {
                     97:        /*
                     98:         * We have two strategies, depending on whether or not the
                     99:         * hard limit has been defined.
                    100:         */
                    101:        if (aix_getlimit(user, aix_limits[n].hard, &val) == 0) {
                    102:            rlim.rlim_max = val == -1 ? RLIM64_INFINITY : val * aix_limits[n].factor;
                    103:            if (aix_getlimit(user, aix_limits[n].soft, &val) == 0)
                    104:                rlim.rlim_cur = val == -1 ? RLIM64_INFINITY : val * aix_limits[n].factor;
                    105:            else
                    106:                rlim.rlim_cur = rlim.rlim_max;  /* soft not specd, use hard */
                    107:        } else {
                    108:            /* No hard limit set, try soft limit. */
                    109:            if (aix_getlimit(user, aix_limits[n].soft, &val) == 0)
                    110:                rlim.rlim_cur = val == -1 ? RLIM64_INFINITY : val * aix_limits[n].factor;
                    111: 
                    112:            /* Set hard limit per AIX /etc/security/limits documentation. */
                    113:            switch (aix_limits[n].resource) {
                    114:                case RLIMIT_CPU:
                    115:                case RLIMIT_FSIZE:
                    116:                    rlim.rlim_max = rlim.rlim_cur;
                    117:                    break;
                    118:                case RLIMIT_STACK:
                    119:                    rlim.rlim_max = RLIM_SAVED_MAX;
                    120:                    break;
                    121:                default:
                    122:                    rlim.rlim_max = RLIM64_INFINITY;
                    123:                    break;
                    124:            }
                    125:        }
                    126:        (void)setrlimit64(aix_limits[n].resource, &rlim);
                    127:     }
                    128:     enduserdb();
                    129: }
                    130: 
                    131: #ifdef HAVE_SETAUTHDB
                    132: /*
                    133:  * Look up administrative domain for user (SYSTEM in /etc/security/user) and
                    134:  * set it as the default for the process.  This ensures that password and
                    135:  * group lookups are made against the correct source (files, NIS, LDAP, etc).
                    136:  */
                    137: void
                    138: aix_setauthdb(char *user)
                    139: {
                    140:     char *registry;
                    141: 
                    142:     if (user != NULL) {
                    143:        if (setuserdb(S_READ) != 0)
                    144:            error(1, _("unable to open userdb"));
                    145:        if (getuserattr(user, S_REGISTRY, &registry, SEC_CHAR) == 0) {
                    146:            if (setauthdb(registry, NULL) != 0)
                    147:                error(1, _("unable to switch to registry \"%s\" for %s"),
                    148:                    registry, user);
                    149:        }
                    150:        enduserdb();
                    151:     }
                    152: }
                    153: 
                    154: /*
                    155:  * Restore the saved administrative domain, if any.
                    156:  */
                    157: void
                    158: aix_restoreauthdb(void)
                    159: {
                    160:     if (setauthdb(NULL, NULL) != 0)
                    161:        error(1, _("unable to restore registry"));
                    162: }
                    163: #endif
                    164: 
                    165: void
                    166: aix_prep_user(char *user, const char *tty)
                    167: {
                    168:     char *info;
                    169:     int len;
                    170: 
                    171:     /* set usrinfo, like login(1) does */
                    172:     len = easprintf(&info, "NAME=%s%cLOGIN=%s%cLOGNAME=%s%cTTY=%s%c",
                    173:        user, '\0', user, '\0', user, '\0', tty ? tty : "", '\0');
                    174:     (void)usrinfo(SETUINFO, info, len);
                    175:     efree(info);
                    176: 
                    177: #ifdef HAVE_SETAUTHDB
                    178:     /* set administrative domain */
                    179:     aix_setauthdb(user);
                    180: #endif
                    181: 
                    182:     /* set resource limits */
                    183:     aix_setlimits(user);
                    184: }
                    185: #endif /* HAVE_GETUSERATTR */

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>