owncloud/core

View on GitHub
core/Controller/LostController.php

Summary

Maintainability
C
1 day
Test Coverage
<?php
/**
 * @author Bernhard Posselt <dev@bernhard-posselt.com>
 * @author Julius Haertl <jus@bitgrid.net>
 * @author Lukas Reschke <lukas@statuscode.ch>
 * @author Morris Jobke <hey@morrisjobke.de>
 * @author Peter Prochaska <info@peter-prochaska.de>
 * @author Thomas Müller <thomas.mueller@tmit.eu>
 * @author Ujjwal Bhardwaj <ujjwalb1996@gmail.com>
 * @author Victor Dubiniuk <dubiniuk@owncloud.com>
 *
 * @copyright Copyright (c) 2018, ownCloud GmbH
 * @license AGPL-3.0
 *
 * This code is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License, version 3,
 * as published by the Free Software Foundation.
 *
 * 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 Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License, version 3,
 * along with this program.  If not, see <http://www.gnu.org/licenses/>
 *
 */

namespace OC\Core\Controller;

use \OCP\AppFramework\Controller;
use \OCP\AppFramework\Http\TemplateResponse;
use OCP\AppFramework\Utility\ITimeFactory;
use OCP\ILogger;
use \OCP\IURLGenerator;
use \OCP\IRequest;
use \OCP\IL10N;
use \OCP\IConfig;
use OCP\IUser;
use OCP\IUserManager;
use OCP\Mail\IMailer;
use OCP\Security\ISecureRandom;
use \OC_Defaults;
use OC\User\Session;

/**
 * Class LostController
 *
 * Successfully changing a password will emit the post_passwordReset hook.
 *
 * @package OC\Core\Controller
 */
class LostController extends Controller {
    /** @var IURLGenerator */
    protected $urlGenerator;
    /** @var IUserManager */
    protected $userManager;
    // FIXME: Inject a non-static factory of OC_Defaults for better unit-testing
    /** @var OC_Defaults */
    protected $defaults;
    /** @var IL10N */
    protected $l10n;
    /** @var string */
    protected $from;
    /** @var bool */
    protected $isDataEncrypted;
    /** @var IConfig */
    protected $config;
    /** @var ISecureRandom */
    protected $secureRandom;
    /** @var IMailer */
    protected $mailer;
    /** @var ITimeFactory */
    protected $timeFactory;
    /** @var ILogger */
    protected $logger;
    /** @var Session */
    private $userSession;

    /**
     * @param string $appName
     * @param IRequest $request
     * @param IURLGenerator $urlGenerator
     * @param IUserManager $userManager
     * @param OC_Defaults $defaults
     * @param IL10N $l10n
     * @param IConfig $config
     * @param ISecureRandom $secureRandom
     * @param string $from
     * @param string $isDataEncrypted
     * @param IMailer $mailer
     * @param ITimeFactory $timeFactory
     * @param ILogger $logger
     * @param Session $userSession
     */
    public function __construct(
        $appName,
        IRequest $request,
        IURLGenerator $urlGenerator,
        IUserManager $userManager,
        OC_Defaults $defaults,
        IL10N $l10n,
        IConfig $config,
        ISecureRandom $secureRandom,
        $from,
        $isDataEncrypted,
        IMailer $mailer,
        ITimeFactory $timeFactory,
        ILogger $logger,
        Session $userSession
    ) {
        parent::__construct($appName, $request);
        $this->urlGenerator = $urlGenerator;
        $this->userManager = $userManager;
        $this->defaults = $defaults;
        $this->l10n = $l10n;
        $this->secureRandom = $secureRandom;
        $this->from = $from;
        $this->isDataEncrypted = $isDataEncrypted;
        $this->config = $config;
        $this->mailer = $mailer;
        $this->timeFactory = $timeFactory;
        $this->logger = $logger;
        $this->userSession = $userSession;
    }

