123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132 |
- <?php
- namespace Symfony\Component\HttpClient;
- use Psr\Log\LoggerAwareInterface;
- use Psr\Log\LoggerInterface;
- use Symfony\Component\HttpClient\Exception\InvalidArgumentException;
- use Symfony\Component\HttpClient\Exception\TransportException;
- use Symfony\Component\HttpFoundation\IpUtils;
- use Symfony\Contracts\HttpClient\HttpClientInterface;
- use Symfony\Contracts\HttpClient\ResponseInterface;
- use Symfony\Contracts\HttpClient\ResponseStreamInterface;
- use Symfony\Contracts\Service\ResetInterface;
- final class NoPrivateNetworkHttpClient implements HttpClientInterface, LoggerAwareInterface, ResetInterface
- {
- use HttpClientTrait;
- private const PRIVATE_SUBNETS = [
- '127.0.0.0/8',
- '10.0.0.0/8',
- '192.168.0.0/16',
- '172.16.0.0/12',
- '169.254.0.0/16',
- '0.0.0.0/8',
- '240.0.0.0/4',
- '::1/128',
- 'fc00::/7',
- 'fe80::/10',
- '::ffff:0:0/96',
- '::/128',
- ];
- private $client;
- private $subnets;
-
- public function __construct(HttpClientInterface $client, $subnets = null)
- {
- if (!(\is_array($subnets) || \is_string($subnets) || null === $subnets)) {
- throw new \TypeError(sprintf('Argument 2 passed to "%s()" must be of the type array, string or null. "%s" given.', __METHOD__, get_debug_type($subnets)));
- }
- if (!class_exists(IpUtils::class)) {
- throw new \LogicException(sprintf('You cannot use "%s" if the HttpFoundation component is not installed. Try running "composer require symfony/http-foundation".', __CLASS__));
- }
- $this->client = $client;
- $this->subnets = $subnets;
- }
-
- public function request(string $method, string $url, array $options = []): ResponseInterface
- {
- $onProgress = $options['on_progress'] ?? null;
- if (null !== $onProgress && !\is_callable($onProgress)) {
- throw new InvalidArgumentException(sprintf('Option "on_progress" must be callable, "%s" given.', get_debug_type($onProgress)));
- }
- $subnets = $this->subnets;
- $lastPrimaryIp = '';
- $options['on_progress'] = function (int $dlNow, int $dlSize, array $info) use ($onProgress, $subnets, &$lastPrimaryIp): void {
- if ($info['primary_ip'] !== $lastPrimaryIp) {
- if ($info['primary_ip'] && IpUtils::checkIp($info['primary_ip'], $subnets ?? self::PRIVATE_SUBNETS)) {
- throw new TransportException(sprintf('IP "%s" is blocked for "%s".', $info['primary_ip'], $info['url']));
- }
- $lastPrimaryIp = $info['primary_ip'];
- }
- null !== $onProgress && $onProgress($dlNow, $dlSize, $info);
- };
- return $this->client->request($method, $url, $options);
- }
-
- public function stream($responses, float $timeout = null): ResponseStreamInterface
- {
- return $this->client->stream($responses, $timeout);
- }
-
- public function setLogger(LoggerInterface $logger): void
- {
- if ($this->client instanceof LoggerAwareInterface) {
- $this->client->setLogger($logger);
- }
- }
-
- public function withOptions(array $options): self
- {
- $clone = clone $this;
- $clone->client = $this->client->withOptions($options);
- return $clone;
- }
- public function reset()
- {
- if ($this->client instanceof ResetInterface) {
- $this->client->reset();
- }
- }
- }
|