391 lines
13 KiB
PHP
391 lines
13 KiB
PHP
<?php
|
|
|
|
/**
|
|
* Copyright (c) 2011-2018 Michael Dowling, https://github.com/mtdowling <mtdowling@gmail.com>
|
|
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
* in the Software without restriction, including without limitation the rights
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
* furnished to do so, subject to the following conditions:
|
|
|
|
* The above copyright notice and this permission notice shall be included in
|
|
* all copies or substantial portions of the Software.
|
|
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
* THE SOFTWARE.
|
|
*/
|
|
|
|
namespace Obs\Internal\Common;
|
|
|
|
use GuzzleHttp\Psr7;
|
|
use GuzzleHttp\Exception\ConnectException;
|
|
use GuzzleHttp\Exception\RequestException;
|
|
use GuzzleHttp\Promise\FulfilledPromise;
|
|
use GuzzleHttp\TransferStats;
|
|
use Psr\Http\Message\RequestInterface;
|
|
use Psr\Http\Message\ResponseInterface;
|
|
use Psr\Http\Message\StreamInterface;
|
|
|
|
class SdkStreamHandler
|
|
{
|
|
private $lastHeaders = [];
|
|
|
|
public function __invoke(RequestInterface $request, array $options)
|
|
{
|
|
if (isset($options['delay'])) {
|
|
usleep($options['delay'] * 1000);
|
|
}
|
|
|
|
$startTime = isset($options['on_stats']) ? microtime(true) : null;
|
|
|
|
try {
|
|
$request = $request->withoutHeader('Expect');
|
|
|
|
if (0 === $request->getBody()->getSize()) {
|
|
$request = $request->withHeader('Content-Length', 0);
|
|
}
|
|
|
|
return $this->createResponse(
|
|
$request,
|
|
$options,
|
|
$this->createStream($request, $options),
|
|
$startTime
|
|
);
|
|
} catch (\InvalidArgumentException $e) {
|
|
throw $e;
|
|
} catch (\Exception $e) {
|
|
$message = $e->getMessage();
|
|
if (strpos($message, 'getaddrinfo')
|
|
|| strpos($message, 'Connection refused')
|
|
|| strpos($message, "couldn't connect to host")
|
|
) {
|
|
$e = new ConnectException($e->getMessage(), $request, $e);
|
|
}
|
|
$e = RequestException::wrapException($request, $e);
|
|
$this->invokeStats($options, $request, $startTime, null, $e);
|
|
|
|
return \GuzzleHttp\Promise\rejection_for($e);
|
|
}
|
|
}
|
|
|
|
private function invokeStats(
|
|
array $options,
|
|
RequestInterface $request,
|
|
$startTime,
|
|
ResponseInterface $response = null,
|
|
$error = null
|
|
) {
|
|
if (isset($options['on_stats'])) {
|
|
$stats = new TransferStats(
|
|
$request,
|
|
$response,
|
|
microtime(true) - $startTime,
|
|
$error,
|
|
[]
|
|
);
|
|
call_user_func($options['on_stats'], $stats);
|
|
}
|
|
}
|
|
|
|
private function createResponse(
|
|
RequestInterface $request,
|
|
array $options,
|
|
$stream,
|
|
$startTime
|
|
) {
|
|
$hdrs = $this->lastHeaders;
|
|
$this->lastHeaders = [];
|
|
$parts = explode(' ', array_shift($hdrs), 3);
|
|
$ver = explode('/', $parts[0])[1];
|
|
$status = $parts[1];
|
|
$reason = isset($parts[2]) ? $parts[2] : null;
|
|
$headers = \GuzzleHttp\headers_from_lines($hdrs);
|
|
list($stream, $headers) = $this->checkDecode($options, $headers, $stream);
|
|
try {
|
|
$stream = Psr7\stream_for($stream);
|
|
} catch (\Throwable $e) {
|
|
$stream = Psr7\Utils::streamFor($stream);
|
|
}
|
|
$sink = $stream;
|
|
|
|
if (strcasecmp('HEAD', $request->getMethod())) {
|
|
$sink = $this->createSink($stream, $options);
|
|
}
|
|
|
|
$response = new Psr7\Response($status, $headers, $sink, $ver, $reason);
|
|
|
|
if (isset($options['on_headers'])) {
|
|
try {
|
|
$options['on_headers']($response);
|
|
} catch (\Exception $e) {
|
|
$msg = 'An error was encountered during the on_headers event';
|
|
$ex = new RequestException($msg, $request, $response, $e);
|
|
return \GuzzleHttp\Promise\rejection_for($ex);
|
|
}
|
|
}
|
|
|
|
if ($sink !== $stream) {
|
|
$this->drain(
|
|
$stream,
|
|
$sink,
|
|
$response->getHeaderLine('Content-Length')
|
|
);
|
|
}
|
|
|
|
$this->invokeStats($options, $request, $startTime, $response, null);
|
|
|
|
return new FulfilledPromise($response);
|
|
}
|
|
|
|
private function createSink(StreamInterface $stream, array $options)
|
|
{
|
|
if (!empty($options['stream'])) {
|
|
return $stream;
|
|
}
|
|
|
|
$sink = isset($options['sink'])
|
|
? $options['sink']
|
|
: fopen('php://temp', 'r+');
|
|
|
|
if (is_string($sink)) {
|
|
return new Psr7\LazyOpenStream($sink, 'w+');
|
|
}
|
|
|
|
try {
|
|
return Psr7\stream_for($sink);
|
|
} catch (\Throwable $e) {
|
|
return Psr7\Utils::streamFor($sink);
|
|
}
|
|
}
|
|
|
|
private function checkDecode(array $options, array $headers, $stream)
|
|
{
|
|
if (!empty($options['decode_content'])) {
|
|
$normalizedKeys = \GuzzleHttp\normalize_header_keys($headers);
|
|
if (isset($normalizedKeys['content-encoding'])) {
|
|
$encoding = $headers[$normalizedKeys['content-encoding']];
|
|
if ($encoding[0] === 'gzip' || $encoding[0] === 'deflate') {
|
|
try {
|
|
$stream = new Psr7\InflateStream(
|
|
Psr7\stream_for($stream)
|
|
);
|
|
} catch (\Throwable $th) {
|
|
$stream = new Psr7\InflateStream(
|
|
Psr7\Utils::streamFor($stream)
|
|
);
|
|
}
|
|
|
|
$headers['x-encoded-content-encoding']
|
|
= $headers[$normalizedKeys['content-encoding']];
|
|
unset($headers[$normalizedKeys['content-encoding']]);
|
|
if (isset($normalizedKeys['content-length'])) {
|
|
$headers['x-encoded-content-length']
|
|
= $headers[$normalizedKeys['content-length']];
|
|
|
|
$length = (int) $stream->getSize();
|
|
if ($length === 0) {
|
|
unset($headers[$normalizedKeys['content-length']]);
|
|
} else {
|
|
$headers[$normalizedKeys['content-length']] = [$length];
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return [$stream, $headers];
|
|
}
|
|
|
|
private function drain(
|
|
StreamInterface $source,
|
|
StreamInterface $sink,
|
|
$contentLength
|
|
) {
|
|
Psr7\copy_to_stream(
|
|
$source,
|
|
$sink,
|
|
(strlen($contentLength) > 0 && (int) $contentLength > 0) ? (int) $contentLength : -1
|
|
);
|
|
|
|
$sink->seek(0);
|
|
$source->close();
|
|
|
|
return $sink;
|
|
}
|
|
|
|
private function createResource(callable $callback)
|
|
{
|
|
$errors = null;
|
|
set_error_handler(function ($_, $msg, $file, $line) use (&$errors) {
|
|
$errors[] = [
|
|
'message' => $msg,
|
|
'file' => $file,
|
|
'line' => $line
|
|
];
|
|
return true;
|
|
});
|
|
|
|
$resource = $callback();
|
|
restore_error_handler();
|
|
|
|
if (!$resource) {
|
|
$message = 'Error creating resource: ';
|
|
foreach ($errors as $err) {
|
|
foreach ($err as $key => $value) {
|
|
$message .= "[$key] $value" . PHP_EOL;
|
|
}
|
|
}
|
|
throw new \UnexpectedValueException(trim($message));
|
|
}
|
|
|
|
return $resource;
|
|
}
|
|
|
|
private function createStream(RequestInterface $request, array $options)
|
|
{
|
|
static $methods;
|
|
if (!$methods) {
|
|
$methods = array_flip(get_class_methods(__CLASS__));
|
|
}
|
|
|
|
if ($request->getProtocolVersion() == '1.1'
|
|
&& !$request->hasHeader('Connection')
|
|
) {
|
|
$request = $request->withHeader('Connection', 'close');
|
|
}
|
|
|
|
if (!isset($options['verify'])) {
|
|
$options['verify'] = true;
|
|
}
|
|
|
|
$params = [];
|
|
$context = $this->getDefaultContext($request, $options);
|
|
|
|
if (isset($options['on_headers']) && !is_callable($options['on_headers'])) {
|
|
throw new \InvalidArgumentException('on_headers must be callable');
|
|
}
|
|
|
|
if (!empty($options)) {
|
|
foreach ($options as $key => $value) {
|
|
$method = "add_{$key}";
|
|
if (isset($methods[$method])) {
|
|
$this->{$method}($request, $context, $value, $params);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (isset($options['stream_context'])) {
|
|
if (!is_array($options['stream_context'])) {
|
|
throw new \InvalidArgumentException('stream_context must be an array');
|
|
}
|
|
$context = array_replace_recursive(
|
|
$context,
|
|
$options['stream_context']
|
|
);
|
|
}
|
|
|
|
if (isset($options['auth'])
|
|
&& is_array($options['auth'])
|
|
&& isset($options['auth'][2])
|
|
&& 'ntlm' == $options['auth'][2]
|
|
) {
|
|
|
|
throw new \InvalidArgumentException('Microsoft NTLM authentication only supported with curl handler');
|
|
}
|
|
|
|
$uri = $this->resolveHost($request, $options);
|
|
|
|
$context = $this->createResource(
|
|
function () use ($context, $params) {
|
|
return stream_context_create($context, $params);
|
|
}
|
|
);
|
|
|
|
return $this->createResource(
|
|
function () use ($uri, &$http_response_header, $context, $options) {
|
|
$resource = fopen((string) $uri, 'r', false, $context);
|
|
$this->lastHeaders = $http_response_header;
|
|
|
|
if (isset($options['read_timeout'])) {
|
|
$readTimeout = $options['read_timeout'];
|
|
$sec = (int) $readTimeout;
|
|
$usec = ($readTimeout - $sec) * 100000;
|
|
stream_set_timeout($resource, $sec, $usec);
|
|
}
|
|
|
|
return $resource;
|
|
}
|
|
);
|
|
}
|
|
|
|
private function resolveHost(RequestInterface $request, array $options)
|
|
{
|
|
$uri = $request->getUri();
|
|
|
|
if (isset($options['force_ip_resolve']) && !filter_var($uri->getHost(), FILTER_VALIDATE_IP)) {
|
|
if ('v4' === $options['force_ip_resolve']) {
|
|
$records = dns_get_record($uri->getHost(), DNS_A);
|
|
if (!isset($records[0]['ip'])) {
|
|
throw new ConnectException(
|
|
sprintf("Could not resolve IPv4 address for host '%s'", $uri->getHost()),
|
|
$request
|
|
);
|
|
}
|
|
$uri = $uri->withHost($records[0]['ip']);
|
|
}
|
|
if ('v6' === $options['force_ip_resolve']) {
|
|
$records = dns_get_record($uri->getHost(), DNS_AAAA);
|
|
if (!isset($records[0]['ipv6'])) {
|
|
throw new ConnectException(
|
|
sprintf("Could not resolve IPv6 address for host '%s'", $uri->getHost()),
|
|
$request
|
|
);
|
|
}
|
|
$uri = $uri->withHost('[' . $records[0]['ipv6'] . ']');
|
|
}
|
|
}
|
|
|
|
return $uri;
|
|
}
|
|
|
|
private function getDefaultContext(RequestInterface $request)
|
|
{
|
|
$headers = '';
|
|
foreach ($request->getHeaders() as $name => $value) {
|
|
foreach ($value as $val) {
|
|
$headers .= "$name: $val\r\n";
|
|
}
|
|
}
|
|
|
|
$context = [
|
|
'http' => [
|
|
'method' => $request->getMethod(),
|
|
'header' => $headers,
|
|
'protocol_version' => $request->getProtocolVersion(),
|
|
'ignore_errors' => true,
|
|
'follow_location' => 0,
|
|
],
|
|
];
|
|
|
|
$body = (string) $request->getBody();
|
|
|
|
if (!empty($body)) {
|
|
$context['http']['content'] = $body;
|
|
if (!$request->hasHeader('Content-Type')) {
|
|
$context['http']['header'] .= "Content-Type:\r\n";
|
|
}
|
|
}
|
|
|
|
$context['http']['header'] = rtrim($context['http']['header']);
|
|
|
|
return $context;
|
|
}
|
|
}
|