Fix: Decrypt TLS requests fully

Add: Handle `Connection: close` header

src/headers/automata.hpp

@@ -165,10 +165,10 @@ void automata_t::operator()() noexcept {
         connection.pipes.mark_submitted_outputs(completed_result);
         if (!connection.pipes.has_remaining_outputs()) {
             connection.exchanges++;
-            // if (connection.exchanges >= server.max_lifetime_exchanges) TODO Why?
-            //     return close_gracefully();
-            // else
-            return receive_next();
+            if (connection.must_close())
+                return close_gracefully();
+            else
+                return receive_next();
         } else {
             connection.pipes.prepare_more_outputs();
             return send_next();

src/headers/connection.hpp

@@ -72,10 +72,15 @@ struct connection_t {
     bool expired() const noexcept {
         return std::chrono::high_resolution_clock::now().time_since_epoch().count() - last_active_ns >
                max_inactive_duration_ns_k;
-    };
+    }
 
     bool is_ready() const noexcept { return tls_ctx == nullptr || ptls_handshake_is_complete(tls_ctx); }
 
+    bool must_close() const noexcept {
+        auto conn = protocol.get_header("Connection");
+        return conn == "Close" || conn == "close";
+    }
+
     bool prepare_step() noexcept {
         if (is_ready())
             return true;
@@ -114,10 +119,17 @@ struct connection_t {
             return;
 
         work_buf.off = 0;
+        int res = 0;
         size_t in_len = pipes.input_span().size();
-        int res = ptls_receive(tls_ctx, &work_buf, pipes.input_span().data(), &in_len);
+        void const* input = pipes.input_span().data();
+        while (in_len != 0 && res != -1) {
+            size_t consumed = in_len;
+            res = ptls_receive(tls_ctx, &work_buf, input, &consumed);
+            in_len -= consumed;
+            input += consumed;
+        }
         if (res != -1) {
-            pipes.release_inputs();
+            pipes.release_current_input();
             std::memcpy(pipes.next_input_address(), work_buf.base, work_buf.off);
             pipes.absorb_input(work_buf.off);
         }

src/headers/containers.hpp

@@ -218,6 +218,9 @@ class exchange_pipes_t {
         input_.dynamic.reset();
         input_.embedded_used = 0;
     }
+
+    void release_current_input() noexcept { input_.embedded_used = 0; }
+
     void release_outputs() noexcept {
         output_.dynamic.reset();
         output_.embedded_used = 0;