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: stable/11/contrib/libarchive/libarchive/archive_string_sprintf.c 316337 2017-03-31 20:16:24Z mm $");
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#ifdef HAVE_ERRNO_H
42#include <errno.h>
43#endif
44#include <stdio.h>
45
46#include "archive_string.h"
47#include "archive_private.h"
48
49/*
50 * Utility functions to format signed/unsigned integers and append
51 * them to an archive_string.
52 */
53static void
54append_uint(struct archive_string *as, uintmax_t d, unsigned base)
55{
56	static const char digits[] = "0123456789abcdef";
57	if (d >= base)
58		append_uint(as, d/base, base);
59	archive_strappend_char(as, digits[d % base]);
60}
61
62static void
63append_int(struct archive_string *as, intmax_t d, unsigned base)
64{
65	uintmax_t ud;
66
67	if (d < 0) {
68		archive_strappend_char(as, '-');
69		ud = (d == INTMAX_MIN) ? (uintmax_t)(INTMAX_MAX) + 1 : (uintmax_t)(-d);
70	} else
71		ud = d;
72	append_uint(as, ud, base);
73}
74
75
76void
77archive_string_sprintf(struct archive_string *as, const char *fmt, ...)
78{
79	va_list ap;
80
81	va_start(ap, fmt);
82	archive_string_vsprintf(as, fmt, ap);
83	va_end(ap);
84}
85
86/*
87 * Like 'vsprintf', but ensures the target is big enough, resizing if
88 * necessary.
89 */
90void
91archive_string_vsprintf(struct archive_string *as, const char *fmt,
92    va_list ap)
93{
94	char long_flag;
95	intmax_t s; /* Signed integer temp. */
96	uintmax_t u; /* Unsigned integer temp. */
97	const char *p, *p2;
98	const wchar_t *pw;
99
100	if (archive_string_ensure(as, 64) == NULL)
101		__archive_errx(1, "Out of memory");
102
103	if (fmt == NULL) {
104		as->s[0] = 0;
105		return;
106	}
107
108	for (p = fmt; *p != '\0'; p++) {
109		const char *saved_p = p;
110
111		if (*p != '%') {
112			archive_strappend_char(as, *p);
113			continue;
114		}
115
116		p++;
117
118		long_flag = '\0';
119		switch(*p) {
120		case 'j':
121		case 'l':
122		case 'z':
123			long_flag = *p;
124			p++;
125			break;
126		}
127
128		switch (*p) {
129		case '%':
130			archive_strappend_char(as, '%');
131			break;
132		case 'c':
133			s = va_arg(ap, int);
134			archive_strappend_char(as, (char)s);
135			break;
136		case 'd':
137			switch(long_flag) {
138			case 'j': s = va_arg(ap, intmax_t); break;
139			case 'l': s = va_arg(ap, long); break;
140			case 'z': s = va_arg(ap, ssize_t); break;
141			default:  s = va_arg(ap, int); break;
142			}
143		        append_int(as, s, 10);
144			break;
145		case 's':
146			switch(long_flag) {
147			case 'l':
148				pw = va_arg(ap, wchar_t *);
149				if (pw == NULL)
150					pw = L"(null)";
151				if (archive_string_append_from_wcs(as, pw,
152				    wcslen(pw)) != 0 && errno == ENOMEM)
153					__archive_errx(1, "Out of memory");
154				break;
155			default:
156				p2 = va_arg(ap, char *);
157				if (p2 == NULL)
158					p2 = "(null)";
159				archive_strcat(as, p2);
160				break;
161			}
162			break;
163		case 'S':
164			pw = va_arg(ap, wchar_t *);
165			if (pw == NULL)
166				pw = L"(null)";
167			if (archive_string_append_from_wcs(as, pw,
168			    wcslen(pw)) != 0 && errno == ENOMEM)
169				__archive_errx(1, "Out of memory");
170			break;
171		case 'o': case 'u': case 'x': case 'X':
172			/* Common handling for unsigned integer formats. */
173			switch(long_flag) {
174			case 'j': u = va_arg(ap, uintmax_t); break;
175			case 'l': u = va_arg(ap, unsigned long); break;
176			case 'z': u = va_arg(ap, size_t); break;
177			default:  u = va_arg(ap, unsigned int); break;
178			}
179			/* Format it in the correct base. */
180			switch (*p) {
181			case 'o': append_uint(as, u, 8); break;
182			case 'u': append_uint(as, u, 10); break;
183			default: append_uint(as, u, 16); break;
184			}
185			break;
186		default:
187			/* Rewind and print the initial '%' literally. */
188			p = saved_p;
189			archive_strappend_char(as, *p);
190		}
191	}
192}
193