added resume support on the server side

1 files changed, 13 insertions, 5 deletions

diff --git a/tupkg/server/tupkgs b/tupkg/server/tupkgs
index 5eac499..0db7aab 100755
--- a/tupkg/server/tupkgs
+++ b/tupkg/server/tupkgs
@@ -40,9 +40,11 @@ class ClientFile:
 	def __init__(self, filename, actual_size, actual_md5):
 		self.pathname = INCOMINGDIR + filename
 		self.filename = filename
-		self.fd = open(self.pathname, "w+b")
+		self.fd = open(self.pathname, "a+b")
 		self.actual_size = actual_size
 		self.actual_md5 = actual_md5
+		self.getSize()
+		self.orig_size = self.size
 
 	def getSize(self):
 		cur = self.fd.tell()
@@ -61,12 +63,16 @@ class ClientFile:
 		self.md5 = md5sum.hexdigest()
 
 	def finishDownload(self):
-		self.fd.close();
+		self.fd.close()
 		newpathname = CACHEDIR + self.filename
 		os.rename(self.pathname, newpathname)
 		self.pathname = newpathname
 		self.fd = open(self.pathname, "a+b")
 
+	def delete(self):
+		self.fd.close()
+		os.remove(self.pathname)
+
 class ClientSocket(threading.Thread):
 	def __init__(self, sock, db, **other):
 		threading.Thread.__init__(self, *other)
@@ -136,14 +142,15 @@ class ClientSocket(threading.Thread):
 		new_files = files.copy()
 		for i in files:
 			if i[:4] == 'size':
-				new_files[i] = '0'
+				clientfile = self.files[int(i[4:])]
+				new_files[i] = str(clientfile.orig_size)
 			if i[:6] == 'md5sum':
 				del new_files[i]
 		self.sendMsg(new_files)
 
 	def readFiles(self):
 		for i in self.files:
-			count = 0
+			count = i.orig_size
 			while count != i.actual_size:
 				if count + 1024 > i.actual_size:
 					i.fd.write(self.reliableRead(i.actual_size - count))
@@ -152,14 +159,15 @@ class ClientSocket(threading.Thread):
 					i.fd.write(self.reliableRead(1024))
 					count += 1024
 			i.fd.flush()
-			i.finishDownload()
 		reply = {'numpkgs': len(self.files)}
 		for i, v in enumerate(self.files):
 			v.makeMd5()
 			if v.actual_md5 == v.md5:
 				reply['md5sum'+str(i)] = "PASS"
+				v.finishDownload()
 			else:
 				reply['md5sum'+str(i)] = "FAIL"
+				v.delete()
 		self.sendMsg(reply)
 		print self.readMsg()