--- /dev/null
+/** @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