Avoid a possible infinite loop using cpu_rec.c when there is a read e…

…rror
airbus-seclab · Apr 24, 2017 · 21dfd6b · 21dfd6b
1 parent 2bf83f8
commit 21dfd6b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cpu_rec.c b/cpu_rec.c
@@ -48,7 +48,7 @@ struct stats count_ngrams(char *filename)
   if (c>1)
     r.m2[ buffer[1] + 256*buffer[0] ] += 1;
   increment(r, buffer, c);
-  while (!feof(f)) {
+  while (c) {
     buffer[0] = buffer[BUFSIZE-2];
     buffer[1] = buffer[BUFSIZE-1];
     c = fread(buffer+2, 1, BUFSIZE-2, f);