TrackManiaControl/core/Sockets/SocketManager.php

<?php
namespace ManiaControl\Sockets;

use ManiaControl\Callbacks\CallbackListener;
use ManiaControl\Callbacks\Callbacks;
use ManiaControl\Callbacks\Listening;
use ManiaControl\Logger;
use ManiaControl\ManiaControl;
use ManiaControl\Settings\Setting;
use ManiaControl\Settings\SettingManager;
use React\EventLoop\Factory;
use React\EventLoop\LoopInterface;
use React\Socket\Connection;
use React\Socket\ConnectionException;
use React\Socket\Server;

/**
 * Class for managing Socket Callbacks
 *
 * @author    ManiaControl Team <mail@maniacontrol.com>
 * @copyright 2014-2015 ManiaControl Team
 * @license   http://www.gnu.org/licenses/ GNU General Public License, Version 3
 */
class SocketManager implements CallbackListener {

	/** @var ManiaControl $maniaControl */
	private $maniaControl = null;

	/** @var LoopInterface $loop */
	private $loop = null;

	/** @var Listening[] $socketListenings */
	private $socketListenings = array();

	/** @var Server $socket */
	private $socket = null;

	const SETTING_SOCKET_ENABLED  = "Activate Socket";
	const SETTING_SOCKET_PASSWORD = "Password for the Socket Connection";
	const SETTING_SOCKET_PORT     = "Socket Port for Server ";

	/**
	 * Create a new Socket Handler Instance
	 *
	 * @param ManiaControl $maniaControl
	 */
	public function __construct(ManiaControl $maniaControl) {
		$this->maniaControl = $maniaControl;

		$this->maniaControl->getCallbackManager()->registerCallbackListener(SettingManager::CB_SETTING_CHANGED, $this, 'updateSettings');
		$this->maniaControl->getCallbackManager()->registerCallbackListener(Callbacks::AFTERINIT, $this, 'initSocketManager');
	}


	/**
	 * Register a new Socket Listener
	 *
	 * @param string         $callbackName
	 * @param SocketListener $listener
	 * @param string         $method
	 * @return bool
	 */
	public function registerSocketListener($echoName, SocketListener $listener, $method) {
		if (!Listening::checkValidCallback($listener, $method)) {
			$listenerClass = get_class($listener);
			trigger_error("Given Listener '{$listenerClass}' can't handle Callback '{$echoName}': No callable Method '{$method}'!");
			return false;
		}

		if (!array_key_exists($echoName, $this->socketListenings)) {
			$this->socketListenings[$echoName] = new Listening($listener, $method);
		} else {
			//TODO say which is already listening and other stuff
			trigger_error("Only one Listener can listen on a specific Socket Message");
		}

		return true;
	}

	/**
	 * Trigger a specific Callback
	 *
	 * @param mixed $callback
	 */
	public function triggerSocketCallback($callbackName) {
		if (!array_key_exists($callbackName, $this->socketListenings)) {
			return null;
		}

		$params = func_get_args();
		$params = array_slice($params, 1, null, true);

		$listening = $this->socketListenings[$callbackName];
		/** @var Listening $listening */
		return $listening->triggerCallbackWithParams($params);
	}


	/**
	 * Unregister a Socket Listener
	 *
	 * @param SocketListener $listener
	 * @return bool
	 */
	public function unregisterEchoListener(SocketListener $listener) {
		return $this->removeSocketListener($this->socketListenings, $listener);
	}


	/**
	 * Remove the Socket Listener from the given Listeners Array
	 *
	 * @param Listening[]    $listeningsArray
	 * @param SocketListener $listener
	 * @return bool
	 */
	private function removeSocketListener(array &$listeningsArray, SocketListener $listener) {
		$removed = false;
		foreach ($listeningsArray as &$listening) {
			if ($listening->listener === $listener) {
				unset($listening);
				$removed = true;
			}
		}
		return $removed;
	}

