Patch failure handling updates, added a 'patch' concrete class, so we can apply patches to quilt-native itself.

git-svn-id: https://svn.o-hand.com/repos/poky/trunk@670 311d38ba-8fff-0310-9ca6-ca027cbcb966

1 files changed, 148 insertions, 89 deletions

diff --git a/meta/classes/patch.bbclass b/meta/classes/patch.bbclass
index 6616cb11a9..ea0182484e 100644
--- a/meta/classes/patch.bbclass
+++ b/meta/classes/patch.bbclass
@@ -1,9 +1,11 @@
-def patch_init():
+# Copyright (C) 2006  OpenedHand LTD
+
+def patch_init(d):
 	import os, sys
-	
+
 	def md5sum(fname):
 		import md5, sys
-	
+
 		f = file(fname, 'rb')
 		m = md5.new()
 		while True:
@@ -13,12 +15,12 @@ def patch_init():
 			m.update(d)
 		f.close()
 		return m.hexdigest()
-	
+
 	class CmdError(Exception):
 		def __init__(self, exitstatus, output):
 			self.status = exitstatus
 			self.output = output
-	
+
 		def __str__(self):
 			return "Command Error: exit status: %d  Output:\n%s" % (self.status, self.output)
 
@@ -27,17 +29,17 @@ def patch_init():
 			self.path = path
 		def __str__(self):
 			return "Error: %s not found." % self.path
-	
+
 	def runcmd(args, dir = None):
 		import commands
-	
+
 		if dir:
 			olddir = os.path.abspath(os.curdir)
 			if not os.path.exists(dir):
 				raise NotFoundError(dir)
 			os.chdir(dir)
 			# print("cwd: %s -> %s" % (olddir, self.dir))
-	
+
 		try:
 			args = [ commands.mkarg(str(arg)) for arg in args ]
 			cmd = " ".join(args)
@@ -46,92 +48,143 @@ def patch_init():
 			if exitstatus != 0:
 				raise CmdError(exitstatus >> 8, output)
 			return output
-	
+
 		finally:
 			if dir:
 				os.chdir(olddir)
-	
+
 	class PatchError(Exception):
 		def __init__(self, msg):
 			self.msg = msg
-	
+
 		def __str__(self):
 			return "Patch Error: %s" % self.msg
-	
+
 	import bb, bb.data, bb.fetch
-	
+
 	class PatchSet(object):
 		defaults = {
 			"strippath": 1
 		}
-	
+
 		def __init__(self, dir, d):
 			self.dir = dir
 			self.d = d
 			self.patches = []
 			self._current = None
-	
+
 		def current(self):
 			return self._current
-	
+
 		def Clean(self):
 			"""
 			Clean out the patch set.  Generally includes unapplying all
 			patches and wiping out all associated metadata.
 			"""
 			raise NotImplementedError()
-	
+
 		def Import(self, patch, force):
 			if not patch.get("file"):
 				if not patch.get("remote"):
 					raise PatchError("Patch file must be specified in patch import.")
 				else:
 					patch["file"] = bb.fetch.localpath(patch["remote"], self.d)
-	
+
 			for param in PatchSet.defaults:
 				if not patch.get(param):
 					patch[param] = PatchSet.defaults[param]
-	
+
 			if patch.get("remote"):
 				patch["file"] = bb.data.expand(bb.fetch.localpath(patch["remote"], self.d), self.d)
-	
+
 			patch["filemd5"] = md5sum(patch["file"])
-	
+
 		def Push(self, force):
 			raise NotImplementedError()
-	
+
 		def Pop(self, force):
 			raise NotImplementedError()
-	
+
 		def Refresh(self, remote = None, all = None):
 			raise NotImplementedError()
-	
-	
+
+
+	class PatchTree(PatchSet):
+		def __init__(self, dir, d):
+			PatchSet.__init__(self, dir, d)
+
+		def Import(self, patch, force = None):
+			""""""
+			PatchSet.Import(self, patch, force)
+
+			self.patches.insert(self._current or 0, patch)
+
+		def _applypatch(self, patch, force = None, reverse = None):
+			shellcmd = ["patch", "<", patch['file'], "-p", patch['strippath']]
+			if reverse:
+				shellcmd.append('-R')
+			shellcmd.append('--dry-run')
+
+			try:
+				output = runcmd(["sh", "-c", " ".join(shellcmd)], self.dir)
+			except CmdError:
+				if force:
+					shellcmd.pop(len(shellcmd) - 1)
+					output = runcmd(["sh", "-c", " ".join(shellcmd)], self.dir)
+				else:
+					import sys
+					raise sys.exc_value
+
+			return output
+
+		def Push(self, force = None, all = None):
+			if all:
+				for i in self.patches:
+					if self._current is not None:
+						self._current = self._current + 1
+					else:
+						self._current = 0
+					self._applypatch(i, force)
+			else:
+				if self._current is not None:
+					self._current = self._current + 1
+				else:
+					self._current = 0
+				self._applypatch(self.patches[self._current], force)
+
+
+		def Pop(self, force = None, all = None):
+			if all:
+				for i in self.patches:
+					self._applypatch(i, force, True)
+			else:
+				self._applypatch(self.patches[self._current], force, True)
+
+		def Clean(self):
+			""""""
+
 	class QuiltTree(PatchSet):
 		def _runcmd(self, args):
 			runcmd(["quilt"] + args, self.dir)
