#include <curses.priv.h>
#include <ctype.h>
MODULE_ID("$Id: safe_sprintf.c,v 1.20 2007/04/21 22:28:06 tom Exp $")
#if USE_SAFE_SPRINTF
typedef enum {
Flags, Width, Prec, Type, Format
} PRINTF;
#define VA_INTGR(type) ival = va_arg(ap, type)
#define VA_FLOAT(type) fval = va_arg(ap, type)
#define VA_POINT(type) pval = (void *)va_arg(ap, type)
static int
_nc_printf_length(const char *fmt, va_list ap)
{
size_t length = BUFSIZ;
char *buffer;
char *format;
int len = 0;
size_t fmt_len;
char fmt_arg[BUFSIZ];
if (fmt == 0 || *fmt == '\0')
return 0;
fmt_len = strlen(fmt) + 1;
if ((format = typeMalloc(char, fmt_len)) == 0)
return -1;
if ((buffer = typeMalloc(char, length)) == 0) {
free(format);
return -1;
}
while (*fmt != '\0') {
if (*fmt == '%') {
static char dummy[] = "";
PRINTF state = Flags;
char *pval = dummy;
double fval = 0.0;
int done = FALSE;
int ival = 0;
int prec = -1;
int type = 0;
int used = 0;
int width = -1;
size_t f = 0;
format[f++] = *fmt;
while (*++fmt != '\0' && len >= 0 && !done) {
format[f++] = *fmt;
if (isdigit(UChar(*fmt))) {
int num = *fmt - '0';
if (state == Flags && num != 0)
state = Width;
if (state == Width) {
if (width < 0)
width = 0;
width = (width * 10) + num;
} else if (state == Prec) {
if (prec < 0)
prec = 0;
prec = (prec * 10) + num;
}
} else if (*fmt == '*') {
VA_INTGR(int);
if (state == Flags)
state = Width;
if (state == Width) {
width = ival;
} else if (state == Prec) {
prec = ival;
}
sprintf(fmt_arg, "%d", ival);
fmt_len += strlen(fmt_arg);
if ((format = realloc(format, fmt_len)) == 0) {
return -1;
}
strcpy(&format[--f], fmt_arg);
f = strlen(format);
} else if (isalpha(UChar(*fmt))) {
done = TRUE;
switch (*fmt) {
case 'Z':
case 'h':
case 'l':
done = FALSE;
type = *fmt;
break;
case 'i':
case 'd':
case 'u':
case 'x':
case 'X':
if (type == 'l')
VA_INTGR(long);
else if (type == 'Z')
VA_INTGR(size_t);
else
VA_INTGR(int);
used = 'i';
break;
case 'f':
case 'e':
case 'E':
case 'g':
case 'G':
VA_FLOAT(double);
used = 'f';
break;
case 'c':
VA_INTGR(int);
used = 'i';
break;
case 's':
VA_POINT(char *);
if (prec < 0)
prec = strlen(pval);
if (prec > (int) length) {
length = length + prec;
buffer = typeRealloc(char, length, buffer);
if (buffer == 0) {
free(format);
return -1;
}
}
used = 'p';
break;
case 'p':
VA_POINT(void *);
used = 'p';
break;
case 'n':
VA_POINT(int *);
used = 0;
break;
default:
break;
}
} else if (*fmt == '.') {
state = Prec;
} else if (*fmt == '%') {
done = TRUE;
used = 'p';
}
}
format[f] = '\0';
switch (used) {
case 'i':
sprintf(buffer, format, ival);
break;
case 'f':
sprintf(buffer, format, fval);
break;
default:
sprintf(buffer, format, pval);
break;
}
len += (int) strlen(buffer);
} else {
fmt++;
len++;
}
}
free(buffer);
free(format);
return len;
}
#endif
#define my_buffer _nc_globals.safeprint_buf
#define my_length _nc_globals.safeprint_used
NCURSES_EXPORT(char *)
_nc_printf_string(const char *fmt, va_list ap)
{
char *result = 0;
if (fmt != 0) {
#if USE_SAFE_SPRINTF
int len = _nc_printf_length(fmt, ap);
if ((int) my_length < len + 1) {
my_length = 2 * (len + 1);
my_buffer = typeRealloc(char, my_length, my_buffer);
}
if (my_buffer != 0) {
*my_buffer = '\0';
if (len >= 0) {
vsprintf(my_buffer, fmt, ap);
}
result = my_buffer;
}
#else
#define MyCols _nc_globals.safeprint_cols
#define MyRows _nc_globals.safeprint_rows
if (screen_lines > MyRows || screen_columns > MyCols) {
if (screen_lines > MyRows)
MyRows = screen_lines;
if (screen_columns > MyCols)
MyCols = screen_columns;
my_length = (MyRows * (MyCols + 1)) + 1;
my_buffer = typeRealloc(char, my_length, my_buffer);
}
if (my_buffer != 0) {
# if HAVE_VSNPRINTF
vsnprintf(my_buffer, my_length, fmt, ap);
# else
vsprintf(my_buffer, fmt, ap);
# endif
result = my_buffer;
}
#endif
} else if (my_buffer != 0) {
free(my_buffer);
my_buffer = 0;
my_length = 0;
}
return result;
}