tox_packet 0.1.1

/*! Top-level DHT udp packets according
    to [Tox spec](https://zetok.github.io/tox-spec/#packet-kind)
*/

#[cfg(test)]
#[macro_use]
mod macros;
pub mod packed_node;
mod ping_request;
mod ping_response;
mod nodes_request;
mod nodes_response;
mod dht_request;
mod cookie_request;
mod cookie_response;
mod bootstrap_info;
mod lan_discovery;
mod crypto_handshake;
mod crypto_data;
mod cookie;
mod errors;

pub use self::ping_request::*;
pub use self::ping_response::*;
pub use self::nodes_request::*;
pub use self::nodes_response::*;
pub use self::dht_request::*;
pub use self::cookie_request::*;
pub use self::cookie_response::*;
pub use self::bootstrap_info::*;
pub use self::lan_discovery::*;
pub use self::crypto_handshake::*;
pub use self::crypto_data::*;
pub use self::cookie::*;
pub use self::errors::*;

use nom::{alt, map};

use nom::{
    named, do_parse, tag, call, take, verify,  eof,
};

use cookie_factory::{
    do_gen,
    gen_slice,
    gen_call,
    gen_cond,
    gen_be_u8,
    gen_be_u16,
    gen_be_u32,
    gen_be_u64,
    gen_many_ref
};

use tox_binary_io::*;
use crate::onion::*;

pub fn unix_time(time: std::time::SystemTime) -> u64 {
    let since_the_epoch = time.duration_since(std::time::UNIX_EPOCH)
        .expect("Current time is earlier than Unix epoch");

    since_the_epoch.as_secs()
}

/// A serialized `Packet` should be not longer than 2048 bytes.
pub const MAX_DHT_PACKET_SIZE: usize = 2048;

/** DHT packet enum that encapsulates all types of DHT packets.
*/
#[derive(Clone, Debug, Eq, PartialEq)]
pub enum Packet {
    /// [`PingRequest`](./struct.PingRequest.html) structure.
    PingRequest(PingRequest),
    /// [`PingResponse`](./struct.PingResponse.html) structure.
    PingResponse(PingResponse),
    /// [`NodesRequest`](./struct.NodesRequest.html) structure.
    NodesRequest(NodesRequest),
    /// [`NodesResponse`](./struct.NodesResponse.html) structure.
    NodesResponse(NodesResponse),
    /// [`CookieRequest`](./struct.CookieRequest.html) structure.
    CookieRequest(CookieRequest),
    /// [`CookieRequest`](./struct.CookieRequest.html) structure.
    CookieResponse(CookieResponse),
    /// [`CryptoHandshake`](./struct.CryptoHandshake.html) structure.
    CryptoHandshake(CryptoHandshake),
    /// [`CryptoData`](./struct.CryptoData.html) structure.
    CryptoData(CryptoData),
    /// [`DhtRequest`](./struct.DhtRequest.html) structure.
    DhtRequest(DhtRequest),
    /// [`LanDiscovery`](./struct.LanDiscovery.html) structure.
    LanDiscovery(LanDiscovery),
    /// [`OnionRequest0`](../onion/struct.OnionRequest0.html) structure.
    OnionRequest0(OnionRequest0),
    /// [`OnionRequest1`](../onion/struct.OnionRequest1.html) structure.
    OnionRequest1(OnionRequest1),
    /// [`OnionRequest2`](../onion/struct.OnionRequest2.html) structure.
    OnionRequest2(OnionRequest2),
    /// [`OnionAnnounceRequest`](../onion/struct.OnionAnnounceRequest.html) structure.
    OnionAnnounceRequest(OnionAnnounceRequest),
    /// [`OnionAnnounceResponse`](../onion/struct.OnionAnnounceResponse.html) structure.
    OnionAnnounceResponse(OnionAnnounceResponse),
    /// [`OnionDataRequest`](../onion/struct.OnionDataRequest.html) structure.
    OnionDataRequest(OnionDataRequest),
    /// [`OnionDataResponse`](../onion/struct.OnionDataResponse.html) structure.
    OnionDataResponse(OnionDataResponse),
    /// [`OnionResponse3`](../onion/struct.OnionResponse3.html) structure.
    OnionResponse3(OnionResponse3),
    /// [`OnionResponse2`](../onion/struct.OnionResponse2.html) structure.
    OnionResponse2(OnionResponse2),
    /// [`OnionResponse1`](../onion/struct.OnionResponse1.html) structure.
    OnionResponse1(OnionResponse1),
    /// [`BootstrapInfo`](./struct.BootstrapInfo.html) structure.
    BootstrapInfo(BootstrapInfo)
}