-	
+
 		def _quiltpatchpath(self, file):
 			return os.path.join(self.dir, "patches", os.path.basename(file))
-	
-	
+
+
 		def __init__(self, dir, d):
 			PatchSet.__init__(self, dir, d)
 			self.initialized = False
-	
+
 		def Clean(self):
 			try:
 				self._runcmd(["pop", "-a", "-f"])
 			except CmdError:
-				if sys.exc_value.output.strip() == "No patch removed":
-					pass
-				else:
-					raise PatchError("Unable to clean patches from tree:\n"+str(sys.exc_value))
+				pass
 			except NotFoundError:
 				pass
-			runcmd(["rm", "-rf", os.path.join(self.dir, "patches"), os.path.join(self.dir, ".pc")])
+			# runcmd(["rm", "-rf", os.path.join(self.dir, "patches"), os.path.join(self.dir, ".pc")])
 			self.initialized = True
-	
+
 		def InitFromDir(self):
 			# read series -> self.patches
 			seriespath = os.path.join(self.dir, 'patches', 'series')
@@ -148,7 +201,7 @@ def patch_init():
 						patch["strippath"] = parts[1][2:]
 					self.patches.append(patch)
 				series.close()
-	
+
 				# determine which patches are applied -> self._current
 				try:
 					output = runcmd(["quilt", "applied"], self.dir)
@@ -162,45 +215,45 @@ def patch_init():
 					if os.path.basename(patch["quiltfile"]) == output[-1]:
 						self._current = self.patches.index(patch)
 			self.initialized = True
-	
+
 		def Import(self, patch, force = None):
 			if not self.initialized:
 				self.InitFromDir()
 			PatchSet.Import(self, patch, force)
-	
+
 			args = ["import", "-p", patch["strippath"]]
 			if force:
 				args.append("-f")
 			args.append(patch["file"])
-	
+
 			self._runcmd(args)
-	
+
 			patch["quiltfile"] = self._quiltpatchpath(patch["file"])
 			patch["quiltfilemd5"] = md5sum(patch["quiltfile"])
-	
+
 			# TODO: determine if the file being imported:
 			#	   1) is already imported, and is the same
 			#	   2) is already imported, but differs
-	
+
 			self.patches.insert(self._current or 0, patch)
-	
-	
+
+
 		def Push(self, force = None, all = None):
 			# quilt push [-f]
-	
+
 			args = ["push"]
 			if force:
 				args.append("-f")
 			if all:
 				args.append("-a")
-	
+
 			self._runcmd(args)
-	
+
 			if self._current is not None:
 				self._current = self._current + 1
 			else:
 				self._current = 0
-	
+
 		def Pop(self, force = None, all = None):
 			# quilt pop [-f]
 			args = ["pop"]
@@ -208,15 +261,15 @@ def patch_init():
 				args.append("-f")
 			if all:
 				args.append("-a")
-	
+
 			self._runcmd(args)
-	
+
 			if self._current == 0:
 				self._current = None
-	
+
 			if self._current is not None:
 				self._current = self._current - 1
-	
+
 		def Refresh(self, **kwargs):
 			if kwargs.get("remote"):
 				patch = self.patches[kwargs["patch"]]
@@ -227,7 +280,7 @@ def patch_init():
 					import shutil
 					if not patch.get("file") and patch.get("remote"):
 						patch["file"] = bb.fetch.localpath(patch["remote"], self.d)
-	
+
 					shutil.copyfile(patch["quiltfile"], patch["file"])
 				else:
 					raise PatchError("Unable to do a remote refresh of %s, unsupported remote url scheme %s." % (os.path.basename(patch["quiltfile"]), type))
@@ -239,31 +292,31 @@ def patch_init():
 				elif kwargs.get("patch"):
 					args.append(os.path.basename(self.patches[kwargs["patch"]]["quiltfile"]))
 				self._runcmd(args)
-	
+
 	class Resolver(object):
 		def __init__(self, patchset):
 			raise NotImplementedError()
-	
+
 		def Resolve(self):
 			raise NotImplementedError()
-	
+
 		def Revert(self):
 			raise NotImplementedError()
-	
+
 		def Finalize(self):
 			raise NotImplementedError()
-	
+
 	# Patch resolver which relies on the user doing all the work involved in the
 	# resolution, with the exception of refreshing the remote copy of the patch
 	# files (the urls).
 	class UserResolver(Resolver):
 		def __init__(self, patchset):
 			self.patchset = patchset
-	
+
 		# Force a push in the patchset, then drop to a shell for the user to
 		# resolve any rejected hunks
 		def Resolve(self):
-	
+
 			olddir = os.path.abspath(os.curdir)
 			os.chdir(self.patchset.dir)
 			try:
