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) || defined(PIPE_OS_ANDROID)
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
os_get_process_name(char * procname,size_t size)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) || defined(PIPE_OS_ANDROID)
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
os_get_command_line(char * cmdline,size_t size)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