add to simple wisp client

This commit is contained in:
Toshit Chawda 2024-09-14 18:14:10 -07:00
parent 577ce71b89
commit 24ccd8d393
No known key found for this signature in database
GPG key ID: 91480ED99E2B3D9D
7 changed files with 82 additions and 9 deletions

View file

@ -7,7 +7,7 @@ use log::LevelFilter;
use regex::RegexSet;
use serde::{Deserialize, Serialize};
use wisp_mux::extensions::{
cert::CertAuthProtocolExtensionBuilder,
cert::{CertAuthProtocolExtension, CertAuthProtocolExtensionBuilder},
motd::MotdProtocolExtensionBuilder,
password::{PasswordProtocolExtension, PasswordProtocolExtensionBuilder},
udp::UdpProtocolExtensionBuilder,
@ -304,6 +304,7 @@ impl WispConfig {
get_certificates_from_paths(self.certificate_extension_keys.clone())
.await?,
)));
required_extensions.push(CertAuthProtocolExtension::ID);
}
None => {}
}

View file

@ -179,9 +179,7 @@ async fn handle_stream(
let id = muxstream.stream_id;
let (mut rx, mut tx) = muxstream.into_io().into_asyncrw().into_split();
match twisp::handle_twisp(id, &mut rx, &mut tx, twisp_map.clone(), pty, cmd)
.await
{
match twisp::handle_twisp(id, &mut rx, &mut tx, twisp_map.clone(), pty, cmd).await {
Ok(()) => {
let _ = closer.close(CloseReason::Voluntary).await;
}