privsep 0.0.2

Privilege Separation for Rust
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

use privsep::{imsg, net::Fd};
use serde_derive::{Deserialize, Serialize};
use std::{io, net::TcpListener, os::unix::io::IntoRawFd, time::Duration};
use tokio::time::interval;

#[derive(Debug, Serialize, Deserialize)]
struct Message {
    id: usize,
    name: String,
}

#[tokio::test(flavor = "multi_thread")]
async fn test_imsg() -> Result<(), io::Error> {
    unix_channel().await
}

async fn unix_channel() -> Result<(), std::io::Error> {
    let (sender, receiver) = imsg::Handler::pair()?;
    let mut count = 3;

    tokio::spawn(async move {
        let mut interval = interval(Duration::from_millis(100));

        let fd = TcpListener::bind("127.0.0.1:1234")
            .ok()
            .map(|stream| stream.into_raw_fd())
            .map(Fd::from);

        for id in 1..=count {
            interval.tick().await;

            let message = Message {
                id,
                name: "test".to_string(),
            };

            if let Err(err) = sender
                .send_message(imsg::Message::min(), fd.as_ref(), &message)
                .await
            {
                eprintln!("Failed to send message: {}", err);
            }
        }
    });

    let res = loop {
        match { receiver.recv_message::<Message>().await } {
            Ok(None) => break Ok(()),
            Ok(Some((imsg, fd, message))) => {
                count -= 1;
                println!(
                    "Received message: {:?}, fd: {:?}, data: {:?}",
                    imsg, fd, message
                );
            }
            Err(err) => {
                eprintln!("Failed to receive message: {}", err);
                break Err(err);
            }
        }
    };

    assert!(count == 0, "did not receive expected messages");

    res
}