123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276 |
- <?php
- namespace Guzzle\Stream;
- use Guzzle\Common\Exception\InvalidArgumentException;
- use Guzzle\Common\Exception\RuntimeException;
- use Guzzle\Http\Message\EntityEnclosingRequestInterface;
- use Guzzle\Http\Message\RequestInterface;
- use Guzzle\Http\Url;
- /**
- * Factory used to create fopen streams using PHP's http and https stream wrappers
- *
- * Note: PHP's http stream wrapper only supports streaming downloads. It does not support streaming uploads.
- */
- class PhpStreamRequestFactory implements StreamRequestFactoryInterface
- {
- /** @var resource Stream context options */
- protected $context;
- /** @var array Stream context */
- protected $contextOptions;
- /** @var Url Stream URL */
- protected $url;
- /** @var array Last response headers received by the HTTP request */
- protected $lastResponseHeaders;
- /**
- * {@inheritdoc}
- *
- * The $params array can contain the following custom keys specific to the PhpStreamRequestFactory:
- * - stream_class: The name of a class to create instead of a Guzzle\Stream\Stream object
- */
- public function fromRequest(RequestInterface $request, $context = array(), array $params = array())
- {
- if (is_resource($context)) {
- $this->contextOptions = stream_context_get_options($context);
- $this->context = $context;
- } elseif (is_array($context) || !$context) {
- $this->contextOptions = $context;
- $this->createContext($params);
- } elseif ($context) {
- throw new InvalidArgumentException('$context must be an array or resource');
- }
- // Dispatch the before send event
- $request->dispatch('request.before_send', array(
- 'request' => $request,
- 'context' => $this->context,
- 'context_options' => $this->contextOptions
- ));
- $this->setUrl($request);
- $this->addDefaultContextOptions($request);
- $this->addSslOptions($request);
- $this->addBodyOptions($request);
- $this->addProxyOptions($request);
- // Create the file handle but silence errors
- return $this->createStream($params)
- ->setCustomData('request', $request)
- ->setCustomData('response_headers', $this->getLastResponseHeaders());
- }
- /**
- * Set an option on the context and the internal options array
- *
- * @param string $wrapper Stream wrapper name of http
- * @param string $name Context name
- * @param mixed $value Context value
- * @param bool $overwrite Set to true to overwrite an existing value
- */
- protected function setContextValue($wrapper, $name, $value, $overwrite = false)
- {
- if (!isset($this->contextOptions[$wrapper])) {
- $this->contextOptions[$wrapper] = array($name => $value);
- } elseif (!$overwrite && isset($this->contextOptions[$wrapper][$name])) {
- return;
- }
- $this->contextOptions[$wrapper][$name] = $value;
- stream_context_set_option($this->context, $wrapper, $name, $value);
- }
- /**
- * Create a stream context
- *
- * @param array $params Parameter array
- */
- protected function createContext(array $params)
- {
- $options = $this->contextOptions;
- $this->context = $this->createResource(function () use ($params, $options) {
- return stream_context_create($options, $params);
- });
- }
- /**
- * Get the last response headers received by the HTTP request
- *
- * @return array
- */
- public function getLastResponseHeaders()
- {
- return $this->lastResponseHeaders;
- }
- /**
- * Adds the default context options to the stream context options
- *
- * @param RequestInterface $request Request
- */
- protected function addDefaultContextOptions(RequestInterface $request)
- {
- $this->setContextValue('http', 'method', $request->getMethod());
- $headers = $request->getHeaderLines();
- // "Connection: close" is required to get streams to work in HTTP 1.1
- if (!$request->hasHeader('Connection')) {
- $headers[] = 'Connection: close';
- }
- $this->setContextValue('http', 'header', $headers);
- $this->setContextValue('http', 'protocol_version', $request->getProtocolVersion());
- $this->setContextValue('http', 'ignore_errors', true);
- }
- /**
- * Set the URL to use with the factory
- *
- * @param RequestInterface $request Request that owns the URL
- */
- protected function setUrl(RequestInterface $request)
- {
- $this->url = $request->getUrl(true);
- // Check for basic Auth username
- if ($request->getUsername()) {
- $this->url->setUsername($request->getUsername());
- }
- // Check for basic Auth password
- if ($request->getPassword()) {
- $this->url->setPassword($request->getPassword());
- }
- }
- /**
- * Add SSL options to the stream context
- *
- * @param RequestInterface $request Request
- */
- protected function addSslOptions(RequestInterface $request)
- {
- if ($request->getCurlOptions()->get(CURLOPT_SSL_VERIFYPEER)) {
- $this->setContextValue('ssl', 'verify_peer', true, true);
- if ($cafile = $request->getCurlOptions()->get(CURLOPT_CAINFO)) {
- $this->setContextValue('ssl', 'cafile', $cafile, true);
- }
- } else {
- $this->setContextValue('ssl', 'verify_peer', false, true);
- }
- }
- /**
- * Add body (content) specific options to the context options
- *
- * @param RequestInterface $request
- */
- protected function addBodyOptions(RequestInterface $request)
- {
- // Add the content for the request if needed
- if (!($request instanceof EntityEnclosingRequestInterface)) {
- return;
- }
- if (count($request->getPostFields())) {
- $this->setContextValue('http', 'content', (string) $request->getPostFields(), true);
- } elseif ($request->getBody()) {
- $this->setContextValue('http', 'content', (string) $request->getBody(), true);
- }
- // Always ensure a content-length header is sent
- if (isset($this->contextOptions['http']['content'])) {
- $headers = isset($this->contextOptions['http']['header']) ? $this->contextOptions['http']['header'] : array();
- $headers[] = 'Content-Length: ' . strlen($this->contextOptions['http']['content']);
- $this->setContextValue('http', 'header', $headers, true);
- }
- }
- /**
- * Add proxy parameters to the context if needed
- *
- * @param RequestInterface $request Request
- */
- protected function addProxyOptions(RequestInterface $request)
- {
- if ($proxy = $request->getCurlOptions()->get(CURLOPT_PROXY)) {
- $this->setContextValue('http', 'proxy', $proxy);
- }
- }
- /**
- * Create the stream for the request with the context options
- *
- * @param array $params Parameters of the stream
- *
- * @return StreamInterface
- */
- protected function createStream(array $params)
- {
- $http_response_header = null;
- $url = $this->url;
- $context = $this->context;
- $fp = $this->createResource(function () use ($context, $url, &$http_response_header) {
- return fopen((string) $url, 'r', false, $context);
- });
- // Determine the class to instantiate
- $className = isset($params['stream_class']) ? $params['stream_class'] : __NAMESPACE__ . '\\Stream';
- /** @var $stream StreamInterface */
- $stream = new $className($fp);
- // Track the response headers of the request
- if (isset($http_response_header)) {
- $this->lastResponseHeaders = $http_response_header;
- $this->processResponseHeaders($stream);
- }
- return $stream;
- }
- /**
- * Process response headers
- *
- * @param StreamInterface $stream
- */
- protected function processResponseHeaders(StreamInterface $stream)
- {
- // Set the size on the stream if it was returned in the response
- foreach ($this->lastResponseHeaders as $header) {
- if ((stripos($header, 'Content-Length:')) === 0) {
- $stream->setSize(trim(substr($header, 15)));
- }
- }
- }
- /**
- * Create a resource and check to ensure it was created successfully
- *
- * @param callable $callback Closure to invoke that must return a valid resource
- *
- * @return resource
- * @throws RuntimeException on error
- */
- protected function createResource($callback)
- {
- // Turn off error reporting while we try to initiate the request
- $level = error_reporting(0);
- $resource = call_user_func($callback);
- error_reporting($level);
- // If the resource could not be created, then grab the last error and throw an exception
- if (false === $resource) {
- $message = 'Error creating resource. ';
- foreach (error_get_last() as $key => $value) {
- $message .= "[{$key}] {$value} ";
- }
- throw new RuntimeException(trim($message));
- }
- return $resource;
- }
- }
|