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

Timestamp:

06/21/2006 11:17:19 PM (18 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;