hydra 0.1.40

A framework for writing fault tolerant, highly scalable applications with the Rust programming language.
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

use std::net::SocketAddr;

use sha2::Sha256;

use hmac::Hmac;
use hmac::Mac;

use bincode::Decode;
use bincode::Encode;

use crate::node_get_cookie;

/// Hmac using sha256.
type HmacSha256 = Hmac<Sha256>;

/// The frame used to handshake with other nodes.
#[derive(Debug, Encode, Decode)]
pub struct Hello {
    pub name: String,
    pub broadcast_address: SocketAddr,
    pub challenge: Vec<u8>,
}

impl Hello {
    /// Constructs a new instance of the [Hello] frame.
    pub fn new(name: String, broadcast_address: SocketAddr) -> Self {
        let mut challenge = {
            let cookie = node_get_cookie();
            let cookie = cookie
                .as_ref()
                .map(|cookie| cookie.as_bytes())
                .unwrap_or(&[0]);

            HmacSha256::new_from_slice(cookie).unwrap()
        };

        challenge.update(name.as_bytes());
        challenge.update(broadcast_address.to_string().as_bytes());

        Self {
            name,
            broadcast_address,
            challenge: challenge.finalize().into_bytes().to_vec(),
        }
    }

    /// Validates this [Hello] frame against our cookie.
    pub fn validate(&mut self) -> bool {
        let mut challenge = {
            let cookie = node_get_cookie();
            let cookie = cookie
                .as_ref()
                .map(|cookie| cookie.as_bytes())
                .unwrap_or(&[0]);

            HmacSha256::new_from_slice(cookie).unwrap()
        };

        challenge.update(self.name.as_bytes());
        challenge.update(self.broadcast_address.to_string().as_bytes());

        let wanted = challenge.finalize().into_bytes();

        let challenge = std::mem::take(&mut self.challenge);

        wanted.as_slice() == challenge
    }
}