quota exceeded

1 files changed, 12 insertions, 0 deletions

diff --git a/www/upload.php b/www/upload.php
index f09b39c..e3b1100 100644
--- a/www/upload.php
+++ b/www/upload.php
@@ -31,6 +31,18 @@ function upload($db){
 	$owner_ar = $owner_db->fetchArray(SQLITE3_NUM);
 	$owner = $owner_ar[0];
 
+	$overall_size_db = $db->query("SELECT size FROM files WHERE owner=" . $owner . " AND size > 0;");
+	$overall_size = 0;
+	$count = 0;
+	while($row = $overall_size_db->fetchArray(SQLITE3_NUM)){
+		$overall_size = $overall_size + $row[$count];
+		$count++;
+	}
+
+	if($overall_size > 2147483648){		// == 2GB
+		error("Quota exceeded");
+	}
+
 	$filename = $_FILES['userfile']['name'];
 	$folder = "FILE";
 	$mime = $_FILES['userfile']['type'];