WordPress.org

Make WordPress Core

Ticket #15729: 15729-4.diff

File 15729-4.diff, 7.3 KB (added by kapeels, 3 years ago)
  • wp-admin/setup-config.php

     
    114114        break; 
    115115 
    116116        case 1: 
    117                 display_header(); 
     117            display_header(); 
     118            if(is_numeric($_POST['error'])) { 
     119                switch($_POST['error']) { 
     120                    case 1 : 
     121                        $error_msg = "Cannot connect to the database server with the provided username,password and host combination."; 
     122                        $focus_element = "uname"; 
     123                    break; 
     124                    case 2 : 
     125                        $error_msg = "Cannot select the database."; 
     126                        $focus_element = "dbname"; 
     127                    break; 
     128                    case 3 : 
     129                        $error_msg = "The table prefix can contain only letters, numbers, and underscores."; 
     130                        $focus_element = "prefix"; 
     131                    break; 
     132                }                 
     133                ?> 
     134                <script type="text/javascript"> 
     135                    function setFocus() { 
     136                        document.getElementById('<?php echo $focus_element; ?>').focus(); 
     137                    } 
     138                </script> 
     139                <p style="color:red;"><?php echo $error_msg; ?></p> 
     140                <?php                 
     141            } 
     142            $dbname =   !empty($_POST['dbname']) ? htmlspecialchars(trim($_POST['dbname']),ENT_QUOTES) : 'wordpress'; 
     143            $uname  =   !empty($_POST['uname']) ? htmlspecialchars(trim($_POST['uname']),ENT_QUOTES) : 'username'; 
     144            // password can be left blank 
     145            $passwrd=   isset($_POST['pwd']) ? htmlspecialchars($_POST['pwd'],ENT_QUOTES) : 'password'; 
     146            $dbhost =   !empty($_POST['dbhost']) ? trim(htmlspecialchars(trim($_POST['dbhost']),ENT_QUOTES)) : 'localhost'; 
     147            $prefix =   !empty($_POST['prefix']) ? trim(htmlspecialchars(trim($_POST['prefix']),ENT_QUOTES)) : 'wp_';             
    118148        ?> 
    119149<form method="post" action="setup-config.php?step=2"> 
    120150        <p>Below you should enter your database connection details. If you're not sure about these, contact your host. </p> 
    121151        <table class="form-table"> 
    122152                <tr> 
    123153                        <th scope="row"><label for="dbname">Database Name</label></th> 
    124                         <td><input name="dbname" id="dbname" type="text" size="25" value="wordpress" /></td> 
     154                        <td><input name="dbname" id="dbname" type="text" size="25" value="<?php echo $dbname; ?>" /></td> 
    125155                        <td>The name of the database you want to run WP in. </td> 
    126156                </tr> 
    127157                <tr> 
    128158                        <th scope="row"><label for="uname">User Name</label></th> 
    129                         <td><input name="uname" id="uname" type="text" size="25" value="username" /></td> 
     159                        <td><input name="uname" id="uname" type="text" size="25" value="<?php echo $uname; ?>" /></td> 
    130160                        <td>Your MySQL username</td> 
    131161                </tr> 
    132162                <tr> 
    133163                        <th scope="row"><label for="pwd">Password</label></th> 
    134                         <td><input name="pwd" id="pwd" type="text" size="25" value="password" /></td> 
     164                        <td><input name="pwd" id="pwd" type="text" size="25" value="<?php echo $passwrd; ?>" /></td> 
    135165                        <td>...and MySQL password.</td> 
    136166                </tr> 
    137167                <tr> 
    138168                        <th scope="row"><label for="dbhost">Database Host</label></th> 
    139                         <td><input name="dbhost" id="dbhost" type="text" size="25" value="localhost" /></td> 
     169                        <td><input name="dbhost" id="dbhost" type="text" size="25" value="<?php echo $dbhost; ?>" /></td> 
    140170                        <td>You should be able to get this info from your web host, if <code>localhost</code> does not work.</td> 
    141171                </tr> 
    142172                <tr> 
    143173                        <th scope="row"><label for="prefix">Table Prefix</label></th> 
    144                         <td><input name="prefix" id="prefix" type="text" id="prefix" value="wp_" size="25" /></td> 
     174                        <td><input name="prefix" id="prefix" type="text" id="prefix" value="<?php echo $prefix; ?>" size="25" /></td> 
    145175                        <td>If you want to run multiple WordPress installations in a single database, change this.</td> 
    146176                </tr> 
    147177        </table> 
     
    159189        $prefix  = trim($_POST['prefix']); 
    160190        if ( empty($prefix) ) 
    161191                $prefix = 'wp_'; 
    162  
     192         
     193        $setup_error    =   null;         
    163194        // Validate $prefix: it can only contain letters, numbers and underscores 
    164         if ( preg_match( '|[^a-z0-9_]|i', $prefix ) ) 
    165                 wp_die( /*WP_I18N_BAD_PREFIX*/'<strong>ERROR</strong>: "Table Prefix" can only contain numbers, letters, and underscores.'/*/WP_I18N_BAD_PREFIX*/ ); 
     195        if ( preg_match( '|[^a-z0-9_]|i', $prefix ) ) {                 
     196                $prefix =   htmlspecialchars($prefix,ENT_QUOTES); 
     197                $setup_error =   array('message' => '<strong>ERROR</strong>: "Table Prefix" can only contain numbers, letters, and underscores.','code' => 3);                                                                 
     198        } 
     199         
     200        // if a prefix related error hasn't happened, test the database credentials 
     201        if( $setup_error==null ) { 
     202            // Test the db connection. 
     203            /**#@+ 
     204             * @ignore 
     205             */ 
     206            define('DB_NAME', $dbname); 
     207            define('DB_USER', $uname); 
     208            define('DB_PASSWORD', $passwrd); 
     209            define('DB_HOST', $dbhost); 
     210            /**#@-*/ 
    166211 
    167         // Test the db connection. 
    168         /**#@+ 
    169          * @ignore 
    170          */ 
    171         define('DB_NAME', $dbname); 
    172         define('DB_USER', $uname); 
    173         define('DB_PASSWORD', $passwrd); 
    174         define('DB_HOST', $dbhost); 
    175         /**#@-*/ 
    176  
    177         // We'll fail here if the values are no good. 
    178         require_wp_db(); 
    179         if ( ! empty( $wpdb->error ) ) { 
    180                 $back = '<p class="step"><a href="setup-config.php?step=1" onclick="javascript:history.go(-1);return false;" class="button">Try Again</a></p>'; 
    181                 wp_die( $wpdb->error->get_error_message() . $back ); 
    182         } 
    183  
     212            // We'll fail here if the values are no good. 
     213            require_wp_db(); 
     214            if ( ! empty( $wpdb->error ) ) {                     
     215                switch(key($wpdb->error->errors)) { 
     216                    case 'db_connect_fail': 
     217                       $setup_error = array('code' => 1); 
     218                    break; 
     219                    case 'db_select_fail': 
     220                       $setup_error = array('code' => 2); 
     221                    break; 
     222                } 
     223                $setup_error['message'] = $wpdb->error->get_error_message(); 
     224            } 
     225        } 
     226        // check if any error occured above         
     227        if( is_array( $setup_error ) ) { 
     228            $try_again = '<p class="step"> 
     229            <form action="setup-config.php?step=1" method="post">   
     230                <input name="error" type="hidden" value="' . $setup_error['code'] . '" /> 
     231                <input name="dbname" type="hidden" value="' .htmlspecialchars($dbname,ENT_QUOTES). '" />                 
     232                <input name="uname" type="hidden" value="' .htmlspecialchars($uname,ENT_QUOTES). '" />           
     233                <input name="pwd" type="hidden" value="' .htmlspecialchars($passwrd,ENT_QUOTES). '" />                   
     234                <input name="dbhost" type="hidden" value="' .htmlspecialchars($dbhost,ENT_QUOTES). '" />                 
     235                <input name="prefix" type="hidden" id="prefix" value="' .$prefix. '" />     
     236                <input type="submit" class="button" value="Try Again" name="try-again"> 
     237            </form> 
     238                  </p>'; 
     239            wp_die($setup_error['message'] . $try_again); 
     240        }        
    184241        // Fetch or generate keys and salts. 
    185242        $no_api = isset( $_POST['noapi'] ); 
    186243        require_once( ABSPATH . WPINC . '/plugin.php' );