safeR 0.1.4

A simple and secure file encryption tool
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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142

use std::sync::Arc;
use arrayref::array_ref;
use futures::future::try_join_all;
use log::{debug, error, info};
use sha2::{Sha256, Digest};
use tokio::sync::Semaphore;
use crate::{args, file, generate_key};

pub async fn handle_encryption(args: args::Args, files: Vec<String>, pb: Option<indicatif::ProgressBar>) {
    if files.is_empty() {
        error!("No files found in {}", args.directory);
        println!("No files found in {}", args.directory);
        return;
    }

    let key = generate_key::generator();
    debug!("Key: {:?}", key);

    let mut hasher = Sha256::new();
    hasher.update(key.as_bytes());
    let result = hasher.finalize();

    let nonce: [u8; 19] = *array_ref![result, 0, 19];
    debug!("Nonce: {:?}", nonce);

    // Create a Vec to hold all the tasks
    let mut tasks = Vec::new();
    let semaphore = Arc::new(Semaphore::new(args.workers));

    // Encrypt all files
    for file_path in files.clone() {
        let key_bytes = key.as_bytes();
        let key_array = <[u8; 32]>::try_from(&key_bytes[0..32]).unwrap();
        let pb = pb.clone();
        let semaphore = Arc::clone(&semaphore);

        // Create a new task for each file
        let task = tokio::spawn(async move {
            let _permit = semaphore.acquire_owned().await;
            match file::encrypt(&file_path, &file_path, &key_array, &nonce) {
                Ok(_) => {
                    info!("File {} encrypted successfully", file_path);
                    if let Some(pb) = &pb {
                        pb.inc(1);
                    }
                }
                Err(e) => {
                    error!("Failed to encrypt file {}: {:?}", file_path, e);
                    if let Some(pb) = &pb {
                        pb.inc(1);
                    }
                }
            }
        });

        // Add the task to the Vec
        tasks.push(task);
    }
    // Wait for all tasks to complete
    let results = try_join_all(tasks).await;

    // Handle errors (if any)
    if let Err(e) = results {
        error!("Failed to encrypt some files: {:?}", e);
    }

    if !args.debug && !args.progress {
        if let Some(pb) = &pb {
            // Finish the progress bar
            pb.finish_with_message("All files encrypted.");
        }
    }
    info!("All {} files in {} encrypted", files.len(),args.directory);
    println!("Key: {}", key);
}

pub async fn handle_decryption(args: args::Args, files: Vec<String>, pb: Option<indicatif::ProgressBar>) {
    if args.key.is_none() {
        error!("Please provide a key to decrypt the files");
        println!("Please provide a key to decrypt the files -k <key>");
    } else {
        let key = args.key.unwrap();
        info!("Key: {}", key);

        let mut hasher = Sha256::new();
        hasher.update(key.as_bytes());
        let result = hasher.finalize();
        let nonce: [u8; 19] = *array_ref![result, 0, 19];
        debug!("Nonce: {:?}", nonce);

        // Create a Vec to hold all the tasks
        let mut tasks = Vec::new();
        let semaphore = Arc::new(Semaphore::new(args.workers));

        // Decrypt all files
        for file_path in files.clone() {
            let key_bytes = key.as_bytes();
            let key_array = <[u8; 32]>::try_from(&key_bytes[0..32]).unwrap();
            let pb = pb.clone();
            let semaphore = Arc::clone(&semaphore);

            // Create a new task for each file
            let task = tokio::spawn(async move {
                let _permit = semaphore.acquire_owned().await;
                match file::decrypt(&file_path, &file_path, &key_array, &nonce) {
                    Ok(_) => {
                        info!("File {} decrypted successfully", file_path);
                        if let Some(pb) = &pb {
                            pb.inc(1);
                        }
                    }
                    Err(e) => {
                        error!("Failed to decrypt file {}: {:?}", file_path, e);
                        if let Some(pb) = &pb {
                            pb.inc(1);
                        }
                    }
                }
            });

            // Add the task to the Vec
            tasks.push(task);
        }

        // Wait for all tasks to complete
        let results = try_join_all(tasks).await;

        // Handle errors (if any)
        if let Err(e) = results {
            error!("Failed to decrypt some files: {:?}", e);
        }

        info!("All {} files in {} decrypted", files.len(),args.directory);

        if !args.debug && !args.progress {
            if let Some(pb) = &pb {
                // Finish the progress bar
                pb.finish_with_message("All files decrypted.");
            }
        }
    }
}