135 lines
4.5 KiB
PHP
135 lines
4.5 KiB
PHP
<?php
|
|
|
|
/*
|
|
* This file is part of the Symfony package.
|
|
*
|
|
* (c) Fabien Potencier <fabien@symfony.com>
|
|
*
|
|
* For the full copyright and license information, please view the LICENSE
|
|
* file that was distributed with this source code.
|
|
*/
|
|
|
|
namespace Symfony\Component\Security\Csrf;
|
|
|
|
use Symfony\Component\HttpFoundation\RequestStack;
|
|
use Symfony\Component\Security\Core\Exception\InvalidArgumentException;
|
|
use Symfony\Component\Security\Csrf\TokenGenerator\UriSafeTokenGenerator;
|
|
use Symfony\Component\Security\Csrf\TokenGenerator\TokenGeneratorInterface;
|
|
use Symfony\Component\Security\Csrf\TokenStorage\NativeSessionTokenStorage;
|
|
use Symfony\Component\Security\Csrf\TokenStorage\TokenStorageInterface;
|
|
|
|
/**
|
|
* Default implementation of {@link CsrfTokenManagerInterface}.
|
|
*
|
|
* @author Bernhard Schussek <bschussek@gmail.com>
|
|
* @author Kévin Dunglas <dunglas@gmail.com>
|
|
*/
|
|
class CsrfTokenManager implements CsrfTokenManagerInterface
|
|
{
|
|
/**
|
|
* @var TokenGeneratorInterface
|
|
*/
|
|
private $generator;
|
|
|
|
/**
|
|
* @var TokenStorageInterface
|
|
*/
|
|
private $storage;
|
|
private $namespace;
|
|
|
|
/**
|
|
* Creates a new CSRF provider using PHP's native session storage.
|
|
*
|
|
* @param null|string|RequestStack|callable $namespace
|
|
* * null: generates a namespace using $_SERVER['HTTPS']
|
|
* * string: uses the given string
|
|
* * RequestStack: generates a namespace using the current master request
|
|
* * callable: uses the result of this callable (must return a string)
|
|
*
|
|
* @param TokenGeneratorInterface|null $generator The token generator
|
|
* @param TokenStorageInterface|null $storage The storage for storing
|
|
* generated CSRF tokens
|
|
*/
|
|
public function __construct(TokenGeneratorInterface $generator = null, TokenStorageInterface $storage = null, $namespace = null)
|
|
{
|
|
$this->generator = $generator ?: new UriSafeTokenGenerator();
|
|
$this->storage = $storage ?: new NativeSessionTokenStorage();
|
|
|
|
$superGlobalNamespaceGenerator = function () {
|
|
return !empty($_SERVER['HTTPS']) && 'off' !== strtolower($_SERVER['HTTPS']) ? 'https-' : '';
|
|
};
|
|
|
|
if (null === $namespace) {
|
|
$this->namespace = $superGlobalNamespaceGenerator;
|
|
} elseif ($namespace instanceof RequestStack) {
|
|
$this->namespace = function () use ($namespace, $superGlobalNamespaceGenerator) {
|
|
if ($request = $namespace->getMasterRequest()) {
|
|
return $request->isSecure() ? 'https-' : '';
|
|
}
|
|
|
|
return $superGlobalNamespaceGenerator();
|
|
};
|
|
} elseif (is_callable($namespace) || is_string($namespace)) {
|
|
$this->namespace = $namespace;
|
|
} else {
|
|
throw new InvalidArgumentException(sprintf('$namespace must be a string, a callable returning a string, null or an instance of "RequestStack". "%s" given.', gettype($namespace)));
|
|
}
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function getToken($tokenId)
|
|
{
|
|
$namespacedId = $this->getNamespace().$tokenId;
|
|
if ($this->storage->hasToken($namespacedId)) {
|
|
$value = $this->storage->getToken($namespacedId);
|
|
} else {
|
|
$value = $this->generator->generateToken();
|
|
|
|
$this->storage->setToken($namespacedId, $value);
|
|
}
|
|
|
|
return new CsrfToken($tokenId, $value);
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function refreshToken($tokenId)
|
|
{
|
|
$namespacedId = $this->getNamespace().$tokenId;
|
|
$value = $this->generator->generateToken();
|
|
|
|
$this->storage->setToken($namespacedId, $value);
|
|
|
|
return new CsrfToken($tokenId, $value);
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function removeToken($tokenId)
|
|
{
|
|
return $this->storage->removeToken($this->getNamespace().$tokenId);
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function isTokenValid(CsrfToken $token)
|
|
{
|
|
$namespacedId = $this->getNamespace().$token->getId();
|
|
if (!$this->storage->hasToken($namespacedId)) {
|
|
return false;
|
|
}
|
|
|
|
return hash_equals($this->storage->getToken($namespacedId), $token->getValue());
|
|
}
|
|
|
|
private function getNamespace()
|
|
{
|
|
return is_callable($ns = $this->namespace) ? $ns() : $ns;
|
|
}
|
|
}
|