rust/neon
1
0
Fork 0
mirror of https://github.com/neondatabase/neon.git synced 2026-05-24 08:30:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix pg raw decoding

Browse Source
This commit is contained in:
Conrad Ludgate
2024-09-13 09:40:51 +01:00
parent 24c48856a2
commit 34696381c5
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
proxy/src/bin/pglb.rs
View File

@@ -590,30 +590,30 @@ impl PgRawMessage {
src.reserve(4);
return Ok(None);
}
let length = src.get_u32() as usize - 4;
let length = u32::from_be_bytes(src[0..4].try_into().unwrap()) as usize;
if src.remaining() < length {
src.reserve(length - src.remaining());
return Ok(None);
}
if length == 4 && src.starts_with(&80877103u32.to_be_bytes()) {
if length == 8 && src[4..8] == 80877103u32.to_be_bytes() {
Ok(Some(PgRawMessage::SslRequest))
} else {
Ok(Some(PgRawMessage::Start(src.split_off(length).to_vec())))
Ok(Some(PgRawMessage::Start(src.split_to(length).to_vec())))
}
} else {
if src.remaining() < 5 {
src.reserve(5);
return Ok(None);
}
let tag = src.get_u8();
let length = src.get_u32() as usize - 4;
let tag = src[0];
let length = u32::from_be_bytes(src[1..5].try_into().unwrap()) as usize - 1;
if src.remaining() < length {
src.reserve(length - src.remaining());
return Ok(None);
}
Ok(Some(PgRawMessage::Generic {
tag,
payload: src.split_off(length).to_vec(),
payload: src.split_to(length).to_vec(),
}))
}
}