332 lines
		
	
	
		
			8.3 KiB
		
	
	
	
		
			Haxe
		
	
	
	
	
	
		
		
			
		
	
	
			332 lines
		
	
	
		
			8.3 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 haxe.io; | ||
|  | 
 | ||
|  | /** | ||
|  | 	An Input is an abstract reader. See other classes in the `haxe.io` package | ||
|  | 	for several possible implementations. | ||
|  | 
 | ||
|  | 	All functions which read data throw `Eof` when the end of the stream | ||
|  | 	is reached. | ||
|  | **/ | ||
|  | class Input { | ||
|  | 	/** | ||
|  | 		Endianness (word byte order) used when reading numbers. | ||
|  | 
 | ||
|  | 		If `true`, big-endian is used, otherwise `little-endian` is used. | ||
|  | 	**/ | ||
|  | 	public var bigEndian(default, set):Bool; | ||
|  | 
 | ||
|  | 	#if cs | ||
|  | 	private var helper:BytesData; | ||
|  | 	#elseif java | ||
|  | 	private var helper:java.nio.ByteBuffer; | ||
|  | 	#end | ||
|  | 
 | ||
|  | 	/** | ||
|  | 		Read and return one byte. | ||
|  | 	**/ | ||
|  | 	public function readByte():Int { | ||
|  | 		#if cpp | ||
|  | 		throw new haxe.exceptions.NotImplementedException(); | ||
|  | 		#else | ||
|  | 		return throw new haxe.exceptions.NotImplementedException(); | ||
|  | 		#end | ||
|  | 	} | ||
|  | 
 | ||
|  | 	/** | ||
|  | 		Read `len` bytes and write them into `s` to the position specified by `pos`. | ||
|  | 
 | ||
|  | 		Returns the actual length of read data that can be smaller than `len`. | ||
|  | 
 | ||
|  | 		See `readFullBytes` that tries to read the exact amount of specified bytes. | ||
|  | 	**/ | ||
|  | 	public function readBytes(s:Bytes, pos:Int, len:Int):Int { | ||
|  | 		var k = len; | ||
|  | 		var b = #if (js || hl) @:privateAccess s.b #else s.getData() #end; | ||
|  | 		if (pos < 0 || len < 0 || pos + len > s.length) | ||
|  | 			throw Error.OutsideBounds; | ||
|  | 		try { | ||
|  | 			while (k > 0) { | ||
|  | 				#if neko | ||
|  | 				untyped __dollar__sset(b, pos, readByte()); | ||
|  | 				#elseif php | ||
|  | 				b.set(pos, readByte()); | ||
|  | 				#elseif cpp | ||
|  | 				b[pos] = untyped readByte(); | ||
|  | 				#else | ||
|  | 				b[pos] = cast readByte(); | ||
|  | 				#end | ||
|  | 				pos++; | ||
|  | 				k--; | ||
|  | 			} | ||
|  | 		} catch (eof:haxe.io.Eof) {} | ||
|  | 		return len - k; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	/** | ||
|  | 		Close the input source. | ||
|  | 
 | ||
|  | 		Behaviour while reading after calling this method is unspecified. | ||
|  | 	**/ | ||
|  | 	public function close():Void {} | ||
|  | 
 | ||
|  | 	function set_bigEndian(b:Bool):Bool { | ||
|  | 		bigEndian = b; | ||
|  | 		return b; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	/* ------------------ API ------------------ */ | ||
|  | 	/** | ||
|  | 		Read and return all available data. | ||
|  | 
 | ||
|  | 		The `bufsize` optional argument specifies the size of chunks by | ||
|  | 		which data is read. Its default value is target-specific. | ||
|  | 	**/ | ||
|  | 	public function readAll(?bufsize:Int):Bytes { | ||
|  | 		if (bufsize == null) | ||
|  | 			#if php | ||
|  | 			bufsize = 8192; // default value for PHP and max under certain circumstances | ||
|  | 			#else | ||
|  | 			bufsize = (1 << 14); // 16 Ko | ||
|  | 			#end | ||
|  | 
 | ||
|  | 		var buf = Bytes.alloc(bufsize); | ||
|  | 		var total = new haxe.io.BytesBuffer(); | ||
|  | 		try { | ||
|  | 			while (true) { | ||
|  | 				var len = readBytes(buf, 0, bufsize); | ||
|  | 				if (len == 0) | ||
|  | 					throw Error.Blocked; | ||
|  | 				total.addBytes(buf, 0, len); | ||
|  | 			} | ||
|  | 		} catch (e:Eof) {} | ||
|  | 		return total.getBytes(); | ||
|  | 	} | ||
|  | 
 | ||
|  | 	/** | ||
|  | 		Read `len` bytes and write them into `s` to the position specified by `pos`. | ||
|  | 
 | ||
|  | 		Unlike `readBytes`, this method tries to read the exact `len` amount of bytes. | ||
|  | 	**/ | ||
|  | 	public function readFullBytes(s:Bytes, pos:Int, len:Int):Void { | ||
|  | 		while (len > 0) { | ||
|  | 			var k = readBytes(s, pos, len); | ||
|  | 			if (k == 0) | ||
|  | 				throw Error.Blocked; | ||
|  | 			pos += k; | ||
|  | 			len -= k; | ||
|  | 		} | ||
|  | 	} | ||
|  | 
 | ||
|  | 	/** | ||
|  | 		Read and return `nbytes` bytes. | ||
|  | 	**/ | ||
|  | 	public function read(nbytes:Int):Bytes { | ||
|  | 		var s = Bytes.alloc(nbytes); | ||
|  | 		var p = 0; | ||
|  | 		while (nbytes > 0) { | ||
|  | 			var k = readBytes(s, p, nbytes); | ||
|  | 			if (k == 0) | ||
|  | 				throw Error.Blocked; | ||
|  | 			p += k; | ||
|  | 			nbytes -= k; | ||
|  | 		} | ||
|  | 		return s; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	/** | ||
|  | 		Read a string until a character code specified by `end` is occurred. | ||
|  | 
 | ||
|  | 		The final character is not included in the resulting string. | ||
|  | 	**/ | ||
|  | 	public function readUntil(end:Int):String { | ||
|  | 		var buf = new BytesBuffer(); | ||
|  | 		var last:Int; | ||
|  | 		while ((last = readByte()) != end) | ||
|  | 			buf.addByte(last); | ||
|  | 		return buf.getBytes().toString(); | ||
|  | 	} | ||
|  | 
 | ||
|  | 	/** | ||
|  | 		Read a line of text separated by CR and/or LF bytes. | ||
|  | 
 | ||
|  | 		The CR/LF characters are not included in the resulting string. | ||
|  | 	**/ | ||
|  | 	public function readLine():String { | ||
|  | 		var buf = new BytesBuffer(); | ||
|  | 		var last:Int; | ||
|  | 		var s; | ||
|  | 		try { | ||
|  | 			while ((last = readByte()) != 10) | ||
|  | 				buf.addByte(last); | ||
|  | 			s = buf.getBytes().toString(); | ||
|  | 			if (s.charCodeAt(s.length - 1) == 13) | ||
|  | 				s = s.substr(0, -1); | ||
|  | 		} catch (e:Eof) { | ||
|  | 			s = buf.getBytes().toString(); | ||
|  | 			if (s.length == 0) | ||
|  | 				#if neko neko.Lib.rethrow #else throw #end (e); | ||
|  | 		} | ||
|  | 		return s; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	/** | ||
|  | 		Read a 32-bit floating point number. | ||
|  | 
 | ||
|  | 		Endianness is specified by the `bigEndian` property. | ||
|  | 	**/ | ||
|  | 	public function readFloat():Float { | ||
|  | 		return FPHelper.i32ToFloat(readInt32()); | ||
|  | 	} | ||
|  | 
 | ||
|  | 	/** | ||
|  | 		Read a 64-bit double-precision floating point number. | ||
|  | 
 | ||
|  | 		Endianness is specified by the `bigEndian` property. | ||
|  | 	**/ | ||
|  | 	public function readDouble():Float { | ||
|  | 		var i1 = readInt32(); | ||
|  | 		var i2 = readInt32(); | ||
|  | 		return bigEndian ? FPHelper.i64ToDouble(i2, i1) : FPHelper.i64ToDouble(i1, i2); | ||
|  | 	} | ||
|  | 
 | ||
|  | 	/** | ||
|  | 		Read a 8-bit signed integer. | ||
|  | 	**/ | ||
|  | 	public function readInt8():Int { | ||
|  | 		var n = readByte(); | ||
|  | 		if (n >= 128) | ||
|  | 			return n - 256; | ||
|  | 		return n; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	/** | ||
|  | 		Read a 16-bit signed integer. | ||
|  | 
 | ||
|  | 		Endianness is specified by the `bigEndian` property. | ||
|  | 	**/ | ||
|  | 	public function readInt16():Int { | ||
|  | 		var ch1 = readByte(); | ||
|  | 		var ch2 = readByte(); | ||
|  | 		var n = bigEndian ? ch2 | (ch1 << 8) : ch1 | (ch2 << 8); | ||
|  | 		if (n & 0x8000 != 0) | ||
|  | 			return n - 0x10000; | ||
|  | 		return n; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	/** | ||
|  | 		Read a 16-bit unsigned integer. | ||
|  | 
 | ||
|  | 		Endianness is specified by the `bigEndian` property. | ||
|  | 	**/ | ||
|  | 	public function readUInt16():Int { | ||
|  | 		var ch1 = readByte(); | ||
|  | 		var ch2 = readByte(); | ||
|  | 		return bigEndian ? ch2 | (ch1 << 8) : ch1 | (ch2 << 8); | ||
|  | 	} | ||
|  | 
 | ||
|  | 	/** | ||
|  | 		Read a 24-bit signed integer. | ||
|  | 
 | ||
|  | 		Endianness is specified by the `bigEndian` property. | ||
|  | 	**/ | ||
|  | 	public function readInt24():Int { | ||
|  | 		var ch1 = readByte(); | ||
|  | 		var ch2 = readByte(); | ||
|  | 		var ch3 = readByte(); | ||
|  | 		var n = bigEndian ? ch3 | (ch2 << 8) | (ch1 << 16) : ch1 | (ch2 << 8) | (ch3 << 16); | ||
|  | 		if (n & 0x800000 != 0) | ||
|  | 			return n - 0x1000000; | ||
|  | 		return n; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	/** | ||
|  | 		Read a 24-bit unsigned integer. | ||
|  | 
 | ||
|  | 		Endianness is specified by the `bigEndian` property. | ||
|  | 	**/ | ||
|  | 	public function readUInt24():Int { | ||
|  | 		var ch1 = readByte(); | ||
|  | 		var ch2 = readByte(); | ||
|  | 		var ch3 = readByte(); | ||
|  | 		return bigEndian ? ch3 | (ch2 << 8) | (ch1 << 16) : ch1 | (ch2 << 8) | (ch3 << 16); | ||
|  | 	} | ||
|  | 
 | ||
|  | 	/** | ||
|  | 		Read a 32-bit signed integer. | ||
|  | 
 | ||
|  | 		Endianness is specified by the `bigEndian` property. | ||
|  | 	**/ | ||
|  | 	public function readInt32():Int { | ||
|  | 		var ch1 = readByte(); | ||
|  | 		var ch2 = readByte(); | ||
|  | 		var ch3 = readByte(); | ||
|  | 		var ch4 = readByte(); | ||
|  | 		#if (php || python) | ||
|  | 		// php will overflow integers.  Convert them back to signed 32-bit ints. | ||
|  | 		var n = bigEndian ? ch4 | (ch3 << 8) | (ch2 << 16) | (ch1 << 24) : ch1 | (ch2 << 8) | (ch3 << 16) | (ch4 << 24); | ||
|  | 		if (n & 0x80000000 != 0) | ||
|  | 			return (n | 0x80000000); | ||
|  | 		else | ||
|  | 			return n; | ||
|  | 		#elseif lua | ||
|  | 		var n = bigEndian ? ch4 | (ch3 << 8) | (ch2 << 16) | (ch1 << 24) : ch1 | (ch2 << 8) | (ch3 << 16) | (ch4 << 24); | ||
|  | 		return lua.Boot.clampInt32(n); | ||
|  | 		#else | ||
|  | 		return bigEndian ? ch4 | (ch3 << 8) | (ch2 << 16) | (ch1 << 24) : ch1 | (ch2 << 8) | (ch3 << 16) | (ch4 << 24); | ||
|  | 		#end | ||
|  | 	} | ||
|  | 
 | ||
|  | 	/** | ||
|  | 		Read and `len` bytes as a string. | ||
|  | 	**/ | ||
|  | 	public function readString(len:Int, ?encoding:Encoding):String { | ||
|  | 		var b = Bytes.alloc(len); | ||
|  | 		readFullBytes(b, 0, len); | ||
|  | 		#if neko | ||
|  | 		return neko.Lib.stringReference(b); | ||
|  | 		#else | ||
|  | 		return b.getString(0, len, encoding); | ||
|  | 		#end | ||
|  | 	} | ||
|  | 
 | ||
|  | 	#if neko | ||
|  | 	static var _float_of_bytes = neko.Lib.load("std", "float_of_bytes", 2); | ||
|  | 	static var _double_of_bytes = neko.Lib.load("std", "double_of_bytes", 2); | ||
|  | 
 | ||
|  | 	static function __init__() | ||
|  | 		untyped { | ||
|  | 			Input.prototype.bigEndian = false; | ||
|  | 		} | ||
|  | 	#end | ||
|  | 
 | ||
|  | 	#if (flash || js || python) | ||
|  | 	function getDoubleSig(bytes:Array<Int>) { | ||
|  | 		return (((bytes[1] & 0xF) << 16) | (bytes[2] << 8) | bytes[3]) * 4294967296. | ||
|  | 			+ (bytes[4] >> 7) * 2147483648 | ||
|  | 			+ (((bytes[4] & 0x7F) << 24) | (bytes[5] << 16) | (bytes[6] << 8) | bytes[7]); | ||
|  | 	} | ||
|  | 	#end | ||
|  | } |