berkus
/
actix-web
mirror of https://github.com/fafhrd91/actix-web
1
0
Fork 0
Code Issues Releases Wiki Activity

check encoder impl consistency

This commit is contained in:
Rob Ede 2021-02-09 06:52:33 +00:00
parent 782e6ed734
commit 14125085d6
No known key found for this signature in database
GPG Key ID: C2A3B36E841A91E6
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
actix-http/src/h1/encoder.rs
View File

@ -148,6 +148,8 @@ pub(crate) trait MessageType: Sized {
for val in value.iter() { for val in value.iter() {
let v = val.as_ref(); let v = val.as_ref();
let v_len = v.len(); let v_len = v.len();
// key length + value length + colon + space + \r\n
let len = k_len + v_len + 4; let len = k_len + v_len + 4;
if len > remaining { if len > remaining {
@ -156,6 +158,7 @@ pub(crate) trait MessageType: Sized {
unsafe { unsafe {
dst.advance_mut(pos); dst.advance_mut(pos);
} }
pos = 0; pos = 0;
dst.reserve(len * 2); dst.reserve(len * 2);
remaining = dst.capacity() - dst.len(); remaining = dst.capacity() - dst.len();
@ -169,6 +172,7 @@ pub(crate) trait MessageType: Sized {
// the cursor matches the number of bytes written // the cursor matches the number of bytes written
unsafe { unsafe {
if camel_case { if camel_case {
// use Camel-Case headers
write_camel_case(k, from_raw_parts_mut(buf, k_len)); write_camel_case(k, from_raw_parts_mut(buf, k_len));
} else { } else {
write_data(k, buf, k_len); write_data(k, buf, k_len);