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

use super::bundle::*;

/******************************
 *
 * CRC
 *
 ******************************/

pub type CRCType = u8;

use byteorder::{BigEndian, ByteOrder};
use crc::{crc16, crc32};

pub const CRC_NO: CRCType = 0;
pub const CRC_16: CRCType = 1;
pub const CRC_32: CRCType = 2;

pub trait CRCFuncations {
    fn to_string(self) -> String;
}
impl CRCFuncations for CRCType {
    fn to_string(self) -> String {
        match self {
            CRC_NO => String::from("no"),
            CRC_16 => String::from("16"),
            CRC_32 => String::from("32"),
            _ => String::from("unknown"),
        }
    }
}
pub fn empty_crc(crc_type: CRCType) -> Result<ByteBuffer, Bp7Error> {
    match crc_type {
        CRC_NO => Ok(Vec::new()),
        CRC_16 => Ok(vec![0; 2]),
        CRC_32 => Ok(vec![0; 4]),
        _ => Err(Bp7Error::CrcError("Unknown CRC type".to_string())),
    }
}

pub fn block_to_bytes<T: Block + Clone>(blck: &T) -> ByteBuffer {
    let temp_blck = &mut blck.clone();
    temp_blck.reset_crc();
    temp_blck.to_cbor()
}
pub fn calculate_crc<T: Block + Clone>(blck: &T) -> ByteBuffer {
    let mut output_crc = empty_crc(blck.crc_type()).unwrap();
    let data = block_to_bytes(blck);
    match blck.crc_type() {
        CRC_NO => return output_crc,
        CRC_16 => {
            let chksm = crc16::checksum_x25(&data);
            BigEndian::write_u16(&mut output_crc, chksm);
        }
        CRC_32 => {
            let chksm = crc32::checksum_castagnoli(&data);
            BigEndian::write_u32(&mut output_crc, chksm);
        }
        _ => {
            panic!("Unknown crc type");
        }
    }

    output_crc
}
pub fn check_crc<T: Block + Clone>(blck: &T) -> bool {
    if !blck.has_crc() {
        return blck.has_crc();
    }
    (&blck).crc() == calculate_crc(blck)
}