xref: /illumos-gate/usr/src/lib/libuutil/common/uu_pname.c (revision 1da57d551424de5a9d469760be7c4b4d4f10a755)
1 /*
2  * CDDL HEADER START
3  *
4  * The contents of this file are subject to the terms of the
5  * Common Development and Distribution License, Version 1.0 only
6  * (the "License").  You may not use this file except in compliance
7  * with the License.
8  *
9  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
10  * or http://www.opensolaris.org/os/licensing.
11  * See the License for the specific language governing permissions
12  * and limitations under the License.
13  *
14  * When distributing Covered Code, include this CDDL HEADER in each
15  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
16  * If applicable, add the following below this CDDL HEADER, with the
17  * fields enclosed by brackets "[]" replaced with your own identifying
18  * information: Portions Copyright [yyyy] [name of copyright owner]
19  *
20  * CDDL HEADER END
21  */
22 /*
23  * Copyright 2005 Sun Microsystems, Inc.  All rights reserved.
24  * Use is subject to license terms.
25  */
26 
27 #include "libuutil_common.h"
28 
29 #include <libintl.h>
30 #include <limits.h>
31 #include <string.h>
32 #include <stdlib.h>
33 #include <stdarg.h>
34 #include <stdio.h>
35 #include <errno.h>
36 #include <wchar.h>
37 #include <unistd.h>
38 
39 static const char PNAME_FMT[] = "%s: ";
40 static const char ERRNO_FMT[] = ": %s\n";
41 
42 static const char *pname;
43 
44 static void
45 uu_die_internal(int status, const char *format, va_list alist) __NORETURN;
46 
47 int uu_exit_ok_value = EXIT_SUCCESS;
48 int uu_exit_fatal_value = EXIT_FAILURE;
49 int uu_exit_usage_value = 2;
50 
51 int *
uu_exit_ok(void)52 uu_exit_ok(void)
53 {
54 	return (&uu_exit_ok_value);
55 }
56 
57 int *
uu_exit_fatal(void)58 uu_exit_fatal(void)
59 {
60 	return (&uu_exit_fatal_value);
61 }
62 
63 int *
uu_exit_usage(void)64 uu_exit_usage(void)
65 {
66 	return (&uu_exit_usage_value);
67 }
68 
69 void
uu_alt_exit(int profile)70 uu_alt_exit(int profile)
71 {
72 	switch (profile) {
73 	case UU_PROFILE_DEFAULT:
74 		uu_exit_ok_value = EXIT_SUCCESS;
75 		uu_exit_fatal_value = EXIT_FAILURE;
76 		uu_exit_usage_value = 2;
77 		break;
78 	case UU_PROFILE_LAUNCHER:
79 		uu_exit_ok_value = EXIT_SUCCESS;
80 		uu_exit_fatal_value = 124;
81 		uu_exit_usage_value = 125;
82 		break;
83 	}
84 }
85 
86 static void
uu_warn_internal(int err,const char * format,va_list alist)87 uu_warn_internal(int err, const char *format, va_list alist)
88 {
89 	if (pname != NULL)
90 		(void) fprintf(stderr, PNAME_FMT, pname);
91 
92 	(void) vfprintf(stderr, format, alist);
93 
94 	if (strrchr(format, '\n') == NULL)
95 		(void) fprintf(stderr, ERRNO_FMT, strerror(err));
96 }
97 
98 void
uu_vwarn(const char * format,va_list alist)99 uu_vwarn(const char *format, va_list alist)
100 {
101 	uu_warn_internal(errno, format, alist);
102 }
103 
104 /*PRINTFLIKE1*/
105 void
uu_warn(const char * format,...)106 uu_warn(const char *format, ...)
107 {
108 	va_list alist;
109 	va_start(alist, format);
110 	uu_warn_internal(errno, format, alist);
111 	va_end(alist);
112 }
113 
114 static void
uu_die_internal(int status,const char * format,va_list alist)115 uu_die_internal(int status, const char *format, va_list alist)
116 {
117 	uu_warn_internal(errno, format, alist);
118 #ifdef DEBUG
119 	{
120 		char *cp;
121 
122 		if (!issetugid()) {
123 			cp = getenv("UU_DIE_ABORTS");
124 			if (cp != NULL && *cp != '\0')
125 				abort();
126 		}
127 	}
128 #endif
129 	exit(status);
130 }
131 
132 void
uu_vdie(const char * format,va_list alist)133 uu_vdie(const char *format, va_list alist)
134 {
135 	uu_die_internal(UU_EXIT_FATAL, format, alist);
136 }
137 
138 /*PRINTFLIKE1*/
139 void
uu_die(const char * format,...)140 uu_die(const char *format, ...)
141 {
142 	va_list alist;
143 	va_start(alist, format);
144 	uu_die_internal(UU_EXIT_FATAL, format, alist);
145 	va_end(alist);
146 }
147 
148 void
uu_vxdie(int status,const char * format,va_list alist)149 uu_vxdie(int status, const char *format, va_list alist)
150 {
151 	uu_die_internal(status, format, alist);
152 }
153 
154 /*PRINTFLIKE2*/
155 void
uu_xdie(int status,const char * format,...)156 uu_xdie(int status, const char *format, ...)
157 {
158 	va_list alist;
159 	va_start(alist, format);
160 	uu_die_internal(status, format, alist);
161 	va_end(alist);
162 }
163 
164 const char *
uu_setpname(char * arg0)165 uu_setpname(char *arg0)
166 {
167 	/*
168 	 * Having a NULL argv[0], while uncommon, is possible.  It
169 	 * makes more sense to handle this event in uu_setpname rather
170 	 * than in each of its consumers.
171 	 */
172 	if (arg0 == NULL) {
173 		pname = getexecname();
174 		if (pname == NULL)
175 			pname = "unknown_command";
176 		return (pname);
177 	}
178 
179 	/*
180 	 * Guard against '/' at end of command invocation.
181 	 */
182 	for (;;) {
183 		char *p = strrchr(arg0, '/');
184 		if (p == NULL) {
185 			pname = arg0;
186 			break;
187 		} else {
188 			if (*(p + 1) == '\0') {
189 				*p = '\0';
190 				continue;
191 			}
192 
193 			pname = p + 1;
194 			break;
195 		}
196 	}
197 
198 	return (pname);
199 }
200 
201 const char *
uu_getpname(void)202 uu_getpname(void)
203 {
204 	return (pname);
205 }
206