    /**
     * Someone wants to reset their password:
     *
     * @PublicPage
     * @NoCSRFRequired
     *
     * @param string $token
     * @param string $userId
     * @return TemplateResponse
     */
    public function resetform($token, $userId) {
        try {
            $this->checkPasswordResetToken($token, $userId);
        } catch (\Exception $e) {
            return new TemplateResponse(
                'core',
                'error',
                [
                    "errors" => [["error" => $e->getMessage()]]
                ],
                'guest'
            );
        }

        return new TemplateResponse(
            'core',
            'lostpassword/resetpassword',
            [
                'link' => $this->urlGenerator->linkToRouteAbsolute('core.lost.setPassword', ['userId' => $userId, 'token' => $token]),
            ],
            'guest'
        );
    }

    /**
     * @param string $token
     * @param string $userId
     * @throws \Exception
     */
    private function checkPasswordResetToken($token, $userId) {
        $user = $this->userManager->get($userId);

        $splittedToken = \explode(':', $this->config->getUserValue($userId, 'owncloud', 'lostpassword', null));
        if (\count($splittedToken) !== 2) {
            $this->config->deleteUserValue($userId, 'owncloud', 'lostpassword');
            throw new \Exception($this->l10n->t('Could not reset password because the token is invalid'));
        }

        if ($splittedToken[0] < ($this->timeFactory->getTime() - 60*60*12) ||
            $user->getLastLogin() > $splittedToken[0]) {
            $this->config->deleteUserValue($userId, 'owncloud', 'lostpassword');
            throw new \Exception($this->l10n->t('Could not reset password because the token expired'));
        }

        if (!\hash_equals($splittedToken[1], $token)) {
            $this->config->deleteUserValue($userId, 'owncloud', 'lostpassword');
            throw new \Exception($this->l10n->t('Could not reset password because the token does not match'));
        }
    }

    /**
     * @param $message
     * @param array $additional
     * @return array
     */
    private function error($message, array $additional= []) {
        return \array_merge(['status' => 'error', 'msg' => \htmlspecialchars($message)], $additional);
    }

    /**
     * @return array
     */
    private function success() {
        return ['status'=>'success'];
    }

    /**
     * @PublicPage
     *
     * @param string $user
     * @return array
     */
    public function email($user) {
        try {
            $userObject = $this->userManager->get($user);
            if ($userObject === null) {
                $users = $this->userManager->getByEmail($user);
                // we only allow login by email if unique
                if (\count($users) === 1) {
                    $user = $users[0]->getUID();
                }
            }
            if ($userObject instanceof IUser) {
                $user = $userObject->getUID();
            }
        } catch (\Exception $e) {
            // Do not disclose any information during User lookup
            return $this->success();
        }
        // FIXME: use HTTP error codes
        try {
            list($link, $token) = $this->generateTokenAndLink($user);
            $this->sendEmail($user, $token, $link);
        } catch (\Exception $e) {
            return $this->error($e->getMessage());
        }

        return $this->success();
    }

    /**
     * @PublicPage
     * @param string $token
     * @param string $userId
     * @param string $password
     * @param boolean $proceed
     * @return array
     */
    public function setPassword($token, $userId, $password, $proceed) {
        if ($this->isDataEncrypted && !$proceed) {
            return $this->error('', ['encryption' => true]);
        }

        try {
            $this->checkPasswordResetToken($token, $userId);
            $user = $this->userManager->get($userId);

            if (!$user->setPassword($password)) {
                throw new \Exception();
            }

            \OC_Hook::emit('\OC\Core\LostPassword\Controller\LostController', 'post_passwordReset', ['uid' => $userId, 'password' => $password]);
            @\OC_User::unsetMagicInCookie();
        } catch (\Exception $e) {
            return $this->error($e->getMessage());
        }

        try {
            $this->sendNotificationMail($userId);
        } catch (\Exception $e) {
            return $this->error($e->getMessage());
        }

        $this->logout();

        return $this->success();
    }

