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

/*! InnerOnionRequest enum
*/

use super::*;

use tox_binary_io::*;

/** Onion requests that can be enclosed in onion packets and sent through onion
path.

Onion allows only two types of packets to be sent as a request through onion
paths: `OnionAnnounceRequest` and `OnionDataRequest`.
*/
#[derive(Clone, Debug, Eq, PartialEq)]
pub enum InnerOnionRequest {
    /// [`InnerOnionAnnounceRequest`](./struct.InnerOnionAnnounceRequest.html) structure.
    InnerOnionAnnounceRequest(InnerOnionAnnounceRequest),
    /// [`InnerOnionDataRequest`](./struct.InnerOnionDataRequest.html) structure.
    InnerOnionDataRequest(InnerOnionDataRequest)
}

impl ToBytes for InnerOnionRequest {
    fn to_bytes<'a>(&self, buf: (&'a mut [u8], usize)) -> Result<(&'a mut [u8], usize), GenError> {
        match *self {
            InnerOnionRequest::InnerOnionAnnounceRequest(ref inner) => inner.to_bytes(buf),
            InnerOnionRequest::InnerOnionDataRequest(ref inner) => inner.to_bytes(buf),
        }
    }
}

impl FromBytes for InnerOnionRequest {
    named!(from_bytes<InnerOnionRequest>, alt!(
        map!(InnerOnionAnnounceRequest::from_bytes, InnerOnionRequest::InnerOnionAnnounceRequest) |
        map!(InnerOnionDataRequest::from_bytes, InnerOnionRequest::InnerOnionDataRequest)
    ));
}

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

    encode_decode_test!(
        tox_crypto::crypto_init().unwrap(),
        inner_onion_announce_request_encode_decode,
        InnerOnionRequest::InnerOnionAnnounceRequest(InnerOnionAnnounceRequest {
            nonce: gen_nonce(),
            pk: gen_keypair().0,
            payload: vec![42; 123]
        })
    );

    encode_decode_test!(
        tox_crypto::crypto_init().unwrap(),
        inner_onion_data_request_encode_decode,
        InnerOnionRequest::InnerOnionDataRequest(InnerOnionDataRequest {
            destination_pk: gen_keypair().0,
            nonce: gen_nonce(),
            temporary_pk: gen_keypair().0,
            payload: vec![42; 123]
        })
    );
}