1
0
Fork 0
mirror of https://github.com/actix/actix-web.git synced 2024-11-05 16:39:36 +00:00
actix-web/src/helpers.rs

269 lines
8.4 KiB
Rust
Raw Normal View History

2018-03-18 18:05:44 +00:00
use std::{mem, ptr, slice};
2017-12-14 00:44:35 +00:00
use std::cell::RefCell;
2017-12-15 03:34:31 +00:00
use std::rc::Rc;
use std::ops::{Deref, DerefMut};
use std::collections::VecDeque;
2017-12-26 03:42:55 +00:00
use bytes::{BufMut, BytesMut};
2017-12-25 21:40:06 +00:00
use http::Version;
2017-12-14 00:44:35 +00:00
use httprequest::HttpInnerMessage;
2017-12-15 03:34:31 +00:00
/// Internal use only! unsafe
pub(crate) struct SharedMessagePool(RefCell<VecDeque<Rc<HttpInnerMessage>>>);
2017-12-15 03:34:31 +00:00
impl SharedMessagePool {
pub fn new() -> SharedMessagePool {
SharedMessagePool(RefCell::new(VecDeque::with_capacity(128)))
}
2018-02-20 01:21:04 +00:00
#[inline]
pub fn get(&self) -> Rc<HttpInnerMessage> {
2017-12-15 03:34:31 +00:00
if let Some(msg) = self.0.borrow_mut().pop_front() {
msg
} else {
Rc::new(HttpInnerMessage::default())
2017-12-15 03:34:31 +00:00
}
}
2018-02-20 01:21:04 +00:00
#[inline]
pub fn release(&self, mut msg: Rc<HttpInnerMessage>) {
2017-12-16 06:49:48 +00:00
let v = &mut self.0.borrow_mut();
if v.len() < 128 {
2017-12-15 03:34:31 +00:00
Rc::get_mut(&mut msg).unwrap().reset();
2017-12-16 06:49:48 +00:00
v.push_front(msg);
2017-12-15 03:34:31 +00:00
}
}
}
pub(crate) struct SharedHttpInnerMessage(
Option<Rc<HttpInnerMessage>>, Option<Rc<SharedMessagePool>>);
2017-12-15 03:34:31 +00:00
impl Drop for SharedHttpInnerMessage {
2017-12-15 03:34:31 +00:00
fn drop(&mut self) {
if let Some(ref pool) = self.1 {
if let Some(msg) = self.0.take() {
if Rc::strong_count(&msg) == 1 {
pool.release(msg);
}
}
}
}
}
impl Deref for SharedHttpInnerMessage {
type Target = HttpInnerMessage;
2017-12-15 03:34:31 +00:00
fn deref(&self) -> &HttpInnerMessage {
2017-12-15 03:34:31 +00:00
self.get_ref()
}
}
impl DerefMut for SharedHttpInnerMessage {
2017-12-15 03:34:31 +00:00
fn deref_mut(&mut self) -> &mut HttpInnerMessage {
2017-12-15 03:34:31 +00:00
self.get_mut()
}
}
impl Clone for SharedHttpInnerMessage {
2017-12-15 03:34:31 +00:00
fn clone(&self) -> SharedHttpInnerMessage {
SharedHttpInnerMessage(self.0.clone(), self.1.clone())
2017-12-15 03:34:31 +00:00
}
}
impl Default for SharedHttpInnerMessage {
2017-12-15 03:34:31 +00:00
fn default() -> SharedHttpInnerMessage {
SharedHttpInnerMessage(Some(Rc::new(HttpInnerMessage::default())), None)
2017-12-15 03:34:31 +00:00
}
}
impl SharedHttpInnerMessage {
2017-12-15 03:34:31 +00:00
pub fn from_message(msg: HttpInnerMessage) -> SharedHttpInnerMessage {
SharedHttpInnerMessage(Some(Rc::new(msg)), None)
2017-12-15 03:34:31 +00:00
}
pub fn new(msg: Rc<HttpInnerMessage>, pool: Rc<SharedMessagePool>) -> SharedHttpInnerMessage {
SharedHttpInnerMessage(Some(msg), Some(pool))
2017-12-15 03:34:31 +00:00
}
#[inline(always)]
#[allow(mutable_transmutes)]
#[cfg_attr(feature = "cargo-clippy", allow(mut_from_ref, inline_always))]
pub fn get_mut(&self) -> &mut HttpInnerMessage {
let r: &HttpInnerMessage = self.0.as_ref().unwrap().as_ref();
2017-12-15 03:34:31 +00:00
unsafe{mem::transmute(r)}
}
2017-12-16 06:49:48 +00:00
#[inline(always)]
#[cfg_attr(feature = "cargo-clippy", allow(inline_always))]
pub fn get_ref(&self) -> &HttpInnerMessage {
2017-12-15 03:34:31 +00:00
self.0.as_ref().unwrap()
}
}
2017-12-14 00:44:35 +00:00
const DEC_DIGITS_LUT: &[u8] =
b"0001020304050607080910111213141516171819\
2021222324252627282930313233343536373839\
4041424344454647484950515253545556575859\
6061626364656667686970717273747576777879\
8081828384858687888990919293949596979899";
2017-12-25 21:40:06 +00:00
pub(crate) fn write_status_line(version: Version, mut n: u16, bytes: &mut BytesMut) {
2017-12-26 03:42:55 +00:00
let mut buf: [u8; 13] = [b'H', b'T', b'T', b'P', b'/', b'1', b'.', b'1',
b' ', b' ', b' ', b' ', b' '];
2017-12-25 21:40:06 +00:00
match version {
Version::HTTP_2 => buf[5] = b'2',
Version::HTTP_10 => buf[7] = b'0',
Version::HTTP_09 => {buf[5] = b'0'; buf[7] = b'9';},
_ => (),
}
2017-12-26 03:42:55 +00:00
let mut curr: isize = 12;
2017-12-14 00:44:35 +00:00
let buf_ptr = buf.as_mut_ptr();
let lut_ptr = DEC_DIGITS_LUT.as_ptr();
2017-12-26 03:42:55 +00:00
let four = n > 999;
2017-12-14 00:44:35 +00:00
unsafe {
// decode 2 more chars, if > 2 chars
2017-12-26 03:42:55 +00:00
let d1 = (n % 100) << 1;
n /= 100;
curr -= 2;
ptr::copy_nonoverlapping(lut_ptr.offset(d1 as isize), buf_ptr.offset(curr), 2);
2017-12-14 00:44:35 +00:00
// decode last 1 or 2 chars
if n < 10 {
curr -= 1;
*buf_ptr.offset(curr) = (n as u8) + b'0';
} else {
let d1 = n << 1;
curr -= 2;
2017-12-26 03:42:55 +00:00
ptr::copy_nonoverlapping(lut_ptr.offset(d1 as isize), buf_ptr.offset(curr), 2);
2017-12-14 00:44:35 +00:00
}
}
2018-03-18 18:05:44 +00:00
bytes.put_slice(&buf);
2017-12-26 03:42:55 +00:00
if four {
bytes.put(b' ');
}
}
pub(crate) fn write_content_length(mut n: usize, bytes: &mut BytesMut) {
if n < 10 {
let mut buf: [u8; 21] = [b'\r',b'\n',b'c',b'o',b'n',b't',b'e',
b'n',b't',b'-',b'l',b'e',b'n',b'g',
b't',b'h',b':',b' ',b'0',b'\r',b'\n'];
buf[18] = (n as u8) + b'0';
2018-03-18 18:05:44 +00:00
bytes.put_slice(&buf);
2017-12-26 03:42:55 +00:00
} else if n < 100 {
let mut buf: [u8; 22] = [b'\r',b'\n',b'c',b'o',b'n',b't',b'e',
b'n',b't',b'-',b'l',b'e',b'n',b'g',
b't',b'h',b':',b' ',b'0',b'0',b'\r',b'\n'];
let d1 = n << 1;
unsafe {
ptr::copy_nonoverlapping(
DEC_DIGITS_LUT.as_ptr().offset(d1 as isize), buf.as_mut_ptr().offset(18), 2);
}
2018-03-18 18:05:44 +00:00
bytes.put_slice(&buf);
2017-12-26 03:42:55 +00:00
} else if n < 1000 {
let mut buf: [u8; 23] = [b'\r',b'\n',b'c',b'o',b'n',b't',b'e',
b'n',b't',b'-',b'l',b'e',b'n',b'g',
b't',b'h',b':',b' ',b'0',b'0',b'0',b'\r',b'\n'];
// decode 2 more chars, if > 2 chars
let d1 = (n % 100) << 1;
n /= 100;
unsafe {ptr::copy_nonoverlapping(
2018-01-02 21:39:32 +00:00
DEC_DIGITS_LUT.as_ptr().offset(d1 as isize), buf.as_mut_ptr().offset(19), 2)};
2017-12-26 03:42:55 +00:00
2018-01-02 21:39:32 +00:00
// decode last 1
buf[18] = (n as u8) + b'0';
2017-12-26 03:42:55 +00:00
2018-03-18 18:05:44 +00:00
bytes.put_slice(&buf);
2017-12-26 03:42:55 +00:00
} else {
2018-03-18 18:05:44 +00:00
bytes.put_slice(b"\r\ncontent-length: ");
2017-12-26 03:42:55 +00:00
convert_usize(n, bytes);
}
2017-12-14 00:44:35 +00:00
}
2017-12-16 15:29:15 +00:00
pub(crate) fn convert_usize(mut n: usize, bytes: &mut BytesMut) {
2017-12-14 00:44:35 +00:00
let mut curr: isize = 39;
2017-12-25 21:40:06 +00:00
let mut buf: [u8; 41] = unsafe { mem::uninitialized() };
buf[39] = b'\r';
buf[40] = b'\n';
2017-12-14 00:44:35 +00:00
let buf_ptr = buf.as_mut_ptr();
let lut_ptr = DEC_DIGITS_LUT.as_ptr();
unsafe {
2017-12-25 21:40:06 +00:00
// eagerly decode 4 characters at a time
while n >= 10_000 {
let rem = (n % 10_000) as isize;
n /= 10_000;
let d1 = (rem / 100) << 1;
let d2 = (rem % 100) << 1;
curr -= 4;
ptr::copy_nonoverlapping(lut_ptr.offset(d1), buf_ptr.offset(curr), 2);
ptr::copy_nonoverlapping(lut_ptr.offset(d2), buf_ptr.offset(curr + 2), 2);
2017-12-14 00:44:35 +00:00
}
// if we reach here numbers are <= 9999, so at most 4 chars long
let mut n = n as isize; // possibly reduce 64bit math
// decode 2 more chars, if > 2 chars
if n >= 100 {
let d1 = (n % 100) << 1;
n /= 100;
curr -= 2;
ptr::copy_nonoverlapping(lut_ptr.offset(d1), buf_ptr.offset(curr), 2);
}
// decode last 1 or 2 chars
if n < 10 {
curr -= 1;
*buf_ptr.offset(curr) = (n as u8) + b'0';
} else {
let d1 = n << 1;
curr -= 2;
ptr::copy_nonoverlapping(lut_ptr.offset(d1), buf_ptr.offset(curr), 2);
}
}
unsafe {
2017-12-16 15:29:15 +00:00
bytes.extend_from_slice(
2017-12-25 21:40:06 +00:00
slice::from_raw_parts(buf_ptr.offset(curr), 41 - curr as usize));
2017-12-14 00:44:35 +00:00
}
}
2017-12-26 03:42:55 +00:00
#[cfg(test)]
mod tests {
use super::*;
2018-01-02 21:39:32 +00:00
#[test]
fn test_write_content_length() {
let mut bytes = BytesMut::new();
write_content_length(0, &mut bytes);
assert_eq!(bytes.take().freeze(), b"\r\ncontent-length: 0\r\n"[..]);
write_content_length(9, &mut bytes);
assert_eq!(bytes.take().freeze(), b"\r\ncontent-length: 9\r\n"[..]);
write_content_length(10, &mut bytes);
assert_eq!(bytes.take().freeze(), b"\r\ncontent-length: 10\r\n"[..]);
write_content_length(99, &mut bytes);
assert_eq!(bytes.take().freeze(), b"\r\ncontent-length: 99\r\n"[..]);
write_content_length(100, &mut bytes);
assert_eq!(bytes.take().freeze(), b"\r\ncontent-length: 100\r\n"[..]);
write_content_length(101, &mut bytes);
assert_eq!(bytes.take().freeze(), b"\r\ncontent-length: 101\r\n"[..]);
write_content_length(998, &mut bytes);
assert_eq!(bytes.take().freeze(), b"\r\ncontent-length: 998\r\n"[..]);
write_content_length(1000, &mut bytes);
assert_eq!(bytes.take().freeze(), b"\r\ncontent-length: 1000\r\n"[..]);
write_content_length(1001, &mut bytes);
assert_eq!(bytes.take().freeze(), b"\r\ncontent-length: 1001\r\n"[..]);
write_content_length(5909, &mut bytes);
assert_eq!(bytes.take().freeze(), b"\r\ncontent-length: 5909\r\n"[..]);
}
2017-12-14 00:44:35 +00:00
}