liblitho 0.2.0

cli tool to flash/clone the images to storage devices
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96

use crate::tui::app::Operation;
use liblitho::io_backend::{clone_io, flash_io, USES_SIMULATED_IO};
use liblitho::progress::{is_operation_cancelled, OperationPhase, OperationProgress};
use log::{error, info};
use std::sync::atomic::{AtomicBool, Ordering};
use std::sync::mpsc::Sender;
use std::sync::Arc;

pub fn spawn_operation(
    operation: Operation,
    device_path: String,
    image_path: String,
    block_size: usize,
    verify: bool,
    cancel: Arc<AtomicBool>,
    tx: Sender<OperationProgress>,
) {
    std::thread::spawn(move || {
        let op_name = match operation {
            Operation::Flash => "flash",
            Operation::Clone => "clone",
        };

        if USES_SIMULATED_IO {
            info!(
                "Starting simulated {op_name}: device={device_path}, image={image_path}, block_size={block_size}"
            );
        } else {
            info!(
                "Starting {op_name}: device={device_path}, image={image_path}, block_size={block_size}"
            );
        }

        let on_progress = |progress: OperationProgress| {
            if cancel.load(Ordering::Relaxed) {
                return;
            }
            let _ = tx.send(progress);
        };

        if !USES_SIMULATED_IO {
            if let Err(error) = liblitho::devices::ensure_device_ready_for_io(&device_path, true) {
                error!("{op_name} preflight failed: {error}");
                let _ = tx.send(
                    OperationProgress::new(OperationPhase::Failed)
                        .with_message(error.to_string()),
                );
                return;
            }
        }

        let cancel_ref = Some(cancel.as_ref());
        let result = match operation {
            Operation::Flash => flash_io(
                &image_path,
                &device_path,
                block_size,
                false,
                verify,
                Some(on_progress),
                cancel_ref,
            ),
            Operation::Clone => clone_io(
                &device_path,
                &image_path,
                block_size,
                false,
                Some(on_progress),
                cancel_ref,
            ),
        };

        match result {
            Err(error) if is_operation_cancelled(&error) => {
                let message = match operation {
                    Operation::Flash => {
                        "Flash cancelled — device may be partially written.".to_string()
                    }
                    Operation::Clone => {
                        "Clone cancelled — incomplete output file removed.".to_string()
                    }
                };
                info!("{op_name} cancelled");
                let _ = tx
                    .send(OperationProgress::new(OperationPhase::Cancelled).with_message(message));
            }
            Err(error) => {
                error!("{op_name} failed: {error}");
                let _ = tx.send(
                    OperationProgress::new(OperationPhase::Failed).with_message(error.to_string()),
                );
            }
            Ok(()) => {}
        }
    });
}