1 /* 2 * nghttp2 - HTTP/2 C Library 3 * 4 * Copyright (c) 2015 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_RATE_LIMIT_H 26 #define SHRPX_RATE_LIMIT_H 27 28 #include "shrpx.h" 29 30 #include <ev.h> 31 32 #include <openssl/ssl.h> 33 34 namespace shrpx { 35 36 struct Connection; 37 38 class RateLimit { 39 public: 40 // We need |conn| object to check that it has unread bytes for TLS 41 // connection. 42 RateLimit(struct ev_loop *loop, ev_io *w, size_t rate, size_t burst, 43 Connection *conn = nullptr); 44 ~RateLimit(); 45 size_t avail() const; 46 void drain(size_t n); 47 void regen(); 48 void startw(); 49 void stopw(); 50 // Feeds event if conn_->tls object has unread bytes. This is 51 // required since it is buffered in conn_->tls object, io event is 52 // not generated unless new incoming data is received. 53 void handle_tls_pending_read(); 54 55 private: 56 ev_timer t_; 57 ev_io *w_; 58 struct ev_loop *loop_; 59 Connection *conn_; 60 size_t rate_; 61 size_t burst_; 62 size_t avail_; 63 bool startw_req_; 64 }; 65 66 } // namespace shrpx 67 68 #endif // SHRPX_RATE_LIMIT_H 69