commit f651bcb296f6ea15217056487c904689d57a0d0f
author James Zern <jzern@google.com> Wed Feb 26 23:27:17 2014 -0800
committer James Zern <jzern@google.com> Wed Feb 26 23:27:17 2014 -0800
tree ed5c6db28e643d05caf6566b7f631b84612e39c9
parent 104d8ec034729183f9c1bb80aea6ae546276e152

y4m_video_source: fix memory leak

Begin() will be called twice with 2-pass encodes, invoking
y4m_input_open which allocates memory; close the old instance first.

Change-Id: Id252a21d286ca9ae998bd87599d43aeb8d7d77aa

test/y4m_video_source.h

diff --git a/test/y4m_video_source.h b/test/y4m_video_source.h
index 20d2be0..7419043 100644
--- a/test/y4m_video_source.h
+++ b/test/y4m_video_source.h
@@ -35,14 +35,11 @@
 
   virtual ~Y4mVideoSource() {
     vpx_img_free(img_.get());
-    y4m_input_close(&y4m_);
-    if (input_file_)
-      fclose(input_file_);
+    CloseSource();
   }
 
   virtual void Begin() {
-    if (input_file_)
-      fclose(input_file_);
+    CloseSource();
     input_file_ = OpenTestDataFile(file_name_);
     ASSERT_TRUE(input_file_ != NULL) << "Input file open failed. Filename: "
         << file_name_;
@@ -89,6 +86,15 @@
   }
 
  protected:
+  void CloseSource() {
+    y4m_input_close(&y4m_);
+    y4m_ = y4m_input();
+    if (input_file_ != NULL) {
+      fclose(input_file_);
+      input_file_ = NULL;
+    }
+  }
+
   std::string file_name_;
   FILE *input_file_;
   testing::internal::scoped_ptr<vpx_image_t> img_;