xref: /freebsd/contrib/tcsh/tc.printf.c (revision 050570efa79efcc9cf5adeb545f1a679c8dc377b) 
1 /* $Header: /p/tcsh/cvsroot/tcsh/tc.printf.c,v 3.35 2006/03/02 18:46:45 christos Exp $ */
2 /*
3  * tc.printf.c: A public-domain, minimal printf/sprintf routine that prints
4  *	       through the putchar() routine.  Feel free to use for
5  *	       anything...  -- 7/17/87 Paul Placeway
6  */
7 /*-
8  * Copyright (c) 1980, 1991 The Regents of the University of California.
9  * All rights reserved.
10  *
11  * Redistribution and use in source and binary forms, with or without
12  * modification, are permitted provided that the following conditions
13  * are met:
14  * 1. Redistributions of source code must retain the above copyright
15  *    notice, this list of conditions and the following disclaimer.
16  * 2. Redistributions in binary form must reproduce the above copyright
17  *    notice, this list of conditions and the following disclaimer in the
18  *    documentation and/or other materials provided with the distribution.
19  * 3. Neither the name of the University nor the names of its contributors
20  *    may be used to endorse or promote products derived from this software
21  *    without specific prior written permission.
22  *
23  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
24  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
25  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
26  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
27  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
28  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
29  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
30  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
31  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
32  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
33  * SUCH DAMAGE.
34  */
35 #include "sh.h"
36 
37 RCSID("$tcsh: tc.printf.c,v 3.35 2006/03/02 18:46:45 christos Exp $")
38 
39 #ifdef lint
40 #undef va_arg
41 #define va_arg(a, b) (a ? (b) 0 : (b) 0)
42 #endif
43 
44 #define INF	INT_MAX		/* should be bigger than any field to print */
45 
46 static char snil[] = "(nil)";
47 
48 static	void	xaddchar	(int);
49 static	void	doprnt		(void (*) (int), const char *, va_list);
50 
51 static void
52 doprnt(void (*addchar) (int), const char *sfmt, va_list ap)
53 {
54     char *bp;
55     const char *f;
56 #ifdef SHORT_STRINGS
57     const Char *Bp;
58 #endif /* SHORT_STRINGS */
59 #ifdef HAVE_LONG_LONG
60     long long l;
61     unsigned long long u;
62 #else
63     long l;
64     unsigned long u;
65 #endif
66     char buf[(CHAR_BIT * sizeof (l) + 2) / 3 + 1]; /* Octal: 3 bits per char */
67     int i;
68     int fmt;
69     unsigned char pad = ' ';
70     int     flush_left = 0, f_width = 0, prec = INF, hash = 0;
71     int	    do_long = 0, do_size_t = 0;
72     int     sign = 0;
73     int     attributes = 0;
74 
75 
76     f = sfmt;
77     for (; *f; f++) {
78 	if (*f != '%') {	/* then just out the char */
79 	    (*addchar) (((unsigned char)*f) | attributes);
80 	}
81 	else {
82 	    f++;		/* skip the % */
83 
84 	    if (*f == '-') {	/* minus: flush left */
85 		flush_left = 1;
86 		f++;
87 	    }
88 
89 	    if (*f == '0' || *f == '.') {
90 		/* padding with 0 rather than blank */
91 		pad = '0';
92 		f++;
93 	    }
94 	    if (*f == '*') {	/* field width */
95 		f_width = va_arg(ap, int);
96 		f++;
97 	    }
98 	    else if (isdigit((unsigned char) *f)) {
99 		f_width = atoi(f);
100 		while (isdigit((unsigned char) *f))
101 		    f++;	/* skip the digits */
102 	    }
103 
104 	    if (*f == '.') {	/* precision */
105 		f++;
106 		if (*f == '*') {
107 		    prec = va_arg(ap, int);
108 		    f++;
109 		}
110 		else if (isdigit((unsigned char) *f)) {
111 		    prec = atoi(f);
112 		    while (isdigit((unsigned char) *f))
113 			f++;	/* skip the digits */
114 		}
115 	    }
116 
117 	    if (*f == '#') {	/* alternate form */
118 		hash = 1;
119 		f++;
120 	    }
121 
122 	    if (*f == 'l') {	/* long format */
123 		do_long++;
124 		f++;
125 		if (*f == 'l') {
126 		    do_long++;
127 		    f++;
128 		}
129 	    }
130 	    if (*f == 'z') {	/* size_t format */
131 		do_size_t++;
132 		f++;
133 	    }
134 
135 	    fmt = (unsigned char) *f;
136 	    if (fmt != 'S' && fmt != 'Q' && isupper(fmt)) {
137 		do_long = 1;
138 		fmt = tolower(fmt);
139 	    }
140 	    bp = buf;
141 	    switch (fmt) {	/* do the format */
142 	    case 'd':
143 		switch (do_long) {
144 		case 0:
145 		    if (do_size_t)
146 			l = (long) (va_arg(ap, size_t));
147 		    else
148 			l = (long) (va_arg(ap, int));
149 		    break;
150 		case 1:
151 #ifndef HAVE_LONG_LONG
152 		default:
153 #endif
154 		    l = va_arg(ap, long);
155 		    break;
156 #ifdef HAVE_LONG_LONG
157 		default:
158 		    l = va_arg(ap, long long);
159 		    break;
160 #endif
161 		}
162 
163 		if (l < 0) {
164 		    sign = 1;
165 		    l = -l;
166 		}
167 		do {
168 		    *bp++ = (char) (l % 10) + '0';
169 		} while ((l /= 10) > 0);
170 		if (sign)
171 		    *bp++ = '-';
172 		f_width = f_width - (int) (bp - buf);
173 		if (!flush_left)
174 		    while (f_width-- > 0)
175 			(*addchar) (pad | attributes);
176 		for (bp--; bp >= buf; bp--)
177 		    (*addchar) (((unsigned char) *bp) | attributes);
178 		if (flush_left)
179 		    while (f_width-- > 0)
180 			(*addchar) (' ' | attributes);
181 		break;
182 
183 	    case 'p':
184 		do_long = 1;
185 		hash = 1;
186 		fmt = 'x';
187 		/*FALLTHROUGH*/
188 	    case 'o':
189 	    case 'x':
190 	    case 'u':
191 		switch (do_long) {
192 		case 0:
193 		    if (do_size_t)
194 			u = va_arg(ap, size_t);
195 		    else
196 			u = va_arg(ap, unsigned int);
197 		    break;
198 		case 1:
199 #ifndef HAVE_LONG_LONG
200 		default:
201 #endif
202 		    u = va_arg(ap, unsigned long);
203 		    break;
204 #ifdef HAVE_LONG_LONG
205 		default:
206 		    u = va_arg(ap, unsigned long long);
207 		    break;
208 #endif
209 		}
210 		if (fmt == 'u') {	/* unsigned decimal */
211 		    do {
212 			*bp++ = (char) (u % 10) + '0';
213 		    } while ((u /= 10) > 0);
214 		}
215 		else if (fmt == 'o') {	/* octal */
216 		    do {
217 			*bp++ = (char) (u % 8) + '0';
218 		    } while ((u /= 8) > 0);
219 		    if (hash)
220 			*bp++ = '0';
221 		}
222 		else if (fmt == 'x') {	/* hex */
223 		    do {
224 			i = (int) (u % 16);
225 			if (i < 10)
226 			    *bp++ = i + '0';
227 			else
228 			    *bp++ = i - 10 + 'a';
229 		    } while ((u /= 16) > 0);
230 		    if (hash) {
231 			*bp++ = 'x';
232 			*bp++ = '0';
233 		    }
234 		}
235 		i = f_width - (int) (bp - buf);
236 		if (!flush_left)
237 		    while (i-- > 0)
238 			(*addchar) (pad | attributes);
239 		for (bp--; bp >= buf; bp--)
240 		    (*addchar) (((unsigned char) *bp) | attributes);
241 		if (flush_left)
242 		    while (i-- > 0)
243 			(*addchar) (' ' | attributes);
244 		break;
245 
246 
247 	    case 'c':
248 		i = va_arg(ap, int);
249 		(*addchar) (i | attributes);
250 		break;
251 
252 	    case 'S':
253 	    case 'Q':
254 #ifdef SHORT_STRINGS
255 		Bp = va_arg(ap, Char *);
256 		if (!Bp) {
257 		    bp = NULL;
258 		    goto lcase_s;
259 	        }
260 		f_width = f_width - Strlen(Bp);
261 		if (!flush_left)
262 		    while (f_width-- > 0)
263 			(*addchar) ((int) (pad | attributes));
264 		for (i = 0; *Bp && i < prec; i++) {
265 		    char cbuf[MB_LEN_MAX];
266 		    size_t pos, len;
267 
268 		    if (fmt == 'Q' && *Bp & QUOTE)
269 			(*addchar) ('\\' | attributes);
270 		    len = one_wctomb(cbuf, *Bp & CHAR);
271 		    for (pos = 0; pos < len; pos++)
272 			(*addchar) ((unsigned char)cbuf[pos] | attributes
273 				    | (*Bp & ATTRIBUTES));
274 		    Bp++;
275 		}
276 		if (flush_left)
277 		    while (f_width-- > 0)
278 			(*addchar) (' ' | attributes);
279 		break;
280 #endif /* SHORT_STRINGS */
281 
282 	    case 's':
283 	    case 'q':
284 		bp = va_arg(ap, char *);
285 lcase_s:
286 		if (!bp)
287 		    bp = snil;
288 		f_width = f_width - strlen(bp);
289 		if (!flush_left)
290 		    while (f_width-- > 0)
291 			(*addchar) (pad | attributes);
292 		for (i = 0; *bp && i < prec; i++) {
293 		    if (fmt == 'q' && *bp & QUOTE)
294 			(*addchar) ('\\' | attributes);
295 		    (*addchar) (((unsigned char) *bp & TRIM) | attributes);
296 		    bp++;
297 		}
298 		if (flush_left)
299 		    while (f_width-- > 0)
300 			(*addchar) (' ' | attributes);
301 		break;
302 
303 	    case 'a':
304 		attributes = va_arg(ap, int);
305 		break;
306 
307 	    case '%':
308 		(*addchar) ('%' | attributes);
309 		break;
310 
311 	    default:
312 		break;
313 	    }
314 	    flush_left = 0, f_width = 0, prec = INF, hash = 0;
315 	    do_size_t = 0, do_long = 0;
316 	    sign = 0;
317 	    pad = ' ';
318 	}
319     }
320 }
321 
322 
323 static char *xstring, *xestring;
324 static void
325 xaddchar(int c)
326 {
327     if (xestring == xstring)
328 	*xstring = '\0';
329     else
330 	*xstring++ = (char) c;
331 }
332 
333 
334 pret_t
335 /*VARARGS*/
336 xsnprintf(char *str, size_t size, const char *fmt, ...)
337 {
338     va_list va;
339     va_start(va, fmt);
340 
341     xstring = str;
342     xestring = str + size - 1;
343     doprnt(xaddchar, fmt, va);
344     va_end(va);
345     *xstring++ = '\0';
346 #ifdef PURIFY
347     return 1;
348 #endif
349 }
350 
351 pret_t
352 /*VARARGS*/
353 xprintf(const char *fmt, ...)
354 {
355     va_list va;
356     va_start(va, fmt);
357     doprnt(xputchar, fmt, va);
358     va_end(va);
359 #ifdef PURIFY
360     return 1;
361 #endif
362 }
363 
364 
365 pret_t
366 xvprintf(const char *fmt, va_list va)
367 {
368     doprnt(xputchar, fmt, va);
369 #ifdef PURIFY
370     return 1;
371 #endif
372 }
373 
374 pret_t
375 xvsnprintf(char *str, size_t size, const char *fmt, va_list va)
376 {
377     xstring = str;
378     xestring = str + size - 1;
379     doprnt(xaddchar, fmt, va);
380     *xstring++ = '\0';
381 #ifdef PURIFY
382     return 1;
383 #endif
384 }
385 
386 char *
387 xvasprintf(const char *fmt, va_list va)
388 {
389     size_t size;
390     char *buf;
391 
392     buf = NULL;
393     size = 2048; /* Arbitrary */
394     for (;;) {
395 	va_list copy;
396 
397 	buf = xrealloc(buf, size);
398 	xstring = buf;
399 	xestring = buf + size - 1;
400 	va_copy(copy, va);
401 	doprnt(xaddchar, fmt, copy);
402 	va_end(copy);
403 	if (xstring < xestring)
404 	    break;
405 	size *= 2;
406     }
407     *xstring++ = '\0';
408     return xrealloc(buf, xstring - buf);
409 }
410 
411 char *
412 xasprintf(const char *fmt, ...)
413 {
414     va_list va;
415     char *ret;
416 
417     va_start (va, fmt);
418     ret = xvasprintf(fmt, va);
419     va_end(va);
420     return ret;
421 }
422 
423 
424 #ifdef PURIFY
425 /* Purify uses (some of..) the following functions to output memory-use
426  * debugging info.  Given all the messing with file descriptors that
427  * tcsh does, the easiest way I could think of to get it (Purify) to
428  * print anything was by replacing some standard functions with
429  * ones that do tcsh output directly - see dumb hook in doreaddirs()
430  * (sh.dir.c) -sg
431  */
432 #ifndef FILE
433 #define FILE int
434 #endif
435 int
436 fprintf(FILE *fp, const char* fmt, ...)
437 {
438     va_list va;
439     va_start(va, fmt);
440     doprnt(xputchar, fmt, va);
441     va_end(va);
442     return 1;
443 }
444 
445 int
446 vfprintf(FILE *fp, const char *fmt, va_list va)
447 {
448     doprnt(xputchar, fmt, va);
449     return 1;
450 }
451 
452 #endif	/* PURIFY */
453