mirror of
https://github.com/fafhrd91/actix-web
synced 2025-07-07 03:10:17 +02:00
Compare commits
105 Commits
web-v2.0.0
...
http-v2.0.
Author | SHA1 | Date | |
---|---|---|---|
1fa02b5f1c | |||
c9fdcc596d | |||
6cc83dbb67 | |||
3b675c9125 | |||
15a2587887 | |||
0173f99726 | |||
f27dd19093 | |||
7ba14fd113 | |||
903ae47baa | |||
95c18dbdf3 | |||
d3ccf46e92 | |||
cd1765035c | |||
ea28219d0f | |||
77058ef779 | |||
e5f2feec45 | |||
0a86907dd2 | |||
78749a4b7e | |||
de815dd99c | |||
e6078bf792 | |||
a84b37199a | |||
c05f9475c5 | |||
69dab0063c | |||
ec5c779732 | |||
2e2ea7ab80 | |||
eeebc653fd | |||
835a00599c | |||
d9c415e540 | |||
09a391a3ca | |||
62aba424e2 | |||
9d04b250f9 | |||
a4148de226 | |||
48ef4d7a26 | |||
71c4bd1b30 | |||
de1d6ad5cb | |||
2a72e8d119 | |||
2a8e5fdc73 | |||
b213c07799 | |||
3d6b8686ad | |||
a4f87a53da | |||
08f172a0aa | |||
7792eaa16e | |||
845ce3cf34 | |||
7daef22e24 | |||
1249262c35 | |||
94da08f506 | |||
d143c44130 | |||
8ec8ccf4fb | |||
c8ccc69b93 | |||
f9f9fb4c84 | |||
1b77963aac | |||
036ffd43f9 | |||
bdccccd536 | |||
060c392c67 | |||
245f96868a | |||
b3f1071aaf | |||
e6811e8818 | |||
809930d36e | |||
f266b44cb0 | |||
31a3515e90 | |||
82b2786d6b | |||
6ab7cfa2be | |||
9b3f7248a8 | |||
a1835d6510 | |||
4484b3f66e | |||
cde3ae5f61 | |||
7d40b66300 | |||
63730c1f73 | |||
53ff3ad099 | |||
6406f56ca2 | |||
728b944360 | |||
3851a377df | |||
fe13789345 | |||
3033f187d2 | |||
276a5a3ee4 | |||
664f9a8b2d | |||
c73c2dc12c | |||
e634e64847 | |||
cdba30d45f | |||
74dcc7366d | |||
d137a8635b | |||
a2d4ff157e | |||
71d11644a7 | |||
8888520d83 | |||
cf3577550c | |||
58844874a0 | |||
78f24dda03 | |||
e17b3accb9 | |||
c6fa007e72 | |||
a3287948d1 | |||
2e9ab0625e | |||
3a5b62b550 | |||
412e54ce10 | |||
bca41f8d40 | |||
7c974ee668 | |||
abb462ef85 | |||
e66312b664 | |||
39f4b2b39e | |||
f6ff056b8a | |||
51ab4fb73d | |||
f5fd6bc49f | |||
2803fcbe22 | |||
67793c5d92 | |||
bcb5086c91 | |||
7bd2270290 | |||
a4ad5e6b69 |
37
.github/ISSUE_TEMPLATE/bug_report.md
vendored
Normal file
37
.github/ISSUE_TEMPLATE/bug_report.md
vendored
Normal file
@ -0,0 +1,37 @@
|
|||||||
|
---
|
||||||
|
name: bug report
|
||||||
|
about: create a bug report
|
||||||
|
---
|
||||||
|
|
||||||
|
Your issue may already be reported!
|
||||||
|
Please search on the [Actix Web issue tracker](https://github.com/actix/actix-web/issues) before creating one.
|
||||||
|
|
||||||
|
## Expected Behavior
|
||||||
|
<!--- If you're describing a bug, tell us what should happen -->
|
||||||
|
<!--- If you're suggesting a change/improvement, tell us how it should work -->
|
||||||
|
|
||||||
|
## Current Behavior
|
||||||
|
<!--- If describing a bug, tell us what happens instead of the expected behavior -->
|
||||||
|
<!--- If suggesting a change/improvement, explain the difference from current behavior -->
|
||||||
|
|
||||||
|
## Possible Solution
|
||||||
|
<!--- Not obligatory, but suggest a fix/reason for the bug, -->
|
||||||
|
<!--- or ideas how to implement the addition or change -->
|
||||||
|
|
||||||
|
## Steps to Reproduce (for bugs)
|
||||||
|
<!--- Provide a link to a live example, or an unambiguous set of steps to -->
|
||||||
|
<!--- reproduce this bug. Include code to reproduce, if relevant -->
|
||||||
|
1.
|
||||||
|
2.
|
||||||
|
3.
|
||||||
|
4.
|
||||||
|
|
||||||
|
## Context
|
||||||
|
<!--- How has this issue affected you? What are you trying to accomplish? -->
|
||||||
|
<!--- Providing context helps us come up with a solution that is most useful in the real world -->
|
||||||
|
|
||||||
|
## Your Environment
|
||||||
|
<!--- Include as many relevant details about the environment you experienced the bug in -->
|
||||||
|
|
||||||
|
* Rust Version (I.e, output of `rustc -V`):
|
||||||
|
* Actix Web Version:
|
47
.github/workflows/bench.yml
vendored
Normal file
47
.github/workflows/bench.yml
vendored
Normal file
@ -0,0 +1,47 @@
|
|||||||
|
name: Benchmark (Linux)
|
||||||
|
|
||||||
|
on: [push, pull_request]
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
check_benchmark:
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
|
||||||
|
steps:
|
||||||
|
- uses: actions/checkout@master
|
||||||
|
|
||||||
|
- name: Install Rust
|
||||||
|
uses: actions-rs/toolchain@v1
|
||||||
|
with:
|
||||||
|
toolchain: nightly
|
||||||
|
profile: minimal
|
||||||
|
override: true
|
||||||
|
|
||||||
|
- name: Generate Cargo.lock
|
||||||
|
uses: actions-rs/cargo@v1
|
||||||
|
with:
|
||||||
|
command: generate-lockfile
|
||||||
|
- name: Cache cargo registry
|
||||||
|
uses: actions/cache@v1
|
||||||
|
with:
|
||||||
|
path: ~/.cargo/registry
|
||||||
|
key: ${{ matrix.version }}-x86_64-unknown-linux-gnu-registry-trimmed-${{ hashFiles('**/Cargo.lock') }}
|
||||||
|
- name: Cache cargo index
|
||||||
|
uses: actions/cache@v1
|
||||||
|
with:
|
||||||
|
path: ~/.cargo/git
|
||||||
|
key: ${{ matrix.version }}-x86_64-unknown-linux-gnu-cargo-index-trimmed-${{ hashFiles('**/Cargo.lock') }}
|
||||||
|
- name: Cache cargo build
|
||||||
|
uses: actions/cache@v1
|
||||||
|
with:
|
||||||
|
path: target
|
||||||
|
key: ${{ matrix.version }}-x86_64-unknown-linux-gnu-cargo-build-trimmed-${{ hashFiles('**/Cargo.lock') }}
|
||||||
|
|
||||||
|
- name: Check benchmark
|
||||||
|
uses: actions-rs/cargo@v1
|
||||||
|
with:
|
||||||
|
command: bench
|
||||||
|
|
||||||
|
- name: Clear the cargo caches
|
||||||
|
run: |
|
||||||
|
cargo install cargo-cache --no-default-features --features ci-autoclean
|
||||||
|
cargo-cache
|
90
.github/workflows/linux.yml
vendored
Normal file
90
.github/workflows/linux.yml
vendored
Normal file
@ -0,0 +1,90 @@
|
|||||||
|
name: CI (Linux)
|
||||||
|
|
||||||
|
on: [push, pull_request]
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
build_and_test:
|
||||||
|
strategy:
|
||||||
|
fail-fast: false
|
||||||
|
matrix:
|
||||||
|
version:
|
||||||
|
- 1.39.0 # MSRV
|
||||||
|
- stable
|
||||||
|
- nightly
|
||||||
|
|
||||||
|
name: ${{ matrix.version }} - x86_64-unknown-linux-gnu
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
|
||||||
|
steps:
|
||||||
|
- uses: actions/checkout@master
|
||||||
|
|
||||||
|
- name: Install ${{ matrix.version }}
|
||||||
|
uses: actions-rs/toolchain@v1
|
||||||
|
with:
|
||||||
|
toolchain: ${{ matrix.version }}-x86_64-unknown-linux-gnu
|
||||||
|
profile: minimal
|
||||||
|
override: true
|
||||||
|
|
||||||
|
- name: Generate Cargo.lock
|
||||||
|
uses: actions-rs/cargo@v1
|
||||||
|
with:
|
||||||
|
command: generate-lockfile
|
||||||
|
- name: Cache cargo registry
|
||||||
|
uses: actions/cache@v1
|
||||||
|
with:
|
||||||
|
path: ~/.cargo/registry
|
||||||
|
key: ${{ matrix.version }}-x86_64-unknown-linux-gnu-cargo-registry-trimmed-${{ hashFiles('**/Cargo.lock') }}
|
||||||
|
- name: Cache cargo index
|
||||||
|
uses: actions/cache@v1
|
||||||
|
with:
|
||||||
|
path: ~/.cargo/git
|
||||||
|
key: ${{ matrix.version }}-x86_64-unknown-linux-gnu-cargo-index-trimmed-${{ hashFiles('**/Cargo.lock') }}
|
||||||
|
- name: Cache cargo build
|
||||||
|
uses: actions/cache@v1
|
||||||
|
with:
|
||||||
|
path: target
|
||||||
|
key: ${{ matrix.version }}-x86_64-unknown-linux-gnu-cargo-build-trimmed-${{ hashFiles('**/Cargo.lock') }}
|
||||||
|
|
||||||
|
- name: check build
|
||||||
|
uses: actions-rs/cargo@v1
|
||||||
|
with:
|
||||||
|
command: check
|
||||||
|
args: --all --bins --examples --tests
|
||||||
|
|
||||||
|
- name: tests
|
||||||
|
uses: actions-rs/cargo@v1
|
||||||
|
timeout-minutes: 40
|
||||||
|
with:
|
||||||
|
command: test
|
||||||
|
args: --all --all-features --no-fail-fast -- --nocapture
|
||||||
|
|
||||||
|
- name: tests (actix-http)
|
||||||
|
uses: actions-rs/cargo@v1
|
||||||
|
timeout-minutes: 40
|
||||||
|
with:
|
||||||
|
command: test
|
||||||
|
args: --package=actix-http --no-default-features --features=rustls -- --nocapture
|
||||||
|
|
||||||
|
- name: tests (awc)
|
||||||
|
uses: actions-rs/cargo@v1
|
||||||
|
timeout-minutes: 40
|
||||||
|
with:
|
||||||
|
command: test
|
||||||
|
args: --package=awc --no-default-features --features=rustls -- --nocapture
|
||||||
|
|
||||||
|
- name: Generate coverage file
|
||||||
|
if: matrix.version == 'stable' && github.ref == 'refs/heads/master'
|
||||||
|
run: |
|
||||||
|
cargo install cargo-tarpaulin
|
||||||
|
cargo tarpaulin --out Xml
|
||||||
|
- name: Upload to Codecov
|
||||||
|
if: matrix.version == 'stable' && github.ref == 'refs/heads/master'
|
||||||
|
uses: codecov/codecov-action@v1
|
||||||
|
with:
|
||||||
|
token: ${{ secrets.CODECOV_TOKEN }}
|
||||||
|
file: cobertura.xml
|
||||||
|
|
||||||
|
- name: Clear the cargo caches
|
||||||
|
run: |
|
||||||
|
cargo install cargo-cache --no-default-features --features ci-autoclean
|
||||||
|
cargo-cache
|
64
.github/workflows/macos.yml
vendored
Normal file
64
.github/workflows/macos.yml
vendored
Normal file
@ -0,0 +1,64 @@
|
|||||||
|
name: CI (macOS)
|
||||||
|
|
||||||
|
on: [push, pull_request]
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
build_and_test:
|
||||||
|
strategy:
|
||||||
|
fail-fast: false
|
||||||
|
matrix:
|
||||||
|
version:
|
||||||
|
- stable
|
||||||
|
- nightly
|
||||||
|
|
||||||
|
name: ${{ matrix.version }} - x86_64-apple-darwin
|
||||||
|
runs-on: macOS-latest
|
||||||
|
|
||||||
|
steps:
|
||||||
|
- uses: actions/checkout@master
|
||||||
|
|
||||||
|
- name: Install ${{ matrix.version }}
|
||||||
|
uses: actions-rs/toolchain@v1
|
||||||
|
with:
|
||||||
|
toolchain: ${{ matrix.version }}-x86_64-apple-darwin
|
||||||
|
profile: minimal
|
||||||
|
override: true
|
||||||
|
|
||||||
|
- name: Generate Cargo.lock
|
||||||
|
uses: actions-rs/cargo@v1
|
||||||
|
with:
|
||||||
|
command: generate-lockfile
|
||||||
|
- name: Cache cargo registry
|
||||||
|
uses: actions/cache@v1
|
||||||
|
with:
|
||||||
|
path: ~/.cargo/registry
|
||||||
|
key: ${{ matrix.version }}-x86_64-apple-darwin-cargo-registry-trimmed-${{ hashFiles('**/Cargo.lock') }}
|
||||||
|
- name: Cache cargo index
|
||||||
|
uses: actions/cache@v1
|
||||||
|
with:
|
||||||
|
path: ~/.cargo/git
|
||||||
|
key: ${{ matrix.version }}-x86_64-apple-darwin-cargo-index-trimmed-${{ hashFiles('**/Cargo.lock') }}
|
||||||
|
- name: Cache cargo build
|
||||||
|
uses: actions/cache@v1
|
||||||
|
with:
|
||||||
|
path: target
|
||||||
|
key: ${{ matrix.version }}-x86_64-apple-darwin-cargo-build-trimmed-${{ hashFiles('**/Cargo.lock') }}
|
||||||
|
|
||||||
|
- name: check build
|
||||||
|
uses: actions-rs/cargo@v1
|
||||||
|
with:
|
||||||
|
command: check
|
||||||
|
args: --all --bins --examples --tests
|
||||||
|
|
||||||
|
- name: tests
|
||||||
|
uses: actions-rs/cargo@v1
|
||||||
|
with:
|
||||||
|
command: test
|
||||||
|
args: --all --all-features --no-fail-fast -- --nocapture
|
||||||
|
--skip=test_h2_content_length
|
||||||
|
--skip=test_reading_deflate_encoding_large_random_rustls
|
||||||
|
|
||||||
|
- name: Clear the cargo caches
|
||||||
|
run: |
|
||||||
|
cargo install cargo-cache --no-default-features --features ci-autoclean
|
||||||
|
cargo-cache
|
67
.github/workflows/main.yml
vendored
67
.github/workflows/main.yml
vendored
@ -1,67 +0,0 @@
|
|||||||
name: CI
|
|
||||||
|
|
||||||
on: [push, pull_request]
|
|
||||||
|
|
||||||
env:
|
|
||||||
VCPKGRS_DYNAMIC: 1
|
|
||||||
|
|
||||||
jobs:
|
|
||||||
build_and_test:
|
|
||||||
strategy:
|
|
||||||
fail-fast: false
|
|
||||||
matrix:
|
|
||||||
toolchain:
|
|
||||||
- x86_64-pc-windows-msvc
|
|
||||||
# - i686-pc-windows-msvc
|
|
||||||
- x86_64-apple-darwin
|
|
||||||
version:
|
|
||||||
- stable
|
|
||||||
- nightly
|
|
||||||
include:
|
|
||||||
- toolchain: x86_64-pc-windows-msvc
|
|
||||||
os: windows-latest
|
|
||||||
arch: x64
|
|
||||||
# - toolchain: i686-pc-windows-msvc
|
|
||||||
# os: windows-latest
|
|
||||||
# arch: x86
|
|
||||||
- toolchain: x86_64-apple-darwin
|
|
||||||
os: macOS-latest
|
|
||||||
|
|
||||||
name: ${{ matrix.version }} - ${{ matrix.toolchain }}
|
|
||||||
runs-on: ${{ matrix.os }}
|
|
||||||
|
|
||||||
steps:
|
|
||||||
- uses: actions/checkout@master
|
|
||||||
|
|
||||||
- name: Install ${{ matrix.version }}
|
|
||||||
uses: actions-rs/toolchain@v1
|
|
||||||
with:
|
|
||||||
toolchain: ${{ matrix.version }}-${{ matrix.toolchain }}
|
|
||||||
default: true
|
|
||||||
|
|
||||||
- name: Install OpenSSL
|
|
||||||
if: matrix.os == 'windows-latest'
|
|
||||||
run: |
|
|
||||||
vcpkg integrate install
|
|
||||||
vcpkg install openssl:${{ matrix.arch }}-windows
|
|
||||||
|
|
||||||
- name: check nightly
|
|
||||||
if: matrix.version == 'nightly'
|
|
||||||
uses: actions-rs/cargo@v1
|
|
||||||
with:
|
|
||||||
command: check
|
|
||||||
args: --all --benches --bins --examples --tests
|
|
||||||
|
|
||||||
- name: check stable
|
|
||||||
if: matrix.version == 'stable'
|
|
||||||
uses: actions-rs/cargo@v1
|
|
||||||
with:
|
|
||||||
command: check
|
|
||||||
args: --all --bins --examples --tests
|
|
||||||
|
|
||||||
- name: tests
|
|
||||||
if: matrix.toolchain != 'x86_64-pc-windows-gnu'
|
|
||||||
uses: actions-rs/cargo@v1
|
|
||||||
with:
|
|
||||||
command: test
|
|
||||||
args: --all --all-features -- --nocapture
|
|
35
.github/workflows/upload-doc.yml
vendored
Normal file
35
.github/workflows/upload-doc.yml
vendored
Normal file
@ -0,0 +1,35 @@
|
|||||||
|
name: Upload documentation
|
||||||
|
|
||||||
|
on:
|
||||||
|
push:
|
||||||
|
branches:
|
||||||
|
- master
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
build:
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
if: github.repository == 'actix/actix-web'
|
||||||
|
|
||||||
|
steps:
|
||||||
|
- uses: actions/checkout@master
|
||||||
|
|
||||||
|
- name: Install Rust
|
||||||
|
uses: actions-rs/toolchain@v1
|
||||||
|
with:
|
||||||
|
toolchain: stable-x86_64-unknown-linux-gnu
|
||||||
|
profile: minimal
|
||||||
|
override: true
|
||||||
|
|
||||||
|
- name: check build
|
||||||
|
uses: actions-rs/cargo@v1
|
||||||
|
with:
|
||||||
|
command: doc
|
||||||
|
args: --no-deps --all-features
|
||||||
|
|
||||||
|
- name: Tweak HTML
|
||||||
|
run: echo "<meta http-equiv=refresh content=0;url=os_balloon/index.html>" > target/doc/index.html
|
||||||
|
|
||||||
|
- name: Upload documentation
|
||||||
|
run: |
|
||||||
|
git clone https://github.com/davisp/ghp-import.git
|
||||||
|
./ghp-import/ghp_import.py -n -p -f -m "Documentation upload" -r https://${{ secrets.GITHUB_TOKEN }}@github.com/"${{ github.repository }}.git" target/doc
|
59
.github/workflows/windows.yml
vendored
Normal file
59
.github/workflows/windows.yml
vendored
Normal file
@ -0,0 +1,59 @@
|
|||||||
|
name: CI (Windows)
|
||||||
|
|
||||||
|
on: [push, pull_request]
|
||||||
|
|
||||||
|
env:
|
||||||
|
VCPKGRS_DYNAMIC: 1
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
build_and_test:
|
||||||
|
strategy:
|
||||||
|
fail-fast: false
|
||||||
|
matrix:
|
||||||
|
version:
|
||||||
|
- stable
|
||||||
|
- nightly
|
||||||
|
|
||||||
|
name: ${{ matrix.version }} - x86_64-pc-windows-msvc
|
||||||
|
runs-on: windows-latest
|
||||||
|
|
||||||
|
steps:
|
||||||
|
- uses: actions/checkout@master
|
||||||
|
|
||||||
|
- name: Install ${{ matrix.version }}
|
||||||
|
uses: actions-rs/toolchain@v1
|
||||||
|
with:
|
||||||
|
toolchain: ${{ matrix.version }}-x86_64-pc-windows-msvc
|
||||||
|
profile: minimal
|
||||||
|
override: true
|
||||||
|
|
||||||
|
- name: Install OpenSSL
|
||||||
|
run: |
|
||||||
|
vcpkg integrate install
|
||||||
|
vcpkg install openssl:x64-windows
|
||||||
|
Copy-Item C:\vcpkg\installed\x64-windows\bin\libcrypto-1_1-x64.dll C:\vcpkg\installed\x64-windows\bin\libcrypto.dll
|
||||||
|
Copy-Item C:\vcpkg\installed\x64-windows\bin\libssl-1_1-x64.dll C:\vcpkg\installed\x64-windows\bin\libssl.dll
|
||||||
|
Get-ChildItem C:\vcpkg\installed\x64-windows\bin
|
||||||
|
Get-ChildItem C:\vcpkg\installed\x64-windows\lib
|
||||||
|
|
||||||
|
- name: check build
|
||||||
|
uses: actions-rs/cargo@v1
|
||||||
|
with:
|
||||||
|
command: check
|
||||||
|
args: --all --bins --examples --tests
|
||||||
|
|
||||||
|
- name: tests
|
||||||
|
uses: actions-rs/cargo@v1
|
||||||
|
with:
|
||||||
|
command: test
|
||||||
|
args: --all --all-features --no-fail-fast -- --nocapture
|
||||||
|
--skip=test_h2_content_length
|
||||||
|
--skip=test_reading_deflate_encoding_large_random_rustls
|
||||||
|
--skip=test_params
|
||||||
|
--skip=test_simple
|
||||||
|
--skip=test_expect_continue
|
||||||
|
--skip=test_http10_keepalive
|
||||||
|
--skip=test_slow_request
|
||||||
|
--skip=test_connection_force_close
|
||||||
|
--skip=test_connection_server_close
|
||||||
|
--skip=test_connection_wait_queue_force_close
|
61
.travis.yml
61
.travis.yml
@ -1,61 +0,0 @@
|
|||||||
language: rust
|
|
||||||
sudo: required
|
|
||||||
dist: trusty
|
|
||||||
|
|
||||||
cache:
|
|
||||||
# cargo: true
|
|
||||||
apt: true
|
|
||||||
|
|
||||||
matrix:
|
|
||||||
include:
|
|
||||||
- rust: stable
|
|
||||||
- rust: beta
|
|
||||||
- rust: nightly-2019-11-20
|
|
||||||
allow_failures:
|
|
||||||
- rust: nightly-2019-11-20
|
|
||||||
|
|
||||||
env:
|
|
||||||
global:
|
|
||||||
# - RUSTFLAGS="-C link-dead-code"
|
|
||||||
- OPENSSL_VERSION=openssl-1.0.2
|
|
||||||
|
|
||||||
before_install:
|
|
||||||
- sudo add-apt-repository -y ppa:0k53d-karl-f830m/openssl
|
|
||||||
- sudo apt-get update -qq
|
|
||||||
- sudo apt-get install -y openssl libssl-dev libelf-dev libdw-dev cmake gcc binutils-dev libiberty-dev
|
|
||||||
|
|
||||||
before_cache: |
|
|
||||||
if [[ "$TRAVIS_RUST_VERSION" == "nightly-2019-11-20" ]]; then
|
|
||||||
RUSTFLAGS="--cfg procmacro2_semver_exempt" cargo install --version 0.6.11 cargo-tarpaulin
|
|
||||||
fi
|
|
||||||
|
|
||||||
# Add clippy
|
|
||||||
before_script:
|
|
||||||
- export PATH=$PATH:~/.cargo/bin
|
|
||||||
|
|
||||||
script:
|
|
||||||
- cargo update
|
|
||||||
- cargo check --all --no-default-features
|
|
||||||
- |
|
|
||||||
if [[ "$TRAVIS_RUST_VERSION" == "stable" || "$TRAVIS_RUST_VERSION" == "beta" ]]; then
|
|
||||||
cargo test --all-features --all -- --nocapture
|
|
||||||
cd actix-http; cargo test --no-default-features --features="rustls" -- --nocapture; cd ..
|
|
||||||
cd awc; cargo test --no-default-features --features="rustls" -- --nocapture; cd ..
|
|
||||||
fi
|
|
||||||
|
|
||||||
# Upload docs
|
|
||||||
after_success:
|
|
||||||
- |
|
|
||||||
if [[ "$TRAVIS_OS_NAME" == "linux" && "$TRAVIS_PULL_REQUEST" = "false" && "$TRAVIS_BRANCH" == "master" && "$TRAVIS_RUST_VERSION" == "stable" ]]; then
|
|
||||||
cargo doc --no-deps --all-features &&
|
|
||||||
echo "<meta http-equiv=refresh content=0;url=os_balloon/index.html>" > target/doc/index.html &&
|
|
||||||
git clone https://github.com/davisp/ghp-import.git &&
|
|
||||||
./ghp-import/ghp_import.py -n -p -f -m "Documentation upload" -r https://"$GH_TOKEN"@github.com/"$TRAVIS_REPO_SLUG.git" target/doc &&
|
|
||||||
echo "Uploaded documentation"
|
|
||||||
fi
|
|
||||||
- |
|
|
||||||
if [[ "$TRAVIS_RUST_VERSION" == "nightly-2019-11-20" ]]; then
|
|
||||||
taskset -c 0 cargo tarpaulin --out Xml --all --all-features
|
|
||||||
bash <(curl -s https://codecov.io/bash)
|
|
||||||
echo "Uploaded code coverage"
|
|
||||||
fi
|
|
15
CHANGES.md
15
CHANGES.md
@ -1,5 +1,20 @@
|
|||||||
# Changes
|
# Changes
|
||||||
|
|
||||||
|
|
||||||
|
## [2.0.NEXT] - 2020-01-xx
|
||||||
|
|
||||||
|
### Added
|
||||||
|
|
||||||
|
* Add helper function for creating routes with `TRACE` method guard `web::trace()`
|
||||||
|
|
||||||
|
### Changed
|
||||||
|
|
||||||
|
* Use `sha-1` crate instead of unmaintained `sha1` crate
|
||||||
|
|
||||||
|
* Skip empty chunks when returning response from a `Stream` #1308
|
||||||
|
|
||||||
|
* Update the `time` dependency to 0.2.7
|
||||||
|
|
||||||
## [2.0.0] - 2019-12-25
|
## [2.0.0] - 2019-12-25
|
||||||
|
|
||||||
### Changed
|
### Changed
|
||||||
|
27
Cargo.toml
27
Cargo.toml
@ -33,8 +33,8 @@ members = [
|
|||||||
"actix-cors",
|
"actix-cors",
|
||||||
"actix-files",
|
"actix-files",
|
||||||
"actix-framed",
|
"actix-framed",
|
||||||
"actix-session",
|
# "actix-session",
|
||||||
"actix-identity",
|
# "actix-identity",
|
||||||
"actix-multipart",
|
"actix-multipart",
|
||||||
"actix-web-actors",
|
"actix-web-actors",
|
||||||
"actix-web-codegen",
|
"actix-web-codegen",
|
||||||
@ -60,9 +60,9 @@ rustls = ["actix-tls/rustls", "awc/rustls", "rust-tls"]
|
|||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
actix-codec = "0.2.0"
|
actix-codec = "0.2.0"
|
||||||
actix-service = "1.0.1"
|
actix-service = "1.0.2"
|
||||||
actix-utils = "1.0.4"
|
actix-utils = "1.0.6"
|
||||||
actix-router = "0.2.1"
|
actix-router = "0.2.4"
|
||||||
actix-rt = "1.0.0"
|
actix-rt = "1.0.0"
|
||||||
actix-server = "1.0.0"
|
actix-server = "1.0.0"
|
||||||
actix-testing = "1.0.0"
|
actix-testing = "1.0.0"
|
||||||
@ -71,7 +71,7 @@ actix-threadpool = "0.3.1"
|
|||||||
actix-tls = "1.0.0"
|
actix-tls = "1.0.0"
|
||||||
|
|
||||||
actix-web-codegen = "0.2.0"
|
actix-web-codegen = "0.2.0"
|
||||||
actix-http = "1.0.1"
|
actix-http = "2.0.0-alpha.1"
|
||||||
awc = { version = "1.0.1", default-features = false }
|
awc = { version = "1.0.1", default-features = false }
|
||||||
|
|
||||||
bytes = "0.5.3"
|
bytes = "0.5.3"
|
||||||
@ -87,18 +87,19 @@ regex = "1.3"
|
|||||||
serde = { version = "1.0", features=["derive"] }
|
serde = { version = "1.0", features=["derive"] }
|
||||||
serde_json = "1.0"
|
serde_json = "1.0"
|
||||||
serde_urlencoded = "0.6.1"
|
serde_urlencoded = "0.6.1"
|
||||||
time = "0.1.42"
|
time = { version = "0.2.7", default-features = false, features = ["std"] }
|
||||||
url = "2.1"
|
url = "2.1"
|
||||||
open-ssl = { version="0.10", package = "openssl", optional = true }
|
open-ssl = { version="0.10", package = "openssl", optional = true }
|
||||||
rust-tls = { version = "0.16.0", package = "rustls", optional = true }
|
rust-tls = { version = "0.16.0", package = "rustls", optional = true }
|
||||||
|
|
||||||
[dev-dependencies]
|
[dev-dependencies]
|
||||||
actix = "0.9.0"
|
actix = "0.10.0-alpha.1"
|
||||||
rand = "0.7"
|
rand = "0.7"
|
||||||
env_logger = "0.6"
|
env_logger = "0.6"
|
||||||
serde_derive = "1.0"
|
serde_derive = "1.0"
|
||||||
brotli2 = "0.3.2"
|
brotli2 = "0.3.2"
|
||||||
flate2 = "1.0.13"
|
flate2 = "1.0.13"
|
||||||
|
criterion = "0.3"
|
||||||
|
|
||||||
[profile.release]
|
[profile.release]
|
||||||
lto = true
|
lto = true
|
||||||
@ -115,4 +116,12 @@ actix-identity = { path = "actix-identity" }
|
|||||||
actix-session = { path = "actix-session" }
|
actix-session = { path = "actix-session" }
|
||||||
actix-files = { path = "actix-files" }
|
actix-files = { path = "actix-files" }
|
||||||
actix-multipart = { path = "actix-multipart" }
|
actix-multipart = { path = "actix-multipart" }
|
||||||
awc = { path = "awc" }
|
awc = { path = "awc" }
|
||||||
|
|
||||||
|
[[bench]]
|
||||||
|
name = "server"
|
||||||
|
harness = false
|
||||||
|
|
||||||
|
[[bench]]
|
||||||
|
name = "service"
|
||||||
|
harness = false
|
||||||
|
36
MIGRATION.md
36
MIGRATION.md
@ -1,3 +1,11 @@
|
|||||||
|
## Unreleased
|
||||||
|
|
||||||
|
* Setting a cookie's SameSite property, explicitly, to `SameSite::None` will now
|
||||||
|
result in `SameSite=None` being sent with the response Set-Cookie header.
|
||||||
|
To create a cookie without a SameSite attribute, remove any calls setting same_site.
|
||||||
|
* actix-http support for Actors messages was moved to actix-http crate and is enabled
|
||||||
|
with feature `actors`
|
||||||
|
|
||||||
## 2.0.0
|
## 2.0.0
|
||||||
|
|
||||||
* `HttpServer::start()` renamed to `HttpServer::run()`. It also possible to
|
* `HttpServer::start()` renamed to `HttpServer::run()`. It also possible to
|
||||||
@ -17,6 +25,34 @@
|
|||||||
* `ResponseError` trait has been reafctored. `ResponseError::error_response()` renders
|
* `ResponseError` trait has been reafctored. `ResponseError::error_response()` renders
|
||||||
http response.
|
http response.
|
||||||
|
|
||||||
|
* Feature `rust-tls` renamed to `rustls`
|
||||||
|
|
||||||
|
instead of
|
||||||
|
|
||||||
|
```rust
|
||||||
|
actix-web = { version = "2.0.0", features = ["rust-tls"] }
|
||||||
|
```
|
||||||
|
|
||||||
|
use
|
||||||
|
|
||||||
|
```rust
|
||||||
|
actix-web = { version = "2.0.0", features = ["rustls"] }
|
||||||
|
```
|
||||||
|
|
||||||
|
* Feature `ssl` renamed to `openssl`
|
||||||
|
|
||||||
|
instead of
|
||||||
|
|
||||||
|
```rust
|
||||||
|
actix-web = { version = "2.0.0", features = ["ssl"] }
|
||||||
|
```
|
||||||
|
|
||||||
|
use
|
||||||
|
|
||||||
|
```rust
|
||||||
|
actix-web = { version = "2.0.0", features = ["openssl"] }
|
||||||
|
```
|
||||||
|
* `Cors` builder now requires that you call `.finish()` to construct the middleware
|
||||||
|
|
||||||
## 1.0.1
|
## 1.0.1
|
||||||
|
|
||||||
|
15
README.md
15
README.md
@ -41,6 +41,16 @@ Actix web is a simple, pragmatic and extremely fast web framework for Rust.
|
|||||||
|
|
||||||
## Example
|
## Example
|
||||||
|
|
||||||
|
Dependencies:
|
||||||
|
|
||||||
|
```toml
|
||||||
|
[dependencies]
|
||||||
|
actix-web = "2"
|
||||||
|
actix-rt = "1"
|
||||||
|
```
|
||||||
|
|
||||||
|
Code:
|
||||||
|
|
||||||
```rust
|
```rust
|
||||||
use actix_web::{get, web, App, HttpServer, Responder};
|
use actix_web::{get, web, App, HttpServer, Responder};
|
||||||
|
|
||||||
@ -65,10 +75,11 @@ async fn main() -> std::io::Result<()> {
|
|||||||
* [Multipart streams](https://github.com/actix/examples/tree/master/multipart/)
|
* [Multipart streams](https://github.com/actix/examples/tree/master/multipart/)
|
||||||
* [Simple websocket](https://github.com/actix/examples/tree/master/websocket/)
|
* [Simple websocket](https://github.com/actix/examples/tree/master/websocket/)
|
||||||
* [Tera](https://github.com/actix/examples/tree/master/template_tera/) /
|
* [Tera](https://github.com/actix/examples/tree/master/template_tera/) /
|
||||||
[Askama](https://github.com/actix/examples/tree/master/template_askama/) templates
|
* [Askama](https://github.com/actix/examples/tree/master/template_askama/) templates
|
||||||
* [Diesel integration](https://github.com/actix/examples/tree/master/diesel/)
|
* [Diesel integration](https://github.com/actix/examples/tree/master/diesel/)
|
||||||
* [r2d2](https://github.com/actix/examples/tree/master/r2d2/)
|
* [r2d2](https://github.com/actix/examples/tree/master/r2d2/)
|
||||||
* [SSL / HTTP/2.0](https://github.com/actix/examples/tree/master/tls/)
|
* [OpenSSL](https://github.com/actix/examples/tree/master/openssl/)
|
||||||
|
* [Rustls](https://github.com/actix/examples/tree/master/rustls/)
|
||||||
* [Tcp/Websocket chat](https://github.com/actix/examples/tree/master/websocket-chat/)
|
* [Tcp/Websocket chat](https://github.com/actix/examples/tree/master/websocket-chat/)
|
||||||
* [Json](https://github.com/actix/examples/tree/master/json/)
|
* [Json](https://github.com/actix/examples/tree/master/json/)
|
||||||
|
|
||||||
|
@ -19,7 +19,7 @@ path = "src/lib.rs"
|
|||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
actix-web = { version = "2.0.0-rc", default-features = false }
|
actix-web = { version = "2.0.0-rc", default-features = false }
|
||||||
actix-http = "1.0.1"
|
actix-http = "2.0.0-alpha.1"
|
||||||
actix-service = "1.0.1"
|
actix-service = "1.0.1"
|
||||||
bitflags = "1"
|
bitflags = "1"
|
||||||
bytes = "0.5.3"
|
bytes = "0.5.3"
|
||||||
|
@ -5,6 +5,7 @@ use derive_more::Display;
|
|||||||
#[derive(Display, Debug, PartialEq)]
|
#[derive(Display, Debug, PartialEq)]
|
||||||
pub enum FilesError {
|
pub enum FilesError {
|
||||||
/// Path is not a directory
|
/// Path is not a directory
|
||||||
|
#[allow(dead_code)]
|
||||||
#[display(fmt = "Path is not a directory. Unable to serve static files")]
|
#[display(fmt = "Path is not a directory. Unable to serve static files")]
|
||||||
IsNotDirectory,
|
IsNotDirectory,
|
||||||
|
|
||||||
|
@ -275,7 +275,7 @@ impl Files {
|
|||||||
///
|
///
|
||||||
/// `File` uses `ThreadPool` for blocking filesystem operations.
|
/// `File` uses `ThreadPool` for blocking filesystem operations.
|
||||||
/// By default pool with 5x threads of available cpus is used.
|
/// By default pool with 5x threads of available cpus is used.
|
||||||
/// Pool size can be changed by setting ACTIX_CPU_POOL environment variable.
|
/// Pool size can be changed by setting ACTIX_THREADPOOL environment variable.
|
||||||
pub fn new<T: Into<PathBuf>>(path: &str, dir: T) -> Files {
|
pub fn new<T: Into<PathBuf>>(path: &str, dir: T) -> Files {
|
||||||
let orig_dir = dir.into();
|
let orig_dir = dir.into();
|
||||||
let dir = match orig_dir.canonicalize() {
|
let dir = match orig_dir.canonicalize() {
|
||||||
|
@ -23,7 +23,7 @@ actix-codec = "0.2.0"
|
|||||||
actix-service = "1.0.1"
|
actix-service = "1.0.1"
|
||||||
actix-router = "0.2.1"
|
actix-router = "0.2.1"
|
||||||
actix-rt = "1.0.0"
|
actix-rt = "1.0.0"
|
||||||
actix-http = "1.0.1"
|
actix-http = "2.0.0-alpha.1"
|
||||||
|
|
||||||
bytes = "0.5.3"
|
bytes = "0.5.3"
|
||||||
futures = "0.3.1"
|
futures = "0.3.1"
|
||||||
|
@ -1,5 +1,21 @@
|
|||||||
# Changes
|
# Changes
|
||||||
|
|
||||||
|
## [2.0.0-alpha.1] - 2020-02-27
|
||||||
|
|
||||||
|
### Changed
|
||||||
|
|
||||||
|
* Update the `time` dependency to 0.2.7.
|
||||||
|
|
||||||
|
* Moved actors messages support from actix crate, enabled with feature `actors`.
|
||||||
|
|
||||||
|
* Breaking change: trait MessageBody requires Unpin and accepting Pin<&mut Self> instead of &mut self in the poll_next().
|
||||||
|
|
||||||
|
* MessageBody is not implemented for &'static [u8] anymore.
|
||||||
|
|
||||||
|
### Fixed
|
||||||
|
|
||||||
|
* Allow `SameSite=None` cookies to be sent in a response.
|
||||||
|
|
||||||
## [1.0.1] - 2019-12-20
|
## [1.0.1] - 2019-12-20
|
||||||
|
|
||||||
### Fixed
|
### Fixed
|
||||||
|
@ -1,6 +1,6 @@
|
|||||||
[package]
|
[package]
|
||||||
name = "actix-http"
|
name = "actix-http"
|
||||||
version = "1.0.1"
|
version = "2.0.0-alpha.1"
|
||||||
authors = ["Nikolay Kim <fafhrd91@gmail.com>"]
|
authors = ["Nikolay Kim <fafhrd91@gmail.com>"]
|
||||||
description = "Actix http primitives"
|
description = "Actix http primitives"
|
||||||
readme = "README.md"
|
readme = "README.md"
|
||||||
@ -15,7 +15,7 @@ license = "MIT/Apache-2.0"
|
|||||||
edition = "2018"
|
edition = "2018"
|
||||||
|
|
||||||
[package.metadata.docs.rs]
|
[package.metadata.docs.rs]
|
||||||
features = ["openssl", "rustls", "failure", "compress", "secure-cookies"]
|
features = ["openssl", "rustls", "failure", "compress", "secure-cookies","actors"]
|
||||||
|
|
||||||
[lib]
|
[lib]
|
||||||
name = "actix_http"
|
name = "actix_http"
|
||||||
@ -39,20 +39,23 @@ failure = ["fail-ure"]
|
|||||||
# support for secure cookies
|
# support for secure cookies
|
||||||
secure-cookies = ["ring"]
|
secure-cookies = ["ring"]
|
||||||
|
|
||||||
|
# support for actix Actor messages
|
||||||
|
actors = ["actix"]
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
actix-service = "1.0.1"
|
actix-service = "1.0.5"
|
||||||
actix-codec = "0.2.0"
|
actix-codec = "0.2.0"
|
||||||
actix-connect = "1.0.1"
|
actix-connect = "1.0.2"
|
||||||
actix-utils = "1.0.3"
|
actix-utils = "1.0.6"
|
||||||
actix-rt = "1.0.0"
|
actix-rt = "1.0.0"
|
||||||
actix-threadpool = "0.3.1"
|
actix-threadpool = "0.3.1"
|
||||||
actix-tls = { version = "1.0.0", optional = true }
|
actix-tls = { version = "1.0.0", optional = true }
|
||||||
|
actix = { version = "0.10.0-alpha.1", optional = true }
|
||||||
|
|
||||||
base64 = "0.11"
|
base64 = "0.11"
|
||||||
bitflags = "1.2"
|
bitflags = "1.2"
|
||||||
bytes = "0.5.3"
|
bytes = "0.5.3"
|
||||||
copyless = "0.1.4"
|
copyless = "0.1.4"
|
||||||
chrono = "0.4.6"
|
|
||||||
derive_more = "0.99.2"
|
derive_more = "0.99.2"
|
||||||
either = "1.5.3"
|
either = "1.5.3"
|
||||||
encoding_rs = "0.8"
|
encoding_rs = "0.8"
|
||||||
@ -74,10 +77,10 @@ rand = "0.7"
|
|||||||
regex = "1.3"
|
regex = "1.3"
|
||||||
serde = "1.0"
|
serde = "1.0"
|
||||||
serde_json = "1.0"
|
serde_json = "1.0"
|
||||||
sha1 = "0.6"
|
sha-1 = "0.8"
|
||||||
slab = "0.4"
|
slab = "0.4"
|
||||||
serde_urlencoded = "0.6.1"
|
serde_urlencoded = "0.6.1"
|
||||||
time = "0.1.42"
|
time = { version = "0.2.7", default-features = false, features = ["std"] }
|
||||||
|
|
||||||
# for secure cookie
|
# for secure cookie
|
||||||
ring = { version = "0.16.9", optional = true }
|
ring = { version = "0.16.9", optional = true }
|
||||||
@ -90,12 +93,17 @@ flate2 = { version = "1.0.13", optional = true }
|
|||||||
fail-ure = { version = "0.1.5", package="failure", optional = true }
|
fail-ure = { version = "0.1.5", package="failure", optional = true }
|
||||||
|
|
||||||
[dev-dependencies]
|
[dev-dependencies]
|
||||||
actix-server = "1.0.0"
|
actix-server = "1.0.1"
|
||||||
actix-connect = { version = "1.0.0", features=["openssl"] }
|
actix-connect = { version = "1.0.2", features=["openssl"] }
|
||||||
actix-http-test = { version = "1.0.0", features=["openssl"] }
|
actix-http-test = { version = "1.0.0", features=["openssl"] }
|
||||||
actix-tls = { version = "1.0.0", features=["openssl"] }
|
actix-tls = { version = "1.0.0", features=["openssl"] }
|
||||||
|
criterion = "0.3"
|
||||||
futures = "0.3.1"
|
futures = "0.3.1"
|
||||||
env_logger = "0.6"
|
env_logger = "0.7"
|
||||||
serde_derive = "1.0"
|
serde_derive = "1.0"
|
||||||
open-ssl = { version="0.10", package = "openssl" }
|
open-ssl = { version="0.10", package = "openssl" }
|
||||||
rust-tls = { version="0.16", package = "rustls" }
|
rust-tls = { version="0.16", package = "rustls" }
|
||||||
|
|
||||||
|
[[bench]]
|
||||||
|
name = "content-length"
|
||||||
|
harness = false
|
||||||
|
267
actix-http/benches/content-length.rs
Normal file
267
actix-http/benches/content-length.rs
Normal file
@ -0,0 +1,267 @@
|
|||||||
|
use criterion::{criterion_group, criterion_main, BenchmarkId, Criterion};
|
||||||
|
|
||||||
|
use bytes::BytesMut;
|
||||||
|
|
||||||
|
// benchmark sending all requests at the same time
|
||||||
|
fn bench_write_content_length(c: &mut Criterion) {
|
||||||
|
let mut group = c.benchmark_group("write_content_length");
|
||||||
|
|
||||||
|
let sizes = [
|
||||||
|
0, 1, 11, 83, 101, 653, 1001, 6323, 10001, 56329, 100001, 123456, 98724245,
|
||||||
|
4294967202,
|
||||||
|
];
|
||||||
|
|
||||||
|
for i in sizes.iter() {
|
||||||
|
group.bench_with_input(BenchmarkId::new("Original (unsafe)", i), i, |b, &i| {
|
||||||
|
b.iter(|| {
|
||||||
|
let mut b = BytesMut::with_capacity(35);
|
||||||
|
_original::write_content_length(i, &mut b)
|
||||||
|
})
|
||||||
|
});
|
||||||
|
|
||||||
|
group.bench_with_input(BenchmarkId::new("New (safe)", i), i, |b, &i| {
|
||||||
|
b.iter(|| {
|
||||||
|
let mut b = BytesMut::with_capacity(35);
|
||||||
|
_new::write_content_length(i, &mut b)
|
||||||
|
})
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
group.finish();
|
||||||
|
}
|
||||||
|
|
||||||
|
criterion_group!(benches, bench_write_content_length);
|
||||||
|
criterion_main!(benches);
|
||||||
|
|
||||||
|
mod _new {
|
||||||
|
use bytes::{BufMut, BytesMut};
|
||||||
|
|
||||||
|
const DIGITS_START: u8 = b'0';
|
||||||
|
|
||||||
|
/// NOTE: bytes object has to contain enough space
|
||||||
|
pub fn write_content_length(n: usize, bytes: &mut BytesMut) {
|
||||||
|
if n == 0 {
|
||||||
|
bytes.put_slice(b"\r\ncontent-length: 0\r\n");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
bytes.put_slice(b"\r\ncontent-length: ");
|
||||||
|
|
||||||
|
if n < 10 {
|
||||||
|
bytes.put_u8(DIGITS_START + (n as u8));
|
||||||
|
} else if n < 100 {
|
||||||
|
let n = n as u8;
|
||||||
|
|
||||||
|
let d10 = n / 10;
|
||||||
|
let d1 = n % 10;
|
||||||
|
|
||||||
|
bytes.put_u8(DIGITS_START + d10);
|
||||||
|
bytes.put_u8(DIGITS_START + d1);
|
||||||
|
} else if n < 1000 {
|
||||||
|
let n = n as u16;
|
||||||
|
|
||||||
|
let d100 = (n / 100) as u8;
|
||||||
|
let d10 = ((n / 10) % 10) as u8;
|
||||||
|
let d1 = (n % 10) as u8;
|
||||||
|
|
||||||
|
bytes.put_u8(DIGITS_START + d100);
|
||||||
|
bytes.put_u8(DIGITS_START + d10);
|
||||||
|
bytes.put_u8(DIGITS_START + d1);
|
||||||
|
} else if n < 10_000 {
|
||||||
|
let n = n as u16;
|
||||||
|
|
||||||
|
let d1000 = (n / 1000) as u8;
|
||||||
|
let d100 = ((n / 100) % 10) as u8;
|
||||||
|
let d10 = ((n / 10) % 10) as u8;
|
||||||
|
let d1 = (n % 10) as u8;
|
||||||
|
|
||||||
|
bytes.put_u8(DIGITS_START + d1000);
|
||||||
|
bytes.put_u8(DIGITS_START + d100);
|
||||||
|
bytes.put_u8(DIGITS_START + d10);
|
||||||
|
bytes.put_u8(DIGITS_START + d1);
|
||||||
|
} else if n < 100_000 {
|
||||||
|
let n = n as u32;
|
||||||
|
|
||||||
|
let d10000 = (n / 10000) as u8;
|
||||||
|
let d1000 = ((n / 1000) % 10) as u8;
|
||||||
|
let d100 = ((n / 100) % 10) as u8;
|
||||||
|
let d10 = ((n / 10) % 10) as u8;
|
||||||
|
let d1 = (n % 10) as u8;
|
||||||
|
|
||||||
|
bytes.put_u8(DIGITS_START + d10000);
|
||||||
|
bytes.put_u8(DIGITS_START + d1000);
|
||||||
|
bytes.put_u8(DIGITS_START + d100);
|
||||||
|
bytes.put_u8(DIGITS_START + d10);
|
||||||
|
bytes.put_u8(DIGITS_START + d1);
|
||||||
|
} else if n < 1_000_000 {
|
||||||
|
let n = n as u32;
|
||||||
|
|
||||||
|
let d100000 = (n / 100000) as u8;
|
||||||
|
let d10000 = ((n / 10000) % 10) as u8;
|
||||||
|
let d1000 = ((n / 1000) % 10) as u8;
|
||||||
|
let d100 = ((n / 100) % 10) as u8;
|
||||||
|
let d10 = ((n / 10) % 10) as u8;
|
||||||
|
let d1 = (n % 10) as u8;
|
||||||
|
|
||||||
|
bytes.put_u8(DIGITS_START + d100000);
|
||||||
|
bytes.put_u8(DIGITS_START + d10000);
|
||||||
|
bytes.put_u8(DIGITS_START + d1000);
|
||||||
|
bytes.put_u8(DIGITS_START + d100);
|
||||||
|
bytes.put_u8(DIGITS_START + d10);
|
||||||
|
bytes.put_u8(DIGITS_START + d1);
|
||||||
|
} else {
|
||||||
|
write_usize(n, bytes);
|
||||||
|
}
|
||||||
|
|
||||||
|
bytes.put_slice(b"\r\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
fn write_usize(n: usize, bytes: &mut BytesMut) {
|
||||||
|
let mut n = n;
|
||||||
|
|
||||||
|
// 20 chars is max length of a usize (2^64)
|
||||||
|
// digits will be added to the buffer from lsd to msd
|
||||||
|
let mut buf = BytesMut::with_capacity(20);
|
||||||
|
|
||||||
|
while n > 9 {
|
||||||
|
// "pop" the least-significant digit
|
||||||
|
let lsd = (n % 10) as u8;
|
||||||
|
|
||||||
|
// remove the lsd from n
|
||||||
|
n = n / 10;
|
||||||
|
|
||||||
|
buf.put_u8(DIGITS_START + lsd);
|
||||||
|
}
|
||||||
|
|
||||||
|
// put msd to result buffer
|
||||||
|
bytes.put_u8(DIGITS_START + (n as u8));
|
||||||
|
|
||||||
|
// put, in reverse (msd to lsd), remaining digits to buffer
|
||||||
|
for i in (0..buf.len()).rev() {
|
||||||
|
bytes.put_u8(buf[i]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
mod _original {
|
||||||
|
use std::{mem, ptr, slice};
|
||||||
|
|
||||||
|
use bytes::{BufMut, BytesMut};
|
||||||
|
|
||||||
|
const DEC_DIGITS_LUT: &[u8] = b"0001020304050607080910111213141516171819\
|
||||||
|
2021222324252627282930313233343536373839\
|
||||||
|
4041424344454647484950515253545556575859\
|
||||||
|
6061626364656667686970717273747576777879\
|
||||||
|
8081828384858687888990919293949596979899";
|
||||||
|
|
||||||
|
/// NOTE: bytes object has to contain enough space
|
||||||
|
pub 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';
|
||||||
|
bytes.put_slice(&buf);
|
||||||
|
} 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().add(d1),
|
||||||
|
buf.as_mut_ptr().offset(18),
|
||||||
|
2,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
bytes.put_slice(&buf);
|
||||||
|
} 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(
|
||||||
|
DEC_DIGITS_LUT.as_ptr().add(d1),
|
||||||
|
buf.as_mut_ptr().offset(19),
|
||||||
|
2,
|
||||||
|
)
|
||||||
|
};
|
||||||
|
|
||||||
|
// decode last 1
|
||||||
|
buf[18] = (n as u8) + b'0';
|
||||||
|
|
||||||
|
bytes.put_slice(&buf);
|
||||||
|
} else {
|
||||||
|
bytes.put_slice(b"\r\ncontent-length: ");
|
||||||
|
convert_usize(n, bytes);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn convert_usize(mut n: usize, bytes: &mut BytesMut) {
|
||||||
|
let mut curr: isize = 39;
|
||||||
|
let mut buf: [u8; 41] = unsafe { mem::MaybeUninit::uninit().assume_init() };
|
||||||
|
buf[39] = b'\r';
|
||||||
|
buf[40] = b'\n';
|
||||||
|
let buf_ptr = buf.as_mut_ptr();
|
||||||
|
let lut_ptr = DEC_DIGITS_LUT.as_ptr();
|
||||||
|
|
||||||
|
// 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;
|
||||||
|
unsafe {
|
||||||
|
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,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 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;
|
||||||
|
unsafe {
|
||||||
|
ptr::copy_nonoverlapping(lut_ptr.offset(d1), buf_ptr.offset(curr), 2);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// decode last 1 or 2 chars
|
||||||
|
if n < 10 {
|
||||||
|
curr -= 1;
|
||||||
|
unsafe {
|
||||||
|
*buf_ptr.offset(curr) = (n as u8) + b'0';
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
let d1 = n << 1;
|
||||||
|
curr -= 2;
|
||||||
|
unsafe {
|
||||||
|
ptr::copy_nonoverlapping(lut_ptr.offset(d1), buf_ptr.offset(curr), 2);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
unsafe {
|
||||||
|
bytes.extend_from_slice(slice::from_raw_parts(
|
||||||
|
buf_ptr.offset(curr),
|
||||||
|
41 - curr as usize,
|
||||||
|
));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
@ -7,7 +7,8 @@ use futures::StreamExt;
|
|||||||
use http::header::HeaderValue;
|
use http::header::HeaderValue;
|
||||||
use log::info;
|
use log::info;
|
||||||
|
|
||||||
fn main() -> io::Result<()> {
|
#[actix_rt::main]
|
||||||
|
async fn main() -> io::Result<()> {
|
||||||
env::set_var("RUST_LOG", "echo=info");
|
env::set_var("RUST_LOG", "echo=info");
|
||||||
env_logger::init();
|
env_logger::init();
|
||||||
|
|
||||||
@ -16,25 +17,21 @@ fn main() -> io::Result<()> {
|
|||||||
HttpService::build()
|
HttpService::build()
|
||||||
.client_timeout(1000)
|
.client_timeout(1000)
|
||||||
.client_disconnect(1000)
|
.client_disconnect(1000)
|
||||||
.finish(|mut req: Request| {
|
.finish(|mut req: Request| async move {
|
||||||
async move {
|
let mut body = BytesMut::new();
|
||||||
let mut body = BytesMut::new();
|
while let Some(item) = req.payload().next().await {
|
||||||
while let Some(item) = req.payload().next().await {
|
body.extend_from_slice(&item?);
|
||||||
body.extend_from_slice(&item?);
|
|
||||||
}
|
|
||||||
|
|
||||||
info!("request body: {:?}", body);
|
|
||||||
Ok::<_, Error>(
|
|
||||||
Response::Ok()
|
|
||||||
.header(
|
|
||||||
"x-head",
|
|
||||||
HeaderValue::from_static("dummy value!"),
|
|
||||||
)
|
|
||||||
.body(body),
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
info!("request body: {:?}", body);
|
||||||
|
Ok::<_, Error>(
|
||||||
|
Response::Ok()
|
||||||
|
.header("x-head", HeaderValue::from_static("dummy value!"))
|
||||||
|
.body(body),
|
||||||
|
)
|
||||||
})
|
})
|
||||||
.tcp()
|
.tcp()
|
||||||
})?
|
})?
|
||||||
.run()
|
.run()
|
||||||
|
.await
|
||||||
}
|
}
|
||||||
|
@ -19,7 +19,8 @@ async fn handle_request(mut req: Request) -> Result<Response, Error> {
|
|||||||
.body(body))
|
.body(body))
|
||||||
}
|
}
|
||||||
|
|
||||||
fn main() -> io::Result<()> {
|
#[actix_rt::main]
|
||||||
|
async fn main() -> io::Result<()> {
|
||||||
env::set_var("RUST_LOG", "echo=info");
|
env::set_var("RUST_LOG", "echo=info");
|
||||||
env_logger::init();
|
env_logger::init();
|
||||||
|
|
||||||
@ -28,4 +29,5 @@ fn main() -> io::Result<()> {
|
|||||||
HttpService::build().finish(handle_request).tcp()
|
HttpService::build().finish(handle_request).tcp()
|
||||||
})?
|
})?
|
||||||
.run()
|
.run()
|
||||||
|
.await
|
||||||
}
|
}
|
||||||
|
@ -6,7 +6,8 @@ use futures::future;
|
|||||||
use http::header::HeaderValue;
|
use http::header::HeaderValue;
|
||||||
use log::info;
|
use log::info;
|
||||||
|
|
||||||
fn main() -> io::Result<()> {
|
#[actix_rt::main]
|
||||||
|
async fn main() -> io::Result<()> {
|
||||||
env::set_var("RUST_LOG", "hello_world=info");
|
env::set_var("RUST_LOG", "hello_world=info");
|
||||||
env_logger::init();
|
env_logger::init();
|
||||||
|
|
||||||
@ -24,4 +25,5 @@ fn main() -> io::Result<()> {
|
|||||||
.tcp()
|
.tcp()
|
||||||
})?
|
})?
|
||||||
.run()
|
.run()
|
||||||
|
.await
|
||||||
}
|
}
|
||||||
|
@ -5,6 +5,7 @@ use std::{fmt, mem};
|
|||||||
|
|
||||||
use bytes::{Bytes, BytesMut};
|
use bytes::{Bytes, BytesMut};
|
||||||
use futures_core::Stream;
|
use futures_core::Stream;
|
||||||
|
use futures_util::ready;
|
||||||
use pin_project::{pin_project, project};
|
use pin_project::{pin_project, project};
|
||||||
|
|
||||||
use crate::error::Error;
|
use crate::error::Error;
|
||||||
@ -35,33 +36,46 @@ impl BodySize {
|
|||||||
pub trait MessageBody {
|
pub trait MessageBody {
|
||||||
fn size(&self) -> BodySize;
|
fn size(&self) -> BodySize;
|
||||||
|
|
||||||
fn poll_next(&mut self, cx: &mut Context<'_>) -> Poll<Option<Result<Bytes, Error>>>;
|
fn poll_next(
|
||||||
|
self: Pin<&mut Self>,
|
||||||
|
cx: &mut Context<'_>,
|
||||||
|
) -> Poll<Option<Result<Bytes, Error>>>;
|
||||||
|
|
||||||
|
downcast_get_type_id!();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
downcast!(MessageBody);
|
||||||
|
|
||||||
impl MessageBody for () {
|
impl MessageBody for () {
|
||||||
fn size(&self) -> BodySize {
|
fn size(&self) -> BodySize {
|
||||||
BodySize::Empty
|
BodySize::Empty
|
||||||
}
|
}
|
||||||
|
|
||||||
fn poll_next(&mut self, _: &mut Context<'_>) -> Poll<Option<Result<Bytes, Error>>> {
|
fn poll_next(
|
||||||
|
self: Pin<&mut Self>,
|
||||||
|
_: &mut Context<'_>,
|
||||||
|
) -> Poll<Option<Result<Bytes, Error>>> {
|
||||||
Poll::Ready(None)
|
Poll::Ready(None)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<T: MessageBody> MessageBody for Box<T> {
|
impl<T: MessageBody + Unpin> MessageBody for Box<T> {
|
||||||
fn size(&self) -> BodySize {
|
fn size(&self) -> BodySize {
|
||||||
self.as_ref().size()
|
self.as_ref().size()
|
||||||
}
|
}
|
||||||
|
|
||||||
fn poll_next(&mut self, cx: &mut Context<'_>) -> Poll<Option<Result<Bytes, Error>>> {
|
fn poll_next(
|
||||||
self.as_mut().poll_next(cx)
|
self: Pin<&mut Self>,
|
||||||
|
cx: &mut Context<'_>,
|
||||||
|
) -> Poll<Option<Result<Bytes, Error>>> {
|
||||||
|
Pin::new(self.get_mut().as_mut()).poll_next(cx)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[pin_project]
|
#[pin_project]
|
||||||
pub enum ResponseBody<B> {
|
pub enum ResponseBody<B> {
|
||||||
Body(B),
|
Body(#[pin] B),
|
||||||
Other(Body),
|
Other(#[pin] Body),
|
||||||
}
|
}
|
||||||
|
|
||||||
impl ResponseBody<Body> {
|
impl ResponseBody<Body> {
|
||||||
@ -97,10 +111,15 @@ impl<B: MessageBody> MessageBody for ResponseBody<B> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn poll_next(&mut self, cx: &mut Context<'_>) -> Poll<Option<Result<Bytes, Error>>> {
|
#[project]
|
||||||
match self {
|
fn poll_next(
|
||||||
ResponseBody::Body(ref mut body) => body.poll_next(cx),
|
self: Pin<&mut Self>,
|
||||||
ResponseBody::Other(ref mut body) => body.poll_next(cx),
|
cx: &mut Context<'_>,
|
||||||
|
) -> Poll<Option<Result<Bytes, Error>>> {
|
||||||
|
#[project]
|
||||||
|
match self.project() {
|
||||||
|
ResponseBody::Body(body) => body.poll_next(cx),
|
||||||
|
ResponseBody::Other(body) => body.poll_next(cx),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -115,12 +134,13 @@ impl<B: MessageBody> Stream for ResponseBody<B> {
|
|||||||
) -> Poll<Option<Self::Item>> {
|
) -> Poll<Option<Self::Item>> {
|
||||||
#[project]
|
#[project]
|
||||||
match self.project() {
|
match self.project() {
|
||||||
ResponseBody::Body(ref mut body) => body.poll_next(cx),
|
ResponseBody::Body(body) => body.poll_next(cx),
|
||||||
ResponseBody::Other(ref mut body) => body.poll_next(cx),
|
ResponseBody::Other(body) => body.poll_next(cx),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[pin_project]
|
||||||
/// Represents various types of http message body.
|
/// Represents various types of http message body.
|
||||||
pub enum Body {
|
pub enum Body {
|
||||||
/// Empty response. `Content-Length` header is not set.
|
/// Empty response. `Content-Length` header is not set.
|
||||||
@ -130,7 +150,7 @@ pub enum Body {
|
|||||||
/// Specific response body.
|
/// Specific response body.
|
||||||
Bytes(Bytes),
|
Bytes(Bytes),
|
||||||
/// Generic message body.
|
/// Generic message body.
|
||||||
Message(Box<dyn MessageBody>),
|
Message(Box<dyn MessageBody + Unpin>),
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Body {
|
impl Body {
|
||||||
@ -140,7 +160,7 @@ impl Body {
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// Create body from generic message body.
|
/// Create body from generic message body.
|
||||||
pub fn from_message<B: MessageBody + 'static>(body: B) -> Body {
|
pub fn from_message<B: MessageBody + Unpin + 'static>(body: B) -> Body {
|
||||||
Body::Message(Box::new(body))
|
Body::Message(Box::new(body))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -155,8 +175,13 @@ impl MessageBody for Body {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn poll_next(&mut self, cx: &mut Context<'_>) -> Poll<Option<Result<Bytes, Error>>> {
|
#[project]
|
||||||
match self {
|
fn poll_next(
|
||||||
|
self: Pin<&mut Self>,
|
||||||
|
cx: &mut Context<'_>,
|
||||||
|
) -> Poll<Option<Result<Bytes, Error>>> {
|
||||||
|
#[project]
|
||||||
|
match self.project() {
|
||||||
Body::None => Poll::Ready(None),
|
Body::None => Poll::Ready(None),
|
||||||
Body::Empty => Poll::Ready(None),
|
Body::Empty => Poll::Ready(None),
|
||||||
Body::Bytes(ref mut bin) => {
|
Body::Bytes(ref mut bin) => {
|
||||||
@ -167,7 +192,7 @@ impl MessageBody for Body {
|
|||||||
Poll::Ready(Some(Ok(mem::replace(bin, Bytes::new()))))
|
Poll::Ready(Some(Ok(mem::replace(bin, Bytes::new()))))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Body::Message(ref mut body) => body.poll_next(cx),
|
Body::Message(ref mut body) => Pin::new(body.as_mut()).poll_next(cx),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -253,7 +278,7 @@ impl From<serde_json::Value> for Body {
|
|||||||
|
|
||||||
impl<S> From<SizedStream<S>> for Body
|
impl<S> From<SizedStream<S>> for Body
|
||||||
where
|
where
|
||||||
S: Stream<Item = Result<Bytes, Error>> + 'static,
|
S: Stream<Item = Result<Bytes, Error>> + Unpin + 'static,
|
||||||
{
|
{
|
||||||
fn from(s: SizedStream<S>) -> Body {
|
fn from(s: SizedStream<S>) -> Body {
|
||||||
Body::from_message(s)
|
Body::from_message(s)
|
||||||
@ -262,7 +287,7 @@ where
|
|||||||
|
|
||||||
impl<S, E> From<BodyStream<S, E>> for Body
|
impl<S, E> From<BodyStream<S, E>> for Body
|
||||||
where
|
where
|
||||||
S: Stream<Item = Result<Bytes, E>> + 'static,
|
S: Stream<Item = Result<Bytes, E>> + Unpin + 'static,
|
||||||
E: Into<Error> + 'static,
|
E: Into<Error> + 'static,
|
||||||
{
|
{
|
||||||
fn from(s: BodyStream<S, E>) -> Body {
|
fn from(s: BodyStream<S, E>) -> Body {
|
||||||
@ -275,11 +300,14 @@ impl MessageBody for Bytes {
|
|||||||
BodySize::Sized(self.len())
|
BodySize::Sized(self.len())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn poll_next(&mut self, _: &mut Context<'_>) -> Poll<Option<Result<Bytes, Error>>> {
|
fn poll_next(
|
||||||
|
self: Pin<&mut Self>,
|
||||||
|
_: &mut Context<'_>,
|
||||||
|
) -> Poll<Option<Result<Bytes, Error>>> {
|
||||||
if self.is_empty() {
|
if self.is_empty() {
|
||||||
Poll::Ready(None)
|
Poll::Ready(None)
|
||||||
} else {
|
} else {
|
||||||
Poll::Ready(Some(Ok(mem::replace(self, Bytes::new()))))
|
Poll::Ready(Some(Ok(mem::replace(self.get_mut(), Bytes::new()))))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -289,11 +317,16 @@ impl MessageBody for BytesMut {
|
|||||||
BodySize::Sized(self.len())
|
BodySize::Sized(self.len())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn poll_next(&mut self, _: &mut Context<'_>) -> Poll<Option<Result<Bytes, Error>>> {
|
fn poll_next(
|
||||||
|
self: Pin<&mut Self>,
|
||||||
|
_: &mut Context<'_>,
|
||||||
|
) -> Poll<Option<Result<Bytes, Error>>> {
|
||||||
if self.is_empty() {
|
if self.is_empty() {
|
||||||
Poll::Ready(None)
|
Poll::Ready(None)
|
||||||
} else {
|
} else {
|
||||||
Poll::Ready(Some(Ok(mem::replace(self, BytesMut::new()).freeze())))
|
Poll::Ready(Some(Ok(
|
||||||
|
mem::replace(self.get_mut(), BytesMut::new()).freeze()
|
||||||
|
)))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -303,41 +336,36 @@ impl MessageBody for &'static str {
|
|||||||
BodySize::Sized(self.len())
|
BodySize::Sized(self.len())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn poll_next(&mut self, _: &mut Context<'_>) -> Poll<Option<Result<Bytes, Error>>> {
|
fn poll_next(
|
||||||
|
self: Pin<&mut Self>,
|
||||||
|
_: &mut Context<'_>,
|
||||||
|
) -> Poll<Option<Result<Bytes, Error>>> {
|
||||||
if self.is_empty() {
|
if self.is_empty() {
|
||||||
Poll::Ready(None)
|
Poll::Ready(None)
|
||||||
} else {
|
} else {
|
||||||
Poll::Ready(Some(Ok(Bytes::from_static(
|
Poll::Ready(Some(Ok(Bytes::from_static(
|
||||||
mem::replace(self, "").as_ref(),
|
mem::replace(self.get_mut(), "").as_ref(),
|
||||||
))))
|
))))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl MessageBody for &'static [u8] {
|
|
||||||
fn size(&self) -> BodySize {
|
|
||||||
BodySize::Sized(self.len())
|
|
||||||
}
|
|
||||||
|
|
||||||
fn poll_next(&mut self, _: &mut Context<'_>) -> Poll<Option<Result<Bytes, Error>>> {
|
|
||||||
if self.is_empty() {
|
|
||||||
Poll::Ready(None)
|
|
||||||
} else {
|
|
||||||
Poll::Ready(Some(Ok(Bytes::from_static(mem::replace(self, b"")))))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl MessageBody for Vec<u8> {
|
impl MessageBody for Vec<u8> {
|
||||||
fn size(&self) -> BodySize {
|
fn size(&self) -> BodySize {
|
||||||
BodySize::Sized(self.len())
|
BodySize::Sized(self.len())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn poll_next(&mut self, _: &mut Context<'_>) -> Poll<Option<Result<Bytes, Error>>> {
|
fn poll_next(
|
||||||
|
self: Pin<&mut Self>,
|
||||||
|
_: &mut Context<'_>,
|
||||||
|
) -> Poll<Option<Result<Bytes, Error>>> {
|
||||||
if self.is_empty() {
|
if self.is_empty() {
|
||||||
Poll::Ready(None)
|
Poll::Ready(None)
|
||||||
} else {
|
} else {
|
||||||
Poll::Ready(Some(Ok(Bytes::from(mem::replace(self, Vec::new())))))
|
Poll::Ready(Some(Ok(Bytes::from(mem::replace(
|
||||||
|
self.get_mut(),
|
||||||
|
Vec::new(),
|
||||||
|
)))))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -347,12 +375,15 @@ impl MessageBody for String {
|
|||||||
BodySize::Sized(self.len())
|
BodySize::Sized(self.len())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn poll_next(&mut self, _: &mut Context<'_>) -> Poll<Option<Result<Bytes, Error>>> {
|
fn poll_next(
|
||||||
|
self: Pin<&mut Self>,
|
||||||
|
_: &mut Context<'_>,
|
||||||
|
) -> Poll<Option<Result<Bytes, Error>>> {
|
||||||
if self.is_empty() {
|
if self.is_empty() {
|
||||||
Poll::Ready(None)
|
Poll::Ready(None)
|
||||||
} else {
|
} else {
|
||||||
Poll::Ready(Some(Ok(Bytes::from(
|
Poll::Ready(Some(Ok(Bytes::from(
|
||||||
mem::replace(self, String::new()).into_bytes(),
|
mem::replace(self.get_mut(), String::new()).into_bytes(),
|
||||||
))))
|
))))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -361,7 +392,7 @@ impl MessageBody for String {
|
|||||||
/// Type represent streaming body.
|
/// Type represent streaming body.
|
||||||
/// Response does not contain `content-length` header and appropriate transfer encoding is used.
|
/// Response does not contain `content-length` header and appropriate transfer encoding is used.
|
||||||
#[pin_project]
|
#[pin_project]
|
||||||
pub struct BodyStream<S, E> {
|
pub struct BodyStream<S: Unpin, E> {
|
||||||
#[pin]
|
#[pin]
|
||||||
stream: S,
|
stream: S,
|
||||||
_t: PhantomData<E>,
|
_t: PhantomData<E>,
|
||||||
@ -369,7 +400,7 @@ pub struct BodyStream<S, E> {
|
|||||||
|
|
||||||
impl<S, E> BodyStream<S, E>
|
impl<S, E> BodyStream<S, E>
|
||||||
where
|
where
|
||||||
S: Stream<Item = Result<Bytes, E>>,
|
S: Stream<Item = Result<Bytes, E>> + Unpin,
|
||||||
E: Into<Error>,
|
E: Into<Error>,
|
||||||
{
|
{
|
||||||
pub fn new(stream: S) -> Self {
|
pub fn new(stream: S) -> Self {
|
||||||
@ -382,26 +413,37 @@ where
|
|||||||
|
|
||||||
impl<S, E> MessageBody for BodyStream<S, E>
|
impl<S, E> MessageBody for BodyStream<S, E>
|
||||||
where
|
where
|
||||||
S: Stream<Item = Result<Bytes, E>>,
|
S: Stream<Item = Result<Bytes, E>> + Unpin,
|
||||||
E: Into<Error>,
|
E: Into<Error>,
|
||||||
{
|
{
|
||||||
fn size(&self) -> BodySize {
|
fn size(&self) -> BodySize {
|
||||||
BodySize::Stream
|
BodySize::Stream
|
||||||
}
|
}
|
||||||
|
|
||||||
fn poll_next(&mut self, cx: &mut Context<'_>) -> Poll<Option<Result<Bytes, Error>>> {
|
/// Attempts to pull out the next value of the underlying [`Stream`].
|
||||||
unsafe { Pin::new_unchecked(self) }
|
///
|
||||||
.project()
|
/// Empty values are skipped to prevent [`BodyStream`]'s transmission being
|
||||||
.stream
|
/// ended on a zero-length chunk, but rather proceed until the underlying
|
||||||
.poll_next(cx)
|
/// [`Stream`] ends.
|
||||||
.map(|res| res.map(|res| res.map_err(std::convert::Into::into)))
|
fn poll_next(
|
||||||
|
self: Pin<&mut Self>,
|
||||||
|
cx: &mut Context<'_>,
|
||||||
|
) -> Poll<Option<Result<Bytes, Error>>> {
|
||||||
|
let mut stream = self.project().stream;
|
||||||
|
loop {
|
||||||
|
let stream = stream.as_mut();
|
||||||
|
return Poll::Ready(match ready!(stream.poll_next(cx)) {
|
||||||
|
Some(Ok(ref bytes)) if bytes.is_empty() => continue,
|
||||||
|
opt => opt.map(|res| res.map_err(Into::into)),
|
||||||
|
});
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Type represent streaming body. This body implementation should be used
|
/// Type represent streaming body. This body implementation should be used
|
||||||
/// if total size of stream is known. Data get sent as is without using transfer encoding.
|
/// if total size of stream is known. Data get sent as is without using transfer encoding.
|
||||||
#[pin_project]
|
#[pin_project]
|
||||||
pub struct SizedStream<S> {
|
pub struct SizedStream<S: Unpin> {
|
||||||
size: u64,
|
size: u64,
|
||||||
#[pin]
|
#[pin]
|
||||||
stream: S,
|
stream: S,
|
||||||
@ -409,7 +451,7 @@ pub struct SizedStream<S> {
|
|||||||
|
|
||||||
impl<S> SizedStream<S>
|
impl<S> SizedStream<S>
|
||||||
where
|
where
|
||||||
S: Stream<Item = Result<Bytes, Error>>,
|
S: Stream<Item = Result<Bytes, Error>> + Unpin,
|
||||||
{
|
{
|
||||||
pub fn new(size: u64, stream: S) -> Self {
|
pub fn new(size: u64, stream: S) -> Self {
|
||||||
SizedStream { size, stream }
|
SizedStream { size, stream }
|
||||||
@ -418,24 +460,38 @@ where
|
|||||||
|
|
||||||
impl<S> MessageBody for SizedStream<S>
|
impl<S> MessageBody for SizedStream<S>
|
||||||
where
|
where
|
||||||
S: Stream<Item = Result<Bytes, Error>>,
|
S: Stream<Item = Result<Bytes, Error>> + Unpin,
|
||||||
{
|
{
|
||||||
fn size(&self) -> BodySize {
|
fn size(&self) -> BodySize {
|
||||||
BodySize::Sized64(self.size)
|
BodySize::Sized64(self.size)
|
||||||
}
|
}
|
||||||
|
|
||||||
fn poll_next(&mut self, cx: &mut Context<'_>) -> Poll<Option<Result<Bytes, Error>>> {
|
/// Attempts to pull out the next value of the underlying [`Stream`].
|
||||||
unsafe { Pin::new_unchecked(self) }
|
///
|
||||||
.project()
|
/// Empty values are skipped to prevent [`SizedStream`]'s transmission being
|
||||||
.stream
|
/// ended on a zero-length chunk, but rather proceed until the underlying
|
||||||
.poll_next(cx)
|
/// [`Stream`] ends.
|
||||||
|
fn poll_next(
|
||||||
|
self: Pin<&mut Self>,
|
||||||
|
cx: &mut Context<'_>,
|
||||||
|
) -> Poll<Option<Result<Bytes, Error>>> {
|
||||||
|
let mut stream: Pin<&mut S> = self.project().stream;
|
||||||
|
loop {
|
||||||
|
let stream = stream.as_mut();
|
||||||
|
return Poll::Ready(match ready!(stream.poll_next(cx)) {
|
||||||
|
Some(Ok(ref bytes)) if bytes.is_empty() => continue,
|
||||||
|
val => val,
|
||||||
|
});
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use super::*;
|
use super::*;
|
||||||
|
use futures::stream;
|
||||||
use futures_util::future::poll_fn;
|
use futures_util::future::poll_fn;
|
||||||
|
use futures_util::pin_mut;
|
||||||
|
|
||||||
impl Body {
|
impl Body {
|
||||||
pub(crate) fn get_ref(&self) -> &[u8] {
|
pub(crate) fn get_ref(&self) -> &[u8] {
|
||||||
@ -463,7 +519,10 @@ mod tests {
|
|||||||
|
|
||||||
assert_eq!("test".size(), BodySize::Sized(4));
|
assert_eq!("test".size(), BodySize::Sized(4));
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
poll_fn(|cx| "test".poll_next(cx)).await.unwrap().ok(),
|
poll_fn(|cx| Pin::new(&mut "test").poll_next(cx))
|
||||||
|
.await
|
||||||
|
.unwrap()
|
||||||
|
.ok(),
|
||||||
Some(Bytes::from("test"))
|
Some(Bytes::from("test"))
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
@ -477,13 +536,12 @@ mod tests {
|
|||||||
BodySize::Sized(4)
|
BodySize::Sized(4)
|
||||||
);
|
);
|
||||||
assert_eq!(Body::from_slice(b"test".as_ref()).get_ref(), b"test");
|
assert_eq!(Body::from_slice(b"test".as_ref()).get_ref(), b"test");
|
||||||
|
let sb = Bytes::from(&b"test"[..]);
|
||||||
|
pin_mut!(sb);
|
||||||
|
|
||||||
assert_eq!((&b"test"[..]).size(), BodySize::Sized(4));
|
assert_eq!(sb.size(), BodySize::Sized(4));
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
poll_fn(|cx| (&b"test"[..]).poll_next(cx))
|
poll_fn(|cx| sb.as_mut().poll_next(cx)).await.unwrap().ok(),
|
||||||
.await
|
|
||||||
.unwrap()
|
|
||||||
.ok(),
|
|
||||||
Some(Bytes::from("test"))
|
Some(Bytes::from("test"))
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
@ -492,10 +550,12 @@ mod tests {
|
|||||||
async fn test_vec() {
|
async fn test_vec() {
|
||||||
assert_eq!(Body::from(Vec::from("test")).size(), BodySize::Sized(4));
|
assert_eq!(Body::from(Vec::from("test")).size(), BodySize::Sized(4));
|
||||||
assert_eq!(Body::from(Vec::from("test")).get_ref(), b"test");
|
assert_eq!(Body::from(Vec::from("test")).get_ref(), b"test");
|
||||||
|
let test_vec = Vec::from("test");
|
||||||
|
pin_mut!(test_vec);
|
||||||
|
|
||||||
assert_eq!(Vec::from("test").size(), BodySize::Sized(4));
|
assert_eq!(test_vec.size(), BodySize::Sized(4));
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
poll_fn(|cx| Vec::from("test").poll_next(cx))
|
poll_fn(|cx| test_vec.as_mut().poll_next(cx))
|
||||||
.await
|
.await
|
||||||
.unwrap()
|
.unwrap()
|
||||||
.ok(),
|
.ok(),
|
||||||
@ -505,41 +565,44 @@ mod tests {
|
|||||||
|
|
||||||
#[actix_rt::test]
|
#[actix_rt::test]
|
||||||
async fn test_bytes() {
|
async fn test_bytes() {
|
||||||
let mut b = Bytes::from("test");
|
let b = Bytes::from("test");
|
||||||
assert_eq!(Body::from(b.clone()).size(), BodySize::Sized(4));
|
assert_eq!(Body::from(b.clone()).size(), BodySize::Sized(4));
|
||||||
assert_eq!(Body::from(b.clone()).get_ref(), b"test");
|
assert_eq!(Body::from(b.clone()).get_ref(), b"test");
|
||||||
|
pin_mut!(b);
|
||||||
|
|
||||||
assert_eq!(b.size(), BodySize::Sized(4));
|
assert_eq!(b.size(), BodySize::Sized(4));
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
poll_fn(|cx| b.poll_next(cx)).await.unwrap().ok(),
|
poll_fn(|cx| b.as_mut().poll_next(cx)).await.unwrap().ok(),
|
||||||
Some(Bytes::from("test"))
|
Some(Bytes::from("test"))
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[actix_rt::test]
|
#[actix_rt::test]
|
||||||
async fn test_bytes_mut() {
|
async fn test_bytes_mut() {
|
||||||
let mut b = BytesMut::from("test");
|
let b = BytesMut::from("test");
|
||||||
assert_eq!(Body::from(b.clone()).size(), BodySize::Sized(4));
|
assert_eq!(Body::from(b.clone()).size(), BodySize::Sized(4));
|
||||||
assert_eq!(Body::from(b.clone()).get_ref(), b"test");
|
assert_eq!(Body::from(b.clone()).get_ref(), b"test");
|
||||||
|
pin_mut!(b);
|
||||||
|
|
||||||
assert_eq!(b.size(), BodySize::Sized(4));
|
assert_eq!(b.size(), BodySize::Sized(4));
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
poll_fn(|cx| b.poll_next(cx)).await.unwrap().ok(),
|
poll_fn(|cx| b.as_mut().poll_next(cx)).await.unwrap().ok(),
|
||||||
Some(Bytes::from("test"))
|
Some(Bytes::from("test"))
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[actix_rt::test]
|
#[actix_rt::test]
|
||||||
async fn test_string() {
|
async fn test_string() {
|
||||||
let mut b = "test".to_owned();
|
let b = "test".to_owned();
|
||||||
assert_eq!(Body::from(b.clone()).size(), BodySize::Sized(4));
|
assert_eq!(Body::from(b.clone()).size(), BodySize::Sized(4));
|
||||||
assert_eq!(Body::from(b.clone()).get_ref(), b"test");
|
assert_eq!(Body::from(b.clone()).get_ref(), b"test");
|
||||||
assert_eq!(Body::from(&b).size(), BodySize::Sized(4));
|
assert_eq!(Body::from(&b).size(), BodySize::Sized(4));
|
||||||
assert_eq!(Body::from(&b).get_ref(), b"test");
|
assert_eq!(Body::from(&b).get_ref(), b"test");
|
||||||
|
pin_mut!(b);
|
||||||
|
|
||||||
assert_eq!(b.size(), BodySize::Sized(4));
|
assert_eq!(b.size(), BodySize::Sized(4));
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
poll_fn(|cx| b.poll_next(cx)).await.unwrap().ok(),
|
poll_fn(|cx| b.as_mut().poll_next(cx)).await.unwrap().ok(),
|
||||||
Some(Bytes::from("test"))
|
Some(Bytes::from("test"))
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
@ -547,14 +610,17 @@ mod tests {
|
|||||||
#[actix_rt::test]
|
#[actix_rt::test]
|
||||||
async fn test_unit() {
|
async fn test_unit() {
|
||||||
assert_eq!(().size(), BodySize::Empty);
|
assert_eq!(().size(), BodySize::Empty);
|
||||||
assert!(poll_fn(|cx| ().poll_next(cx)).await.is_none());
|
assert!(poll_fn(|cx| Pin::new(&mut ()).poll_next(cx))
|
||||||
|
.await
|
||||||
|
.is_none());
|
||||||
}
|
}
|
||||||
|
|
||||||
#[actix_rt::test]
|
#[actix_rt::test]
|
||||||
async fn test_box() {
|
async fn test_box() {
|
||||||
let mut val = Box::new(());
|
let val = Box::new(());
|
||||||
|
pin_mut!(val);
|
||||||
assert_eq!(val.size(), BodySize::Empty);
|
assert_eq!(val.size(), BodySize::Empty);
|
||||||
assert!(poll_fn(|cx| val.poll_next(cx)).await.is_none());
|
assert!(poll_fn(|cx| val.as_mut().poll_next(cx)).await.is_none());
|
||||||
}
|
}
|
||||||
|
|
||||||
#[actix_rt::test]
|
#[actix_rt::test]
|
||||||
@ -589,4 +655,97 @@ mod tests {
|
|||||||
BodySize::Sized(25)
|
BodySize::Sized(25)
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
mod body_stream {
|
||||||
|
use super::*;
|
||||||
|
//use futures::task::noop_waker;
|
||||||
|
//use futures::stream::once;
|
||||||
|
|
||||||
|
#[actix_rt::test]
|
||||||
|
async fn skips_empty_chunks() {
|
||||||
|
let body = BodyStream::new(stream::iter(
|
||||||
|
["1", "", "2"]
|
||||||
|
.iter()
|
||||||
|
.map(|&v| Ok(Bytes::from(v)) as Result<Bytes, ()>),
|
||||||
|
));
|
||||||
|
pin_mut!(body);
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
poll_fn(|cx| body.as_mut().poll_next(cx))
|
||||||
|
.await
|
||||||
|
.unwrap()
|
||||||
|
.ok(),
|
||||||
|
Some(Bytes::from("1")),
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
poll_fn(|cx| body.as_mut().poll_next(cx))
|
||||||
|
.await
|
||||||
|
.unwrap()
|
||||||
|
.ok(),
|
||||||
|
Some(Bytes::from("2")),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Now it does not compile as it should
|
||||||
|
#[actix_rt::test]
|
||||||
|
async fn move_pinned_pointer() {
|
||||||
|
let (sender, receiver) = futures::channel::oneshot::channel();
|
||||||
|
let mut body_stream = Ok(BodyStream::new(once(async {
|
||||||
|
let x = Box::new(0i32);
|
||||||
|
let y = &x;
|
||||||
|
receiver.await.unwrap();
|
||||||
|
let _z = **y;
|
||||||
|
Ok::<_, ()>(Bytes::new())
|
||||||
|
})));
|
||||||
|
|
||||||
|
let waker = noop_waker();
|
||||||
|
let mut context = Context::from_waker(&waker);
|
||||||
|
pin_mut!(body_stream);
|
||||||
|
|
||||||
|
let _ = body_stream.as_mut().unwrap().poll_next(&mut context);
|
||||||
|
sender.send(()).unwrap();
|
||||||
|
let _ = std::mem::replace(&mut body_stream, Err([0; 32])).unwrap().poll_next(&mut context);
|
||||||
|
}*/
|
||||||
|
}
|
||||||
|
|
||||||
|
mod sized_stream {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[actix_rt::test]
|
||||||
|
async fn skips_empty_chunks() {
|
||||||
|
let body = SizedStream::new(
|
||||||
|
2,
|
||||||
|
stream::iter(["1", "", "2"].iter().map(|&v| Ok(Bytes::from(v)))),
|
||||||
|
);
|
||||||
|
pin_mut!(body);
|
||||||
|
assert_eq!(
|
||||||
|
poll_fn(|cx| body.as_mut().poll_next(cx))
|
||||||
|
.await
|
||||||
|
.unwrap()
|
||||||
|
.ok(),
|
||||||
|
Some(Bytes::from("1")),
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
poll_fn(|cx| body.as_mut().poll_next(cx))
|
||||||
|
.await
|
||||||
|
.unwrap()
|
||||||
|
.ok(),
|
||||||
|
Some(Bytes::from("2")),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[actix_rt::test]
|
||||||
|
async fn test_body_casting() {
|
||||||
|
let mut body = String::from("hello cast");
|
||||||
|
let resp_body: &mut dyn MessageBody = &mut body;
|
||||||
|
let body = resp_body.downcast_ref::<String>().unwrap();
|
||||||
|
assert_eq!(body, "hello cast");
|
||||||
|
let body = &mut resp_body.downcast_mut::<String>().unwrap();
|
||||||
|
body.push_str("!");
|
||||||
|
let body = resp_body.downcast_ref::<String>().unwrap();
|
||||||
|
assert_eq!(body, "hello cast!");
|
||||||
|
let not_body = resp_body.downcast_ref::<()>();
|
||||||
|
assert!(not_body.is_none());
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
@ -8,7 +8,7 @@ use bytes::buf::BufMutExt;
|
|||||||
use bytes::{Bytes, BytesMut};
|
use bytes::{Bytes, BytesMut};
|
||||||
use futures_core::Stream;
|
use futures_core::Stream;
|
||||||
use futures_util::future::poll_fn;
|
use futures_util::future::poll_fn;
|
||||||
use futures_util::{SinkExt, StreamExt};
|
use futures_util::{pin_mut, SinkExt, StreamExt};
|
||||||
|
|
||||||
use crate::error::PayloadError;
|
use crate::error::PayloadError;
|
||||||
use crate::h1;
|
use crate::h1;
|
||||||
@ -120,7 +120,7 @@ where
|
|||||||
|
|
||||||
/// send request body to the peer
|
/// send request body to the peer
|
||||||
pub(crate) async fn send_body<I, B>(
|
pub(crate) async fn send_body<I, B>(
|
||||||
mut body: B,
|
body: B,
|
||||||
framed: &mut Framed<I, h1::ClientCodec>,
|
framed: &mut Framed<I, h1::ClientCodec>,
|
||||||
) -> Result<(), SendRequestError>
|
) -> Result<(), SendRequestError>
|
||||||
where
|
where
|
||||||
@ -128,9 +128,10 @@ where
|
|||||||
B: MessageBody,
|
B: MessageBody,
|
||||||
{
|
{
|
||||||
let mut eof = false;
|
let mut eof = false;
|
||||||
|
pin_mut!(body);
|
||||||
while !eof {
|
while !eof {
|
||||||
while !eof && !framed.is_write_buf_full() {
|
while !eof && !framed.is_write_buf_full() {
|
||||||
match poll_fn(|cx| body.poll_next(cx)).await {
|
match poll_fn(|cx| body.as_mut().poll_next(cx)).await {
|
||||||
Some(result) => {
|
Some(result) => {
|
||||||
framed.write(h1::Message::Chunk(Some(result?)))?;
|
framed.write(h1::Message::Chunk(Some(result?)))?;
|
||||||
}
|
}
|
||||||
|
@ -4,6 +4,7 @@ use std::time;
|
|||||||
use actix_codec::{AsyncRead, AsyncWrite};
|
use actix_codec::{AsyncRead, AsyncWrite};
|
||||||
use bytes::Bytes;
|
use bytes::Bytes;
|
||||||
use futures_util::future::poll_fn;
|
use futures_util::future::poll_fn;
|
||||||
|
use futures_util::pin_mut;
|
||||||
use h2::{client::SendRequest, SendStream};
|
use h2::{client::SendRequest, SendStream};
|
||||||
use http::header::{HeaderValue, CONNECTION, CONTENT_LENGTH, TRANSFER_ENCODING};
|
use http::header::{HeaderValue, CONNECTION, CONTENT_LENGTH, TRANSFER_ENCODING};
|
||||||
use http::{request::Request, Method, Version};
|
use http::{request::Request, Method, Version};
|
||||||
@ -123,13 +124,14 @@ where
|
|||||||
}
|
}
|
||||||
|
|
||||||
async fn send_body<B: MessageBody>(
|
async fn send_body<B: MessageBody>(
|
||||||
mut body: B,
|
body: B,
|
||||||
mut send: SendStream<Bytes>,
|
mut send: SendStream<Bytes>,
|
||||||
) -> Result<(), SendRequestError> {
|
) -> Result<(), SendRequestError> {
|
||||||
let mut buf = None;
|
let mut buf = None;
|
||||||
|
pin_mut!(body);
|
||||||
loop {
|
loop {
|
||||||
if buf.is_none() {
|
if buf.is_none() {
|
||||||
match poll_fn(|cx| body.poll_next(cx)).await {
|
match poll_fn(|cx| body.as_mut().poll_next(cx)).await {
|
||||||
Some(Ok(b)) => {
|
Some(Ok(b)) => {
|
||||||
send.reserve_capacity(b.len());
|
send.reserve_capacity(b.len());
|
||||||
buf = Some(b);
|
buf = Some(b);
|
||||||
|
@ -16,6 +16,7 @@ use fxhash::FxHashMap;
|
|||||||
use h2::client::{handshake, Connection, SendRequest};
|
use h2::client::{handshake, Connection, SendRequest};
|
||||||
use http::uri::Authority;
|
use http::uri::Authority;
|
||||||
use indexmap::IndexSet;
|
use indexmap::IndexSet;
|
||||||
|
use pin_project::pin_project;
|
||||||
use slab::Slab;
|
use slab::Slab;
|
||||||
|
|
||||||
use super::connection::{ConnectionType, IoConnection};
|
use super::connection::{ConnectionType, IoConnection};
|
||||||
@ -422,6 +423,7 @@ where
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[pin_project]
|
||||||
struct ConnectorPoolSupport<T, Io>
|
struct ConnectorPoolSupport<T, Io>
|
||||||
where
|
where
|
||||||
Io: AsyncRead + AsyncWrite + Unpin + 'static,
|
Io: AsyncRead + AsyncWrite + Unpin + 'static,
|
||||||
@ -439,7 +441,7 @@ where
|
|||||||
type Output = ();
|
type Output = ();
|
||||||
|
|
||||||
fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
|
fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
|
||||||
let this = unsafe { self.get_unchecked_mut() };
|
let this = self.project();
|
||||||
|
|
||||||
let mut inner = this.inner.as_ref().borrow_mut();
|
let mut inner = this.inner.as_ref().borrow_mut();
|
||||||
inner.waker.register(cx.waker());
|
inner.waker.register(cx.waker());
|
||||||
@ -488,10 +490,12 @@ where
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[pin_project::pin_project(PinnedDrop)]
|
||||||
struct OpenWaitingConnection<F, Io>
|
struct OpenWaitingConnection<F, Io>
|
||||||
where
|
where
|
||||||
Io: AsyncRead + AsyncWrite + Unpin + 'static,
|
Io: AsyncRead + AsyncWrite + Unpin + 'static,
|
||||||
{
|
{
|
||||||
|
#[pin]
|
||||||
fut: F,
|
fut: F,
|
||||||
key: Key,
|
key: Key,
|
||||||
h2: Option<
|
h2: Option<
|
||||||
@ -525,12 +529,13 @@ where
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<F, Io> Drop for OpenWaitingConnection<F, Io>
|
#[pin_project::pinned_drop]
|
||||||
|
impl<F, Io> PinnedDrop for OpenWaitingConnection<F, Io>
|
||||||
where
|
where
|
||||||
Io: AsyncRead + AsyncWrite + Unpin + 'static,
|
Io: AsyncRead + AsyncWrite + Unpin + 'static,
|
||||||
{
|
{
|
||||||
fn drop(&mut self) {
|
fn drop(self: Pin<&mut Self>) {
|
||||||
if let Some(inner) = self.inner.take() {
|
if let Some(inner) = self.project().inner.take() {
|
||||||
let mut inner = inner.as_ref().borrow_mut();
|
let mut inner = inner.as_ref().borrow_mut();
|
||||||
inner.release();
|
inner.release();
|
||||||
inner.check_availibility();
|
inner.check_availibility();
|
||||||
@ -545,8 +550,8 @@ where
|
|||||||
{
|
{
|
||||||
type Output = ();
|
type Output = ();
|
||||||
|
|
||||||
fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
|
fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
|
||||||
let this = unsafe { self.get_unchecked_mut() };
|
let this = self.as_mut().project();
|
||||||
|
|
||||||
if let Some(ref mut h2) = this.h2 {
|
if let Some(ref mut h2) = this.h2 {
|
||||||
return match Pin::new(h2).poll(cx) {
|
return match Pin::new(h2).poll(cx) {
|
||||||
@ -571,7 +576,7 @@ where
|
|||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
match unsafe { Pin::new_unchecked(&mut this.fut) }.poll(cx) {
|
match this.fut.poll(cx) {
|
||||||
Poll::Ready(Err(err)) => {
|
Poll::Ready(Err(err)) => {
|
||||||
let _ = this.inner.take();
|
let _ = this.inner.take();
|
||||||
if let Some(rx) = this.rx.take() {
|
if let Some(rx) = this.rx.take() {
|
||||||
@ -589,8 +594,8 @@ where
|
|||||||
)));
|
)));
|
||||||
Poll::Ready(())
|
Poll::Ready(())
|
||||||
} else {
|
} else {
|
||||||
this.h2 = Some(handshake(io).boxed_local());
|
*this.h2 = Some(handshake(io).boxed_local());
|
||||||
unsafe { Pin::new_unchecked(this) }.poll(cx)
|
self.poll(cx)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Poll::Pending => Poll::Pending,
|
Poll::Pending => Poll::Pending,
|
||||||
|
@ -1,4 +1,4 @@
|
|||||||
use std::cell::UnsafeCell;
|
use std::cell::RefCell;
|
||||||
use std::rc::Rc;
|
use std::rc::Rc;
|
||||||
use std::task::{Context, Poll};
|
use std::task::{Context, Poll};
|
||||||
|
|
||||||
@ -6,37 +6,35 @@ use actix_service::Service;
|
|||||||
|
|
||||||
#[doc(hidden)]
|
#[doc(hidden)]
|
||||||
/// Service that allows to turn non-clone service to a service with `Clone` impl
|
/// Service that allows to turn non-clone service to a service with `Clone` impl
|
||||||
pub(crate) struct CloneableService<T>(Rc<UnsafeCell<T>>);
|
///
|
||||||
|
/// # Panics
|
||||||
|
/// CloneableService might panic with some creative use of thread local storage.
|
||||||
|
/// See https://github.com/actix/actix-web/issues/1295 for example
|
||||||
|
pub(crate) struct CloneableService<T: Service>(Rc<RefCell<T>>);
|
||||||
|
|
||||||
impl<T> CloneableService<T> {
|
impl<T: Service> CloneableService<T> {
|
||||||
pub(crate) fn new(service: T) -> Self
|
pub(crate) fn new(service: T) -> Self {
|
||||||
where
|
Self(Rc::new(RefCell::new(service)))
|
||||||
T: Service,
|
|
||||||
{
|
|
||||||
Self(Rc::new(UnsafeCell::new(service)))
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<T> Clone for CloneableService<T> {
|
impl<T: Service> Clone for CloneableService<T> {
|
||||||
fn clone(&self) -> Self {
|
fn clone(&self) -> Self {
|
||||||
Self(self.0.clone())
|
Self(self.0.clone())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<T> Service for CloneableService<T>
|
impl<T: Service> Service for CloneableService<T> {
|
||||||
where
|
|
||||||
T: Service,
|
|
||||||
{
|
|
||||||
type Request = T::Request;
|
type Request = T::Request;
|
||||||
type Response = T::Response;
|
type Response = T::Response;
|
||||||
type Error = T::Error;
|
type Error = T::Error;
|
||||||
type Future = T::Future;
|
type Future = T::Future;
|
||||||
|
|
||||||
fn poll_ready(&mut self, cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
|
fn poll_ready(&mut self, cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
|
||||||
unsafe { &mut *self.0.as_ref().get() }.poll_ready(cx)
|
self.0.borrow_mut().poll_ready(cx)
|
||||||
}
|
}
|
||||||
|
|
||||||
fn call(&mut self, req: T::Request) -> Self::Future {
|
fn call(&mut self, req: T::Request) -> Self::Future {
|
||||||
unsafe { &mut *self.0.as_ref().get() }.call(req)
|
self.0.borrow_mut().call(req)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -1,4 +1,4 @@
|
|||||||
use std::cell::UnsafeCell;
|
use std::cell::Cell;
|
||||||
use std::fmt::Write;
|
use std::fmt::Write;
|
||||||
use std::rc::Rc;
|
use std::rc::Rc;
|
||||||
use std::time::Duration;
|
use std::time::Duration;
|
||||||
@ -7,7 +7,7 @@ use std::{fmt, net};
|
|||||||
use actix_rt::time::{delay_for, delay_until, Delay, Instant};
|
use actix_rt::time::{delay_for, delay_until, Delay, Instant};
|
||||||
use bytes::BytesMut;
|
use bytes::BytesMut;
|
||||||
use futures_util::{future, FutureExt};
|
use futures_util::{future, FutureExt};
|
||||||
use time;
|
use time::OffsetDateTime;
|
||||||
|
|
||||||
// "Sun, 06 Nov 1994 08:49:37 GMT".len()
|
// "Sun, 06 Nov 1994 08:49:37 GMT".len()
|
||||||
const DATE_VALUE_LENGTH: usize = 29;
|
const DATE_VALUE_LENGTH: usize = 29;
|
||||||
@ -211,7 +211,12 @@ impl Date {
|
|||||||
}
|
}
|
||||||
fn update(&mut self) {
|
fn update(&mut self) {
|
||||||
self.pos = 0;
|
self.pos = 0;
|
||||||
write!(self, "{}", time::at_utc(time::get_time()).rfc822()).unwrap();
|
write!(
|
||||||
|
self,
|
||||||
|
"{}",
|
||||||
|
OffsetDateTime::now().format("%a, %d %b %Y %H:%M:%S GMT")
|
||||||
|
)
|
||||||
|
.unwrap();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -228,24 +233,24 @@ impl fmt::Write for Date {
|
|||||||
struct DateService(Rc<DateServiceInner>);
|
struct DateService(Rc<DateServiceInner>);
|
||||||
|
|
||||||
struct DateServiceInner {
|
struct DateServiceInner {
|
||||||
current: UnsafeCell<Option<(Date, Instant)>>,
|
current: Cell<Option<(Date, Instant)>>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl DateServiceInner {
|
impl DateServiceInner {
|
||||||
fn new() -> Self {
|
fn new() -> Self {
|
||||||
DateServiceInner {
|
DateServiceInner {
|
||||||
current: UnsafeCell::new(None),
|
current: Cell::new(None),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn reset(&self) {
|
fn reset(&self) {
|
||||||
unsafe { (&mut *self.current.get()).take() };
|
self.current.take();
|
||||||
}
|
}
|
||||||
|
|
||||||
fn update(&self) {
|
fn update(&self) {
|
||||||
let now = Instant::now();
|
let now = Instant::now();
|
||||||
let date = Date::new();
|
let date = Date::new();
|
||||||
*(unsafe { &mut *self.current.get() }) = Some((date, now));
|
self.current.set(Some((date, now)));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -255,7 +260,7 @@ impl DateService {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn check_date(&self) {
|
fn check_date(&self) {
|
||||||
if unsafe { (&*self.0.current.get()).is_none() } {
|
if self.0.current.get().is_none() {
|
||||||
self.0.update();
|
self.0.update();
|
||||||
|
|
||||||
// periodic date update
|
// periodic date update
|
||||||
@ -269,12 +274,12 @@ impl DateService {
|
|||||||
|
|
||||||
fn now(&self) -> Instant {
|
fn now(&self) -> Instant {
|
||||||
self.check_date();
|
self.check_date();
|
||||||
unsafe { (&*self.0.current.get()).as_ref().unwrap().1 }
|
self.0.current.get().unwrap().1
|
||||||
}
|
}
|
||||||
|
|
||||||
fn set_date<F: FnMut(&Date)>(&self, mut f: F) {
|
fn set_date<F: FnMut(&Date)>(&self, mut f: F) {
|
||||||
self.check_date();
|
self.check_date();
|
||||||
f(&unsafe { (&*self.0.current.get()).as_ref().unwrap().0 })
|
f(&self.0.current.get().unwrap().0);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -282,6 +287,16 @@ impl DateService {
|
|||||||
mod tests {
|
mod tests {
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
|
// Test modifying the date from within the closure
|
||||||
|
// passed to `set_date`
|
||||||
|
#[test]
|
||||||
|
fn test_evil_date() {
|
||||||
|
let service = DateService::new();
|
||||||
|
// Make sure that `check_date` doesn't try to spawn a task
|
||||||
|
service.0.update();
|
||||||
|
service.set_date(|_| service.0.reset());
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_date_len() {
|
fn test_date_len() {
|
||||||
assert_eq!(DATE_VALUE_LENGTH, "Sun, 06 Nov 1994 08:49:37 GMT".len());
|
assert_eq!(DATE_VALUE_LENGTH, "Sun, 06 Nov 1994 08:49:37 GMT".len());
|
||||||
|
@ -1,7 +1,6 @@
|
|||||||
use std::borrow::Cow;
|
use std::borrow::Cow;
|
||||||
|
|
||||||
use chrono::Duration;
|
use time::{Duration, OffsetDateTime};
|
||||||
use time::Tm;
|
|
||||||
|
|
||||||
use super::{Cookie, SameSite};
|
use super::{Cookie, SameSite};
|
||||||
|
|
||||||
@ -64,13 +63,13 @@ impl CookieBuilder {
|
|||||||
/// use actix_http::cookie::Cookie;
|
/// use actix_http::cookie::Cookie;
|
||||||
///
|
///
|
||||||
/// let c = Cookie::build("foo", "bar")
|
/// let c = Cookie::build("foo", "bar")
|
||||||
/// .expires(time::now())
|
/// .expires(time::OffsetDateTime::now())
|
||||||
/// .finish();
|
/// .finish();
|
||||||
///
|
///
|
||||||
/// assert!(c.expires().is_some());
|
/// assert!(c.expires().is_some());
|
||||||
/// ```
|
/// ```
|
||||||
#[inline]
|
#[inline]
|
||||||
pub fn expires(mut self, when: Tm) -> CookieBuilder {
|
pub fn expires(mut self, when: OffsetDateTime) -> CookieBuilder {
|
||||||
self.cookie.set_expires(when);
|
self.cookie.set_expires(when);
|
||||||
self
|
self
|
||||||
}
|
}
|
||||||
@ -108,7 +107,10 @@ impl CookieBuilder {
|
|||||||
/// ```
|
/// ```
|
||||||
#[inline]
|
#[inline]
|
||||||
pub fn max_age_time(mut self, value: Duration) -> CookieBuilder {
|
pub fn max_age_time(mut self, value: Duration) -> CookieBuilder {
|
||||||
self.cookie.set_max_age(value);
|
// Truncate any nanoseconds from the Duration, as they aren't represented within `Max-Age`
|
||||||
|
// and would cause two otherwise identical `Cookie` instances to not be equivalent to one another.
|
||||||
|
self.cookie
|
||||||
|
.set_max_age(Duration::seconds(value.whole_seconds()));
|
||||||
self
|
self
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -212,7 +214,7 @@ impl CookieBuilder {
|
|||||||
///
|
///
|
||||||
/// ```rust
|
/// ```rust
|
||||||
/// use actix_http::cookie::Cookie;
|
/// use actix_http::cookie::Cookie;
|
||||||
/// use chrono::Duration;
|
/// use time::Duration;
|
||||||
///
|
///
|
||||||
/// let c = Cookie::build("foo", "bar")
|
/// let c = Cookie::build("foo", "bar")
|
||||||
/// .permanent()
|
/// .permanent()
|
||||||
|
@ -10,18 +10,26 @@ use std::fmt;
|
|||||||
/// attribute is "Strict", then the cookie is never sent in cross-site requests.
|
/// attribute is "Strict", then the cookie is never sent in cross-site requests.
|
||||||
/// If the `SameSite` attribute is "Lax", the cookie is only sent in cross-site
|
/// If the `SameSite` attribute is "Lax", the cookie is only sent in cross-site
|
||||||
/// requests with "safe" HTTP methods, i.e, `GET`, `HEAD`, `OPTIONS`, `TRACE`.
|
/// requests with "safe" HTTP methods, i.e, `GET`, `HEAD`, `OPTIONS`, `TRACE`.
|
||||||
/// If the `SameSite` attribute is not present (made explicit via the
|
/// If the `SameSite` attribute is not present then the cookie will be sent as
|
||||||
/// `SameSite::None` variant), then the cookie will be sent as normal.
|
/// normal. In some browsers, this will implicitly handle the cookie as if "Lax"
|
||||||
|
/// and in others, "None". It's best to explicitly set the `SameSite` attribute
|
||||||
|
/// to avoid inconsistent behavior.
|
||||||
|
///
|
||||||
|
/// **Note:** Depending on browser, the `Secure` attribute may be required for
|
||||||
|
/// `SameSite` "None" cookies to be accepted.
|
||||||
///
|
///
|
||||||
/// **Note:** This cookie attribute is an HTTP draft! Its meaning and definition
|
/// **Note:** This cookie attribute is an HTTP draft! Its meaning and definition
|
||||||
/// are subject to change.
|
/// are subject to change.
|
||||||
|
///
|
||||||
|
/// More info about these draft changes can be found in the draft spec:
|
||||||
|
/// - https://tools.ietf.org/html/draft-west-cookie-incrementalism-00
|
||||||
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
|
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
|
||||||
pub enum SameSite {
|
pub enum SameSite {
|
||||||
/// The "Strict" `SameSite` attribute.
|
/// The "Strict" `SameSite` attribute.
|
||||||
Strict,
|
Strict,
|
||||||
/// The "Lax" `SameSite` attribute.
|
/// The "Lax" `SameSite` attribute.
|
||||||
Lax,
|
Lax,
|
||||||
/// No `SameSite` attribute.
|
/// The "None" `SameSite` attribute.
|
||||||
None,
|
None,
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -92,7 +100,7 @@ impl fmt::Display for SameSite {
|
|||||||
match *self {
|
match *self {
|
||||||
SameSite::Strict => write!(f, "Strict"),
|
SameSite::Strict => write!(f, "Strict"),
|
||||||
SameSite::Lax => write!(f, "Lax"),
|
SameSite::Lax => write!(f, "Lax"),
|
||||||
SameSite::None => Ok(()),
|
SameSite::None => write!(f, "None"),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -1,7 +1,7 @@
|
|||||||
use std::collections::HashSet;
|
use std::collections::HashSet;
|
||||||
use std::mem::replace;
|
use std::mem::replace;
|
||||||
|
|
||||||
use chrono::Duration;
|
use time::{Duration, OffsetDateTime};
|
||||||
|
|
||||||
use super::delta::DeltaCookie;
|
use super::delta::DeltaCookie;
|
||||||
use super::Cookie;
|
use super::Cookie;
|
||||||
@ -188,7 +188,7 @@ impl CookieJar {
|
|||||||
///
|
///
|
||||||
/// ```rust
|
/// ```rust
|
||||||
/// use actix_http::cookie::{CookieJar, Cookie};
|
/// use actix_http::cookie::{CookieJar, Cookie};
|
||||||
/// use chrono::Duration;
|
/// use time::Duration;
|
||||||
///
|
///
|
||||||
/// let mut jar = CookieJar::new();
|
/// let mut jar = CookieJar::new();
|
||||||
///
|
///
|
||||||
@ -202,7 +202,7 @@ impl CookieJar {
|
|||||||
/// let delta: Vec<_> = jar.delta().collect();
|
/// let delta: Vec<_> = jar.delta().collect();
|
||||||
/// assert_eq!(delta.len(), 1);
|
/// assert_eq!(delta.len(), 1);
|
||||||
/// assert_eq!(delta[0].name(), "name");
|
/// assert_eq!(delta[0].name(), "name");
|
||||||
/// assert_eq!(delta[0].max_age(), Some(Duration::seconds(0)));
|
/// assert_eq!(delta[0].max_age(), Some(Duration::zero()));
|
||||||
/// ```
|
/// ```
|
||||||
///
|
///
|
||||||
/// Removing a new cookie does not result in a _removal_ cookie:
|
/// Removing a new cookie does not result in a _removal_ cookie:
|
||||||
@ -220,8 +220,8 @@ impl CookieJar {
|
|||||||
pub fn remove(&mut self, mut cookie: Cookie<'static>) {
|
pub fn remove(&mut self, mut cookie: Cookie<'static>) {
|
||||||
if self.original_cookies.contains(cookie.name()) {
|
if self.original_cookies.contains(cookie.name()) {
|
||||||
cookie.set_value("");
|
cookie.set_value("");
|
||||||
cookie.set_max_age(Duration::seconds(0));
|
cookie.set_max_age(Duration::zero());
|
||||||
cookie.set_expires(time::now() - Duration::days(365));
|
cookie.set_expires(OffsetDateTime::now() - Duration::days(365));
|
||||||
self.delta_cookies.replace(DeltaCookie::removed(cookie));
|
self.delta_cookies.replace(DeltaCookie::removed(cookie));
|
||||||
} else {
|
} else {
|
||||||
self.delta_cookies.remove(cookie.name());
|
self.delta_cookies.remove(cookie.name());
|
||||||
@ -239,7 +239,7 @@ impl CookieJar {
|
|||||||
///
|
///
|
||||||
/// ```rust
|
/// ```rust
|
||||||
/// use actix_http::cookie::{CookieJar, Cookie};
|
/// use actix_http::cookie::{CookieJar, Cookie};
|
||||||
/// use chrono::Duration;
|
/// use time::Duration;
|
||||||
///
|
///
|
||||||
/// let mut jar = CookieJar::new();
|
/// let mut jar = CookieJar::new();
|
||||||
///
|
///
|
||||||
@ -533,8 +533,8 @@ mod test {
|
|||||||
#[test]
|
#[test]
|
||||||
#[cfg(feature = "secure-cookies")]
|
#[cfg(feature = "secure-cookies")]
|
||||||
fn delta() {
|
fn delta() {
|
||||||
use chrono::Duration;
|
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
|
use time::Duration;
|
||||||
|
|
||||||
let mut c = CookieJar::new();
|
let mut c = CookieJar::new();
|
||||||
|
|
||||||
@ -556,7 +556,7 @@ mod test {
|
|||||||
assert!(names.get("test2").unwrap().is_none());
|
assert!(names.get("test2").unwrap().is_none());
|
||||||
assert!(names.get("test3").unwrap().is_none());
|
assert!(names.get("test3").unwrap().is_none());
|
||||||
assert!(names.get("test4").unwrap().is_none());
|
assert!(names.get("test4").unwrap().is_none());
|
||||||
assert_eq!(names.get("original").unwrap(), &Some(Duration::seconds(0)));
|
assert_eq!(names.get("original").unwrap(), &Some(Duration::zero()));
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
|
@ -65,9 +65,8 @@ use std::borrow::Cow;
|
|||||||
use std::fmt;
|
use std::fmt;
|
||||||
use std::str::FromStr;
|
use std::str::FromStr;
|
||||||
|
|
||||||
use chrono::Duration;
|
|
||||||
use percent_encoding::{percent_encode, AsciiSet, CONTROLS};
|
use percent_encoding::{percent_encode, AsciiSet, CONTROLS};
|
||||||
use time::Tm;
|
use time::{Duration, OffsetDateTime};
|
||||||
|
|
||||||
pub use self::builder::CookieBuilder;
|
pub use self::builder::CookieBuilder;
|
||||||
pub use self::draft::*;
|
pub use self::draft::*;
|
||||||
@ -172,7 +171,7 @@ pub struct Cookie<'c> {
|
|||||||
/// The cookie's value.
|
/// The cookie's value.
|
||||||
value: CookieStr,
|
value: CookieStr,
|
||||||
/// The cookie's expiration, if any.
|
/// The cookie's expiration, if any.
|
||||||
expires: Option<Tm>,
|
expires: Option<OffsetDateTime>,
|
||||||
/// The cookie's maximum age, if any.
|
/// The cookie's maximum age, if any.
|
||||||
max_age: Option<Duration>,
|
max_age: Option<Duration>,
|
||||||
/// The cookie's domain, if any.
|
/// The cookie's domain, if any.
|
||||||
@ -479,7 +478,7 @@ impl<'c> Cookie<'c> {
|
|||||||
/// assert_eq!(c.max_age(), None);
|
/// assert_eq!(c.max_age(), None);
|
||||||
///
|
///
|
||||||
/// let c = Cookie::parse("name=value; Max-Age=3600").unwrap();
|
/// let c = Cookie::parse("name=value; Max-Age=3600").unwrap();
|
||||||
/// assert_eq!(c.max_age().map(|age| age.num_hours()), Some(1));
|
/// assert_eq!(c.max_age().map(|age| age.whole_hours()), Some(1));
|
||||||
/// ```
|
/// ```
|
||||||
#[inline]
|
#[inline]
|
||||||
pub fn max_age(&self) -> Option<Duration> {
|
pub fn max_age(&self) -> Option<Duration> {
|
||||||
@ -544,10 +543,10 @@ impl<'c> Cookie<'c> {
|
|||||||
/// let expire_time = "Wed, 21 Oct 2017 07:28:00 GMT";
|
/// let expire_time = "Wed, 21 Oct 2017 07:28:00 GMT";
|
||||||
/// let cookie_str = format!("name=value; Expires={}", expire_time);
|
/// let cookie_str = format!("name=value; Expires={}", expire_time);
|
||||||
/// let c = Cookie::parse(cookie_str).unwrap();
|
/// let c = Cookie::parse(cookie_str).unwrap();
|
||||||
/// assert_eq!(c.expires().map(|t| t.tm_year), Some(117));
|
/// assert_eq!(c.expires().map(|t| t.year()), Some(2017));
|
||||||
/// ```
|
/// ```
|
||||||
#[inline]
|
#[inline]
|
||||||
pub fn expires(&self) -> Option<Tm> {
|
pub fn expires(&self) -> Option<OffsetDateTime> {
|
||||||
self.expires
|
self.expires
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -645,7 +644,7 @@ impl<'c> Cookie<'c> {
|
|||||||
///
|
///
|
||||||
/// ```rust
|
/// ```rust
|
||||||
/// use actix_http::cookie::Cookie;
|
/// use actix_http::cookie::Cookie;
|
||||||
/// use chrono::Duration;
|
/// use time::Duration;
|
||||||
///
|
///
|
||||||
/// let mut c = Cookie::new("name", "value");
|
/// let mut c = Cookie::new("name", "value");
|
||||||
/// assert_eq!(c.max_age(), None);
|
/// assert_eq!(c.max_age(), None);
|
||||||
@ -698,18 +697,19 @@ impl<'c> Cookie<'c> {
|
|||||||
///
|
///
|
||||||
/// ```rust
|
/// ```rust
|
||||||
/// use actix_http::cookie::Cookie;
|
/// use actix_http::cookie::Cookie;
|
||||||
|
/// use time::{Duration, OffsetDateTime};
|
||||||
///
|
///
|
||||||
/// let mut c = Cookie::new("name", "value");
|
/// let mut c = Cookie::new("name", "value");
|
||||||
/// assert_eq!(c.expires(), None);
|
/// assert_eq!(c.expires(), None);
|
||||||
///
|
///
|
||||||
/// let mut now = time::now();
|
/// let mut now = OffsetDateTime::now();
|
||||||
/// now.tm_year += 1;
|
/// now += Duration::week();
|
||||||
///
|
///
|
||||||
/// c.set_expires(now);
|
/// c.set_expires(now);
|
||||||
/// assert!(c.expires().is_some())
|
/// assert!(c.expires().is_some())
|
||||||
/// ```
|
/// ```
|
||||||
#[inline]
|
#[inline]
|
||||||
pub fn set_expires(&mut self, time: Tm) {
|
pub fn set_expires(&mut self, time: OffsetDateTime) {
|
||||||
self.expires = Some(time);
|
self.expires = Some(time);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -720,7 +720,7 @@ impl<'c> Cookie<'c> {
|
|||||||
///
|
///
|
||||||
/// ```rust
|
/// ```rust
|
||||||
/// use actix_http::cookie::Cookie;
|
/// use actix_http::cookie::Cookie;
|
||||||
/// use chrono::Duration;
|
/// use time::Duration;
|
||||||
///
|
///
|
||||||
/// let mut c = Cookie::new("foo", "bar");
|
/// let mut c = Cookie::new("foo", "bar");
|
||||||
/// assert!(c.expires().is_none());
|
/// assert!(c.expires().is_none());
|
||||||
@ -733,7 +733,7 @@ impl<'c> Cookie<'c> {
|
|||||||
pub fn make_permanent(&mut self) {
|
pub fn make_permanent(&mut self) {
|
||||||
let twenty_years = Duration::days(365 * 20);
|
let twenty_years = Duration::days(365 * 20);
|
||||||
self.set_max_age(twenty_years);
|
self.set_max_age(twenty_years);
|
||||||
self.set_expires(time::now() + twenty_years);
|
self.set_expires(OffsetDateTime::now() + twenty_years);
|
||||||
}
|
}
|
||||||
|
|
||||||
fn fmt_parameters(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
fn fmt_parameters(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
||||||
@ -746,9 +746,7 @@ impl<'c> Cookie<'c> {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if let Some(same_site) = self.same_site() {
|
if let Some(same_site) = self.same_site() {
|
||||||
if !same_site.is_none() {
|
write!(f, "; SameSite={}", same_site)?;
|
||||||
write!(f, "; SameSite={}", same_site)?;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if let Some(path) = self.path() {
|
if let Some(path) = self.path() {
|
||||||
@ -760,11 +758,11 @@ impl<'c> Cookie<'c> {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if let Some(max_age) = self.max_age() {
|
if let Some(max_age) = self.max_age() {
|
||||||
write!(f, "; Max-Age={}", max_age.num_seconds())?;
|
write!(f, "; Max-Age={}", max_age.whole_seconds())?;
|
||||||
}
|
}
|
||||||
|
|
||||||
if let Some(time) = self.expires() {
|
if let Some(time) = self.expires() {
|
||||||
write!(f, "; Expires={}", time.rfc822())?;
|
write!(f, "; Expires={}", time.format("%a, %d %b %Y %H:%M:%S GMT"))?;
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
@ -992,7 +990,7 @@ impl<'a, 'b> PartialEq<Cookie<'b>> for Cookie<'a> {
|
|||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use super::{Cookie, SameSite};
|
use super::{Cookie, SameSite};
|
||||||
use time::strptime;
|
use time::PrimitiveDateTime;
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn format() {
|
fn format() {
|
||||||
@ -1017,7 +1015,9 @@ mod tests {
|
|||||||
assert_eq!(&cookie.to_string(), "foo=bar; Domain=www.rust-lang.org");
|
assert_eq!(&cookie.to_string(), "foo=bar; Domain=www.rust-lang.org");
|
||||||
|
|
||||||
let time_str = "Wed, 21 Oct 2015 07:28:00 GMT";
|
let time_str = "Wed, 21 Oct 2015 07:28:00 GMT";
|
||||||
let expires = strptime(time_str, "%a, %d %b %Y %H:%M:%S %Z").unwrap();
|
let expires = PrimitiveDateTime::parse(time_str, "%a, %d %b %Y %H:%M:%S")
|
||||||
|
.unwrap()
|
||||||
|
.assume_utc();
|
||||||
let cookie = Cookie::build("foo", "bar").expires(expires).finish();
|
let cookie = Cookie::build("foo", "bar").expires(expires).finish();
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
&cookie.to_string(),
|
&cookie.to_string(),
|
||||||
@ -1037,7 +1037,7 @@ mod tests {
|
|||||||
let cookie = Cookie::build("foo", "bar")
|
let cookie = Cookie::build("foo", "bar")
|
||||||
.same_site(SameSite::None)
|
.same_site(SameSite::None)
|
||||||
.finish();
|
.finish();
|
||||||
assert_eq!(&cookie.to_string(), "foo=bar");
|
assert_eq!(&cookie.to_string(), "foo=bar; SameSite=None");
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
|
@ -5,11 +5,13 @@ use std::error::Error;
|
|||||||
use std::fmt;
|
use std::fmt;
|
||||||
use std::str::Utf8Error;
|
use std::str::Utf8Error;
|
||||||
|
|
||||||
use chrono::Duration;
|
|
||||||
use percent_encoding::percent_decode;
|
use percent_encoding::percent_decode;
|
||||||
|
use time::Duration;
|
||||||
|
|
||||||
use super::{Cookie, CookieStr, SameSite};
|
use super::{Cookie, CookieStr, SameSite};
|
||||||
|
|
||||||
|
use crate::time_parser;
|
||||||
|
|
||||||
/// Enum corresponding to a parsing error.
|
/// Enum corresponding to a parsing error.
|
||||||
#[derive(Debug, PartialEq, Eq, Clone, Copy)]
|
#[derive(Debug, PartialEq, Eq, Clone, Copy)]
|
||||||
pub enum ParseError {
|
pub enum ParseError {
|
||||||
@ -147,7 +149,7 @@ fn parse_inner<'c>(s: &str, decode: bool) -> Result<Cookie<'c>, ParseError> {
|
|||||||
Ok(val) => {
|
Ok(val) => {
|
||||||
// Don't panic if the max age seconds is greater than what's supported by
|
// Don't panic if the max age seconds is greater than what's supported by
|
||||||
// `Duration`.
|
// `Duration`.
|
||||||
let val = cmp::min(val, Duration::max_value().num_seconds());
|
let val = cmp::min(val, Duration::max_value().whole_seconds());
|
||||||
Some(Duration::seconds(val))
|
Some(Duration::seconds(val))
|
||||||
}
|
}
|
||||||
Err(_) => continue,
|
Err(_) => continue,
|
||||||
@ -179,16 +181,14 @@ fn parse_inner<'c>(s: &str, decode: bool) -> Result<Cookie<'c>, ParseError> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
("expires", Some(v)) => {
|
("expires", Some(v)) => {
|
||||||
// Try strptime with three date formats according to
|
// Try parsing with three date formats according to
|
||||||
// http://tools.ietf.org/html/rfc2616#section-3.3.1. Try
|
// http://tools.ietf.org/html/rfc2616#section-3.3.1. Try
|
||||||
// additional ones as encountered in the real world.
|
// additional ones as encountered in the real world.
|
||||||
let tm = time::strptime(v, "%a, %d %b %Y %H:%M:%S %Z")
|
let tm = time_parser::parse_http_date(v)
|
||||||
.or_else(|_| time::strptime(v, "%A, %d-%b-%y %H:%M:%S %Z"))
|
.or_else(|| time::parse(v, "%a, %d-%b-%Y %H:%M:%S").ok());
|
||||||
.or_else(|_| time::strptime(v, "%a, %d-%b-%Y %H:%M:%S %Z"))
|
|
||||||
.or_else(|_| time::strptime(v, "%a %b %d %H:%M:%S %Y"));
|
|
||||||
|
|
||||||
if let Ok(time) = tm {
|
if let Some(time) = tm {
|
||||||
cookie.expires = Some(time)
|
cookie.expires = Some(time.assume_utc())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
_ => {
|
_ => {
|
||||||
@ -216,8 +216,7 @@ where
|
|||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use super::{Cookie, SameSite};
|
use super::{Cookie, SameSite};
|
||||||
use chrono::Duration;
|
use time::{Duration, PrimitiveDateTime};
|
||||||
use time::strptime;
|
|
||||||
|
|
||||||
macro_rules! assert_eq_parse {
|
macro_rules! assert_eq_parse {
|
||||||
($string:expr, $expected:expr) => {
|
($string:expr, $expected:expr) => {
|
||||||
@ -377,7 +376,9 @@ mod tests {
|
|||||||
);
|
);
|
||||||
|
|
||||||
let time_str = "Wed, 21 Oct 2015 07:28:00 GMT";
|
let time_str = "Wed, 21 Oct 2015 07:28:00 GMT";
|
||||||
let expires = strptime(time_str, "%a, %d %b %Y %H:%M:%S %Z").unwrap();
|
let expires = PrimitiveDateTime::parse(time_str, "%a, %d %b %Y %H:%M:%S")
|
||||||
|
.unwrap()
|
||||||
|
.assume_utc();
|
||||||
expected.set_expires(expires);
|
expected.set_expires(expires);
|
||||||
assert_eq_parse!(
|
assert_eq_parse!(
|
||||||
" foo=bar ;HttpOnly; Secure; Max-Age=4; Path=/foo; \
|
" foo=bar ;HttpOnly; Secure; Max-Age=4; Path=/foo; \
|
||||||
@ -386,7 +387,9 @@ mod tests {
|
|||||||
);
|
);
|
||||||
|
|
||||||
unexpected.set_domain("foo.com");
|
unexpected.set_domain("foo.com");
|
||||||
let bad_expires = strptime(time_str, "%a, %d %b %Y %H:%S:%M %Z").unwrap();
|
let bad_expires = PrimitiveDateTime::parse(time_str, "%a, %d %b %Y %H:%S:%M")
|
||||||
|
.unwrap()
|
||||||
|
.assume_utc();
|
||||||
expected.set_expires(bad_expires);
|
expected.set_expires(bad_expires);
|
||||||
assert_ne_parse!(
|
assert_ne_parse!(
|
||||||
" foo=bar ;HttpOnly; Secure; Max-Age=4; Path=/foo; \
|
" foo=bar ;HttpOnly; Secure; Max-Age=4; Path=/foo; \
|
||||||
@ -414,8 +417,16 @@ mod tests {
|
|||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn do_not_panic_on_large_max_ages() {
|
fn do_not_panic_on_large_max_ages() {
|
||||||
let max_seconds = Duration::max_value().num_seconds();
|
let max_duration = Duration::max_value();
|
||||||
let expected = Cookie::build("foo", "bar").max_age(max_seconds).finish();
|
let expected = Cookie::build("foo", "bar")
|
||||||
assert_eq_parse!(format!(" foo=bar; Max-Age={:?}", max_seconds + 1), expected);
|
.max_age_time(max_duration)
|
||||||
|
.finish();
|
||||||
|
let overflow_duration = max_duration
|
||||||
|
.checked_add(Duration::nanoseconds(1))
|
||||||
|
.unwrap_or(max_duration);
|
||||||
|
assert_eq_parse!(
|
||||||
|
format!(" foo=bar; Max-Age={:?}", overflow_duration.whole_seconds()),
|
||||||
|
expected
|
||||||
|
);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -9,6 +9,7 @@ use brotli2::write::BrotliEncoder;
|
|||||||
use bytes::Bytes;
|
use bytes::Bytes;
|
||||||
use flate2::write::{GzEncoder, ZlibEncoder};
|
use flate2::write::{GzEncoder, ZlibEncoder};
|
||||||
use futures_core::ready;
|
use futures_core::ready;
|
||||||
|
use pin_project::{pin_project, project};
|
||||||
|
|
||||||
use crate::body::{Body, BodySize, MessageBody, ResponseBody};
|
use crate::body::{Body, BodySize, MessageBody, ResponseBody};
|
||||||
use crate::http::header::{ContentEncoding, CONTENT_ENCODING};
|
use crate::http::header::{ContentEncoding, CONTENT_ENCODING};
|
||||||
@ -19,8 +20,10 @@ use super::Writer;
|
|||||||
|
|
||||||
const INPLACE: usize = 1024;
|
const INPLACE: usize = 1024;
|
||||||
|
|
||||||
|
#[pin_project]
|
||||||
pub struct Encoder<B> {
|
pub struct Encoder<B> {
|
||||||
eof: bool,
|
eof: bool,
|
||||||
|
#[pin]
|
||||||
body: EncoderBody<B>,
|
body: EncoderBody<B>,
|
||||||
encoder: Option<ContentEncoder>,
|
encoder: Option<ContentEncoder>,
|
||||||
fut: Option<CpuFuture<ContentEncoder, io::Error>>,
|
fut: Option<CpuFuture<ContentEncoder, io::Error>>,
|
||||||
@ -76,67 +79,88 @@ impl<B: MessageBody> Encoder<B> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[pin_project]
|
||||||
enum EncoderBody<B> {
|
enum EncoderBody<B> {
|
||||||
Bytes(Bytes),
|
Bytes(Bytes),
|
||||||
Stream(B),
|
Stream(#[pin] B),
|
||||||
BoxedStream(Box<dyn MessageBody>),
|
BoxedStream(Box<dyn MessageBody + Unpin>),
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<B: MessageBody> MessageBody for EncoderBody<B> {
|
||||||
|
fn size(&self) -> BodySize {
|
||||||
|
match self {
|
||||||
|
EncoderBody::Bytes(ref b) => b.size(),
|
||||||
|
EncoderBody::Stream(ref b) => b.size(),
|
||||||
|
EncoderBody::BoxedStream(ref b) => b.size(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[project]
|
||||||
|
fn poll_next(
|
||||||
|
self: Pin<&mut Self>,
|
||||||
|
cx: &mut Context<'_>,
|
||||||
|
) -> Poll<Option<Result<Bytes, Error>>> {
|
||||||
|
#[project]
|
||||||
|
match self.project() {
|
||||||
|
EncoderBody::Bytes(b) => {
|
||||||
|
if b.is_empty() {
|
||||||
|
Poll::Ready(None)
|
||||||
|
} else {
|
||||||
|
Poll::Ready(Some(Ok(std::mem::replace(b, Bytes::new()))))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
EncoderBody::Stream(b) => b.poll_next(cx),
|
||||||
|
EncoderBody::BoxedStream(ref mut b) => Pin::new(b.as_mut()).poll_next(cx),
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<B: MessageBody> MessageBody for Encoder<B> {
|
impl<B: MessageBody> MessageBody for Encoder<B> {
|
||||||
fn size(&self) -> BodySize {
|
fn size(&self) -> BodySize {
|
||||||
if self.encoder.is_none() {
|
if self.encoder.is_none() {
|
||||||
match self.body {
|
self.body.size()
|
||||||
EncoderBody::Bytes(ref b) => b.size(),
|
|
||||||
EncoderBody::Stream(ref b) => b.size(),
|
|
||||||
EncoderBody::BoxedStream(ref b) => b.size(),
|
|
||||||
}
|
|
||||||
} else {
|
} else {
|
||||||
BodySize::Stream
|
BodySize::Stream
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn poll_next(&mut self, cx: &mut Context<'_>) -> Poll<Option<Result<Bytes, Error>>> {
|
fn poll_next(
|
||||||
|
self: Pin<&mut Self>,
|
||||||
|
cx: &mut Context<'_>,
|
||||||
|
) -> Poll<Option<Result<Bytes, Error>>> {
|
||||||
|
let mut this = self.project();
|
||||||
loop {
|
loop {
|
||||||
if self.eof {
|
if *this.eof {
|
||||||
return Poll::Ready(None);
|
return Poll::Ready(None);
|
||||||
}
|
}
|
||||||
|
|
||||||
if let Some(ref mut fut) = self.fut {
|
if let Some(ref mut fut) = this.fut {
|
||||||
let mut encoder = match ready!(Pin::new(fut).poll(cx)) {
|
let mut encoder = match ready!(Pin::new(fut).poll(cx)) {
|
||||||
Ok(item) => item,
|
Ok(item) => item,
|
||||||
Err(e) => return Poll::Ready(Some(Err(e.into()))),
|
Err(e) => return Poll::Ready(Some(Err(e.into()))),
|
||||||
};
|
};
|
||||||
let chunk = encoder.take();
|
let chunk = encoder.take();
|
||||||
self.encoder = Some(encoder);
|
*this.encoder = Some(encoder);
|
||||||
self.fut.take();
|
this.fut.take();
|
||||||
if !chunk.is_empty() {
|
if !chunk.is_empty() {
|
||||||
return Poll::Ready(Some(Ok(chunk)));
|
return Poll::Ready(Some(Ok(chunk)));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
let result = match self.body {
|
let result = this.body.as_mut().poll_next(cx);
|
||||||
EncoderBody::Bytes(ref mut b) => {
|
|
||||||
if b.is_empty() {
|
|
||||||
Poll::Ready(None)
|
|
||||||
} else {
|
|
||||||
Poll::Ready(Some(Ok(std::mem::replace(b, Bytes::new()))))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
EncoderBody::Stream(ref mut b) => b.poll_next(cx),
|
|
||||||
EncoderBody::BoxedStream(ref mut b) => b.poll_next(cx),
|
|
||||||
};
|
|
||||||
match result {
|
match result {
|
||||||
Poll::Ready(Some(Ok(chunk))) => {
|
Poll::Ready(Some(Ok(chunk))) => {
|
||||||
if let Some(mut encoder) = self.encoder.take() {
|
if let Some(mut encoder) = this.encoder.take() {
|
||||||
if chunk.len() < INPLACE {
|
if chunk.len() < INPLACE {
|
||||||
encoder.write(&chunk)?;
|
encoder.write(&chunk)?;
|
||||||
let chunk = encoder.take();
|
let chunk = encoder.take();
|
||||||
self.encoder = Some(encoder);
|
*this.encoder = Some(encoder);
|
||||||
if !chunk.is_empty() {
|
if !chunk.is_empty() {
|
||||||
return Poll::Ready(Some(Ok(chunk)));
|
return Poll::Ready(Some(Ok(chunk)));
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
self.fut = Some(run(move || {
|
*this.fut = Some(run(move || {
|
||||||
encoder.write(&chunk)?;
|
encoder.write(&chunk)?;
|
||||||
Ok(encoder)
|
Ok(encoder)
|
||||||
}));
|
}));
|
||||||
@ -146,12 +170,12 @@ impl<B: MessageBody> MessageBody for Encoder<B> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
Poll::Ready(None) => {
|
Poll::Ready(None) => {
|
||||||
if let Some(encoder) = self.encoder.take() {
|
if let Some(encoder) = this.encoder.take() {
|
||||||
let chunk = encoder.finish()?;
|
let chunk = encoder.finish()?;
|
||||||
if chunk.is_empty() {
|
if chunk.is_empty() {
|
||||||
return Poll::Ready(None);
|
return Poll::Ready(None);
|
||||||
} else {
|
} else {
|
||||||
self.eof = true;
|
*this.eof = true;
|
||||||
return Poll::Ready(Some(Ok(chunk)));
|
return Poll::Ready(Some(Ok(chunk)));
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
|
@ -1,5 +1,4 @@
|
|||||||
//! Error and Result module
|
//! Error and Result module
|
||||||
use std::any::TypeId;
|
|
||||||
use std::cell::RefCell;
|
use std::cell::RefCell;
|
||||||
use std::io::Write;
|
use std::io::Write;
|
||||||
use std::str::Utf8Error;
|
use std::str::Utf8Error;
|
||||||
@ -15,7 +14,6 @@ use derive_more::{Display, From};
|
|||||||
pub use futures_channel::oneshot::Canceled;
|
pub use futures_channel::oneshot::Canceled;
|
||||||
use http::uri::InvalidUri;
|
use http::uri::InvalidUri;
|
||||||
use http::{header, Error as HttpError, StatusCode};
|
use http::{header, Error as HttpError, StatusCode};
|
||||||
use httparse;
|
|
||||||
use serde::de::value::Error as DeError;
|
use serde::de::value::Error as DeError;
|
||||||
use serde_json::error::Error as JsonError;
|
use serde_json::error::Error as JsonError;
|
||||||
use serde_urlencoded::ser::Error as FormError;
|
use serde_urlencoded::ser::Error as FormError;
|
||||||
@ -83,25 +81,10 @@ pub trait ResponseError: fmt::Debug + fmt::Display {
|
|||||||
resp.set_body(Body::from(buf))
|
resp.set_body(Body::from(buf))
|
||||||
}
|
}
|
||||||
|
|
||||||
#[doc(hidden)]
|
downcast_get_type_id!();
|
||||||
fn __private_get_type_id__(&self) -> TypeId
|
|
||||||
where
|
|
||||||
Self: 'static,
|
|
||||||
{
|
|
||||||
TypeId::of::<Self>()
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
impl dyn ResponseError + 'static {
|
downcast!(ResponseError);
|
||||||
/// Downcasts a response error to a specific type.
|
|
||||||
pub fn downcast_ref<T: ResponseError + 'static>(&self) -> Option<&T> {
|
|
||||||
if self.__private_get_type_id__() == TypeId::of::<T>() {
|
|
||||||
unsafe { Some(&*(self as *const dyn ResponseError as *const T)) }
|
|
||||||
} else {
|
|
||||||
None
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl fmt::Display for Error {
|
impl fmt::Display for Error {
|
||||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
||||||
@ -967,6 +950,16 @@ where
|
|||||||
/// Compatibility for `failure::Error`
|
/// Compatibility for `failure::Error`
|
||||||
impl ResponseError for fail_ure::Error {}
|
impl ResponseError for fail_ure::Error {}
|
||||||
|
|
||||||
|
#[cfg(feature = "actors")]
|
||||||
|
/// `InternalServerError` for `actix::MailboxError`
|
||||||
|
/// This is supported on feature=`actors` only
|
||||||
|
impl ResponseError for actix::MailboxError {}
|
||||||
|
|
||||||
|
#[cfg(feature = "actors")]
|
||||||
|
/// `InternalServerError` for `actix::ResolverError`
|
||||||
|
/// This is supported on feature=`actors` only
|
||||||
|
impl ResponseError for actix::actors::resolver::ResolverError {}
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use super::*;
|
use super::*;
|
||||||
|
@ -28,33 +28,30 @@ impl Extensions {
|
|||||||
|
|
||||||
/// Check if container contains entry
|
/// Check if container contains entry
|
||||||
pub fn contains<T: 'static>(&self) -> bool {
|
pub fn contains<T: 'static>(&self) -> bool {
|
||||||
self.map.get(&TypeId::of::<T>()).is_some()
|
self.map.contains_key(&TypeId::of::<T>())
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get a reference to a type previously inserted on this `Extensions`.
|
/// Get a reference to a type previously inserted on this `Extensions`.
|
||||||
pub fn get<T: 'static>(&self) -> Option<&T> {
|
pub fn get<T: 'static>(&self) -> Option<&T> {
|
||||||
self.map
|
self.map
|
||||||
.get(&TypeId::of::<T>())
|
.get(&TypeId::of::<T>())
|
||||||
.and_then(|boxed| (&**boxed as &(dyn Any + 'static)).downcast_ref())
|
.and_then(|boxed| boxed.downcast_ref())
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get a mutable reference to a type previously inserted on this `Extensions`.
|
/// Get a mutable reference to a type previously inserted on this `Extensions`.
|
||||||
pub fn get_mut<T: 'static>(&mut self) -> Option<&mut T> {
|
pub fn get_mut<T: 'static>(&mut self) -> Option<&mut T> {
|
||||||
self.map
|
self.map
|
||||||
.get_mut(&TypeId::of::<T>())
|
.get_mut(&TypeId::of::<T>())
|
||||||
.and_then(|boxed| (&mut **boxed as &mut (dyn Any + 'static)).downcast_mut())
|
.and_then(|boxed| boxed.downcast_mut())
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Remove a type from this `Extensions`.
|
/// Remove a type from this `Extensions`.
|
||||||
///
|
///
|
||||||
/// If a extension of this type existed, it will be returned.
|
/// If a extension of this type existed, it will be returned.
|
||||||
pub fn remove<T: 'static>(&mut self) -> Option<T> {
|
pub fn remove<T: 'static>(&mut self) -> Option<T> {
|
||||||
self.map.remove(&TypeId::of::<T>()).and_then(|boxed| {
|
self.map
|
||||||
(boxed as Box<dyn Any + 'static>)
|
.remove(&TypeId::of::<T>())
|
||||||
.downcast()
|
.and_then(|boxed| boxed.downcast().ok().map(|boxed| *boxed))
|
||||||
.ok()
|
|
||||||
.map(|boxed| *boxed)
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Clear the `Extensions` of all inserted extensions.
|
/// Clear the `Extensions` of all inserted extensions.
|
||||||
@ -70,6 +67,92 @@ impl fmt::Debug for Extensions {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_remove() {
|
||||||
|
let mut map = Extensions::new();
|
||||||
|
|
||||||
|
map.insert::<i8>(123);
|
||||||
|
assert!(map.get::<i8>().is_some());
|
||||||
|
|
||||||
|
map.remove::<i8>();
|
||||||
|
assert!(map.get::<i8>().is_none());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_clear() {
|
||||||
|
let mut map = Extensions::new();
|
||||||
|
|
||||||
|
map.insert::<i8>(8);
|
||||||
|
map.insert::<i16>(16);
|
||||||
|
map.insert::<i32>(32);
|
||||||
|
|
||||||
|
assert!(map.contains::<i8>());
|
||||||
|
assert!(map.contains::<i16>());
|
||||||
|
assert!(map.contains::<i32>());
|
||||||
|
|
||||||
|
map.clear();
|
||||||
|
|
||||||
|
assert!(!map.contains::<i8>());
|
||||||
|
assert!(!map.contains::<i16>());
|
||||||
|
assert!(!map.contains::<i32>());
|
||||||
|
|
||||||
|
map.insert::<i8>(10);
|
||||||
|
assert_eq!(*map.get::<i8>().unwrap(), 10);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_integers() {
|
||||||
|
let mut map = Extensions::new();
|
||||||
|
|
||||||
|
map.insert::<i8>(8);
|
||||||
|
map.insert::<i16>(16);
|
||||||
|
map.insert::<i32>(32);
|
||||||
|
map.insert::<i64>(64);
|
||||||
|
map.insert::<i128>(128);
|
||||||
|
map.insert::<u8>(8);
|
||||||
|
map.insert::<u16>(16);
|
||||||
|
map.insert::<u32>(32);
|
||||||
|
map.insert::<u64>(64);
|
||||||
|
map.insert::<u128>(128);
|
||||||
|
assert!(map.get::<i8>().is_some());
|
||||||
|
assert!(map.get::<i16>().is_some());
|
||||||
|
assert!(map.get::<i32>().is_some());
|
||||||
|
assert!(map.get::<i64>().is_some());
|
||||||
|
assert!(map.get::<i128>().is_some());
|
||||||
|
assert!(map.get::<u8>().is_some());
|
||||||
|
assert!(map.get::<u16>().is_some());
|
||||||
|
assert!(map.get::<u32>().is_some());
|
||||||
|
assert!(map.get::<u64>().is_some());
|
||||||
|
assert!(map.get::<u128>().is_some());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_composition() {
|
||||||
|
struct Magi<T>(pub T);
|
||||||
|
|
||||||
|
struct Madoka {
|
||||||
|
pub god: bool,
|
||||||
|
}
|
||||||
|
|
||||||
|
struct Homura {
|
||||||
|
pub attempts: usize,
|
||||||
|
}
|
||||||
|
|
||||||
|
struct Mami {
|
||||||
|
pub guns: usize,
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut map = Extensions::new();
|
||||||
|
|
||||||
|
map.insert(Magi(Madoka { god: false }));
|
||||||
|
map.insert(Magi(Homura { attempts: 0 }));
|
||||||
|
map.insert(Magi(Mami { guns: 999 }));
|
||||||
|
|
||||||
|
assert!(!map.get::<Magi<Madoka>>().unwrap().0.god);
|
||||||
|
assert_eq!(0, map.get::<Magi<Homura>>().unwrap().0.attempts);
|
||||||
|
assert_eq!(999, map.get::<Magi<Mami>>().unwrap().0.guns);
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_extensions() {
|
fn test_extensions() {
|
||||||
#[derive(Debug, PartialEq)]
|
#[derive(Debug, PartialEq)]
|
||||||
|
@ -8,7 +8,6 @@ use actix_codec::Decoder;
|
|||||||
use bytes::{Buf, Bytes, BytesMut};
|
use bytes::{Buf, Bytes, BytesMut};
|
||||||
use http::header::{HeaderName, HeaderValue};
|
use http::header::{HeaderName, HeaderValue};
|
||||||
use http::{header, Method, StatusCode, Uri, Version};
|
use http::{header, Method, StatusCode, Uri, Version};
|
||||||
use httparse;
|
|
||||||
use log::{debug, error, trace};
|
use log::{debug, error, trace};
|
||||||
|
|
||||||
use crate::error::ParseError;
|
use crate::error::ParseError;
|
||||||
|
@ -1,3 +1,6 @@
|
|||||||
|
// Because MSRV is 1.39.0.
|
||||||
|
#![allow(clippy::mem_replace_with_default)]
|
||||||
|
|
||||||
use std::collections::VecDeque;
|
use std::collections::VecDeque;
|
||||||
use std::future::Future;
|
use std::future::Future;
|
||||||
use std::pin::Pin;
|
use std::pin::Pin;
|
||||||
@ -10,6 +13,7 @@ use actix_service::Service;
|
|||||||
use bitflags::bitflags;
|
use bitflags::bitflags;
|
||||||
use bytes::{Buf, BytesMut};
|
use bytes::{Buf, BytesMut};
|
||||||
use log::{error, trace};
|
use log::{error, trace};
|
||||||
|
use pin_project::pin_project;
|
||||||
|
|
||||||
use crate::body::{Body, BodySize, MessageBody, ResponseBody};
|
use crate::body::{Body, BodySize, MessageBody, ResponseBody};
|
||||||
use crate::cloneable::CloneableService;
|
use crate::cloneable::CloneableService;
|
||||||
@ -41,6 +45,7 @@ bitflags! {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[pin_project::pin_project]
|
||||||
/// Dispatcher for HTTP/1.1 protocol
|
/// Dispatcher for HTTP/1.1 protocol
|
||||||
pub struct Dispatcher<T, S, B, X, U>
|
pub struct Dispatcher<T, S, B, X, U>
|
||||||
where
|
where
|
||||||
@ -52,9 +57,11 @@ where
|
|||||||
U: Service<Request = (Request, Framed<T, Codec>), Response = ()>,
|
U: Service<Request = (Request, Framed<T, Codec>), Response = ()>,
|
||||||
U::Error: fmt::Display,
|
U::Error: fmt::Display,
|
||||||
{
|
{
|
||||||
|
#[pin]
|
||||||
inner: DispatcherState<T, S, B, X, U>,
|
inner: DispatcherState<T, S, B, X, U>,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[pin_project]
|
||||||
enum DispatcherState<T, S, B, X, U>
|
enum DispatcherState<T, S, B, X, U>
|
||||||
where
|
where
|
||||||
S: Service<Request = Request>,
|
S: Service<Request = Request>,
|
||||||
@ -65,11 +72,11 @@ where
|
|||||||
U: Service<Request = (Request, Framed<T, Codec>), Response = ()>,
|
U: Service<Request = (Request, Framed<T, Codec>), Response = ()>,
|
||||||
U::Error: fmt::Display,
|
U::Error: fmt::Display,
|
||||||
{
|
{
|
||||||
Normal(InnerDispatcher<T, S, B, X, U>),
|
Normal(#[pin] InnerDispatcher<T, S, B, X, U>),
|
||||||
Upgrade(U::Future),
|
Upgrade(#[pin] U::Future),
|
||||||
None,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[pin_project]
|
||||||
struct InnerDispatcher<T, S, B, X, U>
|
struct InnerDispatcher<T, S, B, X, U>
|
||||||
where
|
where
|
||||||
S: Service<Request = Request>,
|
S: Service<Request = Request>,
|
||||||
@ -88,6 +95,7 @@ where
|
|||||||
peer_addr: Option<net::SocketAddr>,
|
peer_addr: Option<net::SocketAddr>,
|
||||||
error: Option<DispatchError>,
|
error: Option<DispatchError>,
|
||||||
|
|
||||||
|
#[pin]
|
||||||
state: State<S, B, X>,
|
state: State<S, B, X>,
|
||||||
payload: Option<PayloadSender>,
|
payload: Option<PayloadSender>,
|
||||||
messages: VecDeque<DispatcherMessage>,
|
messages: VecDeque<DispatcherMessage>,
|
||||||
@ -95,7 +103,7 @@ where
|
|||||||
ka_expire: Instant,
|
ka_expire: Instant,
|
||||||
ka_timer: Option<Delay>,
|
ka_timer: Option<Delay>,
|
||||||
|
|
||||||
io: T,
|
io: Option<T>,
|
||||||
read_buf: BytesMut,
|
read_buf: BytesMut,
|
||||||
write_buf: BytesMut,
|
write_buf: BytesMut,
|
||||||
codec: Codec,
|
codec: Codec,
|
||||||
@ -107,6 +115,7 @@ enum DispatcherMessage {
|
|||||||
Error(Response<()>),
|
Error(Response<()>),
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[pin_project]
|
||||||
enum State<S, B, X>
|
enum State<S, B, X>
|
||||||
where
|
where
|
||||||
S: Service<Request = Request>,
|
S: Service<Request = Request>,
|
||||||
@ -114,9 +123,9 @@ where
|
|||||||
B: MessageBody,
|
B: MessageBody,
|
||||||
{
|
{
|
||||||
None,
|
None,
|
||||||
ExpectCall(X::Future),
|
ExpectCall(#[pin] X::Future),
|
||||||
ServiceCall(S::Future),
|
ServiceCall(#[pin] S::Future),
|
||||||
SendPayload(ResponseBody<B>),
|
SendPayload(#[pin] ResponseBody<B>),
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<S, B, X> State<S, B, X>
|
impl<S, B, X> State<S, B, X>
|
||||||
@ -141,7 +150,6 @@ where
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
enum PollResponse {
|
enum PollResponse {
|
||||||
Upgrade(Request),
|
Upgrade(Request),
|
||||||
DoNothing,
|
DoNothing,
|
||||||
@ -236,7 +244,7 @@ where
|
|||||||
state: State::None,
|
state: State::None,
|
||||||
error: None,
|
error: None,
|
||||||
messages: VecDeque::new(),
|
messages: VecDeque::new(),
|
||||||
io,
|
io: Some(io),
|
||||||
codec,
|
codec,
|
||||||
read_buf,
|
read_buf,
|
||||||
service,
|
service,
|
||||||
@ -278,10 +286,11 @@ where
|
|||||||
}
|
}
|
||||||
|
|
||||||
// if checked is set to true, delay disconnect until all tasks have finished.
|
// if checked is set to true, delay disconnect until all tasks have finished.
|
||||||
fn client_disconnected(&mut self) {
|
fn client_disconnected(self: Pin<&mut Self>) {
|
||||||
self.flags
|
let this = self.project();
|
||||||
|
this.flags
|
||||||
.insert(Flags::READ_DISCONNECT | Flags::WRITE_DISCONNECT);
|
.insert(Flags::READ_DISCONNECT | Flags::WRITE_DISCONNECT);
|
||||||
if let Some(mut payload) = self.payload.take() {
|
if let Some(mut payload) = this.payload.take() {
|
||||||
payload.set_error(PayloadError::Incomplete(None));
|
payload.set_error(PayloadError::Incomplete(None));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -290,17 +299,22 @@ where
|
|||||||
///
|
///
|
||||||
/// true - got whouldblock
|
/// true - got whouldblock
|
||||||
/// false - didnt get whouldblock
|
/// false - didnt get whouldblock
|
||||||
fn poll_flush(&mut self, cx: &mut Context<'_>) -> Result<bool, DispatchError> {
|
#[pin_project::project]
|
||||||
|
fn poll_flush(
|
||||||
|
self: Pin<&mut Self>,
|
||||||
|
cx: &mut Context<'_>,
|
||||||
|
) -> Result<bool, DispatchError> {
|
||||||
if self.write_buf.is_empty() {
|
if self.write_buf.is_empty() {
|
||||||
return Ok(false);
|
return Ok(false);
|
||||||
}
|
}
|
||||||
|
|
||||||
let len = self.write_buf.len();
|
let len = self.write_buf.len();
|
||||||
let mut written = 0;
|
let mut written = 0;
|
||||||
|
#[project]
|
||||||
|
let InnerDispatcher { io, write_buf, .. } = self.project();
|
||||||
|
let mut io = Pin::new(io.as_mut().unwrap());
|
||||||
while written < len {
|
while written < len {
|
||||||
match unsafe { Pin::new_unchecked(&mut self.io) }
|
match io.as_mut().poll_write(cx, &write_buf[written..]) {
|
||||||
.poll_write(cx, &self.write_buf[written..])
|
|
||||||
{
|
|
||||||
Poll::Ready(Ok(0)) => {
|
Poll::Ready(Ok(0)) => {
|
||||||
return Err(DispatchError::Io(io::Error::new(
|
return Err(DispatchError::Io(io::Error::new(
|
||||||
io::ErrorKind::WriteZero,
|
io::ErrorKind::WriteZero,
|
||||||
@ -312,112 +326,118 @@ where
|
|||||||
}
|
}
|
||||||
Poll::Pending => {
|
Poll::Pending => {
|
||||||
if written > 0 {
|
if written > 0 {
|
||||||
self.write_buf.advance(written);
|
write_buf.advance(written);
|
||||||
}
|
}
|
||||||
return Ok(true);
|
return Ok(true);
|
||||||
}
|
}
|
||||||
Poll::Ready(Err(err)) => return Err(DispatchError::Io(err)),
|
Poll::Ready(Err(err)) => return Err(DispatchError::Io(err)),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if written == self.write_buf.len() {
|
if written == write_buf.len() {
|
||||||
unsafe { self.write_buf.set_len(0) }
|
unsafe { write_buf.set_len(0) }
|
||||||
} else {
|
} else {
|
||||||
self.write_buf.advance(written);
|
write_buf.advance(written);
|
||||||
}
|
}
|
||||||
Ok(false)
|
Ok(false)
|
||||||
}
|
}
|
||||||
|
|
||||||
fn send_response(
|
fn send_response(
|
||||||
&mut self,
|
self: Pin<&mut Self>,
|
||||||
message: Response<()>,
|
message: Response<()>,
|
||||||
body: ResponseBody<B>,
|
body: ResponseBody<B>,
|
||||||
) -> Result<State<S, B, X>, DispatchError> {
|
) -> Result<State<S, B, X>, DispatchError> {
|
||||||
self.codec
|
let mut this = self.project();
|
||||||
.encode(Message::Item((message, body.size())), &mut self.write_buf)
|
this.codec
|
||||||
|
.encode(Message::Item((message, body.size())), &mut this.write_buf)
|
||||||
.map_err(|err| {
|
.map_err(|err| {
|
||||||
if let Some(mut payload) = self.payload.take() {
|
if let Some(mut payload) = this.payload.take() {
|
||||||
payload.set_error(PayloadError::Incomplete(None));
|
payload.set_error(PayloadError::Incomplete(None));
|
||||||
}
|
}
|
||||||
DispatchError::Io(err)
|
DispatchError::Io(err)
|
||||||
})?;
|
})?;
|
||||||
|
|
||||||
self.flags.set(Flags::KEEPALIVE, self.codec.keepalive());
|
this.flags.set(Flags::KEEPALIVE, this.codec.keepalive());
|
||||||
match body.size() {
|
match body.size() {
|
||||||
BodySize::None | BodySize::Empty => Ok(State::None),
|
BodySize::None | BodySize::Empty => Ok(State::None),
|
||||||
_ => Ok(State::SendPayload(body)),
|
_ => Ok(State::SendPayload(body)),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn send_continue(&mut self) {
|
fn send_continue(self: Pin<&mut Self>) {
|
||||||
self.write_buf
|
self.project()
|
||||||
|
.write_buf
|
||||||
.extend_from_slice(b"HTTP/1.1 100 Continue\r\n\r\n");
|
.extend_from_slice(b"HTTP/1.1 100 Continue\r\n\r\n");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[pin_project::project]
|
||||||
fn poll_response(
|
fn poll_response(
|
||||||
&mut self,
|
mut self: Pin<&mut Self>,
|
||||||
cx: &mut Context<'_>,
|
cx: &mut Context<'_>,
|
||||||
) -> Result<PollResponse, DispatchError> {
|
) -> Result<PollResponse, DispatchError> {
|
||||||
loop {
|
loop {
|
||||||
let state = match self.state {
|
let mut this = self.as_mut().project();
|
||||||
State::None => match self.messages.pop_front() {
|
#[project]
|
||||||
|
let state = match this.state.project() {
|
||||||
|
State::None => match this.messages.pop_front() {
|
||||||
Some(DispatcherMessage::Item(req)) => {
|
Some(DispatcherMessage::Item(req)) => {
|
||||||
Some(self.handle_request(req, cx)?)
|
Some(self.as_mut().handle_request(req, cx)?)
|
||||||
}
|
|
||||||
Some(DispatcherMessage::Error(res)) => {
|
|
||||||
Some(self.send_response(res, ResponseBody::Other(Body::Empty))?)
|
|
||||||
}
|
}
|
||||||
|
Some(DispatcherMessage::Error(res)) => Some(
|
||||||
|
self.as_mut()
|
||||||
|
.send_response(res, ResponseBody::Other(Body::Empty))?,
|
||||||
|
),
|
||||||
Some(DispatcherMessage::Upgrade(req)) => {
|
Some(DispatcherMessage::Upgrade(req)) => {
|
||||||
return Ok(PollResponse::Upgrade(req));
|
return Ok(PollResponse::Upgrade(req));
|
||||||
}
|
}
|
||||||
None => None,
|
None => None,
|
||||||
},
|
},
|
||||||
State::ExpectCall(ref mut fut) => {
|
State::ExpectCall(fut) => match fut.poll(cx) {
|
||||||
match unsafe { Pin::new_unchecked(fut) }.poll(cx) {
|
Poll::Ready(Ok(req)) => {
|
||||||
Poll::Ready(Ok(req)) => {
|
self.as_mut().send_continue();
|
||||||
self.send_continue();
|
this = self.as_mut().project();
|
||||||
self.state = State::ServiceCall(self.service.call(req));
|
this.state.set(State::ServiceCall(this.service.call(req)));
|
||||||
continue;
|
continue;
|
||||||
}
|
|
||||||
Poll::Ready(Err(e)) => {
|
|
||||||
let res: Response = e.into().into();
|
|
||||||
let (res, body) = res.replace_body(());
|
|
||||||
Some(self.send_response(res, body.into_body())?)
|
|
||||||
}
|
|
||||||
Poll::Pending => None,
|
|
||||||
}
|
}
|
||||||
}
|
Poll::Ready(Err(e)) => {
|
||||||
State::ServiceCall(ref mut fut) => {
|
let res: Response = e.into().into();
|
||||||
match unsafe { Pin::new_unchecked(fut) }.poll(cx) {
|
let (res, body) = res.replace_body(());
|
||||||
Poll::Ready(Ok(res)) => {
|
Some(self.as_mut().send_response(res, body.into_body())?)
|
||||||
let (res, body) = res.into().replace_body(());
|
|
||||||
self.state = self.send_response(res, body)?;
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
Poll::Ready(Err(e)) => {
|
|
||||||
let res: Response = e.into().into();
|
|
||||||
let (res, body) = res.replace_body(());
|
|
||||||
Some(self.send_response(res, body.into_body())?)
|
|
||||||
}
|
|
||||||
Poll::Pending => None,
|
|
||||||
}
|
}
|
||||||
}
|
Poll::Pending => None,
|
||||||
State::SendPayload(ref mut stream) => {
|
},
|
||||||
|
State::ServiceCall(fut) => match fut.poll(cx) {
|
||||||
|
Poll::Ready(Ok(res)) => {
|
||||||
|
let (res, body) = res.into().replace_body(());
|
||||||
|
let state = self.as_mut().send_response(res, body)?;
|
||||||
|
this = self.as_mut().project();
|
||||||
|
this.state.set(state);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
Poll::Ready(Err(e)) => {
|
||||||
|
let res: Response = e.into().into();
|
||||||
|
let (res, body) = res.replace_body(());
|
||||||
|
Some(self.as_mut().send_response(res, body.into_body())?)
|
||||||
|
}
|
||||||
|
Poll::Pending => None,
|
||||||
|
},
|
||||||
|
State::SendPayload(mut stream) => {
|
||||||
loop {
|
loop {
|
||||||
if self.write_buf.len() < HW_BUFFER_SIZE {
|
if this.write_buf.len() < HW_BUFFER_SIZE {
|
||||||
match stream.poll_next(cx) {
|
match stream.as_mut().poll_next(cx) {
|
||||||
Poll::Ready(Some(Ok(item))) => {
|
Poll::Ready(Some(Ok(item))) => {
|
||||||
self.codec.encode(
|
this.codec.encode(
|
||||||
Message::Chunk(Some(item)),
|
Message::Chunk(Some(item)),
|
||||||
&mut self.write_buf,
|
&mut this.write_buf,
|
||||||
)?;
|
)?;
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
Poll::Ready(None) => {
|
Poll::Ready(None) => {
|
||||||
self.codec.encode(
|
this.codec.encode(
|
||||||
Message::Chunk(None),
|
Message::Chunk(None),
|
||||||
&mut self.write_buf,
|
&mut this.write_buf,
|
||||||
)?;
|
)?;
|
||||||
self.state = State::None;
|
this = self.as_mut().project();
|
||||||
|
this.state.set(State::None);
|
||||||
}
|
}
|
||||||
Poll::Ready(Some(Err(_))) => {
|
Poll::Ready(Some(Err(_))) => {
|
||||||
return Err(DispatchError::Unknown)
|
return Err(DispatchError::Unknown)
|
||||||
@ -433,9 +453,11 @@ where
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
this = self.as_mut().project();
|
||||||
|
|
||||||
// set new state
|
// set new state
|
||||||
if let Some(state) = state {
|
if let Some(state) = state {
|
||||||
self.state = state;
|
this.state.set(state);
|
||||||
if !self.state.is_empty() {
|
if !self.state.is_empty() {
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
@ -443,7 +465,7 @@ where
|
|||||||
// if read-backpressure is enabled and we consumed some data.
|
// if read-backpressure is enabled and we consumed some data.
|
||||||
// we may read more data and retry
|
// we may read more data and retry
|
||||||
if self.state.is_call() {
|
if self.state.is_call() {
|
||||||
if self.poll_request(cx)? {
|
if self.as_mut().poll_request(cx)? {
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
} else if !self.messages.is_empty() {
|
} else if !self.messages.is_empty() {
|
||||||
@ -457,16 +479,16 @@ where
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn handle_request(
|
fn handle_request(
|
||||||
&mut self,
|
mut self: Pin<&mut Self>,
|
||||||
req: Request,
|
req: Request,
|
||||||
cx: &mut Context<'_>,
|
cx: &mut Context<'_>,
|
||||||
) -> Result<State<S, B, X>, DispatchError> {
|
) -> Result<State<S, B, X>, DispatchError> {
|
||||||
// Handle `EXPECT: 100-Continue` header
|
// Handle `EXPECT: 100-Continue` header
|
||||||
let req = if req.head().expect() {
|
let req = if req.head().expect() {
|
||||||
let mut task = self.expect.call(req);
|
let mut task = self.as_mut().project().expect.call(req);
|
||||||
match unsafe { Pin::new_unchecked(&mut task) }.poll(cx) {
|
match unsafe { Pin::new_unchecked(&mut task) }.poll(cx) {
|
||||||
Poll::Ready(Ok(req)) => {
|
Poll::Ready(Ok(req)) => {
|
||||||
self.send_continue();
|
self.as_mut().send_continue();
|
||||||
req
|
req
|
||||||
}
|
}
|
||||||
Poll::Pending => return Ok(State::ExpectCall(task)),
|
Poll::Pending => return Ok(State::ExpectCall(task)),
|
||||||
@ -482,7 +504,7 @@ where
|
|||||||
};
|
};
|
||||||
|
|
||||||
// Call service
|
// Call service
|
||||||
let mut task = self.service.call(req);
|
let mut task = self.as_mut().project().service.call(req);
|
||||||
match unsafe { Pin::new_unchecked(&mut task) }.poll(cx) {
|
match unsafe { Pin::new_unchecked(&mut task) }.poll(cx) {
|
||||||
Poll::Ready(Ok(res)) => {
|
Poll::Ready(Ok(res)) => {
|
||||||
let (res, body) = res.into().replace_body(());
|
let (res, body) = res.into().replace_body(());
|
||||||
@ -499,7 +521,7 @@ where
|
|||||||
|
|
||||||
/// Process one incoming requests
|
/// Process one incoming requests
|
||||||
pub(self) fn poll_request(
|
pub(self) fn poll_request(
|
||||||
&mut self,
|
mut self: Pin<&mut Self>,
|
||||||
cx: &mut Context<'_>,
|
cx: &mut Context<'_>,
|
||||||
) -> Result<bool, DispatchError> {
|
) -> Result<bool, DispatchError> {
|
||||||
// limit a mount of non processed requests
|
// limit a mount of non processed requests
|
||||||
@ -508,24 +530,25 @@ where
|
|||||||
}
|
}
|
||||||
|
|
||||||
let mut updated = false;
|
let mut updated = false;
|
||||||
|
let mut this = self.as_mut().project();
|
||||||
loop {
|
loop {
|
||||||
match self.codec.decode(&mut self.read_buf) {
|
match this.codec.decode(&mut this.read_buf) {
|
||||||
Ok(Some(msg)) => {
|
Ok(Some(msg)) => {
|
||||||
updated = true;
|
updated = true;
|
||||||
self.flags.insert(Flags::STARTED);
|
this.flags.insert(Flags::STARTED);
|
||||||
|
|
||||||
match msg {
|
match msg {
|
||||||
Message::Item(mut req) => {
|
Message::Item(mut req) => {
|
||||||
let pl = self.codec.message_type();
|
let pl = this.codec.message_type();
|
||||||
req.head_mut().peer_addr = self.peer_addr;
|
req.head_mut().peer_addr = *this.peer_addr;
|
||||||
|
|
||||||
// set on_connect data
|
// set on_connect data
|
||||||
if let Some(ref on_connect) = self.on_connect {
|
if let Some(ref on_connect) = this.on_connect {
|
||||||
on_connect.set(&mut req.extensions_mut());
|
on_connect.set(&mut req.extensions_mut());
|
||||||
}
|
}
|
||||||
|
|
||||||
if pl == MessageType::Stream && self.upgrade.is_some() {
|
if pl == MessageType::Stream && this.upgrade.is_some() {
|
||||||
self.messages.push_back(DispatcherMessage::Upgrade(req));
|
this.messages.push_back(DispatcherMessage::Upgrade(req));
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
if pl == MessageType::Payload || pl == MessageType::Stream {
|
if pl == MessageType::Payload || pl == MessageType::Stream {
|
||||||
@ -533,41 +556,43 @@ where
|
|||||||
let (req1, _) =
|
let (req1, _) =
|
||||||
req.replace_payload(crate::Payload::H1(pl));
|
req.replace_payload(crate::Payload::H1(pl));
|
||||||
req = req1;
|
req = req1;
|
||||||
self.payload = Some(ps);
|
*this.payload = Some(ps);
|
||||||
}
|
}
|
||||||
|
|
||||||
// handle request early
|
// handle request early
|
||||||
if self.state.is_empty() {
|
if this.state.is_empty() {
|
||||||
self.state = self.handle_request(req, cx)?;
|
let state = self.as_mut().handle_request(req, cx)?;
|
||||||
|
this = self.as_mut().project();
|
||||||
|
this.state.set(state);
|
||||||
} else {
|
} else {
|
||||||
self.messages.push_back(DispatcherMessage::Item(req));
|
this.messages.push_back(DispatcherMessage::Item(req));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Message::Chunk(Some(chunk)) => {
|
Message::Chunk(Some(chunk)) => {
|
||||||
if let Some(ref mut payload) = self.payload {
|
if let Some(ref mut payload) = this.payload {
|
||||||
payload.feed_data(chunk);
|
payload.feed_data(chunk);
|
||||||
} else {
|
} else {
|
||||||
error!(
|
error!(
|
||||||
"Internal server error: unexpected payload chunk"
|
"Internal server error: unexpected payload chunk"
|
||||||
);
|
);
|
||||||
self.flags.insert(Flags::READ_DISCONNECT);
|
this.flags.insert(Flags::READ_DISCONNECT);
|
||||||
self.messages.push_back(DispatcherMessage::Error(
|
this.messages.push_back(DispatcherMessage::Error(
|
||||||
Response::InternalServerError().finish().drop_body(),
|
Response::InternalServerError().finish().drop_body(),
|
||||||
));
|
));
|
||||||
self.error = Some(DispatchError::InternalError);
|
*this.error = Some(DispatchError::InternalError);
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Message::Chunk(None) => {
|
Message::Chunk(None) => {
|
||||||
if let Some(mut payload) = self.payload.take() {
|
if let Some(mut payload) = this.payload.take() {
|
||||||
payload.feed_eof();
|
payload.feed_eof();
|
||||||
} else {
|
} else {
|
||||||
error!("Internal server error: unexpected eof");
|
error!("Internal server error: unexpected eof");
|
||||||
self.flags.insert(Flags::READ_DISCONNECT);
|
this.flags.insert(Flags::READ_DISCONNECT);
|
||||||
self.messages.push_back(DispatcherMessage::Error(
|
this.messages.push_back(DispatcherMessage::Error(
|
||||||
Response::InternalServerError().finish().drop_body(),
|
Response::InternalServerError().finish().drop_body(),
|
||||||
));
|
));
|
||||||
self.error = Some(DispatchError::InternalError);
|
*this.error = Some(DispatchError::InternalError);
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -575,44 +600,49 @@ where
|
|||||||
}
|
}
|
||||||
Ok(None) => break,
|
Ok(None) => break,
|
||||||
Err(ParseError::Io(e)) => {
|
Err(ParseError::Io(e)) => {
|
||||||
self.client_disconnected();
|
self.as_mut().client_disconnected();
|
||||||
self.error = Some(DispatchError::Io(e));
|
this = self.as_mut().project();
|
||||||
|
*this.error = Some(DispatchError::Io(e));
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
if let Some(mut payload) = self.payload.take() {
|
if let Some(mut payload) = this.payload.take() {
|
||||||
payload.set_error(PayloadError::EncodingCorrupted);
|
payload.set_error(PayloadError::EncodingCorrupted);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Malformed requests should be responded with 400
|
// Malformed requests should be responded with 400
|
||||||
self.messages.push_back(DispatcherMessage::Error(
|
this.messages.push_back(DispatcherMessage::Error(
|
||||||
Response::BadRequest().finish().drop_body(),
|
Response::BadRequest().finish().drop_body(),
|
||||||
));
|
));
|
||||||
self.flags.insert(Flags::READ_DISCONNECT);
|
this.flags.insert(Flags::READ_DISCONNECT);
|
||||||
self.error = Some(e.into());
|
*this.error = Some(e.into());
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if updated && self.ka_timer.is_some() {
|
if updated && this.ka_timer.is_some() {
|
||||||
if let Some(expire) = self.codec.config().keep_alive_expire() {
|
if let Some(expire) = this.codec.config().keep_alive_expire() {
|
||||||
self.ka_expire = expire;
|
*this.ka_expire = expire;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Ok(updated)
|
Ok(updated)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// keep-alive timer
|
/// keep-alive timer
|
||||||
fn poll_keepalive(&mut self, cx: &mut Context<'_>) -> Result<(), DispatchError> {
|
fn poll_keepalive(
|
||||||
if self.ka_timer.is_none() {
|
mut self: Pin<&mut Self>,
|
||||||
|
cx: &mut Context<'_>,
|
||||||
|
) -> Result<(), DispatchError> {
|
||||||
|
let mut this = self.as_mut().project();
|
||||||
|
if this.ka_timer.is_none() {
|
||||||
// shutdown timeout
|
// shutdown timeout
|
||||||
if self.flags.contains(Flags::SHUTDOWN) {
|
if this.flags.contains(Flags::SHUTDOWN) {
|
||||||
if let Some(interval) = self.codec.config().client_disconnect_timer() {
|
if let Some(interval) = this.codec.config().client_disconnect_timer() {
|
||||||
self.ka_timer = Some(delay_until(interval));
|
*this.ka_timer = Some(delay_until(interval));
|
||||||
} else {
|
} else {
|
||||||
self.flags.insert(Flags::READ_DISCONNECT);
|
this.flags.insert(Flags::READ_DISCONNECT);
|
||||||
if let Some(mut payload) = self.payload.take() {
|
if let Some(mut payload) = this.payload.take() {
|
||||||
payload.set_error(PayloadError::Incomplete(None));
|
payload.set_error(PayloadError::Incomplete(None));
|
||||||
}
|
}
|
||||||
return Ok(());
|
return Ok(());
|
||||||
@ -622,55 +652,56 @@ where
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
match Pin::new(&mut self.ka_timer.as_mut().unwrap()).poll(cx) {
|
match Pin::new(&mut this.ka_timer.as_mut().unwrap()).poll(cx) {
|
||||||
Poll::Ready(()) => {
|
Poll::Ready(()) => {
|
||||||
// if we get timeout during shutdown, drop connection
|
// if we get timeout during shutdown, drop connection
|
||||||
if self.flags.contains(Flags::SHUTDOWN) {
|
if this.flags.contains(Flags::SHUTDOWN) {
|
||||||
return Err(DispatchError::DisconnectTimeout);
|
return Err(DispatchError::DisconnectTimeout);
|
||||||
} else if self.ka_timer.as_mut().unwrap().deadline() >= self.ka_expire {
|
} else if this.ka_timer.as_mut().unwrap().deadline() >= *this.ka_expire {
|
||||||
// check for any outstanding tasks
|
// check for any outstanding tasks
|
||||||
if self.state.is_empty() && self.write_buf.is_empty() {
|
if this.state.is_empty() && this.write_buf.is_empty() {
|
||||||
if self.flags.contains(Flags::STARTED) {
|
if this.flags.contains(Flags::STARTED) {
|
||||||
trace!("Keep-alive timeout, close connection");
|
trace!("Keep-alive timeout, close connection");
|
||||||
self.flags.insert(Flags::SHUTDOWN);
|
this.flags.insert(Flags::SHUTDOWN);
|
||||||
|
|
||||||
// start shutdown timer
|
// start shutdown timer
|
||||||
if let Some(deadline) =
|
if let Some(deadline) =
|
||||||
self.codec.config().client_disconnect_timer()
|
this.codec.config().client_disconnect_timer()
|
||||||
{
|
{
|
||||||
if let Some(mut timer) = self.ka_timer.as_mut() {
|
if let Some(mut timer) = this.ka_timer.as_mut() {
|
||||||
timer.reset(deadline);
|
timer.reset(deadline);
|
||||||
let _ = Pin::new(&mut timer).poll(cx);
|
let _ = Pin::new(&mut timer).poll(cx);
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
// no shutdown timeout, drop socket
|
// no shutdown timeout, drop socket
|
||||||
self.flags.insert(Flags::WRITE_DISCONNECT);
|
this.flags.insert(Flags::WRITE_DISCONNECT);
|
||||||
return Ok(());
|
return Ok(());
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
// timeout on first request (slow request) return 408
|
// timeout on first request (slow request) return 408
|
||||||
if !self.flags.contains(Flags::STARTED) {
|
if !this.flags.contains(Flags::STARTED) {
|
||||||
trace!("Slow request timeout");
|
trace!("Slow request timeout");
|
||||||
let _ = self.send_response(
|
let _ = self.as_mut().send_response(
|
||||||
Response::RequestTimeout().finish().drop_body(),
|
Response::RequestTimeout().finish().drop_body(),
|
||||||
ResponseBody::Other(Body::Empty),
|
ResponseBody::Other(Body::Empty),
|
||||||
);
|
);
|
||||||
|
this = self.as_mut().project();
|
||||||
} else {
|
} else {
|
||||||
trace!("Keep-alive connection timeout");
|
trace!("Keep-alive connection timeout");
|
||||||
}
|
}
|
||||||
self.flags.insert(Flags::STARTED | Flags::SHUTDOWN);
|
this.flags.insert(Flags::STARTED | Flags::SHUTDOWN);
|
||||||
self.state = State::None;
|
this.state.set(State::None);
|
||||||
}
|
}
|
||||||
} else if let Some(deadline) =
|
} else if let Some(deadline) =
|
||||||
self.codec.config().keep_alive_expire()
|
this.codec.config().keep_alive_expire()
|
||||||
{
|
{
|
||||||
if let Some(mut timer) = self.ka_timer.as_mut() {
|
if let Some(mut timer) = this.ka_timer.as_mut() {
|
||||||
timer.reset(deadline);
|
timer.reset(deadline);
|
||||||
let _ = Pin::new(&mut timer).poll(cx);
|
let _ = Pin::new(&mut timer).poll(cx);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
} else if let Some(mut timer) = self.ka_timer.as_mut() {
|
} else if let Some(mut timer) = this.ka_timer.as_mut() {
|
||||||
timer.reset(self.ka_expire);
|
timer.reset(*this.ka_expire);
|
||||||
let _ = Pin::new(&mut timer).poll(cx);
|
let _ = Pin::new(&mut timer).poll(cx);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -681,20 +712,6 @@ where
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<T, S, B, X, U> Unpin for Dispatcher<T, S, B, X, U>
|
|
||||||
where
|
|
||||||
T: AsyncRead + AsyncWrite + Unpin,
|
|
||||||
S: Service<Request = Request>,
|
|
||||||
S::Error: Into<Error>,
|
|
||||||
S::Response: Into<Response<B>>,
|
|
||||||
B: MessageBody,
|
|
||||||
X: Service<Request = Request, Response = Request>,
|
|
||||||
X::Error: Into<Error>,
|
|
||||||
U: Service<Request = (Request, Framed<T, Codec>), Response = ()>,
|
|
||||||
U::Error: fmt::Display,
|
|
||||||
{
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<T, S, B, X, U> Future for Dispatcher<T, S, B, X, U>
|
impl<T, S, B, X, U> Future for Dispatcher<T, S, B, X, U>
|
||||||
where
|
where
|
||||||
T: AsyncRead + AsyncWrite + Unpin,
|
T: AsyncRead + AsyncWrite + Unpin,
|
||||||
@ -709,22 +726,29 @@ where
|
|||||||
{
|
{
|
||||||
type Output = Result<(), DispatchError>;
|
type Output = Result<(), DispatchError>;
|
||||||
|
|
||||||
|
#[pin_project::project]
|
||||||
#[inline]
|
#[inline]
|
||||||
fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
|
fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
|
||||||
match self.as_mut().inner {
|
let this = self.as_mut().project();
|
||||||
DispatcherState::Normal(ref mut inner) => {
|
#[project]
|
||||||
inner.poll_keepalive(cx)?;
|
match this.inner.project() {
|
||||||
|
DispatcherState::Normal(mut inner) => {
|
||||||
|
inner.as_mut().poll_keepalive(cx)?;
|
||||||
|
|
||||||
if inner.flags.contains(Flags::SHUTDOWN) {
|
if inner.flags.contains(Flags::SHUTDOWN) {
|
||||||
if inner.flags.contains(Flags::WRITE_DISCONNECT) {
|
if inner.flags.contains(Flags::WRITE_DISCONNECT) {
|
||||||
Poll::Ready(Ok(()))
|
Poll::Ready(Ok(()))
|
||||||
} else {
|
} else {
|
||||||
// flush buffer
|
// flush buffer
|
||||||
inner.poll_flush(cx)?;
|
inner.as_mut().poll_flush(cx)?;
|
||||||
if !inner.write_buf.is_empty() {
|
if !inner.write_buf.is_empty() || inner.io.is_none() {
|
||||||
Poll::Pending
|
Poll::Pending
|
||||||
} else {
|
} else {
|
||||||
match Pin::new(&mut inner.io).poll_shutdown(cx) {
|
match Pin::new(inner.project().io)
|
||||||
|
.as_pin_mut()
|
||||||
|
.unwrap()
|
||||||
|
.poll_shutdown(cx)
|
||||||
|
{
|
||||||
Poll::Ready(res) => {
|
Poll::Ready(res) => {
|
||||||
Poll::Ready(res.map_err(DispatchError::from))
|
Poll::Ready(res.map_err(DispatchError::from))
|
||||||
}
|
}
|
||||||
@ -736,53 +760,61 @@ where
|
|||||||
// read socket into a buf
|
// read socket into a buf
|
||||||
let should_disconnect =
|
let should_disconnect =
|
||||||
if !inner.flags.contains(Flags::READ_DISCONNECT) {
|
if !inner.flags.contains(Flags::READ_DISCONNECT) {
|
||||||
read_available(cx, &mut inner.io, &mut inner.read_buf)?
|
let mut inner_p = inner.as_mut().project();
|
||||||
|
read_available(
|
||||||
|
cx,
|
||||||
|
inner_p.io.as_mut().unwrap(),
|
||||||
|
&mut inner_p.read_buf,
|
||||||
|
)?
|
||||||
} else {
|
} else {
|
||||||
None
|
None
|
||||||
};
|
};
|
||||||
|
|
||||||
inner.poll_request(cx)?;
|
inner.as_mut().poll_request(cx)?;
|
||||||
if let Some(true) = should_disconnect {
|
if let Some(true) = should_disconnect {
|
||||||
inner.flags.insert(Flags::READ_DISCONNECT);
|
let inner_p = inner.as_mut().project();
|
||||||
if let Some(mut payload) = inner.payload.take() {
|
inner_p.flags.insert(Flags::READ_DISCONNECT);
|
||||||
|
if let Some(mut payload) = inner_p.payload.take() {
|
||||||
payload.feed_eof();
|
payload.feed_eof();
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
loop {
|
loop {
|
||||||
|
let inner_p = inner.as_mut().project();
|
||||||
let remaining =
|
let remaining =
|
||||||
inner.write_buf.capacity() - inner.write_buf.len();
|
inner_p.write_buf.capacity() - inner_p.write_buf.len();
|
||||||
if remaining < LW_BUFFER_SIZE {
|
if remaining < LW_BUFFER_SIZE {
|
||||||
inner.write_buf.reserve(HW_BUFFER_SIZE - remaining);
|
inner_p.write_buf.reserve(HW_BUFFER_SIZE - remaining);
|
||||||
}
|
}
|
||||||
let result = inner.poll_response(cx)?;
|
let result = inner.as_mut().poll_response(cx)?;
|
||||||
let drain = result == PollResponse::DrainWriteBuf;
|
let drain = result == PollResponse::DrainWriteBuf;
|
||||||
|
|
||||||
// switch to upgrade handler
|
// switch to upgrade handler
|
||||||
if let PollResponse::Upgrade(req) = result {
|
if let PollResponse::Upgrade(req) = result {
|
||||||
if let DispatcherState::Normal(inner) =
|
let inner_p = inner.as_mut().project();
|
||||||
std::mem::replace(&mut self.inner, DispatcherState::None)
|
let mut parts = FramedParts::with_read_buf(
|
||||||
{
|
inner_p.io.take().unwrap(),
|
||||||
let mut parts = FramedParts::with_read_buf(
|
std::mem::replace(inner_p.codec, Codec::default()),
|
||||||
inner.io,
|
std::mem::replace(inner_p.read_buf, BytesMut::default()),
|
||||||
inner.codec,
|
);
|
||||||
inner.read_buf,
|
parts.write_buf = std::mem::replace(
|
||||||
);
|
inner_p.write_buf,
|
||||||
parts.write_buf = inner.write_buf;
|
BytesMut::default(),
|
||||||
let framed = Framed::from_parts(parts);
|
);
|
||||||
self.inner = DispatcherState::Upgrade(
|
let framed = Framed::from_parts(parts);
|
||||||
inner.upgrade.unwrap().call((req, framed)),
|
let upgrade =
|
||||||
);
|
inner_p.upgrade.take().unwrap().call((req, framed));
|
||||||
return self.poll(cx);
|
self.as_mut()
|
||||||
} else {
|
.project()
|
||||||
panic!()
|
.inner
|
||||||
}
|
.set(DispatcherState::Upgrade(upgrade));
|
||||||
|
return self.poll(cx);
|
||||||
}
|
}
|
||||||
|
|
||||||
// we didnt get WouldBlock from write operation,
|
// we didnt get WouldBlock from write operation,
|
||||||
// so data get written to kernel completely (OSX)
|
// so data get written to kernel completely (OSX)
|
||||||
// and we have to write again otherwise response can get stuck
|
// and we have to write again otherwise response can get stuck
|
||||||
if inner.poll_flush(cx)? || !drain {
|
if inner.as_mut().poll_flush(cx)? || !drain {
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -794,25 +826,26 @@ where
|
|||||||
|
|
||||||
let is_empty = inner.state.is_empty();
|
let is_empty = inner.state.is_empty();
|
||||||
|
|
||||||
|
let inner_p = inner.as_mut().project();
|
||||||
// read half is closed and we do not processing any responses
|
// read half is closed and we do not processing any responses
|
||||||
if inner.flags.contains(Flags::READ_DISCONNECT) && is_empty {
|
if inner_p.flags.contains(Flags::READ_DISCONNECT) && is_empty {
|
||||||
inner.flags.insert(Flags::SHUTDOWN);
|
inner_p.flags.insert(Flags::SHUTDOWN);
|
||||||
}
|
}
|
||||||
|
|
||||||
// keep-alive and stream errors
|
// keep-alive and stream errors
|
||||||
if is_empty && inner.write_buf.is_empty() {
|
if is_empty && inner_p.write_buf.is_empty() {
|
||||||
if let Some(err) = inner.error.take() {
|
if let Some(err) = inner_p.error.take() {
|
||||||
Poll::Ready(Err(err))
|
Poll::Ready(Err(err))
|
||||||
}
|
}
|
||||||
// disconnect if keep-alive is not enabled
|
// disconnect if keep-alive is not enabled
|
||||||
else if inner.flags.contains(Flags::STARTED)
|
else if inner_p.flags.contains(Flags::STARTED)
|
||||||
&& !inner.flags.intersects(Flags::KEEPALIVE)
|
&& !inner_p.flags.intersects(Flags::KEEPALIVE)
|
||||||
{
|
{
|
||||||
inner.flags.insert(Flags::SHUTDOWN);
|
inner_p.flags.insert(Flags::SHUTDOWN);
|
||||||
self.poll(cx)
|
self.poll(cx)
|
||||||
}
|
}
|
||||||
// disconnect if shutdown
|
// disconnect if shutdown
|
||||||
else if inner.flags.contains(Flags::SHUTDOWN) {
|
else if inner_p.flags.contains(Flags::SHUTDOWN) {
|
||||||
self.poll(cx)
|
self.poll(cx)
|
||||||
} else {
|
} else {
|
||||||
Poll::Pending
|
Poll::Pending
|
||||||
@ -822,13 +855,10 @@ where
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
DispatcherState::Upgrade(ref mut fut) => {
|
DispatcherState::Upgrade(fut) => fut.poll(cx).map_err(|e| {
|
||||||
unsafe { Pin::new_unchecked(fut) }.poll(cx).map_err(|e| {
|
error!("Upgrade handler error: {}", e);
|
||||||
error!("Upgrade handler error: {}", e);
|
DispatchError::Upgrade
|
||||||
DispatchError::Upgrade
|
}),
|
||||||
})
|
|
||||||
}
|
|
||||||
DispatcherState::None => panic!(),
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -918,9 +948,12 @@ mod tests {
|
|||||||
Poll::Ready(res) => assert!(res.is_err()),
|
Poll::Ready(res) => assert!(res.is_err()),
|
||||||
}
|
}
|
||||||
|
|
||||||
if let DispatcherState::Normal(ref inner) = h1.inner {
|
if let DispatcherState::Normal(ref mut inner) = h1.inner {
|
||||||
assert!(inner.flags.contains(Flags::READ_DISCONNECT));
|
assert!(inner.flags.contains(Flags::READ_DISCONNECT));
|
||||||
assert_eq!(&inner.io.write_buf[..26], b"HTTP/1.1 400 Bad Request\r\n");
|
assert_eq!(
|
||||||
|
&inner.io.take().unwrap().write_buf[..26],
|
||||||
|
b"HTTP/1.1 400 Bad Request\r\n"
|
||||||
|
);
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
.await;
|
.await;
|
||||||
|
@ -364,7 +364,7 @@ where
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// `Service` implementation for HTTP1 transport
|
/// `Service` implementation for HTTP1 transport
|
||||||
pub struct H1ServiceHandler<T, S, B, X, U> {
|
pub struct H1ServiceHandler<T, S: Service, B, X: Service, U: Service> {
|
||||||
srv: CloneableService<S>,
|
srv: CloneableService<S>,
|
||||||
expect: CloneableService<X>,
|
expect: CloneableService<X>,
|
||||||
upgrade: Option<CloneableService<U>>,
|
upgrade: Option<CloneableService<U>>,
|
||||||
|
@ -13,6 +13,7 @@ use crate::response::Response;
|
|||||||
#[pin_project::pin_project]
|
#[pin_project::pin_project]
|
||||||
pub struct SendResponse<T, B> {
|
pub struct SendResponse<T, B> {
|
||||||
res: Option<Message<(Response<()>, BodySize)>>,
|
res: Option<Message<(Response<()>, BodySize)>>,
|
||||||
|
#[pin]
|
||||||
body: Option<ResponseBody<B>>,
|
body: Option<ResponseBody<B>>,
|
||||||
framed: Option<Framed<T, Codec>>,
|
framed: Option<Framed<T, Codec>>,
|
||||||
}
|
}
|
||||||
@ -35,24 +36,27 @@ where
|
|||||||
impl<T, B> Future for SendResponse<T, B>
|
impl<T, B> Future for SendResponse<T, B>
|
||||||
where
|
where
|
||||||
T: AsyncRead + AsyncWrite,
|
T: AsyncRead + AsyncWrite,
|
||||||
B: MessageBody,
|
B: MessageBody + Unpin,
|
||||||
{
|
{
|
||||||
type Output = Result<Framed<T, Codec>, Error>;
|
type Output = Result<Framed<T, Codec>, Error>;
|
||||||
|
|
||||||
|
// TODO: rethink if we need loops in polls
|
||||||
fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
|
fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
|
||||||
let this = self.get_mut();
|
let mut this = self.project();
|
||||||
|
|
||||||
|
let mut body_done = this.body.is_none();
|
||||||
loop {
|
loop {
|
||||||
let mut body_ready = this.body.is_some();
|
let mut body_ready = !body_done;
|
||||||
let framed = this.framed.as_mut().unwrap();
|
let framed = this.framed.as_mut().unwrap();
|
||||||
|
|
||||||
// send body
|
// send body
|
||||||
if this.res.is_none() && this.body.is_some() {
|
if this.res.is_none() && body_ready {
|
||||||
while body_ready && this.body.is_some() && !framed.is_write_buf_full() {
|
while body_ready && !body_done && !framed.is_write_buf_full() {
|
||||||
match this.body.as_mut().unwrap().poll_next(cx)? {
|
match this.body.as_mut().as_pin_mut().unwrap().poll_next(cx)? {
|
||||||
Poll::Ready(item) => {
|
Poll::Ready(item) => {
|
||||||
// body is done
|
// body is done when item is None
|
||||||
if item.is_none() {
|
body_done = item.is_none();
|
||||||
|
if body_done {
|
||||||
let _ = this.body.take();
|
let _ = this.body.take();
|
||||||
}
|
}
|
||||||
framed.write(Message::Chunk(item))?;
|
framed.write(Message::Chunk(item))?;
|
||||||
@ -82,7 +86,7 @@ where
|
|||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
if this.body.is_some() {
|
if !body_done {
|
||||||
if body_ready {
|
if body_ready {
|
||||||
continue;
|
continue;
|
||||||
} else {
|
} else {
|
||||||
|
@ -158,15 +158,17 @@ where
|
|||||||
|
|
||||||
#[pin_project::pin_project]
|
#[pin_project::pin_project]
|
||||||
struct ServiceResponse<F, I, E, B> {
|
struct ServiceResponse<F, I, E, B> {
|
||||||
|
#[pin]
|
||||||
state: ServiceResponseState<F, B>,
|
state: ServiceResponseState<F, B>,
|
||||||
config: ServiceConfig,
|
config: ServiceConfig,
|
||||||
buffer: Option<Bytes>,
|
buffer: Option<Bytes>,
|
||||||
_t: PhantomData<(I, E)>,
|
_t: PhantomData<(I, E)>,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[pin_project::pin_project]
|
||||||
enum ServiceResponseState<F, B> {
|
enum ServiceResponseState<F, B> {
|
||||||
ServiceCall(F, Option<SendResponse<Bytes>>),
|
ServiceCall(#[pin] F, Option<SendResponse<Bytes>>),
|
||||||
SendPayload(SendStream<Bytes>, ResponseBody<B>),
|
SendPayload(SendStream<Bytes>, #[pin] ResponseBody<B>),
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<F, I, E, B> ServiceResponse<F, I, E, B>
|
impl<F, I, E, B> ServiceResponse<F, I, E, B>
|
||||||
@ -247,68 +249,66 @@ where
|
|||||||
{
|
{
|
||||||
type Output = ();
|
type Output = ();
|
||||||
|
|
||||||
|
#[pin_project::project]
|
||||||
fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
|
fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
|
||||||
let mut this = self.as_mut().project();
|
let mut this = self.as_mut().project();
|
||||||
|
|
||||||
match this.state {
|
#[project]
|
||||||
ServiceResponseState::ServiceCall(ref mut call, ref mut send) => {
|
match this.state.project() {
|
||||||
match unsafe { Pin::new_unchecked(call) }.poll(cx) {
|
ServiceResponseState::ServiceCall(call, send) => match call.poll(cx) {
|
||||||
Poll::Ready(Ok(res)) => {
|
Poll::Ready(Ok(res)) => {
|
||||||
let (res, body) = res.into().replace_body(());
|
let (res, body) = res.into().replace_body(());
|
||||||
|
|
||||||
let mut send = send.take().unwrap();
|
let mut send = send.take().unwrap();
|
||||||
let mut size = body.size();
|
let mut size = body.size();
|
||||||
let h2_res =
|
let h2_res = self.as_mut().prepare_response(res.head(), &mut size);
|
||||||
self.as_mut().prepare_response(res.head(), &mut size);
|
this = self.as_mut().project();
|
||||||
this = self.as_mut().project();
|
|
||||||
|
|
||||||
let stream = match send.send_response(h2_res, size.is_eof()) {
|
let stream = match send.send_response(h2_res, size.is_eof()) {
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
trace!("Error sending h2 response: {:?}", e);
|
trace!("Error sending h2 response: {:?}", e);
|
||||||
return Poll::Ready(());
|
return Poll::Ready(());
|
||||||
}
|
|
||||||
Ok(stream) => stream,
|
|
||||||
};
|
|
||||||
|
|
||||||
if size.is_eof() {
|
|
||||||
Poll::Ready(())
|
|
||||||
} else {
|
|
||||||
*this.state =
|
|
||||||
ServiceResponseState::SendPayload(stream, body);
|
|
||||||
self.poll(cx)
|
|
||||||
}
|
}
|
||||||
}
|
Ok(stream) => stream,
|
||||||
Poll::Pending => Poll::Pending,
|
};
|
||||||
Poll::Ready(Err(e)) => {
|
|
||||||
let res: Response = e.into().into();
|
|
||||||
let (res, body) = res.replace_body(());
|
|
||||||
|
|
||||||
let mut send = send.take().unwrap();
|
if size.is_eof() {
|
||||||
let mut size = body.size();
|
Poll::Ready(())
|
||||||
let h2_res =
|
} else {
|
||||||
self.as_mut().prepare_response(res.head(), &mut size);
|
this.state
|
||||||
this = self.as_mut().project();
|
.set(ServiceResponseState::SendPayload(stream, body));
|
||||||
|
self.poll(cx)
|
||||||
let stream = match send.send_response(h2_res, size.is_eof()) {
|
|
||||||
Err(e) => {
|
|
||||||
trace!("Error sending h2 response: {:?}", e);
|
|
||||||
return Poll::Ready(());
|
|
||||||
}
|
|
||||||
Ok(stream) => stream,
|
|
||||||
};
|
|
||||||
|
|
||||||
if size.is_eof() {
|
|
||||||
Poll::Ready(())
|
|
||||||
} else {
|
|
||||||
*this.state = ServiceResponseState::SendPayload(
|
|
||||||
stream,
|
|
||||||
body.into_body(),
|
|
||||||
);
|
|
||||||
self.poll(cx)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
Poll::Pending => Poll::Pending,
|
||||||
|
Poll::Ready(Err(e)) => {
|
||||||
|
let res: Response = e.into().into();
|
||||||
|
let (res, body) = res.replace_body(());
|
||||||
|
|
||||||
|
let mut send = send.take().unwrap();
|
||||||
|
let mut size = body.size();
|
||||||
|
let h2_res = self.as_mut().prepare_response(res.head(), &mut size);
|
||||||
|
this = self.as_mut().project();
|
||||||
|
|
||||||
|
let stream = match send.send_response(h2_res, size.is_eof()) {
|
||||||
|
Err(e) => {
|
||||||
|
trace!("Error sending h2 response: {:?}", e);
|
||||||
|
return Poll::Ready(());
|
||||||
|
}
|
||||||
|
Ok(stream) => stream,
|
||||||
|
};
|
||||||
|
|
||||||
|
if size.is_eof() {
|
||||||
|
Poll::Ready(())
|
||||||
|
} else {
|
||||||
|
this.state.set(ServiceResponseState::SendPayload(
|
||||||
|
stream,
|
||||||
|
body.into_body(),
|
||||||
|
));
|
||||||
|
self.poll(cx)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
ServiceResponseState::SendPayload(ref mut stream, ref mut body) => loop {
|
ServiceResponseState::SendPayload(ref mut stream, ref mut body) => loop {
|
||||||
loop {
|
loop {
|
||||||
if let Some(ref mut buffer) = this.buffer {
|
if let Some(ref mut buffer) = this.buffer {
|
||||||
@ -335,7 +335,7 @@ where
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
match body.poll_next(cx) {
|
match body.as_mut().poll_next(cx) {
|
||||||
Poll::Pending => return Poll::Pending,
|
Poll::Pending => return Poll::Pending,
|
||||||
Poll::Ready(None) => {
|
Poll::Ready(None) => {
|
||||||
if let Err(e) = stream.send_data(Bytes::new(), true) {
|
if let Err(e) = stream.send_data(Bytes::new(), true) {
|
||||||
|
@ -83,13 +83,11 @@ where
|
|||||||
Error = DispatchError,
|
Error = DispatchError,
|
||||||
InitError = S::InitError,
|
InitError = S::InitError,
|
||||||
> {
|
> {
|
||||||
pipeline_factory(fn_factory(|| {
|
pipeline_factory(fn_factory(|| async {
|
||||||
async {
|
Ok::<_, S::InitError>(fn_service(|io: TcpStream| {
|
||||||
Ok::<_, S::InitError>(fn_service(|io: TcpStream| {
|
let peer_addr = io.peer_addr().ok();
|
||||||
let peer_addr = io.peer_addr().ok();
|
ok::<_, DispatchError>((io, peer_addr))
|
||||||
ok::<_, DispatchError>((io, peer_addr))
|
}))
|
||||||
}))
|
|
||||||
}
|
|
||||||
}))
|
}))
|
||||||
.and_then(self)
|
.and_then(self)
|
||||||
}
|
}
|
||||||
@ -246,7 +244,7 @@ where
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// `Service` implementation for http/2 transport
|
/// `Service` implementation for http/2 transport
|
||||||
pub struct H2ServiceHandler<T, S, B> {
|
pub struct H2ServiceHandler<T, S: Service, B> {
|
||||||
srv: CloneableService<S>,
|
srv: CloneableService<S>,
|
||||||
cfg: ServiceConfig,
|
cfg: ServiceConfig,
|
||||||
on_connect: Option<rc::Rc<dyn Fn(&T) -> Box<dyn DataFactory>>>,
|
on_connect: Option<rc::Rc<dyn Fn(&T) -> Box<dyn DataFactory>>>,
|
||||||
|
@ -63,7 +63,7 @@ header! {
|
|||||||
(AcceptCharset, ACCEPT_CHARSET) => (QualityItem<Charset>)+
|
(AcceptCharset, ACCEPT_CHARSET) => (QualityItem<Charset>)+
|
||||||
|
|
||||||
test_accept_charset {
|
test_accept_charset {
|
||||||
/// Test case from RFC
|
// Test case from RFC
|
||||||
test_header!(test1, vec![b"iso-8859-5, unicode-1-1;q=0.8"]);
|
test_header!(test1, vec![b"iso-8859-5, unicode-1-1;q=0.8"]);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -423,7 +423,7 @@ impl ContentDisposition {
|
|||||||
|
|
||||||
/// Return the value of *name* if exists.
|
/// Return the value of *name* if exists.
|
||||||
pub fn get_name(&self) -> Option<&str> {
|
pub fn get_name(&self) -> Option<&str> {
|
||||||
self.parameters.iter().filter_map(|p| p.as_name()).nth(0)
|
self.parameters.iter().filter_map(|p| p.as_name()).next()
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Return the value of *filename* if exists.
|
/// Return the value of *filename* if exists.
|
||||||
@ -431,7 +431,7 @@ impl ContentDisposition {
|
|||||||
self.parameters
|
self.parameters
|
||||||
.iter()
|
.iter()
|
||||||
.filter_map(|p| p.as_filename())
|
.filter_map(|p| p.as_filename())
|
||||||
.nth(0)
|
.next()
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Return the value of *filename\** if exists.
|
/// Return the value of *filename\** if exists.
|
||||||
@ -439,7 +439,7 @@ impl ContentDisposition {
|
|||||||
self.parameters
|
self.parameters
|
||||||
.iter()
|
.iter()
|
||||||
.filter_map(|p| p.as_filename_ext())
|
.filter_map(|p| p.as_filename_ext())
|
||||||
.nth(0)
|
.next()
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Return the value of the parameter which the `name` matches.
|
/// Return the value of the parameter which the `name` matches.
|
||||||
@ -448,7 +448,7 @@ impl ContentDisposition {
|
|||||||
self.parameters
|
self.parameters
|
||||||
.iter()
|
.iter()
|
||||||
.filter_map(|p| p.as_unknown(name))
|
.filter_map(|p| p.as_unknown(name))
|
||||||
.nth(0)
|
.next()
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Return the value of the extended parameter which the `name` matches.
|
/// Return the value of the extended parameter which the `name` matches.
|
||||||
@ -457,7 +457,7 @@ impl ContentDisposition {
|
|||||||
self.parameters
|
self.parameters
|
||||||
.iter()
|
.iter()
|
||||||
.filter_map(|p| p.as_unknown_ext(name))
|
.filter_map(|p| p.as_unknown_ext(name))
|
||||||
.nth(0)
|
.next()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1,59 +1,46 @@
|
|||||||
use std::fmt::{self, Display};
|
use std::fmt::{self, Display};
|
||||||
use std::io::Write;
|
use std::io::Write;
|
||||||
use std::str::FromStr;
|
use std::str::FromStr;
|
||||||
use std::time::{Duration, SystemTime, UNIX_EPOCH};
|
use std::time::{SystemTime, UNIX_EPOCH};
|
||||||
|
|
||||||
use bytes::{buf::BufMutExt, BytesMut};
|
use bytes::{buf::BufMutExt, BytesMut};
|
||||||
use http::header::{HeaderValue, InvalidHeaderValue};
|
use http::header::{HeaderValue, InvalidHeaderValue};
|
||||||
|
use time::{offset, OffsetDateTime, PrimitiveDateTime};
|
||||||
|
|
||||||
use crate::error::ParseError;
|
use crate::error::ParseError;
|
||||||
use crate::header::IntoHeaderValue;
|
use crate::header::IntoHeaderValue;
|
||||||
|
use crate::time_parser;
|
||||||
|
|
||||||
/// A timestamp with HTTP formatting and parsing
|
/// A timestamp with HTTP formatting and parsing
|
||||||
#[derive(Clone, Copy, Debug, PartialEq, Eq, PartialOrd, Ord)]
|
#[derive(Clone, Copy, Debug, PartialEq, Eq, PartialOrd, Ord)]
|
||||||
pub struct HttpDate(time::Tm);
|
pub struct HttpDate(OffsetDateTime);
|
||||||
|
|
||||||
impl FromStr for HttpDate {
|
impl FromStr for HttpDate {
|
||||||
type Err = ParseError;
|
type Err = ParseError;
|
||||||
|
|
||||||
fn from_str(s: &str) -> Result<HttpDate, ParseError> {
|
fn from_str(s: &str) -> Result<HttpDate, ParseError> {
|
||||||
match time::strptime(s, "%a, %d %b %Y %T %Z")
|
match time_parser::parse_http_date(s) {
|
||||||
.or_else(|_| time::strptime(s, "%A, %d-%b-%y %T %Z"))
|
Some(t) => Ok(HttpDate(t.assume_utc())),
|
||||||
.or_else(|_| time::strptime(s, "%c"))
|
None => Err(ParseError::Header),
|
||||||
{
|
|
||||||
Ok(t) => Ok(HttpDate(t)),
|
|
||||||
Err(_) => Err(ParseError::Header),
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Display for HttpDate {
|
impl Display for HttpDate {
|
||||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
||||||
fmt::Display::fmt(&self.0.to_utc().rfc822(), f)
|
fmt::Display::fmt(&self.0.format("%a, %d %b %Y %H:%M:%S GMT"), f)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl From<time::Tm> for HttpDate {
|
impl From<OffsetDateTime> for HttpDate {
|
||||||
fn from(tm: time::Tm) -> HttpDate {
|
fn from(dt: OffsetDateTime) -> HttpDate {
|
||||||
HttpDate(tm)
|
HttpDate(dt)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl From<SystemTime> for HttpDate {
|
impl From<SystemTime> for HttpDate {
|
||||||
fn from(sys: SystemTime) -> HttpDate {
|
fn from(sys: SystemTime) -> HttpDate {
|
||||||
let tmspec = match sys.duration_since(UNIX_EPOCH) {
|
HttpDate(PrimitiveDateTime::from(sys).assume_utc())
|
||||||
Ok(dur) => {
|
|
||||||
time::Timespec::new(dur.as_secs() as i64, dur.subsec_nanos() as i32)
|
|
||||||
}
|
|
||||||
Err(err) => {
|
|
||||||
let neg = err.duration();
|
|
||||||
time::Timespec::new(
|
|
||||||
-(neg.as_secs() as i64),
|
|
||||||
-(neg.subsec_nanos() as i32),
|
|
||||||
)
|
|
||||||
}
|
|
||||||
};
|
|
||||||
HttpDate(time::at_utc(tmspec))
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -62,56 +49,51 @@ impl IntoHeaderValue for HttpDate {
|
|||||||
|
|
||||||
fn try_into(self) -> Result<HeaderValue, Self::Error> {
|
fn try_into(self) -> Result<HeaderValue, Self::Error> {
|
||||||
let mut wrt = BytesMut::with_capacity(29).writer();
|
let mut wrt = BytesMut::with_capacity(29).writer();
|
||||||
write!(wrt, "{}", self.0.rfc822()).unwrap();
|
write!(
|
||||||
|
wrt,
|
||||||
|
"{}",
|
||||||
|
self.0
|
||||||
|
.to_offset(offset!(UTC))
|
||||||
|
.format("%a, %d %b %Y %H:%M:%S GMT")
|
||||||
|
)
|
||||||
|
.unwrap();
|
||||||
HeaderValue::from_maybe_shared(wrt.get_mut().split().freeze())
|
HeaderValue::from_maybe_shared(wrt.get_mut().split().freeze())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl From<HttpDate> for SystemTime {
|
impl From<HttpDate> for SystemTime {
|
||||||
fn from(date: HttpDate) -> SystemTime {
|
fn from(date: HttpDate) -> SystemTime {
|
||||||
let spec = date.0.to_timespec();
|
let dt = date.0;
|
||||||
if spec.sec >= 0 {
|
let epoch = OffsetDateTime::unix_epoch();
|
||||||
UNIX_EPOCH + Duration::new(spec.sec as u64, spec.nsec as u32)
|
|
||||||
} else {
|
UNIX_EPOCH + (dt - epoch)
|
||||||
UNIX_EPOCH - Duration::new(spec.sec as u64, spec.nsec as u32)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use super::HttpDate;
|
use super::HttpDate;
|
||||||
use time::Tm;
|
use time::{date, time, PrimitiveDateTime};
|
||||||
|
|
||||||
const NOV_07: HttpDate = HttpDate(Tm {
|
|
||||||
tm_nsec: 0,
|
|
||||||
tm_sec: 37,
|
|
||||||
tm_min: 48,
|
|
||||||
tm_hour: 8,
|
|
||||||
tm_mday: 7,
|
|
||||||
tm_mon: 10,
|
|
||||||
tm_year: 94,
|
|
||||||
tm_wday: 0,
|
|
||||||
tm_isdst: 0,
|
|
||||||
tm_yday: 0,
|
|
||||||
tm_utcoff: 0,
|
|
||||||
});
|
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_date() {
|
fn test_date() {
|
||||||
|
let nov_07 = HttpDate(
|
||||||
|
PrimitiveDateTime::new(date!(1994 - 11 - 07), time!(8:48:37)).assume_utc(),
|
||||||
|
);
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
"Sun, 07 Nov 1994 08:48:37 GMT".parse::<HttpDate>().unwrap(),
|
"Sun, 07 Nov 1994 08:48:37 GMT".parse::<HttpDate>().unwrap(),
|
||||||
NOV_07
|
nov_07
|
||||||
);
|
);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
"Sunday, 07-Nov-94 08:48:37 GMT"
|
"Sunday, 07-Nov-94 08:48:37 GMT"
|
||||||
.parse::<HttpDate>()
|
.parse::<HttpDate>()
|
||||||
.unwrap(),
|
.unwrap(),
|
||||||
NOV_07
|
nov_07
|
||||||
);
|
);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
"Sun Nov 7 08:48:37 1994".parse::<HttpDate>().unwrap(),
|
"Sun Nov 7 08:48:37 1994".parse::<HttpDate>().unwrap(),
|
||||||
NOV_07
|
nov_07
|
||||||
);
|
);
|
||||||
assert!("this-is-no-date".parse::<HttpDate>().is_err());
|
assert!("this-is-no-date".parse::<HttpDate>().is_err());
|
||||||
}
|
}
|
||||||
|
@ -1,4 +1,4 @@
|
|||||||
use std::{io, mem, ptr, slice};
|
use std::{io, ptr};
|
||||||
|
|
||||||
use bytes::{BufMut, BytesMut};
|
use bytes::{BufMut, BytesMut};
|
||||||
use http::Version;
|
use http::Version;
|
||||||
@ -14,9 +14,7 @@ const DEC_DIGITS_LUT: &[u8] = b"0001020304050607080910111213141516171819\
|
|||||||
pub(crate) const STATUS_LINE_BUF_SIZE: usize = 13;
|
pub(crate) const STATUS_LINE_BUF_SIZE: usize = 13;
|
||||||
|
|
||||||
pub(crate) fn write_status_line(version: Version, mut n: u16, bytes: &mut BytesMut) {
|
pub(crate) fn write_status_line(version: Version, mut n: u16, bytes: &mut BytesMut) {
|
||||||
let mut buf: [u8; STATUS_LINE_BUF_SIZE] = [
|
let mut buf: [u8; STATUS_LINE_BUF_SIZE] = *b"HTTP/1.1 ";
|
||||||
b'H', b'T', b'T', b'P', b'/', b'1', b'.', b'1', b' ', b' ', b' ', b' ', b' ',
|
|
||||||
];
|
|
||||||
match version {
|
match version {
|
||||||
Version::HTTP_2 => buf[5] = b'2',
|
Version::HTTP_2 => buf[5] = b'2',
|
||||||
Version::HTTP_10 => buf[7] = b'0',
|
Version::HTTP_10 => buf[7] = b'0',
|
||||||
@ -64,109 +62,104 @@ pub(crate) fn write_status_line(version: Version, mut n: u16, bytes: &mut BytesM
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const DIGITS_START: u8 = b'0';
|
||||||
|
|
||||||
/// NOTE: bytes object has to contain enough space
|
/// NOTE: bytes object has to contain enough space
|
||||||
pub fn write_content_length(mut n: usize, bytes: &mut BytesMut) {
|
pub fn write_content_length(n: usize, bytes: &mut BytesMut) {
|
||||||
|
bytes.put_slice(b"\r\ncontent-length: ");
|
||||||
|
|
||||||
if n < 10 {
|
if n < 10 {
|
||||||
let mut buf: [u8; 21] = [
|
bytes.put_u8(DIGITS_START + (n as u8));
|
||||||
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';
|
|
||||||
bytes.put_slice(&buf);
|
|
||||||
} else if n < 100 {
|
} else if n < 100 {
|
||||||
let mut buf: [u8; 22] = [
|
let n = n as u8;
|
||||||
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 d10 = n / 10;
|
||||||
];
|
let d1 = n % 10;
|
||||||
let d1 = n << 1;
|
|
||||||
unsafe {
|
bytes.put_u8(DIGITS_START + d10);
|
||||||
ptr::copy_nonoverlapping(
|
bytes.put_u8(DIGITS_START + d1);
|
||||||
DEC_DIGITS_LUT.as_ptr().add(d1),
|
|
||||||
buf.as_mut_ptr().offset(18),
|
|
||||||
2,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
bytes.put_slice(&buf);
|
|
||||||
} else if n < 1000 {
|
} else if n < 1000 {
|
||||||
let mut buf: [u8; 23] = [
|
let n = n as u16;
|
||||||
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(
|
|
||||||
DEC_DIGITS_LUT.as_ptr().add(d1),
|
|
||||||
buf.as_mut_ptr().offset(19),
|
|
||||||
2,
|
|
||||||
)
|
|
||||||
};
|
|
||||||
|
|
||||||
// decode last 1
|
let d100 = (n / 100) as u8;
|
||||||
buf[18] = (n as u8) + b'0';
|
let d10 = ((n / 10) % 10) as u8;
|
||||||
|
let d1 = (n % 10) as u8;
|
||||||
|
|
||||||
bytes.put_slice(&buf);
|
bytes.put_u8(DIGITS_START + d100);
|
||||||
|
bytes.put_u8(DIGITS_START + d10);
|
||||||
|
bytes.put_u8(DIGITS_START + d1);
|
||||||
|
} else if n < 10_000 {
|
||||||
|
let n = n as u16;
|
||||||
|
|
||||||
|
let d1000 = (n / 1000) as u8;
|
||||||
|
let d100 = ((n / 100) % 10) as u8;
|
||||||
|
let d10 = ((n / 10) % 10) as u8;
|
||||||
|
let d1 = (n % 10) as u8;
|
||||||
|
|
||||||
|
bytes.put_u8(DIGITS_START + d1000);
|
||||||
|
bytes.put_u8(DIGITS_START + d100);
|
||||||
|
bytes.put_u8(DIGITS_START + d10);
|
||||||
|
bytes.put_u8(DIGITS_START + d1);
|
||||||
|
} else if n < 100_000 {
|
||||||
|
let n = n as u32;
|
||||||
|
|
||||||
|
let d10000 = (n / 10000) as u8;
|
||||||
|
let d1000 = ((n / 1000) % 10) as u8;
|
||||||
|
let d100 = ((n / 100) % 10) as u8;
|
||||||
|
let d10 = ((n / 10) % 10) as u8;
|
||||||
|
let d1 = (n % 10) as u8;
|
||||||
|
|
||||||
|
bytes.put_u8(DIGITS_START + d10000);
|
||||||
|
bytes.put_u8(DIGITS_START + d1000);
|
||||||
|
bytes.put_u8(DIGITS_START + d100);
|
||||||
|
bytes.put_u8(DIGITS_START + d10);
|
||||||
|
bytes.put_u8(DIGITS_START + d1);
|
||||||
|
} else if n < 1_000_000 {
|
||||||
|
let n = n as u32;
|
||||||
|
|
||||||
|
let d100000 = (n / 100_000) as u8;
|
||||||
|
let d10000 = ((n / 10000) % 10) as u8;
|
||||||
|
let d1000 = ((n / 1000) % 10) as u8;
|
||||||
|
let d100 = ((n / 100) % 10) as u8;
|
||||||
|
let d10 = ((n / 10) % 10) as u8;
|
||||||
|
let d1 = (n % 10) as u8;
|
||||||
|
|
||||||
|
bytes.put_u8(DIGITS_START + d100000);
|
||||||
|
bytes.put_u8(DIGITS_START + d10000);
|
||||||
|
bytes.put_u8(DIGITS_START + d1000);
|
||||||
|
bytes.put_u8(DIGITS_START + d100);
|
||||||
|
bytes.put_u8(DIGITS_START + d10);
|
||||||
|
bytes.put_u8(DIGITS_START + d1);
|
||||||
} else {
|
} else {
|
||||||
bytes.put_slice(b"\r\ncontent-length: ");
|
write_usize(n, bytes);
|
||||||
convert_usize(n, bytes);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
bytes.put_slice(b"\r\n");
|
||||||
}
|
}
|
||||||
|
|
||||||
pub(crate) fn convert_usize(mut n: usize, bytes: &mut BytesMut) {
|
pub(crate) fn write_usize(n: usize, bytes: &mut BytesMut) {
|
||||||
let mut curr: isize = 39;
|
let mut n = n;
|
||||||
let mut buf: [u8; 41] = unsafe { mem::MaybeUninit::uninit().assume_init() };
|
|
||||||
buf[39] = b'\r';
|
|
||||||
buf[40] = b'\n';
|
|
||||||
let buf_ptr = buf.as_mut_ptr();
|
|
||||||
let lut_ptr = DEC_DIGITS_LUT.as_ptr();
|
|
||||||
|
|
||||||
// eagerly decode 4 characters at a time
|
// 20 chars is max length of a usize (2^64)
|
||||||
while n >= 10_000 {
|
// digits will be added to the buffer from lsd to msd
|
||||||
let rem = (n % 10_000) as isize;
|
let mut buf = BytesMut::with_capacity(20);
|
||||||
n /= 10_000;
|
|
||||||
|
|
||||||
let d1 = (rem / 100) << 1;
|
while n > 9 {
|
||||||
let d2 = (rem % 100) << 1;
|
// "pop" the least-significant digit
|
||||||
curr -= 4;
|
let lsd = (n % 10) as u8;
|
||||||
unsafe {
|
|
||||||
ptr::copy_nonoverlapping(lut_ptr.offset(d1), buf_ptr.offset(curr), 2);
|
// remove the lsd from n
|
||||||
ptr::copy_nonoverlapping(lut_ptr.offset(d2), buf_ptr.offset(curr + 2), 2);
|
n /= 10;
|
||||||
}
|
|
||||||
|
buf.put_u8(DIGITS_START + lsd);
|
||||||
}
|
}
|
||||||
|
|
||||||
// if we reach here numbers are <= 9999, so at most 4 chars long
|
// put msd to result buffer
|
||||||
let mut n = n as isize; // possibly reduce 64bit math
|
bytes.put_u8(DIGITS_START + (n as u8));
|
||||||
|
|
||||||
// decode 2 more chars, if > 2 chars
|
// put, in reverse (msd to lsd), remaining digits to buffer
|
||||||
if n >= 100 {
|
for i in (0..buf.len()).rev() {
|
||||||
let d1 = (n % 100) << 1;
|
bytes.put_u8(buf[i]);
|
||||||
n /= 100;
|
|
||||||
curr -= 2;
|
|
||||||
unsafe {
|
|
||||||
ptr::copy_nonoverlapping(lut_ptr.offset(d1), buf_ptr.offset(curr), 2);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// decode last 1 or 2 chars
|
|
||||||
if n < 10 {
|
|
||||||
curr -= 1;
|
|
||||||
unsafe {
|
|
||||||
*buf_ptr.offset(curr) = (n as u8) + b'0';
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
let d1 = n << 1;
|
|
||||||
curr -= 2;
|
|
||||||
unsafe {
|
|
||||||
ptr::copy_nonoverlapping(lut_ptr.offset(d1), buf_ptr.offset(curr), 2);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
unsafe {
|
|
||||||
bytes.extend_from_slice(slice::from_raw_parts(
|
|
||||||
buf_ptr.offset(curr),
|
|
||||||
41 - curr as usize,
|
|
||||||
));
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -231,5 +224,48 @@ mod tests {
|
|||||||
bytes.reserve(50);
|
bytes.reserve(50);
|
||||||
write_content_length(5909, &mut bytes);
|
write_content_length(5909, &mut bytes);
|
||||||
assert_eq!(bytes.split().freeze(), b"\r\ncontent-length: 5909\r\n"[..]);
|
assert_eq!(bytes.split().freeze(), b"\r\ncontent-length: 5909\r\n"[..]);
|
||||||
|
bytes.reserve(50);
|
||||||
|
write_content_length(9999, &mut bytes);
|
||||||
|
assert_eq!(bytes.split().freeze(), b"\r\ncontent-length: 9999\r\n"[..]);
|
||||||
|
bytes.reserve(50);
|
||||||
|
write_content_length(10001, &mut bytes);
|
||||||
|
assert_eq!(bytes.split().freeze(), b"\r\ncontent-length: 10001\r\n"[..]);
|
||||||
|
bytes.reserve(50);
|
||||||
|
write_content_length(59094, &mut bytes);
|
||||||
|
assert_eq!(bytes.split().freeze(), b"\r\ncontent-length: 59094\r\n"[..]);
|
||||||
|
bytes.reserve(50);
|
||||||
|
write_content_length(99999, &mut bytes);
|
||||||
|
assert_eq!(bytes.split().freeze(), b"\r\ncontent-length: 99999\r\n"[..]);
|
||||||
|
|
||||||
|
bytes.reserve(50);
|
||||||
|
write_content_length(590947, &mut bytes);
|
||||||
|
assert_eq!(
|
||||||
|
bytes.split().freeze(),
|
||||||
|
b"\r\ncontent-length: 590947\r\n"[..]
|
||||||
|
);
|
||||||
|
bytes.reserve(50);
|
||||||
|
write_content_length(999999, &mut bytes);
|
||||||
|
assert_eq!(
|
||||||
|
bytes.split().freeze(),
|
||||||
|
b"\r\ncontent-length: 999999\r\n"[..]
|
||||||
|
);
|
||||||
|
bytes.reserve(50);
|
||||||
|
write_content_length(5909471, &mut bytes);
|
||||||
|
assert_eq!(
|
||||||
|
bytes.split().freeze(),
|
||||||
|
b"\r\ncontent-length: 5909471\r\n"[..]
|
||||||
|
);
|
||||||
|
bytes.reserve(50);
|
||||||
|
write_content_length(59094718, &mut bytes);
|
||||||
|
assert_eq!(
|
||||||
|
bytes.split().freeze(),
|
||||||
|
b"\r\ncontent-length: 59094718\r\n"[..]
|
||||||
|
);
|
||||||
|
bytes.reserve(50);
|
||||||
|
write_content_length(4294973728, &mut bytes);
|
||||||
|
assert_eq!(
|
||||||
|
bytes.split().freeze(),
|
||||||
|
b"\r\ncontent-length: 4294973728\r\n"[..]
|
||||||
|
);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -10,6 +10,9 @@
|
|||||||
#[macro_use]
|
#[macro_use]
|
||||||
extern crate log;
|
extern crate log;
|
||||||
|
|
||||||
|
#[macro_use]
|
||||||
|
mod macros;
|
||||||
|
|
||||||
pub mod body;
|
pub mod body;
|
||||||
mod builder;
|
mod builder;
|
||||||
pub mod client;
|
pub mod client;
|
||||||
@ -27,6 +30,7 @@ mod payload;
|
|||||||
mod request;
|
mod request;
|
||||||
mod response;
|
mod response;
|
||||||
mod service;
|
mod service;
|
||||||
|
mod time_parser;
|
||||||
|
|
||||||
pub mod cookie;
|
pub mod cookie;
|
||||||
pub mod error;
|
pub mod error;
|
||||||
|
95
actix-http/src/macros.rs
Normal file
95
actix-http/src/macros.rs
Normal file
@ -0,0 +1,95 @@
|
|||||||
|
#[macro_export]
|
||||||
|
macro_rules! downcast_get_type_id {
|
||||||
|
() => {
|
||||||
|
/// A helper method to get the type ID of the type
|
||||||
|
/// this trait is implemented on.
|
||||||
|
/// This method is unsafe to *implement*, since `downcast_ref` relies
|
||||||
|
/// on the returned `TypeId` to perform a cast.
|
||||||
|
///
|
||||||
|
/// Unfortunately, Rust has no notion of a trait method that is
|
||||||
|
/// unsafe to implement (marking it as `unsafe` makes it unsafe
|
||||||
|
/// to *call*). As a workaround, we require this method
|
||||||
|
/// to return a private type along with the `TypeId`. This
|
||||||
|
/// private type (`PrivateHelper`) has a private constructor,
|
||||||
|
/// making it impossible for safe code to construct outside of
|
||||||
|
/// this module. This ensures that safe code cannot violate
|
||||||
|
/// type-safety by implementing this method.
|
||||||
|
#[doc(hidden)]
|
||||||
|
fn __private_get_type_id__(&self) -> (std::any::TypeId, PrivateHelper)
|
||||||
|
where
|
||||||
|
Self: 'static,
|
||||||
|
{
|
||||||
|
(std::any::TypeId::of::<Self>(), PrivateHelper(()))
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
//Generate implementation for dyn $name
|
||||||
|
#[macro_export]
|
||||||
|
macro_rules! downcast {
|
||||||
|
($name:ident) => {
|
||||||
|
/// A struct with a private constructor, for use with
|
||||||
|
/// `__private_get_type_id__`. Its single field is private,
|
||||||
|
/// ensuring that it can only be constructed from this module
|
||||||
|
#[doc(hidden)]
|
||||||
|
pub struct PrivateHelper(());
|
||||||
|
|
||||||
|
impl dyn $name + 'static {
|
||||||
|
/// Downcasts generic body to a specific type.
|
||||||
|
pub fn downcast_ref<T: $name + 'static>(&self) -> Option<&T> {
|
||||||
|
if self.__private_get_type_id__().0 == std::any::TypeId::of::<T>() {
|
||||||
|
// Safety: external crates cannot override the default
|
||||||
|
// implementation of `__private_get_type_id__`, since
|
||||||
|
// it requires returning a private type. We can therefore
|
||||||
|
// rely on the returned `TypeId`, which ensures that this
|
||||||
|
// case is correct.
|
||||||
|
unsafe { Some(&*(self as *const dyn $name as *const T)) }
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
/// Downcasts a generic body to a mutable specific type.
|
||||||
|
pub fn downcast_mut<T: $name + 'static>(&mut self) -> Option<&mut T> {
|
||||||
|
if self.__private_get_type_id__().0 == std::any::TypeId::of::<T>() {
|
||||||
|
// Safety: external crates cannot override the default
|
||||||
|
// implementation of `__private_get_type_id__`, since
|
||||||
|
// it requires returning a private type. We can therefore
|
||||||
|
// rely on the returned `TypeId`, which ensures that this
|
||||||
|
// case is correct.
|
||||||
|
unsafe {
|
||||||
|
Some(&mut *(self as *const dyn $name as *const T as *mut T))
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
|
||||||
|
trait MB {
|
||||||
|
downcast_get_type_id!();
|
||||||
|
}
|
||||||
|
|
||||||
|
downcast!(MB);
|
||||||
|
|
||||||
|
impl MB for String {}
|
||||||
|
impl MB for () {}
|
||||||
|
|
||||||
|
#[actix_rt::test]
|
||||||
|
async fn test_any_casting() {
|
||||||
|
let mut body = String::from("hello cast");
|
||||||
|
let resp_body: &mut dyn MB = &mut body;
|
||||||
|
let body = resp_body.downcast_ref::<String>().unwrap();
|
||||||
|
assert_eq!(body, "hello cast");
|
||||||
|
let body = &mut resp_body.downcast_mut::<String>().unwrap();
|
||||||
|
body.push_str("!");
|
||||||
|
let body = resp_body.downcast_ref::<String>().unwrap();
|
||||||
|
assert_eq!(body, "hello cast!");
|
||||||
|
let not_body = resp_body.downcast_ref::<()>();
|
||||||
|
assert!(not_body.is_none());
|
||||||
|
}
|
||||||
|
}
|
@ -9,7 +9,6 @@ use std::{fmt, str};
|
|||||||
use bytes::{Bytes, BytesMut};
|
use bytes::{Bytes, BytesMut};
|
||||||
use futures_core::Stream;
|
use futures_core::Stream;
|
||||||
use serde::Serialize;
|
use serde::Serialize;
|
||||||
use serde_json;
|
|
||||||
|
|
||||||
use crate::body::{Body, BodyStream, MessageBody, ResponseBody};
|
use crate::body::{Body, BodyStream, MessageBody, ResponseBody};
|
||||||
use crate::cookie::{Cookie, CookieJar};
|
use crate::cookie::{Cookie, CookieJar};
|
||||||
@ -637,7 +636,7 @@ impl ResponseBuilder {
|
|||||||
/// `ResponseBuilder` can not be used after this call.
|
/// `ResponseBuilder` can not be used after this call.
|
||||||
pub fn streaming<S, E>(&mut self, stream: S) -> Response
|
pub fn streaming<S, E>(&mut self, stream: S) -> Response
|
||||||
where
|
where
|
||||||
S: Stream<Item = Result<Bytes, E>> + 'static,
|
S: Stream<Item = Result<Bytes, E>> + Unpin + 'static,
|
||||||
E: Into<Error> + 'static,
|
E: Into<Error> + 'static,
|
||||||
{
|
{
|
||||||
self.body(Body::from_message(BodyStream::new(stream)))
|
self.body(Body::from_message(BodyStream::new(stream)))
|
||||||
|
@ -443,7 +443,7 @@ where
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// `Service` implementation for http transport
|
/// `Service` implementation for http transport
|
||||||
pub struct HttpServiceHandler<T, S, B, X, U> {
|
pub struct HttpServiceHandler<T, S: Service, B, X: Service, U: Service> {
|
||||||
srv: CloneableService<S>,
|
srv: CloneableService<S>,
|
||||||
expect: CloneableService<X>,
|
expect: CloneableService<X>,
|
||||||
upgrade: Option<CloneableService<U>>,
|
upgrade: Option<CloneableService<U>>,
|
||||||
|
42
actix-http/src/time_parser.rs
Normal file
42
actix-http/src/time_parser.rs
Normal file
@ -0,0 +1,42 @@
|
|||||||
|
use time::{Date, OffsetDateTime, PrimitiveDateTime};
|
||||||
|
|
||||||
|
/// Attempt to parse a `time` string as one of either RFC 1123, RFC 850, or asctime.
|
||||||
|
pub fn parse_http_date(time: &str) -> Option<PrimitiveDateTime> {
|
||||||
|
try_parse_rfc_1123(time)
|
||||||
|
.or_else(|| try_parse_rfc_850(time))
|
||||||
|
.or_else(|| try_parse_asctime(time))
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Attempt to parse a `time` string as a RFC 1123 formatted date time string.
|
||||||
|
fn try_parse_rfc_1123(time: &str) -> Option<PrimitiveDateTime> {
|
||||||
|
time::parse(time, "%a, %d %b %Y %H:%M:%S").ok()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Attempt to parse a `time` string as a RFC 850 formatted date time string.
|
||||||
|
fn try_parse_rfc_850(time: &str) -> Option<PrimitiveDateTime> {
|
||||||
|
match PrimitiveDateTime::parse(time, "%A, %d-%b-%y %H:%M:%S") {
|
||||||
|
Ok(dt) => {
|
||||||
|
// If the `time` string contains a two-digit year, then as per RFC 2616 § 19.3,
|
||||||
|
// we consider the year as part of this century if it's within the next 50 years,
|
||||||
|
// otherwise we consider as part of the previous century.
|
||||||
|
let now = OffsetDateTime::now();
|
||||||
|
let century_start_year = (now.year() / 100) * 100;
|
||||||
|
let mut expanded_year = century_start_year + dt.year();
|
||||||
|
|
||||||
|
if expanded_year > now.year() + 50 {
|
||||||
|
expanded_year -= 100;
|
||||||
|
}
|
||||||
|
|
||||||
|
match Date::try_from_ymd(expanded_year, dt.month(), dt.day()) {
|
||||||
|
Ok(date) => Some(PrimitiveDateTime::new(date, dt.time())),
|
||||||
|
Err(_) => None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Err(_) => None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Attempt to parse a `time` string using ANSI C's `asctime` format.
|
||||||
|
fn try_parse_asctime(time: &str) -> Option<PrimitiveDateTime> {
|
||||||
|
time::parse(time, "%a %b %_d %H:%M:%S %Y").ok()
|
||||||
|
}
|
@ -2,7 +2,6 @@ use std::convert::TryFrom;
|
|||||||
|
|
||||||
use bytes::{Buf, BufMut, BytesMut};
|
use bytes::{Buf, BufMut, BytesMut};
|
||||||
use log::debug;
|
use log::debug;
|
||||||
use rand;
|
|
||||||
|
|
||||||
use crate::ws::mask::apply_mask;
|
use crate::ws::mask::apply_mask;
|
||||||
use crate::ws::proto::{CloseCode, CloseReason, OpCode};
|
use crate::ws::proto::{CloseCode, CloseReason, OpCode};
|
||||||
|
@ -1,5 +1,3 @@
|
|||||||
use base64;
|
|
||||||
use sha1;
|
|
||||||
use std::convert::{From, Into};
|
use std::convert::{From, Into};
|
||||||
use std::fmt;
|
use std::fmt;
|
||||||
|
|
||||||
@ -207,12 +205,13 @@ static WS_GUID: &str = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
|
|||||||
|
|
||||||
// TODO: hash is always same size, we dont need String
|
// TODO: hash is always same size, we dont need String
|
||||||
pub fn hash_key(key: &[u8]) -> String {
|
pub fn hash_key(key: &[u8]) -> String {
|
||||||
|
use sha1::Digest;
|
||||||
let mut hasher = sha1::Sha1::new();
|
let mut hasher = sha1::Sha1::new();
|
||||||
|
|
||||||
hasher.update(key);
|
hasher.input(key);
|
||||||
hasher.update(WS_GUID.as_bytes());
|
hasher.input(WS_GUID.as_bytes());
|
||||||
|
|
||||||
base64::encode(&hasher.digest().bytes())
|
base64::encode(hasher.result().as_ref())
|
||||||
}
|
}
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
@ -277,6 +276,12 @@ mod test {
|
|||||||
assert_eq!(format!("{}", OpCode::Bad), "BAD");
|
assert_eq!(format!("{}", OpCode::Bad), "BAD");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_hash_key() {
|
||||||
|
let hash = hash_key(b"hello actix-web");
|
||||||
|
assert_eq!(&hash, "cR1dlyUUJKp0s/Bel25u5TgvC3E=");
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn closecode_from_u16() {
|
fn closecode_from_u16() {
|
||||||
assert_eq!(CloseCode::from(1000u16), CloseCode::Normal);
|
assert_eq!(CloseCode::from(1000u16), CloseCode::Normal);
|
||||||
|
@ -97,11 +97,9 @@ async fn test_h2_body() -> io::Result<()> {
|
|||||||
let data = "HELLOWORLD".to_owned().repeat(64 * 1024);
|
let data = "HELLOWORLD".to_owned().repeat(64 * 1024);
|
||||||
let mut srv = test_server(move || {
|
let mut srv = test_server(move || {
|
||||||
HttpService::build()
|
HttpService::build()
|
||||||
.h2(|mut req: Request<_>| {
|
.h2(|mut req: Request<_>| async move {
|
||||||
async move {
|
let body = load_body(req.take_payload()).await?;
|
||||||
let body = load_body(req.take_payload()).await?;
|
Ok::<_, Error>(Response::Ok().body(body))
|
||||||
Ok::<_, Error>(Response::Ok().body(body))
|
|
||||||
}
|
|
||||||
})
|
})
|
||||||
.openssl(ssl_acceptor())
|
.openssl(ssl_acceptor())
|
||||||
.map_err(|_| ())
|
.map_err(|_| ())
|
||||||
|
@ -104,11 +104,9 @@ async fn test_h2_body1() -> io::Result<()> {
|
|||||||
let data = "HELLOWORLD".to_owned().repeat(64 * 1024);
|
let data = "HELLOWORLD".to_owned().repeat(64 * 1024);
|
||||||
let mut srv = test_server(move || {
|
let mut srv = test_server(move || {
|
||||||
HttpService::build()
|
HttpService::build()
|
||||||
.h2(|mut req: Request<_>| {
|
.h2(|mut req: Request<_>| async move {
|
||||||
async move {
|
let body = load_body(req.take_payload()).await?;
|
||||||
let body = load_body(req.take_payload()).await?;
|
Ok::<_, Error>(Response::Ok().body(body))
|
||||||
Ok::<_, Error>(Response::Ok().body(body))
|
|
||||||
}
|
|
||||||
})
|
})
|
||||||
.rustls(ssl_acceptor())
|
.rustls(ssl_acceptor())
|
||||||
});
|
});
|
||||||
|
@ -1,5 +1,13 @@
|
|||||||
# Changes
|
# Changes
|
||||||
|
|
||||||
|
## [Unreleased] - 2020-xx-xx
|
||||||
|
|
||||||
|
* Update the `time` dependency to 0.2.5
|
||||||
|
|
||||||
|
## [0.2.1] - 2020-01-10
|
||||||
|
|
||||||
|
* Fix panic with already borrowed: BorrowMutError #1263
|
||||||
|
|
||||||
## [0.2.0] - 2019-12-20
|
## [0.2.0] - 2019-12-20
|
||||||
|
|
||||||
* Use actix-web 2.0
|
* Use actix-web 2.0
|
||||||
|
@ -1,6 +1,6 @@
|
|||||||
[package]
|
[package]
|
||||||
name = "actix-identity"
|
name = "actix-identity"
|
||||||
version = "0.2.0"
|
version = "0.2.1"
|
||||||
authors = ["Nikolay Kim <fafhrd91@gmail.com>"]
|
authors = ["Nikolay Kim <fafhrd91@gmail.com>"]
|
||||||
description = "Identity service for actix web framework."
|
description = "Identity service for actix web framework."
|
||||||
readme = "README.md"
|
readme = "README.md"
|
||||||
@ -10,21 +10,20 @@ repository = "https://github.com/actix/actix-web.git"
|
|||||||
documentation = "https://docs.rs/actix-identity/"
|
documentation = "https://docs.rs/actix-identity/"
|
||||||
license = "MIT/Apache-2.0"
|
license = "MIT/Apache-2.0"
|
||||||
edition = "2018"
|
edition = "2018"
|
||||||
workspace = ".."
|
|
||||||
|
|
||||||
[lib]
|
[lib]
|
||||||
name = "actix_identity"
|
name = "actix_identity"
|
||||||
path = "src/lib.rs"
|
path = "src/lib.rs"
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
actix-web = { version = "2.0.0-rc", default-features = false, features = ["secure-cookies"] }
|
actix-web = { version = "2.0.0", default-features = false, features = ["secure-cookies"] }
|
||||||
actix-service = "1.0.1"
|
actix-service = "1.0.5"
|
||||||
futures = "0.3.1"
|
futures = "0.3.1"
|
||||||
serde = "1.0"
|
serde = "1.0"
|
||||||
serde_json = "1.0"
|
serde_json = "1.0"
|
||||||
time = "0.1.42"
|
time = { version = "0.2.5", default-features = false, features = ["std"] }
|
||||||
|
|
||||||
[dev-dependencies]
|
[dev-dependencies]
|
||||||
actix-rt = "1.0.0"
|
actix-rt = "1.0.0"
|
||||||
actix-http = "1.0.1"
|
actix-http = "1.0.1"
|
||||||
bytes = "0.5.3"
|
bytes = "0.5.4"
|
||||||
|
@ -251,6 +251,15 @@ pub struct IdentityServiceMiddleware<S, T> {
|
|||||||
service: Rc<RefCell<S>>,
|
service: Rc<RefCell<S>>,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl<S, T> Clone for IdentityServiceMiddleware<S, T> {
|
||||||
|
fn clone(&self) -> Self {
|
||||||
|
Self {
|
||||||
|
backend: self.backend.clone(),
|
||||||
|
service: self.service.clone(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
impl<S, T, B> Service for IdentityServiceMiddleware<S, T>
|
impl<S, T, B> Service for IdentityServiceMiddleware<S, T>
|
||||||
where
|
where
|
||||||
B: 'static,
|
B: 'static,
|
||||||
@ -279,7 +288,9 @@ where
|
|||||||
req.extensions_mut()
|
req.extensions_mut()
|
||||||
.insert(IdentityItem { id, changed: false });
|
.insert(IdentityItem { id, changed: false });
|
||||||
|
|
||||||
let mut res = srv.borrow_mut().call(req).await?;
|
// https://github.com/actix/actix-web/issues/1263
|
||||||
|
let fut = { srv.borrow_mut().call(req) };
|
||||||
|
let mut res = fut.await?;
|
||||||
let id = res.request().extensions_mut().remove::<IdentityItem>();
|
let id = res.request().extensions_mut().remove::<IdentityItem>();
|
||||||
|
|
||||||
if let Some(id) = id {
|
if let Some(id) = id {
|
||||||
@ -417,14 +428,14 @@ impl CookieIdentityInner {
|
|||||||
let now = SystemTime::now();
|
let now = SystemTime::now();
|
||||||
if let Some(visit_deadline) = self.visit_deadline {
|
if let Some(visit_deadline) = self.visit_deadline {
|
||||||
if now.duration_since(value.visit_timestamp?).ok()?
|
if now.duration_since(value.visit_timestamp?).ok()?
|
||||||
> visit_deadline.to_std().ok()?
|
> visit_deadline
|
||||||
{
|
{
|
||||||
return None;
|
return None;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if let Some(login_deadline) = self.login_deadline {
|
if let Some(login_deadline) = self.login_deadline {
|
||||||
if now.duration_since(value.login_timestamp?).ok()?
|
if now.duration_since(value.login_timestamp?).ok()?
|
||||||
> login_deadline.to_std().ok()?
|
> login_deadline
|
||||||
{
|
{
|
||||||
return None;
|
return None;
|
||||||
}
|
}
|
||||||
@ -606,9 +617,10 @@ mod tests {
|
|||||||
use std::borrow::Borrow;
|
use std::borrow::Borrow;
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
|
use actix_service::into_service;
|
||||||
use actix_web::http::StatusCode;
|
use actix_web::http::StatusCode;
|
||||||
use actix_web::test::{self, TestRequest};
|
use actix_web::test::{self, TestRequest};
|
||||||
use actix_web::{web, App, Error, HttpResponse};
|
use actix_web::{error, web, App, Error, HttpResponse};
|
||||||
|
|
||||||
const COOKIE_KEY_MASTER: [u8; 32] = [0; 32];
|
const COOKIE_KEY_MASTER: [u8; 32] = [0; 32];
|
||||||
const COOKIE_NAME: &'static str = "actix_auth";
|
const COOKIE_NAME: &'static str = "actix_auth";
|
||||||
@ -843,7 +855,7 @@ mod tests {
|
|||||||
let cv: CookieValue = serde_json::from_str(cookie.value()).unwrap();
|
let cv: CookieValue = serde_json::from_str(cookie.value()).unwrap();
|
||||||
assert_eq!(cv.identity, identity);
|
assert_eq!(cv.identity, identity);
|
||||||
let now = SystemTime::now();
|
let now = SystemTime::now();
|
||||||
let t30sec_ago = now - Duration::seconds(30).to_std().unwrap();
|
let t30sec_ago = now - Duration::seconds(30);
|
||||||
match login_timestamp {
|
match login_timestamp {
|
||||||
LoginTimestampCheck::NoTimestamp => assert_eq!(cv.login_timestamp, None),
|
LoginTimestampCheck::NoTimestamp => assert_eq!(cv.login_timestamp, None),
|
||||||
LoginTimestampCheck::NewTimestamp => assert!(
|
LoginTimestampCheck::NewTimestamp => assert!(
|
||||||
@ -985,7 +997,7 @@ mod tests {
|
|||||||
create_identity_server(|c| c.login_deadline(Duration::days(90))).await;
|
create_identity_server(|c| c.login_deadline(Duration::days(90))).await;
|
||||||
let cookie = login_cookie(
|
let cookie = login_cookie(
|
||||||
COOKIE_LOGIN,
|
COOKIE_LOGIN,
|
||||||
Some(SystemTime::now() - Duration::days(180).to_std().unwrap()),
|
Some(SystemTime::now() - Duration::days(180)),
|
||||||
None,
|
None,
|
||||||
);
|
);
|
||||||
let mut resp = test::call_service(
|
let mut resp = test::call_service(
|
||||||
@ -1011,7 +1023,7 @@ mod tests {
|
|||||||
let cookie = login_cookie(
|
let cookie = login_cookie(
|
||||||
COOKIE_LOGIN,
|
COOKIE_LOGIN,
|
||||||
None,
|
None,
|
||||||
Some(SystemTime::now() - Duration::days(180).to_std().unwrap()),
|
Some(SystemTime::now() - Duration::days(180)),
|
||||||
);
|
);
|
||||||
let mut resp = test::call_service(
|
let mut resp = test::call_service(
|
||||||
&mut srv,
|
&mut srv,
|
||||||
@ -1045,6 +1057,7 @@ mod tests {
|
|||||||
assert_logged_in(resp, Some(COOKIE_LOGIN)).await;
|
assert_logged_in(resp, Some(COOKIE_LOGIN)).await;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// https://github.com/actix/actix-web/issues/1263
|
||||||
#[actix_rt::test]
|
#[actix_rt::test]
|
||||||
async fn test_identity_cookie_updated_on_visit_deadline() {
|
async fn test_identity_cookie_updated_on_visit_deadline() {
|
||||||
let mut srv = create_identity_server(|c| {
|
let mut srv = create_identity_server(|c| {
|
||||||
@ -1052,7 +1065,7 @@ mod tests {
|
|||||||
.login_deadline(Duration::days(90))
|
.login_deadline(Duration::days(90))
|
||||||
})
|
})
|
||||||
.await;
|
.await;
|
||||||
let timestamp = SystemTime::now() - Duration::days(1).to_std().unwrap();
|
let timestamp = SystemTime::now() - Duration::days(1);
|
||||||
let cookie = login_cookie(COOKIE_LOGIN, Some(timestamp), Some(timestamp));
|
let cookie = login_cookie(COOKIE_LOGIN, Some(timestamp), Some(timestamp));
|
||||||
let mut resp = test::call_service(
|
let mut resp = test::call_service(
|
||||||
&mut srv,
|
&mut srv,
|
||||||
@ -1069,4 +1082,47 @@ mod tests {
|
|||||||
);
|
);
|
||||||
assert_logged_in(resp, Some(COOKIE_LOGIN)).await;
|
assert_logged_in(resp, Some(COOKIE_LOGIN)).await;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[actix_rt::test]
|
||||||
|
async fn test_borrowed_mut_error() {
|
||||||
|
use futures::future::{lazy, ok, Ready};
|
||||||
|
|
||||||
|
struct Ident;
|
||||||
|
impl IdentityPolicy for Ident {
|
||||||
|
type Future = Ready<Result<Option<String>, Error>>;
|
||||||
|
type ResponseFuture = Ready<Result<(), Error>>;
|
||||||
|
|
||||||
|
fn from_request(&self, _: &mut ServiceRequest) -> Self::Future {
|
||||||
|
ok(Some("test".to_string()))
|
||||||
|
}
|
||||||
|
|
||||||
|
fn to_response<B>(
|
||||||
|
&self,
|
||||||
|
_: Option<String>,
|
||||||
|
_: bool,
|
||||||
|
_: &mut ServiceResponse<B>,
|
||||||
|
) -> Self::ResponseFuture {
|
||||||
|
ok(())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut srv = IdentityServiceMiddleware {
|
||||||
|
backend: Rc::new(Ident),
|
||||||
|
service: Rc::new(RefCell::new(into_service(|_: ServiceRequest| {
|
||||||
|
async move {
|
||||||
|
actix_rt::time::delay_for(std::time::Duration::from_secs(100)).await;
|
||||||
|
Err::<ServiceResponse, _>(error::ErrorBadRequest("error"))
|
||||||
|
}
|
||||||
|
}))),
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut srv2 = srv.clone();
|
||||||
|
let req = TestRequest::default().to_srv_request();
|
||||||
|
actix_rt::spawn(async move {
|
||||||
|
let _ = srv2.call(req).await;
|
||||||
|
});
|
||||||
|
actix_rt::time::delay_for(std::time::Duration::from_millis(50)).await;
|
||||||
|
|
||||||
|
let _ = lazy(|cx| srv.poll_ready(cx)).await;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
@ -1,5 +1,9 @@
|
|||||||
# Changes
|
# Changes
|
||||||
|
|
||||||
|
## [0.2.1] - 2020-01-xx
|
||||||
|
|
||||||
|
* Remove the unused `time` dependency
|
||||||
|
|
||||||
## [0.2.0] - 2019-12-20
|
## [0.2.0] - 2019-12-20
|
||||||
|
|
||||||
* Release
|
* Release
|
||||||
@ -44,4 +48,4 @@
|
|||||||
|
|
||||||
* Split multipart support to separate crate
|
* Split multipart support to separate crate
|
||||||
|
|
||||||
* Optimize multipart handling #634, #769
|
* Optimize multipart handling #634, #769
|
||||||
|
@ -16,7 +16,7 @@ name = "actix_multipart"
|
|||||||
path = "src/lib.rs"
|
path = "src/lib.rs"
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
actix-web = { version = "2.0.0-rc", default-features = false }
|
actix-web = { version = "2.0.0", default-features = false }
|
||||||
actix-service = "1.0.1"
|
actix-service = "1.0.1"
|
||||||
actix-utils = "1.0.3"
|
actix-utils = "1.0.3"
|
||||||
bytes = "0.5.3"
|
bytes = "0.5.3"
|
||||||
@ -25,9 +25,8 @@ httparse = "1.3"
|
|||||||
futures = "0.3.1"
|
futures = "0.3.1"
|
||||||
log = "0.4"
|
log = "0.4"
|
||||||
mime = "0.3"
|
mime = "0.3"
|
||||||
time = "0.1"
|
|
||||||
twoway = "0.2"
|
twoway = "0.2"
|
||||||
|
|
||||||
[dev-dependencies]
|
[dev-dependencies]
|
||||||
actix-rt = "1.0.0"
|
actix-rt = "1.0.0"
|
||||||
actix-http = "1.0.0"
|
actix-http = "2.0.0-alpha.1"
|
||||||
|
@ -1,5 +1,10 @@
|
|||||||
# Changes
|
# Changes
|
||||||
|
|
||||||
|
## [Unreleased] - 2020-01-xx
|
||||||
|
|
||||||
|
* Update the `time` dependency to 0.2.5
|
||||||
|
* [#1292](https://github.com/actix/actix-web/pull/1292) Long lasting auto-prolonged session
|
||||||
|
|
||||||
## [0.3.0] - 2019-12-20
|
## [0.3.0] - 2019-12-20
|
||||||
|
|
||||||
* Release
|
* Release
|
||||||
|
@ -22,14 +22,14 @@ default = ["cookie-session"]
|
|||||||
cookie-session = ["actix-web/secure-cookies"]
|
cookie-session = ["actix-web/secure-cookies"]
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
actix-web = "2.0.0-rc"
|
actix-web = { version = "2.0.0" }
|
||||||
actix-service = "1.0.1"
|
actix-service = "1.0.5"
|
||||||
bytes = "0.5.3"
|
bytes = "0.5.4"
|
||||||
derive_more = "0.99.2"
|
derive_more = "0.99.2"
|
||||||
futures = "0.3.1"
|
futures = "0.3.1"
|
||||||
serde = "1.0"
|
serde = "1.0"
|
||||||
serde_json = "1.0"
|
serde_json = "1.0"
|
||||||
time = "0.1.42"
|
time = { version = "0.2.5", default-features = false, features = ["std"] }
|
||||||
|
|
||||||
[dev-dependencies]
|
[dev-dependencies]
|
||||||
actix-rt = "1.0.0"
|
actix-rt = "1.0.0"
|
||||||
|
@ -27,6 +27,7 @@ use actix_web::{Error, HttpMessage, ResponseError};
|
|||||||
use derive_more::{Display, From};
|
use derive_more::{Display, From};
|
||||||
use futures::future::{ok, FutureExt, LocalBoxFuture, Ready};
|
use futures::future::{ok, FutureExt, LocalBoxFuture, Ready};
|
||||||
use serde_json::error::Error as JsonError;
|
use serde_json::error::Error as JsonError;
|
||||||
|
use time::{Duration, OffsetDateTime};
|
||||||
|
|
||||||
use crate::{Session, SessionStatus};
|
use crate::{Session, SessionStatus};
|
||||||
|
|
||||||
@ -56,7 +57,8 @@ struct CookieSessionInner {
|
|||||||
domain: Option<String>,
|
domain: Option<String>,
|
||||||
secure: bool,
|
secure: bool,
|
||||||
http_only: bool,
|
http_only: bool,
|
||||||
max_age: Option<time::Duration>,
|
max_age: Option<Duration>,
|
||||||
|
expires_in: Option<Duration>,
|
||||||
same_site: Option<SameSite>,
|
same_site: Option<SameSite>,
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -71,6 +73,7 @@ impl CookieSessionInner {
|
|||||||
secure: true,
|
secure: true,
|
||||||
http_only: true,
|
http_only: true,
|
||||||
max_age: None,
|
max_age: None,
|
||||||
|
expires_in: None,
|
||||||
same_site: None,
|
same_site: None,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -96,6 +99,10 @@ impl CookieSessionInner {
|
|||||||
cookie.set_domain(domain.clone());
|
cookie.set_domain(domain.clone());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if let Some(expires_in) = self.expires_in {
|
||||||
|
cookie.set_expires(OffsetDateTime::now() + expires_in);
|
||||||
|
}
|
||||||
|
|
||||||
if let Some(max_age) = self.max_age {
|
if let Some(max_age) = self.max_age {
|
||||||
cookie.set_max_age(max_age);
|
cookie.set_max_age(max_age);
|
||||||
}
|
}
|
||||||
@ -123,8 +130,8 @@ impl CookieSessionInner {
|
|||||||
fn remove_cookie<B>(&self, res: &mut ServiceResponse<B>) -> Result<(), Error> {
|
fn remove_cookie<B>(&self, res: &mut ServiceResponse<B>) -> Result<(), Error> {
|
||||||
let mut cookie = Cookie::named(self.name.clone());
|
let mut cookie = Cookie::named(self.name.clone());
|
||||||
cookie.set_value("");
|
cookie.set_value("");
|
||||||
cookie.set_max_age(time::Duration::seconds(0));
|
cookie.set_max_age(Duration::zero());
|
||||||
cookie.set_expires(time::now() - time::Duration::days(365));
|
cookie.set_expires(OffsetDateTime::now() - Duration::days(365));
|
||||||
|
|
||||||
let val = HeaderValue::from_str(&cookie.to_string())?;
|
let val = HeaderValue::from_str(&cookie.to_string())?;
|
||||||
res.headers_mut().append(SET_COOKIE, val);
|
res.headers_mut().append(SET_COOKIE, val);
|
||||||
@ -263,7 +270,7 @@ impl CookieSession {
|
|||||||
|
|
||||||
/// Sets the `max-age` field in the session cookie being built.
|
/// Sets the `max-age` field in the session cookie being built.
|
||||||
pub fn max_age(self, seconds: i64) -> CookieSession {
|
pub fn max_age(self, seconds: i64) -> CookieSession {
|
||||||
self.max_age_time(time::Duration::seconds(seconds))
|
self.max_age_time(Duration::seconds(seconds))
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Sets the `max-age` field in the session cookie being built.
|
/// Sets the `max-age` field in the session cookie being built.
|
||||||
@ -271,6 +278,17 @@ impl CookieSession {
|
|||||||
Rc::get_mut(&mut self.0).unwrap().max_age = Some(value);
|
Rc::get_mut(&mut self.0).unwrap().max_age = Some(value);
|
||||||
self
|
self
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Sets the `expires` field in the session cookie being built.
|
||||||
|
pub fn expires_in(self, seconds: i64) -> CookieSession {
|
||||||
|
self.expires_in_time(Duration::seconds(seconds))
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Sets the `expires` field in the session cookie being built.
|
||||||
|
pub fn expires_in_time(mut self, value: Duration) -> CookieSession {
|
||||||
|
Rc::get_mut(&mut self.0).unwrap().expires_in = Some(value);
|
||||||
|
self
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<S, B: 'static> Transform<S> for CookieSession
|
impl<S, B: 'static> Transform<S> for CookieSession
|
||||||
@ -323,6 +341,7 @@ where
|
|||||||
fn call(&mut self, mut req: ServiceRequest) -> Self::Future {
|
fn call(&mut self, mut req: ServiceRequest) -> Self::Future {
|
||||||
let inner = self.inner.clone();
|
let inner = self.inner.clone();
|
||||||
let (is_new, state) = self.inner.load(&req);
|
let (is_new, state) = self.inner.load(&req);
|
||||||
|
let prolong_expiration = self.inner.expires_in.is_some();
|
||||||
Session::set_session(state.into_iter(), &mut req);
|
Session::set_session(state.into_iter(), &mut req);
|
||||||
|
|
||||||
let fut = self.service.call(req);
|
let fut = self.service.call(req);
|
||||||
@ -334,6 +353,9 @@ where
|
|||||||
| (SessionStatus::Renewed, Some(state)) => {
|
| (SessionStatus::Renewed, Some(state)) => {
|
||||||
res.checked_expr(|res| inner.set_cookie(res, state))
|
res.checked_expr(|res| inner.set_cookie(res, state))
|
||||||
}
|
}
|
||||||
|
(SessionStatus::Unchanged, Some(state)) if prolong_expiration => {
|
||||||
|
res.checked_expr(|res| inner.set_cookie(res, state))
|
||||||
|
}
|
||||||
(SessionStatus::Unchanged, _) =>
|
(SessionStatus::Unchanged, _) =>
|
||||||
// set a new session cookie upon first request (new client)
|
// set a new session cookie upon first request (new client)
|
||||||
{
|
{
|
||||||
@ -477,4 +499,47 @@ mod tests {
|
|||||||
let body = test::read_response(&mut app, request).await;
|
let body = test::read_response(&mut app, request).await;
|
||||||
assert_eq!(body, Bytes::from_static(b"counter: 100"));
|
assert_eq!(body, Bytes::from_static(b"counter: 100"));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[actix_rt::test]
|
||||||
|
async fn prolong_expiration() {
|
||||||
|
let mut app = test::init_service(
|
||||||
|
App::new()
|
||||||
|
.wrap(CookieSession::signed(&[0; 32]).secure(false).expires_in(60))
|
||||||
|
.service(web::resource("/").to(|ses: Session| {
|
||||||
|
async move {
|
||||||
|
let _ = ses.set("counter", 100);
|
||||||
|
"test"
|
||||||
|
}
|
||||||
|
}))
|
||||||
|
.service(
|
||||||
|
web::resource("/test/")
|
||||||
|
.to(|| async move { "no-changes-in-session" }),
|
||||||
|
),
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
let request = test::TestRequest::get().to_request();
|
||||||
|
let response = app.call(request).await.unwrap();
|
||||||
|
let expires_1 = response
|
||||||
|
.response()
|
||||||
|
.cookies()
|
||||||
|
.find(|c| c.name() == "actix-session")
|
||||||
|
.expect("Cookie is set")
|
||||||
|
.expires()
|
||||||
|
.expect("Expiration is set");
|
||||||
|
|
||||||
|
actix_rt::time::delay_for(std::time::Duration::from_secs(1)).await;
|
||||||
|
|
||||||
|
let request = test::TestRequest::with_uri("/test/").to_request();
|
||||||
|
let response = app.call(request).await.unwrap();
|
||||||
|
let expires_2 = response
|
||||||
|
.response()
|
||||||
|
.cookies()
|
||||||
|
.find(|c| c.name() == "actix-session")
|
||||||
|
.expect("Cookie is set")
|
||||||
|
.expires()
|
||||||
|
.expect("Expiration is set");
|
||||||
|
|
||||||
|
assert!(expires_2 - expires_1 >= Duration::seconds(1));
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
@ -16,9 +16,9 @@ name = "actix_web_actors"
|
|||||||
path = "src/lib.rs"
|
path = "src/lib.rs"
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
actix = "0.9.0"
|
actix = "0.10.0-alpha.1"
|
||||||
actix-web = "2.0.0-rc"
|
actix-web = "2.0.0"
|
||||||
actix-http = "1.0.1"
|
actix-http = "2.0.0-alpha.1"
|
||||||
actix-codec = "0.2.0"
|
actix-codec = "0.2.0"
|
||||||
bytes = "0.5.2"
|
bytes = "0.5.2"
|
||||||
futures = "0.3.1"
|
futures = "0.3.1"
|
||||||
|
@ -1,5 +1,13 @@
|
|||||||
# Changes
|
# Changes
|
||||||
|
|
||||||
|
## [0.2.1] - 2020-02-25
|
||||||
|
|
||||||
|
* Add `#[allow(missing_docs)]` attribute to generated structs [#1368]
|
||||||
|
* Allow the handler function to be named as `config` [#1290]
|
||||||
|
|
||||||
|
[#1368]: https://github.com/actix/actix-web/issues/1368
|
||||||
|
[#1290]: https://github.com/actix/actix-web/issues/1290
|
||||||
|
|
||||||
## [0.2.0] - 2019-12-13
|
## [0.2.0] - 2019-12-13
|
||||||
|
|
||||||
* Generate code for actix-web 2.0
|
* Generate code for actix-web 2.0
|
||||||
|
@ -1,6 +1,6 @@
|
|||||||
[package]
|
[package]
|
||||||
name = "actix-web-codegen"
|
name = "actix-web-codegen"
|
||||||
version = "0.2.0"
|
version = "0.2.1"
|
||||||
description = "Actix web proc macros"
|
description = "Actix web proc macros"
|
||||||
readme = "README.md"
|
readme = "README.md"
|
||||||
authors = ["Nikolay Kim <fafhrd91@gmail.com>"]
|
authors = ["Nikolay Kim <fafhrd91@gmail.com>"]
|
||||||
@ -17,6 +17,6 @@ syn = { version = "^1", features = ["full", "parsing"] }
|
|||||||
proc-macro2 = "^1"
|
proc-macro2 = "^1"
|
||||||
|
|
||||||
[dev-dependencies]
|
[dev-dependencies]
|
||||||
actix-rt = { version = "1.0.0" }
|
actix-rt = "1.0.0"
|
||||||
actix-web = { version = "2.0.0-rc" }
|
actix-web = "2.0.0"
|
||||||
futures = { version = "0.3.1" }
|
futures = "0.3.1"
|
||||||
|
@ -191,19 +191,19 @@ impl Route {
|
|||||||
let extra_guards = &self.args.guards;
|
let extra_guards = &self.args.guards;
|
||||||
let resource_type = &self.resource_type;
|
let resource_type = &self.resource_type;
|
||||||
let stream = quote! {
|
let stream = quote! {
|
||||||
#[allow(non_camel_case_types)]
|
#[allow(non_camel_case_types, missing_docs)]
|
||||||
pub struct #name;
|
pub struct #name;
|
||||||
|
|
||||||
impl actix_web::dev::HttpServiceFactory for #name {
|
impl actix_web::dev::HttpServiceFactory for #name {
|
||||||
fn register(self, config: &mut actix_web::dev::AppService) {
|
fn register(self, __config: &mut actix_web::dev::AppService) {
|
||||||
#ast
|
#ast
|
||||||
let resource = actix_web::Resource::new(#path)
|
let __resource = actix_web::Resource::new(#path)
|
||||||
.name(#resource_name)
|
.name(#resource_name)
|
||||||
.guard(actix_web::guard::#guard())
|
.guard(actix_web::guard::#guard())
|
||||||
#(.guard(actix_web::guard::fn_guard(#extra_guards)))*
|
#(.guard(actix_web::guard::fn_guard(#extra_guards)))*
|
||||||
.#resource_type(#name);
|
.#resource_type(#name);
|
||||||
|
|
||||||
actix_web::dev::HttpServiceFactory::register(resource, config)
|
actix_web::dev::HttpServiceFactory::register(__resource, __config)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
@ -2,6 +2,12 @@ use actix_web::{http, test, web::Path, App, HttpResponse, Responder};
|
|||||||
use actix_web_codegen::{connect, delete, get, head, options, patch, post, put, trace};
|
use actix_web_codegen::{connect, delete, get, head, options, patch, post, put, trace};
|
||||||
use futures::{future, Future};
|
use futures::{future, Future};
|
||||||
|
|
||||||
|
// Make sure that we can name function as 'config'
|
||||||
|
#[get("/config")]
|
||||||
|
async fn config() -> impl Responder {
|
||||||
|
HttpResponse::Ok()
|
||||||
|
}
|
||||||
|
|
||||||
#[get("/test")]
|
#[get("/test")]
|
||||||
async fn test_handler() -> impl Responder {
|
async fn test_handler() -> impl Responder {
|
||||||
HttpResponse::Ok()
|
HttpResponse::Ok()
|
||||||
|
@ -4,7 +4,6 @@
|
|||||||
|
|
||||||
* Fix compilation with default features off
|
* Fix compilation with default features off
|
||||||
|
|
||||||
|
|
||||||
## [1.0.0] - 2019-12-13
|
## [1.0.0] - 2019-12-13
|
||||||
|
|
||||||
* Release
|
* Release
|
||||||
|
@ -36,7 +36,7 @@ compress = ["actix-http/compress"]
|
|||||||
[dependencies]
|
[dependencies]
|
||||||
actix-codec = "0.2.0"
|
actix-codec = "0.2.0"
|
||||||
actix-service = "1.0.1"
|
actix-service = "1.0.1"
|
||||||
actix-http = "1.0.0"
|
actix-http = "2.0.0-alpha.1"
|
||||||
actix-rt = "1.0.0"
|
actix-rt = "1.0.0"
|
||||||
|
|
||||||
base64 = "0.11"
|
base64 = "0.11"
|
||||||
@ -55,8 +55,8 @@ rust-tls = { version = "0.16.0", package="rustls", optional = true, features = [
|
|||||||
|
|
||||||
[dev-dependencies]
|
[dev-dependencies]
|
||||||
actix-connect = { version = "1.0.1", features=["openssl"] }
|
actix-connect = { version = "1.0.1", features=["openssl"] }
|
||||||
actix-web = { version = "2.0.0-rc", features=["openssl"] }
|
actix-web = { version = "2.0.0", features=["openssl"] }
|
||||||
actix-http = { version = "1.0.1", features=["openssl"] }
|
actix-http = { version = "2.0.0-alpha.1", features=["openssl"] }
|
||||||
actix-http-test = { version = "1.0.0", features=["openssl"] }
|
actix-http-test = { version = "1.0.0", features=["openssl"] }
|
||||||
actix-utils = "1.0.3"
|
actix-utils = "1.0.3"
|
||||||
actix-server = "1.0.0"
|
actix-server = "1.0.0"
|
||||||
|
@ -11,6 +11,7 @@ use futures::future::ok;
|
|||||||
use open_ssl::ssl::{SslAcceptor, SslFiletype, SslMethod, SslVerifyMode};
|
use open_ssl::ssl::{SslAcceptor, SslFiletype, SslMethod, SslVerifyMode};
|
||||||
use rust_tls::ClientConfig;
|
use rust_tls::ClientConfig;
|
||||||
|
|
||||||
|
#[allow(unused)]
|
||||||
fn ssl_acceptor() -> SslAcceptor {
|
fn ssl_acceptor() -> SslAcceptor {
|
||||||
// load ssl keys
|
// load ssl keys
|
||||||
let mut builder = SslAcceptor::mozilla_intermediate(SslMethod::tls()).unwrap();
|
let mut builder = SslAcceptor::mozilla_intermediate(SslMethod::tls()).unwrap();
|
||||||
|
64
benches/server.rs
Normal file
64
benches/server.rs
Normal file
@ -0,0 +1,64 @@
|
|||||||
|
use actix_web::{test, web, App, HttpResponse};
|
||||||
|
use awc::Client;
|
||||||
|
use criterion::{criterion_group, criterion_main, Criterion};
|
||||||
|
use futures::future::join_all;
|
||||||
|
|
||||||
|
const STR: &str = "Hello World Hello World Hello World Hello World Hello World \
|
||||||
|
Hello World Hello World Hello World Hello World Hello World \
|
||||||
|
Hello World Hello World Hello World Hello World Hello World \
|
||||||
|
Hello World Hello World Hello World Hello World Hello World \
|
||||||
|
Hello World Hello World Hello World Hello World Hello World \
|
||||||
|
Hello World Hello World Hello World Hello World Hello World \
|
||||||
|
Hello World Hello World Hello World Hello World Hello World \
|
||||||
|
Hello World Hello World Hello World Hello World Hello World \
|
||||||
|
Hello World Hello World Hello World Hello World Hello World \
|
||||||
|
Hello World Hello World Hello World Hello World Hello World \
|
||||||
|
Hello World Hello World Hello World Hello World Hello World \
|
||||||
|
Hello World Hello World Hello World Hello World Hello World \
|
||||||
|
Hello World Hello World Hello World Hello World Hello World \
|
||||||
|
Hello World Hello World Hello World Hello World Hello World \
|
||||||
|
Hello World Hello World Hello World Hello World Hello World \
|
||||||
|
Hello World Hello World Hello World Hello World Hello World \
|
||||||
|
Hello World Hello World Hello World Hello World Hello World \
|
||||||
|
Hello World Hello World Hello World Hello World Hello World \
|
||||||
|
Hello World Hello World Hello World Hello World Hello World \
|
||||||
|
Hello World Hello World Hello World Hello World Hello World \
|
||||||
|
Hello World Hello World Hello World Hello World Hello World";
|
||||||
|
|
||||||
|
// benchmark sending all requests at the same time
|
||||||
|
fn bench_async_burst(c: &mut Criterion) {
|
||||||
|
let srv = test::start(|| {
|
||||||
|
App::new()
|
||||||
|
.service(web::resource("/").route(web::to(|| HttpResponse::Ok().body(STR))))
|
||||||
|
});
|
||||||
|
|
||||||
|
// We are using System here, since Runtime requires preinitialized tokio
|
||||||
|
// Maybe add to actix_rt docs
|
||||||
|
let url = srv.url("/");
|
||||||
|
let mut rt = actix_rt::System::new("test");
|
||||||
|
|
||||||
|
c.bench_function("get_body_async_burst", move |b| {
|
||||||
|
b.iter_custom(|iters| {
|
||||||
|
let client = Client::new().get(url.clone()).freeze().unwrap();
|
||||||
|
|
||||||
|
let start = std::time::Instant::now();
|
||||||
|
// benchmark body
|
||||||
|
let resps = rt.block_on(async move {
|
||||||
|
let burst = (0..iters).map(|_| client.send());
|
||||||
|
join_all(burst).await
|
||||||
|
});
|
||||||
|
let elapsed = start.elapsed();
|
||||||
|
|
||||||
|
// if there are failed requests that might be an issue
|
||||||
|
let failed = resps.iter().filter(|r| r.is_err()).count();
|
||||||
|
if failed > 0 {
|
||||||
|
eprintln!("failed {} requests (might be bench timeout)", failed);
|
||||||
|
};
|
||||||
|
|
||||||
|
elapsed
|
||||||
|
})
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
criterion_group!(server_benches, bench_async_burst);
|
||||||
|
criterion_main!(server_benches);
|
108
benches/service.rs
Normal file
108
benches/service.rs
Normal file
@ -0,0 +1,108 @@
|
|||||||
|
use actix_service::Service;
|
||||||
|
use actix_web::dev::{ServiceRequest, ServiceResponse};
|
||||||
|
use actix_web::{web, App, Error, HttpResponse};
|
||||||
|
use criterion::{criterion_main, Criterion};
|
||||||
|
use std::cell::RefCell;
|
||||||
|
use std::rc::Rc;
|
||||||
|
|
||||||
|
use actix_web::test::{init_service, ok_service, TestRequest};
|
||||||
|
|
||||||
|
/// Criterion Benchmark for async Service
|
||||||
|
/// Should be used from within criterion group:
|
||||||
|
/// ```rust,ignore
|
||||||
|
/// let mut criterion: ::criterion::Criterion<_> =
|
||||||
|
/// ::criterion::Criterion::default().configure_from_args();
|
||||||
|
/// bench_async_service(&mut criterion, ok_service(), "async_service_direct");
|
||||||
|
/// ```
|
||||||
|
///
|
||||||
|
/// Usable for benching Service wrappers:
|
||||||
|
/// Using minimum service code implementation we first measure
|
||||||
|
/// time to run minimum service, then measure time with wrapper.
|
||||||
|
///
|
||||||
|
/// Sample output
|
||||||
|
/// async_service_direct time: [1.0908 us 1.1656 us 1.2613 us]
|
||||||
|
pub fn bench_async_service<S>(c: &mut Criterion, srv: S, name: &str)
|
||||||
|
where
|
||||||
|
S: Service<Request = ServiceRequest, Response = ServiceResponse, Error = Error>
|
||||||
|
+ 'static,
|
||||||
|
{
|
||||||
|
let mut rt = actix_rt::System::new("test");
|
||||||
|
let srv = Rc::new(RefCell::new(srv));
|
||||||
|
|
||||||
|
let req = TestRequest::default().to_srv_request();
|
||||||
|
assert!(rt
|
||||||
|
.block_on(srv.borrow_mut().call(req))
|
||||||
|
.unwrap()
|
||||||
|
.status()
|
||||||
|
.is_success());
|
||||||
|
|
||||||
|
// start benchmark loops
|
||||||
|
c.bench_function(name, move |b| {
|
||||||
|
b.iter_custom(|iters| {
|
||||||
|
let srv = srv.clone();
|
||||||
|
// exclude request generation, it appears it takes significant time vs call (3us vs 1us)
|
||||||
|
let reqs: Vec<_> = (0..iters)
|
||||||
|
.map(|_| TestRequest::default().to_srv_request())
|
||||||
|
.collect();
|
||||||
|
let start = std::time::Instant::now();
|
||||||
|
// benchmark body
|
||||||
|
rt.block_on(async move {
|
||||||
|
for req in reqs {
|
||||||
|
srv.borrow_mut().call(req).await.unwrap();
|
||||||
|
}
|
||||||
|
});
|
||||||
|
let elapsed = start.elapsed();
|
||||||
|
// check that at least first request succeeded
|
||||||
|
elapsed
|
||||||
|
})
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn index(req: ServiceRequest) -> Result<ServiceResponse, Error> {
|
||||||
|
Ok(req.into_response(HttpResponse::Ok().finish()))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Benchmark basic WebService directly
|
||||||
|
// this approach is usable for benching WebService, though it adds some time to direct service call:
|
||||||
|
// Sample results on MacBook Pro '14
|
||||||
|
// time: [2.0724 us 2.1345 us 2.2074 us]
|
||||||
|
fn async_web_service(c: &mut Criterion) {
|
||||||
|
let mut rt = actix_rt::System::new("test");
|
||||||
|
let srv = Rc::new(RefCell::new(rt.block_on(init_service(
|
||||||
|
App::new().service(web::service("/").finish(index)),
|
||||||
|
))));
|
||||||
|
|
||||||
|
let req = TestRequest::get().uri("/").to_request();
|
||||||
|
assert!(rt
|
||||||
|
.block_on(srv.borrow_mut().call(req))
|
||||||
|
.unwrap()
|
||||||
|
.status()
|
||||||
|
.is_success());
|
||||||
|
|
||||||
|
// start benchmark loops
|
||||||
|
c.bench_function("async_web_service_direct", move |b| {
|
||||||
|
b.iter_custom(|iters| {
|
||||||
|
let srv = srv.clone();
|
||||||
|
let reqs = (0..iters).map(|_| TestRequest::get().uri("/").to_request());
|
||||||
|
|
||||||
|
let start = std::time::Instant::now();
|
||||||
|
// benchmark body
|
||||||
|
rt.block_on(async move {
|
||||||
|
for req in reqs {
|
||||||
|
srv.borrow_mut().call(req).await.unwrap();
|
||||||
|
}
|
||||||
|
});
|
||||||
|
let elapsed = start.elapsed();
|
||||||
|
// check that at least first request succeeded
|
||||||
|
elapsed
|
||||||
|
})
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn service_benches() {
|
||||||
|
let mut criterion: ::criterion::Criterion<_> =
|
||||||
|
::criterion::Criterion::default().configure_from_args();
|
||||||
|
bench_async_service(&mut criterion, ok_service(), "async_service_direct");
|
||||||
|
async_web_service(&mut criterion);
|
||||||
|
}
|
||||||
|
criterion_main!(service_benches);
|
5
codecov.yml
Normal file
5
codecov.yml
Normal file
@ -0,0 +1,5 @@
|
|||||||
|
ignore: # ignore codecoverage on following paths
|
||||||
|
- "**/tests"
|
||||||
|
- "test-server"
|
||||||
|
- "**/benches"
|
||||||
|
- "**/examples"
|
@ -1,6 +1,6 @@
|
|||||||
use actix_web::{
|
use actix_web::{get, web, HttpRequest};
|
||||||
get, middleware, web, App, Error, HttpRequest, HttpResponse, HttpServer,
|
#[cfg(unix)]
|
||||||
};
|
use actix_web::{middleware, App, Error, HttpResponse, HttpServer};
|
||||||
|
|
||||||
#[get("/resource1/{name}/index.html")]
|
#[get("/resource1/{name}/index.html")]
|
||||||
async fn index(req: HttpRequest, name: web::Path<String>) -> String {
|
async fn index(req: HttpRequest, name: web::Path<String>) -> String {
|
||||||
@ -8,6 +8,7 @@ async fn index(req: HttpRequest, name: web::Path<String>) -> String {
|
|||||||
format!("Hello: {}!\r\n", name)
|
format!("Hello: {}!\r\n", name)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[cfg(unix)]
|
||||||
async fn index_async(req: HttpRequest) -> Result<&'static str, Error> {
|
async fn index_async(req: HttpRequest) -> Result<&'static str, Error> {
|
||||||
println!("REQ: {:?}", req);
|
println!("REQ: {:?}", req);
|
||||||
Ok("Hello world!\r\n")
|
Ok("Hello world!\r\n")
|
||||||
|
108
src/extract.rs
108
src/extract.rs
@ -193,57 +193,83 @@ impl FromRequest for () {
|
|||||||
|
|
||||||
macro_rules! tuple_from_req ({$fut_type:ident, $(($n:tt, $T:ident)),+} => {
|
macro_rules! tuple_from_req ({$fut_type:ident, $(($n:tt, $T:ident)),+} => {
|
||||||
|
|
||||||
/// FromRequest implementation for tuple
|
// This module is a trick to get around the inability of
|
||||||
#[doc(hidden)]
|
// `macro_rules!` macros to make new idents. We want to make
|
||||||
#[allow(unused_parens)]
|
// a new `FutWrapper` struct for each distinct invocation of
|
||||||
impl<$($T: FromRequest + 'static),+> FromRequest for ($($T,)+)
|
// this macro. Ideally, we would name it something like
|
||||||
{
|
// `FutWrapper_$fut_type`, but this can't be done in a macro_rules
|
||||||
type Error = Error;
|
// macro.
|
||||||
type Future = $fut_type<$($T),+>;
|
//
|
||||||
type Config = ($($T::Config),+);
|
// Instead, we put everything in a module named `$fut_type`, thus allowing
|
||||||
|
// us to use the name `FutWrapper` without worrying about conflicts.
|
||||||
|
// This macro only exists to generate trait impls for tuples - these
|
||||||
|
// are inherently global, so users don't have to care about this
|
||||||
|
// weird trick.
|
||||||
|
#[allow(non_snake_case)]
|
||||||
|
mod $fut_type {
|
||||||
|
|
||||||
fn from_request(req: &HttpRequest, payload: &mut Payload) -> Self::Future {
|
// Bring everything into scope, so we don't need
|
||||||
$fut_type {
|
// redundant imports
|
||||||
items: <($(Option<$T>,)+)>::default(),
|
use super::*;
|
||||||
futs: ($($T::from_request(req, payload),)+),
|
|
||||||
|
/// A helper struct to allow us to pin-project through
|
||||||
|
/// to individual fields
|
||||||
|
#[pin_project::pin_project]
|
||||||
|
struct FutWrapper<$($T: FromRequest),+>($(#[pin] $T::Future),+);
|
||||||
|
|
||||||
|
/// FromRequest implementation for tuple
|
||||||
|
#[doc(hidden)]
|
||||||
|
#[allow(unused_parens)]
|
||||||
|
impl<$($T: FromRequest + 'static),+> FromRequest for ($($T,)+)
|
||||||
|
{
|
||||||
|
type Error = Error;
|
||||||
|
type Future = $fut_type<$($T),+>;
|
||||||
|
type Config = ($($T::Config),+);
|
||||||
|
|
||||||
|
fn from_request(req: &HttpRequest, payload: &mut Payload) -> Self::Future {
|
||||||
|
$fut_type {
|
||||||
|
items: <($(Option<$T>,)+)>::default(),
|
||||||
|
futs: FutWrapper($($T::from_request(req, payload),)+),
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
#[doc(hidden)]
|
#[doc(hidden)]
|
||||||
#[pin_project::pin_project]
|
#[pin_project::pin_project]
|
||||||
pub struct $fut_type<$($T: FromRequest),+> {
|
pub struct $fut_type<$($T: FromRequest),+> {
|
||||||
items: ($(Option<$T>,)+),
|
items: ($(Option<$T>,)+),
|
||||||
futs: ($($T::Future,)+),
|
#[pin]
|
||||||
}
|
futs: FutWrapper<$($T,)+>,
|
||||||
|
}
|
||||||
|
|
||||||
impl<$($T: FromRequest),+> Future for $fut_type<$($T),+>
|
impl<$($T: FromRequest),+> Future for $fut_type<$($T),+>
|
||||||
{
|
{
|
||||||
type Output = Result<($($T,)+), Error>;
|
type Output = Result<($($T,)+), Error>;
|
||||||
|
|
||||||
fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
|
fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
|
||||||
let this = self.project();
|
let mut this = self.project();
|
||||||
|
|
||||||
let mut ready = true;
|
let mut ready = true;
|
||||||
$(
|
$(
|
||||||
if this.items.$n.is_none() {
|
if this.items.$n.is_none() {
|
||||||
match unsafe { Pin::new_unchecked(&mut this.futs.$n) }.poll(cx) {
|
match this.futs.as_mut().project().$n.poll(cx) {
|
||||||
Poll::Ready(Ok(item)) => {
|
Poll::Ready(Ok(item)) => {
|
||||||
this.items.$n = Some(item);
|
this.items.$n = Some(item);
|
||||||
|
}
|
||||||
|
Poll::Pending => ready = false,
|
||||||
|
Poll::Ready(Err(e)) => return Poll::Ready(Err(e.into())),
|
||||||
}
|
}
|
||||||
Poll::Pending => ready = false,
|
|
||||||
Poll::Ready(Err(e)) => return Poll::Ready(Err(e.into())),
|
|
||||||
}
|
}
|
||||||
}
|
)+
|
||||||
)+
|
|
||||||
|
|
||||||
if ready {
|
if ready {
|
||||||
Poll::Ready(Ok(
|
Poll::Ready(Ok(
|
||||||
($(this.items.$n.take().unwrap(),)+)
|
($(this.items.$n.take().unwrap(),)+)
|
||||||
))
|
))
|
||||||
} else {
|
} else {
|
||||||
Poll::Pending
|
Poll::Pending
|
||||||
}
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
@ -7,6 +7,12 @@
|
|||||||
//! Actix web is a small, pragmatic, and extremely fast web framework
|
//! Actix web is a small, pragmatic, and extremely fast web framework
|
||||||
//! for Rust.
|
//! for Rust.
|
||||||
//!
|
//!
|
||||||
|
//! ## Example
|
||||||
|
//!
|
||||||
|
//! The `#[actix_rt::main]` macro in the example below is provided by the Actix runtime
|
||||||
|
//! crate, [`actix-rt`](https://crates.io/crates/actix-rt). You will need to include
|
||||||
|
//! `actix-rt` in your dependencies for it to run.
|
||||||
|
//!
|
||||||
//! ```rust,no_run
|
//! ```rust,no_run
|
||||||
//! use actix_web::{web, App, Responder, HttpServer};
|
//! use actix_web::{web, App, Responder, HttpServer};
|
||||||
//!
|
//!
|
||||||
@ -47,7 +53,7 @@
|
|||||||
//! configure servers.
|
//! configure servers.
|
||||||
//!
|
//!
|
||||||
//! * [web](web/index.html): This module
|
//! * [web](web/index.html): This module
|
||||||
//! provide essentials helper functions and types for application registration.
|
//! provides essential helper functions and types for application registration.
|
||||||
//!
|
//!
|
||||||
//! * [HttpRequest](struct.HttpRequest.html) and
|
//! * [HttpRequest](struct.HttpRequest.html) and
|
||||||
//! [HttpResponse](struct.HttpResponse.html): These structs
|
//! [HttpResponse](struct.HttpResponse.html): These structs
|
||||||
|
@ -14,7 +14,7 @@ use bytes::Bytes;
|
|||||||
use futures::future::{ok, Ready};
|
use futures::future::{ok, Ready};
|
||||||
use log::debug;
|
use log::debug;
|
||||||
use regex::Regex;
|
use regex::Regex;
|
||||||
use time;
|
use time::OffsetDateTime;
|
||||||
|
|
||||||
use crate::dev::{BodySize, MessageBody, ResponseBody};
|
use crate::dev::{BodySize, MessageBody, ResponseBody};
|
||||||
use crate::error::{Error, Result};
|
use crate::error::{Error, Result};
|
||||||
@ -163,11 +163,11 @@ where
|
|||||||
LoggerResponse {
|
LoggerResponse {
|
||||||
fut: self.service.call(req),
|
fut: self.service.call(req),
|
||||||
format: None,
|
format: None,
|
||||||
time: time::now(),
|
time: OffsetDateTime::now(),
|
||||||
_t: PhantomData,
|
_t: PhantomData,
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
let now = time::now();
|
let now = OffsetDateTime::now();
|
||||||
let mut format = self.inner.format.clone();
|
let mut format = self.inner.format.clone();
|
||||||
|
|
||||||
for unit in &mut format.0 {
|
for unit in &mut format.0 {
|
||||||
@ -192,7 +192,7 @@ where
|
|||||||
{
|
{
|
||||||
#[pin]
|
#[pin]
|
||||||
fut: S::Future,
|
fut: S::Future,
|
||||||
time: time::Tm,
|
time: OffsetDateTime,
|
||||||
format: Option<Format>,
|
format: Option<Format>,
|
||||||
_t: PhantomData<(B,)>,
|
_t: PhantomData<(B,)>,
|
||||||
}
|
}
|
||||||
@ -238,15 +238,20 @@ where
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
use pin_project::{pin_project, pinned_drop};
|
||||||
|
|
||||||
|
#[pin_project(PinnedDrop)]
|
||||||
pub struct StreamLog<B> {
|
pub struct StreamLog<B> {
|
||||||
|
#[pin]
|
||||||
body: ResponseBody<B>,
|
body: ResponseBody<B>,
|
||||||
format: Option<Format>,
|
format: Option<Format>,
|
||||||
size: usize,
|
size: usize,
|
||||||
time: time::Tm,
|
time: OffsetDateTime,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<B> Drop for StreamLog<B> {
|
#[pinned_drop]
|
||||||
fn drop(&mut self) {
|
impl<B> PinnedDrop for StreamLog<B> {
|
||||||
|
fn drop(self: Pin<&mut Self>) {
|
||||||
if let Some(ref format) = self.format {
|
if let Some(ref format) = self.format {
|
||||||
let render = |fmt: &mut Formatter<'_>| {
|
let render = |fmt: &mut Formatter<'_>| {
|
||||||
for unit in &format.0 {
|
for unit in &format.0 {
|
||||||
@ -259,15 +264,17 @@ impl<B> Drop for StreamLog<B> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
impl<B: MessageBody> MessageBody for StreamLog<B> {
|
impl<B: MessageBody> MessageBody for StreamLog<B> {
|
||||||
fn size(&self) -> BodySize {
|
fn size(&self) -> BodySize {
|
||||||
self.body.size()
|
self.body.size()
|
||||||
}
|
}
|
||||||
|
|
||||||
fn poll_next(&mut self, cx: &mut Context<'_>) -> Poll<Option<Result<Bytes, Error>>> {
|
fn poll_next(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<Result<Bytes, Error>>> {
|
||||||
match self.body.poll_next(cx) {
|
let this = self.project();
|
||||||
|
match this.body.poll_next(cx) {
|
||||||
Poll::Ready(Some(Ok(chunk))) => {
|
Poll::Ready(Some(Ok(chunk))) => {
|
||||||
self.size += chunk.len();
|
*this.size += chunk.len();
|
||||||
Poll::Ready(Some(Ok(chunk)))
|
Poll::Ready(Some(Ok(chunk)))
|
||||||
}
|
}
|
||||||
val => val,
|
val => val,
|
||||||
@ -366,20 +373,20 @@ impl FormatText {
|
|||||||
&self,
|
&self,
|
||||||
fmt: &mut Formatter<'_>,
|
fmt: &mut Formatter<'_>,
|
||||||
size: usize,
|
size: usize,
|
||||||
entry_time: time::Tm,
|
entry_time: OffsetDateTime,
|
||||||
) -> Result<(), fmt::Error> {
|
) -> Result<(), fmt::Error> {
|
||||||
match *self {
|
match *self {
|
||||||
FormatText::Str(ref string) => fmt.write_str(string),
|
FormatText::Str(ref string) => fmt.write_str(string),
|
||||||
FormatText::Percent => "%".fmt(fmt),
|
FormatText::Percent => "%".fmt(fmt),
|
||||||
FormatText::ResponseSize => size.fmt(fmt),
|
FormatText::ResponseSize => size.fmt(fmt),
|
||||||
FormatText::Time => {
|
FormatText::Time => {
|
||||||
let rt = time::now() - entry_time;
|
let rt = OffsetDateTime::now() - entry_time;
|
||||||
let rt = (rt.num_nanoseconds().unwrap_or(0) as f64) / 1_000_000_000.0;
|
let rt = rt.as_seconds_f64();
|
||||||
fmt.write_fmt(format_args!("{:.6}", rt))
|
fmt.write_fmt(format_args!("{:.6}", rt))
|
||||||
}
|
}
|
||||||
FormatText::TimeMillis => {
|
FormatText::TimeMillis => {
|
||||||
let rt = time::now() - entry_time;
|
let rt = OffsetDateTime::now() - entry_time;
|
||||||
let rt = (rt.num_nanoseconds().unwrap_or(0) as f64) / 1_000_000.0;
|
let rt = (rt.whole_nanoseconds() as f64) / 1_000_000.0;
|
||||||
fmt.write_fmt(format_args!("{:.6}", rt))
|
fmt.write_fmt(format_args!("{:.6}", rt))
|
||||||
}
|
}
|
||||||
FormatText::EnvironHeader(ref name) => {
|
FormatText::EnvironHeader(ref name) => {
|
||||||
@ -414,7 +421,7 @@ impl FormatText {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn render_request(&mut self, now: time::Tm, req: &ServiceRequest) {
|
fn render_request(&mut self, now: OffsetDateTime, req: &ServiceRequest) {
|
||||||
match *self {
|
match *self {
|
||||||
FormatText::RequestLine => {
|
FormatText::RequestLine => {
|
||||||
*self = if req.query_string().is_empty() {
|
*self = if req.query_string().is_empty() {
|
||||||
@ -436,7 +443,7 @@ impl FormatText {
|
|||||||
}
|
}
|
||||||
FormatText::UrlPath => *self = FormatText::Str(req.path().to_string()),
|
FormatText::UrlPath => *self = FormatText::Str(req.path().to_string()),
|
||||||
FormatText::RequestTime => {
|
FormatText::RequestTime => {
|
||||||
*self = FormatText::Str(now.rfc3339().to_string())
|
*self = FormatText::Str(now.format("%Y-%m-%dT%H:%M:%S"))
|
||||||
}
|
}
|
||||||
FormatText::RequestHeader(ref name) => {
|
FormatText::RequestHeader(ref name) => {
|
||||||
let s = if let Some(val) = req.headers().get(name) {
|
let s = if let Some(val) = req.headers().get(name) {
|
||||||
@ -513,7 +520,7 @@ mod tests {
|
|||||||
.uri("/test/route/yeah")
|
.uri("/test/route/yeah")
|
||||||
.to_srv_request();
|
.to_srv_request();
|
||||||
|
|
||||||
let now = time::now();
|
let now = OffsetDateTime::now();
|
||||||
for unit in &mut format.0 {
|
for unit in &mut format.0 {
|
||||||
unit.render_request(now, &req);
|
unit.render_request(now, &req);
|
||||||
}
|
}
|
||||||
@ -544,7 +551,7 @@ mod tests {
|
|||||||
)
|
)
|
||||||
.to_srv_request();
|
.to_srv_request();
|
||||||
|
|
||||||
let now = time::now();
|
let now = OffsetDateTime::now();
|
||||||
for unit in &mut format.0 {
|
for unit in &mut format.0 {
|
||||||
unit.render_request(now, &req);
|
unit.render_request(now, &req);
|
||||||
}
|
}
|
||||||
@ -554,7 +561,7 @@ mod tests {
|
|||||||
unit.render_response(&resp);
|
unit.render_response(&resp);
|
||||||
}
|
}
|
||||||
|
|
||||||
let entry_time = time::now();
|
let entry_time = OffsetDateTime::now();
|
||||||
let render = |fmt: &mut Formatter<'_>| {
|
let render = |fmt: &mut Formatter<'_>| {
|
||||||
for unit in &format.0 {
|
for unit in &format.0 {
|
||||||
unit.render(fmt, 1024, entry_time)?;
|
unit.render(fmt, 1024, entry_time)?;
|
||||||
@ -572,7 +579,7 @@ mod tests {
|
|||||||
let mut format = Format::new("%t");
|
let mut format = Format::new("%t");
|
||||||
let req = TestRequest::default().to_srv_request();
|
let req = TestRequest::default().to_srv_request();
|
||||||
|
|
||||||
let now = time::now();
|
let now = OffsetDateTime::now();
|
||||||
for unit in &mut format.0 {
|
for unit in &mut format.0 {
|
||||||
unit.render_request(now, &req);
|
unit.render_request(now, &req);
|
||||||
}
|
}
|
||||||
@ -589,6 +596,6 @@ mod tests {
|
|||||||
Ok(())
|
Ok(())
|
||||||
};
|
};
|
||||||
let s = format!("{}", FormatDisplay(&render));
|
let s = format!("{}", FormatDisplay(&render));
|
||||||
assert!(s.contains(&format!("{}", now.rfc3339())));
|
assert!(s.contains(&format!("{}", now.format("%Y-%m-%dT%H:%M:%S"))));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -206,8 +206,14 @@ impl HttpRequest {
|
|||||||
&self.0.config
|
&self.0.config
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get an application data stored with `App::extension()` method during
|
/// Get an application data object stored with `App::data` or `App::app_data`
|
||||||
/// application configuration.
|
/// methods during application configuration.
|
||||||
|
///
|
||||||
|
/// If `App::data` was used to store object, use `Data<T>`:
|
||||||
|
///
|
||||||
|
/// ```rust,ignore
|
||||||
|
/// let opt_t = req.app_data::<Data<T>>();
|
||||||
|
/// ```
|
||||||
pub fn app_data<T: 'static>(&self) -> Option<&T> {
|
pub fn app_data<T: 'static>(&self) -> Option<&T> {
|
||||||
if let Some(st) = self.0.app_data.get::<T>() {
|
if let Some(st) = self.0.app_data.get::<T>() {
|
||||||
Some(&st)
|
Some(&st)
|
||||||
|
@ -443,8 +443,6 @@ where
|
|||||||
|
|
||||||
#[cfg(unix)]
|
#[cfg(unix)]
|
||||||
/// Start listening for unix domain connections on existing listener.
|
/// Start listening for unix domain connections on existing listener.
|
||||||
///
|
|
||||||
/// This method is available with `uds` feature.
|
|
||||||
pub fn listen_uds(
|
pub fn listen_uds(
|
||||||
mut self,
|
mut self,
|
||||||
lst: std::os::unix::net::UnixListener,
|
lst: std::os::unix::net::UnixListener,
|
||||||
@ -483,8 +481,6 @@ where
|
|||||||
|
|
||||||
#[cfg(unix)]
|
#[cfg(unix)]
|
||||||
/// Start listening for incoming unix domain connections.
|
/// Start listening for incoming unix domain connections.
|
||||||
///
|
|
||||||
/// This method is available with `uds` feature.
|
|
||||||
pub fn bind_uds<A>(mut self, addr: A) -> io::Result<Self>
|
pub fn bind_uds<A>(mut self, addr: A) -> io::Result<Self>
|
||||||
where
|
where
|
||||||
A: AsRef<std::path::Path>,
|
A: AsRef<std::path::Path>,
|
||||||
|
27
src/test.rs
27
src/test.rs
@ -95,11 +95,10 @@ where
|
|||||||
/// Calls service and waits for response future completion.
|
/// Calls service and waits for response future completion.
|
||||||
///
|
///
|
||||||
/// ```rust
|
/// ```rust
|
||||||
/// use actix_web::{test, App, HttpResponse, http::StatusCode};
|
/// use actix_web::{test, web, App, HttpResponse, http::StatusCode};
|
||||||
/// use actix_service::Service;
|
|
||||||
///
|
///
|
||||||
/// #[test]
|
/// #[actix_rt::test]
|
||||||
/// fn test_response() {
|
/// async fn test_response() {
|
||||||
/// let mut app = test::init_service(
|
/// let mut app = test::init_service(
|
||||||
/// App::new()
|
/// App::new()
|
||||||
/// .service(web::resource("/test").to(|| async {
|
/// .service(web::resource("/test").to(|| async {
|
||||||
@ -151,7 +150,7 @@ where
|
|||||||
pub async fn read_response<S, B>(app: &mut S, req: Request) -> Bytes
|
pub async fn read_response<S, B>(app: &mut S, req: Request) -> Bytes
|
||||||
where
|
where
|
||||||
S: Service<Request = Request, Response = ServiceResponse<B>, Error = Error>,
|
S: Service<Request = Request, Response = ServiceResponse<B>, Error = Error>,
|
||||||
B: MessageBody,
|
B: MessageBody + Unpin,
|
||||||
{
|
{
|
||||||
let mut resp = app
|
let mut resp = app
|
||||||
.call(req)
|
.call(req)
|
||||||
@ -194,7 +193,7 @@ where
|
|||||||
/// ```
|
/// ```
|
||||||
pub async fn read_body<B>(mut res: ServiceResponse<B>) -> Bytes
|
pub async fn read_body<B>(mut res: ServiceResponse<B>) -> Bytes
|
||||||
where
|
where
|
||||||
B: MessageBody,
|
B: MessageBody + Unpin,
|
||||||
{
|
{
|
||||||
let mut body = res.take_body();
|
let mut body = res.take_body();
|
||||||
let mut bytes = BytesMut::new();
|
let mut bytes = BytesMut::new();
|
||||||
@ -252,7 +251,7 @@ where
|
|||||||
pub async fn read_response_json<S, B, T>(app: &mut S, req: Request) -> T
|
pub async fn read_response_json<S, B, T>(app: &mut S, req: Request) -> T
|
||||||
where
|
where
|
||||||
S: Service<Request = Request, Response = ServiceResponse<B>, Error = Error>,
|
S: Service<Request = Request, Response = ServiceResponse<B>, Error = Error>,
|
||||||
B: MessageBody,
|
B: MessageBody + Unpin,
|
||||||
T: DeserializeOwned,
|
T: DeserializeOwned,
|
||||||
{
|
{
|
||||||
let body = read_response(app, req).await;
|
let body = read_response(app, req).await;
|
||||||
@ -730,7 +729,7 @@ where
|
|||||||
.map_err(|e| log::error!("Can not set alpn protocol: {:?}", e));
|
.map_err(|e| log::error!("Can not set alpn protocol: {:?}", e));
|
||||||
Connector::new()
|
Connector::new()
|
||||||
.conn_lifetime(time::Duration::from_secs(0))
|
.conn_lifetime(time::Duration::from_secs(0))
|
||||||
.timeout(time::Duration::from_millis(3000))
|
.timeout(time::Duration::from_millis(30000))
|
||||||
.ssl(builder.build())
|
.ssl(builder.build())
|
||||||
.finish()
|
.finish()
|
||||||
}
|
}
|
||||||
@ -738,7 +737,7 @@ where
|
|||||||
{
|
{
|
||||||
Connector::new()
|
Connector::new()
|
||||||
.conn_lifetime(time::Duration::from_secs(0))
|
.conn_lifetime(time::Duration::from_secs(0))
|
||||||
.timeout(time::Duration::from_millis(3000))
|
.timeout(time::Duration::from_millis(30000))
|
||||||
.finish()
|
.finish()
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
@ -954,7 +953,6 @@ impl Drop for TestServer {
|
|||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use actix_http::httpmessage::HttpMessage;
|
use actix_http::httpmessage::HttpMessage;
|
||||||
use futures::FutureExt;
|
|
||||||
use serde::{Deserialize, Serialize};
|
use serde::{Deserialize, Serialize};
|
||||||
use std::time::SystemTime;
|
use std::time::SystemTime;
|
||||||
|
|
||||||
@ -1164,6 +1162,13 @@ mod tests {
|
|||||||
assert!(res.status().is_success());
|
assert!(res.status().is_success());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
|
||||||
|
Comment out until actix decoupled of actix-http:
|
||||||
|
https://github.com/actix/actix/issues/321
|
||||||
|
|
||||||
|
use futures::FutureExt;
|
||||||
|
|
||||||
#[actix_rt::test]
|
#[actix_rt::test]
|
||||||
async fn test_actor() {
|
async fn test_actor() {
|
||||||
use actix::Actor;
|
use actix::Actor;
|
||||||
@ -1184,7 +1189,6 @@ mod tests {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
let addr = MyActor.start();
|
|
||||||
|
|
||||||
let mut app = init_service(App::new().service(web::resource("/index.html").to(
|
let mut app = init_service(App::new().service(web::resource("/index.html").to(
|
||||||
move || {
|
move || {
|
||||||
@ -1206,4 +1210,5 @@ mod tests {
|
|||||||
let res = app.call(req).await.unwrap();
|
let res = app.call(req).await.unwrap();
|
||||||
assert!(res.status().is_success());
|
assert!(res.status().is_success());
|
||||||
}
|
}
|
||||||
|
*/
|
||||||
}
|
}
|
||||||
|
32
src/web.rs
32
src/web.rs
@ -96,7 +96,7 @@ pub fn route() -> Route {
|
|||||||
/// );
|
/// );
|
||||||
/// ```
|
/// ```
|
||||||
///
|
///
|
||||||
/// In the above example, one `GET` route get added:
|
/// In the above example, one `GET` route gets added:
|
||||||
/// * /{project_id}
|
/// * /{project_id}
|
||||||
///
|
///
|
||||||
pub fn get() -> Route {
|
pub fn get() -> Route {
|
||||||
@ -114,7 +114,7 @@ pub fn get() -> Route {
|
|||||||
/// );
|
/// );
|
||||||
/// ```
|
/// ```
|
||||||
///
|
///
|
||||||
/// In the above example, one `POST` route get added:
|
/// In the above example, one `POST` route gets added:
|
||||||
/// * /{project_id}
|
/// * /{project_id}
|
||||||
///
|
///
|
||||||
pub fn post() -> Route {
|
pub fn post() -> Route {
|
||||||
@ -132,7 +132,7 @@ pub fn post() -> Route {
|
|||||||
/// );
|
/// );
|
||||||
/// ```
|
/// ```
|
||||||
///
|
///
|
||||||
/// In the above example, one `PUT` route get added:
|
/// In the above example, one `PUT` route gets added:
|
||||||
/// * /{project_id}
|
/// * /{project_id}
|
||||||
///
|
///
|
||||||
pub fn put() -> Route {
|
pub fn put() -> Route {
|
||||||
@ -150,7 +150,7 @@ pub fn put() -> Route {
|
|||||||
/// );
|
/// );
|
||||||
/// ```
|
/// ```
|
||||||
///
|
///
|
||||||
/// In the above example, one `PATCH` route get added:
|
/// In the above example, one `PATCH` route gets added:
|
||||||
/// * /{project_id}
|
/// * /{project_id}
|
||||||
///
|
///
|
||||||
pub fn patch() -> Route {
|
pub fn patch() -> Route {
|
||||||
@ -168,7 +168,7 @@ pub fn patch() -> Route {
|
|||||||
/// );
|
/// );
|
||||||
/// ```
|
/// ```
|
||||||
///
|
///
|
||||||
/// In the above example, one `DELETE` route get added:
|
/// In the above example, one `DELETE` route gets added:
|
||||||
/// * /{project_id}
|
/// * /{project_id}
|
||||||
///
|
///
|
||||||
pub fn delete() -> Route {
|
pub fn delete() -> Route {
|
||||||
@ -186,13 +186,31 @@ pub fn delete() -> Route {
|
|||||||
/// );
|
/// );
|
||||||
/// ```
|
/// ```
|
||||||
///
|
///
|
||||||
/// In the above example, one `HEAD` route get added:
|
/// In the above example, one `HEAD` route gets added:
|
||||||
/// * /{project_id}
|
/// * /{project_id}
|
||||||
///
|
///
|
||||||
pub fn head() -> Route {
|
pub fn head() -> Route {
|
||||||
method(Method::HEAD)
|
method(Method::HEAD)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Create *route* with `TRACE` method guard.
|
||||||
|
///
|
||||||
|
/// ```rust
|
||||||
|
/// use actix_web::{web, App, HttpResponse};
|
||||||
|
///
|
||||||
|
/// let app = App::new().service(
|
||||||
|
/// web::resource("/{project_id}")
|
||||||
|
/// .route(web::trace().to(|| HttpResponse::Ok()))
|
||||||
|
/// );
|
||||||
|
/// ```
|
||||||
|
///
|
||||||
|
/// In the above example, one `HEAD` route gets added:
|
||||||
|
/// * /{project_id}
|
||||||
|
///
|
||||||
|
pub fn trace() -> Route {
|
||||||
|
method(Method::TRACE)
|
||||||
|
}
|
||||||
|
|
||||||
/// Create *route* and add method guard.
|
/// Create *route* and add method guard.
|
||||||
///
|
///
|
||||||
/// ```rust
|
/// ```rust
|
||||||
@ -204,7 +222,7 @@ pub fn head() -> Route {
|
|||||||
/// );
|
/// );
|
||||||
/// ```
|
/// ```
|
||||||
///
|
///
|
||||||
/// In the above example, one `GET` route get added:
|
/// In the above example, one `GET` route gets added:
|
||||||
/// * /{project_id}
|
/// * /{project_id}
|
||||||
///
|
///
|
||||||
pub fn method(method: Method) -> Route {
|
pub fn method(method: Method) -> Route {
|
||||||
|
@ -1,5 +1,9 @@
|
|||||||
# Changes
|
# Changes
|
||||||
|
|
||||||
|
## [Unreleased] - 2020-xx-xx
|
||||||
|
|
||||||
|
* Update the `time` dependency to 0.2.7
|
||||||
|
|
||||||
## [1.0.0] - 2019-12-13
|
## [1.0.0] - 2019-12-13
|
||||||
|
|
||||||
### Changed
|
### Changed
|
||||||
|
@ -37,7 +37,7 @@ actix-utils = "1.0.3"
|
|||||||
actix-rt = "1.0.0"
|
actix-rt = "1.0.0"
|
||||||
actix-server = "1.0.0"
|
actix-server = "1.0.0"
|
||||||
actix-testing = "1.0.0"
|
actix-testing = "1.0.0"
|
||||||
awc = "1.0.0"
|
awc = "1.0.1"
|
||||||
|
|
||||||
base64 = "0.11"
|
base64 = "0.11"
|
||||||
bytes = "0.5.3"
|
bytes = "0.5.3"
|
||||||
@ -51,9 +51,9 @@ serde_json = "1.0"
|
|||||||
sha1 = "0.6"
|
sha1 = "0.6"
|
||||||
slab = "0.4"
|
slab = "0.4"
|
||||||
serde_urlencoded = "0.6.1"
|
serde_urlencoded = "0.6.1"
|
||||||
time = "0.1"
|
time = { version = "0.2.7", default-features = false, features = ["std"] }
|
||||||
open-ssl = { version="0.10", package="openssl", optional = true }
|
open-ssl = { version="0.10", package="openssl", optional = true }
|
||||||
|
|
||||||
[dev-dependencies]
|
[dev-dependencies]
|
||||||
actix-web = "2.0.0-rc"
|
actix-web = "2.0.0"
|
||||||
actix-http = "1.0.1"
|
actix-http = "2.0.0-alpha.1"
|
||||||
|
@ -77,7 +77,7 @@ pub fn test_server<F: ServiceFactory<TcpStream>>(factory: F) -> TestServer {
|
|||||||
.map_err(|e| log::error!("Can not set alpn protocol: {:?}", e));
|
.map_err(|e| log::error!("Can not set alpn protocol: {:?}", e));
|
||||||
Connector::new()
|
Connector::new()
|
||||||
.conn_lifetime(time::Duration::from_secs(0))
|
.conn_lifetime(time::Duration::from_secs(0))
|
||||||
.timeout(time::Duration::from_millis(3000))
|
.timeout(time::Duration::from_millis(30000))
|
||||||
.ssl(builder.build())
|
.ssl(builder.build())
|
||||||
.finish()
|
.finish()
|
||||||
}
|
}
|
||||||
@ -85,7 +85,7 @@ pub fn test_server<F: ServiceFactory<TcpStream>>(factory: F) -> TestServer {
|
|||||||
{
|
{
|
||||||
Connector::new()
|
Connector::new()
|
||||||
.conn_lifetime(time::Duration::from_secs(0))
|
.conn_lifetime(time::Duration::from_secs(0))
|
||||||
.timeout(time::Duration::from_millis(3000))
|
.timeout(time::Duration::from_millis(30000))
|
||||||
.finish()
|
.finish()
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
30
tests/test_weird_poll.rs
Normal file
30
tests/test_weird_poll.rs
Normal file
@ -0,0 +1,30 @@
|
|||||||
|
// Regression test for #/1321
|
||||||
|
|
||||||
|
use futures::task::{noop_waker, Context};
|
||||||
|
use futures::stream::once;
|
||||||
|
use actix_http::body::{MessageBody, BodyStream};
|
||||||
|
use bytes::Bytes;
|
||||||
|
|
||||||
|
/*
|
||||||
|
Disable weird poll until actix-web is based on actix-http 2.0.0
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn weird_poll() {
|
||||||
|
let (sender, receiver) = futures::channel::oneshot::channel();
|
||||||
|
let mut body_stream = Ok(BodyStream::new(once(async {
|
||||||
|
let x = Box::new(0);
|
||||||
|
let y = &x;
|
||||||
|
receiver.await.unwrap();
|
||||||
|
let _z = **y;
|
||||||
|
Ok::<_, ()>(Bytes::new())
|
||||||
|
})));
|
||||||
|
|
||||||
|
let waker = noop_waker();
|
||||||
|
let mut context = Context::from_waker(&waker);
|
||||||
|
|
||||||
|
let _ = body_stream.as_mut().unwrap().poll_next(&mut context);
|
||||||
|
sender.send(()).unwrap();
|
||||||
|
let _ = std::mem::replace(&mut body_stream, Err([0; 32])).unwrap().poll_next(&mut context);
|
||||||
|
}
|
||||||
|
|
||||||
|
*/
|
Reference in New Issue
Block a user