ncurses 5.6 - patch 20070224
[ncurses.git] / ncurses / base / safe_sprintf.c
index cb48365f7f51074571dd196b8dfebae1161409b7..81fe44f0c783d6945e0618af7eee095d0ea8914a 100644 (file)
@@ -1,5 +1,5 @@
 /****************************************************************************
- * Copyright (c) 1998,1999,2000 Free Software Foundation, Inc.              *
+ * Copyright (c) 1998-2001,2003 Free Software Foundation, Inc.              *
  *                                                                          *
  * Permission is hereby granted, free of charge, to any person obtaining a  *
  * copy of this software and associated documentation files (the            *
@@ -33,7 +33,7 @@
 #include <curses.priv.h>
 #include <ctype.h>
 
-MODULE_ID("$Id: safe_sprintf.c,v 1.14 2001/07/08 00:58:34 tom Exp $")
+MODULE_ID("$Id: safe_sprintf.c,v 1.18 2003/08/09 21:52:04 tom Exp $")
 
 #if USE_SAFE_SPRINTF
 
@@ -56,10 +56,13 @@ _nc_printf_length(const char *fmt, va_list ap)
     char *buffer;
     char *format;
     int len = 0;
+    size_t fmt_len;
+    char fmt_arg[BUFSIZ];
 
     if (fmt == 0 || *fmt == '\0')
-       return -1;
-    if ((format = typeMalloc(char, strlen(fmt) + 1)) == 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);
@@ -106,7 +109,12 @@ _nc_printf_length(const char *fmt, va_list ap)
                    } else if (state == Prec) {
                        prec = ival;
                    }
-                   sprintf(&format[--f], "%d", 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;
@@ -203,42 +211,52 @@ _nc_printf_length(const char *fmt, va_list ap)
  * Wrapper for vsprintf that allocates a buffer big enough to hold the result.
  */
 NCURSES_EXPORT(char *)
-_nc_printf_string
-(const char *fmt, va_list ap)
+_nc_printf_string(const char *fmt, va_list ap)
 {
+    static char *buf;
+    static size_t used;
+    char *result = 0;
+
+    if (fmt != 0) {
 #if USE_SAFE_SPRINTF
-    char *buf = 0;
-    int len = _nc_printf_length(fmt, ap);
+       int len = _nc_printf_length(fmt, ap);
 
-    if (len > 0) {
-       if ((buf = typeMalloc(char, len + 1)) == 0)
-             return (0);
-       vsprintf(buf, fmt, ap);
-    }
+       if ((int) used < len + 1) {
+           used = 2 * (len + 1);
+           buf = typeRealloc(char, used, buf);
+       }
+       if (buf != 0) {
+           *buf = '\0';
+           if (len >= 0) {
+               vsprintf(buf, fmt, ap);
+           }
+           result = buf;
+       }
 #else
-    static int rows, cols;
-    static char *buf;
-    static size_t len;
-
-    if (screen_lines > rows || screen_columns > cols) {
-       if (screen_lines > rows)
-           rows = screen_lines;
-       if (screen_columns > cols)
-           cols = screen_columns;
-       len = (rows * (cols + 1)) + 1;
-       buf = typeRealloc(char, len, buf);
-       if (buf == 0) {
-           return (0);
+       static int rows, cols;
+
+       if (screen_lines > rows || screen_columns > cols) {
+           if (screen_lines > rows)
+               rows = screen_lines;
+           if (screen_columns > cols)
+               cols = screen_columns;
+           used = (rows * (cols + 1)) + 1;
+           buf = typeRealloc(char, used, buf);
        }
-    }
 
-    if (buf != 0) {
+       if (buf != 0) {
 # if HAVE_VSNPRINTF
-       vsnprintf(buf, len, fmt, ap);   /* GNU extension */
+           vsnprintf(buf, used, fmt, ap);      /* GNU extension */
 # else
-       vsprintf(buf, fmt, ap); /* ANSI */
+           vsprintf(buf, fmt, ap);     /* ANSI */
 # endif
-    }
+           result = buf;
+       }
 #endif
-    return buf;
+    } else if (buf != 0) {     /* see _nc_freeall() */
+       free(buf);
+       buf = 0;
+       used = 0;
+    }
+    return result;
 }