vendor/contao/core-bundle/src/Csrf/MemoryTokenStorage.php line 45

Open in your IDE?
  1. <?php
  3. declare(strict_types=1);
  5. /*
  6.  * This file is part of Contao.
  7.  *
  8.  * (c) Leo Feyer
  9.  *
  10.  * @license LGPL-3.0-or-later
  11.  */
  13. namespace Contao\CoreBundle\Csrf;
  15. use Symfony\Component\Security\Csrf\Exception\TokenNotFoundException;
  16. use Symfony\Component\Security\Csrf\TokenStorage\TokenStorageInterface;
  17. use Symfony\Contracts\Service\ResetInterface;
  19. class MemoryTokenStorage implements TokenStorageInterface, ResetInterface
  20. {
  21.     private ?array $tokens = null;
  22.     private array $usedTokens = [];
  24.     public function getToken(string $tokenId): string
  25.     {
  26.         $this->assertInitialized();
  28.         if (empty($this->tokens[$tokenId])) {
  29.             throw new TokenNotFoundException(sprintf('The CSRF token ID "%s" does not exist.', $tokenId));
  30.         }
  32.         $this->usedTokens[$tokenId] = true;
  34.         return $this->tokens[$tokenId];
  35.     }
  37.     public function setToken(string $tokenId, string $token): void
  38.     {
  39.         $this->assertInitialized();
  41.         $this->usedTokens[$tokenId] = true;
  42.         $this->tokens[$tokenId] = $token;
  43.     }
  45.     public function hasToken(string $tokenId): bool
  46.     {
  47.         $this->assertInitialized();
  49.         return !empty($this->tokens[$tokenId]);
  50.     }
  52.     public function removeToken(string $tokenId): ?string
  53.     {
  54.         $this->assertInitialized();
  56.         $token = null;
  58.         if (isset($this->tokens[$tokenId])) {
  59.             $token = $this->tokens[$tokenId];
  60.             $this->tokens[$tokenId] = null;
  61.         }
  63.         $this->usedTokens[$tokenId] = true;
  65.         return $token;
  66.     }
  68.     public function initialize(array $tokens): void
  69.     {
  70.         $this->tokens = $tokens;
  71.     }
  73.     public function getUsedTokens(): array
  74.     {
  75.         if (null === $this->tokens) {
  76.             return [];
  77.         }
  79.         return array_intersect_key($this->tokens, $this->usedTokens);
  80.     }
  82.     public function reset(): void
  83.     {
  84.         $this->tokens = null;
  85.         $this->usedTokens = [];
  86.     }
  88.     private function assertInitialized(): void
  89.     {
  90.         if (null === $this->tokens) {
  91.             throw new \LogicException('MemoryTokenStorage must not be accessed before it was initialized.');
  92.         }
  93.     }
  94. }