]> git.proxmox.com Git - mirror_edk2.git/blobdiff - Tools/Source/FrameworkTasks/org/tianocore/framework/tasks/PeiReBaseTask.java
Add follows warpped tianotools to frameworktask:
[mirror_edk2.git] / Tools / Source / FrameworkTasks / org / tianocore / framework / tasks / PeiReBaseTask.java
diff --git a/Tools/Source/FrameworkTasks/org/tianocore/framework/tasks/PeiReBaseTask.java b/Tools/Source/FrameworkTasks/org/tianocore/framework/tasks/PeiReBaseTask.java
new file mode 100644 (file)
index 0000000..c033c0d
--- /dev/null
@@ -0,0 +1,267 @@
+/** @file\r
+ PeiReBaseTask class.\r
+\r
+ PeiReBaseTask is used to call PeiReBase.exe to rebase efi fv file.\r
\r
\r
+ Copyright (c) 2006, Intel Corporation\r
+ All rights reserved. This program and the accompanying materials\r
+ are licensed and made available under the terms and conditions of the BSD License\r
+ which accompanies this distribution.  The full text of the license may be found at\r
+ http://opensource.org/licenses/bsd-license.php\r
\r
+ THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
+ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
+\r
+ **/\r
+package org.tianocore.framework.tasks;\r
+\r
+import java.io.File;\r
+\r
+import org.apache.tools.ant.Task;\r
+import org.apache.tools.ant.Project;\r
+import org.apache.tools.ant.BuildException;\r
+import org.apache.tools.ant.taskdefs.Execute;\r
+import org.apache.tools.ant.taskdefs.LogStreamHandler;\r
+import org.apache.tools.ant.types.Commandline;\r
+import org.tianocore.logger.EdkLog;\r
+\r
+/**\r
+  PeiReBaseTask class.\r
+\r
+  PeiReBaseTask is used to call PeiReBase.exe to rebase efi fv file.\r
+**/\r
+public class PeiReBaseTask extends Task implements EfiDefine {\r
+    ///\r
+    /// tool name\r
+    ///\r
+    private String toolName = "PeiReBase";\r
+    // /\r
+    // / Input file\r
+    // /\r
+    private String inputFile = "";\r
+\r
+    // /\r
+    // / Output file\r
+    // /\r
+    private String outputFile = "";\r
+\r
+    // /\r
+    // / Output directory, this variable is added by jave wrap\r
+    // /\r
+    private String outputDir = "";\r
+    \r
+    ///\r
+    /// Base address\r
+    ///\r
+    private String baseAddr = "";\r
+    \r
+    ///\r
+    /// Architecture \r
+    ///\r
+    private String arch = "";\r
+    \r
+    /**\r
+     * execute\r
+     * \r
+     * PeiReBaseTask execute function is to assemble tool command line & execute\r
+     * tool command line\r
+     * \r
+     * @throws BuidException\r
+     */\r
+    public void execute() throws BuildException {\r
+\r
+        Project project = this.getOwningTarget().getProject();\r
+        //\r
+        // set Logger\r
+        //\r
+        FrameworkLogger logger = new FrameworkLogger(project, toolName.toLowerCase());\r
+        EdkLog.setLogLevel(project.getProperty("env.LOGLEVEL"));\r
+        EdkLog.setLogger(logger);\r
+        \r
+        //\r
+        // absolute path of efi tools\r
+        //\r
+        String path = project.getProperty("env.FRAMEWORK_TOOLS_PATH");\r
+        String command;\r
+        String argument;\r
+        if (this.arch.equalsIgnoreCase("IA32")){\r
+            command = toolName + "_IA32";\r
+        }else if (this.arch.equalsIgnoreCase("X64")){\r
+            command = toolName + "_X64";\r
+        }else if (this.arch.equalsIgnoreCase("IPF")){\r
+            command = toolName + "_IPF";\r
+        }else {\r
+            command = toolName + "_IA32";\r
+        }\r
+        if (path != null) {\r
+            command = path + File.separatorChar + command;\r
+        }\r
+        \r
+        //\r
+        // argument of tools\r
+        //\r
+        File file = new File(outputFile);\r
+        if (!file.isAbsolute() && (!this.outputDir.equalsIgnoreCase(""))) {\r
+            argument = inputFile + " " +  "-O " + outputDir + File.separatorChar\r
+                    + outputFile + " " + this.baseAddr;\r
+        } else {\r
+            argument = inputFile + " " + "-O " + outputFile + " " + this.baseAddr;\r
+        }\r
+        \r
+        //\r
+        // return value of fwimage execution\r
+        //\r
+        int revl = -1;\r
+\r
+        try {\r
+            Commandline cmdline = new Commandline();\r
+            cmdline.setExecutable(command);\r
+            cmdline.createArgument().setLine(argument);\r
+\r
+            LogStreamHandler streamHandler = new LogStreamHandler(this,\r
+                    Project.MSG_INFO, Project.MSG_WARN);\r
+            Execute runner = new Execute(streamHandler, null);\r
+\r
+            runner.setAntRun(project);\r
+            runner.setCommandline(cmdline.getCommandline());\r
+            //\r
+            // Set debug log information.\r
+            //\r
+            EdkLog.log(EdkLog.EDK_INFO, Commandline.toString(cmdline.getCommandline()));\r
+            \r
+            revl = runner.execute();\r
+            \r
+            if (EFI_SUCCESS == revl) {\r
+                //\r
+                // command execution success\r
+                //\r
+                EdkLog.log(EdkLog.EDK_INFO,"PeiBase succeeded!");\r
+            } else {\r
+                //\r
+                // command execution fail\r
+                //\r
+                EdkLog.log(EdkLog.EDK_ERROR, "PeiBase failed. (error="\r
+                        + Integer.toHexString(revl) + ")");\r
+                throw new BuildException("PeiBase failed. (error="\r
+                        + Integer.toHexString(revl) + ")");\r
+\r
+            }\r
+        } catch (Exception e) {\r
+            throw new BuildException(e.getMessage());\r
+        }\r
+    }\r
+\r
+    /**\r
+     * getInputFile\r
+     * \r
+     * This function is to get class member "inputFile".\r
+     * \r
+     * @return string of input file name.\r
+     */\r
+    public String getInputFile() {\r
+        return inputFile;\r
+    }\r
+\r
+    /**\r
+     * setComponentType\r
+     * \r
+     * This function is to set class member "inputFile".\r
+     * \r
+     * @param inputFile\r
+     *            string of input file name.\r
+     */\r
+    public void setInputFile(String inputFile) {\r
+        this.inputFile = "-I " + inputFile;\r
+    }\r
+\r
+    /**\r
+     * getOutputFile\r
+     * \r
+     * This function is to get class member "outputFile"\r
+     * \r
+     * @return outputFile string of output file name.\r
+     */\r
+    public String getOutputFile() {\r
+        return outputFile;\r
+    }\r
+\r
+    /**\r
+     * setOutputFile\r
+     * \r
+     * This function is to set class member "outputFile"\r
+     * \r
+     * @param outputFile\r
+     *            string of output file name.\r
+     */\r
+    public void setOutputFile(String outputFile) {\r
+        this.outputFile = outputFile;\r
+    }\r
+\r
+    /**\r
+     * getOutputDir\r
+     * \r
+     * This function is to get class member "outputDir"\r
+     * \r
+     * @return outputDir string of output directory.\r
+     */\r
+    public String getOutputDir() {\r
+        return outputDir;\r
+    }\r
+\r
+    /**\r
+     * setOutputDir\r
+     * \r
+     * This function is to set class member "outputDir"\r
+     * \r
+     * @param outputDir\r
+     *            string of output directory.\r
+     */\r
+    public void setOutputDir(String outputDir) {\r
+        this.outputDir = outputDir;\r
+    }\r
+\r
+    /**\r
+     * getBaseAddr\r
+     * \r
+     * This function is to get class member "baseAddr"\r
+     * \r
+     * @return baseAddr   string of base address.\r
+     */\r
+    public String getBaseAddr() {\r
+        return baseAddr;\r
+    }\r
+\r
+    /**\r
+     * setBaseAddr\r
+     * \r
+     * This function is to set class member "baseAddr"\r
+     * \r
+     * @param baseAddr    string of base address\r
+     */\r
+    public void setBaseAddr(String baseAddr) {\r
+        this.baseAddr = "-B " +  baseAddr;\r
+    }\r
+\r
+    /**\r
+     * getArch\r
+     * \r
+     * This function is to get class member "arch".\r
+     * \r
+     * @return arch       Architecture\r
+     */\r
+    public String getArch() {\r
+        return arch;\r
+    }\r
+\r
+    /**\r
+     * setArch\r
+     * \r
+     * This function is to set class member "arch" \r
+     * \r
+     * @param arch         Architecture\r
+     */\r
+    public void setArch(String arch) {\r
+        this.arch = arch;\r
+    }\r
+}\r