Lua, dealing with non-ascii byte streams, byteorder change

前端 未结 3 1556
广开言路
广开言路 2020-12-30 16:11

Need to encode & decode byte-stream (containing non-ascii characters possibly), from/into uint16, uint32, uint64 (their typical C/C++ meaning), taking care of endianness

3条回答
  •  无人及你
    2020-12-30 16:36

    for converting from bytes to int (taking care of endianness at byte level, and signedness):

    function bytes_to_int(str,endian,signed) -- use length of string to determine 8,16,32,64 bits
        local t={str:byte(1,-1)}
        if endian=="big" then --reverse bytes
            local tt={}
            for k=1,#t do
                tt[#t-k+1]=t[k]
            end
            t=tt
        end
        local n=0
        for k=1,#t do
            n=n+t[k]*2^((k-1)*8)
        end
        if signed then
            n = (n > 2^(#t*8-1) -1) and (n - 2^(#t*8)) or n -- if last bit set, negative.
        end
        return n
    end
    

    And while we're at it also the other direction:

    function int_to_bytes(num,endian,signed)
        if num<0 and not signed then num=-num print"warning, dropping sign from number converting to unsigned" end
        local res={}
        local n = math.ceil(select(2,math.frexp(num))/8) -- number of bytes to be used.
        if signed and num < 0 then
            num = num + 2^n
        end
        for k=n,1,-1 do -- 256 = 2^8 bits per char.
            local mul=2^(8*(k-1))
            res[k]=math.floor(num/mul)
            num=num-res[k]*mul
        end
        assert(num==0)
        if endian == "big" then
            local t={}
            for k=1,n do
                t[k]=res[n-k+1]
            end
            res=t
        end
        return string.char(unpack(res))
    end
    

    Any remarks are welcome, it's tested, but not too thoroughly...

提交回复
热议问题