Ticket #21181: 21181.2.patch

wp-includes/ms-functions.php

@@ -1475 +1475 @@
         if ( empty( $spaceAllowed ) || !is_numeric( $spaceAllowed ) )
                 $spaceAllowed = 10;     // Default space allowed is 10 MB
 
-        $size = get_dirsize( BLOGUPLOADDIR ) / 1024 / 1024;
+        $size = apply_filters( 'blog_space_used', false );
+        if ( false === $size )
+                $size = get_dirsize( BLOGUPLOADDIR );
 
+        $size = $size / 1024 / 1024;
+
         if ( ($spaceAllowed-$size) < 0 ) {
                 if ( $echo )
                         _e( 'Sorry, you have used your space allocation. Please delete some files to upload more files.' ); // No space left
@@ -1556 +1560 @@
                 return 0;
 
         $spaceAllowed = 1024 * 1024 * get_space_allowed();
-        $dirsize = get_dirsize( BLOGUPLOADDIR );
+        $dirsize = apply_filters( 'blog_space_used', false );
+        if ( false === $dirsize )
+                $dirsize = get_dirsize( BLOGUPLOADDIR );
+
         if ( $size > $spaceAllowed - $dirsize )
                 return $spaceAllowed - $dirsize; // remaining space
         else

wp-admin/includes/ms.php

@@ -26 +26 @@
                 return $file;
 
         $space_allowed = 1048576 * get_space_allowed();
-        $space_used = get_dirsize( BLOGUPLOADDIR );
+        $space_used = apply_filters( 'blog_space_used', false );
+        if ( false === $space_used )
+                $space_used = get_dirsize( BLOGUPLOADDIR );
         $space_left = $space_allowed - $space_used;
         $file_size = filesize( $file['tmp_name'] );
         if ( $space_left < $file_size )
@@ -310 +312 @@
         if ( get_site_option( 'upload_space_check_disabled' ) )
                 return $space_allowed;
 
+        $space_available = apply_filters( 'pre_get_upload_space_available', false );
+        if ( false !== $space_available )
+                return $space_available;
+
         $dir_name = trailingslashit( BLOGUPLOADDIR );
         if ( !( is_dir( $dir_name) && is_readable( $dir_name ) ) )
                 return $space_allowed;
@@ -353 +359 @@
 
 function display_space_usage() {
         $space = get_space_allowed();
-        $used = get_dirsize( BLOGUPLOADDIR ) / 1024 / 1024;
+        $used = apply_filters( 'blog_space_used', false );
+        if ( false === $used )
+                $used = get_dirsize( BLOGUPLOADDIR ) / 1024 / 1024;
 
         $percentused = ( $used / $space ) * 100;