1 /*
2 Copyright (c) 2012, Broadcom Europe Ltd
3 All rights reserved.
4 
5 Redistribution and use in source and binary forms, with or without
6 modification, are permitted provided that the following conditions are met:
7     * Redistributions of source code must retain the above copyright
8       notice, this list of conditions and the following disclaimer.
9     * Redistributions in binary form must reproduce the above copyright
10       notice, this list of conditions and the following disclaimer in the
11       documentation and/or other materials provided with the distribution.
12     * Neither the name of the copyright holder nor the
13       names of its contributors may be used to endorse or promote products
14       derived from this software without specific prior written permission.
15 
16 THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
17 ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
18 WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
19 DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY
20 DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
21 (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
22 LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
23 ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24 (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
25 SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26 */
27 
28 #include "mmal_types.h"
29 #include "mmal_format.h"
30 
31 #define MMAL_ES_FORMAT_MAGIC MMAL_FOURCC('m','a','g','f')
32 #define EXTRADATA_SIZE_DEFAULT 32
33 #define EXTRADATA_SIZE_MAX (10*1024)
34 
35 typedef struct MMAL_ES_FORMAT_PRIVATE_T
36 {
37    MMAL_ES_FORMAT_T format;
38    MMAL_ES_SPECIFIC_FORMAT_T es;
39 
40    uint32_t magic;
41 
42    unsigned int extradata_size;
43    uint8_t *extradata;
44 
45    uint8_t buffer[EXTRADATA_SIZE_DEFAULT];
46 
47 } MMAL_ES_FORMAT_PRIVATE_T;
48 
49 /** Allocate a format structure */
mmal_format_alloc(void)50 MMAL_ES_FORMAT_T *mmal_format_alloc(void)
51 {
52    MMAL_ES_FORMAT_PRIVATE_T *private;
53 
54    private = vcos_malloc(sizeof(*private), "mmal format");
55    if(!private) return 0;
56    memset(private, 0, sizeof(*private));
57 
58    private->magic = MMAL_ES_FORMAT_MAGIC;
59    private->format.es = (void *)&private->es;
60    private->extradata_size = EXTRADATA_SIZE_DEFAULT;
61 
62    return &private->format;
63 }
64 
65 /** Free a format structure */
mmal_format_free(MMAL_ES_FORMAT_T * format)66 void mmal_format_free(MMAL_ES_FORMAT_T *format)
67 {
68    MMAL_ES_FORMAT_PRIVATE_T *private = (MMAL_ES_FORMAT_PRIVATE_T *)format;
69    vcos_assert(private->magic == MMAL_ES_FORMAT_MAGIC);
70    if(private->extradata) vcos_free(private->extradata);
71    vcos_free(private);
72 }
73 
74 /** Copy a format structure */
mmal_format_copy(MMAL_ES_FORMAT_T * fmt_dst,MMAL_ES_FORMAT_T * fmt_src)75 void mmal_format_copy(MMAL_ES_FORMAT_T *fmt_dst, MMAL_ES_FORMAT_T *fmt_src)
76 {
77    void *backup = fmt_dst->es;
78    *fmt_dst->es = *fmt_src->es;
79    *fmt_dst = *fmt_src;
80    fmt_dst->es = backup;
81    fmt_dst->extradata = 0;
82    fmt_dst->extradata_size = 0;
83 }
84 
85 /** Full copy of a format structure (including extradata) */
mmal_format_full_copy(MMAL_ES_FORMAT_T * fmt_dst,MMAL_ES_FORMAT_T * fmt_src)86 MMAL_STATUS_T mmal_format_full_copy(MMAL_ES_FORMAT_T *fmt_dst, MMAL_ES_FORMAT_T *fmt_src)
87 {
88    mmal_format_copy(fmt_dst, fmt_src);
89 
90    if (fmt_src->extradata_size)
91    {
92       MMAL_STATUS_T status = mmal_format_extradata_alloc(fmt_dst, fmt_src->extradata_size);
93       if (status != MMAL_SUCCESS)
94          return status;
95       fmt_dst->extradata_size = fmt_src->extradata_size;
96       memcpy(fmt_dst->extradata, fmt_src->extradata, fmt_src->extradata_size);
97    }
98    return MMAL_SUCCESS;
99 }
100 
101 /** Compare 2 format structures */
mmal_format_compare(MMAL_ES_FORMAT_T * fmt1,MMAL_ES_FORMAT_T * fmt2)102 uint32_t mmal_format_compare(MMAL_ES_FORMAT_T *fmt1, MMAL_ES_FORMAT_T *fmt2)
103 {
104    MMAL_VIDEO_FORMAT_T *video1, *video2;
105    uint32_t result = 0;
106 
107    if (fmt1->type != fmt2->type)
108       return MMAL_ES_FORMAT_COMPARE_FLAG_TYPE;
109 
110    if (fmt1->encoding != fmt2->encoding)
111       result |= MMAL_ES_FORMAT_COMPARE_FLAG_ENCODING;
112    if (fmt1->bitrate != fmt2->bitrate)
113       result |= MMAL_ES_FORMAT_COMPARE_FLAG_BITRATE;
114    if (fmt1->flags != fmt2->flags)
115       result |= MMAL_ES_FORMAT_COMPARE_FLAG_FLAGS;
116    if (fmt1->extradata_size != fmt2->extradata_size ||
117        (fmt1->extradata_size && (!fmt1->extradata || !fmt2->extradata)) ||
118        memcmp(fmt1->extradata, fmt2->extradata, fmt1->extradata_size))
119       result |= MMAL_ES_FORMAT_COMPARE_FLAG_EXTRADATA;
120 
121    /* Compare the ES specific information */
122    switch (fmt1->type)
123    {
124    case MMAL_ES_TYPE_VIDEO:
125       video1 = &fmt1->es->video;
126       video2 = &fmt2->es->video;
127       if (video1->width != video2->width || video1->height != video2->height)
128          result |= MMAL_ES_FORMAT_COMPARE_FLAG_VIDEO_RESOLUTION;
129       if (memcmp(&video1->crop, &video2->crop, sizeof(video1->crop)))
130          result |= MMAL_ES_FORMAT_COMPARE_FLAG_VIDEO_CROPPING;
131       if (memcmp(&video1->par, &video2->par, sizeof(video1->par)))
132          result |= MMAL_ES_FORMAT_COMPARE_FLAG_VIDEO_ASPECT_RATIO;
133       if (memcmp(&video1->frame_rate, &video2->frame_rate, sizeof(video1->frame_rate)))
134          result |= MMAL_ES_FORMAT_COMPARE_FLAG_VIDEO_FRAME_RATE;
135       if (video1->color_space != video2->color_space)
136          result |= MMAL_ES_FORMAT_COMPARE_FLAG_VIDEO_COLOR_SPACE;
137       /* coverity[overrun-buffer-arg] We're comparing the rest of the video format structure */
138       if (memcmp(((char*)&video1->color_space) + sizeof(video1->color_space),
139                  ((char*)&video2->color_space) + sizeof(video2->color_space),
140                  sizeof(*video1) - offsetof(MMAL_VIDEO_FORMAT_T, color_space) - sizeof(video1->color_space)))
141          result |= MMAL_ES_FORMAT_COMPARE_FLAG_ES_OTHER;
142       break;
143    case MMAL_ES_TYPE_AUDIO:
144       if (memcmp(fmt1->es, fmt2->es, sizeof(MMAL_AUDIO_FORMAT_T)))
145          result |= MMAL_ES_FORMAT_COMPARE_FLAG_ES_OTHER;
146       break;
147    case MMAL_ES_TYPE_SUBPICTURE:
148       if (memcmp(fmt1->es, fmt2->es, sizeof(MMAL_SUBPICTURE_FORMAT_T)))
149          result |= MMAL_ES_FORMAT_COMPARE_FLAG_ES_OTHER;
150       break;
151    default:
152       break;
153    }
154 
155    return result;
156 }
157 
158 /** */
mmal_format_extradata_alloc(MMAL_ES_FORMAT_T * format,unsigned int size)159 MMAL_STATUS_T mmal_format_extradata_alloc(MMAL_ES_FORMAT_T *format, unsigned int size)
160 {
161    MMAL_ES_FORMAT_PRIVATE_T *private = (MMAL_ES_FORMAT_PRIVATE_T *)format;
162 
163    /* Sanity check the size requested */
164    if(size > EXTRADATA_SIZE_MAX)
165       return MMAL_EINVAL;
166 
167    /* Allocate memory if needed */
168    if(private->extradata_size < size)
169    {
170       if(private->extradata) vcos_free(private->extradata);
171       private->extradata = vcos_malloc(size, "mmal format extradata");
172       if(!private->extradata)
173          return MMAL_ENOMEM;
174       private->extradata_size = size;
175    }
176 
177    /* Set the fields in the actual format structure */
178    if(private->extradata) private->format.extradata = private->extradata;
179    else private->format.extradata = private->buffer;
180 
181    return MMAL_SUCCESS;
182 }
183