add --prove

1 files changed, 8 insertions, 0 deletions

diff --git a/src/tty.c b/src/tty.c
index 104c780..a1e8590 100644
--- a/src/tty.c
+++ b/src/tty.c
@@ -34,6 +34,7 @@
 #include "list.h"
 #include "tty.h"
 #include "tcp.h"
+#include "prove.h"
 
 #ifndef USE_SGTTY
 #  ifdef APOLLO
@@ -815,6 +816,13 @@ void input_insert_follow_chars(char *str, int n)
 
 void tty_puts(const char *s)
 {
+#ifdef ENABLE_PROVE
+    if (is_proving()) {
+        prove_save_tty_output(s);
+        return;
+    }
+#endif
+
     while (*s)
         tty_putc(*s++);
 }