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
54 static char *path = NULL;
55
__freeProgramPath()56 static void __freeProgramPath()
57 {
58 free(path);
59 path = NULL;
60 }
61
62 static const char *
__getProgramName()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
114 static const char *
__getProgramName()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)
135 static const char *
__getProgramName()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>
153 extern char **__libc_argv;
154 extern int __libc_argc;
155
156 static const char *
__getProgramName()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. */
177 static const char *
__getProgramName()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
194 const char *
util_get_process_name(void)195 util_get_process_name(void)
196 {
197 return GET_PROGRAM_NAME();
198 }
199
200 size_t
util_get_process_exec_path(char * process_path,size_t len)201 util_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;
228 success:
229 if (r == len)
230 return 0;
231
232 process_path[r] = '\0';
233 return r;
234
235 #endif
236 return 0;
237 }
238