44 lines
No EOL
1.6 KiB
Rust
44 lines
No EOL
1.6 KiB
Rust
use std::future::Future;
|
|
use std::io::Cursor;
|
|
use std::pin::Pin;
|
|
use std::sync::Arc;
|
|
use log::error;
|
|
use crate::prudp::packet::PRUDPPacket;
|
|
use crate::prudp::socket::{ConnectionData, SocketData};
|
|
use crate::rmc::message::RMCMessage;
|
|
use crate::rmc::response::{RMCResponse, RMCResponseResult, send_response};
|
|
use crate::rmc::response::ErrorCode::Core_NotImplemented;
|
|
|
|
type ContainedProtocolList = Box<[Box<dyn for<'a> Fn(&'a RMCMessage, &'a mut ConnectionData) -> Pin<Box<dyn Future<Output = Option<RMCResponse>> + Send + 'a>> + Send + Sync>]>;
|
|
|
|
pub struct RMCProtocolServer(ContainedProtocolList);
|
|
|
|
impl RMCProtocolServer{
|
|
pub fn new(protocols: ContainedProtocolList) -> Arc<Self>{
|
|
Arc::new(Self(protocols))
|
|
}
|
|
|
|
pub async fn process_message(&self, packet: PRUDPPacket, socket: &SocketData, connection: &mut ConnectionData){
|
|
let Ok(rmc) = RMCMessage::new(&mut Cursor::new(&packet.payload)) else {
|
|
error!("error reading rmc message");
|
|
return;
|
|
};
|
|
|
|
for proto in &self.0 {
|
|
if let Some(response) = proto(&rmc, connection).await {
|
|
send_response(&packet, &socket, connection, response).await;
|
|
return;
|
|
}
|
|
}
|
|
|
|
error!("tried to send message to unimplemented protocol {} with method id {}", rmc.protocol_id, rmc.method_id);
|
|
|
|
send_response(&packet, &socket, connection, RMCResponse{
|
|
protocol_id: rmc.protocol_id as u8,
|
|
response_result: RMCResponseResult::Error {
|
|
call_id: rmc.call_id,
|
|
error_code: Core_NotImplemented
|
|
}
|
|
}).await;
|
|
}
|
|
} |