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
__freeProgramPath()51 static void __freeProgramPath()
52 {
53 free(path);
54 path = NULL;
55 }
56
57 static const char *
__getProgramName()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 *
__getProgramName()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 *
__getProgramName()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 *
util_get_process_name(void)163 util_get_process_name(void)
164 {
165 return GET_PROGRAM_NAME();
166 }
167
168 size_t
util_get_process_exec_path(char * process_path,size_t len)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 if (r == len)
191 return 0;
192
193 process_path[r] = '\0';
194 return r;
195
196 #endif
197 return 0;
198 }
199