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 "detect_os.h"
  30 #include "macros.h"
  31 #include <string.h>
  32 #include <errno.h>
  33 #include <stdlib.h>
  34
  35 #undef GET_PROGRAM_NAME
  36
  37 #if DETECT_OS_WINDOWS
  38 #include <windows.h>
  39 #else
  40 #include <unistd.h>
  41 #endif
  42
  43 #if DETECT_OS_APPLE
  44 #include <mach-o/dyld.h>
  45 #endif
  46
  47 #if defined(__linux__) && defined(HAVE_PROGRAM_INVOCATION_NAME)
  48
  49 static char *path = NULL;
  50
  51 static void __freeProgramPath()
  52 {
  53    free(path);
  54    path = NULL;
  55 }
  56
  57 static const char *
  58 __getProgramName()
  59 {
  60    char * arg = strrchr(program_invocation_name, '/');
  61    if (arg) {
  62       /* If the / character was found this is likely a linux path or
  63        * an invocation path for a 64-bit wine program.
  64        *
  65        * However, some programs pass command line arguments into argv[0].
  66        * Strip these arguments out by using the realpath only if it was
  67        * a prefix of the invocation name.
  68        */
  69       if (!path) {
  70          path = realpath("/proc/self/exe", NULL);
  71          atexit(__freeProgramPath);
  72       }
  73
  74       if (path && strncmp(path, program_invocation_name, strlen(path)) == 0) {
  75          /* This shouldn't be null because path is a a prefix,
  76           * but check it anyway since path is static. */
  77          char * name = strrchr(path, '/');
  78          if (name)
  79             return name + 1;
  80       }
  81
  82       return arg+1;
  83    }
  84
  85    /* If there was no '/' at all we likely have a windows like path from
  86     * a wine application.
  87     */
  88    arg = strrchr(program_invocation_name, '\\');
  89    if (arg)
  90       return arg+1;
  91
  92    return program_invocation_name;
  93 }
  94 #    define GET_PROGRAM_NAME() __getProgramName()
  95 #elif defined(HAVE_PROGRAM_INVOCATION_NAME)
  96 #    define GET_PROGRAM_NAME() program_invocation_short_name
  97 #elif defined(__FreeBSD__) && (__FreeBSD__ >= 2)
  98 #    include <osreldate.h>
  99 #    if (__FreeBSD_version >= 440000)
 100 #        define GET_PROGRAM_NAME() getprogname()
 101 #    endif
 102 #elif defined(__NetBSD__)
 103 #    include <sys/param.h>
 104 #    if defined(__NetBSD_Version__) && (__NetBSD_Version__ >= 106000100)
 105 #        define GET_PROGRAM_NAME() getprogname()
 106 #    endif
 107 #elif defined(__DragonFly__)
 108 #    define GET_PROGRAM_NAME() getprogname()
 109 #elif defined(__APPLE__)
 110 #    define GET_PROGRAM_NAME() getprogname()
 111 #elif defined(ANDROID)
 112 #    define GET_PROGRAM_NAME() getprogname()
 113 #elif defined(__sun)
 114 /* Solaris has getexecname() which returns the full path - return just
 115    the basename to match BSD getprogname() */
 116 #    include <libgen.h>
 117
 118 static const char *
 119 __getProgramName()
 120 {
 121     static const char *progname;
 122
 123     if (progname == NULL) {
 124         const char *e = getexecname();
 125         if (e != NULL) {
 126             /* Have to make a copy since getexecname can return a readonly
 127                string, but basename expects to be able to modify its arg. */
 128             char *n = strdup(e);
 129             if (n != NULL) {
 130                 progname = basename(n);
 131             }
 132         }
 133     }
 134     return progname;
 135 }
 136
 137 #    define GET_PROGRAM_NAME() __getProgramName()
 138 #endif
 139
 140 #if !defined(GET_PROGRAM_NAME)
 141 #    if defined(__OpenBSD__) || defined(__NetBSD__) || defined(__UCLIBC__) || defined(ANDROID)
 142 /* This is a hack. It's said to work on OpenBSD, NetBSD and GNU.
 143  * Rogelio M.Serrano Jr. reported it's also working with UCLIBC. It's
 144  * used as a last resort, if there is no documented facility available. */
 145 static const char *
 146 __getProgramName()
 147 {
 148     extern const char *__progname;
 149     char * arg = strrchr(__progname, '/');
 150     if (arg)
 151         return arg+1;
 152     else
 153         return __progname;
 154 }
 155 #        define GET_PROGRAM_NAME() __getProgramName()
 156 #    else
 157 #        define GET_PROGRAM_NAME() ""
 158 #        pragma message ( "Warning: Per application configuration won't work with your OS version." )
 159 #    endif
 160 #endif
 161
 162 const char *
 163 util_get_process_name(void)
 164 {
 165    return GET_PROGRAM_NAME();
 166 }
 167
 168 size_t
 169 util_get_process_exec_path(char* process_path, size_t len)
 170 {
 171 #if DETECT_OS_WINDOWS
 172    return GetModuleFileNameA(NULL, process_path, len);
 173 #elif DETECT_OS_APPLE
 174    uint32_t bufSize = len;
 175    int result = _NSGetExecutablePath(process_path, &bufSize);
 176
 177    return (result == 0) ? strlen(process_path) : 0;
 178 #elif DETECT_OS_UNIX
 179    ssize_t r;
 180
 181    if ((r = readlink("/proc/self/exe", process_path, len)) > 0)
 182       goto success;
 183    if ((r = readlink("/proc/curproc/exe", process_path, len)) > 0)
 184       goto success;
 185    if ((r = readlink("/proc/curproc/file", process_path, len)) > 0)
 186       goto success;
 187
 188     return 0;
 189 success:
 190     process_path[r] = '\0';
 191     return r;
 192
 193 #endif
 194    return 0;
 195 }