1/*-
2 * Copyright (c) 2003-2007 Tim Kientzle
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 *
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR
15 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
16 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
17 * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT,
18 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
19 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
20 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
21 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
23 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24 */
25
26#include "archive_platform.h"
27__FBSDID("$FreeBSD: head/lib/libarchive/archive_string_sprintf.c 189435 2009-03-06 05:14:55Z kientzle $");
28
29/*
30 * The use of printf()-family functions can be troublesome
31 * for space-constrained applications.  In addition, correctly
32 * implementing this function in terms of vsnprintf() requires
33 * two calls (one to determine the size, another to format the
34 * result), which in turn requires duplicating the argument list
35 * using va_copy, which isn't yet universally available. <sigh>
36 *
37 * So, I've implemented a bare minimum of printf()-like capability
38 * here.  This is only used to format error messages, so doesn't
39 * require any floating-point support or field-width handling.
40 */
41
42#include <stdio.h>
43
44#include "archive_string.h"
45#include "archive_private.h"
46
47/*
48 * Utility functions to format signed/unsigned integers and append
49 * them to an archive_string.
50 */
51static void
52append_uint(struct archive_string *as, uintmax_t d, unsigned base)
53{
54	static const char *digits = "0123456789abcdef";
55	if (d >= base)
56		append_uint(as, d/base, base);
57	archive_strappend_char(as, digits[d % base]);
58}
59
60static void
61append_int(struct archive_string *as, intmax_t d, unsigned base)
62{
63	if (d < 0) {
64		archive_strappend_char(as, '-');
65		d = -d;
66	}
67	append_uint(as, d, base);
68}
69
70
71void
72__archive_string_sprintf(struct archive_string *as, const char *fmt, ...)
73{
74	va_list ap;
75
76	va_start(ap, fmt);
77	archive_string_vsprintf(as, fmt, ap);
78	va_end(ap);
79}
80
81/*
82 * Like 'vsprintf', but ensures the target is big enough, resizing if
83 * necessary.
84 */
85void
86__archive_string_vsprintf(struct archive_string *as, const char *fmt,
87    va_list ap)
88{
89	char long_flag;
90	intmax_t s; /* Signed integer temp. */
91	uintmax_t u; /* Unsigned integer temp. */
92	const char *p, *p2;
93
94	if (__archive_string_ensure(as, 64) == NULL)
95		__archive_errx(1, "Out of memory");
96
97	if (fmt == NULL) {
98		as->s[0] = 0;
99		return;
100	}
101
102	for (p = fmt; *p != '\0'; p++) {
103		const char *saved_p = p;
104
105		if (*p != '%') {
106			archive_strappend_char(as, *p);
107			continue;
108		}
109
110		p++;
111
112		long_flag = '\0';
113		switch(*p) {
114		case 'j':
115			long_flag = 'j';
116			p++;
117			break;
118		case 'l':
119			long_flag = 'l';
120			p++;
121			break;
122		}
123
124		switch (*p) {
125		case '%':
126			__archive_strappend_char(as, '%');
127			break;
128		case 'c':
129			s = va_arg(ap, int);
130			__archive_strappend_char(as, s);
131			break;
132		case 'd':
133			switch(long_flag) {
134			case 'j': s = va_arg(ap, intmax_t); break;
135			case 'l': s = va_arg(ap, long); break;
136			default:  s = va_arg(ap, int); break;
137			}
138		        append_int(as, s, 10);
139			break;
140		case 's':
141			p2 = va_arg(ap, char *);
142			archive_strcat(as, p2);
143			break;
144		case 'o': case 'u': case 'x': case 'X':
145			/* Common handling for unsigned integer formats. */
146			switch(long_flag) {
147			case 'j': u = va_arg(ap, uintmax_t); break;
148			case 'l': u = va_arg(ap, unsigned long); break;
149			default:  u = va_arg(ap, unsigned int); break;
150			}
151			/* Format it in the correct base. */
152			switch (*p) {
153			case 'o': append_uint(as, u, 8); break;
154			case 'u': append_uint(as, u, 10); break;
155			default: append_uint(as, u, 16); break;
156			}
157			break;
158		default:
159			/* Rewind and print the initial '%' literally. */
160			p = saved_p;
161			archive_strappend_char(as, *p);
162		}
163	}
164}
165