* @author Joas Schilling * @author Morris Jobke * @author Normal Ra * @author Olivier Paroz * @author Roeland Jago Douma * @author Thomas Müller * @author Victor Dubiniuk * @author Vincent Petry * * @license AGPL-3.0 * * This code is free software: you can redistribute it and/or modify * it under the terms of the GNU Affero General Public License, version 3, * as published by the Free Software Foundation. * * 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 Affero General Public License for more details. * * You should have received a copy of the GNU Affero General Public License, version 3, * along with this program. If not, see * */ namespace OC\Repair; use OCP\Migration\IOutput; use OCP\Migration\IRepairStep; class RepairMimeTypes implements IRepairStep { /** * @var \OCP\IConfig */ protected $config; /** * @var int */ protected $folderMimeTypeId; /** * @param \OCP\IConfig $config */ public function __construct($config) { $this->config = $config; } public function getName() { return 'Repair mime types'; } private static function existsStmt() { return \OC_DB::prepare(' SELECT count(`mimetype`) FROM `*PREFIX*mimetypes` WHERE `mimetype` = ? '); } private static function getIdStmt() { return \OC_DB::prepare(' SELECT `id` FROM `*PREFIX*mimetypes` WHERE `mimetype` = ? '); } private static function insertStmt() { return \OC_DB::prepare(' INSERT INTO `*PREFIX*mimetypes` ( `mimetype` ) VALUES ( ? ) '); } private static function updateWrongStmt() { return \OC_DB::prepare(' UPDATE `*PREFIX*filecache` SET `mimetype` = ( SELECT `id` FROM `*PREFIX*mimetypes` WHERE `mimetype` = ? ) WHERE `mimetype` = ? '); } private static function deleteStmt() { return \OC_DB::prepare(' DELETE FROM `*PREFIX*mimetypes` WHERE `id` = ? '); } private static function updateByNameStmt() { return \OC_DB::prepare(' UPDATE `*PREFIX*filecache` SET `mimetype` = ? WHERE `mimetype` <> ? AND `mimetype` <> ? AND `name` ILIKE ? '); } private function repairMimetypes($wrongMimetypes) { foreach ($wrongMimetypes as $wrong => $correct) { // do we need to remove a wrong mimetype? $result = \OC_DB::executeAudited(self::getIdStmt(), array($wrong)); $wrongId = $result->fetchOne(); if ($wrongId !== false) { // do we need to insert the correct mimetype? $result = \OC_DB::executeAudited(self::existsStmt(), array($correct)); $exists = $result->fetchOne(); if (!is_null($correct)) { if (!$exists) { // insert mimetype \OC_DB::executeAudited(self::insertStmt(), array($correct)); } // change wrong mimetype to correct mimetype in filecache \OC_DB::executeAudited(self::updateWrongStmt(), array($correct, $wrongId)); } // delete wrong mimetype \OC_DB::executeAudited(self::deleteStmt(), array($wrongId)); } } } private function updateMimetypes($updatedMimetypes) { if (empty($this->folderMimeTypeId)) { $result = \OC_DB::executeAudited(self::getIdStmt(), array('httpd/unix-directory')); $this->folderMimeTypeId = (int)$result->fetchOne(); } foreach ($updatedMimetypes as $extension => $mimetype) { $result = \OC_DB::executeAudited(self::existsStmt(), array($mimetype)); $exists = $result->fetchOne(); if (!$exists) { // insert mimetype \OC_DB::executeAudited(self::insertStmt(), array($mimetype)); } // get target mimetype id $result = \OC_DB::executeAudited(self::getIdStmt(), array($mimetype)); $mimetypeId = $result->fetchOne(); // change mimetype for files with x extension \OC_DB::executeAudited(self::updateByNameStmt(), array($mimetypeId, $this->folderMimeTypeId, $mimetypeId, '%.' . $extension)); } } private function fixOfficeMimeTypes() { // update wrong mimetypes $wrongMimetypes = array( 'application/mspowerpoint' => 'application/vnd.ms-powerpoint', 'application/msexcel' => 'application/vnd.ms-excel', ); self::repairMimetypes($wrongMimetypes); $updatedMimetypes = array( 'docx' => 'application/vnd.openxmlformats-officedocument.wordprocessingml.document', 'xlsx' => 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet', 'pptx' => 'application/vnd.openxmlformats-officedocument.presentationml.presentation', ); // separate doc from docx etc self::updateMimetypes($updatedMimetypes); } private function fixApkMimeType() { $updatedMimetypes = array( 'apk' => 'application/vnd.android.package-archive', ); self::updateMimetypes($updatedMimetypes); } private function fixFontsMimeTypes() { // update wrong mimetypes $wrongMimetypes = array( 'font' => null, 'font/opentype' => 'application/font-sfnt', 'application/x-font-ttf' => 'application/font-sfnt', ); self::repairMimetypes($wrongMimetypes); $updatedMimetypes = array( 'ttf' => 'application/font-sfnt', 'otf' => 'application/font-sfnt', 'pfb' => 'application/x-font', ); self::updateMimetypes($updatedMimetypes); } private function fixPostscriptMimeType() { $updatedMimetypes = array( 'eps' => 'application/postscript', 'ps' => 'application/postscript', ); self::updateMimetypes($updatedMimetypes); } private function introduceRawMimeType() { $updatedMimetypes = array( 'arw' => 'image/x-dcraw', 'cr2' => 'image/x-dcraw', 'dcr' => 'image/x-dcraw', 'dng' => 'image/x-dcraw', 'erf' => 'image/x-dcraw', 'iiq' => 'image/x-dcraw', 'k25' => 'image/x-dcraw', 'kdc' => 'image/x-dcraw', 'mef' => 'image/x-dcraw', 'nef' => 'image/x-dcraw', 'orf' => 'image/x-dcraw', 'pef' => 'image/x-dcraw', 'raf' => 'image/x-dcraw', 'rw2' => 'image/x-dcraw', 'srf' => 'image/x-dcraw', 'sr2' => 'image/x-dcraw', 'xrf' => 'image/x-dcraw', ); self::updateMimetypes($updatedMimetypes); } private function introduce3dImagesMimeType() { $updatedMimetypes = array( 'jps' => 'image/jpeg', 'mpo' => 'image/jpeg', ); self::updateMimetypes($updatedMimetypes); } private function introduceConfMimeType() { $updatedMimetypes = array( 'conf' => 'text/plain', 'cnf' => 'text/plain', ); self::updateMimetypes($updatedMimetypes); } private function introduceYamlMimeType() { $updatedMimetypes = array( 'yaml' => 'application/yaml', 'yml' => 'application/yaml', ); self::updateMimetypes($updatedMimetypes); } private function introduceJavaMimeType() { $updatedMimetypes = array( 'class' => 'application/java', 'java' => 'text/x-java-source', ); self::updateMimetypes($updatedMimetypes); } private function introduceHppMimeType() { $updatedMimetypes = array( 'hpp' => 'text/x-h', ); self::updateMimetypes($updatedMimetypes); } private function introduceRssMimeType() { $updatedMimetypes = array( 'rss' => 'application/rss+xml', ); self::updateMimetypes($updatedMimetypes); } private function introduceRtfMimeType() { $updatedMimetypes = array( 'rtf' => 'text/rtf', ); self::updateMimetypes($updatedMimetypes); } private function introduceRichDocumentsMimeTypes() { $updatedMimetypes = array( 'lwp' => 'application/vnd.lotus-wordpro', 'one' => 'application/msonenote', 'vsd' => 'application/vnd.visio', 'wpd' => 'application/vnd.wordperfect', ); self::updateMimetypes($updatedMimetypes); } /** * Fix mime types */ public function run(IOutput $out) { $ocVersionFromBeforeUpdate = $this->config->getSystemValue('version', '0.0.0'); // NOTE TO DEVELOPERS: when adding new mime types, please make sure to // add a version comparison to avoid doing it every time // only update mime types if necessary as it can be expensive if (version_compare($ocVersionFromBeforeUpdate, '8.2.0', '<')) { if ($this->fixOfficeMimeTypes()) { $out->info('Fixed office mime types'); } if ($this->fixApkMimeType()) { $out->info('Fixed APK mime type'); } if ($this->fixFontsMimeTypes()) { $out->info('Fixed fonts mime types'); } if ($this->fixPostscriptMimeType()) { $out->info('Fixed Postscript mime types'); } if ($this->introduceRawMimeType()) { $out->info('Fixed Raw mime types'); } if ($this->introduce3dImagesMimeType()) { $out->info('Fixed 3D images mime types'); } if ($this->introduceConfMimeType()) { $out->info('Fixed Conf/cnf mime types'); } if ($this->introduceYamlMimeType()) { $out->info('Fixed Yaml/Yml mime types'); } } // Mimetype updates from #19272 if (version_compare($ocVersionFromBeforeUpdate, '8.2.0.8', '<')) { if ($this->introduceJavaMimeType()) { $out->info('Fixed java/class mime types'); } if ($this->introduceHppMimeType()) { $out->info('Fixed hpp mime type'); } if ($this->introduceRssMimeType()) { $out->info('Fixed rss mime type'); } if ($this->introduceRtfMimeType()) { $out->info('Fixed rtf mime type'); } } if (version_compare($ocVersionFromBeforeUpdate, '9.0.0.10', '<')) { if ($this->introduceRichDocumentsMimeTypes()) { $out->info('Fixed richdocuments additional office mime types'); } } } }