Wordpress 3.3.1-scripts

[autoinstalls/wordpress.git] / wp-admin / update.php

diff --git a/wp-admin/update.php b/wp-admin/update.php
index e529064eed2ce5521b95cc9133674cf83624934c..71708d044b0a692ba0bbc69967afdf6ad0fc86e3 100644 (file)
--- a/wp-admin/update.php
+++ b/wp-admin/update.php
@@ -134,11 +134,14 @@ if ( isset($_GET['action']) ) {
 
                $title = sprintf( __('Installing Plugin from uploaded file: %s'), basename( $file_upload->filename ) );
                $nonce = 'plugin-upload';
-               $url = add_query_arg(array('package' => $file_upload->filename ), 'update.php?action=upload-plugin');
+               $url = add_query_arg(array('package' => $file_upload->id), 'update.php?action=upload-plugin');
                $type = 'upload'; //Install plugin type, From Web or an Upload.
 
                $upgrader = new Plugin_Upgrader( new Plugin_Installer_Skin( compact('type', 'title', 'nonce', 'url') ) );
-               $upgrader->install( $file_upload->package );
+               $result = $upgrader->install( $file_upload->package );
+
+               if ( $result || is_wp_error($result) )
+                       $file_upload->cleanup();
 
                include(ABSPATH . 'wp-admin/admin-footer.php');
 
@@ -236,11 +239,14 @@ if ( isset($_GET['action']) ) {
 
                $title = sprintf( __('Installing Theme from uploaded file: %s'), basename( $file_upload->filename ) );
                $nonce = 'theme-upload';
-               $url = add_query_arg(array('package' => $file_upload->filename), 'update.php?action=upload-theme');
+               $url = add_query_arg(array('package' => $file_upload->id), 'update.php?action=upload-theme');
                $type = 'upload'; //Install plugin type, From Web or an Upload.
 
                $upgrader = new Theme_Upgrader( new Theme_Installer_Skin( compact('type', 'title', 'nonce', 'url') ) );
-               $upgrader->install( $file_upload->package );
+               $result = $upgrader->install( $file_upload->package );
+
+               if ( $result || is_wp_error($result) )
+                       $file_upload->cleanup();
 
                include(ABSPATH . 'wp-admin/admin-footer.php');