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