Bug #212152 - Check available space in DISTDIR before trying to

fetch/copy a file. (trunk r9425, r9502, and 9504:9506)

svn path=/main/branches/2.1.2/; revision=9544

1 files changed, 20 insertions, 2 deletions

diff --git a/pym/portage.py b/pym/portage.py
index 681749402..684838035 100644
--- a/pym/portage.py
+++ b/pym/portage.py
@@ -3184,10 +3184,28 @@ def fetch(myuris, mysettings, listonly=0, fetchonly=0, locks_in_subdir=".locks",
 		"""
 		myfile_path = os.path.join(mysettings["DISTDIR"], myfile)
 		fetched=0
+		has_space = True
 		file_lock = None
 		if listonly:
 			writemsg_stdout("\n", noiselevel=-1)
 		else:
+			# check if there is enough space in DISTDIR to completely store myfile
+			# overestimate the filesize so we aren't bitten by FS overhead
+			if hasattr(os, "statvfs"):
+				vfs_stat = os.statvfs(mysettings["DISTDIR"])
+				try:
+					mysize = os.stat(myfile_path).st_size
+				except OSError, e:
+					if e.errno != errno.ENOENT:
+						raise
+					del e
+					mysize = 0
+				if myfile in mydigests \
+					and (mydigests[myfile]["size"] - mysize + vfs_stat.f_bsize) >= \
+					(vfs_stat.f_bsize * vfs_stat.f_bavail):
+					writemsg("!!! Insufficient space to store %s in %s\n" % (myfile, mysettings["DISTDIR"]), noiselevel=-1)
+					has_space = False
+
 			if use_locks and can_fetch:
 				waiting_msg = None
 				if "parallel-fetch" in features:
@@ -3204,7 +3222,7 @@ def fetch(myuris, mysettings, listonly=0, fetchonly=0, locks_in_subdir=".locks",
 						waiting_msg=waiting_msg)
 		try:
 			if not listonly:
-				if fsmirrors and not os.path.exists(myfile_path):
+				if fsmirrors and not os.path.exists(myfile_path) and has_space:
 					for mydir in fsmirrors:
 						mirror_file = os.path.join(mydir, myfile)
 						try:
@@ -3325,7 +3343,7 @@ def fetch(myuris, mysettings, listonly=0, fetchonly=0, locks_in_subdir=".locks",
 					else:
 						continue
 
-				if fetched != 2:
+				if fetched != 2 and has_space:
 					#we either need to resume or start the download
 					#you can't use "continue" when you're inside a "try" block
 					if fetched==1: