Modify GenFfsFileTask.

[mirror_edk2.git] / Tools / Source / FrameworkTasks / org / tianocore / framework / tasks / GenSectionTask.java

diff --git a/Tools/Source/FrameworkTasks/org/tianocore/framework/tasks/GenSectionTask.java b/Tools/Source/FrameworkTasks/org/tianocore/framework/tasks/GenSectionTask.java
index b1bf83c2a786accd6833a641054016aff642c5d7..4cd7a96cdbd76fc71dd2e0ea72cf5e2dcebf8418 100644 (file)
--- a/Tools/Source/FrameworkTasks/org/tianocore/framework/tasks/GenSectionTask.java
+++ b/Tools/Source/FrameworkTasks/org/tianocore/framework/tasks/GenSectionTask.java
@@ -19,7 +19,6 @@ package org.tianocore.framework.tasks;
 import java.io.ByteArrayOutputStream;\r
 import java.io.DataOutputStream;\r
 import java.io.File;\r
-import java.io.file;\r
 import java.util.ArrayList;\r
 import java.util.Iterator;\r
 import java.util.List;\r
@@ -304,11 +303,12 @@ public class GenSectionTask extends Task implements EfiDefine, Section, FfsTypes
                         System.out.print(e.getMessage());\r
                         throw new BuildException(\r
                                 "Compress.toBuffer failed at section");\r
+                    } finally {\r
+                        if (Do != null){\r
+                            Do.close();\r
+                        }\r
                     }\r
-\r
                 }\r
-                Do.close();\r
-\r
                 //\r
                 // Call compress\r
                 //\r
@@ -365,7 +365,7 @@ public class GenSectionTask extends Task implements EfiDefine, Section, FfsTypes
                 }\r
             } catch (Exception e) {\r
                 throw new BuildException("compress.toBuffer failed!\n");\r
-            }\r
+            } \r
         } else {\r
             Section sect;\r
             Iterator sectionIter = this.sectFileList.iterator();\r