impl ToBytes for Packet {
    fn to_bytes<'a>(&self, buf: (&'a mut [u8], usize)) -> Result<(&'a mut [u8], usize), GenError> {
        match *self {
            Packet::PingRequest(ref p) => p.to_bytes(buf),
            Packet::PingResponse(ref p) => p.to_bytes(buf),
            Packet::NodesRequest(ref p) => p.to_bytes(buf),
            Packet::NodesResponse(ref p) => p.to_bytes(buf),
            Packet::CookieRequest(ref p) => p.to_bytes(buf),
            Packet::CookieResponse(ref p) => p.to_bytes(buf),
            Packet::CryptoHandshake(ref p) => p.to_bytes(buf),
            Packet::CryptoData(ref p) => p.to_bytes(buf),
            Packet::DhtRequest(ref p) => p.to_bytes(buf),
            Packet::LanDiscovery(ref p) => p.to_bytes(buf),
            Packet::OnionRequest0(ref p) => p.to_bytes(buf),
            Packet::OnionRequest1(ref p) => p.to_bytes(buf),
            Packet::OnionRequest2(ref p) => p.to_bytes(buf),
            Packet::OnionAnnounceRequest(ref p) => p.to_bytes(buf),
            Packet::OnionAnnounceResponse(ref p) => p.to_bytes(buf),
            Packet::OnionDataRequest(ref p) => p.to_bytes(buf),
            Packet::OnionDataResponse(ref p) => p.to_bytes(buf),
            Packet::OnionResponse3(ref p) => p.to_bytes(buf),
            Packet::OnionResponse2(ref p) => p.to_bytes(buf),
            Packet::OnionResponse1(ref p) => p.to_bytes(buf),
            Packet::BootstrapInfo(ref p) => p.to_bytes(buf)
        }
    }
}

impl FromBytes for Packet {
    named!(from_bytes<Packet>, alt!(
        map!(PingRequest::from_bytes, Packet::PingRequest) |
        map!(PingResponse::from_bytes, Packet::PingResponse) |
        map!(NodesRequest::from_bytes, Packet::NodesRequest) |
        map!(NodesResponse::from_bytes, Packet::NodesResponse) |
        map!(CookieRequest::from_bytes, Packet::CookieRequest) |
        map!(CookieResponse::from_bytes, Packet::CookieResponse) |
        map!(CryptoHandshake::from_bytes, Packet::CryptoHandshake) |
        map!(CryptoData::from_bytes, Packet::CryptoData) |
        map!(DhtRequest::from_bytes, Packet::DhtRequest) |
        map!(LanDiscovery::from_bytes, Packet::LanDiscovery) |
        map!(OnionRequest0::from_bytes, Packet::OnionRequest0) |
        map!(OnionRequest1::from_bytes, Packet::OnionRequest1) |
        map!(OnionRequest2::from_bytes, Packet::OnionRequest2) |
        map!(OnionAnnounceRequest::from_bytes, Packet::OnionAnnounceRequest) |
        map!(OnionAnnounceResponse::from_bytes, Packet::OnionAnnounceResponse) |
        map!(OnionDataRequest::from_bytes, Packet::OnionDataRequest) |
        map!(OnionDataResponse::from_bytes, Packet::OnionDataResponse) |
        map!(OnionResponse3::from_bytes, Packet::OnionResponse3) |
        map!(OnionResponse2::from_bytes, Packet::OnionResponse2) |
        map!(OnionResponse1::from_bytes, Packet::OnionResponse1) |
        map!(BootstrapInfo::from_bytes, Packet::BootstrapInfo)
    ));
}