@@ -274,21 +327,21 @@ def patch_init():
 					return
 				print(sys.exc_value)
 				print('NOTE: dropping user into a shell, so that patch rejects can be fixed manually.')
-	
+
 				os.system('/bin/sh')
-	
+
 				# Construct a new PatchSet after the user's changes, compare the
 				# sets, checking patches for modifications, and doing a remote
 				# refresh on each.
 				oldpatchset = self.patchset
 				self.patchset = oldpatchset.__class__(self.patchset.dir, self.patchset.d)
-	
+
 				for patch in self.patchset.patches:
 					oldpatch = None
 					for opatch in oldpatchset.patches:
 						if opatch["quiltfile"] == patch["quiltfile"]:
 							oldpatch = opatch
-	
+
 					if oldpatch:
 						patch["remote"] = oldpatch["remote"]
 						if patch["quiltfile"] == oldpatch["quiltfile"]:
@@ -303,17 +356,18 @@ def patch_init():
 				os.chdir(olddir)
 				raise
 			os.chdir(olddir)
-	
+
 		# Throw away the changes to the patches in the patchset made by resolve()
 		def Revert(self):
 			raise NotImplementedError()
-	
+
 		# Apply the changes to the patches in the patchset made by resolve()
 		def Finalize(self):
 			raise NotImplementedError()
 
 	g = globals()
 	g["PatchSet"] = PatchSet
+	g["PatchTree"] = PatchTree
 	g["QuiltTree"] = QuiltTree
 	g["Resolver"] = Resolver
 	g["UserResolver"] = UserResolver
@@ -322,45 +376,48 @@ def patch_init():
 
 addtask patch after do_unpack
 do_patch[dirs] = "${WORKDIR}"
-python base_do_patch() {
+python patch_do_patch() {
 	import re
 	import bb.fetch
 
-	patch_init()
-	
+	patch_init(d)
+
 	src_uri = (bb.data.getVar('SRC_URI', d, 1) or '').split()
 	if not src_uri:
 		return
-	
+
 	patchsetmap = {
+		"patch": PatchTree,
 		"quilt": QuiltTree,
 	}
 
 	cls = patchsetmap[bb.data.getVar('PATCHTOOL', d, 1) or 'quilt']
-	
+
 	resolvermap = {
 		"user": UserResolver,
 	}
-	
+
 	rcls = resolvermap[bb.data.getVar('PATCHRESOLVE', d, 1) or 'user']
-	
+
 	s = bb.data.getVar('S', d, 1)
-	
+
+	path = os.getenv('PATH')
+	os.putenv('PATH', bb.data.getVar('PATH', d, 1))
 	patchset = cls(s, d)
 	patchset.Clean()
 
 	resolver = rcls(patchset)
-	
+
 	workdir = bb.data.getVar('WORKDIR', d, 1)
 	for url in src_uri:
 		(type, host, path, user, pswd, parm) = bb.decodeurl(url)
 		if not "patch" in parm:
 			continue
-	
+
 		bb.fetch.init([url],d)
 		url = bb.encodeurl((type, host, path, user, pswd, []))
 		local = os.path.join('/', bb.fetch.localpath(url, d))
-	
+
 		# did it need to be unpacked?
 		dots = os.path.basename(local).split(".")
 		if dots[-1] in ['gz', 'bz2', 'Z']:
@@ -368,45 +425,45 @@ python base_do_patch() {
 		else:
 			unpacked = local
 		unpacked = bb.data.expand(unpacked, d)
-	
+
 		if "pnum" in parm:
 			pnum = parm["pnum"]
 		else:
 			pnum = "1"
-	
+
 		if "pname" in parm:
 			pname = parm["pname"]
 		else:
 			pname = os.path.basename(unpacked)
-	
+
 		if "mindate" in parm:
 			mindate = parm["mindate"]
 		else:
 			mindate = 0
-	
+
 		if "maxdate" in parm:
 			maxdate = parm["maxdate"]
 		else:
 			maxdate = "20711226"
-	
+
 		pn = bb.data.getVar('PN', d, 1)
 		srcdate = bb.data.getVar('SRCDATE_%s' % pn, d, 1)
-	
+
 		if not srcdate:
 			srcdate = bb.data.getVar('SRCDATE', d, 1)
-	
-		if srcdate == "now": 
+
+		if srcdate == "now":
 			srcdate = bb.data.getVar('DATE', d, 1)
-	
+
 		if (maxdate < srcdate) or (mindate > srcdate):
 			if (maxdate < srcdate):
 				bb.note("Patch '%s' is outdated" % pname)
-	
+
 			if (mindate > srcdate):
 				bb.note("Patch '%s' is predated" % pname)
-	
+
 			continue
-	
+
 		bb.note("Applying patch '%s'" % pname)
 		try:
 			patchset.Import({"file":unpacked, "remote":url, "strippath": pnum}, True)
@@ -415,3 +472,5 @@ python base_do_patch() {
 			raise bb.build.FuncFailed(str(sys.exc_value))
 		resolver.Resolve()
 }
+
+EXPORT_FUNCTIONS do_patch