FSFileBackend.php

Go to the documentation of this file.

<?php
class FSFileBackend extends FileBackendStore {
        protected $basePath; // string; directory holding the container directories
        protected $containerPaths = array(); // for custom container paths
        protected $fileMode; // integer; file permission mode

        protected $hadWarningErrors = array();

        public function __construct( array $config ) {
                parent::__construct( $config );

                // Remove any possible trailing slash from directories
                if ( isset( $config['basePath'] ) ) {
                        $this->basePath = rtrim( $config['basePath'], '/' ); // remove trailing slash
                } else {
                        $this->basePath = null; // none; containers must have explicit paths
                }

                if ( isset( $config['containerPaths'] ) ) {
                        $this->containerPaths = (array)$config['containerPaths'];
                        foreach ( $this->containerPaths as &$path ) {
                                $path = rtrim( $path, '/' );  // remove trailing slash
                        }
                }

                $this->fileMode = isset( $config['fileMode'] )
                        ? $config['fileMode']
                        : 0644;
        }

        protected function resolveContainerPath( $container, $relStoragePath ) {
                // Check that container has a root directory
                if ( isset( $this->containerPaths[$container] ) || isset( $this->basePath ) ) {
                        // Check for sane relative paths (assume the base paths are OK)
                        if ( $this->isLegalRelPath( $relStoragePath ) ) {
                                return $relStoragePath;
                        }
                }
                return null;
        }

        protected function isLegalRelPath( $path ) {
                // Check for file names longer than 255 chars
                if ( preg_match( '![^/]{256}!', $path ) ) { // ext3/NTFS
                        return false;
                }
                if ( wfIsWindows() ) { // NTFS
                        return !preg_match( '![:*?"<>|]!', $path );
                } else {
                        return true;
                }
        }

        protected function containerFSRoot( $shortCont, $fullCont ) {
                if ( isset( $this->containerPaths[$shortCont] ) ) {
                        return $this->containerPaths[$shortCont]; 
                } elseif ( isset( $this->basePath ) ) {
                        return "{$this->basePath}/{$fullCont}";
                }
                return null; // no container base path defined
        }

        protected function resolveToFSPath( $storagePath ) {
                list( $fullCont, $relPath ) = $this->resolveStoragePathReal( $storagePath );
                if ( $relPath === null ) {
                        return null; // invalid
                }
                list( $b, $shortCont, $r ) = FileBackend::splitStoragePath( $storagePath );
                $fsPath = $this->containerFSRoot( $shortCont, $fullCont ); // must be valid
                if ( $relPath != '' ) {
                        $fsPath .= "/{$relPath}";
                }
                return $fsPath;
        }

        public function isPathUsableInternal( $storagePath ) {
                $fsPath = $this->resolveToFSPath( $storagePath );
                if ( $fsPath === null ) {
                        return false; // invalid
                }
                $parentDir = dirname( $fsPath );

                if ( file_exists( $fsPath ) ) {
                        $ok = is_file( $fsPath ) && is_writable( $fsPath );
                } else {
                        $ok = is_dir( $parentDir ) && is_writable( $parentDir );
                }

                return $ok;
        }

        protected function doStoreInternal( array $params ) {
                $status = Status::newGood();

                $dest = $this->resolveToFSPath( $params['dst'] );
                if ( $dest === null ) {
                        $status->fatal( 'backend-fail-invalidpath', $params['dst'] );
                        return $status;
                }

                if ( file_exists( $dest ) ) {
                        if ( !empty( $params['overwrite'] ) ) {
                                $ok = unlink( $dest );
                                if ( !$ok ) {
                                        $status->fatal( 'backend-fail-delete', $params['dst'] );
                                        return $status;
                                }
                        } else {
                                $status->fatal( 'backend-fail-alreadyexists', $params['dst'] );
                                return $status;
                        }
                }

                $ok = copy( $params['src'], $dest );
                if ( !$ok ) {
                        $status->fatal( 'backend-fail-store', $params['src'], $params['dst'] );
                        return $status;
                }

                $this->chmod( $dest );

                return $status;
        }

