YGGverse/ggemini
1
0
Fork 0
mirror of https://github.com/YGGverse/ggemini.git synced 2026-03-31 09:05:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add HEADER_BYTES_LEN

Browse source
This commit is contained in:
yggverse 2024-10-28 13:40:16 +02:00
parent 2a01232f6a
commit 7b0bc23a9f
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/client/response/header.rs
View file

@ -14,6 +14,8 @@ use gio::{
};
use glib::{Bytes, Priority};
pub const HEADER_BYTES_LEN: usize = 0x400; // 1024
pub struct Header {
status: Status,
meta: Option<Meta>,
@ -34,7 +36,7 @@ impl Header {
) {
// Take header buffer from input stream
Self::read_from_socket_connection_async(
Vec::with_capacity(1024),
Vec::with_capacity(HEADER_BYTES_LEN),
socket_connection,
match cancellable {
Some(value) => Some(value),
@ -105,7 +107,7 @@ impl Header {
move |result| match result {
Ok(bytes) => {
// Expect valid header length
if bytes.len() == 0 || buffer.len() + 1 > 1024 {
if bytes.len() == 0 || buffer.len() >= HEADER_BYTES_LEN {
return callback(Err((Error::Protocol, None)));
}