	/**
	 * Inits the Socket Manager after ManiaControl Startup
	 */
	public function initSocketManager() {
		$this->maniaControl->getSettingManager()->initSetting($this, self::SETTING_SOCKET_ENABLED, false);
		$this->maniaControl->getSettingManager()->initSetting($this, self::SETTING_SOCKET_PASSWORD, "");

		$servers = $this->maniaControl->getServer()->getAllServers();
		foreach ($servers as $server) {
			$this->maniaControl->getSettingManager()->initSetting($this, self::SETTING_SOCKET_PORT . $server->login, 31500 + $server->index);
		}


		$this->createSocket();
	}

	/**
	 * Update Setting
	 *
	 * @param Setting $setting
	 */
	public function updateSettings(Setting $setting) {
		if (!$setting->belongsToClass($this)) {
			return;
		}

		$socketEnabled = $this->maniaControl->getSettingManager()->getSettingValue($this, self::SETTING_SOCKET_ENABLED);

		if ($socketEnabled && !$this->socket) {
			$this->createSocket();
		}

		if (!$socketEnabled) {
			unset ($this->socket);
		}
	}

	/**
	 * Creates The Socket
	 */
	private function createSocket() {
		$socketEnabled = $this->maniaControl->getSettingManager()->getSettingValue($this, self::SETTING_SOCKET_ENABLED);
		if ($socketEnabled) {

			Logger::log("[SocketManager] Trying to create Socket");

			// Check for MySQLi
			$message = '[SocketManager] Checking for installed openssl ... ';
			if (!extension_loaded('openssl')) {
				Logger::log($message . 'NOT FOUND!');
				Logger::log(" -- You don't have openssl installed! Check: http://www.php.net/manual/en/openssl.installation.php");
				return;
			} else {
				Logger::log($message . 'FOUND!');
			}

			$serverLogin = $this->maniaControl->getServer()->login;
			$socketPort  = $this->maniaControl->getSettingManager()->getSettingValue($this, self::SETTING_SOCKET_PORT . $serverLogin);

			try {
				$this->loop   = Factory::create();
				$this->socket = new Server($this->loop);

				$this->socket->on('error', function ($e) {
					Logger::log("[SocketManager] Socket Error" . $e);
				});

				$this->socket->on('connection', function (Connection $connection) {
					$buffer = '';
					$connection->on('data', function ($data) use (&$buffer, &$connection) {
						$buffer .= $data;
						$arr = explode("\n", $buffer, 2); // much haxy.
						while (count($arr) == 2 && strlen($arr[1]) >= (int) $arr[0]) {
							// received full message
							$len    = (int) $arr[0];
							$msg    = substr($arr[1], 0, $len); // clip msg
							$buffer = substr($buffer, strlen((string) $len) + 1 /* newline */ + $len); // clip buffer

							// Decode Message
							$data = openssl_decrypt($msg, 'aes-192-cbc', 'testpass123', OPENSSL_RAW_DATA, 'kZ2Kt0CzKUjN2MJX');
							$data = json_decode($data);

							if ($data == null) {
								$data = array("error" => true, "data" => "Data is not provided as an valid AES-196-encrypted encrypted JSON");
							} else if (!property_exists($data, "method") || !property_exists($data, "data")) {
								$data = array("error" => true, "data" => "Invalid Message");
							} else {
								$answer = $this->triggerSocketCallback($data->method, $data);
								//Prepare Response
								if (!$answer) {
									$data = array("error" => true, "data" => "No listener or response on the given Message");
								} else {
									$data = array("error" => false, "data" => $answer);
								}
							}

							//Encode, Encrypt and Send Response
							$data = json_encode($data);
							$data = openssl_encrypt($data, 'aes-192-cbc', 'testpass123', OPENSSL_RAW_DATA, 'kZ2Kt0CzKUjN2MJX');
							$connection->write(strlen($data) . "\n" . $data);

							// next msg
							$arr = explode("\n", $buffer, 2);
						}
					});
				});
				//TODO check if port is closed
				$this->socket->listen($socketPort, getHostByName(getHostName()));

				Logger::log("[SocketManager] Socket " . getHostByName(getHostName()) . ":" . $this->socket->getPort() . " Successfully created!");
			} catch (ConnectionException $e) {
				Logger::log("[SocketManager] Exception: " . $e->getMessage());
			}
		}
	}


	/**
	 * Processes Data on every ManiaControl Tick, don't call this Method
	 */
	public function tick() {
		if ($this->loop) {
			$this->loop->tick();
		}
	}
}