        protected function doCopyInternal( array $params ) {
                $status = Status::newGood();

                $source = $this->resolveToFSPath( $params['src'] );
                if ( $source === null ) {
                        $status->fatal( 'backend-fail-invalidpath', $params['src'] );
                        return $status;
                }

                $dest = $this->resolveToFSPath( $params['dst'] );
                if ( $dest === null ) {
                        $status->fatal( 'backend-fail-invalidpath', $params['dst'] );
                        return $status;
                }

                if ( file_exists( $dest ) ) {
                        if ( !empty( $params['overwrite'] ) ) {
                                $ok = unlink( $dest );
                                if ( !$ok ) {
                                        $status->fatal( 'backend-fail-delete', $params['dst'] );
                                        return $status;
                                }
                        } else {
                                $status->fatal( 'backend-fail-alreadyexists', $params['dst'] );
                                return $status;
                        }
                }

                $ok = copy( $source, $dest );
                if ( !$ok ) {
                        $status->fatal( 'backend-fail-copy', $params['src'], $params['dst'] );
                        return $status;
                }

                $this->chmod( $dest );

                return $status;
        }

        protected function doMoveInternal( array $params ) {
                $status = Status::newGood();

                $source = $this->resolveToFSPath( $params['src'] );
                if ( $source === null ) {
                        $status->fatal( 'backend-fail-invalidpath', $params['src'] );
                        return $status;
                }

                $dest = $this->resolveToFSPath( $params['dst'] );
                if ( $dest === null ) {
                        $status->fatal( 'backend-fail-invalidpath', $params['dst'] );
                        return $status;
                }

                if ( file_exists( $dest ) ) {
                        if ( !empty( $params['overwrite'] ) ) {
                                // Windows does not support moving over existing files
                                if ( wfIsWindows() ) {
                                        $ok = unlink( $dest );
                                        if ( !$ok ) {
                                                $status->fatal( 'backend-fail-delete', $params['dst'] );
                                                return $status;
                                        }
                                }
                        } else {
                                $status->fatal( 'backend-fail-alreadyexists', $params['dst'] );
                                return $status;
                        }
                }

                $ok = rename( $source, $dest );
                clearstatcache(); // file no longer at source
                if ( !$ok ) {
                        $status->fatal( 'backend-fail-move', $params['src'], $params['dst'] );
                        return $status;
                }

                return $status;
        }

        protected function doDeleteInternal( array $params ) {
                $status = Status::newGood();

                $source = $this->resolveToFSPath( $params['src'] );
                if ( $source === null ) {
                        $status->fatal( 'backend-fail-invalidpath', $params['src'] );
                        return $status;
                }

                if ( !is_file( $source ) ) {
                        if ( empty( $params['ignoreMissingSource'] ) ) {
                                $status->fatal( 'backend-fail-delete', $params['src'] );
                        }
                        return $status; // do nothing; either OK or bad status
                }

                $ok = unlink( $source );
                if ( !$ok ) {
                        $status->fatal( 'backend-fail-delete', $params['src'] );
                        return $status;
                }

                return $status;
        }

        protected function doCreateInternal( array $params ) {
                $status = Status::newGood();

                $dest = $this->resolveToFSPath( $params['dst'] );
                if ( $dest === null ) {
                        $status->fatal( 'backend-fail-invalidpath', $params['dst'] );
                        return $status;
                }

                if ( file_exists( $dest ) ) {
                        if ( !empty( $params['overwrite'] ) ) {
                                $ok = unlink( $dest );
                                if ( !$ok ) {
                                        $status->fatal( 'backend-fail-delete', $params['dst'] );
                                        return $status;
                                }
                        } else {
                                $status->fatal( 'backend-fail-alreadyexists', $params['dst'] );
                                return $status;
                        }
                }

                $bytes = file_put_contents( $dest, $params['content'] );
                if ( $bytes === false ) {
                        $status->fatal( 'backend-fail-create', $params['dst'] );
                        return $status;
                }

                $this->chmod( $dest );

                return $status;
        }

