]> git.proxmox.com Git - mirror_edk2.git/blobdiff - Tools/Python/MkFar.py
Import BaseMemoryLibMmx;
[mirror_edk2.git] / Tools / Python / MkFar.py
index 7531527f582bfc522110fee95b5dd77f88f73abe..2848b1de6596093c432236a13fe9eb897f3ee653 100755 (executable)
@@ -1,5 +1,8 @@
 #!/usr/bin/env python
 
+"""This is a python script that takes user input from the command line and
+creates a far (Framework Archive Manifest) file for distribution."""
+
 import os, sys, getopt, string, xml.dom.minidom, zipfile, md5
 from XmlRoutines import *
 from WorkspaceRoutines import *
@@ -12,16 +15,28 @@ class Far:
     far.FarName=""
     far.Version=""
     far.License=""
+    far.Abstract=""
     far.Description=""
     far.Copyright=""
-    far.SpdFiles=""
-    far.FpdFile=""
-    far.ExtraFile=""
+    far.SpdFiles=[]
+    far.FpdFiles=[]
+    far.ExtraFiles=[]
 
 far = Far()
+"""The far object is constructed from the template file the user passed in."""
+
+def AddToZip(zip, infile):
+
+  """Add a file to a zip file, provided it is not already there."""
+
+  if not infile in zip.namelist():
+    zip.write(inWorkspace(infile), infile)
 
 def parseMsa(msaFile, spdDir):
 
+  """Parse an msa file and return a list of all the files that this msa
+  includes."""
+
   filelist = [msaFile]
 
   msaDir = os.path.dirname(msaFile)
@@ -31,7 +46,7 @@ def parseMsa(msaFile, spdDir):
   xmlPaths = [
     "/ModuleSurfaceArea/SourceFiles/Filename",
     "/ModuleSurfaceArea/NonProcessedFiles/Filename" ]
-    
+
   for xmlPath in xmlPaths:
     for f in XmlList(msa, xmlPath):
       filelist.append(str(os.path.join(msaDir, XmlElementData(f))))
@@ -40,16 +55,25 @@ def parseMsa(msaFile, spdDir):
 
 def parseSpd(spdFile):
 
+  """Parse an spd file and return a list of all the files that this spd
+  includes."""
+
   files = []
 
   spdDir = os.path.dirname(spdFile)
 
   spd = xml.dom.minidom.parse(inWorkspace(spdFile))
 
+  # We are currently ignoring these hints.
+  readonly = XmlElement(spd, "/PackageSurfaceArea/PackageDefinitions/ReadOnly") != "false"
+  repackage = XmlElement(spd, "/PackageSurfaceArea/PackageDefinitions/RePackage") != "false"
+
   xmlPaths = [
     "/PackageSurfaceArea/LibraryClassDeclarations/LibraryClass/IncludeHeader",
-    "/PackageSurfaceArea/IndustryStdIncludes/IndustryStdHeader/IncludeHeader",
-    "/PackageSurfaceArea/PackageHeaders/IncludePkgHeader" ]
+    "/PackageSurfaceArea/IndustryStdIncludes/IndustryStdHeader/IncludeHeader" ]
+
+    # These are covered by the Industry Standard Includes.
+    # "/PackageSurfaceArea/PackageHeaders/IncludePkgHeader"
 
   for xmlPath in xmlPaths:
     for f in XmlList(spd, xmlPath):
@@ -62,7 +86,8 @@ def parseSpd(spdFile):
   cwd = os.getcwd()
   os.chdir(inWorkspace(spdDir))
   for root, dirs, entries in os.walk("Include"):
-    for r  in ["CVS", ".svn"]:
+    # Some files need to be skipped.
+    for r in ["CVS", ".svn"]:
       if r in dirs:
         dirs.remove(r)
     for entry in entries:
@@ -73,42 +98,34 @@ def parseSpd(spdFile):
 
 def makeFarHeader(doc):
 
