1 | /* printf implementation for the dynamic loader. |
2 | Copyright (C) 1997-2023 Free Software Foundation, Inc. |
3 | This file is part of the GNU C Library. |
4 | |
5 | The GNU C Library is free software; you can redistribute it and/or |
6 | modify it under the terms of the GNU Lesser General Public |
7 | License as published by the Free Software Foundation; either |
8 | version 2.1 of the License, or (at your option) any later version. |
9 | |
10 | The GNU C Library is distributed in the hope that it will be useful, |
11 | but WITHOUT ANY WARRANTY; without even the implied warranty of |
12 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
13 | Lesser General Public License for more details. |
14 | |
15 | You should have received a copy of the GNU Lesser General Public |
16 | License along with the GNU C Library; if not, see |
17 | <https://www.gnu.org/licenses/>. */ |
18 | |
19 | #include <_itoa.h> |
20 | #include <assert.h> |
21 | #include <dl-writev.h> |
22 | #include <ldsodefs.h> |
23 | #include <limits.h> |
24 | #include <stdarg.h> |
25 | #include <stdint.h> |
26 | #include <stdlib.h> |
27 | #include <string.h> |
28 | #include <sys/uio.h> |
29 | #include <unistd.h> |
30 | #include <intprops.h> |
31 | |
32 | /* Bare-bones printf implementation. This function only knows about |
33 | the formats and flags needed and can handle only up to 64 stripes in |
34 | the output. */ |
35 | static void |
36 | _dl_debug_vdprintf (int fd, int tag_p, const char *fmt, va_list arg) |
37 | { |
38 | # define NIOVMAX 64 |
39 | struct iovec iov[NIOVMAX]; |
40 | /* Maximum size for 'd', 'u', and 'x' including padding. */ |
41 | enum { IFMTSIZE = INT_STRLEN_BOUND(void *) }; |
42 | char ifmtbuf[NIOVMAX][IFMTSIZE]; |
43 | int niov = 0; |
44 | pid_t pid = 0; |
45 | char pidbuf[12]; |
46 | |
47 | while (*fmt != '\0') |
48 | { |
49 | const char *startp = fmt; |
50 | |
51 | if (tag_p > 0) |
52 | { |
53 | /* Generate the tag line once. It consists of the PID and a |
54 | colon followed by a tab. */ |
55 | if (pid == 0) |
56 | { |
57 | char *p; |
58 | pid = __getpid (); |
59 | assert (pid >= 0 && sizeof (pid_t) <= 4); |
60 | p = _itoa (pid, &pidbuf[10], 10, 0); |
61 | while (p > pidbuf) |
62 | *--p = ' '; |
63 | pidbuf[10] = ':'; |
64 | pidbuf[11] = '\t'; |
65 | } |
66 | |
67 | /* Append to the output. */ |
68 | assert (niov < NIOVMAX); |
69 | iov[niov].iov_len = 12; |
70 | iov[niov++].iov_base = pidbuf; |
71 | |
72 | /* No more tags until we see the next newline. */ |
73 | tag_p = -1; |
74 | } |
75 | |
76 | /* Skip everything except % and \n (if tags are needed). */ |
77 | while (*fmt != '\0' && *fmt != '%' && (! tag_p || *fmt != '\n')) |
78 | ++fmt; |
79 | |
80 | /* Append constant string. */ |
81 | assert (niov < NIOVMAX); |
82 | if ((iov[niov].iov_len = fmt - startp) != 0) |
83 | iov[niov++].iov_base = (char *) startp; |
84 | |
85 | if (*fmt == '%') |
86 | { |
87 | /* It is a format specifier. */ |
88 | char fill = ' '; |
89 | int width = -1; |
90 | int prec = -1; |
91 | #if LONG_MAX != INT_MAX |
92 | int long_mod = 0; |
93 | #endif |
94 | |
95 | /* Recognize zero-digit fill flag. */ |
96 | if (*++fmt == '0') |
97 | { |
98 | fill = '0'; |
99 | ++fmt; |
100 | } |
101 | |
102 | /* See whether with comes from a parameter. Note that no other |
103 | way to specify the width is implemented. */ |
104 | if (*fmt == '*') |
105 | { |
106 | width = va_arg (arg, int); |
107 | /* The maximum padding accepted is up to pointer size. */ |
108 | assert (width < IFMTSIZE); |
109 | ++fmt; |
110 | } |
111 | |
112 | /* Handle precision. */ |
113 | if (*fmt == '.' && fmt[1] == '*') |
114 | { |
115 | prec = va_arg (arg, int); |
116 | fmt += 2; |
117 | } |
118 | |
119 | /* Recognize the l modifier. It is only important on some |
120 | platforms where long and int have a different size. We |
121 | can use the same code for size_t. */ |
122 | if (*fmt == 'l' || *fmt == 'z') |
123 | { |
124 | #if LONG_MAX != INT_MAX |
125 | long_mod = 1; |
126 | #endif |
127 | ++fmt; |
128 | } |
129 | |
130 | switch (*fmt) |
131 | { |
132 | /* Integer formatting. */ |
133 | case 'd': |
134 | case 'u': |
135 | case 'x': |
136 | { |
137 | /* We have to make a difference if long and int have a |
138 | different size. */ |
139 | #if LONG_MAX != INT_MAX |
140 | unsigned long int num = (long_mod |
141 | ? va_arg (arg, unsigned long int) |
142 | : va_arg (arg, unsigned int)); |
143 | #else |
144 | unsigned long int num = va_arg (arg, unsigned int); |
145 | #endif |
146 | bool negative = false; |
147 | if (*fmt == 'd') |
148 | { |
149 | #if LONG_MAX != INT_MAX |
150 | if (long_mod) |
151 | { |
152 | if ((long int) num < 0) |
153 | negative = true; |
154 | } |
155 | else |
156 | { |
157 | if ((int) num < 0) |
158 | { |
159 | num = (unsigned int) num; |
160 | negative = true; |
161 | } |
162 | } |
163 | #else |
164 | if ((int) num < 0) |
165 | negative = true; |
166 | #endif |
167 | } |
168 | |
169 | char *endp = &ifmtbuf[niov][IFMTSIZE]; |
170 | char *cp = _itoa (num, endp, *fmt == 'x' ? 16 : 10, 0); |
171 | |
172 | /* Pad to the width the user specified. */ |
173 | if (width != -1) |
174 | while (endp - cp < width) |
175 | *--cp = fill; |
176 | |
177 | if (negative) |
178 | *--cp = '-'; |
179 | |
180 | iov[niov].iov_base = cp; |
181 | iov[niov].iov_len = endp - cp; |
182 | ++niov; |
183 | } |
184 | break; |
185 | |
186 | case 's': |
187 | /* Get the string argument. */ |
188 | iov[niov].iov_base = va_arg (arg, char *); |
189 | iov[niov].iov_len = strlen (iov[niov].iov_base); |
190 | if (prec != -1) |
191 | iov[niov].iov_len = MIN ((size_t) prec, iov[niov].iov_len); |
192 | ++niov; |
193 | break; |
194 | |
195 | case '%': |
196 | iov[niov].iov_base = (void *) fmt; |
197 | iov[niov].iov_len = 1; |
198 | ++niov; |
199 | break; |
200 | |
201 | default: |
202 | assert (! "invalid format specifier" ); |
203 | } |
204 | ++fmt; |
205 | } |
206 | else if (*fmt == '\n') |
207 | { |
208 | /* See whether we have to print a single newline character. */ |
209 | if (fmt == startp) |
210 | { |
211 | iov[niov].iov_base = (char *) startp; |
212 | iov[niov++].iov_len = 1; |
213 | } |
214 | else |
215 | /* No, just add it to the rest of the string. */ |
216 | ++iov[niov - 1].iov_len; |
217 | |
218 | /* Next line, print a tag again. */ |
219 | tag_p = 1; |
220 | ++fmt; |
221 | } |
222 | } |
223 | |
224 | /* Finally write the result. */ |
225 | _dl_writev (fd, iov, niov); |
226 | } |
227 | |
228 | |
229 | /* Write to debug file. */ |
230 | void |
231 | _dl_debug_printf (const char *fmt, ...) |
232 | { |
233 | va_list arg; |
234 | |
235 | va_start (arg, fmt); |
236 | _dl_debug_vdprintf (GLRO(dl_debug_fd), 1, fmt, arg); |
237 | va_end (arg); |
238 | } |
239 | |
240 | |
241 | /* Write to debug file but don't start with a tag. */ |
242 | void |
243 | _dl_debug_printf_c (const char *fmt, ...) |
244 | { |
245 | va_list arg; |
246 | |
247 | va_start (arg, fmt); |
248 | _dl_debug_vdprintf (GLRO(dl_debug_fd), -1, fmt, arg); |
249 | va_end (arg); |
250 | } |
251 | |
252 | |
253 | /* Write the given file descriptor. */ |
254 | void |
255 | _dl_dprintf (int fd, const char *fmt, ...) |
256 | { |
257 | va_list arg; |
258 | |
259 | va_start (arg, fmt); |
260 | _dl_debug_vdprintf (fd, 0, fmt, arg); |
261 | va_end (arg); |
262 | } |
263 | |
264 | void |
265 | _dl_printf (const char *fmt, ...) |
266 | { |
267 | va_list arg; |
268 | |
269 | va_start (arg, fmt); |
270 | _dl_debug_vdprintf (STDOUT_FILENO, 0, fmt, arg); |
271 | va_end (arg); |
272 | } |
273 | |
274 | void |
275 | _dl_error_printf (const char *fmt, ...) |
276 | { |
277 | va_list arg; |
278 | |
279 | va_start (arg, fmt); |
280 | _dl_debug_vdprintf (STDERR_FILENO, 0, fmt, arg); |
281 | va_end (arg); |
282 | } |
283 | |
284 | void |
285 | _dl_fatal_printf (const char *fmt, ...) |
286 | { |
287 | va_list arg; |
288 | |
289 | va_start (arg, fmt); |
290 | _dl_debug_vdprintf (STDERR_FILENO, 0, fmt, arg); |
291 | va_end (arg); |
292 | _exit (127); |
293 | } |
294 | rtld_hidden_def (_dl_fatal_printf) |
295 | |