--- /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
+\r
+import org.tianocore.common.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 FileArg inputFile = new FileArg();\r
+ //\r
+ // Output file\r
+ //\r
+ private FileArg outputFile = new FileArg();\r
+ //\r
+ // Base address\r
+ //\r
+ private ToolArg baseAddr = new ToolArg();\r
+ //\r
+ // \r
+ // \r
+ private FileArg mapFile = new FileArg();\r
+ //\r
+ // Architecture\r
+ //\r
+ private String arch = "IA32";\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
+ if (isUptodate()) {\r
+ EdkLog.log(this, EdkLog.EDK_VERBOSE, outputFile.toFileList() + " is up-to-date!");\r
+ return;\r
+ }\r
+\r
+ Project project = this.getOwningTarget().getProject();\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.separator + command;\r
+ }\r
+\r
+ //\r
+ // argument of tools\r
+ //\r
+ if (mapFile.getValue().length() == 0) {\r
+ mapFile.setArg(" -M ", outputFile.getValue() + ".map");\r
+ }\r
+ argument = "" + inputFile + outputFile + baseAddr + mapFile;\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(this, EdkLog.EDK_VERBOSE, Commandline.toString(cmdline.getCommandline()));\r
+ EdkLog.log(this, EdkLog.EDK_INFO, inputFile.toFileList() + " => " \r
+ + outputFile.toFileList()\r
+ + mapFile.toFileList());\r
+\r
+ revl = runner.execute();\r
+\r
+ if (EFI_SUCCESS == revl) {\r
+ //\r
+ // command execution success\r
+ //\r
+ EdkLog.log(this, EdkLog.EDK_VERBOSE, "PeiReBase succeeded!");\r
+ } else {\r
+ //\r
+ // command execution fail\r
+ //\r
+ EdkLog.log(this, EdkLog.EDK_INFO, "ERROR = " + Integer.toHexString(revl));\r
+ throw new BuildException("PeiReBase failed!");\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.getValue();\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.setArg(" -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.getValue();\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.setArg(" -O ", outputFile);\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.getValue();\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.setArg(" -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
+ /**\r
+ Get the value of map file\r
+\r
+ @return String The map file path\r
+ **/\r
+ public String getMapFile() {\r
+ return mapFile.getValue();\r
+ }\r
+\r
+ /**\r
+ Set "-M MapFile" argument\r
+\r
+ @param mapFile The path of map file\r
+ **/\r
+ public void setMapFile(String mapFile) {\r
+ this.mapFile.setArg(" -M ", mapFile);\r
+ }\r
+\r
+ //\r
+ // Dependency check\r
+ // \r
+ private boolean isUptodate() {\r
+ File srcFile = new File(inputFile.getValue());\r
+ File dstFile = new File(outputFile.getValue());\r
+\r
+ if (srcFile.lastModified() > dstFile.lastModified()) {\r
+ return false;\r
+ }\r
+\r
+ return true;\r
+ }\r
+}\r