One Level Up Top Level

src/http/v3/ngx_http_v3.c - nginx source code

Functions defined

Source code


  2. /*
  3. * Copyright (C) Roman Arutyunyan
  4. * Copyright (C) Nginx, Inc.
  5. */


  8. #include <ngx_config.h>
  9. #include <ngx_core.h>
  10. #include <ngx_http.h>


  13. static void ngx_http_v3_keepalive_handler(ngx_event_t *ev);
  14. static void ngx_http_v3_cleanup_session(void *data);


  17. ngx_int_t
  18. ‌ngx_http_v3_init_session(ngx_connection_t *c)
  19. {
  20.     ngx_pool_cleanup_t     *cln;
  21.     ngx_http_connection_t  *hc;
  22.     ngx_http_v3_session_t  *h3c;

  24.     hc = c->data;

  26.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, c->log, 0, "http3 init session");

  28.     h3c = ngx_pcalloc(c->pool, sizeof(ngx_http_v3_session_t));
  29.     if (h3c == NULL) {
  30.         goto failed;
  31.     }

  33.     h3c->http_connection = hc;

  35.     ngx_queue_init(&h3c->blocked);

  37.     h3c->keepalive.log = c->log;
  38.     h3c->keepalive.data = c;
  39.     h3c->keepalive.handler = ngx_http_v3_keepalive_handler;

  41.     h3c->table.send_insert_count.log = c->log;
  42.     h3c->table.send_insert_count.data = c;
  43.     h3c->table.send_insert_count.handler = ngx_http_v3_inc_insert_count_handler;

  45.     cln = ngx_pool_cleanup_add(c->pool, 0);
  46.     if (cln == NULL) {
  47.         goto failed;
  48.     }

  50.     cln->handler = ngx_http_v3_cleanup_session;
  51.     cln->data = h3c;

  53.     c->data = h3c;

  55.     return NGX_OK;

  57. failed:

  59.     ngx_log_error(NGX_LOG_ERR, c->log, 0, "failed to create http3 session");
  60.     return NGX_ERROR;
  61. }


  64. static void
  65. ‌ngx_http_v3_keepalive_handler(ngx_event_t *ev)
  66. {
  67.     ngx_connection_t  *c;

  69.     c = ev->data;

  71.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, c->log, 0, "http3 keepalive handler");

  73.     ngx_http_v3_finalize_connection(c, NGX_HTTP_V3_ERR_NO_ERROR,
  74.                                     "keepalive timeout");
  75. }


  78. static void
  79. ‌ngx_http_v3_cleanup_session(void *data)
  80. {
  81.     ngx_http_v3_session_t  *h3c = data;

  83.     ngx_http_v3_cleanup_table(h3c);

  85.     if (h3c->keepalive.timer_set) {
  86.         ngx_del_timer(&h3c->keepalive);
  87.     }

  89.     if (h3c->table.send_insert_count.posted) {
  90.         ngx_delete_posted_event(&h3c->table.send_insert_count);
  91.     }
  92. }


  95. ngx_int_t
  96. ‌ngx_http_v3_check_flood(ngx_connection_t *c)
  97. {
  98.     ngx_http_v3_session_t  *h3c;

  100.     h3c = ngx_http_v3_get_session(c);

  102.     if (h3c->total_bytes / 8 > h3c->payload_bytes + 1048576) {
  103.         ngx_log_error(NGX_LOG_INFO, c->log, 0, "http3 flood detected");

  105.         ngx_http_v3_finalize_connection(c, NGX_HTTP_V3_ERR_NO_ERROR,
  106.                                         "HTTP/3 flood detected");
  107.         return NGX_ERROR;
  108.     }

  110.     return NGX_OK;
  111. }

One Level Up Top Level