From ab4186e9aec67d61177da89a5ccf712727d4731d Mon Sep 17 00:00:00 2001
From: "Alfred E. Heggestad" <114750+alfredh@users.noreply.github.com>
Date: Mon, 22 Jul 2024 15:01:33 +0200
Subject: [PATCH] test: check if header and payload is set

---
 test/h264.c | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/test/h264.c b/test/h264.c
index 2b8168cc2..9db0e0fef 100644
--- a/test/h264.c
+++ b/test/h264.c
@@ -670,7 +670,7 @@ static int packet_handler(bool marker, uint64_t rtp_ts,
 {
 	struct state *state = arg;
 	struct mbuf *mb_pkt = mbuf_alloc(hdr_len + pld_len);
-	int err;
+	int err = 0;
 
 	if (!mb_pkt)
 		return ENOMEM;
@@ -679,8 +679,12 @@ static int packet_handler(bool marker, uint64_t rtp_ts,
 
 	++state->count;
 
-	err  = mbuf_write_mem(mb_pkt, hdr, hdr_len);
-	err |= mbuf_write_mem(mb_pkt, pld, pld_len);
+	if (hdr && hdr_len)
+		err |= mbuf_write_mem(mb_pkt, hdr, hdr_len);
+
+	if (pld && pld_len)
+		err |= mbuf_write_mem(mb_pkt, pld, pld_len);
+
 	if (err)
 		goto out;