demos/http3: Use `SSL_write_ex2()` together with `SSL_WRITE_FLAG_CONCLUDE`
authorAlexandr Nedvedicky <sashan@openssl.org>
Thu, 15 Feb 2024 11:07:26 +0000 (12:07 +0100)
committerTomas Mraz <tomas@openssl.org>
Fri, 1 Mar 2024 09:56:16 +0000 (10:56 +0100)
These calls were introduced by PR #23343.

Change also does a minor tweak to Makefile so CFLAGS and LDFLAGS variables
from the environment are respected.

Reviewed-by: Tom Cosgrove <tom.cosgrove@arm.com>
Reviewed-by: Tomas Mraz <tomas@openssl.org>
(Merged from https://github.com/openssl/openssl/pull/23602)

demos/http3/Makefile
demos/http3/ossl-nghttp3.c

index aeff1e9e4f70e55128317939ea471d0561bfbd06..c2275bd755fa94aa8cd364357947ec2e3a4ee4c2 100644 (file)
@@ -4,8 +4,8 @@
 #
 #    LD_LIBRARY_PATH=../.. ./ossl-nghttp3-demo www.example.com:443
 
-CFLAGS  = -I../../include -g -Wall -Wsign-compare
-LDFLAGS = -L../..
+CFLAGS  += -I../../include -g -Wall -Wsign-compare
+LDFLAGS += -L../..
 LDLIBS  = -lcrypto -lssl -lnghttp3
 
 all: ossl-nghttp3-demo
index 2b2259f34fb16884a54d0291bf9e2113eb3d016f..c8c8aa8d428388b61c941c4df0115ff6a6f77c42 100644 (file)
@@ -567,6 +567,7 @@ int OSSL_DEMO_H3_CONN_handle_events(OSSL_DEMO_H3_CONN *conn)
     int ec, fin;
     size_t i, num_vecs, written, total_written, total_len;
     int64_t stream_id;
+    uint64_t flags;
     nghttp3_vec vecs[8] = {0};
     OSSL_DEMO_H3_STREAM key, *s;
     SSL *snew;
@@ -613,6 +614,12 @@ int OSSL_DEMO_H3_CONN_handle_events(OSSL_DEMO_H3_CONN *conn)
         if (ec == 0)
             break;
 
+        /*
+        * we let SSL_write_ex2(3) to conclude the stream for us (send FIN)
+        * after all data are written.
+         */
+        flags = (fin == 0) ? 0 : SSL_WRITE_FLAG_CONCLUDE;
+
         /* For each of the vectors returned, pass it to OpenSSL QUIC. */
         key.id = stream_id;
         if ((s = lh_OSSL_DEMO_H3_STREAM_retrieve(conn->streams, &key)) == NULL) {
@@ -631,7 +638,7 @@ int OSSL_DEMO_H3_CONN_handle_events(OSSL_DEMO_H3_CONN *conn)
             if (s->s == NULL) {
                 /* Already did STOP_SENDING and threw away stream, ignore */
                 written = vecs[i].len;
-            } else if (!SSL_write_ex(s->s, vecs[i].base, vecs[i].len, &written)) {
+            } else if (!SSL_write_ex2(s->s, vecs[i].base, vecs[i].len, flags, &written)) {
                 if (SSL_get_error(s->s, 0) == SSL_ERROR_WANT_WRITE) {
                     /*
                      * We have filled our send buffer so tell nghttp3 to stop
@@ -676,11 +683,6 @@ int OSSL_DEMO_H3_CONN_handle_events(OSSL_DEMO_H3_CONN *conn)
         }
 
         if (fin && total_written == total_len) {
-            /*
-             * We have written all the data so mark the stream as concluded
-             * (FIN).
-             */
-            SSL_stream_conclude(s->s, 0);
 
             if (total_len == 0) {
                 /*