apt-swarm 0.6.0

🥸 Experimental p2p gossip network for OpenPGP signature transparency 🥸
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

use crate::db;
use crate::db::compression;
use crate::db::header::BlockHeader;
use crate::errors::*;
use async_trait::async_trait;
use tokio::io::{AsyncRead, AsyncReadExt, AsyncSeek, AsyncSeekExt, SeekFrom};

#[async_trait]
pub trait Consume {
    type Item: Ord + 'static;

    async fn consume<R: AsyncRead + AsyncSeek + Unpin + Send>(
        mut reader: R,
        header: &BlockHeader,
    ) -> Result<Self::Item>;
}

/// Read and return data
pub struct ReadValue;

#[async_trait]
impl Consume for ReadValue {
    type Item = db::Value;

    async fn consume<R: AsyncRead + AsyncSeek + Unpin + Send>(
        mut reader: R,
        header: &BlockHeader,
    ) -> Result<Self::Item> {
        let mut compressed = vec![0u8; header.data_length as usize];
        reader
            .read_exact(&mut compressed)
            .await
            .context("Failed to read block data")?;

        let data = compression::decompress(&compressed)
            .await
            .context("Failed to decompress block data")?;

        Ok(data)
    }
}

/// Skip over data immediately, possibly beyond EOF
pub struct FastSkipValue;

#[async_trait]
impl Consume for FastSkipValue {
    type Item = ();

    async fn consume<R: AsyncRead + AsyncSeek + Unpin + Send>(
        mut reader: R,
        header: &BlockHeader,
    ) -> Result<Self::Item> {
        reader
            .seek(SeekFrom::Current(header.data_length as i64))
            .await
            .context("Failed to seek over block data")?;
        trace!("Seeked forward by {} bytes", header.data_length);
        Ok(())
    }
}

/// Skip over data, verify enough data is present before seek
pub struct CheckedSkipValue;

#[async_trait]
impl Consume for CheckedSkipValue {
    type Item = ();

    async fn consume<R: AsyncRead + AsyncSeek + Unpin + Send>(
        mut reader: R,
        header: &BlockHeader,
    ) -> Result<Self::Item> {
        // determine if enough data is available
        let pos = reader
            .stream_position()
            .await
            .context("Failed to determine stream position")?;
        let end = reader
            .seek(SeekFrom::End(0))
            .await
            .context("Failed to seek to file end")?;

        let remaining = end - pos;
        if remaining < header.data_length {
            bail!(
                "Seek failed, missing {} bytes",
                header.data_length - remaining
            );
        }

        // we have enough data, perform seek
        let new = reader
            .seek(SeekFrom::Start(pos + header.data_length))
            .await
            .context("Failed to seek over block data")?;
        trace!("Seeked forward by {} bytes", new - pos);
        Ok(())
    }
}