WordPress.org

Make WordPress Core

Ticket #5367: secure_cookie.3.diff

File secure_cookie.3.diff, 11.6 KB (added by ryan, 10 years ago)

Allow POST requests a grace period

  • wp-login.php

     
    288288default: 
    289289        $user_login = ''; 
    290290        $user_pass = ''; 
    291         $using_cookie = FALSE; 
    292291 
    293292        if ( !isset( $_REQUEST['redirect_to'] ) || is_user_logged_in() ) 
    294293                $redirect_to = 'wp-admin/'; 
    … …  
    296295                $redirect_to = $_REQUEST['redirect_to']; 
    297296 
    298297        if ( $http_post ) { 
     298                // If cookies are disabled we can't log in even with a valid user+pass 
     299                if ( empty($_COOKIE[TEST_COOKIE]) ) 
     300                        $errors['test_cookie'] = __('<strong>ERROR</strong>: WordPress requires Cookies but your browser does not support them or they are blocked.'); 
     301                 
    299302                $user_login = $_POST['log']; 
    300303                $user_login = sanitize_user( $user_login ); 
    301304                $user_pass  = $_POST['pwd']; 
    302305                $rememberme = $_POST['rememberme']; 
     306 
     307                do_action_ref_array('wp_authenticate', array(&$user_login, &$user_pass)); 
    303308        } else { 
    304                 $cookie_login = wp_get_cookie_login(); 
    305                 if ( ! empty($cookie_login) ) { 
    306                         $using_cookie = true; 
    307                         $user_login = $cookie_login['login']; 
    308                         $user_pass = $cookie_login['password']; 
     309                $user = wp_validate_auth_cookie(); 
     310                if ( !$user ) { 
     311                        $errors['expiredsession'] = __('Your session has expired.'); 
     312                } else { 
     313                        $user = new WP_User($user); 
     314 
     315                        // If the user can't edit posts, send them to their profile. 
     316                        if ( !$user->has_cap('edit_posts') && ( empty( $redirect_to ) || $redirect_to == 'wp-admin/' ) ) 
     317                                $redirect_to = get_option('siteurl') . '/wp-admin/profile.php'; 
     318                        wp_safe_redirect($redirect_to); 
     319                        exit(); 
    309320                } 
    310321        } 
    311322 
    312         do_action_ref_array('wp_authenticate', array(&$user_login, &$user_pass)); 
    313  
    314         // If cookies are disabled we can't log in even with a valid user+pass 
    315         if ( $http_post && empty($_COOKIE[TEST_COOKIE]) ) 
    316                 $errors['test_cookie'] = __('<strong>ERROR</strong>: WordPress requires Cookies but your browser does not support them or they are blocked.'); 
    317  
    318323        if ( $user_login && $user_pass && empty( $errors ) ) { 
    319324                $user = new WP_User(0, $user_login); 
    320325 
    … …  
    322327                if ( !$user->has_cap('edit_posts') && ( empty( $redirect_to ) || $redirect_to == 'wp-admin/' ) ) 
    323328                        $redirect_to = get_option('siteurl') . '/wp-admin/profile.php'; 
    324329 
    325                 if ( wp_login($user_login, $user_pass, $using_cookie) ) { 
    326                         if ( !$using_cookie ) 
    327                                 wp_setcookie($user_login, $user_pass, false, '', '', $rememberme); 
     330                if ( wp_login($user_login, $user_pass) ) { 
     331                        wp_setcookie($user_login, $user_pass, false, '', '', $rememberme); 
    328332                        do_action('wp_login', $user_login); 
    329333                        wp_safe_redirect($redirect_to); 
    330334                        exit(); 
    331                 } else { 
    332                         if ( $using_cookie ) 
    333                                 $errors['expiredsession'] = __('Your session has expired.'); 
    334335                } 
    335336        } 
    336337 

  • wp-includes/compat.php

     
    147147        } 
    148148} 
    149149 
     150if ( ! function_exists('hash_hmac') ): 
     151function hash_hmac($algo, $data, $key, $raw_output = false) { 
     152        $packs = array('md5' => 'H32', 'sha1' => 'H40'); 
     153 
     154        if ( !isset($packs[$algo]) ) 
     155                return false; 
     156 
     157        $pack = $packs[$algo]; 
     158 
     159        if (strlen($key) > 64) 
     160                $key = pack($pack, $algo($key)); 
     161        else if (strlen($key) < 64) 
     162                $key = str_pad($key, 64, chr(0)); 
     163         
     164        $ipad = (substr($key, 0, 64) ^ str_repeat(chr(0x36), 64)); 
     165        $opad = (substr($key, 0, 64) ^ str_repeat(chr(0x5C), 64)); 
     166 
     167        return $algo($opad . pack($pack, $algo($ipad . $data))); 
     168} 
     169endif; 
     170 
    150171// Added in PHP 4.3.0? 
    151172if (!defined('IMAGETYPE_GIF')) 
    152173    define('IMAGETYPE_GIF', 1); 

  • wp-includes/pluggable.php

     
    4646        if ( ! empty($current_user) ) 
    4747                return; 
    4848 
    49         if ( empty($_COOKIE[USER_COOKIE]) || empty($_COOKIE[PASS_COOKIE]) || 
    50                 !wp_login($_COOKIE[USER_COOKIE], $_COOKIE[PASS_COOKIE], true) ) { 
     49        if ( ! $user = wp_validate_auth_cookie() ) { 
    5150                wp_set_current_user(0); 
    5251                return false; 
    5352        } 
    5453 
    55         $user_login = $_COOKIE[USER_COOKIE]; 
    56         wp_set_current_user(0, $user_login); 
     54        wp_set_current_user($user); 
    5755} 
    5856endif; 
    5957 
    … …  
    293291endif; 
    294292 
    295293if ( !function_exists('wp_login') ) : 
    296 function wp_login($username, $password, $already_md5 = false) { 
     294function wp_login($username, $password, $deprecated = false) { 
    297295        global $wpdb, $error; 
    298296 
    299297        $username = sanitize_user($username); 
    … …  
    313311                return false; 
    314312        } 
    315313 
    316         // If the password is already_md5, it has been double hashed. 
    317         // Otherwise, it is plain text. 
    318         if ( !$already_md5 ) { 
    319                 if ( wp_check_password($password, $login->user_pass) ) { 
    320                         // If using old md5 password, rehash. 
    321                         if ( strlen($login->user_pass) <= 32 ) { 
    322                                 $hash = wp_hash_password($password); 
    323                                 $wpdb->query("UPDATE $wpdb->users SET user_pass = '$hash', user_activation_key = '' WHERE ID = '$login->ID'"); 
    324                                 wp_cache_delete($login->ID, 'users'); 
    325                         } 
     314        if ( !wp_check_password($password, $login->user_pass) ) { 
     315                $error = __('<strong>ERROR</strong>: Incorrect password.'); 
     316                return false; 
     317        } 
    326318 
    327                         return true; 
    328                 } 
     319        // If using old md5 password, rehash. 
     320        if ( strlen($login->user_pass) <= 32 ) { 
     321                $hash = wp_hash_password($password); 
     322                $wpdb->query("UPDATE $wpdb->users SET user_pass = '$hash', user_activation_key = '' WHERE ID = '$login->ID'"); 
     323                wp_cache_delete($login->ID, 'users'); 
     324        } 
     325 
     326        return true; 
     327} 
     328endif; 
     329 
     330if ( !function_exists('wp_validate_auth_cookie') ) : 
     331function wp_validate_auth_cookie($cookie = '') { 
     332        if ( empty($cookie) ) { 
     333                if ( empty($_COOKIE[AUTH_COOKIE]) ) 
     334                        return false; 
     335                $cookie = $_COOKIE[AUTH_COOKIE]; 
     336        } 
     337 
     338        list($username, $expiration, $hmac) = explode('|', $cookie); 
     339 
     340        // Allow a grace period for POST requests 
     341        if ( 'POST' == $_SERVER['REQUEST_METHOD'] ) 
     342                $expiration += 3600; 
     343 
     344        if ( $expiration < time() ) 
     345                return false; 
     346 
     347        $key = wp_hash($username . $expiration); 
     348        $hash = hash_hmac('md5', $username . $expiration, $key); 
     349         
     350        if ( $hmac != $hash ) 
     351                return false; 
     352 
     353        $user = get_userdatabylogin($username); 
     354        if ( ! $user ) 
     355                return false; 
     356 
     357        return $user->ID; 
     358} 
     359endif; 
     360 
     361if ( !function_exists('wp_set_auth_cookie') ) : 
     362function wp_set_auth_cookie($user_id, $remember = false) { 
     363        $user = get_userdata($user_id); 
     364 
     365        if ( $remember ) { 
     366                $expiration = $expire = time() + 1209600; 
    329367        } else { 
    330                 if ( md5($login->user_pass) == $password ) 
    331                         return true; 
     368                $expiration = time() + 172800; 
     369                $expire = 0; 
    332370        } 
    333371 
    334         $error = __('<strong>ERROR</strong>: Incorrect password.'); 
    335         return false; 
     372        $key = wp_hash($user->user_login . $expiration); 
     373        $hash = hash_hmac('md5', $user->user_login . $expiration, $key); 
     374 
     375        $cookie = $user->user_login . '|' . $expiration . '|' . $hash; 
     376 
     377        setcookie(AUTH_COOKIE, $cookie, $expire, COOKIEPATH, COOKIE_DOMAIN); 
     378        if ( COOKIEPATH != SITECOOKIEPATH ) 
     379                setcookie(AUTH_COOKIE, $cookie, $expire, SITECOOKIEPATH, COOKIE_DOMAIN); 
    336380} 
    337381endif; 
    338382 
    … …  
    350394if ( !function_exists('auth_redirect') ) : 
    351395function auth_redirect() { 
    352396        // Checks if a user is logged in, if not redirects them to the login page 
    353         if ( (!empty($_COOKIE[USER_COOKIE]) && 
    354                                 !wp_login($_COOKIE[USER_COOKIE], $_COOKIE[PASS_COOKIE], true)) || 
    355                          (empty($_COOKIE[USER_COOKIE])) ) { 
     397        if ( (!empty($_COOKIE[AUTH_COOKIE]) && 
     398                                !wp_validate_auth_cookie($_COOKIE[AUTH_COOKIE])) || 
     399                         (empty($_COOKIE[AUTH_COOKIE])) ) { 
    356400                nocache_headers(); 
    357401 
    358402                wp_redirect(get_option('siteurl') . '/wp-login.php?redirect_to=' . urlencode($_SERVER['REQUEST_URI'])); 
    … …  
    379423        if ( !wp_verify_nonce( $nonce, $action ) ) { 
    380424                $current_name = ''; 
    381425                if ( ( $current = wp_get_current_user() ) && $current->ID ) 
    382                         $current_name = $current->data->user_login; 
     426                        $current_name = $current->user_login; 
    383427                if ( !$current_name ) 
    384428                        die('-1'); 
    385429 
     430                $auth_cookie = ''; 
    386431                $cookie = explode('; ', urldecode(empty($_POST['cookie']) ? $_GET['cookie'] : $_POST['cookie'])); // AJAX scripts must pass cookie=document.cookie 
    387432                foreach ( $cookie as $tasty ) { 
    388                         if ( false !== strpos($tasty, USER_COOKIE) ) 
    389                                 $user = substr(strstr($tasty, '='), 1); 
    390                         if ( false !== strpos($tasty, PASS_COOKIE) ) 
    391                                 $pass = substr(strstr($tasty, '='), 1); 
     433                        if ( false !== strpos($tasty, AUTH_COOKIE) ) 
     434                                $auth_cookie = substr(strstr($tasty, '='), 1); 
    392435                } 
    393436 
    394                 if ( $current_name != $user || !wp_login( $user, $pass, true ) ) 
     437                if ( $current_name != $user || empty($auth_cookie) || !wp_validate_auth_cookie( $auth_cookie ) ) 
    395438                        die('-1'); 
    396439        } 
    397440        do_action('check_ajax_referer'); 
    … …  
    483526endif; 
    484527 
    485528if ( !function_exists('wp_setcookie') ) : 
    486 function wp_setcookie($username, $password, $already_md5 = false, $home = '', $siteurl = '', $remember = false) { 
     529function wp_setcookie($username, $password = '', $already_md5 = false, $home = '', $siteurl = '', $remember = false) { 
    487530        $user = get_userdatabylogin($username); 
    488         if ( !$already_md5) { 
    489                 $password = md5($user->user_pass); // Double hash the password in the cookie. 
    490         } 
    491  
    492         if ( empty($home) ) 
    493                 $cookiepath = COOKIEPATH; 
    494         else 
    495                 $cookiepath = preg_replace('|https?://[^/]+|i', '', $home . '/' ); 
    496  
    497         if ( empty($siteurl) ) { 
    498                 $sitecookiepath = SITECOOKIEPATH; 
    499                 $cookiehash = COOKIEHASH; 
    500         } else { 
    501                 $sitecookiepath = preg_replace('|https?://[^/]+|i', '', $siteurl . '/' ); 
    502                 $cookiehash = md5($siteurl); 
    503         } 
    504  
    505         if ( $remember ) 
    506                 $expire = time() + 31536000; 
    507         else 
    508                 $expire = 0; 
    509  
    510         setcookie(USER_COOKIE, $username, $expire, $cookiepath, COOKIE_DOMAIN); 
    511         setcookie(PASS_COOKIE, $password, $expire, $cookiepath, COOKIE_DOMAIN); 
    512  
    513         if ( $cookiepath != $sitecookiepath ) { 
    514                 setcookie(USER_COOKIE, $username, $expire, $sitecookiepath, COOKIE_DOMAIN); 
    515                 setcookie(PASS_COOKIE, $password, $expire, $sitecookiepath, COOKIE_DOMAIN); 
    516         } 
     531        wp_set_auth_cookie($user->ID, $remember); 
    517532} 
    518533endif; 
    519534 
    520535if ( !function_exists('wp_clearcookie') ) : 
    521536function wp_clearcookie() { 
     537        setcookie(AUTH_COOKIE, ' ', time() - 31536000, COOKIEPATH, COOKIE_DOMAIN); 
     538        setcookie(AUTH_COOKIE, ' ', time() - 31536000, SITECOOKIEPATH, COOKIE_DOMAIN); 
     539 
     540        // Old cookies 
    522541        setcookie(USER_COOKIE, ' ', time() - 31536000, COOKIEPATH, COOKIE_DOMAIN); 
    523542        setcookie(PASS_COOKIE, ' ', time() - 31536000, COOKIEPATH, COOKIE_DOMAIN); 
    524543        setcookie(USER_COOKIE, ' ', time() - 31536000, SITECOOKIEPATH, COOKIE_DOMAIN); 
    … …  
    695714        if ( empty($salt) ) 
    696715                $salt = DB_PASSWORD . DB_USER . DB_NAME . DB_HOST . ABSPATH; 
    697716 
    698         return $salt; 
     717        if ( ! defined('SECRET_KEY') ) 
     718                $secret_key = 'shhhh'; 
     719        else 
     720                $secret_key = SECRET_KEY; 
     721                 
     722        return $salt . $secret_key; 
    699723} 
    700724endif; 
    701725 

  • wp-config-sample.php

     
    66define('DB_HOST', 'localhost');    // 99% chance you won't need to change this value 
    77define('DB_CHARSET', 'utf8'); 
    88define('DB_COLLATE', ''); 
     9define('SECRET_KEY', 'shhhh'); // Change this to something unique 
    910 
    1011// You can have multiple installations in one database if you give each a unique prefix 
    1112$table_prefix  = 'wp_';   // Only numbers, letters, and underscores please! 

  • wp-settings.php

     
    186186} 
    187187 
    188188if ( !defined('USER_COOKIE') ) 
    189         define('USER_COOKIE', 'wordpressuser_'. COOKIEHASH); 
     189        define('USER_COOKIE', 'wordpressuser_' . COOKIEHASH); 
    190190if ( !defined('PASS_COOKIE') ) 
    191         define('PASS_COOKIE', 'wordpresspass_'. COOKIEHASH); 
     191        define('PASS_COOKIE', 'wordpresspass_' . COOKIEHASH); 
     192if ( !defined('AUTH_COOKIE') ) 
     193        define('AUTH_COOKIE', 'wordpress_' . COOKIEHASH); 
    192194if ( !defined('TEST_COOKIE') ) 
    193195        define('TEST_COOKIE', 'wordpress_test_cookie'); 
    194196if ( !defined('COOKIEPATH') )