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 #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 */
53 static void
append_uint(struct archive_string * as,uintmax_t d,unsigned base)54 append_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
62 static void
append_int(struct archive_string * as,intmax_t d,unsigned base)63 append_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
76 void
archive_string_sprintf(struct archive_string * as,const char * fmt,...)77 archive_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 */
90 void
archive_string_vsprintf(struct archive_string * as,const char * fmt,va_list ap)91 archive_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