-  header = doc.createElement("FarHeader")
-  name = doc.createElement("FarName")
-  name.appendChild(doc.createTextNode(far.FarName))
-  header.appendChild(name)
-  guidVal = doc.createElement("GuidValue")
-  guidVal.appendChild(doc.createTextNode(genguid()))
-  header.appendChild(guidVal)
-  ver = doc.createElement("Version")
-  ver.appendChild(doc.createTextNode(far.Version))
-  header.appendChild(ver)
-  abstract = doc.createElement("Abstract")
-  abstract.appendChild(doc.createTextNode(far.Abstract))
-  header.appendChild(abstract)
-  desc = doc.createElement("Description")
-  desc.appendChild(doc.createTextNode(far.Description))
-  header.appendChild(desc)
-  copy = doc.createElement("Copyright")
-  copy.appendChild(doc.createTextNode(far.Copyright))
-  header.appendChild(copy)
-  lic = doc.createElement("License")
-  lic.appendChild(doc.createTextNode(far.License))
-  header.appendChild(lic)
-  spec = doc.createElement("Specification")
-  spec.appendChild(doc.createTextNode("FRAMEWORK_BUILD_PACKAGING_SPECIFICATION 0x00000052"))
-  header.appendChild(spec)
+  """Create a dom tree for the Far Header. It will use information from the
+  template file passed on the command line, if present."""
+  header = XmlAppendChildElement(doc.documentElement, "FarHeader")
+
+  XmlAppendChildElement(header, "FarName", far.FarName)
+  XmlAppendChildElement(header, "GuidValue", genguid())
+  XmlAppendChildElement(header, "Version", far.Version)
+  XmlAppendChildElement(header, "Abstract", far.Abstract)
+  XmlAppendChildElement(header, "Description", far.Description)
+  XmlAppendChildElement(header, "Copyright", far.Copyright)
+  XmlAppendChildElement(header, "License", far.License)
+  XmlAppendChildElement(header, "Specification", "FRAMEWORK_BUILD_PACKAGING_SPECIFICATION 0x00000052")
 
   return header
 
 def getSpdGuidVersion(spdFile):
 
+  """Returns a tuple (guid, version) which is read from the given spdFile."""
+
   spd = xml.dom.minidom.parse(inWorkspace(spdFile))
 
   return (XmlElement(spd, "/PackageSurfaceArea/SpdHeader/GuidValue"),
           XmlElement(spd, "/PackageSurfaceArea/SpdHeader/Version"))
 
-def makeFar(filelist, farname):
+def makeFar(files, farname):
+
+  """Make a far out of the given filelist and writes it to the file farname."""
 
   domImpl = xml.dom.minidom.getDOMImplementation()
   man = domImpl.createDocument(None, "FrameworkArchiveManifest", None)
@@ -116,22 +133,19 @@ def makeFar(filelist, farname):
 
   top_element.appendChild(makeFarHeader(man))
 
-  packList = man.createElement("FarPackageList")
-  top_element.appendChild(packList)
-
-  platList = man.createElement("FarPlatformList")
-  top_element.appendChild(platList)
-
-  contents = man.createElement("Contents")
-  top_element.appendChild(contents)
-
-  exts = man.createElement("UserExtensions")
-  top_element.appendChild(exts)
+  packList = XmlAppendChildElement(top_element, "FarPackageList")
+  platList = XmlAppendChildElement(top_element, "FarPlatformList")
+  contents = XmlAppendChildElement(top_element, "Contents")
+  XmlAppendChildElement(top_element, "UserExtensions")
 
-  zip = zipfile.ZipFile(farname, "w")
-  for infile in filelist:
+  try:
+    zip = zipfile.ZipFile(farname, "w", zipfile.ZIP_DEFLATED)
+  except:
+    zip = zipfile.ZipFile(farname, "w", zipfile.ZIP_STORED)
+  for infile in set(files):
     if not os.path.exists(inWorkspace(infile)):
-      print "Skipping non-existent file '%s'." % infile
+      print "Error: Non-existent file '%s'." % infile
+      sys.exit()
     (_, extension) = os.path.splitext(infile)
     if extension == ".spd":
       filelist = parseSpd(infile)
@@ -139,68 +153,34 @@ def makeFar(filelist, farname):
 
       (spdGuid, spdVersion) = getSpdGuidVersion(infile)
 
-      package = man.createElement("FarPackage")
-      packList.appendChild(package)
-
-      spdfilename = farFileNode(man, inWorkspace(infile))
-      zip.write(inWorkspace(infile), infile)
-      spdfilename.appendChild(man.createTextNode(infile))
-      package.appendChild(spdfilename)
-
-      guidValue = man.createElement("GuidValue")
-      guidValue.appendChild(man.createTextNode(spdGuid))
-      package.appendChild(guidValue)
-
-      version = man.createElement("Version")
-      version.appendChild(man.createTextNode(spdVersion))
-      package.appendChild(version)
-
-      defaultPath = man.createElement("DefaultPath")
-      defaultPath.appendChild(man.createTextNode(spdDir))
-      package.appendChild(defaultPath)
-
-      farPlatformList = man.createElement("FarPlatformList")
-      package.appendChild(farPlatformList)
-
-      packContents = man.createElement("Contents")
-      package.appendChild(packContents)
-
-      ue = man.createElement("UserExtensions")
-      package.appendChild(ue)
+      package = XmlAppendChildElement(packList, "FarPackage")
+      XmlAppendChildElement(package, "FarFilename", lean(infile), {"Md5Sum": Md5(inWorkspace(infile))})
+      AddToZip(zip, infile)
+      XmlAppendChildElement(package, "GuidValue", spdGuid)
+      XmlAppendChildElement(package, "Version", spdVersion)
+      XmlAppendChildElement(package, "DefaultPath", spdDir)
+      XmlAppendChildElement(package, "FarPlatformList")
+      packContents = XmlAppendChildElement(package, "Contents")
+      XmlAppendChildElement(package, "UserExtensions")
 
       for spdfile in filelist:
