Fix EOF detection

Fixes issue with Ctrl-D having to be pressed twice when reading from
terminal.

1 files changed, 6 insertions, 0 deletions

diff --git a/src/main.c b/src/main.c
index 5d46e00..6134b13 100644
--- a/src/main.c
+++ b/src/main.c
@@ -71,6 +71,9 @@ int main(int argc, char *argv[])
 		start_timer();
 		while ((bytes = fread(buffer, 1, sizeof(buffer), fp)) > 0) {
 			cmark_parser_feed(parser, buffer, bytes);
+			if (bytes < sizeof(buffer)) {
+				break;
+			}
 		}
 		end_timer("processing lines");
 
@@ -86,6 +89,9 @@ int main(int argc, char *argv[])
 
 		while ((bytes = fread(buffer, 1, sizeof(buffer), stdin)) > 0) {
 			cmark_parser_feed(parser, buffer, bytes);
+			if (bytes < sizeof(buffer)) {
+				break;
+			}
 		}
 	}