Adding several dependency checks for far installation. Redoing the XML output.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@2299 6f19259b-4bc3-4df7-8a09-765794883524
This commit is contained in:
bbahnsen
2007-01-24 18:14:09 +00:00
parent 92a154063a
commit 312ffece7b
3 changed files with 149 additions and 149 deletions

View File

@ -7,11 +7,18 @@ import os, sys, getopt, string, xml.dom.minidom, zipfile, md5
from XmlRoutines import *
from WorkspaceRoutines import *
verbose = False
force = False
class Flags:
"""Keep track of some command line flags and operating modes."""
def __init__(self):
self.verbose = False
self.force = False
self.reinstall = False
class Database:
"""This class encapsulates the FrameworkDatabase file for the workspace we
are operating on."""
def __init__(self, filename="Tools/Conf/FrameworkDatabase.db"):
# First try to get a lock file.
@ -35,13 +42,13 @@ class Database:
for spdfile in XmlList(self.dom, "/FrameworkDatabase/PackageList/Filename"):
filename = str(XmlElementData(spdfile))
spd = XmlParseFileSection(inWorkspace(filename), "SpdHeader")
self.installedPackages[XmlElement(spd, "/SpdHeader/GuidValue"), XmlElement(spd, "/SpdHeader/Version")] = \
self.installedPackages[GetSpdGuidVersion(spd, 1)] = \
XmlElement(spd, "/SpdHeader/PackageName")
for fpdfile in XmlList(self.dom, "/FrameworkDatabase/PlatformList/Filename"):
filename = str(XmlElementData(fpdfile))
fpd = XmlParseFileSection(inWorkspace(filename), "PlatformHeader")
self.installedPlatforms[XmlElement(fpd, "/PlatformHeader/GuidValue"), XmlElement(fpd, "/PlatformHeader/Version") ] = \
self.installedPlatforms[GetFpdGuidVersion(fpd, 1)] = \
XmlElement(fpd, "/PlatformHeader/PlatformName")
for farfile in XmlList(self.dom, "/FrameworkDatabase/FarList/Filename"):
@ -80,22 +87,19 @@ class Database:
return self.installedFars.has_key(farguid)
def AddPackage(self, f):
filename = self.dom.createElement("Filename")
filename.appendChild(self.dom.createTextNode(f))
self.packageList.appendChild(filename)
"""Put this package in the database"""
XmlAppendChildElement(self.packageList, "Filename", f)
def AddPlatform(self, f):
filename = self.dom.createElement("Filename")
filename.appendChild(self.dom.createTextNode(f))
self.platformList.appendChild(filename)
"""Put this platform in the database"""
XmlAppendChildElement(self.platformList, "Filename", f)
def AddFar(self, f, guid=""):
filename = self.dom.createElement("Filename")
filename.setAttribute("FarGuid", guid)
filename.appendChild(self.dom.createTextNode(f))
self.farList.appendChild(filename)
"""Put this far in the database"""
XmlAppendChildElement(self.farList, "Filename", f, {"FarGuid":guid} )
def Write(self):
"""Save the Xml tree out to the file."""
if True:
XmlSaveFile(self.dom, self.DBFile)
else:
@ -103,9 +107,9 @@ class Database:
f.write(self.dom.toprettyxml(2*" "))
f.close()
def ExtractFile(zip, file, workspaceLocation=""):
if verbose:
def ExtractFile(zip, file, defaultDir="", workspaceLocation="", md5sum=""):
"""Unzip a file."""
if flags.verbose:
print "Extracting ", file
destFile = os.path.join(inWorkspace(workspaceLocation), str(file))
@ -114,25 +118,37 @@ def ExtractFile(zip, file, workspaceLocation=""):
mkdir(destDir)
f = open(destFile, "w")
f.write(zip.read(file))
contents = zip.read(os.path.join(defaultDir,file))
if md5sum and (md5.md5(contents).hexdigest() != md5sum):
print "Error: The md5 sum does not match on file %s." % file
f.write(contents)
f.close()
def GetFpdGuidVersion(Dom):
def GetFpdGuidVersion(Dom, strip=0):
"""Get the Guid and version of the fpd from a dom object."""
return XmlElement(Dom, "/PlatformSurfaceArea/PlatformHeader/GuidValue"), \
XmlElement(Dom, "/PlatformSurfaceArea/PlatformHeader/Version")
gpath = ["PlatformSurfaceArea", "PlatformHeader", "GuidValue"]
vpath = ["PlatformSurfaceArea", "PlatformHeader", "Version"]
def GetSpdGuidVersion(Dom):
return string.lower(XmlElement(Dom, "/".join(gpath[strip:]))), \
XmlElement(Dom, "/".join(vpath[strip:]))
def GetSpdGuidVersion(Dom, strip=0):
"""Get the Guid and version of the spd from a dom object."""
return XmlElement(Dom, "/PackageSurfaceArea/SpdHeader/GuidValue"), \
XmlElement(Dom, "/PackageSurfaceArea/SpdHeader/Version")
gpath = ["PackageSurfaceArea", "SpdHeader", "GuidValue"]
vpath = ["PackageSurfaceArea", "SpdHeader", "Version"]
return string.lower(XmlElement(Dom, "/".join(gpath[strip:]))), \
XmlElement(Dom, "/".join(vpath[strip:]))
def InstallFar(farfile, workspaceLocation=""):
"""Unpack the far an install it in the workspace. We need to adhere to the
rules of far handling."""
far = zipfile.ZipFile(farfile, "r")
# Use this list to make sure we get everything from the far.
@ -151,21 +167,21 @@ def InstallFar(farfile, workspaceLocation=""):
# Check the packages
for farPackage in XmlList(manifest, "/FrameworkArchiveManifest/FarPackageList/FarPackage/FarFilename"):
spdfile = str(XmlElementData(farPackage))
spd = XmlParseString(far.read(spdfile))
packageGV = GetSpdGuidVersion(spd)
spd = XmlParseStringSection(far.read(spdfile), "SpdHeader")
packageGV = GetSpdGuidVersion(spd, 1)
if fdb.HasPackage(packageGV):
print "Error: This package is already installed: ", spdfile
installError = True
if not flags.reinstall:
print "Error: This package is already installed: ", spdfile
installError = True
# Build up a list of the package guid versions that this far is bringing in.
# This is needed to satisfy dependencies of msas that are in the other packages of
# this far.
farSpds.append(packageGV)
spdDoms.append(spd)
spdDoms.append((spd, spdfile))
for spd in spdDoms:
for spd, spdfile in spdDoms:
# Now we need to get a list of every msa in this spd and check the package dependencies.
for msafile in XmlList(spd, "/PackageSurfaceArea/MsaFiles/Filename"):
msafilePath = str(os.path.join(os.path.dirname(spdfile), XmlElementData(msafile)))
@ -176,29 +192,52 @@ def InstallFar(farfile, workspaceLocation=""):
guid = package.getAttribute("PackageGuid")
version = package.getAttribute("PackageVersion")
# Does anyone provide this package?
if not fdb.HasPackage((guid, version)) and not (guid, version) in farSpds:
print "The module %s depends on the package guid % version %s, which is not installed in the workspace." \
% (msafilePath, guid, version)
print ("Error: The module %s depends on the package guid %s version %s, which " + \
"is not installed in the workspace, nor is it provided by this far.") \
% (msafilePath, guid, version)
installError = True
# Check the platforms
for farPlatform in XmlList(manifest, "/FrameworkArchiveManifest/FarPlatformList/FarPlatform/FarFilename"):
fpdfile = str(XmlElementData(farPlatform))
fpd = XmlParseString(far.read(fpdfile))
if fdb.HasPlatform(GetFpdGuidVersion(fpd)):
print "Error: This platform is already installed: ", fpdfile
installError = True
if fdb.HasPlatform(GetFpdGuidVersion(fpd, 0)):
if not flags.reinstall:
print "Error: This platform is already installed: ", fpdfile
installError = True
# Now we need to check that all the Platforms (and modules?) that are
# referenced by this fpd are installed in the workspace or are in this far.
packagesNeeded = set()
# Go through the dependencies
for dependency in XmlList(fpd, "/PlatformSurfaceArea/FrameworkModules/ModuleSA") + \
XmlList(fpd, "/PlatformSurfaceArea/FrameworkModules/ModuleSA/Libraries/Instance"):
packagesNeeded.add((string.lower(dependency.getAttribute("PackageGuid")),
dependency.getAttribute("PackageVersion")))
# Let's see if all the packages are in the workspace
for guid, version in packagesNeeded:
# Does anyone provide this package?
if not fdb.HasPackage((guid, version)) and not (guid, version) in farSpds:
print ("Error: The fpd %s depends on the package guid %s version %s, which " + \
"is not installed in the workspace, nor is it provided by this far.") \
% (fpdfile, guid, version)
installError = True
# Check the fars
thisFarGuid = XmlElement(manifest, "/FrameworkArchiveManifest/FarHeader/GuidValue")
thisFarGuid = string.lower(XmlElement(manifest, "/FrameworkArchiveManifest/FarHeader/GuidValue"))
if fdb.HasFar(thisFarGuid):
print "Error: There is a far with this guid already installed."
installError = True
if not flags.reinstall:
print "Error: There is a far with this guid already installed."
installError = True
# We can not do the install
if installError:
if force:
print "Ignoring previous errors as you requested."
if flags.force:
print "Warning: Ignoring previous errors as you requested."
else:
return False
@ -206,36 +245,41 @@ def InstallFar(farfile, workspaceLocation=""):
for farPackage in XmlList(manifest, "/FrameworkArchiveManifest/FarPackageList/FarPackage"):
filename = XmlElement(farPackage, "FarPackage/FarFilename")
fdb.AddPackage(filename)
if not flags.reinstall:
fdb.AddPackage(filename)
ExtractFile(far, filename, workspaceLocation)
zipContents.remove(filename)
DefaultPath = XmlElement(farPackage, "FarPackage/DefaultPath")
for content in XmlList(farPackage, "FarPackage/Contents/FarFilename"):
filename = XmlElementData(content)
ExtractFile(far, filename, workspaceLocation)
zipContents.remove(filename)
ExtractFile(far, filename, DefaultPath, workspaceLocation, md5sum=content.getAttribute("Md5Sum"))
zipContents.remove(os.path.join(DefaultPath, filename))
# Install the platforms
for farPlatform in XmlList(manifest, "/FrameworkArchiveManifest/FarPlatformList/FarPlatform"):
filename = XmlElement(farPlatform, "FarPlatform/FarFilename")
fdb.AddPlatform(filename)
ExtractFile(far, filename, workspaceLocation)
if not flags.reinstall:
fdb.AddPlatform(filename)
ExtractFile(far, filename, "", workspaceLocation)
zipContents.remove(filename)
# Install the Contents
for content in XmlList(manifest, "/FrameworkArchiveManifest/Contents/FarFilename"):
filename = XmlElementData(content)
ExtractFile(far, filename, workspaceLocation)
ExtractFile(far, filename, "", workspaceLocation)
zipContents.remove(filename)
# What if there are more files in the far?
if not zipContents == []:
print "There are still files in the far:", zipContents
print "Warning: There are files in the far that were not expected: ", zipContents
fdb.AddFar(farfile, thisFarGuid)
if not flags.reinstall:
fdb.AddFar(farfile, thisFarGuid)
# If everything has gone well, we can put the manifest file in a safe place...
farDir = inWorkspace("Tools/Conf/InstalledFars/")
@ -245,7 +289,8 @@ def InstallFar(farfile, workspaceLocation=""):
f.close()
# Write out the new database
fdb.Write()
if not flags.reinstall:
fdb.Write()
far.close()
@ -253,22 +298,26 @@ def InstallFar(farfile, workspaceLocation=""):
# into another script.
if __name__ == '__main__':
flags = Flags()
# Process the command line args.
optlist, args = getopt.getopt(sys.argv[1:], '?hvf', ['help', 'verbose', 'force'])
optlist, args = getopt.getopt(sys.argv[1:], '?hvf', ['help', 'verbose', 'force', 'reinstall'])
# First pass through the options list.
for o, a in optlist:
if o in ["-h", "--help"]:
if o in ["-h", "-?", "--help"]:
print """
Install a far (Framework Archive) into the current workspace.
%s: Install a far (Framework Archive) into the current workspace.
""" % os.path.basename(sys.argv[0])
sys.exit()
optlist.remove((o,a))
if o in ["-v", "--verbose"]:
verbose = True
flags.verbose = True
if o in ["-f", "--force"]:
force = True
flags.force = True
if o in ["--reinstall"]:
flags.reinstall = True
for f in args:
InstallFar(f)