1 files changed, 23 insertions, 1 deletions
diff --git a/Tools/Java/Source/FrameworkTasks/org/tianocore/framework/tasks/PeiReBaseTask.java b/Tools/Java/Source/FrameworkTasks/org/tianocore/framework/tasks/PeiReBaseTask.java
index 1f8a4d1f3c..07154921d2 100644
--- a/Tools/Java/Source/FrameworkTasks/org/tianocore/framework/tasks/PeiReBaseTask.java
+++ b/Tools/Java/Source/FrameworkTasks/org/tianocore/framework/tasks/PeiReBaseTask.java
@@ -50,7 +50,11 @@ public class PeiReBaseTask extends Task implements EfiDefine {
     //
     private ToolArg baseAddr = new ToolArg();
     //
+    // Fv.inf file
     // 
+    private FileArg fvinfFile = new FileArg();
+    //
+    // map file
     // 
     private FileArg mapFile = new FileArg();
     //
@@ -99,7 +103,7 @@ public class PeiReBaseTask extends Task implements EfiDefine {
         if (mapFile.getValue().length() == 0) {
             mapFile.setArg(" -M ", outputFile.getValue() + ".map");
         }
-        argument = "" + inputFile + outputFile + baseAddr + mapFile;
+        argument = "" + inputFile + outputFile + baseAddr + fvinfFile + mapFile;
 
         //
         // return value of fwimage execution
@@ -235,6 +239,24 @@ public class PeiReBaseTask extends Task implements EfiDefine {
     }
 
     /**
+       Get the value of fv.inf file
+
+       @return String   The fv.inf file path
+     **/
+    public String getFvInfFile() {
+        return fvinfFile.getValue();
+    }
+
+    /**
+       Set "-F FvinfFile" argument
+
+       @param fvinfFile   The path of fv.inf file
+     **/
+    public void setFvInfFile(String fvinfFile) {
+        this.fvinfFile.setArg(" -F ", fvinfFile);
+    }
+
+    /**
        Get the value of map file
 
        @return String   The map file path