ignatenkobrain / rpms / nginx

Forked from rpms/nginx 2 years ago
Clone
d22de4
diff --git a/src/http/v2/ngx_http_v2.c b/src/http/v2/ngx_http_v2.c
d22de4
index 12214e1..3b7f892 100644
d22de4
--- a/src/http/v2/ngx_http_v2.c
d22de4
+++ b/src/http/v2/ngx_http_v2.c
d22de4
@@ -4335,6 +4335,8 @@ ngx_http_v2_close_stream(ngx_http_v2_stream_t *stream, ngx_int_t rc)
d22de4
      */
d22de4
     pool = stream->pool;
d22de4
 
d22de4
+    h2c->frames -= stream->frames;
d22de4
+
d22de4
     ngx_http_free_request(stream->request, rc);
d22de4
 
d22de4
     if (pool != h2c->state.pool) {
d22de4
diff --git a/src/http/v2/ngx_http_v2.h b/src/http/v2/ngx_http_v2.h
d22de4
index bec2216..715b7d3 100644
d22de4
--- a/src/http/v2/ngx_http_v2.h
d22de4
+++ b/src/http/v2/ngx_http_v2.h
d22de4
@@ -192,6 +192,8 @@ struct ngx_http_v2_stream_s {
d22de4
 
d22de4
     ngx_buf_t                       *preread;
d22de4
 
d22de4
+    ngx_uint_t                       frames;
d22de4
+
d22de4
     ngx_http_v2_out_frame_t         *free_frames;
d22de4
     ngx_chain_t                     *free_frame_headers;
d22de4
     ngx_chain_t                     *free_bufs;
d22de4
diff --git a/src/http/v2/ngx_http_v2_filter_module.c b/src/http/v2/ngx_http_v2_filter_module.c
d22de4
index 029e8ec..c7ee553 100644
d22de4
--- a/src/http/v2/ngx_http_v2_filter_module.c
d22de4
+++ b/src/http/v2/ngx_http_v2_filter_module.c
d22de4
@@ -1661,22 +1661,34 @@ static ngx_http_v2_out_frame_t *
d22de4
 ngx_http_v2_filter_get_data_frame(ngx_http_v2_stream_t *stream,
d22de4
     size_t len, ngx_chain_t *first, ngx_chain_t *last)
d22de4
 {
d22de4
-    u_char                    flags;
d22de4
-    ngx_buf_t                *buf;
d22de4
-    ngx_chain_t              *cl;
d22de4
-    ngx_http_v2_out_frame_t  *frame;
d22de4
+    u_char                     flags;
d22de4
+    ngx_buf_t                 *buf;
d22de4
+    ngx_chain_t               *cl;
d22de4
+    ngx_http_v2_out_frame_t   *frame;
d22de4
+    ngx_http_v2_connection_t  *h2c;
d22de4
 
d22de4
     frame = stream->free_frames;
d22de4
+    h2c = stream->connection;
d22de4
 
d22de4
     if (frame) {
d22de4
         stream->free_frames = frame->next;
d22de4
 
d22de4
-    } else {
d22de4
+    } else if (h2c->frames < 10000) {
d22de4
         frame = ngx_palloc(stream->request->pool,
d22de4
                            sizeof(ngx_http_v2_out_frame_t));
d22de4
         if (frame == NULL) {
d22de4
             return NULL;
d22de4
         }
d22de4
+
d22de4
+        stream->frames++;
d22de4
+        h2c->frames++;
d22de4
+
d22de4
+    } else {
d22de4
+        ngx_log_error(NGX_LOG_INFO, h2c->connection->log, 0,
d22de4
+                      "http2 flood detected");
d22de4
+
d22de4
+        h2c->connection->error = 1;
d22de4
+        return NULL;
d22de4
     }
d22de4
 
d22de4
     flags = last->buf->last_buf ? NGX_HTTP_V2_END_STREAM_FLAG : 0;