[PATCH 8 of 8] QUIC: application init() callback
Roman Arutyunyan
arut at nginx.com
Thu Jun 23 15:58:46 UTC 2022
# HG changeset patch
# User Roman Arutyunyan <arut at nginx.com>
# Date 1655999826 -14400
# Thu Jun 23 19:57:06 2022 +0400
# Branch quic
# Node ID 9be89f499d1f4f550fd995aaa59866edcf386bac
# Parent 03f1661db9601dd552c83643a6198375762b0615
QUIC: application init() callback.
It's called after handshake completion or prior to the first early data stream
creation. The callback should initialize application-level data before
creating streams.
HTTP/3 callback implementation sets keepalive timer and marks connection as
reusable.
Also, this allows to limit max handshake time in ngx_http_v3_init_stream().
diff --git a/src/event/quic/ngx_event_quic.h b/src/event/quic/ngx_event_quic.h
--- a/src/event/quic/ngx_event_quic.h
+++ b/src/event/quic/ngx_event_quic.h
@@ -28,6 +28,7 @@
#define NGX_QUIC_STREAM_UNIDIRECTIONAL 0x02
+typedef ngx_int_t (*ngx_quic_init_pt)(ngx_connection_t *c);
typedef void (*ngx_quic_shutdown_pt)(ngx_connection_t *c);
@@ -77,6 +78,7 @@ typedef struct {
ngx_int_t stream_reject_code_uni;
ngx_int_t stream_reject_code_bidi;
+ ngx_quic_init_pt init;
ngx_quic_shutdown_pt shutdown;
u_char av_token_key[NGX_QUIC_AV_KEY_LEN];
diff --git a/src/event/quic/ngx_event_quic_streams.c b/src/event/quic/ngx_event_quic_streams.c
--- a/src/event/quic/ngx_event_quic_streams.c
+++ b/src/event/quic/ngx_event_quic_streams.c
@@ -21,6 +21,7 @@ static ngx_quic_stream_t *ngx_quic_get_s
static ngx_int_t ngx_quic_reject_stream(ngx_connection_t *c, uint64_t id);
static void ngx_quic_init_stream_handler(ngx_event_t *ev);
static void ngx_quic_init_streams_handler(ngx_connection_t *c);
+static ngx_int_t ngx_quic_do_init_streams(ngx_connection_t *c);
static ngx_quic_stream_t *ngx_quic_create_stream(ngx_connection_t *c,
uint64_t id);
static void ngx_quic_empty_handler(ngx_event_t *ev);
@@ -571,15 +572,22 @@ ngx_quic_init_streams(ngx_connection_t *
return NGX_OK;
}
- ngx_quic_init_streams_handler(c);
-
- return NGX_OK;
+ return ngx_quic_do_init_streams(c);
}
static void
ngx_quic_init_streams_handler(ngx_connection_t *c)
{
+ if (ngx_quic_do_init_streams(c) != NGX_OK) {
+ ngx_quic_close_connection(c, NGX_ERROR);
+ }
+}
+
+
+static ngx_int_t
+ngx_quic_do_init_streams(ngx_connection_t *c)
+{
ngx_queue_t *q;
ngx_quic_stream_t *qs;
ngx_quic_connection_t *qc;
@@ -588,6 +596,12 @@ ngx_quic_init_streams_handler(ngx_connec
qc = ngx_quic_get_connection(c);
+ if (qc->conf->init) {
+ if (qc->conf->init(c) != NGX_OK) {
+ return NGX_ERROR;
+ }
+ }
+
for (q = ngx_queue_head(&qc->streams.uninitialized);
q != ngx_queue_sentinel(&qc->streams.uninitialized);
q = ngx_queue_next(q))
@@ -597,6 +611,8 @@ ngx_quic_init_streams_handler(ngx_connec
}
qc->streams.initialized = 1;
+
+ return NGX_OK;
}
diff --git a/src/http/v3/ngx_http_v3.c b/src/http/v3/ngx_http_v3.c
--- a/src/http/v3/ngx_http_v3.c
+++ b/src/http/v3/ngx_http_v3.c
@@ -17,21 +17,15 @@ static void ngx_http_v3_cleanup_session(
ngx_int_t
ngx_http_v3_init_session(ngx_connection_t *c)
{
- ngx_connection_t *pc;
ngx_pool_cleanup_t *cln;
ngx_http_connection_t *hc;
ngx_http_v3_session_t *h3c;
- pc = c->quic->parent;
- hc = pc->data;
-
- if (hc->v3_session) {
- return NGX_OK;
- }
+ hc = c->data;
ngx_log_debug0(NGX_LOG_DEBUG_HTTP, c->log, 0, "http3 init session");
- h3c = ngx_pcalloc(pc->pool, sizeof(ngx_http_v3_session_t));
+ h3c = ngx_pcalloc(c->pool, sizeof(ngx_http_v3_session_t));
if (h3c == NULL) {
goto failed;
}
@@ -42,16 +36,16 @@ ngx_http_v3_init_session(ngx_connection_
ngx_queue_init(&h3c->blocked);
ngx_queue_init(&h3c->pushing);
- h3c->keepalive.log = pc->log;
- h3c->keepalive.data = pc;
+ h3c->keepalive.log = c->log;
+ h3c->keepalive.data = c;
h3c->keepalive.handler = ngx_http_v3_keepalive_handler;
h3c->keepalive.cancelable = 1;
- h3c->table.send_insert_count.log = pc->log;
- h3c->table.send_insert_count.data = pc;
+ h3c->table.send_insert_count.log = c->log;
+ h3c->table.send_insert_count.data = c;
h3c->table.send_insert_count.handler = ngx_http_v3_inc_insert_count_handler;
- cln = ngx_pool_cleanup_add(pc->pool, 0);
+ cln = ngx_pool_cleanup_add(c->pool, 0);
if (cln == NULL) {
goto failed;
}
@@ -83,7 +77,7 @@ ngx_http_v3_keepalive_handler(ngx_event_
ngx_log_debug0(NGX_LOG_DEBUG_HTTP, c->log, 0, "http3 keepalive handler");
ngx_http_v3_finalize_connection(c, NGX_HTTP_V3_ERR_NO_ERROR,
- "keepalive timeout");
+ "client timed out");
}
diff --git a/src/http/v3/ngx_http_v3.h b/src/http/v3/ngx_http_v3.h
--- a/src/http/v3/ngx_http_v3.h
+++ b/src/http/v3/ngx_http_v3.h
@@ -152,6 +152,7 @@ void ngx_http_v3_init_stream(ngx_connect
void ngx_http_v3_reset_stream(ngx_connection_t *c);
ngx_int_t ngx_http_v3_init_session(ngx_connection_t *c);
ngx_int_t ngx_http_v3_check_flood(ngx_connection_t *c);
+ngx_int_t ngx_http_v3_init(ngx_connection_t *c);
void ngx_http_v3_shutdown(ngx_connection_t *c);
ngx_int_t ngx_http_v3_read_request_body(ngx_http_request_t *r);
diff --git a/src/http/v3/ngx_http_v3_module.c b/src/http/v3/ngx_http_v3_module.c
--- a/src/http/v3/ngx_http_v3_module.c
+++ b/src/http/v3/ngx_http_v3_module.c
@@ -249,6 +249,7 @@ ngx_http_v3_create_srv_conf(ngx_conf_t *
h3scf->quic.stream_reject_code_bidi = NGX_HTTP_V3_ERR_REQUEST_REJECTED;
h3scf->quic.active_connection_id_limit = NGX_CONF_UNSET_UINT;
+ h3scf->quic.init = ngx_http_v3_init;
h3scf->quic.shutdown = ngx_http_v3_shutdown;
return h3scf;
diff --git a/src/http/v3/ngx_http_v3_request.c b/src/http/v3/ngx_http_v3_request.c
--- a/src/http/v3/ngx_http_v3_request.c
+++ b/src/http/v3/ngx_http_v3_request.c
@@ -61,19 +61,29 @@ static const struct {
void
ngx_http_v3_init_stream(ngx_connection_t *c)
{
+ ngx_http_connection_t *hc, *phc;
ngx_http_v3_session_t *h3c;
- ngx_http_connection_t *hc, *phc;
ngx_http_v3_srv_conf_t *h3scf;
ngx_http_core_loc_conf_t *clcf;
+ ngx_http_core_srv_conf_t *cscf;
hc = c->data;
hc->ssl = 1;
clcf = ngx_http_get_module_loc_conf(hc->conf_ctx, ngx_http_core_module);
+ cscf = ngx_http_get_module_srv_conf(hc->conf_ctx, ngx_http_core_module);
h3scf = ngx_http_get_module_srv_conf(hc->conf_ctx, ngx_http_v3_module);
if (c->quic == NULL) {
+ if (ngx_http_v3_init_session(c) != NGX_OK) {
+ ngx_http_close_connection(c);
+ return;
+ }
+
+ h3c = hc->v3_session;
+ ngx_add_timer(&h3c->keepalive, cscf->client_header_timeout);
+
c->idle = 1;
h3scf->quic.timeout = clcf->keepalive_timeout;
ngx_quic_run(c, &h3scf->quic);
@@ -89,13 +99,6 @@ ngx_http_v3_init_stream(ngx_connection_t
ngx_set_connection_log(c, clcf->error_log);
}
- h3c = hc->v3_session;
-
- if (h3c == NULL && ngx_http_v3_init_session(c) != NGX_OK) {
- ngx_http_close_connection(c);
- return;
- }
-
#if (NGX_HTTP_V3_HQ)
if (h3scf->hq) {
ngx_http_v3_init_hq_stream(c);
@@ -103,11 +106,6 @@ ngx_http_v3_init_stream(ngx_connection_t
}
#endif
- if (h3c == NULL && ngx_http_v3_send_settings(c)) {
- ngx_http_close_connection(c);
- return;
- }
-
if (c->quic->id & NGX_QUIC_STREAM_UNIDIRECTIONAL) {
ngx_http_v3_init_uni_stream(c);
@@ -117,6 +115,43 @@ ngx_http_v3_init_stream(ngx_connection_t
}
+ngx_int_t
+ngx_http_v3_init(ngx_connection_t *c)
+{
+ ngx_http_v3_session_t *h3c;
+#if (NGX_HTTP_V3_HQ)
+ ngx_http_connection_t *hc;
+ ngx_http_v3_srv_conf_t *h3scf;
+#endif
+ ngx_http_core_loc_conf_t *clcf;
+
+ ngx_log_debug0(NGX_LOG_DEBUG_HTTP, c->log, 0, "http3 init");
+
+ if (ngx_terminate || ngx_exiting) {
+ return NGX_ERROR;
+ }
+
+ ngx_reusable_connection(c, 1);
+
+ h3c = ngx_http_v3_get_session(c);
+ clcf = ngx_http_v3_get_module_loc_conf(c, ngx_http_core_module);
+ ngx_add_timer(&h3c->keepalive, clcf->keepalive_timeout);
+
+#if (NGX_HTTP_V3_HQ)
+
+ hc = c->data;
+ h3scf = ngx_http_get_module_srv_conf(hc->conf_ctx, ngx_http_v3_module);
+
+ if (h3scf->hq) {
+ return NGX_OK;
+ }
+
+#endif
+
+ return ngx_http_v3_send_settings(c);
+}
+
+
void
ngx_http_v3_shutdown(ngx_connection_t *c)
{
More information about the nginx-devel
mailing list