Use traps to ensure tty settings get restored

cloudposse · Jan 1, 2025 · d9b0786 · d9b0786
1 parent 94e2add
commit d9b0786
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/rootfs/etc/profile.d/_01-launch-warning.sh b/rootfs/etc/profile.d/_01-launch-warning.sh
@@ -19,7 +19,10 @@
 # so the cursor never advances. Then we eval the message with backspaces removed.
 # We have to then add some extra characters to erase the eval command.
 function warn_if_piped() {
-	local saved_stty=$(stty -g)
+	local saved_stty
+	saved_stty=$(stty -g)
+	trap 'stty "$saved_stty"' EXIT
+
 	stty -echo -opost
 
 	local mesg cmd xb m fx bx feval beval
@@ -53,6 +56,7 @@ function warn_if_piped() {
 	echo -n "$beval$feval$beval"
 
 	stty "$saved_stty"
+	trap - EXIT
 }
 
 warn_if_piped