add retries to OCI network hiccups

Signed-off-by: Benny Zlotnik <bzlotnik@redhat.com>
Assisted-by: claude-opus-4.6

Author: bennyz

Cargo.lock

@@ -21,6 +21,7 @@ nix = { version = "0.29", features = ["ioctl"] }
 # This ensures OpenSSL builds from source with musl compatibility
 openssl = { version = "0.10", features = ["vendored"] }
 xz2 = "0.1"
+http-body = "1.0.1"
 
 [dev-dependencies]
 wiremock = "0.6"

Cargo.toml

@@ -35,17 +35,43 @@ pub(crate) async fn start_decompressor_process(
 ) -> Result<(Child, &'static str), Box<dyn std::error::Error>> {
     let cmd = get_decompressor_command(url);
 
-    // Check if binary is available before attempting to spawn
     check_binary_available(cmd)?;
+    println!("Using decompressor: {}", cmd);
+    spawn_decompressor(cmd)
+}
+
+/// Maps a Compression enum to the corresponding decompressor command
+pub(crate) fn decompressor_for_compression(
+    compression: crate::fls::compression::Compression,
+) -> &'static str {
+    use crate::fls::compression::Compression;
+    match compression {
+        Compression::Gzip => "zcat",
+        Compression::Xz => "xzcat",
+        Compression::Zstd => "zstdcat",
+        Compression::None => "cat",
+    }
+}
 
+/// Starts a decompressor process based on detected compression type
+pub(crate) fn start_decompressor_for_compression(
+    compression: crate::fls::compression::Compression,
+) -> Result<(Child, &'static str), Box<dyn std::error::Error>> {
+    let cmd = decompressor_for_compression(compression);
+    check_binary_available(cmd)?;
     println!("Using decompressor: {}", cmd);
+    spawn_decompressor(cmd)
+}
 
+/// Spawns a decompressor subprocess with piped stdin/stdout/stderr
+fn spawn_decompressor(
+    cmd: &'static str,
+) -> Result<(Child, &'static str), Box<dyn std::error::Error>> {
     let process = Command::new(cmd)
         .stdin(std::process::Stdio::piped())
         .stdout(std::process::Stdio::piped())
         .stderr(std::process::Stdio::piped())
         .spawn()?;
-
     Ok((process, cmd))
 }
 

src/fls/decompress.rs

@@ -231,6 +231,42 @@ impl fmt::Display for DownloadError {
 
 impl std::error::Error for DownloadError {}
 
+/// Shared retry handler for download errors.
+///
+/// Returns `Some(Duration)` with the delay to wait before retrying, or `None` if
+/// the error is non-retryable or max retries have been exceeded.
+pub fn handle_download_retry(
+    error: &DownloadError,
+    retry_count: &mut usize,
+    max_retries: usize,
+    default_retry_delay_secs: u64,
+) -> Option<Duration> {
+    if !error.is_retryable() {
+        eprintln!(
+            "\nDownload failed with non-retryable error: {}",
+            error.format_error()
+        );
+        return None;
+    }
+    if *retry_count >= max_retries {
+        eprintln!("\nMax retries ({}) reached, giving up", max_retries);
+        eprintln!("Last error: {}", error.format_error());
+        return None;
+    }
+    let retry_delay = error
+        .suggested_retry_delay()
+        .unwrap_or_else(|| Duration::from_secs(default_retry_delay_secs));
+    eprintln!("\nDownload failed: {}", error.format_error());
+    eprintln!(
+        "Retrying in {} seconds... (attempt {}/{})",
+        retry_delay.as_secs(),
+        *retry_count + 1,
+        max_retries
+    );
+    *retry_count += 1;
+    Some(retry_delay)
+}
+
 #[cfg(test)]
 mod tests {
     use super::*;

src/fls/download_error.rs

@@ -86,6 +86,8 @@ fn build_oci_options(options: &FastbootOptions) -> crate::fls::options::OciOptio
         username: options.username.clone(),
         password: options.password.clone(),
         file_pattern: None,
+        max_retries: crate::fls::options::DEFAULT_MAX_RETRIES,
+        retry_delay_secs: crate::fls::options::DEFAULT_RETRY_DELAY_SECS,
     }
 }
 

src/fls/fastboot.rs

@@ -49,44 +49,7 @@ async fn get_decompressor_error(
     }
 }
 
-/// Handles retry logic for download errors.
-///
-/// Returns `Some(Duration)` with the delay to wait before retrying, or `None` if
-/// the error is non-retryable or max retries have been exceeded.
-fn handle_retry_error(
-    error: &DownloadError,
-    retry_count: &mut usize,
-    max_retries: usize,
-    default_retry_delay_secs: u64,
-) -> Option<Duration> {
-    if !error.is_retryable() {
-        eprintln!(
-            "\nDownload failed with non-retryable error: {}",
-            error.format_error()
-        );
-        return None;
-    }
-
-    if *retry_count >= max_retries {
-        eprintln!("\nMax retries ({}) reached, giving up", max_retries);
-        eprintln!("Last error: {}", error.format_error());
-        return None;
-    }
-
-    let retry_delay = error
-        .suggested_retry_delay()
-        .unwrap_or_else(|| Duration::from_secs(default_retry_delay_secs));
-
-    eprintln!("\nDownload failed: {}", error.format_error());
-    eprintln!(
-        "Retrying in {} seconds... (attempt {}/{})",
-        retry_delay.as_secs(),
-        *retry_count + 1,
-        max_retries
-    );
-    *retry_count += 1;
-    Some(retry_delay)
-}
+use crate::fls::download_error::handle_download_retry;
 
 /// Execute a sequence of write commands on the block writer
 async fn execute_write_commands(
@@ -407,7 +370,7 @@ pub async fn flash_from_url(
             match start_download(url, &client, resume_from, &options.headers, debug).await {
                 Ok(r) => r,
                 Err(e) => {
-                    match handle_retry_error(
+                    match handle_download_retry(
                         &e,
                         &mut retry_count,
                         options.max_retries,
@@ -551,7 +514,7 @@ pub async fn flash_from_url(
 
             if let Some(e) = connection_error {
                 eprintln!("\nConnection interrupted: {}", e.format_error());
-                match handle_retry_error(
+                match handle_download_retry(
                     &e,
                     &mut retry_count,
                     options.max_retries,

src/fls/from_url.rs

@@ -24,6 +24,7 @@ pub use from_url::flash_from_url;
 pub use oci::flash_from_oci;
 pub use options::{
     BlockFlashOptions, FastbootOptions, FlashOptions, HttpClientOptions, OciOptions,
+    DEFAULT_MAX_RETRIES, DEFAULT_RETRY_DELAY_SECS,
 };
 
 #[cfg(test)]