subr_prf.c revision 1.20 1 /* $NetBSD: subr_prf.c,v 1.20 2011/05/20 16:33:07 tsutsui Exp $ */
2
3 /*-
4 * Copyright (c) 1993
5 * The Regents of the University of California. All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
15 * 3. Neither the name of the University nor the names of its contributors
16 * may be used to endorse or promote products derived from this software
17 * without specific prior written permission.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
23 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29 * SUCH DAMAGE.
30 *
31 * @(#)printf.c 8.1 (Berkeley) 6/11/93
32 */
33
34 /*
35 * Scaled down version of printf(3).
36 */
37
38 #include <sys/cdefs.h>
39 #include <sys/types.h>
40 #include <sys/stdint.h> /* XXX: for intptr_t */
41 #include <machine/stdarg.h>
42
43 #include "stand.h"
44
45 #ifdef LIBSA_PRINTF_LONGLONG_SUPPORT
46 #define INTMAX_T longlong_t
47 #define UINTMAX_T u_longlong_t
48 #else
49 #define INTMAX_T long
50 #define UINTMAX_T u_long
51 #endif
52
53 #if 0 /* XXX: abuse intptr_t until the situation with ptrdiff_t is clear */
54 #define PTRDIFF_T ptrdiff_t
55 #else
56 #define PTRDIFF_T intptr_t
57 #endif
58
59 #ifdef LIBSA_PRINTF_WIDTH_SUPPORT
60 static void kprintn(void (*)(int), UINTMAX_T, int, int, int);
61 #else
62 static void kprintn(void (*)(int), UINTMAX_T, int);
63 #endif
64 static void sputchar(int);
65 static void kdoprnt(void (*)(int), const char *, va_list);
66
67 static char *sbuf, *ebuf;
68
69 const char hexdigits[16] = "0123456789abcdef";
70
71 #define LONG 0x01
72 #ifdef LIBSA_PRINTF_LONGLONG_SUPPORT
73 #define LLONG 0x02
74 #endif
75 #ifdef LIBSA_PRINTF_WIDTH_SUPPORT
76 #define ALT 0x04
77 #define SPACE 0x08
78 #define LADJUST 0x10
79 #define SIGN 0x20
80 #define ZEROPAD 0x40
81 #define NEGATIVE 0x80
82 #define KPRINTN(base) kprintn(put, ul, base, lflag, width)
83 #define RZERO() \
84 do { \
85 if ((lflag & (ZEROPAD|LADJUST)) == ZEROPAD) { \
86 while (width-- > 0) \
87 put('0'); \
88 } \
89 } while (/*CONSTCOND*/0)
90 #define RPAD() \
91 do { \
92 if (lflag & LADJUST) { \
93 while (width-- > 0) \
94 put(' '); \
95 } \
96 } while (/*CONSTCOND*/0)
97 #define LPAD() \
98 do { \
99 if ((lflag & (ZEROPAD|LADJUST)) == 0) { \
100 while (width-- > 0) \
101 put(' '); \
102 } \
103 } while (/*CONSTCOND*/0)
104 #else /* LIBSA_PRINTF_WIDTH_SUPPORT */
105 #define KPRINTN(base) kprintn(put, ul, base)
106 #define RZERO() /**/
107 #define RPAD() /**/
108 #define LPAD() /**/
109 #endif /* LIBSA_PRINTF_WIDTH_SUPPORT */
110
111 #ifdef LIBSA_PRINTF_LONGLONG_SUPPORT
112 #define KPRINT(base) \
113 do { \
114 ul = (lflag & LLONG) \
115 ? va_arg(ap, u_longlong_t) \
116 : (lflag & LONG) \
117 ? va_arg(ap, u_long) \
118 : va_arg(ap, u_int); \
119 KPRINTN(base); \
120 } while (/*CONSTCOND*/0)
121 #else /* LIBSA_PRINTF_LONGLONG_SUPPORT */
122 #define KPRINT(base) \
123 do { \
124 ul = (lflag & LONG) \
125 ? va_arg(ap, u_long) : va_arg(ap, u_int); \
126 KPRINTN(base); \
127 } while (/*CONSTCOND*/0)
128 #endif /* LIBSA_PRINTF_LONGLONG_SUPPORT */
129
130 static void
131 sputchar(int c)
132 {
133
134 if (sbuf < ebuf)
135 *sbuf++ = c;
136 }
137
138 void
139 vprintf(const char *fmt, va_list ap)
140 {
141
142 kdoprnt(putchar, fmt, ap);
143 }
144
145 int
146 vsnprintf(char *buf, size_t size, const char *fmt, va_list ap)
147 {
148
149 sbuf = buf;
150 ebuf = buf + size - 1;
151 kdoprnt(sputchar, fmt, ap);
152 *sbuf = '\0';
153 return sbuf - buf;
154 }
155
156 static void
157 kdoprnt(void (*put)(int), const char *fmt, va_list ap)
158 {
159 char *p;
160 int ch;
161 UINTMAX_T ul;
162 int lflag;
163 #ifdef LIBSA_PRINTF_WIDTH_SUPPORT
164 int width;
165 char *q;
166 #endif
167
168 for (;;) {
169 while ((ch = *fmt++) != '%') {
170 if (ch == '\0')
171 return;
172 put(ch);
173 }
174 lflag = 0;
175 #ifdef LIBSA_PRINTF_WIDTH_SUPPORT
176 width = 0;
177 #endif
178 reswitch:
179 switch (ch = *fmt++) {
180 #ifdef LIBSA_PRINTF_WIDTH_SUPPORT
181 case '#':
182 lflag |= ALT;
183 goto reswitch;
184 case ' ':
185 lflag |= SPACE;
186 goto reswitch;
187 case '-':
188 lflag |= LADJUST;
189 goto reswitch;
190 case '+':
191 lflag |= SIGN;
192 goto reswitch;
193 case '0':
194 lflag |= ZEROPAD;
195 goto reswitch;
196 case '1': case '2': case '3': case '4': case '5':
197 case '6': case '7': case '8': case '9':
198 for (;;) {
199 width *= 10;
200 width += ch - '0';
201 ch = *fmt;
202 if ((unsigned)ch - '0' > 9)
203 break;
204 ++fmt;
205 }
206 #endif
207 goto reswitch;
208 case 'l':
209 #ifdef LIBSA_PRINTF_LONGLONG_SUPPORT
210 if (*fmt == 'l') {
211 ++fmt;
212 lflag |= LLONG;
213 } else
214 #endif
215 lflag |= LONG;
216 goto reswitch;
217 case 't':
218 if (sizeof(PTRDIFF_T) == sizeof(long))
219 lflag |= LONG;
220 goto reswitch;
221 case 'z':
222 if (sizeof(ssize_t) == sizeof(long))
223 lflag |= LONG;
224 goto reswitch;
225 case 'c':
226 ch = va_arg(ap, int);
227 #ifdef LIBSA_PRINTF_WIDTH_SUPPORT
228 --width;
229 #endif
230 RPAD();
231 put(ch & 0xFF);
232 LPAD();
233 break;
234 case 's':
235 p = va_arg(ap, char *);
236 #ifdef LIBSA_PRINTF_WIDTH_SUPPORT
237 for (q = p; *q != '\0'; ++q)
238 continue;
239 width -= q - p;
240 #endif
241 RPAD();
242 while ((ch = (unsigned char)*p++))
243 put(ch);
244 LPAD();
245 break;
246 case 'd':
247 ul =
248 #ifdef LIBSA_PRINTF_LONGLONG_SUPPORT
249 (lflag & LLONG) ? va_arg(ap, longlong_t) :
250 #endif
251 (lflag & LONG) ? va_arg(ap, long) : va_arg(ap, int);
252 if ((INTMAX_T)ul < 0) {
253 ul = -(INTMAX_T)ul;
254 #ifdef LIBSA_PRINTF_WIDTH_SUPPORT
255 lflag |= NEGATIVE;
256 #else
257 put('-');
258 #endif
259 }
260 KPRINTN(10);
261 break;
262 case 'o':
263 KPRINT(8);
264 break;
265 case 'u':
266 KPRINT(10);
267 break;
268 case 'p':
269 #ifdef LIBSA_PRINTF_WIDTH_SUPPORT
270 lflag |= (LONG|ALT);
271 #else
272 put('0');
273 put('x');
274 #endif
275 /* FALLTHROUGH */
276 case 'x':
277 KPRINT(16);
278 break;
279 default:
280 if (ch == '\0')
281 return;
282 put(ch);
283 break;
284 }
285 }
286 }
287
288 static void
289 #ifdef LIBSA_PRINTF_WIDTH_SUPPORT
290 kprintn(void (*put)(int), UINTMAX_T ul, int base, int lflag, int width)
291 #else
292 kprintn(void (*put)(int), UINTMAX_T ul, int base)
293 #endif
294 {
295 /* hold a INTMAX_T in base 8 */
296 char *p, buf[(sizeof(INTMAX_T) * NBBY / 3) + 1 + 2 /* ALT + SIGN */];
297 #ifdef LIBSA_PRINTF_WIDTH_SUPPORT
298 char *q;
299 #endif
300
301 p = buf;
302 do {
303 *p++ = hexdigits[ul % base];
304 } while (ul /= base);
305 #ifdef LIBSA_PRINTF_WIDTH_SUPPORT
306 q = p;
307 if (lflag & ALT && *(p - 1) != '0') {
308 if (base == 8) {
309 *p++ = '0';
310 } else if (base == 16) {
311 *p++ = 'x';
312 *p++ = '0';
313 }
314 }
315 if (lflag & NEGATIVE)
316 *p++ = '-';
317 else if (lflag & SIGN)
318 *p++ = '+';
319 else if (lflag & SPACE)
320 *p++ = ' ';
321 width -= p - buf;
322 if ((lflag & LADJUST) == 0) {
323 while (p > q)
324 put(*--p);
325 }
326 #endif
327 RPAD();
328 RZERO();
329 do {
330 put(*--p);
331 } while (p > buf);
332 LPAD();
333 }
334