rsa_msg_packets/file/chunk/
index.rs

1use anyhow::anyhow;
2
3use openssl::pkey::{Public, Private};
4use openssl::rsa::Rsa;
5use uuid::Uuid;
6
7use crate::encryption::sign::validate_signature;
8use crate::other::key_iv::KeyIVPair;
9use crate::util::tools::{u64_from_vec, usize_to_vec};
10use crate::util::{tools::{vec_to_usize, uuid_from_vec}, vec::extract_vec};
11
12
13#[derive(Debug, Clone)]
14pub struct ChunkMsg {
15    pub signature: Vec<u8>,
16    pub encrypted: Vec<u8>,
17    pub key: KeyIVPair,
18    pub uuid: Uuid,
19    pub chunk_index: u64
20}
21
22
23impl ChunkMsg {
24    pub fn serialize(&self, receiver_key: &Rsa<Public>) -> anyhow::Result<Vec<u8>> {
25        let mut merged = Vec::new();
26        let mut b_encrypted = self.encrypted.clone();
27        let mut b_key = self.key.serialize(receiver_key)?;
28
29        let signature_size = usize_to_vec(self.signature.len())?;
30        let mut b_uuid = self.uuid.clone().as_bytes().to_vec();
31
32        let mut b_chunk_index = self.chunk_index.clone().to_le_bytes().to_vec();
33
34        merged.append(&mut signature_size.clone());
35        merged.append(&mut self.signature.clone());
36        merged.append(&mut b_uuid);
37        merged.append(&mut b_chunk_index);
38        merged.append(&mut b_key);
39        merged.append(&mut b_encrypted);
40
41        return Ok(merged);
42    }
43
44    pub fn deserialize(data: &Vec<u8>, sender_pubkey: &Rsa<Public>, receiver_key: &Rsa<Private>) -> anyhow::Result<Self> {
45        let mut data = data.clone();
46
47        let signature_size = vec_to_usize(&mut data)?;
48        let signature = extract_vec(0..signature_size, &mut data)?;
49
50        let uuid = uuid_from_vec(&mut data)?;
51        let chunk_index = u64_from_vec(&mut data)?;
52
53        let key = KeyIVPair::deserialize_mut(&mut data, receiver_key)?;
54
55        let valid = validate_signature(&data, &signature, sender_pubkey)?;
56        if !valid {
57            return Err(anyhow!("Invalid signature in ChunkByteMsg"));
58        }
59
60        return Ok(ChunkMsg {
61            signature,
62            encrypted: data,
63            uuid,
64            chunk_index,
65            key
66        });
67    }
68}
rsa_msg_packets/file/chunk/index.rs

rsa_msg_packets/file/chunk/
index.rs