        protected function doPrepareInternal( $fullCont, $dirRel, array $params ) {
                $status = Status::newGood();
                list( $b, $shortCont, $r ) = FileBackend::splitStoragePath( $params['dir'] );
                $contRoot = $this->containerFSRoot( $shortCont, $fullCont ); // must be valid
                $dir = ( $dirRel != '' ) ? "{$contRoot}/{$dirRel}" : $contRoot;
                if ( !wfMkdirParents( $dir ) ) { // make directory and its parents
                        $status->fatal( 'directorycreateerror', $params['dir'] );
                } elseif ( !is_writable( $dir ) ) {
                        $status->fatal( 'directoryreadonlyerror', $params['dir'] );
                } elseif ( !is_readable( $dir ) ) {
                        $status->fatal( 'directorynotreadableerror', $params['dir'] );
                }
                return $status;
        }

        protected function doSecureInternal( $fullCont, $dirRel, array $params ) {
                $status = Status::newGood();
                list( $b, $shortCont, $r ) = FileBackend::splitStoragePath( $params['dir'] );
                $contRoot = $this->containerFSRoot( $shortCont, $fullCont ); // must be valid
                $dir = ( $dirRel != '' ) ? "{$contRoot}/{$dirRel}" : $contRoot;
                // Seed new directories with a blank index.html, to prevent crawling...
                if ( !empty( $params['noListing'] ) && !file_exists( "{$dir}/index.html" ) ) {
                        $bytes = file_put_contents( "{$dir}/index.html", '' );
                        if ( !$bytes ) {
                                $status->fatal( 'backend-fail-create', $params['dir'] . '/index.html' );
                                return $status;
                        }
                }
                // Add a .htaccess file to the root of the container...
                if ( !empty( $params['noAccess'] ) ) {
                        if ( !file_exists( "{$contRoot}/.htaccess" ) ) {
                                $bytes = file_put_contents( "{$contRoot}/.htaccess", "Deny from all\n" );
                                if ( !$bytes ) {
                                        $storeDir = "mwstore://{$this->name}/{$shortCont}";
                                        $status->fatal( 'backend-fail-create', "{$storeDir}/.htaccess" );
                                        return $status;
                                }
                        }
                }
                return $status;
        }

        protected function doCleanInternal( $fullCont, $dirRel, array $params ) {
                $status = Status::newGood();
                list( $b, $shortCont, $r ) = FileBackend::splitStoragePath( $params['dir'] );
                $contRoot = $this->containerFSRoot( $shortCont, $fullCont ); // must be valid
                $dir = ( $dirRel != '' ) ? "{$contRoot}/{$dirRel}" : $contRoot;
                wfSuppressWarnings();
                if ( is_dir( $dir ) ) {
                        rmdir( $dir ); // remove directory if empty
                }
                wfRestoreWarnings();
                return $status;
        }

        protected function doGetFileStat( array $params ) {
                $source = $this->resolveToFSPath( $params['src'] );
                if ( $source === null ) {
                        return false; // invalid storage path
                }

                $this->trapWarnings(); // don't trust 'false' if there were errors
                $stat = is_file( $source ) ? stat( $source ) : false; // regular files only
                $hadError = $this->untrapWarnings();

                if ( $stat ) {
                        return array(
                                'mtime' => wfTimestamp( TS_MW, $stat['mtime'] ),
                                'size'  => $stat['size']
                        );
                } elseif ( !$hadError ) {
                        return false; // file does not exist
                } else {
                        return null; // failure
                }
        }

        protected function doClearCache( array $paths = null ) {
                clearstatcache(); // clear the PHP file stat cache
        }

        public function getFileListInternal( $fullCont, $dirRel, array $params ) {
                list( $b, $shortCont, $r ) = FileBackend::splitStoragePath( $params['dir'] );
                $contRoot = $this->containerFSRoot( $shortCont, $fullCont ); // must be valid
                $dir = ( $dirRel != '' ) ? "{$contRoot}/{$dirRel}" : $contRoot;
                $exists = is_dir( $dir );
                if ( !$exists ) {
                        wfDebug( __METHOD__ . "() given directory does not exist: '$dir'\n" );
                        return array(); // nothing under this dir
                }
                $readable = is_readable( $dir );
                if ( !$readable ) {
                        wfDebug( __METHOD__ . "() given directory is unreadable: '$dir'\n" );
                        return null; // bad permissions?
                }
                return new FSFileBackendFileList( $dir );
        }

