avutil/aes-test: fix memleak

Signed-off-by: James Almer <jamrial@gmail.com>

diff --git a/libavutil/aes-test.c b/libavutil/aes-test.c
index 339aa5a6e4ef7a07ffccfe5b46af7f6b9ef7d5ab..137df3060e2432fea039440a2d125a9476d37eb2 100644 (file)
--- a/libavutil/aes-test.c
+++ b/libavutil/aes-test.c
@@ -22,6 +22,7 @@
 #include "aes.h"
 #include "lfg.h"
 #include "log.h"
+#include "mem.h"
 
 int main(int argc, char **argv)
 {
@@ -64,6 +65,7 @@ int main(int argc, char **argv)
             }
         }
     }
+    av_free(b);
 
     if (argc > 1 && !strcmp(argv[1], "-t")) {
         struct AVAES *ae, *ad;
@@ -108,6 +110,8 @@ int main(int argc, char **argv)
                 }
             }
         }
+        av_free(ae);
+        av_free(ad);
     }
     return err;
 }