psp_security 0.3.1

A Rust implementation of the PSP Security Protocol
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
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

use pnet_macros::packet;
use pnet_macros_support::types::{u1, u2, u32be, u4, u6, u64be};


/// Represents a PSP Packet.
///
/// PSP Security Payload (PSP) Header
///
/// |      0        |       1       |       2       |      3        |
/// |0 1 2 3 4 5 6 7|8 9 0 1 2 3 4 5|6 7 8 9 0 1 2 3|4 5 6 7 8 9 0 1|
/// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
/// |  Next Header  |  Hdr Ext Len  | R |Crypt Off  |S|D|Version|V|1|
/// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
/// |                Security Parameters Index (SPI)                |
/// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
/// |                                                               |
/// |                  Initialization Vector (IV)                   |
/// |                                                               |
/// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
/// |                                                               |
/// |             Virtualization Cookie (VC) [Optional]             |
/// |                                                               |
/// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
///
#[allow(dead_code)]
#[packet]
pub struct Psp {
    pub next_hdr: u8,
    pub hdr_ext_len: u8,
    _r: u2,
    pub crypt_offset: u6,
    pub s: u1,
    pub d: u1,
    pub version: u4,
    pub v: u1,
    pub always1: u1,
    pub spi: u32be,
    pub iv: u64be,
    #[length_fn = "psp_vc_length"]
    pub vc: Vec<u64be>,
    #[payload]
    pub payload: Vec<u8>,
}

fn psp_vc_length(psp: &PspPacket) -> usize {
    match psp.get_v() {
        1 => 8,
        _ => 0,
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use pnet_macros_support::packet::Packet;

    #[test]
    fn psp_header_test() {
        let buf = vec![
            0x11, 0x01, 0x01, 0x01, 0x00, 0x00, 0x00, 0x03, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06,
            0x07, 0x08, 0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F, 0x00, 0x01, 0x02, 0x03, 0x04,
            0x05, 0x06, 0x07, 0x08, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08,
        ];
        let psp = PspPacket::new(&buf).unwrap();
        assert_eq!(psp.get_next_hdr(), 0x11);
        assert_eq!(psp.get_hdr_ext_len(), 1);
        assert_eq!(psp.get_crypt_offset(), 1);
        assert_eq!(psp.get_version(), 0);
        assert_eq!(psp.get_s(), 0);
        assert_eq!(psp.get_d(), 0);
        assert_eq!(psp.get_v(), 0);
        assert_eq!(psp.get_spi(), 3);
        assert_eq!(psp.get_iv(), 0x0102030405060708);
        assert_eq!(psp.payload().len(), 24);
        assert_eq!(psp.packet().len(), 40);
    }

    #[test]
    fn psp_header_vc_test() {
        let buf = vec![
            0x11, 0x01, 0x01, 0x03, 0x00, 0x00, 0x00, 0x03, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06,
            0x07, 0x08, 0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F, 0x00, 0x01, 0x02, 0x03, 0x04,
            0x05, 0x06, 0x07, 0x08, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08,
        ];
        let psp = PspPacket::new(&buf).unwrap();
        assert_eq!(psp.get_next_hdr(), 0x11);
        assert_eq!(psp.get_hdr_ext_len(), 1);
        assert_eq!(psp.get_crypt_offset(), 1);
        assert_eq!(psp.get_version(), 0);
        assert_eq!(psp.get_s(), 0);
        assert_eq!(psp.get_d(), 0);
        assert_eq!(psp.get_v(), 1);
        assert_eq!(psp.get_spi(), 3);
        assert_eq!(psp.get_iv(), 0x0102030405060708);
        assert_eq!(psp.get_vc(), vec![0x090A0B0C0D0E0F00]);
        assert_eq!(psp.payload().len(), 16);
        assert_eq!(psp.packet().len(), 40);
    }
}