![[tokkee]](http://tokkee.org/images/avatar.png){kind=avatar}
index 73a0da5f68e22948d9ea9b2f65491b35518f603b..4442e1c5017729efdbccdfd6513655a279c079e7 100644 (file)
<?php
+/*
+ * This code is part of GOsa (http://www.gosa-project.org)
+ * Copyright (C) 2003-2008 GONICUS GmbH
+ *
+ * ID: $$Id$$
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ */
class Socket_Client
{
- private $host = "";
- private $port = "";
- private $timeout= "";
- private $errno = "";
- private $errstr = "";
- private $b_data_send = FALSE;
- private $handle = NULL;
- private $bytes_read = 0;
- private $error = "";
-
- /* Crypto information */
- private $td= NULL;
- private $ckey= "";
- private $ks;
- private $iv;
-
-
- public function __construct($host, $port, $connect = TRUE, $timeout = 3){
- $this->host= $host;
- $this->port= $port;
- $this->timeout= $timeout;
-
- /* Connect if needed */
- if($connect){
- $this->open();
- }
- }
-
-
- public function setEncryptionKey($key)
- {
- if(!function_exists("mcrypt_get_iv_size")){
- $this->error = _("The mcrypt module was not found. Please install php5-mcrypt.");
- $this->ckey = "";
- return FALSE ;
- }
-
- if ($this->connected()){
- $this->ckey = substr(md5($key), 0, $this->ks);
- return TRUE;
- }
-
- return FALSE;
- }
-
-
- private function encrypt($data)
- {
- mcrypt_generic_init($this->td, $this->ckey, $this->iv);
- return base64_encode(mcrypt_generic($this->td, $data));
- }
-
-
- private function decrypt($data)
- {
- /* decrypt data */
- $data = base64_decode($data);
- mcrypt_generic_init($this->td, $this->ckey, $this->iv);
- return mdecrypt_generic($this->td, $data);
- }
-
-
- public function connected()
- {
- return ($this->handle == TRUE);
- }
-
-
- public function open()
- {
- $this->handle = @fsockopen($this->host, $this->port, $this->errno, $this->errstr, $this->timeout);
- if(!$this->handle){
- $this->handle = NULL;
- $this->error = $this->errstr;
- }else{
- $this->b_data_send = TRUE;
-
- /* Open the cipher */
- $this->td = mcrypt_module_open('rijndael-128', '', 'cbc', '');
-
- /* Create the IV and determine the keysize length */
- $this->iv = substr(md5('GONICUS GmbH'),0, mcrypt_enc_get_iv_size($this->td));
- $this->ks = mcrypt_enc_get_key_size($this->td);
- }
- }
-
-
- public function get_error()
- {
- return $this->error;
- }
-
-
- public function write($data){
- if($this->handle){
- $data = $this->encrypt($data);
- fputs($this->handle, $data."\n");
- $this->b_data_send = TRUE;
- }else{
- $this->b_data_send = FALSE;
- }
-
- return $this->b_data_send;
- }
-
-
- public function read()
- {
- $str = FALSE;
- if($this->handle){
-
- /* Check if there is something to read for us */
- $read = array("0"=>$this->handle);
- $write = array();
- $accept = array();
- $num = @stream_select($read,$write,$accept,floor($this->timeout), ceil($this->timeout*1000000));
- $str = "";
-
- /* Read data if necessary */
- while($num && $this->b_data_send){
- $str.= fread($this->handle, 1024000);
- $read = array("0"=>$this->handle);
- $num = @stream_select($read,$write,$accept,floor($this->timeout), ceil($this->timeout*1000000));
- }
- $this->bytes_read = strlen($str);
- $this->b_data_send = FALSE;
- $str = $this->decrypt($str);
- }
- return $str;
- }
-
-
- public function bytes_read()
- {
- return $this->bytes_read;
- }
-
-
- public function close()
- {
- if($this->handle){
- fclose($this->handle);
- }
-
- /* Terminate decryption handle and close module */
- mcrypt_generic_deinit($this->td);
- }
-
+ private $host = "";
+ private $port = "";
+ private $timeout= "";
+ private $errno = "";
+ private $errstr = "";
+ private $b_data_send = FALSE;
+ private $handle = NULL;
+ private $bytes_read = 0;
+ private $error = "";
+ private $is_error = FALSE;
+ private $b_encrypt = FALSE;
+
+ /* Crypto information */
+ private $td= NULL;
+ private $ckey= "";
+ private $ks;
+ private $iv;
+
+
+ public function __construct($host, $port, $connect = TRUE, $timeout = 3){
+ $this->host= $host;
+ $this->port= $port;
+ $this->timeout= $timeout;
+ $this->reset_error();
+
+ /* Connect if needed */
+ if($connect){
+ $this->open();
+ }
+ }
+
+
+ public function setEncryptionKey($key)
+ {
+ if(!function_exists("mcrypt_get_iv_size")){
+ $this->set_error(_("The mcrypt module was not found. Please install php5-mcrypt."));
+ $this->ckey = "";
+ $this->b_encrypt = FALSE;
+ }
+
+ if ($this->connected()){
+ $this->ckey = substr(md5($key), 0, $this->ks);
+ $this->b_encrypt = TRUE;
+ }
+
+ return($this->b_encrypt);
+ }
+
+
+ private function encrypt($data)
+ {
+ if($this->b_encrypt){
+ mcrypt_generic_init($this->td, $this->ckey, $this->iv);
+ $data = base64_encode(mcrypt_generic($this->td, $data));
+ }
+ return($data);
+ }
+
+
+ private function decrypt($data)
+ {
+ /* decrypt data */
+ if($this->b_encrypt && strlen($data)){
+ $data = base64_decode($data);
+ mcrypt_generic_init($this->td, $this->ckey, $this->iv);
+ $data = mdecrypt_generic($this->td, $data);
+ }
+ return($data);
+ }
+
+
+ public function connected()
+ {
+ return ($this->handle == TRUE);
+ }
+
+
+ public function open()
+ {
+ $this->reset_error();
+ $this->handle = @fsockopen($this->host, $this->port, $this->errno, $this->errstr, $this->timeout);
+ if(!$this->handle){
+ $this->handle = NULL;
+ $this->set_error(sprintf(_("Socket connection to host '%s:%s' failed: %s"),$this->host,$this->port,$this->errstr));
+ }else{
+ $this->b_data_send = TRUE;
+
+ /* Open the cipher */
+ $this->td = mcrypt_module_open('rijndael-128', '', 'cbc', '');
+
+ /* Create the IV and determine the keysize length */
+ $this->iv = substr(md5('GONICUS GmbH'),0, mcrypt_enc_get_iv_size($this->td));
+ $this->ks = mcrypt_enc_get_key_size($this->td);
+ }
+ }
+
+
+ public function set_error($str)
+ {
+ $this->is_error =TRUE;
+ $this->error=$str;
+ }
+
+
+ public function reset_error()
+ {
+ $this->is_error =FALSE;
+ $this->error = "";
+ }
+
+
+ public function is_error()
+ {
+ return($this->is_error);
+ }
+
+
+ public function get_error()
+ {
+ return $this->error;
+ }
+
+
+ public function write($data){
+ if($this->handle){
+ $data = $this->encrypt($data);
+ fputs($this->handle, $data."\n");
+ $this->b_data_send = TRUE;
+ }else{
+ $this->b_data_send = FALSE;
+ }
+
+ return $this->b_data_send;
+ }
+
+
+ public function read()
+ {
+ // Output the request results
+ $this->reset_error();
+ $str = "";
+ $data = "test";
+ socket_set_timeout($this->handle,$this->timeout);
+ stream_set_blocking($this->handle,0);
+ $start = microtime(TRUE);
+
+ /* Read while
+ * nothing was read yet
+ * the timelimit reached
+ * there is not data left on the socket.
+ */
+ while(TRUE){
+ usleep(10000);
+ $data = fread($this->handle, 1024000);
+ if($data && strlen($data)>0) {
+ $str .= $data;
+ } else {
+ if(strlen($str) != 0){
+
+ /* We got <xml> but </xml> is still missing, keep on reading */
+ if(preg_match("/<\/xml>/",$this->decrypt($str))){
+ break;
+ }
+ }
+ }
+ if((microtime(TRUE) - $start) > $this->timeout ){
+ $this->set_error(sprintf(_("Socket timeout of %s seconds reached."),$this->timeout));
+ break;
+ }
+ }
+ $this->bytes_read = strlen($str);
+ $this->b_data_send = FALSE;
+ $str = $this->decrypt($str);
+ return($str);
+ }
+
+
+ public function bytes_read()
+ {
+ return $this->bytes_read;
+ }
+
+
+ public function close()
+ {
+ if($this->handle){
+ fclose($this->handle);
+ }
+
+ /* Terminate decryption handle and close module */
+ @mcrypt_generic_deinit($this->td);
+ }
}
-
-
+// vim:tabstop=2:expandtab:shiftwidth=2:filetype=php:syntax:ruler:
?>