timw4mail/HummingBirdAnimeClient
1
1
Fork 1
Code Issues 5 Pull Requests Projects 1 Releases Wiki Activity
762bdba724
HummingBirdAnimeClient/src/API/HummingbirdClient.php

1311 lines
33 KiB
PHP
Raw Blame History

<?php // declare(strict_types=1);
/**
* Hummingbird Anime List Client
*
* An API client for Kitsu to manage anime and manga watch lists
*
* PHP version 7.1
*
* @package HummingbirdAnimeClient
* @author Timothy J. Warren <tim@timshomepage.net>
* @copyright 2015 - 2018 Timothy J. Warren
* @license http://www.opensource.org/licenses/mit-license.html MIT License
* @version 4.1
* @link https://git.timshomepage.net/timw4mail/HummingBirdAnimeClient
*/
namespace Aviat\AnimeClient\API;
use Amp\{
CancellationToken,
CancelledException,
Deferred,
Delayed,
Emitter,
Failure,
Loop,
NullCancellationToken,
Promise,
Success,
TimeoutCancellationToken
};
use Amp\Artax\{
ConnectionInfo,
Client,
DnsException,
HttpException,
HttpSocketPool,
MetaInfo,
ParseException,
RequestBody,
Response,
Request,
SocketException,
TimeoutException,
TlsInfo,
TooManyRedirectsException
};
use Amp\Artax\Cookie\{
Cookie,
CookieFormatException,
CookieJar,
NullCookieJar
};
use Amp\Artax\Internal\{
CombinedCancellationToken,
Parser,
PublicSuffixList,
RequestCycle
};
use Amp\ByteStream\{
InputStream, IteratorStream, Message, ZlibInputStream
};
use Amp\Dns\ResolutionException;
use Amp\Socket\{
ClientSocket, ClientTlsContext, ConnectException
};
use Amp\Uri\{
InvalidUriException, Uri
};
use const Aviat\AnimeClient\USER_AGENT;
use function Amp\{
asyncCall, call
};
/**
* Re-implementation of Artax's default client
*/
class HummingbirdClient implements Client {
const DEFAULT_USER_AGENT = USER_AGENT;
private $cookieJar;
private $socketPool;
private $tlsContext;
private $hasZlib;
private $options = [
self::OP_AUTO_ENCODING => TRUE,
self::OP_TRANSFER_TIMEOUT => 100000,
self::OP_MAX_REDIRECTS => 5,
self::OP_AUTO_REFERER => TRUE,
self::OP_DISCARD_BODY => FALSE,
self::OP_DEFAULT_HEADERS => [],
self::OP_MAX_HEADER_BYTES => Parser::DEFAULT_MAX_HEADER_BYTES,
self::OP_MAX_BODY_BYTES => Parser::DEFAULT_MAX_BODY_BYTES,
];
public function __construct(
CookieJar $cookieJar = NULL,
HttpSocketPool $socketPool = NULL,
ClientTlsContext $tlsContext = NULL
)
{
$this->cookieJar = $cookieJar ?? new NullCookieJar;
$this->tlsContext = $tlsContext ?? new ClientTlsContext;
$this->socketPool = $socketPool ?? new HttpSocketPool;
$this->hasZlib = \extension_loaded('zlib');
}
/** @inheritdoc */
public function request($uriOrRequest, array $options = [], CancellationToken $cancellation = NULL): Promise
{
return call(function () use ($uriOrRequest, $options, $cancellation) {
$cancellation = $cancellation ?? new NullCancellationToken;
foreach ($options as $option => $value)
{
$this->validateOption($option, $value);
}
/** @var Request $request */
list($request, $uri) = $this->generateRequestFromUri($uriOrRequest);
$options = $options ? array_merge($this->options, $options) : $this->options;
foreach ($this->options[self::OP_DEFAULT_HEADERS] as $name => $header)
{
if ( ! $request->hasHeader($name))
{
$request = $request->withHeaders([$name => $header]);
}
}
/** @var array $headers */
$headers = yield $request->getBody()->getHeaders();
foreach ($headers as $name => $header)
{
if ( ! $request->hasHeader($name))
{
$request = $request->withHeaders([$name => $header]);
}
}
$originalUri = $uri;
$previousResponse = NULL;
$maxRedirects = $options[self::OP_MAX_REDIRECTS];
$requestNr = 1;
do
{
/** @var Request $request */
$request = yield from $this->normalizeRequestBodyHeaders($request);
$request = $this->normalizeRequestHeaders($request, $uri, $options);
// Always normalize this as last item, because we need to strip sensitive headers
$request = $this->normalizeTraceRequest($request);
/** @var Response $response */
$response = yield $this->doRequest($request, $uri, $options, $previousResponse, $cancellation);
// Explicit $maxRedirects !== 0 check to not consume redirect bodies if redirect following is disabled
if ($maxRedirects !== 0 && $redirectUri = $this->getRedirectUri($response))
{
// Discard response body of redirect responses
$body = $response->getBody();
while (NULL !== yield $body->read()) ;
/**
* If this is a 302/303 we need to follow the location with a GET if the original request wasn't
* GET. Otherwise we need to send the body again.
*
* We won't resend the body nor any headers on redirects to other hosts for security reasons.
*
* @link http://www.w3.org/Protocols/rfc2616/rfc2616-sec10.html#sec10.3.3
*/
$method = $request->getMethod();
$status = $response->getStatus();
$isSameHost = $redirectUri->getAuthority(FALSE) === $originalUri->getAuthority(FALSE);
if ($isSameHost)
{
$request = $request->withUri($redirectUri);
if ($status >= 300 && $status <= 303 && $method !== 'GET')
{
$request = $request->withMethod('GET');
$request = $request->withoutHeader('Transfer-Encoding');
$request = $request->withoutHeader('Content-Length');
$request = $request->withoutHeader('Content-Type');
$request = $request->withBody(NULL);
}
} else
{
// We ALWAYS follow with a GET and without any set headers or body for redirects to other hosts.
$optionsWithoutHeaders = $options;
unset($optionsWithoutHeaders[self::OP_DEFAULT_HEADERS]);
$request = new Request((string)$redirectUri);
$request = $this->normalizeRequestHeaders($request, $redirectUri, $optionsWithoutHeaders);
}
if ($options[self::OP_AUTO_REFERER])
{
$request = $this->assignRedirectRefererHeader($request, $originalUri, $redirectUri);
}
$previousResponse = $response;
$originalUri = $redirectUri;
$uri = $redirectUri;
} else
{
break;
}
} while (++$requestNr <= $maxRedirects + 1);
if ($maxRedirects !== 0 && $redirectUri = $this->getRedirectUri($response))
{
throw new TooManyRedirectsException($response);
}
return $response;
});
}
private function validateOption(string $option, $value)
{
switch ($option)
{
case self::OP_AUTO_ENCODING:
if ( ! \is_bool($value))
{
throw new \TypeError("Invalid value for OP_AUTO_ENCODING, bool expected");
}
break;
case self::OP_TRANSFER_TIMEOUT:
if ( ! \is_int($value) || $value < 0)
{
throw new \Error("Invalid value for OP_TRANSFER_TIMEOUT, int >= 0 expected");
}
break;
case self::OP_MAX_REDIRECTS:
if ( ! \is_int($value) || $value < 0)
{
throw new \Error("Invalid value for OP_MAX_REDIRECTS, int >= 0 expected");
}
break;
case self::OP_AUTO_REFERER:
if ( ! \is_bool($value))
{
throw new \TypeError("Invalid value for OP_AUTO_REFERER, bool expected");
}
break;
case self::OP_DISCARD_BODY:
if ( ! \is_bool($value))
{
throw new \TypeError("Invalid value for OP_DISCARD_BODY, bool expected");
}
break;
case self::OP_DEFAULT_HEADERS:
// We attempt to set the headers here, because they're automatically validated then.
(new Request("https://example.com/"))->withHeaders($value);
break;
case self::OP_MAX_HEADER_BYTES:
if ( ! \is_int($value) || $value < 0)
{
throw new \Error("Invalid value for OP_MAX_HEADER_BYTES, int >= 0 expected");
}
break;
case self::OP_MAX_BODY_BYTES:
if ( ! \is_int($value) || $value < 0)
{
throw new \Error("Invalid value for OP_MAX_BODY_BYTES, int >= 0 expected");
}
break;
default:
throw new \Error(
sprintf("Unknown option: %s", $option)
);
}
}
private function generateRequestFromUri($uriOrRequest)
{
if (is_string($uriOrRequest))
{
$uri = $this->buildUriFromString($uriOrRequest);
$request = new Request($uri);
} elseif ($uriOrRequest instanceof Request)
{
$uri = $this->buildUriFromString($uriOrRequest->getUri());
$request = $uriOrRequest;
} else
{
throw new HttpException(
'Request must be a valid HTTP URI or Amp\Artax\Request instance'
);
}
return [$request, $uri];
}
private function buildUriFromString($str): Uri
{
try
{
$uri = new Uri($str);
$scheme = $uri->getScheme();
if (($scheme === "http" || $scheme === "https") && $uri->getHost())
{
return $uri;
}
throw new HttpException("Request must specify a valid HTTP URI");
} catch (InvalidUriException $e)
{
throw new HttpException("Request must specify a valid HTTP URI", 0, $e);
}
}
private function normalizeRequestBodyHeaders(Request $request): \Generator
{
if ($request->hasHeader("Transfer-Encoding"))
{
return $request->withoutHeader("Content-Length");
}
if ($request->hasHeader("Content-Length"))
{
return $request;
}
/** @var RequestBody $body */
$body = $request->getBody();
$bodyLength = yield $body->getBodyLength();
if ($bodyLength === 0)
{
$request = $request->withHeader('Content-Length', '0');
$request = $request->withoutHeader('Transfer-Encoding');
} else
{
if ($bodyLength > 0)
{
$request = $request->withHeader("Content-Length", (string)$bodyLength);
$request = $request->withoutHeader("Transfer-Encoding");
} else
{
$request = $request->withHeader("Transfer-Encoding", "chunked");
}
}
return $request;
}
private function normalizeRequestHeaders($request, $uri, $options)
{
$request = $this->normalizeRequestEncodingHeaderForZlib($request, $options);
$request = $this->normalizeRequestHostHeader($request, $uri);
$request = $this->normalizeRequestUserAgent($request);
$request = $this->normalizeRequestAcceptHeader($request);
$request = $this->assignApplicableRequestCookies($request);
return $request;
}
private function normalizeRequestEncodingHeaderForZlib(Request $request, array $options): Request
{
$autoEncoding = $options[self::OP_AUTO_ENCODING];
if ( ! $autoEncoding)
{
return $request;
}
if ($this->hasZlib)
{
return $request->withHeader('Accept-Encoding', 'gzip, deflate, identity');
}
return $request->withoutHeader('Accept-Encoding');
}
private function normalizeRequestHostHeader(Request $request, Uri $uri): Request
{
if ($request->hasHeader('Host'))
{
return $request;
}
$authority = $this->generateAuthorityFromUri($uri);
$request = $request->withHeader('Host', $this->normalizeHostHeader($authority));
return $request;
}
private function generateAuthorityFromUri(Uri $uri): string
{
$host = $uri->getHost();
$port = $uri->getPort();
return "{$host}:{$port}";
}
private function normalizeHostHeader(string $host): string
{
// Though servers are supposed to be able to handle standard port names on the end of the
// Host header some fail to do this correctly. As a result, we strip the port from the end
// if it's a standard 80 or 443
if (strpos($host, ':80') === strlen($host) - 3)
{
return substr($host, 0, -3);
} elseif (strpos($host, ':443') === strlen($host) - 4)
{
return substr($host, 0, -4);
}
return $host;
}
private function normalizeRequestUserAgent(Request $request): Request
{
if ($request->hasHeader('User-Agent'))
{
return $request;
}
return $request->withHeader('User-Agent', self::DEFAULT_USER_AGENT);
}
private function normalizeRequestAcceptHeader(Request $request): Request
{
if ($request->hasHeader('Accept'))
{
return $request;
}
return $request->withHeader('Accept', '*/*');
}
private function assignApplicableRequestCookies(Request $request): Request
{
$uri = new Uri($request->getUri());
$domain = $uri->getHost();
$path = $uri->getPath();
if ( ! $applicableCookies = $this->cookieJar->get($domain, $path))
{
// No cookies matched our request; we're finished.
return $request->withoutHeader("Cookie");
}
$isRequestSecure = strcasecmp($uri->getScheme(), "https") === 0;
$cookiePairs = [];
/** @var Cookie $cookie */
foreach ($applicableCookies as $cookie)
{
if ( ! $cookie->isSecure() || $isRequestSecure)
{
$cookiePairs[] = $cookie->getName() . "=" . $cookie->getValue();
}
}
if ($cookiePairs)
{
return $request->withHeader("Cookie", \implode("; ", $cookiePairs));
}
return $request->withoutHeader("Cookie");
}
private function normalizeTraceRequest(Request $request): Request
{
$method = $request->getMethod();
if ($method !== 'TRACE')
{
return $request;
}
// https://tools.ietf.org/html/rfc7231#section-4.3.8
/** @var Request $request */
$request = $request->withBody(NULL);
// Remove all body and sensitive headers
$request = $request->withHeaders([
"Transfer-Encoding" => [],
"Content-Length" => [],
"Authorization" => [],
"Proxy-Authorization" => [],
"Cookie" => [],
]);
return $request;
}
private function doRequest(Request $request, Uri $uri, array $options, Response $previousResponse = NULL, CancellationToken $cancellation): Promise
{
$deferred = new Deferred;
$requestCycle = new RequestCycle;
$requestCycle->request = $request;
$requestCycle->uri = $uri;
$requestCycle->options = $options;
$requestCycle->previousResponse = $previousResponse;
$requestCycle->deferred = $deferred;
$requestCycle->bodyDeferred = new Deferred;
$requestCycle->body = new Emitter;
$requestCycle->cancellation = $cancellation;
$protocolVersions = $request->getProtocolVersions();
if (\in_array("1.1", $protocolVersions, TRUE))
{
$requestCycle->protocolVersion = "1.1";
} elseif (\in_array("1.0", $protocolVersions, TRUE))
{
$requestCycle->protocolVersion = "1.0";
} else
{
return new Failure(new HttpException(
"None of the requested protocol versions are supported: " . \implode(", ", $protocolVersions)
));
}
asyncCall(function () use ($requestCycle) {
try
{
yield from $this->doWrite($requestCycle);
} catch (\Throwable $e)
{
$this->fail($requestCycle, $e);
}
});
return $deferred->promise();
}
private function doWrite(RequestCycle $requestCycle)
{
$timeout = $requestCycle->options[self::OP_TRANSFER_TIMEOUT];
$timeoutToken = new NullCancellationToken;
if ($timeout > 0)
{
$transferTimeoutWatcher = Loop::delay($timeout, function () use ($requestCycle, $timeout) {
$this->fail($requestCycle, new TimeoutException(
sprintf('Allowed transfer timeout exceeded: %d ms', $timeout)
));
});
$requestCycle->bodyDeferred->promise()->onResolve(static function () use ($transferTimeoutWatcher) {
Loop::cancel($transferTimeoutWatcher);
});
$timeoutToken = new TimeoutCancellationToken($timeout);
}
$authority = $this->generateAuthorityFromUri($requestCycle->uri);
$socketCheckoutUri = $requestCycle->uri->getScheme() . "://{$authority}";
$connectTimeoutToken = new CombinedCancellationToken($requestCycle->cancellation, $timeoutToken);
try
{
/** @var ClientSocket $socket */
$socket = yield $this->socketPool->checkout($socketCheckoutUri, $connectTimeoutToken);
$requestCycle->socket = $socket;
} catch (ResolutionException $dnsException)
{
throw new DnsException(\sprintf("Resolving the specified domain failed: '%s'", $requestCycle->uri->getHost()), 0, $dnsException);
} catch (ConnectException $e)
{
throw new SocketException(\sprintf("Connection to '%s' failed", $authority), 0, $e);
} catch (CancelledException $e)
{
// In case of a user cancellation request, throw the expected exception
$requestCycle->cancellation->throwIfRequested();
// Otherwise we ran into a timeout of our TimeoutCancellationToken
throw new SocketException(\sprintf("Connection to '%s' timed out", $authority), 0, $e);
}
$cancellation = $requestCycle->cancellation->subscribe(function ($error) use ($requestCycle) {
$this->fail($requestCycle, $error);
});
try
{
if ($requestCycle->uri->getScheme() === 'https')
{
$tlsContext = $this->tlsContext
->withPeerName($requestCycle->uri->getHost())
->withPeerCapturing();
yield $socket->enableCrypto($tlsContext);
}
// Collect this here, because it fails in case the remote closes the connection directly.
$connectionInfo = $this->collectConnectionInfo($socket);
$rawHeaders = $this->generateRawRequestHeaders($requestCycle->request, $requestCycle->protocolVersion);
yield $socket->write($rawHeaders);
$body = $requestCycle->request->getBody()->createBodyStream();
$chunking = $requestCycle->request->getHeader("transfer-encoding") === "chunked";
$remainingBytes = $requestCycle->request->getHeader("content-length");
if ($chunking && $requestCycle->protocolVersion === "1.0")
{
throw new HttpException("Can't send chunked bodies over HTTP/1.0");
}
// We always buffer the last chunk to make sure we don't write $contentLength bytes if the body is too long.
$buffer = "";
while (NULL !== $chunk = yield $body->read())
{
$requestCycle->cancellation->throwIfRequested();
if ($chunk === "")
{
continue;
}
if ($chunking)
{
$chunk = \dechex(\strlen($chunk)) . "\r\n" . $chunk . "\r\n";
}/* elseif ($remainingBytes !== null) {
$remainingBytes -= \strlen($chunk);
if ($remainingBytes < 0) {
throw new HttpException("Body contained more bytes than specified in Content-Length, aborting request");
}
}*/
yield $socket->write($buffer);
$buffer = $chunk;
}
// Flush last buffered chunk.
yield $socket->write($buffer);
if ($chunking)
{
yield $socket->write("0\r\n\r\n");
}/* elseif ($remainingBytes !== null && $remainingBytes > 0) {
throw new HttpException("Body contained fewer bytes than specified in Content-Length, aborting request");
}*/
yield from $this->doRead($requestCycle, $socket, $connectionInfo);
} finally
{
$requestCycle->cancellation->unsubscribe($cancellation);
}
}
private function fail(RequestCycle $requestCycle, \Throwable $error)
{
$toFails = [];
$socket = NULL;
if ($requestCycle->deferred)
{
$toFails[] = $requestCycle->deferred;
$requestCycle->deferred = NULL;
}
if ($requestCycle->body)
{
$toFails[] = $requestCycle->body;
$requestCycle->body = NULL;
}
if ($requestCycle->bodyDeferred)
{
$toFails[] = $requestCycle->bodyDeferred;
$requestCycle->bodyDeferred = NULL;
}
if ($requestCycle->socket)
{
$this->socketPool->clear($requestCycle->socket);
$socket = $requestCycle->socket;
$requestCycle->socket = NULL;
$socket->close();
}
foreach ($toFails as $toFail)
{
$toFail->fail($error);
}
}
private function collectConnectionInfo(ClientSocket $socket): ConnectionInfo
{
$crypto = \stream_get_meta_data($socket->getResource())["crypto"] ?? NULL;
return new ConnectionInfo(
$socket->getLocalAddress(),
$socket->getRemoteAddress(),
$crypto ? TlsInfo::fromMetaData($crypto, \stream_context_get_options($socket->getResource())["ssl"]) : NULL
);
}
/**
* @param Request $request
* @param string $protocolVersion
*
* @return string
*
* @TODO Send absolute URIs in the request line when using a proxy server
* Right now this doesn't matter because all proxy requests use a CONNECT
* tunnel but this likely will not always be the case.
*/
private function generateRawRequestHeaders(Request $request, string $protocolVersion): string
{
$uri = $request->getUri();
$uri = new Uri($uri);
$requestUri = $uri->getPath() ?: '/';
if ($query = $uri->getQuery())
{
$requestUri .= '?' . $query;
}
$head = $request->getMethod() . ' ' . $requestUri . ' HTTP/' . $protocolVersion . "\r\n";
$headers = $request->getHeaders(TRUE);
foreach ($headers as $field => $values)
{
if (\strcspn($field, "\r\n") !== \strlen($field))
{
throw new HttpException("Blocked header injection attempt for header '{$field}'");
}
foreach ($values as $value)
{
if (\strcspn($value, "\r\n") !== \strlen($value))
{
throw new HttpException("Blocked header injection attempt for header '{$field}' with value '{$value}'");
}
$head .= "{$field}: {$value}\r\n";
}
}
$head .= "\r\n";
return $head;
}
private function doRead(RequestCycle $requestCycle, ClientSocket $socket, ConnectionInfo $connectionInfo): \Generator
{
try
{
$backpressure = new Success;
$bodyCallback = $requestCycle->options[self::OP_DISCARD_BODY]
? NULL
: static function ($data) use ($requestCycle, &$backpressure) {
$backpressure = $requestCycle->body->emit($data);
};
$parser = new Parser($bodyCallback);
$parser->enqueueResponseMethodMatch($requestCycle->request->getMethod());
$parser->setAllOptions([
Parser::OP_MAX_HEADER_BYTES => $requestCycle->options[self::OP_MAX_HEADER_BYTES],
Parser::OP_MAX_BODY_BYTES => $requestCycle->options[self::OP_MAX_BODY_BYTES],
]);
while (NULL !== $chunk = yield $socket->read())
{
$requestCycle->cancellation->throwIfRequested();
$parseResult = $parser->parse($chunk);
if ( ! $parseResult)
{
continue;
}
$parseResult["headers"] = \array_change_key_case($parseResult["headers"], \CASE_LOWER);
$response = $this->finalizeResponse($requestCycle, $parseResult, $connectionInfo);
$shouldCloseSocketAfterResponse = $this->shouldCloseSocketAfterResponse($response);
$ignoreIncompleteBodyCheck = FALSE;
$responseHeaders = $response->getHeaders();
if ($requestCycle->deferred)
{
$deferred = $requestCycle->deferred;
$requestCycle->deferred = NULL;
$deferred->resolve($response);
$response = NULL; // clear references
$deferred = NULL; // there's also a reference in the deferred
} else
{
return;
}
// Required, otherwise responses without body hang
if ($parseResult["headersOnly"])
{
// Directly parse again in case we already have the full body but aborted parsing
// to resolve promise with headers.
$chunk = NULL;
do
{
try
{
$parseResult = $parser->parse($chunk);
} catch (ParseException $e)
{
$this->fail($requestCycle, $e);
throw $e;
}
if ($parseResult)
{
break;
}
if ( ! $backpressure instanceof Success)
{
yield $this->withCancellation($backpressure, $requestCycle->cancellation);
}
if ($requestCycle->bodyTooLarge)
{
throw new HttpException("Response body exceeded the specified size limit");
}
} while (NULL !== $chunk = yield $socket->read());
$parserState = $parser->getState();
if ($parserState !== Parser::AWAITING_HEADERS)
{
// Ignore check if neither content-length nor chunked encoding are given.
$ignoreIncompleteBodyCheck = $parserState === Parser::BODY_IDENTITY_EOF &&
! isset($responseHeaders["content-length"]) &&
strcasecmp('identity', $responseHeaders['transfer-encoding'][0] ?? "");
if ( ! $ignoreIncompleteBodyCheck)
{
throw new SocketException(sprintf(
'Socket disconnected prior to response completion (Parser state: %s)',
$parserState
));
}
}
}
if ($shouldCloseSocketAfterResponse || $ignoreIncompleteBodyCheck)
{
$this->socketPool->clear($socket);
$socket->close();
} else
{
$this->socketPool->checkin($socket);
}
$requestCycle->socket = NULL;
// Complete body AFTER socket checkin, so the socket can be reused for a potential redirect
$body = $requestCycle->body;
$requestCycle->body = NULL;
$bodyDeferred = $requestCycle->bodyDeferred;
$requestCycle->bodyDeferred = NULL;
$body->complete();
$bodyDeferred->resolve();
return;
}
} catch (\Throwable $e)
{
$this->fail($requestCycle, $e);
return;
}
if ($socket->getResource() !== NULL)
{
$requestCycle->socket = NULL;
$this->socketPool->clear($socket);
$socket->close();
}
// Required, because if the write fails, the read() call immediately resolves.
yield new Delayed(0);
if ($requestCycle->deferred === NULL)
{
return;
}
$parserState = $parser->getState();
if ($parserState === Parser::AWAITING_HEADERS && $requestCycle->retryCount < 1)
{
$requestCycle->retryCount++;
yield from $this->doWrite($requestCycle);
} else
{
$this->fail($requestCycle, new SocketException(sprintf(
'Socket disconnected prior to response completion (Parser state: %s)',
$parserState
)));
}
}
private function finalizeResponse(RequestCycle $requestCycle, array $parserResult, ConnectionInfo $connectionInfo)
{
$body = new IteratorStream($requestCycle->body->iterate());
if ($encoding = $this->determineCompressionEncoding($parserResult["headers"]))
{
$body = new ZlibInputStream($body, $encoding);
}
// Wrap the input stream so we can discard the body in case it's destructed but hasn't been consumed.
// This allows reusing the connection for further requests. It's important to have __destruct in InputStream and
// not in Message, because an InputStream might be pulled out of Message and used separately.
$body = new class($body, $requestCycle, $this->socketPool) implements InputStream {
private $body;
private $bodySize = 0;
private $requestCycle;
private $socketPool;
private $successfulEnd = FALSE;
public function __construct(InputStream $body, RequestCycle $requestCycle, HttpSocketPool $socketPool)
{
$this->body = $body;
$this->requestCycle = $requestCycle;
$this->socketPool = $socketPool;
}
public function read(): Promise
{
$promise = $this->body->read();
$promise->onResolve(function ($error, $value) {
if ($value !== NULL)
{
$this->bodySize += \strlen($value);
$maxBytes = $this->requestCycle->options[Client::OP_MAX_BODY_BYTES];
if ($maxBytes !== 0 && $this->bodySize >= $maxBytes)
{
$this->requestCycle->bodyTooLarge = TRUE;
}
} elseif ($error === NULL)
{
$this->successfulEnd = TRUE;
}
});
return $promise;
}
public function __destruct()
{
if ( ! $this->successfulEnd && $this->requestCycle->socket)
{
$this->socketPool->clear($this->requestCycle->socket);
$socket = $this->requestCycle->socket;
$this->requestCycle->socket = NULL;
$socket->close();
}
}
};
$response = new class($parserResult["protocol"], $parserResult["status"], $parserResult["reason"], $parserResult["headers"], $body, $requestCycle->request, $requestCycle->previousResponse, new MetaInfo($connectionInfo)) implements Response {
private $protocolVersion;
private $status;
private $reason;
private $request;
private $previousResponse;
private $headers;
private $body;
private $metaInfo;
public function __construct(
string $protocolVersion,
int $status,
string $reason,
array $headers,
InputStream $body,
Request $request,
Response $previousResponse = NULL,
MetaInfo $metaInfo
)
{
$this->protocolVersion = $protocolVersion;
$this->status = $status;
$this->reason = $reason;
$this->headers = $headers;
$this->body = new Message($body);
$this->request = $request;
$this->previousResponse = $previousResponse;
$this->metaInfo = $metaInfo;
}
public function getProtocolVersion(): string
{
return $this->protocolVersion;
}
public function getStatus(): int
{
return $this->status;
}
public function getReason(): string
{
return $this->reason;
}
public function getRequest(): Request
{
return $this->request;
}
public function getOriginalRequest(): Request
{
if (empty($this->previousResponse))
{
return $this->request;
}
return $this->previousResponse->getOriginalRequest();
}
public function getPreviousResponse()
{
return $this->previousResponse;
}
public function hasHeader(string $field): bool
{
return isset($this->headers[\strtolower($field)]);
}
public function getHeader(string $field)
{
return $this->headers[\strtolower($field)][0] ?? NULL;
}
public function getHeaderArray(string $field): array
{
return $this->headers[\strtolower($field)] ?? [];
}
public function getHeaders(): array
{
return $this->headers;
}
public function getBody(): Message
{
return $this->body;
}
public function getMetaInfo(): MetaInfo
{
return $this->metaInfo;
}
};
if ($response->hasHeader('Set-Cookie'))
{
$requestDomain = $requestCycle->uri->getHost();
$cookies = $response->getHeaderArray('Set-Cookie');
foreach ($cookies as $rawCookieStr)
{
$this->storeResponseCookie($requestDomain, $rawCookieStr);
}
}
return $response;
}
private function determineCompressionEncoding(array $responseHeaders): int
{
if ( ! $this->hasZlib)
{
return 0;
}
if ( ! isset($responseHeaders["content-encoding"]))
{
return 0;
}
$contentEncodingHeader = \trim(\current($responseHeaders["content-encoding"]));
if (strcasecmp($contentEncodingHeader, 'gzip') === 0)
{
return \ZLIB_ENCODING_GZIP;
}
if (strcasecmp($contentEncodingHeader, 'deflate') === 0)
{
return \ZLIB_ENCODING_DEFLATE;
}
return 0;
}
private function storeResponseCookie(string $requestDomain, string $rawCookieStr)
{
try
{
$cookie = Cookie::fromString($rawCookieStr);
if ( ! $cookie->getDomain())
{
$cookie = $cookie->withDomain($requestDomain);
} else
{
// https://tools.ietf.org/html/rfc6265#section-4.1.2.3
$cookieDomain = $cookie->getDomain();
// If a domain is set, left dots are ignored and it's always a wildcard
$cookieDomain = \ltrim($cookieDomain, ".");
if ($cookieDomain !== $requestDomain)
{
// ignore cookies on domains that are public suffixes
if (PublicSuffixList::isPublicSuffix($cookieDomain))
{
return;
}
// cookie origin would not be included when sending the cookie
if (\substr($requestDomain, 0, -\strlen($cookieDomain) - 1) . "." . $cookieDomain !== $requestDomain)
{
return;
}
}
// always add the dot, it's used internally for wildcard matching when an explicit domain is sent
$cookie = $cookie->withDomain("." . $cookieDomain);
}
$this->cookieJar->store($cookie);
} catch (CookieFormatException $e)
{
// Ignore malformed Set-Cookie headers
}
}
private function shouldCloseSocketAfterResponse(Response $response)
{
$request = $response->getRequest();
$requestConnHeader = $request->getHeader('Connection');
$responseConnHeader = $response->getHeader('Connection');
if ($requestConnHeader && ! strcasecmp($requestConnHeader, 'close'))
{
return TRUE;
} elseif ($responseConnHeader && ! strcasecmp($responseConnHeader, 'close'))
{
return TRUE;
} elseif ($response->getProtocolVersion() === '1.0' && ! $responseConnHeader)
{
return TRUE;
}
return FALSE;
}
private function withCancellation(Promise $promise, CancellationToken $cancellationToken): Promise
{
$deferred = new Deferred;
$newPromise = $deferred->promise();
$promise->onResolve(function ($error, $value) use (&$deferred) {
if ($deferred)
{
if ($error)
{
$deferred->fail($error);
$deferred = NULL;
} else
{
$deferred->resolve($value);
$deferred = NULL;
}
}
});
$cancellationSubscription = $cancellationToken->subscribe(function ($e) use (&$deferred) {
if ($deferred)
{
$deferred->fail($e);
$deferred = NULL;
}
});
$newPromise->onResolve(function () use ($cancellationToken, $cancellationSubscription) {
$cancellationToken->unsubscribe($cancellationSubscription);
});
return $newPromise;
}
private function getRedirectUri(Response $response)
{
if ( ! $response->hasHeader('Location'))
{
return NULL;
}
$request = $response->getRequest();
$status = $response->getStatus();
$method = $request->getMethod();
if ($status < 300 || $status > 399 || $method === 'HEAD')
{
return NULL;
}
$requestUri = new Uri($request->getUri());
$redirectLocation = $response->getHeader('Location');
try
{
return $requestUri->resolve($redirectLocation);
} catch (InvalidUriException $e)
{
return NULL;
}
}
/**
* Clients must not add a Referer header when leaving an unencrypted resource and redirecting to an encrypted
* resource.
*
* @param Request $request
* @param string $refererUri
* @param string $newUri
*
* @return Request
*
* @link http://www.w3.org/Protocols/rfc2616/rfc2616-sec15.html#sec15.1.3
*/
private function assignRedirectRefererHeader(Request $request, string $refererUri, string $newUri): Request
{
$refererIsEncrypted = (\stripos($refererUri, 'https') === 0);
$destinationIsEncrypted = (\stripos($newUri, 'https') === 0);
if ( ! $refererIsEncrypted || $destinationIsEncrypted)
{
return $request->withHeader('Referer', $refererUri);
}
return $request->withoutHeader('Referer');
}
/**
* Set multiple options at once.
*
* @param array $options An array of the form [OP_CONSTANT => $value]
*
* @throws \Error On unknown option key or invalid value.
*/
public function setOptions(array $options)
{
foreach ($options as $option => $value)
{
$this->setOption($option, $value);
}
}
/**
* Set an option.
*
* @param string $option A Client option constant
* @param mixed $value The option value to assign
*
* @throws \Error On unknown option key or invalid value.
*/
public function setOption(string $option, $value)
{
$this->validateOption($option, $value);
$this->options[$option] = $value;
}
}
Reference in New Issue View Git Blame Copy Permalink