--- /dev/null
+<?php
+
+/*
+ * This file is part of the Symfony package.
+ *
+ * (c) Fabien Potencier <fabien@symfony.com>
+ *
+ * This code is partially based on the Rack-Cache library by Ryan Tomayko,
+ * which is released under the MIT license.
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Symfony\Component\HttpKernel\HttpCache;
+
+use Symfony\Component\HttpFoundation\Request;
+use Symfony\Component\HttpFoundation\Response;
+
+/**
+ * Store implements all the logic for storing cache metadata (Request and Response headers).
+ *
+ * @author Fabien Potencier <fabien@symfony.com>
+ */
+class Store implements StoreInterface
+{
+ protected $root;
+ private $keyCache;
+ private $locks;
+
+ /**
+ * Constructor.
+ *
+ * @param string $root The path to the cache directory
+ *
+ * @throws \RuntimeException
+ */
+ public function __construct($root)
+ {
+ $this->root = $root;
+ if (!file_exists($this->root) && !@mkdir($this->root, 0777, true) && !is_dir($this->root)) {
+ throw new \RuntimeException(sprintf('Unable to create the store directory (%s).', $this->root));
+ }
+ $this->keyCache = new \SplObjectStorage();
+ $this->locks = array();
+ }
+
+ /**
+ * Cleanups storage.
+ */
+ public function cleanup()
+ {
+ // unlock everything
+ foreach ($this->locks as $lock) {
+ flock($lock, LOCK_UN);
+ fclose($lock);
+ }
+
+ $this->locks = array();
+ }
+
+ /**
+ * Tries to lock the cache for a given Request, without blocking.
+ *
+ * @param Request $request A Request instance
+ *
+ * @return bool|string true if the lock is acquired, the path to the current lock otherwise
+ */
+ public function lock(Request $request)
+ {
+ $key = $this->getCacheKey($request);
+
+ if (!isset($this->locks[$key])) {
+ $path = $this->getPath($key);
+ if (!file_exists(dirname($path)) && false === @mkdir(dirname($path), 0777, true) && !is_dir(dirname($path))) {
+ return $path;
+ }
+ $h = fopen($path, 'cb');
+ if (!flock($h, LOCK_EX | LOCK_NB)) {
+ fclose($h);
+
+ return $path;
+ }
+
+ $this->locks[$key] = $h;
+ }
+
+ return true;
+ }
+
+ /**
+ * Releases the lock for the given Request.
+ *
+ * @param Request $request A Request instance
+ *
+ * @return bool False if the lock file does not exist or cannot be unlocked, true otherwise
+ */
+ public function unlock(Request $request)
+ {
+ $key = $this->getCacheKey($request);
+
+ if (isset($this->locks[$key])) {
+ flock($this->locks[$key], LOCK_UN);
+ fclose($this->locks[$key]);
+ unset($this->locks[$key]);
+
+ return true;
+ }
+
+ return false;
+ }
+
+ public function isLocked(Request $request)
+ {
+ $key = $this->getCacheKey($request);
+
+ if (isset($this->locks[$key])) {
+ return true; // shortcut if lock held by this process
+ }
+
+ if (!file_exists($path = $this->getPath($key))) {
+ return false;
+ }
+
+ $h = fopen($path, 'rb');
+ flock($h, LOCK_EX | LOCK_NB, $wouldBlock);
+ flock($h, LOCK_UN); // release the lock we just acquired
+ fclose($h);
+
+ return (bool) $wouldBlock;
+ }
+
+ /**
+ * Locates a cached Response for the Request provided.
+ *
+ * @param Request $request A Request instance
+ *
+ * @return Response|null A Response instance, or null if no cache entry was found
+ */
+ public function lookup(Request $request)
+ {
+ $key = $this->getCacheKey($request);
+
+ if (!$entries = $this->getMetadata($key)) {
+ return;
+ }
+
+ // find a cached entry that matches the request.
+ $match = null;
+ foreach ($entries as $entry) {
+ if ($this->requestsMatch(isset($entry[1]['vary'][0]) ? implode(', ', $entry[1]['vary']) : '', $request->headers->all(), $entry[0])) {
+ $match = $entry;
+
+ break;
+ }
+ }
+
+ if (null === $match) {
+ return;
+ }
+
+ list($req, $headers) = $match;
+ if (file_exists($body = $this->getPath($headers['x-content-digest'][0]))) {
+ return $this->restoreResponse($headers, $body);
+ }
+
+ // TODO the metaStore referenced an entity that doesn't exist in
+ // the entityStore. We definitely want to return nil but we should
+ // also purge the entry from the meta-store when this is detected.
+ }
+
+ /**
+ * Writes a cache entry to the store for the given Request and Response.
+ *
+ * Existing entries are read and any that match the response are removed. This
+ * method calls write with the new list of cache entries.
+ *
+ * @param Request $request A Request instance
+ * @param Response $response A Response instance
+ *
+ * @return string The key under which the response is stored
+ *
+ * @throws \RuntimeException
+ */
+ public function write(Request $request, Response $response)
+ {
+ $key = $this->getCacheKey($request);
+ $storedEnv = $this->persistRequest($request);
+
+ // write the response body to the entity store if this is the original response
+ if (!$response->headers->has('X-Content-Digest')) {
+ $digest = $this->generateContentDigest($response);
+
+ if (false === $this->save($digest, $response->getContent())) {
+ throw new \RuntimeException('Unable to store the entity.');
+ }
+
+ $response->headers->set('X-Content-Digest', $digest);
+
+ if (!$response->headers->has('Transfer-Encoding')) {
+ $response->headers->set('Content-Length', strlen($response->getContent()));
+ }
+ }
+
+ // read existing cache entries, remove non-varying, and add this one to the list
+ $entries = array();
+ $vary = $response->headers->get('vary');
+ foreach ($this->getMetadata($key) as $entry) {
+ if (!isset($entry[1]['vary'][0])) {
+ $entry[1]['vary'] = array('');
+ }
+
+ if ($vary != $entry[1]['vary'][0] || !$this->requestsMatch($vary, $entry[0], $storedEnv)) {
+ $entries[] = $entry;
+ }
+ }
+
+ $headers = $this->persistResponse($response);
+ unset($headers['age']);
+
+ array_unshift($entries, array($storedEnv, $headers));
+
+ if (false === $this->save($key, serialize($entries))) {
+ throw new \RuntimeException('Unable to store the metadata.');
+ }
+
+ return $key;
+ }
+
+ /**
+ * Returns content digest for $response.
+ *
+ * @param Response $response
+ *
+ * @return string
+ */
+ protected function generateContentDigest(Response $response)
+ {
+ return 'en'.hash('sha256', $response->getContent());
+ }
+
+ /**
+ * Invalidates all cache entries that match the request.
+ *
+ * @param Request $request A Request instance
+ *
+ * @throws \RuntimeException
+ */
+ public function invalidate(Request $request)
+ {
+ $modified = false;
+ $key = $this->getCacheKey($request);
+
+ $entries = array();
+ foreach ($this->getMetadata($key) as $entry) {
+ $response = $this->restoreResponse($entry[1]);
+ if ($response->isFresh()) {
+ $response->expire();
+ $modified = true;
+ $entries[] = array($entry[0], $this->persistResponse($response));
+ } else {
+ $entries[] = $entry;
+ }
+ }
+
+ if ($modified && false === $this->save($key, serialize($entries))) {
+ throw new \RuntimeException('Unable to store the metadata.');
+ }
+ }
+
+ /**
+ * Determines whether two Request HTTP header sets are non-varying based on
+ * the vary response header value provided.
+ *
+ * @param string $vary A Response vary header
+ * @param array $env1 A Request HTTP header array
+ * @param array $env2 A Request HTTP header array
+ *
+ * @return bool true if the two environments match, false otherwise
+ */
+ private function requestsMatch($vary, $env1, $env2)
+ {
+ if (empty($vary)) {
+ return true;
+ }
+
+ foreach (preg_split('/[\s,]+/', $vary) as $header) {
+ $key = str_replace('_', '-', strtolower($header));
+ $v1 = isset($env1[$key]) ? $env1[$key] : null;
+ $v2 = isset($env2[$key]) ? $env2[$key] : null;
+ if ($v1 !== $v2) {
+ return false;
+ }
+ }
+
+ return true;
+ }
+
+ /**
+ * Gets all data associated with the given key.
+ *
+ * Use this method only if you know what you are doing.
+ *
+ * @param string $key The store key
+ *
+ * @return array An array of data associated with the key
+ */
+ private function getMetadata($key)
+ {
+ if (!$entries = $this->load($key)) {
+ return array();
+ }
+
+ return unserialize($entries);
+ }
+
+ /**
+ * Purges data for the given URL.
+ *
+ * This method purges both the HTTP and the HTTPS version of the cache entry.
+ *
+ * @param string $url A URL
+ *
+ * @return bool true if the URL exists with either HTTP or HTTPS scheme and has been purged, false otherwise
+ */
+ public function purge($url)
+ {
+ $http = preg_replace('#^https:#', 'http:', $url);
+ $https = preg_replace('#^http:#', 'https:', $url);
+
+ $purgedHttp = $this->doPurge($http);
+ $purgedHttps = $this->doPurge($https);
+
+ return $purgedHttp || $purgedHttps;
+ }
+
+ /**
+ * Purges data for the given URL.
+ *
+ * @param string $url A URL
+ *
+ * @return bool true if the URL exists and has been purged, false otherwise
+ */
+ private function doPurge($url)
+ {
+ $key = $this->getCacheKey(Request::create($url));
+ if (isset($this->locks[$key])) {
+ flock($this->locks[$key], LOCK_UN);
+ fclose($this->locks[$key]);
+ unset($this->locks[$key]);
+ }
+
+ if (file_exists($path = $this->getPath($key))) {
+ unlink($path);
+
+ return true;
+ }
+
+ return false;
+ }
+
+ /**
+ * Loads data for the given key.
+ *
+ * @param string $key The store key
+ *
+ * @return string The data associated with the key
+ */
+ private function load($key)
+ {
+ $path = $this->getPath($key);
+
+ return file_exists($path) ? file_get_contents($path) : false;
+ }
+
+ /**
+ * Save data for the given key.
+ *
+ * @param string $key The store key
+ * @param string $data The data to store
+ *
+ * @return bool
+ */
+ private function save($key, $data)
+ {
+ $path = $this->getPath($key);
+
+ if (isset($this->locks[$key])) {
+ $fp = $this->locks[$key];
+ @ftruncate($fp, 0);
+ @fseek($fp, 0);
+ $len = @fwrite($fp, $data);
+ if (strlen($data) !== $len) {
+ @ftruncate($fp, 0);
+
+ return false;
+ }
+ } else {
+ if (!file_exists(dirname($path)) && false === @mkdir(dirname($path), 0777, true) && !is_dir(dirname($path))) {
+ return false;
+ }
+
+ $tmpFile = tempnam(dirname($path), basename($path));
+ if (false === $fp = @fopen($tmpFile, 'wb')) {
+ return false;
+ }
+ @fwrite($fp, $data);
+ @fclose($fp);
+
+ if ($data != file_get_contents($tmpFile)) {
+ return false;
+ }
+
+ if (false === @rename($tmpFile, $path)) {
+ return false;
+ }
+ }
+
+ @chmod($path, 0666 & ~umask());
+ }
+
+ public function getPath($key)
+ {
+ return $this->root.DIRECTORY_SEPARATOR.substr($key, 0, 2).DIRECTORY_SEPARATOR.substr($key, 2, 2).DIRECTORY_SEPARATOR.substr($key, 4, 2).DIRECTORY_SEPARATOR.substr($key, 6);
+ }
+
+ /**
+ * Generates a cache key for the given Request.
+ *
+ * This method should return a key that must only depend on a
+ * normalized version of the request URI.
+ *
+ * If the same URI can have more than one representation, based on some
+ * headers, use a Vary header to indicate them, and each representation will
+ * be stored independently under the same cache key.
+ *
+ * @param Request $request A Request instance
+ *
+ * @return string A key for the given Request
+ */
+ protected function generateCacheKey(Request $request)
+ {
+ return 'md'.hash('sha256', $request->getUri());
+ }
+
+ /**
+ * Returns a cache key for the given Request.
+ *
+ * @param Request $request A Request instance
+ *
+ * @return string A key for the given Request
+ */
+ private function getCacheKey(Request $request)
+ {
+ if (isset($this->keyCache[$request])) {
+ return $this->keyCache[$request];
+ }
+
+ return $this->keyCache[$request] = $this->generateCacheKey($request);
+ }
+
+ /**
+ * Persists the Request HTTP headers.
+ *
+ * @param Request $request A Request instance
+ *
+ * @return array An array of HTTP headers
+ */
+ private function persistRequest(Request $request)
+ {
+ return $request->headers->all();
+ }
+
+ /**
+ * Persists the Response HTTP headers.
+ *
+ * @param Response $response A Response instance
+ *
+ * @return array An array of HTTP headers
+ */
+ private function persistResponse(Response $response)
+ {
+ $headers = $response->headers->all();
+ $headers['X-Status'] = array($response->getStatusCode());
+
+ return $headers;
+ }
+
+ /**
+ * Restores a Response from the HTTP headers and body.
+ *
+ * @param array $headers An array of HTTP headers for the Response
+ * @param string $body The Response body
+ *
+ * @return Response
+ */
+ private function restoreResponse($headers, $body = null)
+ {
+ $status = $headers['X-Status'][0];
+ unset($headers['X-Status']);
+
+ if (null !== $body) {
+ $headers['X-Body-File'] = array($body);
+ }
+
+ return new Response($body, $status, $headers);
+ }
+}