123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542 |
- <?php
- /**
- * cURL HTTP transport
- *
- * @package Requests
- * @subpackage Transport
- */
- /**
- * cURL HTTP transport
- *
- * @package Requests
- * @subpackage Transport
- */
- class Requests_Transport_cURL implements Requests_Transport {
- const CURL_7_10_5 = 0x070A05;
- const CURL_7_16_2 = 0x071002;
- /**
- * Raw HTTP data
- *
- * @var string
- */
- public $headers = '';
- /**
- * Raw body data
- *
- * @var string
- */
- public $response_data = '';
- /**
- * Information on the current request
- *
- * @var array cURL information array, see {@see https://secure.php.net/curl_getinfo}
- */
- public $info;
- /**
- * Version string
- *
- * @var long
- */
- public $version;
- /**
- * cURL handle
- *
- * @var resource
- */
- protected $handle;
- /**
- * Hook dispatcher instance
- *
- * @var Requests_Hooks
- */
- protected $hooks;
- /**
- * Have we finished the headers yet?
- *
- * @var boolean
- */
- protected $done_headers = false;
- /**
- * If streaming to a file, keep the file pointer
- *
- * @var resource
- */
- protected $stream_handle;
- /**
- * How many bytes are in the response body?
- *
- * @var int
- */
- protected $response_bytes;
- /**
- * What's the maximum number of bytes we should keep?
- *
- * @var int|bool Byte count, or false if no limit.
- */
- protected $response_byte_limit;
- /**
- * Constructor
- */
- public function __construct() {
- $curl = curl_version();
- $this->version = $curl['version_number'];
- $this->handle = curl_init();
- curl_setopt($this->handle, CURLOPT_HEADER, false);
- curl_setopt($this->handle, CURLOPT_RETURNTRANSFER, 1);
- if ($this->version >= self::CURL_7_10_5) {
- curl_setopt($this->handle, CURLOPT_ENCODING, '');
- }
- if (defined('CURLOPT_PROTOCOLS')) {
- curl_setopt($this->handle, CURLOPT_PROTOCOLS, CURLPROTO_HTTP | CURLPROTO_HTTPS);
- }
- if (defined('CURLOPT_REDIR_PROTOCOLS')) {
- curl_setopt($this->handle, CURLOPT_REDIR_PROTOCOLS, CURLPROTO_HTTP | CURLPROTO_HTTPS);
- }
- }
- /**
- * Destructor
- */
- public function __destruct() {
- if (is_resource($this->handle)) {
- curl_close($this->handle);
- }
- }
- /**
- * Perform a request
- *
- * @throws Requests_Exception On a cURL error (`curlerror`)
- *
- * @param string $url URL to request
- * @param array $headers Associative array of request headers
- * @param string|array $data Data to send either as the POST body, or as parameters in the URL for a GET/HEAD
- * @param array $options Request options, see {@see Requests::response()} for documentation
- * @return string Raw HTTP result
- */
- public function request($url, $headers = array(), $data = array(), $options = array()) {
- $this->hooks = $options['hooks'];
- $this->setup_handle($url, $headers, $data, $options);
- $options['hooks']->dispatch('curl.before_send', array(&$this->handle));
- if ($options['filename'] !== false) {
- $this->stream_handle = fopen($options['filename'], 'wb');
- }
- $this->response_data = '';
- $this->response_bytes = 0;
- $this->response_byte_limit = false;
- if ($options['max_bytes'] !== false) {
- $this->response_byte_limit = $options['max_bytes'];
- }
- if (isset($options['verify'])) {
- if ($options['verify'] === false) {
- curl_setopt($this->handle, CURLOPT_SSL_VERIFYHOST, 0);
- curl_setopt($this->handle, CURLOPT_SSL_VERIFYPEER, 0);
- }
- elseif (is_string($options['verify'])) {
- curl_setopt($this->handle, CURLOPT_CAINFO, $options['verify']);
- }
- }
- if (isset($options['verifyname']) && $options['verifyname'] === false) {
- curl_setopt($this->handle, CURLOPT_SSL_VERIFYHOST, 0);
- }
- curl_exec($this->handle);
- $response = $this->response_data;
- $options['hooks']->dispatch('curl.after_send', array());
- if (curl_errno($this->handle) === 23 || curl_errno($this->handle) === 61) {
- // Reset encoding and try again
- curl_setopt($this->handle, CURLOPT_ENCODING, 'none');
- $this->response_data = '';
- $this->response_bytes = 0;
- curl_exec($this->handle);
- $response = $this->response_data;
- }
- $this->process_response($response, $options);
- // Need to remove the $this reference from the curl handle.
- // Otherwise Requests_Transport_cURL wont be garbage collected and the curl_close() will never be called.
- curl_setopt($this->handle, CURLOPT_HEADERFUNCTION, null);
- curl_setopt($this->handle, CURLOPT_WRITEFUNCTION, null);
- return $this->headers;
- }
- /**
- * Send multiple requests simultaneously
- *
- * @param array $requests Request data
- * @param array $options Global options
- * @return array Array of Requests_Response objects (may contain Requests_Exception or string responses as well)
- */
- public function request_multiple($requests, $options) {
- // If you're not requesting, we can't get any responses ¯\_(ツ)_/¯
- if (empty($requests)) {
- return array();
- }
- $multihandle = curl_multi_init();
- $subrequests = array();
- $subhandles = array();
- $class = get_class($this);
- foreach ($requests as $id => $request) {
- $subrequests[$id] = new $class();
- $subhandles[$id] = $subrequests[$id]->get_subrequest_handle($request['url'], $request['headers'], $request['data'], $request['options']);
- $request['options']['hooks']->dispatch('curl.before_multi_add', array(&$subhandles[$id]));
- curl_multi_add_handle($multihandle, $subhandles[$id]);
- }
- $completed = 0;
- $responses = array();
- $request['options']['hooks']->dispatch('curl.before_multi_exec', array(&$multihandle));
- do {
- $active = false;
- do {
- $status = curl_multi_exec($multihandle, $active);
- }
- while ($status === CURLM_CALL_MULTI_PERFORM);
- $to_process = array();
- // Read the information as needed
- while ($done = curl_multi_info_read($multihandle)) {
- $key = array_search($done['handle'], $subhandles, true);
- if (!isset($to_process[$key])) {
- $to_process[$key] = $done;
- }
- }
- // Parse the finished requests before we start getting the new ones
- foreach ($to_process as $key => $done) {
- $options = $requests[$key]['options'];
- if (CURLE_OK !== $done['result']) {
- //get error string for handle.
- $reason = curl_error($done['handle']);
- $exception = new Requests_Exception_Transport_cURL(
- $reason,
- Requests_Exception_Transport_cURL::EASY,
- $done['handle'],
- $done['result']
- );
- $responses[$key] = $exception;
- $options['hooks']->dispatch('transport.internal.parse_error', array(&$responses[$key], $requests[$key]));
- }
- else {
- $responses[$key] = $subrequests[$key]->process_response($subrequests[$key]->response_data, $options);
- $options['hooks']->dispatch('transport.internal.parse_response', array(&$responses[$key], $requests[$key]));
- }
- curl_multi_remove_handle($multihandle, $done['handle']);
- curl_close($done['handle']);
- if (!is_string($responses[$key])) {
- $options['hooks']->dispatch('multiple.request.complete', array(&$responses[$key], $key));
- }
- $completed++;
- }
- }
- while ($active || $completed < count($subrequests));
- $request['options']['hooks']->dispatch('curl.after_multi_exec', array(&$multihandle));
- curl_multi_close($multihandle);
- return $responses;
- }
- /**
- * Get the cURL handle for use in a multi-request
- *
- * @param string $url URL to request
- * @param array $headers Associative array of request headers
- * @param string|array $data Data to send either as the POST body, or as parameters in the URL for a GET/HEAD
- * @param array $options Request options, see {@see Requests::response()} for documentation
- * @return resource Subrequest's cURL handle
- */
- public function &get_subrequest_handle($url, $headers, $data, $options) {
- $this->setup_handle($url, $headers, $data, $options);
- if ($options['filename'] !== false) {
- $this->stream_handle = fopen($options['filename'], 'wb');
- }
- $this->response_data = '';
- $this->response_bytes = 0;
- $this->response_byte_limit = false;
- if ($options['max_bytes'] !== false) {
- $this->response_byte_limit = $options['max_bytes'];
- }
- $this->hooks = $options['hooks'];
- return $this->handle;
- }
- /**
- * Setup the cURL handle for the given data
- *
- * @param string $url URL to request
- * @param array $headers Associative array of request headers
- * @param string|array $data Data to send either as the POST body, or as parameters in the URL for a GET/HEAD
- * @param array $options Request options, see {@see Requests::response()} for documentation
- */
- protected function setup_handle($url, $headers, $data, $options) {
- $options['hooks']->dispatch('curl.before_request', array(&$this->handle));
- // Force closing the connection for old versions of cURL (<7.22).
- if ( ! isset( $headers['Connection'] ) ) {
- $headers['Connection'] = 'close';
- }
- $headers = Requests::flatten($headers);
- if (!empty($data)) {
- $data_format = $options['data_format'];
- if ($data_format === 'query') {
- $url = self::format_get($url, $data);
- $data = '';
- }
- elseif (!is_string($data)) {
- $data = http_build_query($data, null, '&');
- }
- }
- switch ($options['type']) {
- case Requests::POST:
- curl_setopt($this->handle, CURLOPT_POST, true);
- curl_setopt($this->handle, CURLOPT_POSTFIELDS, $data);
- break;
- case Requests::HEAD:
- curl_setopt($this->handle, CURLOPT_CUSTOMREQUEST, $options['type']);
- curl_setopt($this->handle, CURLOPT_NOBODY, true);
- break;
- case Requests::TRACE:
- curl_setopt($this->handle, CURLOPT_CUSTOMREQUEST, $options['type']);
- break;
- case Requests::PATCH:
- case Requests::PUT:
- case Requests::DELETE:
- case Requests::OPTIONS:
- default:
- curl_setopt($this->handle, CURLOPT_CUSTOMREQUEST, $options['type']);
- if (!empty($data)) {
- curl_setopt($this->handle, CURLOPT_POSTFIELDS, $data);
- }
- }
- // cURL requires a minimum timeout of 1 second when using the system
- // DNS resolver, as it uses `alarm()`, which is second resolution only.
- // There's no way to detect which DNS resolver is being used from our
- // end, so we need to round up regardless of the supplied timeout.
- //
- // https://github.com/curl/curl/blob/4f45240bc84a9aa648c8f7243be7b79e9f9323a5/lib/hostip.c#L606-L609
- $timeout = max($options['timeout'], 1);
- if (is_int($timeout) || $this->version < self::CURL_7_16_2) {
- curl_setopt($this->handle, CURLOPT_TIMEOUT, ceil($timeout));
- }
- else {
- curl_setopt($this->handle, CURLOPT_TIMEOUT_MS, round($timeout * 1000));
- }
- if (is_int($options['connect_timeout']) || $this->version < self::CURL_7_16_2) {
- curl_setopt($this->handle, CURLOPT_CONNECTTIMEOUT, ceil($options['connect_timeout']));
- }
- else {
- curl_setopt($this->handle, CURLOPT_CONNECTTIMEOUT_MS, round($options['connect_timeout'] * 1000));
- }
- curl_setopt($this->handle, CURLOPT_URL, $url);
- curl_setopt($this->handle, CURLOPT_REFERER, $url);
- curl_setopt($this->handle, CURLOPT_USERAGENT, $options['useragent']);
- if (!empty($headers)) {
- curl_setopt($this->handle, CURLOPT_HTTPHEADER, $headers);
- }
- if ($options['protocol_version'] === 1.1) {
- curl_setopt($this->handle, CURLOPT_HTTP_VERSION, CURL_HTTP_VERSION_1_1);
- }
- else {
- curl_setopt($this->handle, CURLOPT_HTTP_VERSION, CURL_HTTP_VERSION_1_0);
- }
- if (true === $options['blocking']) {
- curl_setopt($this->handle, CURLOPT_HEADERFUNCTION, array(&$this, 'stream_headers'));
- curl_setopt($this->handle, CURLOPT_WRITEFUNCTION, array(&$this, 'stream_body'));
- curl_setopt($this->handle, CURLOPT_BUFFERSIZE, Requests::BUFFER_SIZE);
- }
- }
- /**
- * Process a response
- *
- * @param string $response Response data from the body
- * @param array $options Request options
- * @return string HTTP response data including headers
- */
- public function process_response($response, $options) {
- if ($options['blocking'] === false) {
- $fake_headers = '';
- $options['hooks']->dispatch('curl.after_request', array(&$fake_headers));
- return false;
- }
- if ($options['filename'] !== false) {
- fclose($this->stream_handle);
- $this->headers = trim($this->headers);
- }
- else {
- $this->headers .= $response;
- }
- if (curl_errno($this->handle)) {
- $error = sprintf(
- 'cURL error %s: %s',
- curl_errno($this->handle),
- curl_error($this->handle)
- );
- throw new Requests_Exception($error, 'curlerror', $this->handle);
- }
- $this->info = curl_getinfo($this->handle);
- $options['hooks']->dispatch('curl.after_request', array(&$this->headers, &$this->info));
- return $this->headers;
- }
- /**
- * Collect the headers as they are received
- *
- * @param resource $handle cURL resource
- * @param string $headers Header string
- * @return integer Length of provided header
- */
- public function stream_headers($handle, $headers) {
- // Why do we do this? cURL will send both the final response and any
- // interim responses, such as a 100 Continue. We don't need that.
- // (We may want to keep this somewhere just in case)
- if ($this->done_headers) {
- $this->headers = '';
- $this->done_headers = false;
- }
- $this->headers .= $headers;
- if ($headers === "\r\n") {
- $this->done_headers = true;
- }
- return strlen($headers);
- }
- /**
- * Collect data as it's received
- *
- * @since 1.6.1
- *
- * @param resource $handle cURL resource
- * @param string $data Body data
- * @return integer Length of provided data
- */
- public function stream_body($handle, $data) {
- $this->hooks->dispatch('request.progress', array($data, $this->response_bytes, $this->response_byte_limit));
- $data_length = strlen($data);
- // Are we limiting the response size?
- if ($this->response_byte_limit) {
- if ($this->response_bytes === $this->response_byte_limit) {
- // Already at maximum, move on
- return $data_length;
- }
- if (($this->response_bytes + $data_length) > $this->response_byte_limit) {
- // Limit the length
- $limited_length = ($this->response_byte_limit - $this->response_bytes);
- $data = substr($data, 0, $limited_length);
- }
- }
- if ($this->stream_handle) {
- fwrite($this->stream_handle, $data);
- }
- else {
- $this->response_data .= $data;
- }
- $this->response_bytes += strlen($data);
- return $data_length;
- }
- /**
- * Format a URL given GET data
- *
- * @param string $url
- * @param array|object $data Data to build query using, see {@see https://secure.php.net/http_build_query}
- * @return string URL with data
- */
- protected static function format_get($url, $data) {
- if (!empty($data)) {
- $url_parts = parse_url($url);
- if (empty($url_parts['query'])) {
- $query = $url_parts['query'] = '';
- }
- else {
- $query = $url_parts['query'];
- }
- $query .= '&' . http_build_query($data, null, '&');
- $query = trim($query, '&');
- if (empty($url_parts['query'])) {
- $url .= '?' . $query;
- }
- else {
- $url = str_replace($url_parts['query'], $query, $url);
- }
- }
- return $url;
- }
- /**
- * Whether this transport is valid
- *
- * @codeCoverageIgnore
- * @return boolean True if the transport is valid, false otherwise.
- */
- public static function test($capabilities = array()) {
- if (!function_exists('curl_init') || !function_exists('curl_exec')) {
- return false;
- }
- // If needed, check that our installed curl version supports SSL
- if (isset($capabilities['ssl']) && $capabilities['ssl']) {
- $curl_version = curl_version();
- if (!(CURL_VERSION_SSL & $curl_version['features'])) {
- return false;
- }
- }
- return true;
- }
- }
|