actix/actix-extras
1
0
Fork 0
mirror of https://github.com/actix/actix-extras.git synced 2025-06-28 02:49:02 +02:00
Code Issues Releases Wiki Activity

drop connection if handler get dropped without consuming payload

Browse Source
This commit is contained in:
Nikolay Kim
2018-02-27 16:08:57 -08:00
parent 9b06eac720
commit e2c8f17c2c
5 changed files with 35 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/server/h2.rs
View File

@ -21,7 +21,7 @@ use error::PayloadError;
use httpcodes::HTTPNotFound;
use httpmessage::HttpMessage;
use httprequest::HttpRequest;
use payload::{Payload, PayloadWriter};
use payload::{Payload, PayloadWriter, PayloadStatus};
use super::h2writer::H2Writer;
use super::encoding::PayloadType;
@ -105,7 +105,7 @@ impl<T, H> Http2<T, H>
item.poll_payload();
if !item.flags.contains(EntryFlags::EOF) {
let retry = item.payload.need_read();
let retry = item.payload.need_read() == PayloadStatus::Read;
loop {
match item.task.poll_io(&mut item.stream) {
Ok(Async::Ready(ready)) => {
@ -116,7 +116,8 @@ impl<T, H> Http2<T, H>
not_ready = false;
},
Ok(Async::NotReady) => {
if item.payload.need_read() && !retry {
if item.payload.need_read() == PayloadStatus::Read && !retry
{
continue
}
},
@ -307,7 +308,7 @@ impl Entry {
fn poll_payload(&mut self) {
if !self.flags.contains(EntryFlags::REOF) {
if self.payload.need_read() {
if self.payload.need_read() == PayloadStatus::Read {
if let Err(err) = self.recv.release_capacity().release_capacity(32_768) {
self.payload.set_error(PayloadError::Http2(err))
}