From 0238b5b31debb7f28ff860027656eae9c12de422 Mon Sep 17 00:00:00 2001
From: Sven Slootweg <jamsoftgamedev@gmail.com>
Date: Fri, 26 Oct 2012 18:50:49 +0200
Subject: [PATCH] Add support for password-protected downloads

---
 resolv/resolvers/mediafire.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/resolv/resolvers/mediafire.py b/resolv/resolvers/mediafire.py
index a3452af..228c32d 100644
--- a/resolv/resolvers/mediafire.py
+++ b/resolv/resolvers/mediafire.py
@@ -11,10 +11,28 @@ class MediafireTask(Task):
 			self.state = "failed"
 			raise ResolverError("Could not retrieve the specified URL.")
 		
+		if '<form name="form_password"' in contents:
+			# The file is password-protected
+			self.state = "need_password"
+			return self
+		else:
+			return self._find_link(contents)
+	
+	def verify_password(self, password):
+		contents = self.post_page(self.url, {'downloadp': password})
+		
+		if '<form name="form_password"' in contents:
+			self.state = "password_invalid"
+			return self
+		else:
+			return self._find_link(contents)
+
+	def _find_link(self, contents):
 		matches = re.search('kNO = "([^"]+)";', contents)
 		
 		if matches is None:
 			self.state = "failed"
+			print contents
 			raise ResolverError("No download was found on the given URL; the server for this file may be in maintenance mode, or the given URL may not be valid. It is also possible that you have been blocked - CAPTCHA support is not yet present.")
 		
 		file_url = matches.group(1)