SpecialPasswordReset.php

Go to the documentation of this file.

<?php
class SpecialPasswordReset extends FormSpecialPage {

        private $email;

        private $result;

        public function __construct() {
                parent::__construct( 'PasswordReset' );
        }

        public function userCanExecute( User $user ) {
                return $this->canChangePassword( $user ) === true && parent::userCanExecute( $user );
        }

        public function checkExecutePermissions( User $user ) {
                $error = $this->canChangePassword( $user );
                if ( is_string( $error ) ) {
                        throw new ErrorPageError( 'internalerror', $error );
                } elseif ( !$error ) {
                        throw new ErrorPageError( 'internalerror', 'resetpass_forbidden' );
                }

                return parent::checkExecutePermissions( $user );
        }

        protected function getFormFields() {
                global $wgPasswordResetRoutes, $wgAuth;
                $a = array();
                if ( isset( $wgPasswordResetRoutes['username'] ) && $wgPasswordResetRoutes['username'] ) {
                        $a['Username'] = array(
                                'type' => 'text',
                                'label-message' => 'passwordreset-username',
                        );
                        if( $this->getUser()->isLoggedIn() ) {
                                $a['Username']['default'] = $this->getUser()->getName();
                        }
                }

                if ( isset( $wgPasswordResetRoutes['email'] ) && $wgPasswordResetRoutes['email'] ) {
                        $a['Email'] = array(
                                'type' => 'email',
                                'label-message' => 'passwordreset-email',
                        );
                }

                if ( isset( $wgPasswordResetRoutes['domain'] ) && $wgPasswordResetRoutes['domain'] ) {
                        $domains = $wgAuth->domainList();
                        $a['Domain'] = array(
                                'type' => 'select',
                                'options' => $domains,
                                'label-message' => 'passwordreset-domain',
                        );
                }

                if( $this->getUser()->isAllowed( 'passwordreset' ) ){
                        $a['Capture'] = array(
                                'type' => 'check',
                                'label-message' => 'passwordreset-capture',
                                'help-message' => 'passwordreset-capture-help',
                        );
                }

                return $a;
        }

        public function alterForm( HTMLForm $form ) {
                $form->setSubmitTextMsg( 'mailmypassword' );
        }

        protected function preText() {
                global $wgPasswordResetRoutes;
                $i = 0;
                if ( isset( $wgPasswordResetRoutes['username'] ) && $wgPasswordResetRoutes['username'] ) {
                        $i++;
                }
                if ( isset( $wgPasswordResetRoutes['email'] ) && $wgPasswordResetRoutes['email'] ) {
                        $i++;
                }
                if ( isset( $wgPasswordResetRoutes['domain'] ) && $wgPasswordResetRoutes['domain'] ) {
                        $i++;
                }
                return $this->msg( 'passwordreset-pretext', $i )->parseAsBlock();
        }

