k.inc

<?php
if(PHP_VERSION_ID > 50603) define('can_int64',true);
class KException extends Exception {};
class KScan {
	private $s,$i,$n;
	public function __construct($s) {
		$this->s=$s;
		$this->i=0;
		$this->n=strlen($s);
	}
	public function r8() {
		if($this->i >= $this->n) return -1; $i=$this->i; $this->i++;
		$x=ord($this->s{$i});
		if($x>=128)$x-=256;
		return $x;
	}
	public function rn($s,$n) {$a=unpack($s,$this->sn($n));return $a[1];}
	public function s1() { if($this->i >= $this->n) return chr(0); $i=$this->i; $this->i++; return $this->s{$i}; }
	public function sn($n) {
		if($this->i >= $this->n) return str_repeat(chr(0),$n);
		$i=$this->i;
		$this->i += $n;
		if($this->i <= $this->n) return substr($this->s, $i, $n);
		return substr($this->s, $i, $this->n - $i);
	}
}
class KTLS extends K {
	public function __construct($context,$hostname,$p=1234,$up="anonymous") {
		if(is_string($context)) $context = array("ssl" => array("cafile" => $context));
		$context["ssl"]["verify_peer"] = TRUE;
		$context["ssl"]["verify_peer_name"] = TRUE;
		$context["ssl"]["verify_depth"] = 12;
		$context["ssl"]["ciphers"] = "HIGH";

		$sock = stream_socket_client("tls://{$hostname}:{$p}", $errno, $errstr, -1, STREAM_CLIENT_CONNECT, stream_context_create($context));
		if(false === $sock)throw new Exception($errstr, $errno);
		parent::__construct($sock, 0, $up);
		$this->i_();
	}
	public static function server_loop($context=null, $p=1234, $k=null, $h="0") { // "Declaration of KTLS::server_loop(...) should be compatible with K::server_loop(...)"
		$context["ssl"]["ciphers"] = "HIGH";
		$sock = stream_socket_server("tls://{$h}:{$p}", $errno, $errstr, STREAM_SERVER_BIND | STREAM_SERVER_LISTEN, stream_context_create($context));
		if(false === $sock)throw new Exception($errstr, $errno);
		K::server_loop($sock, $k, "ignored");
	}
}
class K {
	private static $be = -1;
	private $h,$b;

	private static function init() {
		if(self::$be === -1) {
			self::$be = (pack("N",-69) === pack("l",-69));
		}
	}

	public static function server_loop($p=1234, $k=null, $h="localhost") {//args server_loop(port,receiver) or server_loop(hostname,port,receiver)
		if(is_integer($k)){$h=$p;$p=$k;$k=$h;}
		self::init();$KS=array($k,'ks');$K=array($k,'k');
		$client = array(); $klient = array();
		if(is_integer($p)) {
			$server = stream_socket_server("tcp://{$h}:{$p}", $errno, $errstr);
		} else {
			$server = $p;
		}
		if(false === $server)throw new Exception($errstr, $errno);
		for(;;) {
			$r = $client;$r[] = $server;$w=null;$e=null;if(0>=stream_select($r,$w,$e,null))continue;
			foreach($r as $x) {
				if($x === $server){$i=(int)($y=stream_socket_accept($x,0));stream_set_blocking($y,0);$client[$i]=$y;$klient[$i]=null;continue;}
				$i=(int)$x;$y=$klient[$i];
				if(is_null($y)){$y=$client[$i];while(false!==($c=fgetc($y))&&ord($c)>0);if($c!==false){self::w_($y,pack("c",0));$klient[$i]=new K($y,0);}continue;}
				$g=array();$a=$y->kr($g); if(feof($x)){unset($client[$i]);unset($klient[$i]);continue;} if($a===false)continue; 
				if(!is_array($a))$a=array($a);if($g[1]==257)self::w_($x,self::enc(2,call_user_func_array($K,$a)));else call_user_func_array($KS,$a);
			}
		}
	}
	protected function i_() {
		self::w_($this->h, $up.pack("c", 0));
		if(ord(fgetc($this->h)) != 0) throw new Exception("Can't connect to k: $h:$p");
	}

	public function __construct($h="localhost",$p=1234,$up="anonymous") {
		self::init();
		$this->b="";
		if($p>0) {
			$this->h = fsockopen($h,$p);
			$this->i_();
		}else{
			$this->h=$h;
		}
	}
	function __destruct() {
		fclose($this->h);
	}
	private static function w_($h, $x) {
		$r=fwrite($h,$x);
		if($r === false) throw new Exception("Write failed");
		if($r < strlen($x)) return self::w_($h, substr($x, $r));
	}
	public function ks($s) {
		if(func_num_args()===1){$x=self::enc(0,$s);}else{$x=self::enc(0,func_get_args());}
		self::w_($this->h, $x);
	}
	public function k($s) {
		if(func_num_args()===1){$x=self::enc(1,$s);}else{$x=self::enc(1,func_get_args());}
		self::w_($this->h, $x);
		return $this->kr();
	}
	public function ka($s) {
		if(func_num_args()===1){$x=self::enc(1,$s);}else{$x=self::enc(1,func_get_args());}
		self::w_($this->h, $x);
	}
	private function rd($n,&$nb){
		if(isset($nb)){$r=strlen($nb);if($r>=$n){$x=substr($nb,0,$n);$nb=substr($nb,$n,$r-$n);return $x;}}else{$r=0;}
		$x = stream_get_contents($this->h, $n-$r);
		if($x===false||$x==="")return false;
		if(isset($nb)){$r=strlen($x=$nb.$x);if($r>=$n){$nb=substr($x,$n,$r-$n);$x=substr($x,0,$n);}else{$nb=$x;$x=false;}}return $x;
	}
	public function krs($x){
		$this->b.=$x;
		return $this->kr();
	}
	public function kr(&$b=null) {
		if(false===($s=$this->rd(8,$this->b)))return $s;$a=unpack("l2", $s);
		if(isset($b))$b=$a;$r=$a[2]-8;if($r<=0)return null;//hup.
		if(false===($g=$this->rd($r,$this->b))){$this->b=$s.$this->b;return $g;}
		return self::dec(new KScan($g));
	}

