Make sure to check return from fflush()

gavinhoward · Oct 10, 2018 · 8c62e60 · 8c62e60
1 parent 2ce361e
commit 8c62e60
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/program.c b/src/program.c
@@ -1953,7 +1953,7 @@ BcStatus bc_program_printInst(BcProgram *p, char *code, size_t *bgn) {
 		if (inst == BC_INST_CALL) s = bc_program_printIndex(code, bgn);
 	}
 
-	if (!s && fflush(stdout) < 0) s = BC_STATUS_IO_ERR;
+	if (!s && fflush(stdout) == EOF) s = BC_STATUS_IO_ERR;
 
 	return s;
 }

diff --git a/src/vm.c b/src/vm.c
@@ -172,7 +172,7 @@ BcStatus bc_vm_process(BcVm *vm, const char *text) {
 
 	if (BC_PARSE_CAN_EXEC(&vm->prs)) {
 		s = bc_program_exec(&vm->prog);
-		if (bcg.tty) fflush(stdout);
+		if (!s && bcg.tty && fflush(stdout) == EOF) s = BC_STATUS_IO_ERR;
 		if (s && s != BC_STATUS_QUIT)
 			s = bc_vm_error(bc_program_reset(&vm->prog, s), vm->prs.l.f, 0);
 	}