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

//! protobuf protocol server implementation

use std::net::SocketAddr;

use futures::prelude::*;
use thiserror::Error;
use tokio::net::TcpStream;
use tokio_util::codec::Framed;

use crate::{
    api::proto::{self, message, ProtoApiError},
    global::{Global, InputSourceName, PriorityGuard},
};

mod codec;
use codec::*;

#[derive(Debug, Error)]
pub enum ProtoServerError {
    #[error("i/o error: {0}")]
    Io(#[from] futures_io::Error),
    #[error("decode error: {0}")]
    Codec(#[from] ProtoCodecError),
    #[error(transparent)]
    Api(#[from] ProtoApiError),
}

fn success_response(peer_addr: SocketAddr) -> message::HyperionReply {
    let reply = message::HyperionReply {
        r#type: message::hyperion_reply::Type::Reply.into(),
        success: Some(true),
        ..Default::default()
    };

    trace!("({}) sending success: {:?}", peer_addr, reply);
    reply
}

fn error_response(peer_addr: SocketAddr, error: impl std::fmt::Display) -> message::HyperionReply {
    let reply = message::HyperionReply {
        r#type: message::hyperion_reply::Type::Reply.into(),
        success: Some(false),
        error: Some(error.to_string()),
        ..Default::default()
    };

    trace!("({}) sending error: {:?}", peer_addr, reply);
    reply
}

pub async fn handle_client(
    (socket, peer_addr): (TcpStream, SocketAddr),
    global: Global,
) -> Result<(), ProtoServerError> {
    debug!("accepted new connection from {}", peer_addr);

    let (mut writer, mut reader) = Framed::new(socket, ProtoCodec::new()).split();

    // unwrap: cannot fail because the priority is None
    let source = global
        .register_input_source(InputSourceName::Protobuf { peer_addr }, None)
        .await
        .unwrap();

    let mut priority_guard = PriorityGuard::new_broadcast(&source);

    while let Some(request) = reader.next().await {
        let request = match request {
            Ok(rb) => rb,
            Err(error) => {
                error!("({}) error reading frame: {}", peer_addr, error);
                continue;
            }
        };

        trace!("({}) got request: {:?}", peer_addr, request);

        let reply = match proto::handle_request(peer_addr, request, &source, &mut priority_guard) {
            Ok(()) => success_response(peer_addr),
            Err(error) => {
                error!("({}) error processing request: {}", peer_addr, error);

                error_response(peer_addr, error)
            }
        };

        writer.send(reply).await?;
        writer.flush().await?;
    }

    Ok(())
}