Improve error handling during switch reinstall

Signed-off-by: Pau Capdevila <pau@githedgehog.com>

Author: pau-hedgehog

pkg/hhfab/vlabhelpers_reinstall.exp

@@ -169,7 +169,6 @@ if { ! $POWER_RESET} {
 
 expect -timeout 150 -ex "GNU GRUB" {
 	log_message "INF" $SW_NAME "GRUB Menu detected"
-	# Select the ONIE option, and finally the Install OS option
 	sleep 1
 	send -- "$KEY_HOME"
 	set timeout 1
@@ -185,6 +184,10 @@ expect -timeout 150 -ex "GNU GRUB" {
 				"ONIE: OS Install Mode ..." {
 					log_message "INF" $SW_NAME "Entering OS Install Mode..."
 				}
+				eof {
+					log_message "ERR" $SW_NAME "Connection lost before confirming ONIE install mode"
+					exit $ERROR_CONSOLE
+				}
 				timeout {
 					send "\r"
 				}
@@ -194,6 +197,10 @@ expect -timeout 150 -ex "GNU GRUB" {
 			send -- "$KEY_DOWN"
 			exp_continue
 		}
+		eof {
+			log_message "ERR" $SW_NAME "Connection lost during GRUB menu expect"
+			exit $ERROR_CONSOLE
+		}
 	}
 } timeout {
 	log_message "ERR" $SW_NAME "Failed to reset. GRUB not detected."