118 lines
2.6 KiB
PHP
118 lines
2.6 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\TokenStorage;
|
|
|
|
use Symfony\Component\Security\Csrf\Exception\TokenNotFoundException;
|
|
|
|
/**
|
|
* Token storage that uses PHP's native session handling.
|
|
*
|
|
* @author Bernhard Schussek <bschussek@gmail.com>
|
|
*/
|
|
class NativeSessionTokenStorage implements TokenStorageInterface
|
|
{
|
|
/**
|
|
* The namespace used to store values in the session.
|
|
*
|
|
* @var string
|
|
*/
|
|
const SESSION_NAMESPACE = '_csrf';
|
|
|
|
/**
|
|
* @var bool
|
|
*/
|
|
private $sessionStarted = false;
|
|
|
|
/**
|
|
* @var string
|
|
*/
|
|
private $namespace;
|
|
|
|
/**
|
|
* Initializes the storage with a session namespace.
|
|
*
|
|
* @param string $namespace The namespace under which the token is stored
|
|
* in the session
|
|
*/
|
|
public function __construct($namespace = self::SESSION_NAMESPACE)
|
|
{
|
|
$this->namespace = $namespace;
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function getToken($tokenId)
|
|
{
|
|
if (!$this->sessionStarted) {
|
|
$this->startSession();
|
|
}
|
|
|
|
if (!isset($_SESSION[$this->namespace][$tokenId])) {
|
|
throw new TokenNotFoundException('The CSRF token with ID '.$tokenId.' does not exist.');
|
|
}
|
|
|
|
return (string) $_SESSION[$this->namespace][$tokenId];
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function setToken($tokenId, $token)
|
|
{
|
|
if (!$this->sessionStarted) {
|
|
$this->startSession();
|
|
}
|
|
|
|
$_SESSION[$this->namespace][$tokenId] = (string) $token;
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function hasToken($tokenId)
|
|
{
|
|
if (!$this->sessionStarted) {
|
|
$this->startSession();
|
|
}
|
|
|
|
return isset($_SESSION[$this->namespace][$tokenId]);
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function removeToken($tokenId)
|
|
{
|
|
if (!$this->sessionStarted) {
|
|
$this->startSession();
|
|
}
|
|
|
|
$token = isset($_SESSION[$this->namespace][$tokenId])
|
|
? (string) $_SESSION[$this->namespace][$tokenId]
|
|
: null;
|
|
|
|
unset($_SESSION[$this->namespace][$tokenId]);
|
|
|
|
return $token;
|
|
}
|
|
|
|
private function startSession()
|
|
{
|
|
if (PHP_SESSION_NONE === session_status()) {
|
|
session_start();
|
|
}
|
|
|
|
$this->sessionStarted = true;
|
|
}
|
|
}
|