MAINHACK

Viewing / Editing File: TransportResponseTrait.php

<?php

/*
 * This file is part of the Symfony package.
 *
 * (c) Fabien Potencier <fabien@symfony.com>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Symfony\Component\HttpClient\Response;

use Psr\Log\LoggerInterface;
use Symfony\Component\HttpClient\Chunk\DataChunk;
use Symfony\Component\HttpClient\Chunk\ErrorChunk;
use Symfony\Component\HttpClient\Chunk\FirstChunk;
use Symfony\Component\HttpClient\Chunk\LastChunk;
use Symfony\Component\HttpClient\Exception\TransportException;
use Symfony\Component\HttpClient\Internal\Canary;
use Symfony\Component\HttpClient\Internal\ClientState;

/**
 * Implements common logic for transport-level response classes.
 *
 * @author Nicolas Grekas <p@tchwork.com>
 *
 * @internal
 */
trait TransportResponseTrait
{
    private Canary $canary;
    private array $headers = [];
    private array $info = [
        'response_headers' => [],
        'http_code' => 0,
        'error' => null,
        'canceled' => false,
    ];

/** @var object|resource|null */
    private $handle;
    private int|string $id;
    private ?float $timeout = 0;
    private \InflateContext|bool|null $inflate = null;
    private ?array $finalInfo = null;
    private ?LoggerInterface $logger = null;

public function getStatusCode(): int
    {
        if ($this->initializer) {
            self::initialize($this);
        }

return $this->info['http_code'];
    }

public function getHeaders(bool $throw = true): array
    {
        if ($this->initializer) {
            self::initialize($this);
        }

if ($throw) {
            $this->checkStatusCode();
        }

return $this->headers;
    }

public function cancel(): void
    {
        $this->info['canceled'] = true;
        $this->info['error'] = 'Response has been canceled.';
        $this->close();
    }

/**
     * Closes the response and all its network handles.
     */
    protected function close(): void
    {
        $this->canary->cancel();
        $this->inflate = null;
    }

/**
     * Adds pending responses to the activity list.
     */
    abstract protected static function schedule(self $response, array &$runningResponses): void;

/**
     * Performs all pending non-blocking operations.
     */
    abstract protected static function perform(ClientState $multi, array $responses): void;

/**
     * Waits for network activity.
     */
    abstract protected static function select(ClientState $multi, float $timeout): int;

private static function addResponseHeaders(array $responseHeaders, array &$info, array &$headers, string &$debug = ''): void
    {
        foreach ($responseHeaders as $h) {
            if (11 <= \strlen($h) && '/' === $h[4] && preg_match('#^HTTP/\d+(?:\.\d+)? (\d\d\d)(?: |$)#', $h, $m)) {
                if ($headers) {
                    $debug .= "< \r\n";
                    $headers = [];
                }
                $info['http_code'] = (int) $m[1];
            } elseif (2 === \count($m = explode(':', $h, 2))) {
                $headers[strtolower($m[0])][] = ltrim($m[1]);
            }

$debug .= "< {$h}\r\n";
            $info['response_headers'][] = $h;
        }

$debug .= "< \r\n";
    }

/**
     * Ensures the request is always sent and that the response code was checked.
     */
    private function doDestruct(): void
    {
        $this->shouldBuffer = true;

if ($this->initializer && null === $this->info['error']) {
            self::initialize($this);
            $this->checkStatusCode();
        }
    }

/**
     * Implements an event loop based on a buffer activity queue.
     *
     * @param iterable<array-key, self> $responses
     *
     * @internal
     */
    public static function stream(iterable $responses, ?float $timeout = null): \Generator
    {
        $runningResponses = [];

foreach ($responses as $response) {
            self::schedule($response, $runningResponses);
        }

$lastActivity = hrtime(true) / 1E9;
        $elapsedTimeout = 0;

if ((0.0 === $timeout && '-0' === (string) $timeout) || 0 > $timeout) {
            $timeout = $timeout ? -$timeout : null;

/** @var ClientState $multi */
            foreach ($runningResponses as [$multi]) {
                if (null !== $multi->lastTimeout) {
                    $elapsedTimeout = max($elapsedTimeout, $lastActivity - $multi->lastTimeout);
                }
            }
        }

while (true) {
            $hasActivity = false;
            $timeoutMax = 0;
            $timeoutMin = $timeout ?? \INF;

/** @var ClientState $multi */
            foreach ($runningResponses as $i => [$multi, &$responses]) {
                self::perform($multi, $responses);

foreach ($responses as $j => $response) {
                    $timeoutMax = $timeout ?? max($timeoutMax, $response->timeout);
                    $timeoutMin = min($timeoutMin, $response->timeout, 1);
                    $chunk = false;

if (isset($multi->handlesActivity[$j])) {
                        $multi->lastTimeout = null;
                        $elapsedTimeout = 0;
                    } elseif (!isset($multi->openHandles[$j])) {
                        $hasActivity = true;
                        unset($responses[$j]);
                        continue;
                    } elseif ($elapsedTimeout >= $timeoutMax) {
                        $multi->handlesActivity[$j] = [new ErrorChunk($response->offset, \sprintf('Idle timeout reached for "%s".', $response->getInfo('url')))];
                        $multi->lastTimeout ??= $lastActivity;
                        $elapsedTimeout = $timeoutMax;
                    } else {
                        continue;
                    }

$lastActivity = null;
                    $hasActivity = true;

while ($multi->handlesActivity[$j] ?? false) {
                        if (\is_string($chunk = array_shift($multi->handlesActivity[$j]))) {
                            if (null !== $response->inflate && false === $chunk = @inflate_add($response->inflate, $chunk)) {
                                $multi->handlesActivity[$j] = [null, new TransportException(\sprintf('Error while processing content unencoding for "%s".', $response->getInfo('url')))];
                                continue;
                            }

if ('' !== $chunk && null !== $response->content && \strlen($chunk) !== fwrite($response->content, $chunk)) {
                                $multi->handlesActivity[$j] = [null, new TransportException(\sprintf('Failed writing %d bytes to the response buffer.', \strlen($chunk)))];
                                continue;
                            }

$chunkLen = \strlen($chunk);
                            $chunk = new DataChunk($response->offset, $chunk);
                            $response->offset += $chunkLen;
                        } elseif (null === $chunk) {
                            $e = $multi->handlesActivity[$j][0];
                            unset($responses[$j], $multi->handlesActivity[$j]);
                            $response->close();

if (null !== $e) {
                                $response->info['error'] = $e->getMessage();

if ($e instanceof \Error) {
                                    throw $e;
                                }

$chunk = new ErrorChunk($response->offset, $e);
                            } else {
                                if (0 === $response->offset && null === $response->content) {
                                    $response->content = fopen('php://memory', 'w+');
                                }

$chunk = new LastChunk($response->offset);
                            }
                        } elseif ($chunk instanceof ErrorChunk) {
                            unset($responses[$j]);
                        } elseif ($chunk instanceof FirstChunk) {
                            if ($response->logger) {
                                $info = $response->getInfo();
                                $response->logger->info('Response: "{http_code} {url}" {total_time} seconds', [
                                    'http_code' => $info['http_code'],
                                    'url' => $info['url'],
                                    'total_time' => $info['total_time'],
                                ]);
                            }

$response->inflate = \extension_loaded('zlib') && $response->inflate && 'gzip' === ($response->headers['content-encoding'][0] ?? null) ? inflate_init(\ZLIB_ENCODING_GZIP) : null;

if ($response->shouldBuffer instanceof \Closure) {
                                try {
                                    $response->shouldBuffer = ($response->shouldBuffer)($response->headers);

if (null !== $response->info['error']) {
                                        throw new TransportException($response->info['error']);
                                    }
                                } catch (\Throwable $e) {
                                    $response->close();
                                    $multi->handlesActivity[$j] = [null, $e];
                                }
                            }

if (true === $response->shouldBuffer) {
                                $response->content = fopen('php://temp', 'w+');
                            } elseif (\is_resource($response->shouldBuffer)) {
                                $response->content = $response->shouldBuffer;
                            }
                            $response->shouldBuffer = null;

yield $response => $chunk;

if ($response->initializer && null === $response->info['error']) {
                                // Ensure the HTTP status code is always checked
                                $response->getHeaders(true);
                            }

continue;
                        }

yield $response => $chunk;
                    }

unset($multi->handlesActivity[$j]);

if ($chunk instanceof ErrorChunk && !$chunk->didThrow()) {
                        // Ensure transport exceptions are always thrown
                        $chunk->getContent();
                        throw new \LogicException('A transport exception should have been thrown.');
                    }
                }

if (!$responses) {
                    $hasActivity = true;
                    unset($runningResponses[$i]);
                }

// Prevent memory leaks
                $multi->handlesActivity = $multi->handlesActivity ?: [];
                $multi->openHandles = $multi->openHandles ?: [];
            }

if (!$runningResponses) {
                break;
            }

if ($hasActivity) {
                $lastActivity ??= hrtime(true) / 1E9;
                continue;
            }

if (-1 === self::select($multi, min($timeoutMin, $timeoutMax - $elapsedTimeout))) {
                usleep((int) min(500, 1E6 * $timeoutMin));
            }

$elapsedTimeout = hrtime(true) / 1E9 - $lastActivity;
        }
    }
}

Cancel / Back

[ MAINHACK ]

Viewing / Editing File: TransportResponseTrait.php

Server Info

System Features

User Info