X-Git-Url: http://git.zerfleddert.de/cgi-bin/gitweb.cgi/proxmark3-svn/blobdiff_plain/a553f2674865fa544d04ca7b8194e30c008644ae..e6b8c965bb27650b3ead5738ba46dd4c82f937de:/client/ui.c

diff --git a/client/ui.c b/client/ui.c
index 0b4183df..554230cb 100644
--- a/client/ui.c
+++ b/client/ui.c
@@ -1,4 +1,5 @@
 //-----------------------------------------------------------------------------
+// Copyright (C) 2009 Michael Gernoth <michael at gernoth.net>
 // Copyright (C) 2010 iZsh <izsh at fail0verflow.com>
 //
 // This code is licensed to you under the terms of the GNU GPL, version 2 or,
@@ -9,8 +10,10 @@
 //-----------------------------------------------------------------------------
 
 #include <stdarg.h>
+#include <stdlib.h>
 #include <stdio.h>
 #include <time.h>
+#include <readline/readline.h>
 
 #include "ui.h"
 
@@ -22,6 +25,8 @@ static char *logfilename = "proxmark3.log";
 
 void PrintAndLog(char *fmt, ...)
 {
+	char *saved_line;
+	int saved_point;
   va_list argptr, argptr2;
   static FILE *logfile = NULL;
   static int logging=1;
@@ -33,12 +38,32 @@ void PrintAndLog(char *fmt, ...)
       logging=0;
     }
   }
+	
+	int need_hack = (rl_readline_state & RL_STATE_READCMD) > 0;
 
+	if (need_hack) {
+		saved_point = rl_point;
+		saved_line = rl_copy_text(0, rl_end);
+		rl_save_prompt();
+		rl_replace_line("", 0);
+		rl_redisplay();
+	}
+	
   va_start(argptr, fmt);
   va_copy(argptr2, argptr);
   vprintf(fmt, argptr);
+  printf("          "); // cleaning prompt
   va_end(argptr);
   printf("\n");
+
+	if (need_hack) {
+		rl_restore_prompt();
+		rl_replace_line(saved_line, 0);
+		rl_point = saved_point;
+		rl_redisplay();
+		free(saved_line);
+	}
+	
   if (logging && logfile) {
     vfprintf(logfile, fmt, argptr2);
     fprintf(logfile,"\n");