Subversion Repositories SmartDukaan

<?php

/**

 * CakePHP Socket connection class.

 *

 * CakePHP(tm) : Rapid Development Framework (http://cakephp.org)

 * Copyright (c) Cake Software Foundation, Inc. (http://cakefoundation.org)

 *

 * Licensed under The MIT License

 * For full copyright and license information, please see the LICENSE.txt

 * Redistributions of files must retain the above copyright notice.

 *

 * @copyright     Copyright (c) Cake Software Foundation, Inc. (http://cakefoundation.org)

 * @link          http://cakephp.org CakePHP(tm) Project

 * @package       Cake.Network

 * @since         CakePHP(tm) v 1.2.0

 * @license       http://www.opensource.org/licenses/mit-license.php MIT License

 */

App::uses('Validation', 'Utility');

/**

 * CakePHP network socket connection class.

 *

 * Core base class for network communication.

 *

 * @package       Cake.Network

 */

class CakeSocket {

/**

 * Object description

 *

 * @var string

 */

	public $description = 'Remote DataSource Network Socket Interface';

/**

 * Base configuration settings for the socket connection

 *

 * @var array

 */

	protected $_baseConfig = array(

		'persistent' => false,

		'host' => 'localhost',

		'protocol' => 'tcp',

		'port' => 80,

		'timeout' => 30

	);

/**

 * Configuration settings for the socket connection

 *

 * @var array

 */

	public $config = array();

/**

 * Reference to socket connection resource

 *

 * @var resource

 */

	public $connection = null;

/**

 * This boolean contains the current state of the CakeSocket class

 *

 * @var bool

 */

	public $connected = false;

/**

 * This variable contains an array with the last error number (num) and string (str)

 *

 * @var array

 */

	public $lastError = array();

/**

 * True if the socket stream is encrypted after a CakeSocket::enableCrypto() call

 *

 * @var bool

 */

	public $encrypted = false;

/**

 * Contains all the encryption methods available

 *

 * @var array

 */

	protected $_encryptMethods = array(

		// @codingStandardsIgnoreStart

		'sslv2_client' => STREAM_CRYPTO_METHOD_SSLv2_CLIENT,

		'sslv3_client' => STREAM_CRYPTO_METHOD_SSLv3_CLIENT,

		'sslv23_client' => STREAM_CRYPTO_METHOD_SSLv23_CLIENT,

		'tls_client' => STREAM_CRYPTO_METHOD_TLS_CLIENT,

		'sslv2_server' => STREAM_CRYPTO_METHOD_SSLv2_SERVER,

		'sslv3_server' => STREAM_CRYPTO_METHOD_SSLv3_SERVER,

		'sslv23_server' => STREAM_CRYPTO_METHOD_SSLv23_SERVER,

		'tls_server' => STREAM_CRYPTO_METHOD_TLS_SERVER

		// @codingStandardsIgnoreEnd

	);

/**

 * Used to capture connection warnings which can happen when there are

 * SSL errors for example.

 *

 * @var array

 */

	protected $_connectionErrors = array();

/**

 * Constructor.

 *

 * @param array $config Socket configuration, which will be merged with the base configuration

 * @see CakeSocket::$_baseConfig

 */

	public function __construct($config = array()) {

		$this->config = array_merge($this->_baseConfig, $config);

	}

/**

 * Connects the socket to the given host and port.

 *

 * @return bool Success

 * @throws SocketException

 */

	public function connect() {

		if ($this->connection) {

			$this->disconnect();

		}

		$scheme = null;

		if (!empty($this->config['protocol']) && strpos($this->config['host'], '://') === false && empty($this->config['proxy'])) {

			$scheme = $this->config['protocol'] . '://';

		}

		if (!empty($this->config['context'])) {

			$context = stream_context_create($this->config['context']);

		} else {

			$context = stream_context_create();

		}

		$connectAs = STREAM_CLIENT_CONNECT;

		if ($this->config['persistent']) {

			$connectAs |= STREAM_CLIENT_PERSISTENT;

		}

		set_error_handler(array($this, '_connectionErrorHandler'));

		$this->connection = stream_socket_client(

			$scheme . $this->config['host'] . ':' . $this->config['port'],

			$errNum,

			$errStr,

			$this->config['timeout'],

			$connectAs,

			$context

		);

		restore_error_handler();

		if (!empty($errNum) || !empty($errStr)) {

			$this->setLastError($errNum, $errStr);

			throw new SocketException($errStr, $errNum);

		}

		if (!$this->connection && $this->_connectionErrors) {

			$message = implode("\n", $this->_connectionErrors);

			throw new SocketException($message, E_WARNING);

		}

		$this->connected = is_resource($this->connection);

		if ($this->connected) {

			stream_set_timeout($this->connection, $this->config['timeout']);

			if (!empty($this->config['request']) &&

				$this->config['request']['uri']['scheme'] === 'https' &&

				!empty($this->config['proxy'])

			) {

				$req = array();

				$req[] = 'CONNECT ' . $this->config['request']['uri']['host'] . ':' .

					$this->config['request']['uri']['port'] . ' HTTP/1.1';

				$req[] = 'Host: ' . $this->config['host'];

				$req[] = 'User-Agent: php proxy';

				fwrite($this->connection, implode("\r\n", $req) . "\r\n\r\n");

				while (!feof($this->connection)) {

					$s = rtrim(fgets($this->connection, 4096));

					if (preg_match('/^$/', $s)) {

						break;

					}

				}

				$this->enableCrypto('tls', 'client');

			}

		}

		return $this->connected;

	}

/**

 * socket_stream_client() does not populate errNum, or $errStr when there are

 * connection errors, as in the case of SSL verification failure.

 *

 * Instead we need to handle those errors manually.

 *

 * @param int $code Code.

 * @param string $message Message.

 * @return void

 */

