rsa_msg_packets/file/question/
index.rs

1use std::collections::VecDeque;
2
3use log::trace;
4use uuid::Uuid;
5
6use crate::{
7    types::ByteMessage,
8    util::{
9        converter::{str_to_decque, uuid_to_decque},
10        modes::Modes,
11        tools::{u64_from_vec, uuid_from_vec},
12        vec::{decque_to_vec, vec_to_decque, extract_vec},
13    }, consts::MSG_DIGEST,
14};
15
16#[derive(Debug, Clone)]
17pub struct FileQuestionMsg {
18    pub filename: String,
19    pub sender: Uuid,
20    pub receiver: Uuid,
21    pub uuid: Uuid,
22    // Uses sha256 so 32 bytes
23    pub hash: Vec<u8>,
24    pub size: u64
25}
26
27impl ByteMessage for FileQuestionMsg {
28    fn serialize(&self) -> Vec<u8> {
29        let mut merged: VecDeque<u8> = VecDeque::new();
30        let mut b_filename = str_to_decque(&self.filename);
31
32        let mut b_uuid = uuid_to_decque(&self.uuid);
33        let mut b_receiver = uuid_to_decque(&self.receiver);
34        let mut b_sender = uuid_to_decque(&self.sender);
35        let mut b_size = vec_to_decque(self.size.to_le_bytes().to_vec());
36        let mut b_hash = vec_to_decque(self.hash.clone());
37
38        merged.append(&mut b_uuid);
39        merged.append(&mut b_receiver);
40        merged.append(&mut b_sender);
41        merged.append(&mut b_size);
42        merged.append(&mut b_hash);
43        merged.append(&mut b_filename);
44
45        return Modes::SendFileQuestion.get_send(&decque_to_vec(merged));
46    }
47
48    fn deserialize(data: &Vec<u8>) -> anyhow::Result<Self> {
49        trace!("Parsing file question...");
50        let mut data = data.clone();
51
52        let uuid = uuid_from_vec(&mut data)?;
53        let receiver = uuid_from_vec(&mut data)?;
54        let sender = uuid_from_vec(&mut data)?;
55        let size = u64_from_vec(&mut data)?;
56        let hash = extract_vec(0..MSG_DIGEST.size(), &mut data)?;
57
58        let filename = String::from_utf8(data)?;
59
60        let msg = FileQuestionMsg {
61            filename,
62            uuid,
63            sender,
64            receiver,
65            size,
66            hash
67        };
68
69        trace!("FileQuestion info is {:?}", msg);
70        return Ok(msg);
71    }
72}