diff options
| -rw-r--r-- | lib/libc/stdio/snprintf.c | 10 | ||||
| -rw-r--r-- | lib/libc/stdio/vsnprintf.c | 10 |
2 files changed, 10 insertions, 10 deletions
diff --git a/lib/libc/stdio/snprintf.c b/lib/libc/stdio/snprintf.c index 8a7584c..bfcf72c 100644 --- a/lib/libc/stdio/snprintf.c +++ b/lib/libc/stdio/snprintf.c @@ -39,7 +39,7 @@ static char sccsid[] = "@(#)snprintf.c 8.1 (Berkeley) 6/4/93"; #endif static const char rcsid[] = - "$Id: snprintf.c,v 1.6 1997/12/24 12:31:31 ache Exp $"; + "$Id: snprintf.c,v 1.7 1997/12/24 14:32:39 ache Exp $"; #endif /* LIBC_SCCS and not lint */ #include <stdio.h> @@ -48,6 +48,7 @@ static const char rcsid[] = #else #include <varargs.h> #endif +#include <limits.h> #if __STDC__ int @@ -65,9 +66,10 @@ snprintf(str, n, fmt, va_alist) va_list ap; FILE f; - if ((int)n < 1) + if (n == 0) + return (0); + if (--n > INT_MAX) return (EOF); - n--; #if __STDC__ va_start(ap, fmt); #else @@ -80,7 +82,5 @@ snprintf(str, n, fmt, va_alist) ret = vfprintf(&f, fmt, ap); *f._p = 0; va_end(ap); - if (ret == EOF) - return (ret); return (ret > (int)n ? n : ret); } diff --git a/lib/libc/stdio/vsnprintf.c b/lib/libc/stdio/vsnprintf.c index 8d824fe..738dd21 100644 --- a/lib/libc/stdio/vsnprintf.c +++ b/lib/libc/stdio/vsnprintf.c @@ -39,10 +39,11 @@ static char sccsid[] = "@(#)vsnprintf.c 8.1 (Berkeley) 6/4/93"; #endif static const char rcsid[] = - "$Id: vsnprintf.c,v 1.6 1997/12/24 12:31:32 ache Exp $"; + "$Id: vsnprintf.c,v 1.7 1997/12/24 14:32:40 ache Exp $"; #endif /* LIBC_SCCS and not lint */ #include <stdio.h> +#include <limits.h> int vsnprintf(str, n, fmt, ap) @@ -54,16 +55,15 @@ vsnprintf(str, n, fmt, ap) int ret; FILE f; - if ((int)n < 1) + if (n == 0) + return (0); + if (--n > INT_MAX) return (EOF); - n--; f._file = -1; f._flags = __SWR | __SSTR; f._bf._base = f._p = (unsigned char *)str; f._bf._size = f._w = n; ret = vfprintf(&f, fmt, ap); *f._p = 0; - if (ret == EOF) - return (ret); return (ret > (int)n ? n : ret); } |
