emit wrote_all flag

h2o · Jan 29, 2025 · f59a382 · f59a382
1 parent ebcfde2
commit f59a382
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/lib/quicly.c b/lib/quicly.c
@@ -4348,12 +4348,13 @@ quicly_error_t quicly_send_stream(quicly_stream_t *stream, quicly_send_context_t
         is_fin = 0;
     }
 
-    QUICLY_PROBE(STREAM_AFTER_SEND_EMIT, stream->conn, stream->conn->stash.now, stream, *s->dst, off, dst, len);
+    QUICLY_PROBE(STREAM_AFTER_SEND_EMIT, stream->conn, stream->conn->stash.now, stream, *s->dst, off, dst, len, wrote_all);
     QUICLY_LOG_CONN(stream_after_send_emit, stream->conn, {
         PTLS_LOG_ELEMENT_SIGNED(stream_id, stream->stream_id);
         PTLS_LOG_ELEMENT_UNSIGNED(frame_type, *s->dst);
         PTLS_LOG_ELEMENT_UNSIGNED(frame_offset, off);
         PTLS_LOG_APPDATA_ELEMENT_HEXDUMP(data, dst - len, len);
+        PTLS_LOG_ELEMENT_BOOL(wrote_all, wrote_all);
     });
 
     /* update s->dst now that frame construction is complete */

diff --git a/quicly-probes.d b/quicly-probes.d
@@ -158,7 +158,7 @@ provider quicly {
     probe stream_on_send_emit(struct st_quicly_conn_t *conn, int64_t at, struct st_quicly_stream_t *stream, size_t off,
                               size_t capacity);
     probe stream_after_send_emit(struct st_quicly_conn_t *conn, int64_t at, struct st_quicly_stream_t *stream, uint8_t frame_type,
-                                 uint64_t frame_offset, const void *data, size_t data_len);
+                                 uint64_t frame_offset, const void *data, size_t data_len, int wrote_all);
     probe stream_on_send_stop(struct st_quicly_conn_t *conn, int64_t at, struct st_quicly_stream_t *stream, int64_t err);
     probe stream_on_receive(struct st_quicly_conn_t *conn, int64_t at, struct st_quicly_stream_t *stream, size_t off,
                             const void *src, size_t src_len);