my_getchar() parameter: allow return without new character

1 files changed, 3 insertions, 3 deletions

diff --git a/avr/con-utils.c b/avr/con-utils.c
index 02657b9..b8017ed 100644
--- a/avr/con-utils.c
+++ b/avr/con-utils.c
@@ -12,14 +12,14 @@ uint_fast8_t tstc(void)
 	return serial_tstc();
 }
 
-int my_getchar(void)
+int my_getchar(uint_fast8_t waitforchar)
 {
 	int c;
 	
 	do {
 		bg_shed();
 		c = serial_getc();
-	} while (c < 0);
+	} while ((c < 0) && waitforchar);
 
 	return c;
 }
@@ -57,7 +57,7 @@ uint_fast8_t confirm_yesno(void)
 		;
 	i = 0;
 	while (i < sizeof(str_input)) {
-		str_input[i] = my_getchar();
+		str_input[i] = my_getchar(1);
 		putchar(str_input[i]);
 		if (str_input[i] == '\r')
 			break;