about summary refs log tree commit diff
diff options
context:
space:
mode:
-rw-r--r--src/internal/shgetc.c7
1 files changed, 3 insertions, 4 deletions
diff --git a/src/internal/shgetc.c b/src/internal/shgetc.c
index e85d38ac..96f72b6a 100644
--- a/src/internal/shgetc.c
+++ b/src/internal/shgetc.c
@@ -4,7 +4,7 @@ void __shlim(FILE *f, off_t lim)
 {
 	f->shlim = lim;
 	f->shcnt = f->rend - f->rpos;
-	if (lim && f->rend - f->rpos > lim)
+	if (lim && f->shcnt > lim)
 		f->shend = f->rpos + lim;
 	else
 		f->shend = f->rend;
@@ -13,15 +13,14 @@ void __shlim(FILE *f, off_t lim)
 int __shgetc(FILE *f)
 {
 	int c;
-	if (f->shlim && f->shcnt >= f->shlim) {
+	if (f->shlim && f->shcnt >= f->shlim || (c=__uflow(f)) < 0) {
 		f->shend = 0;
 		return EOF;
 	}
-	c = __uflow(f);
 	if (f->shlim && f->rend - f->rpos > f->shlim - f->shcnt - 1)
 		f->shend = f->rpos + (f->shlim - f->shcnt - 1);
 	else
 		f->shend = f->rend;
-	if (f->rend) f->shcnt += f->rend - f->buf;
+	if (f->rend) f->shcnt += f->rend - f->rpos + 1;
 	return c;
 }