<?php

/**

 * Zend Framework

 *

 * LICENSE

 *

 * This source file is subject to the new BSD license that is bundled

 * with this package in the file LICENSE.txt.

 * It is also available through the world-wide-web at this URL:

 * http://framework.zend.com/license/new-bsd

 * If you did not receive a copy of the license and are unable to

 * obtain it through the world-wide-web, please send an email

 * to license@zend.com so we can send you a copy immediately.

 *

 * @category   Zend

 * @package    Zend_Http

 * @subpackage Client_Adapter

 * @version    $Id: Curl.php 22216 2010-05-20 21:12:05Z dragonbe $

 * @copyright  Copyright (c) 2005-2010 Zend Technologies USA Inc. (http://www.zend.com)

 * @license    http://framework.zend.com/license/new-bsd     New BSD License

 */

/**

 * @see Zend_Uri_Http

 */

require_once 'Zend/Uri/Http.php';

/**

 * @see Zend_Http_Client_Adapter_Interface

 */

require_once 'Zend/Http/Client/Adapter/Interface.php';

/**

 * @see Zend_Http_Client_Adapter_Stream

 */

require_once 'Zend/Http/Client/Adapter/Stream.php';

/**

 * An adapter class for Zend_Http_Client based on the curl extension.

 * Curl requires libcurl. See for full requirements the PHP manual: http://php.net/curl

 *

 * @category   Zend

 * @package    Zend_Http

 * @subpackage Client_Adapter

 * @copyright  Copyright (c) 2005-2010 Zend Technologies USA Inc. (http://www.zend.com)

 * @license    http://framework.zend.com/license/new-bsd     New BSD License

 */

class Zend_Http_Client_Adapter_Curl implements Zend_Http_Client_Adapter_Interface, Zend_Http_Client_Adapter_Stream

{

    /**

     * Parameters array

     *

     * @var array

     */

    protected $_config = array();

    /**

     * What host/port are we connected to?

     *

     * @var array

     */

    protected $_connected_to = array(null, null);

    /**

     * The curl session handle

     *

     * @var resource|null

     */

    protected $_curl = null;

    /**

     * List of cURL options that should never be overwritten

     *

     * @var array

     */

    protected $_invalidOverwritableCurlOptions;

    /**

     * Response gotten from server

     *

     * @var string

     */

    protected $_response = null;

    /**

     * Stream for storing output

     *

     * @var resource

     */

    protected $out_stream;

    /**

     * Adapter constructor

     *

     * Config is set using setConfig()

     *

     * @return void

     * @throws Zend_Http_Client_Adapter_Exception

     */

    public function __construct()

    {

        if (!extension_loaded('curl')) {

            require_once 'Zend/Http/Client/Adapter/Exception.php';

            throw new Zend_Http_Client_Adapter_Exception('cURL extension has to be loaded to use this Zend_Http_Client adapter.');

        }

        $this->_invalidOverwritableCurlOptions = array(

            CURLOPT_HTTPGET,

            CURLOPT_POST,

            CURLOPT_PUT,

            CURLOPT_CUSTOMREQUEST,

            CURLOPT_HEADER,

            CURLOPT_RETURNTRANSFER,

            CURLOPT_HTTPHEADER,

            CURLOPT_POSTFIELDS,

            CURLOPT_INFILE,

            CURLOPT_INFILESIZE,

            CURLOPT_PORT,

            CURLOPT_MAXREDIRS,

            CURLOPT_CONNECTTIMEOUT,

            CURL_HTTP_VERSION_1_1,

            CURL_HTTP_VERSION_1_0,

        );

    }

    /**

     * Set the configuration array for the adapter

     *

     * @throws Zend_Http_Client_Adapter_Exception

     * @param  Zend_Config | array $config

     * @return Zend_Http_Client_Adapter_Curl

     */

    public function setConfig($config = array())

    {

        if ($config instanceof Zend_Config) {

            $config = $config->toArray();

        } elseif (! is_array($config)) {

            require_once 'Zend/Http/Client/Adapter/Exception.php';

            throw new Zend_Http_Client_Adapter_Exception(

                'Array or Zend_Config object expected, got ' . gettype($config)

            );

        }

        if(isset($config['proxy_user']) && isset($config['proxy_pass'])) {

            $this->setCurlOption(CURLOPT_PROXYUSERPWD, $config['proxy_user'].":".$config['proxy_pass']);

            unset($config['proxy_user'], $config['proxy_pass']);

        }

        foreach ($config as $k => $v) {

            $option = strtolower($k);

            switch($option) {

                case 'proxy_host':

                    $this->setCurlOption(CURLOPT_PROXY, $v);

                    break;

                case 'proxy_port':

                    $this->setCurlOption(CURLOPT_PROXYPORT, $v);

                    break;

                default:

                    $this->_config[$option] = $v;

                    break;

            }

        }

        return $this;

    }

    /**

      * Retrieve the array of all configuration options

      *

      * @return array

      */

     public function getConfig()

