]> scripts.mit.edu Git - autoinstallsdev/mediawiki.git/blobdiff - includes/filerepo/FSRepo.php
MediaWiki 1.17.0
[autoinstallsdev/mediawiki.git] / includes / filerepo / FSRepo.php
index 84ec9a27095a1939376105300e94d2abb1a5cce7..e2251b2bacdae9cd7d4b083a7e83b57a10b31ca3 100644 (file)
@@ -1,12 +1,18 @@
 <?php
+/**
+ * A repository for files accessible via the local filesystem.
+ *
+ * @file
+ * @ingroup FileRepo
+ */
 
 /**
  * A repository for files accessible via the local filesystem. Does not support
  * database access or registration.
+ * @ingroup FileRepo
  */
-
 class FSRepo extends FileRepo {
-       var $directory, $deletedDir, $url, $hashLevels, $deletedHashLevels;
+       var $directory, $deletedDir, $deletedHashLevels, $fileMode;
        var $fileFactory = array( 'UnregisteredLocalFile', 'newFromTitle' );
        var $oldFileFactory = false;
        var $pathDisclosureProtection = 'simple';
@@ -20,9 +26,20 @@ class FSRepo extends FileRepo {
 
                // Optional settings
                $this->hashLevels = isset( $info['hashLevels'] ) ? $info['hashLevels'] : 2;
-               $this->deletedHashLevels = isset( $info['deletedHashLevels'] ) ? 
+               $this->deletedHashLevels = isset( $info['deletedHashLevels'] ) ?
                        $info['deletedHashLevels'] : $this->hashLevels;
                $this->deletedDir = isset( $info['deletedDir'] ) ? $info['deletedDir'] : false;
+               $this->fileMode = isset( $info['fileMode'] ) ? $info['fileMode'] : 0644;
+               if ( isset( $info['thumbDir'] ) ) {
+                       $this->thumbDir =  $info['thumbDir'];
+               } else {
+                       $this->thumbDir = "{$this->directory}/thumb";
+               }
+               if ( isset( $info['thumbUrl'] ) ) {
+                       $this->thumbUrl = $info['thumbUrl'];
+               } else {
+                       $this->thumbUrl = "{$this->url}/thumb";
+               }
        }
 
        /**
@@ -57,13 +74,15 @@ class FSRepo extends FileRepo {
                                return "{$this->directory}/temp";
                        case 'deleted':
                                return $this->deletedDir;
+                       case 'thumb':
+                               return $this->thumbDir;
                        default:
                                return false;
                }
        }
 
        /**
-        * Get the URL corresponding to one of the three basic zones
+        * @see FileRepo::getZoneUrl()
         */
        function getZoneUrl( $zone ) {
                switch ( $zone ) {
@@ -72,15 +91,17 @@ class FSRepo extends FileRepo {
                        case 'temp':
                                return "{$this->url}/temp";
                        case 'deleted':
-                               return false; // no public URL
+                               return parent::getZoneUrl( $zone ); // no public URL
+                       case 'thumb':
+                               return $this->thumbUrl;
                        default:
-                               return false;
+                               return parent::getZoneUrl( $zone );
                }
        }
 
        /**
         * Get a URL referring to this repository, with the private mwrepo protocol.
-        * The suffix, if supplied, is considered to be unencoded, and will be 
+        * The suffix, if supplied, is considered to be unencoded, and will be
         * URL-encoded before being returned.
         */
        function getVirtualUrl( $suffix = false ) {
@@ -117,14 +138,17 @@ class FSRepo extends FileRepo {
        /**
         * Store a batch of files
         *
-        * @param array $triplets (src,zone,dest) triplets as per store()
-        * @param integer $flags Bitwise combination of the following flags:
+        * @param $triplets Array: (src,zone,dest) triplets as per store()
+        * @param $flags Integer: bitwise combination of the following flags:
         *     self::DELETE_SOURCE     Delete the source file after upload
         *     self::OVERWRITE         Overwrite an existing destination file instead of failing
-        *     self::OVERWRITE_SAME    Overwrite the file if the destination exists and has the 
+        *     self::OVERWRITE_SAME    Overwrite the file if the destination exists and has the
         *                             same contents as the source
         */
        function storeBatch( $triplets, $flags = 0 ) {
+               if ( !wfMkdirParents( $this->directory ) ) {
+                       return $this->newFatal( 'upload_directory_missing', $this->directory );
+               }
                if ( !is_writable( $this->directory ) ) {
                        return $this->newFatal( 'upload_directory_read_only', $this->directory );
                }
@@ -146,13 +170,11 @@ class FSRepo extends FileRepo {
                                if ( !wfMkdirParents( $dstDir ) ) {
                                        return $this->newFatal( 'directorycreateerror', $dstDir );
                                }
-                               // In the deleted zone, seed new directories with a blank 
-                               // index.html, to prevent crawling
                                if ( $dstZone == 'deleted' ) {
-                                       file_put_contents( "$dstDir/index.html", '' );
+                                       $this->initDeletedDir( $dstDir );
                                }
                        }
-                       
+
                        if ( self::isVirtualUrl( $srcPath ) ) {
                                $srcPath = $triplets[$i][0] = $this->resolveVirtualUrl( $srcPath );
                        }
@@ -202,7 +224,7 @@ class FSRepo extends FileRepo {
                                }
                        }
                        if ( $good ) {
-                               chmod( $dstPath, 0644 );
+                               $this->chmod( $dstPath );
                                $status->successCount++;
                        } else {
                                $status->failCount++;
@@ -211,11 +233,89 @@ class FSRepo extends FileRepo {
                return $status;
        }
 
+       function append( $srcPath, $toAppendPath, $flags = 0 ) {
+               $status = $this->newGood();
+
+               // Resolve the virtual URL
+               if ( self::isVirtualUrl( $srcPath ) ) {
+                       $srcPath = $this->resolveVirtualUrl( $srcPath );
+               }
+               // Make sure the files are there
+               if ( !is_file( $srcPath ) )
+                       $status->fatal( 'filenotfound', $srcPath );
+
+               if ( !is_file( $toAppendPath ) )
+                       $status->fatal( 'filenotfound', $toAppendPath );
+
+               if ( !$status->isOk() ) return $status;
+
+               // Do the append
+               $chunk = file_get_contents( $toAppendPath );
+               if( $chunk === false ) {
+                       $status->fatal( 'fileappenderrorread', $toAppendPath );
+               }
+
+               if( $status->isOk() ) {
+                       if ( file_put_contents( $srcPath, $chunk, FILE_APPEND ) ) {
+                               $status->value = $srcPath;
+                       } else {
+                               $status->fatal( 'fileappenderror', $toAppendPath,  $srcPath);
+                       }
+               }
+
+               if ( $flags & self::DELETE_SOURCE ) {
+                       unlink( $toAppendPath );
+               }
+
+               return $status;
+       }
+
+       /**
+        * Checks existence of specified array of files.
+        *
+        * @param $files Array: URLs of files to check
+        * @param $flags Integer: bitwise combination of the following flags:
+        *     self::FILES_ONLY     Mark file as existing only if it is a file (not directory)
+        * @return Either array of files and existence flags, or false
+        */
+       function fileExistsBatch( $files, $flags = 0 ) {
+               if ( !file_exists( $this->directory ) || !is_readable( $this->directory ) ) {
+                       return false;
+               }
+               $result = array();
+               foreach ( $files as $key => $file ) {
+                       if ( self::isVirtualUrl( $file ) ) {
+                               $file = $this->resolveVirtualUrl( $file );
+                       }
+                       if( $flags & self::FILES_ONLY ) {
+                               $result[$key] = is_file( $file );
+                       } else {
+                               $result[$key] = file_exists( $file );
+                       }
+               }
+
+               return $result;
+       }
+
+       /**
+        * Take all available measures to prevent web accessibility of new deleted
+        * directories, in case the user has not configured offline storage
+        */
+       protected function initDeletedDir( $dir ) {
+               // Add a .htaccess file to the root of the deleted zone
+               $root = $this->getZonePath( 'deleted' );
+               if ( !file_exists( "$root/.htaccess" ) ) {
+                       file_put_contents( "$root/.htaccess", "Deny from all\n" );
+               }
+               // Seed new directories with a blank index.html, to prevent crawling
+               file_put_contents( "$dir/index.html", '' );
+       }
+
        /**
         * Pick a random name in the temp zone and store a file to it.
-        * @param string $originalName The base name of the file as specified 
+        * @param $originalName String: the base name of the file as specified
         *     by the user. The file extension will be maintained.
-        * @param string $srcPath The current location of the file.
+        * @param $srcPath String: the current location of the file.
         * @return FileRepoStatus object with the URL in the value.
         */
        function storeTemp( $originalName, $srcPath ) {
@@ -231,8 +331,8 @@ class FSRepo extends FileRepo {
 
        /**
         * Remove a temporary file or mark it for garbage collection
-        * @param string $virtualUrl The virtual URL returned by storeTemp
-        * @return boolean True on success, false on failure
+        * @param $virtualUrl String: the virtual URL returned by storeTemp
+        * @return Boolean: true on success, false on failure
         */
        function freeTemp( $virtualUrl ) {
                $temp = "mwrepo://{$this->name}/temp";
@@ -249,12 +349,15 @@ class FSRepo extends FileRepo {
 
        /**
         * Publish a batch of files
-        * @param array $triplets (source,dest,archive) triplets as per publish()
-        * @param integer $flags Bitfield, may be FileRepo::DELETE_SOURCE to indicate
+        * @param $triplets Array: (source,dest,archive) triplets as per publish()
+        * @param $flags Integer: bitfield, may be FileRepo::DELETE_SOURCE to indicate
         *        that the source files should be deleted if possible
         */
        function publishBatch( $triplets, $flags = 0 ) {
                // Perform initial checks
+               if ( !wfMkdirParents( $this->directory ) ) {
+                       return $this->newFatal( 'upload_directory_missing', $this->directory );
+               }
                if ( !is_writable( $this->directory ) ) {
                        return $this->newFatal( 'upload_directory_read_only', $this->directory );
                }
@@ -273,7 +376,7 @@ class FSRepo extends FileRepo {
                        }
                        $dstPath = "{$this->directory}/$dstRel";
                        $archivePath = "{$this->directory}/$archiveRel";
-                       
+
                        $dstDir = dirname( $dstPath );
                        $archiveDir = dirname( $archivePath );
                        // Abort immediately on directory creation errors since they're likely to be repetitive
@@ -292,7 +395,7 @@ class FSRepo extends FileRepo {
                if ( !$status->ok ) {
                        return $status;
                }
-               
+
                foreach ( $triplets as $i => $triplet ) {
                        list( $srcPath, $dstRel, $archiveRel ) = $triplet;
                        $dstPath = "{$this->directory}/$dstRel";
@@ -302,8 +405,8 @@ class FSRepo extends FileRepo {
                        if( is_file( $dstPath ) ) {
                                // Check if the archive file exists
                                // This is a sanity check to avoid data loss. In UNIX, the rename primitive
-                               // unlinks the destination file if it exists. DB-based synchronisation in 
-                               // publishBatch's caller should prevent races. In Windows there's no 
+                               // unlinks the destination file if it exists. DB-based synchronisation in
+                               // publishBatch's caller should prevent races. In Windows there's no
                                // problem because the rename primitive fails if the destination exists.
                                if ( is_file( $archivePath ) ) {
                                        $success = false;
@@ -344,7 +447,7 @@ class FSRepo extends FileRepo {
                                $status->successCount++;
                                wfDebug(__METHOD__.": wrote tempfile $srcPath to $dstPath\n");
                                // Thread-safe override for umask
-                               chmod( $dstPath, 0644 );
+                               $this->chmod( $dstPath );
                        } else {
                                $status->failCount++;
                        }
@@ -354,12 +457,12 @@ class FSRepo extends FileRepo {
 
        /**
         * Move a group of files to the deletion archive.
-        * If no valid deletion archive is configured, this may either delete the 
+        * If no valid deletion archive is configured, this may either delete the
         * file or throw an exception, depending on the preference of the repository.
         *
-        * @param array $sourceDestPairs Array of source/destination pairs. Each element 
+        * @param $sourceDestPairs Array of source/destination pairs. Each element
         *        is a two-element array containing the source file path relative to the
-        *        public root in the first element, and the archive file path relative 
+        *        public root in the first element, and the archive file path relative
         *        to the deleted zone root in the second element.
         * @return FileRepoStatus
         */
@@ -387,8 +490,7 @@ class FSRepo extends FileRepo {
                                        $status->fatal( 'directorycreateerror', $archiveDir );
                                        continue;
                                }
-                               // Seed new directories with a blank index.html, to prevent crawling
-                               file_put_contents( "$archiveDir/index.html", '' );
+                               $this->initDeletedDir( $archiveDir );
                        }
                        // Check if the archive directory is writable
                        // This doesn't appear to work on NTFS
@@ -403,7 +505,7 @@ class FSRepo extends FileRepo {
 
                /**
                 * Move the files
-                * We're now committed to returning an OK result, which will lead to 
+                * We're now committed to returning an OK result, which will lead to
                 * the files being moved in the DB also.
                 */
                foreach ( $sourceDestPairs as $pair ) {
@@ -422,7 +524,7 @@ class FSRepo extends FileRepo {
                                        $status->error( 'filerenameerror', $srcPath, $archivePath );
                                        $good = false;
                                } else {
-                                       chmod( $archivePath, 0644 );
+                                       $this->chmod( $archivePath );
                                }
                        }
                        if ( $good ) {
@@ -433,17 +535,9 @@ class FSRepo extends FileRepo {
                }
                return $status;
        }
-       
-       /**
-        * Get a relative path including trailing slash, e.g. f/fa/
-        * If the repo is not hashed, returns an empty string
-        */
-       function getHashPath( $name ) {
-               return FileRepo::getHashPathForLevel( $name, $this->hashLevels );
-       }
 
        /**
-        * Get a relative path for a deletion archive key, 
+        * Get a relative path for a deletion archive key,
         * e.g. s/z/a/ for sza251lrxrc1jad41h5mgilp8nysje52.jpg
         */
        function getDeletedHashPath( $key ) {
@@ -453,7 +547,7 @@ class FSRepo extends FileRepo {
                }
                return $path;
        }
-       
+
        /**
         * Call a callback function for every file in the repository.
         * Uses the filesystem even in child classes.
@@ -525,6 +619,14 @@ class FSRepo extends FileRepo {
                return strtr( $param, $this->simpleCleanPairs );
        }
 
-}
-
+       /**
+        * Chmod a file, supressing the warnings.
+        * @param $path String: the path to change
+        */
+       protected function chmod( $path ) {
+               wfSuppressWarnings();
+               chmod( $path, $this->fileMode );
+               wfRestoreWarnings();
+       }
 
+}