move verification token logic out of lost password controller

- to make it reusable
- needed for local email verification

Signed-off-by: Arthur Schiwon <blizzz@arthur-schiwon.de>

1 files changed, 111 insertions, 0 deletions

diff --git a/lib/private/Security/VerificationToken/VerificationToken.php b/lib/private/Security/VerificationToken/VerificationToken.php
new file mode 100644
index 00000000000..4ac5605eecf
--- /dev/null
+++ b/lib/private/Security/VerificationToken/VerificationToken.php
@@ -0,0 +1,111 @@
+<?php
+
+declare(strict_types=1);
+
+/**
+ * @copyright Copyright (c) 2021 Arthur Schiwon <blizzz@arthur-schiwon.de>
+ *
+ * @author Arthur Schiwon <blizzz@arthur-schiwon.de>
+ *
+ * @license GNU AGPL version 3 or any later version
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as
+ * published by the Free Software Foundation, either version 3 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 Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <https://www.gnu.org/licenses/>.
+ *
+ */
+
+namespace OC\Security\VerificationToken;
+
+use OCP\AppFramework\Utility\ITimeFactory;
+use OCP\IConfig;
+use OCP\IUser;
+use OCP\Security\ICrypto;
+use OCP\Security\ISecureRandom;
+use OCP\Security\VerificationToken\InvalidTokenException;
+use OCP\Security\VerificationToken\IVerificationToken;
+
+class VerificationToken implements IVerificationToken {
+
+	/** @var IConfig */
+	private $config;
+	/** @var ICrypto */
+	private $crypto;
+	/** @var ITimeFactory */
+	private $timeFactory;
+	/** @var ISecureRandom */
+	private $secureRandom;
+
+	public function __construct(
+		IConfig $config,
+		ICrypto $crypto,
+		ITimeFactory $timeFactory,
+		ISecureRandom $secureRandom
+	) {
+		$this->config = $config;
+		$this->crypto = $crypto;
+		$this->timeFactory = $timeFactory;
+		$this->secureRandom = $secureRandom;
+	}
+
+	/**
+	 * @throws InvalidTokenException
+	 */
+	protected function throwInvalidTokenException(int $code): void {
+		throw new InvalidTokenException($code);
+	}
+
+	public function check(string $token, ?IUser $user, string $subject, string $passwordPrefix = ''): void {
+		if ($user === null || !$user->isEnabled()) {
+			$this->throwInvalidTokenException(InvalidTokenException::USER_UNKNOWN);
+		}
+
+		$encryptedToken = $this->config->getUserValue($user->getUID(), 'core', $subject, null);
+		if ($encryptedToken === null) {
+			$this->throwInvalidTokenException(InvalidTokenException::TOKEN_NOT_FOUND);
+		}
+
+		try {
+			$decryptedToken = $this->crypto->decrypt($encryptedToken, $passwordPrefix.$this->config->getSystemValue('secret'));
+		} catch (\Exception $e) {
+			$this->throwInvalidTokenException(InvalidTokenException::TOKEN_DECRYPTION_ERROR);
+		}
+
+		$splitToken = explode(':', $decryptedToken ?? '');
+		if (count($splitToken) !== 2) {
+			$this->throwInvalidTokenException(InvalidTokenException::TOKEN_INVALID_FORMAT);
+		}
+
+		if ($splitToken[0] < ($this->timeFactory->getTime() - 60 * 60 * 24 * 7) ||
+			$user->getLastLogin() > $splitToken[0]) {
+			$this->throwInvalidTokenException(InvalidTokenException::TOKEN_EXPIRED);
+		}
+
+		if (!hash_equals($splitToken[1], $token)) {
+			$this->throwInvalidTokenException(InvalidTokenException::TOKEN_MISMATCH);
+		}
+	}
+
+	public function create(IUser $user, string $subject, string $passwordPrefix = ''): string {
+		$token = $this->secureRandom->generate(
+			21,
+			ISecureRandom::CHAR_DIGITS.
+			ISecureRandom::CHAR_LOWER.
+			ISecureRandom::CHAR_UPPER
+		);
+		$tokenValue = $this->timeFactory->getTime() .':'. $token;
+		$encryptedValue = $this->crypto->encrypt($tokenValue, $passwordPrefix . $this->config->getSystemValue('secret'));
+		$this->config->setUserValue($user->getUID(), 'core', $subject, $encryptedValue);
+
+		return $token;
+	}
+}