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 "ssl_compat.h" 33 34 #ifdef NGHTTP2_OPENSSL_IS_WOLFSSL 35 # include <wolfssl/options.h> 36 # include <wolfssl/openssl/ssl.h> 37 #else // !NGHTTP2_OPENSSL_IS_WOLFSSL 38 # include <openssl/ssl.h> 39 #endif // !NGHTTP2_OPENSSL_IS_WOLFSSL 40 41 namespace shrpx { 42 43 struct Connection; 44 45 class RateLimit { 46 public: 47 // We need |conn| object to check that it has unread bytes for TLS 48 // connection. 49 RateLimit(struct ev_loop *loop, ev_io *w, size_t rate, size_t burst, 50 Connection *conn = nullptr); 51 ~RateLimit(); 52 size_t avail() const; 53 void drain(size_t n); 54 void regen(); 55 void startw(); 56 void stopw(); 57 // Feeds event if conn_->tls object has unread bytes. This is 58 // required since it is buffered in conn_->tls object, io event is 59 // not generated unless new incoming data is received. 60 void handle_tls_pending_read(); 61 62 private: 63 ev_timer t_; 64 ev_io *w_; 65 struct ev_loop *loop_; 66 Connection *conn_; 67 size_t rate_; 68 size_t burst_; 69 size_t avail_; 70 bool startw_req_; 71 }; 72 73 } // namespace shrpx 74 75 #endif // SHRPX_RATE_LIMIT_H 76