     {

         return $this->_config;

     }

    /**

     * Direct setter for cURL adapter related options.

     *

     * @param  string|int $option

     * @param  mixed $value

     * @return Zend_Http_Adapter_Curl

     */

    public function setCurlOption($option, $value)

    {

        if (!isset($this->_config['curloptions'])) {

            $this->_config['curloptions'] = array();

        }

        $this->_config['curloptions'][$option] = $value;

        return $this;

    }

    /**

     * Initialize curl

     *

     * @param  string  $host

     * @param  int     $port

     * @param  boolean $secure

     * @return void

     * @throws Zend_Http_Client_Adapter_Exception if unable to connect

     */

    public function connect($host, $port = 80, $secure = false)

    {

        // If we're already connected, disconnect first

        if ($this->_curl) {

            $this->close();

        }

        // If we are connected to a different server or port, disconnect first

        if ($this->_curl

            && is_array($this->_connected_to)

            && ($this->_connected_to[0] != $host

            || $this->_connected_to[1] != $port)

        ) {

            $this->close();

        }

        // Do the actual connection

        $this->_curl = curl_init();

        if ($port != 80) {

            curl_setopt($this->_curl, CURLOPT_PORT, intval($port));

        }

        // Set timeout

        curl_setopt($this->_curl, CURLOPT_CONNECTTIMEOUT, $this->_config['timeout']);

        // Set Max redirects

        curl_setopt($this->_curl, CURLOPT_MAXREDIRS, $this->_config['maxredirects']);

        if (!$this->_curl) {

            $this->close();

            require_once 'Zend/Http/Client/Adapter/Exception.php';

            throw new Zend_Http_Client_Adapter_Exception('Unable to Connect to ' .  $host . ':' . $port);

        }

        if ($secure !== false) {

            // Behave the same like Zend_Http_Adapter_Socket on SSL options.

            if (isset($this->_config['sslcert'])) {

                curl_setopt($this->_curl, CURLOPT_SSLCERT, $this->_config['sslcert']);

            }

            if (isset($this->_config['sslpassphrase'])) {

                curl_setopt($this->_curl, CURLOPT_SSLCERTPASSWD, $this->_config['sslpassphrase']);

            }

        }

        // Update connected_to

        $this->_connected_to = array($host, $port);

    }

    /**

     * Send request to the remote server

     *

     * @param  string        $method

     * @param  Zend_Uri_Http $uri

     * @param  float         $http_ver

     * @param  array         $headers

     * @param  string        $body

     * @return string        $request

     * @throws Zend_Http_Client_Adapter_Exception If connection fails, connected to wrong host, no PUT file defined, unsupported method, or unsupported cURL option

     */

    public function write($method, $uri, $httpVersion = 1.1, $headers = array(), $body = '')

