166 lines
		
	
	
		
			4.9 KiB
		
	
	
	
		
			Haxe
		
	
	
	
	
	
		
		
			
		
	
	
			166 lines
		
	
	
		
			4.9 KiB
		
	
	
	
		
			Haxe
		
	
	
	
	
	
|  | /* | ||
|  |  * Copyright (C)2005-2019 Haxe Foundation | ||
|  |  * | ||
|  |  * Permission is hereby granted, free of charge, to any person obtaining a | ||
|  |  * copy of this software and associated documentation files (the "Software"), | ||
|  |  * to deal in the Software without restriction, including without limitation | ||
|  |  * the rights to use, copy, modify, merge, publish, distribute, sublicense, | ||
|  |  * and/or sell copies of the Software, and to permit persons to whom the | ||
|  |  * Software is furnished to do so, subject to the following conditions: | ||
|  |  * | ||
|  |  * The above copyright notice and this permission notice shall be included in | ||
|  |  * all copies or substantial portions of the Software. | ||
|  |  * | ||
|  |  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | ||
|  |  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | ||
|  |  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE | ||
|  |  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | ||
|  |  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING | ||
|  |  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER | ||
|  |  * DEALINGS IN THE SOFTWARE. | ||
|  |  */ | ||
|  | 
 | ||
|  | package php.net; | ||
|  | 
 | ||
|  | import php.*; | ||
|  | import php.Global.*; | ||
|  | import php.Const.*; | ||
|  | import sys.io.FileInput; | ||
|  | import sys.io.FileOutput; | ||
|  | import sys.net.Host; | ||
|  | 
 | ||
