setup-sshd: improve error handling

repeat the question til we get a valid answer

1 files changed, 4 insertions, 4 deletions

diff --git a/setup-sshd.in b/setup-sshd.in
index 00e9bb8..67cb8a0 100644
--- a/setup-sshd.in
+++ b/setup-sshd.in
@@ -24,10 +24,10 @@ while getopts "hc:" opt; do
 	esac
 done
 
-if [ "$sshdchoice" = "" ]; then
-	echo -n "Which SSH server? ('openssh', 'dropbear' or 'none') [openssh] "
-	default_read sshdchoice "openssh"
-fi
+while ! isin "$sshdchoice" openssh dropbear none; do
+	ask "Which SSH server? ('openssh', 'dropbear' or 'none')" openssh
+	sshdchoice="$resp"
+done
 
 if [ "$sshdchoice" = "none" ]; then
 	exit 0