]> git.mxchange.org Git - quix0rs-gnu-social.git/blobdiff - lib/mediafile.php
Only accept filenames for existing files in getUploadedFileType
[quix0rs-gnu-social.git] / lib / mediafile.php
index a41d7c76b52d190e066e3b093570b67ad08d1f26..e01d2eaeabc9b896be6f301288c5d125daceee17 100644 (file)
@@ -36,21 +36,17 @@ if (!defined('STATUSNET') && !defined('LACONICA')) {
 
 class MediaFile
 {
+    protected $scoped   = null;
 
     var $filename      = null;
     var $fileRecord    = null;
-    var $user          = null;
     var $fileurl       = null;
     var $short_fileurl = null;
     var $mimetype      = null;
 
-    function __construct($user = null, $filename = null, $mimetype = null)
+    function __construct(Profile $scoped, $filename = null, $mimetype = null)
     {
-        if ($user == null) {
-            $this->user = common_current_user();
-        } else {
-            $this->user = $user;
-        }
+        $this->scoped = $scoped;
 
         $this->filename   = $filename;
         $this->mimetype   = $mimetype;
@@ -124,7 +120,7 @@ class MediaFile
             return null;
         }
 
-        $outname = File::filename($this->user->getProfile(), 'thumb-' . $this->filename, $this->mimetype);
+        $outname = File::filename($this->scoped, 'thumb-' . $this->filename, $this->mimetype);
         $outpath = File::path($outname);
 
         $maxWidth = common_config('attachments', 'thumb_width');
@@ -158,7 +154,7 @@ class MediaFile
 
     function maybeAddRedir($file_id, $url)
     {
-        $file_redir = File_redirection::staticGet('url', $url);
+        $file_redir = File_redirection::getKV('url', $url);
 
         if (empty($file_redir)) {
 
@@ -176,10 +172,10 @@ class MediaFile
         }
     }
 
-    static function fromUpload($param = 'media', $user = null)
+    static function fromUpload($param = 'media', Profile $scoped)
     {
-        if (empty($user)) {
-            $user = common_current_user();
+        if (is_null($scoped)) {
+            $scoped = Profile::current();
         }
 
         if (!isset($_FILES[$param]['error'])){
@@ -229,85 +225,49 @@ class MediaFile
             return;
         }
 
-        if (!MediaFile::respectsQuota($user, $_FILES[$param]['size'])) {
-
-            // Should never actually get here
-
-            @unlink($_FILES[$param]['tmp_name']);
-            // TRANS: Client exception thrown when a file upload operation would cause a user to exceed a set quota.
-            throw new ClientException(_('File exceeds user\'s quota.'));
-            return;
-        }
+        // Throws exception if additional size does not respect quota
+        File::respectsQuota($scoped, $_FILES[$param]['size']);
 
         $mimetype = MediaFile::getUploadedFileType($_FILES[$param]['tmp_name'],
                                                    $_FILES[$param]['name']);
 
-        $filename = null;
-
-        if (isset($mimetype)) {
-
-            $basename = basename($_FILES[$param]['name']);
-            $filename = File::filename($user->getProfile(), $basename, $mimetype);
-            $filepath = File::path($filename);
-
-            $result = move_uploaded_file($_FILES[$param]['tmp_name'], $filepath);
+        $basename = basename($_FILES[$param]['name']);
+        $filename = File::filename($scoped, $basename, $mimetype);
+        $filepath = File::path($filename);
 
-            if (!$result) {
-                // TRANS: Client exception thrown when a file upload operation fails because the file could
-                // TRANS: not be moved from the temporary folder to the permanent file location.
-                throw new ClientException(_('File could not be moved to destination directory.'));
-                return;
-            }
+        $result = move_uploaded_file($_FILES[$param]['tmp_name'], $filepath);
 
-        } else {
-            // TRANS: Client exception thrown when a file upload operation has been stopped because the MIME
-            // TRANS: type of the uploaded file could not be determined.
-            throw new ClientException(_('Could not determine file\'s MIME type.'));
-            return;
+        if (!$result) {
+            // TRANS: Client exception thrown when a file upload operation fails because the file could
+            // TRANS: not be moved from the temporary folder to the permanent file location.
+            throw new ClientException(_('File could not be moved to destination directory.'));
         }
 
-        return new MediaFile($user, $filename, $mimetype);
+        return new MediaFile($scoped, $filename, $mimetype);
     }
 
-    static function fromFilehandle($fh, $user) {
+    static function fromFilehandle($fh, Profile $scoped) {
 
         $stream = stream_get_meta_data($fh);
 
-        if (!MediaFile::respectsQuota($user, filesize($stream['uri']))) {
-
-            // Should never actually get here
-
-            // TRANS: Client exception thrown when a file upload operation would cause a user to exceed a set quota.
-            throw new ClientException(_('File exceeds user\'s quota.'));
-            return;
-        }
-
-        $mimetype = MediaFile::getUploadedFileType($fh);
-
-        $filename = null;
+        File::respectsQuota($scoped, filesize($stream['uri']));
 
-        if (isset($mimetype)) {
+        $mimetype = MediaFile::getUploadedFileType($stream['uri']);
 
-            $filename = File::filename($user->getProfile(), "email", $mimetype);
+        $filename = File::filename($scoped, "email", $mimetype);
 
-            $filepath = File::path($filename);
+        $filepath = File::path($filename);
 
-            $result = copy($stream['uri'], $filepath) && chmod($filepath, 0664);
+        $result = copy($stream['uri'], $filepath) && chmod($filepath, 0664);
 
-            if (!$result) {
-                // TRANS: Client exception thrown when a file upload operation fails because the file could
-                // TRANS: not be moved from the temporary folder to the permanent file location.
-                throw new ClientException(_('File could not be moved to destination directory.' .
-                    $stream['uri'] . ' ' . $filepath));
-            }
-        } else {
-            // TRANS: Client exception thrown when a file upload operation has been stopped because the MIME
-            // TRANS: type of the uploaded file could not be determined.
-            throw new ClientException(_('Could not determine file\'s MIME type.'));
-            return;
+        if (!$result) {
+            // TRANS: Client exception thrown when a file upload operation fails because the file could
+            // TRANS: not be moved from the temporary folder to the permanent file location.
+            throw new ClientException(_('File could not be moved to destination directory.' .
+                $stream['uri'] . ' ' . $filepath));
         }
 
-        return new MediaFile($user, $filename, $mimetype);
+        return new MediaFile($scoped, $filename, $mimetype);
     }
 
     /**
@@ -325,14 +285,16 @@ class MediaFile
      * @throws ClientException if type is known, but not supported for local uploads
      */
     static function getUploadedFileType($f, $originalFilename=false) {
+        global $_PEAR;
+
         require_once 'MIME/Type.php';
         require_once 'MIME/Type/Extension.php';
 
         // We have to disable auto handling of PEAR errors
-        PEAR::staticPushErrorHandling(PEAR_ERROR_RETURN);
+        $_PEAR->staticPushErrorHandling(PEAR_ERROR_RETURN);
         $mte = new MIME_Type_Extension();
 
-        $cmd = &PEAR::getStaticProperty('MIME_Type', 'fileCmd');
+        $cmd = &$_PEAR->getStaticProperty('MIME_Type', 'fileCmd');
         $cmd = common_config('attachments', 'filecommand');
 
         $filetype = null;
@@ -342,19 +304,8 @@ class MediaFile
         // are unambiguous for most image files, but nearly useless
         // for office document formats.
 
-        if (is_string($f)) {
-
-            // assuming a filename
-
-            $filetype = MIME_Type::autoDetect($f);
-
-        } else {
-
-            // assuming a filehandle
-
-            $stream  = stream_get_meta_data($f);
-            $filetype = MIME_Type::autoDetect($stream['uri']);
-        }
+        // We only accept filenames to existing files
+        $filetype = MIME_Type::autoDetect($f);
 
         // The content-based sources for MIME_Type::autoDetect()
         // are wildly unreliable for office-type documents. If we've
@@ -362,7 +313,9 @@ class MediaFile
         // we'll try detecting a type from its extension...
         $unclearTypes = array('application/octet-stream',
                               'application/vnd.ms-office',
-                              'application/zip');
+                              'application/zip',
+                              // TODO: for XML we could do better content-based sniffing too
+                              'text/xml');
 
         if ($originalFilename && (!$filetype || in_array($filetype, $unclearTypes))) {
             $type = $mte->getMIMEType($originalFilename);
@@ -383,7 +336,7 @@ class MediaFile
         }
         if ($supported === true || in_array($filetype, $supported)) {
             // Restore PEAR error handlers for our DB code...
-            PEAR::staticPopErrorHandling();
+            $_PEAR->staticPopErrorHandling();
             return $filetype;
         }
         $media = MIME_Type::getMedia($filetype);
@@ -399,19 +352,7 @@ class MediaFile
             $hint = sprintf(_('"%s" is not a supported file type on this server.'), $filetype);
         }
         // Restore PEAR error handlers for our DB code...
-        PEAR::staticPopErrorHandling();
+        $_PEAR->staticPopErrorHandling();
         throw new ClientException($hint);
     }
-
-    static function respectsQuota($user, $filesize)
-    {
-        $file = new File;
-        $result = $file->isRespectsQuota($user, $filesize);
-        if ($result === true) {
-            return true;
-        } else {
-            throw new ClientException($result);
-        }
-    }
-
 }