Return to mktemp.c CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / sudo / compat

Annotation of embedaddon/sudo/compat/mktemp.c, revision 1.1.1.1

1.1       misho       1: /*
                      2:  * Copyright (c) 2001, 2003, 2004, 2008-2011
                      3:  *     Todd C. Miller <Todd.Miller@courtesan.com>
                      4:  *
                      5:  * Permission to use, copy, modify, and distribute this software for any
                      6:  * purpose with or without fee is hereby granted, provided that the above
                      7:  * copyright notice and this permission notice appear in all copies.
                      8:  *
                      9:  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
                     10:  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
                     11:  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
                     12:  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
                     13:  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
                     14:  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
                     15:  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
                     16:  */
                     17: 
                     18: #include <config.h>
                     19: 
                     20: #include <sys/types.h>
                     21: #include <sys/time.h>
                     22: #include <sys/stat.h>
                     23: 
                     24: #include <errno.h>
                     25: #include <fcntl.h>
                     26: #include <limits.h>
                     27: #include <stdio.h>
                     28: #ifdef HAVE_STDLIB_H
                     29: # include <stdlib.h>
                     30: #endif /* HAVE_STDLIB_H */
                     31: #include <ctype.h>
                     32: #ifdef HAVE_UNISTD_H
                     33: # include <unistd.h>
                     34: #endif /* HAVE_UNISTD_H */
                     35: #if TIME_WITH_SYS_TIME
                     36: # include <time.h>
                     37: #endif
                     38: 
                     39: #include "missing.h"
                     40: 
                     41: #define MKTEMP_FILE    1
                     42: #define MKTEMP_DIR     2
                     43: 
                     44: #define TEMPCHARS      "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789"
                     45: #define NUM_CHARS      (sizeof(TEMPCHARS) - 1)
                     46: 
                     47: #ifndef INT_MAX
                     48: #define INT_MAX        0x7fffffff
                     49: #endif
                     50: 
                     51: #ifdef HAVE_RANDOM
                     52: # define RAND          random
                     53: # define SRAND         srandom
                     54: # define SEED_T                unsigned int
                     55: #else
                     56: # ifdef HAVE_LRAND48
                     57: #  define RAND         lrand48
                     58: #  define SRAND                srand48
                     59: #  define SEED_T       long
                     60: # else
                     61: #  define RAND         rand
                     62: #  define SRAND                srand
                     63: #  define SEED_T       unsigned int
                     64: # endif
                     65: #endif
                     66: 
                     67: static void
                     68: seed_random(void)
                     69: {
                     70:        SEED_T seed;
                     71:        struct timeval tv;
                     72: 
                     73:        /*
                     74:         * Seed from time of day and process id multiplied by small primes.
                     75:         */
                     76:        (void) gettimeofday(&tv, NULL);
                     77:        seed = (tv.tv_sec % 10000) * 523 + tv.tv_usec * 13 +
                     78:            (getpid() % 1000) * 983;
                     79:        SRAND(seed);
                     80: }
                     81: 
                     82: static unsigned int
                     83: get_random(void)
                     84: {
                     85:        static int initialized;
                     86: 
                     87:        if (!initialized) {
                     88:                seed_random();
                     89:                initialized = 1;
                     90:        }
                     91: 
                     92:        return RAND() & 0xffffffff;
                     93: }
                     94: 
                     95: static int
                     96: mktemp_internal(char *path, int slen, int mode)
                     97: {
                     98:        char *start, *cp, *ep;
                     99:        const char *tempchars = TEMPCHARS;
                    100:        unsigned int r, tries;
                    101:        int fd;
                    102: 
                    103:        for (ep = path; *ep; ep++)
                    104:                ;
                    105:        if (path + slen >= ep) {
                    106:                errno = EINVAL;
                    107:                return -1;
                    108:        }
                    109:        ep -= slen;
                    110: 
                    111:        tries = 1;
                    112:        for (start = ep; start > path && start[-1] == 'X'; start--) {
                    113:                if (tries < INT_MAX / NUM_CHARS)
                    114:                        tries *= NUM_CHARS;
                    115:        }
                    116:        tries *= 2;
                    117: 
                    118:        do {
                    119:                for (cp = start; *cp; cp++) {
                    120:                        r = get_random() % NUM_CHARS;
                    121:                        *cp = tempchars[r];
                    122:                }
                    123: 
                    124:                switch (mode) {
                    125:                case MKTEMP_FILE:
                    126:                        fd = open(path, O_CREAT|O_EXCL|O_RDWR, S_IRUSR|S_IWUSR);
                    127:                        if (fd != -1 || errno != EEXIST)
                    128:                                return fd;
                    129:                        break;
                    130:                case MKTEMP_DIR:
                    131:                        if (mkdir(path, S_IRWXU) == 0)
                    132:                                return 0;
                    133:                        if (errno != EEXIST)
                    134:                                return -1;
                    135:                        break;
                    136:                }
                    137:        } while (--tries);
                    138: 
                    139:        errno = EEXIST;
                    140:        return -1;
                    141: }
                    142: 
                    143: #ifndef HAVE_MKSTEMPS
                    144: int
                    145: mkstemps(char *path, int slen)
                    146: {
                    147:        return mktemp_internal(path, slen, MKTEMP_FILE);
                    148: }
                    149: #endif /* HAVE_MKSTEMPS */
                    150: 
                    151: #ifndef HAVE_MKDTEMP
                    152: char *
                    153: mkdtemp(char *path)
                    154: {
                    155:        if (mktemp_internal(path, 0, MKTEMP_DIR) == -1)
                    156:                return NULL;
                    157:        return path;
                    158: }
                    159: #endif /* HAVE_MKDTEMP */