summaryrefslogtreecommitdiffstats
path: root/usr.bin
diff options
context:
space:
mode:
authorjh <jh@FreeBSD.org>2009-12-17 17:03:47 +0000
committerjh <jh@FreeBSD.org>2009-12-17 17:03:47 +0000
commit060a87e0f55f660dac1fdc467a1e5bb520f953e0 (patch)
treead817a52fcb0d3b32db89c4148efc8bcbf4de12d /usr.bin
parent4491896aee42d649ffebd4db8fe3607e1debffc9 (diff)
downloadFreeBSD-src-060a87e0f55f660dac1fdc467a1e5bb520f953e0.zip
FreeBSD-src-060a87e0f55f660dac1fdc467a1e5bb520f953e0.tar.gz
Sync getline() with comm(1):
- Prevent overflowing of the buffer length variable in getline() by limiting its maximum value. - Exit if reallocf(3) fails in getline(). Failure was silently considered as end-of-file. Reviewed by: ghelmer Approved by: trasz (mentor)
Diffstat (limited to 'usr.bin')
-rw-r--r--usr.bin/uniq/uniq.c23
1 files changed, 14 insertions, 9 deletions
diff --git a/usr.bin/uniq/uniq.c b/usr.bin/uniq/uniq.c
index bc04db7..2b11fe4 100644
--- a/usr.bin/uniq/uniq.c
+++ b/usr.bin/uniq/uniq.c
@@ -52,6 +52,7 @@ static const char rcsid[] =
#include <err.h>
#include <limits.h>
#include <locale.h>
+#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
@@ -59,7 +60,8 @@ static const char rcsid[] =
#include <wchar.h>
#include <wctype.h>
-#define MAXLINELEN (LINE_MAX + 1)
+#define INITLINELEN (LINE_MAX + 1)
+#define MAXLINELEN ((SIZE_MAX / sizeof(wchar_t)) / 2)
int cflag, dflag, uflag;
int numchars, numfields, repeats;
@@ -137,8 +139,8 @@ main (int argc, char *argv[])
if (argc > 1)
ofp = file(argv[1], "w");
- prevbuflen = MAXLINELEN;
- thisbuflen = MAXLINELEN;
+ prevbuflen = INITLINELEN;
+ thisbuflen = INITLINELEN;
prevline = malloc(prevbuflen * sizeof(*prevline));
thisline = malloc(thisbuflen * sizeof(*thisline));
if (prevline == NULL || thisline == NULL)
@@ -198,16 +200,19 @@ getline(wchar_t *buf, size_t *buflen, FILE *fp)
bufpos = 0;
while ((ch = getwc(fp)) != WEOF && ch != '\n') {
- if (bufpos + 2 >= *buflen) {
+ if (bufpos + 1 >= *buflen) {
*buflen = *buflen * 2;
+ if (*buflen > MAXLINELEN)
+ errx(1,
+ "Maximum line buffer length (%zu) exceeded",
+ MAXLINELEN);
buf = reallocf(buf, *buflen * sizeof(*buf));
if (buf == NULL)
- return (NULL);
+ err(1, "reallocf");
}
buf[bufpos++] = ch;
}
- if (bufpos + 1 != *buflen)
- buf[bufpos] = '\0';
+ buf[bufpos] = '\0';
return (bufpos != 0 || ch == '\n' ? buf : NULL);
}
@@ -305,13 +310,13 @@ wcsicoll(wchar_t *s1, wchar_t *s2)
new_l2_buflen = wcsicoll_l2_buflen;
while (new_l1_buflen < l1) {
if (new_l1_buflen == 0)
- new_l1_buflen = MAXLINELEN;
+ new_l1_buflen = INITLINELEN;
else
new_l1_buflen *= 2;
}
while (new_l2_buflen < l2) {
if (new_l2_buflen == 0)
- new_l2_buflen = MAXLINELEN;
+ new_l2_buflen = INITLINELEN;
else
new_l2_buflen *= 2;
}
OpenPOWER on IntegriCloud