mercuryworkshop-mirror/epoxy-tls
1
0
Fork 0
mirror of https://github.com/MercuryWorkshop/epoxy-tls.git synced 2025-05-13 06:20:02 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

wisp-server-rust

Browse source
This commit is contained in:
r58Playz 2024-01-24 13:19:57 -08:00
parent 24d145cc66
commit 379e07d643
3 changed files with 15 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

23
server/src/main.rs
View file

@ -122,25 +122,24 @@ async fn accept_ws(
fut: upgrade::UpgradeFut,
addr: String,
) -> Result<(), Box<dyn std::error::Error>> {
let ws_stream = FragmentCollector::new(fut.await?);
let ws_stream = lockedws::LockedWebSocket::new(ws_stream);
let (mut rx, tx) = fut.await?.split(tokio::io::split);
let tx = lockedws::LockedWebSocketWrite::new(tx);
let stream_map = Arc::new(dashmap::DashMap::<u32, mpsc::UnboundedSender<WsEvent>>::new());
println!("{:?}: connected", addr);
ws_stream
.write_frame(ws::Frame::from(Packet::new_continue(0, u32::MAX)).into())
tx.write_frame(ws::Frame::from(Packet::new_continue(0, u32::MAX)).into())
.await?;
while let Ok(frame) = ws_stream.read_frame().await {
while let Ok(frame) = rx.read_frame(&mut |x| tx.write_frame(x)).await {
if let Ok(packet) = Packet::try_from(ws::Frame::try_from(frame)?) {
use PacketType::*;
match packet.packet {
Connect(inner_packet) => {
let (ch_tx, mut ch_rx) = mpsc::unbounded_channel::<WsEvent>();
stream_map.clone().insert(packet.stream_id, ch_tx);
let ws_stream_cloned = ws_stream.clone();
let tx_cloned = tx.clone();
tokio::spawn(async move {
let tcp_stream = match TcpStream::connect(format!(
"{}:{}",
@ -150,7 +149,7 @@ async fn accept_ws(
{
Ok(stream) => stream,
Err(err) => {
ws_stream_cloned
tx_cloned
.write_frame(
ws::Frame::from(Packet::new_close(packet.stream_id, 0x03))
.into(),
@ -160,16 +159,14 @@ async fn accept_ws(
return Err(Box::new(err));
}
};
println!("muxing");
let mut tcp_stream = Framed::new(tcp_stream, BytesCodec::new());
loop {
tokio::select! {
event = tcp_stream.next() => {
println!("recvd");
if let Some(res) = event {
match res {
Ok(buf) => {
ws_stream_cloned.write_frame(
tx_cloned.write_frame(
ws::Frame::from(
Packet::new_data(
packet.stream_id,
@ -179,7 +176,7 @@ async fn accept_ws(
).await.map_err(std::io::Error::other)?;
}
Err(err) => {
ws_stream_cloned
tx_cloned
.write_frame(
ws::Frame::from(Packet::new_close(
packet.stream_id,
@ -199,8 +196,7 @@ async fn accept_ws(
match event {
WsEvent::Send(buf) => {
tcp_stream.send(buf).await?;
println!("sending");
ws_stream_cloned
tx_cloned
.write_frame(
ws::Frame::from(
Packet::new_continue(
@ -209,7 +205,6 @@ async fn accept_ws(
)
).into()
).await.map_err(std::io::Error::other)?;
println!("sent");
}
WsEvent::Close => {
break;