invert without namespace doesn't do anything, so ignore it
[lhc/web/wiklou.git] / includes / specials / SpecialUserlogin.php
index 4337f98..5e5f6a6 100644 (file)
@@ -1,11 +1,28 @@
 <?php
 /**
+ * Implements Special:UserLogin
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
+ * http://www.gnu.org/copyleft/gpl.html
+ *
  * @file
  * @ingroup SpecialPage
  */
 
 /**
- * constructor
+ * Constructor
  */
 function wfSpecialUserlogin( $par = '' ) {
        global $wgRequest;
@@ -18,7 +35,8 @@ function wfSpecialUserlogin( $par = '' ) {
 }
 
 /**
- * implements Special:Login
+ * Implements Special:UserLogin
+ *
  * @ingroup SpecialPage
  */
 class LoginForm {
@@ -34,19 +52,23 @@ class LoginForm {
        const ABORTED = 8;
        const CREATE_BLOCKED = 9;
        const THROTTLED = 10;
+       const USER_BLOCKED = 11;
+       const NEED_TOKEN = 12;
+       const WRONG_TOKEN = 13;
 
        var $mName, $mPassword, $mRetype, $mReturnTo, $mCookieCheck, $mPosted;
        var $mAction, $mCreateaccount, $mCreateaccountMail, $mMailmypassword;
        var $mLoginattempt, $mRemember, $mEmail, $mDomain, $mLanguage;
-       var $mSkipCookieCheck, $mReturnToQuery;
+       var $mSkipCookieCheck, $mReturnToQuery, $mToken;
 
        private $mExtUser = null;
 
        /**
         * Constructor
-        * @param WebRequest $request A WebRequest object passed by reference
+        * @param $request WebRequest: a WebRequest object passed by reference
+        * @param $par String: subpage parameter
         */
-       function LoginForm( &$request, $par = '' ) {
+       function __construct( &$request, $par = '' ) {
                global $wgAuth, $wgHiddenPrefs, $wgEnableEmail, $wgRedirectOnLogin;
 
                $this->mType = ( $par == 'signup' ) ? $par : $request->getText( 'type' ); # Check for [[Special:Userlogin/signup]]
@@ -54,6 +76,7 @@ class LoginForm {
                $this->mPassword = $request->getText( 'wpPassword' );
                $this->mRetype = $request->getText( 'wpRetype' );
                $this->mDomain = $request->getText( 'wpDomain' );
+               $this->mReason = $request->getText( 'wpReason' );
                $this->mReturnTo = $request->getVal( 'returnto' );
                $this->mReturnToQuery = $request->getVal( 'returntoquery' );
                $this->mCookieCheck = $request->getVal( 'wpCookieCheck' );
@@ -68,6 +91,7 @@ class LoginForm {
                $this->mRemember = $request->getCheck( 'wpRemember' );
                $this->mLanguage = $request->getText( 'uselang' );
                $this->mSkipCookieCheck = $request->getCheck( 'wpSkipCookieCheck' );
+               $this->mToken = ( $this->mType == 'signup' ) ? $request->getVal( 'wpCreateaccountToken' ) : $request->getVal( 'wpLoginToken' );
 
                if ( $wgRedirectOnLogin ) {
                        $this->mReturnTo = $wgRedirectOnLogin;
@@ -102,14 +126,14 @@ class LoginForm {
                if ( !is_null( $this->mCookieCheck ) ) {
                        $this->onCookieRedirectCheck( $this->mCookieCheck );
                        return;
-               } else if( $this->mPosted ) {
+               } elseif( $this->mPosted ) {
                        if( $this->mCreateaccount ) {
                                return $this->addNewAccount();
-                       } else if ( $this->mCreateaccountMail ) {
+                       } elseif ( $this->mCreateaccountMail ) {
                                return $this->addNewAccountMailPassword();
-                       } else if ( $this->mMailmypassword ) {
+                       } elseif ( $this->mMailmypassword ) {
                                return $this->mailPassword();
-                       } else if ( ( 'submitlogin' == $this->mAction ) || $this->mLoginattempt ) {
+                       } elseif ( ( 'submitlogin' == $this->mAction ) || $this->mLoginattempt ) {
                                return $this->processLogin();
                        }
                }
@@ -122,14 +146,14 @@ class LoginForm {
        function addNewAccountMailPassword() {
                global $wgOut;
 
-               if ('' == $this->mEmail) {
-                       $this->mainLoginForm( wfMsg( 'noemail', htmlspecialchars( $this->mName ) ) );
+               if ( $this->mEmail == '' ) {
+                       $this->mainLoginForm( wfMsgExt( 'noemail', array( 'parsemag', 'escape' ), $this->mName ) );
                        return;
                }
 
                $u = $this->addNewaccountInternal();
 
-               if ($u == null) {
+               if ( $u == null ) {
                        return;
                }
 
@@ -139,7 +163,7 @@ class LoginForm {
                $result = $this->mailPasswordInternal( $u, false, 'createaccount-title', 'createaccount-text' );
 
                wfRunHooks( 'AddNewAccount', array( $u, true ) );
-               $u->addNewUserLogEntry();
+               $u->addNewUserLogEntry( true, $this->mReason );
 
                $wgOut->setPageTitle( wfMsg( 'accmailtitle' ) );
                $wgOut->setRobotPolicy( 'noindex,nofollow' );
@@ -154,27 +178,27 @@ class LoginForm {
                $u = 0;
        }
 
-
        /**
         * @private
         */
        function addNewAccount() {
-               global $wgUser, $wgEmailAuthentication;
+               global $wgUser, $wgEmailAuthentication, $wgOut;
 
                # Create the account and abort if there's a problem doing so
                $u = $this->addNewAccountInternal();
-               if( $u == null )
+               if( $u == null ) {
                        return;
+               }
 
                # If we showed up language selection links, and one was in use, be
                # smart (and sensible) and save that language as the user's preference
                global $wgLoginLanguageSelector;
-               if( $wgLoginLanguageSelector && $this->mLanguage )
+               if( $wgLoginLanguageSelector && $this->mLanguage ) {
                        $u->setOption( 'language', $this->mLanguage );
+               }
 
                # Send out an email authentication message if needed
                if( $wgEmailAuthentication && User::isValidEmailAddr( $u->getEmail() ) ) {
-                       global $wgOut;
                        $error = $u->sendConfirmationMail();
                        if( WikiError::isError( $error ) ) {
                                $wgOut->addWikiMsg( 'confirmemail_sendfailed', $error->getMessage() );
@@ -201,7 +225,6 @@ class LoginForm {
                        }
                } else {
                        # Confirm that the account was created
-                       global $wgOut;
                        $self = SpecialPage::getTitleFor( 'Userlogin' );
                        $wgOut->setPageTitle( wfMsgHtml( 'accountcreated' ) );
                        $wgOut->setArticleRelated( false );
@@ -209,7 +232,7 @@ class LoginForm {
                        $wgOut->addHTML( wfMsgWikiHtml( 'accountcreatedtext', $u->getName() ) );
                        $wgOut->returnToMain( false, $self );
                        wfRunHooks( 'AddNewAccount', array( $u, false ) );
-                       $u->addNewUserLogEntry();
+                       $u->addNewUserLogEntry( false, $this->mReason );
                        return true;
                }
        }
@@ -234,7 +257,7 @@ class LoginForm {
                // cation server before they create an account (otherwise, they can
                // create a local account and login as any domain user). We only need
                // to check this for domains that aren't local.
-               if( 'local' != $this->mDomain && '' != $this->mDomain ) {
+               if( 'local' != $this->mDomain && $this->mDomain != '' ) {
                        if( !$wgAuth->canCreateAccounts() && ( !$wgAuth->userExists( $this->mName ) || !$wgAuth->authenticate( $this->mName, $this->mPassword ) ) ) {
                                $this->mainLoginForm( wfMsg( 'wrongpassword' ) );
                                return false;
@@ -246,6 +269,25 @@ class LoginForm {
                        return false;
                }
 
+               # Request forgery checks.
+               if ( !self::getCreateaccountToken() ) {
+                       self::setCreateaccountToken();
+                       $this->mainLoginForm( wfMsgExt( 'nocookiesnew', array( 'parseinline' ) ) );;
+                       return false;
+               }
+
+               # The user didn't pass a createaccount token
+               if ( !$this->mToken ) {
+                       $this->mainLoginForm( wfMsg( 'sessionfailure' ) );
+                       return false;
+               }
+
+               # Validate the createaccount token
+               if ( $this->mToken !== self::getCreateaccountToken() ) {
+                       $this->mainLoginForm( wfMsg( 'sessionfailure' ) );
+                       return false;
+               }
+
                # Check permissions
                if ( !$wgUser->isAllowed( 'createaccount' ) ) {
                        $this->userNotPrivilegedMessage();
@@ -258,17 +300,12 @@ class LoginForm {
                $ip = wfGetIP();
                if ( $wgUser->isDnsBlacklisted( $ip, true /* check $wgProxyWhitelist */ ) ) {
                        $this->mainLoginForm( wfMsg( 'sorbs_create_account_reason' ) . ' (' . htmlspecialchars( $ip ) . ')' );
-                       return;
+                       return false;
                }
 
                # Now create a dummy user ($u) and check if it is valid
                $name = trim( $this->mName );
                $u = User::newFromName( $name, 'creatable' );
-               if ( WikiError::isError( $u ) ) {
-                       $this->mainLoginForm( wfMsg( $u->getMessage() ) );
-                       return false;
-               }
-
                if ( !is_object( $u ) ) {
                        $this->mainLoginForm( wfMsg( 'noname' ) );
                        return false;
@@ -340,6 +377,7 @@ class LoginForm {
                        return false;
                }
 
+               self::clearCreateaccountToken();
                return $this->initUser( $u, false );
        }
 
@@ -393,11 +431,27 @@ class LoginForm {
         * creation.
         */
        public function authenticateUserData() {
-               global $wgUser, $wgAuth;
-               if ( '' == $this->mName ) {
+               global $wgUser, $wgAuth, $wgMemc;
+
+               if ( $this->mName == '' ) {
                        return self::NO_NAME;
                }
-               
+
+               // We require a login token to prevent login CSRF
+               // Handle part of this before incrementing the throttle so
+               // token-less login attempts don't count towards the throttle
+               // but wrong-token attempts do.
+
+               // If the user doesn't have a login token yet, set one.
+               if ( !self::getLoginToken() ) {
+                       self::setLoginToken();
+                       return self::NEED_TOKEN;
+               }
+               // If the user didn't pass a login token, tell them we need one
+               if ( !$this->mToken ) {
+                       return self::NEED_TOKEN;
+               }
+
                global $wgPasswordAttemptThrottle;
 
                $throttleCount = 0;
@@ -405,18 +459,22 @@ class LoginForm {
                        $throttleKey = wfMemcKey( 'password-throttle', wfGetIP(), md5( $this->mName ) );
                        $count = $wgPasswordAttemptThrottle['count'];
                        $period = $wgPasswordAttemptThrottle['seconds'];
-                       
-                       global $wgMemc;
+
                        $throttleCount = $wgMemc->get( $throttleKey );
                        if ( !$throttleCount ) {
                                $wgMemc->add( $throttleKey, 1, $period ); // start counter
-                       } else if ( $throttleCount < $count ) {
-                               $wgMemc->incr($throttleKey);
-                       } else if ( $throttleCount >= $count ) {
+                       } elseif ( $throttleCount < $count ) {
+                               $wgMemc->incr( $throttleKey );
+                       } elseif ( $throttleCount >= $count ) {
                                return self::THROTTLED;
                        }
                }
 
+               // Validate the login token
+               if ( $this->mToken !== self::getLoginToken() ) {
+                       return self::WRONG_TOKEN;
+               }
+
                // Load $wgUser now, and check to see if we're logging in as the same
                // name. This is necessary because loading $wgUser (say by calling
                // getName()) calls the UserLoadFromSession hook, which potentially
@@ -424,7 +482,7 @@ class LoginForm {
                // for user existence using User::newFromName($name)->getId() below
                // will effectively be using stale data.
                if ( $wgUser->getName() === $this->mName ) {
-                       wfDebug( __METHOD__.": already logged in as {$this->mName}\n" );
+                       wfDebug( __METHOD__ . ": already logged in as {$this->mName}\n" );
                        return self::SUCCESS;
                }
 
@@ -433,7 +491,7 @@ class LoginForm {
                # TODO: Allow some magic here for invalid external names, e.g., let the
                # user choose a different wiki name.
                $u = User::newFromName( $this->mName );
-               if( is_null( $u ) || !User::isUsableName( $u->getName() ) ) {
+               if( !( $u instanceof User ) || !User::isUsableName( $u->getName() ) ) {
                        return self::ILLEGAL;
                }
 
@@ -464,7 +522,8 @@ class LoginForm {
                        return $abort;
                }
 
-               if (!$u->checkPassword( $this->mPassword )) {
+               global $wgBlockDisablesLogin;
+               if ( !$u->checkPassword( $this->mPassword ) ) {
                        if( $u->checkTemporaryPassword( $this->mPassword ) ) {
                                // The e-mailed temporary password should not be used for actu-
                                // al logins; that's a very sloppy habit, and insecure if an
@@ -492,15 +551,18 @@ class LoginForm {
                                // faces etc will probably just fail cleanly here.
                                $retval = self::RESET_PASS;
                        } else {
-                               $retval = '' == $this->mPassword ? self::EMPTY_PASS : self::WRONG_PASS;
+                               $retval = ( $this->mPassword  == '' ) ? self::EMPTY_PASS : self::WRONG_PASS;
                        }
+               } elseif ( $wgBlockDisablesLogin && $u->isBlocked() ) {
+                       // If we've enabled it, make it so that a blocked user cannot login
+                       $retval = self::USER_BLOCKED;
                } else {
                        $wgAuth->updateUser( $u );
                        $wgUser = $u;
 
                        // Please reset throttle for successful logins, thanks!
-                       if($throttleCount) {
-                               $wgMemc->delete($throttleKey);
+                       if( $throttleCount ) {
+                               $wgMemc->delete( $throttleKey );
                        }
 
                        if ( $isAutoCreated ) {
@@ -523,7 +585,7 @@ class LoginForm {
                global $wgAuth, $wgUser, $wgAutocreatePolicy;
 
                if ( $wgUser->isBlockedFromCreateAccount() ) {
-                       wfDebug( __METHOD__.": user is blocked from account creation\n" );
+                       wfDebug( __METHOD__ . ": user is blocked from account creation\n" );
                        return self::CREATE_BLOCKED;
                }
 
@@ -547,22 +609,22 @@ class LoginForm {
                                return self::NOT_EXISTS;
                        }
                        if ( !$wgAuth->userExists( $user->getName() ) ) {
-                               wfDebug( __METHOD__.": user does not exist\n" );
+                               wfDebug( __METHOD__ . ": user does not exist\n" );
                                return self::NOT_EXISTS;
                        }
                        if ( !$wgAuth->authenticate( $user->getName(), $this->mPassword ) ) {
-                               wfDebug( __METHOD__.": \$wgAuth->authenticate() returned false, aborting\n" );
+                               wfDebug( __METHOD__ . ": \$wgAuth->authenticate() returned false, aborting\n" );
                                return self::WRONG_PLUGIN_PASS;
                        }
                }
 
-               wfDebug( __METHOD__.": creating account\n" );
+               wfDebug( __METHOD__ . ": creating account\n" );
                $user = $this->initUser( $user, true );
                return self::SUCCESS;
        }
 
        function processLogin() {
-               global $wgUser, $wgAuth;
+               global $wgUser;
 
                switch ( $this->authenticateUserData() ) {
                        case self::SUCCESS:
@@ -574,6 +636,7 @@ class LoginForm {
                                        $wgUser->invalidateCache();
                                }
                                $wgUser->setCookies();
+                               self::clearLoginToken();
 
                                // Reset the throttle
                                $key = wfMemcKey( 'password-throttle', wfGetIP(), md5( $this->mName ) );
@@ -593,6 +656,12 @@ class LoginForm {
                                }
                                break;
 
+                       case self::NEED_TOKEN:
+                               $this->mainLoginForm( wfMsgExt( 'nocookieslogin', array( 'parseinline' ) ) );
+                               break;
+                       case self::WRONG_TOKEN:
+                               $this->mainLoginForm( wfMsg( 'sessionfailure' ) );
+                               break;
                        case self::NO_NAME:
                        case self::ILLEGAL:
                                $this->mainLoginForm( wfMsg( 'noname' ) );
@@ -601,7 +670,7 @@ class LoginForm {
                                $this->mainLoginForm( wfMsg( 'wrongpassword' ) );
                                break;
                        case self::NOT_EXISTS:
-                               if( $wgUser->isAllowed( 'createaccount' ) ){
+                               if( $wgUser->isAllowed( 'createaccount' ) ) {
                                        $this->mainLoginForm( wfMsgWikiHtml( 'nosuchuser', htmlspecialchars( $this->mName ) ) );
                                } else {
                                        $this->mainLoginForm( wfMsg( 'nosuchusershort', htmlspecialchars( $this->mName ) ) );
@@ -622,8 +691,12 @@ class LoginForm {
                        case self::THROTTLED:
                                $this->mainLoginForm( wfMsg( 'login-throttled' ) );
                                break;
+                       case self::USER_BLOCKED:
+                               $this->mainLoginForm( wfMsgExt( 'login-userblocked',
+                                       array( 'parsemag', 'escape' ), $this->mName ) );
+                               break;
                        default:
-                               throw new MWException( "Unhandled case value" );
+                               throw new MWException( 'Unhandled case value' );
                }
        }
 
@@ -639,43 +712,56 @@ class LoginForm {
         */
        function mailPassword() {
                global $wgUser, $wgOut, $wgAuth;
-               
+
                if ( wfReadOnly() ) {
                        $wgOut->readOnlyPage();
                        return false;
                }
-               
+
                if( !$wgAuth->allowPasswordChange() ) {
                        $this->mainLoginForm( wfMsg( 'resetpass_forbidden' ) );
                        return;
                }
 
-               # Check against blocked IPs
-               # fixme -- should we not?
+               # Check against blocked IPs so blocked users can't flood admins
+               # with password resets
                if( $wgUser->isBlocked() ) {
                        $this->mainLoginForm( wfMsg( 'blocked-mailpassword' ) );
                        return;
                }
-               
-               // Check for hooks
+
+               # Check for hooks
                $error = null;
-               if ( ! wfRunHooks( 'UserLoginMailPassword', array( $this->mName, &$error ) ) ) {
+               if ( !wfRunHooks( 'UserLoginMailPassword', array( $this->mName, &$error ) ) ) {
                        $this->mainLoginForm( $error );
                        return;
                }
 
+               # If the user doesn't have a login token yet, set one.
+               if ( !self::getLoginToken() ) {
+                       self::setLoginToken();
+                       $this->mainLoginForm( wfMsg( 'sessionfailure' ) );
+                       return;
+               }
+
+               # If the user didn't pass a login token, tell them we need one
+               if ( !$this->mToken ) {
+                       $this->mainLoginForm( wfMsg( 'sessionfailure' ) );
+                       return;
+               }
+
                # Check against the rate limiter
                if( $wgUser->pingLimiter( 'mailpassword' ) ) {
                        $wgOut->rateLimited();
                        return;
                }
 
-               if ( '' == $this->mName ) {
+               if ( $this->mName == '' ) {
                        $this->mainLoginForm( wfMsg( 'noname' ) );
                        return;
                }
                $u = User::newFromName( $this->mName );
-               if( is_null( $u ) ) {
+               if( !$u instanceof User ) {
                        $this->mainLoginForm( wfMsg( 'noname' ) );
                        return;
                }
@@ -684,6 +770,12 @@ class LoginForm {
                        return;
                }
 
+               # Validate the login token
+               if ( $this->mToken !== self::getLoginToken() ) {
+                       $this->mainLoginForm( wfMsg( 'sessionfailure' ) );
+                       return;
+               }
+
                # Check against password throttle
                if ( $u->isPasswordReminderThrottled() ) {
                        global $wgPasswordReminderResendTime;
@@ -699,38 +791,39 @@ class LoginForm {
                        $this->mainLoginForm( wfMsg( 'mailerror', $result->getMessage() ) );
                } else {
                        $this->mainLoginForm( wfMsg( 'passwordsent', $u->getName() ), 'success' );
+                       self::clearLoginToken();
                }
        }
 
 
        /**
-        * @param object user
-        * @param bool throttle
-        * @param string message name of email title
-        * @param string message name of email text
-        * @return mixed true on success, WikiError on failure
+        * @param $u User object
+        * @param $throttle Boolean
+        * @param $emailTitle String: message name of email title
+        * @param $emailText String: message name of email text
+        * @return Mixed: true on success, WikiError on failure
         * @private
         */
        function mailPasswordInternal( $u, $throttle = true, $emailTitle = 'passwordremindertitle', $emailText = 'passwordremindertext' ) {
                global $wgServer, $wgScript, $wgUser, $wgNewPasswordExpiry;
 
-               if ( '' == $u->getEmail() ) {
+               if ( $u->getEmail() == '' ) {
                        return new WikiError( wfMsg( 'noemail', $u->getName() ) );
                }
                $ip = wfGetIP();
                if( !$ip ) {
                        return new WikiError( wfMsg( 'badipaddress' ) );
                }
-               
-               wfRunHooks( 'User::mailPasswordInternal', array(&$wgUser, &$ip, &$u) );
+
+               wfRunHooks( 'User::mailPasswordInternal', array( &$wgUser, &$ip, &$u ) );
 
                $np = $u->randomPassword();
                $u->setNewpassword( $np, $throttle );
                $u->saveSettings();
-
-               $m = wfMsgExt( $emailText, array( 'parsemag' ), $ip, $u->getName(), $np,
+               $userLanguage = $u->getOption( 'language' );
+               $m = wfMsgExt( $emailText, array( 'parsemag', 'language' => $userLanguage ), $ip, $u->getName(), $np,
                                $wgServer . $wgScript, round( $wgNewPasswordExpiry / 86400 ) );
-               $result = $u->sendMail( wfMsg( $emailTitle ), $m );
+               $result = $u->sendMail( wfMsgExt( $emailTitle, array( 'parsemag', 'language' => $userLanguage ) ), $m );
 
                return $result;
        }
@@ -751,7 +844,7 @@ class LoginForm {
 
                # Run any hooks; display injected HTML if any, else redirect
                $injected_html = '';
-               wfRunHooks('UserLoginComplete', array(&$wgUser, &$injected_html));
+               wfRunHooks( 'UserLoginComplete', array( &$wgUser, &$injected_html ) );
 
                if( $injected_html !== '' ) {
                        $this->displaySuccessfulLogin( 'loginsuccess', $injected_html );
@@ -771,11 +864,10 @@ class LoginForm {
         * @private
         */
        function successfulCreation() {
-               global $wgUser, $wgOut;
-
+               global $wgUser;
                # Run any hooks; display injected HTML
                $injected_html = '';
-               wfRunHooks('UserLoginComplete', array(&$wgUser, &$injected_html));
+               wfRunHooks( 'UserLoginComplete', array( &$wgUser, &$injected_html ) );
 
                $this->displaySuccessfulLogin( 'welcomecreation', $injected_html );
        }
@@ -807,7 +899,7 @@ class LoginForm {
                $wgOut->setRobotPolicy( 'noindex,nofollow' );
                $wgOut->setArticleRelated( false );
 
-               $wgOut->addWikitext( $wgOut->formatPermissionsErrorMessage( $errors, 'createaccount' ) );
+               $wgOut->addWikiText( $wgOut->formatPermissionsErrorMessage( $errors, 'createaccount' ) );
                // Stuff that might want to be added at the end. For example, instruc-
                // tions if blocked.
                $wgOut->addWikiMsg( 'cantcreateaccount-nonblock-text' );
@@ -847,14 +939,14 @@ class LoginForm {
         */
        function mainLoginForm( $msg, $msgtype = 'error' ) {
                global $wgUser, $wgOut, $wgHiddenPrefs, $wgEnableEmail;
-               global $wgCookiePrefix, $wgLoginLanguageSelector;
+               global $wgRequest, $wgLoginLanguageSelector;
                global $wgAuth, $wgEmailConfirmToEdit, $wgCookieExpiration;
-               
+
                $titleObj = SpecialPage::getTitleFor( 'Userlogin' );
-               
+
                if ( $this->mType == 'signup' ) {
-                       // Block signup here if in readonly. Keeps user from 
-                       // going through the process (filling out data, etc) 
+                       // Block signup here if in readonly. Keeps user from
+                       // going through the process (filling out data, etc)
                        // and being informed later.
                        if ( wfReadOnly() ) {
                                $wgOut->readOnlyPage();
@@ -868,17 +960,21 @@ class LoginForm {
                        }
                }
 
-               if ( '' == $this->mName ) {
+               if ( $this->mName == '' ) {
                        if ( $wgUser->isLoggedIn() ) {
                                $this->mName = $wgUser->getName();
                        } else {
-                               $this->mName = isset( $_COOKIE[$wgCookiePrefix.'UserName'] ) ? $_COOKIE[$wgCookiePrefix.'UserName'] : null;
+                               $this->mName = $wgRequest->getCookie( 'UserName' );
                        }
                }
 
                $titleObj = SpecialPage::getTitleFor( 'Userlogin' );
 
                if ( $this->mType == 'signup' ) {
+                       global $wgLivePasswordStrengthChecks;
+                       if ( $wgLivePasswordStrengthChecks ) {
+                               $wgOut->addPasswordSecurity( 'wpPassword2', 'wpRetype' );
+                       }
                        $template = new UsercreateTemplate();
                        $q = 'action=submitlogin&type=signup';
                        $linkq = 'type=login';
@@ -892,26 +988,29 @@ class LoginForm {
 
                if ( !empty( $this->mReturnTo ) ) {
                        $returnto = '&returnto=' . wfUrlencode( $this->mReturnTo );
-                       if ( !empty( $this->mReturnToQuery ) )
+                       if ( !empty( $this->mReturnToQuery ) ) {
                                $returnto .= '&returntoquery=' .
                                        wfUrlencode( $this->mReturnToQuery );
+                       }
                        $q .= $returnto;
                        $linkq .= $returnto;
                }
 
                # Pass any language selection on to the mode switch link
-               if( $wgLoginLanguageSelector && $this->mLanguage )
+               if( $wgLoginLanguageSelector && $this->mLanguage ) {
                        $linkq .= '&uselang=' . $this->mLanguage;
+               }
 
-               $link = '<a href="' . htmlspecialchars ( $titleObj->getLocalUrl( $linkq ) ) . '">';
+               $link = '<a href="' . htmlspecialchars ( $titleObj->getLocalURL( $linkq ) ) . '">';
                $link .= wfMsgHtml( $linkmsg . 'link' ); # Calling either 'gotaccountlink' or 'nologinlink'
                $link .= '</a>';
 
                # Don't show a "create account" link if the user can't
-               if( $this->showCreateOrLoginLink( $wgUser ) )
-                       $template->set( 'link', wfMsgWikiHtml( $linkmsg, $link ) );
-               else
+               if( $this->showCreateOrLoginLink( $wgUser ) ) {
+                       $template->set( 'link', wfMsgExt( $linkmsg, array( 'parseinline', 'replaceafter' ), $link ) );
+               } else {
                        $template->set( 'link', '' );
+               }
 
                $template->set( 'header', '' );
                $template->set( 'name', $this->mName );
@@ -920,8 +1019,9 @@ class LoginForm {
                $template->set( 'email', $this->mEmail );
                $template->set( 'realname', $this->mRealName );
                $template->set( 'domain', $this->mDomain );
+               $template->set( 'reason', $this->mReason );
 
-               $template->set( 'action', $titleObj->getLocalUrl( $q ) );
+               $template->set( 'action', $titleObj->getLocalURL( $q ) );
                $template->set( 'message', $msg );
                $template->set( 'messagetype', $msgtype );
                $template->set( 'createemail', $wgEnableEmail && $wgUser->isLoggedIn() );
@@ -930,7 +1030,20 @@ class LoginForm {
                $template->set( 'emailrequired', $wgEmailConfirmToEdit );
                $template->set( 'canreset', $wgAuth->allowPasswordChange() );
                $template->set( 'canremember', ( $wgCookieExpiration > 0 ) );
-               $template->set( 'remember', $wgUser->getOption( 'rememberpassword' ) or $this->mRemember  );
+               $template->set( 'usereason', $wgUser->isLoggedIn() );
+               $template->set( 'remember', $wgUser->getOption( 'rememberpassword' ) || $this->mRemember );
+
+               if ( $this->mType == 'signup' ) {
+                       if ( !self::getCreateaccountToken() ) {
+                               self::setCreateaccountToken();
+                       }
+                       $template->set( 'token', self::getCreateaccountToken() );
+               } else {
+                       if ( !self::getLoginToken() ) {
+                               self::setLoginToken();
+                       }
+                       $template->set( 'token', self::getLoginToken() );
+               }
 
                # Prepare language selection links as needed
                if( $wgLoginLanguageSelector ) {
@@ -947,7 +1060,7 @@ class LoginForm {
                        wfRunHooks( 'UserLoginForm', array( &$template ) );
                }
 
-               //Changes the title depending on permissions for creating account
+               // Changes the title depending on permissions for creating account
                if ( $wgUser->isAllowed( 'createaccount' ) ) {
                        $wgOut->setPageTitle( wfMsg( 'userlogin' ) );
                } else {
@@ -956,7 +1069,7 @@ class LoginForm {
 
                $wgOut->setRobotPolicy( 'noindex,nofollow' );
                $wgOut->setArticleRelated( false );
-               $wgOut->disallowUserJs();  // just in case...
+               $wgOut->disallowUserJs(); // just in case...
                $wgOut->addTemplate( $template );
        }
 
@@ -987,6 +1100,56 @@ class LoginForm {
                return $wgDisableCookieCheck ? true : $wgRequest->checkSessionCookie();
        }
 
+       /**
+        * Get the login token from the current session
+        */
+       public static function getLoginToken() {
+               global $wgRequest;
+               return $wgRequest->getSessionData( 'wsLoginToken' );
+       }
+
+       /**
+        * Randomly generate a new login token and attach it to the current session
+        */
+       public static function setLoginToken() {
+               global $wgRequest;
+               // Use User::generateToken() instead of $user->editToken()
+               // because the latter reuses $_SESSION['wsEditToken']
+               $wgRequest->setSessionData( 'wsLoginToken', User::generateToken() );
+       }
+
+       /**
+        * Remove any login token attached to the current session
+        */
+       public static function clearLoginToken() {
+               global $wgRequest;
+               $wgRequest->setSessionData( 'wsLoginToken', null );
+       }
+
+       /**
+        * Get the createaccount token from the current session
+        */
+       public static function getCreateaccountToken() {
+               global $wgRequest;
+               return $wgRequest->getSessionData( 'wsCreateaccountToken' );
+       }
+
+       /**
+        * Randomly generate a new createaccount token and attach it to the current session
+        */
+       public static function setCreateaccountToken() {
+               global $wgRequest;
+               $wgRequest->setSessionData( 'wsCreateaccountToken', User::generateToken() );
+       }
+
+       /**
+        * Remove any createaccount token attached to the current session
+        */
+       public static function clearCreateaccountToken() {
+               global $wgRequest;
+               $wgRequest->setSessionData( 'wsCreateaccountToken', null );
+       }
+
        /**
         * @private
         */
@@ -995,7 +1158,9 @@ class LoginForm {
 
                $titleObj = SpecialPage::getTitleFor( 'Userlogin' );
                $query = array( 'wpCookieCheck' => $type );
-               if ( $this->mReturnTo ) $query['returnto'] = $this->mReturnTo;
+               if ( $this->mReturnTo ) {
+                       $query['returnto'] = $this->mReturnTo;
+               }
                $check = $titleObj->getFullURL( $query );
 
                return $wgOut->redirect( $check );
@@ -1008,7 +1173,7 @@ class LoginForm {
                if ( !$this->hasSessionCookie() ) {
                        if ( $type == 'new' ) {
                                return $this->mainLoginForm( wfMsgExt( 'nocookiesnew', array( 'parseinline' ) ) );
-                       } else if ( $type == 'login' ) {
+                       } elseif ( $type == 'login' ) {
                                return $this->mainLoginForm( wfMsgExt( 'nocookieslogin', array( 'parseinline' ) ) );
                        } else {
                                # shouldn't happen
@@ -1042,7 +1207,7 @@ class LoginForm {
                        foreach( $langs as $lang ) {
                                $lang = trim( $lang, '* ' );
                                $parts = explode( '|', $lang );
-                               if (count($parts) >= 2) {
+                               if ( count( $parts ) >= 2 ) {
                                        $links[] = $this->makeLanguageSelectorLink( $parts[0], $parts[1] );
                                }
                        }
@@ -1063,10 +1228,12 @@ class LoginForm {
                global $wgUser;
                $self = SpecialPage::getTitleFor( 'Userlogin' );
                $attr = array( 'uselang' => $lang );
-               if( $this->mType == 'signup' )
+               if( $this->mType == 'signup' ) {
                        $attr['type'] = 'signup';
-               if( $this->mReturnTo )
+               }
+               if( $this->mReturnTo ) {
                        $attr['returnto'] = $this->mReturnTo;
+               }
                $skin = $wgUser->getSkin();
                return $skin->linkKnown(
                        $self,