As well as the AV_SIDE_DATA_PROP_STRUCT prop, to define types that describe
a fixed-size C struct and not a flat byte array.
This excludes types like VIDEO_ENC_PARAMS as the struct it describes may have
extra bytes allocated past the end of the struct.

Signed-off-by: James Almer <jamr...@gmail.com>
---
 libavutil/ambient_viewing_environment.c |  9 ++-
 libavutil/frame.h                       | 17 ++++++
 libavutil/mastering_display_metadata.c  |  9 ++-
 libavutil/side_data.c                   | 77 +++++++++++++++++++++----
 libavutil/side_data.h                   |  4 ++
 libavutil/spherical.c                   | 10 +++-
 6 files changed, 107 insertions(+), 19 deletions(-)

diff --git a/libavutil/ambient_viewing_environment.c 
b/libavutil/ambient_viewing_environment.c
index e359727776..ee2e9427cd 100644
--- a/libavutil/ambient_viewing_environment.c
+++ b/libavutil/ambient_viewing_environment.c
@@ -20,9 +20,12 @@
 
 #include "ambient_viewing_environment.h"
 #include "mem.h"
+#include "side_data.h"
 
-static void get_defaults(AVAmbientViewingEnvironment *env)
+void ff_ave_get_defaults(void *obj)
 {
+    AVAmbientViewingEnvironment *env = obj;
+
     env->ambient_illuminance =
     env->ambient_light_x     =
     env->ambient_light_y     = (AVRational) { 0, 1 };
@@ -35,7 +38,7 @@ AVAmbientViewingEnvironment 
*av_ambient_viewing_environment_alloc(size_t *size)
     if (!env)
         return NULL;
 
-    get_defaults(env);
+    ff_ave_get_defaults(env);
 
      if (size)
         *size = sizeof(*env);
@@ -53,7 +56,7 @@ AVAmbientViewingEnvironment 
*av_ambient_viewing_environment_create_side_data(AVF
         return NULL;
 
     memset(side_data->data, 0, side_data->size);
-    get_defaults((AVAmbientViewingEnvironment *)side_data->data);
+    ff_ave_get_defaults(side_data->data);
 
     return (AVAmbientViewingEnvironment *)side_data->data;
 }
diff --git a/libavutil/frame.h b/libavutil/frame.h
index a707b35087..cd6b5194d1 100644
--- a/libavutil/frame.h
+++ b/libavutil/frame.h
@@ -304,6 +304,13 @@ enum AVSideDataProps {
      * or adjusted to the new layout.
      */
     AV_SIDE_DATA_PROP_CHANNEL_DEPENDENT = (1 << 4),
+
+    /**
+     * Side data stores a fixed-size C struct and not a flat byte array.
+     * Entries of this type should be allocated with
+     * @ref av_frame_side_data_new_struct().
+     */
+    AV_SIDE_DATA_PROP_STRUCT = (1 << 5),
 };
 
 /**
@@ -1121,6 +1128,16 @@ AVFrameSideData *av_frame_side_data_new(AVFrameSideData 
***sd, int *nb_sd,
                                         enum AVFrameSideDataType type,
                                         size_t size, unsigned int flags);
 
+/**
+ * Variant of @ref av_frame_side_data_new() to add side data entires of a type
+ * with the AV_SIDE_DATA_PROP_STRUCT prop.
+ *
+ * @return newly added side data on success, NULL on error.
+ */
+AVFrameSideData *av_frame_side_data_new_struct(AVFrameSideData ***sd, int 
*nb_sd,
+                                               enum AVFrameSideDataType type,
+                                               unsigned int flags);
+
 /**
  * Add a new side data entry to an array from an existing AVBufferRef.
  *
diff --git a/libavutil/mastering_display_metadata.c 
b/libavutil/mastering_display_metadata.c
index dd37ed7d0e..4948f30523 100644
--- a/libavutil/mastering_display_metadata.c
+++ b/libavutil/mastering_display_metadata.c
@@ -24,9 +24,12 @@
 
 #include "mastering_display_metadata.h"
 #include "mem.h"
+#include "side_data.h"
 
-static void get_defaults(AVMasteringDisplayMetadata *mastering)
+void ff_mdm_get_defaults(void *obj)
 {
+    AVMasteringDisplayMetadata *mastering = obj;
+
     for (int i = 0; i < 3; i++)
         for (int j = 0; j < 2; j++)
             mastering->display_primaries[i][j] = (AVRational) { 0, 1 };
@@ -47,7 +50,7 @@ AVMasteringDisplayMetadata 
*av_mastering_display_metadata_alloc_size(size_t *siz
     if (!mastering)
         return NULL;
 
-    get_defaults(mastering);
+    ff_mdm_get_defaults(mastering);
 
     if (size)
         *size = sizeof(*mastering);
@@ -64,7 +67,7 @@ AVMasteringDisplayMetadata 
*av_mastering_display_metadata_create_side_data(AVFra
         return NULL;
 
     memset(side_data->data, 0, sizeof(AVMasteringDisplayMetadata));
-    get_defaults((AVMasteringDisplayMetadata *)side_data->data);
+    ff_mdm_get_defaults(side_data->data);
 
     return (AVMasteringDisplayMetadata *)side_data->data;
 }
diff --git a/libavutil/side_data.c b/libavutil/side_data.c
index beb8ea3212..49197e321d 100644
--- a/libavutil/side_data.c
+++ b/libavutil/side_data.c
@@ -24,43 +24,78 @@
 #include "mem.h"
 #include "side_data.h"
 
+// headers for struct sizes
+#include "libavcodec/defs.h"
+#include "ambient_viewing_environment.h"
+#include "downmix_info.h"
+#include "hdr_dynamic_metadata.h"
+#include "hdr_dynamic_vivid_metadata.h"
+#include "mastering_display_metadata.h"
+#include "motion_vector.h"
+#include "replaygain.h"
+#include "spherical.h"
+#include "stereo3d.h"
+
 typedef struct FFSideDataDescriptor {
     AVSideDataDescriptor p;
+
+    void (*init)(void *obj);
+
+    size_t size;
 } FFSideDataDescriptor;
 
 static const FFSideDataDescriptor sd_props[] = {
-    [AV_FRAME_DATA_PANSCAN]                     = { .p = { "AVPanScan",        
                            AV_SIDE_DATA_PROP_SIZE_DEPENDENT } },
+    [AV_FRAME_DATA_PANSCAN]                     = { .p = { "AVPanScan",        
                            AV_SIDE_DATA_PROP_STRUCT | 
AV_SIDE_DATA_PROP_SIZE_DEPENDENT },
+                                                    .size = sizeof(AVPanScan) 
},
     [AV_FRAME_DATA_A53_CC]                      = { .p = { "ATSC A53 Part 4 
Closed Captions" } },
     [AV_FRAME_DATA_MATRIXENCODING]              = { .p = { "AVMatrixEncoding", 
                            AV_SIDE_DATA_PROP_CHANNEL_DEPENDENT } },
-    [AV_FRAME_DATA_DOWNMIX_INFO]                = { .p = { "Metadata relevant 
to a downmix procedure",     AV_SIDE_DATA_PROP_CHANNEL_DEPENDENT } },
+    [AV_FRAME_DATA_DOWNMIX_INFO]                = { .p = { "Metadata relevant 
to a downmix procedure",     AV_SIDE_DATA_PROP_STRUCT | 
AV_SIDE_DATA_PROP_CHANNEL_DEPENDENT },
+                                                    .size = 
sizeof(AVDownmixInfo) },
     [AV_FRAME_DATA_AFD]                         = { .p = { "Active format 
description" } },
-    [AV_FRAME_DATA_MOTION_VECTORS]              = { .p = { "Motion vectors",   
                            AV_SIDE_DATA_PROP_SIZE_DEPENDENT } },
+    [AV_FRAME_DATA_MOTION_VECTORS]              = { .p = { "Motion vectors",   
                            AV_SIDE_DATA_PROP_STRUCT | 
AV_SIDE_DATA_PROP_SIZE_DEPENDENT },
+                                                    .size = 
sizeof(AVMotionVector) },
     [AV_FRAME_DATA_SKIP_SAMPLES]                = { .p = { "Skip samples" } },
     [AV_FRAME_DATA_GOP_TIMECODE]                = { .p = { "GOP timecode" } },
     [AV_FRAME_DATA_S12M_TIMECODE]               = { .p = { "SMPTE 12-1 
timecode" } },
-    [AV_FRAME_DATA_DYNAMIC_HDR_PLUS]            = { .p = { "HDR Dynamic 
Metadata SMPTE2094-40 (HDR10+)",   AV_SIDE_DATA_PROP_COLOR_DEPENDENT } },
-    [AV_FRAME_DATA_DYNAMIC_HDR_VIVID]           = { .p = { "HDR Dynamic 
Metadata CUVA 005.1 2021 (Vivid)", AV_SIDE_DATA_PROP_COLOR_DEPENDENT } },
+    [AV_FRAME_DATA_DYNAMIC_HDR_PLUS]            = { .p = { "HDR Dynamic 
Metadata SMPTE2094-40 (HDR10+)",   AV_SIDE_DATA_PROP_STRUCT| 
AV_SIDE_DATA_PROP_COLOR_DEPENDENT },
+                                                    .size = 
sizeof(AVDynamicHDRPlus) },
+    [AV_FRAME_DATA_DYNAMIC_HDR_VIVID]           = { .p = { "HDR Dynamic 
Metadata CUVA 005.1 2021 (Vivid)", AV_SIDE_DATA_PROP_STRUCT | 
AV_SIDE_DATA_PROP_COLOR_DEPENDENT },
+                                                    .size = 
sizeof(AVDynamicHDRVivid) },
     [AV_FRAME_DATA_REGIONS_OF_INTEREST]         = { .p = { "Regions Of 
Interest",                          AV_SIDE_DATA_PROP_SIZE_DEPENDENT } },
     [AV_FRAME_DATA_VIDEO_ENC_PARAMS]            = { .p = { "Video encoding 
parameters" } },
-    [AV_FRAME_DATA_FILM_GRAIN_PARAMS]           = { .p = { "Film grain 
parameters" } },
+    [AV_FRAME_DATA_FILM_GRAIN_PARAMS]           = { .p = { "Film grain 
parameters",                        AV_SIDE_DATA_PROP_STRUCT } },
     [AV_FRAME_DATA_DETECTION_BBOXES]            = { .p = { "Bounding boxes for 
object detection and classification", AV_SIDE_DATA_PROP_SIZE_DEPENDENT } },
     [AV_FRAME_DATA_DOVI_RPU_BUFFER]             = { .p = { "Dolby Vision RPU 
Data",                        AV_SIDE_DATA_PROP_COLOR_DEPENDENT } },
     [AV_FRAME_DATA_DOVI_METADATA]               = { .p = { "Dolby Vision 
Metadata",                        AV_SIDE_DATA_PROP_COLOR_DEPENDENT } },
     [AV_FRAME_DATA_LCEVC]                       = { .p = { "LCEVC NAL data",   
                            AV_SIDE_DATA_PROP_SIZE_DEPENDENT } },
     [AV_FRAME_DATA_VIEW_ID]                     = { .p = { "View ID" } },
-    [AV_FRAME_DATA_STEREO3D]                    = { .p = { "Stereo 3D",        
                            AV_SIDE_DATA_PROP_GLOBAL } },
-    [AV_FRAME_DATA_REPLAYGAIN]                  = { .p = { "AVReplayGain",     
                            AV_SIDE_DATA_PROP_GLOBAL } },
+    [AV_FRAME_DATA_STEREO3D]                    = { .p = { "Stereo 3D",        
                            AV_SIDE_DATA_PROP_GLOBAL | AV_SIDE_DATA_PROP_STRUCT 
},
+                                                    .size = sizeof(AVStereo3D) 
},
+    [AV_FRAME_DATA_REPLAYGAIN]                  = { .p = { "AVReplayGain",     
                            AV_SIDE_DATA_PROP_GLOBAL | AV_SIDE_DATA_PROP_STRUCT 
},
+                                                    .size = 
sizeof(AVReplayGain) },
     [AV_FRAME_DATA_DISPLAYMATRIX]               = { .p = { "3x3 
displaymatrix",                            AV_SIDE_DATA_PROP_GLOBAL } },
     [AV_FRAME_DATA_AUDIO_SERVICE_TYPE]          = { .p = { "Audio service 
type",                           AV_SIDE_DATA_PROP_GLOBAL } },
-    [AV_FRAME_DATA_MASTERING_DISPLAY_METADATA]  = { .p = { "Mastering display 
metadata",                   AV_SIDE_DATA_PROP_GLOBAL | 
AV_SIDE_DATA_PROP_COLOR_DEPENDENT } },
-    [AV_FRAME_DATA_CONTENT_LIGHT_LEVEL]         = { .p = { "Content light 
level metadata",                 AV_SIDE_DATA_PROP_GLOBAL | 
AV_SIDE_DATA_PROP_COLOR_DEPENDENT } },
-    [AV_FRAME_DATA_AMBIENT_VIEWING_ENVIRONMENT] = { .p = { "Ambient viewing 
environment",                  AV_SIDE_DATA_PROP_GLOBAL } },
-    [AV_FRAME_DATA_SPHERICAL]                   = { .p = { "Spherical 
Mapping",                            AV_SIDE_DATA_PROP_GLOBAL | 
AV_SIDE_DATA_PROP_SIZE_DEPENDENT } },
+    [AV_FRAME_DATA_MASTERING_DISPLAY_METADATA]  = { .p = { "Mastering display 
metadata",                   AV_SIDE_DATA_PROP_GLOBAL | 
AV_SIDE_DATA_PROP_STRUCT | AV_SIDE_DATA_PROP_COLOR_DEPENDENT },
+                                                    .init = 
ff_mdm_get_defaults,
+                                                    .size = 
sizeof(AVMasteringDisplayMetadata) },
+    [AV_FRAME_DATA_CONTENT_LIGHT_LEVEL]         = { .p = { "Content light 
level metadata",                 AV_SIDE_DATA_PROP_GLOBAL | 
AV_SIDE_DATA_PROP_STRUCT | AV_SIDE_DATA_PROP_COLOR_DEPENDENT },
+                                                    .size = 
sizeof(AVContentLightMetadata) },
+    [AV_FRAME_DATA_AMBIENT_VIEWING_ENVIRONMENT] = { .p = { "Ambient viewing 
environment",                  AV_SIDE_DATA_PROP_GLOBAL | 
AV_SIDE_DATA_PROP_STRUCT },
+                                                    .init = 
ff_ave_get_defaults,
+                                                    .size = 
sizeof(AVAmbientViewingEnvironment) },
+    [AV_FRAME_DATA_SPHERICAL]                   = { .p = { "Spherical 
Mapping",                            AV_SIDE_DATA_PROP_GLOBAL | 
AV_SIDE_DATA_PROP_STRUCT | AV_SIDE_DATA_PROP_SIZE_DEPENDENT },
+                                                    .init = 
ff_spherical_get_defaults,
+                                                    .size = 
sizeof(AVSphericalMapping) },
     [AV_FRAME_DATA_ICC_PROFILE]                 = { .p = { "ICC profile",      
                            AV_SIDE_DATA_PROP_GLOBAL | 
AV_SIDE_DATA_PROP_COLOR_DEPENDENT } },
     [AV_FRAME_DATA_SEI_UNREGISTERED]            = { .p = { "H.26[45] User Data 
Unregistered SEI message",  AV_SIDE_DATA_PROP_MULTI } },
     [AV_FRAME_DATA_VIDEO_HINT]                  = { .p = { "Encoding video 
hint",                          AV_SIDE_DATA_PROP_SIZE_DEPENDENT } },
 };
 
+static const FFSideDataDescriptor *dp_from_desc(const AVSideDataDescriptor 
*desc)
+{
+    return (const FFSideDataDescriptor *)desc;
+}
+
 const AVSideDataDescriptor *av_frame_side_data_desc(enum AVFrameSideDataType 
type)
 {
     unsigned t = type;
@@ -247,6 +282,24 @@ AVFrameSideData *av_frame_side_data_add(AVFrameSideData 
***sd, int *nb_sd,
     return sd_dst;
 }
 
+AVFrameSideData *av_frame_side_data_new_struct(AVFrameSideData ***sd, int 
*nb_sd,
+                                               enum AVFrameSideDataType type,
+                                               unsigned int flags)
+{
+    const AVSideDataDescriptor *desc = av_frame_side_data_desc(type);
+    const FFSideDataDescriptor *dp = dp_from_desc(desc);
+    AVFrameSideData *ret;
+
+    if (!desc || !(desc->props & AV_SIDE_DATA_PROP_STRUCT))
+        return NULL;
+
+    av_assert0(dp->size);
+    ret = av_frame_side_data_new(sd, nb_sd, type, dp->size, flags);
+    if (ret && dp->init)
+         dp->init(ret->data);
+    return ret;
+}
+
 int av_frame_side_data_clone(AVFrameSideData ***sd, int *nb_sd,
                              const AVFrameSideData *src, unsigned int flags)
 {
diff --git a/libavutil/side_data.h b/libavutil/side_data.h
index 8275aa35a5..5d25833882 100644
--- a/libavutil/side_data.h
+++ b/libavutil/side_data.h
@@ -27,4 +27,8 @@ AVFrameSideData 
*ff_frame_side_data_add_from_buf(AVFrameSideData ***sd,
                                                  enum AVFrameSideDataType type,
                                                  AVBufferRef *buf);
 
+void ff_mdm_get_defaults(void *obj);
+void ff_ave_get_defaults(void *obj);
+void ff_spherical_get_defaults(void *obj);
+
 #endif // AVUTIL_SIDE_DATA_H
diff --git a/libavutil/spherical.c b/libavutil/spherical.c
index 64ade1d0ec..4ef2c225c2 100644
--- a/libavutil/spherical.c
+++ b/libavutil/spherical.c
@@ -21,15 +21,23 @@
 #include "avstring.h"
 #include "macros.h"
 #include "mem.h"
+#include "side_data.h"
 #include "spherical.h"
 
+void ff_spherical_get_defaults(void *obj)
+{
+    AVSphericalMapping *spherical = obj;
+
+    spherical->projection = AV_SPHERICAL_RECTILINEAR;
+}
+
 AVSphericalMapping *av_spherical_alloc(size_t *size)
 {
     AVSphericalMapping *spherical = av_mallocz(sizeof(AVSphericalMapping));
     if (!spherical)
         return NULL;
 
-    spherical->projection = AV_SPHERICAL_RECTILINEAR;
+    ff_spherical_get_defaults(spherical);
 
     if (size)
         *size = sizeof(*spherical);
-- 
2.48.1

_______________________________________________
ffmpeg-devel mailing list
ffmpeg-devel@ffmpeg.org
https://ffmpeg.org/mailman/listinfo/ffmpeg-devel

To unsubscribe, visit link above, or email
ffmpeg-devel-requ...@ffmpeg.org with subject "unsubscribe".

Reply via email to