Only add 1 byte to av_malloc(0) when it actually returned NULL

Signed-off-by: Michael Niedermayer <michaelni@gmx.at>

1 files changed, 2 insertions, 2 deletions

diff --git a/libavutil/mem.c b/libavutil/mem.c
index 83e8c35561..42beb503a1 100644
--- a/libavutil/mem.c
+++ b/libavutil/mem.c
@@ -71,8 +71,6 @@ void *av_malloc(size_t size)
     /* let's disallow possible ambiguous cases */
     if(size > (INT_MAX-32) )
         return NULL;
-    else if(!size)
-        size= 1;
 
 #if CONFIG_MEMALIGN_HACK
     ptr = malloc(size+32);
@@ -113,6 +111,8 @@ void *av_malloc(size_t size)
 #else
     ptr = malloc(size);
 #endif
+    if(!ptr && !size)
+        ptr= av_malloc(1);
     return ptr;
 }