Annotation of embedaddon/sudo/common/progname.c, revision 1.1

1.1     ! misho       1: /*
        !             2:  * Copyright (c) 2013 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: /* Large files not supported by procfs.h */
        !            20: #if defined(HAVE_PROCFS_H) || defined(HAVE_SYS_PROCFS_H)
        !            21: # undef _FILE_OFFSET_BITS
        !            22: # undef _LARGE_FILES
        !            23: #endif
        !            24: 
        !            25: #include <sys/types.h>
        !            26: #ifdef HAVE_PSTAT_GETPROC
        !            27: # include <sys/param.h>
        !            28: # include <sys/pstat.h>
        !            29: #endif
        !            30: #if defined(HAVE_PROCFS_H)
        !            31: # include <procfs.h>
        !            32: #elif defined(HAVE_SYS_PROCFS_H)
        !            33: # include <sys/procfs.h>
        !            34: #endif
        !            35: 
        !            36: #include <stdio.h>
        !            37: #ifdef STDC_HEADERS
        !            38: # include <stdlib.h>
        !            39: # include <stddef.h>
        !            40: #else
        !            41: # ifdef HAVE_STDLIB_H
        !            42: #  include <stdlib.h>
        !            43: # endif
        !            44: #endif /* STDC_HEADERS */
        !            45: #ifdef HAVE_STRING_H
        !            46: # include <string.h>
        !            47: #endif /* HAVE_STRING_H */
        !            48: #ifdef HAVE_STRINGS_H
        !            49: # include <strings.h>
        !            50: #endif /* HAVE_STRINGS_H */
        !            51: #include <errno.h>
        !            52: #include <fcntl.h>
        !            53: 
        !            54: #include "missing.h"
        !            55: #include "sudo_util.h"
        !            56: 
        !            57: #if defined(HAVE_GETPROGNAME) || defined(HAVE___PROGNAME)
        !            58: 
        !            59: /* STUB */
        !            60: void
        !            61: initprogname(const char *name)
        !            62: {
        !            63:     return;
        !            64: }
        !            65: 
        !            66: #else
        !            67: 
        !            68: static const char *progname = "";
        !            69: 
        !            70: void
        !            71: initprogname(const char *name)
        !            72: {
        !            73:     const char *base;
        !            74: #ifdef HAVE_PSTAT_GETPROC
        !            75:     static char ucomm[PST_UCOMMLEN];
        !            76:     struct pst_status pstat;
        !            77:     int rc;
        !            78: 
        !            79:     /*
        !            80:      * Determine the progname from pst_ucomm in struct pst_status.
        !            81:      * We may get EOVERFLOW if the whole thing doesn't fit but that is OK.
        !            82:      */
        !            83:     rc = pstat_getproc(&pstat, sizeof(pstat), (size_t)0, (int)getpid());
        !            84:     if (rc != -1 || errno == EOVERFLOW) {
        !            85:         strlcpy(ucomm, pstat.pst_ucomm, sizeof(ucomm));
        !            86:        progname = ucomm;
        !            87:        return;
        !            88:     }
        !            89: #elif defined(HAVE_PROCFS_H) || defined(HAVE_SYS_PROCFS_H)
        !            90:     /* XXX - configure check for psinfo.pr_fname */
        !            91:     static char ucomm[PRFNSZ];
        !            92:     struct psinfo psinfo;
        !            93:     char path[PATH_MAX];
        !            94:     ssize_t nread;
        !            95:     int fd;
        !            96: 
        !            97:     /* Try to determine the tty from pr_ttydev in /proc/pid/psinfo. */
        !            98:     snprintf(path, sizeof(path), "/proc/%u/psinfo", (unsigned int)getpid());
        !            99:     if ((fd = open(path, O_RDONLY, 0)) != -1) {
        !           100:        nread = read(fd, &psinfo, sizeof(psinfo));
        !           101:        close(fd);
        !           102:        if (nread == (ssize_t)sizeof(psinfo)) {
        !           103:            strlcpy(ucomm, psinfo.pr_fname, sizeof(ucomm));
        !           104:            progname = ucomm;
        !           105:            return;
        !           106:        }
        !           107:     }
        !           108: #endif /* HAVE_PSTAT_GETPROC */
        !           109: 
        !           110:     if ((base = strrchr(name, '/')) != NULL) {
        !           111:        base++;
        !           112:     } else {
        !           113:        base = name;
        !           114:     }
        !           115:     progname = base;
        !           116: }
        !           117: 
        !           118: const char *
        !           119: getprogname(void)
        !           120: {
        !           121:     return progname;
        !           122: }
        !           123: 
        !           124: #endif /* !HAVE_GETPROGNAME && !HAVE___PROGNAME */

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