esafenet.py

esafenet.py.patch

diff --git a/esafenet.py b/esafenet.py
index e91c284..80cf7dc 100644
--- a/esafenet.py
+++ b/esafenet.py
@@ -124,10 +124,28 @@ class Esafenet:
 				if not os.path.isdir(dest_folder + "/" + nr + d):
 					os.mkdir(dest_folder + "/" + nr + d)
 			for f in files:
-				fl = open(root + "/" + f, "rb")
-				dec = Esafenet.decrypt_file(fl.read(), key)
+                                # It's dirty and I know it
+				try:
+				    fl = open(root + "/" + f, "rb")
+				except:
+				    continue
+
+				fbyte = fl.read(3)
+				if len(fbyte) >= 3:
+					fchar = struct.unpack('ccc', fbyte)
+					fl.seek(0)
+
+					if fchar[0] == 'b' and fchar[2] == '#':
+						# Encrypted
+						dec = Esafenet.decrypt_file(fl.read(), key)
+						with open(dest_folder + "/" + nr + f, "wb") as fh:
+						    fh.write(dec)
+
+						continue
+
+				# Not encrypted
 				with open(dest_folder + "/" + nr + f, "wb") as fh:
-					fh.write(dec)
+				    fh.write(fl.read())
 
 
 	"""