Remove some unused code and imports

This commit is contained in:
2024-02-23 13:24:37 +01:00
parent 467e2b6d21
commit 0d39ea71c7
4 changed files with 9 additions and 21 deletions

View File

@@ -1,17 +1,13 @@
use actix::prelude::*; use actix::prelude::*;
use uuid::Uuid; use uuid::Uuid;
use std::collections::HashMap; use std::collections::HashMap;
use std::io::Error;
use actix_tls::accept::rustls_0_22::{TlsStream};
use actix_rt::net::TcpStream;
use futures::SinkExt; use futures::SinkExt;
use thiserror::Error; use thiserror::Error;
use tracing::{info, error, warn, debug}; use tracing::{info, error, warn};
use libbonknet::ToPeerDataStream; use libbonknet::ToPeerDataStream;
use crate::TransportStream; use crate::TransportStream;
type RawStream = TlsStream<TcpStream>; #[allow(dead_code)]
#[derive(Error, Debug)] #[derive(Error, Debug)]
pub enum DataConnManagerError { pub enum DataConnManagerError {
#[error("Generic Failure")] #[error("Generic Failure")]

View File

@@ -160,8 +160,8 @@ impl Handler<TryStartDataStream> for PendingDataConnManager {
fn handle(&mut self, msg: TryStartDataStream, _ctx: &mut Self::Context) -> Self::Result { fn handle(&mut self, msg: TryStartDataStream, _ctx: &mut Self::Context) -> Self::Result {
use RegisterKind::*; use RegisterKind::*;
let idx = match msg.kind { let idx = match msg.kind {
Server => self.db.iter().enumerate().find(|(i, x)| x.server.conn_id == msg.conn_id), Server => self.db.iter().enumerate().find(|(_i, x)| x.server.conn_id == msg.conn_id),
Client => self.db.iter().enumerate().find(|(i, x)| x.client.conn_id == msg.conn_id), Client => self.db.iter().enumerate().find(|(_i, x)| x.client.conn_id == msg.conn_id),
}; };
if let Some((idx, record)) = idx { if let Some((idx, record)) = idx {
if record.client.transport.is_some() && record.server.transport.is_some() { if record.client.transport.is_some() && record.server.transport.is_some() {

View File

@@ -15,12 +15,6 @@ use crate::{TransportStream, TransportStreamRx, TransportStreamTx};
use uuid::Uuid; use uuid::Uuid;
use crate::pendingdataconndb::*; use crate::pendingdataconndb::*;
#[derive(MessageResponse)]
enum SendMsgResult {
Accepted,
Rejected,
}
#[derive(Error, Debug)] #[derive(Error, Debug)]
pub enum SendMsgError { pub enum SendMsgError {
#[error("Generic Failure")] #[error("Generic Failure")]
@@ -35,7 +29,7 @@ struct SendMsg {
struct ServerTransporter { struct ServerTransporter {
rx: Option<TransportStreamRx>, rx: Option<TransportStreamRx>,
tx: Arc<tokio::sync::Mutex<TransportStreamTx>>, tx: Arc<Mutex<TransportStreamTx>>,
timeout: Option<SpawnHandle>, timeout: Option<SpawnHandle>,
reply_channels: HashMap<u64, oneshot::Sender<FromServerReplyBody>>, reply_channels: HashMap<u64, oneshot::Sender<FromServerReplyBody>>,
} }
@@ -49,7 +43,7 @@ impl ServerTransporter {
let tx = FramedWrite::new(stx, codec.clone()); let tx = FramedWrite::new(stx, codec.clone());
ServerTransporter { ServerTransporter {
rx: Some(rx), rx: Some(rx),
tx: Arc::new(tokio::sync::Mutex::new(tx)), tx: Arc::new(Mutex::new(tx)),
timeout: None, timeout: None,
reply_channels: HashMap::new(), reply_channels: HashMap::new(),
} }
@@ -78,7 +72,7 @@ impl Actor for ServerTransporter {
let arc_tx = Arc::clone(&s.tx); let arc_tx = Arc::clone(&s.tx);
c.spawn(async move { c.spawn(async move {
arc_tx.lock().await.send(payload).await arc_tx.lock().await.send(payload).await
}.into_actor(s).map(|res, a, ctx| { }.into_actor(s).map(|res, _a, ctx| {
match res { match res {
Ok(_) => { Ok(_) => {
info!("Ping sent!"); info!("Ping sent!");
@@ -115,7 +109,7 @@ impl Handler<SendMsg> for ServerTransporter {
let payload: Bytes = rmp_serde::to_vec(&msg).unwrap().into(); let payload: Bytes = rmp_serde::to_vec(&msg).unwrap().into();
let arc_tx = self.tx.clone(); let arc_tx = self.tx.clone();
Box::pin(async move { Box::pin(async move {
arc_tx.lock().await.send(payload).await.map_err(|e| SendMsgError::GenericFailure)?; arc_tx.lock().await.send(payload).await.map_err(|_e| SendMsgError::GenericFailure)?;
info!("msg sent"); info!("msg sent");
let r = reply_channel_rx.await.unwrap(); let r = reply_channel_rx.await.unwrap();
info!("reply received"); info!("reply received");
@@ -226,7 +220,7 @@ impl Handler<StartTransporter> for ServerManager {
type Result = ResponseFuture<Result<(), ServerManagerError>>; type Result = ResponseFuture<Result<(), ServerManagerError>>;
fn handle(&mut self, msg: StartTransporter, _ctx: &mut Self::Context) -> Self::Result { fn handle(&mut self, msg: StartTransporter, _ctx: &mut Self::Context) -> Self::Result {
let mut entries = Arc::clone(&self.entries); let entries = Arc::clone(&self.entries);
Box::pin(async move { Box::pin(async move {
let mut entries_mg = entries.lock().await; let mut entries_mg = entries.lock().await;
if entries_mg.contains_key(&msg.name) { if entries_mg.contains_key(&msg.name) {

View File

@@ -5,12 +5,10 @@ use tokio::net::TcpStream;
use tokio_rustls::rustls::{ClientConfig, RootCertStore}; use tokio_rustls::rustls::{ClientConfig, RootCertStore};
use tokio_rustls::rustls::pki_types::{ServerName, CertificateDer, PrivatePkcs8KeyDer}; use tokio_rustls::rustls::pki_types::{ServerName, CertificateDer, PrivatePkcs8KeyDer};
use tokio_rustls::TlsConnector; use tokio_rustls::TlsConnector;
use tokio_util::bytes::BytesMut;
use tokio_util::codec::{Framed, LengthDelimitedCodec}; use tokio_util::codec::{Framed, LengthDelimitedCodec};
use libbonknet::*; use libbonknet::*;
use uuid::Uuid; use uuid::Uuid;
use tracing::{info, error}; use tracing::{info, error};
use libbonknet::ToPeerDataStream::{OkDataStreamOpen, OkDataStreamRequestAccepted, Refused, Revoked};
async fn datastream(tlsconfig: Arc<ClientConfig>, conn_id: Uuid) -> std::io::Result<()> { async fn datastream(tlsconfig: Arc<ClientConfig>, conn_id: Uuid) -> std::io::Result<()> {