Home | History | Annotate | Line # | Download | only in libsa
subr_prf.c revision 1.27.20.1
      1 /*	$NetBSD: subr_prf.c,v 1.27.20.1 2019/06/10 22:09:05 christos 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 
     42 #include "stand.h"
     43 
     44 #ifdef LIBSA_PRINTF_LONGLONG_SUPPORT
     45 #define INTMAX_T	longlong_t
     46 #define UINTMAX_T	u_longlong_t
     47 #else
     48 #define INTMAX_T	long
     49 #define UINTMAX_T	u_long
     50 #endif
     51 
     52 #if 0 /* XXX: abuse intptr_t until the situation with ptrdiff_t is clear */
     53 #define PTRDIFF_T	ptrdiff_t
     54 #else
     55 #define PTRDIFF_T	intptr_t
     56 #endif
     57 
     58 #ifdef LIBSA_PRINTF_WIDTH_SUPPORT
     59 static void kprintn(void (*)(int), UINTMAX_T, int, int, int);
     60 #else
     61 static void kprintn(void (*)(int), UINTMAX_T, int);
     62 #endif
     63 static void sputchar(int);
     64 static void kdoprnt(void (*)(int), const char *, va_list);
     65 
     66 static char *sbuf, *ebuf;
     67 
     68 const char hexdigits[16] = "0123456789abcdef";
     69 
     70 #define LONG		0x01
     71 #ifdef LIBSA_PRINTF_LONGLONG_SUPPORT
     72 #define LLONG		0x02
     73 #endif
     74 #ifdef LIBSA_PRINTF_WIDTH_SUPPORT
     75 #define ALT		0x04
     76 #define SPACE		0x08
     77 #define LADJUST		0x10
     78 #define SIGN		0x20
     79 #define ZEROPAD		0x40
     80 #define NEGATIVE	0x80
     81 #define KPRINTN(base)	kprintn(put, ul, base, lflag, width)
     82 #define RADJUSTZEROPAD()					\
     83 do {								\
     84 	if ((lflag & (ZEROPAD|LADJUST)) == ZEROPAD) {		\
     85 		while (width-- > 0)				\
     86 			put('0');				\
     87 	}							\
     88 } while (/*CONSTCOND*/0)
     89 #define LADJUSTPAD()						\
     90 do {								\
     91 	if (lflag & LADJUST) {					\
     92 		while (width-- > 0)				\
     93 			put(' ');				\
     94 	}							\
     95 } while (/*CONSTCOND*/0)
     96 #define RADJUSTPAD()						\
     97 do {								\
     98 	if ((lflag & (ZEROPAD|LADJUST)) == 0) {			\
     99 		while (width-- > 0)				\
    100 			put(' ');				\
    101 	}							\
    102 } while (/*CONSTCOND*/0)
    103 #else	/* LIBSA_PRINTF_WIDTH_SUPPORT */
    104 #define KPRINTN(base)	kprintn(put, ul, base)
    105 #define RADJUSTZEROPAD()	/**/
    106 #define LADJUSTPAD()		/**/
    107 #define RADJUSTPAD()		/**/
    108 #endif	/* LIBSA_PRINTF_WIDTH_SUPPORT */
    109 
    110 #ifdef LIBSA_PRINTF_LONGLONG_SUPPORT
    111 #define KPRINT(base)						\
    112 do {								\
    113 	ul = (lflag & LLONG)					\
    114 	    ? va_arg(ap, u_longlong_t)				\
    115 	    : (lflag & LONG)					\
    116 		? va_arg(ap, u_long)				\
    117 		: va_arg(ap, u_int);				\
    118 	KPRINTN(base);						\
    119 } while (/*CONSTCOND*/0)
    120 #else	/* LIBSA_PRINTF_LONGLONG_SUPPORT */
    121 #define KPRINT(base)						\
    122 do {								\
    123 	ul = (lflag & LONG)					\
    124 	    ? va_arg(ap, u_long) : va_arg(ap, u_int);		\
    125 	KPRINTN(base);						\
    126 } while (/*CONSTCOND*/0)
    127 #endif	/* LIBSA_PRINTF_LONGLONG_SUPPORT */
    128 
    129 static void
    130 sputchar(int c)
    131 {
    132 
    133 	if (sbuf < ebuf)
    134 		*sbuf++ = c;
    135 }
    136 
    137 void
    138 vprintf(const char *fmt, va_list ap)
    139 {
    140 
    141 	kdoprnt(putchar, fmt, ap);
    142 }
    143 
    144 int
    145 vsnprintf(char *buf, size_t size, const char *fmt, va_list ap)
    146 {
    147 
    148 	sbuf = buf;
    149 	ebuf = buf + size - 1;
    150 	kdoprnt(sputchar, fmt, ap);
    151 	*sbuf = '\0';
    152 	return sbuf - buf;
    153 }
    154 
    155 static void
    156 kdoprnt(void (*put)(int), const char *fmt, va_list ap)
    157 {
    158 	char *p;
    159 	int ch;
    160 	UINTMAX_T ul;
    161 	int lflag;
    162 #ifdef LIBSA_PRINTF_WIDTH_SUPPORT
    163 	int width;
    164 	char *q;
    165 #endif
    166 
    167 	for (;;) {
    168 		while ((ch = *fmt++) != '%') {
    169 			if (ch == '\0')
    170 				return;
    171 			put(ch);
    172 		}
    173 		lflag = 0;
    174 #ifdef LIBSA_PRINTF_WIDTH_SUPPORT
    175 		width = 0;
    176 #endif
    177 reswitch:
    178 		switch (ch = *fmt++) {
    179 #ifdef LIBSA_PRINTF_WIDTH_SUPPORT
    180 		case '#':
    181 			lflag |= ALT;
    182 			goto reswitch;
    183 		case ' ':
    184 			lflag |= SPACE;
    185 			goto reswitch;
    186 		case '-':
    187 			lflag |= LADJUST;
    188 			goto reswitch;
    189 		case '+':
    190 			lflag |= SIGN;
    191 			goto reswitch;
    192 		case '0':
    193 			lflag |= ZEROPAD;
    194 			goto reswitch;
    195 		case '1': case '2': case '3': case '4': case '5':
    196 		case '6': case '7': case '8': case '9':
    197 			for (;;) {
    198 				width *= 10;
    199 				width += ch - '0';
    200 				ch = *fmt;
    201 				if ((unsigned)ch - '0' > 9)
    202 					break;
    203 				++fmt;
    204 			}
    205 #endif
    206 		case 'l':
    207 #ifdef LIBSA_PRINTF_LONGLONG_SUPPORT
    208 			if (*fmt == 'l') {
    209 				++fmt;
    210 				lflag |= LLONG;
    211 			} else
    212 #endif
    213 				lflag |= LONG;
    214 			goto reswitch;
    215 		case 'j':
    216 #ifdef LIBSA_PRINTF_LONGLONG_SUPPORT
    217 			if (sizeof(intmax_t) == sizeof(long long))
    218 				lflag |= LLONG;
    219 			else
    220 #endif
    221 			if (sizeof(intmax_t) == sizeof(long))
    222 				lflag |= LONG;
    223 			goto reswitch;
    224 		case 't':
    225 			if (sizeof(PTRDIFF_T) == sizeof(long))
    226 				lflag |= LONG;
    227 			goto reswitch;
    228 		case 'z':
    229 			if (sizeof(ssize_t) == sizeof(long))
    230 				lflag |= LONG;
    231 			goto reswitch;
    232 		case 'c':
    233 			ch = va_arg(ap, int);
    234 #ifdef LIBSA_PRINTF_WIDTH_SUPPORT
    235 			--width;
    236 #endif
    237 			RADJUSTPAD();
    238 			put(ch & 0xFF);
    239 			LADJUSTPAD();
    240 			break;
    241 		case 's':
    242 			p = va_arg(ap, char *);
    243 #ifdef LIBSA_PRINTF_WIDTH_SUPPORT
    244 			for (q = p; *q != '\0'; ++q)
    245 				continue;
    246 			width -= q - p;
    247 #endif
    248 			RADJUSTPAD();
    249 			while ((ch = (unsigned char)*p++))
    250 				put(ch);
    251 			LADJUSTPAD();
    252 			break;
    253 		case 'd':
    254 			ul =
    255 #ifdef LIBSA_PRINTF_LONGLONG_SUPPORT
    256 			(lflag & LLONG) ? va_arg(ap, longlong_t) :
    257 #endif
    258 			(lflag & LONG) ? va_arg(ap, long) : va_arg(ap, int);
    259 			if ((INTMAX_T)ul < 0) {
    260 				ul = -(INTMAX_T)ul;
    261 #ifdef LIBSA_PRINTF_WIDTH_SUPPORT
    262 				lflag |= NEGATIVE;
    263 #else
    264 				put('-');
    265 #endif
    266 			}
    267 			KPRINTN(10);
    268 			break;
    269 		case 'o':
    270 			KPRINT(8);
    271 			break;
    272 		case 'u':
    273 			KPRINT(10);
    274 			break;
    275 		case 'p':
    276 #ifdef LIBSA_PRINTF_WIDTH_SUPPORT
    277 			lflag |= (LONG|ALT);
    278 #else
    279 			put('0');
    280 			put('x');
    281 #endif
    282 			/* FALLTHROUGH */
    283 		case 'x':
    284 			KPRINT(16);
    285 			break;
    286 		default:
    287 			if (ch == '\0')
    288 				return;
    289 			put(ch);
    290 			break;
    291 		}
    292 	}
    293 }
    294 
    295 static void
    296 #ifdef LIBSA_PRINTF_WIDTH_SUPPORT
    297 kprintn(void (*put)(int), UINTMAX_T ul, int base, int lflag, int width)
    298 #else
    299 kprintn(void (*put)(int), UINTMAX_T ul, int base)
    300 #endif
    301 {
    302 					/* hold a INTMAX_T in base 8 */
    303 	char *p, buf[(sizeof(INTMAX_T) * NBBY / 3) + 1 + 2 /* ALT + SIGN */];
    304 #ifdef LIBSA_PRINTF_WIDTH_SUPPORT
    305 	char *q;
    306 #endif
    307 
    308 	p = buf;
    309 	do {
    310 		*p++ = hexdigits[ul % base];
    311 	} while (ul /= base);
    312 #ifdef LIBSA_PRINTF_WIDTH_SUPPORT
    313 	q = p;
    314 	if (lflag & ALT && *(p - 1) != '0') {
    315 		if (base == 8) {
    316 			*p++ = '0';
    317 		} else if (base == 16) {
    318 			*p++ = 'x';
    319 			*p++ = '0';
    320 		}
    321 	}
    322 	if (lflag & NEGATIVE)
    323 		*p++ = '-';
    324 	else if (lflag & SIGN)
    325 		*p++ = '+';
    326 	else if (lflag & SPACE)
    327 		*p++ = ' ';
    328 	width -= p - buf;
    329 	if (lflag & ZEROPAD) {
    330 		while (p > q)
    331 			put(*--p);
    332 	}
    333 #endif
    334 	RADJUSTPAD();
    335 	RADJUSTZEROPAD();
    336 	do {
    337 		put(*--p);
    338 	} while (p > buf);
    339 	LADJUSTPAD();
    340 }
    341