Big oops - merged to wrong branch.
[lhc/web/wiklou.git] / includes / upload / UploadStash.php
index ce73b8b..12531c2 100644 (file)
 <?php
-/** 
+/**
+ * Temporary storage for uploaded files.
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
+ * http://www.gnu.org/copyleft/gpl.html
+ *
+ * @file
+ * @ingroup Upload
+ */
+
+/**
  * UploadStash is intended to accomplish a few things:
  *   - enable applications to temporarily stash files without publishing them to the wiki.
  *      - Several parts of MediaWiki do this in similar ways: UploadBase, UploadWizard, and FirefoggChunkedExtension
  *        And there are several that reimplement stashing from scratch, in idiosyncratic ways. The idea is to unify them all here.
  *       Mostly all of them are the same except for storing some custom fields, which we subsume into the data array.
- *   - enable applications to find said files later, as long as the session or temp files haven't been purged. 
+ *   - enable applications to find said files later, as long as the db table or temp files haven't been purged.
  *   - enable the uploading user (and *ONLY* the uploading user) to access said files, and thumbnails of said files, via a URL.
- *     We accomplish this by making the session serve as a URL->file mapping, on the assumption that nobody else can access 
- *     the session, even the uploading user. See SpecialUploadStash, which implements a web interface to some files stored this way.
+ *     We accomplish this using a database table, with ownership checking as you might expect. See SpecialUploadStash, which
+ *     implements a web interface to some files stored this way.
+ *
+ * UploadStash right now is *mostly* intended to show you one user's slice of the entire stash. The user parameter is only optional
+ * because there are few cases where we clean out the stash from an automated script. In the future we might refactor this.
+ *
+ * UploadStash represents the entire stash of temporary files.
+ * UploadStashFile is a filestore for the actual physical disk files.
+ * UploadFromStash extends UploadBase, and represents a single stashed file as it is moved from the stash to the regular file repository
  *
+ * @ingroup Upload
  */
 class UploadStash {
 
        // Format of the key for files -- has to be suitable as a filename itself (e.g. ab12cd34ef.jpg)
-       const KEY_FORMAT_REGEX = '/^[\w-]+\.\w+$/';
+       const KEY_FORMAT_REGEX = '/^[\w-\.]+\.\w*$/';
 
-       // repository that this uses to store temp files
-       // public because we sometimes need to get a LocalFile within the same repo.
-       public $repo; 
-       
-       // array of initialized objects obtained from session (lazily initialized upon getFile())
-       private $files = array();  
+       /**
+        * repository that this uses to store temp files
+        * public because we sometimes need to get a LocalFile within the same repo.
+        *
+        * @var LocalRepo
+        */
+       public $repo;
+
+       // array of initialized repo objects
+       protected $files = array();
 
-       // TODO: Once UploadBase starts using this, switch to use these constants rather than UploadBase::SESSION*
-       // const SESSION_VERSION = 2;
-       // const SESSION_KEYNAME = 'wsUploadData';
+       // cache of the file metadata that's stored in the database
+       protected $fileMetadata = array();
+
+       // fileprops cache
+       protected $fileProps = array();
+
+       // current user
+       protected $user, $userId, $isLoggedIn;
 
        /**
-        * Represents the session which contains temporarily stored files.
+        * Represents a temporary filestore, with metadata in the database.
         * Designed to be compatible with the session stashing code in UploadBase (should replace it eventually)
+        *
+        * @param $repo FileRepo
         */
-       public function __construct( $repo ) { 
-
+       public function __construct( FileRepo $repo, $user = null ) {
                // this might change based on wiki's configuration.
                $this->repo = $repo;
 
-               if ( ! isset( $_SESSION ) ) {
-                       throw new UploadStashNotAvailableException( 'no session variable' );
+               // if a user was passed, use it. otherwise, attempt to use the global.
+               // this keeps FileRepo from breaking when it creates an UploadStash object
+               if ( $user ) {
+                       $this->user = $user;
+               } else {
+                       global $wgUser;
+                       $this->user = $wgUser;
                }
 
-               if ( !isset( $_SESSION[UploadBase::SESSION_KEYNAME] ) ) {
-                       $_SESSION[UploadBase::SESSION_KEYNAME] = array();
+               if ( is_object( $this->user ) ) {
+                       $this->userId = $this->user->getId();
+                       $this->isLoggedIn = $this->user->isLoggedIn();
                }
-               
        }
 
-
        /**
         * Get a file and its metadata from the stash.
-        * May throw exception if session data cannot be parsed due to schema change, or key not found.
+        * The noAuth param is a bit janky but is required for automated scripts which clean out the stash.
         *
-        * @param $key Integer: key
+        * @param $key String: key under which file information is stored
+        * @param $noAuth Boolean (optional) Don't check authentication. Used by maintenance scripts.
         * @throws UploadStashFileNotFoundException
-        * @throws UploadStashBadVersionException
+        * @throws UploadStashNotLoggedInException
+        * @throws UploadStashWrongOwnerException
+        * @throws UploadStashBadPathException
         * @return UploadStashFile
         */
-       public function getFile( $key ) {
+       public function getFile( $key, $noAuth = false ) {
+
                if ( ! preg_match( self::KEY_FORMAT_REGEX, $key ) ) {
                        throw new UploadStashBadPathException( "key '$key' is not in a proper format" );
-               } 
-               if ( !isset( $this->files[$key] ) ) {
-                       if ( !isset( $_SESSION[UploadBase::SESSION_KEYNAME][$key] ) ) {
-                               throw new UploadStashFileNotFoundException( "key '$key' not found in stash" );
+               }
+
+               if ( !$noAuth ) {
+                       if ( !$this->isLoggedIn ) {
+                               throw new UploadStashNotLoggedInException( __METHOD__ . ' No user is logged in, files must belong to users' );
                        }
+               }
 
-                       $data = $_SESSION[UploadBase::SESSION_KEYNAME][$key];
-                       // guards against PHP class changing while session data doesn't
-                       if ($data['version'] !== UploadBase::SESSION_VERSION ) {
-                               throw new UploadStashBadVersionException( $data['version'] . " does not match current version " . UploadBase::SESSION_VERSION );
+               if ( !isset( $this->fileMetadata[$key] ) ) {
+                       if ( !$this->fetchFileMetadata( $key ) ) {
+                               // If nothing was received, it's likely due to replication lag.  Check the master to see if the record is there.
+                               $this->fetchFileMetadata( $key, DB_MASTER );
                        }
-               
-                       // separate the stashData into the path, and then the rest of the data
-                       $path = $data['mTempPath'];
-                       unset( $data['mTempPath'] );
-
-                       $file = new UploadStashFile( $this, $this->repo, $path, $key, $data );
-                       if ( $file->getSize() === 0 ) {
-                               throw new UploadStashZeroLengthFileException( "File is zero length" );
+
+                       if ( !isset( $this->fileMetadata[$key] ) ) {
+                               throw new UploadStashFileNotFoundException( "key '$key' not found in stash" );
                        }
-                       $this->files[$key] = $file;
 
+                       // create $this->files[$key]
+                       $this->initFile( $key );
+
+                       // fetch fileprops
+                       $path = $this->fileMetadata[$key]['us_path'];
+                       $this->fileProps[$key] = $this->repo->getFileProps( $path );
+               }
+
+               if ( ! $this->files[$key]->exists() ) {
+                       wfDebug( __METHOD__ . " tried to get file at $key, but it doesn't exist\n" );
+                       throw new UploadStashBadPathException( "path doesn't exist" );
+               }
+
+               if ( !$noAuth ) {
+                       if ( $this->fileMetadata[$key]['us_user'] != $this->userId ) {
+                               throw new UploadStashWrongOwnerException( "This file ($key) doesn't belong to the current user." );
+                       }
                }
+
                return $this->files[$key];
        }
 
        /**
-        * Stash a file in a temp directory and record that we did this in the session, along with other metadata.
-        * We store data in a flat key-val namespace because that's how UploadBase did it. This also means we have to
-        * ensure that the key-val pairs in $data do not overwrite other required fields.
+        * Getter for file metadata.
+        *
+        * @param key String: key under which file information is stored
+        * @return Array
+        */
+       public function getMetadata ( $key ) {
+               $this->getFile( $key );
+               return $this->fileMetadata[$key];
+       }
+
+       /**
+        * Getter for fileProps
+        *
+        * @param key String: key under which file information is stored
+        * @return Array
+        */
+       public function getFileProps ( $key ) {
+               $this->getFile( $key );
+               return $this->fileProps[$key];
+       }
+
+       /**
+        * Stash a file in a temp directory and record that we did this in the database, along with other metadata.
         *
         * @param $path String: path to file you want stashed
-        * @param $data Array: optional, other data you want associated with the file. Do not use 'mTempPath', 'mFileProps', 'mFileSize', or 'version' as keys here
-        * @param $key String: optional, unique key for this file in this session. Used for directory hashing when storing, otherwise not important
+        * @param $sourceType String: the type of upload that generated this file (currently, I believe, 'file' or null)
         * @throws UploadStashBadPathException
         * @throws UploadStashFileException
+        * @throws UploadStashNotLoggedInException
         * @return UploadStashFile: file, or null on failure
         */
-       public function stashFile( $path, $data = array(), $key = null ) {
+       public function stashFile( $path, $sourceType = null ) {
                if ( ! file_exists( $path ) ) {
-                       wfDebug( "UploadStash: tried to stash file at '$path', but it doesn't exist\n" );
+                       wfDebug( __METHOD__ . " tried to stash file at '$path', but it doesn't exist\n" );
                        throw new UploadStashBadPathException( "path doesn't exist" );
                }
-                $fileProps = File::getPropsFromPath( $path );
+               $fileProps = FSFile::getPropsFromPath( $path );
+               wfDebug( __METHOD__ . " stashing file at '$path'\n" );
 
                // we will be initializing from some tmpnam files that don't have extensions.
                // most of MediaWiki assumes all uploaded files have good extensions. So, we fix this.
@@ -114,93 +196,226 @@ class UploadStash {
                                throw new UploadStashFileException( "couldn't rename $path to have a better extension at $pathWithGoodExtension" );
                        }
                        $path = $pathWithGoodExtension;
-               } 
-
-               // If no key was supplied, use content hash. Also has the nice property of collapsing multiple identical files
-               // uploaded this session, which could happen if uploads had failed.
-               if ( is_null( $key ) ) {
-                       $key = $fileProps['sha1'] . "." . $extension;
                }
 
+               // If no key was supplied, make one.  a mysql insertid would be totally reasonable here, except
+               // that for historical reasons, the key is this random thing instead.  At least it's not guessable.
+               //
+               // some things that when combined will make a suitably unique key.
+               // see: http://www.jwz.org/doc/mid.html
+               list ($usec, $sec) = explode( ' ', microtime() );
+               $usec = substr($usec, 2);
+               $key = wfBaseConvert( $sec . $usec, 10, 36 ) . '.' .
+                       wfBaseConvert( mt_rand(), 10, 36 ) . '.'.
+                       $this->userId . '.' .
+                       $extension;
+
+               $this->fileProps[$key] = $fileProps;
+
                if ( ! preg_match( self::KEY_FORMAT_REGEX, $key ) ) {
                        throw new UploadStashBadPathException( "key '$key' is not in a proper format" );
-               } 
+               }
 
+               wfDebug( __METHOD__ . " key for '$path': $key\n" );
 
                // if not already in a temporary area, put it there
-               $status = $this->repo->storeTemp( basename( $path ), $path );
+               $storeStatus = $this->repo->storeTemp( basename( $path ), $path );
 
-               if( ! $status->isOK() ) {
+               if ( ! $storeStatus->isOK() ) {
                        // It is a convention in MediaWiki to only return one error per API exception, even if multiple errors
                        // are available. We use reset() to pick the "first" thing that was wrong, preferring errors to warnings.
-                       // This is a bit lame, as we may have more info in the $status and we're throwing it away, but to fix it means
+                       // This is a bit lame, as we may have more info in the $storeStatus and we're throwing it away, but to fix it means
                        // redesigning API errors significantly.
-                       // $status->value just contains the virtual URL (if anything) which is probably useless to the caller
-                       $error = reset( $status->getErrorsArray() );
+                       // $storeStatus->value just contains the virtual URL (if anything) which is probably useless to the caller
+                       $error = $storeStatus->getErrorsArray();
+                       $error = reset( $error );
                        if ( ! count( $error ) ) {
-                               $error = reset( $status->getWarningsArray() );
+                               $error = $storeStatus->getWarningsArray();
+                               $error = reset( $error );
                                if ( ! count( $error ) ) {
                                        $error = array( 'unknown', 'no error recorded' );
                                }
                        }
-                       throw new UploadStashFileException( "error storing file in '$path': " . implode( '; ', $error ) );
+                       // at this point, $error should contain the single "most important" error, plus any parameters.
+                       throw new UploadStashFileException( "Error storing file in '$path': " . wfMessage( $error )->text() );
                }
-               $stashPath = $status->value;
-
-               // required info we always store. Must trump any other application info in $data
-               // 'mTempPath', 'mFileSize', and 'mFileProps' are arbitrary names
-               // chosen for compatibility with UploadBase's way of doing this.
-               $requiredData = array( 
-                       'mTempPath' => $stashPath,
-                       'mFileSize' => $fileProps['size'],
-                       'mFileProps' => $fileProps,
-                       'version' => UploadBase::SESSION_VERSION
+               $stashPath = $storeStatus->value;
+
+               // fetch the current user ID
+               if ( !$this->isLoggedIn ) {
+                       throw new UploadStashNotLoggedInException( __METHOD__ . ' No user is logged in, files must belong to users' );
+               }
+
+               // insert the file metadata into the db.
+               wfDebug( __METHOD__ . " inserting $stashPath under $key\n" );
+               $dbw = $this->repo->getMasterDb();
+
+               $this->fileMetadata[$key] = array(
+                       'us_id' => $dbw->nextSequenceValue( 'uploadstash_us_id_seq' ),
+                       'us_user' => $this->userId,
+                       'us_key' => $key,
+                       'us_orig_path' => $path,
+                       'us_path' => $stashPath, // virtual URL
+                       'us_size' => $fileProps['size'],
+                       'us_sha1' => $fileProps['sha1'],
+                       'us_mime' => $fileProps['mime'],
+                       'us_media_type' => $fileProps['media_type'],
+                       'us_image_width' => $fileProps['width'],
+                       'us_image_height' => $fileProps['height'],
+                       'us_image_bits' => $fileProps['bits'],
+                       'us_source_type' => $sourceType,
+                       'us_timestamp' => $dbw->timestamp(),
+                       'us_status' => 'finished'
+               );
+
+               $dbw->insert(
+                       'uploadstash',
+                       $this->fileMetadata[$key],
+                       __METHOD__
                );
 
-               // now, merge required info and extra data into the session. (The extra data changes from application to application.
-               // UploadWizard wants different things than say FirefoggChunkedUpload.)
-               wfDebug( __METHOD__ . " storing under $key\n" );
-               $_SESSION[UploadBase::SESSION_KEYNAME][$key] = array_merge( $data, $requiredData );
-               
+               // store the insertid in the class variable so immediate retrieval (possibly laggy) isn't necesary.
+               $this->fileMetadata[$key]['us_id'] = $dbw->insertId();
+
+               # create the UploadStashFile object for this file.
+               $this->initFile( $key );
+
                return $this->getFile( $key );
        }
 
        /**
         * Remove all files from the stash.
         * Does not clean up files in the repo, just the record of them.
+        *
+        * @throws UploadStashNotLoggedInException
         * @return boolean: success
         */
        public function clear() {
-               $_SESSION[UploadBase::SESSION_KEYNAME] = array();
+               if ( !$this->isLoggedIn ) {
+                       throw new UploadStashNotLoggedInException( __METHOD__ . ' No user is logged in, files must belong to users' );
+               }
+
+               wfDebug( __METHOD__ . ' clearing all rows for user ' . $this->userId . "\n" );
+               $dbw = $this->repo->getMasterDb();
+               $dbw->delete(
+                       'uploadstash',
+                       array( 'us_user' => $this->userId ),
+                       __METHOD__
+               );
+
+               # destroy objects.
+               $this->files = array();
+               $this->fileMetadata = array();
+
                return true;
        }
 
-
        /**
-        * Remove a particular file from the stash.
-        * Does not clean up file in the repo, just the record of it.
+        * Remove a particular file from the stash.  Also removes it from the repo.
+        *
+        * @throws UploadStashNotLoggedInException
+        * @throws UploadStashWrongOwnerException
         * @return boolean: success
         */
        public function removeFile( $key ) {
-               unset ( $_SESSION[UploadBase::SESSION_KEYNAME][$key] );
+               if ( !$this->isLoggedIn ) {
+                       throw new UploadStashNotLoggedInException( __METHOD__ . ' No user is logged in, files must belong to users' );
+               }
+
+               $dbw = $this->repo->getMasterDb();
+
+               // this is a cheap query. it runs on the master so that this function still works when there's lag.
+               // it won't be called all that often.
+               $row = $dbw->selectRow(
+                       'uploadstash',
+                       'us_user',
+                       array( 'us_key' => $key ),
+                       __METHOD__
+               );
+
+               if( !$row ) {
+                       throw new UploadStashNoSuchKeyException( "No such key ($key), cannot remove" );
+               }
+
+               if ( $row->us_user != $this->userId ) {
+                       throw new UploadStashWrongOwnerException( "Can't delete: the file ($key) doesn't belong to this user." );
+               }
+
+               return $this->removeFileNoAuth( $key );
+       }
+
+
+       /**
+        * Remove a file (see removeFile), but doesn't check ownership first.
+        *
+        * @return boolean: success
+        */
+       public function removeFileNoAuth( $key ) {
+               wfDebug( __METHOD__ . " clearing row $key\n" );
+
+               $dbw = $this->repo->getMasterDb();
+
+               // this gets its own transaction since it's called serially by the cleanupUploadStash maintenance script
+               $dbw->begin( __METHOD__ );
+               $dbw->delete(
+                       'uploadstash',
+                       array( 'us_key' => $key ),
+                       __METHOD__
+               );
+               $dbw->commit( __METHOD__ );
+
+               // TODO: look into UnregisteredLocalFile and find out why the rv here is sometimes wrong (false when file was removed)
+               // for now, ignore.
+               $this->files[$key]->remove();
+
+               unset( $this->files[$key] );
+               unset( $this->fileMetadata[$key] );
+
                return true;
        }
 
        /**
         * List all files in the stash.
+        *
+        * @throws UploadStashNotLoggedInException
+        * @return Array
         */
        public function listFiles() {
-               return array_keys( $_SESSION[UploadBase::SESSION_KEYNAME] );
+               if ( !$this->isLoggedIn ) {
+                       throw new UploadStashNotLoggedInException( __METHOD__ . ' No user is logged in, files must belong to users' );
+               }
+
+               $dbr = $this->repo->getSlaveDb();
+               $res = $dbr->select(
+                       'uploadstash',
+                       'us_key',
+                       array( 'us_user' => $this->userId ),
+                       __METHOD__
+               );
+
+               if ( !is_object( $res ) || $res->numRows() == 0 ) {
+                       // nothing to do.
+                       return false;
+               }
+
+               // finish the read before starting writes.
+               $keys = array();
+               foreach ( $res as $row ) {
+                       array_push( $keys, $row->us_key );
+               }
+
+               return $keys;
        }
 
        /**
         * Find or guess extension -- ensuring that our extension matches our mime type.
-        * Since these files are constructed from php tempnames they may not start off 
+        * Since these files are constructed from php tempnames they may not start off
         * with an extension.
-        * XXX this is somewhat redundant with the checks that ApiUpload.php does with incoming 
+        * XXX this is somewhat redundant with the checks that ApiUpload.php does with incoming
         * uploads versus the desired filename. Maybe we can get that passed to us...
+        * @return string
         */
-       public static function getExtensionForPath( $path ) {   
+       public static function getExtensionForPath( $path ) {
                // Does this have an extension?
                $n = strrpos( $path, '.' );
                $extension = null;
@@ -211,8 +426,8 @@ class UploadStash {
                        $magic = MimeMagic::singleton();
                        $mimeType = $magic->guessMimeType( $path );
                        $extensions = explode( ' ', MimeMagic::singleton()->getExtensionsForType( $mimeType ) );
-                       if ( count( $extensions ) ) { 
-                               $extension = $extensions[0];    
+                       if ( count( $extensions ) ) {
+                               $extension = $extensions[0];
                        }
                }
 
@@ -223,48 +438,93 @@ class UploadStash {
                return File::normalizeExtension( $extension );
        }
 
+       /**
+        * Helper function: do the actual database query to fetch file metadata.
+        *
+        * @param $key String: key
+        * @return boolean
+        */
+       protected function fetchFileMetadata( $key, $readFromDB = DB_SLAVE ) {
+               // populate $fileMetadata[$key]
+               $dbr = null;
+               if( $readFromDB === DB_MASTER ) {
+                       // sometimes reading from the master is necessary, if there's replication lag.
+                       $dbr = $this->repo->getMasterDb();
+               } else {
+                       $dbr = $this->repo->getSlaveDb();
+               }
+
+               $row = $dbr->selectRow(
+                       'uploadstash',
+                       '*',
+                       array( 'us_key' => $key ),
+                       __METHOD__
+               );
+
+               if ( !is_object( $row ) ) {
+                       // key wasn't present in the database. this will happen sometimes.
+                       return false;
+               }
+
+               $this->fileMetadata[$key] = (array)$row;
+
+               return true;
+       }
+
+       /**
+        * Helper function: Initialize the UploadStashFile for a given file.
+        *
+        * @param $path String: path to file
+        * @param $key String: key under which to store the object
+        * @throws UploadStashZeroLengthFileException
+        * @return bool
+        */
+       protected function initFile( $key ) {
+               $file = new UploadStashFile( $this->repo, $this->fileMetadata[$key]['us_path'], $key );
+               if ( $file->getSize() === 0 ) {
+                       throw new UploadStashZeroLengthFileException( "File is zero length" );
+               }
+               $this->files[$key] = $file;
+               return true;
+       }
 }
 
 class UploadStashFile extends UnregisteredLocalFile {
-       private $sessionStash;
-       private $sessionKey;
-       private $sessionData;
+       private $fileKey;
        private $urlName;
+       protected $url;
 
        /**
         * A LocalFile wrapper around a file that has been temporarily stashed, so we can do things like create thumbnails for it
         * Arguably UnregisteredLocalFile should be handling its own file repo but that class is a bit retarded currently
         *
-        * @param $stash UploadStash: useful for obtaining config, stashing transformed files
         * @param $repo FileRepo: repository where we should find the path
         * @param $path String: path to file
         * @param $key String: key to store the path and any stashed data under
-        * @param $data String: any other data we want stored with this file
         * @throws UploadStashBadPathException
         * @throws UploadStashFileNotFoundException
         */
-       public function __construct( $stash, $repo, $path, $key, $data ) { 
-               $this->sessionStash = $stash;
-               $this->sessionKey = $key;
-               $this->sessionData = $data;
+       public function __construct( $repo, $path, $key ) {
+               $this->fileKey = $key;
 
                // resolve mwrepo:// urls
                if ( $repo->isVirtualUrl( $path ) ) {
-                       $path = $repo->resolveVirtualUrl( $path );      
-               }
+                       $path = $repo->resolveVirtualUrl( $path );
+               } else {
 
-               // check if path appears to be sane, no parent traversals, and is in this repo's temp zone.
-               $repoTempPath = $repo->getZonePath( 'temp' );
-               if ( ( ! $repo->validateFilename( $path ) ) || 
-                               ( strpos( $path, $repoTempPath ) !== 0 ) ) {
-                       wfDebug( "UploadStash: tried to construct an UploadStashFile from a file that should already exist at '$path', but path is not valid\n" );
-                       throw new UploadStashBadPathException( 'path is not valid' );
-               }
+                       // check if path appears to be sane, no parent traversals, and is in this repo's temp zone.
+                       $repoTempPath = $repo->getZonePath( 'temp' );
+                       if ( ( ! $repo->validateFilename( $path ) ) ||
+                                       ( strpos( $path, $repoTempPath ) !== 0 ) ) {
+                               wfDebug( "UploadStash: tried to construct an UploadStashFile from a file that should already exist at '$path', but path is not valid\n" );
+                               throw new UploadStashBadPathException( 'path is not valid' );
+                       }
 
-               // check if path exists! and is a plain file.
-               if ( ! $repo->fileExists( $path, FileRepo::FILES_ONLY ) ) {
-                       wfDebug( "UploadStash: tried to construct an UploadStashFile from a file that should already exist at '$path', but path is not found\n" );
-                       throw new UploadStashFileNotFoundException( 'cannot find path, or not a plain file' );
+                       // check if path exists! and is a plain file.
+                       if ( ! $repo->fileExists( $path ) ) {
+                               wfDebug( "UploadStash: tried to construct an UploadStashFile from a file that should already exist at '$path', but path is not found\n" );
+                               throw new UploadStashFileNotFoundException( 'cannot find path, or not a plain file' );
+                       }
                }
 
                parent::__construct( false, $repo, $path, false );
@@ -286,13 +546,13 @@ class UploadStashFile extends UnregisteredLocalFile {
 
        /**
         * Get the path for the thumbnail (actually any transformation of this file)
-        * The actual argument is the result of thumbName although we seem to have 
+        * The actual argument is the result of thumbName although we seem to have
         * buggy code elsewhere that expects a boolean 'suffix'
         *
         * @param $thumbName String: name of thumbnail (e.g. "120px-123456.jpg" ), or false to just get the path
         * @return String: path thumbnail should take on filesystem, or containing directory if thumbname is false
         */
-       public function getThumbPath( $thumbName = false ) { 
+       public function getThumbPath( $thumbName = false ) {
                $path = dirname( $this->path );
                if ( $thumbName !== false ) {
                        $path .= "/$thumbName";
@@ -301,8 +561,8 @@ class UploadStashFile extends UnregisteredLocalFile {
        }
 
        /**
-        * Return the file/url base name of a thumbnail with the specified parameters. 
-        * We override this because we want to use the pretty url name instead of the 
+        * Return the file/url base name of a thumbnail with the specified parameters.
+        * We override this because we want to use the pretty url name instead of the
         * ugly file name.
         *
         * @param $params Array: handler-specific parameters
@@ -315,35 +575,35 @@ class UploadStashFile extends UnregisteredLocalFile {
        /**
         * Helper function -- given a 'subpage', return the local URL e.g. /wiki/Special:UploadStash/subpage
         * @param {String} $subPage
-        * @return {String} local URL for this subpage in the Special:UploadStash space. 
+        * @return {String} local URL for this subpage in the Special:UploadStash space.
         */
        private function getSpecialUrl( $subPage ) {
                return SpecialPage::getTitleFor( 'UploadStash', $subPage )->getLocalURL();
        }
 
-       /** 
-        * Get a URL to access the thumbnail 
-        * This is required because the model of how files work requires that 
+       /**
+        * Get a URL to access the thumbnail
+        * This is required because the model of how files work requires that
         * the thumbnail urls be predictable. However, in our model the URL is not based on the filename
-        * (that's hidden in the session)
+        * (that's hidden in the db)
         *
         * @param $thumbName String: basename of thumbnail file -- however, we don't want to use the file exactly
         * @return String: URL to access thumbnail, or URL with partial path
         */
-       public function getThumbUrl( $thumbName = false ) { 
+       public function getThumbUrl( $thumbName = false ) {
                wfDebug( __METHOD__ . " getting for $thumbName \n" );
                return $this->getSpecialUrl( 'thumb/' . $this->getUrlName() . '/' . $thumbName );
        }
 
-       /** 
+       /**
         * The basename for the URL, which we want to not be related to the filename.
         * Will also be used as the lookup key for a thumbnail file.
         *
         * @return String: base url name, like '120px-123456.jpg'
         */
-       public function getUrlName() { 
+       public function getUrlName() {
                if ( ! $this->urlName ) {
-                       $this->urlName = $this->sessionKey;
+                       $this->urlName = $this->fileKey;
                }
                return $this->urlName;
        }
@@ -362,22 +622,22 @@ class UploadStashFile extends UnregisteredLocalFile {
        }
 
        /**
-        * Parent classes use this method, for no obvious reason, to return the path (relative to wiki root, I assume). 
+        * Parent classes use this method, for no obvious reason, to return the path (relative to wiki root, I assume).
         * But with this class, the URL is unrelated to the path.
         *
         * @return String: url
         */
-       public function getFullUrl() { 
+       public function getFullUrl() {
                return $this->getUrl();
        }
 
        /**
-        * Getter for session key (the session-unique id by which this file's location & metadata is stored in the session)
+        * Getter for file key (the unique id by which this file's location & metadata is stored in the db)
         *
-        * @return String: session key
+        * @return String: file key
         */
-       public function getSessionKey() {
-               return $this->sessionKey;
+       public function getFileKey() {
+               return $this->fileKey;
        }
 
        /**
@@ -385,15 +645,25 @@ class UploadStashFile extends UnregisteredLocalFile {
         * @return Status: success
         */
        public function remove() {
+               if ( !$this->repo->fileExists( $this->path ) ) {
+                       // Maybe the file's already been removed? This could totally happen in UploadBase.
+                       return true;
+               }
+
                return $this->repo->freeTemp( $this->path );
        }
 
+       public function exists() {
+               return $this->repo->fileExists( $this->path );
+       }
+
 }
 
 class UploadStashNotAvailableException extends MWException {};
 class UploadStashFileNotFoundException extends MWException {};
 class UploadStashBadPathException extends MWException {};
-class UploadStashBadVersionException extends MWException {};
 class UploadStashFileException extends MWException {};
 class UploadStashZeroLengthFileException extends MWException {};
-
+class UploadStashNotLoggedInException extends MWException {};
+class UploadStashWrongOwnerException extends MWException {};
+class UploadStashNoSuchKeyException extends MWException {};