From b535210626649582a771d67b303d063f516f58cf Mon Sep 17 00:00:00 2001
From: "Eric Johansson (neku)" <neku@pomf.se>
Date: Wed, 7 Oct 2015 22:56:03 +0200
Subject: [PATCH] Update core.php

---
 includes/core.php | 18 ++++++++++++++----
 1 file changed, 14 insertions(+), 4 deletions(-)

diff --git a/includes/core.php b/includes/core.php
index 29cfdb0..7a1e4d6 100644
--- a/includes/core.php
+++ b/includes/core.php
@@ -1,5 +1,5 @@
 <?php
-function save_file ($file, $name, $arg){
+function save_file ($file, $name, $arg, $type){
     //Where to save
     $path='/home/neku/www/files/';
     $block = array('exe', 'scr', 'rar', 'zip', 'com', 'vbs', 'bat', 'cmd', 'html', 'htm', 'msi');
@@ -9,8 +9,13 @@ function save_file ($file, $name, $arg){
             $ext = pathinfo($file.$name, PATHINFO_EXTENSION);
             $ext = strtolower($ext);
             if(in_array($ext, $block)){
+                if($type==='normal'){
                 include_once('error_meow.php');
-                exit(0);}
+                exit(0);
+                }else{
+                    exit('File type not allowed.');
+                }
+                }
             $file_name = gen_name('random', $ext);
             while(file_exists($path.$file_name)){
                 $file_name = gen_name('random', $ext);
@@ -23,8 +28,13 @@ function save_file ($file, $name, $arg){
                 $ext = pathinfo($file_name, PATHINFO_EXTENSION);
                 $ext = strtolower($ext);
                 if(in_array($ext, $block)){
-                   include_once('error_meow.php');
-                   exit(0);}
+                if($type==='normal'){
+                include_once('error_meow.php');
+                exit(0);
+                }else{
+                    exit('File type not allowed.');
+                }
+                }
             while(file_exists($path.$file_name)){
                 $file_name = gen_name('custom_original', $name);
             }