TCP reassembler: make sure additional packet array entries are NULL

salcock · Oct 14, 2024 · d8ef7dd · d8ef7dd
1 parent 16232b6
commit d8ef7dd
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/collector/reassembler.c b/src/collector/reassembler.c
@@ -504,6 +504,7 @@ int update_tcp_reassemble_stream(tcp_reassemble_stream_t *stream,
 
     tcp_reass_segment_t *seg, *existing;
     uint8_t *endptr;
+    int i;
 
     HASH_FIND(hh, stream->segments, &seqno, sizeof(seqno), existing);
     if (existing) {
@@ -560,6 +561,9 @@ int update_tcp_reassemble_stream(tcp_reassemble_stream_t *stream,
         if (stream->pkt_cnt == stream->pkt_alloc) {
             stream->packets = realloc(stream->packets,
                     (stream->pkt_alloc + 4) * sizeof(libtrace_packet_t *));
+            for (i = stream->pkt_alloc; i < stream->pkt_alloc + 4; i++) {
+                stream->packets[i] = NULL;
+            }
             stream->pkt_alloc += 4;
         }
         stream->packets[stream->pkt_cnt] = pkt;