MockFileSessionStorage.php 4.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160
  1. <?php
  2. /*
  3. * This file is part of the Symfony package.
  4. *
  5. * (c) Fabien Potencier <fabien@symfony.com>
  6. *
  7. * For the full copyright and license information, please view the LICENSE
  8. * file that was distributed with this source code.
  9. */
  10. namespace Symfony\Component\HttpFoundation\Session\Storage;
  11. /**
  12. * MockFileSessionStorage is used to mock sessions for
  13. * functional testing where you may need to persist session data
  14. * across separate PHP processes.
  15. *
  16. * No PHP session is actually started since a session can be initialized
  17. * and shutdown only once per PHP execution cycle and this class does
  18. * not pollute any session related globals, including session_*() functions
  19. * or session.* PHP ini directives.
  20. *
  21. * @author Drak <drak@zikula.org>
  22. */
  23. class MockFileSessionStorage extends MockArraySessionStorage
  24. {
  25. private $savePath;
  26. /**
  27. * @param string|null $savePath Path of directory to save session files
  28. */
  29. public function __construct(string $savePath = null, string $name = 'MOCKSESSID', MetadataBag $metaBag = null)
  30. {
  31. if (null === $savePath) {
  32. $savePath = sys_get_temp_dir();
  33. }
  34. if (!is_dir($savePath) && !@mkdir($savePath, 0777, true) && !is_dir($savePath)) {
  35. throw new \RuntimeException(sprintf('Session Storage was not able to create directory "%s".', $savePath));
  36. }
  37. $this->savePath = $savePath;
  38. parent::__construct($name, $metaBag);
  39. }
  40. /**
  41. * {@inheritdoc}
  42. */
  43. public function start()
  44. {
  45. if ($this->started) {
  46. return true;
  47. }
  48. if (!$this->id) {
  49. $this->id = $this->generateId();
  50. }
  51. $this->read();
  52. $this->started = true;
  53. return true;
  54. }
  55. /**
  56. * {@inheritdoc}
  57. */
  58. public function regenerate(bool $destroy = false, int $lifetime = null)
  59. {
  60. if (!$this->started) {
  61. $this->start();
  62. }
  63. if ($destroy) {
  64. $this->destroy();
  65. }
  66. return parent::regenerate($destroy, $lifetime);
  67. }
  68. /**
  69. * {@inheritdoc}
  70. */
  71. public function save()
  72. {
  73. if (!$this->started) {
  74. throw new \RuntimeException('Trying to save a session that was not started yet or was already closed.');
  75. }
  76. $data = $this->data;
  77. foreach ($this->bags as $bag) {
  78. if (empty($data[$key = $bag->getStorageKey()])) {
  79. unset($data[$key]);
  80. }
  81. }
  82. if ([$key = $this->metadataBag->getStorageKey()] === array_keys($data)) {
  83. unset($data[$key]);
  84. }
  85. try {
  86. if ($data) {
  87. $path = $this->getFilePath();
  88. $tmp = $path.bin2hex(random_bytes(6));
  89. file_put_contents($tmp, serialize($data));
  90. rename($tmp, $path);
  91. } else {
  92. $this->destroy();
  93. }
  94. } finally {
  95. $this->data = $data;
  96. }
  97. // this is needed when the session object is re-used across multiple requests
  98. // in functional tests.
  99. $this->started = false;
  100. }
  101. /**
  102. * Deletes a session from persistent storage.
  103. * Deliberately leaves session data in memory intact.
  104. */
  105. private function destroy(): void
  106. {
  107. set_error_handler(static function () {});
  108. try {
  109. unlink($this->getFilePath());
  110. } finally {
  111. restore_error_handler();
  112. }
  113. }
  114. /**
  115. * Calculate path to file.
  116. */
  117. private function getFilePath(): string
  118. {
  119. return $this->savePath.'/'.$this->id.'.mocksess';
  120. }
  121. /**
  122. * Reads session from storage and loads session.
  123. */
  124. private function read(): void
  125. {
  126. set_error_handler(static function () {});
  127. try {
  128. $data = file_get_contents($this->getFilePath());
  129. } finally {
  130. restore_error_handler();
  131. }
  132. $this->data = $data ? unserialize($data) : [];
  133. $this->loadSession();
  134. }
  135. }