-        content = farFileNode(man, inWorkspace(os.path.join(spdDir, spdfile))) 
-        zip.write(inWorkspace(os.path.join(spdDir, spdfile)), spdfile)
-        content.appendChild(man.createTextNode(spdfile))
-        packContents.appendChild(content)
+        XmlAppendChildElement(packContents, "FarFilename", lean(spdfile), {"Md5Sum": Md5(inWorkspace(os.path.join(spdDir, spdfile)))})
+        AddToZip(zip, os.path.join(spdDir,spdfile))
 
     elif extension == ".fpd":
 
-      platform = man.createElement("FarPlatform")
-      platList.appendChild(platform)
-
-      fpdfilename = farFileNode(man, inWorkspace(infile))
-      zip.write(inWorkspace(infile), infile)
-      platform.appendChild(fpdfilename)
-      fpdfilename.appendChild( man.createTextNode(infile) )
+      platform = XmlAppendChildElement(platList, "FarPlatform")
+      XmlAppendChildElement(platform, "FarFilename", lean(infile), {"Md5Sum": Md5(inWorkspace(infile))})
+      AddToZip(zip, infile)
 
     else:
-      content = farFileNode(man, inWorkspace(infile))
-      zip.write(inWorkspace(infile), infile)
-      content.appendChild(man.createTextNode(infile))
-      contents.appendChild(content)
+      XmlAppendChildElement(contents, "FarFilename", lean(infile), {"Md5Sum": Md5(inWorkspace(infile))})
+      AddToZip(zip, infile)
 
-  zip.writestr("FrameworkArchiveManifest.xml", man.toprettyxml(2*" "))
+  zip.writestr("FrameworkArchiveManifest.xml", man.toxml('UTF-8'))
   zip.close()
   return
 
-def farFileNode(doc, filename):
-  content = doc.createElement("FarFilename")
-  f=open(filename, "rb")
-  content.setAttribute("Md5sum", md5.md5(f.read()).hexdigest())
-  f.close()
-  return content
-
 # This acts like the main() function for the script, unless it is 'import'ed
 # into another script.
 if __name__ == '__main__':
@@ -209,30 +189,42 @@ if __name__ == '__main__':
   # pp = pprint.PrettyPrinter(indent=2)
 
   # Process the command line args.
-  optlist, args = getopt.getopt(sys.argv[1:], 'hf:t:', [ 'template=', 'far=', 'help'])
+  optlist, args = getopt.getopt(sys.argv[1:], 'ho:t:v', [ 'template=', 'output=', 'far=', 'help', 'debug', 'verbose', 'version'])
 
+  # First pass through the options list.
   for o, a in optlist:
     if o in ["-h", "--help"]:
       print """
 Pass a list of .spd and .fpd files to be placed into a far for distribution.
 You may give the name of the far with a -f or --far option. For example:
 
-  %s --far library.far MdePkg/MdePkg.spd
+  %s --template far-template --far library.far MdePkg/MdePkg.spd
 
 The file paths of .spd and .fpd are treated as relative to the WORKSPACE
-envirnonment variable which must be set to a valid workspace root directory.
+environment variable which must be set to a valid workspace root directory.
+
+A template file may be passed in with the --template option. This template file
+is a text file that allows more contol over the contents of the far.
 """ % os.path.basename(sys.argv[0])
 
       sys.exit()
+      optlist.remove((o,a))
     if o in ["-t", "--template"]:
       # The template file is processed first, so that command line options can
       # override it.
       templateName = a
       execfile(templateName)
-    if o in ["-f", "--far"]:
+      optlist.remove((o,a))
+
+  # Second pass through the options list. These can override the first pass.
+  for o, a in optlist:
+    if o in ["-o", "--far", "--output"]:
       far.FileName = a
-      if os.path.exists(far.FileName):
-        print "Error: File %s exists. Not overwriting." % far.FileName
-        sys.exit()
 
-  makeFar(args, far.FileName)
+  # Let's err on the side of caution and not let people blow away data 
+  # accidentally.
+  if os.path.exists(far.FileName):
+    print "Error: File %s exists. Not overwriting." % far.FileName
+    sys.exit()
+
+  makeFar(far.SpdFiles + far.FpdFiles + far.ExtraFiles + args, far.FileName)