1 | /* Program name management. |
2 | Copyright (C) 2016-2019 Free Software Foundation, Inc. |
3 | |
4 | This program is free software: you can redistribute it and/or modify |
5 | it under the terms of the GNU General Public License as published by |
6 | the Free Software Foundation; either version 3 of the License, or |
7 | (at your option) any later version. |
8 | |
9 | This program is distributed in the hope that it will be useful, |
10 | but WITHOUT ANY WARRANTY; without even the implied warranty of |
11 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
12 | GNU General Public License for more details. |
13 | |
14 | You should have received a copy of the GNU General Public License |
15 | along with this program. If not, see <https://www.gnu.org/licenses/>. */ |
16 | |
17 | #include <config.h> |
18 | |
19 | /* Specification. */ |
20 | #include "getprogname.h" |
21 | |
22 | #include <errno.h> /* get program_invocation_name declaration */ |
23 | #include <stdlib.h> /* get __argv declaration */ |
24 | |
25 | #ifdef _AIX |
26 | # include <unistd.h> |
27 | # include <procinfo.h> |
28 | # include <string.h> |
29 | #endif |
30 | |
31 | #ifdef __MVS__ |
32 | # ifndef _OPEN_SYS |
33 | # define _OPEN_SYS |
34 | # endif |
35 | # include <string.h> |
36 | # include <sys/ps.h> |
37 | #endif |
38 | |
39 | #ifdef __hpux |
40 | # include <unistd.h> |
41 | # include <sys/param.h> |
42 | # include <sys/pstat.h> |
43 | # include <string.h> |
44 | #endif |
45 | |
46 | #ifdef __sgi |
47 | # include <string.h> |
48 | # include <unistd.h> |
49 | # include <stdio.h> |
50 | # include <fcntl.h> |
51 | # include <sys/procfs.h> |
52 | #endif |
53 | |
54 | #include "dirname.h" |
55 | |
56 | #ifndef HAVE_GETPROGNAME /* not Mac OS X, FreeBSD, NetBSD, OpenBSD >= 5.4, Cygwin */ |
57 | char const * |
58 | getprogname (void) |
59 | { |
60 | # if HAVE_DECL_PROGRAM_INVOCATION_SHORT_NAME /* glibc, BeOS */ |
61 | /* https://www.gnu.org/software/libc/manual/html_node/Error-Messages.html */ |
62 | return program_invocation_short_name; |
63 | # elif HAVE_DECL_PROGRAM_INVOCATION_NAME /* glibc, BeOS */ |
64 | /* https://www.gnu.org/software/libc/manual/html_node/Error-Messages.html */ |
65 | return last_component (program_invocation_name); |
66 | # elif HAVE_GETEXECNAME /* Solaris */ |
67 | /* https://docs.oracle.com/cd/E19253-01/816-5168/6mbb3hrb1/index.html */ |
68 | const char *p = getexecname (); |
69 | if (!p) |
70 | p = "?" ; |
71 | return last_component (p); |
72 | # elif HAVE_DECL___ARGV /* mingw, MSVC */ |
73 | /* https://docs.microsoft.com/en-us/cpp/c-runtime-library/argc-argv-wargv */ |
74 | const char *p = __argv && __argv[0] ? __argv[0] : "?" ; |
75 | return last_component (p); |
76 | # elif HAVE_VAR___PROGNAME /* OpenBSD, Android, QNX */ |
77 | /* https://man.openbsd.org/style.9 */ |
78 | /* http://www.qnx.de/developers/docs/6.5.0/index.jsp?topic=%2Fcom.qnx.doc.neutrino_lib_ref%2Fp%2F__progname.html */ |
79 | /* Be careful to declare this only when we absolutely need it |
80 | (OpenBSD 5.1), rather than when it's available. Otherwise, |
81 | its mere declaration makes program_invocation_short_name |
82 | malfunction (have zero length) with Fedora 25's glibc. */ |
83 | extern char *__progname; |
84 | const char *p = __progname; |
85 | # if defined __ANDROID__ |
86 | return last_component (p); |
87 | # else |
88 | return p && p[0] ? p : "?" ; |
89 | # endif |
90 | # elif _AIX /* AIX */ |
91 | /* Idea by Bastien ROUCARIÈS, |
92 | https://lists.gnu.org/r/bug-gnulib/2010-12/msg00095.html |
93 | Reference: https://www.ibm.com/support/knowledgecenter/en/ssw_aix_61/com.ibm.aix.basetrf1/getprocs.htm |
94 | */ |
95 | static char *p; |
96 | static int first = 1; |
97 | if (first) |
98 | { |
99 | first = 0; |
100 | pid_t pid = getpid (); |
101 | struct procentry64 procs; |
102 | p = (0 < getprocs64 (&procs, sizeof procs, NULL, 0, &pid, 1) |
103 | ? strdup (procs.pi_comm) |
104 | : NULL); |
105 | if (!p) |
106 | p = "?" ; |
107 | } |
108 | return p; |
109 | # elif defined __hpux |
110 | static char *p; |
111 | static int first = 1; |
112 | if (first) |
113 | { |
114 | first = 0; |
115 | pid_t pid = getpid (); |
116 | struct pst_status status; |
117 | if (pstat_getproc (&status, sizeof status, 0, pid) > 0) |
118 | { |
119 | char *ucomm = status.pst_ucomm; |
120 | char *cmd = status.pst_cmd; |
121 | if (strlen (ucomm) < PST_UCOMMLEN - 1) |
122 | p = ucomm; |
123 | else |
124 | { |
125 | /* ucomm is truncated to length PST_UCOMMLEN - 1. |
126 | Look at cmd instead. */ |
127 | char *space = strchr (cmd, ' '); |
128 | if (space != NULL) |
129 | *space = '\0'; |
130 | p = strrchr (cmd, '/'); |
131 | if (p != NULL) |
132 | p++; |
133 | else |
134 | p = cmd; |
135 | if (strlen (p) > PST_UCOMMLEN - 1 |
136 | && memcmp (p, ucomm, PST_UCOMMLEN - 1) == 0) |
137 | /* p is less truncated than ucomm. */ |
138 | ; |
139 | else |
140 | p = ucomm; |
141 | } |
142 | p = strdup (p); |
143 | } |
144 | else |
145 | { |
146 | # if !defined __LP64__ |
147 | /* Support for 32-bit programs running in 64-bit HP-UX. |
148 | The documented way to do this is to use the same source code |
149 | as above, but in a compilation unit where '#define _PSTAT64 1' |
150 | is in effect. I prefer a single compilation unit; the struct |
151 | size and the offsets are not going to change. */ |
152 | char status64[1216]; |
153 | if (__pstat_getproc64 (status64, sizeof status64, 0, pid) > 0) |
154 | { |
155 | char *ucomm = status64 + 288; |
156 | char *cmd = status64 + 168; |
157 | if (strlen (ucomm) < PST_UCOMMLEN - 1) |
158 | p = ucomm; |
159 | else |
160 | { |
161 | /* ucomm is truncated to length PST_UCOMMLEN - 1. |
162 | Look at cmd instead. */ |
163 | char *space = strchr (cmd, ' '); |
164 | if (space != NULL) |
165 | *space = '\0'; |
166 | p = strrchr (cmd, '/'); |
167 | if (p != NULL) |
168 | p++; |
169 | else |
170 | p = cmd; |
171 | if (strlen (p) > PST_UCOMMLEN - 1 |
172 | && memcmp (p, ucomm, PST_UCOMMLEN - 1) == 0) |
173 | /* p is less truncated than ucomm. */ |
174 | ; |
175 | else |
176 | p = ucomm; |
177 | } |
178 | p = strdup (p); |
179 | } |
180 | else |
181 | # endif |
182 | p = NULL; |
183 | } |
184 | if (!p) |
185 | p = "?" ; |
186 | } |
187 | return p; |
188 | # elif __MVS__ /* z/OS */ |
189 | /* https://www.ibm.com/support/knowledgecenter/SSLTBW_2.1.0/com.ibm.zos.v2r1.bpxbd00/rtwgetp.htm */ |
190 | static char *p = "?" ; |
191 | static int first = 1; |
192 | if (first) |
193 | { |
194 | pid_t pid = getpid (); |
195 | int token; |
196 | W_PSPROC buf; |
197 | first = 0; |
198 | memset (&buf, 0, sizeof(buf)); |
199 | buf.ps_cmdptr = (char *) malloc (buf.ps_cmdlen = PS_CMDBLEN_LONG); |
200 | buf.ps_conttyptr = (char *) malloc (buf.ps_conttylen = PS_CONTTYBLEN); |
201 | buf.ps_pathptr = (char *) malloc (buf.ps_pathlen = PS_PATHBLEN); |
202 | if (buf.ps_cmdptr && buf.ps_conttyptr && buf.ps_pathptr) |
203 | { |
204 | for (token = 0; token >= 0; |
205 | token = w_getpsent (token, &buf, sizeof(buf))) |
206 | { |
207 | if (token > 0 && buf.ps_pid == pid) |
208 | { |
209 | char *s = strdup (last_component (buf.ps_pathptr)); |
210 | if (s) |
211 | p = s; |
212 | break; |
213 | } |
214 | } |
215 | } |
216 | free (buf.ps_cmdptr); |
217 | free (buf.ps_conttyptr); |
218 | free (buf.ps_pathptr); |
219 | } |
220 | return p; |
221 | # elif defined __sgi /* IRIX */ |
222 | char filename[50]; |
223 | int fd; |
224 | |
225 | sprintf (filename, "/proc/pinfo/%d" , (int) getpid ()); |
226 | fd = open (filename, O_RDONLY); |
227 | if (0 <= fd) |
228 | { |
229 | prpsinfo_t buf; |
230 | int ioctl_ok = 0 <= ioctl (fd, PIOCPSINFO, &buf); |
231 | close (fd); |
232 | if (ioctl_ok) |
233 | { |
234 | char *name = buf.pr_fname; |
235 | size_t namesize = sizeof buf.pr_fname; |
236 | /* It may not be NUL-terminated. */ |
237 | char *namenul = memchr (name, '\0', namesize); |
238 | size_t namelen = namenul ? namenul - name : namesize; |
239 | char *namecopy = malloc (namelen + 1); |
240 | if (namecopy) |
241 | { |
242 | namecopy[namelen] = '\0'; |
243 | return memcpy (namecopy, name, namelen); |
244 | } |
245 | } |
246 | } |
247 | return NULL; |
248 | # else |
249 | # error "getprogname module not ported to this OS" |
250 | # endif |
251 | } |
252 | |
253 | #endif |
254 | |
255 | /* |
256 | * Hey Emacs! |
257 | * Local Variables: |
258 | * coding: utf-8 |
259 | * End: |
260 | */ |
261 | |