Return to progname.c CVS log

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

Annotation of embedaddon/sudo/common/progname.c, revision 1.1.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 */