    /**
     * @param string $userId
     * @throws \Exception
     */
    protected function sendNotificationMail($userId) {
        $user = $this->userManager->get($userId);
        $email = $user->getEMailAddress();
        $name = $user->getDisplayName();

        if ($email !== '') {
            $tmpl = new \OC_Template('core', 'lostpassword/notify');
            $msg = $tmpl->fetchPage();
            $tmplAlt = new \OC_Template('core', 'lostpassword/altnotify');
            $msgAlt = $tmplAlt->fetchPage();

            try {
                $message = $this->mailer->createMessage();
                $message->setTo([$email => $name]);
                $message->setSubject($this->l10n->t('%s password changed successfully', [$this->defaults->getName()]));
                $message->setPlainBody($msgAlt);
                $message->setHtmlBody($msg);
                $message->setFrom([$this->from => $this->defaults->getName()]);
                $this->mailer->send($message);
            } catch (\Exception $e) {
                throw new \Exception($this->l10n->t(
                    $e->getMessage()
                ));
            }
        }
    }

    /**
     * @param string $userId
     * @return array
     */
    public function generateTokenAndLink($userId) {
        $token = $this->secureRandom->generate(
            21,
            ISecureRandom::CHAR_DIGITS .
            ISecureRandom::CHAR_LOWER .
            ISecureRandom::CHAR_UPPER
        );

        $link = $this->urlGenerator->linkToRouteAbsolute('core.lost.resetform', ['userId' => $userId, 'token' => $token]);
        return [$link, $token];
    }

    /**
     * @param string $user
     * @param string $token
     * @param string $link
     * @throws \Exception
     * @return boolean
     */
    public function sendEmail($user, $token, $link) {
        if ($this->userManager->userExists($user)) {
            $userObject = $this->userManager->get($user);
            $email = $userObject->getEMailAddress();

            if (empty($email)) {
                $this->logger->error('Could not send reset email because there is no email address for this username. User: {user}', ['app' => 'core', 'user' => $user]);
                return false;
            }
        } else {
            $users = $this->userManager->getByEmail($user);

            switch (\count($users)) {
                case 0:
                    $this->logger->error('Could not send reset email because User does not exist. User: {user}', ['app' => 'core', 'user' => $user]);
                    return false;
                case 1:
                    $this->logger->info('User with input as email address found. User: {user}', ['app' => 'core', 'user' => $user]);
                    $userObject = $users[0];
                    $email = $userObject->getEMailAddress();
                    $user = $userObject->getUID();
                    break;
                default:
                    $this->logger->error('Could not send reset email because the email id is not unique. User: {user}', ['app' => 'core', 'user' => $user]);
                    return false;
            }
        }

        $getToken = $this->config->getUserValue($user, 'owncloud', 'lostpassword');
        if ($getToken !== '') {
            $splittedToken = \explode(':', $getToken);
            if ((\count($splittedToken)) === 2 && $splittedToken[0] > ($this->timeFactory->getTime() - 60 * 5)) {
                $this->logger->alert('The email is not sent because a password reset email was sent recently.');
                return false;
            }
        }

        $this->config->setUserValue($user, 'owncloud', 'lostpassword', $this->timeFactory->getTime() . ':' . $token);

        $tmpl = new \OC_Template('core', 'lostpassword/email');
        $tmpl->assign('link', $link);
        $msg = $tmpl->fetchPage();
        $tmplAlt = new \OC_Template('core', 'lostpassword/altemail');
        $tmplAlt->assign('link', $link);
        $msgAlt = $tmplAlt->fetchPage();
        $name = $userObject->getDisplayName();

        try {
            $message = $this->mailer->createMessage();
            $message->setTo([$email => $name]);
            $message->setSubject($this->l10n->t('%s password reset', [$this->defaults->getName()]));
            $message->setPlainBody($msgAlt);
            $message->setHtmlBody($msg);
            $message->setFrom([$this->from => $this->defaults->getName()]);
            $this->mailer->send($message);
        } catch (\Exception $e) {
            throw new \Exception($this->l10n->t(
                'Couldn\'t send reset email. Please contact your administrator.'
            ));
        }

        return true;
    }

    private function logout() {
        $loginToken = $this->request->getCookie('oc_token');
        if ($loginToken !== null) {
            $this->config->deleteUserValue($this->userSession->getUser()->getUID(), 'login_token', $loginToken);
        }
        $this->userSession->logout();
    }
}