diff options
Diffstat (limited to 'lib/getdelim.c')
-rw-r--r-- | lib/getdelim.c | 30 |
1 files changed, 18 insertions, 12 deletions
diff --git a/lib/getdelim.c b/lib/getdelim.c index 8498b7556..42cac8f55 100644 --- a/lib/getdelim.c +++ b/lib/getdelim.c @@ -23,11 +23,15 @@ # include <config.h> #endif +#include "getdelim.h" + +#include <limits.h> #include <stdlib.h> #include <errno.h> -#include "getdelim.h" - +#ifndef SSIZE_MAX +# define SSIZE_MAX ((ssize_t) (SIZE_MAX / 2)) +#endif #if !HAVE_FLOCKFILE # undef flockfile # define flockfile(x) ((void) 0) @@ -46,9 +50,8 @@ ssize_t getdelim (char **lineptr, size_t *n, int delimiter, FILE *fp) { - int result = 0; - ssize_t cur_len = 0; - ssize_t len; + ssize_t result; + size_t cur_len = 0; if (lineptr == NULL || n == NULL || fp == NULL) { @@ -71,23 +74,26 @@ getdelim (char **lineptr, size_t *n, int delimiter, FILE *fp) for (;;) { - char *t; int i; i = getc (fp); if (i == EOF) - { - result = -1; - break; - } + { + result = -1; + break; + } /* Make enough space for len+1 (for final NUL) bytes. */ if (cur_len + 1 >= *n) { - size_t needed = 2 * (cur_len + 1) + 1; /* Be generous. */ + size_t needed_max = + SSIZE_MAX < SIZE_MAX ? (size_t) SSIZE_MAX + 1 : SIZE_MAX; + size_t needed = 2 * *n + 1; /* Be generous. */ char *new_lineptr; - if (needed < cur_len) + if (needed_max < needed) + needed = needed_max; + if (cur_len + 1 >= needed) { result = -1; goto unlock_return; |