hibana 0.5.2

Const-projected Affine Multiparty Session Types for choreography-first Rust protocols
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

//! Internal generativity helpers used to brand rendezvous instances and lanes.
//!
//! The B+ execution plan requires that rendezvous be the exclusive authority for
//! minting control-plane witnesses.  We model this by giving every rendezvous a
//! zero-sized brand token and handing out `Guard<'brand>` projections to code
//! that must prove it is operating within that rendezvous instance.  Each call
//! to [`with_brand`] produces a fresh, unforgeable brand lifetime.

use core::marker::PhantomData;

/// Unique brand token.  The invariant is that a `Brand<'brand>` can only be
/// created inside [`with_brand`] and therefore cannot be forged by user code.
#[derive(Clone, Copy)]
pub(crate) struct Brand<'brand> {
    _marker: PhantomData<&'brand mut &'brand ()>,
}

#[cfg(test)]
impl<'brand> Brand<'brand> {
    #[inline]
    const fn new() -> Self {
        Self {
            _marker: PhantomData,
        }
    }

    #[inline]
    pub(crate) fn guard(&self) -> Guard<'brand> {
        Guard::new()
    }
}

/// Lightweight projection of a brand that can be stored inside data
/// structures without exposing the brand type itself.
#[derive(Clone, Copy)]
pub(crate) struct Guard<'brand> {
    _marker: PhantomData<&'brand mut &'brand ()>,
}

impl<'brand> Guard<'brand> {
    #[inline]
    pub(crate) const fn new() -> Self {
        Self {
            _marker: PhantomData,
        }
    }
}

/// Execute `f` with a freshly minted brand.  Each invocation receives a brand
/// that is unique to the call site, ensuring that witnesses cannot outlive the
/// rendezvous that produced them.
#[cfg(test)]
pub(crate) fn with_brand<R>(f: impl for<'brand> FnOnce(Brand<'brand>) -> R) -> R {
    struct Wrapper<F>(F);

    impl<F> Wrapper<F> {
        #[inline]
        fn run<R>(self) -> R
        where
            F: for<'brand> FnOnce(Brand<'brand>) -> R,
        {
            fn call<'brand, R>(f: impl FnOnce(Brand<'brand>) -> R) -> R {
                f(Brand::new())
            }

            call(self.0)
        }
    }

    Wrapper(f).run()
}

#[cfg(test)]
mod tests {
    use super::with_brand;

    #[test]
    fn brands_are_unique() {
        with_brand(|brand_a| {
            let guard_a = brand_a.guard();
            with_brand(|brand_b| {
                let guard_b = brand_b.guard();
                // Guards from distinct brands must not be interchangeable.  The
                // following line would fail to compile if we attempted to pass
                // both guards to a function expecting the same lifetime.
                let _ = guard_a;
                let _ = guard_b;
            });
        });
    }
}