zedmq 0.7.0

A lightweight, safe, pure-Rust ØMQ/ZMTP library 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
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

use std::io::{self, Read, Write};

use crate::stream::Transport;

use super::FrameBuf;

/// A builder struct used to handle `greeting` and `handshake` steps.
pub(crate) struct ZMTP {
    // security: Option<()>,
    transport: Transport,
}

impl ZMTP {
    /// The "connect" side of the connection.
    pub fn connect<F>(produce_transport: F) -> io::Result<Self>
    where
        F: FnOnce() -> io::Result<Transport>,
    {
        let transport = produce_transport()?;

        Ok(Self {
            // security: None,
            transport,
        })
    }

    /// Perform the greeting step of the ZMTP spec.
    pub fn greet(mut self, (_major, _minor, _): (u8, u8, u8), as_server: bool) -> io::Result<Self> {
        let (partial, remaining) = {
            let mut greeting = crate::codec::Greeting::build();
            greeting.as_server(as_server);
            // greeting.security(self.security)
            greeting.into_parts()
        };

        // Send partial greeting
        self.transport.write(&partial)?;

        // Inspect remote partial greeting.
        {
            let mut buf = [0u8; 12];
            let n = self.transport.read(&mut buf)?;
            assert_eq!(n, 12, "{:?}", buf);

            // let peer_major = buf[10];

            // if peer_major != major {
            //     return Err(io::Error::new(io::ErrorKind::ConnectionAborted, "peer major is not the same as us."))
            // }

            // let peer_minor = buf[11];

            // if peer_minor > minor {
            //     return Err(io::Error::new(io::ErrorKind::ConnectionAborted, "peer minor is higher than us."))
            // }
        }

        // Send remaining greeting
        self.transport.write(&remaining)?;

        Ok(self)
    }

    pub fn ready<'b>(self, socket_type: &'b str) -> io::Result<Transport> {
        let Self { mut transport, .. } = self;

        {
            // Read the remaining remote greeting.
            let mut buf = [0u8; 52];
            let n = transport.read(&mut buf[..])?;
            assert_eq!(n, 52);
            // TODO: parse, this contains the security mechanism (by default NULL) and some extra metadata.

            // Inspect remote handshake.
            let mut buf = [0u8; 64];
            let _n = transport.read(&mut buf)?;

            // dbg!((super::Frame { bytes: &buf[..n] }.try_into_command()).unwrap());

            // TODO: validate handshake, this contains (for NULL security mechanisms) the following properties:
            //  - Socket-Type {type} i.e. PUSH, PULL, DEALER, ROUTER, PAIR
            //  - Identity; only if WE are ROUTER and they are using a ROUTER compatible socket type with a custom routing id.
        }

        // Send handshake

        let handshake = {
            let properties = vec![("Socket-Type", socket_type)];

            FrameBuf::short_command("READY", Some(properties))
        };

        transport.write(handshake.as_ref())?;

        Ok(transport)
    }
}