don't just assume we get a valid title object
[lhc/web/wiklou.git] / includes / SpecialUserlogin.php
index 89eedcb..b392a38 100644 (file)
@@ -6,12 +6,7 @@
  */
 
 /**
- *
- */
-require_once('UserMailer.php');
-
-/**
- * consutrctor
+ * constructor
  */
 function wfSpecialUserlogin() {
        global $wgCommandLineMode;
@@ -32,30 +27,48 @@ function wfSpecialUserlogin() {
 class LoginForm {
        var $mName, $mPassword, $mRetype, $mReturnto, $mCookieCheck, $mPosted;
        var $mAction, $mCreateaccount, $mCreateaccountMail, $mMailmypassword;
-       var $mLoginattempt, $mRemember, $mEmail;
+       var $mLoginattempt, $mRemember, $mEmail, $mDomain;
        
+       /**
+        * Constructor
+        * @param webrequest $request A webrequest object passed by reference
+        */
        function LoginForm( &$request ) {
-               global $wgLang, $wgAllowRealName;
+               global $wgLang, $wgAllowRealName, $wgEnableEmail;
+               global $wgAuth;
 
                $this->mName = $request->getText( 'wpName' );
                $this->mPassword = $request->getText( 'wpPassword' );
                $this->mRetype = $request->getText( 'wpRetype' );
+               $this->mDomain = $request->getText( 'wpDomain' );
                $this->mReturnto = $request->getVal( 'returnto' );
                $this->mCookieCheck = $request->getVal( 'wpCookieCheck' );
                $this->mPosted = $request->wasPosted();
                $this->mCreateaccount = $request->getCheck( 'wpCreateaccount' );
-               $this->mCreateaccountMail = $request->getCheck( 'wpCreateaccountMail' );
-               $this->mMailmypassword = $request->getCheck( 'wpMailmypassword' );
+               $this->mCreateaccountMail = $request->getCheck( 'wpCreateaccountMail' )
+                                           && $wgEnableEmail;
+               $this->mMailmypassword = $request->getCheck( 'wpMailmypassword' )
+                                        && $wgEnableEmail;
                $this->mLoginattempt = $request->getCheck( 'wpLoginattempt' );
                $this->mAction = $request->getVal( 'action' );
                $this->mRemember = $request->getCheck( 'wpRemember' );
-               $this->mEmail = $request->getText( 'wpEmail' );
-               if ($wgAllowRealName) {
+               
+               if( $wgEnableEmail ) {
+                       $this->mEmail = $request->getText( 'wpEmail' );
+               } else {
+                       $this->mEmail = '';
+               }
+               if( $wgAllowRealName ) {
                    $this->mRealName = $request->getText( 'wpRealName' );
                } else {
                    $this->mRealName = '';
                }
-           
+               
+               if( !$wgAuth->validDomain( $this->mDomain ) ) {
+                       $this->mDomain = 'invaliddomain';
+               }
+               $wgAuth->setDomain( $this->mDomain );
                # When switching accounts, it sucks to get automatically logged out
                if( $this->mReturnto == $wgLang->specialPage( 'Userlogout' ) ) {
                        $this->mReturnto = '';
@@ -73,7 +86,7 @@ class LoginForm {
                                return $this->addNewAccountMailPassword();
                        } else if ( $this->mMailmypassword ) {
                                return $this->mailPassword();
-                       } else if ( ( 'submit' == $this->mAction ) || $this->mLoginattempt ) {
+                       } else if ( ( 'submitlogin' == $this->mAction ) || $this->mLoginattempt ) {
                                return $this->processLogin();
                        }
                }
@@ -98,19 +111,18 @@ class LoginForm {
                }
 
                $u->saveSettings();
-               $error = $this->mailPasswordInternal($u);
-
+               $result = $this->mailPasswordInternal($u);
+               
                $wgOut->setPageTitle( wfMsg( 'accmailtitle' ) );
                $wgOut->setRobotpolicy( 'noindex,nofollow' );
                $wgOut->setArticleRelated( false );
        
-               if ( $error === '' ) {
+               if( WikiError::isError( $result ) ) {
+                       $this->mainLoginForm( wfMsg( 'mailerror', $result->getMessage() ) );
+               } else {
                        $wgOut->addWikiText( wfMsg( 'accmailtext', $u->getName(), $u->getEmail() ) );
                        $wgOut->returnToMain( false );
-               } else {
-                       $this->mainLoginForm( wfMsg( 'mailerror', $error ) );
                }
-
                $u = 0;
        }
 
@@ -119,8 +131,7 @@ class LoginForm {
         * @access private
         */
        function addNewAccount() {
-               global $wgUser, $wgOut;
-               global $wgDeferredUpdateList;
+               global $wgUser, $wgOut, $wgEmailAuthentication;
 
                $u = $this->addNewAccountInternal();
 
@@ -131,8 +142,10 @@ class LoginForm {
                $wgUser = $u;
                $wgUser->setCookies();
 
-               $up = new UserUpdate();
-               array_push( $wgDeferredUpdateList, $up );
+               $wgUser->saveSettings();
+               if( $wgEmailAuthentication && $wgUser->isValidEmailAddr( $wgUser->getEmail() ) ) {
+                       $wgUser->sendConfirmationMail();
+               }
 
                if( $this->hasSessionCookie() ) {
                        return $this->successfulLogin( wfMsg( 'welcomecreation', $wgUser->getName() ) );
@@ -141,45 +154,72 @@ class LoginForm {
                }
        }
 
-
        /**
         * @access private
         */
        function addNewAccountInternal() {
                global $wgUser, $wgOut;
-               global $wgMaxNameChars;
+               global $wgUseLatin1, $wgEnableSorbs, $wgProxyWhitelist;
                global $wgMemc, $wgAccountCreationThrottle, $wgDBname, $wgIP;
+               global $wgAuth;
+
+               // If the user passes an invalid domain, something is fishy
+               if( !$wgAuth->validDomain( $this->mDomain ) ) {
+                       $this->mainLoginForm( wfMsg( 'wrongpassword' ) );
+                       return false;
+               }
+
+               // If we are not allowing users to login locally, we should
+               // be checking to see if the user is actually able to
+               // authenticate to the authentication 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( !$wgAuth->canCreateAccounts() && ( !$wgAuth->userExists( $this->mName ) || !$wgAuth->authenticate( $this->mName, $this->mPassword ) ) ) {
+                               $this->mainLoginForm( wfMsg( 'wrongpassword' ) );
+                               return false;
+                       }
+               }
 
                if (!$wgUser->isAllowedToCreateAccount()) {
                        $this->userNotPrivilegedMessage();
+                       return false;
+               }
+
+               if ( $wgEnableSorbs && !in_array( $wgIP, $wgProxyWhitelist ) && 
+                 $wgUser->inSorbsBlacklist( $wgIP ) ) 
+               {
+                       $this->mainLoginForm( wfMsg( 'sorbs_create_account_reason' ) );
                        return;
                }
 
+
                if ( 0 != strcmp( $this->mPassword, $this->mRetype ) ) {
                        $this->mainLoginForm( wfMsg( 'badretype' ) );
-                       return;
+                       return false;
                }
                
                $name = trim( $this->mName );
                $u = User::newFromName( $name );
-               if ( is_null( $u ) ||
-                 ( '' == $name ) ||
-                 preg_match( "/\\d{1,3}\\.\\d{1,3}\\.\\d{1,3}\\.\\d{1,3}/", $name ) ||
-                 (strpos( $name, "/" ) !== false) ||
-                 (strlen( $name ) > $wgMaxNameChars) ||
-                 ucFirst($name) != $u->getName() ) 
-               {
+               if ( is_null( $u ) ) {
                        $this->mainLoginForm( wfMsg( 'noname' ) );
-                       return;
+                       return false;
                }
+               
                if ( wfReadOnly() ) {
                        $wgOut->readOnlyPage();
-                       return;
+                       return false;
                }
                
                if ( 0 != $u->idForName() ) {
                        $this->mainLoginForm( wfMsg( 'userexists' ) );
-                       return;
+                       return false;
+               }
+
+               if ( !$wgUser->isValidPassword( $this->mPassword ) ) {
+                       $this->mainLoginForm( wfMsg( 'passwordtooshort', $wgMinimalPasswordLength ) );
+                       return false;
                }
 
                if ( $wgAccountCreationThrottle ) {
@@ -190,14 +230,39 @@ class LoginForm {
                        }
                        if ( $value > $wgAccountCreationThrottle ) {
                                $this->throttleHit( $wgAccountCreationThrottle );
-                               return;
+                               return false;
                        }
                }
 
+               if( !$wgAuth->addUser( $u, $this->mPassword ) ) {
+                       $this->mainLoginForm( wfMsg( 'externaldberror' ) );
+                       return false;
+               }
+
+               # Update user count
+               $ssUpdate = new SiteStatsUpdate( 0, 0, 0, 0, 1 );
+               $ssUpdate->doUpdate();
+
+               return $this->initUser( $u );
+       }
+       
+       /**
+        * Actually add a user to the database.
+        * Give it a User object that has been initialised with a name.
+        *
+        * @param User $u
+        * @return User
+        * @access private
+        */
+       function &initUser( &$u ) {
                $u->addToDatabase();
                $u->setPassword( $this->mPassword );
                $u->setEmail( $this->mEmail );
                $u->setRealName( $this->mRealName );
+               $u->setToken();
+               
+               global $wgAuth;
+               $wgAuth->initUser( $u );
 
                if ( $this->mRemember ) { $r = 1; }
                else { $r = 0; }
@@ -211,7 +276,7 @@ class LoginForm {
         */
        function processLogin() {
                global $wgUser;
-               global $wgDeferredUpdateList;
+               global $wgAuth;
 
                if ( '' == $this->mName ) {
                        $this->mainLoginForm( wfMsg( 'noname' ) );
@@ -222,13 +287,28 @@ class LoginForm {
                        $this->mainLoginForm( wfMsg( 'noname' ) );
                        return;
                }
-               $id = $u->idForName();
-               if ( 0 == $id ) {
-                       $this->mainLoginForm( wfMsg( 'nosuchuser', $u->getName() ) );
-                       return;
+               if ( 0 == $u->getID() ) {
+                       global $wgAuth;
+                       /**
+                        * If the external authentication plugin allows it,
+                        * automatically create a new account for users that
+                        * are externally defined but have not yet logged in.
+                        */
+                       if ( $wgAuth->autoCreate() && $wgAuth->userExists( $u->getName() ) ) {
+                               if ( $wgAuth->authenticate( $u->getName(), $this->mPassword ) ) {
+                                       $u =& $this->initUser( $u );
+                               } else {
+                                       $this->mainLoginForm( wfMsg( 'wrongpassword' ) );
+                                       return;
+                               }
+                       } else {
+                               $this->mainLoginForm( wfMsg( 'nosuchuser', $u->getName() ) );
+                               return;
+                       }
+               } else {
+                       $u->loadFromDatabase();
                }
-               $u->setId( $id );
-               $u->loadFromDatabase();
+
                if (!$u->checkPassword( $this->mPassword )) {
                        $this->mainLoginForm( wfMsg( 'wrongpassword' ) );
                        return;
@@ -243,12 +323,13 @@ class LoginForm {
                }
                $u->setOption( 'rememberpassword', $r );
 
+               $wgAuth->updateUser( $u );
+
                $wgUser = $u;
                $wgUser->setCookies();
 
-               $up = new UserUpdate();
-               array_push( $wgDeferredUpdateList, $up );
-
+               $wgUser->saveSettings();
+               
                if( $this->hasSessionCookie() ) {
                        return $this->successfulLogin( wfMsg( 'loginsuccess', $wgUser->getName() ) );
                } else {
@@ -272,39 +353,39 @@ class LoginForm {
                        $this->mainLoginForm( wfMsg( 'noname' ) );
                        return;
                }
-               $id = $u->idForName();
-               if ( 0 == $id ) {
+               if ( 0 == $u->getID() ) {
                        $this->mainLoginForm( wfMsg( 'nosuchuser', $u->getName() ) );
                        return;
                }
-               $u->setId( $id );
+
                $u->loadFromDatabase();
 
-               $error = $this->mailPasswordInternal( $u );
-               if ($error === '') {
-                       $this->mainLoginForm( wfMsg( 'passwordsent', $u->getName() ) );
+               $result = $this->mailPasswordInternal( $u );
+               if( WikiError::isError( $result ) ) {
+                       $this->mainLoginForm( wfMsg( 'mailerror', $result->getMessage() ) );
                } else {
-                       $this->mainLoginForm( wfMsg( 'mailerror', $error ) );
+                       $this->mainLoginForm( wfMsg( 'passwordsent', $u->getName() ) );
                }
-
        }
 
 
        /**
+        * @return mixed true on success, WikiError on failure
         * @access private
         */
        function mailPasswordInternal( $u ) {
-               global $wgDeferredUpdateList, $wgOutputEncoding;
                global $wgPasswordSender, $wgDBname, $wgIP;
                global $wgCookiePath, $wgCookieDomain;
 
                if ( '' == $u->getEmail() ) {
                        return wfMsg( 'noemail', $u->getName() );
                }
-               $np = User::randomPassword();
+
+               $np = $u->randomPassword();
                $u->setNewpassword( $np );
 
                setcookie( "{$wgDBname}Token", '', time() - 3600, $wgCookiePath, $wgCookieDomain );
+
                $u->saveSettings();
 
                $ip = $wgIP;
@@ -312,20 +393,23 @@ class LoginForm {
 
                $m = wfMsg( 'passwordremindertext', $ip, $u->getName(), $np );
 
-               $error = userMailer( $u->getEmail(), $wgPasswordSender, wfMsg( 'passwordremindertitle' ), $m );
-               
-               return htmlspecialchars( $error );
+               $result = $u->sendMail( wfMsg( 'passwordremindertitle' ), $m );
+               return $result;
        }
 
 
        /**
+        * @param string $msg Message that will be shown on success.
         * @access private
         */
        function successfulLogin( $msg ) {
                global $wgUser;
-               global $wgDeferredUpdateList;
                global $wgOut;
 
+               # Run any hooks; ignore results
+               
+               wfRunHooks('UserLoginComplete', array(&$wgUser));
+               
                $wgOut->setPageTitle( wfMsg( 'loginsuccesstitle' ) );
                $wgOut->setRobotpolicy( 'noindex,nofollow' );
                $wgOut->setArticleRelated( false );
@@ -333,8 +417,9 @@ class LoginForm {
                $wgOut->returnToMain();
        }
 
+       /** */
        function userNotPrivilegedMessage() {
-               global $wgOut, $wgUser, $wgLang;
+               global $wgOut;
                
                $wgOut->setPageTitle( wfMsg( 'whitelistacctitle' ) );
                $wgOut->setRobotpolicy( 'noindex,nofollow' );
@@ -350,135 +435,46 @@ class LoginForm {
         */
        function mainLoginForm( $err ) {
                global $wgUser, $wgOut, $wgLang;
-               global $wgDBname, $wgAllowRealName;
-
-               $le = wfMsg( 'loginerror' );
-               $yn = wfMsg( 'yourname' );
-               $yp = wfMsg( 'yourpassword' );
-               $ypa = wfMsg( 'yourpasswordagain' );
-               $rmp = wfMsg( 'remembermypassword' );
-               $nuo = wfMsg( 'newusersonly' );
-               $li = wfMsg( 'login' );
-               $ca = wfMsg( 'createaccount' );
-               $cam = wfMsg( 'createaccountmail' );
-               $ye = wfMsg( 'youremail' );
-               if( $wgAllowRealName ) {
-                   $yrn = wfMsg( 'yourrealname' );
-               } else {
-                   $yrn = '';
-               }
-               $efl = wfMsg( 'emailforlost' );
-               $mmp = wfMsg( 'mailmypassword' );
-               $endText = wfMsg( 'loginend' );
-
-               if ( $endText == '<loginend>' ) {
-                       $endText = '';
-               }
+               global $wgDBname, $wgAllowRealName, $wgEnableEmail;
+               global $wgAuth;
 
                if ( '' == $this->mName ) {
-                       if ( 0 != $wgUser->getID() ) {
+                       if ( $wgUser->isLoggedIn() ) {
                                $this->mName = $wgUser->getName();
                        } else {
                                $this->mName = @$_COOKIE[$wgDBname.'UserName'];
                        }
                }
 
-               $wgOut->setPageTitle( wfMsg( 'userlogin' ) );
-               $wgOut->setRobotpolicy( 'noindex,nofollow' );
-               $wgOut->setArticleRelated( false );
-
-               if ( '' == $err ) {
-                       $lp = wfMsg( 'loginprompt' );
-                       $wgOut->addHTML( "<h2>$li:</h2>\n<p>$lp</p>" );
-               } else {
-                       $wgOut->addHTML( "<h2>$le:</h2>\n<font size='+1' 
-       color='red'>$err</font>\n" );
-               }
-               if ( 1 == $wgUser->getOption( 'rememberpassword' ) ) {
-                       $checked = ' checked';
-               } else {
-                       $checked = '';
-               }
-               
-               $q = 'action=submit';
+               $q = 'action=submitlogin';
                if ( !empty( $this->mReturnto ) ) {
                        $q .= '&returnto=' . wfUrlencode( $this->mReturnto );
                }
-               
                $titleObj = Title::makeTitle( NS_SPECIAL, 'Userlogin' );
-               $action = $titleObj->escapeLocalUrl( $q );
 
-               $encName = htmlspecialchars( $this->mName );
-               $encPassword = htmlspecialchars( $this->mPassword );
-               $encRetype = htmlspecialchars( $this->mRetype );
-               $encEmail = htmlspecialchars( $this->mEmail );
-               $encRealName = htmlspecialchars( $this->mRealName );
-
-               if ($wgUser->getID() != 0) {
-                       $cambutton = "<input tabindex='6' type='submit' name=\"wpCreateaccountMail\" value=\"{$cam}\" />";
-               } else {
-                       $cambutton = '';
-               }
-
-               $wgOut->addHTML( "
-       <form name=\"userlogin\" id=\"userlogin\" method=\"post\" action=\"{$action}\">
-       <table border='0'><tr>
-       <td align='right'>$yn:</td>
-       <td align='left'>
-       <input tabindex='1' type='text' name=\"wpName\" value=\"{$encName}\" size='20' />
-       </td>
-       <td align='left'>
-       <input tabindex='3' type='submit' name=\"wpLoginattempt\" value=\"{$li}\" />
-       </td>
-       </tr>
-       <tr>
-       <td align='right'>$yp:</td>
-       <td align='left'>
-       <input tabindex='2' type='password' name=\"wpPassword\" value=\"{$encPassword}\" size='20' />
-       </td>
-       <td align='left'>
-       <input tabindex='4' type='checkbox' name=\"wpRemember\" value=\"1\" id=\"wpRemember\"$checked /><label for=\"wpRemember\">$rmp</label>
-       </td>
-       </tr>");
-
-               if ($wgUser->isAllowedToCreateAccount()) {
-                       $encRetype = htmlspecialchars( $this->mRetype );
-                       $encEmail = htmlspecialchars( $this->mEmail );
-       $wgOut->addHTML("<tr><td colspan='3'>&nbsp;</td></tr><tr>
-       <td align='right'>$ypa:</td>
-       <td align='left'>
-       <input tabindex='5' type='password' name=\"wpRetype\" value=\"{$encRetype}\" 
-       size='20' />
-       </td><td>$nuo</td></tr>
-       <tr>
-       <td align='right'>$ye:</td>
-       <td align='left'>
-       <input tabindex='7' type='text' name=\"wpEmail\" value=\"{$encEmail}\" size='20' />
-       </td>");
-                   
-       if ($wgAllowRealName) {
-           $wgOut->addHTML("<td>&nbsp;</td>
-                             </tr><tr>
-                            <td align='right'>$yrn:</td>
-                            <td align='left'>
-                             <input tabindex='8' type='text' name=\"wpRealName\" value=\"{$encRealName}\" size='20' />
-                             </td>");
-       }
-                   
-       $wgOut->addHTML("<td align='left'>
-       <input tabindex='9' type='submit' name=\"wpCreateaccount\" value=\"{$ca}\" />
-       $cambutton
-       </td></tr>");
-               }
-           
-               $wgOut->addHTML("
-       <tr><td colspan='3'>&nbsp;</td></tr><tr>
-       <td colspan='3' align='left'>
-       <p>$efl<br />
-       <input tabindex='10' type='submit' name=\"wpMailmypassword\" value=\"{$mmp}\" /></p>
-       </td></tr></table>
-       </form>\n" );
-               $wgOut->addHTML( $endText );
+               require_once( 'templates/Userlogin.php' );
+               $template =& new UserloginTemplate();
+               
+               $template->set( 'name', $this->mName );
+               $template->set( 'password', $this->mPassword );
+               $template->set( 'retype', $this->mRetype );
+               $template->set( 'email', $this->mEmail );
+               $template->set( 'realname', $this->mRealName );
+               $template->set( 'domain', $this->mDomain );
+
+               $template->set( 'action', $titleObj->getLocalUrl( $q ) );
+               $template->set( 'error', $err );
+               $template->set( 'create', $wgUser->isAllowedToCreateAccount() );
+               $template->set( 'createemail', $wgEnableEmail && $wgUser->isLoggedIn() );
+               $template->set( 'userealname', $wgAllowRealName );
+               $template->set( 'useemail', $wgEnableEmail );
+               $template->set( 'remember', $wgUser->getOption( 'rememberpassword' ) or $this->mRemember  );
+               $wgAuth->modifyUITemplate( $template );
+               
+               $wgOut->setPageTitle( wfMsg( 'userlogin' ) );
+               $wgOut->setRobotpolicy( 'noindex,nofollow' );
+               $wgOut->setArticleRelated( false );
+               $wgOut->addTemplate( $template );
        }
 
        /**