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 #ifndef SHRPX_IO_CONTROL_H 26 #define SHRPX_IO_CONTROL_H 27 28 #include "shrpx.h" 29 30 #include <cinttypes> 31 #include <vector> 32 33 #include <ev.h> 34 35 #include "shrpx_rate_limit.h" 36 37 namespace shrpx { 38 39 enum IOCtrlReason { SHRPX_NO_BUFFER = 1 << 0, SHRPX_MSG_BLOCK = 1 << 1 }; 40 41 class IOControl { 42 public: 43 IOControl(RateLimit *lim); 44 ~IOControl(); 45 void pause_read(IOCtrlReason reason); 46 // Returns true if read operation is enabled after this call 47 bool resume_read(IOCtrlReason reason); 48 // Clear all pause flags and enable read 49 void force_resume_read(); 50 51 private: 52 RateLimit *lim_; 53 uint32_t rdbits_; 54 }; 55 56 } // namespace shrpx 57 58 #endif // SHRPX_IO_CONTROL_H 59