]> ncurses.scripts.mit.edu Git - ncurses.git/blobdiff - ncurses/base/safe_sprintf.c
ncurses 5.9 - patch 20130119
[ncurses.git] / ncurses / base / safe_sprintf.c
index e4d525272627b7c153c99d4bdb036b21e6965254..34abd2f8cf47409bfe9973a1dfba78f96b861f06 100644 (file)
@@ -1,5 +1,5 @@
 /****************************************************************************
- * Copyright (c) 1998,1999 Free Software Foundation, Inc.                   *
+ * Copyright (c) 1998-2012,2013 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            *
  ****************************************************************************/
 
 /****************************************************************************
- *  Author: Thomas E. Dickey <dickey@clark.net> 1997                        *
+ *  Author: Thomas E. Dickey        1997-on                                 *
  ****************************************************************************/
 
 #include <curses.priv.h>
 #include <ctype.h>
 
-MODULE_ID("$Id: safe_sprintf.c,v 1.11 1999/09/11 18:03:27 tom Exp $")
+MODULE_ID("$Id: safe_sprintf.c,v 1.27 2013/01/20 01:04:32 tom Exp $")
 
 #if USE_SAFE_SPRINTF
 
-typedef enum { Flags, Width, Prec, Type, Format } PRINTF;
+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)
@@ -50,190 +52,232 @@ typedef enum { Flags, Width, Prec, Type, Format } PRINTF;
 static int
 _nc_printf_length(const char *fmt, va_list ap)
 {
-       size_t length = BUFSIZ;
-       char *buffer;
-       char *format;
-       int len = 0;
-
-       if (fmt == 0 || *fmt == '\0')
-               return -1;
-       if ((format = typeMalloc(char, strlen(fmt)+1)) == 0)
-               return -1;
-       if ((buffer = typeMalloc(char, length)) == 0) {
-               free(format);
-               return -1;
-       }
+    size_t length = BUFSIZ;
+    char *buffer;
+    char *format;
+    int len = 0;
+    size_t fmt_len;
+    char fmt_arg[BUFSIZ];
 
-       while (*fmt != '\0') {
-               if (*fmt == '%') {
-                       static char dummy[] = "";
-                       PRINTF state = Flags;
-                       char *pval   = dummy;   /* avoid const-cast */
-                       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(*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(&format[--f], "%d", ival);
-                                       f = strlen(format);
-                               } else if (isalpha(*fmt)) {
-                                       done = TRUE;
-                                       switch (*fmt) {
-                                       case 'Z': /* FALLTHRU */
-                                       case 'h': /* FALLTHRU */
-                                       case 'l': /* FALLTHRU */
-                                               done = FALSE;
-                                               type = *fmt;
-                                               break;
-                                       case 'i': /* FALLTHRU */
-                                       case 'd': /* FALLTHRU */
-                                       case 'u': /* FALLTHRU */
-                                       case 'x': /* FALLTHRU */
-                                       case 'X': /* FALLTHRU */
-                                               if (type == 'l')
-                                                       VA_INTGR(long);
-                                               else if (type == 'Z')
-                                                       VA_INTGR(size_t);
-                                               else
-                                                       VA_INTGR(int);
-                                               used = 'i';
-                                               break;
-                                       case 'f': /* FALLTHRU */
-                                       case 'e': /* FALLTHRU */
-                                       case 'E': /* FALLTHRU */
-                                       case 'g': /* FALLTHRU */
-                                       case 'G': /* FALLTHRU */
-                                               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;
+    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; /* avoid const-cast */
+           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;
+                   }
+                   _nc_SPRINTF(fmt_arg,
+                               _nc_SLIMIT(sizeof(fmt_arg))
+                               "%d", ival);
+                   fmt_len += strlen(fmt_arg);
+                   if ((format = _nc_doalloc(format, fmt_len)) == 0) {
+                       free(buffer);
+                       return -1;
+                   }
+                   --f;
+                   _nc_STRCPY(&format[f], fmt_arg, fmt_len - f);
+                   f = strlen(format);
+               } else if (isalpha(UChar(*fmt))) {
+                   done = TRUE;
+                   switch (*fmt) {
+                   case 'Z':   /* FALLTHRU */
+                   case 'h':   /* FALLTHRU */
+                   case 'l':   /* FALLTHRU */
+                       done = FALSE;
+                       type = *fmt;
+                       break;
+                   case 'i':   /* FALLTHRU */
+                   case 'd':   /* FALLTHRU */
+                   case 'u':   /* FALLTHRU */
+                   case 'x':   /* FALLTHRU */
+                   case 'X':   /* FALLTHRU */
+                       if (type == 'l')
+                           VA_INTGR(long);
+                       else if (type == 'Z')
+                           VA_INTGR(size_t);
+                       else
+                           VA_INTGR(int);
+                       used = 'i';
+                       break;
+                   case 'f':   /* FALLTHRU */
+                   case 'e':   /* FALLTHRU */
+                   case 'E':   /* FALLTHRU */
+                   case 'g':   /* FALLTHRU */
+                   case 'G':   /* FALLTHRU */
+                       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;
+                           }
                        }
-                       len += (int)strlen(buffer);
-               } else {
-                       fmt++;
-                       len++;
+                       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':
+               _nc_SPRINTF(buffer, _nc_SLIMIT(length) format, ival);
+               break;
+           case 'f':
+               _nc_SPRINTF(buffer, _nc_SLIMIT(length) format, fval);
+               break;
+           default:
+               _nc_SPRINTF(buffer, _nc_SLIMIT(length) format, pval);
+               break;
+           }
+           len += (int) strlen(buffer);
+       } else {
+           fmt++;
+           len++;
        }
+    }
 