	protected function _connectionErrorHandler($code, $message) {

		$this->_connectionErrors[] = $message;

	}

/**

 * Gets the connection context.

 *

 * @return null|array Null when there is no connection, an array when there is.

 */

	public function context() {

		if (!$this->connection) {

			return;

		}

		return stream_context_get_options($this->connection);

	}

/**

 * Gets the host name of the current connection.

 *

 * @return string Host name

 */

	public function host() {

		if (Validation::ip($this->config['host'])) {

			return gethostbyaddr($this->config['host']);

		}

		return gethostbyaddr($this->address());

	}

/**

 * Gets the IP address of the current connection.

 *

 * @return string IP address

 */

	public function address() {

		if (Validation::ip($this->config['host'])) {

			return $this->config['host'];

		}

		return gethostbyname($this->config['host']);

	}

/**

 * Gets all IP addresses associated with the current connection.

 *

 * @return array IP addresses

 */

	public function addresses() {

		if (Validation::ip($this->config['host'])) {

			return array($this->config['host']);

		}

		return gethostbynamel($this->config['host']);

	}

/**

 * Gets the last error as a string.

 *

 * @return string|null Last error

 */

	public function lastError() {

		if (!empty($this->lastError)) {

			return $this->lastError['num'] . ': ' . $this->lastError['str'];

		}

		return null;

	}

/**

 * Sets the last error.

 *

 * @param int $errNum Error code

 * @param string $errStr Error string

 * @return void

 */

	public function setLastError($errNum, $errStr) {

		$this->lastError = array('num' => $errNum, 'str' => $errStr);

	}

/**

 * Writes data to the socket.

 *

 * @param string $data The data to write to the socket

 * @return bool Success

 */

	public function write($data) {

		if (!$this->connected) {

			if (!$this->connect()) {

				return false;

			}

		}

		$totalBytes = strlen($data);

		for ($written = 0, $rv = 0; $written < $totalBytes; $written += $rv) {

			$rv = fwrite($this->connection, substr($data, $written));

			if ($rv === false || $rv === 0) {

				return $written;

			}

		}

		return $written;

	}

/**

 * Reads data from the socket. Returns false if no data is available or no connection could be

 * established.

 *

 * @param int $length Optional buffer length to read; defaults to 1024

 * @return mixed Socket data

 */

	public function read($length = 1024) {

		if (!$this->connected) {

			if (!$this->connect()) {

				return false;

			}

		}

		if (!feof($this->connection)) {

			$buffer = fread($this->connection, $length);

			$info = stream_get_meta_data($this->connection);

			if ($info['timed_out']) {

				$this->setLastError(E_WARNING, __d('cake_dev', 'Connection timed out'));

				return false;

			}

			return $buffer;

		}

		return false;

	}

/**

 * Disconnects the socket from the current connection.

 *

 * @return bool Success

 */

	public function disconnect() {

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

			$this->connected = false;

			return true;

		}

		$this->connected = !fclose($this->connection);

		if (!$this->connected) {

			$this->connection = null;

		}

		return !$this->connected;

	}

/**

 * Destructor, used to disconnect from current connection.

 */

	public function __destruct() {

		$this->disconnect();

	}

/**

 * Resets the state of this Socket instance to it's initial state (before Object::__construct got executed)

 *

 * @param array $state Array with key and values to reset

 * @return bool True on success

 */

	public function reset($state = null) {

		if (empty($state)) {

			static $initalState = array();

			if (empty($initalState)) {

				$initalState = get_class_vars(__CLASS__);

			}

			$state = $initalState;

		}

		foreach ($state as $property => $value) {

			$this->{$property} = $value;

		}

		return true;

	}

/**

 * Encrypts current stream socket, using one of the defined encryption methods.

 *

 * @param string $type Type which can be one of 'sslv2', 'sslv3', 'sslv23' or 'tls'.

 * @param string $clientOrServer Can be one of 'client', 'server'. Default is 'client'.

 * @param bool $enable Enable or disable encryption. Default is true (enable)

 * @return bool True on success

 * @throws InvalidArgumentException When an invalid encryption scheme is chosen.

 * @throws SocketException When attempting to enable SSL/TLS fails.

 * @see stream_socket_enable_crypto

 */

	public function enableCrypto($type, $clientOrServer = 'client', $enable = true) {

		if (!array_key_exists($type . '_' . $clientOrServer, $this->_encryptMethods)) {

			throw new InvalidArgumentException(__d('cake_dev', 'Invalid encryption scheme chosen'));

		}

		$enableCryptoResult = false;

		try {

			$enableCryptoResult = stream_socket_enable_crypto($this->connection, $enable,

				$this->_encryptMethods[$type . '_' . $clientOrServer]);

		} catch (Exception $e) {

			$this->setLastError(null, $e->getMessage());

			throw new SocketException($e->getMessage());

		}

		if ($enableCryptoResult === true) {

			$this->encrypted = $enable;

			return true;

		}

		$errorMessage = __d('cake_dev', 'Unable to perform enableCrypto operation on CakeSocket');

		$this->setLastError(null, $errorMessage);

		throw new SocketException($errorMessage);

	}

}