Browse Source

demos/http3: Use `SSL_write_ex2()` together with `SSL_WRITE_FLAG_CONCLUDE`

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)
Alexandr Nedvedicky 2 months ago
parent
commit
c18c301deb
2 changed files with 10 additions and 8 deletions
  1. 2 2
      demos/http3/Makefile
  2. 8 6
      demos/http3/ossl-nghttp3.c

+ 2 - 2
demos/http3/Makefile

@@ -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

+ 8 - 6
demos/http3/ossl-nghttp3.c

@@ -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) {
                 /*