loco-protocol 6.5.0

Loco protocol implementation
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

/*
 * Created on Sun Sep 10 2023
 *
 * Copyright (c) storycraft. Licensed under the MIT Licence.
 */

use core::mem::swap;

use loco_protocol::secure::{client::LocoClientSecureLayer, SecurePacket};
use rand::RngCore;
use rsa::{RsaPrivateKey, RsaPublicKey};

#[test]
pub fn test_handshake() {
    let mut layer = LocoClientSecureLayer::new([0_u8; 16]);

    let priv_key =
        RsaPrivateKey::new(&mut rand::thread_rng(), 2048).expect("failed to generate a key");
    let pub_key = RsaPublicKey::from(&priv_key);

    layer.handshake(&pub_key);

    assert_eq!(layer.write_buffer.len(), 12 + 256);
}

#[test]
pub fn read_write_test() {
    let key = {
        let mut arr = [0_u8; 16];
        rand::thread_rng().fill_bytes(&mut arr);

        arr
    };

    let mut layer = LocoClientSecureLayer::new(key);

    let packet = SecurePacket {
        iv: [0_u8; 16],
        data: Box::new([0_u8, 1, 2]) as Box<[u8]>,
    };

    layer.send(packet.clone());

    swap(&mut layer.read_buffer, &mut layer.write_buffer);

    assert_eq!(layer.read(), Some(packet));
}