[FFmpeg-cvslog] avformat/iamf_writer: ensure the stream groups are not empty

[James Almer]

ffmpeg | branch: release/7.1 | James Almer <jamrial at gmail.com> | Fri Dec 27 16:17:51 2024 -0300| [4f5769e0526252fb80c3c5a1d46d0ea414643fd6] | committer: James Almer

avformat/iamf_writer: ensure the stream groups are not empty

Signed-off-by: James Almer <jamrial at gmail.com>
(cherry picked from commit cedd9151f863e3db9c325aff2f9375c39adbbec2)

> http://git.videolan.org/gitweb.cgi/ffmpeg.git/?a=commit;h=4f5769e0526252fb80c3c5a1d46d0ea414643fd6
---

 libavformat/iamf_writer.c | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/libavformat/iamf_writer.c b/libavformat/iamf_writer.c
index 0508c676d1..27926f859a 100644
--- a/libavformat/iamf_writer.c
+++ b/libavformat/iamf_writer.c
@@ -217,6 +217,10 @@ int ff_iamf_add_audio_element(IAMFContext *iamf, const AVStreamGroup *stg, void
 
     if (stg->type != AV_STREAM_GROUP_PARAMS_IAMF_AUDIO_ELEMENT)
         return AVERROR(EINVAL);
+    if (!stg->nb_streams) {
+        av_log(log_ctx, AV_LOG_ERROR, "Audio Element id %"PRId64" has no streams\n", stg->id);
+        return AVERROR(EINVAL);
+    }
 
     iamf_audio_element = stg->params.iamf_audio_element;
     if (iamf_audio_element->audio_element_type == AV_IAMF_AUDIO_ELEMENT_TYPE_SCENE) {
@@ -381,6 +385,10 @@ int ff_iamf_add_mix_presentation(IAMFContext *iamf, const AVStreamGroup *stg, vo
 
     if (stg->type != AV_STREAM_GROUP_PARAMS_IAMF_MIX_PRESENTATION)
         return AVERROR(EINVAL);
+    if (!stg->nb_streams) {
+        av_log(log_ctx, AV_LOG_ERROR, "Mix Presentation id %"PRId64" has no streams\n", stg->id);
+        return AVERROR(EINVAL);
+    }
 
     for (int i = 0; i < iamf->nb_mix_presentations; i++) {
         if (stg->id == iamf->mix_presentations[i]->mix_presentation_id) {