From e7c95267245d226f4966f055289b061dd1bc2f91 Mon Sep 17 00:00:00 2001
From: RaNaN <Mast3rRaNaN@hotmail.de>
Date: Thu, 14 Oct 2010 16:02:08 +0200
Subject: more fixes

---
 module/network/Request.py | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

(limited to 'module/network/Request.py')

diff --git a/module/network/Request.py b/module/network/Request.py
index f9941d74c..c76590afc 100755
--- a/module/network/Request.py
+++ b/module/network/Request.py
@@ -126,7 +126,7 @@ class Request:
         return None
     
     def load(self, url, get={}, post={}, ref=True, cookies=True, just_header=False, no_post_encode=False):
-        
+
         self.pycurl.setopt(pycurl.NOPROGRESS, 1)
         
         url = self.__myquote(str(url))
@@ -144,7 +144,7 @@ class Request:
             get = ""
 
         self.pycurl.setopt(pycurl.URL, url)
-        self.pycurl.setopt(pycurl.WRITEFUNCTION, self.rep.write)
+        self.pycurl.setopt(pycurl.WRITEFUNCTION, self.write_rep)
         
         if cookies:
             self.curl_enable_cookies()
@@ -333,6 +333,14 @@ class Request:
     def write_header(self, string):
         self.header += string
 
+    def write_rep(self, buf):
+        if self.rep.tell() > 180000 or self.abort:
+            if self.abort: raise Abort
+            print self.rep.getvalue()
+            raise Exception("Loaded Url exceeded limit")
+
+        self.rep.write(buf)
+
     def get_rep(self):
         value = self.rep.getvalue()
         self.rep.close()
-- 
cgit v1.2.3