Skip to content

Commit 6cae3cd

Browse files
Aman Sharmafacebook-github-bot
authored andcommitted
Use ResetsRateLimitFilter to rate limit resets
Summary: We're going to rate limit resets separately from other kinds of control messages. Reviewed By: afrind Differential Revision: D50751362 fbshipit-source-id: 5fdbe215d664bf952a0bb00f8de808ef5ea39329
1 parent 7591428 commit 6cae3cd

File tree

2 files changed

+2
-7
lines changed

2 files changed

+2
-7
lines changed

proxygen/lib/http/codec/ControlMessageRateLimitFilter.h

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -26,13 +26,6 @@ class ControlMessageRateLimitFilter : public RateLimitFilter {
2626
timeoutDuration_ = kDefaultTimeoutDuration;
2727
}
2828

29-
void onAbort(HTTPCodec::StreamID streamID, ErrorCode code) override {
30-
if (!incrementNumEventsInCurrentInterval()) {
31-
callback_->onAbort(streamID, code);
32-
} else {
33-
sendErrorCallback(http2::FrameType::RST_STREAM);
34-
}
35-
}
3629
void onPingRequest(uint64_t data) override {
3730
if (!incrementNumEventsInCurrentInterval()) {
3831
callback_->onPingRequest(data);

proxygen/lib/http/session/HTTPSession.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
#include <proxygen/lib/http/codec/HTTP2Codec.h>
2222
#include <proxygen/lib/http/codec/HTTPChecks.h>
2323
#include <proxygen/lib/http/codec/HeadersRateLimitFilter.h>
24+
#include <proxygen/lib/http/codec/ResetsRateLimitFilter.h>
2425
#include <proxygen/lib/http/session/HTTPSessionController.h>
2526
#include <proxygen/lib/http/session/HTTPSessionStats.h>
2627
#include <wangle/acceptor/ConnectionManager.h>
@@ -228,6 +229,7 @@ void HTTPSession::setupCodec() {
228229
addRateLimitFilter(RateLimitFilter::Type::HEADERS);
229230
addRateLimitFilter(RateLimitFilter::Type::DIRECT_ERROR_HANDLING);
230231
addRateLimitFilter(RateLimitFilter::Type::MISC_CONTROL_MSGS);
232+
addRateLimitFilter(RateLimitFilter::Type::RSTS);
231233
}
232234

233235
codec_.setCallback(this);

0 commit comments

Comments
 (0)