    {

        // Make sure we're properly connected

        if (!$this->_curl) {

            require_once 'Zend/Http/Client/Adapter/Exception.php';

            throw new Zend_Http_Client_Adapter_Exception("Trying to write but we are not connected");

        }

        if ($this->_connected_to[0] != $uri->getHost() || $this->_connected_to[1] != $uri->getPort()) {

            require_once 'Zend/Http/Client/Adapter/Exception.php';

            throw new Zend_Http_Client_Adapter_Exception("Trying to write but we are connected to the wrong host");

        }

        // set URL

        curl_setopt($this->_curl, CURLOPT_URL, $uri->__toString());

        // ensure correct curl call

        $curlValue = true;

        switch ($method) {

            case Zend_Http_Client::GET:

                $curlMethod = CURLOPT_HTTPGET;

                break;

            case Zend_Http_Client::POST:

                $curlMethod = CURLOPT_POST;

                break;

            case Zend_Http_Client::PUT:

                // There are two different types of PUT request, either a Raw Data string has been set

                // or CURLOPT_INFILE and CURLOPT_INFILESIZE are used.

                if(is_resource($body)) {

                    $this->_config['curloptions'][CURLOPT_INFILE] = $body;

                }

                if (isset($this->_config['curloptions'][CURLOPT_INFILE])) {

                    // Now we will probably already have Content-Length set, so that we have to delete it

                    // from $headers at this point:

                    foreach ($headers AS $k => $header) {

                        if (preg_match('/Content-Length:\s*(\d+)/i', $header, $m)) {

                            if(is_resource($body)) {

                                $this->_config['curloptions'][CURLOPT_INFILESIZE] = (int)$m[1];

                            }

                            unset($headers[$k]);

                        }

                    }

                    if (!isset($this->_config['curloptions'][CURLOPT_INFILESIZE])) {

                        require_once 'Zend/Http/Client/Adapter/Exception.php';

                        throw new Zend_Http_Client_Adapter_Exception("Cannot set a file-handle for cURL option CURLOPT_INFILE without also setting its size in CURLOPT_INFILESIZE.");

                    }

                    if(is_resource($body)) {

                        $body = '';

                    }

                    $curlMethod = CURLOPT_PUT;

                } else {

                    $curlMethod = CURLOPT_CUSTOMREQUEST;

                    $curlValue = "PUT";

                }

                break;

            case Zend_Http_Client::DELETE:

                $curlMethod = CURLOPT_CUSTOMREQUEST;

                $curlValue = "DELETE";

                break;

            case Zend_Http_Client::OPTIONS:

                $curlMethod = CURLOPT_CUSTOMREQUEST;

                $curlValue = "OPTIONS";

                break;

            case Zend_Http_Client::TRACE:

                $curlMethod = CURLOPT_CUSTOMREQUEST;

                $curlValue = "TRACE";

                break;

            case Zend_Http_Client::HEAD:

                $curlMethod = CURLOPT_CUSTOMREQUEST;

                $curlValue = "HEAD";

                break;

            default:

                // For now, through an exception for unsupported request methods

                require_once 'Zend/Http/Client/Adapter/Exception.php';

                throw new Zend_Http_Client_Adapter_Exception("Method currently not supported");

        }

        if(is_resource($body) && $curlMethod != CURLOPT_PUT) {

            require_once 'Zend/Http/Client/Adapter/Exception.php';

            throw new Zend_Http_Client_Adapter_Exception("Streaming requests are allowed only with PUT");

        }

        // get http version to use

        $curlHttp = ($httpVersion == 1.1) ? CURL_HTTP_VERSION_1_1 : CURL_HTTP_VERSION_1_0;

        // mark as HTTP request and set HTTP method

        curl_setopt($this->_curl, $curlHttp, true);

        curl_setopt($this->_curl, $curlMethod, $curlValue);

        if($this->out_stream) {

            // headers will be read into the response

            curl_setopt($this->_curl, CURLOPT_HEADER, false);

            curl_setopt($this->_curl, CURLOPT_HEADERFUNCTION, array($this, "readHeader"));

            // and data will be written into the file

            curl_setopt($this->_curl, CURLOPT_FILE, $this->out_stream);

        } else {

            // ensure headers are also returned

            curl_setopt($this->_curl, CURLOPT_HEADER, true);

            // ensure actual response is returned

            curl_setopt($this->_curl, CURLOPT_RETURNTRANSFER, true);

        }

        // set additional headers

        $headers['Accept'] = '';

        curl_setopt($this->_curl, CURLOPT_HTTPHEADER, $headers);

        /**

         * Make sure POSTFIELDS is set after $curlMethod is set:

         * @link http://de2.php.net/manual/en/function.curl-setopt.php#81161

         */

        if ($method == Zend_Http_Client::POST) {

            curl_setopt($this->_curl, CURLOPT_POSTFIELDS, $body);

        } elseif ($curlMethod == CURLOPT_PUT) {

            // this covers a PUT by file-handle:

            // Make the setting of this options explicit (rather than setting it through the loop following a bit lower)

            // to group common functionality together.

            curl_setopt($this->_curl, CURLOPT_INFILE, $this->_config['curloptions'][CURLOPT_INFILE]);

            curl_setopt($this->_curl, CURLOPT_INFILESIZE, $this->_config['curloptions'][CURLOPT_INFILESIZE]);

            unset($this->_config['curloptions'][CURLOPT_INFILE]);

            unset($this->_config['curloptions'][CURLOPT_INFILESIZE]);

        } elseif ($method == Zend_Http_Client::PUT) {

            // This is a PUT by a setRawData string, not by file-handle

            curl_setopt($this->_curl, CURLOPT_POSTFIELDS, $body);

        }

        // set additional curl options

        if (isset($this->_config['curloptions'])) {

            foreach ((array)$this->_config['curloptions'] as $k => $v) {

                if (!in_array($k, $this->_invalidOverwritableCurlOptions)) {

                    if (curl_setopt($this->_curl, $k, $v) == false) {

                        require_once 'Zend/Http/Client/Exception.php';

                        throw new Zend_Http_Client_Exception(sprintf("Unknown or erroreous cURL option '%s' set", $k));

                    }

                }

            }

        }

        // send the request

        $response = curl_exec($this->_curl);

        // if we used streaming, headers are already there

        if(!is_resource($this->out_stream)) {

            $this->_response = $response;

        }

        $request  = curl_getinfo($this->_curl, CURLINFO_HEADER_OUT);

        $request .= $body;

        if (empty($this->_response)) {

            require_once 'Zend/Http/Client/Exception.php';

            throw new Zend_Http_Client_Exception("Error in cURL request: " . curl_error($this->_curl));

        }

        // cURL automatically decodes chunked-messages, this means we have to disallow the Zend_Http_Response to do it again

        if (stripos($this->_response, "Transfer-Encoding: chunked\r\n")) {

            $this->_response = str_ireplace("Transfer-Encoding: chunked\r\n", '', $this->_response);

        }

        // Eliminate multiple HTTP responses.

        do {

            $parts  = preg_split('|(?:\r?\n){2}|m', $this->_response, 2);