Skip to content

Commit 21dfd6b

Browse files
committed
Avoid a possible infinite loop using cpu_rec.c when there is a read error
1 parent 2bf83f8 commit 21dfd6b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

cpu_rec.c

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ struct stats count_ngrams(char *filename)
4848
if (c>1)
4949
r.m2[ buffer[1] + 256*buffer[0] ] += 1;
5050
increment(r, buffer, c);
51-
while (!feof(f)) {
51+
while (c) {
5252
buffer[0] = buffer[BUFSIZE-2];
5353
buffer[1] = buffer[BUFSIZE-1];
5454
c = fread(buffer+2, 1, BUFSIZE-2, f);

0 commit comments

Comments
 (0)