	public static function S($x) {
		$s = pack("c", -11) . $x . pack("c", 0);
		return (object)array("s" => $s);
	}
	public static function G($x=null) {
		if($x === null) {
			if(function_exists('openssl_random_pseudo_bytes')) $x = openssl_random_pseudo_bytes(16);
			else{$x="";for($i=0;$i<16;++$i)$x.=chr(mt_rand(0,255));}
		} else { $x = hex2bin(str_replace("-","",$x)); }
		return (object)array("s" => pack("c", -2) . $x);
	}

	private static function aa($x) {
		$n=min(100,count($x));for($i=0;$i<$n;++$i)if(!isset($x[$i]))return true;
		return false;
	}

	private static function e1($x) {
		if(is_null($x)){return pack("cc",101,0);}
		if($x === true){return pack("cc",-1,1);}
		if($x === false){return pack("cc",-1,0);}
		if(is_string($x)){return pack("ccl",10,0,strlen($x)).$x;}
		if(is_integer($x)){return pack("cl",-6,$x);}
		if(is_float($x)){return pack("cd",-9,$x);}
		if(is_object($x)){return $x->s;}
		if(!is_array($x)){throw new Exception("Can't encode this: " . var_export($x));}
		$a="";
		if(self::aa($x)){//associative array->dict
			$n=count($x);$a=pack("cccl",99,0,0,$n);$v=array();
			foreach($x as $k => $_){$a.=pack("c",-11).$k.pack("c",0);$v[]=$_;}
			$x=$v;//fall through
		}
		$a.=pack("ccl",0,0,count($x));foreach($x as $b){$a.=self::e1($b);}
		return $a;
	}
	private static function enc($b,$x) {
		$x=self::e1($x);
		return pack("ccccl",!self::$be,$b,0,0,8+strlen($x)).$x;
	}
	private static function flip(&$c, &$x) {
		if(!is_array($x))return;
		foreach($x as$k=>&$v){
			if(is_string($v))for($i=0,$n=strlen($v);$i<$n;++$i)$c[$i][$k]=substr($v,$i,1);
			else if(!is_array($v))foreach($c as $i=>&$_)$c[$i][$k]=$v;//atom
			else for($i=0,$n=count($v);$i<$n;++$i)$c[$i][$k]=$v[$i];
		}
	}
	private static function om(&$c) {
		$o=array();foreach($c as$i=>&$x)$o[$i]=(object)$x;return $o;
	}
	private static function date($x) {
		return (864e5*10957)+$x;
	}
	private static function d1($t,$x) {
		switch($t){
		case 0:return self::dec($x);
		case 1:return 1==$x->r8();
		case 2:return (object)array("s" => pack('c',-2) . $x->sn(16));
		case 4:return $x->r8();
		case 5:return $x->rn("s",2);
		case 6:return $x->rn("l",4);
		case 7:
			if(defined('can_int64'))return $x->rn('q',8);
			$a=unpack("l2",$x->sn(8));if($a[2]==-1&&$a[1]<0)return $a[1];if($a[2])throw new Exception("Number too long(64)");return $a[1];
		case 8:return $x->rn("f",4);
		case 9:return $x->rn("d",8);
		case 10:return $x->s1();
		case 11:$s="";while(ord($c=$x->s1()))$s.=$c;return $s;
		case 12:
			if(defined('can_int64'))return self::date($x->rn('q',8)/1e6);
			$a=unpack("l2",$x->sn(8));if($a[2]==-1&&$a[1]<0)return self::date($a[1]/1e6);
			if($a[2])throw new Exception("Number too long(64; upgrade your php)");
			return self::date($a[1]/1e6);
		case 14:return self::date(864e5*$x->rn("l",4));
		case 15:return self::date($x->rn("f",8)*864e5);
		case 16:
			if(defined('can_int64'))return self::date($x->rn('q',8)*1e6);
			$a=unpack("l2",$x->sn(8));if($a[2]==-1&&$a[1]<0)return self::date(1e6*$a[1]);
			if($a[2])throw new Exception("Number too long(64; upgrade your php)");
			return self::date(1e6*$a[1]);
		case 17:return self::date($x->rn("l",4)*6e4);
		case 18:return self::date($x->rn("l",4)/1e3);
		case 19:return self::date($x->rn("l",4));
		case 98:$x->s1();$k=self::dec($x);$v=self::dec($x);$c=array();self::flip($c,$k);self::flip($c,$v);return self::om($c);
		case 99:$k=self::dec($x);$v=self::dec($x);if(empty($v))return $k;return array_combine($k,$v);
		case 101:$x->r8();return null;
		case 128:throw new KException(self::d1(11,$x));
		default:throw new Exception("Can't decode this $t yet");
		} 
	}
	private static function dec($x) {
		$t=$x->r8();
		if($t<0)return self::d1(-$t,$x);
		if($t>=98)return self::d1($t,$x);
		$x->r8();
		$n=self::d1(6,$x);
		if($t===10)return $x->sn($n);
		$a=array();for($i=0;$i<$n;++$i)$a[]=self::d1($t,$x);
		return $a;
	}
}