-       free(buffer);
-       free(format);
-       return len;
+    free(buffer);
+    free(format);
+    return len;
 }
 #endif
 
+#define my_buffer _nc_globals.safeprint_buf
+#define my_length _nc_globals.safeprint_used
+
 /*
  * Wrapper for vsprintf that allocates a buffer big enough to hold the result.
  */
-char *
-_nc_printf_string(const char *fmt, va_list ap)
+NCURSES_EXPORT(char *)
+NCURSES_SP_NAME(_nc_printf_string) (NCURSES_SP_DCLx
+                                   const char *fmt,
+                                   va_list ap)
 {
+    char *result = 0;
+
+    if (fmt != 0) {
 #if USE_SAFE_SPRINTF
-       char *buf = 0;
-       int len = _nc_printf_length(fmt, ap);
+       va_list ap2;
+       int len;
 
-       if (len > 0) {
-               if ((buf = typeMalloc(char, len+1)) == 0)
-                       return(0);
-               vsprintf(buf, fmt, ap);
+       begin_va_copy(ap2, ap);
+       len = _nc_printf_length(fmt, ap2);
+       end_va_copy(ap2);
+
+       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
-       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);
-               }
+#define MyCols _nc_globals.safeprint_cols
+#define MyRows _nc_globals.safeprint_rows
+
+       if (screen_lines(SP_PARM) > MyRows || screen_columns(SP_PARM) > MyCols) {
+           if (screen_lines(SP_PARM) > MyRows)
+               MyRows = screen_lines(SP_PARM);
+           if (screen_columns(SP_PARM) > MyCols)
+               MyCols = screen_columns(SP_PARM);
+           my_length = (size_t) (MyRows * (MyCols + 1)) + 1;
+           my_buffer = typeRealloc(char, my_length, my_buffer);
        }
 
-       if (buf != 0) {
+       if (my_buffer != 0) {
 # if HAVE_VSNPRINTF
-               vsnprintf(buf, len, fmt, ap);   /* GNU extension */
+           vsnprintf(my_buffer, my_length, fmt, ap);   /* GNU extension */
 # else
-               vsprintf(buf, fmt, ap);         /* ANSI */
+           vsprintf(my_buffer, fmt, ap);       /* ANSI */
 # endif
+           result = my_buffer;
        }
 #endif
-       return buf;
+    } else if (my_buffer != 0) {       /* see _nc_freeall() */
+       free(my_buffer);
+       my_buffer = 0;
+       my_length = 0;
+    }
+    return result;
 }
+
+#if NCURSES_SP_FUNCS
+NCURSES_EXPORT(char *)
+_nc_printf_string(const char *fmt, va_list ap)
+{
+    return NCURSES_SP_NAME(_nc_printf_string) (CURRENT_SCREEN, fmt, ap);
+}
+#endif