cli: fix closing incorrectly if updated failed (microsoft#215595)

aaronchucarroll · Jun 14, 2024 · 03d781a · 03d781a
1 parent d32e093
commit 03d781a
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/cli/src/tunnels/control_server.rs b/cli/src/tunnels/control_server.rs
@@ -920,9 +920,14 @@ async fn handle_update(
 
 	info!(log, "Updating CLI to {}", latest_release);
 
-	updater
+	let r = updater
 		.do_update(&latest_release, SilentCopyProgress())
-		.await?;
+		.await;
+
+	if let Err(e) = r {
+		did_update.store(false, Ordering::SeqCst);
+		return Err(e);
+	}
 
 	Ok(UpdateResult {
 		up_to_date: true,