2020-03-04 15:26:49 +11:00
local mtar = { }
2021-05-26 17:49:42 +10:00
local versions = { }
versions [ 0 ] = { nlf = " >I2 " , flf = " >I2 " } -- original version format
versions [ 1 ] = { nlf = " >I2 " , flf = " >I8 " } -- extended file size format
mtar.versions = versions
2020-03-04 15:26:49 +11:00
2021-05-26 17:49:42 +10:00
local function cleanPath ( path )
local pt = { }
for segment in path : gmatch ( " [^/]+ " ) do
if segment == " .. " then
pt [ # pt ] = nil
elseif segment ~= " . " then
pt [ # pt + 1 ] = segment
end
2020-03-04 15:26:49 +11:00
end
2021-05-26 17:49:42 +10:00
return table.concat ( pt , " / " )
2020-03-04 15:26:49 +11:00
end
2021-05-26 17:49:42 +10:00
function mtar . genHeader ( fname , len , version ) -- string number -- string -- generate a header for file *fname* when provided with file length *len*
version = version or 1
return string.format ( " \255 \255 %s%s%s%s " , string.char ( version ) , string.pack ( versions [ version ] . nlf , fname : len ( ) ) , fname , string.pack ( versions [ version ] . flf , len ) )
2020-03-04 15:26:49 +11:00
end
2020-05-29 10:54:00 +10:00
function mtar . iter ( stream ) -- table -- function -- Given buffer *stream*, returns an iterator suitable for use with *for* that returns, for each iteration, the file name, a function to read from the file, and the length of the file.
local remain = 0
local function read ( n )
local rb = stream : read ( math.min ( n , remain ) )
remain = remain - rb : len ( )
return rb
end
return function ( )
2021-05-26 17:49:42 +10:00
while remain > 0 do
remain = remain -# stream : read ( math.min ( remain , 2048 ) )
end
local version = 0
local nlen = string.unpack ( " >I2 " , stream : read ( 2 ) or " \0 \0 " )
2020-05-29 10:54:00 +10:00
if nlen == 0 then
return
2021-05-26 17:49:42 +10:00
elseif nlen == 65535 then -- versioned header
version = string.byte ( stream : read ( 1 ) )
nlen = string.unpack ( versions [ version ] . nlf , stream : read ( string.packsize ( versions [ version ] . nlf ) ) )
2020-05-29 10:54:00 +10:00
end
2021-05-26 17:49:42 +10:00
local name = cleanPath ( stream : read ( nlen ) )
remain = string.unpack ( versions [ version ] . flf , stream : read ( string.packsize ( versions [ version ] . flf ) ) )
return name , read , remain
2020-05-29 10:54:00 +10:00
end
end
2020-03-04 15:26:49 +11:00
return mtar