silverstripe / silverstripe-mimevalidator

src/MimeUploadValidator.php

Indentation +154 added lines, -154 removed lines

@@ -19,158 +19,158 @@
  */
 class MimeUploadValidator extends Upload_Validator
 {
-    /**
-     * The preg_replace() pattern to use against MIME types. Used to strip out
-     * useless characters so matching of MIME types can be fuzzy.
-     *
-     * @var string Regexp pattern
-     */
-    protected $filterPattern = '/.*[\/\.\-\+]/i';
-
-    /**
-     * @param string $pattern
-     */
-    public function setFilterPattern($pattern)
-    {
-        $this->filterPattern = $pattern;
-    }
-
-    /**
-     * @return string
-     */
-    public function getFilterPattern()
-    {
-        return $this->filterPattern;
-    }
-
-    /**
-     * Check if the temporary file has a valid MIME type for it's extension.
-     *
-     * @uses finfo php extension
-     * @return bool|null
-     * @throws MimeUploadValidator_Exception
-     */
-    public function isValidMime()
-    {
-        $extension = strtolower(pathinfo($this->tmpFile['name'], PATHINFO_EXTENSION));
-
-        // we can't check filenames without an extension or no temp file path, let them pass validation.
-        if (!$extension || !$this->tmpFile['tmp_name']) {
-            return true;
-        }
-
-        $expectedMimes = $this->getExpectedMimeTypes($this->tmpFile);
-        if (empty($expectedMimes)) {
-            throw new MimeUploadValidator_Exception(
-                sprintf('Could not find a MIME type for extension %s', $extension)
-            );
-        }
-
-        $fileInfo = new finfo(FILEINFO_MIME_TYPE);
-        $foundMime = $fileInfo->file($this->tmpFile['tmp_name']);
-        if (!$foundMime) {
-            throw new MimeUploadValidator_Exception(
-                sprintf('Could not find a MIME type for file %s', $this->tmpFile['tmp_name'])
-            );
-        }
-
-        foreach ($expectedMimes as $expected) {
-            if ($this->compareMime($foundMime, $expected)) {
-                return true;
-            }
-        }
-
-        return false;
-    }
-
-    /**
-     * Fetches an array of valid mimetypes.
-     *
-     * @param $file
-     * @return array
-     * @throws MimeUploadValidator_Exception
-     */
-    public function getExpectedMimeTypes($file)
-    {
-        $extension = strtolower(pathinfo($file['name'], PATHINFO_EXTENSION));
-
-        // if the finfo php extension isn't loaded, we can't complete this check.
-        if (!class_exists('finfo')) {
-            throw new MimeUploadValidator_Exception('PHP extension finfo is not loaded');
-        }
-
-        // Attempt to figure out which mime types are expected/acceptable here.
-        $expectedMimes = array();
-
-        // Get the mime types set in framework core
-        $knownMimes = Config::inst()->get(HTTP::class, 'MimeTypes');
-        if (isset($knownMimes[$extension])) {
-            $expectedMimes[] = $knownMimes[$extension];
-        }
-
-        // Get the mime types and their variations from mime validator
-        $knownMimes = $this->config()->get('MimeTypes');
-        if (isset($knownMimes[$extension])) {
-          $mimes = (array) $knownMimes[$extension];
-
-          foreach($mimes as $mime){
-              if (!in_array($mime,$expectedMimes)) {
-                  $expectedMimes[] = $mime;
-              }
-          }
-        }
-
-        return $expectedMimes;
-    }
-
-    /**
-     * Check two MIME types roughly match eachother.
-     *
-     * Before we check MIME types, remove known prefixes "vnd.", "x-" etc.
-     * If there is a suffix, we'll use that to compare. Examples:
-     *
-     * application/x-json = json
-     * application/json = json
-     * application/xhtml+xml = xml
-     * application/xml = xml
-     *
-     * @param string $first The first MIME type to compare to the second
-     * @param string $second The second MIME type to compare to the first
-     * @return boolean
-     */
-    public function compareMime($first, $second)
-    {
-        return preg_replace($this->filterPattern, '', $first) === preg_replace($this->filterPattern, '', $second);
-    }
-
-    public function validate()
-    {
-        if (parent::validate() === false) {
-            return false;
-        }
-
-        try {
-            $result = $this->isValidMime();
-
-            if ($result === false) {
-                $this->errors[] = _t(
-                    __CLASS__ . '.INVALIDMIME',
-                    'File extension does not match known MIME type'
-                );
-
-                return false;
-            }
-        } catch (MimeUploadValidator_Exception $e) {
-            $this->errors[] = _t(
-                __CLASS__ . '.FAILEDMIMECHECK',
-                'MIME validation failed: {message}',
-                'Argument 1: Message about why MIME type detection failed',
-                ['message' => $e->getMessage()]
-            );
-
-            return false;
-        }
-
-        return true;
-    }
+	/**
+	 * The preg_replace() pattern to use against MIME types. Used to strip out
+	 * useless characters so matching of MIME types can be fuzzy.
+	 *
+	 * @var string Regexp pattern
+	 */
+	protected $filterPattern = '/.*[\/\.\-\+]/i';
+
+	/**
+	 * @param string $pattern
+	 */
+	public function setFilterPattern($pattern)
+	{
+		$this->filterPattern = $pattern;
+	}
+
+	/**
+	 * @return string
+	 */
+	public function getFilterPattern()
+	{
+		return $this->filterPattern;
+	}
+
+	/**
+	 * Check if the temporary file has a valid MIME type for it's extension.
+	 *
+	 * @uses finfo php extension
+	 * @return bool|null
+	 * @throws MimeUploadValidator_Exception
+	 */
+	public function isValidMime()
+	{
+		$extension = strtolower(pathinfo($this->tmpFile['name'], PATHINFO_EXTENSION));
+
+		// we can't check filenames without an extension or no temp file path, let them pass validation.
+		if (!$extension || !$this->tmpFile['tmp_name']) {
+			return true;
+		}
+
+		$expectedMimes = $this->getExpectedMimeTypes($this->tmpFile);
+		if (empty($expectedMimes)) {
+			throw new MimeUploadValidator_Exception(
+				sprintf('Could not find a MIME type for extension %s', $extension)
+			);
+		}
+
+		$fileInfo = new finfo(FILEINFO_MIME_TYPE);
+		$foundMime = $fileInfo->file($this->tmpFile['tmp_name']);
+		if (!$foundMime) {
+			throw new MimeUploadValidator_Exception(
+				sprintf('Could not find a MIME type for file %s', $this->tmpFile['tmp_name'])
+			);
+		}
+
+		foreach ($expectedMimes as $expected) {
+			if ($this->compareMime($foundMime, $expected)) {
+				return true;
+			}
+		}
+
+		return false;
+	}
+
+	/**
+	 * Fetches an array of valid mimetypes.
+	 *
+	 * @param $file
+	 * @return array
+	 * @throws MimeUploadValidator_Exception
+	 */
+	public function getExpectedMimeTypes($file)
+	{
+		$extension = strtolower(pathinfo($file['name'], PATHINFO_EXTENSION));
+
+		// if the finfo php extension isn't loaded, we can't complete this check.
+		if (!class_exists('finfo')) {
+			throw new MimeUploadValidator_Exception('PHP extension finfo is not loaded');
+		}
+
+		// Attempt to figure out which mime types are expected/acceptable here.
+		$expectedMimes = array();
+
+		// Get the mime types set in framework core
+		$knownMimes = Config::inst()->get(HTTP::class, 'MimeTypes');
+		if (isset($knownMimes[$extension])) {
+			$expectedMimes[] = $knownMimes[$extension];
+		}
+
+		// Get the mime types and their variations from mime validator
+		$knownMimes = $this->config()->get('MimeTypes');
+		if (isset($knownMimes[$extension])) {
+		  $mimes = (array) $knownMimes[$extension];
+
+		  foreach($mimes as $mime){
+			  if (!in_array($mime,$expectedMimes)) {
+				  $expectedMimes[] = $mime;
+			  }
+		  }
+		}
+
+		return $expectedMimes;
+	}
+
+	/**
+	 * Check two MIME types roughly match eachother.
+	 *
+	 * Before we check MIME types, remove known prefixes "vnd.", "x-" etc.
+	 * If there is a suffix, we'll use that to compare. Examples:
+	 *
+	 * application/x-json = json
+	 * application/json = json
+	 * application/xhtml+xml = xml
+	 * application/xml = xml
+	 *
+	 * @param string $first The first MIME type to compare to the second
+	 * @param string $second The second MIME type to compare to the first
+	 * @return boolean
+	 */
+	public function compareMime($first, $second)
+	{
+		return preg_replace($this->filterPattern, '', $first) === preg_replace($this->filterPattern, '', $second);
+	}
+
+	public function validate()
+	{
+		if (parent::validate() === false) {
+			return false;
+		}
+
+		try {
+			$result = $this->isValidMime();
+
+			if ($result === false) {
+				$this->errors[] = _t(
+					__CLASS__ . '.INVALIDMIME',
+					'File extension does not match known MIME type'
+				);
+
+				return false;
+			}
+		} catch (MimeUploadValidator_Exception $e) {
+			$this->errors[] = _t(
+				__CLASS__ . '.FAILEDMIMECHECK',
+				'MIME validation failed: {message}',
+				'Argument 1: Message about why MIME type detection failed',
+				['message' => $e->getMessage()]
+			);
+
+			return false;
+		}
+
+		return true;
+	}
 }

