YGGverse/aquatic
1
0
Fork 0
mirror of https://github.com/YGGverse/aquatic.git synced 2026-04-02 02:35:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

aquatic_ws: add settings for max ws message sizes, set low defaults

Browse source
This commit is contained in:
Joakim Frostegård 2020-05-23 14:44:03 +02:00
parent 0f6d6d4b21
commit a30335730b
3 changed files with 27 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

10
aquatic_ws/src/lib/network/mod.rs
View file

@ -6,6 +6,7 @@ use log::info;
use native_tls::TlsAcceptor;
use mio::{Events, Poll, Interest, Token};
use mio::net::TcpListener;
use tungstenite::protocol::WebSocketConfig;
use crate::common::*;
use crate::config::Config;
@ -54,6 +55,11 @@ pub fn run_poll_loop(
let poll_timeout = Duration::from_millis(
config.network.poll_timeout_milliseconds
);
let ws_config = WebSocketConfig {
max_message_size: Some(config.network.websocket_max_message_size),
max_frame_size: Some(config.network.websocket_max_frame_size),
max_send_queue: None,
};
let mut listener = TcpListener::from_std(listener);
let mut poll = Poll::new().expect("create poll");
@ -79,6 +85,7 @@ pub fn run_poll_loop(
if token.0 == 0 {
accept_new_streams(
ws_config,
&mut listener,
&mut poll,
&mut connections,
@ -113,6 +120,7 @@ pub fn run_poll_loop(
fn accept_new_streams(
ws_config: WebSocketConfig,
listener: &mut TcpListener,
poll: &mut Poll,
connections: &mut ConnectionMap,
@ -136,7 +144,7 @@ fn accept_new_streams(
.register(&mut stream, token, Interest::READABLE)
.unwrap();
let connection = Connection::new(valid_until, stream);
let connection = Connection::new(ws_config, valid_until, stream);
connections.insert(token, connection);
},