        public function onSubmit( array $data ) {
                global $wgAuth;

                if ( isset( $data['Domain'] ) ) {
                        if ( $wgAuth->validDomain( $data['Domain'] ) ) {
                                $wgAuth->setDomain( $data['Domain'] );
                        } else {
                                $wgAuth->setDomain( 'invaliddomain' );
                        }
                }

                if( isset( $data['Capture'] ) && !$this->getUser()->isAllowed( 'passwordreset' ) ){
                        // The user knows they don't have the passwordreset permission, but they tried to spoof the form.  That's naughty
                        throw new PermissionsError( 'passwordreset' );
                }

                if ( isset( $data['Username'] ) && $data['Username'] !== '' ) {
                        $method = 'username';
                        $users = array( User::newFromName( $data['Username'] ) );
                } elseif ( isset( $data['Email'] )
                        && $data['Email'] !== ''
                        && Sanitizer::validateEmail( $data['Email'] ) )
                {
                        $method = 'email';
                        $res = wfGetDB( DB_SLAVE )->select(
                                'user',
                                User::selectFields(),
                                array( 'user_email' => $data['Email'] ),
                                __METHOD__
                        );
                        if ( $res ) {
                                $users = array();
                                foreach( $res as $row ){
                                        $users[] = User::newFromRow( $row );
                                }
                        } else {
                                // Some sort of database error, probably unreachable
                                throw new MWException( 'Unknown database error in ' . __METHOD__ );
                        }
                } else {
                        // The user didn't supply any data
                        return false;
                }

                // Check for hooks (captcha etc), and allow them to modify the users list
                $error = array();
                if ( !wfRunHooks( 'SpecialPasswordResetOnSubmit', array( &$users, $data, &$error ) ) ) {
                        return array( $error );
                }

                if( count( $users ) == 0 ){
                        if( $method == 'email' ){
                                // Don't reveal whether or not an email address is in use
                                return true;
                        } else {
                                return array( 'noname' );
                        }
                }

                $firstUser = $users[0];

                if ( !$firstUser instanceof User || !$firstUser->getID() ) {
                        return array( array( 'nosuchuser', $data['Username'] ) );
                }

                // Check against the rate limiter
                if ( $this->getUser()->pingLimiter( 'mailpassword' ) ) {
                        throw new ThrottledError;
                }

                // Check against password throttle
                foreach ( $users as $user ) {
                        if ( $user->isPasswordReminderThrottled() ) {
                                global $wgPasswordReminderResendTime;
                                # Round the time in hours to 3 d.p., in case someone is specifying
                                # minutes or seconds.
                                return array( array( 'throttled-mailpassword', round( $wgPasswordReminderResendTime, 3 ) ) );
                        }
                }

                global $wgNewPasswordExpiry;

                // All the users will have the same email address
                if ( $firstUser->getEmail() == '' ) {
                        // This won't be reachable from the email route, so safe to expose the username
                        return array( array( 'noemail', $firstUser->getName() ) );
                }

                // We need to have a valid IP address for the hook, but per bug 18347, we should
                // send the user's name if they're logged in.
                $ip = $this->getRequest()->getIP();
                if ( !$ip ) {
                        return array( 'badipaddress' );
                }
                $caller = $this->getUser();
                wfRunHooks( 'User::mailPasswordInternal', array( &$caller, &$ip, &$firstUser ) );
                $username = $caller->getName();
                $msg = IP::isValid( $username )
                        ? 'passwordreset-emailtext-ip'
                        : 'passwordreset-emailtext-user';

                // Send in the user's language; which should hopefully be the same
                $userLanguage = $firstUser->getOption( 'language' );

                $passwords = array();
                foreach ( $users as $user ) {
                        $password = $user->randomPassword();
                        $user->setNewpassword( $password );
                        $user->saveSettings();
                        $passwords[] = $this->msg( 'passwordreset-emailelement', $user->getName(), $password
                                )->inLanguage( $userLanguage )->text(); // We'll escape the whole thing later
                }
                $passwordBlock = implode( "\n\n", $passwords );

                $this->email = $this->msg( $msg )->inLanguage( $userLanguage );
                $this->email->params(
                        $username,
                        $passwordBlock,
                        count( $passwords ),
                        Title::newMainPage()->getCanonicalUrl(),
                        round( $wgNewPasswordExpiry / 86400 )
                );

                $title = $this->msg( 'passwordreset-emailtitle' );

                $this->result = $firstUser->sendMail( $title->escaped(), $this->email->escaped() );

                // Blank the email if the user is not supposed to see it
                if( !isset( $data['Capture'] ) || !$data['Capture'] ) {
                        $this->email = null;
                }

                if ( $this->result->isGood() ) {
                        return true;
                } elseif( isset( $data['Capture'] ) && $data['Capture'] ){
                        // The email didn't send, but maybe they knew that and that's why they captured it
                        return true;
                } else {
                        // @todo FIXME: The email didn't send, but we have already set the password throttle
                        // timestamp, so they won't be able to try again until it expires...  :(
                        return array( array( 'mailerror', $this->result->getMessage() ) );
                }
        }

        public function onSuccess() {
                if( $this->getUser()->isAllowed( 'passwordreset' ) && $this->email != null ){
                        // @todo: Logging

                        if( $this->result->isGood() ){
                                $this->getOutput()->addWikiMsg( 'passwordreset-emailsent-capture' );
                        } else {
                                $this->getOutput()->addWikiMsg( 'passwordreset-emailerror-capture', $this->result->getMessage() );
                        }

                        $this->getOutput()->addHTML( Html::rawElement( 'pre', array(), $this->email->escaped() ) );
                }

                $this->getOutput()->addWikiMsg( 'passwordreset-emailsent' );
                $this->getOutput()->returnToMain();
        }

        protected function canChangePassword( User $user ) {
                global $wgPasswordResetRoutes, $wgAuth;

                // Maybe password resets are disabled, or there are no allowable routes
                if ( !is_array( $wgPasswordResetRoutes ) ||
                         !in_array( true, array_values( $wgPasswordResetRoutes ) ) )
                {
                        return 'passwordreset-disabled';
                }

                // Maybe the external auth plugin won't allow local password changes
                if ( !$wgAuth->allowPasswordChange() ) {
                        return 'resetpass_forbidden';
                }

                // Maybe the user is blocked (check this here rather than relying on the parent
                // method as we have a more specific error message to use here
                if ( $user->isBlocked() ) {
                        return 'blocked-mailpassword';
                }

                return true;
        }

        function isListed() {
                if ( $this->canChangePassword( $this->getUser() ) === true ) {
                        return parent::isListed();
                }

                return false;
        }
}