Added (back in?) "wp_handle_upload_prefilter" filter to filter uploads before they go in blog's directory, fixes #478

git-svn-id: http://svn.automattic.com/wordpress-mu/trunk@1133 7be80a69-a1ef-0310-a953-fb0f7c49ff36

1 files changed, 24 insertions, 0 deletions

diff --git a/wp-admin/includes/mu.php b/wp-admin/includes/mu.php
index cbf5c9d..90d0343 100644
--- a/wp-admin/includes/mu.php
+++ b/wp-admin/includes/mu.php
@@ -1,5 +1,29 @@
 <?php
 
+function check_upload_size( $file ) {
+
+	if( $file[ 'error' ] != '0' ) // there's already an error
+		return $file;
+
+	$space_allowed = 1048576*get_space_allowed();
+	$space_used = get_dirsize( constant( "ABSPATH" ) . constant( "UPLOADS" ) );
+	$space_left = $space_allowed - $space_used;
+	$file_size = filesize( $file[ 'tmp_name' ]);
+	if( $space_left < $file_size )
+		$file[ 'error' ] = sprintf( __( 'Not enough space to upload. %1$sKb needed.' ), number_format( ($file_size - $space_left) /1024 ) );
+	if( $file_size > ( 1024 * get_site_option( 'fileupload_maxk', 1500 ) ) )
+		$file['error'] = sprintf(__('This file is too big. Files must be less than %1$s Kb in size.'), get_site_option( 'fileupload_maxk', 1500 ) );
+	if( upload_is_user_over_quota( false ) ) {
+		$file['error'] = __('You have used your space quota. Please delete files before uploading.');
+	}
+	if( $file[ 'error' ] != '0' )
+		wp_die( $file[ 'error' ] . ' <a href="javascript:history.go(-1)">' . __( 'Back' ) . '</a>' );
+
+	return $file;
+
+}
+add_filter( 'wp_handle_upload_prefilter', 'check_upload_size' );
+
 function wpmu_delete_blog($blog_id, $drop = false) {
 	global $wpdb, $wpmuBaseTablePrefix;