includes/MimeMagic.php

   1 <?php
   2 /** Module defining helper functions for detecting and dealing with mime types.
   3  *
   4  */
   5
   6  /** Defines a set of well known mime types
   7  * This is used as a fallback to mime.types files.
   8  * An extensive list of well known mime types is provided by
   9  * the file mime.types in the includes directory.
  10  */
  11 define('MM_WELL_KNOWN_MIME_TYPES',<<<END_STRING
  12 application/ogg ogg ogm
  13 application/pdf pdf
  14 application/x-javascript js
  15 application/x-shockwave-flash swf
  16 audio/midi mid midi kar
  17 audio/mpeg mpga mpa mp2 mp3
  18 audio/x-aiff aif aiff aifc
  19 audio/x-wav wav
  20 audio/ogg ogg
  21 image/x-bmp bmp
  22 image/gif gif
  23 image/jpeg jpeg jpg jpe
  24 image/png png
  25 image/svg+xml image/svg svg
  26 image/tiff tiff tif
  27 image/vnd.djvu image/x.djvu image/x-djvu djvu
  28 image/x-portable-pixmap ppm
  29 image/x-xcf xcf
  30 text/plain txt
  31 text/html html htm
  32 video/ogg ogm ogg
  33 video/mpeg mpg mpeg
  34 END_STRING
  35 );
  36
  37  /** Defines a set of well known mime info entries
  38  * This is used as a fallback to mime.info files.
  39  * An extensive list of well known mime types is provided by
  40  * the file mime.info in the includes directory.
  41  */
  42 define('MM_WELL_KNOWN_MIME_INFO', <<<END_STRING
  43 application/pdf [OFFICE]
  44 text/javascript application/x-javascript [EXECUTABLE]
  45 application/x-shockwave-flash [MULTIMEDIA]
  46 audio/midi [AUDIO]
  47 audio/x-aiff [AUDIO]
  48 audio/x-wav [AUDIO]
  49 audio/mp3 audio/mpeg [AUDIO]
  50 application/ogg audio/ogg video/ogg [MULTIMEDIA]
  51 image/x-bmp image/bmp [BITMAP]
  52 image/gif [BITMAP]
  53 image/jpeg [BITMAP]
  54 image/png [BITMAP]
  55 image/svg+xml [DRAWING]
  56 image/tiff [BITMAP]
  57 image/vnd.djvu [BITMAP]
  58 image/x-xcf [BITMAP]
  59 image/x-portable-pixmap [BITMAP]
  60 text/plain [TEXT]
  61 text/html [TEXT]
  62 video/ogg [VIDEO]
  63 video/mpeg [VIDEO]
  64 unknown/unknown application/octet-stream application/x-empty [UNKNOWN]
  65 END_STRING
  66 );
  67
  68 #note: because this file is possibly included by a function,
  69 #we need to access the global scope explicitely!
  70 global $wgLoadFileinfoExtension;
  71
  72 if ($wgLoadFileinfoExtension) {
  73         if(!extension_loaded('fileinfo')) dl('fileinfo.' . PHP_SHLIB_SUFFIX);
  74 }
  75
  76 /**
  77  * Implements functions related to mime types such as detection and mapping to
  78  * file extension.
  79  *
  80  * Instances of this class are stateles, there only needs to be one global instance
  81  * of MimeMagic. Please use MimeMagic::singleton() to get that instance.
  82  */
  83 class MimeMagic {
  84
  85         /**
  86         * Mapping of media types to arrays of mime types.
  87         * This is used by findMediaType and getMediaType, respectively
  88         */
  89         var $mMediaTypes= NULL;
  90
  91         /** Map of mime type aliases
  92         */
  93         var $mMimeTypeAliases= NULL;
  94
  95         /** map of mime types to file extensions (as a space seprarated list)
  96         */
  97         var $mMimeToExt= NULL;
  98
  99         /** map of file extensions types to mime types (as a space seprarated list)
 100         */
 101         var $mExtToMime= NULL;
 102
 103         /** The singleton instance
 104          */
 105         private static $instance;
 106
 107         /** Initializes the MimeMagic object. This is called by MimeMagic::singleton().
 108         *
 109         * This constructor parses the mime.types and mime.info files and build internal mappings.
 110         */
 111         function __construct() {
 112                 /*
 113                 *   --- load mime.types ---
 114                 */
 115
 116                 global $wgMimeTypeFile, $IP;
 117
 118                 $types = MM_WELL_KNOWN_MIME_TYPES;
 119
 120                 if ( $wgMimeTypeFile == 'includes/mime.types' ) {
 121                         $wgMimeTypeFile = "$IP/$wgMimeTypeFile";
 122                 }
 123
 124                 if ( $wgMimeTypeFile ) {
 125                         if ( is_file( $wgMimeTypeFile ) and is_readable( $wgMimeTypeFile ) ) {
 126                                 wfDebug( __METHOD__.": loading mime types from $wgMimeTypeFile\n" );
 127                                 $types .= "\n";
 128                                 $types .= file_get_contents( $wgMimeTypeFile );
 129                         } else {
 130                                 wfDebug( __METHOD__.": can't load mime types from $wgMimeTypeFile\n" );
 131                         }
 132                 } else {
 133                         wfDebug( __METHOD__.": no mime types file defined, using build-ins only.\n" );
 134                 }
 135
 136                 $types = str_replace( array( "\r\n", "\n\r", "\n\n", "\r\r", "\r" ), "\n", $types );
 137                 $types = str_replace( "\t", " ", $types );
 138
 139                 $this->mMimeToExt = array();
 140                 $this->mToMime = array();
 141
 142                 $lines = explode( "\n",$types );
 143                 foreach ( $lines as $s ) {
 144                         $s = trim( $s );
 145                         if ( empty( $s ) ) continue;
 146                         if ( strpos( $s, '#' ) === 0 ) continue;
 147
 148                         $s = strtolower( $s );
 149                         $i = strpos( $s, ' ' );
 150
 151                         if ( $i === false ) continue;
 152
 153                         #print "processing MIME line $s<br>";
 154
 155                         $mime = substr( $s, 0, $i );
 156                         $ext = trim( substr($s, $i+1 ) );
 157
 158                         if ( empty( $ext ) ) continue;
 159
 160                         if ( !empty( $this->mMimeToExt[$mime] ) ) {
 161                                 $this->mMimeToExt[$mime] .= ' ' . $ext;
 162                         } else {
 163                                 $this->mMimeToExt[$mime] = $ext;
 164                         }
 165
 166                         $extensions = explode( ' ', $ext );
 167
 168                         foreach ( $extensions as $e ) {
 169                                 $e = trim( $e );
 170                                 if ( empty( $e ) ) continue;
 171
 172                                 if ( !empty( $this->mExtToMime[$e] ) ) {
 173                                         $this->mExtToMime[$e] .= ' ' . $mime;
 174                                 } else {
 175                                         $this->mExtToMime[$e] = $mime;
 176                                 }
 177                         }
 178                 }
 179
 180                 /*
 181                 *   --- load mime.info ---
 182                 */
 183
 184                 global $wgMimeInfoFile;
 185                 if ( $wgMimeInfoFile == 'includes/mime.info' ) {
 186                         $wgMimeInfoFile = "$IP/$wgMimeInfoFile";
 187                 }
 188
 189                 $info = MM_WELL_KNOWN_MIME_INFO;
 190
 191                 if ( $wgMimeInfoFile ) {
 192                         if ( is_file( $wgMimeInfoFile ) and is_readable( $wgMimeInfoFile ) ) {
 193                                 wfDebug( __METHOD__.": loading mime info from $wgMimeInfoFile\n" );
 194                                 $info .= "\n";
 195                                 $info .= file_get_contents( $wgMimeInfoFile );
 196                         } else {
 197                                 wfDebug(__METHOD__.": can't load mime info from $wgMimeInfoFile\n");
 198                         }
 199                 } else {
 200                         wfDebug(__METHOD__.": no mime info file defined, using build-ins only.\n");
 201                 }
 202
 203                 $info = str_replace( array( "\r\n", "\n\r", "\n\n", "\r\r", "\r" ), "\n", $info);
 204                 $info = str_replace( "\t", " ", $info );
 205
 206                 $this->mMimeTypeAliases = array();
 207                 $this->mMediaTypes = array();
 208
 209                 $lines = explode( "\n", $info );
 210                 foreach ( $lines as $s ) {
 211                         $s = trim( $s );
 212                         if ( empty( $s ) ) continue;
 213                         if ( strpos( $s, '#' ) === 0 ) continue;
 214
 215                         $s = strtolower( $s );
 216                         $i = strpos( $s, ' ' );
 217
 218                         if ( $i === false ) continue;
 219
 220                         #print "processing MIME INFO line $s<br>";
 221
 222                         $match = array();
 223                         if ( preg_match( '!\[\s*(\w+)\s*\]!', $s, $match ) ) {
 224                                 $s = preg_replace( '!\[\s*(\w+)\s*\]!', '', $s );
 225                                 $mtype = trim( strtoupper( $match[1] ) );
 226                         } else {
 227                                 $mtype = MEDIATYPE_UNKNOWN;
 228                         }
 229
 230                         $m = explode( ' ', $s );
 231
 232                         if ( !isset( $this->mMediaTypes[$mtype] ) ) {
 233                                 $this->mMediaTypes[$mtype] = array();
 234                         }
 235
 236                         foreach ( $m as $mime ) {
 237                                 $mime = trim( $mime );
 238                                 if ( empty( $mime ) ) continue;
 239
 240                                 $this->mMediaTypes[$mtype][] = $mime;
 241                         }
 242
 243                         if ( sizeof( $m ) > 1 ) {
 244                                 $main = $m[0];
 245                                 for ( $i=1; $i<sizeof($m); $i += 1 ) {
 246                                         $mime = $m[$i];
 247                                         $this->mMimeTypeAliases[$mime] = $main;
 248                                 }
 249                         }
 250                 }
 251
 252         }
 253
 254         /**
 255          * Get an instance of this class
 256          */
 257         static function &singleton() {
 258                 if ( !isset( self::$instance ) ) {
 259                         self::$instance = new MimeMagic;
 260                 }
 261                 return self::$instance;
 262         }
 263
 264         /** returns a list of file extensions for a given mime type
 265         * as a space separated string.
 266         */
 267         function getExtensionsForType( $mime ) {
 268                 $mime = strtolower( $mime );
 269
 270                 $r = @$this->mMimeToExt[$mime];
 271
 272                 if ( @!$r and isset( $this->mMimeTypeAliases[$mime] ) ) {
 273                         $mime = $this->mMimeTypeAliases[$mime];
 274                         $r = @$this->mMimeToExt[$mime];
 275                 }
 276
 277                 return $r;
 278         }
 279
 280         /** returns a list of mime types for a given file extension
 281         * as a space separated string.
 282         */
 283         function getTypesForExtension( $ext ) {
 284                 $ext = strtolower( $ext );
 285
 286                 $r = isset( $this->mExtToMime[$ext] ) ? $this->mExtToMime[$ext] : null;
 287                 return $r;
 288         }
 289
 290         /** returns a single mime type for a given file extension.
 291         * This is always the first type from the list returned by getTypesForExtension($ext).
 292         */
 293         function guessTypesForExtension( $ext ) {
 294                 $m = $this->getTypesForExtension( $ext );
 295                 if ( is_null( $m ) ) return NULL;
 296
 297                 $m = trim( $m );
 298                 $m = preg_replace( '/\s.*$/', '', $m );
 299
 300                 return $m;
 301         }
 302
 303
 304         /** tests if the extension matches the given mime type.
 305         * returns true if a match was found, NULL if the mime type is unknown,
 306         * and false if the mime type is known but no matches where found.
 307         */
 308         function isMatchingExtension( $extension, $mime ) {
 309                 $ext = $this->getExtensionsForType( $mime );
 310
 311                 if ( !$ext ) {
 312                         return NULL;  //unknown
 313                 }
 314
 315                 $ext = explode( ' ', $ext );
 316
 317                 $extension = strtolower( $extension );
 318                 if ( in_array( $extension, $ext ) ) {
 319                         return true;
 320                 }
 321
 322                 return false;
 323         }
 324
 325         /** returns true if the mime type is known to represent
 326         * an image format supported by the PHP GD library.
 327         */
 328         function isPHPImageType( $mime ) {
 329                 #as defined by imagegetsize and image_type_to_mime
 330                 static $types = array(
 331                         'image/gif', 'image/jpeg', 'image/png',
 332                         'image/x-bmp', 'image/xbm', 'image/tiff',
 333                         'image/jp2', 'image/jpeg2000', 'image/iff',
 334                         'image/xbm', 'image/x-xbitmap',
 335                         'image/vnd.wap.wbmp', 'image/vnd.xiff',
 336                         'image/x-photoshop',
 337                         'application/x-shockwave-flash',
 338                 );
 339
 340                 return in_array( $mime, $types );
 341         }
 342
 343         /**
 344          * Returns true if the extension represents a type which can
 345          * be reliably detected from its content. Use this to determine
 346          * whether strict content checks should be applied to reject
 347          * invalid uploads; if we can't identify the type we won't
 348          * be able to say if it's invalid.
 349          *
 350          * @todo Be more accurate when using fancy mime detector plugins;
 351          *       right now this is the bare minimum getimagesize() list.
 352          * @return bool
 353          */
 354         function isRecognizableExtension( $extension ) {
 355                 static $types = array(
 356                         // Types recognized by getimagesize()
 357                         'gif', 'jpeg', 'jpg', 'png', 'swf', 'psd',
 358                         'bmp', 'tiff', 'tif', 'jpc', 'jp2',
 359                         'jpx', 'jb2', 'swc', 'iff', 'wbmp',
 360                         'xbm',
 361
 362                         // Formats we recognize magic numbers for
 363                         'djvu', 'ogg', 'mid', 'pdf', 'wmf', 'xcf',
 364
 365                         // XML formats we sure hope we recognize reliably
 366                         'svg',
 367                 );
 368                 return in_array( strtolower( $extension ), $types );
 369         }
 370
 371
 372         /** mime type detection. This uses detectMimeType to detect the mime type of the file,
 373         * but applies additional checks to determine some well known file formats that may be missed
 374         * or misinterpreter by the default mime detection (namely xml based formats like XHTML or SVG).
 375         *
 376         * @param string $file The file to check
 377         * @param mixed $ext The file extension, or true to extract it from the filename.
 378         *                   Set it to false to ignore the extension.
 379         *
 380         * @return string the mime type of $file
 381         */
 382         function guessMimeType( $file, $ext = true ) {
 383                 $mime = $this->doGuessMimeType( $file, $ext );
 384
 385                 if( !$mime ) {
 386                         wfDebug( __METHOD__.": internal type detection failed for $file (.$ext)...\n" );
 387                         $mime = $this->detectMimeType( $file, $ext );
 388                 }
 389
 390                 if ( isset( $this->mMimeTypeAliases[$mime] ) ) {
 391                         $mime = $this->mMimeTypeAliases[$mime];
 392                 }
 393
 394                 wfDebug(__METHOD__.": final mime type of $file: $mime\n");
 395                 return $mime;
 396         }
 397
 398         function doGuessMimeType( $file, $ext = true ) {
 399                 // Read a chunk of the file
 400                 wfSuppressWarnings();
 401                 $f = fopen( $file, "rt" );
 402                 wfRestoreWarnings();
 403                 if( !$f ) return "unknown/unknown";
 404                 $head = fread( $f, 1024 );
 405                 fclose( $f );
 406
 407                 // Hardcode a few magic number checks...
 408                 $headers = array(
 409                         // Multimedia...
 410                         'MThd'             => 'audio/midi',
 411                         'OggS'             => 'application/ogg',
 412
 413                         // Image formats...
 414                         // Note that WMF may have a bare header, no magic number.
 415                         "\x01\x00\x09\x00" => 'application/x-msmetafile', // Possibly prone to false positives?
 416                         "\xd7\xcd\xc6\x9a" => 'application/x-msmetafile',
 417                         '%PDF'             => 'application/pdf',
 418                         'gimp xcf'         => 'image/x-xcf',
 419
 420                         // Some forbidden fruit...
 421                         'MZ'               => 'application/octet-stream', // DOS/Windows executable
 422                         "\xca\xfe\xba\xbe" => 'application/octet-stream', // Mach-O binary
 423                         "\x7fELF"          => 'application/octet-stream', // ELF binary
 424                 );
 425
 426                 foreach( $headers as $magic => $candidate ) {
 427                         if( strncmp( $head, $magic, strlen( $magic ) ) == 0 ) {
 428                                 wfDebug( __METHOD__ . ": magic header in $file recognized as $candidate\n" );
 429                                 return $candidate;
 430                         }
 431                 }
 432
 433                 /*
 434                  * look for PHP
 435                  * Check for this before HTML/XML...
 436                  * Warning: this is a heuristic, and won't match a file with a lot of non-PHP before.
 437                  * It will also match text files which could be PHP. :)
 438                  */
 439                 if( ( strpos( $head, '<?php' ) !== false ) ||
 440                     ( strpos( $head, '<? ' ) !== false ) ||
 441                     ( strpos( $head, "<?\n" ) !== false ) ||
 442                     ( strpos( $head, "<?\t" ) !== false ) ||
 443                     ( strpos( $head, "<?=" ) !== false ) ||
 444
 445                     ( strpos( $head, "<\x00?\x00p\x00h\x00p" ) !== false ) ||
 446                     ( strpos( $head, "<\x00?\x00 " ) !== false ) ||
 447                     ( strpos( $head, "<\x00?\x00\n" ) !== false ) ||
 448                     ( strpos( $head, "<\x00?\x00\t" ) !== false ) ||
 449                     ( strpos( $head, "<\x00?\x00=" ) !== false ) ) {
 450
 451                         wfDebug( __METHOD__ . ": recognized $file as application/x-php\n" );
 452                         return "application/x-php";
 453                 }
 454
 455                 /*
 456                  * look for XML formats (XHTML and SVG)
 457                  */
 458                 $xml = new XmlTypeCheck( $file );
 459                 if( $xml->wellFormed ) {
 460                         $types = array(
 461                                 'http://www.w3.org/2000/svg:svg'    => 'image/svg+xml',
 462                                 'svg'                               => 'image/svg+xml',
 463                                 'http://www.w3.org/1999/xhtml:html' => 'text/html', // application/xhtml+xml?
 464                                 'html'                              => 'text/html', // application/xhtml+xml?
 465                         );
 466                         if( isset( $types[$xml->rootElement] ) ) {
 467                                 $mime = $types[$xml->rootElement];
 468                                 return $mime;
 469                         } else {
 470                                 /// Fixme -- this would be the place to allow additional XML type checks
 471                                 return 'application/xml';
 472                         }
 473                 }
 474
 475                 /*
 476                  * look for shell scripts
 477                  */
 478                 $script_type = NULL;
 479
 480                 # detect by shebang
 481                 if ( substr( $head, 0, 2) == "#!" ) {
 482                         $script_type = "ASCII";
 483                 } elseif ( substr( $head, 0, 5) == "\xef\xbb\xbf#!" ) {
 484                         $script_type = "UTF-8";
 485                 } elseif ( substr( $head, 0, 7) == "\xfe\xff\x00#\x00!" ) {
 486                         $script_type = "UTF-16BE";
 487                 } elseif ( substr( $head, 0, 7 ) == "\xff\xfe#\x00!" ) {
 488                         $script_type= "UTF-16LE";
 489                 }
 490
 491                 if ( $script_type ) {
 492                         if ( $script_type !== "UTF-8" && $script_type !== "ASCII") {
 493                                 // Quick and dirty fold down to ASCII!
 494                                 $pack = array( 'UTF-16BE' => 'n*', 'UTF-16LE' => 'v*' );
 495                                 $chars = unpack( $pack[$script_type], substr( $head, 2 ) );
 496                                 $head = '';
 497                                 foreach( $chars as $codepoint ) {
 498                                         if( $codepoint < 128 ) {
 499                                                 $head .= chr( $codepoint );
 500                                         } else {
 501                                                 $head .= '?';
 502                                         }
 503                                 }
 504                         }
 505
 506                         $match = array();
 507
 508                         if ( preg_match( '%/?([^\s]+/)(\w+)%', $head, $match ) ) {
 509                                 $mime = "application/x-{$match[2]}";
 510                                 wfDebug( __METHOD__.": shell script recognized as $mime\n" );
 511                                 return $mime;
 512                         }
 513                 }
 514
 515                 wfSuppressWarnings();
 516                 $gis = getimagesize( $file );
 517                 wfRestoreWarnings();
 518
 519                 if( $gis && isset( $gis['mime'] ) ) {
 520                         $mime = $gis['mime'];
 521                         wfDebug( __METHOD__.": getimagesize detected $file as $mime\n" );
 522                         return $mime;
 523                 } else {
 524                         return false;
 525                 }
 526
 527                 // Also test DjVu
 528                 $deja = new DjVuImage( $file );
 529                 if( $deja->isValid() ) {
 530                         wfDebug( __METHOD__.": detected $file as image/vnd.djvu\n" );
 531                         return 'image/vnd.djvu';
 532                 }
 533         }
 534
 535         /** Internal mime type detection, please use guessMimeType() for application code instead.
 536         * Detection is done using an external program, if $wgMimeDetectorCommand is set.
 537         * Otherwise, the fileinfo extension and mime_content_type are tried (in this order), if they are available.
 538         * If the dections fails and $ext is not false, the mime type is guessed from the file extension, using
 539         * guessTypesForExtension.
 540         * If the mime type is still unknown, getimagesize is used to detect the mime type if the file is an image.
 541         * If no mime type can be determined, this function returns "unknown/unknown".
 542         *
 543         * @param string $file The file to check
 544         * @param mixed $ext The file extension, or true to extract it from the filename.
 545         *                   Set it to false to ignore the extension.
 546         *
 547         * @return string the mime type of $file
 548         * @access private
 549         */
 550         function detectMimeType( $file, $ext = true ) {
 551                 global $wgMimeDetectorCommand;
 552
 553                 $m = NULL;
 554                 if ( $wgMimeDetectorCommand ) {
 555                         $fn = wfEscapeShellArg( $file );
 556                         $m = `$wgMimeDetectorCommand $fn`;
 557                 } elseif ( function_exists( "finfo_open" ) && function_exists( "finfo_file" ) ) {
 558
 559                         # This required the fileinfo extension by PECL,
 560                         # see http://pecl.php.net/package/fileinfo
 561                         # This must be compiled into PHP
 562                         #
 563                         # finfo is the official replacement for the deprecated
 564                         # mime_content_type function, see below.
 565                         #
 566                         # If you may need to load the fileinfo extension at runtime, set
 567                         # $wgLoadFileinfoExtension in LocalSettings.php
 568
 569                         $mime_magic_resource = finfo_open(FILEINFO_MIME); /* return mime type ala mimetype extension */
 570
 571                         if ($mime_magic_resource) {
 572                                 $m = finfo_file( $mime_magic_resource, $file );
 573                                 finfo_close( $mime_magic_resource );
 574                         } else {
 575                                 wfDebug( __METHOD__.": finfo_open failed on ".FILEINFO_MIME."!\n" );
 576                         }
 577                 } elseif ( function_exists( "mime_content_type" ) ) {
 578
 579                         # NOTE: this function is available since PHP 4.3.0, but only if
 580                         # PHP was compiled with --with-mime-magic or, before 4.3.2, with --enable-mime-magic.
 581                         #
 582                         # On Windows, you must set mime_magic.magicfile in php.ini to point to the mime.magic file bundeled with PHP;
 583                         # sometimes, this may even be needed under linus/unix.
 584                         #
 585                         # Also note that this has been DEPRECATED in favor of the fileinfo extension by PECL, see above.
 586                         # see http://www.php.net/manual/en/ref.mime-magic.php for details.
 587
 588                         $m = mime_content_type($file);
 589                 } else {
 590                         wfDebug( __METHOD__.": no magic mime detector found!\n" );
 591                 }
 592
 593                 if ( $m ) {
 594                         # normalize
 595                         $m = preg_replace( '![;, ].*$!', '', $m ); #strip charset, etc
 596                         $m = trim( $m );
 597                         $m = strtolower( $m );
 598
 599                         if ( strpos( $m, 'unknown' ) !== false ) {
 600                                 $m = NULL;
 601                         } else {
 602                                 wfDebug( __METHOD__.": magic mime type of $file: $m\n" );
 603                                 return $m;
 604                         }
 605                 }
 606
 607                 # if desired, look at extension as a fallback.
 608                 if ( $ext === true ) {
 609                         $i = strrpos( $file, '.' );
 610                         $ext = strtolower( $i ? substr( $file, $i + 1 ) : '' );
 611                 }
 612                 if ( $ext ) {
 613                         if( $this->isRecognizableExtension( $ext ) ) {
 614                                 wfDebug( __METHOD__. ": refusing to guess mime type for .$ext file, we should have recognized it\n" );
 615                         } else {
 616                                 $m = $this->guessTypesForExtension( $ext );
 617                                 if ( $m ) {
 618                                         wfDebug( __METHOD__.": extension mime type of $file: $m\n" );
 619                                         return $m;
 620                                 }
 621                         }
 622                 }
 623
 624                 #unknown type
 625                 wfDebug( __METHOD__.": failed to guess mime type for $file!\n" );
 626                 return "unknown/unknown";
 627         }
 628
 629         /**
 630         * Determine the media type code for a file, using its mime type, name and possibly
 631         * its contents.
 632         *
 633         * This function relies on the findMediaType(), mapping extensions and mime
 634         * types to media types.
 635         *
 636         * @todo analyse file if need be
 637         * @todo look at multiple extension, separately and together.
 638         *
 639         * @param string $path full path to the image file, in case we have to look at the contents
 640         *        (if null, only the mime type is used to determine the media type code).
 641         * @param string $mime mime type. If null it will be guessed using guessMimeType.
 642         *
 643         * @return (int?string?) a value to be used with the MEDIATYPE_xxx constants.
 644         */
 645         function getMediaType( $path = NULL, $mime = NULL ) {
 646                 if( !$mime && !$path ) return MEDIATYPE_UNKNOWN;
 647
 648                 # If mime type is unknown, guess it
 649                 if( !$mime ) $mime = $this->guessMimeType( $path, false );
 650
 651                 # Special code for ogg - detect if it's video (theora),
 652                 # else label it as sound.
 653                 if( $mime == "application/ogg" && file_exists( $path ) ) {
 654
 655                         // Read a chunk of the file
 656                         $f = fopen( $path, "rt" );
 657                         if ( !$f ) return MEDIATYPE_UNKNOWN;
 658                         $head = fread( $f, 256 );
 659                         fclose( $f );
 660
 661                         $head = strtolower( $head );
 662
 663                         # This is an UGLY HACK, file should be parsed correctly
 664                         if ( strpos( $head, 'theora' ) !== false ) return MEDIATYPE_VIDEO;
 665                         elseif ( strpos( $head, 'vorbis' ) !== false ) return MEDIATYPE_AUDIO;
 666                         elseif ( strpos( $head, 'flac' ) !== false ) return MEDIATYPE_AUDIO;
 667                         elseif ( strpos( $head, 'speex' ) !== false ) return MEDIATYPE_AUDIO;
 668                         else return MEDIATYPE_MULTIMEDIA;
 669                 }
 670
 671                 # check for entry for full mime type
 672                 if( $mime ) {
 673                         $type = $this->findMediaType( $mime );
 674                         if( $type !== MEDIATYPE_UNKNOWN ) return $type;
 675                 }
 676
 677                 # Check for entry for file extension
 678                 $e = NULL;
 679                 if ( $path ) {
 680                         $i = strrpos( $path, '.' );
 681                         $e = strtolower( $i ? substr( $path, $i + 1 ) : '' );
 682
 683                         # TODO: look at multi-extension if this fails, parse from full path
 684
 685                         $type = $this->findMediaType( '.' . $e );
 686                         if ( $type !== MEDIATYPE_UNKNOWN ) return $type;
 687                 }
 688
 689                 # Check major mime type
 690                 if( $mime ) {
 691                         $i = strpos( $mime, '/' );
 692                         if( $i !== false ) {
 693                                 $major = substr( $mime, 0, $i );
 694                                 $type = $this->findMediaType( $major );
 695                                 if( $type !== MEDIATYPE_UNKNOWN ) return $type;
 696                         }
 697                 }
 698
 699                 if( !$type ) $type = MEDIATYPE_UNKNOWN;
 700
 701                 return $type;
 702         }
 703
 704         /** returns a media code matching the given mime type or file extension.
 705         * File extensions are represented by a string starting with a dot (.) to
 706         * distinguish them from mime types.
 707         *
 708         * This funktion relies on the mapping defined by $this->mMediaTypes
 709         * @access private
 710         */
 711         function findMediaType( $extMime ) {
 712                 if ( strpos( $extMime, '.' ) === 0 ) { #if it's an extension, look up the mime types
 713                         $m = $this->getTypesForExtension( substr( $extMime, 1 ) );
 714                         if ( !$m ) return MEDIATYPE_UNKNOWN;
 715
 716                         $m = explode( ' ', $m );
 717                 } else {
 718                         # Normalize mime type
 719                         if ( isset( $this->mMimeTypeAliases[$extMime] ) ) {
 720                                 $extMime = $this->mMimeTypeAliases[$extMime];
 721                         }
 722
 723                         $m = array($extMime);
 724                 }
 725
 726                 foreach ( $m as $mime ) {
 727                         foreach ( $this->mMediaTypes as $type => $codes ) {
 728                                 if ( in_array($mime, $codes, true ) ) {
 729                                         return $type;
 730                                 }
 731                         }
 732                 }
 733
 734                 return MEDIATYPE_UNKNOWN;
 735         }
 736 }
 737
 738