Changeset 3894 for trunk/wp-includes/functions.php

Timestamp:

06/21/2006 11:17:19 PM (20 years ago)

Author:

ryan

Message:

wp_check_filetype() from skeltoac.

File:

• trunk/wp-includes/functions.php (modified) (2 diffs)

trunk/wp-includes/functions.php

@@ -964 +964 @@
 function wp_upload_bits($name, $type, $bits) {
         if ( empty($name) )
-                return array('error' => "Empty filename");
+                return array('error' => __("Empty filename"));
+
+        $wp_filetype = wp_check_filetype($name);
+        if ( !$wp_filetype['ext'] )
+                return array('error' => __("Invalid file type"));
 
         $upload = wp_upload_dir();
@@ -1010 +1014 @@
 }
 
+function wp_check_filetype($filename, $mimes = null) {
+        // Accepted MIME types are set here as PCRE unless provided.
+        $mimes = is_array($mimes) ? $mimes : apply_filters('upload_mimes', array (
+                'jpg|jpeg|jpe' => 'image/jpeg',
+                'gif' => 'image/gif',
+                'png' => 'image/png',
+                'bmp' => 'image/bmp',
+                'tif|tiff' => 'image/tiff',
+                'ico' => 'image/x-icon',
+                'asf|asx|wax|wmv|wmx' => 'video/asf',
+                'avi' => 'video/avi',
+                'mov|qt' => 'video/quicktime',
+                'mpeg|mpg|mpe' => 'video/mpeg',
+                'txt|c|cc|h' => 'text/plain',
+                'rtx' => 'text/richtext',
+                'css' => 'text/css',
+                'htm|html' => 'text/html',
+                'mp3|mp4' => 'audio/mpeg',
+                'ra|ram' => 'audio/x-realaudio',
+                'wav' => 'audio/wav',
+                'ogg' => 'audio/ogg',
+                'mid|midi' => 'audio/midi',
+                'wma' => 'audio/wma',
+                'rtf' => 'application/rtf',
+                'js' => 'application/javascript',
+                'pdf' => 'application/pdf',
+                'doc' => 'application/msword',
+                'pot|pps|ppt' => 'application/vnd.ms-powerpoint',
+                'wri' => 'application/vnd.ms-write',
+                'xla|xls|xlt|xlw' => 'application/vnd.ms-excel',
+                'mdb' => 'application/vnd.ms-access',
+                'mpp' => 'application/vnd.ms-project',
+                'swf' => 'application/x-shockwave-flash',
+                'class' => 'application/java',
+                'tar' => 'application/x-tar',
+                'zip' => 'application/zip',
+                'gz|gzip' => 'application/x-gzip',
+                'exe' => 'application/x-msdownload'
+        ));
+
+        $type = false;
+        $ext = false;
+
+        foreach ($mimes as $ext_preg => $mime_match) {
+                $ext_preg = '!\.(' . $ext_preg . ')$!i';
+                if ( preg_match($ext_preg, $filename, $ext_matches) ) {
+                        $type = $mime_match;
+                        $ext = $ext_matches[1];
+                        break;
+                }
+        }
+
+        return compact('ext', 'type');
+}
+
 function do_trackbacks($post_id) {
         global $wpdb;