vendor/mpdf/mpdf/src/AssetFetcher.php line 51

Open in your IDE?
  1. <?php
  3. namespace Mpdf;
  5. use Mpdf\File\LocalContentLoaderInterface;
  6. use Mpdf\File\StreamWrapperChecker;
  7. use Mpdf\Http\ClientInterface;
  8. use Mpdf\Http\Request;
  9. use Mpdf\Log\Context as LogContext;
  10. use Psr\Log\LoggerInterface;
  12. class AssetFetcher implements \Psr\Log\LoggerAwareInterface
  13. {
  15.     private $mpdf;
  17.     private $contentLoader;
  19.     private $http;
  21.     private $logger;
  23.     public function __construct(Mpdf $mpdf, LocalContentLoaderInterface $contentLoader, ClientInterface $http, LoggerInterface $logger)
  24.     {
  25.         $this->mpdf = $mpdf;
  26.         $this->contentLoader = $contentLoader;
  27.         $this->http = $http;
  28.         $this->logger = $logger;
  29.     }
  31.     public function fetchDataFromPath($path, $originalSrc = null)
  32.     {
  33.         /**
  34.          * Prevents insecure PHP object injection through phar:// wrapper
  35.          * @see https://github.com/mpdf/mpdf/issues/949
  36.          * @see https://github.com/mpdf/mpdf/issues/1381
  37.          */
  38.         $wrapperChecker = new StreamWrapperChecker($this->mpdf);
  40.         if ($wrapperChecker->hasBlacklistedStreamWrapper($path)) {
  41.             throw new \Mpdf\Exception\AssetFetchingException('File contains an invalid stream. Only ' . implode(', ', $wrapperChecker->getWhitelistedStreamWrappers()) . ' streams are allowed.');
  42.         }
  44.         if ($originalSrc && $wrapperChecker->hasBlacklistedStreamWrapper($originalSrc)) {
  45.             throw new \Mpdf\Exception\AssetFetchingException('File contains an invalid stream. Only ' . implode(', ', $wrapperChecker->getWhitelistedStreamWrappers()) . ' streams are allowed.');
  46.         }
  48.         $this->mpdf->GetFullPath($path);
  50.         return $this->isPathLocal($path) || ($originalSrc !== null && $this->isPathLocal($originalSrc))
  51.             ? $this->fetchLocalContent($path, $originalSrc)
  52.             : $this->fetchRemoteContent($path);
  53.     }
  55.     public function fetchLocalContent($path, $originalSrc)
  56.     {
  57.         $data = '';
  59.         if ($originalSrc && $this->mpdf->basepathIsLocal && $check = @fopen($originalSrc, 'rb')) {
  60.             fclose($check);
  61.             $path = $originalSrc;
  62.             $this->logger->debug(sprintf('Fetching content of file "%s" with local basepath', $path), ['context' => LogContext::REMOTE_CONTENT]);
  64.             return $this->contentLoader->load($path);
  65.         }
  67.         if ($path && $check = @fopen($path, 'rb')) {
  68.             fclose($check);
  69.             $this->logger->debug(sprintf('Fetching content of file "%s" with non-local basepath', $path), ['context' => LogContext::REMOTE_CONTENT]);
  71.             return $this->contentLoader->load($path);
  72.         }
  74.         return $data;
  75.     }
  77.     public function fetchRemoteContent($path)
  78.     {
  79.         $data = '';
  81.         try {
  83.             $this->logger->debug(sprintf('Fetching remote content of file "%s"', $path), ['context' => LogContext::REMOTE_CONTENT]);
  85.             /** @var \Mpdf\Http\Response $response */
  86.             $response = $this->http->sendRequest(new Request('GET', $path));
  88.             if ($response->getStatusCode() !== 200) {
  90.                 $message = sprintf('Non-OK HTTP response "%s" on fetching remote content "%s" because of an error', $response->getStatusCode(), $path);
  91.                 if ($this->mpdf->debug) {
  92.                     throw new \Mpdf\MpdfException($message);
  93.                 }
  95.                 $this->logger->info($message);
  97.                 return $data;
  98.             }
  100.             $data = $response->getBody()->getContents();
  102.         } catch (\InvalidArgumentException $e) {
  103.             $message = sprintf('Unable to fetch remote content "%s" because of an error "%s"', $path, $e->getMessage());
  104.             if ($this->mpdf->debug) {
  105.                 throw new \Mpdf\MpdfException($message, 0, $e);
  106.             }
  108.             $this->logger->warning($message);
  109.         }
  111.         return $data;
  112.     }
  114.     public function isPathLocal($path)
  115.     {
  116.         return strpos($path, '://') === false; // @todo More robust implementation
  117.     }
  119.     public function setLogger(LoggerInterface $logger)
  120.     {
  121.         $this->logger = $logger;
  122.     }
  124. }