1 /*
2 * nghttp2 - HTTP/2 C Library
3 *
4 * Copyright (c) 2012 Tatsuhiro Tsujikawa
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining
7 * a copy of this software and associated documentation files (the
8 * "Software"), to deal in the Software without restriction, including
9 * without limitation the rights to use, copy, modify, merge, publish,
10 * distribute, sublicense, and/or sell copies of the Software, and to
11 * permit persons to whom the Software is furnished to do so, subject to
12 * the following conditions:
13 *
14 * The above copyright notice and this permission notice shall be
15 * included in all copies or substantial portions of the Software.
16 *
17 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
18 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
19 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
20 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
21 * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
22 * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
23 * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
24 */
25 #include "shrpx_io_control.h"
26
27 #include <algorithm>
28
29 #include "shrpx_rate_limit.h"
30 #include "util.h"
31
32 using namespace nghttp2;
33
34 namespace shrpx {
35
IOControl(RateLimit * lim)36 IOControl::IOControl(RateLimit *lim) : lim_(lim), rdbits_(0) {}
37
~IOControl()38 IOControl::~IOControl() {}
39
pause_read(IOCtrlReason reason)40 void IOControl::pause_read(IOCtrlReason reason) {
41 rdbits_ |= reason;
42 if (lim_) {
43 lim_->stopw();
44 }
45 }
46
resume_read(IOCtrlReason reason)47 bool IOControl::resume_read(IOCtrlReason reason) {
48 rdbits_ &= ~reason;
49 if (rdbits_ == 0) {
50 if (lim_) {
51 lim_->startw();
52 }
53 return true;
54 }
55
56 return false;
57 }
58
force_resume_read()59 void IOControl::force_resume_read() {
60 rdbits_ = 0;
61 if (lim_) {
62 lim_->startw();
63 }
64 }
65
66 } // namespace shrpx
67