Controller.php 13 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452
  1. <?php
  2. /**
  3. * @package Grav\Plugin\Login
  4. *
  5. * @copyright Copyright (C) 2014 - 2017 RocketTheme, LLC. All rights reserved.
  6. * @license MIT License; see LICENSE file for details.
  7. */
  8. namespace Grav\Plugin\Login;
  9. use Grav\Common\Grav;
  10. use Grav\Common\Language\Language;
  11. use Grav\Common\Uri;
  12. use Grav\Common\User\User;
  13. use Grav\Common\Utils;
  14. use Grav\Plugin\Email\Utils as EmailUtils;
  15. /**
  16. * Class Controller
  17. * @package Grav\Plugin\Login
  18. */
  19. class Controller
  20. {
  21. /**
  22. * @var Grav
  23. */
  24. public $grav;
  25. /**
  26. * @var string
  27. */
  28. public $action;
  29. /**
  30. * @var array
  31. */
  32. public $post;
  33. /**
  34. * @var string
  35. */
  36. protected $redirect;
  37. /**
  38. * @var int
  39. */
  40. protected $redirectCode;
  41. /**
  42. * @var string
  43. */
  44. protected $prefix = 'task';
  45. /**
  46. * @var RememberMe\RememberMe
  47. * @deprecated 2.0 Use $grav['login']->rememberMe() instead
  48. */
  49. protected $rememberMe;
  50. /**
  51. * @var Login
  52. */
  53. protected $login;
  54. /**
  55. * @param Grav $grav
  56. * @param string $action
  57. * @param array $post
  58. */
  59. public function __construct(Grav $grav, $action, $post = null)
  60. {
  61. $this->grav = $grav;
  62. $this->action = $action;
  63. $this->login = $this->grav['login'];
  64. $this->post = $post ? $this->getPost($post) : [];
  65. $this->rememberMe();
  66. }
  67. /**
  68. * Performs an action.
  69. * @throws \RuntimeException
  70. */
  71. public function execute()
  72. {
  73. $messages = $this->grav['messages'];
  74. // Set redirect if available.
  75. if (isset($this->post['_redirect'])) {
  76. $redirect = $this->post['_redirect'];
  77. unset($this->post['_redirect']);
  78. }
  79. $success = false;
  80. $method = $this->prefix . ucfirst($this->action);
  81. if (!method_exists($this, $method)) {
  82. throw new \RuntimeException('Page Not Found', 404);
  83. }
  84. try {
  85. $success = call_user_func([$this, $method]);
  86. } catch (\RuntimeException $e) {
  87. $messages->add($e->getMessage(), 'error');
  88. }
  89. if (!$this->redirect && isset($redirect)) {
  90. $this->setRedirect($redirect);
  91. }
  92. return $success;
  93. }
  94. /**
  95. * Handle login.
  96. *
  97. * @return bool True if the action was performed.
  98. */
  99. public function taskLogin()
  100. {
  101. /** @var Language $t */
  102. $t = $this->grav['language'];
  103. $messages = $this->grav['messages'];
  104. $userKey = isset($this->post['username']) ? (string)$this->post['username'] : '';
  105. $ipKey = Uri::ip();
  106. $rateLimiter = $this->login->getRateLimiter('login_attempts');
  107. // Check if the current IP has been used in failed login attempts.
  108. $attempts = count($rateLimiter->getAttempts($ipKey, 'ip'));
  109. $rateLimiter->registerRateLimitedAction($ipKey, 'ip')->registerRateLimitedAction($userKey);
  110. // Check rate limit for both IP and user, but allow each IP a single try even if user is already rate limited.
  111. if ($rateLimiter->isRateLimited($ipKey, 'ip') || ($attempts && $rateLimiter->isRateLimited($userKey))) {
  112. $messages->add($t->translate(['PLUGIN_LOGIN.TOO_MANY_LOGIN_ATTEMPTS', $rateLimiter->getInterval()]), 'error');
  113. $this->setRedirect($this->grav['config']->get('plugins.login.route', '/'));
  114. return true;
  115. }
  116. if ($this->authenticate($this->post)) {
  117. $rateLimiter->resetRateLimit($ipKey, 'ip')->resetRateLimit($userKey);
  118. $messages->add($t->translate('PLUGIN_LOGIN.LOGIN_SUCCESSFUL'), 'info');
  119. $redirect = $this->grav['config']->get('plugins.login.redirect_after_login');
  120. if (!$redirect) {
  121. $redirect = $this->grav['session']->redirect_after_login ?: $this->grav['uri']->referrer('/');
  122. }
  123. $this->setRedirect($redirect);
  124. } else {
  125. /** @var User $user */
  126. $user = $this->grav['user'];
  127. if ($user->username) {
  128. $messages->add($t->translate('PLUGIN_LOGIN.ACCESS_DENIED'), 'error');
  129. $this->setRedirect($this->grav['config']->get('plugins.login.route_unauthorized', '/'));
  130. } else {
  131. $messages->add($t->translate('PLUGIN_LOGIN.LOGIN_FAILED'), 'error');
  132. }
  133. }
  134. return true;
  135. }
  136. /**
  137. * Handle logout.
  138. *
  139. * @return bool True if the action was performed.
  140. */
  141. public function taskLogout()
  142. {
  143. $this->login->logout(['remember_me' => true]);
  144. $this->setRedirect('/');
  145. return true;
  146. }
  147. /**
  148. * Handle the email password recovery procedure.
  149. *
  150. * @return bool True if the action was performed.
  151. */
  152. protected function taskForgot()
  153. {
  154. $param_sep = $this->grav['config']->get('system.param_sep', ':');
  155. $data = $this->post;
  156. $email = isset($data['email']) ? $data['email'] : '';
  157. $user = !empty($email) ? User::find($email, ['email']) : null;
  158. /** @var Language $language */
  159. $language = $this->grav['language'];
  160. $messages = $this->grav['messages'];
  161. if (!isset($this->grav['Email'])) {
  162. $messages->add($language->translate('PLUGIN_LOGIN.FORGOT_EMAIL_NOT_CONFIGURED'), 'error');
  163. $this->setRedirect($this->grav['config']->get('plugins.login.route_forgot', '/'));
  164. return true;
  165. }
  166. if (!$user || !$user->exists()) {
  167. $messages->add($language->translate('PLUGIN_LOGIN.FORGOT_INSTRUCTIONS_SENT_VIA_EMAIL'), 'info');
  168. $this->setRedirect($this->grav['config']->get('plugins.login.route_forgot', '/'));
  169. return true;
  170. }
  171. if (empty($user->email)) {
  172. $messages->add($language->translate(['PLUGIN_LOGIN.FORGOT_CANNOT_RESET_EMAIL_NO_EMAIL', $email]),
  173. 'error');
  174. $this->setRedirect($this->grav['config']->get('plugins.login.route_forgot', '/'));
  175. return true;
  176. }
  177. $from = $this->grav['config']->get('plugins.email.from');
  178. if (empty($from)) {
  179. $messages->add($language->translate('PLUGIN_LOGIN.FORGOT_EMAIL_NOT_CONFIGURED'), 'error');
  180. $this->setRedirect($this->grav['config']->get('plugins.login.route_forgot', '/'));
  181. return true;
  182. }
  183. $userKey = $user->username;
  184. $rateLimiter = $this->login->getRateLimiter('pw_resets');
  185. $rateLimiter->registerRateLimitedAction($userKey);
  186. if ($rateLimiter->isRateLimited($userKey)) {
  187. $messages->add($language->translate(['PLUGIN_LOGIN.FORGOT_CANNOT_RESET_IT_IS_BLOCKED', $email, $rateLimiter->getInterval()]), 'error');
  188. $this->setRedirect($this->grav['config']->get('plugins.login.route', '/'));
  189. return true;
  190. }
  191. $token = md5(uniqid(mt_rand(), true));
  192. $expire = time() + 604800; // next week
  193. $user->reset = $token . '::' . $expire;
  194. $user->save();
  195. $author = $this->grav['config']->get('site.author.name', '');
  196. $fullname = $user->fullname ?: $user->username;
  197. $reset_link = $this->grav['base_url_absolute'] . $this->grav['config']->get('plugins.login.route_reset') . '/task:login.reset/token' . $param_sep . $token . '/user' . $param_sep . $user->username . '/nonce' . $param_sep . Utils::getNonce('reset-form');
  198. $sitename = $this->grav['config']->get('site.title', 'Website');
  199. $to = $user->email;
  200. $subject = $language->translate(['PLUGIN_LOGIN.FORGOT_EMAIL_SUBJECT', $sitename]);
  201. $content = $language->translate(['PLUGIN_LOGIN.FORGOT_EMAIL_BODY', $fullname, $reset_link, $author, $sitename]);
  202. $sent = EmailUtils::sendEmail($subject, $content, $to);
  203. if ($sent < 1) {
  204. $messages->add($language->translate('PLUGIN_LOGIN.FORGOT_FAILED_TO_EMAIL'), 'error');
  205. } else {
  206. $messages->add($language->translate('PLUGIN_LOGIN.FORGOT_INSTRUCTIONS_SENT_VIA_EMAIL'), 'info');
  207. }
  208. $this->setRedirect($this->grav['config']->get('plugins.login.route', '/'));
  209. return true;
  210. }
  211. /**
  212. * Handle the reset password action.
  213. *
  214. * @return bool True if the action was performed.
  215. * @throws \Exception
  216. */
  217. public function taskReset()
  218. {
  219. $data = $this->post;
  220. $language = $this->grav['language'];
  221. $messages = $this->grav['messages'];
  222. if (isset($data['password'])) {
  223. $username = isset($data['username']) ? $data['username'] : null;
  224. $user = !empty($username) ? User::find($username) : null;
  225. $password = isset($data['password']) ? $data['password'] : null;
  226. $token = isset($data['token']) ? $data['token'] : null;
  227. if ($user && !empty($user->reset) && $user->exists()) {
  228. list($good_token, $expire) = explode('::', $user->reset);
  229. if ($good_token === $token) {
  230. if (time() > $expire) {
  231. $messages->add($language->translate('PLUGIN_LOGIN.RESET_LINK_EXPIRED'), 'error');
  232. $this->grav->redirect($this->grav['config']->get('plugins.login.route_forgot', '/'));
  233. return true;
  234. }
  235. unset($user->hashed_password, $user->reset);
  236. $user->password = $password;
  237. $user->validate();
  238. $user->filter();
  239. $user->save();
  240. $messages->add($language->translate('PLUGIN_LOGIN.RESET_PASSWORD_RESET'), 'info');
  241. $this->setRedirect($this->grav['config']->get('plugins.login.route', '/'));
  242. return true;
  243. }
  244. }
  245. $messages->add($language->translate('PLUGIN_LOGIN.RESET_INVALID_LINK'), 'error');
  246. $this->grav->redirect($this->grav['config']->get('plugins.login.route_forgot'));
  247. return true;
  248. }
  249. $user = $this->grav['uri']->param('user');
  250. $token = $this->grav['uri']->param('token');
  251. if (!$user || !$token) {
  252. $messages->add($language->translate('PLUGIN_LOGIN.RESET_INVALID_LINK'), 'error');
  253. $this->grav->redirect($this->grav['config']->get('plugins.login.route_forgot'));
  254. return true;
  255. }
  256. return true;
  257. }
  258. /**
  259. * Authenticate user.
  260. *
  261. * @param array $form Form fields.
  262. *
  263. * @return bool
  264. */
  265. protected function authenticate($form)
  266. {
  267. // Remove login nonce.
  268. $form = array_diff_key($form, ['login-form-nonce' => true]);
  269. return $this->login->login($form, ['remember_me' => true])->authenticated;
  270. }
  271. /**
  272. * Redirects an action
  273. */
  274. public function redirect()
  275. {
  276. if ($this->redirect) {
  277. $this->grav->redirect($this->redirect, $this->redirectCode);
  278. }
  279. }
  280. /**
  281. * Set redirect.
  282. *
  283. * @param $path
  284. * @param int $code
  285. */
  286. public function setRedirect($path, $code = 303)
  287. {
  288. $this->redirect = $path;
  289. $this->redirectCode = $code;
  290. }
  291. /**
  292. * Prepare and return POST data.
  293. *
  294. * @param array $post
  295. *
  296. * @return array
  297. */
  298. protected function &getPost(array $post)
  299. {
  300. unset($post[$this->prefix]);
  301. // Decode JSON encoded fields and merge them to data.
  302. if (isset($post['_json'])) {
  303. $post = array_merge_recursive($post, $this->jsonDecode($post['_json']));
  304. unset($post['_json']);
  305. }
  306. return $post;
  307. }
  308. /**
  309. * Recursively JSON decode data.
  310. *
  311. * @param array $data
  312. *
  313. * @return array
  314. */
  315. protected function jsonDecode(array $data)
  316. {
  317. foreach ($data as &$value) {
  318. if (is_array($value)) {
  319. $value = $this->jsonDecode($value);
  320. } else {
  321. $value = json_decode($value, true);
  322. }
  323. }
  324. return $data;
  325. }
  326. /**
  327. * Gets and sets the RememberMe class
  328. *
  329. * @param mixed $var A rememberMe instance to set
  330. *
  331. * @return RememberMe\RememberMe Returns the current rememberMe instance
  332. * @deprecated 3.0 Use $grav['login']->rememberMe() instead
  333. */
  334. public function rememberMe($var = null)
  335. {
  336. $this->rememberMe = $this->login->rememberMe($var);
  337. return $this->rememberMe;
  338. }
  339. /**
  340. * Check if user may use password reset functionality.
  341. *
  342. * @param User $user
  343. * @param $field
  344. * @param $count
  345. * @param $interval
  346. * @return bool
  347. * @deprecated 3.0 Use $grav['login']->getRateLimiter($context) instead. See Grav\Plugin\Login\RateLimiter class.
  348. */
  349. protected function isUserRateLimited(User $user, $field, $count, $interval)
  350. {
  351. return $this->login->isUserRateLimited($user, $field, $count, $interval);
  352. }
  353. /**
  354. * Reset the rate limit counter
  355. *
  356. * @param User $user
  357. * @param $field
  358. * @deprecated 3.0 Use $grav['login']->getRateLimiter($context) instead. See Grav\Plugin\Login\RateLimiter class.
  359. */
  360. protected function resetRateLimit(User $user, $field)
  361. {
  362. $this->login->resetRateLimit($user, $field);
  363. }
  364. }