src/util/u_process.c

   1 /*
   2  * Copyright © 2003 Felix Kuehling
   3  * Copyright © 2018 Advanced Micro Devices, Inc.
   4  * All Rights Reserved.
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining
   7  * a copy of this software and associated documentation files (the
   8  * "Software"), to deal in the Software without restriction, including
   9  * without limitation the rights to use, copy, modify, merge, publish,
  10  * distribute, sub license, and/or sell copies of the Software, and to
  11  * permit persons to whom the Software is furnished to do so, subject to
  12  * the following conditions:
  13  *
  14  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  15  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
  16  * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  17  * NON-INFRINGEMENT. IN NO EVENT SHALL THE COPYRIGHT HOLDERS, AUTHORS
  18  * AND/OR ITS SUPPLIERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
  20  * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
  21  * USE OR OTHER DEALINGS IN THE SOFTWARE.
  22  *
  23  * The above copyright notice and this permission notice (including the
  24  * next paragraph) shall be included in all copies or substantial portions
  25  * of the Software.
  26  */
  27
  28 #include "u_process.h"
  29 #include <string.h>
  30 #include <errno.h>
  31 #include <stdlib.h>
  32
  33 #undef GET_PROGRAM_NAME
  34
  35 #if (defined(__GNU_LIBRARY__) || defined(__GLIBC__)) && !defined(__UCLIBC__)
  36 #    if !defined(__GLIBC__) || (__GLIBC__ < 2)
  37 /* These aren't declared in any libc5 header */
  38 extern char *program_invocation_name, *program_invocation_short_name;
  39 #    endif
  40 static const char *
  41 __getProgramName()
  42 {
  43    char * arg = strrchr(program_invocation_name, '/');
  44    if (arg) {
  45       /* If the / character was found this is likely a linux path or
  46        * an invocation path for a 64-bit wine program.
  47        *
  48        * However, some programs pass command line arguments into argv[0].
  49        * Strip these arguments out by using the realpath only if it was
  50        * a prefix of the invocation name.
  51        */
  52       static char *path;
  53
  54       if (!path)
  55          /* Note: realpath() allocates memory that we will keep around for
  56           * the lifetime of the app, and then leak as the app closes.
  57           * FIXME: we should find a way to clean this properly
  58           */
  59          path = realpath("/proc/self/exe", NULL);
  60
  61       if (path && strncmp(path, program_invocation_name, strlen(path)) == 0) {
  62          /* This shouldn't be null because path is a a prefix,
  63           * but check it anyway since path is static. */
  64          char * name = strrchr(path, '/');
  65          if (name)
  66             return name + 1;
  67       }
  68
  69       return arg+1;
  70    }
  71
  72    /* If there was no '/' at all we likely have a windows like path from
  73     * a wine application.
  74     */
  75    arg = strrchr(program_invocation_name, '\\');
  76    if (arg)
  77       return arg+1;
  78
  79    return program_invocation_name;
  80 }
  81 #    define GET_PROGRAM_NAME() __getProgramName()
  82 #elif defined(__CYGWIN__)
  83 #    define GET_PROGRAM_NAME() program_invocation_short_name
  84 #elif defined(__FreeBSD__) && (__FreeBSD__ >= 2)
  85 #    include <osreldate.h>
  86 #    if (__FreeBSD_version >= 440000)
  87 #        define GET_PROGRAM_NAME() getprogname()
  88 #    endif
  89 #elif defined(__NetBSD__) && defined(__NetBSD_Version__) && (__NetBSD_Version__ >= 106000100)
  90 #    define GET_PROGRAM_NAME() getprogname()
  91 #elif defined(__DragonFly__)
  92 #    define GET_PROGRAM_NAME() getprogname()
  93 #elif defined(__APPLE__)
  94 #    define GET_PROGRAM_NAME() getprogname()
  95 #elif defined(ANDROID)
  96 #    define GET_PROGRAM_NAME() getprogname()
  97 #elif defined(__sun)
  98 /* Solaris has getexecname() which returns the full path - return just
  99    the basename to match BSD getprogname() */
 100 #    include <libgen.h>
 101
 102 static const char *
 103 __getProgramName()
 104 {
 105     static const char *progname;
 106
 107     if (progname == NULL) {
 108         const char *e = getexecname();
 109         if (e != NULL) {
 110             /* Have to make a copy since getexecname can return a readonly
 111                string, but basename expects to be able to modify its arg. */
 112             char *n = strdup(e);
 113             if (n != NULL) {
 114                 progname = basename(n);
 115             }
 116         }
 117     }
 118     return progname;
 119 }
 120
 121 #    define GET_PROGRAM_NAME() __getProgramName()
 122 #endif
 123
 124 #if !defined(GET_PROGRAM_NAME)
 125 #    if defined(__OpenBSD__) || defined(NetBSD) || defined(__UCLIBC__) || defined(ANDROID)
 126 /* This is a hack. It's said to work on OpenBSD, NetBSD and GNU.
 127  * Rogelio M.Serrano Jr. reported it's also working with UCLIBC. It's
 128  * used as a last resort, if there is no documented facility available. */
 129 static const char *
 130 __getProgramName()
 131 {
 132     extern const char *__progname;
 133     char * arg = strrchr(__progname, '/');
 134     if (arg)
 135         return arg+1;
 136     else
 137         return __progname;
 138 }
 139 #        define GET_PROGRAM_NAME() __getProgramName()
 140 #    else
 141 #        define GET_PROGRAM_NAME() ""
 142 #        pragma message ( "Warning: Per application configuration won't work with your OS version." )
 143 #    endif
 144 #endif
 145
 146 const char *
 147 util_get_process_name(void)
 148 {
 149    return GET_PROGRAM_NAME();
 150 }