1 |
|
|
/* $OpenBSD: safe_sprintf.c,v 1.6 2010/10/18 18:22:35 nicm Exp $ */ |
2 |
|
|
|
3 |
|
|
/**************************************************************************** |
4 |
|
|
* Copyright (c) 1998-2003,2007 Free Software Foundation, Inc. * |
5 |
|
|
* * |
6 |
|
|
* Permission is hereby granted, free of charge, to any person obtaining a * |
7 |
|
|
* copy of this software and associated documentation files (the * |
8 |
|
|
* "Software"), to deal in the Software without restriction, including * |
9 |
|
|
* without limitation the rights to use, copy, modify, merge, publish, * |
10 |
|
|
* distribute, distribute with modifications, sublicense, and/or sell * |
11 |
|
|
* copies of the Software, and to permit persons to whom the Software is * |
12 |
|
|
* furnished to do so, subject to the following conditions: * |
13 |
|
|
* * |
14 |
|
|
* The above copyright notice and this permission notice shall be included * |
15 |
|
|
* in all copies or substantial portions of the Software. * |
16 |
|
|
* * |
17 |
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS * |
18 |
|
|
* OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF * |
19 |
|
|
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. * |
20 |
|
|
* IN NO EVENT SHALL THE ABOVE COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, * |
21 |
|
|
* DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR * |
22 |
|
|
* OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR * |
23 |
|
|
* THE USE OR OTHER DEALINGS IN THE SOFTWARE. * |
24 |
|
|
* * |
25 |
|
|
* Except as contained in this notice, the name(s) of the above copyright * |
26 |
|
|
* holders shall not be used in advertising or otherwise to promote the * |
27 |
|
|
* sale, use or other dealings in this Software without prior written * |
28 |
|
|
* authorization. * |
29 |
|
|
****************************************************************************/ |
30 |
|
|
|
31 |
|
|
/**************************************************************************** |
32 |
|
|
* Author: Thomas E. Dickey <dickey@clark.net> 1997 * |
33 |
|
|
****************************************************************************/ |
34 |
|
|
|
35 |
|
|
#include <curses.priv.h> |
36 |
|
|
#include <ctype.h> |
37 |
|
|
|
38 |
|
|
MODULE_ID("$Id: safe_sprintf.c,v 1.6 2010/10/18 18:22:35 nicm Exp $") |
39 |
|
|
|
40 |
|
|
#if USE_SAFE_SPRINTF |
41 |
|
|
|
42 |
|
|
typedef enum { |
43 |
|
|
Flags, Width, Prec, Type, Format |
44 |
|
|
} PRINTF; |
45 |
|
|
|
46 |
|
|
#define VA_INTGR(type) ival = va_arg(ap, type) |
47 |
|
|
#define VA_FLOAT(type) fval = va_arg(ap, type) |
48 |
|
|
#define VA_POINT(type) pval = (void *)va_arg(ap, type) |
49 |
|
|
|
50 |
|
|
/* |
51 |
|
|
* Scan a variable-argument list for printf to determine the number of |
52 |
|
|
* characters that would be emitted. |
53 |
|
|
*/ |
54 |
|
|
static int |
55 |
|
|
_nc_printf_length(const char *fmt, va_list ap) |
56 |
|
|
{ |
57 |
|
|
size_t length = BUFSIZ; |
58 |
|
|
char *buffer; |
59 |
|
|
char *format; |
60 |
|
|
char *tmp_format; |
61 |
|
|
int len = 0; |
62 |
|
|
size_t fmt_len; |
63 |
|
|
char fmt_arg[BUFSIZ]; |
64 |
|
|
|
65 |
|
|
if (fmt == 0 || *fmt == '\0') |
66 |
|
|
return 0; |
67 |
|
|
fmt_len = strlen(fmt) + 1; |
68 |
|
|
if ((format = typeMalloc(char, fmt_len)) == 0) |
69 |
|
|
return -1; |
70 |
|
|
if ((buffer = typeMalloc(char, length)) == 0) { |
71 |
|
|
free(format); |
72 |
|
|
return -1; |
73 |
|
|
} |
74 |
|
|
|
75 |
|
|
while (*fmt != '\0') { |
76 |
|
|
if (*fmt == '%') { |
77 |
|
|
static char dummy[] = ""; |
78 |
|
|
PRINTF state = Flags; |
79 |
|
|
char *pval = dummy; /* avoid const-cast */ |
80 |
|
|
double fval = 0.0; |
81 |
|
|
int done = FALSE; |
82 |
|
|
int ival = 0; |
83 |
|
|
int prec = -1; |
84 |
|
|
int type = 0; |
85 |
|
|
int used = 0; |
86 |
|
|
int width = -1; |
87 |
|
|
size_t f = 0; |
88 |
|
|
|
89 |
|
|
format[f++] = *fmt; |
90 |
|
|
while (*++fmt != '\0' && len >= 0 && !done) { |
91 |
|
|
format[f++] = *fmt; |
92 |
|
|
|
93 |
|
|
if (isdigit(UChar(*fmt))) { |
94 |
|
|
int num = *fmt - '0'; |
95 |
|
|
if (state == Flags && num != 0) |
96 |
|
|
state = Width; |
97 |
|
|
if (state == Width) { |
98 |
|
|
if (width < 0) |
99 |
|
|
width = 0; |
100 |
|
|
width = (width * 10) + num; |
101 |
|
|
} else if (state == Prec) { |
102 |
|
|
if (prec < 0) |
103 |
|
|
prec = 0; |
104 |
|
|
prec = (prec * 10) + num; |
105 |
|
|
} |
106 |
|
|
} else if (*fmt == '*') { |
107 |
|
|
VA_INTGR(int); |
108 |
|
|
if (state == Flags) |
109 |
|
|
state = Width; |
110 |
|
|
if (state == Width) { |
111 |
|
|
width = ival; |
112 |
|
|
} else if (state == Prec) { |
113 |
|
|
prec = ival; |
114 |
|
|
} |
115 |
|
|
sprintf(fmt_arg, "%d", ival); |
116 |
|
|
fmt_len += strlen(fmt_arg); |
117 |
|
|
if ((tmp_format = realloc(format, fmt_len)) == 0) { |
118 |
|
|
free(buffer); |
119 |
|
|
free(format); |
120 |
|
|
return -1; |
121 |
|
|
} |
122 |
|
|
format = tmp_format; |
123 |
|
|
strcpy(&format[--f], fmt_arg); |
124 |
|
|
f = strlen(format); |
125 |
|
|
} else if (isalpha(UChar(*fmt))) { |
126 |
|
|
done = TRUE; |
127 |
|
|
switch (*fmt) { |
128 |
|
|
case 'Z': /* FALLTHRU */ |
129 |
|
|
case 'h': /* FALLTHRU */ |
130 |
|
|
case 'l': /* FALLTHRU */ |
131 |
|
|
done = FALSE; |
132 |
|
|
type = *fmt; |
133 |
|
|
break; |
134 |
|
|
case 'i': /* FALLTHRU */ |
135 |
|
|
case 'd': /* FALLTHRU */ |
136 |
|
|
case 'u': /* FALLTHRU */ |
137 |
|
|
case 'x': /* FALLTHRU */ |
138 |
|
|
case 'X': /* FALLTHRU */ |
139 |
|
|
if (type == 'l') |
140 |
|
|
VA_INTGR(long); |
141 |
|
|
else if (type == 'Z') |
142 |
|
|
VA_INTGR(size_t); |
143 |
|
|
else |
144 |
|
|
VA_INTGR(int); |
145 |
|
|
used = 'i'; |
146 |
|
|
break; |
147 |
|
|
case 'f': /* FALLTHRU */ |
148 |
|
|
case 'e': /* FALLTHRU */ |
149 |
|
|
case 'E': /* FALLTHRU */ |
150 |
|
|
case 'g': /* FALLTHRU */ |
151 |
|
|
case 'G': /* FALLTHRU */ |
152 |
|
|
VA_FLOAT(double); |
153 |
|
|
used = 'f'; |
154 |
|
|
break; |
155 |
|
|
case 'c': |
156 |
|
|
VA_INTGR(int); |
157 |
|
|
used = 'i'; |
158 |
|
|
break; |
159 |
|
|
case 's': |
160 |
|
|
VA_POINT(char *); |
161 |
|
|
if (prec < 0) |
162 |
|
|
prec = strlen(pval); |
163 |
|
|
if (prec > (int) length) { |
164 |
|
|
length = length + prec; |
165 |
|
|
buffer = typeRealloc(char, length, buffer); |
166 |
|
|
if (buffer == 0) { |
167 |
|
|
free(format); |
168 |
|
|
return -1; |
169 |
|
|
} |
170 |
|
|
} |
171 |
|
|
used = 'p'; |
172 |
|
|
break; |
173 |
|
|
case 'p': |
174 |
|
|
VA_POINT(void *); |
175 |
|
|
used = 'p'; |
176 |
|
|
break; |
177 |
|
|
case 'n': |
178 |
|
|
VA_POINT(int *); |
179 |
|
|
used = 0; |
180 |
|
|
break; |
181 |
|
|
default: |
182 |
|
|
break; |
183 |
|
|
} |
184 |
|
|
} else if (*fmt == '.') { |
185 |
|
|
state = Prec; |
186 |
|
|
} else if (*fmt == '%') { |
187 |
|
|
done = TRUE; |
188 |
|
|
used = 'p'; |
189 |
|
|
} |
190 |
|
|
} |
191 |
|
|
format[f] = '\0'; |
192 |
|
|
switch (used) { |
193 |
|
|
case 'i': |
194 |
|
|
sprintf(buffer, format, ival); |
195 |
|
|
break; |
196 |
|
|
case 'f': |
197 |
|
|
sprintf(buffer, format, fval); |
198 |
|
|
break; |
199 |
|
|
default: |
200 |
|
|
sprintf(buffer, format, pval); |
201 |
|
|
break; |
202 |
|
|
} |
203 |
|
|
len += (int) strlen(buffer); |
204 |
|
|
} else { |
205 |
|
|
fmt++; |
206 |
|
|
len++; |
207 |
|
|
} |
208 |
|
|
} |
209 |
|
|
|
210 |
|
|
free(buffer); |
211 |
|
|
free(format); |
212 |
|
|
return len; |
213 |
|
|
} |
214 |
|
|
#endif |
215 |
|
|
|
216 |
|
|
#define my_buffer _nc_globals.safeprint_buf |
217 |
|
|
#define my_length _nc_globals.safeprint_used |
218 |
|
|
|
219 |
|
|
/* |
220 |
|
|
* Wrapper for vsprintf that allocates a buffer big enough to hold the result. |
221 |
|
|
*/ |
222 |
|
|
NCURSES_EXPORT(char *) |
223 |
|
|
_nc_printf_string(const char *fmt, va_list ap) |
224 |
|
|
{ |
225 |
|
|
char *result = 0; |
226 |
|
|
|
227 |
|
|
if (fmt != 0) { |
228 |
|
|
#if USE_SAFE_SPRINTF |
229 |
|
|
int len = _nc_printf_length(fmt, ap); |
230 |
|
|
|
231 |
|
|
if ((int) my_length < len + 1) { |
232 |
|
|
my_length = 2 * (len + 1); |
233 |
|
|
my_buffer = typeRealloc(char, my_length, my_buffer); |
234 |
|
|
} |
235 |
|
|
if (my_buffer != 0) { |
236 |
|
|
*my_buffer = '\0'; |
237 |
|
|
if (len >= 0) { |
238 |
|
|
vsprintf(my_buffer, fmt, ap); |
239 |
|
|
} |
240 |
|
|
result = my_buffer; |
241 |
|
|
} |
242 |
|
|
#else |
243 |
|
|
#define MyCols _nc_globals.safeprint_cols |
244 |
|
|
#define MyRows _nc_globals.safeprint_rows |
245 |
|
|
|
246 |
|
|
if (screen_lines > MyRows || screen_columns > MyCols) { |
247 |
|
|
if (screen_lines > MyRows) |
248 |
|
|
MyRows = screen_lines; |
249 |
|
|
if (screen_columns > MyCols) |
250 |
|
|
MyCols = screen_columns; |
251 |
|
|
my_length = (MyRows * (MyCols + 1)) + 1; |
252 |
|
|
my_buffer = typeRealloc(char, my_length, my_buffer); |
253 |
|
|
} |
254 |
|
|
|
255 |
|
|
if (my_buffer != 0) { |
256 |
|
|
# if HAVE_VSNPRINTF |
257 |
|
|
vsnprintf(my_buffer, my_length, fmt, ap); /* GNU extension */ |
258 |
|
|
# else |
259 |
|
|
vsprintf(my_buffer, fmt, ap); /* ANSI */ |
260 |
|
|
# endif |
261 |
|
|
result = my_buffer; |
262 |
|
|
} |
263 |
|
|
#endif |
264 |
|
|
} else if (my_buffer != 0) { /* see _nc_freeall() */ |
265 |
|
|
free(my_buffer); |
266 |
|
|
my_buffer = 0; |
267 |
|
|
my_length = 0; |
268 |
|
|
} |
269 |
|
|
return result; |
270 |
|
|
} |