moparisthebest/xmpp-proxy
1
0
Fork 0
Code Issues Pull Requests Projects Releases 3 Wiki Activity

Send header Access-Control-Allow-Origin: * for outgoing WebSocket connections, thanks singpolyma

Browse Source
This commit is contained in:
Travis Burtrum 2023-01-31 22:01:05 -05:00
parent 30ca8609d5
commit 2a6dea344f
1 changed files with 18 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/websocket/mod.rs
View File

@ -1,7 +1,14 @@
use anyhow::Result; use anyhow::Result;
use futures::StreamExt; use futures::StreamExt;
use futures_util::stream::{SplitSink, SplitStream}; use futures_util::stream::{SplitSink, SplitStream};
use tokio_tungstenite::{tungstenite::protocol::WebSocketConfig, WebSocketStream}; use tokio_tungstenite::{
tungstenite::{
handshake::server::{Request, Response},
http::header::ACCESS_CONTROL_ALLOW_ORIGIN,
protocol::WebSocketConfig,
},
WebSocketStream,
};
#[cfg(feature = "incoming")] #[cfg(feature = "incoming")]
pub mod incoming; pub mod incoming;
@ -29,8 +36,16 @@ impl<T: tokio::io::AsyncRead + tokio::io::AsyncWrite> AsyncReadAndWrite for T {}
pub async fn incoming_websocket_connection(stream: Box<dyn AsyncReadAndWrite + Unpin + Send>, max_stanza_size_bytes: usize) -> Result<(StanzaRead, StanzaWrite)> { pub async fn incoming_websocket_connection(stream: Box<dyn AsyncReadAndWrite + Unpin + Send>, max_stanza_size_bytes: usize) -> Result<(StanzaRead, StanzaWrite)> {
// accept the websocket // accept the websocket
// todo: check SEC_WEBSOCKET_PROTOCOL or ORIGIN ? let stream = tokio_tungstenite::accept_hdr_async_with_config(
let stream = tokio_tungstenite::accept_async_with_config(stream, ws_cfg(max_stanza_size_bytes)).await?; stream,
|_request: &Request, mut response: Response| {
// todo: check SEC_WEBSOCKET_PROTOCOL or ORIGIN ?
response.headers_mut().append(ACCESS_CONTROL_ALLOW_ORIGIN, "*".parse().expect("known to be good value"));
Ok(response)
},
ws_cfg(max_stanza_size_bytes),
)
.await?;
let (in_wr, in_rd) = stream.split(); let (in_wr, in_rd) = stream.split();