Store.php 13 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460
  1. <?php
  2. /*
  3. * This file is part of the Symfony package.
  4. *
  5. * (c) Fabien Potencier <fabien@symfony.com>
  6. *
  7. * This code is partially based on the Rack-Cache library by Ryan Tomayko,
  8. * which is released under the MIT license.
  9. *
  10. * For the full copyright and license information, please view the LICENSE
  11. * file that was distributed with this source code.
  12. */
  13. namespace Symfony\Component\HttpKernel\HttpCache;
  14. use Symfony\Component\HttpFoundation\Request;
  15. use Symfony\Component\HttpFoundation\Response;
  16. /**
  17. * Store implements all the logic for storing cache metadata (Request and Response headers).
  18. *
  19. * @author Fabien Potencier <fabien@symfony.com>
  20. */
  21. class Store implements StoreInterface
  22. {
  23. protected $root;
  24. private $keyCache;
  25. private $locks;
  26. /**
  27. * @throws \RuntimeException
  28. */
  29. public function __construct(string $root)
  30. {
  31. $this->root = $root;
  32. if (!file_exists($this->root) && !@mkdir($this->root, 0777, true) && !is_dir($this->root)) {
  33. throw new \RuntimeException(sprintf('Unable to create the store directory (%s).', $this->root));
  34. }
  35. $this->keyCache = new \SplObjectStorage();
  36. $this->locks = [];
  37. }
  38. /**
  39. * Cleanups storage.
  40. */
  41. public function cleanup()
  42. {
  43. // unlock everything
  44. foreach ($this->locks as $lock) {
  45. flock($lock, LOCK_UN);
  46. fclose($lock);
  47. }
  48. $this->locks = [];
  49. }
  50. /**
  51. * Tries to lock the cache for a given Request, without blocking.
  52. *
  53. * @return bool|string true if the lock is acquired, the path to the current lock otherwise
  54. */
  55. public function lock(Request $request)
  56. {
  57. $key = $this->getCacheKey($request);
  58. if (!isset($this->locks[$key])) {
  59. $path = $this->getPath($key);
  60. if (!file_exists(\dirname($path)) && false === @mkdir(\dirname($path), 0777, true) && !is_dir(\dirname($path))) {
  61. return $path;
  62. }
  63. $h = fopen($path, 'cb');
  64. if (!flock($h, LOCK_EX | LOCK_NB)) {
  65. fclose($h);
  66. return $path;
  67. }
  68. $this->locks[$key] = $h;
  69. }
  70. return true;
  71. }
  72. /**
  73. * Releases the lock for the given Request.
  74. *
  75. * @return bool False if the lock file does not exist or cannot be unlocked, true otherwise
  76. */
  77. public function unlock(Request $request)
  78. {
  79. $key = $this->getCacheKey($request);
  80. if (isset($this->locks[$key])) {
  81. flock($this->locks[$key], LOCK_UN);
  82. fclose($this->locks[$key]);
  83. unset($this->locks[$key]);
  84. return true;
  85. }
  86. return false;
  87. }
  88. public function isLocked(Request $request)
  89. {
  90. $key = $this->getCacheKey($request);
  91. if (isset($this->locks[$key])) {
  92. return true; // shortcut if lock held by this process
  93. }
  94. if (!file_exists($path = $this->getPath($key))) {
  95. return false;
  96. }
  97. $h = fopen($path, 'rb');
  98. flock($h, LOCK_EX | LOCK_NB, $wouldBlock);
  99. flock($h, LOCK_UN); // release the lock we just acquired
  100. fclose($h);
  101. return (bool) $wouldBlock;
  102. }
  103. /**
  104. * Locates a cached Response for the Request provided.
  105. *
  106. * @return Response|null A Response instance, or null if no cache entry was found
  107. */
  108. public function lookup(Request $request)
  109. {
  110. $key = $this->getCacheKey($request);
  111. if (!$entries = $this->getMetadata($key)) {
  112. return null;
  113. }
  114. // find a cached entry that matches the request.
  115. $match = null;
  116. foreach ($entries as $entry) {
  117. if ($this->requestsMatch(isset($entry[1]['vary'][0]) ? implode(', ', $entry[1]['vary']) : '', $request->headers->all(), $entry[0])) {
  118. $match = $entry;
  119. break;
  120. }
  121. }
  122. if (null === $match) {
  123. return null;
  124. }
  125. $headers = $match[1];
  126. if (file_exists($body = $this->getPath($headers['x-content-digest'][0]))) {
  127. return $this->restoreResponse($headers, $body);
  128. }
  129. // TODO the metaStore referenced an entity that doesn't exist in
  130. // the entityStore. We definitely want to return nil but we should
  131. // also purge the entry from the meta-store when this is detected.
  132. return null;
  133. }
  134. /**
  135. * Writes a cache entry to the store for the given Request and Response.
  136. *
  137. * Existing entries are read and any that match the response are removed. This
  138. * method calls write with the new list of cache entries.
  139. *
  140. * @return string The key under which the response is stored
  141. *
  142. * @throws \RuntimeException
  143. */
  144. public function write(Request $request, Response $response)
  145. {
  146. $key = $this->getCacheKey($request);
  147. $storedEnv = $this->persistRequest($request);
  148. // write the response body to the entity store if this is the original response
  149. if (!$response->headers->has('X-Content-Digest')) {
  150. $digest = $this->generateContentDigest($response);
  151. if (!$this->save($digest, $response->getContent())) {
  152. throw new \RuntimeException('Unable to store the entity.');
  153. }
  154. $response->headers->set('X-Content-Digest', $digest);
  155. if (!$response->headers->has('Transfer-Encoding')) {
  156. $response->headers->set('Content-Length', \strlen($response->getContent()));
  157. }
  158. }
  159. // read existing cache entries, remove non-varying, and add this one to the list
  160. $entries = [];
  161. $vary = $response->headers->get('vary');
  162. foreach ($this->getMetadata($key) as $entry) {
  163. if (!isset($entry[1]['vary'][0])) {
  164. $entry[1]['vary'] = [''];
  165. }
  166. if ($entry[1]['vary'][0] != $vary || !$this->requestsMatch($vary ?? '', $entry[0], $storedEnv)) {
  167. $entries[] = $entry;
  168. }
  169. }
  170. $headers = $this->persistResponse($response);
  171. unset($headers['age']);
  172. array_unshift($entries, [$storedEnv, $headers]);
  173. if (!$this->save($key, serialize($entries))) {
  174. throw new \RuntimeException('Unable to store the metadata.');
  175. }
  176. return $key;
  177. }
  178. /**
  179. * Returns content digest for $response.
  180. *
  181. * @return string
  182. */
  183. protected function generateContentDigest(Response $response)
  184. {
  185. return 'en'.hash('sha256', $response->getContent());
  186. }
  187. /**
  188. * Invalidates all cache entries that match the request.
  189. *
  190. * @throws \RuntimeException
  191. */
  192. public function invalidate(Request $request)
  193. {
  194. $modified = false;
  195. $key = $this->getCacheKey($request);
  196. $entries = [];
  197. foreach ($this->getMetadata($key) as $entry) {
  198. $response = $this->restoreResponse($entry[1]);
  199. if ($response->isFresh()) {
  200. $response->expire();
  201. $modified = true;
  202. $entries[] = [$entry[0], $this->persistResponse($response)];
  203. } else {
  204. $entries[] = $entry;
  205. }
  206. }
  207. if ($modified && !$this->save($key, serialize($entries))) {
  208. throw new \RuntimeException('Unable to store the metadata.');
  209. }
  210. }
  211. /**
  212. * Determines whether two Request HTTP header sets are non-varying based on
  213. * the vary response header value provided.
  214. *
  215. * @param string|null $vary A Response vary header
  216. * @param array $env1 A Request HTTP header array
  217. * @param array $env2 A Request HTTP header array
  218. */
  219. private function requestsMatch(?string $vary, array $env1, array $env2): bool
  220. {
  221. if (empty($vary)) {
  222. return true;
  223. }
  224. foreach (preg_split('/[\s,]+/', $vary) as $header) {
  225. $key = str_replace('_', '-', strtolower($header));
  226. $v1 = isset($env1[$key]) ? $env1[$key] : null;
  227. $v2 = isset($env2[$key]) ? $env2[$key] : null;
  228. if ($v1 !== $v2) {
  229. return false;
  230. }
  231. }
  232. return true;
  233. }
  234. /**
  235. * Gets all data associated with the given key.
  236. *
  237. * Use this method only if you know what you are doing.
  238. */
  239. private function getMetadata(string $key): array
  240. {
  241. if (!$entries = $this->load($key)) {
  242. return [];
  243. }
  244. return unserialize($entries);
  245. }
  246. /**
  247. * Purges data for the given URL.
  248. *
  249. * This method purges both the HTTP and the HTTPS version of the cache entry.
  250. *
  251. * @return bool true if the URL exists with either HTTP or HTTPS scheme and has been purged, false otherwise
  252. */
  253. public function purge(string $url)
  254. {
  255. $http = preg_replace('#^https:#', 'http:', $url);
  256. $https = preg_replace('#^http:#', 'https:', $url);
  257. $purgedHttp = $this->doPurge($http);
  258. $purgedHttps = $this->doPurge($https);
  259. return $purgedHttp || $purgedHttps;
  260. }
  261. /**
  262. * Purges data for the given URL.
  263. */
  264. private function doPurge(string $url): bool
  265. {
  266. $key = $this->getCacheKey(Request::create($url));
  267. if (isset($this->locks[$key])) {
  268. flock($this->locks[$key], LOCK_UN);
  269. fclose($this->locks[$key]);
  270. unset($this->locks[$key]);
  271. }
  272. if (file_exists($path = $this->getPath($key))) {
  273. unlink($path);
  274. return true;
  275. }
  276. return false;
  277. }
  278. /**
  279. * Loads data for the given key.
  280. */
  281. private function load(string $key): ?string
  282. {
  283. $path = $this->getPath($key);
  284. return file_exists($path) && false !== ($contents = file_get_contents($path)) ? $contents : null;
  285. }
  286. /**
  287. * Save data for the given key.
  288. */
  289. private function save(string $key, string $data): bool
  290. {
  291. $path = $this->getPath($key);
  292. if (isset($this->locks[$key])) {
  293. $fp = $this->locks[$key];
  294. @ftruncate($fp, 0);
  295. @fseek($fp, 0);
  296. $len = @fwrite($fp, $data);
  297. if (\strlen($data) !== $len) {
  298. @ftruncate($fp, 0);
  299. return false;
  300. }
  301. } else {
  302. if (!file_exists(\dirname($path)) && false === @mkdir(\dirname($path), 0777, true) && !is_dir(\dirname($path))) {
  303. return false;
  304. }
  305. $tmpFile = tempnam(\dirname($path), basename($path));
  306. if (false === $fp = @fopen($tmpFile, 'wb')) {
  307. @unlink($tmpFile);
  308. return false;
  309. }
  310. @fwrite($fp, $data);
  311. @fclose($fp);
  312. if ($data != file_get_contents($tmpFile)) {
  313. @unlink($tmpFile);
  314. return false;
  315. }
  316. if (false === @rename($tmpFile, $path)) {
  317. @unlink($tmpFile);
  318. return false;
  319. }
  320. }
  321. @chmod($path, 0666 & ~umask());
  322. return true;
  323. }
  324. public function getPath(string $key)
  325. {
  326. 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);
  327. }
  328. /**
  329. * Generates a cache key for the given Request.
  330. *
  331. * This method should return a key that must only depend on a
  332. * normalized version of the request URI.
  333. *
  334. * If the same URI can have more than one representation, based on some
  335. * headers, use a Vary header to indicate them, and each representation will
  336. * be stored independently under the same cache key.
  337. *
  338. * @return string A key for the given Request
  339. */
  340. protected function generateCacheKey(Request $request)
  341. {
  342. return 'md'.hash('sha256', $request->getUri());
  343. }
  344. /**
  345. * Returns a cache key for the given Request.
  346. */
  347. private function getCacheKey(Request $request): string
  348. {
  349. if (isset($this->keyCache[$request])) {
  350. return $this->keyCache[$request];
  351. }
  352. return $this->keyCache[$request] = $this->generateCacheKey($request);
  353. }
  354. /**
  355. * Persists the Request HTTP headers.
  356. */
  357. private function persistRequest(Request $request): array
  358. {
  359. return $request->headers->all();
  360. }
  361. /**
  362. * Persists the Response HTTP headers.
  363. */
  364. private function persistResponse(Response $response): array
  365. {
  366. $headers = $response->headers->all();
  367. $headers['X-Status'] = [$response->getStatusCode()];
  368. return $headers;
  369. }
  370. /**
  371. * Restores a Response from the HTTP headers and body.
  372. */
  373. private function restoreResponse(array $headers, string $body = null): Response
  374. {
  375. $status = $headers['X-Status'][0];
  376. unset($headers['X-Status']);
  377. if (null !== $body) {
  378. $headers['X-Body-File'] = [$body];
  379. }
  380. return new Response($body, $status, $headers);
  381. }
  382. }