Spacing +4 added lines, -4 removed lines

@@ -113,8 +113,8 @@ 
         if (isset($knownMimes[$extension])) {
           $mimes = (array) $knownMimes[$extension];
 
-          foreach($mimes as $mime){
-              if (!in_array($mime,$expectedMimes)) {
+          foreach ($mimes as $mime) {
+              if (!in_array($mime, $expectedMimes)) {
                   $expectedMimes[] = $mime;
               }
           }
@@ -154,7 +154,7 @@ 
 
             if ($result === false) {
                 $this->errors[] = _t(
-                    __CLASS__ . '.INVALIDMIME',
+                    __CLASS__.'.INVALIDMIME',
                     'File extension does not match known MIME type'
                 );
 
@@ -162,7 +162,7 @@ 
             }
         } catch (MimeUploadValidator_Exception $e) {
             $this->errors[] = _t(
-                __CLASS__ . '.FAILEDMIMECHECK',
+                __CLASS__.'.FAILEDMIMECHECK',
                 'MIME validation failed: {message}',
                 'Argument 1: Message about why MIME type detection failed',
                 ['message' => $e->getMessage()]

tests/MimeUploadValidatorTest.php

Indentation +92 added lines, -92 removed lines

@@ -12,96 +12,96 @@
  */
 class MimeUploadValidatorTest extends SapphireTest
 {
-    public function testInvalidFileExtensionValidatingMimeType()
-    {
-        // setup plaintext file with invalid extension
-        $tmpFileName = 'UploadTest-testUpload.jpg';
-        $tmpFilePath = TEMP_FOLDER . '/' . $tmpFileName;
-        $tmpFileContent = '';
-
-        for ($i = 0; $i < 10000; $i++) {
-            $tmpFileContent .= '0';
-        }
-
-        file_put_contents($tmpFilePath, $tmpFileContent);
-
-        // emulates the $_FILES array
-        $tmpFile = [
-            'name' => $tmpFileName,
-            'size' => filesize($tmpFilePath),
-            'tmp_name' => $tmpFilePath,
-            'extension' => 'jpg',
-            'error' => UPLOAD_ERR_OK,
-        ];
-
-        $upload = Upload::create();
-        $upload->setValidator(MimeUploadValidator::create());
-        $result = $upload->load($tmpFile);
-        $errors = $upload->getErrors();
-
-        $this->assertFalse($result, 'Load failed because file extension does not match excepted MIME type');
-        $this->assertEquals('File is not a valid upload', $errors[0]);
-
-        unlink($tmpFilePath);
-    }
-
-    public function testGetExpectedMimeTypes()
-    {
-        // Setup a file with a capitalised extension and try to match it against a lowercase file.
-        $tmpFileName = 'text.TXT';
-        $tmpFilePath = TEMP_FOLDER . '/' . $tmpFileName;
-        $tmpFileContent = '';
-
-        for ($i = 0; $i < 10000; $i++) {
-            $tmpFileContent .= '0';
-        }
-
-        file_put_contents($tmpFilePath, $tmpFileContent);
-
-        $validator = MimeUploadValidator::create();
-        $tmpFile = [
-            'name' => $tmpFileName,
-            'tmp_name' => $tmpFilePath,
-        ];
-
-        $expected = $validator->getExpectedMimeTypes($tmpFile);
-        $this->assertCount(1, $expected);
-        $this->assertContains('text/plain', $expected);
-
-        unlink($tmpFilePath);
-
-        // Test a physical ico file with capitalised extension
-        $tmpFile = [
-            'name' => 'favicon.ICO',
-            'tmp_name' => 'assets/favicon.ICO',
-        ];
-
-        // Ensure that site configuration doesn't interfere with the test
-        $icoMimes = [
-            'ico' => [
-                'image/vnd.microsoft.icon',
-                'image/x-icon',
-                'image/x-ico'
-            ]
-        ];
-        Config::modify()->set(MimeUploadValidator::class, 'MimeTypes', $icoMimes);
-
-        $expected = $validator->getExpectedMimeTypes($tmpFile);
-        $this->assertCount(3, $expected);
-        $this->assertContains('image/x-icon', $expected);
-    }
-
-    public function testMimeComparison()
-    {
-        $validator = MimeUploadValidator::create();
-
-        $this->assertTrue($validator->compareMime('application/xhtml+xml', 'application/xml'));
-        $this->assertTrue($validator->compareMime('application/vnd.text', 'application/text'));
-        $this->assertTrue($validator->compareMime('application/vnd.vnd.text', 'application/text'));
-        $this->assertTrue($validator->compareMime('application/x-text', 'application/text'));
-        $this->assertTrue($validator->compareMime('application/gzip', 'application/gzip'));
-        $this->assertTrue($validator->compareMime('application/x-gzip', 'application/gzip'));
-        $this->assertFalse($validator->compareMime('application/png', 'application/json'));
-        $this->assertFalse($validator->compareMime('text/plain', 'text/json'));
-    }
+	public function testInvalidFileExtensionValidatingMimeType()
+	{
+		// setup plaintext file with invalid extension
+		$tmpFileName = 'UploadTest-testUpload.jpg';
+		$tmpFilePath = TEMP_FOLDER . '/' . $tmpFileName;
+		$tmpFileContent = '';
+
+		for ($i = 0; $i < 10000; $i++) {
+			$tmpFileContent .= '0';
+		}
+
+		file_put_contents($tmpFilePath, $tmpFileContent);
+
+		// emulates the $_FILES array
+		$tmpFile = [
+			'name' => $tmpFileName,
+			'size' => filesize($tmpFilePath),
+			'tmp_name' => $tmpFilePath,
+			'extension' => 'jpg',
+			'error' => UPLOAD_ERR_OK,
+		];
+
+		$upload = Upload::create();
+		$upload->setValidator(MimeUploadValidator::create());
+		$result = $upload->load($tmpFile);
+		$errors = $upload->getErrors();
+
+		$this->assertFalse($result, 'Load failed because file extension does not match excepted MIME type');
+		$this->assertEquals('File is not a valid upload', $errors[0]);
+
+		unlink($tmpFilePath);
+	}
+
+	public function testGetExpectedMimeTypes()
+	{
+		// Setup a file with a capitalised extension and try to match it against a lowercase file.
+		$tmpFileName = 'text.TXT';
+		$tmpFilePath = TEMP_FOLDER . '/' . $tmpFileName;
+		$tmpFileContent = '';
+
+		for ($i = 0; $i < 10000; $i++) {
+			$tmpFileContent .= '0';
+		}
+
+		file_put_contents($tmpFilePath, $tmpFileContent);
+
+		$validator = MimeUploadValidator::create();
+		$tmpFile = [
+			'name' => $tmpFileName,
+			'tmp_name' => $tmpFilePath,
+		];
+
+		$expected = $validator->getExpectedMimeTypes($tmpFile);
+		$this->assertCount(1, $expected);
+		$this->assertContains('text/plain', $expected);
+
+		unlink($tmpFilePath);
+
+		// Test a physical ico file with capitalised extension
+		$tmpFile = [
+			'name' => 'favicon.ICO',
+			'tmp_name' => 'assets/favicon.ICO',
+		];
+
+		// Ensure that site configuration doesn't interfere with the test
+		$icoMimes = [
+			'ico' => [
+				'image/vnd.microsoft.icon',
+				'image/x-icon',
+				'image/x-ico'
+			]
+		];
+		Config::modify()->set(MimeUploadValidator::class, 'MimeTypes', $icoMimes);
+
+		$expected = $validator->getExpectedMimeTypes($tmpFile);
+		$this->assertCount(3, $expected);
+		$this->assertContains('image/x-icon', $expected);
+	}
+
+	public function testMimeComparison()
+	{
+		$validator = MimeUploadValidator::create();
+
+		$this->assertTrue($validator->compareMime('application/xhtml+xml', 'application/xml'));
+		$this->assertTrue($validator->compareMime('application/vnd.text', 'application/text'));
+		$this->assertTrue($validator->compareMime('application/vnd.vnd.text', 'application/text'));
+		$this->assertTrue($validator->compareMime('application/x-text', 'application/text'));
+		$this->assertTrue($validator->compareMime('application/gzip', 'application/gzip'));
+		$this->assertTrue($validator->compareMime('application/x-gzip', 'application/gzip'));
+		$this->assertFalse($validator->compareMime('application/png', 'application/json'));
+		$this->assertFalse($validator->compareMime('text/plain', 'text/json'));
+	}
 }

Spacing +2 added lines, -2 removed lines

@@ -16,7 +16,7 @@ 
     {
         // setup plaintext file with invalid extension
         $tmpFileName = 'UploadTest-testUpload.jpg';
-        $tmpFilePath = TEMP_FOLDER . '/' . $tmpFileName;
+        $tmpFilePath = TEMP_FOLDER.'/'.$tmpFileName;
         $tmpFileContent = '';
 
         for ($i = 0; $i < 10000; $i++) {
@@ -49,7 +49,7 @@ 
     {
         // Setup a file with a capitalised extension and try to match it against a lowercase file.
         $tmpFileName = 'text.TXT';
-        $tmpFilePath = TEMP_FOLDER . '/' . $tmpFileName;
+        $tmpFilePath = TEMP_FOLDER.'/'.$tmpFileName;
         $tmpFileContent = '';
 
         for ($i = 0; $i < 10000; $i++) {