        public function getLocalReference( array $params ) {
                $source = $this->resolveToFSPath( $params['src'] );
                if ( $source === null ) {
                        return null;
                }
                return new FSFile( $source );
        }

        public function getLocalCopy( array $params ) {
                $source = $this->resolveToFSPath( $params['src'] );
                if ( $source === null ) {
                        return null;
                }

                // Create a new temporary file with the same extension...
                $ext = FileBackend::extensionFromPath( $params['src'] );
                $tmpFile = TempFSFile::factory( wfBaseName( $source ) . '_', $ext );
                if ( !$tmpFile ) {
                        return null;
                }
                $tmpPath = $tmpFile->getPath();

                // Copy the source file over the temp file
                $ok = copy( $source, $tmpPath );
                if ( !$ok ) {
                        return null;
                }

                $this->chmod( $tmpPath );

                return $tmpFile;
        }

        protected function chmod( $path ) {
                wfSuppressWarnings();
                $ok = chmod( $path, $this->fileMode );
                wfRestoreWarnings();

                return $ok;
        }

        protected function trapWarnings() {
                $this->hadWarningErrors[] = false; // push to stack
                set_error_handler( array( $this, 'handleWarning' ), E_WARNING );
                return false; // invoke normal PHP error handler
        }

        protected function untrapWarnings() {
                restore_error_handler(); // restore previous handler
                return array_pop( $this->hadWarningErrors ); // pop from stack
        }

        private function handleWarning() {
                $this->hadWarningErrors[count( $this->hadWarningErrors ) - 1] = true;
                return true; // suppress from PHP handler
        }
}

class FSFileBackendFileList implements Iterator {
        protected $iter;
        protected $suffixStart; // integer
        protected $pos = 0; // integer

        public function __construct( $dir ) {
                $dir = realpath( $dir ); // normalize
                $this->suffixStart = strlen( $dir ) + 1; // size of "path/to/dir/"
                try {
                        # Get an iterator that will return leaf nodes (non-directories)
                        if ( MWInit::classExists( 'FilesystemIterator' ) ) { // PHP >= 5.3
                                # RecursiveDirectoryIterator extends FilesystemIterator.
                                # FilesystemIterator::SKIP_DOTS default is inconsistent in PHP 5.3.x.
                                $flags = FilesystemIterator::CURRENT_AS_FILEINFO | FilesystemIterator::SKIP_DOTS;
                                $this->iter = new RecursiveIteratorIterator( 
                                        new RecursiveDirectoryIterator( $dir, $flags ) );
                        } else { // PHP < 5.3
                                # RecursiveDirectoryIterator extends DirectoryIterator
                                $this->iter = new RecursiveIteratorIterator( 
                                        new RecursiveDirectoryIterator( $dir ) );
                        }
                } catch ( UnexpectedValueException $e ) {
                        $this->iter = null; // bad permissions? deleted?
                }
        }

        public function current() {
                // Return only the relative path and normalize slashes to FileBackend-style
                // Make sure to use the realpath since the suffix is based upon that
                return str_replace( '\\', '/',
                        substr( realpath( $this->iter->current() ), $this->suffixStart ) );
        }

        public function key() {
                return $this->pos;
        }

        public function next() {
                try {
                        $this->iter->next();
                } catch ( UnexpectedValueException $e ) {
                        $this->iter = null;
                }
                ++$this->pos;
        }

        public function rewind() {
                $this->pos = 0;
                try {
                        $this->iter->rewind();
                } catch ( UnexpectedValueException $e ) {
                        $this->iter = null;
                }
        }

        public function valid() {
                return $this->iter && $this->iter->valid();
        }
}