Projects/muscl
12
4
Fork 0
Code Issues 69 Pull Requests Actions Packages Releases Activity

core/protocol: set request/response max size

Browse Source
This commit is contained in:
Oystein Kristoffer Tveit
2025-12-16 18:50:22 +09:00
parent 10ef171c91
commit 97908ce887
1 changed files with 17 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/core/protocol/commands.rs
View File

@@ -55,13 +55,26 @@ pub type ClientToServerMessageStream = SerdeFramed<
Bincode<Response, Request>,
>;
pub fn create_server_to_client_message_stream(socket: UnixStream) -> ServerToClientMessageStream {
let length_delimited = Framed::new(socket, LengthDelimitedCodec::new());
const MAX_REQUEST_FRAME_LENGTH: usize = 100 * 1024; // 100 KB
const MAX_RESPONSE_FRAME_LENGTH: usize = 1024 * 1024; // 1 MB
pub fn create_client_to_server_message_stream(socket: UnixStream) -> ClientToServerMessageStream {
let codec = {
let mut codec = LengthDelimitedCodec::new();
codec.set_max_frame_length(MAX_REQUEST_FRAME_LENGTH);
codec
};
let length_delimited = Framed::new(socket, codec);
tokio_serde::Framed::new(length_delimited, Bincode::default())
}
pub fn create_client_to_server_message_stream(socket: UnixStream) -> ClientToServerMessageStream {
let length_delimited = Framed::new(socket, LengthDelimitedCodec::new());
pub fn create_server_to_client_message_stream(socket: UnixStream) -> ServerToClientMessageStream {
let codec = {
let mut codec = LengthDelimitedCodec::new();
codec.set_max_frame_length(MAX_RESPONSE_FRAME_LENGTH);
codec
};
let length_delimited = Framed::new(socket, codec);
tokio_serde::Framed::new(length_delimited, Bincode::default())
}