366 lines
13 KiB
Lua
366 lines
13 KiB
Lua
|
do
|
|||
|
|
require("public.class")
|
|||
|
|
require("bio.BioType")
|
|||
|
|
BioInputStream = {}
|
|||
|
|
|
|||
|
|
local B2Type = BioType.B2Type;
|
|||
|
|
local IntB2TypeValueMap = BioType.IntB2TypeValueMap;
|
|||
|
|
local StringB2TypeLenMap = BioType.StringB2TypeLenMap
|
|||
|
|
local MapB2TypeSizeMap = BioType.MapB2TypeSizeMap
|
|||
|
|
local ListB2TypeSizeMap = BioType.ListB2TypeSizeMap
|
|||
|
|
|
|||
|
|
|
|||
|
|
local type = type
|
|||
|
|
local sub = string.sub
|
|||
|
|
local strbyte = string.byte
|
|||
|
|
local strchar = string.char
|
|||
|
|
|
|||
|
|
--===================================================
|
|||
|
|
--===================================================
|
|||
|
|
--===================================================
|
|||
|
|
-- 数据流
|
|||
|
|
LuaB2InputStream = class("LuaB2InputStream");
|
|||
|
|
function LuaB2InputStream:ctor(v)
|
|||
|
|
self.bytes = v;
|
|||
|
|
self.pos = 1;
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
function LuaB2InputStream:init(v)
|
|||
|
|
self.bytes = v;
|
|||
|
|
self.pos = 1;
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
function LuaB2InputStream:readByte()
|
|||
|
|
local ret = strbyte(self.bytes, self.pos);
|
|||
|
|
self.pos = self.pos + 1;
|
|||
|
|
return ret;
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
function LuaB2InputStream:readBytes(len)
|
|||
|
|
local ret = strbyte(self.bytes, self.pos, self.pos + len - 1);
|
|||
|
|
self.pos = self.pos + len;
|
|||
|
|
return ret;
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
function LuaB2InputStream:readString(len)
|
|||
|
|
local ret = sub(self.bytes, self.pos, self.pos + len - 1);
|
|||
|
|
self.pos = self.pos + len;
|
|||
|
|
return ret;
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
function LuaB2InputStream:release()
|
|||
|
|
self.bytes = nil;
|
|||
|
|
self.pos = 1;
|
|||
|
|
end
|
|||
|
|
--===================================================
|
|||
|
|
--===================================================
|
|||
|
|
--===================================================
|
|||
|
|
---public Void readObject (LuaB2InputStream s)
|
|||
|
|
---@param optional LuaB2InputStream s
|
|||
|
|
--[[
|
|||
|
|
local is = LuaB2InputStream.new(os:toBytes());
|
|||
|
|
local result = BioInputStream.readObject(is)
|
|||
|
|
--]]
|
|||
|
|
function BioInputStream.readObject (s)
|
|||
|
|
local tag = BioInputStream.ReadByteTag(s);
|
|||
|
|
if tag == B2Type.NULL then
|
|||
|
|
return nil;
|
|||
|
|
else
|
|||
|
|
-- decode map
|
|||
|
|
local mapSize = MapB2TypeSizeMap[tag]
|
|||
|
|
if mapSize then
|
|||
|
|
return BioInputStream.readMap(s, mapSize);
|
|||
|
|
elseif tag == B2Type.HASHTABLE then
|
|||
|
|
local len = BioInputStream.readInt(s);
|
|||
|
|
return BioInputStream.readMap(s, len);
|
|||
|
|
end
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
-- decode int
|
|||
|
|
local intVal = IntB2TypeValueMap[tag]
|
|||
|
|
if intVal then
|
|||
|
|
return intVal;
|
|||
|
|
elseif tag == B2Type.INT_N1 then
|
|||
|
|
return -1;
|
|||
|
|
elseif tag == B2Type.INT_8B then
|
|||
|
|
local int = 0;
|
|||
|
|
local int0 = BioInputStream.ReadByte(s)
|
|||
|
|
if int0 >= 128 then
|
|||
|
|
--字节是负数,里面存放的1字节是负数的补码,比如读出来是128， 128=1000 0000，对应signed char 是 -127，而127=0111 1111
|
|||
|
|
int = int0 - 256;
|
|||
|
|
else
|
|||
|
|
int = int0;
|
|||
|
|
end
|
|||
|
|
return int;
|
|||
|
|
elseif tag == B2Type.INT_16B then
|
|||
|
|
local int = 0;
|
|||
|
|
--INT_16B
|
|||
|
|
local int0 = BioInputStream.ReadByte(s)
|
|||
|
|
local int1 = BioInputStream.ReadByte(s)
|
|||
|
|
if int0 >= 128 then
|
|||
|
|
--字节是负数,里面存放的2字节是负数的补码
|
|||
|
|
int = (int0 * 256 + int1) - 65536;
|
|||
|
|
else
|
|||
|
|
int = (int0 * 256 + int1);
|
|||
|
|
end
|
|||
|
|
return int;
|
|||
|
|
elseif tag == B2Type.INT_32B then
|
|||
|
|
local int = 0;
|
|||
|
|
--INT_32B
|
|||
|
|
local int0 = BioInputStream.ReadByte(s)
|
|||
|
|
local int1 = BioInputStream.ReadByte(s)
|
|||
|
|
local int2 = BioInputStream.ReadByte(s)
|
|||
|
|
local int3 = BioInputStream.ReadByte(s)
|
|||
|
|
--字节是负数,里面存放的4字节是负数的补码
|
|||
|
|
if int0 >= 128 then
|
|||
|
|
int = (int0 * 16777216 + int1 * 65536 + int2 * 256 + int3) - 4294967296;
|
|||
|
|
else
|
|||
|
|
int = (int0 * 16777216 + int1 * 65536 + int2 * 256 + int3);
|
|||
|
|
end
|
|||
|
|
return int;
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
-- decode string
|
|||
|
|
local strLen = StringB2TypeLenMap[tag]
|
|||
|
|
if strLen then
|
|||
|
|
return s:readString(strLen);
|
|||
|
|
elseif tag == B2Type.STR then
|
|||
|
|
strLen = BioInputStream.readInt(s);
|
|||
|
|
return s:readString(strLen);
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
if tag == B2Type.BOOLEAN_TRUE then
|
|||
|
|
return true;
|
|||
|
|
elseif tag == B2Type.BOOLEAN_FALSE then
|
|||
|
|
return false;
|
|||
|
|
elseif tag == B2Type.BYTE_0 then
|
|||
|
|
return 0;
|
|||
|
|
elseif tag == B2Type.BYTE then
|
|||
|
|
return BioInputStream.ReadByte(s);
|
|||
|
|
elseif tag == B2Type.BYTES_0 then
|
|||
|
|
return 0;
|
|||
|
|
elseif tag == B2Type.BYTES then
|
|||
|
|
local len = BioInputStream.readInt(s);
|
|||
|
|
return BioInputStream.ReadBytes(s, len);
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
-- decode list
|
|||
|
|
local listLen = ListB2TypeSizeMap[tag]
|
|||
|
|
if listLen then
|
|||
|
|
return BioInputStream.readList(s, listLen);
|
|||
|
|
elseif tag == B2Type.VECTOR then
|
|||
|
|
local len = BioInputStream.readInt(s);
|
|||
|
|
return BioInputStream.readList(s, len);
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
if tag == B2Type.SHORT_0 then
|
|||
|
|
return 0;
|
|||
|
|
elseif tag == B2Type.SHORT_8B then
|
|||
|
|
local int = 0;
|
|||
|
|
local int0 = BioInputStream.ReadByte(s)
|
|||
|
|
if int0 >= 128 then
|
|||
|
|
--字节是负数,里面存放的1字节是负数的补码,比如读出来是128， 128=1000 0000，对应signed char 是 -127，而127=0111 1111
|
|||
|
|
int = int0 - 256;
|
|||
|
|
else
|
|||
|
|
int = int0;
|
|||
|
|
end
|
|||
|
|
return int;
|
|||
|
|
elseif tag == B2Type.SHORT_16B then
|
|||
|
|
local int = 0;
|
|||
|
|
--INT_16B
|
|||
|
|
local int0 = BioInputStream.ReadByte(s)
|
|||
|
|
local int1 = BioInputStream.ReadByte(s)
|
|||
|
|
if int0 >= 128 then
|
|||
|
|
--字节是负数,里面存放的2字节是负数的补码
|
|||
|
|
int = (int0 * 256 + int1) - 65536;
|
|||
|
|
else
|
|||
|
|
int = (int0 * 256 + int1);
|
|||
|
|
end
|
|||
|
|
return int;
|
|||
|
|
elseif tag == B2Type.LONG_0 then
|
|||
|
|
return 0;
|
|||
|
|
elseif tag == B2Type.LONG_8B then
|
|||
|
|
local int = 0;
|
|||
|
|
local int0 = BioInputStream.ReadByte(s)
|
|||
|
|
if int0 >= 128 then
|
|||
|
|
--字节是负数,里面存放的1字节是负数的补码,比如读出来是128， 128=1000 0000，对应signed char 是 -127，而127=0111 1111
|
|||
|
|
int = int0 - 256;
|
|||
|
|
else
|
|||
|
|
int = int0;
|
|||
|
|
end
|
|||
|
|
return int;
|
|||
|
|
elseif tag == B2Type.LONG_16B then
|
|||
|
|
local int = 0;
|
|||
|
|
--INT_16B
|
|||
|
|
local int0 = BioInputStream.ReadByte(s)
|
|||
|
|
local int1 = BioInputStream.ReadByte(s)
|
|||
|
|
if int0 >= 128 then
|
|||
|
|
--字节是负数,里面存放的2字节是负数的补码
|
|||
|
|
int = (int0 * 256 + int1) - 65536;
|
|||
|
|
else
|
|||
|
|
int = (int0 * 256 + int1);
|
|||
|
|
end
|
|||
|
|
return int;
|
|||
|
|
elseif tag == B2Type.LONG_32B then
|
|||
|
|
local int = 0;
|
|||
|
|
--INT_32B
|
|||
|
|
local int0 = BioInputStream.ReadByte(s)
|
|||
|
|
local int1 = BioInputStream.ReadByte(s)
|
|||
|
|
local int2 = BioInputStream.ReadByte(s)
|
|||
|
|
local int3 = BioInputStream.ReadByte(s)
|
|||
|
|
--字节是负数,里面存放的4字节是负数的补码
|
|||
|
|
if int0 >= 128 then
|
|||
|
|
int = (int0 * 16777216 + int1 * 65536 + int2 * 256 + int3) - 4294967296;
|
|||
|
|
else
|
|||
|
|
int = (int0 * 16777216 + int1 * 65536 + int2 * 256 + int3);
|
|||
|
|
end
|
|||
|
|
return int;
|
|||
|
|
elseif tag == B2Type.LONG_64B then
|
|||
|
|
local int = 0
|
|||
|
|
local int0 = BioInputStream.ReadByte(s)
|
|||
|
|
local int1 = BioInputStream.ReadByte(s)
|
|||
|
|
local int2 = BioInputStream.ReadByte(s)
|
|||
|
|
local int3 = BioInputStream.ReadByte(s)
|
|||
|
|
local int4 = BioInputStream.ReadByte(s)
|
|||
|
|
local int5 = BioInputStream.ReadByte(s)
|
|||
|
|
local int6 = BioInputStream.ReadByte(s)
|
|||
|
|
local int7 = BioInputStream.ReadByte(s)
|
|||
|
|
--字节是负数,里面存放的4字节是负数的补码
|
|||
|
|
if int0 >= 128 then
|
|||
|
|
--? 1 00000000 00000000 00000000 00000000 00000000 00000000 00000000 00000000 = (9223372036854775807 * 4 + 3) ，
|
|||
|
|
--lua里表示不出来了，只能这样写。
|
|||
|
|
--实际上 大于9007199254740991（52bit) 的 long long 64在lua这边已经不是精确值了
|
|||
|
|
int = (int0 * 72057594037927936 + int1 * 281474976710656 + int2 * 1099511627776 + int3 * 4294967296 + int4 * 16777216 + int5 * 65536 + int6 * 256 + int7) - (9223372036854775807 * 4 + 3);
|
|||
|
|
else
|
|||
|
|
int = (int0 * 72057594037927936 + int1 * 281474976710656 + int2 * 1099511627776 + int3 * 4294967296 + int4 * 16777216 + int5 * 65536 + int6 * 256 + int7)
|
|||
|
|
end
|
|||
|
|
return int
|
|||
|
|
elseif tag == B2Type.DOUBLE_0 then
|
|||
|
|
return 0;
|
|||
|
|
elseif tag == B2Type.DOUBLE_64B then
|
|||
|
|
local str = BioInputStream.readString(s)
|
|||
|
|
return tonumber(str);
|
|||
|
|
elseif tag == B2Type.INT_B2 then
|
|||
|
|
return BioInputStream.readInt(s);
|
|||
|
|
else
|
|||
|
|
--//throw new IOException("unknow tag error then" + tag);
|
|||
|
|
print("bio2 unknon type then==" .. tostring(tag));
|
|||
|
|
end
|
|||
|
|
return 0;
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
function BioInputStream.ReadByte(s)
|
|||
|
|
if s == nil then
|
|||
|
|
return nil;
|
|||
|
|
end
|
|||
|
|
return s:readByte();
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
function BioInputStream.ReadByteTag(s)
|
|||
|
|
if s == nil then
|
|||
|
|
return nil;
|
|||
|
|
end
|
|||
|
|
local int0 = s:readByte()
|
|||
|
|
local int = 0;
|
|||
|
|
if int0 >= 128 then
|
|||
|
|
--字节是负数,里面存放的1字节是负数的补码,比如读出来是128， 128=1000 0000，对应signed char 是 -127，而127=0111 1111
|
|||
|
|
int = int0 - 256;
|
|||
|
|
else
|
|||
|
|
int = int0;
|
|||
|
|
end
|
|||
|
|
return int;
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
function BioInputStream.ReadBytes(s, len)
|
|||
|
|
return s:readBytes(len);
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
function BioInputStream.readMap(s, n)
|
|||
|
|
if n == nil or n <= 0 then
|
|||
|
|
return {}
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
local ret = {};
|
|||
|
|
for i = 1, n do
|
|||
|
|
local key = BioInputStream.readObject(s);
|
|||
|
|
local val = BioInputStream.readObject(s);
|
|||
|
|
ret [key] = val;
|
|||
|
|
end
|
|||
|
|
return ret;
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
function BioInputStream.readString(s)
|
|||
|
|
local tag = BioInputStream.ReadByteTag(s);
|
|||
|
|
local strLen = StringB2TypeLenMap[tag]
|
|||
|
|
if strLen then
|
|||
|
|
return s:readString(strLen)
|
|||
|
|
elseif tag == B2Type.STR then
|
|||
|
|
strLen = BioInputStream.readInt(s);
|
|||
|
|
return s:readString(strLen)
|
|||
|
|
end
|
|||
|
|
return "";
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
function BioInputStream.readInt(s)
|
|||
|
|
local tag = BioInputStream.ReadByteTag(s);
|
|||
|
|
local intVal = IntB2TypeValueMap[tag]
|
|||
|
|
if intVal then
|
|||
|
|
return intVal;
|
|||
|
|
elseif tag == B2Type.INT_N1 then
|
|||
|
|
return -1;
|
|||
|
|
elseif tag == B2Type.INT_8B then
|
|||
|
|
local int = 0;
|
|||
|
|
local int0 = BioInputStream.ReadByte(s)
|
|||
|
|
if int0 >= 128 then
|
|||
|
|
--字节是负数,里面存放的1字节是负数的补码,比如读出来是128， 128=1000 0000，对应signed char 是 -127，而127=0111 1111
|
|||
|
|
int = int0 - 256;
|
|||
|
|
else
|
|||
|
|
int = int0;
|
|||
|
|
end
|
|||
|
|
return int;
|
|||
|
|
elseif tag == B2Type.INT_16B then
|
|||
|
|
local int = 0;
|
|||
|
|
--INT_16B
|
|||
|
|
local int0 = BioInputStream.ReadByte(s)
|
|||
|
|
local int1 = BioInputStream.ReadByte(s)
|
|||
|
|
if int0 >= 128 then
|
|||
|
|
--字节是负数,里面存放的2字节是负数的补码
|
|||
|
|
int = (int0 * 256 + int1) - 65536;
|
|||
|
|
else
|
|||
|
|
int = (int0 * 256 + int1);
|
|||
|
|
end
|
|||
|
|
return int;
|
|||
|
|
elseif tag == B2Type.INT_32B then
|
|||
|
|
local int = 0;
|
|||
|
|
--INT_32B
|
|||
|
|
local int0 = BioInputStream.ReadByte(s)
|
|||
|
|
local int1 = BioInputStream.ReadByte(s)
|
|||
|
|
local int2 = BioInputStream.ReadByte(s)
|
|||
|
|
local int3 = BioInputStream.ReadByte(s)
|
|||
|
|
--字节是负数,里面存放的4字节是负数的补码
|
|||
|
|
if int0 >= 128 then
|
|||
|
|
int = (int0 * 16777216 + int1 * 65536 + int2 * 256 + int3) - 4294967296;
|
|||
|
|
else
|
|||
|
|
int = (int0 * 16777216 + int1 * 65536 + int2 * 256 + int3);
|
|||
|
|
end
|
|||
|
|
return int;
|
|||
|
|
end
|
|||
|
|
return 0;
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
function BioInputStream.readList(s, n)
|
|||
|
|
if n == nil or n <= 0 then
|
|||
|
|
return {}
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
local ret = {};
|
|||
|
|
for i = 1, n do
|
|||
|
|
local val = BioInputStream.readObject(s);
|
|||
|
|
table.insert(ret, val);
|
|||
|
|
end
|
|||
|
|
return ret;
|
|||
|
|
end
|
|||
|
|
--------------------------------------------
|
|||
|
|
return BioInputStream;
|
|||
|
|
end
|