1
0
pretty/cdata.lua

84 lines
2.1 KiB
Lua
Raw Normal View History

-- Import
local ffi = require 'ffi'
-- Constants
--------------------------------------------------------------------------------
-- Util
local NUMBER_TO_HEX = {
[00] = '0', [01] = '1', [02] = '2', [03] = '3', [04] = '4', [05] = '5',
[06] = '6', [07] = '7', [08] = '8', [09] = '9', [10] = 'A', [11] = 'B',
[12] = 'C', [13] = 'D', [14] = 'E', [15] = 'F',
}
local function to_hex (str)
local l = {}
for i = 1, #str do
local v = str:byte(i)
l[#l+1] = NUMBER_TO_HEX[math.floor(v / 16)]
l[#l+1] = NUMBER_TO_HEX[v % 16]
l[#l+1] = ' '
end
l[#l] = nil
return table.concat(l, '')
end
local function is_nice_ascii_string (str)
for i = 1, #str do
local byte = str:byte(i)
if not (32 <= byte and byte <= 126) then return false end
end
return true
end
local function get_type_and_size_of_singular ( ctype )
local nr_elements = 1
while true do
local etype, elements = ctype:match('(.+)%[(%d*)%]$')
if not elements then break end
ctype, nr_elements = etype, nr_elements * elements
end
return ctype, nr_elements
end
--------------------------------------------------------------------------------
local CDATA_REPR_MATCHER = 'cdata<(.+)>: (0x%w+)'
return function (value, options, depth, l)
local native_repr = tostring(value)
local data_length = ffi.sizeof(value)
local ctype, addr = native_repr:match(CDATA_REPR_MATCHER)
l[#l+1] = 'cdata {'
--l[#l+1] = '\n\tnative = \'' .. native_repr .. '\','
l[#l+1] = '\n\ttype = ' .. ctype .. ','
l[#l+1] = '\n\taddr = ' .. addr .. ','
if data_length then
-- Size
local str = ffi.string(value, data_length)
l[#l+1] = '\n\tsize = ' .. data_length .. ','
-- Element size and type
local element_type, nr_elements = get_type_and_size_of_singular(ctype)
local element_size = data_length / nr_elements
l[#l+1] = '\n\tnr_e = ' .. nr_elements .. ','
l[#l+1] = '\n\ttype_e = ' .. element_type .. ','
l[#l+1] = '\n\tsize_e = ' .. element_size .. ','
--
if is_nice_ascii_string(str) then
l[#l+1] = '\n\tstr = ' .. str .. ','
end
l[#l+1] = '\n\tbin = ' .. to_hex(str) .. ','
end
l[#l+1] = '\n}'
end