|  | class Socket extends sys.net.Socket { | ||
|  | 	var connected = false; | ||
|  | 	var timeout:Null<Float> = null; | ||
|  | 
 | ||
|  | 	public function new():Void { | ||
|  | 		super(); | ||
|  | 		protocol = "tcp"; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	override function initSocket():Void { | ||
|  | 		// do not init socket with `socket_*` functions, because we will use streams instead. | ||
|  | 	} | ||
|  | 
 | ||
|  | 	override function assignHandler():Void { | ||
|  | 		@:privateAccess (cast input : FileInput).__f = __s; | ||
|  | 		@:privateAccess (cast output : FileOutput).__f = __s; | ||
|  | 		connected = true; | ||
|  | 		if (timeout != null) { | ||
|  | 			setTimeout(timeout); | ||
|  | 		} | ||
|  | 	} | ||
|  | 
 | ||
|  | 	override public function close():Void { | ||
|  | 		connected = false; | ||
|  | 		fclose(__s); | ||
|  | 		@:privateAccess (cast input : FileInput).__f = null; | ||
|  | 		@:privateAccess (cast output : FileOutput).__f = null; | ||
|  | 		input.close(); | ||
|  | 		output.close(); | ||
|  | 	} | ||
|  | 
 | ||
|  | 	override public function read():String { | ||
|  | 		var b = ''; | ||
|  | 		while (!feof(__s)) | ||
|  | 			b += fgets(__s); | ||
|  | 		return b; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	override public function write(content:String):Void { | ||
|  | 		fwrite(__s, content); | ||
|  | 	} | ||
|  | 
 | ||
|  | 	override public function connect(host:Host, port:Int):Void { | ||
|  | 		var errs = null; | ||
|  | 		var errn = null; | ||
|  | 		var r = stream_socket_client(protocol + '://' + host.host + ':' + port, errn, errs); | ||
|  | 		checkError(r, errn, errs); | ||
|  | 		__s = r; | ||
|  | 		assignHandler(); | ||
|  | 	} | ||
|  | 
 | ||
|  | 	override public function listen(connections:Int):Void { | ||
|  | 		throw new haxe.exceptions.NotImplementedException(); | ||
|  | 		/* TODO: ?????? | ||
|  | 			var r = socket_listen(__s, connections); | ||
|  | 			checkError(r); | ||
|  | 		 */ | ||
|  | 	} | ||
|  | 
 | ||
|  | 	override public function shutdown(read:Bool, write:Bool):Void { | ||
|  | 		var rw = read && write ? 2 : (write ? 1 : (read ? 0 : 2)); | ||
|  | 		var r = stream_socket_shutdown(__s, rw); | ||
|  | 		checkError(r, 0, 'Unable to Shutdown'); | ||
|  | 	} | ||
|  | 
 | ||
|  | 	override public function bind(host:Host, port:Int):Void { | ||
|  | 		var errs = Boot.deref(null); | ||
|  | 		var errn = Boot.deref(null); | ||
|  | 		var r = stream_socket_server(protocol + '://' + host.host + ':' + port, errn, errs, | ||
|  | 			(protocol == "udp" ? STREAM_SERVER_BIND : STREAM_SERVER_BIND | STREAM_SERVER_LISTEN)); | ||
|  | 		Socket.checkError(r, errn, errs); | ||
|  | 		__s = cast r; | ||
|  | 		assignHandler(); | ||
|  | 	} | ||
|  | 
 | ||
|  | 	override public function accept():Socket { | ||
|  | 		var r = stream_socket_accept(__s); | ||
|  | 		checkError(r, 0, 'Unable to accept connections on socket'); | ||
|  | 		var s = new Socket(); | ||
|  | 		s.__s = r; | ||
|  | 		s.assignHandler(); | ||
|  | 		return s; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	private function hpOfString(s:String):{host:Host, port:Int} { | ||
|  | 		var parts = s.split(':'); | ||
|  | 		if (parts.length == 2) { | ||
|  | 			return {host: new Host(parts[0]), port: Std.parseInt(parts[1])}; | ||
|  | 		} else { | ||
|  | 			return {host: new Host(parts[1].substr(2)), port: Std.parseInt(parts[2])}; | ||
|  | 		} | ||
|  | 	} | ||
|  | 
 | ||
|  | 	override public function peer():{host:Host, port:Int} { | ||
|  | 		var r = stream_socket_get_name(__s, true); | ||
|  | 		checkError(r, 0, 'Unable to retrieve the peer name'); | ||
|  | 		return hpOfString(r); | ||
|  | 	} | ||
|  | 
 | ||
|  | 	override public function host():{host:Host, port:Int} { | ||
|  | 		var r = stream_socket_get_name(__s, false); | ||
|  | 		checkError(r, 0, 'Unable to retrieve the host name'); | ||
|  | 		return hpOfString(r); | ||
|  | 	} | ||
|  | 
 | ||
|  | 	override public function setTimeout(timeout:Float):Void { | ||
|  | 		if (!connected) { | ||
|  | 			this.timeout = timeout; | ||
|  | 			return; | ||
|  | 		} | ||
|  | 		var s = Std.int(timeout); | ||
|  | 		var ms = Std.int((timeout - s) * 1000000); | ||
|  | 		var r = stream_set_timeout(__s, s, ms); | ||
|  | 		checkError(r, 0, 'Unable to set timeout'); | ||
|  | 	} | ||
|  | 
 | ||
|  | 	private static function getType(isUdp:Bool):Int { | ||
|  | 		return isUdp ? SOCK_DGRAM : SOCK_STREAM; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	private static function getProtocol(protocol:String):Int { | ||
|  | 		return getprotobyname(protocol); | ||
|  | 	} | ||
|  | 
 | ||
|  | 	public static function select(read:Array<Socket>, write:Array<Socket>, others:Array<Socket>, | ||
|  | 			?timeout:Float):{read:Array<Socket>, write:Array<Socket>, others:Array<Socket>} { | ||
|  | 		throw new haxe.exceptions.NotImplementedException(); | ||
|  | 		return null; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	private static inline function checkError(r:Bool, code:Int, msg:String):Void { | ||
|  | 		return sys.net.Socket.checkError(r, code, msg); | ||
|  | 	} | ||
|  | } |