avdevice/dshow_enummediatypes: check return of av_malloc

Untested. Reviewed-by: Michael Niedermayer <[email protected]> Signed-off-by: Ganesh Ajjanagadde <[email protected]>
Bingnan · Nov 6, 2015 · 6f1ddc7 · 6f1ddc7
1 parent b456ece
commit 6f1ddc7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/libavdevice/dshow_enummediatypes.c b/libavdevice/dshow_enummediatypes.c
@@ -37,6 +37,8 @@ libAVEnumMediaTypes_Next(libAVEnumMediaTypes *this, unsigned long n,
     if (!this->pos && n == 1) {
         if (!IsEqualGUID(&this->type.majortype, &GUID_NULL)) {
             AM_MEDIA_TYPE *type = av_malloc(sizeof(AM_MEDIA_TYPE));
+            if (!type)
+                return E_OUTOFMEMORY;
             ff_copy_dshow_media_type(type, &this->type);
             *types = type;
             count = 1;