1 /************************************************************************** 2 * 3 * Copyright 2013 VMware, Inc. 4 * All Rights Reserved. 5 * 6 * Permission is hereby granted, free of charge, to any person obtaining a 7 * 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 above copyright notice and this permission notice (including the 15 * next paragraph) shall be included in all copies or substantial portions 16 * of the Software. 17 * 18 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS 19 * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF 20 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. 21 * IN NO EVENT SHALL THE AUTHORS AND/OR ITS SUPPLIERS BE LIABLE FOR 22 * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, 23 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE 24 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. 25 * 26 **************************************************************************/ 27 28 29 #include "pipe/p_config.h" 30 #include "os/os_process.h" 31 #include "util/u_memory.h" 32 33 #if defined(PIPE_SUBSYSTEM_WINDOWS_USER) 34 # include <windows.h> 35 #elif defined(__GLIBC__) || defined(__CYGWIN__) 36 # include <errno.h> 37 #elif defined(PIPE_OS_BSD) || defined(PIPE_OS_APPLE) 38 # include <stdlib.h> 39 #elif defined(PIPE_OS_HAIKU) 40 # include <kernel/OS.h> 41 # include <kernel/image.h> 42 #else 43 #warning unexpected platform in os_process.c 44 #endif 45 46 #if defined(PIPE_OS_LINUX) 47 # include <fcntl.h> 48 #endif 49 50 51 /** 52 * Return the name of the current process. 53 * \param procname returns the process name 54 * \param size size of the procname buffer 55 * \return TRUE or FALSE for success, failure 56 */ 57 boolean 58 os_get_process_name(char *procname, size_t size) 59 { 60 const char *name; 61 62 /* First, check if the GALLIUM_PROCESS_NAME env var is set to 63 * override the normal process name query. 64 */ 65 name = os_get_option("GALLIUM_PROCESS_NAME"); 66 67 if (!name) { 68 /* do normal query */ 69 70 #if defined(PIPE_SUBSYSTEM_WINDOWS_USER) 71 char szProcessPath[MAX_PATH]; 72 char *lpProcessName; 73 char *lpProcessExt; 74 75 GetModuleFileNameA(NULL, szProcessPath, ARRAY_SIZE(szProcessPath)); 76 77 lpProcessName = strrchr(szProcessPath, '\\'); 78 lpProcessName = lpProcessName ? lpProcessName + 1 : szProcessPath; 79 80 lpProcessExt = strrchr(lpProcessName, '.'); 81 if (lpProcessExt) { 82 *lpProcessExt = '\0'; 83 } 84 85 name = lpProcessName; 86 87 #elif defined(__GLIBC__) || defined(__CYGWIN__) 88 name = program_invocation_short_name; 89 #elif defined(PIPE_OS_BSD) || defined(PIPE_OS_APPLE) 90 /* *BSD and OS X */ 91 name = getprogname(); 92 #elif defined(PIPE_OS_HAIKU) 93 image_info info; 94 get_image_info(B_CURRENT_TEAM, &info); 95 name = info.name; 96 #else 97 #warning unexpected platform in os_process.c 98 return FALSE; 99 #endif 100 101 } 102 103 assert(size > 0); 104 assert(procname); 105 106 if (name && procname && size > 0) { 107 strncpy(procname, name, size); 108 procname[size - 1] = '\0'; 109 return TRUE; 110 } 111 else { 112 return FALSE; 113 } 114 } 115 116 117 /** 118 * Return the command line for the calling process. This is basically 119 * the argv[] array with the arguments separated by spaces. 120 * \param cmdline returns the command line string 121 * \param size size of the cmdline buffer 122 * \return TRUE or FALSE for success, failure 123 */ 124 boolean 125 os_get_command_line(char *cmdline, size_t size) 126 { 127 #if defined(PIPE_SUBSYSTEM_WINDOWS_USER) 128 const char *args = GetCommandLine(); 129 if (args) { 130 strncpy(cmdline, args, size); 131 // make sure we terminate the string 132 cmdline[size - 1] = 0; 133 return TRUE; 134 } 135 #elif defined(PIPE_OS_LINUX) 136 int f = open("/proc/self/cmdline", O_RDONLY); 137 if (f) { 138 const int n = read(f, cmdline, size - 1); 139 int i; 140 assert(n < size); 141 // The arguments are separated by '\0' chars. Convert them to spaces. 142 for (i = 0; i < n; i++) { 143 if (cmdline[i] == 0) { 144 cmdline[i] = ' '; 145 } 146 } 147 // terminate the string 148 cmdline[n] = 0; 149 close(f); 150 return TRUE; 151 } 152 #endif 153 154 /* XXX to-do: implement this function for other operating systems */ 155 156 cmdline[0] = 0; 157 return FALSE; 158 } 159