demikernel 1.5.13

Kernel-Bypass LibOS Architecture
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
98
99
100
101
102
103
104
105

// Copyright (c) Microsoft Corporation.
// Licensed under the MIT license.

#![deny(clippy::all)]

//======================================================================================================================
// Modules
//======================================================================================================================

mod args;
mod async_close;
mod close;
mod create_pipe;
mod open_pipe;
mod pop_wait;
mod push_wait;
mod wait;

//======================================================================================================================
// Imports
//======================================================================================================================

use self::args::ProgramArguments;
use anyhow::Result;
use demikernel::{
    LibOS,
    LibOSName,
};

//======================================================================================================================
// Standalone Functions
//======================================================================================================================

fn main() -> Result<()> {
    let mut result: Vec<(String, String, Result<(), anyhow::Error>)> = Vec::new();

    let args: ProgramArguments = ProgramArguments::new(
        "pipe-test",
        "Pedro Henrique Penna <ppenna@microsoft.com>",
        "Integration test for pipe queues.",
    )?;

    let mut libos: LibOS = {
        let libos_name: LibOSName = LibOSName::from_env()?.into();
        LibOS::new(libos_name, None)?
    };

    match args.run_mode().as_str() {
        "standalone" => {
            demikernel::collect_test!(result, create_pipe::run(&mut libos, &args.pipe_name()));
            demikernel::collect_test!(result, open_pipe::run(&mut libos, &args.pipe_name()));
            demikernel::collect_test!(result, close::run(&mut libos, &args.pipe_name()));
            demikernel::collect_test!(result, wait::run(&mut libos));
            demikernel::collect_test!(result, async_close::run(&mut libos, &args.pipe_name()));

            // Dump results.
            demikernel::dump_test!(result)
        },
        "push-wait" => match args.peer_type().ok_or(anyhow::anyhow!("missing peer type"))?.as_str() {
            "client" => {
                let mut client: push_wait::PipeClient = push_wait::PipeClient::new(libos, args.pipe_name())?;
                client.run()
            },
            "server" => {
                let mut server: push_wait::PipeServer = push_wait::PipeServer::new(libos, args.pipe_name())?;
                server.run()
            },
            _ => anyhow::bail!("invalid peer type"),
        },
        "pop-wait" => match args.peer_type().ok_or(anyhow::anyhow!("missing peer type"))?.as_str() {
            "client" => {
                let mut client: pop_wait::PipeClient = pop_wait::PipeClient::new(libos, args.pipe_name())?;
                client.run()
            },
            "server" => {
                let mut server: pop_wait::PipeServer = pop_wait::PipeServer::new(libos, args.pipe_name())?;
                server.run()
            },
            _ => anyhow::bail!("invalid peer type"),
        },
        "push-wait-async" => match args.peer_type().ok_or(anyhow::anyhow!("missing peer type"))?.as_str() {
            "client" => {
                let mut client: push_wait::PipeClient = push_wait::PipeClient::new(libos, args.pipe_name())?;
                client.run_aynsc()
            },
            "server" => {
                let mut server: push_wait::PipeServer = push_wait::PipeServer::new(libos, args.pipe_name())?;
                server.run()
            },
            _ => anyhow::bail!("invalid peer type"),
        },
        "pop-wait-async" => match args.peer_type().ok_or(anyhow::anyhow!("missing peer type"))?.as_str() {
            "client" => {
                let mut client: pop_wait::PipeClient = pop_wait::PipeClient::new(libos, args.pipe_name())?;
                client.run()
            },
            "server" => {
                let mut server: pop_wait::PipeServer = pop_wait::PipeServer::new(libos, args.pipe_name())?;
                server.run_async()
            },
            _ => anyhow::bail!("invalid peer type"),
        },
        _ => anyhow::bail!("invalid run mode"),
    }
}