From 159a7cf066c57e6d564d62b5021769fbe89da8a1 Mon Sep 17 00:00:00 2001
From: Guillaume Pellerin <yomguy@parisson.com>
Date: Tue, 10 Feb 2015 10:50:18 +0100
Subject: [PATCH] Auto reject script if file is too large

---
 teleforma/exam/models.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/teleforma/exam/models.py b/teleforma/exam/models.py
index 73bf468a..b2781503 100644
--- a/teleforma/exam/models.py
+++ b/teleforma/exam/models.py
@@ -78,6 +78,8 @@ REJECT_REASON = (('unreadable', _('unreadable')),
 cache_path = settings.MEDIA_ROOT + 'cache/'
 script_path = settings.MEDIA_ROOT + 'scripts/'
 
+SCRIPT_MAX_SIZE = 50000000
+
 
 def sha1sum_file(filename):
     '''
@@ -379,6 +381,10 @@ class Script(BaseResource):
             self.auto_reject('wrong format')
             return
 
+        if os.stat(self.file.path).st_size > SCRIPT_MAX_SIZE:
+            self.auto_reject('file too large')
+            return
+
         if not self.status == 0 and self.file:
             if not self.box_uuid:
                 self.uuid_link()
-- 
2.39.5